s3:winbindd/idmap make idmap modules loadable again
[rusty/samba.git] / source3 / winbindd / idmap_ad.c
index 8144d876d4061a4083ad96693e33877f0176c730..a3ecc476e62591d25799d20032d4f71580931c24 100644 (file)
@@ -9,6 +9,7 @@
  * Copyright (C) Andrew Bartlett <abartlet@samba.org> 2003
  * Copyright (C) Gerald (Jerry) Carter 2004-2007
  * Copyright (C) Luke Howard 2001-2004
+ * Copyright (C) Michael Adam 2008,2010
  *
  * 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
  */
 
 #include "includes.h"
+#include "winbindd.h"
+#include "../libds/common/flags.h"
+#include "ads.h"
+#include "libads/ldap_schema.h"
+#include "nss_info.h"
+#include "secrets.h"
+#include "idmap.h"
+#include "../libcli/ldap/ldap_ndr.h"
+#include "../libcli/security/security.h"
 
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_IDMAP
 } while (0)
 
 struct idmap_ad_context {
-       uint32_t filter_low_id;
-       uint32_t filter_high_id;
+       ADS_STRUCT *ads;
+       struct posix_schema *ad_schema;
+       enum wb_posix_mapping ad_map_type; /* WB_POSIX_MAP_UNKNOWN */
 };
 
 NTSTATUS init_module(void);
 
-static ADS_STRUCT *ad_idmap_ads = NULL;
-static struct posix_schema *ad_schema = NULL;
-static enum wb_posix_mapping ad_map_type = WB_POSIX_MAP_UNKNOWN;
-
 /************************************************************************
  ***********************************************************************/
 
-static ADS_STRUCT *ad_idmap_cached_connection_internal(void)
+static ADS_STATUS ad_idmap_cached_connection_internal(struct idmap_domain *dom)
 {
        ADS_STRUCT *ads;
        ADS_STATUS status;
        bool local = False;
        fstring dc_name;
        struct sockaddr_storage dc_ip;
+       struct idmap_ad_context *ctx;
+       char *ldap_server = NULL;
+       char *realm = NULL;
+       struct winbindd_domain *wb_dom;
+
+       DEBUG(10, ("ad_idmap_cached_connection: called for domain '%s'\n",
+                  dom->name));
 
-       if (ad_idmap_ads != NULL) {
+       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
+
+       if (ctx->ads != NULL) {
 
                time_t expire;
                time_t now = time(NULL);
 
-               ads = ad_idmap_ads;
+               ads = ctx->ads;
 
                expire = MIN(ads->auth.tgt_expire, ads->auth.tgs_expire);
 
@@ -76,15 +92,14 @@ static ADS_STRUCT *ad_idmap_cached_connection_internal(void)
                          (uint32)expire-(uint32)now, (uint32) expire, (uint32) now));
 
                if ( ads->config.realm && (expire > time(NULL))) {
-                       return ads;
+                       return ADS_SUCCESS;
                } else {
                        /* we own this ADS_STRUCT so make sure it goes away */
                        DEBUG(7,("Deleting expired krb5 credential cache\n"));
                        ads->is_mine = True;
                        ads_destroy( &ads );
                        ads_kdestroy(WINBIND_CCACHE_NAME);
-                       ad_idmap_ads = NULL;
-                       TALLOC_FREE( ad_schema );                       
+                       ctx->ads = NULL;
                }
        }
 
@@ -93,9 +108,28 @@ static ADS_STRUCT *ad_idmap_cached_connection_internal(void)
                setenv("KRB5CCNAME", WINBIND_CCACHE_NAME, 1);
        }
 
-       if ( (ads = ads_init(lp_realm(), lp_workgroup(), NULL)) == NULL ) {
+       /*
+        * At this point we only have the NetBIOS domain name.
+        * Check if we can get server nam and realm from SAF cache
+        * and the domain list.
+        */
+       ldap_server = saf_fetch(dom->name);
+       DEBUG(10, ("ldap_server from saf cache: '%s'\n", ldap_server?ldap_server:""));
+
+       wb_dom = find_domain_from_name_noinit(dom->name);
+       if (wb_dom == NULL) {
+               DEBUG(10, ("find_domain_from_name_noinit did not find domain '%s'\n",
+                          dom->name));
+               realm = NULL;
+       } else {
+               DEBUG(10, ("find_domain_from_name_noinit found realm '%s' for "
+                         " domain '%s'\n", wb_dom->alt_name, dom->name));
+               realm = wb_dom->alt_name;
+       }
+
+       if ( (ads = ads_init(realm, dom->name, ldap_server)) == NULL ) {
                DEBUG(1,("ads_init failed\n"));
-               return NULL;
+               return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
        }
 
        /* the machine acct password might have change - fetch it every time */
@@ -107,100 +141,107 @@ static ADS_STRUCT *ad_idmap_cached_connection_internal(void)
 
        /* setup server affinity */
 
-       get_dc_name( NULL, ads->auth.realm, dc_name, &dc_ip );
-       
+       get_dc_name(dom->name, realm, dc_name, &dc_ip );
+
        status = ads_connect(ads);
        if (!ADS_ERR_OK(status)) {
-               DEBUG(1, ("ad_idmap_init: failed to connect to AD\n"));
+               DEBUG(1, ("ad_idmap_cached_connection_internal: failed to "
+                         "connect to AD\n"));
                ads_destroy(&ads);
-               return NULL;
+               return status;
        }
 
        ads->is_mine = False;
 
-       ad_idmap_ads = ads;
+       ctx->ads = ads;
 
-       return ads;
+       return ADS_SUCCESS;
 }
 
 /************************************************************************
  ***********************************************************************/
 
-static ADS_STRUCT *ad_idmap_cached_connection(void)
+static ADS_STATUS ad_idmap_cached_connection(struct idmap_domain *dom)
 {
-       ADS_STRUCT *ads = ad_idmap_cached_connection_internal();
-       
-       if ( !ads )
-               return NULL;
+       ADS_STATUS status;
+       struct idmap_ad_context * ctx;
+
+       status = ad_idmap_cached_connection_internal(dom);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
 
        /* if we have a valid ADS_STRUCT and the schema model is
           defined, then we can return here. */
 
-       if ( ad_schema )
-               return ads;
+       if ( ctx->ad_schema ) {
+               return ADS_SUCCESS;
+       }
 
        /* Otherwise, set the schema model */
 
-       if ( (ad_map_type ==  WB_POSIX_MAP_SFU) ||
-            (ad_map_type ==  WB_POSIX_MAP_SFU20) || 
-            (ad_map_type ==  WB_POSIX_MAP_RFC2307) ) 
+       if ( (ctx->ad_map_type ==  WB_POSIX_MAP_SFU) ||
+            (ctx->ad_map_type ==  WB_POSIX_MAP_SFU20) ||
+            (ctx->ad_map_type ==  WB_POSIX_MAP_RFC2307) )
        {
-               ADS_STATUS schema_status;
-               
-               schema_status = ads_check_posix_schema_mapping( NULL, ads, ad_map_type, &ad_schema);
-               if ( !ADS_ERR_OK(schema_status) ) {
+               status = ads_check_posix_schema_mapping(
+                       ctx, ctx->ads, ctx->ad_map_type, &ctx->ad_schema);
+               if ( !ADS_ERR_OK(status) ) {
                        DEBUG(2,("ad_idmap_cached_connection: Failed to obtain schema details!\n"));
-                       return NULL;                    
                }
        }
-       
-       return ads;
+
+       return status;
+}
+
+static int idmap_ad_context_destructor(struct idmap_ad_context *ctx)
+{
+       if (ctx->ads != NULL) {
+               /* we own this ADS_STRUCT so make sure it goes away */
+               ctx->ads->is_mine = True;
+               ads_destroy( &ctx->ads );
+               ctx->ads = NULL;
+       }
+       return 0;
 }
 
 /************************************************************************
  ***********************************************************************/
 
-static NTSTATUS idmap_ad_initialize(struct idmap_domain *dom,
-                                   const char *params)
+static NTSTATUS idmap_ad_initialize(struct idmap_domain *dom)
 {
        struct idmap_ad_context *ctx;
        char *config_option;
-       const char *range = NULL;
        const char *schema_mode = NULL; 
 
-       if ( (ctx = TALLOC_ZERO_P(dom, struct idmap_ad_context)) == NULL ) {
+       ctx = talloc_zero(dom, struct idmap_ad_context);
+       if (ctx == NULL) {
                DEBUG(0, ("Out of memory!\n"));
                return NT_STATUS_NO_MEMORY;
        }
+       talloc_set_destructor(ctx, idmap_ad_context_destructor);
 
-       if ( (config_option = talloc_asprintf(ctx, "idmap config %s", dom->name)) == NULL ) {
+       config_option = talloc_asprintf(ctx, "idmap config %s", dom->name);
+       if (config_option == NULL) {
                DEBUG(0, ("Out of memory!\n"));
                talloc_free(ctx);
                return NT_STATUS_NO_MEMORY;
        }
 
-       /* load ranges */
-       range = lp_parm_const_string(-1, config_option, "range", NULL);
-       if (range && range[0]) {
-               if ((sscanf(range, "%u - %u", &ctx->filter_low_id, &ctx->filter_high_id) != 2) ||
-                   (ctx->filter_low_id > ctx->filter_high_id)) {
-                       DEBUG(1, ("ERROR: invalid filter range [%s]", range));
-                       ctx->filter_low_id = 0;
-                       ctx->filter_high_id = 0;
-               }
-       }
+       /* default map type */
+       ctx->ad_map_type = WB_POSIX_MAP_RFC2307;
 
        /* schema mode */
-       if ( ad_map_type == WB_POSIX_MAP_UNKNOWN )
-               ad_map_type = WB_POSIX_MAP_RFC2307;
        schema_mode = lp_parm_const_string(-1, config_option, "schema_mode", NULL);
        if ( schema_mode && schema_mode[0] ) {
                if ( strequal(schema_mode, "sfu") )
-                       ad_map_type = WB_POSIX_MAP_SFU;
+                       ctx->ad_map_type = WB_POSIX_MAP_SFU;
                else if ( strequal(schema_mode, "sfu20" ) )
-                       ad_map_type = WB_POSIX_MAP_SFU20;
+                       ctx->ad_map_type = WB_POSIX_MAP_SFU20;
                else if ( strequal(schema_mode, "rfc2307" ) )
-                       ad_map_type = WB_POSIX_MAP_RFC2307;
+                       ctx->ad_map_type = WB_POSIX_MAP_RFC2307;
                else
                        DEBUG(0,("idmap_ad_initialize: Unknown schema_mode (%s)\n",
                                 schema_mode));
@@ -234,12 +275,12 @@ static struct id_map *find_map_by_id(struct id_map **maps, enum id_type type, ui
  Search up to IDMAP_AD_MAX_IDS entries in maps for a match
  ***********************************************************************/
 
-static struct id_map *find_map_by_sid(struct id_map **maps, DOM_SID *sid)
+static struct id_map *find_map_by_sid(struct id_map **maps, struct dom_sid *sid)
 {
        int i;
 
        for (i = 0; maps[i] && i<IDMAP_AD_MAX_IDS; i++) {
-               if (sid_equal(maps[i]->sid, sid)) {
+               if (dom_sid_equal(maps[i]->sid, sid)) {
                        return maps[i];
                }
        }
@@ -256,7 +297,6 @@ static NTSTATUS idmap_ad_unixids_to_sids(struct idmap_domain *dom, struct id_map
        TALLOC_CTX *memctx;
        struct idmap_ad_context *ctx;
        ADS_STATUS rc;
-       ADS_STRUCT *ads;
        const char *attrs[] = { "sAMAccountType", 
                                "objectSid",
                                NULL, /* uidnumber */
@@ -272,6 +312,11 @@ static NTSTATUS idmap_ad_unixids_to_sids(struct idmap_domain *dom, struct id_map
        char *u_filter = NULL;
        char *g_filter = NULL;
 
+       /* initialize the status to avoid suprise */
+       for (i = 0; ids[i]; i++) {
+               ids[i]->status = ID_UNKNOWN;
+       }
+
        /* Only do query if we are online */
        if (idmap_is_offline()) {
                return NT_STATUS_FILE_IS_OFFLINE;
@@ -284,14 +329,16 @@ static NTSTATUS idmap_ad_unixids_to_sids(struct idmap_domain *dom, struct id_map
                return NT_STATUS_NO_MEMORY;
        }
 
-       if ( (ads = ad_idmap_cached_connection()) == NULL ) {
-               DEBUG(1, ("ADS uninitialized\n"));
+       rc = ad_idmap_cached_connection(dom);
+       if (!ADS_ERR_OK(rc)) {
+               DEBUG(1, ("ADS uninitialized: %s\n", ads_errstr(rc)));
                ret = NT_STATUS_UNSUCCESSFUL;
+               /* ret = ads_ntstatus(rc); */
                goto done;
        }
 
-       attrs[2] = ad_schema->posix_uidnumber_attr;
-       attrs[3] = ad_schema->posix_gidnumber_attr;
+       attrs[2] = ctx->ad_schema->posix_uidnumber_attr;
+       attrs[3] = ctx->ad_schema->posix_gidnumber_attr;
 
 again:
        bidx = idx;
@@ -308,11 +355,11 @@ again:
                                                           ATYPE_INTERDOMAIN_TRUST);
                        }
                        u_filter = talloc_asprintf_append_buffer(u_filter, "(%s=%lu)",
-                                                         ad_schema->posix_uidnumber_attr,
+                                                         ctx->ad_schema->posix_uidnumber_attr,
                                                          (unsigned long)ids[idx]->xid.id);
                        CHECK_ALLOC_DONE(u_filter);
                        break;
-                               
+
                case ID_TYPE_GID:
                        if ( ! g_filter) {
                                g_filter = talloc_asprintf(memctx, "(&(|"
@@ -322,7 +369,7 @@ again:
                                                           ATYPE_SECURITY_LOCAL_GROUP);
                        }
                        g_filter = talloc_asprintf_append_buffer(g_filter, "(%s=%lu)",
-                                                         ad_schema->posix_gidnumber_attr,
+                                                         ctx->ad_schema->posix_gidnumber_attr,
                                                          (unsigned long)ids[idx]->xid.id);
                        CHECK_ALLOC_DONE(g_filter);
                        break;
@@ -348,29 +395,29 @@ again:
        filter = talloc_asprintf_append_buffer(filter, ")");
        CHECK_ALLOC_DONE(filter);
 
-       rc = ads_search_retry(ads, &res, filter, attrs);
+       rc = ads_search_retry(ctx->ads, &res, filter, attrs);
        if (!ADS_ERR_OK(rc)) {
                DEBUG(1, ("ERROR: ads search returned: %s\n", ads_errstr(rc)));
                ret = NT_STATUS_UNSUCCESSFUL;
                goto done;
        }
 
-       if ( (count = ads_count_replies(ads, res)) == 0 ) {
+       if ( (count = ads_count_replies(ctx->ads, res)) == 0 ) {
                DEBUG(10, ("No IDs found\n"));
        }
 
        entry = res;
        for (i = 0; (i < count) && entry; i++) {
-               DOM_SID sid;
+               struct dom_sid sid;
                enum id_type type;
                struct id_map *map;
                uint32_t id;
                uint32_t atype;
 
                if (i == 0) { /* first entry */
-                       entry = ads_first_entry(ads, entry);
+                       entry = ads_first_entry(ctx->ads, entry);
                } else { /* following ones */
-                       entry = ads_next_entry(ads, entry);
+                       entry = ads_next_entry(ctx->ads, entry);
                }
 
                if ( !entry ) {
@@ -379,13 +426,13 @@ again:
                }
 
                /* first check if the SID is present */
-               if (!ads_pull_sid(ads, entry, "objectSid", &sid)) {
+               if (!ads_pull_sid(ctx->ads, entry, "objectSid", &sid)) {
                        DEBUG(2, ("Could not retrieve SID from entry\n"));
                        continue;
                }
 
                /* get type */
-               if (!ads_pull_uint32(ads, entry, "sAMAccountType", &atype)) {
+               if (!ads_pull_uint32(ctx->ads, entry, "sAMAccountType", &atype)) {
                        DEBUG(1, ("could not get SAM account type\n"));
                        continue;
                }
@@ -405,20 +452,18 @@ again:
                        continue;
                }
 
-               if (!ads_pull_uint32(ads, entry, (type==ID_TYPE_UID) ? 
-                                                ad_schema->posix_uidnumber_attr : 
-                                                ad_schema->posix_gidnumber_attr, 
+               if (!ads_pull_uint32(ctx->ads, entry, (type==ID_TYPE_UID) ?
+                                                ctx->ad_schema->posix_uidnumber_attr :
+                                                ctx->ad_schema->posix_gidnumber_attr,
                                     &id)) 
                {
                        DEBUG(1, ("Could not get unix ID\n"));
                        continue;
                }
 
-               if ((id == 0) ||
-                   (ctx->filter_low_id && (id < ctx->filter_low_id)) ||
-                   (ctx->filter_high_id && (id > ctx->filter_high_id))) {
+               if (!idmap_unix_id_is_in_range(id, dom)) {
                        DEBUG(5, ("Requested id (%u) out of range (%u - %u). Filtered!\n",
-                               id, ctx->filter_low_id, ctx->filter_high_id));
+                               id, dom->low_id, dom->high_id));
                        continue;
                }
 
@@ -439,7 +484,7 @@ again:
        }
 
        if (res) {
-               ads_msgfree(ads, res);
+               ads_msgfree(ctx->ads, res);
        }
 
        if (ids[idx]) { /* still some values to map */
@@ -468,7 +513,6 @@ static NTSTATUS idmap_ad_sids_to_unixids(struct idmap_domain *dom, struct id_map
        TALLOC_CTX *memctx;
        struct idmap_ad_context *ctx;
        ADS_STATUS rc;
-       ADS_STRUCT *ads;
        const char *attrs[] = { "sAMAccountType", 
                                "objectSid",
                                NULL, /* attr_uidnumber */
@@ -483,6 +527,11 @@ static NTSTATUS idmap_ad_sids_to_unixids(struct idmap_domain *dom, struct id_map
        int i;
        char *sidstr;
 
+       /* initialize the status to avoid suprise */
+       for (i = 0; ids[i]; i++) {
+               ids[i]->status = ID_UNKNOWN;
+       }
+
        /* Only do query if we are online */
        if (idmap_is_offline()) {
                return NT_STATUS_FILE_IS_OFFLINE;
@@ -495,14 +544,22 @@ static NTSTATUS idmap_ad_sids_to_unixids(struct idmap_domain *dom, struct id_map
                return NT_STATUS_NO_MEMORY;
        }
 
-       if ( (ads = ad_idmap_cached_connection()) == NULL ) {
-               DEBUG(1, ("ADS uninitialized\n"));
+       rc = ad_idmap_cached_connection(dom);
+       if (!ADS_ERR_OK(rc)) {
+               DEBUG(1, ("ADS uninitialized: %s\n", ads_errstr(rc)));
                ret = NT_STATUS_UNSUCCESSFUL;
+               /* ret = ads_ntstatus(rc); */
                goto done;
        }
 
-       attrs[2] = ad_schema->posix_uidnumber_attr;
-       attrs[3] = ad_schema->posix_gidnumber_attr;
+       if (ctx->ad_schema == NULL) {
+               DEBUG(0, ("haven't got ctx->ad_schema ! \n"));
+               ret = NT_STATUS_UNSUCCESSFUL;
+               goto done;
+       }
+
+       attrs[2] = ctx->ad_schema->posix_uidnumber_attr;
+       attrs[3] = ctx->ad_schema->posix_gidnumber_attr;
 
 again:
        filter = talloc_asprintf(memctx, "(&(|"
@@ -511,45 +568,47 @@ again:
                                 ")(|",
                                 ATYPE_NORMAL_ACCOUNT, ATYPE_WORKSTATION_TRUST, ATYPE_INTERDOMAIN_TRUST,
                                 ATYPE_SECURITY_GLOBAL_GROUP, ATYPE_SECURITY_LOCAL_GROUP);
-               
+
        CHECK_ALLOC_DONE(filter);
 
        bidx = idx;
        for (i = 0; (i < IDMAP_AD_MAX_IDS) && ids[idx]; i++, idx++) {
 
-               sidstr = sid_binstring(ids[idx]->sid);
+               ids[idx]->status = ID_UNKNOWN;
+
+               sidstr = ldap_encode_ndr_dom_sid(talloc_tos(), ids[idx]->sid);
                filter = talloc_asprintf_append_buffer(filter, "(objectSid=%s)", sidstr);
-                       
-               free(sidstr);
+
+               TALLOC_FREE(sidstr);
                CHECK_ALLOC_DONE(filter);
        }
        filter = talloc_asprintf_append_buffer(filter, "))");
        CHECK_ALLOC_DONE(filter);
        DEBUG(10, ("Filter: [%s]\n", filter));
 
-       rc = ads_search_retry(ads, &res, filter, attrs);
+       rc = ads_search_retry(ctx->ads, &res, filter, attrs);
        if (!ADS_ERR_OK(rc)) {
                DEBUG(1, ("ERROR: ads search returned: %s\n", ads_errstr(rc)));
                ret = NT_STATUS_UNSUCCESSFUL;
                goto done;
        }
 
-       if ( (count = ads_count_replies(ads, res)) == 0 ) {
+       if ( (count = ads_count_replies(ctx->ads, res)) == 0 ) {
                DEBUG(10, ("No IDs found\n"));
        }
 
        entry = res;    
        for (i = 0; (i < count) && entry; i++) {
-               DOM_SID sid;
+               struct dom_sid sid;
                enum id_type type;
                struct id_map *map;
                uint32_t id;
                uint32_t atype;
 
                if (i == 0) { /* first entry */
-                       entry = ads_first_entry(ads, entry);
+                       entry = ads_first_entry(ctx->ads, entry);
                } else { /* following ones */
-                       entry = ads_next_entry(ads, entry);
+                       entry = ads_next_entry(ctx->ads, entry);
                }
 
                if ( !entry ) {
@@ -558,7 +617,7 @@ again:
                }
 
                /* first check if the SID is present */
-               if (!ads_pull_sid(ads, entry, "objectSid", &sid)) {
+               if (!ads_pull_sid(ctx->ads, entry, "objectSid", &sid)) {
                        DEBUG(2, ("Could not retrieve SID from entry\n"));
                        continue;
                }
@@ -570,7 +629,7 @@ again:
                }
 
                /* get type */
-               if (!ads_pull_uint32(ads, entry, "sAMAccountType", &atype)) {
+               if (!ads_pull_uint32(ctx->ads, entry, "sAMAccountType", &atype)) {
                        DEBUG(1, ("could not get SAM account type\n"));
                        continue;
                }
@@ -590,19 +649,17 @@ again:
                        continue;
                }
 
-               if (!ads_pull_uint32(ads, entry, (type==ID_TYPE_UID) ? 
-                                                ad_schema->posix_uidnumber_attr : 
-                                                ad_schema->posix_gidnumber_attr, 
+               if (!ads_pull_uint32(ctx->ads, entry, (type==ID_TYPE_UID) ?
+                                                ctx->ad_schema->posix_uidnumber_attr :
+                                                ctx->ad_schema->posix_gidnumber_attr,
                                     &id)) 
                {
                        DEBUG(1, ("Could not get unix ID\n"));
                        continue;
                }
-               if ((id == 0) ||
-                   (ctx->filter_low_id && (id < ctx->filter_low_id)) ||
-                   (ctx->filter_high_id && (id > ctx->filter_high_id))) {
+               if (!idmap_unix_id_is_in_range(id, dom)) {
                        DEBUG(5, ("Requested id (%u) out of range (%u - %u). Filtered!\n",
-                               id, ctx->filter_low_id, ctx->filter_high_id));
+                               id, dom->low_id, dom->high_id));
                        continue;
                }
 
@@ -617,7 +674,7 @@ again:
        }
 
        if (res) {
-               ads_msgfree(ads, res);
+               ads_msgfree(ctx->ads, res);
        }
 
        if (ids[idx]) { /* still some values to map */
@@ -626,7 +683,7 @@ again:
 
        ret = NT_STATUS_OK;
 
-       /* mark all unknwoni/expired ones as unmapped */
+       /* mark all unknown/expired ones as unmapped */
        for (i = 0; ids[i]; i++) {
                if (ids[i]->status != ID_MAPPED) 
                        ids[i]->status = ID_UNMAPPED;
@@ -637,25 +694,6 @@ done:
        return ret;
 }
 
-/************************************************************************
- ***********************************************************************/
-
-static NTSTATUS idmap_ad_close(struct idmap_domain *dom)
-{
-       ADS_STRUCT *ads = ad_idmap_ads;
-
-       if (ads != NULL) {
-               /* we own this ADS_STRUCT so make sure it goes away */
-               ads->is_mine = True;
-               ads_destroy( &ads );
-               ad_idmap_ads = NULL;
-       }
-
-       TALLOC_FREE( ad_schema );
-       
-       return NT_STATUS_OK;
-}
-
 /*
  * nss_info_{sfu,sfu20,rfc2307}
  */
@@ -664,74 +702,112 @@ static NTSTATUS idmap_ad_close(struct idmap_domain *dom)
  Initialize the {sfu,sfu20,rfc2307} state
  ***********************************************************************/
 
-static NTSTATUS nss_sfu_init( struct nss_domain_entry *e )
-{
-       /* Sanity check if we have previously been called with a
-          different schema model */
+static const char *wb_posix_map_unknown_string = "WB_POSIX_MAP_UNKNOWN";
+static const char *wb_posix_map_template_string = "WB_POSIX_MAP_TEMPLATE";
+static const char *wb_posix_map_sfu_string = "WB_POSIX_MAP_SFU";
+static const char *wb_posix_map_sfu20_string = "WB_POSIX_MAP_SFU20";
+static const char *wb_posix_map_rfc2307_string = "WB_POSIX_MAP_RFC2307";
+static const char *wb_posix_map_unixinfo_string = "WB_POSIX_MAP_UNIXINFO";
 
-       if ( (ad_map_type != WB_POSIX_MAP_UNKNOWN) &&
-            (ad_map_type != WB_POSIX_MAP_SFU) ) 
-       {
-               DEBUG(0,("nss_sfu_init: Posix Map type has already been set.  "
-                        "Mixed schema models not supported!\n"));
-               return NT_STATUS_NOT_SUPPORTED;
+static const char *ad_map_type_string(enum wb_posix_mapping map_type)
+{
+       switch (map_type) {
+               case WB_POSIX_MAP_TEMPLATE:
+                       return wb_posix_map_template_string;
+               case WB_POSIX_MAP_SFU:
+                       return wb_posix_map_sfu_string;
+               case WB_POSIX_MAP_SFU20:
+                       return wb_posix_map_sfu20_string;
+               case WB_POSIX_MAP_RFC2307:
+                       return wb_posix_map_rfc2307_string;
+               case WB_POSIX_MAP_UNIXINFO:
+                       return wb_posix_map_unixinfo_string;
+               default:
+                       return wb_posix_map_unknown_string;
        }
-       
-       ad_map_type = WB_POSIX_MAP_SFU; 
-
-       return NT_STATUS_OK;
 }
 
-static NTSTATUS nss_sfu20_init( struct nss_domain_entry *e )
+static NTSTATUS nss_ad_generic_init(struct nss_domain_entry *e,
+                                   enum wb_posix_mapping new_ad_map_type)
 {
-       /* Sanity check if we have previously been called with a
-          different schema model */
+       struct idmap_domain *dom;
+       struct idmap_ad_context *ctx;
+
+       if (e->state != NULL) {
+               dom = talloc_get_type(e->state, struct idmap_domain);
+       } else {
+               dom = talloc_zero(e, struct idmap_domain);
+               if (dom == NULL) {
+                       DEBUG(0, ("Out of memory!\n"));
+                       return NT_STATUS_NO_MEMORY;
+               }
+               e->state = dom;
+       }
+
+       if (e->domain != NULL) {
+               dom->name = talloc_strdup(dom, e->domain);
+               if (dom->name == NULL) {
+                       DEBUG(0, ("Out of memory!\n"));
+                       return NT_STATUS_NO_MEMORY;
+               }
+       }
+
+       if (dom->private_data != NULL) {
+               ctx = talloc_get_type(dom->private_data,
+                                     struct idmap_ad_context);
+       } else {
+               ctx = talloc_zero(dom, struct idmap_ad_context);
+               if (ctx == NULL) {
+                       DEBUG(0, ("Out of memory!\n"));
+                       return NT_STATUS_NO_MEMORY;
+               }
+               ctx->ad_map_type = WB_POSIX_MAP_RFC2307;
+               dom->private_data = ctx;
+       }
 
-       if ( (ad_map_type != WB_POSIX_MAP_UNKNOWN) &&
-            (ad_map_type != WB_POSIX_MAP_SFU20) )
+       if ((ctx->ad_map_type != WB_POSIX_MAP_UNKNOWN) &&
+           (ctx->ad_map_type != new_ad_map_type))
        {
-               DEBUG(0,("nss_sfu20_init: Posix Map type has already been set.  "
-                        "Mixed schema models not supported!\n"));
-               return NT_STATUS_NOT_SUPPORTED;
+               DEBUG(2, ("nss_ad_generic_init: "
+                         "Warning: overriding previously set posix map type "
+                         "%s for domain %s with map type %s.\n",
+                         ad_map_type_string(ctx->ad_map_type),
+                         dom->name,
+                         ad_map_type_string(new_ad_map_type)));
        }
-       
-       ad_map_type = WB_POSIX_MAP_SFU20;       
+
+       ctx->ad_map_type = new_ad_map_type;
 
        return NT_STATUS_OK;
 }
 
-static NTSTATUS nss_rfc2307_init( struct nss_domain_entry *e )
+static NTSTATUS nss_sfu_init( struct nss_domain_entry *e )
 {
-       /* Sanity check if we have previously been called with a
-          different schema model */
-        
-       if ( (ad_map_type != WB_POSIX_MAP_UNKNOWN) &&
-            (ad_map_type != WB_POSIX_MAP_RFC2307) ) 
-       {
-               DEBUG(0,("nss_rfc2307_init: Posix Map type has already been set.  "
-                        "Mixed schema models not supported!\n"));
-               return NT_STATUS_NOT_SUPPORTED;
-       }
-       
-       ad_map_type = WB_POSIX_MAP_RFC2307;
+       return nss_ad_generic_init(e, WB_POSIX_MAP_SFU);
+}
 
-       return NT_STATUS_OK;
+static NTSTATUS nss_sfu20_init( struct nss_domain_entry *e )
+{
+       return nss_ad_generic_init(e, WB_POSIX_MAP_SFU20);
+}
+
+static NTSTATUS nss_rfc2307_init( struct nss_domain_entry *e )
+{
+       return nss_ad_generic_init(e, WB_POSIX_MAP_RFC2307);
 }
 
 
 /************************************************************************
  ***********************************************************************/
+
 static NTSTATUS nss_ad_get_info( struct nss_domain_entry *e, 
-                                 const DOM_SID *sid, 
-                                 TALLOC_CTX *ctx,
-                                 ADS_STRUCT *ads, 
-                                 LDAPMessage *msg,
-                                 char **homedir,
-                                 char **shell, 
-                                 char **gecos,
+                                 const struct dom_sid *sid,
+                                 TALLOC_CTX *mem_ctx,
+                                 const char **homedir,
+                                 const char **shell,
+                                 const char **gecos,
                                  uint32 *gid )
 {
-       ADS_STRUCT *ads_internal = NULL;
        const char *attrs[] = {NULL, /* attr_homedir */
                               NULL, /* attr_shell */
                               NULL, /* attr_gecos */
@@ -742,18 +818,27 @@ static NTSTATUS nss_ad_get_info( struct nss_domain_entry *e,
        ADS_STATUS ads_status = ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
        char *sidstr = NULL;
+       struct idmap_domain *dom;
+       struct idmap_ad_context *ctx;
+
+       DEBUG(10, ("nss_ad_get_info called for sid [%s] in domain '%s'\n",
+                  sid_string_dbg(sid), e->domain?e->domain:"NULL"));
 
        /* Only do query if we are online */
        if (idmap_is_offline()) {
                return NT_STATUS_FILE_IS_OFFLINE;
        }
 
-       /* We are assuming that the internal ADS_STRUCT is for the 
-          same forest as the incoming *ads pointer */
+       dom = talloc_get_type(e->state, struct idmap_domain);
+       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
 
-       ads_internal = ad_idmap_cached_connection();
+       ads_status = ad_idmap_cached_connection(dom);
+       if (!ADS_ERR_OK(ads_status)) {
+               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
+       }
 
-       if ( !ads_internal || !ad_schema ) {
+       if (!ctx->ad_schema) {
+               DEBUG(10, ("nss_ad_get_info: no ad_schema configured!\n"));
                return NT_STATUS_OBJECT_NAME_NOT_FOUND;
        }
 
@@ -761,50 +846,37 @@ static NTSTATUS nss_ad_get_info( struct nss_domain_entry *e,
                return NT_STATUS_INVALID_PARAMETER;
        }
 
-       /* See if we can use the ADS connection struct swe were given */
-
-       if (ads) {
-               *homedir = ads_pull_string( ads, ctx, msg, ad_schema->posix_homedir_attr );
-               *shell   = ads_pull_string( ads, ctx, msg, ad_schema->posix_shell_attr );
-               *gecos   = ads_pull_string( ads, ctx, msg, ad_schema->posix_gecos_attr );
-
-               if (gid) {
-                       if ( !ads_pull_uint32(ads, msg, ad_schema->posix_gidnumber_attr, gid ) )
-                               *gid = (uint32)-1;
-               }
-
-               nt_status = NT_STATUS_OK;
-               goto done;
-       }
-
        /* Have to do our own query */
 
-       attrs[0] = ad_schema->posix_homedir_attr;
-       attrs[1] = ad_schema->posix_shell_attr;
-       attrs[2] = ad_schema->posix_gecos_attr;
-       attrs[3] = ad_schema->posix_gidnumber_attr;
+       DEBUG(10, ("nss_ad_get_info: no ads connection given, doing our "
+                  "own query\n"));
 
-       sidstr = sid_binstring(sid);
-       filter = talloc_asprintf(ctx, "(objectSid=%s)", sidstr);
-       SAFE_FREE(sidstr);
+       attrs[0] = ctx->ad_schema->posix_homedir_attr;
+       attrs[1] = ctx->ad_schema->posix_shell_attr;
+       attrs[2] = ctx->ad_schema->posix_gecos_attr;
+       attrs[3] = ctx->ad_schema->posix_gidnumber_attr;
+
+       sidstr = ldap_encode_ndr_dom_sid(mem_ctx, sid);
+       filter = talloc_asprintf(mem_ctx, "(objectSid=%s)", sidstr);
+       TALLOC_FREE(sidstr);
 
        if (!filter) {
                nt_status = NT_STATUS_NO_MEMORY;
                goto done;
        }
 
-       ads_status = ads_search_retry(ads_internal, &msg_internal, filter, attrs);
+       ads_status = ads_search_retry(ctx->ads, &msg_internal, filter, attrs);
        if (!ADS_ERR_OK(ads_status)) {
                nt_status = ads_ntstatus(ads_status);
                goto done;
        }
 
-       *homedir = ads_pull_string(ads_internal, ctx, msg_internal, ad_schema->posix_homedir_attr);
-       *shell   = ads_pull_string(ads_internal, ctx, msg_internal, ad_schema->posix_shell_attr);
-       *gecos   = ads_pull_string(ads_internal, ctx, msg_internal, ad_schema->posix_gecos_attr);
+       *homedir = ads_pull_string(ctx->ads, mem_ctx, msg_internal, ctx->ad_schema->posix_homedir_attr);
+       *shell   = ads_pull_string(ctx->ads, mem_ctx, msg_internal, ctx->ad_schema->posix_shell_attr);
+       *gecos   = ads_pull_string(ctx->ads, mem_ctx, msg_internal, ctx->ad_schema->posix_gecos_attr);
 
        if (gid) {
-               if (!ads_pull_uint32(ads_internal, msg_internal, ad_schema->posix_gidnumber_attr, gid))
+               if (!ads_pull_uint32(ctx->ads, msg_internal, ctx->ad_schema->posix_gidnumber_attr, gid))
                        *gid = (uint32)-1;
        }
 
@@ -812,7 +884,7 @@ static NTSTATUS nss_ad_get_info( struct nss_domain_entry *e,
 
 done:
        if (msg_internal) {
-               ads_msgfree(ads_internal, msg_internal);
+               ads_msgfree(ctx->ads, msg_internal);
        }
 
        return nt_status;
@@ -822,21 +894,22 @@ done:
  *********************************************************************/
 
 static NTSTATUS nss_ad_map_to_alias(TALLOC_CTX *mem_ctx,
-                                   const char *domain,
+                                   struct nss_domain_entry *e,
                                    const char *name,
                                    char **alias)
 {
-       ADS_STRUCT *ads_internal = NULL;
        const char *attrs[] = {NULL, /* attr_uid */
                               NULL };
        char *filter = NULL;
        LDAPMessage *msg = NULL;
        ADS_STATUS ads_status = ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
+       struct idmap_domain *dom;
+       struct idmap_ad_context *ctx = NULL;
 
        /* Check incoming parameters */
 
-       if ( !domain || !name || !*alias) {
+       if ( !e || !e->domain || !name || !*alias) {
                nt_status = NT_STATUS_INVALID_PARAMETER;
                goto done;
        }
@@ -848,14 +921,20 @@ static NTSTATUS nss_ad_map_to_alias(TALLOC_CTX *mem_ctx,
                goto done;
        }
 
-       ads_internal = ad_idmap_cached_connection();
+       dom = talloc_get_type(e->state, struct idmap_domain);
+       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
 
-       if (!ads_internal || !ad_schema) {
+       ads_status = ad_idmap_cached_connection(dom);
+       if (!ADS_ERR_OK(ads_status)) {
+               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
+       }
+
+       if (!ctx->ad_schema) {
                nt_status = NT_STATUS_OBJECT_PATH_NOT_FOUND;
                goto done;
        }
 
-       attrs[0] = ad_schema->posix_uid_attr;
+       attrs[0] = ctx->ad_schema->posix_uid_attr;
 
        filter = talloc_asprintf(mem_ctx,
                                 "(sAMAccountName=%s)",
@@ -865,13 +944,13 @@ static NTSTATUS nss_ad_map_to_alias(TALLOC_CTX *mem_ctx,
                goto done;
        }
 
-       ads_status = ads_search_retry(ads_internal, &msg, filter, attrs);
+       ads_status = ads_search_retry(ctx->ads, &msg, filter, attrs);
        if (!ADS_ERR_OK(ads_status)) {
                nt_status = ads_ntstatus(ads_status);
                goto done;
        }
 
-       *alias = ads_pull_string(ads_internal, mem_ctx, msg, ad_schema->posix_uid_attr );
+       *alias = ads_pull_string(ctx->ads, mem_ctx, msg, ctx->ad_schema->posix_uid_attr);
 
        if (!*alias) {
                return NT_STATUS_OBJECT_NAME_NOT_FOUND;
@@ -884,7 +963,7 @@ done:
                talloc_destroy(filter);
        }
        if (msg) {
-               ads_msgfree(ads_internal, msg);
+               ads_msgfree(ctx->ads, msg);
        }
 
        return nt_status;
@@ -894,11 +973,10 @@ done:
  *********************************************************************/
 
 static NTSTATUS nss_ad_map_from_alias( TALLOC_CTX *mem_ctx,
-                                            const char *domain,
+                                            struct nss_domain_entry *e,
                                             const char *alias,
                                             char **name )
 {
-       ADS_STRUCT *ads_internal = NULL;
        const char *attrs[] = {"sAMAccountName",
                               NULL };
        char *filter = NULL;
@@ -906,6 +984,8 @@ static NTSTATUS nss_ad_map_from_alias( TALLOC_CTX *mem_ctx,
        ADS_STATUS ads_status = ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
        char *username;
+       struct idmap_domain *dom;
+       struct idmap_ad_context *ctx = NULL;
 
        /* Check incoming parameters */
 
@@ -921,29 +1001,35 @@ static NTSTATUS nss_ad_map_from_alias( TALLOC_CTX *mem_ctx,
                goto done;
        }
 
-       ads_internal = ad_idmap_cached_connection();
+       dom = talloc_get_type(e->state, struct idmap_domain);
+       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
+
+       ads_status = ad_idmap_cached_connection(dom);
+       if (!ADS_ERR_OK(ads_status)) {
+               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
+       }
 
-       if (!ads_internal || !ad_schema) {
+       if (!ctx->ad_schema) {
                nt_status = NT_STATUS_OBJECT_PATH_NOT_FOUND;
                goto done;
        }
 
        filter = talloc_asprintf(mem_ctx,
                                 "(%s=%s)",
-                                ad_schema->posix_uid_attr,
+                                ctx->ad_schema->posix_uid_attr,
                                 alias);
        if (!filter) {
                nt_status = NT_STATUS_NO_MEMORY;
                goto done;
        }
 
-       ads_status = ads_search_retry(ads_internal, &msg, filter, attrs);
+       ads_status = ads_search_retry(ctx->ads, &msg, filter, attrs);
        if (!ADS_ERR_OK(ads_status)) {
                nt_status = ads_ntstatus(ads_status);
                goto done;
        }
 
-       username = ads_pull_string(ads_internal, mem_ctx, msg,
+       username = ads_pull_string(ctx->ads, mem_ctx, msg,
                                   "sAMAccountName");
        if (!username) {
                return NT_STATUS_OBJECT_NAME_NOT_FOUND;
@@ -964,23 +1050,12 @@ done:
                talloc_destroy(filter);
        }
        if (msg) {
-               ads_msgfree(ads_internal, msg);
+               ads_msgfree(ctx->ads, msg);
        }
 
        return nt_status;
 }
 
-
-/************************************************************************
- ***********************************************************************/
-
-static NTSTATUS nss_ad_close( void )
-{
-       /* nothing to do.  All memory is free()'d by the idmap close_fn() */
-
-       return NT_STATUS_OK;
-}
-
 /************************************************************************
  Function dispatch tables for the idmap and nss plugins
  ***********************************************************************/
@@ -989,18 +1064,16 @@ static struct idmap_methods ad_methods = {
        .init            = idmap_ad_initialize,
        .unixids_to_sids = idmap_ad_unixids_to_sids,
        .sids_to_unixids = idmap_ad_sids_to_unixids,
-       .close_fn        = idmap_ad_close
 };
 
 /* The SFU and RFC2307 NSS plugins share everything but the init
    function which sets the intended schema model to use */
-  
+
 static struct nss_info_methods nss_rfc2307_methods = {
        .init           = nss_rfc2307_init,
        .get_nss_info   = nss_ad_get_info,
        .map_to_alias   = nss_ad_map_to_alias,
        .map_from_alias = nss_ad_map_from_alias,
-       .close_fn       = nss_ad_close
 };
 
 static struct nss_info_methods nss_sfu_methods = {
@@ -1008,7 +1081,6 @@ static struct nss_info_methods nss_sfu_methods = {
        .get_nss_info   = nss_ad_get_info,
        .map_to_alias   = nss_ad_map_to_alias,
        .map_from_alias = nss_ad_map_from_alias,
-       .close_fn       = nss_ad_close
 };
 
 static struct nss_info_methods nss_sfu20_methods = {
@@ -1016,7 +1088,6 @@ static struct nss_info_methods nss_sfu20_methods = {
        .get_nss_info   = nss_ad_get_info,
        .map_to_alias   = nss_ad_map_to_alias,
        .map_from_alias = nss_ad_map_from_alias,
-       .close_fn       = nss_ad_close
 };
 
 
@@ -1025,7 +1096,7 @@ static struct nss_info_methods nss_sfu20_methods = {
  Initialize the plugins
  ***********************************************************************/
 
-NTSTATUS idmap_ad_init(void)
+NTSTATUS samba_init_module(void)
 {
        static NTSTATUS status_idmap_ad = NT_STATUS_UNSUCCESSFUL;
        static NTSTATUS status_nss_rfc2307 = NT_STATUS_UNSUCCESSFUL;
@@ -1041,7 +1112,7 @@ NTSTATUS idmap_ad_init(void)
                if ( !NT_STATUS_IS_OK(status_idmap_ad) )
                        return status_idmap_ad;         
        }
-       
+
        if ( !NT_STATUS_IS_OK( status_nss_rfc2307 ) ) {
                status_nss_rfc2307 = smb_register_idmap_nss(SMB_NSS_INFO_INTERFACE_VERSION,
                                                            "rfc2307",  &nss_rfc2307_methods );