winbindd: Fix indentation
[sfrench/samba-autobuild/.git] / source3 / winbindd / idmap_ad.c
index ca1cf81ca9a97227a9bd1b2687de7c84e08a8628..315a9444a19ae16f66d1cdf899c567d31504bd99 100644 (file)
@@ -1,14 +1,7 @@
 /*
- *  idmap_ad: map between Active Directory and RFC 2307 or "Services for Unix" (SFU) Accounts
+ * idmap_ad: map between Active Directory and RFC 2307 accounts
  *
- * Unix SMB/CIFS implementation.
- *
- * Winbind ADS backend functions
- *
- * Copyright (C) Andrew Tridgell 2001
- * Copyright (C) Andrew Bartlett <abartlet@samba.org> 2003
- * Copyright (C) Gerald (Jerry) Carter 2004-2007
- * Copyright (C) Luke Howard 2001-2004
+ * Copyright (C) Volker Lendecke 2015
  *
  * 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 "idmap.h"
+#include "tldap_gensec_bind.h"
+#include "tldap_util.h"
+#include "passdb.h"
+#include "lib/param/param.h"
+#include "utils/net.h"
+#include "auth/gensec/gensec.h"
+#include "librpc/gen_ndr/ndr_netlogon.h"
+#include "libads/ldap_schema_oids.h"
+#include "../libds/common/flags.h"
+#include "libcli/ldap/ldap_ndr.h"
+#include "libcli/security/dom_sid.h"
+
+struct idmap_ad_schema_names;
 
-#undef DBGC_CLASS
-#define DBGC_CLASS DBGC_IDMAP
+struct idmap_ad_context {
+       struct idmap_domain *dom;
+       struct tldap_context *ld;
+       struct idmap_ad_schema_names *schema;
+       const char *default_nc;
 
-#define WINBIND_CCACHE_NAME "MEMORY:winbind_ccache"
+       bool unix_primary_group;
+       bool unix_nss_info;
+};
 
-#define IDMAP_AD_MAX_IDS 30
-#define CHECK_ALLOC_DONE(mem) do { \
-     if (!mem) { \
-           DEBUG(0, ("Out of memory!\n")); \
-           ret = NT_STATUS_NO_MEMORY; \
-           goto done; \
-      } \
-} while (0)
+static NTSTATUS idmap_ad_get_context(struct idmap_domain *dom,
+                                    struct idmap_ad_context **pctx);
 
-struct idmap_ad_context {
-       uint32_t filter_low_id;
-       uint32_t filter_high_id;
-};
+static char *get_schema_path(TALLOC_CTX *mem_ctx, struct tldap_context *ld)
+{
+       struct tldap_message *rootdse;
 
-NTSTATUS init_module(void);
+       rootdse = tldap_rootdse(ld);
+       if (rootdse == NULL) {
+               return NULL;
+       }
 
-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;
+       return tldap_talloc_single_attribute(rootdse, "schemaNamingContext",
+                                            mem_ctx);
+}
+
+static char *get_default_nc(TALLOC_CTX *mem_ctx, struct tldap_context *ld)
+{
+       struct tldap_message *rootdse;
+
+       rootdse = tldap_rootdse(ld);
+       if (rootdse == NULL) {
+               return NULL;
+       }
 
-/************************************************************************
- ***********************************************************************/
+       return tldap_talloc_single_attribute(rootdse, "defaultNamingContext",
+                                            mem_ctx);
+}
 
-static ADS_STRUCT *ad_idmap_cached_connection_internal(void)
+struct idmap_ad_schema_names {
+       char *name;
+       char *uid;
+       char *gid;
+       char *gecos;
+       char *dir;
+       char *shell;
+};
+
+static TLDAPRC get_attrnames_by_oids(struct tldap_context *ld,
+                                    TALLOC_CTX *mem_ctx,
+                                    const char *schema_path,
+                                    size_t num_oids,
+                                    const char **oids,
+                                    char **names)
 {
-       ADS_STRUCT *ads;
-       ADS_STATUS status;
-       bool local = False;
-       fstring dc_name;
-       struct sockaddr_storage dc_ip;
+       char *filter;
+       const char *attrs[] = { "lDAPDisplayName", "attributeId" };
+       size_t i;
+       TLDAPRC rc;
+       struct tldap_message **msgs;
+       size_t num_msgs;
+
+       filter = talloc_strdup(mem_ctx, "(|");
+       if (filter == NULL) {
+               return TLDAP_NO_MEMORY;
+       }
+
+       for (i=0; i<num_oids; i++) {
+               filter = talloc_asprintf_append_buffer(
+                       filter, "(attributeId=%s)", oids[i]);
+               if (filter == NULL) {
+                       return TLDAP_NO_MEMORY;
+               }
+       }
+
+       filter = talloc_asprintf_append_buffer(filter, ")");
+       if (filter == NULL) {
+               return TLDAP_NO_MEMORY;
+       }
+
+       rc = tldap_search(ld, schema_path, TLDAP_SCOPE_SUB, filter,
+                         attrs, ARRAY_SIZE(attrs), 0, NULL, 0, NULL, 0,
+                         0, 0, 0, mem_ctx, &msgs);;
+       TALLOC_FREE(filter);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               return rc;
+       }
 
-       if (ad_idmap_ads != NULL) {
+       for (i=0; i<num_oids; i++) {
+               names[i] = NULL;
+       }
 
-               time_t expire;
-               time_t now = time(NULL);
+       num_msgs = talloc_array_length(msgs);
 
-               ads = ad_idmap_ads;
+       for (i=0; i<num_msgs; i++) {
+               struct tldap_message *msg = msgs[i];
+               char *oid;
+               size_t j;
 
-               expire = MIN(ads->auth.tgt_expire, ads->auth.tgs_expire);
+               if (tldap_msg_type(msg) != TLDAP_RES_SEARCH_ENTRY) {
+                       /* Could be a TLDAP_RES_SEARCH_REFERENCE */
+                       continue;
+               }
 
-               /* check for a valid structure */
-               DEBUG(7, ("Current tickets expire in %d seconds (at %d, time is now %d)\n",
-                         (uint32)expire-(uint32)now, (uint32) expire, (uint32) now));
+               oid = tldap_talloc_single_attribute(
+                       msg, "attributeId", msg);
+               if (oid == NULL) {
+                       continue;
+               }
 
-               if ( ads->config.realm && (expire > time(NULL))) {
-                       return ads;
+               for (j=0; j<num_oids; j++) {
+                       if (strequal(oid, oids[j])) {
+                               break;
+                       }
                }
+               TALLOC_FREE(oid);
 
-               /* 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 );
+               if (j == num_oids) {
+                       /* not found */
+                       continue;
+               }
+
+               names[j] = tldap_talloc_single_attribute(
+                       msg, "lDAPDisplayName", mem_ctx);
        }
 
-       if (!local) {
-               /* we don't want this to affect the users ccache */
-               setenv("KRB5CCNAME", WINBIND_CCACHE_NAME, 1);
+       TALLOC_FREE(msgs);
+
+       return TLDAP_SUCCESS;
+}
+
+static TLDAPRC get_posix_schema_names(struct tldap_context *ld,
+                                     const char *schema_mode,
+                                     TALLOC_CTX *mem_ctx,
+                                     struct idmap_ad_schema_names **pschema)
+{
+       char *schema_path;
+       struct idmap_ad_schema_names *schema;
+       char *names[6];
+       const char *oids_sfu[] = {
+               ADS_ATTR_SFU_UIDNUMBER_OID,
+               ADS_ATTR_SFU_GIDNUMBER_OID,
+               ADS_ATTR_SFU_HOMEDIR_OID,
+               ADS_ATTR_SFU_SHELL_OID,
+               ADS_ATTR_SFU_GECOS_OID,
+               ADS_ATTR_SFU_UID_OID
+       };
+       const char *oids_sfu20[] = {
+               ADS_ATTR_SFU20_UIDNUMBER_OID,
+               ADS_ATTR_SFU20_GIDNUMBER_OID,
+               ADS_ATTR_SFU20_HOMEDIR_OID,
+               ADS_ATTR_SFU20_SHELL_OID,
+               ADS_ATTR_SFU20_GECOS_OID,
+               ADS_ATTR_SFU20_UID_OID
+       };
+       const char *oids_rfc2307[] = {
+               ADS_ATTR_RFC2307_UIDNUMBER_OID,
+               ADS_ATTR_RFC2307_GIDNUMBER_OID,
+               ADS_ATTR_RFC2307_HOMEDIR_OID,
+               ADS_ATTR_RFC2307_SHELL_OID,
+               ADS_ATTR_RFC2307_GECOS_OID,
+               ADS_ATTR_RFC2307_UID_OID
+       };
+       const char **oids;
+
+       TLDAPRC rc;
+
+       schema = talloc(mem_ctx, struct idmap_ad_schema_names);
+       if (schema == NULL) {
+               return TLDAP_NO_MEMORY;
+       }
+
+       schema_path = get_schema_path(schema, ld);
+       if (schema_path == NULL) {
+               TALLOC_FREE(schema);
+               return TLDAP_NO_MEMORY;
+       }
+
+       oids = oids_rfc2307;
+
+       if ((schema_mode != NULL) && (schema_mode[0] != '\0')) {
+               if (strequal(schema_mode, "sfu")) {
+                       oids = oids_sfu;
+               } else if (strequal(schema_mode, "sfu20")) {
+                       oids = oids_sfu20;
+               } else if (strequal(schema_mode, "rfc2307" )) {
+                       oids = oids_rfc2307;
+               } else {
+                       DBG_WARNING("Unknown schema mode %s\n", schema_mode);
+               }
        }
 
-       if ( (ads = ads_init(lp_realm(), lp_workgroup(), NULL)) == NULL ) {
-               DEBUG(1,("ads_init failed\n"));
-               return NULL;
+       rc = get_attrnames_by_oids(ld, schema, schema_path, 6, oids, names);
+       TALLOC_FREE(schema_path);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               TALLOC_FREE(schema);
+               return rc;
        }
 
-       /* the machine acct password might have changed - fetch it every
-        * time */
-       SAFE_FREE(ads->auth.password);
-       ads->auth.password = secrets_fetch_machine_password(lp_workgroup(), NULL, NULL);
+       schema->uid = names[0];
+       schema->gid = names[1];
+       schema->dir = names[2];
+       schema->shell = names[3];
+       schema->gecos = names[4];
+       schema->name = names[5];
+
+       *pschema = schema;
 
-       SAFE_FREE(ads->auth.realm);
-       ads->auth.realm = SMB_STRDUP(lp_realm());
+       return TLDAP_SUCCESS;
+}
 
-       /* setup server affinity */
+static NTSTATUS idmap_ad_get_tldap_ctx(TALLOC_CTX *mem_ctx,
+                                      const char *domname,
+                                      struct tldap_context **pld)
+{
+       struct netr_DsRGetDCNameInfo *dcinfo;
+       struct sockaddr_storage dcaddr;
+       struct cli_credentials *creds;
+       struct loadparm_context *lp_ctx;
+       struct tldap_context *ld;
+       int fd;
+       NTSTATUS status;
+       bool ok;
+       TLDAPRC rc;
 
-       get_dc_name( NULL, ads->auth.realm, dc_name, &dc_ip );
+       status = wb_dsgetdcname_gencache_get(mem_ctx, domname, &dcinfo);
+       if (!NT_STATUS_IS_OK(status)) {
+               DBG_DEBUG("Could not get dcinfo for %s: %s\n", domname,
+                         nt_errstr(status));
+               return status;
+       }
 
-       status = ads_connect(ads);
-       if (!ADS_ERR_OK(status)) {
-               DEBUG(1, ("ad_idmap_init: failed to connect to AD\n"));
-               ads_destroy(&ads);
-               return NULL;
+       if (dcinfo->dc_unc == NULL) {
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
+       }
+       if (dcinfo->dc_unc[0] == '\\') {
+               dcinfo->dc_unc += 1;
+       }
+       if (dcinfo->dc_unc[0] == '\\') {
+               dcinfo->dc_unc += 1;
+       }
+
+       ok = resolve_name(dcinfo->dc_unc, &dcaddr, 0x20, true);
+       if (!ok) {
+               DBG_DEBUG("Could not resolve name %s\n", dcinfo->dc_unc);
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
        }
 
-       ads->is_mine = False;
+       status = open_socket_out(&dcaddr, 389, 10000, &fd);
+       if (!NT_STATUS_IS_OK(status)) {
+               DBG_DEBUG("open_socket_out failed: %s\n", nt_errstr(status));
+               TALLOC_FREE(dcinfo);
+               return status;
+       }
+
+       ld = tldap_context_create(dcinfo, fd);
+       if (ld == NULL) {
+               DBG_DEBUG("tldap_context_create failed\n");
+               close(fd);
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       /*
+        * Here we use or own machine account as
+        * we run as domain member.
+        */
+       status = pdb_get_trust_credentials(lp_workgroup(),
+                                          lp_realm(),
+                                          dcinfo,
+                                          &creds);
+       if (!NT_STATUS_IS_OK(status)) {
+               DBG_DEBUG("pdb_get_trust_credentials() failed - %s\n",
+                         nt_errstr(status));
+               TALLOC_FREE(dcinfo);
+               return status;
+       }
+
+       lp_ctx = loadparm_init_s3(dcinfo, loadparm_s3_helpers());
+       if (lp_ctx == NULL) {
+               DBG_DEBUG("loadparm_init_s3 failed\n");
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       rc = tldap_gensec_bind(ld, creds, "ldap", dcinfo->dc_unc, NULL, lp_ctx,
+                              GENSEC_FEATURE_SIGN | GENSEC_FEATURE_SEAL);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               DBG_DEBUG("tldap_gensec_bind failed: %s\n",
+                         tldap_errstr(dcinfo, ld, rc));
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
+       }
 
-       ad_idmap_ads = ads;
+       rc = tldap_fetch_rootdse(ld);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               DBG_DEBUG("tldap_fetch_rootdse failed: %s\n",
+                         tldap_errstr(dcinfo, ld, rc));
+               TALLOC_FREE(dcinfo);
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
+       }
 
-       return ads;
+       *pld = talloc_move(mem_ctx, &ld);
+       TALLOC_FREE(dcinfo);
+       return NT_STATUS_OK;
 }
 
-/************************************************************************
- ***********************************************************************/
+static int idmap_ad_context_destructor(struct idmap_ad_context *ctx)
+{
+       if ((ctx->dom != NULL) && (ctx->dom->private_data == ctx)) {
+               ctx->dom->private_data = NULL;
+       }
+       return 0;
+}
 
-static ADS_STRUCT *ad_idmap_cached_connection(void)
+static NTSTATUS idmap_ad_context_create(TALLOC_CTX *mem_ctx,
+                                       struct idmap_domain *dom,
+                                       const char *domname,
+                                       struct idmap_ad_context **pctx)
 {
-       ADS_STRUCT *ads = ad_idmap_cached_connection_internal();
+       struct idmap_ad_context *ctx;
+       const char *schema_mode;
+       NTSTATUS status;
+       TLDAPRC rc;
 
-       if ( !ads )
-               return NULL;
+       ctx = talloc(mem_ctx, struct idmap_ad_context);
+       if (ctx == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       ctx->dom = dom;
 
-       /* if we have a valid ADS_STRUCT and the schema model is
-          defined, then we can return here. */
+       talloc_set_destructor(ctx, idmap_ad_context_destructor);
+
+       status = idmap_ad_get_tldap_ctx(ctx, domname, &ctx->ld);
+       if (!NT_STATUS_IS_OK(status)) {
+               DBG_DEBUG("idmap_ad_get_tldap_ctx failed: %s\n",
+                         nt_errstr(status));
+               TALLOC_FREE(ctx);
+               return status;
+       }
 
-       if ( ad_schema )
-               return ads;
+       ctx->default_nc = get_default_nc(ctx, ctx->ld);
+       if (ctx->default_nc == NULL) {
+               DBG_DEBUG("No default nc\n");
+               TALLOC_FREE(ctx);
+               return status;
+       }
 
-       /* Otherwise, set the schema model */
+       ctx->unix_primary_group = idmap_config_bool(
+               domname, "unix_primary_group", false);
+       ctx->unix_nss_info = idmap_config_bool(
+               domname, "unix_nss_info", false);
 
-       if ( (ad_map_type ==  WB_POSIX_MAP_SFU) ||
-            (ad_map_type ==  WB_POSIX_MAP_SFU20) || 
-            (ad_map_type ==  WB_POSIX_MAP_RFC2307) ) 
-       {
-               ADS_STATUS schema_status;
+       schema_mode = idmap_config_const_string(
+               domname, "schema_mode", "rfc2307");
 
-               schema_status = ads_check_posix_schema_mapping( NULL, ads, ad_map_type, &ad_schema);
-               if ( !ADS_ERR_OK(schema_status) ) {
-                       DEBUG(2,("ad_idmap_cached_connection: Failed to obtain schema details!\n"));
-                       return NULL;                    
-               }
+       rc = get_posix_schema_names(ctx->ld, schema_mode, ctx, &ctx->schema);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               DBG_DEBUG("get_posix_schema_names failed: %s\n",
+                         tldap_errstr(ctx, ctx->ld, rc));
+               TALLOC_FREE(ctx);
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
        }
 
-       return ads;
+       *pctx = ctx;
+       return NT_STATUS_OK;
 }
 
-/************************************************************************
- ***********************************************************************/
-
-static NTSTATUS idmap_ad_initialize(struct idmap_domain *dom)
+static NTSTATUS idmap_ad_query_user(struct idmap_domain *domain,
+                                   struct wbint_userinfo *info)
 {
        struct idmap_ad_context *ctx;
-       char *config_option;
-       const char *range = NULL;
-       const char *schema_mode = NULL; 
+       TLDAPRC rc;
+       NTSTATUS status;
+       char *sidstr, *filter;
+       const char *attrs[4];
+       size_t i, num_msgs;
+       struct tldap_message **msgs;
 
-       if ( (ctx = TALLOC_ZERO_P(dom, struct idmap_ad_context)) == NULL ) {
-               DEBUG(0, ("Out of memory!\n"));
-               return NT_STATUS_NO_MEMORY;
+       status = idmap_ad_get_context(domain, &ctx);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+
+       if (!(ctx->unix_primary_group || ctx->unix_nss_info)) {
+               return NT_STATUS_OK;
        }
 
-       if ( (config_option = talloc_asprintf(ctx, "idmap config %s", dom->name)) == NULL ) {
-               DEBUG(0, ("Out of memory!\n"));
-               talloc_free(ctx);
+       attrs[0] = ctx->schema->gid;
+       attrs[1] = ctx->schema->gecos;
+       attrs[2] = ctx->schema->dir;
+       attrs[3] = ctx->schema->shell;
+
+       sidstr = ldap_encode_ndr_dom_sid(talloc_tos(), &info->user_sid);
+       if (sidstr == NULL) {
                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;
-               }
+       filter = talloc_asprintf(talloc_tos(), "(objectsid=%s)", sidstr);
+       TALLOC_FREE(sidstr);
+       if (filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
        }
 
-       /* 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;
-               else if ( strequal(schema_mode, "sfu20" ) )
-                       ad_map_type = WB_POSIX_MAP_SFU20;
-               else if ( strequal(schema_mode, "rfc2307" ) )
-                       ad_map_type = WB_POSIX_MAP_RFC2307;
-               else
-                       DEBUG(0,("idmap_ad_initialize: Unknown schema_mode (%s)\n",
-                                schema_mode));
+       DBG_DEBUG("Filter: [%s]\n", filter);
+
+       rc = tldap_search(ctx->ld, ctx->default_nc, TLDAP_SCOPE_SUB, filter,
+                         attrs, ARRAY_SIZE(attrs), 0, NULL, 0, NULL, 0,
+                         0, 0, 0, talloc_tos(), &msgs);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
        }
 
-       dom->private_data = ctx;
-       dom->initialized = True;
+       TALLOC_FREE(filter);
+
+       num_msgs = talloc_array_length(msgs);
+
+       for (i=0; i<num_msgs; i++) {
+               struct tldap_message *msg = msgs[i];
+
+               if (tldap_msg_type(msg) != TLDAP_RES_SEARCH_ENTRY) {
+                       continue;
+               }
+
+               if (ctx->unix_primary_group) {
+                       bool ok;
+                       uint32_t gid;
+
+                       ok = tldap_pull_uint32(msg, ctx->schema->gid, &gid);
+                       if (ok) {
+                               DBG_DEBUG("Setting primary group "
+                                         "to %"PRIu32" from attr %s\n",
+                                         gid, ctx->schema->gid);
+                               info->primary_gid = gid;
+                       }
+               }
+
+               if (ctx->unix_nss_info) {
+                       char *attr;
+
+                       attr = tldap_talloc_single_attribute(
+                               msg, ctx->schema->dir, talloc_tos());
+                       if (attr != NULL) {
+                               info->homedir = talloc_move(info, &attr);
+                       }
+                       TALLOC_FREE(attr);
+
+                       attr = tldap_talloc_single_attribute(
+                               msg, ctx->schema->shell, talloc_tos());
+                       if (attr != NULL) {
+                               info->shell = talloc_move(info, &attr);
+                       }
+                       TALLOC_FREE(attr);
 
-       talloc_free(config_option);
+                       attr = tldap_talloc_single_attribute(
+                               msg, ctx->schema->gecos, talloc_tos());
+                       if (attr != NULL) {
+                               info->full_name = talloc_move(info, &attr);
+                       }
+                       TALLOC_FREE(attr);
+               }
+       }
 
        return NT_STATUS_OK;
 }
 
-/************************************************************************
- Search up to IDMAP_AD_MAX_IDS entries in maps for a match.
- ***********************************************************************/
-
-static struct id_map *find_map_by_id(struct id_map **maps, enum id_type type, uint32_t id)
+static NTSTATUS idmap_ad_query_user_retry(struct idmap_domain *domain,
+                                         struct wbint_userinfo *info)
 {
-       int i;
+       const NTSTATUS status_server_down =
+               NT_STATUS_LDAP(TLDAP_RC_V(TLDAP_SERVER_DOWN));
+       NTSTATUS status;
 
-       for (i = 0; maps[i] && i<IDMAP_AD_MAX_IDS; i++) {
-               if ((maps[i]->xid.type == type) && (maps[i]->xid.id == id)) {
-                       return maps[i];
-               }
+       status = idmap_ad_query_user(domain, info);
+
+       if (NT_STATUS_EQUAL(status, status_server_down)) {
+               TALLOC_FREE(domain->private_data);
+               status = idmap_ad_query_user(domain, info);
        }
 
-       return NULL;    
+       return status;
 }
 
-/************************************************************************
- Search up to IDMAP_AD_MAX_IDS entries in maps for a match
- ***********************************************************************/
+static NTSTATUS idmap_ad_initialize(struct idmap_domain *dom)
+{
+       dom->query_user = idmap_ad_query_user_retry;
+       dom->private_data = NULL;
+       return NT_STATUS_OK;
+}
 
-static struct id_map *find_map_by_sid(struct id_map **maps, DOM_SID *sid)
+static NTSTATUS idmap_ad_get_context(struct idmap_domain *dom,
+                                    struct idmap_ad_context **pctx)
 {
-       int i;
+       struct idmap_ad_context *ctx = NULL;
+       NTSTATUS status;
 
-       for (i = 0; maps[i] && i<IDMAP_AD_MAX_IDS; i++) {
-               if (sid_equal(maps[i]->sid, sid)) {
-                       return maps[i];
-               }
+       if (dom->private_data != NULL) {
+               *pctx = talloc_get_type_abort(dom->private_data,
+                                             struct idmap_ad_context);
+               return NT_STATUS_OK;
        }
 
-       return NULL;    
-}
+       status = idmap_ad_context_create(dom, dom, dom->name, &ctx);
+       if (!NT_STATUS_IS_OK(status)) {
+               DBG_DEBUG("idmap_ad_context_create failed: %s\n",
+                         nt_errstr(status));
+               return status;
+       }
 
-/************************************************************************
- ***********************************************************************/
+       dom->private_data = ctx;
+       *pctx = ctx;
+       return NT_STATUS_OK;
+}
 
-static NTSTATUS idmap_ad_unixids_to_sids(struct idmap_domain *dom, struct id_map **ids)
+static NTSTATUS idmap_ad_unixids_to_sids(struct idmap_domain *dom,
+                                        struct id_map **ids)
 {
-       NTSTATUS ret;
-       TALLOC_CTX *memctx;
        struct idmap_ad_context *ctx;
-       ADS_STATUS rc;
-       ADS_STRUCT *ads;
-       const char *attrs[] = { "sAMAccountType", 
-                               "objectSid",
-                               NULL, /* uidnumber */
-                               NULL, /* gidnumber */
-                               NULL };
-       LDAPMessage *res = NULL;
-       LDAPMessage *entry = NULL;
-       char *filter = NULL;
-       int idx = 0;
-       int bidx = 0;
-       int count;
-       int i;
-       char *u_filter = NULL;
-       char *g_filter = NULL;
-
-       /* Only do query if we are online */
-       if (idmap_is_offline()) {
-               return NT_STATUS_FILE_IS_OFFLINE;
-       }
-
-       /* Initilization my have been deferred because we were offline */
-       if ( ! dom->initialized) {
-               ret = idmap_ad_initialize(dom);
-               if ( ! NT_STATUS_IS_OK(ret)) {
-                       return ret;
-               }
+       TLDAPRC rc;
+       NTSTATUS status;
+       struct tldap_message **msgs;
+
+       size_t i, num_msgs;
+       char *u_filter, *g_filter, *filter;
+
+       const char *attrs[] = {
+               "sAMAccountType",
+               "objectSid",
+               NULL, /* attr_uidnumber */
+               NULL, /* attr_gidnumber */
+       };
+
+       status = idmap_ad_get_context(dom, &ctx);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);
+       attrs[2] = ctx->schema->uid;
+       attrs[3] = ctx->schema->gid;
 
-       if ( (memctx = talloc_new(ctx)) == NULL ) {
-               DEBUG(0, ("Out of memory!\n"));
+       u_filter = talloc_strdup(talloc_tos(), "");
+       if (u_filter == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
 
-       if ( (ads = ad_idmap_cached_connection()) == NULL ) {
-               DEBUG(1, ("ADS uninitialized\n"));
-               ret = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
-
-       attrs[2] = ad_schema->posix_uidnumber_attr;
-       attrs[3] = ad_schema->posix_gidnumber_attr;
-
-again:
-       bidx = idx;
-       for (i = 0; (i < IDMAP_AD_MAX_IDS) && ids[idx]; i++, idx++) {
-               switch (ids[idx]->xid.type) {
-               case ID_TYPE_UID:     
-                       if ( ! u_filter) {
-                               u_filter = talloc_asprintf(memctx, "(&(|"
-                                                          "(sAMAccountType=%d)"
-                                                          "(sAMAccountType=%d)"
-                                                          "(sAMAccountType=%d))(|",
-                                                          ATYPE_NORMAL_ACCOUNT,
-                                                          ATYPE_WORKSTATION_TRUST,
-                                                          ATYPE_INTERDOMAIN_TRUST);
-                       }
-                       u_filter = talloc_asprintf_append_buffer(u_filter, "(%s=%lu)",
-                                                         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, "(&(|"
-                                                          "(sAMAccountType=%d)"
-                                                          "(sAMAccountType=%d))(|",
-                                                          ATYPE_SECURITY_GLOBAL_GROUP,
-                                                          ATYPE_SECURITY_LOCAL_GROUP);
-                       }
-                       g_filter = talloc_asprintf_append_buffer(g_filter, "(%s=%lu)",
-                                                         ad_schema->posix_gidnumber_attr,
-                                                         (unsigned long)ids[idx]->xid.id);
-                       CHECK_ALLOC_DONE(g_filter);
-                       break;
+       g_filter = talloc_strdup(talloc_tos(), "");
+       if (g_filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-               default:
-                       DEBUG(3, ("Error: mapping requested but Unknown ID type\n"));
-                       ids[idx]->status = ID_UNKNOWN;
-                       continue;
+       for (i=0; ids[i] != NULL; i++) {
+               struct id_map *id = ids[i];
+
+               id->status = ID_UNKNOWN;
+
+               switch (id->xid.type) {
+                   case ID_TYPE_UID: {
+                           u_filter = talloc_asprintf_append_buffer(
+                                   u_filter, "(%s=%ju)", ctx->schema->uid,
+                                   (uintmax_t)id->xid.id);
+                           if (u_filter == NULL) {
+                                   return NT_STATUS_NO_MEMORY;
+                           }
+                           break;
+                   }
+
+                   case ID_TYPE_GID: {
+                           g_filter = talloc_asprintf_append_buffer(
+                                   g_filter, "(%s=%ju)", ctx->schema->gid,
+                                   (uintmax_t)id->xid.id);
+                           if (g_filter == NULL) {
+                                   return NT_STATUS_NO_MEMORY;
+                           }
+                           break;
+                   }
+
+                   default:
+                           DBG_WARNING("Unknown id type: %u\n",
+                                       (unsigned)id->xid.type);
+                           break;
                }
        }
-       filter = talloc_asprintf(memctx, "(|");
-       CHECK_ALLOC_DONE(filter);
-       if ( u_filter) {
-               filter = talloc_asprintf_append_buffer(filter, "%s))", u_filter);
-               CHECK_ALLOC_DONE(filter);
-                       TALLOC_FREE(u_filter);
+
+       filter = talloc_strdup(talloc_tos(), "(|");
+       if (filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
        }
-       if ( g_filter) {
-               filter = talloc_asprintf_append_buffer(filter, "%s))", g_filter);
-               CHECK_ALLOC_DONE(filter);
-               TALLOC_FREE(g_filter);
+
+       if (*u_filter != '\0') {
+               filter = talloc_asprintf_append_buffer(
+                       filter,
+                       "(&(|(sAMAccountType=%d)(sAMAccountType=%d)"
+                       "(sAMAccountType=%d))(|%s))",
+                       ATYPE_NORMAL_ACCOUNT, ATYPE_WORKSTATION_TRUST,
+                       ATYPE_INTERDOMAIN_TRUST, u_filter);
+               if (filter == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
-       filter = talloc_asprintf_append_buffer(filter, ")");
-       CHECK_ALLOC_DONE(filter);
+       TALLOC_FREE(u_filter);
+
+       if (*g_filter != '\0') {
+               filter = talloc_asprintf_append_buffer(
+                       filter,
+                       "(&(|(sAMAccountType=%d)(sAMAccountType=%d))(|%s))",
+                       ATYPE_SECURITY_GLOBAL_GROUP,
+                       ATYPE_SECURITY_LOCAL_GROUP,
+                       g_filter);
+               if (filter == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+       }
+       TALLOC_FREE(g_filter);
 
-       rc = ads_search_retry(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;
+       filter = talloc_asprintf_append_buffer(filter, ")");
+       if (filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
        }
 
-       if ( (count = ads_count_replies(ads, res)) == 0 ) {
-               DEBUG(10, ("No IDs found\n"));
+       DBG_DEBUG("Filter: [%s]\n", filter);
+
+       rc = tldap_search(ctx->ld, ctx->default_nc, TLDAP_SCOPE_SUB, filter,
+                         attrs, ARRAY_SIZE(attrs), 0, NULL, 0, NULL, 0,
+                         0, 0, 0, talloc_tos(), &msgs);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
        }
 
-       entry = res;
-       for (i = 0; (i < count) && entry; i++) {
-               DOM_SID sid;
-               enum id_type type;
-               struct id_map *map;
-               uint32_t id;
-               uint32_t atype;
+       TALLOC_FREE(filter);
 
-               if (i == 0) { /* first entry */
-                       entry = ads_first_entry(ads, entry);
-               } else { /* following ones */
-                       entry = ads_next_entry(ads, entry);
-               }
+       num_msgs = talloc_array_length(msgs);
 
-               if ( !entry ) {
-                       DEBUG(2, ("ERROR: Unable to fetch ldap entries from results\n"));
-                       break;
+       for (i=0; i<num_msgs; i++) {
+               struct tldap_message *msg = msgs[i];
+               char *dn;
+               struct id_map *map;
+               struct dom_sid sid;
+               size_t j;
+               bool ok;
+               uint32_t atype, xid;
+               enum id_type type;
+
+               if (tldap_msg_type(msg) != TLDAP_RES_SEARCH_ENTRY) {
+                       continue;
                }
 
-               /* first check if the SID is present */
-               if (!ads_pull_sid(ads, entry, "objectSid", &sid)) {
-                       DEBUG(2, ("Could not retrieve SID from entry\n"));
+               ok = tldap_entry_dn(msg, &dn);
+               if (!ok) {
+                       DBG_DEBUG("No dn found in msg %zu\n", i);
                        continue;
                }
 
-               /* get type */
-               if (!ads_pull_uint32(ads, entry, "sAMAccountType", &atype)) {
-                       DEBUG(1, ("could not get SAM account type\n"));
+               ok = tldap_pull_uint32(msg, "sAMAccountType", &atype);
+               if (!ok) {
+                       DBG_DEBUG("No atype in object %s\n", dn);
                        continue;
                }
 
                switch (atype & 0xF0000000) {
-               case ATYPE_SECURITY_GLOBAL_GROUP:
-               case ATYPE_SECURITY_LOCAL_GROUP:
-                       type = ID_TYPE_GID;
-                       break;
-               case ATYPE_NORMAL_ACCOUNT:
-               case ATYPE_WORKSTATION_TRUST:
-               case ATYPE_INTERDOMAIN_TRUST:
-                       type = ID_TYPE_UID;
-                       break;
-               default:
-                       DEBUG(1, ("unrecognized SAM account type %08x\n", atype));
+                   case ATYPE_SECURITY_GLOBAL_GROUP:
+                   case ATYPE_SECURITY_LOCAL_GROUP:
+                           type = ID_TYPE_GID;
+                           break;
+                   case ATYPE_NORMAL_ACCOUNT:
+                   case ATYPE_WORKSTATION_TRUST:
+                   case ATYPE_INTERDOMAIN_TRUST:
+                           type = ID_TYPE_UID;
+                           break;
+                   default:
+                           DBG_WARNING("unrecognized SAM account type %08x\n",
+                                       atype);
                        continue;
                }
 
-               if (!ads_pull_uint32(ads, entry, (type==ID_TYPE_UID) ? 
-                                                ad_schema->posix_uidnumber_attr : 
-                                                ad_schema->posix_gidnumber_attr, 
-                                    &id)) 
-               {
-                       DEBUG(1, ("Could not get unix ID\n"));
+               ok = tldap_pull_uint32(msg, (type == ID_TYPE_UID) ?
+                                      ctx->schema->uid : ctx->schema->gid,
+                                      &xid);
+               if (!ok) {
+                       DBG_WARNING("No unix id in object %s\n", dn);
                        continue;
                }
 
-               if ((id == 0) ||
-                   (ctx->filter_low_id && (id < ctx->filter_low_id)) ||
-                   (ctx->filter_high_id && (id > ctx->filter_high_id))) {
-                       DEBUG(5, ("Requested id (%u) out of range (%u - %u). Filtered!\n",
-                               id, ctx->filter_low_id, ctx->filter_high_id));
+               ok = tldap_pull_binsid(msg, "objectSid", &sid);
+               if (!ok) {
+                       DBG_DEBUG("No objectSid in object %s\n", dn);
                        continue;
                }
 
-               map = find_map_by_id(&ids[bidx], type, id);
-               if (!map) {
-                       DEBUG(2, ("WARNING: couldn't match result with requested ID\n"));
+               map = NULL;
+               for (j=0; ids[j]; j++) {
+                       if ((type == ids[j]->xid.type) &&
+                           (xid == ids[j]->xid.id)) {
+                               map = ids[j];
+                               break;
+                       }
+               }
+               if (map == NULL) {
+                       DBG_DEBUG("Got unexpected sid %s from object %s\n",
+                                 sid_string_tos(&sid), dn);
                        continue;
                }
 
                sid_copy(map->sid, &sid);
-
-               /* mapped */
                map->status = ID_MAPPED;
 
-               DEBUG(10, ("Mapped %s -> %lu (%d)\n", sid_string_dbg(map->sid),
-                          (unsigned long)map->xid.id,
-                          map->xid.type));
-       }
-
-       if (res) {
-               ads_msgfree(ads, res);
-       }
-
-       if (ids[idx]) { /* still some values to map */
-               goto again;
+               DBG_DEBUG("Mapped %s -> %ju (%d)\n", sid_string_dbg(map->sid),
+                         (uintmax_t)map->xid.id, map->xid.type);
        }
 
-       ret = NT_STATUS_OK;
-
-       /* mark all unknown/expired ones as unmapped */
-       for (i = 0; ids[i]; i++) {
-               if (ids[i]->status != ID_MAPPED) 
-                       ids[i]->status = ID_UNMAPPED;
-       }
+       TALLOC_FREE(msgs);
 
-done:
-       talloc_free(memctx);
-       return ret;
+       return NT_STATUS_OK;
 }
 
-/************************************************************************
- ***********************************************************************/
-
-static NTSTATUS idmap_ad_sids_to_unixids(struct idmap_domain *dom, struct id_map **ids)
+static NTSTATUS idmap_ad_sids_to_unixids(struct idmap_domain *dom,
+                                        struct id_map **ids)
 {
-       NTSTATUS ret;
-       TALLOC_CTX *memctx;
        struct idmap_ad_context *ctx;
-       ADS_STATUS rc;
-       ADS_STRUCT *ads;
-       const char *attrs[] = { "sAMAccountType", 
-                               "objectSid",
-                               NULL, /* attr_uidnumber */
-                               NULL, /* attr_gidnumber */
-                               NULL };
-       LDAPMessage *res = NULL;
-       LDAPMessage *entry = NULL;
-       char *filter = NULL;
-       int idx = 0;
-       int bidx = 0;
-       int count;
-       int i;
-       char *sidstr;
-
-       /* Only do query if we are online */
-       if (idmap_is_offline()) {
-               return NT_STATUS_FILE_IS_OFFLINE;
-       }
-
-       /* Initilization my have been deferred because we were offline */
-       if ( ! dom->initialized) {
-               ret = idmap_ad_initialize(dom);
-               if ( ! NT_STATUS_IS_OK(ret)) {
-                       return ret;
-               }
-       }
-
-       ctx = talloc_get_type(dom->private_data, struct idmap_ad_context);      
-
-       if ( (memctx = talloc_new(ctx)) == NULL ) {             
-               DEBUG(0, ("Out of memory!\n"));
+       TLDAPRC rc;
+       NTSTATUS status;
+       struct tldap_message **msgs;
+
+       char *filter;
+       size_t i, num_msgs;
+
+       const char *attrs[] = {
+               "sAMAccountType",
+               "objectSid",
+               NULL, /* attr_uidnumber */
+               NULL, /* attr_gidnumber */
+       };
+
+       status = idmap_ad_get_context(dom, &ctx);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+
+       attrs[2] = ctx->schema->uid;
+       attrs[3] = ctx->schema->gid;
+
+       filter = talloc_asprintf(
+               talloc_tos(),
+               "(&(|(sAMAccountType=%d)(sAMAccountType=%d)(sAMAccountType=%d)"
+               "(sAMAccountType=%d)(sAMAccountType=%d))(|",
+               ATYPE_NORMAL_ACCOUNT, ATYPE_WORKSTATION_TRUST,
+               ATYPE_INTERDOMAIN_TRUST, ATYPE_SECURITY_GLOBAL_GROUP,
+               ATYPE_SECURITY_LOCAL_GROUP);
+       if (filter == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
 
-       if ( (ads = ad_idmap_cached_connection()) == NULL ) {
-               DEBUG(1, ("ADS uninitialized\n"));
-               ret = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
+       for (i=0; ids[i]; i++) {
+               char *sidstr;
 
-       attrs[2] = ad_schema->posix_uidnumber_attr;
-       attrs[3] = ad_schema->posix_gidnumber_attr;
+               ids[i]->status = ID_UNKNOWN;
 
-again:
-       filter = talloc_asprintf(memctx, "(&(|"
-                                "(sAMAccountType=%d)(sAMAccountType=%d)(sAMAccountType=%d)" /* user account types */
-                                "(sAMAccountType=%d)(sAMAccountType=%d)" /* group account types */
-                                ")(|",
-                                ATYPE_NORMAL_ACCOUNT, ATYPE_WORKSTATION_TRUST, ATYPE_INTERDOMAIN_TRUST,
-                                ATYPE_SECURITY_GLOBAL_GROUP, ATYPE_SECURITY_LOCAL_GROUP);
+               sidstr = ldap_encode_ndr_dom_sid(talloc_tos(), ids[i]->sid);
+               if (sidstr == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
 
-       CHECK_ALLOC_DONE(filter);
+               filter = talloc_asprintf_append_buffer(
+                       filter, "(objectSid=%s)", sidstr);
+               TALLOC_FREE(sidstr);
+               if (filter == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+       }
 
-       bidx = idx;
-       for (i = 0; (i < IDMAP_AD_MAX_IDS) && ids[idx]; i++, idx++) {
+       filter = talloc_asprintf_append_buffer(filter, "))");
+       if (filter == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-               sidstr = sid_binstring(ids[idx]->sid);
-               filter = talloc_asprintf_append_buffer(filter, "(objectSid=%s)", sidstr);
+       DBG_DEBUG("Filter: [%s]\n", filter);
 
-               free(sidstr);
-               CHECK_ALLOC_DONE(filter);
+       rc = tldap_search(ctx->ld, ctx->default_nc, TLDAP_SCOPE_SUB, filter,
+                         attrs, ARRAY_SIZE(attrs), 0, NULL, 0, NULL, 0,
+                         0, 0, 0, talloc_tos(), &msgs);
+       if (!TLDAP_RC_IS_SUCCESS(rc)) {
+               return NT_STATUS_LDAP(TLDAP_RC_V(rc));
        }
-       filter = talloc_asprintf_append_buffer(filter, "))");
-       CHECK_ALLOC_DONE(filter);
-       DEBUG(10, ("Filter: [%s]\n", filter));
 
-       rc = ads_search_retry(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;
-       }
+       TALLOC_FREE(filter);
 
-       if ( (count = ads_count_replies(ads, res)) == 0 ) {
-               DEBUG(10, ("No IDs found\n"));
-       }
+       num_msgs = talloc_array_length(msgs);
 
-       entry = res;    
-       for (i = 0; (i < count) && entry; i++) {
-               DOM_SID sid;
-               enum id_type type;
+       for (i=0; i<num_msgs; i++) {
+               struct tldap_message *msg = msgs[i];
+               char *dn;
                struct id_map *map;
-               uint32_t id;
-               uint32_t atype;
+               struct dom_sid sid;
+               size_t j;
+               bool ok;
+               uint64_t account_type, xid;
+               enum id_type type;
 
-               if (i == 0) { /* first entry */
-                       entry = ads_first_entry(ads, entry);
-               } else { /* following ones */
-                       entry = ads_next_entry(ads, entry);
+               if (tldap_msg_type(msg) != TLDAP_RES_SEARCH_ENTRY) {
+                       continue;
                }
 
-               if ( !entry ) {
-                       DEBUG(2, ("ERROR: Unable to fetch ldap entries from results\n"));
-                       break;
+               ok = tldap_entry_dn(msg, &dn);
+               if (!ok) {
+                       DBG_DEBUG("No dn found in msg %zu\n", i);
+                       continue;
                }
 
-               /* first check if the SID is present */
-               if (!ads_pull_sid(ads, entry, "objectSid", &sid)) {
-                       DEBUG(2, ("Could not retrieve SID from entry\n"));
+               ok = tldap_pull_binsid(msg, "objectSid", &sid);
+               if (!ok) {
+                       DBG_DEBUG("No objectSid in object %s\n", dn);
                        continue;
                }
 
-               map = find_map_by_sid(&ids[bidx], &sid);
-               if (!map) {
-                       DEBUG(2, ("WARNING: couldn't match result with requested SID\n"));
+               map = NULL;
+               for (j=0; ids[j]; j++) {
+                       if (dom_sid_equal(&sid, ids[j]->sid)) {
+                               map = ids[j];
+                               break;
+                       }
+               }
+               if (map == NULL) {
+                       DBG_DEBUG("Got unexpected sid %s from object %s\n",
+                                 sid_string_tos(&sid), dn);
                        continue;
                }
 
-               /* get type */
-               if (!ads_pull_uint32(ads, entry, "sAMAccountType", &atype)) {
-                       DEBUG(1, ("could not get SAM account type\n"));
+               ok = tldap_pull_uint64(msg, "sAMAccountType", &account_type);
+               if (!ok) {
+                       DBG_DEBUG("No sAMAccountType in %s\n", dn);
                        continue;
                }
 
-               switch (atype & 0xF0000000) {
+               switch (account_type & 0xF0000000) {
                case ATYPE_SECURITY_GLOBAL_GROUP:
                case ATYPE_SECURITY_LOCAL_GROUP:
                        type = ID_TYPE_GID;
@@ -603,267 +879,89 @@ again:
                        type = ID_TYPE_UID;
                        break;
                default:
-                       DEBUG(1, ("unrecognized SAM account type %08x\n", atype));
+                       DBG_WARNING("unrecognized SAM account type %"PRIu64"\n",
+                                   account_type);
                        continue;
                }
 
-               if (!ads_pull_uint32(ads, entry, (type==ID_TYPE_UID) ? 
-                                                ad_schema->posix_uidnumber_attr : 
-                                                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))) {
-                       DEBUG(5, ("Requested id (%u) out of range (%u - %u). Filtered!\n",
-                               id, ctx->filter_low_id, ctx->filter_high_id));
+               ok = tldap_pull_uint64(msg,
+                                      type == ID_TYPE_UID ?
+                                      ctx->schema->uid : ctx->schema->gid,
+                                      &xid);
+               if (!ok) {
+                       DBG_DEBUG("No xid in %s\n", dn);
                        continue;
                }
 
                /* mapped */
                map->xid.type = type;
-               map->xid.id = id;
+               map->xid.id = xid;
                map->status = ID_MAPPED;
 
                DEBUG(10, ("Mapped %s -> %lu (%d)\n", sid_string_dbg(map->sid),
-                          (unsigned long)map->xid.id,
-                          map->xid.type));
-       }
-
-       if (res) {
-               ads_msgfree(ads, res);
-       }
-
-       if (ids[idx]) { /* still some values to map */
-               goto again;
-       }
-
-       ret = NT_STATUS_OK;
-
-       /* mark all unknwoni/expired ones as unmapped */
-       for (i = 0; ids[i]; i++) {
-               if (ids[i]->status != ID_MAPPED) 
-                       ids[i]->status = ID_UNMAPPED;
-       }
-
-done:
-       talloc_free(memctx);
-       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}
- */
-
-/************************************************************************
- 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 */
-
-       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;
+                          (unsigned long)map->xid.id, map->xid.type));
        }
 
-       ad_map_type = WB_POSIX_MAP_SFU; 
+       TALLOC_FREE(msgs);
 
        return NT_STATUS_OK;
 }
 
-static NTSTATUS nss_sfu20_init( struct nss_domain_entry *e )
+static NTSTATUS idmap_ad_unixids_to_sids_retry(struct idmap_domain *dom,
+                                              struct id_map **ids)
 {
-       /* 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_SFU20) )
-       {
-               DEBUG(0,("nss_sfu20_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_SFU20;       
-
-       return NT_STATUS_OK;
-}
+       const NTSTATUS status_server_down =
+               NT_STATUS_LDAP(TLDAP_RC_V(TLDAP_SERVER_DOWN));
+       NTSTATUS status;
 
-static NTSTATUS nss_rfc2307_init( struct nss_domain_entry *e )
-{
-       /* Sanity check if we have previously been called with a
-          different schema model */
+       status = idmap_ad_unixids_to_sids(dom, ids);
 
-       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;
+       if (NT_STATUS_EQUAL(status, status_server_down)) {
+               TALLOC_FREE(dom->private_data);
+               status = idmap_ad_unixids_to_sids(dom, ids);
        }
 
-       ad_map_type = WB_POSIX_MAP_RFC2307;
-
-       return NT_STATUS_OK;
+       return status;
 }
 
-
-/************************************************************************
- ***********************************************************************/
-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,
-                                 uint32 *gid )
+static NTSTATUS idmap_ad_sids_to_unixids_retry(struct idmap_domain *dom,
+                                              struct id_map **ids)
 {
-       ADS_STRUCT *ads_internal = 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 */
+       const NTSTATUS status_server_down =
+               NT_STATUS_LDAP(TLDAP_RC_V(TLDAP_SERVER_DOWN));
+       NTSTATUS status;
 
-       ads_internal = ad_idmap_cached_connection();
+       status = idmap_ad_sids_to_unixids(dom, ids);
 
-       if ( !ads_internal || !ad_schema )
-               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
-
-       if ( !homedir || !shell || !gecos )
-               return NT_STATUS_INVALID_PARAMETER;
-
-       *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;              
+       if (NT_STATUS_EQUAL(status, status_server_down)) {
+               TALLOC_FREE(dom->private_data);
+               status = idmap_ad_sids_to_unixids(dom, ids);
        }
 
-       return NT_STATUS_OK;
-}
-
-/************************************************************************
- ***********************************************************************/
-
-static NTSTATUS nss_ad_close( void )
-{
-       /* nothing to do.  All memory is free()'d by the idmap close_fn() */
-
-       return NT_STATUS_OK;
+       return status;
 }
 
-/************************************************************************
- Function dispatch tables for the idmap and nss plugins
- ***********************************************************************/
-
 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,
-       .close_fn     = nss_ad_close
-};
-
-static struct nss_info_methods nss_sfu_methods = {
-       .init         = nss_sfu_init,
-       .get_nss_info = nss_ad_get_info,
-       .close_fn     = nss_ad_close
+       .unixids_to_sids = idmap_ad_unixids_to_sids_retry,
+       .sids_to_unixids = idmap_ad_sids_to_unixids_retry,
 };
 
-static struct nss_info_methods nss_sfu20_methods = {
-       .init         = nss_sfu20_init,
-       .get_nss_info = nss_ad_get_info,
-       .close_fn     = nss_ad_close
-};
-
-
-
-/************************************************************************
- Initialize the plugins
- ***********************************************************************/
-
-NTSTATUS idmap_ad_init(void)
+static_decl_idmap;
+NTSTATUS idmap_ad_init(TALLOC_CTX *ctx)
 {
-       static NTSTATUS status_idmap_ad = NT_STATUS_UNSUCCESSFUL;
-       static NTSTATUS status_nss_rfc2307 = NT_STATUS_UNSUCCESSFUL;
-       static NTSTATUS status_nss_sfu = NT_STATUS_UNSUCCESSFUL;
-       static NTSTATUS status_nss_sfu20 = NT_STATUS_UNSUCCESSFUL;
-
-       /* Always register the AD method first in order to get the
-          idmap_domain interface called */
-
-       if ( !NT_STATUS_IS_OK(status_idmap_ad) ) {
-               status_idmap_ad = smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION, 
-                                                    "ad", &ad_methods);
-               if ( !NT_STATUS_IS_OK(status_idmap_ad) )
-                       return status_idmap_ad;         
-       }
+       NTSTATUS status;
 
-       if ( !NT_STATUS_IS_OK( status_nss_rfc2307 ) ) {
-               status_nss_rfc2307 = smb_register_idmap_nss(SMB_NSS_INFO_INTERFACE_VERSION,
-                                                           "rfc2307",  &nss_rfc2307_methods );         
-               if ( !NT_STATUS_IS_OK(status_nss_rfc2307) )
-                       return status_nss_rfc2307;
+       status = smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION,
+                                   "ad", &ad_methods);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       if ( !NT_STATUS_IS_OK( status_nss_sfu ) ) {
-               status_nss_sfu = smb_register_idmap_nss(SMB_NSS_INFO_INTERFACE_VERSION,
-                                                       "sfu",  &nss_sfu_methods );             
-               if ( !NT_STATUS_IS_OK(status_nss_sfu) )
-                       return status_nss_sfu;          
+       status = idmap_ad_nss_init(ctx);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       if ( !NT_STATUS_IS_OK( status_nss_sfu20 ) ) {
-               status_nss_sfu20 = smb_register_idmap_nss(SMB_NSS_INFO_INTERFACE_VERSION,
-                                                       "sfu20",  &nss_sfu20_methods );         
-               if ( !NT_STATUS_IS_OK(status_nss_sfu20) )
-                       return status_nss_sfu20;                
-       }
-
-       return NT_STATUS_OK;    
+       return NT_STATUS_OK;
 }
-