[GLUE] Rsync SAMBA_3_2_0 SVN r25598 in order to create the v3-2-test branch.
[samba.git] / source / libads / ldap_utils.c
index 4392cde00bf5d1cd19b824ab5c37040300c898eb..18f7da81d71bcb5dcaf54b5cdafbd930a2ca8b7f 100644 (file)
@@ -4,10 +4,11 @@
    Some Helpful wrappers on LDAP 
 
    Copyright (C) Andrew Tridgell 2001
+   Copyright (C) Guenther Deschner 2006,2007
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
@@ -16,8 +17,7 @@
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
@@ -38,22 +38,30 @@ static ADS_STATUS ads_do_search_retry_internal(ADS_STRUCT *ads, const char *bind
 
        *res = NULL;
 
-       if (!ads->ld &&
-           time(NULL) - ads->last_attempt < ADS_RECONNECT_TIME) {
+       if (!ads->ldap.ld &&
+           time(NULL) - ads->ldap.last_attempt < ADS_RECONNECT_TIME) {
                return ADS_ERROR(LDAP_SERVER_DOWN);
        }
 
        bp = SMB_STRDUP(bind_path);
 
        if (!bp) {
-               return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
+               return ADS_ERROR(LDAP_NO_MEMORY);
        }
 
        *res = NULL;
-       status = ads_do_search_all_args(ads, bp, scope, expr, attrs, args, res);
+
+       /* when binding anonymously, we cannot use the paged search LDAP
+        * control - Guenther */
+
+       if (ads->auth.flags & ADS_AUTH_ANON_BIND) {
+               status = ads_do_search(ads, bp, scope, expr, attrs, res);
+       } else {
+               status = ads_do_search_all_args(ads, bp, scope, expr, attrs, args, res);
+       }
        if (ADS_ERR_OK(status)) {
-               DEBUG(5,("Search for %s gave %d replies\n",
-                        expr, ads_count_replies(ads, *res)));
+               DEBUG(5,("Search for %s in <%s> gave %d replies\n",
+                        expr, bp, ads_count_replies(ads, *res)));
                SAFE_FREE(bp);
                return status;
        }
@@ -67,11 +75,7 @@ static ADS_STATUS ads_do_search_retry_internal(ADS_STRUCT *ads, const char *bind
                DEBUG(3,("Reopening ads connection to realm '%s' after error %s\n", 
                         ads->config.realm, ads_errstr(status)));
                         
-               if (ads->ld) {
-                       ldap_unbind(ads->ld); 
-               }
-               
-               ads->ld = NULL;
+               ads_disconnect(ads);
                status = ads_connect(ads);
                
                if (!ADS_ERR_OK(status)) {
@@ -83,20 +87,29 @@ static ADS_STATUS ads_do_search_retry_internal(ADS_STRUCT *ads, const char *bind
                }
 
                *res = NULL;
-               status = ads_do_search_all_args(ads, bp, scope, expr, attrs, args, res);
+
+               /* when binding anonymously, we cannot use the paged search LDAP
+                * control - Guenther */
+
+               if (ads->auth.flags & ADS_AUTH_ANON_BIND) {
+                       status = ads_do_search(ads, bp, scope, expr, attrs, res);
+               } else {
+                       status = ads_do_search_all_args(ads, bp, scope, expr, attrs, args, res);
+               }
+
                if (ADS_ERR_OK(status)) {
-                       DEBUG(5,("Search for %s gave %d replies\n",
-                                expr, ads_count_replies(ads, *res)));
+                       DEBUG(5,("Search for filter: %s, base: %s gave %d replies\n",
+                                expr, bp, ads_count_replies(ads, *res)));
                        SAFE_FREE(bp);
                        return status;
                }
        }
         SAFE_FREE(bp);
 
-       if (!ADS_ERR_OK(status))
+       if (!ADS_ERR_OK(status)) {
                DEBUG(1,("ads reopen failed after error %s\n", 
                         ads_errstr(status)));
-
+       }
        return status;
 }
 
@@ -144,6 +157,45 @@ static ADS_STATUS ads_do_search_retry_internal(ADS_STRUCT *ads, const char *bind
 
        return ads_do_search_retry_args(ads, dn, LDAP_SCOPE_BASE,
                                        "(objectclass=*)", attrs, &args, res);
+}
+
+ ADS_STATUS ads_search_retry_dn_sd_flags(ADS_STRUCT *ads, LDAPMessage **res, 
+                                        uint32 sd_flags,
+                                        const char *dn, 
+                                        const char **attrs)
+{
+       ads_control args;
+
+       args.control = ADS_SD_FLAGS_OID;
+       args.val = sd_flags;
+       args.critical = True;
+
+       return ads_do_search_retry_args(ads, dn, LDAP_SCOPE_BASE,
+                                       "(objectclass=*)", attrs, &args, res);
+}
+
+ ADS_STATUS ads_search_retry_extended_dn_ranged(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, 
+                                               const char *dn, 
+                                               const char **attrs,
+                                               enum ads_extended_dn_flags flags,
+                                               char ***strings,
+                                               size_t *num_strings)
+{
+       ads_control args;
+
+       args.control = ADS_EXTENDED_DN_OID;
+       args.val = flags;
+       args.critical = True;
+
+       /* we can only range process one attribute */
+       if (!attrs || !attrs[0] || attrs[1]) {
+               return ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
+       }
+
+       return ads_ranged_search(ads, mem_ctx, LDAP_SCOPE_BASE, dn, 
+                                "(objectclass=*)", &args, attrs[0],
+                                strings, num_strings);
+
 }
 
  ADS_STATUS ads_search_retry_sid(ADS_STRUCT *ads, LDAPMessage **res, 
@@ -170,4 +222,152 @@ static ADS_STATUS ads_do_search_retry_internal(ADS_STRUCT *ads, const char *bind
        return status;
 }
 
+ADS_STATUS ads_ranged_search(ADS_STRUCT *ads, 
+                            TALLOC_CTX *mem_ctx,
+                            int scope,
+                            const char *base,
+                            const char *filter,
+                            void *args,
+                            const char *range_attr,
+                            char ***strings,
+                            size_t *num_strings)
+{
+       ADS_STATUS status;
+       uint32 first_usn;
+       int num_retries = 0;
+       const char **attrs;
+       BOOL more_values = False;
+
+       *num_strings = 0;
+       *strings = NULL;
+
+       attrs = TALLOC_ARRAY(mem_ctx, const char *, 3);
+       ADS_ERROR_HAVE_NO_MEMORY(attrs);
+
+       attrs[0] = talloc_strdup(mem_ctx, range_attr);
+       attrs[1] = talloc_strdup(mem_ctx, "usnChanged");
+       attrs[2] = NULL;
+
+       ADS_ERROR_HAVE_NO_MEMORY(attrs[0]);
+       ADS_ERROR_HAVE_NO_MEMORY(attrs[1]);
+
+       do {
+               status = ads_ranged_search_internal(ads, mem_ctx, 
+                                                   scope, base, filter, 
+                                                   attrs, args, range_attr, 
+                                                   strings, num_strings,
+                                                   &first_usn, &num_retries, 
+                                                   &more_values);
+
+               if (NT_STATUS_EQUAL(STATUS_MORE_ENTRIES, ads_ntstatus(status))) {
+                       continue;
+               }
+
+               if (!ADS_ERR_OK(status)) {
+                       *num_strings = 0;
+                       strings = NULL;
+                       goto done;
+               }
+
+       } while (more_values);
+
+ done:
+       DEBUG(10,("returning with %d strings\n", (int)*num_strings));
+
+       return status;
+}
+
+ADS_STATUS ads_ranged_search_internal(ADS_STRUCT *ads, 
+                                     TALLOC_CTX *mem_ctx,
+                                     int scope,
+                                     const char *base,
+                                     const char *filter,
+                                     const char **attrs,
+                                     void *args,
+                                     const char *range_attr,
+                                     char ***strings,
+                                     size_t *num_strings,
+                                     uint32 *first_usn,
+                                     int *num_retries,
+                                     BOOL *more_values)
+{
+       LDAPMessage *res = NULL;
+       ADS_STATUS status;
+       int count;
+       uint32 current_usn;
+
+       DEBUG(10, ("Searching for attrs[0] = %s, attrs[1] = %s\n", attrs[0], attrs[1]));
+
+       *more_values = False;
+
+       status = ads_do_search_retry_internal(ads, base, scope, filter, attrs, args, &res);
+
+       if (!ADS_ERR_OK(status)) {
+               DEBUG(1,("ads_search: %s\n",
+                        ads_errstr(status)));
+               return status;
+       }
+       
+       if (!res) {
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       count = ads_count_replies(ads, res);
+       if (count == 0) {
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_SUCCESS);
+       }
+
+       if (*num_strings == 0) {
+               if (!ads_pull_uint32(ads, res, "usnChanged", first_usn)) {
+                       DEBUG(1, ("could not pull first usnChanged!\n"));
+                       ads_msgfree(ads, res);
+                       return ADS_ERROR(LDAP_NO_MEMORY);
+               }
+       }
+
+       if (!ads_pull_uint32(ads, res, "usnChanged", &current_usn)) {
+               DEBUG(1, ("could not pull current usnChanged!\n"));
+               ads_msgfree(ads, res);
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       if (*first_usn != current_usn) {
+               DEBUG(5, ("USN on this record changed"
+                         " - restarting search\n"));
+               if (*num_retries < 5) {
+                       (*num_retries)++;
+                       *num_strings = 0;
+                       ads_msgfree(ads, res);
+                       return ADS_ERROR_NT(STATUS_MORE_ENTRIES);
+               } else {
+                       DEBUG(5, ("USN on this record changed"
+                                 " - restarted search too many times, aborting!\n"));
+                       ads_msgfree(ads, res);
+                       return ADS_ERROR(LDAP_NO_MEMORY);
+               }
+       }
+
+       *strings = ads_pull_strings_range(ads, mem_ctx, res,
+                                        range_attr,
+                                        *strings,
+                                        &attrs[0],
+                                        num_strings,
+                                        more_values);
+
+       ads_msgfree(ads, res);
+
+       /* paranoia checks */
+       if (*strings == NULL && *more_values) {
+               DEBUG(0,("no strings found but more values???\n"));
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+       if (*num_strings == 0 && *more_values) {
+               DEBUG(0,("no strings found but more values???\n"));
+               return ADS_ERROR(LDAP_NO_MEMORY);
+       }
+
+       return (*more_values) ? ADS_ERROR_NT(STATUS_MORE_ENTRIES) : ADS_ERROR(LDAP_SUCCESS);
+}
+
 #endif