r17943: The horror, the horror. Add KDC site support by
[tprouty/samba.git] / source / libads / ldap.c
index 99227f6574846df4fc11e3771e908f6e7f69456a..1d192895d905891a5fc908efa7c8c341c1f3ac3b 100644 (file)
@@ -4,6 +4,8 @@
    Copyright (C) Andrew Tridgell 2001
    Copyright (C) Remus Koos 2001
    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2002
+   Copyright (C) Guenther Deschner 2005
+   Copyright (C) Gerald Carter 2006
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
  **/
 
 
+#define LDAP_SERVER_TREE_DELETE_OID    "1.2.840.113556.1.4.805"
+
+static SIG_ATOMIC_T gotalarm;
+                                                                                                                   
+/***************************************************************
+ Signal function to tell us we timed out.
+****************************************************************/
+
+static void gotalarm_sig(void)
+{
+       gotalarm = 1;
+}
+
+ LDAP *ldap_open_with_timeout(const char *server, int port, unsigned int to)
+{
+       LDAP *ldp = NULL;
+
+       /* Setup timeout */
+       gotalarm = 0;
+       CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
+       alarm(to);
+       /* End setup timeout. */
+
+       ldp = ldap_open(server, port);
+
+       if (ldp == NULL) {
+               DEBUG(2,("Could not open LDAP connection to %s:%d: %s\n",
+                        server, port, strerror(errno)));
+       }
+
+       /* Teardown timeout. */
+       CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
+       alarm(0);
+
+       return ldp;
+}
+
+static int ldap_search_with_timeout(LDAP *ld,
+                                   LDAP_CONST char *base,
+                                   int scope,
+                                   LDAP_CONST char *filter,
+                                   char **attrs,
+                                   int attrsonly,
+                                   LDAPControl **sctrls,
+                                   LDAPControl **cctrls,
+                                   int sizelimit,
+                                   LDAPMessage **res )
+{
+       struct timeval timeout;
+       int result;
+
+       /* Setup timeout for the ldap_search_ext_s call - local and remote. */
+       timeout.tv_sec = lp_ldap_timeout();
+       timeout.tv_usec = 0;
+
+       /* Setup alarm timeout.... Do we need both of these ? JRA. */
+       gotalarm = 0;
+       CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
+       alarm(lp_ldap_timeout());
+       /* End setup timeout. */
+
+       result = ldap_search_ext_s(ld, base, scope, filter, attrs,
+                                  attrsonly, sctrls, cctrls, &timeout,
+                                  sizelimit, res);
+
+       /* Teardown timeout. */
+       CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
+       alarm(0);
+
+       if (gotalarm != 0)
+               return LDAP_TIMELIMIT_EXCEEDED;
+
+       return result;
+}
+
 /*
   try a connection to a given ldap server, returning True and setting the servers IP
   in the ads struct if successful
-  
-  TODO : add a negative connection cache in here leveraged off of the one
-  found in the rpc code.  --jerry
  */
-static BOOL ads_try_connect(ADS_STRUCT *ads, const char *server, unsigned port)
+BOOL ads_try_connect(ADS_STRUCT *ads, const char *server )
 {
        char *srv;
+       struct cldap_netlogon_reply cldap_reply;
 
        if (!server || !*server) {
                return False;
        }
-
-       DEBUG(5,("ads_try_connect: trying ldap server '%s' port %u\n", server, port));
+       
+       DEBUG(5,("ads_try_connect: sending CLDAP request to %s (realm: %s)\n", 
+               server, ads->server.realm));
 
        /* this copes with inet_ntoa brokenness */
-       srv = strdup(server);
+       
+       srv = SMB_STRDUP(server);
 
-       ads->ld = ldap_open(srv, port);
-       if (!ads->ld) {
-               free(srv);
+       ZERO_STRUCT( cldap_reply );
+
+       if ( !ads_cldap_netlogon( srv, ads->server.realm, &cldap_reply ) ) {
+               DEBUG(3,("ads_try_connect: CLDAP request %s failed.\n", srv));
+               SAFE_FREE( srv );
                return False;
        }
-       ads->ldap_port = port;
-       ads->ldap_ip = *interpret_addr2(srv);
-       free(srv);
-
-       return True;
-}
 
-/*
-  try a connection to a given ldap server, based on URL, returning True if successful
- */
-static BOOL ads_try_connect_uri(ADS_STRUCT *ads)
-{
-#if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
-       DEBUG(5,("ads_try_connect: trying ldap server at URI '%s'\n", 
-                ads->server.ldap_uri));
+       /* Check the CLDAP reply flags */
 
-       
-       if (ldap_initialize((LDAP**)&(ads->ld), ads->server.ldap_uri) == LDAP_SUCCESS) {
-               return True;
+       if ( !(cldap_reply.flags & ADS_LDAP) ) {
+               DEBUG(1,("ads_try_connect: %s's CLDAP reply says it is not an LDAP server!\n",
+                       srv));
+               SAFE_FREE( srv );
+               return False;
        }
-       DEBUG(0, ("ldap_initialize: %s\n", strerror(errno)));
-       
-#else 
 
-       DEBUG(1, ("no URL support in LDAP libs!\n"));
-#endif
+       /* Fill in the ads->config values */
 
-       return False;
+       SAFE_FREE(ads->config.realm);
+       SAFE_FREE(ads->config.bind_path);
+       SAFE_FREE(ads->config.ldap_server_name);
+       SAFE_FREE(ads->server.workgroup);
+
+       ads->config.flags              = cldap_reply.flags;
+       ads->config.ldap_server_name   = SMB_STRDUP(cldap_reply.hostname);
+       strupper_m(cldap_reply.domain);
+       ads->config.realm              = SMB_STRDUP(cldap_reply.domain);
+       ads->config.bind_path          = ads_build_dn(ads->config.realm);
+       ads->server.workgroup          = SMB_STRDUP(cldap_reply.netbios_domain);
+
+       ads->ldap_port = LDAP_PORT;
+       ads->ldap_ip = *interpret_addr2(srv);
+       SAFE_FREE(srv);
+       
+       /* Store our site name. */
+       sitename_store( cldap_reply.client_site_name );
+
+       return True;
 }
 
 /**********************************************************************
@@ -99,7 +182,7 @@ static BOOL ads_try_connect_uri(ADS_STRUCT *ads)
  disabled
 **********************************************************************/
 
-static BOOL ads_find_dc(ADS_STRUCT *ads)
+static NTSTATUS ads_find_dc(ADS_STRUCT *ads)
 {
        const char *c_realm;
        int count, i=0;
@@ -107,6 +190,7 @@ static BOOL ads_find_dc(ADS_STRUCT *ads)
        pstring realm;
        BOOL got_realm = False;
        BOOL use_own_domain = False;
+       NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
 
        /* if the realm and workgroup are both empty, assume they are ours */
 
@@ -137,7 +221,7 @@ again:
                
                if ( !c_realm || !*c_realm ) {
                        DEBUG(0,("ads_find_dc: no realm or workgroup!  Don't know what to do\n"));
-                       return False;
+                       return NT_STATUS_INVALID_PARAMETER; /* rather need MISSING_PARAMETER ... */
                }
        }
        
@@ -146,20 +230,19 @@ again:
        DEBUG(6,("ads_find_dc: looking for %s '%s'\n", 
                (got_realm ? "realm" : "domain"), realm));
 
-       if ( !get_sorted_dc_list(realm, &ip_list, &count, got_realm) ) {
+       status = get_sorted_dc_list(realm, &ip_list, &count, got_realm);
+       if (!NT_STATUS_IS_OK(status)) {
                /* fall back to netbios if we can */
                if ( got_realm && !lp_disable_netbios() ) {
                        got_realm = False;
                        goto again;
                }
                
-               return False;
+               return status;
        }
-                       
+
        /* if we fail this loop, then giveup since all the IP addresses returned were dead */
        for ( i=0; i<count; i++ ) {
-               /* since this is an ads conection request, default to LDAP_PORT is not set */
-               int port = (ip_list[i].port!=PORT_NONE) ? ip_list[i].port : LDAP_PORT;
                fstring server;
                
                fstrcpy( server, inet_ntoa(ip_list[i].ip) );
@@ -167,9 +250,9 @@ again:
                if ( !NT_STATUS_IS_OK(check_negative_conn_cache(realm, server)) )
                        continue;
                        
-               if ( ads_try_connect(ads, server, port) ) {
+               if ( ads_try_connect(ads, server) ) {
                        SAFE_FREE(ip_list);
-                       return True;
+                       return NT_STATUS_OK;
                }
                
                /* keep track of failures */
@@ -178,7 +261,7 @@ again:
 
        SAFE_FREE(ip_list);
        
-       return False;
+       return NT_STATUS_NO_LOGON_SERVERS;
 }
 
 
@@ -191,54 +274,41 @@ ADS_STATUS ads_connect(ADS_STRUCT *ads)
 {
        int version = LDAP_VERSION3;
        ADS_STATUS status;
+       NTSTATUS ntstatus;
 
        ads->last_attempt = time(NULL);
        ads->ld = NULL;
 
-       /* try with a URL based server */
-
-       if (ads->server.ldap_uri &&
-           ads_try_connect_uri(ads)) {
-               goto got_connection;
-       }
-
        /* try with a user specified server */
+
        if (ads->server.ldap_server && 
-           ads_try_connect(ads, ads->server.ldap_server, LDAP_PORT)) {
+           ads_try_connect(ads, ads->server.ldap_server)) {
                goto got_connection;
        }
 
-       if (ads_find_dc(ads)) {
+       ntstatus = ads_find_dc(ads);
+       if (NT_STATUS_IS_OK(ntstatus)) {
                goto got_connection;
        }
 
-       return ADS_ERROR_SYSTEM(errno?errno:ENOENT);
+       return ADS_ERROR_NT(ntstatus);
 
 got_connection:
        DEBUG(3,("Connected to LDAP server %s\n", inet_ntoa(ads->ldap_ip)));
 
-       status = ads_server_info(ads);
-       if (!ADS_ERR_OK(status)) {
-               DEBUG(1,("Failed to get ldap server info\n"));
-               return status;
-       }
-
-       ldap_set_option(ads->ld, LDAP_OPT_PROTOCOL_VERSION, &version);
-
        if (!ads->auth.user_name) {
-               /* by default use the machine account */
-               fstring myname;
-               fstrcpy(myname, global_myname());
-               strlower_m(myname);
-               asprintf(&ads->auth.user_name, "HOST/%s", myname);
+               /* Must use the userPrincipalName value here or sAMAccountName
+                  and not servicePrincipalName; found by Guenther Deschner */
+
+               asprintf(&ads->auth.user_name, "%s$", global_myname() );
        }
 
        if (!ads->auth.realm) {
-               ads->auth.realm = strdup(ads->config.realm);
+               ads->auth.realm = SMB_STRDUP(ads->config.realm);
        }
 
        if (!ads->auth.kdc_server) {
-               ads->auth.kdc_server = strdup(inet_ntoa(ads->ldap_ip));
+               ads->auth.kdc_server = SMB_STRDUP(inet_ntoa(ads->ldap_ip));
        }
 
 #if KRB5_DNS_HACK
@@ -252,10 +322,39 @@ got_connection:
        }
 #endif
 
+       /* If the caller() requested no LDAP bind, then we are done */
+       
        if (ads->auth.flags & ADS_AUTH_NO_BIND) {
                return ADS_SUCCESS;
        }
+       
+       /* Otherwise setup the TCP LDAP session */
+
+       if ( (ads->ld = ldap_open_with_timeout(ads->config.ldap_server_name, 
+               LDAP_PORT, lp_ldap_timeout())) == NULL )
+       {
+               return ADS_ERROR(LDAP_OPERATIONS_ERROR);
+       }
+
+       /* cache the successful connection */
+       saf_store( ads->server.workgroup, inet_ntoa(ads->ldap_ip));
+
+       ldap_set_option(ads->ld, LDAP_OPT_PROTOCOL_VERSION, &version);
+
+       status = ADS_ERROR(smb_ldap_start_tls(ads->ld, version));
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       /* fill in the current time and offsets */
+       
+       status = ads_current_time( ads );
+       if ( !ADS_ERR_OK(status) ) {
+               return status;
+       }
 
+       /* Now do the bind */
+       
        if (ads->auth.flags & ADS_AUTH_ANON_BIND) {
                return ADS_ERROR(ldap_simple_bind_s( ads->ld, NULL, NULL));
        }
@@ -276,13 +375,13 @@ static struct berval *dup_berval(TALLOC_CTX *ctx, const struct berval *in_val)
 
        if (!in_val) return NULL;
 
-       value = talloc_zero(ctx, sizeof(struct berval));
+       value = TALLOC_ZERO_P(ctx, struct berval);
        if (value == NULL)
                return NULL;
        if (in_val->bv_len == 0) return value;
 
        value->bv_len = in_val->bv_len;
-       value->bv_val = talloc_memdup(ctx, in_val->bv_val, in_val->bv_len);
+       value->bv_val = TALLOC_MEMDUP(ctx, in_val->bv_val, in_val->bv_len);
        return value;
 }
 
@@ -296,9 +395,9 @@ static struct berval **ads_dup_values(TALLOC_CTX *ctx,
        int i;
        
        if (!in_vals) return NULL;
-       for (i=0; in_vals[i]; i++); /* count values */
-       values = (struct berval **) talloc_zero(ctx, 
-                                               (i+1)*sizeof(struct berval *));
+       for (i=0; in_vals[i]; i++)
+               ; /* count values */
+       values = TALLOC_ZERO_ARRAY(ctx, struct berval *, i+1);
        if (!values) return NULL;
 
        for (i=0; in_vals[i]; i++) {
@@ -316,8 +415,9 @@ static char **ads_push_strvals(TALLOC_CTX *ctx, const char **in_vals)
        int i;
        
        if (!in_vals) return NULL;
-       for (i=0; in_vals[i]; i++); /* count values */
-       values = (char ** ) talloc_zero(ctx, (i+1)*sizeof(char *));
+       for (i=0; in_vals[i]; i++)
+               ; /* count values */
+       values = TALLOC_ZERO_ARRAY(ctx, char *, i+1);
        if (!values) return NULL;
 
        for (i=0; in_vals[i]; i++) {
@@ -335,8 +435,9 @@ static char **ads_pull_strvals(TALLOC_CTX *ctx, const char **in_vals)
        int i;
        
        if (!in_vals) return NULL;
-       for (i=0; in_vals[i]; i++); /* count values */
-       values = (char **) talloc_zero(ctx, (i+1)*sizeof(char *));
+       for (i=0; in_vals[i]; i++)
+               ; /* count values */
+       values = TALLOC_ZERO_ARRAY(ctx, char *, i+1);
        if (!values) return NULL;
 
        for (i=0; in_vals[i]; i++) {
@@ -359,21 +460,25 @@ static char **ads_pull_strvals(TALLOC_CTX *ctx, const char **in_vals)
  * @param cookie The paged results cookie to be returned on subsequent calls
  * @return status of search
  **/
-ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
-                              int scope, const char *expr,
-                              const char **attrs, void **res, 
-                              int *count, void **cookie)
+ADS_STATUS ads_do_paged_search_args(ADS_STRUCT *ads, const char *bind_path,
+                                   int scope, const char *expr,
+                                   const char **attrs, void *args, void **res, 
+                                   int *count, void **cookie)
 {
        int rc, i, version;
        char *utf8_expr, *utf8_path, **search_attrs;
-       LDAPControl PagedResults, NoReferrals, *controls[3], **rcontrols; 
+       LDAPControl PagedResults, NoReferrals, ExtendedDn, *controls[4], **rcontrols;
        BerElement *cookie_be = NULL;
        struct berval *cookie_bv= NULL;
+       BerElement *extdn_be = NULL;
+       struct berval *extdn_bv= NULL;
+
        TALLOC_CTX *ctx;
+       ads_control *external_control = (ads_control *) args;
 
        *res = NULL;
 
-       if (!(ctx = talloc_init("ads_do_paged_search")))
+       if (!(ctx = talloc_init("ads_do_paged_search_args")))
                return ADS_ERROR(LDAP_NO_MEMORY);
 
        /* 0 means the conversion worked but the result was empty 
@@ -405,7 +510,7 @@ ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
        }
 
        cookie_be = ber_alloc_t(LBER_USE_DER);
-       if (cookie && *cookie) {
+       if (*cookie) {
                ber_printf(cookie_be, "{iO}", (ber_int_t) 1000, *cookie);
                ber_bvfree(*cookie); /* don't need it from last time */
                *cookie = NULL;
@@ -413,22 +518,57 @@ ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
                ber_printf(cookie_be, "{io}", (ber_int_t) 1000, "", 0);
        }
        ber_flatten(cookie_be, &cookie_bv);
-       PagedResults.ldctl_oid = ADS_PAGE_CTL_OID;
+       PagedResults.ldctl_oid = CONST_DISCARD(char *, ADS_PAGE_CTL_OID);
        PagedResults.ldctl_iscritical = (char) 1;
        PagedResults.ldctl_value.bv_len = cookie_bv->bv_len;
        PagedResults.ldctl_value.bv_val = cookie_bv->bv_val;
 
-       NoReferrals.ldctl_oid = ADS_NO_REFERRALS_OID;
+       NoReferrals.ldctl_oid = CONST_DISCARD(char *, ADS_NO_REFERRALS_OID);
        NoReferrals.ldctl_iscritical = (char) 0;
        NoReferrals.ldctl_value.bv_len = 0;
-       NoReferrals.ldctl_value.bv_val = "";
+       NoReferrals.ldctl_value.bv_val = CONST_DISCARD(char *, "");
 
+       if (external_control && strequal(external_control->control, ADS_EXTENDED_DN_OID)) {
 
-       controls[0] = &NoReferrals;
-       controls[1] = &PagedResults;
-       controls[2] = NULL;
+               ExtendedDn.ldctl_oid = CONST_DISCARD(char *, external_control->control);
+               ExtendedDn.ldctl_iscritical = (char) external_control->critical;
 
-       *res = NULL;
+               /* win2k does not accept a ldctl_value beeing passed in */
+
+               if (external_control->val != 0) {
+
+                       if ((extdn_be = ber_alloc_t(LBER_USE_DER)) == NULL ) {
+                               rc = LDAP_NO_MEMORY;
+                               goto done;
+                       }
+
+                       if ((ber_printf(extdn_be, "{i}", (ber_int_t) external_control->val)) == -1) {
+                               rc = LDAP_NO_MEMORY;
+                               goto done;
+                       }
+                       if ((ber_flatten(extdn_be, &extdn_bv)) == -1) {
+                               rc = LDAP_NO_MEMORY;
+                               goto done;
+                       }
+
+                       ExtendedDn.ldctl_value.bv_len = extdn_bv->bv_len;
+                       ExtendedDn.ldctl_value.bv_val = extdn_bv->bv_val;
+
+               } else {
+                       ExtendedDn.ldctl_value.bv_len = 0;
+                       ExtendedDn.ldctl_value.bv_val = NULL;
+               }
+
+               controls[0] = &NoReferrals;
+               controls[1] = &PagedResults;
+               controls[2] = &ExtendedDn;
+               controls[3] = NULL;
+
+       } else {
+               controls[0] = &NoReferrals;
+               controls[1] = &PagedResults;
+               controls[2] = NULL;
+       }
 
        /* we need to disable referrals as the openldap libs don't
           handle them and paged results at the same time.  Using them
@@ -440,15 +580,17 @@ ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
        */
        ldap_set_option(ads->ld, LDAP_OPT_REFERRALS, LDAP_OPT_OFF);
 
-       rc = ldap_search_ext_s(ads->ld, utf8_path, scope, utf8_expr, 
-                              search_attrs, 0, controls,
-                              NULL, NULL, LDAP_NO_LIMIT, (LDAPMessage **)res);
+       rc = ldap_search_with_timeout(ads->ld, utf8_path, scope, utf8_expr, 
+                                     search_attrs, 0, controls,
+                                     NULL, LDAP_NO_LIMIT,
+                                     (LDAPMessage **)res);
 
        ber_free(cookie_be, 1);
        ber_bvfree(cookie_bv);
 
        if (rc) {
-               DEBUG(3,("ldap_search_ext_s(%s) -> %s\n", expr, ldap_err2string(rc)));
+               DEBUG(3,("ads_do_paged_search_args: ldap_search_with_timeout(%s) -> %s\n", expr,
+                        ldap_err2string(rc)));
                goto done;
        }
 
@@ -479,12 +621,29 @@ ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
 
 done:
        talloc_destroy(ctx);
+
+       if (extdn_be) {
+               ber_free(extdn_be, 1);
+       }
+
+       if (extdn_bv) {
+               ber_bvfree(extdn_bv);
+       }
        /* if/when we decide to utf8-encode attrs, take out this next line */
        str_list_free(&search_attrs);
 
        return ADS_ERROR(rc);
 }
 
+ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
+                              int scope, const char *expr,
+                              const char **attrs, void **res, 
+                              int *count, void **cookie)
+{
+       return ads_do_paged_search_args(ads, bind_path, scope, expr, attrs, NULL, res, count, cookie);
+}
+
 
 /**
  * Get all results for a search.  This uses ads_do_paged_search() to return 
@@ -497,26 +656,29 @@ done:
  * @param res ** which will contain results - free res* with ads_msgfree()
  * @return status of search
  **/
-ADS_STATUS ads_do_search_all(ADS_STRUCT *ads, const char *bind_path,
-                            int scope, const char *expr,
-                            const char **attrs, void **res)
+ADS_STATUS ads_do_search_all_args(ADS_STRUCT *ads, const char *bind_path,
+                                 int scope, const char *expr,
+                                 const char **attrs, void *args, void **res)
 {
        void *cookie = NULL;
        int count = 0;
        ADS_STATUS status;
 
-       status = ads_do_paged_search(ads, bind_path, scope, expr, attrs, res,
+       *res = NULL;
+       status = ads_do_paged_search_args(ads, bind_path, scope, expr, attrs, args, res,
                                     &count, &cookie);
 
-       if (!ADS_ERR_OK(status)) return status;
+       if (!ADS_ERR_OK(status)) 
+               return status;
 
+#ifdef HAVE_LDAP_ADD_RESULT_ENTRY
        while (cookie) {
                void *res2 = NULL;
                ADS_STATUS status2;
                LDAPMessage *msg, *next;
 
-               status2 = ads_do_paged_search(ads, bind_path, scope, expr, 
-                                             attrs, &res2, &count, &cookie);
+               status2 = ads_do_paged_search_args(ads, bind_path, scope, expr, 
+                                             attrs, args, &res2, &count, &cookie);
 
                if (!ADS_ERR_OK(status2)) break;
 
@@ -529,10 +691,21 @@ ADS_STATUS ads_do_search_all(ADS_STRUCT *ads, const char *bind_path,
                /* note that we do not free res2, as the memory is now
                    part of the main returned list */
        }
+#else
+       DEBUG(0, ("no ldap_add_result_entry() support in LDAP libs!\n"));
+       status = ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
+#endif
 
        return status;
 }
 
+ADS_STATUS ads_do_search_all(ADS_STRUCT *ads, const char *bind_path,
+                            int scope, const char *expr,
+                            const char **attrs, void **res)
+{
+       return ads_do_search_all_args(ads, bind_path, scope, expr, attrs, NULL, res);
+}
+
 /**
  * Run a function on all results for a search.  Uses ads_do_paged_search() and
  *  runs the function as each page is returned, using ads_process_results()
@@ -590,11 +763,11 @@ ADS_STATUS ads_do_search(ADS_STRUCT *ads, const char *bind_path, int scope,
                         const char *expr,
                         const char **attrs, void **res)
 {
-       struct timeval timeout;
        int rc;
        char *utf8_expr, *utf8_path, **search_attrs = NULL;
        TALLOC_CTX *ctx;
 
+       *res = NULL;
        if (!(ctx = talloc_init("ads_do_search"))) {
                DEBUG(1,("ads_do_search: talloc_init() failed!"));
                return ADS_ERROR(LDAP_NO_MEMORY);
@@ -623,16 +796,13 @@ ADS_STATUS ads_do_search(ADS_STRUCT *ads, const char *bind_path, int scope,
                }
        }
 
-       timeout.tv_sec = ADS_SEARCH_TIMEOUT;
-       timeout.tv_usec = 0;
-       *res = NULL;
-
        /* see the note in ads_do_paged_search - we *must* disable referrals */
        ldap_set_option(ads->ld, LDAP_OPT_REFERRALS, LDAP_OPT_OFF);
 
-       rc = ldap_search_ext_s(ads->ld, utf8_path, scope, utf8_expr,
-                              search_attrs, 0, NULL, NULL, 
-                              &timeout, LDAP_NO_LIMIT, (LDAPMessage **)res);
+       rc = ldap_search_with_timeout(ads->ld, utf8_path, scope, utf8_expr,
+                                     search_attrs, 0, NULL, NULL, 
+                                     LDAP_NO_LIMIT,
+                                     (LDAPMessage **)res);
 
        if (rc == LDAP_SIZELIMIT_EXCEEDED) {
                DEBUG(3,("Warning! sizelimit exceeded in ldap. Truncating.\n"));
@@ -669,10 +839,11 @@ ADS_STATUS ads_search(ADS_STRUCT *ads, void **res,
  * @param attrs Attributes to retrieve
  * @return status of search
  **/
-ADS_STATUS ads_search_dn(ADS_STRUCT *ads, void **res, 
+ADS_STATUS ads_search_dn(ADS_STRUCT *ads, void *_res, 
                         const char *dn, 
                         const char **attrs)
 {
+       void **res = (void **)_res;
        return ads_do_search(ads, dn, LDAP_SCOPE_BASE, "(objectclass=*)", attrs, res);
 }
 
@@ -714,7 +885,7 @@ char *ads_get_dn(ADS_STRUCT *ads, void *msg)
                return NULL;
        }
 
-       if (pull_utf8_allocate((void **) &unix_dn, utf8_dn) == (size_t)-1) {
+       if (pull_utf8_allocate(&unix_dn, utf8_dn) == (size_t)-1) {
                DEBUG(0,("ads_get_dn: string conversion failure utf8 [%s]\n",
                        utf8_dn ));
                return NULL;
@@ -723,6 +894,43 @@ char *ads_get_dn(ADS_STRUCT *ads, void *msg)
        return unix_dn;
 }
 
+/**
+ * Get a canonical dn from search results
+ * @param ads connection to ads server
+ * @param msg Search result
+ * @return dn string
+ **/
+char *ads_get_dn_canonical(ADS_STRUCT *ads, void *msg)
+{
+#ifdef HAVE_LDAP_DN2AD_CANONICAL
+       return ldap_dn2ad_canonical(ads_get_dn(ads, msg));
+#else
+       return NULL;
+#endif
+}
+
+/**
+ * Get the parent from a dn
+ * @param dn the dn to return the parent from
+ * @return parent dn string
+ **/
+char *ads_parent_dn(const char *dn)
+{
+       char *p;
+
+       if (dn == NULL) {
+               return NULL;
+       }
+
+       p = strchr(dn, ',');
+
+       if (p == NULL) {
+               return NULL;
+       }
+
+       return p+1;
+}
+
 /**
  * Find a machine account given a hostname
  * @param ads connection to ads server
@@ -730,21 +938,23 @@ char *ads_get_dn(ADS_STRUCT *ads, void *msg)
  * @param host Hostname to search for
  * @return status of search
  **/
-ADS_STATUS ads_find_machine_acct(ADS_STRUCT *ads, void **res, const char *host)
+ADS_STATUS ads_find_machine_acct(ADS_STRUCT *ads, void **res, const char *machine)
 {
        ADS_STATUS status;
        char *expr;
        const char *attrs[] = {"*", "nTSecurityDescriptor", NULL};
 
+       *res = NULL;
+
        /* the easiest way to find a machine account anywhere in the tree
           is to look for hostname$ */
-       if (asprintf(&expr, "(samAccountName=%s$)", host) == -1) {
+       if (asprintf(&expr, "(samAccountName=%s$)", machine) == -1) {
                DEBUG(1, ("asprintf failed!\n"));
                return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
        }
        
        status = ads_search(ads, res, expr, attrs);
-       free(expr);
+       SAFE_FREE(expr);
        return status;
 }
 
@@ -758,13 +968,12 @@ ADS_MODLIST ads_init_mods(TALLOC_CTX *ctx)
 #define ADS_MODLIST_ALLOC_SIZE 10
        LDAPMod **mods;
        
-       if ((mods = (LDAPMod **) talloc_zero(ctx, sizeof(LDAPMod *) * 
-                                            (ADS_MODLIST_ALLOC_SIZE + 1))))
+       if ((mods = TALLOC_ZERO_ARRAY(ctx, LDAPMod *, ADS_MODLIST_ALLOC_SIZE + 1)))
                /* -1 is safety to make sure we don't go over the end.
                   need to reset it to NULL before doing ldap modify */
                mods[ADS_MODLIST_ALLOC_SIZE] = (LDAPMod *) -1;
        
-       return mods;
+       return (ADS_MODLIST)mods;
 }
 
 
@@ -773,8 +982,9 @@ ADS_MODLIST ads_init_mods(TALLOC_CTX *ctx)
 */
 static ADS_STATUS ads_modlist_add(TALLOC_CTX *ctx, ADS_MODLIST *mods, 
                                  int mod_op, const char *name, 
-                                 const void **invals)
+                                 const void *_invals)
 {
+       const void **invals = (const void **)_invals;
        int curmod;
        LDAPMod **modlist = (LDAPMod **) *mods;
        struct berval **ber_values = NULL;
@@ -795,16 +1005,16 @@ static ADS_STATUS ads_modlist_add(TALLOC_CTX *ctx, ADS_MODLIST *mods,
        for (curmod=0; modlist[curmod] && modlist[curmod] != (LDAPMod *) -1;
             curmod++);
        if (modlist[curmod] == (LDAPMod *) -1) {
-               if (!(modlist = talloc_realloc(ctx, modlist, 
-                       (curmod+ADS_MODLIST_ALLOC_SIZE+1)*sizeof(LDAPMod *))))
+               if (!(modlist = TALLOC_REALLOC_ARRAY(ctx, modlist, LDAPMod *,
+                               curmod+ADS_MODLIST_ALLOC_SIZE+1)))
                        return ADS_ERROR(LDAP_NO_MEMORY);
                memset(&modlist[curmod], 0, 
                       ADS_MODLIST_ALLOC_SIZE*sizeof(LDAPMod *));
                modlist[curmod+ADS_MODLIST_ALLOC_SIZE] = (LDAPMod *) -1;
-               *mods = modlist;
+               *mods = (ADS_MODLIST)modlist;
        }
                
-       if (!(modlist[curmod] = talloc_zero(ctx, sizeof(LDAPMod))))
+       if (!(modlist[curmod] = TALLOC_ZERO_P(ctx, LDAPMod)))
                return ADS_ERROR(LDAP_NO_MEMORY);
        modlist[curmod]->mod_type = talloc_strdup(ctx, name);
        if (mod_op & LDAP_MOD_BVALUES) {
@@ -837,8 +1047,7 @@ ADS_STATUS ads_mod_str(TALLOC_CTX *ctx, ADS_MODLIST *mods,
 
        if (!val)
                return ads_modlist_add(ctx, mods, LDAP_MOD_DELETE, name, NULL);
-       return ads_modlist_add(ctx, mods, LDAP_MOD_REPLACE, name, 
-                              (const void **) values);
+       return ads_modlist_add(ctx, mods, LDAP_MOD_REPLACE, name, values);
 }
 
 /**
@@ -858,6 +1067,7 @@ ADS_STATUS ads_mod_strlist(TALLOC_CTX *ctx, ADS_MODLIST *mods,
                               name, (const void **) vals);
 }
 
+#if 0
 /**
  * Add a single ber-encoded value to a mod list
  * @param ctx An initialized TALLOC_CTX
@@ -878,6 +1088,7 @@ static ADS_STATUS ads_mod_ber(TALLOC_CTX *ctx, ADS_MODLIST *mods,
        return ads_modlist_add(ctx, mods, LDAP_MOD_REPLACE|LDAP_MOD_BVALUES,
                               name, (const void **) values);
 }
+#endif
 
 /**
  * Perform an ldap modify
@@ -895,7 +1106,7 @@ ADS_STATUS ads_gen_mod(ADS_STRUCT *ads, const char *mod_dn, ADS_MODLIST mods)
           non-existent attribute (but allowable for the object) to run
        */
        LDAPControl PermitModify = {
-               ADS_PERMIT_MODIFY_OID,
+                CONST_DISCARD(char *, ADS_PERMIT_MODIFY_OID),
                {0, NULL},
                (char) 1};
        LDAPControl *controls[2];
@@ -939,7 +1150,7 @@ ADS_STATUS ads_gen_add(ADS_STRUCT *ads, const char *new_dn, ADS_MODLIST mods)
        /* make sure the end of the list is NULL */
        mods[i] = NULL;
 
-       ret = ldap_add_s(ads->ld, utf8_dn, mods);
+       ret = ldap_add_s(ads->ld, utf8_dn, (LDAPMod**)mods);
        SAFE_FREE(utf8_dn);
        return ADS_ERROR(ret);
 }
@@ -966,178 +1177,432 @@ ADS_STATUS ads_del_dn(ADS_STRUCT *ads, char *del_dn)
 /**
  * Build an org unit string
  *  if org unit is Computers or blank then assume a container, otherwise
- *  assume a \ separated list of organisational units
+ *  assume a / separated list of organisational units.
+ * jmcd: '\' is now used for escapes so certain chars can be in the ou (e.g. #)
+ * @param ads connection to ads server
  * @param org_unit Organizational unit
  * @return org unit string - caller must free
  **/
-char *ads_ou_string(const char *org_unit)
-{      
-       if (!org_unit || !*org_unit || strequal(org_unit, "Computers")) {
-               return strdup("cn=Computers");
-       }
+char *ads_ou_string(ADS_STRUCT *ads, const char *org_unit)
+{
+       char *ret = NULL;
 
-       return ads_build_path(org_unit, "\\/", "ou=", 1);
-}
+       if (!org_unit || !*org_unit) {
+
+               ret = ads_default_ou_string(ads, WELL_KNOWN_GUID_COMPUTERS);
 
+               /* samba4 might not yet respond to a wellknownobject-query */
+               return ret ? ret : SMB_STRDUP("cn=Computers");
+       }
+       
+       if (strequal(org_unit, "Computers")) {
+               return SMB_STRDUP("cn=Computers");
+       }
 
+       /* jmcd: removed "\\" from the separation chars, because it is
+          needed as an escape for chars like '#' which are valid in an
+          OU name */
+       return ads_build_path(org_unit, "/", "ou=", 1);
+}
 
-/*
-  add a machine account to the ADS server
-*/
-static ADS_STATUS ads_add_machine_acct(ADS_STRUCT *ads, const char *hostname, 
-                                      uint32 account_type,
-                                      const char *org_unit)
+/**
+ * Get a org unit string for a well-known GUID
+ * @param ads connection to ads server
+ * @param wknguid Well known GUID
+ * @return org unit string - caller must free
+ **/
+char *ads_default_ou_string(ADS_STRUCT *ads, const char *wknguid)
 {
-       ADS_STATUS ret, status;
-       char *host_spn, *host_upn, *new_dn, *samAccountName, *controlstr;
-       char *ou_str;
-       TALLOC_CTX *ctx;
-       ADS_MODLIST mods;
-       const char *objectClass[] = {"top", "person", "organizationalPerson",
-                                    "user", "computer", NULL};
-       const char *servicePrincipalName[5] = {NULL, NULL, NULL, NULL, NULL};
-       char *psp, *psp2;
-       unsigned acct_control;
-       unsigned exists=0;
-       LDAPMessage *res;
+       ADS_STATUS status;
+       void *res;
+       char *base, *wkn_dn, *ret, **wkn_dn_exp, **bind_dn_exp;
+       const char *attrs[] = {"distinguishedName", NULL};
+       int new_ln, wkn_ln, bind_ln, i;
 
-       status = ads_find_machine_acct(ads, (void **)&res, hostname);
-       if (ADS_ERR_OK(status) && ads_count_replies(ads, res) == 1) {
-               DEBUG(0, ("Host account for %s already exists - modifying old account\n", hostname));
-               exists=1;
+       if (wknguid == NULL) {
+               return NULL;
        }
 
-       if (!(ctx = talloc_init("machine_account")))
-               return ADS_ERROR(LDAP_NO_MEMORY);
+       if (asprintf(&base, "<WKGUID=%s,%s>", wknguid, ads->config.bind_path ) == -1) {
+               DEBUG(1, ("asprintf failed!\n"));
+               return NULL;
+       }
 
-       ret = ADS_ERROR(LDAP_NO_MEMORY);
+       status = ads_search_dn(ads, &res, base, attrs);
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(1,("Failed while searching for: %s\n", base));
+               SAFE_FREE(base);
+               return NULL;
+       }
+       SAFE_FREE(base);
 
-       if (!(host_spn = talloc_asprintf(ctx, "HOST/%s", hostname)))
-               goto done;
-       if (!(host_upn = talloc_asprintf(ctx, "%s@%s", host_spn, ads->config.realm)))
-               goto done;
-       ou_str = ads_ou_string(org_unit);
-       if (!ou_str) {
-               DEBUG(1, ("ads_ou_string returned NULL (malloc failure?)\n"));
-               goto done;
+       if (ads_count_replies(ads, res) != 1) {
+               return NULL;
        }
-       new_dn = talloc_asprintf(ctx, "cn=%s,%s,%s", hostname, ou_str, 
-                                ads->config.bind_path);
-       servicePrincipalName[0] = talloc_asprintf(ctx, "HOST/%s", hostname);
-       psp = talloc_asprintf(ctx, "HOST/%s.%s", 
-                             hostname, 
-                             ads->config.realm);
-       strlower_m(&psp[5]);
-       servicePrincipalName[1] = psp;
-       servicePrincipalName[2] = talloc_asprintf(ctx, "CIFS/%s", hostname);
-       psp2 = talloc_asprintf(ctx, "CIFS/%s.%s", 
-                              hostname, 
-                              ads->config.realm);
-       strlower_m(&psp2[5]);
-       servicePrincipalName[3] = psp2;
-
-       free(ou_str);
-       if (!new_dn)
-               goto done;
 
-       if (!(samAccountName = talloc_asprintf(ctx, "%s$", hostname)))
-               goto done;
+       /* substitute the bind-path from the well-known-guid-search result */
+       wkn_dn = ads_get_dn(ads, res);
+       wkn_dn_exp = ldap_explode_dn(wkn_dn, 0);
+       bind_dn_exp = ldap_explode_dn(ads->config.bind_path, 0);
 
-       acct_control = account_type | UF_DONT_EXPIRE_PASSWD;
-#ifndef ENCTYPE_ARCFOUR_HMAC
-       acct_control |= UF_USE_DES_KEY_ONLY;
-#endif
+       for (wkn_ln=0; wkn_dn_exp[wkn_ln]; wkn_ln++)
+               ;
+       for (bind_ln=0; bind_dn_exp[bind_ln]; bind_ln++)
+               ;
 
-       if (!(controlstr = talloc_asprintf(ctx, "%u", acct_control)))
-               goto done;
+       new_ln = wkn_ln - bind_ln;
 
-       if (!(mods = ads_init_mods(ctx)))
-               goto done;
+       ret = wkn_dn_exp[0];
 
-       if (!exists) {
-               ads_mod_str(ctx, &mods, "cn", hostname);
-               ads_mod_str(ctx, &mods, "sAMAccountName", samAccountName);
-               ads_mod_str(ctx, &mods, "userAccountControl", controlstr);
-               ads_mod_strlist(ctx, &mods, "objectClass", objectClass);
+       for (i=1; i < new_ln; i++) {
+               char *s;
+               asprintf(&s, "%s,%s", ret, wkn_dn_exp[i]);
+               ret = SMB_STRDUP(s);
+               free(s);
        }
-       ads_mod_str(ctx, &mods, "dNSHostName", hostname);
-       ads_mod_str(ctx, &mods, "userPrincipalName", host_upn);
-       ads_mod_strlist(ctx, &mods, "servicePrincipalName", servicePrincipalName);
-       ads_mod_str(ctx, &mods, "operatingSystem", "Samba");
-       ads_mod_str(ctx, &mods, "operatingSystemVersion", SAMBA_VERSION_STRING);
-
-       if (!exists) 
-               ret = ads_gen_add(ads, new_dn, mods);
-       else
-               ret = ads_gen_mod(ads, new_dn, mods);
 
-       if (!ADS_ERR_OK(ret))
-               goto done;
+       ads_memfree(ads, wkn_dn);
+       ldap_value_free(wkn_dn_exp);
+       ldap_value_free(bind_dn_exp);
 
-       /* Do not fail if we can't set security descriptor
-        * it shouldn't be mandatory and probably we just 
-        * don't have enough rights to do it.
-        */
-       if (!exists) {
-               status = ads_set_machine_sd(ads, hostname, new_dn);
-       
-               if (!ADS_ERR_OK(status)) {
-                       DEBUG(0, ("Warning: ads_set_machine_sd: %s\n",
-                                       ads_errstr(status)));
-               }
-       }
-done:
-       talloc_destroy(ctx);
        return ret;
 }
 
-/*
-  dump a binary result from ldap
-*/
-static void dump_binary(const char *field, struct berval **values)
+/**
+ * Adds (appends) an item to an attribute array, rather then
+ * replacing the whole list
+ * @param ctx An initialized TALLOC_CTX
+ * @param mods An initialized ADS_MODLIST
+ * @param name name of the ldap attribute to append to
+ * @param vals an array of values to add
+ * @return status of addition
+ **/
+
+ADS_STATUS ads_add_strlist(TALLOC_CTX *ctx, ADS_MODLIST *mods,
+                               const char *name, const char **vals)
 {
-       int i, j;
-       for (i=0; values[i]; i++) {
-               printf("%s: ", field);
-               for (j=0; j<values[i]->bv_len; j++) {
-                       printf("%02X", (unsigned char)values[i]->bv_val[j]);
-               }
-               printf("\n");
-       }
+       return ads_modlist_add(ctx, mods, LDAP_MOD_ADD, name, (const void **) vals);
 }
 
-struct uuid {
-        uint32   i1;
-        uint16   i2;
-        uint16   i3;
-        uint8    s[8];
-};
+/**
+ * Determines the computer account's current KVNO via an LDAP lookup
+ * @param ads An initialized ADS_STRUCT
+ * @param machine_name the NetBIOS name of the computer, which is used to identify the computer account.
+ * @return the kvno for the computer account, or -1 in case of a failure.
+ **/
 
-static void dump_guid(const char *field, struct berval **values)
+uint32 ads_get_kvno(ADS_STRUCT *ads, const char *machine_name)
 {
-       int i;
-       GUID guid;
-       for (i=0; values[i]; i++) {
-               memcpy(guid.info, values[i]->bv_val, sizeof(guid.info));
-               printf("%s: %s\n", field, smb_uuid_string_static(guid));
+       LDAPMessage *res = NULL;
+       uint32 kvno = (uint32)-1;      /* -1 indicates a failure */
+       char *filter;
+       const char *attrs[] = {"msDS-KeyVersionNumber", NULL};
+       char *dn_string = NULL;
+       ADS_STATUS ret = ADS_ERROR(LDAP_SUCCESS);
+
+       DEBUG(5,("ads_get_kvno: Searching for host %s\n", machine_name));
+       if (asprintf(&filter, "(samAccountName=%s$)", machine_name) == -1) {
+               return kvno;
+       }
+       ret = ads_search(ads, (void**)(void *)&res, filter, attrs);
+       SAFE_FREE(filter);
+       if (!ADS_ERR_OK(ret) && ads_count_replies(ads, res)) {
+               DEBUG(1,("ads_get_kvno: Computer Account For %s not found.\n", machine_name));
+               ads_msgfree(ads, res);
+               return kvno;
        }
-}
 
-/*
-  dump a sid result from ldap
-*/
-static void dump_sid(const char *field, struct berval **values)
-{
-       int i;
-       for (i=0; values[i]; i++) {
-               DOM_SID sid;
-               sid_parse(values[i]->bv_val, values[i]->bv_len, &sid);
-               printf("%s: %s\n", field, sid_string_static(&sid));
+       dn_string = ads_get_dn(ads, res);
+       if (!dn_string) {
+               DEBUG(0,("ads_get_kvno: out of memory.\n"));
+               ads_msgfree(ads, res);
+               return kvno;
+       }
+       DEBUG(5,("ads_get_kvno: Using: %s\n", dn_string));
+       ads_memfree(ads, dn_string);
+
+       /* ---------------------------------------------------------
+        * 0 is returned as a default KVNO from this point on...
+        * This is done because Windows 2000 does not support key
+        * version numbers.  Chances are that a failure in the next
+        * step is simply due to Windows 2000 being used for a
+        * domain controller. */
+       kvno = 0;
+
+       if (!ads_pull_uint32(ads, res, "msDS-KeyVersionNumber", &kvno)) {
+               DEBUG(3,("ads_get_kvno: Error Determining KVNO!\n"));
+               DEBUG(3,("ads_get_kvno: Windows 2000 does not support KVNO's, so this may be normal.\n"));
+               ads_msgfree(ads, res);
+               return kvno;
        }
+
+       /* Success */
+       DEBUG(5,("ads_get_kvno: Looked Up KVNO of: %d\n", kvno));
+       ads_msgfree(ads, res);
+       return kvno;
 }
 
-/*
-  dump ntSecurityDescriptor
+/**
+ * This clears out all registered spn's for a given hostname
+ * @param ads An initilaized ADS_STRUCT
+ * @param machine_name the NetBIOS name of the computer.
+ * @return 0 upon success, non-zero otherwise.
+ **/
+
+ADS_STATUS ads_clear_service_principal_names(ADS_STRUCT *ads, const char *machine_name)
+{
+       TALLOC_CTX *ctx;
+       LDAPMessage *res = NULL;
+       ADS_MODLIST mods;
+       const char *servicePrincipalName[1] = {NULL};
+       ADS_STATUS ret = ADS_ERROR(LDAP_SUCCESS);
+       char *dn_string = NULL;
+
+       ret = ads_find_machine_acct(ads, (void **)(void *)&res, machine_name);
+       if (!ADS_ERR_OK(ret) || ads_count_replies(ads, res) != 1) {
+               DEBUG(5,("ads_clear_service_principal_names: WARNING: Host Account for %s not found... skipping operation.\n", machine_name));
+               DEBUG(5,("ads_clear_service_principal_names: WARNING: Service Principals for %s have NOT been cleared.\n", machine_name));
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
+
+       DEBUG(5,("ads_clear_service_principal_names: Host account for %s found\n", machine_name));
+       ctx = talloc_init("ads_clear_service_principal_names");
+       if (!ctx) {
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       if (!(mods = ads_init_mods(ctx))) {
+               talloc_destroy(ctx);
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+       ret = ads_mod_strlist(ctx, &mods, "servicePrincipalName", servicePrincipalName);
+       if (!ADS_ERR_OK(ret)) {
+               DEBUG(1,("ads_clear_service_principal_names: Error creating strlist.\n"));
+               ads_msgfree(ads, res);
+               talloc_destroy(ctx);
+               return ret;
+       }
+       dn_string = ads_get_dn(ads, res);
+       if (!dn_string) {
+               talloc_destroy(ctx);
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+       ret = ads_gen_mod(ads, dn_string, mods);
+       ads_memfree(ads,dn_string);
+       if (!ADS_ERR_OK(ret)) {
+               DEBUG(1,("ads_clear_service_principal_names: Error: Updating Service Principals for machine %s in LDAP\n",
+                       machine_name));
+               ads_msgfree(ads, res);
+               talloc_destroy(ctx);
+               return ret;
+       }
+
+       ads_msgfree(ads, res);
+       talloc_destroy(ctx);
+       return ret;
+}
+
+/**
+ * This adds a service principal name to an existing computer account
+ * (found by hostname) in AD.
+ * @param ads An initialized ADS_STRUCT
+ * @param machine_name the NetBIOS name of the computer, which is used to identify the computer account.
+ * @param my_fqdn The fully qualified DNS name of the machine
+ * @param spn A string of the service principal to add, i.e. 'host'
+ * @return 0 upon sucess, or non-zero if a failure occurs
+ **/
+
+ADS_STATUS ads_add_service_principal_name(ADS_STRUCT *ads, const char *machine_name, 
+                                          const char *my_fqdn, const char *spn)
+{
+       ADS_STATUS ret;
+       TALLOC_CTX *ctx;
+       LDAPMessage *res = NULL;
+       char *psp1, *psp2;
+       ADS_MODLIST mods;
+       char *dn_string = NULL;
+       const char *servicePrincipalName[3] = {NULL, NULL, NULL};
+
+       ret = ads_find_machine_acct(ads, (void **)(void *)&res, machine_name);
+       if (!ADS_ERR_OK(ret) || ads_count_replies(ads, res) != 1) {
+               DEBUG(1,("ads_add_service_principal_name: WARNING: Host Account for %s not found... skipping operation.\n",
+                       machine_name));
+               DEBUG(1,("ads_add_service_principal_name: WARNING: Service Principal '%s/%s@%s' has NOT been added.\n",
+                       spn, machine_name, ads->config.realm));
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
+
+       DEBUG(1,("ads_add_service_principal_name: Host account for %s found\n", machine_name));
+       if (!(ctx = talloc_init("ads_add_service_principal_name"))) {
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       /* add short name spn */
+       
+       if ( (psp1 = talloc_asprintf(ctx, "%s/%s", spn, machine_name)) == NULL ) {
+               talloc_destroy(ctx);
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+       strupper_m(psp1);
+       strlower_m(&psp1[strlen(spn)]);
+       servicePrincipalName[0] = psp1;
+       
+       DEBUG(5,("ads_add_service_principal_name: INFO: Adding %s to host %s\n", 
+               psp1, machine_name));
+
+
+       /* add fully qualified spn */
+       
+       if ( (psp2 = talloc_asprintf(ctx, "%s/%s", spn, my_fqdn)) == NULL ) {
+               ret = ADS_ERROR(LDAP_NO_MEMORY);
+               goto out;
+       }
+       strupper_m(psp2);
+       strlower_m(&psp2[strlen(spn)]);
+       servicePrincipalName[1] = psp2;
+
+       DEBUG(5,("ads_add_service_principal_name: INFO: Adding %s to host %s\n", 
+               psp2, machine_name));
+
+       if ( (mods = ads_init_mods(ctx)) == NULL ) {
+               ret = ADS_ERROR(LDAP_NO_MEMORY);
+               goto out;
+       }
+       
+       ret = ads_add_strlist(ctx, &mods, "servicePrincipalName", servicePrincipalName);
+       if (!ADS_ERR_OK(ret)) {
+               DEBUG(1,("ads_add_service_principal_name: Error: Updating Service Principals in LDAP\n"));
+               goto out;
+       }
+       
+       if ( (dn_string = ads_get_dn(ads, res)) == NULL ) {
+               ret = ADS_ERROR(LDAP_NO_MEMORY);
+               goto out;
+       }
+       
+       ret = ads_gen_mod(ads, dn_string, mods);
+       ads_memfree(ads,dn_string);
+       if (!ADS_ERR_OK(ret)) {
+               DEBUG(1,("ads_add_service_principal_name: Error: Updating Service Principals in LDAP\n"));
+               goto out;
+       }
+
+ out:
+       TALLOC_FREE( ctx );
+       ads_msgfree(ads, res);
+       return ret;
+}
+
+/**
+ * adds a machine account to the ADS server
+ * @param ads An intialized ADS_STRUCT
+ * @param machine_name - the NetBIOS machine name of this account.
+ * @param account_type A number indicating the type of account to create
+ * @param org_unit The LDAP path in which to place this account
+ * @return 0 upon success, or non-zero otherwise
+**/
+
+ADS_STATUS ads_create_machine_acct(ADS_STRUCT *ads, const char *machine_name, 
+                                   const char *org_unit)
+{
+       ADS_STATUS ret;
+       char *samAccountName, *controlstr;
+       TALLOC_CTX *ctx;
+       ADS_MODLIST mods;
+       char *new_dn;
+       const char *objectClass[] = {"top", "person", "organizationalPerson",
+                                    "user", "computer", NULL};
+       LDAPMessage *res = NULL;
+       uint32 acct_control = ( UF_WORKSTATION_TRUST_ACCOUNT |\
+                               UF_DONT_EXPIRE_PASSWD |\
+                               UF_ACCOUNTDISABLE );
+                             
+       if (!(ctx = talloc_init("ads_add_machine_acct")))
+               return ADS_ERROR(LDAP_NO_MEMORY);
+
+       ret = ADS_ERROR(LDAP_NO_MEMORY);
+               
+       new_dn = talloc_asprintf(ctx, "cn=%s,%s", machine_name, org_unit);
+       samAccountName = talloc_asprintf(ctx, "%s$", machine_name);
+
+       if ( !new_dn || !samAccountName ) {
+               goto done;
+       }
+       
+#ifndef ENCTYPE_ARCFOUR_HMAC
+       acct_control |= UF_USE_DES_KEY_ONLY;
+#endif
+
+       if (!(controlstr = talloc_asprintf(ctx, "%u", acct_control))) {
+               goto done;
+       }
+
+       if (!(mods = ads_init_mods(ctx))) {
+               goto done;
+       }
+       
+       ads_mod_str(ctx, &mods, "cn", machine_name);
+       ads_mod_str(ctx, &mods, "sAMAccountName", samAccountName);
+       ads_mod_strlist(ctx, &mods, "objectClass", objectClass);
+       ads_mod_str(ctx, &mods, "userAccountControl", controlstr);
+
+       ret = ads_gen_add(ads, new_dn, mods);
+
+done:
+       ads_msgfree(ads, res);
+       talloc_destroy(ctx);
+       
+       return ret;
+}
+
+/*
+  dump a binary result from ldap
+*/
+static void dump_binary(const char *field, struct berval **values)
+{
+       int i, j;
+       for (i=0; values[i]; i++) {
+               printf("%s: ", field);
+               for (j=0; j<values[i]->bv_len; j++) {
+                       printf("%02X", (unsigned char)values[i]->bv_val[j]);
+               }
+               printf("\n");
+       }
+}
+
+static void dump_guid(const char *field, struct berval **values)
+{
+       int i;
+       UUID_FLAT guid;
+       for (i=0; values[i]; i++) {
+               memcpy(guid.info, values[i]->bv_val, sizeof(guid.info));
+               printf("%s: %s\n", field, 
+                      smb_uuid_string_static(smb_uuid_unpack_static(guid)));
+       }
+}
+
+/*
+  dump a sid result from ldap
+*/
+static void dump_sid(const char *field, struct berval **values)
+{
+       int i;
+       for (i=0; values[i]; i++) {
+               DOM_SID sid;
+               sid_parse(values[i]->bv_val, values[i]->bv_len, &sid);
+               printf("%s: %s\n", field, sid_string_static(&sid));
+       }
+}
+
+/*
+  dump ntSecurityDescriptor
 */
 static void dump_sd(const char *filed, struct berval **values)
 {
@@ -1190,10 +1655,13 @@ static BOOL ads_dump_field(char *field, void **values, void *data_area)
                void (*handler)(const char *, struct berval **);
        } handlers[] = {
                {"objectGUID", False, dump_guid},
+               {"netbootGUID", False, dump_guid},
                {"nTSecurityDescriptor", False, dump_sd},
                {"dnsRecord", False, dump_binary},
                {"objectSid", False, dump_sid},
                {"tokenGroups", False, dump_sid},
+               {"tokenGroupsNoGCAcceptable", False, dump_sid},
+               {"tokengroupsGlobalandUniversal", False, dump_sid},
                {NULL, True, NULL}
        };
        int i;
@@ -1287,7 +1755,7 @@ void ads_process_results(ADS_STRUCT *ads, void *res,
                        ldap_memfree(utf8_field);
                }
                ber_free(b, 0);
-               talloc_destroy_pool(ctx);
+               talloc_free_children(ctx);
                fn(NULL, NULL, data_area); /* completed an entry */
 
        }
@@ -1305,218 +1773,6 @@ int ads_count_replies(ADS_STRUCT *ads, void *res)
        return ldap_count_entries(ads->ld, (LDAPMessage *)res);
 }
 
-/**
- * Join a machine to a realm
- *  Creates the machine account and sets the machine password
- * @param ads connection to ads server
- * @param hostname name of host to add
- * @param org_unit Organizational unit to place machine in
- * @return status of join
- **/
-ADS_STATUS ads_join_realm(ADS_STRUCT *ads, const char *hostname, 
-                         uint32 account_type, const char *org_unit)
-{
-       ADS_STATUS status;
-       LDAPMessage *res;
-       char *host;
-
-       /* hostname must be lowercase */
-       host = strdup(hostname);
-       strlower_m(host);
-
-       /*
-       status = ads_find_machine_acct(ads, (void **)&res, host);
-       if (ADS_ERR_OK(status) && ads_count_replies(ads, res) == 1) {
-               DEBUG(0, ("Host account for %s already exists - deleting old account\n", host));
-               status = ads_leave_realm(ads, host);
-               if (!ADS_ERR_OK(status)) {
-                       DEBUG(0, ("Failed to delete host '%s' from the '%s' realm.\n", 
-                                 host, ads->config.realm));
-                       return status;
-               }
-       }
-       */
-
-       status = ads_add_machine_acct(ads, host, account_type, org_unit);
-       if (!ADS_ERR_OK(status)) {
-               DEBUG(0, ("ads_add_machine_acct: %s\n", ads_errstr(status)));
-               return status;
-       }
-
-       status = ads_find_machine_acct(ads, (void **)&res, host);
-       if (!ADS_ERR_OK(status)) {
-               DEBUG(0, ("Host account test failed\n"));
-               return status;
-       }
-
-       free(host);
-
-       return status;
-}
-
-/**
- * Delete a machine from the realm
- * @param ads connection to ads server
- * @param hostname Machine to remove
- * @return status of delete
- **/
-ADS_STATUS ads_leave_realm(ADS_STRUCT *ads, const char *hostname)
-{
-       ADS_STATUS status;
-       void *res, *msg;
-       char *hostnameDN, *host; 
-       int rc;
-
-       /* hostname must be lowercase */
-       host = strdup(hostname);
-       strlower_m(host);
-
-       status = ads_find_machine_acct(ads, &res, host);
-       if (!ADS_ERR_OK(status)) {
-           DEBUG(0, ("Host account for %s does not exist.\n", host));
-           return status;
-       }
-
-       msg = ads_first_entry(ads, res);
-       if (!msg) {
-               return ADS_ERROR_SYSTEM(ENOENT);
-       }
-
-       hostnameDN = ads_get_dn(ads, (LDAPMessage *)msg);
-       rc = ldap_delete_s(ads->ld, hostnameDN);
-       ads_memfree(ads, hostnameDN);
-       if (rc != LDAP_SUCCESS) {
-               return ADS_ERROR(rc);
-       }
-
-       status = ads_find_machine_acct(ads, &res, host);
-       if (ADS_ERR_OK(status) && ads_count_replies(ads, res) == 1) {
-               DEBUG(0, ("Failed to remove host account.\n"));
-               return status;
-       }
-
-       free(host);
-
-       return status;
-}
-
-/**
- * add machine account to existing security descriptor 
- * @param ads connection to ads server
- * @param hostname machine to add
- * @param dn DN of security descriptor
- * @return status
- **/
-ADS_STATUS ads_set_machine_sd(ADS_STRUCT *ads, const char *hostname, char *dn)
-{
-       const char     *attrs[] = {"nTSecurityDescriptor", "objectSid", 0};
-       char           *expr     = 0;
-       size_t          sd_size = 0;
-       struct berval   bval = {0, NULL};
-       prs_struct      ps_wire;
-       char           *escaped_hostname = escape_ldap_string_alloc(hostname);
-
-       LDAPMessage *res  = 0;
-       LDAPMessage *msg  = 0;
-       ADS_MODLIST  mods = 0;
-
-       NTSTATUS    status;
-       ADS_STATUS  ret;
-       DOM_SID     sid;
-       SEC_DESC   *psd = NULL;
-       TALLOC_CTX *ctx = NULL; 
-
-       /* Avoid segmentation fault in prs_mem_free if
-        * we have to bail out before prs_init */
-       ps_wire.is_dynamic = False;
-
-       if (!ads) return ADS_ERROR(LDAP_SERVER_DOWN);
-
-       ret = ADS_ERROR(LDAP_SUCCESS);
-
-       if (!escaped_hostname) {
-               return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
-       }
-
-       if (asprintf(&expr, "(samAccountName=%s$)", escaped_hostname) == -1) {
-               DEBUG(1, ("ads_set_machine_sd: asprintf failed!\n"));
-               SAFE_FREE(escaped_hostname);
-               return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
-       }
-
-       SAFE_FREE(escaped_hostname);
-
-       ret = ads_search(ads, (void *) &res, expr, attrs);
-
-       if (!ADS_ERR_OK(ret)) return ret;
-
-       if ( !(msg = ads_first_entry(ads, res) )) {
-               ret = ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
-               goto ads_set_sd_error;
-       }
-
-       if (!ads_pull_sid(ads, msg, attrs[1], &sid)) {
-               ret = ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
-               goto ads_set_sd_error;
-       }
-
-       if (!(ctx = talloc_init("sec_io_desc"))) {
-               ret =  ADS_ERROR(LDAP_NO_MEMORY);
-               goto ads_set_sd_error;
-       }
-
-       if (!ads_pull_sd(ads, ctx, msg, attrs[0], &psd)) {
-               ret = ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
-               goto ads_set_sd_error;
-       }
-
-       status = sec_desc_add_sid(ctx, &psd, &sid, SEC_RIGHTS_FULL_CTRL, &sd_size);
-
-       if (!NT_STATUS_IS_OK(status)) {
-               ret = ADS_ERROR_NT(status);
-               goto ads_set_sd_error;
-       }
-
-       if (!prs_init(&ps_wire, sd_size, ctx, MARSHALL)) {
-               ret = ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
-       }
-
-       if (!sec_io_desc("sd_wire", &psd, &ps_wire, 1)) {
-               ret = ADS_ERROR(LDAP_NO_MEMORY);
-               goto ads_set_sd_error;
-       }
-
-#if 0
-       file_save("/tmp/sec_desc.new", ps_wire.data_p, sd_size);
-#endif
-       if (!(mods = ads_init_mods(ctx))) return ADS_ERROR(LDAP_NO_MEMORY);
-
-       bval.bv_len = prs_offset(&ps_wire);
-       bval.bv_val = talloc(ctx, bval.bv_len);
-       if (!bval.bv_val) {
-               ret = ADS_ERROR(LDAP_NO_MEMORY);
-               goto ads_set_sd_error;
-       }
-
-       prs_set_offset(&ps_wire, 0);
-
-       if (!prs_copy_data_out(bval.bv_val, &ps_wire, bval.bv_len)) {
-               ret = ADS_ERROR(LDAP_NO_MEMORY);
-               goto ads_set_sd_error;          
-       }
-
-       ret = ads_mod_ber(ctx, &mods, attrs[0], &bval);
-       if (ADS_ERR_OK(ret)) {
-               ret = ads_gen_mod(ads, dn, mods);
-       }
-
-ads_set_sd_error:
-       ads_msgfree(ads, res);
-       prs_mem_free(&ps_wire);
-       talloc_destroy(ctx);
-       return ret;
-}
-
 /**
  * pull the first entry from a ADS result
  * @param ads connection to ads server
@@ -1579,27 +1835,26 @@ char *ads_pull_string(ADS_STRUCT *ads,
  * @return Result strings in talloc context
  **/
 char **ads_pull_strings(ADS_STRUCT *ads, 
-                      TALLOC_CTX *mem_ctx, void *msg, const char *field)
+                       TALLOC_CTX *mem_ctx, void *msg, const char *field,
+                       size_t *num_values)
 {
        char **values;
        char **ret = NULL;
-       int i, n;
+       int i;
 
        values = ldap_get_values(ads->ld, msg, field);
        if (!values)
                return NULL;
 
-       for (i=0;values[i];i++)
-               /* noop */ ;
-       n = i;
+       *num_values = ldap_count_values(values);
 
-       ret = talloc(mem_ctx, sizeof(char *) * (n+1));
+       ret = TALLOC_ARRAY(mem_ctx, char *, *num_values + 1);
        if (!ret) {
                ldap_value_free(values);
                return NULL;
        }
 
-       for (i=0;i<n;i++) {
+       for (i=0;i<*num_values;i++) {
                if (pull_utf8_talloc(mem_ctx, &ret[i], values[i]) == -1) {
                        ldap_value_free(values);
                        return NULL;
@@ -1611,17 +1866,144 @@ char **ads_pull_strings(ADS_STRUCT *ads,
        return ret;
 }
 
-
 /**
- * pull a single uint32 from a ADS result
+ * pull an array of strings from a ADS result 
+ *  (handle large multivalue attributes with range retrieval)
  * @param ads connection to ads server
+ * @param mem_ctx TALLOC_CTX to use for allocating result string
  * @param msg Results of search
  * @param field Attribute to retrieve
- * @param v Pointer to int to store result
- * @return boolean inidicating success
-*/
-BOOL ads_pull_uint32(ADS_STRUCT *ads, 
-                    void *msg, const char *field, uint32 *v)
+ * @param current_strings strings returned by a previous call to this function
+ * @param next_attribute The next query should ask for this attribute
+ * @param num_values How many values did we get this time?
+ * @param more_values Are there more values to get?
+ * @return Result strings in talloc context
+ **/
+char **ads_pull_strings_range(ADS_STRUCT *ads, 
+                             TALLOC_CTX *mem_ctx,
+                             void *msg, const char *field,
+                             char **current_strings,
+                             const char **next_attribute,
+                             size_t *num_strings,
+                             BOOL *more_strings)
+{
+       char *attr;
+       char *expected_range_attrib, *range_attr;
+       BerElement *ptr = NULL;
+       char **strings;
+       char **new_strings;
+       size_t num_new_strings;
+       unsigned long int range_start;
+       unsigned long int range_end;
+       
+       /* we might have been given the whole lot anyway */
+       if ((strings = ads_pull_strings(ads, mem_ctx, msg, field, num_strings))) {
+               *more_strings = False;
+               return strings;
+       }
+
+       expected_range_attrib = talloc_asprintf(mem_ctx, "%s;Range=", field);
+
+       /* look for Range result */
+       for (attr = ldap_first_attribute(ads->ld, (LDAPMessage *)msg, &ptr); 
+            attr; 
+            attr = ldap_next_attribute(ads->ld, (LDAPMessage *)msg, ptr)) {
+               /* we ignore the fact that this is utf8, as all attributes are ascii... */
+               if (strnequal(attr, expected_range_attrib, strlen(expected_range_attrib))) {
+                       range_attr = attr;
+                       break;
+               }
+               ldap_memfree(attr);
+       }
+       if (!attr) {
+               ber_free(ptr, 0);
+               /* nothing here - this field is just empty */
+               *more_strings = False;
+               return NULL;
+       }
+       
+       if (sscanf(&range_attr[strlen(expected_range_attrib)], "%lu-%lu", 
+                  &range_start, &range_end) == 2) {
+               *more_strings = True;
+       } else {
+               if (sscanf(&range_attr[strlen(expected_range_attrib)], "%lu-*", 
+                          &range_start) == 1) {
+                       *more_strings = False;
+               } else {
+                       DEBUG(1, ("ads_pull_strings_range:  Cannot parse Range attriubte (%s)\n", 
+                                 range_attr));
+                       ldap_memfree(range_attr);
+                       *more_strings = False;
+                       return NULL;
+               }
+       }
+
+       if ((*num_strings) != range_start) {
+               DEBUG(1, ("ads_pull_strings_range: Range attribute (%s) doesn't start at %u, but at %lu"
+                         " - aborting range retreival\n",
+                         range_attr, (unsigned int)(*num_strings) + 1, range_start));
+               ldap_memfree(range_attr);
+               *more_strings = False;
+               return NULL;
+       }
+
+       new_strings = ads_pull_strings(ads, mem_ctx, msg, range_attr, &num_new_strings);
+       
+       if (*more_strings && ((*num_strings + num_new_strings) != (range_end + 1))) {
+               DEBUG(1, ("ads_pull_strings_range: Range attribute (%s) tells us we have %lu "
+                         "strings in this bunch, but we only got %lu - aborting range retreival\n",
+                         range_attr, (unsigned long int)range_end - range_start + 1, 
+                         (unsigned long int)num_new_strings));
+               ldap_memfree(range_attr);
+               *more_strings = False;
+               return NULL;
+       }
+
+       strings = TALLOC_REALLOC_ARRAY(mem_ctx, current_strings, char *,
+                                *num_strings + num_new_strings);
+       
+       if (strings == NULL) {
+               ldap_memfree(range_attr);
+               *more_strings = False;
+               return NULL;
+       }
+       
+       if (new_strings && num_new_strings) {
+               memcpy(&strings[*num_strings], new_strings,
+                      sizeof(*new_strings) * num_new_strings);
+       }
+
+       (*num_strings) += num_new_strings;
+
+       if (*more_strings) {
+               *next_attribute = talloc_asprintf(mem_ctx,
+                                                 "%s;range=%d-*", 
+                                                 field,
+                                                 (int)*num_strings);
+               
+               if (!*next_attribute) {
+                       DEBUG(1, ("talloc_asprintf for next attribute failed!\n"));
+                       ldap_memfree(range_attr);
+                       *more_strings = False;
+                       return NULL;
+               }
+       }
+
+       ldap_memfree(range_attr);
+
+       return strings;
+}
+
+/**
+ * pull a single uint32 from a ADS result
+ * @param ads connection to ads server
+ * @param msg Results of search
+ * @param field Attribute to retrieve
+ * @param v Pointer to int to store result
+ * @return boolean inidicating success
+*/
+BOOL ads_pull_uint32(ADS_STRUCT *ads, 
+                    void *msg, const char *field, uint32 *v)
 {
        char **values;
 
@@ -1646,16 +2028,18 @@ BOOL ads_pull_uint32(ADS_STRUCT *ads,
  * @return boolean indicating success
  **/
 BOOL ads_pull_guid(ADS_STRUCT *ads,
-                  void *msg, GUID *guid)
+                  void *msg, struct uuid *guid)
 {
        char **values;
+       UUID_FLAT flat_guid;
 
        values = ldap_get_values(ads->ld, msg, "objectGUID");
        if (!values)
                return False;
        
        if (values[0]) {
-               memcpy(guid, values[0], sizeof(GUID));
+               memcpy(&flat_guid.info, values[0], sizeof(UUID_FLAT));
+               smb_uuid_unpack(flat_guid, guid);
                ldap_value_free(values);
                return True;
        }
@@ -1715,7 +2099,7 @@ int ads_pull_sids(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
        for (i=0; values[i]; i++)
                /* nop */ ;
 
-       (*sids) = talloc(mem_ctx, sizeof(DOM_SID) * i);
+       (*sids) = TALLOC_ARRAY(mem_ctx, DOM_SID, i);
        if (!(*sids)) {
                ldap_value_free_len(values);
                return 0;
@@ -1779,13 +2163,19 @@ BOOL ads_pull_sd(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
  */
 char *ads_pull_username(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, void *msg)
 {
+#if 0  /* JERRY */
        char *ret, *p;
 
+       /* lookup_name() only works on the sAMAccountName to 
+          returning the username portion of userPrincipalName
+          breaks winbindd_getpwnam() */
+
        ret = ads_pull_string(ads, mem_ctx, msg, "userPrincipalName");
-       if (ret && (p = strchr(ret, '@'))) {
+       if (ret && (p = strchr_m(ret, '@'))) {
                *p = 0;
                return ret;
        }
+#endif
        return ads_pull_string(ads, mem_ctx, msg, "sAMAccountName");
 }
 
@@ -1836,83 +2226,123 @@ static time_t ads_parse_time(const char *str)
        return timegm(&tm);
 }
 
+/********************************************************************
+********************************************************************/
 
-/**
- * Find the servers name and realm - this can be done before authentication 
- *  The ldapServiceName field on w2k  looks like this:
- *    vnet3.home.samba.org:win2000-vnet3$@VNET3.HOME.SAMBA.ORG
- * @param ads connection to ads server
- * @return status of search
- **/
-ADS_STATUS ads_server_info(ADS_STRUCT *ads)
+ADS_STATUS ads_current_time(ADS_STRUCT *ads)
 {
-       const char *attrs[] = {"ldapServiceName", "currentTime", NULL};
+       const char *attrs[] = {"currentTime", NULL};
        ADS_STATUS status;
        void *res;
-       char *value;
-       char *p;
        char *timestr;
        TALLOC_CTX *ctx;
+       ADS_STRUCT *ads_s = ads;
 
-       if (!(ctx = talloc_init("ads_server_info"))) {
+       if (!(ctx = talloc_init("ads_current_time"))) {
                return ADS_ERROR(LDAP_NO_MEMORY);
        }
 
-       status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
-       if (!ADS_ERR_OK(status)) return status;
+        /* establish a new ldap tcp session if necessary */
 
-       value = ads_pull_string(ads, ctx, res, "ldapServiceName");
-       if (!value) {
-               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+       if ( !ads->ld ) {
+               if ( (ads_s = ads_init( ads->server.realm, ads->server.workgroup, 
+                       ads->server.ldap_server )) == NULL )
+               {
+                       goto done;
+               }
+               ads_s->auth.flags = ADS_AUTH_ANON_BIND;
+               status = ads_connect( ads_s );
+               if ( !ADS_ERR_OK(status))
+                       goto done;
+       }
+
+       status = ads_do_search(ads_s, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               goto done;
        }
 
-       timestr = ads_pull_string(ads, ctx, res, "currentTime");
+       timestr = ads_pull_string(ads_s, ctx, res, "currentTime");
        if (!timestr) {
-               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+               ads_msgfree(ads, res);
+               status = ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+               goto done;
        }
 
-       ldap_msgfree(res);
+       /* but save the time and offset in the original ADS_STRUCT */   
+       
+       ads->config.current_time = ads_parse_time(timestr);
 
-       p = strchr(value, ':');
-       if (!p) {
-               talloc_destroy(ctx);
-               DEBUG(1, ("ads_server_info: returned ldap server name did not contain a ':' so was deemed invalid\n"));
-               return ADS_ERROR(LDAP_DECODING_ERROR);
+       if (ads->config.current_time != 0) {
+               ads->auth.time_offset = ads->config.current_time - time(NULL);
+               DEBUG(4,("time offset is %d seconds\n", ads->auth.time_offset));
        }
 
-       SAFE_FREE(ads->config.ldap_server_name);
+       ads_msgfree(ads, res);
 
-       ads->config.ldap_server_name = strdup(p+1);
-       p = strchr(ads->config.ldap_server_name, '$');
-       if (!p || p[1] != '@') {
-               talloc_destroy(ctx);
-               DEBUG(1, ("ads_server_info: returned ldap server name (%s) does not contain '$@' so was deemed invalid\n", ads->config.ldap_server_name));
-               SAFE_FREE(ads->config.ldap_server_name);
-               return ADS_ERROR(LDAP_DECODING_ERROR);
+       status = ADS_SUCCESS;
+
+done:
+       /* free any temporary ads connections */
+       if ( ads_s != ads ) {
+               ads_destroy( &ads_s );
        }
+       talloc_destroy(ctx);
 
-       *p = 0;
+       return status;
+}
 
-       SAFE_FREE(ads->config.realm);
-       SAFE_FREE(ads->config.bind_path);
+/********************************************************************
+********************************************************************/
 
-       ads->config.realm = strdup(p+2);
-       ads->config.bind_path = ads_build_dn(ads->config.realm);
+ADS_STATUS ads_domain_func_level(ADS_STRUCT *ads, uint32 *val)
+{
+       const char *attrs[] = {"domainFunctionality", NULL};
+       ADS_STATUS status;
+       void *res;
+       ADS_STRUCT *ads_s = ads;
+       
+       *val = DS_DOMAIN_FUNCTION_2000;
 
-       DEBUG(3,("got ldap server name %s@%s, using bind path: %s\n", 
-                ads->config.ldap_server_name, ads->config.realm,
-                ads->config.bind_path));
+        /* establish a new ldap tcp session if necessary */
 
-       ads->config.current_time = ads_parse_time(timestr);
+       if ( !ads->ld ) {
+               if ( (ads_s = ads_init( ads->server.realm, ads->server.workgroup, 
+                       ads->server.ldap_server )) == NULL )
+               {
+                       goto done;
+               }
+               ads_s->auth.flags = ADS_AUTH_ANON_BIND;
+               status = ads_connect( ads_s );
+               if ( !ADS_ERR_OK(status))
+                       goto done;
+       }
 
-       if (ads->config.current_time != 0) {
-               ads->auth.time_offset = ads->config.current_time - time(NULL);
-               DEBUG(4,("time offset is %d seconds\n", ads->auth.time_offset));
+       /* If the attribute does not exist assume it is a Windows 2000 
+          functional domain */
+          
+       status = ads_do_search(ads_s, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               if ( status.err.rc == LDAP_NO_SUCH_ATTRIBUTE ) {
+                       status = ADS_SUCCESS;
+               }
+               goto done;
        }
 
-       talloc_destroy(ctx);
+       if ( !ads_pull_uint32(ads_s, res, "domainFunctionality", val) ) {
+               DEBUG(5,("ads_domain_func_level: Failed to pull the domainFunctionality attribute.\n"));
+       }
+       DEBUG(3,("ads_domain_func_level: %d\n", *val));
 
-       return ADS_SUCCESS;
+       
+       ads_msgfree(ads, res);
+
+done:
+       /* free any temporary ads connections */
+       if ( ads_s != ads ) {
+               ads_destroy( &ads_s );
+       }
+
+       return status;
 }
 
 /**
@@ -1931,6 +2361,7 @@ ADS_STATUS ads_domain_sid(ADS_STRUCT *ads, DOM_SID *sid)
                           attrs, &res);
        if (!ADS_ERR_OK(rc)) return rc;
        if (!ads_pull_sid(ads, res, "objectSid", sid)) {
+               ads_msgfree(ads, res);
                return ADS_ERROR_SYSTEM(ENOENT);
        }
        ads_msgfree(ads, res);
@@ -1938,73 +2369,386 @@ ADS_STATUS ads_domain_sid(ADS_STRUCT *ads, DOM_SID *sid)
        return ADS_SUCCESS;
 }
 
-/* this is rather complex - we need to find the allternate (netbios) name
-   for the domain, but there isn't a simple query to do this. Instead
-   we look for the principle names on the DCs account and find one that has 
-   the right form, then extract the netbios name of the domain from that
+/**
+ * find our site name 
+ * @param ads connection to ads server
+ * @param mem_ctx Pointer to talloc context
+ * @param site_name Pointer to the sitename
+ * @return status of search
+ **/
+ADS_STATUS ads_site_dn(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, const char **site_name)
+{
+       ADS_STATUS status;
+       void *res;
+       const char *dn, *service_name;
+       const char *attrs[] = { "dsServiceName", NULL };
+
+       status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       service_name = ads_pull_string(ads, mem_ctx, res, "dsServiceName");
+       if (service_name == NULL) {
+               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+       }
 
-   NOTE! better method is this:
+       /* go up three levels */
+       dn = ads_parent_dn(ads_parent_dn(ads_parent_dn(service_name)));
+       if (dn == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
-bin/net -Uadministrator%XXXXX ads search '(&(objectclass=crossref)(dnsroot=VNET3.HOME.SAMBA.ORG))'  nETBIOSName 
+       *site_name = talloc_strdup(mem_ctx, dn);
+       if (*site_name == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
-but you need to force the bind path to match the configurationNamingContext from the rootDSE
+       ads_msgfree(ads, res);
 
-*/
-ADS_STATUS ads_workgroup_name(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, const char **workgroup)
+       return status;
+       /*
+       dsServiceName: CN=NTDS Settings,CN=W2K3DC,CN=Servers,CN=Default-First-Site-Name,CN=Sites,CN=Configuration,DC=ber,DC=suse,DC=de
+       */                                               
+}
+
+/**
+ * find the site dn where a machine resides
+ * @param ads connection to ads server
+ * @param mem_ctx Pointer to talloc context
+ * @param computer_name name of the machine
+ * @param site_name Pointer to the sitename
+ * @return status of search
+ **/
+ADS_STATUS ads_site_dn_for_machine(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, const char *computer_name, const char **site_dn)
 {
-       char *expr;
-       ADS_STATUS rc;
-       char **principles;
-       char *prefix;
-       int prefix_length;
-       int i;
+       ADS_STATUS status;
        void *res;
-       const char *attrs[] = {"servicePrincipalName", NULL};
+       const char *parent, *config_context, *filter;
+       const char *attrs[] = { "configurationNamingContext", NULL };
+       char *dn;
 
-       (*workgroup) = NULL;
+       /* shortcut a query */
+       if (strequal(computer_name, ads->config.ldap_server_name)) {
+               return ads_site_dn(ads, mem_ctx, site_dn);
+       }
 
-       asprintf(&expr, "(&(objectclass=computer)(dnshostname=%s.%s))", 
-                ads->config.ldap_server_name, ads->config.realm);
-       rc = ads_search(ads, &res, expr, attrs);
-       free(expr);
+       status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
 
-       if (!ADS_ERR_OK(rc)) {
-               return rc;
+       config_context = ads_pull_string(ads, mem_ctx, res, "configurationNamingContext");
+       if (config_context == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
        }
 
-       principles = ads_pull_strings(ads, mem_ctx, res, "servicePrincipalName");
+       filter = talloc_asprintf(mem_ctx, "(cn=%s)", computer_name);
+       if (filter == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
+       status = ads_do_search(ads, config_context, LDAP_SCOPE_SUBTREE, filter, NULL, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       if (ads_count_replies(ads, res) != 1) {
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
+
+       dn = ads_get_dn(ads, res);
+       if (dn == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       /* go up three levels */
+       parent = ads_parent_dn(ads_parent_dn(ads_parent_dn(dn)));
+       if (parent == NULL) {
+               ads_memfree(ads, dn);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       *site_dn = talloc_strdup(mem_ctx, parent);
+       if (*site_dn == NULL) {
+               ads_memfree(ads, dn);
+               ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       ads_memfree(ads, dn);
        ads_msgfree(ads, res);
 
-       if (!principles) {
-               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+       return status;
+}
+
+/**
+ * get the upn suffixes for a domain
+ * @param ads connection to ads server
+ * @param mem_ctx Pointer to talloc context
+ * @param suffixes Pointer to an array of suffixes
+ * @param site_name Pointer to the number of suffixes
+ * @return status of search
+ **/
+ADS_STATUS ads_upn_suffixes(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, char **suffixes, size_t *num_suffixes)
+{
+       ADS_STATUS status;
+       void *res;
+       const char *config_context, *base;
+       const char *attrs[] = { "configurationNamingContext", NULL };
+       const char *attrs2[] = { "uPNSuffixes", NULL };
+
+       status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       config_context = ads_pull_string(ads, mem_ctx, res, "configurationNamingContext");
+       if (config_context == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
        }
 
-       asprintf(&prefix, "HOST/%s.%s/", 
-                ads->config.ldap_server_name, 
-                ads->config.realm);
+       base = talloc_asprintf(mem_ctx, "cn=Partitions,%s", config_context);
+       if (base == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       status = ads_search_dn(ads, &res, base, attrs2); 
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
 
-       prefix_length = strlen(prefix);
+       if (ads_count_replies(ads, res) != 1) {
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
 
-       for (i=0;principles[i]; i++) {
-               if (strnequal(principles[i], prefix, prefix_length) &&
-                   !strequal(ads->config.realm, principles[i]+prefix_length) &&
-                   !strchr(principles[i]+prefix_length, '.')) {
-                       /* found an alternate (short) name for the domain. */
-                       DEBUG(3,("Found alternate name '%s' for realm '%s'\n",
-                                principles[i]+prefix_length, 
-                                ads->config.realm));
-                       (*workgroup) = talloc_strdup(mem_ctx, principles[i]+prefix_length);
-                       break;
+       suffixes = ads_pull_strings(ads, mem_ctx, &res, "uPNSuffixes", num_suffixes);
+       if (suffixes == NULL) {
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       ads_msgfree(ads, res);
+
+       return status;
+}
+
+/**
+ * pull a DOM_SID from an extended dn string
+ * @param mem_ctx TALLOC_CTX 
+ * @param flags string type of extended_dn
+ * @param sid pointer to a DOM_SID
+ * @return boolean inidicating success
+ **/
+BOOL ads_get_sid_from_extended_dn(TALLOC_CTX *mem_ctx, 
+                                 const char *dn, 
+                                 enum ads_extended_dn_flags flags, 
+                                 DOM_SID *sid)
+{
+       char *p, *q;
+
+       if (!dn) {
+               return False;
+       }
+
+       /* 
+        * ADS_EXTENDED_DN_HEX_STRING:
+        * <GUID=238e1963cb390f4bb032ba0105525a29>;<SID=010500000000000515000000bb68c8fd6b61b427572eb04556040000>;CN=gd,OU=berlin,OU=suse,DC=ber,DC=suse,DC=de
+        *
+        * ADS_EXTENDED_DN_STRING (only with w2k3):
+       <GUID=63198e23-39cb-4b0f-b032-ba0105525a29>;<SID=S-1-5-21-4257769659-666132843-1169174103-1110>;CN=gd,OU=berlin,OU=suse,DC=ber,DC=suse,DC=de
+        */
+
+       p = strchr(dn, ';');
+       if (!p) {
+               return False;
+       }
+
+       if (strncmp(p, ";<SID=", strlen(";<SID=")) != 0) {
+               return False;
+       }
+
+       p += strlen(";<SID=");
+
+       q = strchr(p, '>');
+       if (!q) {
+               return False;
+       }
+       
+       *q = '\0';
+
+       DEBUG(100,("ads_get_sid_from_extended_dn: sid string is %s\n", p));
+
+       switch (flags) {
+       
+       case ADS_EXTENDED_DN_STRING:
+               if (!string_to_sid(sid, p)) {
+                       return False;
+               }
+               break;
+       case ADS_EXTENDED_DN_HEX_STRING: {
+               pstring buf;
+               size_t buf_len;
+
+               buf_len = strhex_to_str(buf, strlen(p), p);
+               if (buf_len == 0) {
+                       return False;
+               }
+
+               if (!sid_parse(buf, buf_len, sid)) {
+                       DEBUG(10,("failed to parse sid\n"));
+                       return False;
+               }
+               break;
                }
+       default:
+               DEBUG(10,("unknown extended dn format\n"));
+               return False;
        }
-       free(prefix);
 
-       if (!*workgroup) {
-               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+       return True;
+}
+
+/**
+ * pull an array of DOM_SIDs from a ADS result
+ * @param ads connection to ads server
+ * @param mem_ctx TALLOC_CTX for allocating sid array
+ * @param msg Results of search
+ * @param field Attribute to retrieve
+ * @param flags string type of extended_dn
+ * @param sids pointer to sid array to allocate
+ * @return the count of SIDs pulled
+ **/
+int ads_pull_sids_from_extendeddn(ADS_STRUCT *ads, 
+                                 TALLOC_CTX *mem_ctx, 
+                                 void *msg, 
+                                 const char *field,
+                                 enum ads_extended_dn_flags flags,
+                                 DOM_SID **sids)
+{
+       int i;
+       size_t dn_count;
+       char **dn_strings;
+
+       if ((dn_strings = ads_pull_strings(ads, mem_ctx, msg, field, 
+                                          &dn_count)) == NULL) {
+               return 0;
+       }
+
+       (*sids) = TALLOC_ZERO_ARRAY(mem_ctx, DOM_SID, dn_count + 1);
+       if (!(*sids)) {
+               TALLOC_FREE(dn_strings);
+               return 0;
        }
+
+       for (i=0; i<dn_count; i++) {
+
+               if (!ads_get_sid_from_extended_dn(mem_ctx, dn_strings[i], 
+                                                 flags, &(*sids)[i])) {
+                       TALLOC_FREE(*sids);
+                       TALLOC_FREE(dn_strings);
+                       return 0;
+               }
+       }
+
+       TALLOC_FREE(dn_strings);
+
+       return dn_count;
+}
+
+/********************************************************************
+********************************************************************/
+
+char* ads_get_dnshostname( ADS_STRUCT *ads, TALLOC_CTX *ctx, const char *machine_name )
+{
+       LDAPMessage *res = NULL;
+       ADS_STATUS status;
+       int count = 0;
+       char *name = NULL;
        
-       return ADS_SUCCESS;
+       status = ads_find_machine_acct(ads, (void **)(void *)&res, global_myname());
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(0,("ads_get_dnshostname: Failed to find account for %s\n",
+                       global_myname()));
+               goto out;
+       }
+               
+       if ( (count = ads_count_replies(ads, res)) != 1 ) {
+               DEBUG(1,("ads_get_dnshostname: %d entries returned!\n", count));
+               goto out;
+       }
+               
+       if ( (name = ads_pull_string(ads, ctx, res, "dNSHostName")) == NULL ) {
+               DEBUG(0,("ads_get_dnshostname: No dNSHostName attribute!\n"));
+       }
+
+out:
+       ads_msgfree(ads, res);
+       
+       return name;
+}
+
+/********************************************************************
+********************************************************************/
+
+char* ads_get_upn( ADS_STRUCT *ads, TALLOC_CTX *ctx, const char *machine_name )
+{
+       LDAPMessage *res = NULL;
+       ADS_STATUS status;
+       int count = 0;
+       char *name = NULL;
+       
+       status = ads_find_machine_acct(ads, (void **)(void *)&res, global_myname());
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(0,("ads_get_upn: Failed to find account for %s\n",
+                       global_myname()));
+               goto out;
+       }
+               
+       if ( (count = ads_count_replies(ads, res)) != 1 ) {
+               DEBUG(1,("ads_get_upn: %d entries returned!\n", count));
+               goto out;
+       }
+               
+       if ( (name = ads_pull_string(ads, ctx, res, "userPrincipalName")) == NULL ) {
+               DEBUG(2,("ads_get_upn: No userPrincipalName attribute!\n"));
+       }
+
+out:
+       ads_msgfree(ads, res);
+       
+       return name;
+}
+
+/********************************************************************
+********************************************************************/
+
+char* ads_get_samaccountname( ADS_STRUCT *ads, TALLOC_CTX *ctx, const char *machine_name )
+{
+       LDAPMessage *res = NULL;
+       ADS_STATUS status;
+       int count = 0;
+       char *name = NULL;
+       
+       status = ads_find_machine_acct(ads, (void **)(void *)&res, global_myname());
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(0,("ads_get_dnshostname: Failed to find account for %s\n",
+                       global_myname()));
+               goto out;
+       }
+               
+       if ( (count = ads_count_replies(ads, res)) != 1 ) {
+               DEBUG(1,("ads_get_dnshostname: %d entries returned!\n", count));
+               goto out;
+       }
+               
+       if ( (name = ads_pull_string(ads, ctx, res, "sAMAccountName")) == NULL ) {
+               DEBUG(0,("ads_get_dnshostname: No sAMAccountName attribute!\n"));
+       }
+
+out:
+       ads_msgfree(ads, res);
+       
+       return name;
 }
 
 #endif