Fix valgrind memory leak in bug #6814 - Fixes for problems reported by valgrind
[nivanova/samba-autobuild/.git] / source3 / lib / smbldap.c
1 /* 
2    Unix SMB/CIFS implementation.
3    LDAP protocol helper functions for SAMBA
4    Copyright (C) Jean Fran├žois Micouleau       1998
5    Copyright (C) Gerald Carter                  2001-2003
6    Copyright (C) Shahms King                    2001
7    Copyright (C) Andrew Bartlett                2002-2003
8    Copyright (C) Stefan (metze) Metzmacher      2002-2003
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22
23 */
24
25 #include "includes.h"
26 #include "smbldap.h"
27
28 #ifndef LDAP_OPT_SUCCESS
29 #define LDAP_OPT_SUCCESS 0
30 #endif
31
32 /* Try not to hit the up or down server forever */
33
34 #define SMBLDAP_DONT_PING_TIME 10       /* ping only all 10 seconds */
35 #define SMBLDAP_NUM_RETRIES 8           /* retry only 8 times */
36
37 #define SMBLDAP_IDLE_TIME 150           /* After 2.5 minutes disconnect */
38
39
40 /* attributes used by Samba 2.2 */
41
42 ATTRIB_MAP_ENTRY attrib_map_v22[] = {
43         { LDAP_ATTR_UID,                "uid"           },
44         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
45         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
46         { LDAP_ATTR_UNIX_HOME,          "homeDirectory" },
47         { LDAP_ATTR_PWD_LAST_SET,       "pwdLastSet"    },
48         { LDAP_ATTR_PWD_CAN_CHANGE,     "pwdCanChange"  },
49         { LDAP_ATTR_PWD_MUST_CHANGE,    "pwdMustChange" },
50         { LDAP_ATTR_LOGON_TIME,         "logonTime"     },
51         { LDAP_ATTR_LOGOFF_TIME,        "logoffTime"    },
52         { LDAP_ATTR_KICKOFF_TIME,       "kickoffTime"   },
53         { LDAP_ATTR_CN,                 "cn"            },
54         { LDAP_ATTR_SN,                 "sn"            },
55         { LDAP_ATTR_DISPLAY_NAME,       "displayName"   },
56         { LDAP_ATTR_HOME_PATH,          "smbHome"       },
57         { LDAP_ATTR_HOME_DRIVE,         "homeDrive"     },
58         { LDAP_ATTR_LOGON_SCRIPT,       "scriptPath"    },
59         { LDAP_ATTR_PROFILE_PATH,       "profilePath"   },
60         { LDAP_ATTR_DESC,               "description"   },
61         { LDAP_ATTR_USER_WKS,           "userWorkstations"},
62         { LDAP_ATTR_USER_RID,           "rid"           },
63         { LDAP_ATTR_PRIMARY_GROUP_RID,  "primaryGroupID"},
64         { LDAP_ATTR_LMPW,               "lmPassword"    },
65         { LDAP_ATTR_NTPW,               "ntPassword"    },
66         { LDAP_ATTR_DOMAIN,             "domain"        },
67         { LDAP_ATTR_OBJCLASS,           "objectClass"   },
68         { LDAP_ATTR_ACB_INFO,           "acctFlags"     },
69         { LDAP_ATTR_MOD_TIMESTAMP,      "modifyTimestamp"       },
70         { LDAP_ATTR_LIST_END,           NULL            }
71 };
72
73 ATTRIB_MAP_ENTRY attrib_map_to_delete_v22[] = {
74         { LDAP_ATTR_PWD_LAST_SET,       "pwdLastSet"    },
75         { LDAP_ATTR_PWD_CAN_CHANGE,     "pwdCanChange"  },
76         { LDAP_ATTR_PWD_MUST_CHANGE,    "pwdMustChange" },
77         { LDAP_ATTR_LOGON_TIME,         "logonTime"     },
78         { LDAP_ATTR_LOGOFF_TIME,        "logoffTime"    },
79         { LDAP_ATTR_KICKOFF_TIME,       "kickoffTime"   },
80         { LDAP_ATTR_DISPLAY_NAME,       "displayName"   },
81         { LDAP_ATTR_HOME_PATH,          "smbHome"       },
82         { LDAP_ATTR_HOME_DRIVE,         "homeDrives"    },
83         { LDAP_ATTR_LOGON_SCRIPT,       "scriptPath"    },
84         { LDAP_ATTR_PROFILE_PATH,       "profilePath"   },
85         { LDAP_ATTR_USER_WKS,           "userWorkstations"},
86         { LDAP_ATTR_USER_RID,           "rid"           },
87         { LDAP_ATTR_PRIMARY_GROUP_RID,  "primaryGroupID"},
88         { LDAP_ATTR_LMPW,               "lmPassword"    },
89         { LDAP_ATTR_NTPW,               "ntPassword"    },
90         { LDAP_ATTR_DOMAIN,             "domain"        },
91         { LDAP_ATTR_ACB_INFO,           "acctFlags"     },
92         { LDAP_ATTR_LIST_END,           NULL            }
93 };
94
95 /* attributes used by Samba 3.0's sambaSamAccount */
96
97 ATTRIB_MAP_ENTRY attrib_map_v30[] = {
98         { LDAP_ATTR_UID,                "uid"                   },
99         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
100         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
101         { LDAP_ATTR_UNIX_HOME,          "homeDirectory"         },
102         { LDAP_ATTR_PWD_LAST_SET,       "sambaPwdLastSet"       },
103         { LDAP_ATTR_PWD_CAN_CHANGE,     "sambaPwdCanChange"     },
104         { LDAP_ATTR_PWD_MUST_CHANGE,    "sambaPwdMustChange"    },
105         { LDAP_ATTR_LOGON_TIME,         "sambaLogonTime"        },
106         { LDAP_ATTR_LOGOFF_TIME,        "sambaLogoffTime"       },
107         { LDAP_ATTR_KICKOFF_TIME,       "sambaKickoffTime"      },
108         { LDAP_ATTR_CN,                 "cn"                    },
109         { LDAP_ATTR_SN,                 "sn"                    },
110         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
111         { LDAP_ATTR_HOME_DRIVE,         "sambaHomeDrive"        },
112         { LDAP_ATTR_HOME_PATH,          "sambaHomePath"         },
113         { LDAP_ATTR_LOGON_SCRIPT,       "sambaLogonScript"      },
114         { LDAP_ATTR_PROFILE_PATH,       "sambaProfilePath"      },
115         { LDAP_ATTR_DESC,               "description"           },
116         { LDAP_ATTR_USER_WKS,           "sambaUserWorkstations" },
117         { LDAP_ATTR_USER_SID,           LDAP_ATTRIBUTE_SID      },
118         { LDAP_ATTR_PRIMARY_GROUP_SID,  "sambaPrimaryGroupSID"  },
119         { LDAP_ATTR_LMPW,               "sambaLMPassword"       },
120         { LDAP_ATTR_NTPW,               "sambaNTPassword"       },
121         { LDAP_ATTR_DOMAIN,             "sambaDomainName"       },
122         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
123         { LDAP_ATTR_ACB_INFO,           "sambaAcctFlags"        },
124         { LDAP_ATTR_MUNGED_DIAL,        "sambaMungedDial"       },
125         { LDAP_ATTR_BAD_PASSWORD_COUNT, "sambaBadPasswordCount" },
126         { LDAP_ATTR_BAD_PASSWORD_TIME,  "sambaBadPasswordTime"  },
127         { LDAP_ATTR_PWD_HISTORY,        "sambaPasswordHistory"  },
128         { LDAP_ATTR_MOD_TIMESTAMP,      "modifyTimestamp"       },
129         { LDAP_ATTR_LOGON_HOURS,        "sambaLogonHours"       },
130         { LDAP_ATTR_LIST_END,           NULL                    }
131 };
132
133 ATTRIB_MAP_ENTRY attrib_map_to_delete_v30[] = {
134         { LDAP_ATTR_PWD_LAST_SET,       "sambaPwdLastSet"       },
135         { LDAP_ATTR_PWD_CAN_CHANGE,     "sambaPwdCanChange"     },
136         { LDAP_ATTR_PWD_MUST_CHANGE,    "sambaPwdMustChange"    },
137         { LDAP_ATTR_LOGON_TIME,         "sambaLogonTime"        },
138         { LDAP_ATTR_LOGOFF_TIME,        "sambaLogoffTime"       },
139         { LDAP_ATTR_KICKOFF_TIME,       "sambaKickoffTime"      },
140         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
141         { LDAP_ATTR_HOME_DRIVE,         "sambaHomeDrive"        },
142         { LDAP_ATTR_HOME_PATH,          "sambaHomePath"         },
143         { LDAP_ATTR_LOGON_SCRIPT,       "sambaLogonScript"      },
144         { LDAP_ATTR_PROFILE_PATH,       "sambaProfilePath"      },
145         { LDAP_ATTR_USER_WKS,           "sambaUserWorkstations" },
146         { LDAP_ATTR_USER_SID,           LDAP_ATTRIBUTE_SID      },
147         { LDAP_ATTR_PRIMARY_GROUP_SID,  "sambaPrimaryGroupSID"  },
148         { LDAP_ATTR_LMPW,               "sambaLMPassword"       },
149         { LDAP_ATTR_NTPW,               "sambaNTPassword"       },
150         { LDAP_ATTR_DOMAIN,             "sambaDomainName"       },
151         { LDAP_ATTR_ACB_INFO,           "sambaAcctFlags"        },
152         { LDAP_ATTR_MUNGED_DIAL,        "sambaMungedDial"       },
153         { LDAP_ATTR_BAD_PASSWORD_COUNT, "sambaBadPasswordCount" },
154         { LDAP_ATTR_BAD_PASSWORD_TIME,  "sambaBadPasswordTime"  },
155         { LDAP_ATTR_PWD_HISTORY,        "sambaPasswordHistory"  },
156         { LDAP_ATTR_LOGON_HOURS,        "sambaLogonHours"       },
157         { LDAP_ATTR_LIST_END,           NULL                    }
158 };
159
160 /* attributes used for allocating RIDs */
161
162 ATTRIB_MAP_ENTRY dominfo_attr_list[] = {
163         { LDAP_ATTR_DOMAIN,             "sambaDomainName"       },
164         { LDAP_ATTR_NEXT_RID,           "sambaNextRid"          },
165         { LDAP_ATTR_NEXT_USERRID,       "sambaNextUserRid"      },
166         { LDAP_ATTR_NEXT_GROUPRID,      "sambaNextGroupRid"     },
167         { LDAP_ATTR_DOM_SID,            LDAP_ATTRIBUTE_SID      },
168         { LDAP_ATTR_ALGORITHMIC_RID_BASE,"sambaAlgorithmicRidBase"},
169         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
170         { LDAP_ATTR_LIST_END,           NULL                    },
171 };
172
173 /* Samba 3.0 group mapping attributes */
174
175 ATTRIB_MAP_ENTRY groupmap_attr_list[] = {
176         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
177         { LDAP_ATTR_GROUP_SID,          LDAP_ATTRIBUTE_SID      },
178         { LDAP_ATTR_GROUP_TYPE,         "sambaGroupType"        },
179         { LDAP_ATTR_SID_LIST,           "sambaSIDList"          },
180         { LDAP_ATTR_DESC,               "description"           },
181         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
182         { LDAP_ATTR_CN,                 "cn"                    },
183         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
184         { LDAP_ATTR_LIST_END,           NULL                    }       
185 };
186
187 ATTRIB_MAP_ENTRY groupmap_attr_list_to_delete[] = {
188         { LDAP_ATTR_GROUP_SID,          LDAP_ATTRIBUTE_SID      },
189         { LDAP_ATTR_GROUP_TYPE,         "sambaGroupType"        },
190         { LDAP_ATTR_DESC,               "description"           },
191         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
192         { LDAP_ATTR_SID_LIST,           "sambaSIDList"          },
193         { LDAP_ATTR_LIST_END,           NULL                    }       
194 };
195
196 /* idmap_ldap sambaUnixIdPool */
197
198 ATTRIB_MAP_ENTRY idpool_attr_list[] = {
199         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
200         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
201         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
202         { LDAP_ATTR_LIST_END,           NULL                    }       
203 };
204
205 ATTRIB_MAP_ENTRY sidmap_attr_list[] = {
206         { LDAP_ATTR_SID,                LDAP_ATTRIBUTE_SID      },
207         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
208         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
209         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
210         { LDAP_ATTR_LIST_END,           NULL                    }       
211 };
212
213 /**********************************************************************
214  perform a simple table lookup and return the attribute name 
215  **********************************************************************/
216  
217  const char* get_attr_key2string( ATTRIB_MAP_ENTRY table[], int key )
218 {
219         int i = 0;
220
221         while ( table[i].attrib != LDAP_ATTR_LIST_END ) {
222                 if ( table[i].attrib == key )
223                         return table[i].name;
224                 i++;
225         }
226
227         return NULL;
228 }
229
230
231 /**********************************************************************
232  Return the list of attribute names from a mapping table
233  **********************************************************************/
234
235  const char** get_attr_list( TALLOC_CTX *mem_ctx, ATTRIB_MAP_ENTRY table[] )
236 {
237         const char **names;
238         int i = 0;
239
240         while ( table[i].attrib != LDAP_ATTR_LIST_END )
241                 i++;
242         i++;
243
244         names = TALLOC_ARRAY( mem_ctx, const char*, i );
245         if ( !names ) {
246                 DEBUG(0,("get_attr_list: out of memory\n"));
247                 return NULL;
248         }
249
250         i = 0;
251         while ( table[i].attrib != LDAP_ATTR_LIST_END ) {
252                 names[i] = talloc_strdup( names, table[i].name );
253                 i++;
254         }
255         names[i] = NULL;
256
257         return names;
258 }
259
260 /*******************************************************************
261  Search an attribute and return the first value found.
262 ******************************************************************/
263
264  bool smbldap_get_single_attribute (LDAP * ldap_struct, LDAPMessage * entry,
265                                     const char *attribute, char *value,
266                                     int max_len)
267 {
268         char **values;
269
270         if ( !attribute )
271                 return False;
272
273         value[0] = '\0';
274
275         if ((values = ldap_get_values (ldap_struct, entry, attribute)) == NULL) {
276                 DEBUG (10, ("smbldap_get_single_attribute: [%s] = [<does not exist>]\n", attribute));
277
278                 return False;
279         }
280
281         if (convert_string(CH_UTF8, CH_UNIX,values[0], -1, value, max_len, False) == (size_t)-1) {
282                 DEBUG(1, ("smbldap_get_single_attribute: string conversion of [%s] = [%s] failed!\n", 
283                           attribute, values[0]));
284                 ldap_value_free(values);
285                 return False;
286         }
287
288         ldap_value_free(values);
289 #ifdef DEBUG_PASSWORDS
290         DEBUG (100, ("smbldap_get_single_attribute: [%s] = [%s]\n", attribute, value));
291 #endif  
292         return True;
293 }
294
295  char * smbldap_talloc_single_attribute(LDAP *ldap_struct, LDAPMessage *entry,
296                                         const char *attribute,
297                                         TALLOC_CTX *mem_ctx)
298 {
299         char **values;
300         char *result;
301         size_t converted_size;
302
303         if (attribute == NULL) {
304                 return NULL;
305         }
306
307         values = ldap_get_values(ldap_struct, entry, attribute);
308
309         if (values == NULL) {
310                 DEBUG(10, ("attribute %s does not exist\n", attribute));
311                 return NULL;
312         }
313
314         if (ldap_count_values(values) != 1) {
315                 DEBUG(10, ("attribute %s has %d values, expected only one\n",
316                            attribute, ldap_count_values(values)));
317                 ldap_value_free(values);
318                 return NULL;
319         }
320
321         if (!pull_utf8_talloc(mem_ctx, &result, values[0], &converted_size)) {
322                 DEBUG(10, ("pull_utf8_talloc failed\n"));
323                 ldap_value_free(values);
324                 return NULL;
325         }
326
327         ldap_value_free(values);
328
329 #ifdef DEBUG_PASSWORDS
330         DEBUG (100, ("smbldap_get_single_attribute: [%s] = [%s]\n",
331                      attribute, result));
332 #endif  
333         return result;
334 }
335
336  char * smbldap_talloc_smallest_attribute(LDAP *ldap_struct, LDAPMessage *entry,
337                                           const char *attribute,
338                                           TALLOC_CTX *mem_ctx)
339 {
340         char **values;
341         char *result;
342         size_t converted_size;
343         int i, num_values;
344
345         if (attribute == NULL) {
346                 return NULL;
347         }
348
349         values = ldap_get_values(ldap_struct, entry, attribute);
350
351         if (values == NULL) {
352                 DEBUG(10, ("attribute %s does not exist\n", attribute));
353                 return NULL;
354         }
355
356         if (!pull_utf8_talloc(mem_ctx, &result, values[0], &converted_size)) {
357                 DEBUG(10, ("pull_utf8_talloc failed\n"));
358                 ldap_value_free(values);
359                 return NULL;
360         }
361
362         num_values = ldap_count_values(values);
363
364         for (i=1; i<num_values; i++) {
365                 char *tmp;
366
367                 if (!pull_utf8_talloc(mem_ctx, &tmp, values[i],
368                                       &converted_size)) {
369                         DEBUG(10, ("pull_utf8_talloc failed\n"));
370                         TALLOC_FREE(result);
371                         ldap_value_free(values);
372                         return NULL;
373                 }
374
375                 if (StrCaseCmp(tmp, result) < 0) {
376                         TALLOC_FREE(result);
377                         result = tmp;
378                 } else {
379                         TALLOC_FREE(tmp);
380                 }
381         }
382
383         ldap_value_free(values);
384
385 #ifdef DEBUG_PASSWORDS
386         DEBUG (100, ("smbldap_get_single_attribute: [%s] = [%s]\n",
387                      attribute, result));
388 #endif
389         return result;
390 }
391
392  bool smbldap_talloc_single_blob(TALLOC_CTX *mem_ctx, LDAP *ld,
393                                  LDAPMessage *msg, const char *attrib,
394                                  DATA_BLOB *blob)
395 {
396         struct berval **values;
397
398         values = ldap_get_values_len(ld, msg, attrib);
399         if (!values) {
400                 return false;
401         }
402
403         if (ldap_count_values_len(values) != 1) {
404                 DEBUG(10, ("Expected one value for %s, got %d\n", attrib,
405                            ldap_count_values_len(values)));
406                 return false;
407         }
408
409         *blob = data_blob_talloc(mem_ctx, values[0]->bv_val,
410                                  values[0]->bv_len);
411         ldap_value_free_len(values);
412
413         return (blob->data != NULL);
414 }
415
416  bool smbldap_pull_sid(LDAP *ld, LDAPMessage *msg, const char *attrib,
417                        struct dom_sid *sid)
418 {
419         DATA_BLOB blob;
420         bool ret;
421
422         if (!smbldap_talloc_single_blob(talloc_tos(), ld, msg, attrib,
423                                         &blob)) {
424                 return false;
425         }
426         ret = sid_parse((char *)blob.data, blob.length, sid);
427         TALLOC_FREE(blob.data);
428         return ret;
429 }
430
431  static int ldapmsg_destructor(LDAPMessage **result) {
432         ldap_msgfree(*result);
433         return 0;
434 }
435
436  void talloc_autofree_ldapmsg(TALLOC_CTX *mem_ctx, LDAPMessage *result)
437 {
438         LDAPMessage **handle;
439
440         if (result == NULL) {
441                 return;
442         }
443
444         handle = TALLOC_P(mem_ctx, LDAPMessage *);
445         SMB_ASSERT(handle != NULL);
446
447         *handle = result;
448         talloc_set_destructor(handle, ldapmsg_destructor);
449 }
450
451  static int ldapmod_destructor(LDAPMod ***mod) {
452         ldap_mods_free(*mod, True);
453         return 0;
454 }
455
456  void talloc_autofree_ldapmod(TALLOC_CTX *mem_ctx, LDAPMod **mod)
457 {
458         LDAPMod ***handle;
459
460         if (mod == NULL) {
461                 return;
462         }
463
464         handle = TALLOC_P(mem_ctx, LDAPMod **);
465         SMB_ASSERT(handle != NULL);
466
467         *handle = mod;
468         talloc_set_destructor(handle, ldapmod_destructor);
469 }
470
471 /************************************************************************
472  Routine to manage the LDAPMod structure array
473  manage memory used by the array, by each struct, and values
474  ***********************************************************************/
475
476  void smbldap_set_mod (LDAPMod *** modlist, int modop, const char *attribute, const char *value)
477 {
478         LDAPMod **mods;
479         int i;
480         int j;
481
482         mods = *modlist;
483
484         /* sanity checks on the mod values */
485
486         if (attribute == NULL || *attribute == '\0') {
487                 return; 
488         }
489
490 #if 0   /* commented out after discussion with abartlet.  Do not reenable.
491            left here so other do not re-add similar code   --jerry */
492         if (value == NULL || *value == '\0')
493                 return;
494 #endif
495
496         if (mods == NULL) {
497                 mods = SMB_MALLOC_P(LDAPMod *);
498                 if (mods == NULL) {
499                         smb_panic("smbldap_set_mod: out of memory!");
500                         /* notreached. */
501                 }
502                 mods[0] = NULL;
503         }
504
505         for (i = 0; mods[i] != NULL; ++i) {
506                 if (mods[i]->mod_op == modop && strequal(mods[i]->mod_type, attribute))
507                         break;
508         }
509
510         if (mods[i] == NULL) {
511                 mods = SMB_REALLOC_ARRAY (mods, LDAPMod *, i + 2);
512                 if (mods == NULL) {
513                         smb_panic("smbldap_set_mod: out of memory!");
514                         /* notreached. */
515                 }
516                 mods[i] = SMB_MALLOC_P(LDAPMod);
517                 if (mods[i] == NULL) {
518                         smb_panic("smbldap_set_mod: out of memory!");
519                         /* notreached. */
520                 }
521                 mods[i]->mod_op = modop;
522                 mods[i]->mod_values = NULL;
523                 mods[i]->mod_type = SMB_STRDUP(attribute);
524                 mods[i + 1] = NULL;
525         }
526
527         if (value != NULL) {
528                 char *utf8_value = NULL;
529                 size_t converted_size;
530
531                 j = 0;
532                 if (mods[i]->mod_values != NULL) {
533                         for (; mods[i]->mod_values[j] != NULL; j++);
534                 }
535                 mods[i]->mod_values = SMB_REALLOC_ARRAY(mods[i]->mod_values, char *, j + 2);
536
537                 if (mods[i]->mod_values == NULL) {
538                         smb_panic("smbldap_set_mod: out of memory!");
539                         /* notreached. */
540                 }
541
542                 if (!push_utf8_talloc(talloc_tos(), &utf8_value, value, &converted_size)) {
543                         smb_panic("smbldap_set_mod: String conversion failure!");
544                         /* notreached. */
545                 }
546
547                 mods[i]->mod_values[j] = SMB_STRDUP(utf8_value);
548                 TALLOC_FREE(utf8_value);
549                 SMB_ASSERT(mods[i]->mod_values[j] != NULL);
550
551                 mods[i]->mod_values[j + 1] = NULL;
552         }
553         *modlist = mods;
554 }
555
556 /**********************************************************************
557   Set attribute to newval in LDAP, regardless of what value the
558   attribute had in LDAP before.
559 *********************************************************************/
560
561  void smbldap_make_mod(LDAP *ldap_struct, LDAPMessage *existing,
562                       LDAPMod ***mods,
563                       const char *attribute, const char *newval)
564 {
565         char oldval[2048]; /* current largest allowed value is mungeddial */
566         bool existed;
567
568         if (attribute == NULL) {
569                 /* This can actually happen for ldapsam_compat where we for
570                  * example don't have a password history */
571                 return;
572         }
573
574         if (existing != NULL) {
575                 existed = smbldap_get_single_attribute(ldap_struct, existing, attribute, oldval, sizeof(oldval));
576         } else {
577                 existed = False;
578                 *oldval = '\0';
579         }
580
581         /* all of our string attributes are case insensitive */
582
583         if (existed && newval && (StrCaseCmp(oldval, newval) == 0)) {
584
585                 /* Believe it or not, but LDAP will deny a delete and
586                    an add at the same time if the values are the
587                    same... */
588                 DEBUG(10,("smbldap_make_mod: attribute |%s| not changed.\n", attribute));
589                 return;
590         }
591
592         if (existed) {
593                 /* There has been no value before, so don't delete it.
594                  * Here's a possible race: We might end up with
595                  * duplicate attributes */
596                 /* By deleting exactly the value we found in the entry this
597                  * should be race-free in the sense that the LDAP-Server will
598                  * deny the complete operation if somebody changed the
599                  * attribute behind our back. */
600                 /* This will also allow modifying single valued attributes 
601                  * in Novell NDS. In NDS you have to first remove attribute and then
602                  * you could add new value */
603
604                 DEBUG(10,("smbldap_make_mod: deleting attribute |%s| values |%s|\n", attribute, oldval));
605                 smbldap_set_mod(mods, LDAP_MOD_DELETE, attribute, oldval);
606         }
607
608         /* Regardless of the real operation (add or modify)
609            we add the new value here. We rely on deleting
610            the old value, should it exist. */
611
612         if ((newval != NULL) && (strlen(newval) > 0)) {
613                 DEBUG(10,("smbldap_make_mod: adding attribute |%s| value |%s|\n", attribute, newval));
614                 smbldap_set_mod(mods, LDAP_MOD_ADD, attribute, newval);
615         }
616 }
617
618 /**********************************************************************
619  Some varients of the LDAP rebind code do not pass in the third 'arg' 
620  pointer to a void*, so we try and work around it by assuming that the 
621  value of the 'LDAP *' pointer is the same as the one we had passed in
622  **********************************************************************/
623
624 struct smbldap_state_lookup {
625         LDAP *ld;
626         struct smbldap_state *smbldap_state;
627         struct smbldap_state_lookup *prev, *next;
628 };
629
630 static struct smbldap_state_lookup *smbldap_state_lookup_list;
631
632 static struct smbldap_state *smbldap_find_state(LDAP *ld) 
633 {
634         struct smbldap_state_lookup *t;
635
636         for (t = smbldap_state_lookup_list; t; t = t->next) {
637                 if (t->ld == ld) {
638                         return t->smbldap_state;
639                 }
640         }
641         return NULL;
642 }
643
644 static void smbldap_delete_state(struct smbldap_state *smbldap_state) 
645 {
646         struct smbldap_state_lookup *t;
647
648         for (t = smbldap_state_lookup_list; t; t = t->next) {
649                 if (t->smbldap_state == smbldap_state) {
650                         DLIST_REMOVE(smbldap_state_lookup_list, t);
651                         SAFE_FREE(t);
652                         return;
653                 }
654         }
655 }
656
657 static void smbldap_store_state(LDAP *ld, struct smbldap_state *smbldap_state) 
658 {
659         struct smbldap_state *tmp_ldap_state;
660         struct smbldap_state_lookup *t;
661
662         if ((tmp_ldap_state = smbldap_find_state(ld))) {
663                 SMB_ASSERT(tmp_ldap_state == smbldap_state);
664                 return;
665         }
666
667         t = SMB_XMALLOC_P(struct smbldap_state_lookup);
668         ZERO_STRUCTP(t);
669
670         DLIST_ADD_END(smbldap_state_lookup_list, t, struct smbldap_state_lookup *);
671         t->ld = ld;
672         t->smbldap_state = smbldap_state;
673 }
674
675 /********************************************************************
676  start TLS on an existing LDAP connection
677 *******************************************************************/
678
679 int smb_ldap_start_tls(LDAP *ldap_struct, int version)
680
681 #ifdef LDAP_OPT_X_TLS
682         int rc;
683 #endif
684
685         if (lp_ldap_ssl() != LDAP_SSL_START_TLS) {
686                 return LDAP_SUCCESS;
687         }
688
689 #ifdef LDAP_OPT_X_TLS
690         if (version != LDAP_VERSION3) {
691                 DEBUG(0, ("Need LDAPv3 for Start TLS\n"));
692                 return LDAP_OPERATIONS_ERROR;
693         }
694
695         if ((rc = ldap_start_tls_s (ldap_struct, NULL, NULL)) != LDAP_SUCCESS)  {
696                 DEBUG(0,("Failed to issue the StartTLS instruction: %s\n",
697                          ldap_err2string(rc)));
698                 return rc;
699         }
700
701         DEBUG (3, ("StartTLS issued: using a TLS connection\n"));
702         return LDAP_SUCCESS;
703 #else
704         DEBUG(0,("StartTLS not supported by LDAP client libraries!\n"));
705         return LDAP_OPERATIONS_ERROR;
706 #endif
707 }
708
709 /********************************************************************
710  setup a connection to the LDAP server based on a uri
711 *******************************************************************/
712
713 int smb_ldap_setup_conn(LDAP **ldap_struct, const char *uri)
714 {
715         int rc;
716
717         DEBUG(10, ("smb_ldap_setup_connection: %s\n", uri));
718
719 #ifdef HAVE_LDAP_INITIALIZE
720
721         rc = ldap_initialize(ldap_struct, uri);
722         if (rc) {
723                 DEBUG(0, ("ldap_initialize: %s\n", ldap_err2string(rc)));
724                 return rc;
725         }
726
727         if (lp_ldap_ref_follow() != Auto) {
728                 rc = ldap_set_option(*ldap_struct, LDAP_OPT_REFERRALS,
729                      lp_ldap_ref_follow() ? LDAP_OPT_ON : LDAP_OPT_OFF);
730                 if (rc != LDAP_SUCCESS)
731                         DEBUG(0, ("Failed to set LDAP_OPT_REFERRALS: %s\n",
732                                 ldap_err2string(rc)));
733         }
734
735         return LDAP_SUCCESS;
736 #else 
737
738         /* Parse the string manually */
739
740         {
741                 int port = 0;
742                 fstring protocol;
743                 fstring host;
744                 SMB_ASSERT(sizeof(protocol)>10 && sizeof(host)>254);
745
746
747                 /* skip leading "URL:" (if any) */
748                 if ( strnequal( uri, "URL:", 4 ) ) {
749                         uri += 4;
750                 }
751
752                 sscanf(uri, "%10[^:]://%254[^:/]:%d", protocol, host, &port);
753
754                 if (port == 0) {
755                         if (strequal(protocol, "ldap")) {
756                                 port = LDAP_PORT;
757                         } else if (strequal(protocol, "ldaps")) {
758                                 port = LDAPS_PORT;
759                         } else {
760                                 DEBUG(0, ("unrecognised protocol (%s)!\n", protocol));
761                         }
762                 }
763
764                 if ((*ldap_struct = ldap_init(host, port)) == NULL)     {
765                         DEBUG(0, ("ldap_init failed !\n"));
766                         return LDAP_OPERATIONS_ERROR;
767                 }
768
769                 if (strequal(protocol, "ldaps")) {
770 #ifdef LDAP_OPT_X_TLS
771                         int tls = LDAP_OPT_X_TLS_HARD;
772                         if (ldap_set_option (*ldap_struct, LDAP_OPT_X_TLS, &tls) != LDAP_SUCCESS)
773                         {
774                                 DEBUG(0, ("Failed to setup a TLS session\n"));
775                         }
776
777                         DEBUG(3,("LDAPS option set...!\n"));
778 #else
779                         DEBUG(0,("smbldap_open_connection: Secure connection not supported by LDAP client libraries!\n"));
780                         return LDAP_OPERATIONS_ERROR;
781 #endif /* LDAP_OPT_X_TLS */
782                 }
783         }
784 #endif /* HAVE_LDAP_INITIALIZE */
785
786         /* now set connection timeout */
787 #ifdef LDAP_X_OPT_CONNECT_TIMEOUT /* Netscape */
788         {
789                 int ct = lp_ldap_connection_timeout()*1000;
790                 rc = ldap_set_option(*ldap_struct, LDAP_X_OPT_CONNECT_TIMEOUT, &ct);
791                 if (rc != LDAP_SUCCESS) {
792                         DEBUG(0,("Failed to setup an ldap connection timeout %d: %s\n",
793                                 ct, ldap_err2string(rc)));
794                 }
795         }
796 #elif defined (LDAP_OPT_NETWORK_TIMEOUT) /* OpenLDAP */
797         {
798                 struct timeval ct;
799                 ct.tv_usec = 0;
800                 ct.tv_sec = lp_ldap_connection_timeout();
801                 rc = ldap_set_option(*ldap_struct, LDAP_OPT_NETWORK_TIMEOUT, &ct);
802                 if (rc != LDAP_SUCCESS) {
803                         DEBUG(0,("Failed to setup an ldap connection timeout %d: %s\n",
804                                 (int)ct.tv_sec, ldap_err2string(rc)));
805                 }
806         }
807 #endif
808
809         return LDAP_SUCCESS;
810 }
811
812 /********************************************************************
813  try to upgrade to Version 3 LDAP if not already, in either case return current
814  version 
815  *******************************************************************/
816
817 int smb_ldap_upgrade_conn(LDAP *ldap_struct, int *new_version) 
818 {
819         int version;
820         int rc;
821
822         /* assume the worst */
823         *new_version = LDAP_VERSION2;
824
825         rc = ldap_get_option(ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
826         if (rc) {
827                 return rc;
828         }
829
830         if (version == LDAP_VERSION3) {
831                 *new_version = LDAP_VERSION3;
832                 return LDAP_SUCCESS;
833         }
834
835         /* try upgrade */
836         version = LDAP_VERSION3;
837         rc = ldap_set_option (ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
838         if (rc) {
839                 return rc;
840         }
841
842         *new_version = LDAP_VERSION3;
843         return LDAP_SUCCESS;
844 }
845
846 /*******************************************************************
847  open a connection to the ldap server (just until the bind)
848  ******************************************************************/
849
850 int smb_ldap_setup_full_conn(LDAP **ldap_struct, const char *uri)
851 {
852         int rc, version;
853
854         rc = smb_ldap_setup_conn(ldap_struct, uri);
855         if (rc) {
856                 return rc;
857         }
858
859         rc = smb_ldap_upgrade_conn(*ldap_struct, &version);
860         if (rc) {
861                 return rc;
862         }
863
864         rc = smb_ldap_start_tls(*ldap_struct, version);
865         if (rc) {
866                 return rc;
867         }
868
869         return LDAP_SUCCESS;
870 }
871
872 /*******************************************************************
873  open a connection to the ldap server.
874 ******************************************************************/
875 static int smbldap_open_connection (struct smbldap_state *ldap_state)
876
877 {
878         int rc = LDAP_SUCCESS;
879         int version;
880         LDAP **ldap_struct = &ldap_state->ldap_struct;
881
882         rc = smb_ldap_setup_conn(ldap_struct, ldap_state->uri);
883         if (rc) {
884                 return rc;
885         }
886
887         /* Store the LDAP pointer in a lookup list */
888
889         smbldap_store_state(*ldap_struct, ldap_state);
890
891         /* Upgrade to LDAPv3 if possible */
892
893         rc = smb_ldap_upgrade_conn(*ldap_struct, &version);
894         if (rc) {
895                 return rc;
896         }
897
898         /* Start TLS if required */
899
900         rc = smb_ldap_start_tls(*ldap_struct, version);
901         if (rc) {
902                 return rc;
903         }
904
905         DEBUG(2, ("smbldap_open_connection: connection opened\n"));
906         return rc;
907 }
908
909 /*******************************************************************
910  a rebind function for authenticated referrals
911  This version takes a void* that we can shove useful stuff in :-)
912 ******************************************************************/
913 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
914 #else
915 static int rebindproc_with_state  (LDAP * ld, char **whop, char **credp, 
916                                    int *methodp, int freeit, void *arg)
917 {
918         struct smbldap_state *ldap_state = arg;
919
920         /** @TODO Should we be doing something to check what servers we rebind to?
921             Could we get a referral to a machine that we don't want to give our
922             username and password to? */
923
924         if (freeit) {
925                 SAFE_FREE(*whop);
926                 if (*credp) {
927                         memset(*credp, '\0', strlen(*credp));
928                 }
929                 SAFE_FREE(*credp);
930         } else {
931                 DEBUG(5,("rebind_proc_with_state: Rebinding as \"%s\"\n", 
932                           ldap_state->bind_dn?ldap_state->bind_dn:"[Anonymous bind]"));
933
934                 if (ldap_state->anonymous) {
935                         *whop = NULL;
936                         *credp = NULL;
937                 } else {
938                         *whop = SMB_STRDUP(ldap_state->bind_dn);
939                         if (!*whop) {
940                                 return LDAP_NO_MEMORY;
941                         }
942                         *credp = SMB_STRDUP(ldap_state->bind_secret);
943                         if (!*credp) {
944                                 SAFE_FREE(*whop);
945                                 return LDAP_NO_MEMORY;
946                         }
947                 }
948                 *methodp = LDAP_AUTH_SIMPLE;
949         }
950
951         GetTimeOfDay(&ldap_state->last_rebind);
952
953         return 0;
954 }
955 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
956
957 /*******************************************************************
958  a rebind function for authenticated referrals
959  This version takes a void* that we can shove useful stuff in :-)
960  and actually does the connection.
961 ******************************************************************/
962 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
963 static int rebindproc_connect_with_state (LDAP *ldap_struct, 
964                                           LDAP_CONST char *url, 
965                                           ber_tag_t request,
966                                           ber_int_t msgid, void *arg)
967 {
968         struct smbldap_state *ldap_state =
969                 (struct smbldap_state *)arg;
970         int rc;
971         int version;
972
973         DEBUG(5,("rebindproc_connect_with_state: Rebinding to %s as \"%s\"\n", 
974                  url, ldap_state->bind_dn?ldap_state->bind_dn:"[Anonymous bind]"));
975
976         /* call START_TLS again (ldaps:// is handled by the OpenLDAP library
977          * itself) before rebinding to another LDAP server to avoid to expose
978          * our credentials. At least *try* to secure the connection - Guenther */
979
980         smb_ldap_upgrade_conn(ldap_struct, &version);
981         smb_ldap_start_tls(ldap_struct, version);
982
983         /** @TODO Should we be doing something to check what servers we rebind to?
984             Could we get a referral to a machine that we don't want to give our
985             username and password to? */
986
987         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
988
989         /* only set the last rebind timestamp when we did rebind after a
990          * non-read LDAP operation. That way we avoid the replication sleep
991          * after a simple redirected search operation - Guenther */
992
993         switch (request) {
994
995                 case LDAP_REQ_MODIFY:
996                 case LDAP_REQ_ADD:
997                 case LDAP_REQ_DELETE:
998                 case LDAP_REQ_MODDN:
999                 case LDAP_REQ_EXTENDED:
1000                         DEBUG(10,("rebindproc_connect_with_state: "
1001                                 "setting last_rebind timestamp "
1002                                 "(req: 0x%02x)\n", (unsigned int)request));
1003                         GetTimeOfDay(&ldap_state->last_rebind);
1004                         break;
1005                 default:
1006                         ZERO_STRUCT(ldap_state->last_rebind);
1007                         break;
1008         }
1009
1010         return rc;
1011 }
1012 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1013
1014 /*******************************************************************
1015  Add a rebind function for authenticated referrals
1016 ******************************************************************/
1017 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1018 #else
1019 # if LDAP_SET_REBIND_PROC_ARGS == 2
1020 static int rebindproc (LDAP *ldap_struct, char **whop, char **credp,
1021                        int *method, int freeit )
1022 {
1023         struct smbldap_state *ldap_state = smbldap_find_state(ldap_struct);
1024
1025         return rebindproc_with_state(ldap_struct, whop, credp,
1026                                      method, freeit, ldap_state);
1027 }
1028 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
1029 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1030
1031 /*******************************************************************
1032  a rebind function for authenticated referrals
1033  this also does the connection, but no void*.
1034 ******************************************************************/
1035 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1036 # if LDAP_SET_REBIND_PROC_ARGS == 2
1037 static int rebindproc_connect (LDAP * ld, LDAP_CONST char *url, int request,
1038                                ber_int_t msgid)
1039 {
1040         struct smbldap_state *ldap_state = smbldap_find_state(ld);
1041
1042         return rebindproc_connect_with_state(ld, url, (ber_tag_t)request, msgid, 
1043                                              ldap_state);
1044 }
1045 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
1046 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1047
1048 /*******************************************************************
1049  connect to the ldap server under system privilege.
1050 ******************************************************************/
1051 static int smbldap_connect_system(struct smbldap_state *ldap_state, LDAP * ldap_struct)
1052 {
1053         int rc;
1054         int version;
1055
1056         if (!ldap_state->anonymous && !ldap_state->bind_dn) {
1057                 char *bind_dn = NULL;
1058                 char *bind_secret = NULL;
1059
1060                 /* get the default dn and password only if they are not set already */
1061                 if (!fetch_ldap_pw(&bind_dn, &bind_secret)) {
1062                         DEBUG(0, ("ldap_connect_system: Failed to retrieve password from secrets.tdb\n"));
1063                         return LDAP_INVALID_CREDENTIALS;
1064                 }
1065                 smbldap_set_creds(ldap_state, false, bind_dn, bind_secret);
1066                 SAFE_FREE(bind_dn);
1067                 memset(bind_secret, '\0', strlen(bind_secret));
1068                 SAFE_FREE(bind_secret);
1069         }
1070
1071         /* removed the sasl_bind_s "EXTERNAL" stuff, as my testsuite 
1072            (OpenLDAP) doesnt' seem to support it */
1073
1074         DEBUG(10,("ldap_connect_system: Binding to ldap server %s as \"%s\"\n",
1075                   ldap_state->uri, ldap_state->bind_dn));
1076
1077 #ifdef HAVE_LDAP_SET_REBIND_PROC
1078 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1079 # if LDAP_SET_REBIND_PROC_ARGS == 2     
1080         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect); 
1081 # endif
1082 # if LDAP_SET_REBIND_PROC_ARGS == 3     
1083         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect_with_state, (void *)ldap_state);  
1084 # endif
1085 #else /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1086 # if LDAP_SET_REBIND_PROC_ARGS == 2     
1087         ldap_set_rebind_proc(ldap_struct, &rebindproc); 
1088 # endif
1089 # if LDAP_SET_REBIND_PROC_ARGS == 3     
1090         ldap_set_rebind_proc(ldap_struct, &rebindproc_with_state, (void *)ldap_state);  
1091 # endif
1092 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1093 #endif
1094
1095         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
1096
1097         if (rc != LDAP_SUCCESS) {
1098                 char *ld_error = NULL;
1099                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING,
1100                                 &ld_error);
1101                 DEBUG(ldap_state->num_failures ? 2 : 0,
1102                       ("failed to bind to server %s with dn=\"%s\" Error: %s\n\t%s\n",
1103                                ldap_state->uri,
1104                                ldap_state->bind_dn ? ldap_state->bind_dn : "[Anonymous bind]",
1105                                ldap_err2string(rc),
1106                                ld_error ? ld_error : "(unknown)"));
1107                 SAFE_FREE(ld_error);
1108                 ldap_state->num_failures++;
1109                 return rc;
1110         }
1111
1112         ldap_state->num_failures = 0;
1113         ldap_state->paged_results = False;
1114
1115         ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
1116
1117         if (smbldap_has_control(ldap_state->ldap_struct, ADS_PAGE_CTL_OID) && version == 3) {
1118                 ldap_state->paged_results = True;
1119         }
1120
1121         DEBUG(3, ("ldap_connect_system: successful connection to the LDAP server\n"));
1122         DEBUGADD(10, ("ldap_connect_system: LDAP server %s support paged results\n", 
1123                 ldap_state->paged_results ? "does" : "does not"));
1124         return rc;
1125 }
1126
1127 static void smbldap_idle_fn(struct event_context *event_ctx,
1128                             struct timed_event *te,
1129                             struct timeval now,
1130                             void *private_data);
1131
1132 /**********************************************************************
1133  Connect to LDAP server (called before every ldap operation)
1134 *********************************************************************/
1135 static int smbldap_open(struct smbldap_state *ldap_state)
1136 {
1137         int rc, opt_rc;
1138         bool reopen = False;
1139         SMB_ASSERT(ldap_state);
1140
1141         if ((ldap_state->ldap_struct != NULL) && ((ldap_state->last_ping + SMBLDAP_DONT_PING_TIME) < time(NULL))) {
1142
1143 #ifdef HAVE_UNIXSOCKET
1144                 struct sockaddr_un addr;
1145 #else
1146                 struct sockaddr addr;
1147 #endif
1148                 socklen_t len = sizeof(addr);
1149                 int sd;
1150
1151                 opt_rc = ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_DESC, &sd);
1152                 if (opt_rc == 0 && (getpeername(sd, (struct sockaddr *) &addr, &len)) < 0 )
1153                         reopen = True;
1154
1155 #ifdef HAVE_UNIXSOCKET
1156                 if (opt_rc == 0 && addr.sun_family == AF_UNIX)
1157                         reopen = True;
1158 #endif
1159                 if (reopen) {
1160                         /* the other end has died. reopen. */
1161                         ldap_unbind(ldap_state->ldap_struct);
1162                         ldap_state->ldap_struct = NULL;
1163                         ldap_state->last_ping = (time_t)0;
1164                 } else {
1165                         ldap_state->last_ping = time(NULL);
1166                 } 
1167         }
1168
1169         if (ldap_state->ldap_struct != NULL) {
1170                 DEBUG(11,("smbldap_open: already connected to the LDAP server\n"));
1171                 return LDAP_SUCCESS;
1172         }
1173
1174         if ((rc = smbldap_open_connection(ldap_state))) {
1175                 return rc;
1176         }
1177
1178         if ((rc = smbldap_connect_system(ldap_state, ldap_state->ldap_struct))) {
1179                 ldap_unbind(ldap_state->ldap_struct);
1180                 ldap_state->ldap_struct = NULL;
1181                 return rc;
1182         }
1183
1184
1185         ldap_state->last_ping = time(NULL);
1186         ldap_state->pid = sys_getpid();
1187
1188         TALLOC_FREE(ldap_state->idle_event);
1189
1190         if (ldap_state->event_context != NULL) {
1191                 ldap_state->idle_event = event_add_timed(
1192                         ldap_state->event_context, NULL,
1193                         timeval_current_ofs(SMBLDAP_IDLE_TIME, 0),
1194                         smbldap_idle_fn, ldap_state);
1195         }
1196
1197         DEBUG(4,("The LDAP server is successfully connected\n"));
1198
1199         return LDAP_SUCCESS;
1200 }
1201
1202 /**********************************************************************
1203 Disconnect from LDAP server 
1204 *********************************************************************/
1205 static NTSTATUS smbldap_close(struct smbldap_state *ldap_state)
1206 {
1207         if (!ldap_state)
1208                 return NT_STATUS_INVALID_PARAMETER;
1209
1210         if (ldap_state->ldap_struct != NULL) {
1211                 ldap_unbind(ldap_state->ldap_struct);
1212                 ldap_state->ldap_struct = NULL;
1213         }
1214
1215         smbldap_delete_state(ldap_state);
1216
1217         DEBUG(5,("The connection to the LDAP server was closed\n"));
1218         /* maybe free the results here --metze */
1219
1220         return NT_STATUS_OK;
1221 }
1222
1223 static bool got_alarm;
1224
1225 static void (*old_handler)(int);
1226
1227 static void gotalarm_sig(int dummy)
1228 {
1229         got_alarm = True;
1230 }
1231
1232 static int another_ldap_try(struct smbldap_state *ldap_state, int *rc,
1233                             int *attempts, time_t endtime)
1234 {
1235         time_t now = time(NULL);
1236         int open_rc = LDAP_SERVER_DOWN;
1237
1238         if (*rc != LDAP_SERVER_DOWN)
1239                 goto no_next;
1240
1241         if (now >= endtime) {
1242                 smbldap_close(ldap_state);
1243                 *rc = LDAP_TIMEOUT;
1244                 goto no_next;
1245         }
1246
1247         if (*attempts == 0) {
1248                 got_alarm = False;
1249                 old_handler = CatchSignal(SIGALRM, gotalarm_sig);
1250                 alarm(endtime - now);
1251
1252                 if (ldap_state->pid != sys_getpid())
1253                         smbldap_close(ldap_state);
1254         }
1255
1256         while (1) {
1257
1258                 if (*attempts != 0)
1259                         smb_msleep(1000);
1260
1261                 *attempts += 1;
1262
1263                 open_rc = smbldap_open(ldap_state);
1264
1265                 if (open_rc == LDAP_SUCCESS) {
1266                         ldap_state->last_use = now;
1267                         return True;
1268                 }
1269
1270                 if (open_rc == LDAP_INSUFFICIENT_ACCESS) {
1271                         /* The fact that we are non-root or any other
1272                          * access-denied condition will not change in the next
1273                          * round of trying */
1274                         *rc = open_rc;
1275                         break;
1276                 }
1277
1278                 if (got_alarm) {
1279                         *rc = LDAP_TIMEOUT;
1280                         break;
1281                 }
1282
1283                 if (open_rc != LDAP_SUCCESS) {
1284                         DEBUG(1, ("Connection to LDAP server failed for the "
1285                                   "%d try!\n", *attempts));
1286                 }
1287         }
1288
1289  no_next:
1290         CatchSignal(SIGALRM, old_handler);
1291         alarm(0);
1292         ldap_state->last_use = now;
1293         return False;
1294 }
1295
1296 /*********************************************************************
1297  ********************************************************************/
1298
1299 static int smbldap_search_ext(struct smbldap_state *ldap_state,
1300                               const char *base, int scope, const char *filter, 
1301                               const char *attrs[], int attrsonly,
1302                               LDAPControl **sctrls, LDAPControl **cctrls, 
1303                               int sizelimit, LDAPMessage **res)
1304 {
1305         int             rc = LDAP_SERVER_DOWN;
1306         int             attempts = 0;
1307         char           *utf8_filter;
1308         time_t          endtime = time(NULL)+lp_ldap_timeout();
1309         struct          timeval timeout;
1310         size_t          converted_size;
1311
1312         SMB_ASSERT(ldap_state);
1313
1314         DEBUG(5,("smbldap_search_ext: base => [%s], filter => [%s], "
1315                  "scope => [%d]\n", base, filter, scope));
1316
1317         if (ldap_state->last_rebind.tv_sec > 0) {
1318                 struct timeval  tval;
1319                 int64_t tdiff = 0;
1320                 int             sleep_time = 0;
1321
1322                 ZERO_STRUCT(tval);
1323                 GetTimeOfDay(&tval);
1324
1325                 tdiff = usec_time_diff(&tval, &ldap_state->last_rebind);
1326                 tdiff /= 1000; /* Convert to milliseconds. */
1327
1328                 sleep_time = lp_ldap_replication_sleep()-(int)tdiff;
1329                 sleep_time = MIN(sleep_time, MAX_LDAP_REPLICATION_SLEEP_TIME);
1330
1331                 if (sleep_time > 0) {
1332                         /* we wait for the LDAP replication */
1333                         DEBUG(5,("smbldap_search_ext: waiting %d milliseconds "
1334                                  "for LDAP replication.\n",sleep_time));
1335                         smb_msleep(sleep_time);
1336                         DEBUG(5,("smbldap_search_ext: go on!\n"));
1337                 }
1338                 ZERO_STRUCT(ldap_state->last_rebind);
1339         }
1340
1341         if (!push_utf8_talloc(talloc_tos(), &utf8_filter, filter, &converted_size)) {
1342                 return LDAP_NO_MEMORY;
1343         }
1344
1345         /* Setup timeout for the ldap_search_ext_s call - local and remote. */
1346         timeout.tv_sec = lp_ldap_timeout();
1347         timeout.tv_usec = 0;
1348
1349         /* Setup alarm timeout.... Do we need both of these ? JRA.
1350          * Yes, I think we do need both of these. The server timeout only
1351          * covers the case where the server's operation takes too long. It
1352          * does not cover the case where the request hangs on its way to the
1353          * server. The server side timeout is not strictly necessary, it's
1354          * just a bit more kind to the server. VL. */
1355
1356         got_alarm = 0;
1357         CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
1358         alarm(lp_ldap_timeout());
1359         /* End setup timeout. */
1360
1361         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1362                 rc = ldap_search_ext_s(ldap_state->ldap_struct, base, scope, 
1363                                        utf8_filter,
1364                                        CONST_DISCARD(char **, attrs),
1365                                        attrsonly, sctrls, cctrls, &timeout,
1366                                        sizelimit, res);
1367                 if (rc != LDAP_SUCCESS) {
1368                         char *ld_error = NULL;
1369                         int ld_errno;
1370
1371                         ldap_get_option(ldap_state->ldap_struct,
1372                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1373
1374                         ldap_get_option(ldap_state->ldap_struct,
1375                                         LDAP_OPT_ERROR_STRING, &ld_error);
1376                         DEBUG(10, ("Failed search for base: %s, error: %d (%s) "
1377                                    "(%s)\n", base, ld_errno,
1378                                    ldap_err2string(rc),
1379                                    ld_error ? ld_error : "unknown"));
1380                         SAFE_FREE(ld_error);
1381
1382                         if (ld_errno == LDAP_SERVER_DOWN) {
1383                                 ldap_unbind(ldap_state->ldap_struct);
1384                                 ldap_state->ldap_struct = NULL;
1385                         }
1386                 }
1387         }
1388
1389         TALLOC_FREE(utf8_filter);
1390
1391         /* Teardown timeout. */
1392         CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
1393         alarm(0);
1394
1395         if (got_alarm != 0)
1396                 return LDAP_TIMELIMIT_EXCEEDED;
1397
1398         return rc;
1399 }
1400
1401 int smbldap_search(struct smbldap_state *ldap_state, 
1402                    const char *base, int scope, const char *filter, 
1403                    const char *attrs[], int attrsonly, 
1404                    LDAPMessage **res)
1405 {
1406         return smbldap_search_ext(ldap_state, base, scope, filter, attrs,
1407                                   attrsonly, NULL, NULL, LDAP_NO_LIMIT, res);
1408 }
1409
1410 int smbldap_search_paged(struct smbldap_state *ldap_state, 
1411                          const char *base, int scope, const char *filter, 
1412                          const char **attrs, int attrsonly, int pagesize,
1413                          LDAPMessage **res, void **cookie)
1414 {
1415         LDAPControl     pr;
1416         LDAPControl     **rcontrols;
1417         LDAPControl     *controls[2] = { NULL, NULL};
1418         BerElement      *cookie_be = NULL;
1419         struct berval   *cookie_bv = NULL;
1420         int             tmp = 0, i, rc;
1421         bool            critical = True;
1422
1423         *res = NULL;
1424
1425         DEBUG(3,("smbldap_search_paged: base => [%s], filter => [%s],"
1426                  "scope => [%d], pagesize => [%d]\n",
1427                  base, filter, scope, pagesize));
1428
1429         cookie_be = ber_alloc_t(LBER_USE_DER);
1430         if (cookie_be == NULL) {
1431                 DEBUG(0,("smbldap_create_page_control: ber_alloc_t returns "
1432                          "NULL\n"));
1433                 return LDAP_NO_MEMORY;
1434         }
1435
1436         /* construct cookie */
1437         if (*cookie != NULL) {
1438                 ber_printf(cookie_be, "{iO}", (ber_int_t) pagesize, *cookie);
1439                 ber_bvfree((struct berval *)*cookie); /* don't need it from last time */
1440                 *cookie = NULL;
1441         } else {
1442                 ber_printf(cookie_be, "{io}", (ber_int_t) pagesize, "", 0);
1443         }
1444         ber_flatten(cookie_be, &cookie_bv);
1445
1446         pr.ldctl_oid = CONST_DISCARD(char *, ADS_PAGE_CTL_OID);
1447         pr.ldctl_iscritical = (char) critical;
1448         pr.ldctl_value.bv_len = cookie_bv->bv_len;
1449         pr.ldctl_value.bv_val = cookie_bv->bv_val;
1450
1451         controls[0] = &pr;
1452         controls[1] = NULL;
1453
1454         rc = smbldap_search_ext(ldap_state, base, scope, filter, attrs, 
1455                                  0, controls, NULL, LDAP_NO_LIMIT, res);
1456
1457         ber_free(cookie_be, 1);
1458         ber_bvfree(cookie_bv);
1459
1460         if (rc != 0) {
1461                 DEBUG(3,("smbldap_search_paged: smbldap_search_ext(%s) "
1462                          "failed with [%s]\n", filter, ldap_err2string(rc)));
1463                 goto done;
1464         }
1465
1466         DEBUG(3,("smbldap_search_paged: search was successful\n"));
1467
1468         rc = ldap_parse_result(ldap_state->ldap_struct, *res, NULL, NULL, 
1469                                NULL, NULL, &rcontrols,  0);
1470         if (rc != 0) {
1471                 DEBUG(3,("smbldap_search_paged: ldap_parse_result failed " \
1472                          "with [%s]\n", ldap_err2string(rc)));
1473                 goto done;
1474         }
1475
1476         if (rcontrols == NULL)
1477                 goto done;
1478
1479         for (i=0; rcontrols[i]; i++) {
1480
1481                 if (strcmp(ADS_PAGE_CTL_OID, rcontrols[i]->ldctl_oid) != 0)
1482                         continue;
1483
1484                 cookie_be = ber_init(&rcontrols[i]->ldctl_value);
1485                 ber_scanf(cookie_be,"{iO}", &tmp, &cookie_bv);
1486                 /* the berval is the cookie, but must be freed when it is all
1487                    done */
1488                 if (cookie_bv->bv_len)
1489                         *cookie=ber_bvdup(cookie_bv);
1490                 else
1491                         *cookie=NULL;
1492                 ber_bvfree(cookie_bv);
1493                 ber_free(cookie_be, 1);
1494                 break;
1495         }
1496         ldap_controls_free(rcontrols);
1497 done:   
1498         return rc;
1499 }
1500
1501 int smbldap_modify(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1502 {
1503         int             rc = LDAP_SERVER_DOWN;
1504         int             attempts = 0;
1505         char           *utf8_dn;
1506         time_t          endtime = time(NULL)+lp_ldap_timeout();
1507         size_t          converted_size;
1508
1509         SMB_ASSERT(ldap_state);
1510
1511         DEBUG(5,("smbldap_modify: dn => [%s]\n", dn ));
1512
1513         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1514                 return LDAP_NO_MEMORY;
1515         }
1516
1517         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1518                 rc = ldap_modify_s(ldap_state->ldap_struct, utf8_dn, attrs);
1519                 if (rc != LDAP_SUCCESS) {
1520                         char *ld_error = NULL;
1521                         int ld_errno;
1522
1523                         ldap_get_option(ldap_state->ldap_struct,
1524                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1525
1526                         ldap_get_option(ldap_state->ldap_struct,
1527                                         LDAP_OPT_ERROR_STRING, &ld_error);
1528                         DEBUG(10, ("Failed to modify dn: %s, error: %d (%s) "
1529                                    "(%s)\n", dn, ld_errno,
1530                                    ldap_err2string(rc),
1531                                    ld_error ? ld_error : "unknown"));
1532                         SAFE_FREE(ld_error);
1533
1534                         if (ld_errno == LDAP_SERVER_DOWN) {
1535                                 ldap_unbind(ldap_state->ldap_struct);
1536                                 ldap_state->ldap_struct = NULL;
1537                         }
1538                 }
1539         }
1540
1541         TALLOC_FREE(utf8_dn);
1542         return rc;
1543 }
1544
1545 int smbldap_add(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1546 {
1547         int             rc = LDAP_SERVER_DOWN;
1548         int             attempts = 0;
1549         char           *utf8_dn;
1550         time_t          endtime = time(NULL)+lp_ldap_timeout();
1551         size_t          converted_size;
1552
1553         SMB_ASSERT(ldap_state);
1554
1555         DEBUG(5,("smbldap_add: dn => [%s]\n", dn ));
1556
1557         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1558                 return LDAP_NO_MEMORY;
1559         }
1560
1561         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1562                 rc = ldap_add_s(ldap_state->ldap_struct, utf8_dn, attrs);
1563                 if (rc != LDAP_SUCCESS) {
1564                         char *ld_error = NULL;
1565                         int ld_errno;
1566
1567                         ldap_get_option(ldap_state->ldap_struct,
1568                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1569
1570                         ldap_get_option(ldap_state->ldap_struct,
1571                                         LDAP_OPT_ERROR_STRING, &ld_error);
1572                         DEBUG(10, ("Failed to add dn: %s, error: %d (%s) "
1573                                    "(%s)\n", dn, ld_errno,
1574                                    ldap_err2string(rc),
1575                                    ld_error ? ld_error : "unknown"));
1576                         SAFE_FREE(ld_error);
1577
1578                         if (ld_errno == LDAP_SERVER_DOWN) {
1579                                 ldap_unbind(ldap_state->ldap_struct);
1580                                 ldap_state->ldap_struct = NULL;
1581                         }
1582                 }
1583         }
1584
1585         TALLOC_FREE(utf8_dn);
1586         return rc;
1587 }
1588
1589 int smbldap_delete(struct smbldap_state *ldap_state, const char *dn)
1590 {
1591         int             rc = LDAP_SERVER_DOWN;
1592         int             attempts = 0;
1593         char           *utf8_dn;
1594         time_t          endtime = time(NULL)+lp_ldap_timeout();
1595         size_t          converted_size;
1596
1597         SMB_ASSERT(ldap_state);
1598
1599         DEBUG(5,("smbldap_delete: dn => [%s]\n", dn ));
1600
1601         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1602                 return LDAP_NO_MEMORY;
1603         }
1604
1605         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1606                 rc = ldap_delete_s(ldap_state->ldap_struct, utf8_dn);
1607                 if (rc != LDAP_SUCCESS) {
1608                         char *ld_error = NULL;
1609                         int ld_errno;
1610
1611                         ldap_get_option(ldap_state->ldap_struct,
1612                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1613
1614                         ldap_get_option(ldap_state->ldap_struct,
1615                                         LDAP_OPT_ERROR_STRING, &ld_error);
1616                         DEBUG(10, ("Failed to delete dn: %s, error: %d (%s) "
1617                                    "(%s)\n", dn, ld_errno,
1618                                    ldap_err2string(rc),
1619                                    ld_error ? ld_error : "unknown"));
1620                         SAFE_FREE(ld_error);
1621
1622                         if (ld_errno == LDAP_SERVER_DOWN) {
1623                                 ldap_unbind(ldap_state->ldap_struct);
1624                                 ldap_state->ldap_struct = NULL;
1625                         }
1626                 }
1627         }
1628
1629         TALLOC_FREE(utf8_dn);
1630         return rc;
1631 }
1632
1633 int smbldap_extended_operation(struct smbldap_state *ldap_state, 
1634                                LDAP_CONST char *reqoid, struct berval *reqdata, 
1635                                LDAPControl **serverctrls, LDAPControl **clientctrls, 
1636                                char **retoidp, struct berval **retdatap)
1637 {
1638         int             rc = LDAP_SERVER_DOWN;
1639         int             attempts = 0;
1640         time_t          endtime = time(NULL)+lp_ldap_timeout();
1641
1642         if (!ldap_state)
1643                 return (-1);
1644
1645         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1646                 rc = ldap_extended_operation_s(ldap_state->ldap_struct, reqoid,
1647                                                reqdata, serverctrls,
1648                                                clientctrls, retoidp, retdatap);
1649                 if (rc != LDAP_SUCCESS) {
1650                         char *ld_error = NULL;
1651                         int ld_errno;
1652
1653                         ldap_get_option(ldap_state->ldap_struct,
1654                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1655
1656                         ldap_get_option(ldap_state->ldap_struct,
1657                                         LDAP_OPT_ERROR_STRING, &ld_error);
1658                         DEBUG(10, ("Extended operation failed with error: "
1659                                    "%d (%s) (%s)\n", ld_errno,
1660                                    ldap_err2string(rc),
1661                                    ld_error ? ld_error : "unknown"));
1662                         SAFE_FREE(ld_error);
1663
1664                         if (ld_errno == LDAP_SERVER_DOWN) {
1665                                 ldap_unbind(ldap_state->ldap_struct);
1666                                 ldap_state->ldap_struct = NULL;
1667                         }
1668                 }
1669         }
1670
1671         return rc;
1672 }
1673
1674 /*******************************************************************
1675  run the search by name.
1676 ******************************************************************/
1677 int smbldap_search_suffix (struct smbldap_state *ldap_state,
1678                            const char *filter, const char **search_attr,
1679                            LDAPMessage ** result)
1680 {
1681         return smbldap_search(ldap_state, lp_ldap_suffix(), LDAP_SCOPE_SUBTREE,
1682                               filter, search_attr, 0, result);
1683 }
1684
1685 static void smbldap_idle_fn(struct event_context *event_ctx,
1686                             struct timed_event *te,
1687                             struct timeval now,
1688                             void *private_data)
1689 {
1690         struct smbldap_state *state = (struct smbldap_state *)private_data;
1691
1692         TALLOC_FREE(state->idle_event);
1693
1694         if (state->ldap_struct == NULL) {
1695                 DEBUG(10,("ldap connection not connected...\n"));
1696                 return;
1697         }
1698
1699         if ((state->last_use+SMBLDAP_IDLE_TIME) > now.tv_sec) {
1700                 DEBUG(10,("ldap connection not idle...\n"));
1701
1702                 state->idle_event = event_add_timed(
1703                         event_ctx, NULL,
1704                         timeval_add(&now, SMBLDAP_IDLE_TIME, 0),
1705                         smbldap_idle_fn,
1706                         private_data);
1707                 return;
1708         }
1709
1710         DEBUG(7,("ldap connection idle...closing connection\n"));
1711         smbldap_close(state);
1712 }
1713
1714 /**********************************************************************
1715  Housekeeping
1716  *********************************************************************/
1717
1718 void smbldap_free_struct(struct smbldap_state **ldap_state) 
1719 {
1720         smbldap_close(*ldap_state);
1721
1722         if ((*ldap_state)->bind_secret) {
1723                 memset((*ldap_state)->bind_secret, '\0', strlen((*ldap_state)->bind_secret));
1724         }
1725
1726         SAFE_FREE((*ldap_state)->bind_dn);
1727         SAFE_FREE((*ldap_state)->bind_secret);
1728
1729         TALLOC_FREE((*ldap_state)->idle_event);
1730
1731         *ldap_state = NULL;
1732
1733         /* No need to free any further, as it is talloc()ed */
1734 }
1735
1736
1737 /**********************************************************************
1738  Intitalise the 'general' ldap structures, on which ldap operations may be conducted
1739  *********************************************************************/
1740
1741 NTSTATUS smbldap_init(TALLOC_CTX *mem_ctx, struct event_context *event_ctx,
1742                       const char *location,
1743                       struct smbldap_state **smbldap_state)
1744 {
1745         *smbldap_state = TALLOC_ZERO_P(mem_ctx, struct smbldap_state);
1746         if (!*smbldap_state) {
1747                 DEBUG(0, ("talloc() failed for ldapsam private_data!\n"));
1748                 return NT_STATUS_NO_MEMORY;
1749         }
1750
1751         if (location) {
1752                 (*smbldap_state)->uri = talloc_strdup(mem_ctx, location);
1753         } else {
1754                 (*smbldap_state)->uri = "ldap://localhost";
1755         }
1756
1757         (*smbldap_state)->event_context = event_ctx;
1758
1759         return NT_STATUS_OK;
1760 }
1761
1762  char *smbldap_talloc_dn(TALLOC_CTX *mem_ctx, LDAP *ld,
1763                          LDAPMessage *entry)
1764 {
1765         char *utf8_dn, *unix_dn;
1766         size_t converted_size;
1767
1768         utf8_dn = ldap_get_dn(ld, entry);
1769         if (!utf8_dn) {
1770                 DEBUG (5, ("smbldap_talloc_dn: ldap_get_dn failed\n"));
1771                 return NULL;
1772         }
1773         if (!pull_utf8_talloc(mem_ctx, &unix_dn, utf8_dn, &converted_size)) {
1774                 DEBUG (0, ("smbldap_talloc_dn: String conversion failure utf8 "
1775                            "[%s]\n", utf8_dn));
1776                 return NULL;
1777         }
1778         ldap_memfree(utf8_dn);
1779         return unix_dn;
1780 }
1781
1782 /*******************************************************************
1783  Check if root-dse has a certain Control or Extension
1784 ********************************************************************/
1785
1786 static bool smbldap_check_root_dse(LDAP *ld, const char **attrs, const char *value) 
1787 {
1788         LDAPMessage *msg = NULL;
1789         LDAPMessage *entry = NULL;
1790         char **values = NULL;
1791         int rc, num_result, num_values, i;
1792         bool result = False;
1793
1794         if (!attrs[0]) {
1795                 DEBUG(3,("smbldap_check_root_dse: nothing to look for\n"));
1796                 return False;
1797         }
1798
1799         if (!strequal(attrs[0], "supportedExtension") && 
1800             !strequal(attrs[0], "supportedControl") && 
1801             !strequal(attrs[0], "namingContexts")) {
1802                 DEBUG(3,("smbldap_check_root_dse: no idea what to query root-dse for: %s ?\n", attrs[0]));
1803                 return False;
1804         }
1805
1806         rc = ldap_search_s(ld, "", LDAP_SCOPE_BASE, 
1807                            "(objectclass=*)", CONST_DISCARD(char **, attrs), 0 , &msg);
1808
1809         if (rc != LDAP_SUCCESS) {
1810                 DEBUG(3,("smbldap_check_root_dse: Could not search rootDSE\n"));
1811                 return False;
1812         }
1813
1814         num_result = ldap_count_entries(ld, msg);
1815
1816         if (num_result != 1) {
1817                 DEBUG(3,("smbldap_check_root_dse: Expected one rootDSE, got %d\n", num_result));
1818                 goto done;
1819         }
1820
1821         entry = ldap_first_entry(ld, msg);
1822
1823         if (entry == NULL) {
1824                 DEBUG(3,("smbldap_check_root_dse: Could not retrieve rootDSE\n"));
1825                 goto done;
1826         }
1827
1828         values = ldap_get_values(ld, entry, attrs[0]);
1829
1830         if (values == NULL) {
1831                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not support any %s\n", attrs[0]));
1832                 goto done;
1833         }
1834
1835         num_values = ldap_count_values(values);
1836
1837         if (num_values == 0) {
1838                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not have any %s\n", attrs[0]));
1839                 goto done;
1840         }
1841
1842         for (i=0; i<num_values; i++) {
1843                 if (strcmp(values[i], value) == 0)
1844                         result = True;
1845         }
1846
1847
1848  done:
1849         if (values != NULL)
1850                 ldap_value_free(values);
1851         if (msg != NULL)
1852                 ldap_msgfree(msg);
1853
1854         return result;
1855
1856 }
1857
1858 /*******************************************************************
1859  Check if LDAP-Server supports a certain Control (OID in string format)
1860 ********************************************************************/
1861
1862 bool smbldap_has_control(LDAP *ld, const char *control)
1863 {
1864         const char *attrs[] = { "supportedControl", NULL };
1865         return smbldap_check_root_dse(ld, attrs, control);
1866 }
1867
1868 /*******************************************************************
1869  Check if LDAP-Server supports a certain Extension (OID in string format)
1870 ********************************************************************/
1871
1872 bool smbldap_has_extension(LDAP *ld, const char *extension)
1873 {
1874         const char *attrs[] = { "supportedExtension", NULL };
1875         return smbldap_check_root_dse(ld, attrs, extension);
1876 }
1877
1878 /*******************************************************************
1879  Check if LDAP-Server holds a given namingContext
1880 ********************************************************************/
1881
1882 bool smbldap_has_naming_context(LDAP *ld, const char *naming_context)
1883 {
1884         const char *attrs[] = { "namingContexts", NULL };
1885         return smbldap_check_root_dse(ld, attrs, naming_context);
1886 }
1887
1888 bool smbldap_set_creds(struct smbldap_state *ldap_state, bool anon, const char *dn, const char *secret)
1889 {
1890         ldap_state->anonymous = anon;
1891
1892         /* free any previously set credential */
1893
1894         SAFE_FREE(ldap_state->bind_dn);
1895         if (ldap_state->bind_secret) {
1896                 /* make sure secrets are zeroed out of memory */
1897                 memset(ldap_state->bind_secret, '\0', strlen(ldap_state->bind_secret));
1898                 SAFE_FREE(ldap_state->bind_secret);
1899         }
1900
1901         if ( ! anon) {
1902                 ldap_state->bind_dn = SMB_STRDUP(dn);
1903                 ldap_state->bind_secret = SMB_STRDUP(secret);
1904         }
1905
1906         return True;
1907 }