s4:gensec: allow clearing local and remote address by passing NULL

[ira/wip.git] / source4 / auth / gensec / gensec.c

diff --git a/source4/auth/gensec/gensec.c b/source4/auth/gensec/gensec.c
index 8de7b4b7e663798b9b266c27b182ab9dd29478b5..ecd0778ee67dbdc175fda8ee84b4edac1691aaf4 100644 (file)
--- a/source4/auth/gensec/gensec.c
+++ b/source4/auth/gensec/gensec.c
@@ -1184,6 +1184,11 @@ _PUBLIC_ NTSTATUS gensec_set_local_address(struct gensec_security *gensec_securi
                const struct tsocket_address *local)
 {
        TALLOC_FREE(gensec_security->local_addr);
+
+       if (local == NULL) {
+               return NT_STATUS_OK;
+       }
+
        gensec_security->local_addr = tsocket_address_copy(local, gensec_security);
        if (gensec_security->local_addr == NULL) {
                return NT_STATUS_NO_MEMORY;
@@ -1206,6 +1211,11 @@ _PUBLIC_ NTSTATUS gensec_set_remote_address(struct gensec_security *gensec_secur
                const struct tsocket_address *remote)
 {
        TALLOC_FREE(gensec_security->remote_addr);
+
+       if (remote == NULL) {
+               return NT_STATUS_OK;
+       }
+
        gensec_security->remote_addr = tsocket_address_copy(remote, gensec_security);
        if (gensec_security->remote_addr == NULL) {
                return NT_STATUS_NO_MEMORY;