r25446: Merge some changes I made on the way home from SFO:
[kai/samba.git] / source4 / auth / gensec / gensec.c
index a23846f3adf43516d0bedeb490a9d57db7c91531..9bbf31a7c797f52399b344f47f9cb64db6768afa 100644 (file)
@@ -4,11 +4,11 @@
    Generic Authentication Interface
 
    Copyright (C) Andrew Tridgell 2003
-   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
+   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2006
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
 #include "auth/auth.h"
 #include "lib/events/events.h"
+#include "build.h"
+#include "librpc/rpc/dcerpc.h"
+#include "auth/credentials/credentials.h"
+#include "auth/gensec/gensec.h"
+#include "param/param.h"
 
 /* the list of currently registered GENSEC backends */
-const static struct gensec_security_ops **generic_security_ops;
+static struct gensec_security_ops **generic_security_ops;
 static int gensec_num_backends;
 
-static const struct gensec_security_ops *gensec_security_by_authtype(uint8_t auth_type)
+/* Return all the registered mechs.  Don't modify the return pointer,
+ * but you may talloc_reference it if convient */
+struct gensec_security_ops **gensec_security_all(void)
+{
+       return generic_security_ops;
+}
+
+/* Sometimes we want to force only kerberos, sometimes we want to
+ * force it's avoidance.  The old list could be either
+ * gensec_security_all(), or from cli_credentials_gensec_list() (ie,
+ * an existing list we have trimmed down) */
+
+struct gensec_security_ops **gensec_use_kerberos_mechs(TALLOC_CTX *mem_ctx, 
+                                                      struct gensec_security_ops **old_gensec_list, 
+                                                      struct cli_credentials *creds)
+{
+       struct gensec_security_ops **new_gensec_list;
+       int i, j, num_mechs_in;
+       enum credentials_use_kerberos use_kerberos = CRED_AUTO_USE_KERBEROS;
+
+       if (creds) {
+               use_kerberos = cli_credentials_get_kerberos_state(creds);
+       }
+
+       if (use_kerberos == CRED_AUTO_USE_KERBEROS) {
+               if (!talloc_reference(mem_ctx, old_gensec_list)) {
+                       return NULL;
+               }
+               return old_gensec_list;
+       }
+
+       for (num_mechs_in=0; old_gensec_list && old_gensec_list[num_mechs_in]; num_mechs_in++) {
+               /* noop */
+       }
+
+       new_gensec_list = talloc_array(mem_ctx, struct gensec_security_ops *, num_mechs_in + 1);
+       if (!new_gensec_list) {
+               return NULL;
+       }
+
+       j = 0;
+       for (i=0; old_gensec_list && old_gensec_list[i]; i++) {
+               int oid_idx;
+               for (oid_idx = 0; old_gensec_list[i]->oid && old_gensec_list[i]->oid[oid_idx]; oid_idx++) {
+                       if (strcmp(old_gensec_list[i]->oid[oid_idx], GENSEC_OID_SPNEGO) == 0) {
+                               new_gensec_list[j] = old_gensec_list[i];
+                               j++;
+                               break;
+                       }
+               }
+               switch (use_kerberos) {
+               case CRED_DONT_USE_KERBEROS:
+                       if (old_gensec_list[i]->kerberos == False) {
+                               new_gensec_list[j] = old_gensec_list[i];
+                               j++;
+                       }
+                       break;
+               case CRED_MUST_USE_KERBEROS:
+                       if (old_gensec_list[i]->kerberos == True) {
+                               new_gensec_list[j] = old_gensec_list[i];
+                               j++;
+                       }
+                       break;
+               default:
+                       /* Can't happen or invalid parameter */
+                       return NULL;
+               }
+       }
+       new_gensec_list[j] = NULL; 
+       
+       return new_gensec_list;
+}
+
+struct gensec_security_ops **gensec_security_mechs(struct gensec_security *gensec_security,
+                                                  TALLOC_CTX *mem_ctx) 
+{
+       struct gensec_security_ops **backends;
+       backends = gensec_security_all();
+       if (!gensec_security) {
+               if (!talloc_reference(mem_ctx, backends)) {
+                       return NULL;
+               }
+               return backends;
+       } else {
+               struct cli_credentials *creds = gensec_get_credentials(gensec_security);
+               if (!creds) {
+                       if (!talloc_reference(mem_ctx, backends)) {
+                               return NULL;
+                       }
+                       return backends;
+               }
+               return gensec_use_kerberos_mechs(mem_ctx, backends, creds);
+       }
+}
+
+static const struct gensec_security_ops *gensec_security_by_authtype(struct gensec_security *gensec_security,
+                                                                    uint8_t auth_type)
 {
        int i;
-       for (i=0; i < gensec_num_backends; i++) {
-               if (generic_security_ops[i]->auth_type == auth_type) {
-                       return generic_security_ops[i];
+       struct gensec_security_ops **backends;
+       const struct gensec_security_ops *backend;
+       TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
+       if (!mem_ctx) {
+               return NULL;
+       }
+       backends = gensec_security_mechs(gensec_security, mem_ctx);
+       for (i=0; backends && backends[i]; i++) {
+               if (backends[i]->auth_type == auth_type) {
+                       backend = backends[i];
+                       talloc_free(mem_ctx);
+                       return backend;
                }
        }
+       talloc_free(mem_ctx);
 
        return NULL;
 }
 
-static const struct gensec_security_ops *gensec_security_by_oid(const char *oid_string)
+const struct gensec_security_ops *gensec_security_by_oid(struct gensec_security *gensec_security,
+                                                        const char *oid_string)
 {
        int i, j;
-       for (i=0; i < gensec_num_backends; i++) {
-               if (generic_security_ops[i]->oid) {
-                       for (j=0; generic_security_ops[i]->oid[j]; j++) { 
-                               if (generic_security_ops[i]->oid[j] &&
-                                   (strcmp(generic_security_ops[i]->oid[j], oid_string) == 0)) {
-                                       return generic_security_ops[i];
+       struct gensec_security_ops **backends;
+       const struct gensec_security_ops *backend;
+       TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
+       if (!mem_ctx) {
+               return NULL;
+       }
+       backends = gensec_security_mechs(gensec_security, mem_ctx);
+       for (i=0; backends && backends[i]; i++) {
+               if (backends[i]->oid) {
+                       for (j=0; backends[i]->oid[j]; j++) { 
+                               if (backends[i]->oid[j] &&
+                                   (strcmp(backends[i]->oid[j], oid_string) == 0)) {
+                                       backend = backends[i];
+                                       talloc_free(mem_ctx);
+                                       return backend;
                                }
                        }
                }
        }
+       talloc_free(mem_ctx);
 
        return NULL;
 }
 
-static const struct gensec_security_ops *gensec_security_by_sasl_name(const char *sasl_name)
+const struct gensec_security_ops *gensec_security_by_sasl_name(struct gensec_security *gensec_security,
+                                                              const char *sasl_name)
 {
        int i;
-       for (i=0; i < gensec_num_backends; i++) {
-               if (generic_security_ops[i]->sasl_name 
-                   && (strcmp(generic_security_ops[i]->sasl_name, sasl_name) == 0)) {
-                       return generic_security_ops[i];
+       struct gensec_security_ops **backends;
+       const struct gensec_security_ops *backend;
+       TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
+       if (!mem_ctx) {
+               return NULL;
+       }
+       backends = gensec_security_mechs(gensec_security, mem_ctx);
+       for (i=0; backends && backends[i]; i++) {
+               if (backends[i]->sasl_name 
+                   && (strcmp(backends[i]->sasl_name, sasl_name) == 0)) {
+                       backend = backends[i];
+                       talloc_free(mem_ctx);
+                       return backend;
                }
        }
+       talloc_free(mem_ctx);
 
        return NULL;
 }
 
-static const struct gensec_security_ops *gensec_security_by_name(const char *name)
+static const struct gensec_security_ops *gensec_security_by_name(struct gensec_security *gensec_security,
+                                                                const char *name)
 {
        int i;
-       for (i=0; i < gensec_num_backends; i++) {
-               if (generic_security_ops[i]->name 
-                   && (strcmp(generic_security_ops[i]->name, name) == 0)) {
-                       return generic_security_ops[i];
+       struct gensec_security_ops **backends;
+       const struct gensec_security_ops *backend;
+       TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
+       if (!mem_ctx) {
+               return NULL;
+       }
+       backends = gensec_security_mechs(gensec_security, mem_ctx);
+       for (i=0; backends && backends[i]; i++) {
+               if (backends[i]->name 
+                   && (strcmp(backends[i]->name, name) == 0)) {
+                       backend = backends[i];
+                       talloc_free(mem_ctx);
+                       return backend;
                }
        }
-
+       talloc_free(mem_ctx);
        return NULL;
 }
 
-const struct gensec_security_ops **gensec_security_all(int *num_backends_out)
-{
-       *num_backends_out = gensec_num_backends;
-       return generic_security_ops;
-}
-
 /**
  * Return a unique list of security subsystems from those specified in
- * the OID list.  That is, where two OIDs refer to the same module,
- * return that module only once 
+ * the list of SASL names.   
  *
- * The list is in the exact order of the OIDs asked for, where available.
+ * Use the list of enabled GENSEC mechanisms from the credentials
+ * attached to the gensec_security, and return in our preferred order.
  */
 
-const struct gensec_security_ops **gensec_security_by_sasl(TALLOC_CTX *mem_ctx, 
-                                                         const char **sasl_names)
+const struct gensec_security_ops **gensec_security_by_sasl_list(struct gensec_security *gensec_security,
+                                                               TALLOC_CTX *mem_ctx, 
+                                                               const char **sasl_names)
 {
        const struct gensec_security_ops **backends_out;
-       const struct gensec_security_ops **backends;
+       struct gensec_security_ops **backends;
        int i, k, sasl_idx;
        int num_backends_out = 0;
-       int num_backends;
 
        if (!sasl_names) {
                return NULL;
        }
 
-       backends = gensec_security_all(&num_backends);
+       backends = gensec_security_mechs(gensec_security, mem_ctx);
 
        backends_out = talloc_array(mem_ctx, const struct gensec_security_ops *, 1);
        if (!backends_out) {
@@ -121,7 +257,7 @@ const struct gensec_security_ops **gensec_security_by_sasl(TALLOC_CTX *mem_ctx,
 
        /* Find backends in our preferred order, by walking our list,
         * then looking in the supplied list */
-       for (i=0; i < num_backends; i++) {
+       for (i=0; backends && backends[i]; i++) {
                for (sasl_idx = 0; sasl_names[sasl_idx]; sasl_idx++) {
                        if (!backends[i]->sasl_name ||
                            !(strcmp(backends[i]->sasl_name, 
@@ -158,26 +294,27 @@ const struct gensec_security_ops **gensec_security_by_sasl(TALLOC_CTX *mem_ctx,
 /**
  * Return a unique list of security subsystems from those specified in
  * the OID list.  That is, where two OIDs refer to the same module,
- * return that module only once 
+ * return that module only once. 
  *
- * The list is in the exact order of the OIDs asked for, where available.
+ * Use the list of enabled GENSEC mechanisms from the credentials
+ * attached to the gensec_security, and return in our preferred order.
  */
 
-const struct gensec_security_ops_wrapper *gensec_security_by_oid_list(TALLOC_CTX *mem_ctx, 
+const struct gensec_security_ops_wrapper *gensec_security_by_oid_list(struct gensec_security *gensec_security,
+                                                                     TALLOC_CTX *mem_ctx, 
                                                                      const char **oid_strings,
                                                                      const char *skip)
 {
        struct gensec_security_ops_wrapper *backends_out;
-       const struct gensec_security_ops **backends;
+       struct gensec_security_ops **backends;
        int i, j, k, oid_idx;
        int num_backends_out = 0;
-       int num_backends;
 
        if (!oid_strings) {
                return NULL;
        }
 
-       backends = gensec_security_all(&num_backends);
+       backends = gensec_security_mechs(gensec_security, gensec_security);
 
        backends_out = talloc_array(mem_ctx, struct gensec_security_ops_wrapper, 1);
        if (!backends_out) {
@@ -188,7 +325,7 @@ const struct gensec_security_ops_wrapper *gensec_security_by_oid_list(TALLOC_CTX
 
        /* Find backends in our preferred order, by walking our list,
         * then looking in the supplied list */
-       for (i=0; i < num_backends; i++) {
+       for (i=0; backends && backends[i]; i++) {
                if (!backends[i]->oid) {
                        continue;
                }
@@ -238,8 +375,7 @@ const struct gensec_security_ops_wrapper *gensec_security_by_oid_list(TALLOC_CTX
  */
 
 const char **gensec_security_oids_from_ops(TALLOC_CTX *mem_ctx, 
-                                          const struct gensec_security_ops **ops,                                 
-                                          int num_backends,
+                                          struct gensec_security_ops **ops,                               
                                           const char *skip) 
 {
        int i;
@@ -254,7 +390,7 @@ const char **gensec_security_oids_from_ops(TALLOC_CTX *mem_ctx,
                return NULL;
        }
        
-       for (i=0; i<num_backends; i++) {
+       for (i=0; ops && ops[i]; i++) {
                if (!ops[i]->oid) {
                        continue;
                }
@@ -315,15 +451,20 @@ const char **gensec_security_oids_from_ops_wrapped(TALLOC_CTX *mem_ctx,
 
 
 /**
- * Return all the security subsystems currently enabled in GENSEC 
+ * Return all the security subsystems currently enabled on a GENSEC context.
+ * 
+ * This is taken from a list attached to the cli_credentails, and
+ * skips the OID in 'skip'.  (Typically the SPNEGO OID)
+ * 
  */
 
-const char **gensec_security_oids(TALLOC_CTX *mem_ctx, const char *skip) 
+const char **gensec_security_oids(struct gensec_security *gensec_security, 
+                                 TALLOC_CTX *mem_ctx, 
+                                 const char *skip) 
 {
-       int num_backends;
-       const struct gensec_security_ops **ops = gensec_security_all(&num_backends);
-       return gensec_security_oids_from_ops(mem_ctx, ops, 
-                                            num_backends, skip);
+       struct gensec_security_ops **ops
+               = gensec_security_mechs(gensec_security, mem_ctx);
+       return gensec_security_oids_from_ops(mem_ctx, ops, skip);
 }
 
 
@@ -335,8 +476,9 @@ const char **gensec_security_oids(TALLOC_CTX *mem_ctx, const char *skip)
   @note  The mem_ctx is only a parent and may be NULL.
 */
 static NTSTATUS gensec_start(TALLOC_CTX *mem_ctx, 
-                            struct gensec_security **gensec_security,
-                            struct event_context *ev) 
+                            struct event_context *ev,
+                            struct messaging_context *msg,
+                            struct gensec_security **gensec_security)
 {
        (*gensec_security) = talloc(mem_ctx, struct gensec_security);
        NT_STATUS_HAVE_NO_MEMORY(*gensec_security);
@@ -344,6 +486,8 @@ static NTSTATUS gensec_start(TALLOC_CTX *mem_ctx,
        (*gensec_security)->ops = NULL;
 
        ZERO_STRUCT((*gensec_security)->target);
+       ZERO_STRUCT((*gensec_security)->peer_addr);
+       ZERO_STRUCT((*gensec_security)->my_addr);
 
        (*gensec_security)->subcontext = False;
        (*gensec_security)->want_features = 0;
@@ -357,6 +501,7 @@ static NTSTATUS gensec_start(TALLOC_CTX *mem_ctx,
        }
 
        (*gensec_security)->event_ctx = ev;
+       (*gensec_security)->msg_ctx = msg;
 
        return NT_STATUS_OK;
 }
@@ -369,7 +514,7 @@ static NTSTATUS gensec_start(TALLOC_CTX *mem_ctx,
  * @note Used by SPNEGO in particular, for the actual implementation mechanism
  */
 
-NTSTATUS gensec_subcontext_start(TALLOC_CTX *mem_ctx, 
+_PUBLIC_ NTSTATUS gensec_subcontext_start(TALLOC_CTX *mem_ctx, 
                                 struct gensec_security *parent, 
                                 struct gensec_security **gensec_security)
 {
@@ -382,6 +527,7 @@ NTSTATUS gensec_subcontext_start(TALLOC_CTX *mem_ctx,
 
        (*gensec_security)->subcontext = True;
        (*gensec_security)->event_ctx = parent->event_ctx;
+       (*gensec_security)->msg_ctx = parent->msg_ctx;
 
        return NT_STATUS_OK;
 }
@@ -392,15 +538,25 @@ NTSTATUS gensec_subcontext_start(TALLOC_CTX *mem_ctx,
   @param gensec_security Returned GENSEC context pointer.
   @note  The mem_ctx is only a parent and may be NULL.
 */
-NTSTATUS gensec_client_start(TALLOC_CTX *mem_ctx, 
+_PUBLIC_ NTSTATUS gensec_client_start(TALLOC_CTX *mem_ctx, 
                             struct gensec_security **gensec_security,
                             struct event_context *ev)
 {
        NTSTATUS status;
-       status = gensec_start(mem_ctx, gensec_security, ev);
+       struct event_context *new_ev = NULL;
+
+       if (ev == NULL) {
+               new_ev = event_context_init(mem_ctx);
+               NT_STATUS_HAVE_NO_MEMORY(new_ev);
+               ev = new_ev;
+       }
+
+       status = gensec_start(mem_ctx, ev, NULL, gensec_security);
        if (!NT_STATUS_IS_OK(status)) {
+               talloc_free(new_ev);
                return status;
        }
+       talloc_steal((*gensec_security), new_ev);
        (*gensec_security)->gensec_role = GENSEC_CLIENT;
 
        return status;
@@ -413,11 +569,23 @@ NTSTATUS gensec_client_start(TALLOC_CTX *mem_ctx,
   @note  The mem_ctx is only a parent and may be NULL.
 */
 NTSTATUS gensec_server_start(TALLOC_CTX *mem_ctx, 
-                            struct gensec_security **gensec_security,
-                            struct event_context *ev)
+                            struct event_context *ev,
+                            struct messaging_context *msg,
+                            struct gensec_security **gensec_security)
 {
        NTSTATUS status;
-       status = gensec_start(mem_ctx, gensec_security, ev);
+
+       if (!ev) {
+               DEBUG(0,("gensec_server_start: no event context given!\n"));
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+
+       if (!msg) {
+               DEBUG(0,("gensec_server_start: no messaging context given!\n"));
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+
+       status = gensec_start(mem_ctx, ev, msg, gensec_security);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
@@ -442,6 +610,7 @@ static NTSTATUS gensec_start_mech(struct gensec_security *gensec_security)
                        }
                        return status;
                }
+               break;
        case GENSEC_SERVER:
                if (gensec_security->ops->server_start) {
                        status = gensec_security->ops->server_start(gensec_security);
@@ -451,6 +620,7 @@ static NTSTATUS gensec_start_mech(struct gensec_security *gensec_security)
                        }
                        return status;
                }
+               break;
        }
        return NT_STATUS_INVALID_PARAMETER;
 }
@@ -465,7 +635,7 @@ static NTSTATUS gensec_start_mech(struct gensec_security *gensec_security)
 NTSTATUS gensec_start_mech_by_authtype(struct gensec_security *gensec_security, 
                                       uint8_t auth_type, uint8_t auth_level) 
 {
-       gensec_security->ops = gensec_security_by_authtype(auth_type);
+       gensec_security->ops = gensec_security_by_authtype(gensec_security, auth_type);
        if (!gensec_security->ops) {
                DEBUG(3, ("Could not find GENSEC backend for auth_type=%d\n", (int)auth_type));
                return NT_STATUS_INVALID_PARAMETER;
@@ -491,7 +661,7 @@ NTSTATUS gensec_start_mech_by_authtype(struct gensec_security *gensec_security,
 const char *gensec_get_name_by_authtype(uint8_t authtype) 
 {
        const struct gensec_security_ops *ops;
-       ops = gensec_security_by_authtype(authtype);
+       ops = gensec_security_by_authtype(NULL, authtype);
        if (ops) {
                return ops->name;
        }
@@ -502,7 +672,7 @@ const char *gensec_get_name_by_authtype(uint8_t authtype)
 const char *gensec_get_name_by_oid(const char *oid_string) 
 {
        const struct gensec_security_ops *ops;
-       ops = gensec_security_by_oid(oid_string);
+       ops = gensec_security_by_oid(NULL, oid_string);
        if (ops) {
                return ops->name;
        }
@@ -532,7 +702,7 @@ NTSTATUS gensec_start_mech_by_ops(struct gensec_security *gensec_security,
 NTSTATUS gensec_start_mech_by_oid(struct gensec_security *gensec_security, 
                                  const char *mech_oid) 
 {
-       gensec_security->ops = gensec_security_by_oid(mech_oid);
+       gensec_security->ops = gensec_security_by_oid(gensec_security, mech_oid);
        if (!gensec_security->ops) {
                DEBUG(3, ("Could not find GENSEC backend for oid=%s\n", mech_oid));
                return NT_STATUS_INVALID_PARAMETER;
@@ -548,7 +718,7 @@ NTSTATUS gensec_start_mech_by_oid(struct gensec_security *gensec_security,
 NTSTATUS gensec_start_mech_by_sasl_name(struct gensec_security *gensec_security, 
                                        const char *sasl_name) 
 {
-       gensec_security->ops = gensec_security_by_sasl_name(sasl_name);
+       gensec_security->ops = gensec_security_by_sasl_name(gensec_security, sasl_name);
        if (!gensec_security->ops) {
                DEBUG(3, ("Could not find GENSEC backend for sasl_name=%s\n", sasl_name));
                return NT_STATUS_INVALID_PARAMETER;
@@ -556,6 +726,39 @@ NTSTATUS gensec_start_mech_by_sasl_name(struct gensec_security *gensec_security,
        return gensec_start_mech(gensec_security);
 }
 
+/** 
+ * Start a GENSEC sub-mechanism with the preferred option from a SASL name list
+ *
+ */
+
+_PUBLIC_ NTSTATUS gensec_start_mech_by_sasl_list(struct gensec_security *gensec_security, 
+                                                const char **sasl_names) 
+{
+       NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
+       TALLOC_CTX *mem_ctx = talloc_new(gensec_security);
+       const struct gensec_security_ops **ops;
+       int i;
+       if (!mem_ctx) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       ops = gensec_security_by_sasl_list(gensec_security, mem_ctx, sasl_names);
+       if (!ops || !*ops) {
+               DEBUG(3, ("Could not find GENSEC backend for any of sasl_name = %s\n", 
+                         str_list_join(mem_ctx, 
+                                       sasl_names, ' ')));
+               talloc_free(mem_ctx);
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+       for (i=0; ops[i]; i++) {
+               nt_status = gensec_start_mech_by_ops(gensec_security, ops[i]);
+               if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER)) {
+                       break;
+               }
+       }
+       talloc_free(mem_ctx);
+       return nt_status;
+}
+
 /** 
  * Start a GENSEC sub-mechanism by an internal name
  *
@@ -564,7 +767,7 @@ NTSTATUS gensec_start_mech_by_sasl_name(struct gensec_security *gensec_security,
 NTSTATUS gensec_start_mech_by_name(struct gensec_security *gensec_security, 
                                        const char *name) 
 {
-       gensec_security->ops = gensec_security_by_name(name);
+       gensec_security->ops = gensec_security_by_name(gensec_security, name);
        if (!gensec_security->ops) {
                DEBUG(3, ("Could not find GENSEC backend for name=%s\n", name));
                return NT_STATUS_INVALID_PARAMETER;
@@ -654,6 +857,24 @@ size_t gensec_sig_size(struct gensec_security *gensec_security, size_t data_size
        return gensec_security->ops->sig_size(gensec_security, data_size);
 }
 
+size_t gensec_max_wrapped_size(struct gensec_security *gensec_security) 
+{
+       if (!gensec_security->ops->max_wrapped_size) {
+               return (1 << 17);
+       }
+       
+       return gensec_security->ops->max_wrapped_size(gensec_security);
+}
+
+size_t gensec_max_input_size(struct gensec_security *gensec_security) 
+{
+       if (!gensec_security->ops->max_input_size) {
+               return (1 << 17) - gensec_sig_size(gensec_security, 1 << 17);
+       }
+       
+       return gensec_security->ops->max_input_size(gensec_security);
+}
+
 NTSTATUS gensec_wrap(struct gensec_security *gensec_security, 
                     TALLOC_CTX *mem_ctx, 
                     const DATA_BLOB *in, 
@@ -719,18 +940,86 @@ NTSTATUS gensec_session_info(struct gensec_security *gensec_security,
  *                or NT_STATUS_OK if the user is authenticated. 
  */
 
-NTSTATUS gensec_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
+_PUBLIC_ NTSTATUS gensec_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
                       const DATA_BLOB in, DATA_BLOB *out) 
 {
        return gensec_security->ops->update(gensec_security, out_mem_ctx, in, out);
 }
 
+static void gensec_update_async_timed_handler(struct event_context *ev, struct timed_event *te,
+                                             struct timeval t, void *ptr)
+{
+       struct gensec_update_request *req = talloc_get_type(ptr, struct gensec_update_request);
+       req->status = req->gensec_security->ops->update(req->gensec_security, req, req->in, &req->out);
+       req->callback.fn(req, req->callback.private_data);
+}
+
+/**
+ * Next state function for the GENSEC state machine async version
+ * 
+ * @param gensec_security GENSEC State
+ * @param in The request, as a DATA_BLOB
+ * @param callback The function that will be called when the operation is
+ *                 finished, it should return gensec_update_recv() to get output
+ * @param private_data A private pointer that will be passed to the callback function
+ */
+
+_PUBLIC_ void gensec_update_send(struct gensec_security *gensec_security, const DATA_BLOB in,
+                                void (*callback)(struct gensec_update_request *req, void *private_data),
+                                void *private_data)
+{
+       struct gensec_update_request *req = NULL;
+       struct timed_event *te = NULL;
+
+       req = talloc(gensec_security, struct gensec_update_request);
+       if (!req) goto failed;
+       req->gensec_security            = gensec_security;
+       req->in                         = in;
+       req->out                        = data_blob(NULL, 0);
+       req->callback.fn                = callback;
+       req->callback.private_data      = private_data;
+
+       te = event_add_timed(gensec_security->event_ctx, req,
+                            timeval_zero(),
+                            gensec_update_async_timed_handler, req);
+       if (!te) goto failed;
+
+       return;
+
+failed:
+       talloc_free(req);
+       callback(NULL, private_data);
+}
+
+/**
+ * Next state function for the GENSEC state machine
+ * 
+ * @param req GENSEC update request state
+ * @param out_mem_ctx The TALLOC_CTX for *out to be allocated on
+ * @param out The reply, as an talloc()ed DATA_BLOB, on *out_mem_ctx
+ * @return Error, MORE_PROCESSING_REQUIRED if a reply is sent, 
+ *                or NT_STATUS_OK if the user is authenticated. 
+ */
+_PUBLIC_ NTSTATUS gensec_update_recv(struct gensec_update_request *req, TALLOC_CTX *out_mem_ctx, DATA_BLOB *out)
+{
+       NTSTATUS status;
+
+       NT_STATUS_HAVE_NO_MEMORY(req);
+
+       *out = req->out;
+       talloc_steal(out_mem_ctx, out->data);
+       status = req->status;
+
+       talloc_free(req);
+       return status;
+}
+
 /** 
  * Set the requirement for a certain feature on the connection
  *
  */
 
-void gensec_want_feature(struct gensec_security *gensec_security,
+_PUBLIC_ void gensec_want_feature(struct gensec_security *gensec_security,
                         uint32_t feature) 
 {
        gensec_security->want_features |= feature;
@@ -741,38 +1030,41 @@ void gensec_want_feature(struct gensec_security *gensec_security,
  *
  */
 
-BOOL gensec_have_feature(struct gensec_security *gensec_security,
+_PUBLIC_ BOOL gensec_have_feature(struct gensec_security *gensec_security,
                         uint32_t feature) 
 {
        if (!gensec_security->ops->have_feature) {
                return False;
        }
        
-       /* Can only 'have' a feature if you already 'want'ed it */
-       if (gensec_security->want_features & feature) {
-               return gensec_security->ops->have_feature(gensec_security, feature);
-       }
-       return False;
+       /* We might 'have' features that we don't 'want', because the
+        * other end demanded them, or we can't neotiate them off */
+       return gensec_security->ops->have_feature(gensec_security, feature);
 }
 
 /** 
- * Associate a credentails structure with a GENSEC context - talloc_reference()s it to the context 
+ * Associate a credentials structure with a GENSEC context - talloc_reference()s it to the context 
  *
  */
 
-NTSTATUS gensec_set_credentials(struct gensec_security *gensec_security, struct cli_credentials *credentials) 
+_PUBLIC_ NTSTATUS gensec_set_credentials(struct gensec_security *gensec_security, struct cli_credentials *credentials) 
 {
        gensec_security->credentials = talloc_reference(gensec_security, credentials);
+       NT_STATUS_HAVE_NO_MEMORY(gensec_security->credentials);
+       gensec_want_feature(gensec_security, cli_credentials_get_gensec_features(gensec_security->credentials));
        return NT_STATUS_OK;
 }
 
 /** 
- * Return the credentails structure associated with a GENSEC context
+ * Return the credentials structure associated with a GENSEC context
  *
  */
 
 struct cli_credentials *gensec_get_credentials(struct gensec_security *gensec_security) 
 {
+       if (!gensec_security) {
+               return NULL;
+       }
        return gensec_security->credentials;
 }
 
@@ -781,7 +1073,7 @@ struct cli_credentials *gensec_get_credentials(struct gensec_security *gensec_se
  *
  */
 
-NTSTATUS gensec_set_target_service(struct gensec_security *gensec_security, const char *service) 
+_PUBLIC_ NTSTATUS gensec_set_target_service(struct gensec_security *gensec_security, const char *service) 
 {
        gensec_security->target.service = talloc_strdup(gensec_security, service);
        if (!gensec_security->target.service) {
@@ -790,7 +1082,7 @@ NTSTATUS gensec_set_target_service(struct gensec_security *gensec_security, cons
        return NT_STATUS_OK;
 }
 
-const char *gensec_get_target_service(struct gensec_security *gensec_security) 
+_PUBLIC_ const char *gensec_get_target_service(struct gensec_security *gensec_security) 
 {
        if (gensec_security->target.service) {
                return gensec_security->target.service;
@@ -804,25 +1096,84 @@ const char *gensec_get_target_service(struct gensec_security *gensec_security)
  *
  */
 
-NTSTATUS gensec_set_target_hostname(struct gensec_security *gensec_security, const char *hostname) 
+_PUBLIC_ NTSTATUS gensec_set_target_hostname(struct gensec_security *gensec_security, const char *hostname) 
 {
        gensec_security->target.hostname = talloc_strdup(gensec_security, hostname);
-       if (!gensec_security->target.hostname) {
+       if (hostname && !gensec_security->target.hostname) {
                return NT_STATUS_NO_MEMORY;
        }
        return NT_STATUS_OK;
 }
 
-const char *gensec_get_target_hostname(struct gensec_security *gensec_security) 
+_PUBLIC_ const char *gensec_get_target_hostname(struct gensec_security *gensec_security) 
 {
+       /* We allow the target hostname to be overriden for testing purposes */
+       const char *target_hostname = lp_parm_string(global_loadparm, NULL, "gensec", "target_hostname");
+       if (target_hostname) {
+               return target_hostname;
+       }
+
        if (gensec_security->target.hostname) {
                return gensec_security->target.hostname;
        }
 
-       /* TODO: Add a 'set sockaddr' call, and do a reverse lookup */
+       /* We could add use the 'set sockaddr' call, and do a reverse
+        * lookup, but this would be both insecure (compromising the
+        * way kerberos works) and add DNS timeouts */
        return NULL;
 }
 
+/** 
+ * Set (and talloc_reference) local and peer socket addresses onto a socket context on the GENSEC context 
+ *
+ * This is so that kerberos can include these addresses in
+ * cryptographic tokens, to avoid certain attacks.
+ */
+
+NTSTATUS gensec_set_my_addr(struct gensec_security *gensec_security, struct socket_address *my_addr) 
+{
+       gensec_security->my_addr = my_addr;
+       if (my_addr && !talloc_reference(gensec_security, my_addr)) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       return NT_STATUS_OK;
+}
+
+NTSTATUS gensec_set_peer_addr(struct gensec_security *gensec_security, struct socket_address *peer_addr) 
+{
+       gensec_security->peer_addr = peer_addr;
+       if (peer_addr && !talloc_reference(gensec_security, peer_addr)) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       return NT_STATUS_OK;
+}
+
+struct socket_address *gensec_get_my_addr(struct gensec_security *gensec_security) 
+{
+       if (gensec_security->my_addr) {
+               return gensec_security->my_addr;
+       }
+
+       /* We could add a 'set sockaddr' call, and do a lookup.  This
+        * would avoid needing to do system calls if nothing asks. */
+       return NULL;
+}
+
+struct socket_address *gensec_get_peer_addr(struct gensec_security *gensec_security) 
+{
+       if (gensec_security->peer_addr) {
+               return gensec_security->peer_addr;
+       }
+
+       /* We could add a 'set sockaddr' call, and do a lookup.  This
+        * would avoid needing to do system calls if nothing asks.
+        * However, this is not appropriate for the peer addres on
+        * datagram sockets */
+       return NULL;
+}
+
+
+
 /** 
  * Set the target principal (assuming it it known, say from the SPNEGO reply)
  *  - ensures it is talloc()ed 
@@ -853,30 +1204,29 @@ const char *gensec_get_target_principal(struct gensec_security *gensec_security)
   The 'name' can be later used by other backends to find the operations
   structure for this backend.
 */
-NTSTATUS gensec_register(const void *_ops)
+NTSTATUS gensec_register(const struct gensec_security_ops *ops)
 {
-       const struct gensec_security_ops *ops = _ops;
-       
-       if (!lp_parm_bool(-1, "gensec", ops->name, ops->enabled)) {
+       if (!lp_parm_bool(global_loadparm, NULL, "gensec", ops->name, ops->enabled)) {
                DEBUG(2,("gensec subsystem %s is disabled\n", ops->name));
                return NT_STATUS_OK;
        }
 
-       if (gensec_security_by_name(ops->name) != NULL) {
+       if (gensec_security_by_name(NULL, ops->name) != NULL) {
                /* its already registered! */
                DEBUG(0,("GENSEC backend '%s' already registered\n", 
                         ops->name));
                return NT_STATUS_OBJECT_NAME_COLLISION;
        }
 
-       generic_security_ops = realloc_p(generic_security_ops, 
-                                        const struct gensec_security_ops *, 
-                                        gensec_num_backends+2);
+       generic_security_ops = talloc_realloc(talloc_autofree_context(), 
+                                             generic_security_ops, 
+                                             struct gensec_security_ops *, 
+                                             gensec_num_backends+2);
        if (!generic_security_ops) {
-               smb_panic("out of memory in gensec_register");
+               return NT_STATUS_NO_MEMORY;
        }
 
-       generic_security_ops[gensec_num_backends] = ops;
+       generic_security_ops[gensec_num_backends] = discard_const_p(struct gensec_security_ops, ops);
        gensec_num_backends++;
        generic_security_ops[gensec_num_backends] = NULL;
 
@@ -902,10 +1252,31 @@ const struct gensec_critical_sizes *gensec_interface_version(void)
        return &critical_sizes;
 }
 
+static int sort_gensec(struct gensec_security_ops **gs1, struct gensec_security_ops **gs2) {
+       return (*gs2)->priority - (*gs1)->priority;
+}
+
 /*
   initialise the GENSEC subsystem
 */
 NTSTATUS gensec_init(void)
 {
+       static BOOL initialized = False;
+
+       init_module_fn static_init[] = STATIC_gensec_MODULES;
+       init_module_fn *shared_init;
+
+       if (initialized) return NT_STATUS_OK;
+       initialized = True;
+       
+       shared_init = load_samba_modules(NULL, global_loadparm, "gensec");
+
+       run_init_functions(static_init);
+       run_init_functions(shared_init);
+
+       talloc_free(shared_init);
+
+       qsort(generic_security_ops, gensec_num_backends, sizeof(*generic_security_ops), QSORT_CAST sort_gensec);
+       
        return NT_STATUS_OK;
 }