r18663: Fix one more uuid -> GUID.
[sfrench/samba-autobuild/.git] / source3 / libads / ldap.c
index 79f267a6a8fc744ef7664d3f0eaae9f6c3393ef5..f9bdb9c651cbb2308b142499eb98236bfc06f3c0 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;
+}
+
+/**********************************************
+ Do client and server sitename match ?
+**********************************************/
+
+BOOL ads_sitename_match(ADS_STRUCT *ads)
+{
+       if (ads->config.server_site_name == NULL &&
+           ads->config.client_site_name == NULL ) {
+               DEBUG(10,("ads_sitename_match: both null\n"));
+               return True;
+       }
+       if (ads->config.server_site_name &&
+           ads->config.client_site_name &&
+           strequal(ads->config.server_site_name,
+                    ads->config.client_site_name)) {
+               DEBUG(10,("ads_sitename_match: name %s match\n", ads->config.server_site_name));
+               return True;
+       }
+       DEBUG(10,("ads_sitename_match: no match %s %s\n",
+               ads->config.server_site_name ? ads->config.server_site_name : "NULL",
+               ads->config.client_site_name ? ads->config.client_site_name : "NULL"));
+       return False;
+}
+
 /*
   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);
+
+       ZERO_STRUCT( cldap_reply );
 
-       ads->ld = ldap_open(srv, port);
-       if (!ads->ld) {
-               free(srv);
+       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;
-}
+       /* Check the CLDAP reply flags */
 
-/*
-  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));
+       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;
+       }
 
-       
-       if (ldap_initialize((LDAP**)&(ads->ld), ads->server.ldap_uri) == LDAP_SUCCESS) {
-               return True;
+       /* Fill in the ads->config values */
+
+       SAFE_FREE(ads->config.realm);
+       SAFE_FREE(ads->config.bind_path);
+       SAFE_FREE(ads->config.ldap_server_name);
+       SAFE_FREE(ads->config.server_site_name);
+       SAFE_FREE(ads->config.client_site_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);
+       if (*cldap_reply.server_site_name) {
+               ads->config.server_site_name =
+                       SMB_STRDUP(cldap_reply.server_site_name);
+       }
+       if (*cldap_reply.client_site_name) {
+               ads->config.client_site_name =
+                       SMB_STRDUP(cldap_reply.client_site_name);
        }
-       DEBUG(0, ("ldap_initialize: %s\n", strerror(errno)));
-       
-#else 
+               
+       ads->server.workgroup          = SMB_STRDUP(cldap_reply.netbios_domain);
 
-       DEBUG(1, ("no URL support in LDAP libs!\n"));
-#endif
+       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 False;
+       return True;
 }
 
 /**********************************************************************
@@ -99,7 +217,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 +225,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 +256,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,30 +265,49 @@ 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) );
                
                if ( !NT_STATUS_IS_OK(check_negative_conn_cache(realm, server)) )
                        continue;
+
+               if (!got_realm) {
+                       /* realm in this case is a workgroup name. We need
+                          to ignore any IP addresses in the negative connection
+                          cache that match ip addresses returned in the ad realm
+                          case. It sucks that I have to reproduce the logic above... */
+                       c_realm = ads->server.realm;
+                       if ( !c_realm || !*c_realm ) {
+                               if ( !ads->server.workgroup || !*ads->server.workgroup ) {
+                                       c_realm = lp_realm();
+                               }
+                       }
+                       if (c_realm && *c_realm &&
+                                       !NT_STATUS_IS_OK(check_negative_conn_cache(c_realm, server))) {
+                               /* Ensure we add the workgroup name for this
+                                  IP address as negative too. */
+                               add_failed_connection_entry( realm, server, NT_STATUS_UNSUCCESSFUL );
+                               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 +316,7 @@ again:
 
        SAFE_FREE(ip_list);
        
-       return False;
+       return NT_STATUS_NO_LOGON_SERVERS;
 }
 
 
@@ -191,54 +329,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 +377,42 @@ 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 for workgroup and realm */
+       if (ads_sitename_match(ads)) {
+               saf_store( ads->server.workgroup, inet_ntoa(ads->ldap_ip));
+               saf_store( ads->server.realm, 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 +433,14 @@ 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 = (char *)TALLOC_MEMDUP(ctx, in_val->bv_val,
+                                             in_val->bv_len);
        return value;
 }
 
@@ -296,9 +454,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 +474,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 +494,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 +519,27 @@ 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)
+static ADS_STATUS ads_do_paged_search_args(ADS_STRUCT *ads,
+                                          const char *bind_path,
+                                          int scope, const char *expr,
+                                          const char **attrs, void *args,
+                                          LDAPMessage **res, 
+                                          int *count, struct berval **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 +571,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 +579,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 +641,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 +682,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);
 }
 
+static ADS_STATUS ads_do_paged_search(ADS_STRUCT *ads, const char *bind_path,
+                                     int scope, const char *expr,
+                                     const char **attrs, LDAPMessage **res, 
+                                     int *count, struct berval **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 +717,30 @@ 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,
+                                  LDAPMessage **res)
 {
-       void *cookie = NULL;
+       struct berval *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;
+               LDAPMessage *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 +753,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, LDAPMessage **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()
@@ -550,10 +785,10 @@ ADS_STATUS ads_do_search_all_fn(ADS_STRUCT *ads, const char *bind_path,
                                BOOL(*fn)(char *, void **, void *), 
                                void *data_area)
 {
-       void *cookie = NULL;
+       struct berval *cookie = NULL;
        int count = 0;
        ADS_STATUS status;
-       void *res;
+       LDAPMessage *res;
 
        status = ads_do_paged_search(ads, bind_path, scope, expr, attrs, &res,
                                     &count, &cookie);
@@ -586,15 +821,15 @@ ADS_STATUS ads_do_search_all_fn(ADS_STRUCT *ads, const char *bind_path,
  * @param res ** which will contain results - free res* with ads_msgfree()
  * @return status of search
  **/
-ADS_STATUS ads_do_search(ADS_STRUCT *ads, const char *bind_path, int scope, 
-                        const char *expr,
-                        const char **attrs, void **res)
+ ADS_STATUS ads_do_search(ADS_STRUCT *ads, const char *bind_path, int scope, 
+                         const char *expr,
+                         const char **attrs, LDAPMessage **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 +858,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"));
@@ -653,9 +885,8 @@ ADS_STATUS ads_do_search(ADS_STRUCT *ads, const char *bind_path, int scope,
  * @param attrs Attributes to retrieve
  * @return status of search
  **/
-ADS_STATUS ads_search(ADS_STRUCT *ads, void **res, 
-                     const char *expr, 
-                     const char **attrs)
+ ADS_STATUS ads_search(ADS_STRUCT *ads, LDAPMessage **res, 
+                      const char *expr, const char **attrs)
 {
        return ads_do_search(ads, ads->config.bind_path, LDAP_SCOPE_SUBTREE, 
                             expr, attrs, res);
@@ -669,11 +900,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, 
-                        const char *dn, 
-                        const char **attrs)
+ ADS_STATUS ads_search_dn(ADS_STRUCT *ads, LDAPMessage **res, 
+                         const char *dn, const char **attrs)
 {
-       return ads_do_search(ads, dn, LDAP_SCOPE_BASE, "(objectclass=*)", attrs, res);
+       return ads_do_search(ads, dn, LDAP_SCOPE_BASE, "(objectclass=*)",
+                            attrs, res);
 }
 
 /**
@@ -681,7 +912,7 @@ ADS_STATUS ads_search_dn(ADS_STRUCT *ads, void **res,
  * @param ads connection to ads server
  * @param msg Search results to free
  **/
-void ads_msgfree(ADS_STRUCT *ads, void *msg)
+ void ads_msgfree(ADS_STRUCT *ads, LDAPMessage *msg)
 {
        if (!msg) return;
        ldap_msgfree(msg);
@@ -703,7 +934,7 @@ void ads_memfree(ADS_STRUCT *ads, void *mem)
  * @param msg Search result
  * @return dn string
  **/
-char *ads_get_dn(ADS_STRUCT *ads, void *msg)
+ char *ads_get_dn(ADS_STRUCT *ads, LDAPMessage *msg)
 {
        char *utf8_dn, *unix_dn;
 
@@ -723,6 +954,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, LDAPMessage *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 +998,24 @@ 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, LDAPMessage **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 +1029,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 +1043,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 +1066,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 +1108,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 +1128,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 +1149,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 +1167,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 +1211,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,162 +1238,423 @@ 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");
+       }
 
-/*
-  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)
+       /* 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);
+}
+
+/**
+ * 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;
+       ADS_STATUS status;
        LDAPMessage *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);
+       ads_memfree(ads, wkn_dn);
+       ldap_value_free(wkn_dn_exp);
+       ldap_value_free(bind_dn_exp);
 
-       if (!ADS_ERR_OK(ret))
-               goto done;
-
-       /* 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);
 }
 
-static void dump_guid(const char *field, struct berval **values)
+/**
+ * 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.
+ **/
+
+uint32 ads_get_kvno(ADS_STRUCT *ads, const char *machine_name)
 {
-       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)));
+       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, &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;
+       }
+
+       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 a sid result from ldap
-*/
-static void dump_sid(const char *field, struct berval **values)
-{
+/**
+ * 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, &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, &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;
@@ -1184,10 +1717,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;
@@ -1220,7 +1756,7 @@ static BOOL ads_dump_field(char *field, void **values, void *data_area)
  * @param res Results to dump
  **/
 
-void ads_dump(ADS_STRUCT *ads, void *res)
+ void ads_dump(ADS_STRUCT *ads, LDAPMessage *res)
 {
        ads_process_results(ads, res, ads_dump_field, NULL);
 }
@@ -1236,11 +1772,11 @@ void ads_dump(ADS_STRUCT *ads, void *res)
  * @param fn Function for processing each result
  * @param data_area user-defined area to pass to function
  **/
-void ads_process_results(ADS_STRUCT *ads, void *res,
-                        BOOL(*fn)(char *, void **, void *),
-                        void *data_area)
+ void ads_process_results(ADS_STRUCT *ads, LDAPMessage *res,
+                         BOOL(*fn)(char *, void **, void *),
+                         void *data_area)
 {
-       void *msg;
+       LDAPMessage *msg;
        TALLOC_CTX *ctx;
 
        if (!(ctx = talloc_init("ads_process_results")))
@@ -1281,7 +1817,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 */
 
        }
@@ -1300,305 +1836,93 @@ int ads_count_replies(ADS_STRUCT *ads, void *res)
 }
 
 /**
- * Join a machine to a realm
- *  Creates the machine account and sets the machine password
+ * pull the first entry from a ADS result
  * @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
+ * @param res Results of search
+ * @return first entry from result
  **/
-ADS_STATUS ads_join_realm(ADS_STRUCT *ads, const char *hostname, 
-                         uint32 account_type, const char *org_unit)
+ LDAPMessage *ads_first_entry(ADS_STRUCT *ads, LDAPMessage *res)
 {
-       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;
+       return ldap_first_entry(ads->ld, res);
 }
 
 /**
- * Delete a machine from the realm
+ * pull the next entry from a ADS result
  * @param ads connection to ads server
- * @param hostname Machine to remove
- * @return status of delete
+ * @param res Results of search
+ * @return next entry from result
  **/
-ADS_STATUS ads_leave_realm(ADS_STRUCT *ads, const char *hostname)
+ LDAPMessage *ads_next_entry(ADS_STRUCT *ads, LDAPMessage *res)
 {
-       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);
-       }
+       return ldap_next_entry(ads->ld, res);
+}
 
-       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);
-       }
+/**
+ * pull a single string from a ADS result
+ * @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
+ * @return Result string in talloc context
+ **/
+ char *ads_pull_string(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, LDAPMessage *msg,
+                      const char *field)
+{
+       char **values;
+       char *ret = NULL;
+       char *ux_string;
+       size_t 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;
+       values = ldap_get_values(ads->ld, msg, field);
+       if (!values)
+               return NULL;
+       
+       if (values[0]) {
+               rc = pull_utf8_talloc(mem_ctx, &ux_string, 
+                                     values[0]);
+               if (rc != (size_t)-1)
+                       ret = ux_string;
+               
        }
-
-       free(host);
-
-       return status;
+       ldap_value_free(values);
+       return ret;
 }
 
 /**
- * add machine account to existing security descriptor 
+ * pull an array of strings from a ADS result
  * @param ads connection to ads server
- * @param hostname machine to add
- * @param dn DN of security descriptor
- * @return status
+ * @param mem_ctx TALLOC_CTX to use for allocating result string
+ * @param msg Results of search
+ * @param field Attribute to retrieve
+ * @return Result strings in talloc context
  **/
-ADS_STATUS ads_set_machine_sd(ADS_STRUCT *ads, const char *hostname, char *dn)
+ char **ads_pull_strings(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
+                        LDAPMessage *msg, const char *field,
+                        size_t *num_values)
 {
-       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;
+       char **values;
+       char **ret = NULL;
+       int i;
 
-       if (!ads) return ADS_ERROR(LDAP_SERVER_DOWN);
+       values = ldap_get_values(ads->ld, msg, field);
+       if (!values)
+               return NULL;
 
-       ret = ADS_ERROR(LDAP_SUCCESS);
+       *num_values = ldap_count_values(values);
 
-       if (!escaped_hostname) {
-               return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
+       ret = TALLOC_ARRAY(mem_ctx, char *, *num_values + 1);
+       if (!ret) {
+               ldap_value_free(values);
+               return NULL;
        }
 
-       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);
+       for (i=0;i<*num_values;i++) {
+               if (pull_utf8_talloc(mem_ctx, &ret[i], values[i]) == -1) {
+                       ldap_value_free(values);
+                       return NULL;
+               }
        }
-
-       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
- * @param res Results of search
- * @return first entry from result
- **/
-void *ads_first_entry(ADS_STRUCT *ads, void *res)
-{
-       return (void *)ldap_first_entry(ads->ld, (LDAPMessage *)res);
-}
-
-/**
- * pull the next entry from a ADS result
- * @param ads connection to ads server
- * @param res Results of search
- * @return next entry from result
- **/
-void *ads_next_entry(ADS_STRUCT *ads, void *res)
-{
-       return (void *)ldap_next_entry(ads->ld, (LDAPMessage *)res);
-}
-
-/**
- * pull a single string from a ADS result
- * @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
- * @return Result string in talloc context
- **/
-char *ads_pull_string(ADS_STRUCT *ads, 
-                     TALLOC_CTX *mem_ctx, void *msg, const char *field)
-{
-       char **values;
-       char *ret = NULL;
-       char *ux_string;
-       size_t rc;
-
-       values = ldap_get_values(ads->ld, msg, field);
-       if (!values)
-               return NULL;
-       
-       if (values[0]) {
-               rc = pull_utf8_talloc(mem_ctx, &ux_string, 
-                                     values[0]);
-               if (rc != (size_t)-1)
-                       ret = ux_string;
-               
-       }
-       ldap_value_free(values);
-       return ret;
-}
-
-/**
- * pull an array of strings from a ADS result
- * @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
- * @return Result strings in talloc context
- **/
-char **ads_pull_strings(ADS_STRUCT *ads, 
-                       TALLOC_CTX *mem_ctx, void *msg, const char *field,
-                       size_t *num_values)
-{
-       char **values;
-       char **ret = NULL;
-       int i;
-
-       values = ldap_get_values(ads->ld, msg, field);
-       if (!values)
-               return NULL;
-
-       *num_values = ldap_count_values(values);
-
-       ret = talloc(mem_ctx, sizeof(char *) * (*num_values+1));
-       if (!ret) {
-               ldap_value_free(values);
-               return NULL;
-       }
-
-       for (i=0;i<*num_values;i++) {
-               if (pull_utf8_talloc(mem_ctx, &ret[i], values[i]) == -1) {
-                       ldap_value_free(values);
-                       return NULL;
-               }
-       }
-       ret[i] = NULL;
+       ret[i] = NULL;
 
        ldap_value_free(values);
        return ret;
@@ -1617,13 +1941,13 @@ char **ads_pull_strings(ADS_STRUCT *ads,
  * @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 **ads_pull_strings_range(ADS_STRUCT *ads, 
+                              TALLOC_CTX *mem_ctx,
+                              LDAPMessage *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;
@@ -1655,7 +1979,7 @@ char **ads_pull_strings_range(ADS_STRUCT *ads,
        }
        if (!attr) {
                ber_free(ptr, 0);
-               /* nothing here - this feild is just empty */
+               /* nothing here - this field is just empty */
                *more_strings = False;
                return NULL;
        }
@@ -1679,7 +2003,7 @@ char **ads_pull_strings_range(ADS_STRUCT *ads,
        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, *num_strings + 1, range_start));
+                         range_attr, (unsigned int)(*num_strings) + 1, range_start));
                ldap_memfree(range_attr);
                *more_strings = False;
                return NULL;
@@ -1697,9 +2021,8 @@ char **ads_pull_strings_range(ADS_STRUCT *ads,
                return NULL;
        }
 
-       strings = talloc_realloc(mem_ctx, current_strings,
-                                sizeof(*current_strings) *
-                                (*num_strings + num_new_strings));
+       strings = TALLOC_REALLOC_ARRAY(mem_ctx, current_strings, char *,
+                                *num_strings + num_new_strings);
        
        if (strings == NULL) {
                ldap_memfree(range_attr);
@@ -1707,15 +2030,18 @@ char **ads_pull_strings_range(ADS_STRUCT *ads,
                return NULL;
        }
        
-       memcpy(&strings[*num_strings], new_strings,
-              sizeof(*new_strings) * num_new_strings);
+       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,
-                                                 "member;range=%d-*", 
-                                                 *num_strings);
+                                                 "%s;range=%d-*", 
+                                                 field,
+                                                 (int)*num_strings);
                
                if (!*next_attribute) {
                        DEBUG(1, ("talloc_asprintf for next attribute failed!\n"));
@@ -1738,8 +2064,8 @@ char **ads_pull_strings_range(ADS_STRUCT *ads,
  * @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)
+ BOOL ads_pull_uint32(ADS_STRUCT *ads, LDAPMessage *msg, const char *field,
+                     uint32 *v)
 {
        char **values;
 
@@ -1763,8 +2089,7 @@ BOOL ads_pull_uint32(ADS_STRUCT *ads,
  * @param guid 37-byte area to receive text guid
  * @return boolean indicating success
  **/
-BOOL ads_pull_guid(ADS_STRUCT *ads,
-                  void *msg, struct uuid *guid)
+ BOOL ads_pull_guid(ADS_STRUCT *ads, LDAPMessage *msg, struct GUID *guid)
 {
        char **values;
        UUID_FLAT flat_guid;
@@ -1793,8 +2118,8 @@ BOOL ads_pull_guid(ADS_STRUCT *ads,
  * @param sid Pointer to sid to store result
  * @return boolean inidicating success
 */
-BOOL ads_pull_sid(ADS_STRUCT *ads, 
-                 void *msg, const char *field, DOM_SID *sid)
+ BOOL ads_pull_sid(ADS_STRUCT *ads, LDAPMessage *msg, const char *field,
+                  DOM_SID *sid)
 {
        struct berval **values;
        BOOL ret = False;
@@ -1820,8 +2145,8 @@ BOOL ads_pull_sid(ADS_STRUCT *ads,
  * @param sids pointer to sid array to allocate
  * @return the count of SIDs pulled
  **/
-int ads_pull_sids(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
-                 void *msg, const char *field, DOM_SID **sids)
+ int ads_pull_sids(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
+                  LDAPMessage *msg, const char *field, DOM_SID **sids)
 {
        struct berval **values;
        BOOL ret;
@@ -1835,7 +2160,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;
@@ -1864,8 +2189,8 @@ int ads_pull_sids(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
  * @param sd Pointer to *SEC_DESC to store result (talloc()ed)
  * @return boolean inidicating success
 */
-BOOL ads_pull_sd(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
-                 void *msg, const char *field, SEC_DESC **sd)
+ BOOL ads_pull_sd(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
+                 LDAPMessage *msg, const char *field, SEC_DESC **sd)
 {
        struct berval **values;
        prs_struct      ps;
@@ -1897,15 +2222,22 @@ BOOL ads_pull_sd(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
  * @param msg Results of search
  * @return the username
  */
-char *ads_pull_username(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, void *msg)
+ char *ads_pull_username(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
+                        LDAPMessage *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");
 }
 
@@ -1921,17 +2253,22 @@ ADS_STATUS ads_USN(ADS_STRUCT *ads, uint32 *usn)
 {
        const char *attrs[] = {"highestCommittedUSN", NULL};
        ADS_STATUS status;
-       void *res;
+       LDAPMessage *res;
 
        status = ads_do_search_retry(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
        if (!ADS_ERR_OK(status)) 
                return status;
 
        if (ads_count_replies(ads, res) != 1) {
+               ads_msgfree(ads, res);
                return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
        }
 
-       ads_pull_uint32(ads, res, "highestCommittedUSN", usn);
+       if (!ads_pull_uint32(ads, res, "highestCommittedUSN", usn)) {
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_SUCH_ATTRIBUTE);
+       }
+
        ads_msgfree(ads, res);
        return ADS_SUCCESS;
 }
@@ -1956,85 +2293,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;
+       LDAPMessage *res;
        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_s, 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);
+
+       return status;
+}
 
-       *p = 0;
+/********************************************************************
+********************************************************************/
 
-       SAFE_FREE(ads->config.realm);
-       SAFE_FREE(ads->config.bind_path);
+ADS_STATUS ads_domain_func_level(ADS_STRUCT *ads, uint32 *val)
+{
+       const char *attrs[] = {"domainFunctionality", NULL};
+       ADS_STATUS status;
+       LDAPMessage *res;
+       ADS_STRUCT *ads_s = ads;
+       
+       *val = DS_DOMAIN_FUNCTION_2000;
 
-       ads->config.realm = strdup(p+2);
-       ads->config.bind_path = ads_build_dn(ads->config.realm);
+        /* establish a new ldap tcp session if necessary */
 
-       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));
+       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;
+       }
 
-       ads->config.current_time = ads_parse_time(timestr);
+       /* 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;
+       }
 
-       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 ( !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));
 
-       talloc_destroy(ctx);
+       
+       ads_msgfree(ads, res);
 
-       return ADS_SUCCESS;
+done:
+       /* free any temporary ads connections */
+       if ( ads_s != ads ) {
+               ads_destroy( &ads_s );
+       }
+
+       return status;
 }
 
 /**
@@ -2046,13 +2421,14 @@ ADS_STATUS ads_server_info(ADS_STRUCT *ads)
 ADS_STATUS ads_domain_sid(ADS_STRUCT *ads, DOM_SID *sid)
 {
        const char *attrs[] = {"objectSid", NULL};
-       void *res;
+       LDAPMessage *res;
        ADS_STATUS rc;
 
        rc = ads_do_search_retry(ads, ads->config.bind_path, LDAP_SCOPE_BASE, "(objectclass=*)", 
                           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);
@@ -2060,75 +2436,565 @@ 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;
+       LDAPMessage *res;
+       const char *dn, *service_name;
+       const char *attrs[] = { "dsServiceName", NULL };
 
-   NOTE! better method is this:
+       status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
 
-bin/net -Uadministrator%XXXXX ads search '(&(objectclass=crossref)(dnsroot=VNET3.HOME.SAMBA.ORG))'  nETBIOSName 
+       service_name = ads_pull_string(ads, mem_ctx, res, "dsServiceName");
+       if (service_name == NULL) {
+               return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
+       }
 
-but you need to force the bind path to match the configurationNamingContext from the rootDSE
+       /* 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);
+       }
 
-*/
-ADS_STATUS ads_workgroup_name(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, const char **workgroup)
+       *site_name = talloc_strdup(mem_ctx, dn);
+       if (*site_name == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       ads_msgfree(ads, res);
+
+       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;
-       void *res;
-       const char *attrs[] = {"servicePrincipalName", NULL};
-       int num_principals;
+       ADS_STATUS status;
+       LDAPMessage *res;
+       const char *parent, *config_context, *filter;
+       const char *attrs[] = { "configurationNamingContext", NULL };
+       char *dn;
+
+       /* shortcut a query */
+       if (strequal(computer_name, ads->config.ldap_server_name)) {
+               return ads_site_dn(ads, mem_ctx, site_dn);
+       }
+
+       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);
+       }
+
+       filter = talloc_asprintf(mem_ctx, "(cn=%s)", computer_name);
+       if (filter == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
-       (*workgroup) = NULL;
+       status = ads_do_search(ads, config_context, LDAP_SCOPE_SUBTREE, filter, NULL, &res);
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
 
-       asprintf(&expr, "(&(objectclass=computer)(dnshostname=%s.%s))", 
-                ads->config.ldap_server_name, ads->config.realm);
-       rc = ads_search(ads, &res, expr, attrs);
-       free(expr);
+       if (ads_count_replies(ads, res) != 1) {
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
 
-       if (!ADS_ERR_OK(rc)) {
-               return rc;
+       dn = ads_get_dn(ads, res);
+       if (dn == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
        }
 
-       principles = ads_pull_strings(ads, mem_ctx, res,
-                                     "servicePrincipalName", &num_principals);
+       /* 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;
+       LDAPMessage *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;
        }
 
-       asprintf(&prefix, "HOST/%s.%s/", 
-                ads->config.ldap_server_name, 
-                ads->config.realm);
+       config_context = ads_pull_string(ads, mem_ctx, res, "configurationNamingContext");
+       if (config_context == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
-       prefix_length = strlen(prefix);
+       base = talloc_asprintf(mem_ctx, "cn=Partitions,%s", config_context);
+       if (base == NULL) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
 
-       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;
+       status = ads_search_dn(ads, &res, base, attrs2); 
+       if (!ADS_ERR_OK(status)) {
+               return status;
+       }
+
+       if (ads_count_replies(ads, res) != 1) {
+               return ADS_ERROR(LDAP_NO_SUCH_OBJECT);
+       }
+
+       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, 
+                                  LDAPMessage *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;
+       
+       status = ads_find_machine_acct(ads, &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 ADS_SUCCESS;
+       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, &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, &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;
+}
+
+#if 0
+
+   SAVED CODE - we used to join via ldap - remember how we did this. JRA.
+
+/**
+ * Join a machine to a realm
+ *  Creates the machine account and sets the machine password
+ * @param ads connection to ads server
+ * @param machine 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 *machine_name,
+                       uint32 account_type, const char *org_unit)
+{
+       ADS_STATUS status;
+       LDAPMessage *res = NULL;
+       char *machine;
+
+       /* machine name must be lowercase */
+       machine = SMB_STRDUP(machine_name);
+       strlower_m(machine);
+
+       /*
+       status = ads_find_machine_acct(ads, (void **)&res, machine);
+       if (ADS_ERR_OK(status) && ads_count_replies(ads, res) == 1) {
+               DEBUG(0, ("Host account for %s already exists - deleting old account\n", machine));
+               status = ads_leave_realm(ads, machine);
+               if (!ADS_ERR_OK(status)) {
+                       DEBUG(0, ("Failed to delete host '%s' from the '%s' realm.\n",
+                               machine, ads->config.realm));
+                       return status;
+               }
+       }
+       */
+       status = ads_add_machine_acct(ads, machine, account_type, org_unit);
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(0, ("ads_join_realm: ads_add_machine_acct failed (%s): %s\n", machine, ads_errstr(status)));
+               SAFE_FREE(machine);
+               return status;
+       }
+
+       status = ads_find_machine_acct(ads, (void **)(void *)&res, machine);
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(0, ("ads_join_realm: Host account test failed for machine %s\n", machine));
+               SAFE_FREE(machine);
+               return status;
+       }
+
+       SAFE_FREE(machine);
+       ads_msgfree(ads, res);
+
+       return status;
+}
+#endif
+
+#ifdef HAVE_LDAP
+
+/**
+ * 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 *msg;
+       LDAPMessage *res;
+       char *hostnameDN, *host;
+       int rc;
+       LDAPControl ldap_control;
+       LDAPControl  * pldap_control[2] = {NULL, NULL};
+
+       pldap_control[0] = &ldap_control;
+       memset(&ldap_control, 0, sizeof(LDAPControl));
+       ldap_control.ldctl_oid = (char *)LDAP_SERVER_TREE_DELETE_OID;
+
+       /* hostname must be lowercase */
+       host = SMB_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));
+               SAFE_FREE(host);
+               return status;
+       }
+
+       msg = ads_first_entry(ads, res);
+       if (!msg) {
+               SAFE_FREE(host);
+               return ADS_ERROR_SYSTEM(ENOENT);
+       }
+
+       hostnameDN = ads_get_dn(ads, (LDAPMessage *)msg);
+
+       rc = ldap_delete_ext_s(ads->ld, hostnameDN, pldap_control, NULL);
+       if (rc) {
+               DEBUG(3,("ldap_delete_ext_s failed with error code %d\n", rc));
+       }else {
+               DEBUG(3,("ldap_delete_ext_s succeeded with error code %d\n", rc));
+       }
+
+       if (rc != LDAP_SUCCESS) {
+               const char *attrs[] = { "cn", NULL };
+               LDAPMessage *msg_sub;
+
+               /* we only search with scope ONE, we do not expect any further
+                * objects to be created deeper */
+
+               status = ads_do_search_retry(ads, hostnameDN,
+                                            LDAP_SCOPE_ONELEVEL,
+                                            "(objectclass=*)", attrs, &res);
+
+               if (!ADS_ERR_OK(status)) {
+                       SAFE_FREE(host);
+                       ads_memfree(ads, hostnameDN);
+                       return status;
+               }
+
+               for (msg_sub = ads_first_entry(ads, res); msg_sub;
+                       msg_sub = ads_next_entry(ads, msg_sub)) {
+
+                       char *dn = NULL;
+
+                       if ((dn = ads_get_dn(ads, msg_sub)) == NULL) {
+                               SAFE_FREE(host);
+                               ads_memfree(ads, hostnameDN);
+                               return ADS_ERROR(LDAP_NO_MEMORY);
+                       }
+
+                       status = ads_del_dn(ads, dn);
+                       if (!ADS_ERR_OK(status)) {
+                               DEBUG(3,("failed to delete dn %s: %s\n", dn, ads_errstr(status)));
+                               SAFE_FREE(host);
+                               ads_memfree(ads, dn);
+                               ads_memfree(ads, hostnameDN);
+                               return status;
+                       }
+
+                       ads_memfree(ads, dn);
+               }
+
+               /* there should be no subordinate objects anymore */
+               status = ads_do_search_retry(ads, hostnameDN,
+                                            LDAP_SCOPE_ONELEVEL,
+                                            "(objectclass=*)", attrs, &res);
+
+               if (!ADS_ERR_OK(status) || ( (ads_count_replies(ads, res)) > 0 ) ) {
+                       SAFE_FREE(host);
+                       ads_memfree(ads, hostnameDN);
+                       return status;
+               }
+
+               /* delete hostnameDN now */
+               status = ads_del_dn(ads, hostnameDN);
+               if (!ADS_ERR_OK(status)) {
+                       SAFE_FREE(host);
+                       DEBUG(3,("failed to delete dn %s: %s\n", hostnameDN, ads_errstr(status)));
+                       ads_memfree(ads, hostnameDN);
+                       return status;
+               }
+       }
+
+       ads_memfree(ads, hostnameDN);
+
+       status = ads_find_machine_acct(ads, &res, host);
+       if (ADS_ERR_OK(status) && ads_count_replies(ads, res) == 1) {
+               DEBUG(3, ("Failed to remove host account.\n"));
+               SAFE_FREE(host);
+               return status;
+       }
+
+       SAFE_FREE(host);
+       return status;
+}
+#endif
+
 #endif