Update my copyrights according to my agreement with IBM
[bbaumbach/samba-autobuild/.git] / source3 / libads / ldap_utils.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Some Helpful wrappers on LDAP 
5
6    Copyright (C) Andrew Tridgell 2001
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24
25 #ifdef HAVE_LDAP
26 /*
27   a wrapper around ldap_search_s that retries depending on the error code
28   this is supposed to catch dropped connections and auto-reconnect
29 */
30 ADS_STATUS ads_do_search_retry(ADS_STRUCT *ads, const char *bind_path, int scope, 
31                                const char *expr,
32                                const char **attrs, void **res)
33 {
34         ADS_STATUS status;
35         int count = 3;
36         char *bp;
37
38         if (!ads->ld &&
39             time(NULL) - ads->last_attempt < ADS_RECONNECT_TIME) {
40                 return ADS_ERROR(LDAP_SERVER_DOWN);
41         }
42
43         bp = strdup(bind_path);
44
45         if (!bp) 
46                 return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
47
48         while (count--) {
49                 status = ads_do_search_all(ads, bp, scope, expr, attrs, res);
50                 if (ADS_ERR_OK(status)) {
51                         DEBUG(5,("Search for %s gave %d replies\n",
52                                  expr, ads_count_replies(ads, *res)));
53                         free(bp);
54                         return status;
55                 }
56
57                 if (*res) ads_msgfree(ads, *res);
58                 *res = NULL;
59                 DEBUG(3,("Reopening ads connection to realm '%s' after error %s\n", 
60                          ads->config.realm, ads_errstr(status)));
61                 if (ads->ld) {
62                         ldap_unbind(ads->ld); 
63                 }
64                 ads->ld = NULL;
65                 status = ads_connect(ads);
66                 if (!ADS_ERR_OK(status)) {
67                         DEBUG(1,("ads_search_retry: failed to reconnect (%s)\n",
68                                  ads_errstr(status)));
69                         ads_destroy(&ads);
70                         free(bp);
71                         return status;
72                 }
73         }
74         free(bp);
75
76         DEBUG(1,("ads reopen failed after error %s\n", ads_errstr(status)));
77         return status;
78 }
79
80
81 ADS_STATUS ads_search_retry(ADS_STRUCT *ads, void **res, 
82                             const char *expr, 
83                             const char **attrs)
84 {
85         return ads_do_search_retry(ads, ads->config.bind_path, LDAP_SCOPE_SUBTREE,
86                                    expr, attrs, res);
87 }
88
89 ADS_STATUS ads_search_retry_dn(ADS_STRUCT *ads, void **res, 
90                                const char *dn, 
91                                const char **attrs)
92 {
93         return ads_do_search_retry(ads, dn, LDAP_SCOPE_BASE,
94                                    "(objectclass=*)", attrs, res);
95 }
96 #endif