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
 548                 mods[i]->mod_values[j] = SMB_STRDUP(utf8_value);
 549                 TALLOC_FREE(utf8_value);
 550                 SMB_ASSERT(mods[i]->mod_values[j] != NULL);
 551
 552                 mods[i]->mod_values[j + 1] = NULL;
 553         }
 554         *modlist = mods;
 555 }
 556
 557 /**********************************************************************
 558   Set attribute to newval in LDAP, regardless of what value the
 559   attribute had in LDAP before.
 560 *********************************************************************/
 561
 562  void smbldap_make_mod(LDAP *ldap_struct, LDAPMessage *existing,
 563                       LDAPMod ***mods,
 564                       const char *attribute, const char *newval)
 565 {
 566         char oldval[2048]; /* current largest allowed value is mungeddial */
 567         bool existed;
 568
 569         if (attribute == NULL) {
 570                 /* This can actually happen for ldapsam_compat where we for
 571                  * example don't have a password history */
 572                 return;
 573         }
 574
 575         if (existing != NULL) {
 576                 existed = smbldap_get_single_attribute(ldap_struct, existing, attribute, oldval, sizeof(oldval));
 577         } else {
 578                 existed = False;
 579                 *oldval = '\0';
 580         }
 581
 582         /* all of our string attributes are case insensitive */
 583
 584         if (existed && newval && (StrCaseCmp(oldval, newval) == 0)) {
 585
 586                 /* Believe it or not, but LDAP will deny a delete and
 587                    an add at the same time if the values are the
 588                    same... */
 589                 DEBUG(10,("smbldap_make_mod: attribute |%s| not changed.\n", attribute));
 590                 return;
 591         }
 592
 593         if (existed) {
 594                 /* There has been no value before, so don't delete it.
 595                  * Here's a possible race: We might end up with
 596                  * duplicate attributes */
 597                 /* By deleting exactly the value we found in the entry this
 598                  * should be race-free in the sense that the LDAP-Server will
 599                  * deny the complete operation if somebody changed the
 600                  * attribute behind our back. */
 601                 /* This will also allow modifying single valued attributes
 602                  * in Novell NDS. In NDS you have to first remove attribute and then
 603                  * you could add new value */
 604
 605                 DEBUG(10,("smbldap_make_mod: deleting attribute |%s| values |%s|\n", attribute, oldval));
 606                 smbldap_set_mod(mods, LDAP_MOD_DELETE, attribute, oldval);
 607         }
 608
 609         /* Regardless of the real operation (add or modify)
 610            we add the new value here. We rely on deleting
 611            the old value, should it exist. */
 612
 613         if ((newval != NULL) && (strlen(newval) > 0)) {
 614                 DEBUG(10,("smbldap_make_mod: adding attribute |%s| value |%s|\n", attribute, newval));
 615                 smbldap_set_mod(mods, LDAP_MOD_ADD, attribute, newval);
 616         }
 617 }
 618
 619 /**********************************************************************
 620  Some varients of the LDAP rebind code do not pass in the third 'arg'
 621  pointer to a void*, so we try and work around it by assuming that the
 622  value of the 'LDAP *' pointer is the same as the one we had passed in
 623  **********************************************************************/
 624
 625 struct smbldap_state_lookup {
 626         LDAP *ld;
 627         struct smbldap_state *smbldap_state;
 628         struct smbldap_state_lookup *prev, *next;
 629 };
 630
 631 static struct smbldap_state_lookup *smbldap_state_lookup_list;
 632
 633 static struct smbldap_state *smbldap_find_state(LDAP *ld)
 634 {
 635         struct smbldap_state_lookup *t;
 636
 637         for (t = smbldap_state_lookup_list; t; t = t->next) {
 638                 if (t->ld == ld) {
 639                         return t->smbldap_state;
 640                 }
 641         }
 642         return NULL;
 643 }
 644
 645 static void smbldap_delete_state(struct smbldap_state *smbldap_state)
 646 {
 647         struct smbldap_state_lookup *t;
 648
 649         for (t = smbldap_state_lookup_list; t; t = t->next) {
 650                 if (t->smbldap_state == smbldap_state) {
 651                         DLIST_REMOVE(smbldap_state_lookup_list, t);
 652                         SAFE_FREE(t);
 653                         return;
 654                 }
 655         }
 656 }
 657
 658 static void smbldap_store_state(LDAP *ld, struct smbldap_state *smbldap_state)
 659 {
 660         struct smbldap_state *tmp_ldap_state;
 661         struct smbldap_state_lookup *t;
 662
 663         if ((tmp_ldap_state = smbldap_find_state(ld))) {
 664                 SMB_ASSERT(tmp_ldap_state == smbldap_state);
 665                 return;
 666         }
 667
 668         t = SMB_XMALLOC_P(struct smbldap_state_lookup);
 669         ZERO_STRUCTP(t);
 670
 671         DLIST_ADD_END(smbldap_state_lookup_list, t, struct smbldap_state_lookup *);
 672         t->ld = ld;
 673         t->smbldap_state = smbldap_state;
 674 }
 675
 676 /********************************************************************
 677  start TLS on an existing LDAP connection
 678 *******************************************************************/
 679
 680 int smb_ldap_start_tls(LDAP *ldap_struct, int version)
 681 {
 682 #ifdef LDAP_OPT_X_TLS
 683         int rc;
 684 #endif
 685
 686         if (lp_ldap_ssl() != LDAP_SSL_START_TLS) {
 687                 return LDAP_SUCCESS;
 688         }
 689
 690 #ifdef LDAP_OPT_X_TLS
 691         if (version != LDAP_VERSION3) {
 692                 DEBUG(0, ("Need LDAPv3 for Start TLS\n"));
 693                 return LDAP_OPERATIONS_ERROR;
 694         }
 695
 696         if ((rc = ldap_start_tls_s (ldap_struct, NULL, NULL)) != LDAP_SUCCESS)  {
 697                 DEBUG(0,("Failed to issue the StartTLS instruction: %s\n",
 698                          ldap_err2string(rc)));
 699                 return rc;
 700         }
 701
 702         DEBUG (3, ("StartTLS issued: using a TLS connection\n"));
 703         return LDAP_SUCCESS;
 704 #else
 705         DEBUG(0,("StartTLS not supported by LDAP client libraries!\n"));
 706         return LDAP_OPERATIONS_ERROR;
 707 #endif
 708 }
 709
 710 /********************************************************************
 711  setup a connection to the LDAP server based on a uri
 712 *******************************************************************/
 713
 714 int smb_ldap_setup_conn(LDAP **ldap_struct, const char *uri)
 715 {
 716         int rc;
 717
 718         DEBUG(10, ("smb_ldap_setup_connection: %s\n", uri));
 719
 720 #ifdef HAVE_LDAP_INITIALIZE
 721
 722         rc = ldap_initialize(ldap_struct, uri);
 723         if (rc) {
 724                 DEBUG(0, ("ldap_initialize: %s\n", ldap_err2string(rc)));
 725         }
 726
 727         return rc;
 728 #else
 729
 730         /* Parse the string manually */
 731
 732         {
 733                 int port = 0;
 734                 fstring protocol;
 735                 fstring host;
 736                 SMB_ASSERT(sizeof(protocol)>10 && sizeof(host)>254);
 737
 738
 739                 /* skip leading "URL:" (if any) */
 740                 if ( strnequal( uri, "URL:", 4 ) ) {
 741                         uri += 4;
 742                 }
 743
 744                 sscanf(uri, "%10[^:]://%254[^:/]:%d", protocol, host, &port);
 745
 746                 if (port == 0) {
 747                         if (strequal(protocol, "ldap")) {
 748                                 port = LDAP_PORT;
 749                         } else if (strequal(protocol, "ldaps")) {
 750                                 port = LDAPS_PORT;
 751                         } else {
 752                                 DEBUG(0, ("unrecognised protocol (%s)!\n", protocol));
 753                         }
 754                 }
 755
 756                 if ((*ldap_struct = ldap_init(host, port)) == NULL)     {
 757                         DEBUG(0, ("ldap_init failed !\n"));
 758                         return LDAP_OPERATIONS_ERROR;
 759                 }
 760
 761                 if (strequal(protocol, "ldaps")) {
 762 #ifdef LDAP_OPT_X_TLS
 763                         int tls = LDAP_OPT_X_TLS_HARD;
 764                         if (ldap_set_option (*ldap_struct, LDAP_OPT_X_TLS, &tls) != LDAP_SUCCESS)
 765                         {
 766                                 DEBUG(0, ("Failed to setup a TLS session\n"));
 767                         }
 768
 769                         DEBUG(3,("LDAPS option set...!\n"));
 770 #else
 771                         DEBUG(0,("smbldap_open_connection: Secure connection not supported by LDAP client libraries!\n"));
 772                         return LDAP_OPERATIONS_ERROR;
 773 #endif /* LDAP_OPT_X_TLS */
 774                 }
 775         }
 776 #endif /* HAVE_LDAP_INITIALIZE */
 777
 778
 779         /* now set connection timeout */
 780 #ifdef LDAP_X_OPT_CONNECT_TIMEOUT /* Netscape */
 781         {
 782                 int ct = lp_ldap_connection_timeout()*1000;
 783                 rc = ldap_set_option(*ldap_struct, LDAP_X_OPT_CONNECT_TIMEOUT, &ct);
 784                 if (rc != LDAP_SUCCESS) {
 785                         DEBUG(0,("Failed to setup an ldap connection timeout %d: %s\n",
 786                                 ct, ldap_err2string(rc)));
 787                 }
 788         }
 789 #elif defined (LDAP_OPT_NETWORK_TIMEOUT) /* OpenLDAP */
 790         {
 791                 struct timeval ct;
 792                 ct.tv_usec = 0;
 793                 ct.tv_sec = lp_ldap_connection_timeout();
 794                 rc = ldap_set_option(*ldap_struct, LDAP_OPT_NETWORK_TIMEOUT, &ct);
 795                 if (rc != LDAP_SUCCESS) {
 796                         DEBUG(0,("Failed to setup an ldap connection timeout %d: %s\n",
 797                                 (int)ct.tv_sec, ldap_err2string(rc)));
 798                 }
 799         }
 800 #endif
 801
 802         return LDAP_SUCCESS;
 803 }
 804
 805 /********************************************************************
 806  try to upgrade to Version 3 LDAP if not already, in either case return current
 807  version
 808  *******************************************************************/
 809
 810 int smb_ldap_upgrade_conn(LDAP *ldap_struct, int *new_version)
 811 {
 812         int version;
 813         int rc;
 814
 815         /* assume the worst */
 816         *new_version = LDAP_VERSION2;
 817
 818         rc = ldap_get_option(ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
 819         if (rc) {
 820                 return rc;
 821         }
 822
 823         if (version == LDAP_VERSION3) {
 824                 *new_version = LDAP_VERSION3;
 825                 return LDAP_SUCCESS;
 826         }
 827
 828         /* try upgrade */
 829         version = LDAP_VERSION3;
 830         rc = ldap_set_option (ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
 831         if (rc) {
 832                 return rc;
 833         }
 834
 835         *new_version = LDAP_VERSION3;
 836         return LDAP_SUCCESS;
 837 }
 838
 839 /*******************************************************************
 840  open a connection to the ldap server (just until the bind)
 841  ******************************************************************/
 842
 843 int smb_ldap_setup_full_conn(LDAP **ldap_struct, const char *uri)
 844 {
 845         int rc, version;
 846
 847         rc = smb_ldap_setup_conn(ldap_struct, uri);
 848         if (rc) {
 849                 return rc;
 850         }
 851
 852         rc = smb_ldap_upgrade_conn(*ldap_struct, &version);
 853         if (rc) {
 854                 return rc;
 855         }
 856
 857         rc = smb_ldap_start_tls(*ldap_struct, version);
 858         if (rc) {
 859                 return rc;
 860         }
 861
 862         return LDAP_SUCCESS;
 863 }
 864
 865 /*******************************************************************
 866  open a connection to the ldap server.
 867 ******************************************************************/
 868 static int smbldap_open_connection (struct smbldap_state *ldap_state)
 869
 870 {
 871         int rc = LDAP_SUCCESS;
 872         int version;
 873         LDAP **ldap_struct = &ldap_state->ldap_struct;
 874
 875         rc = smb_ldap_setup_conn(ldap_struct, ldap_state->uri);
 876         if (rc) {
 877                 return rc;
 878         }
 879
 880         /* Store the LDAP pointer in a lookup list */
 881
 882         smbldap_store_state(*ldap_struct, ldap_state);
 883
 884         /* Upgrade to LDAPv3 if possible */
 885
 886         rc = smb_ldap_upgrade_conn(*ldap_struct, &version);
 887         if (rc) {
 888                 return rc;
 889         }
 890
 891         /* Start TLS if required */
 892
 893         rc = smb_ldap_start_tls(*ldap_struct, version);
 894         if (rc) {
 895                 return rc;
 896         }
 897
 898         DEBUG(2, ("smbldap_open_connection: connection opened\n"));
 899         return rc;
 900 }
 901
 902 /*******************************************************************
 903  a rebind function for authenticated referrals
 904  This version takes a void* that we can shove useful stuff in :-)
 905 ******************************************************************/
 906 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 907 #else
 908 static int rebindproc_with_state  (LDAP * ld, char **whop, char **credp,
 909                                    int *methodp, int freeit, void *arg)
 910 {
 911         struct smbldap_state *ldap_state = arg;
 912
 913         /** @TODO Should we be doing something to check what servers we rebind to?
 914             Could we get a referral to a machine that we don't want to give our
 915             username and password to? */
 916
 917         if (freeit) {
 918                 SAFE_FREE(*whop);
 919                 if (*credp) {
 920                         memset(*credp, '\0', strlen(*credp));
 921                 }
 922                 SAFE_FREE(*credp);
 923         } else {
 924                 DEBUG(5,("rebind_proc_with_state: Rebinding as \"%s\"\n",
 925                           ldap_state->bind_dn?ldap_state->bind_dn:"[Anonymous bind]"));
 926
 927                 if (ldap_state->anonymous) {
 928                         *whop = NULL;
 929                         *credp = NULL;
 930                 } else {
 931                         *whop = SMB_STRDUP(ldap_state->bind_dn);
 932                         if (!*whop) {
 933                                 return LDAP_NO_MEMORY;
 934                         }
 935                         *credp = SMB_STRDUP(ldap_state->bind_secret);
 936                         if (!*credp) {
 937                                 SAFE_FREE(*whop);
 938                                 return LDAP_NO_MEMORY;
 939                         }
 940                 }
 941                 *methodp = LDAP_AUTH_SIMPLE;
 942         }
 943
 944         GetTimeOfDay(&ldap_state->last_rebind);
 945
 946         return 0;
 947 }
 948 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 949
 950 /*******************************************************************
 951  a rebind function for authenticated referrals
 952  This version takes a void* that we can shove useful stuff in :-)
 953  and actually does the connection.
 954 ******************************************************************/
 955 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 956 static int rebindproc_connect_with_state (LDAP *ldap_struct,
 957                                           LDAP_CONST char *url,
 958                                           ber_tag_t request,
 959                                           ber_int_t msgid, void *arg)
 960 {
 961         struct smbldap_state *ldap_state =
 962                 (struct smbldap_state *)arg;
 963         int rc;
 964         int version;
 965
 966         DEBUG(5,("rebindproc_connect_with_state: Rebinding to %s as \"%s\"\n",
 967                  url, ldap_state->bind_dn?ldap_state->bind_dn:"[Anonymous bind]"));
 968
 969         /* call START_TLS again (ldaps:// is handled by the OpenLDAP library
 970          * itself) before rebinding to another LDAP server to avoid to expose
 971          * our credentials. At least *try* to secure the connection - Guenther */
 972
 973         smb_ldap_upgrade_conn(ldap_struct, &version);
 974         smb_ldap_start_tls(ldap_struct, version);
 975
 976         /** @TODO Should we be doing something to check what servers we rebind to?
 977             Could we get a referral to a machine that we don't want to give our
 978             username and password to? */
 979
 980         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
 981
 982         /* only set the last rebind timestamp when we did rebind after a
 983          * non-read LDAP operation. That way we avoid the replication sleep
 984          * after a simple redirected search operation - Guenther */
 985
 986         switch (request) {
 987
 988                 case LDAP_REQ_MODIFY:
 989                 case LDAP_REQ_ADD:
 990                 case LDAP_REQ_DELETE:
 991                 case LDAP_REQ_MODDN:
 992                 case LDAP_REQ_EXTENDED:
 993                         DEBUG(10,("rebindproc_connect_with_state: "
 994                                 "setting last_rebind timestamp "
 995                                 "(req: 0x%02x)\n", (unsigned int)request));
 996                         GetTimeOfDay(&ldap_state->last_rebind);
 997                         break;
 998                 default:
 999                         ZERO_STRUCT(ldap_state->last_rebind);
1000                         break;
1001         }
1002
1003         return rc;
1004 }
1005 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1006
1007 /*******************************************************************
1008  Add a rebind function for authenticated referrals
1009 ******************************************************************/
1010 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1011 #else
1012 # if LDAP_SET_REBIND_PROC_ARGS == 2
1013 static int rebindproc (LDAP *ldap_struct, char **whop, char **credp,
1014                        int *method, int freeit )
1015 {
1016         struct smbldap_state *ldap_state = smbldap_find_state(ldap_struct);
1017
1018         return rebindproc_with_state(ldap_struct, whop, credp,
1019                                      method, freeit, ldap_state);
1020
1021 }
1022 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
1023 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1024
1025 /*******************************************************************
1026  a rebind function for authenticated referrals
1027  this also does the connection, but no void*.
1028 ******************************************************************/
1029 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1030 # if LDAP_SET_REBIND_PROC_ARGS == 2
1031 static int rebindproc_connect (LDAP * ld, LDAP_CONST char *url, int request,
1032                                ber_int_t msgid)
1033 {
1034         struct smbldap_state *ldap_state = smbldap_find_state(ld);
1035
1036         return rebindproc_connect_with_state(ld, url, (ber_tag_t)request, msgid,
1037                                              ldap_state);
1038 }
1039 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
1040 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1041
1042 /*******************************************************************
1043  connect to the ldap server under system privilege.
1044 ******************************************************************/
1045 static int smbldap_connect_system(struct smbldap_state *ldap_state, LDAP * ldap_struct)
1046 {
1047         int rc;
1048         int version;
1049
1050         if (!ldap_state->anonymous && !ldap_state->bind_dn) {
1051
1052                 /* get the default dn and password only if they are not set already */
1053                 if (!fetch_ldap_pw(&ldap_state->bind_dn, &ldap_state->bind_secret)) {
1054                         DEBUG(0, ("ldap_connect_system: Failed to retrieve password from secrets.tdb\n"));
1055                         return LDAP_INVALID_CREDENTIALS;
1056                 }
1057         }
1058
1059         /* removed the sasl_bind_s "EXTERNAL" stuff, as my testsuite
1060            (OpenLDAP) doesnt' seem to support it */
1061
1062         DEBUG(10,("ldap_connect_system: Binding to ldap server %s as \"%s\"\n",
1063                   ldap_state->uri, ldap_state->bind_dn));
1064
1065 #ifdef HAVE_LDAP_SET_REBIND_PROC
1066 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1067 # if LDAP_SET_REBIND_PROC_ARGS == 2
1068         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect);
1069 # endif
1070 # if LDAP_SET_REBIND_PROC_ARGS == 3
1071         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect_with_state, (void *)ldap_state);
1072 # endif
1073 #else /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1074 # if LDAP_SET_REBIND_PROC_ARGS == 2
1075         ldap_set_rebind_proc(ldap_struct, &rebindproc);
1076 # endif
1077 # if LDAP_SET_REBIND_PROC_ARGS == 3
1078         ldap_set_rebind_proc(ldap_struct, &rebindproc_with_state, (void *)ldap_state);
1079 # endif
1080 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1081 #endif
1082
1083         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
1084
1085         if (rc != LDAP_SUCCESS) {
1086                 char *ld_error = NULL;
1087                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING,
1088                                 &ld_error);
1089                 DEBUG(ldap_state->num_failures ? 2 : 0,
1090                       ("failed to bind to server %s with dn=\"%s\" Error: %s\n\t%s\n",
1091                                ldap_state->uri,
1092                                ldap_state->bind_dn ? ldap_state->bind_dn : "[Anonymous bind]",
1093                                ldap_err2string(rc),
1094                                ld_error ? ld_error : "(unknown)"));
1095                 SAFE_FREE(ld_error);
1096                 ldap_state->num_failures++;
1097                 return rc;
1098         }
1099
1100         ldap_state->num_failures = 0;
1101         ldap_state->paged_results = False;
1102
1103         ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
1104
1105         if (smbldap_has_control(ldap_state->ldap_struct, ADS_PAGE_CTL_OID) && version == 3) {
1106                 ldap_state->paged_results = True;
1107         }
1108
1109         DEBUG(3, ("ldap_connect_system: successful connection to the LDAP server\n"));
1110         DEBUGADD(10, ("ldap_connect_system: LDAP server %s support paged results\n",
1111                 ldap_state->paged_results ? "does" : "does not"));
1112         return rc;
1113 }
1114
1115 static void smbldap_idle_fn(struct event_context *event_ctx,
1116                             struct timed_event *te,
1117                             struct timeval now,
1118                             void *private_data);
1119
1120 /**********************************************************************
1121  Connect to LDAP server (called before every ldap operation)
1122 *********************************************************************/
1123 static int smbldap_open(struct smbldap_state *ldap_state)
1124 {
1125         int rc, opt_rc;
1126         bool reopen = False;
1127         SMB_ASSERT(ldap_state);
1128
1129         if ((ldap_state->ldap_struct != NULL) && ((ldap_state->last_ping + SMBLDAP_DONT_PING_TIME) < time(NULL))) {
1130
1131 #ifdef HAVE_UNIXSOCKET
1132                 struct sockaddr_un addr;
1133 #else
1134                 struct sockaddr addr;
1135 #endif
1136                 socklen_t len = sizeof(addr);
1137                 int sd;
1138
1139                 opt_rc = ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_DESC, &sd);
1140                 if (opt_rc == 0 && (getpeername(sd, (struct sockaddr *) &addr, &len)) < 0 )
1141                         reopen = True;
1142
1143 #ifdef HAVE_UNIXSOCKET
1144                 if (opt_rc == 0 && addr.sun_family == AF_UNIX)
1145                         reopen = True;
1146 #endif
1147                 if (reopen) {
1148                         /* the other end has died. reopen. */
1149                         ldap_unbind(ldap_state->ldap_struct);
1150                         ldap_state->ldap_struct = NULL;
1151                         ldap_state->last_ping = (time_t)0;
1152                 } else {
1153                         ldap_state->last_ping = time(NULL);
1154                 }
1155         }
1156
1157         if (ldap_state->ldap_struct != NULL) {
1158                 DEBUG(11,("smbldap_open: already connected to the LDAP server\n"));
1159                 return LDAP_SUCCESS;
1160         }
1161
1162         if ((rc = smbldap_open_connection(ldap_state))) {
1163                 return rc;
1164         }
1165
1166         if ((rc = smbldap_connect_system(ldap_state, ldap_state->ldap_struct))) {
1167                 ldap_unbind(ldap_state->ldap_struct);
1168                 ldap_state->ldap_struct = NULL;
1169                 return rc;
1170         }
1171
1172
1173         ldap_state->last_ping = time(NULL);
1174         ldap_state->pid = sys_getpid();
1175
1176         TALLOC_FREE(ldap_state->idle_event);
1177
1178         if (ldap_state->event_context != NULL) {
1179                 ldap_state->idle_event = event_add_timed(
1180                         ldap_state->event_context, NULL,
1181                         timeval_current_ofs(SMBLDAP_IDLE_TIME, 0),
1182                         smbldap_idle_fn, ldap_state);
1183         }
1184
1185         DEBUG(4,("The LDAP server is successfully connected\n"));
1186
1187         return LDAP_SUCCESS;
1188 }
1189
1190 /**********************************************************************
1191 Disconnect from LDAP server
1192 *********************************************************************/
1193 static NTSTATUS smbldap_close(struct smbldap_state *ldap_state)
1194 {
1195         if (!ldap_state)
1196                 return NT_STATUS_INVALID_PARAMETER;
1197
1198         if (ldap_state->ldap_struct != NULL) {
1199                 ldap_unbind(ldap_state->ldap_struct);
1200                 ldap_state->ldap_struct = NULL;
1201         }
1202
1203         smbldap_delete_state(ldap_state);
1204
1205         DEBUG(5,("The connection to the LDAP server was closed\n"));
1206         /* maybe free the results here --metze */
1207
1208         return NT_STATUS_OK;
1209 }
1210
1211 static bool got_alarm;
1212
1213 static void (*old_handler)(int);
1214
1215 static void gotalarm_sig(int dummy)
1216 {
1217         got_alarm = True;
1218 }
1219
1220 static int another_ldap_try(struct smbldap_state *ldap_state, int *rc,
1221                             int *attempts, time_t endtime)
1222 {
1223         time_t now = time(NULL);
1224         int open_rc = LDAP_SERVER_DOWN;
1225
1226         if (*rc != LDAP_SERVER_DOWN)
1227                 goto no_next;
1228
1229         if (now >= endtime) {
1230                 smbldap_close(ldap_state);
1231                 *rc = LDAP_TIMEOUT;
1232                 goto no_next;
1233         }
1234
1235         if (*attempts == 0) {
1236                 got_alarm = False;
1237                 old_handler = CatchSignal(SIGALRM, gotalarm_sig);
1238                 alarm(endtime - now);
1239
1240                 if (ldap_state->pid != sys_getpid())
1241                         smbldap_close(ldap_state);
1242         }
1243
1244         while (1) {
1245
1246                 if (*attempts != 0)
1247                         smb_msleep(1000);
1248
1249                 *attempts += 1;
1250
1251                 open_rc = smbldap_open(ldap_state);
1252
1253                 if (open_rc == LDAP_SUCCESS) {
1254                         ldap_state->last_use = now;
1255                         return True;
1256                 }
1257
1258                 if (open_rc == LDAP_INSUFFICIENT_ACCESS) {
1259                         /* The fact that we are non-root or any other
1260                          * access-denied condition will not change in the next
1261                          * round of trying */
1262                         *rc = open_rc;
1263                         break;
1264                 }
1265
1266                 if (got_alarm) {
1267                         *rc = LDAP_TIMEOUT;
1268                         break;
1269                 }
1270
1271                 if (open_rc != LDAP_SUCCESS) {
1272                         DEBUG(1, ("Connection to LDAP server failed for the "
1273                                   "%d try!\n", *attempts));
1274                 }
1275         }
1276
1277  no_next:
1278         CatchSignal(SIGALRM, old_handler);
1279         alarm(0);
1280         ldap_state->last_use = now;
1281         return False;
1282 }
1283
1284 /*********************************************************************
1285  ********************************************************************/
1286
1287 static int smbldap_search_ext(struct smbldap_state *ldap_state,
1288                               const char *base, int scope, const char *filter,
1289                               const char *attrs[], int attrsonly,
1290                               LDAPControl **sctrls, LDAPControl **cctrls,
1291                               int sizelimit, LDAPMessage **res)
1292 {
1293         int             rc = LDAP_SERVER_DOWN;
1294         int             attempts = 0;
1295         char           *utf8_filter;
1296         time_t          endtime = time(NULL)+lp_ldap_timeout();
1297         struct          timeval timeout;
1298         size_t          converted_size;
1299
1300         SMB_ASSERT(ldap_state);
1301
1302         DEBUG(5,("smbldap_search_ext: base => [%s], filter => [%s], "
1303                  "scope => [%d]\n", base, filter, scope));
1304
1305         if (ldap_state->last_rebind.tv_sec > 0) {
1306                 struct timeval  tval;
1307                 int64_t tdiff = 0;
1308                 int             sleep_time = 0;
1309
1310                 ZERO_STRUCT(tval);
1311                 GetTimeOfDay(&tval);
1312
1313                 tdiff = usec_time_diff(&tval, &ldap_state->last_rebind);
1314                 tdiff /= 1000; /* Convert to milliseconds. */
1315
1316                 sleep_time = lp_ldap_replication_sleep()-(int)tdiff;
1317                 sleep_time = MIN(sleep_time, MAX_LDAP_REPLICATION_SLEEP_TIME);
1318
1319                 if (sleep_time > 0) {
1320                         /* we wait for the LDAP replication */
1321                         DEBUG(5,("smbldap_search_ext: waiting %d milliseconds "
1322                                  "for LDAP replication.\n",sleep_time));
1323                         smb_msleep(sleep_time);
1324                         DEBUG(5,("smbldap_search_ext: go on!\n"));
1325                 }
1326                 ZERO_STRUCT(ldap_state->last_rebind);
1327         }
1328
1329         if (!push_utf8_talloc(talloc_tos(), &utf8_filter, filter, &converted_size)) {
1330                 return LDAP_NO_MEMORY;
1331         }
1332
1333         /* Setup timeout for the ldap_search_ext_s call - local and remote. */
1334         timeout.tv_sec = lp_ldap_timeout();
1335         timeout.tv_usec = 0;
1336
1337         /* Setup alarm timeout.... Do we need both of these ? JRA.
1338          * Yes, I think we do need both of these. The server timeout only
1339          * covers the case where the server's operation takes too long. It
1340          * does not cover the case where the request hangs on its way to the
1341          * server. The server side timeout is not strictly necessary, it's
1342          * just a bit more kind to the server. VL. */
1343
1344         got_alarm = 0;
1345         CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
1346         alarm(lp_ldap_timeout());
1347         /* End setup timeout. */
1348
1349         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1350                 rc = ldap_search_ext_s(ldap_state->ldap_struct, base, scope,
1351                                        utf8_filter,
1352                                        CONST_DISCARD(char **, attrs),
1353                                        attrsonly, sctrls, cctrls, &timeout,
1354                                        sizelimit, res);
1355                 if (rc != LDAP_SUCCESS) {
1356                         char *ld_error = NULL;
1357                         int ld_errno;
1358
1359                         ldap_get_option(ldap_state->ldap_struct,
1360                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1361
1362                         ldap_get_option(ldap_state->ldap_struct,
1363                                         LDAP_OPT_ERROR_STRING, &ld_error);
1364                         DEBUG(10, ("Failed search for base: %s, error: %d (%s) "
1365                                    "(%s)\n", base, ld_errno,
1366                                    ldap_err2string(rc),
1367                                    ld_error ? ld_error : "unknown"));
1368                         SAFE_FREE(ld_error);
1369
1370                         if (ld_errno == LDAP_SERVER_DOWN) {
1371                                 ldap_unbind(ldap_state->ldap_struct);
1372                                 ldap_state->ldap_struct = NULL;
1373                         }
1374                 }
1375         }
1376
1377         TALLOC_FREE(utf8_filter);
1378
1379         /* Teardown timeout. */
1380         CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
1381         alarm(0);
1382
1383         if (got_alarm != 0)
1384                 return LDAP_TIMELIMIT_EXCEEDED;
1385
1386         return rc;
1387 }
1388
1389 int smbldap_search(struct smbldap_state *ldap_state,
1390                    const char *base, int scope, const char *filter,
1391                    const char *attrs[], int attrsonly,
1392                    LDAPMessage **res)
1393 {
1394         return smbldap_search_ext(ldap_state, base, scope, filter, attrs,
1395                                   attrsonly, NULL, NULL, LDAP_NO_LIMIT, res);
1396 }
1397
1398 int smbldap_search_paged(struct smbldap_state *ldap_state,
1399                          const char *base, int scope, const char *filter,
1400                          const char **attrs, int attrsonly, int pagesize,
1401                          LDAPMessage **res, void **cookie)
1402 {
1403         LDAPControl     pr;
1404         LDAPControl     **rcontrols;
1405         LDAPControl     *controls[2] = { NULL, NULL};
1406         BerElement      *cookie_be = NULL;
1407         struct berval   *cookie_bv = NULL;
1408         int             tmp = 0, i, rc;
1409         bool            critical = True;
1410
1411         *res = NULL;
1412
1413         DEBUG(3,("smbldap_search_paged: base => [%s], filter => [%s],"
1414                  "scope => [%d], pagesize => [%d]\n",
1415                  base, filter, scope, pagesize));
1416
1417         cookie_be = ber_alloc_t(LBER_USE_DER);
1418         if (cookie_be == NULL) {
1419                 DEBUG(0,("smbldap_create_page_control: ber_alloc_t returns "
1420                          "NULL\n"));
1421                 return LDAP_NO_MEMORY;
1422         }
1423
1424         /* construct cookie */
1425         if (*cookie != NULL) {
1426                 ber_printf(cookie_be, "{iO}", (ber_int_t) pagesize, *cookie);
1427                 ber_bvfree((struct berval *)*cookie); /* don't need it from last time */
1428                 *cookie = NULL;
1429         } else {
1430                 ber_printf(cookie_be, "{io}", (ber_int_t) pagesize, "", 0);
1431         }
1432         ber_flatten(cookie_be, &cookie_bv);
1433
1434         pr.ldctl_oid = CONST_DISCARD(char *, ADS_PAGE_CTL_OID);
1435         pr.ldctl_iscritical = (char) critical;
1436         pr.ldctl_value.bv_len = cookie_bv->bv_len;
1437         pr.ldctl_value.bv_val = cookie_bv->bv_val;
1438
1439         controls[0] = &pr;
1440         controls[1] = NULL;
1441
1442         rc = smbldap_search_ext(ldap_state, base, scope, filter, attrs,
1443                                  0, controls, NULL, LDAP_NO_LIMIT, res);
1444
1445         ber_free(cookie_be, 1);
1446         ber_bvfree(cookie_bv);
1447
1448         if (rc != 0) {
1449                 DEBUG(3,("smbldap_search_paged: smbldap_search_ext(%s) "
1450                          "failed with [%s]\n", filter, ldap_err2string(rc)));
1451                 goto done;
1452         }
1453
1454         DEBUG(3,("smbldap_search_paged: search was successfull\n"));
1455
1456         rc = ldap_parse_result(ldap_state->ldap_struct, *res, NULL, NULL,
1457                                NULL, NULL, &rcontrols,  0);
1458         if (rc != 0) {
1459                 DEBUG(3,("smbldap_search_paged: ldap_parse_result failed " \
1460                          "with [%s]\n", ldap_err2string(rc)));
1461                 goto done;
1462         }
1463
1464         if (rcontrols == NULL)
1465                 goto done;
1466
1467         for (i=0; rcontrols[i]; i++) {
1468
1469                 if (strcmp(ADS_PAGE_CTL_OID, rcontrols[i]->ldctl_oid) != 0)
1470                         continue;
1471
1472                 cookie_be = ber_init(&rcontrols[i]->ldctl_value);
1473                 ber_scanf(cookie_be,"{iO}", &tmp, &cookie_bv);
1474                 /* the berval is the cookie, but must be freed when it is all
1475                    done */
1476                 if (cookie_bv->bv_len)
1477                         *cookie=ber_bvdup(cookie_bv);
1478                 else
1479                         *cookie=NULL;
1480                 ber_bvfree(cookie_bv);
1481                 ber_free(cookie_be, 1);
1482                 break;
1483         }
1484         ldap_controls_free(rcontrols);
1485 done:
1486         return rc;
1487 }
1488
1489 int smbldap_modify(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1490 {
1491         int             rc = LDAP_SERVER_DOWN;
1492         int             attempts = 0;
1493         char           *utf8_dn;
1494         time_t          endtime = time(NULL)+lp_ldap_timeout();
1495         size_t          converted_size;
1496
1497         SMB_ASSERT(ldap_state);
1498
1499         DEBUG(5,("smbldap_modify: dn => [%s]\n", dn ));
1500
1501         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1502                 return LDAP_NO_MEMORY;
1503         }
1504
1505         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1506                 rc = ldap_modify_s(ldap_state->ldap_struct, utf8_dn, attrs);
1507                 if (rc != LDAP_SUCCESS) {
1508                         char *ld_error = NULL;
1509                         int ld_errno;
1510
1511                         ldap_get_option(ldap_state->ldap_struct,
1512                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1513
1514                         ldap_get_option(ldap_state->ldap_struct,
1515                                         LDAP_OPT_ERROR_STRING, &ld_error);
1516                         DEBUG(10, ("Failed to modify dn: %s, error: %d (%s) "
1517                                    "(%s)\n", dn, ld_errno,
1518                                    ldap_err2string(rc),
1519                                    ld_error ? ld_error : "unknown"));
1520                         SAFE_FREE(ld_error);
1521
1522                         if (ld_errno == LDAP_SERVER_DOWN) {
1523                                 ldap_unbind(ldap_state->ldap_struct);
1524                                 ldap_state->ldap_struct = NULL;
1525                         }
1526                 }
1527         }
1528
1529         TALLOC_FREE(utf8_dn);
1530         return rc;
1531 }
1532
1533 int smbldap_add(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1534 {
1535         int             rc = LDAP_SERVER_DOWN;
1536         int             attempts = 0;
1537         char           *utf8_dn;
1538         time_t          endtime = time(NULL)+lp_ldap_timeout();
1539         size_t          converted_size;
1540
1541         SMB_ASSERT(ldap_state);
1542
1543         DEBUG(5,("smbldap_add: dn => [%s]\n", dn ));
1544
1545         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1546                 return LDAP_NO_MEMORY;
1547         }
1548
1549         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1550                 rc = ldap_add_s(ldap_state->ldap_struct, utf8_dn, attrs);
1551                 if (rc != LDAP_SUCCESS) {
1552                         char *ld_error = NULL;
1553                         int ld_errno;
1554
1555                         ldap_get_option(ldap_state->ldap_struct,
1556                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1557
1558                         ldap_get_option(ldap_state->ldap_struct,
1559                                         LDAP_OPT_ERROR_STRING, &ld_error);
1560                         DEBUG(10, ("Failed to add dn: %s, error: %d (%s) "
1561                                    "(%s)\n", dn, ld_errno,
1562                                    ldap_err2string(rc),
1563                                    ld_error ? ld_error : "unknown"));
1564                         SAFE_FREE(ld_error);
1565
1566                         if (ld_errno == LDAP_SERVER_DOWN) {
1567                                 ldap_unbind(ldap_state->ldap_struct);
1568                                 ldap_state->ldap_struct = NULL;
1569                         }
1570                 }
1571         }
1572
1573         TALLOC_FREE(utf8_dn);
1574         return rc;
1575 }
1576
1577 int smbldap_delete(struct smbldap_state *ldap_state, const char *dn)
1578 {
1579         int             rc = LDAP_SERVER_DOWN;
1580         int             attempts = 0;
1581         char           *utf8_dn;
1582         time_t          endtime = time(NULL)+lp_ldap_timeout();
1583         size_t          converted_size;
1584
1585         SMB_ASSERT(ldap_state);
1586
1587         DEBUG(5,("smbldap_delete: dn => [%s]\n", dn ));
1588
1589         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1590                 return LDAP_NO_MEMORY;
1591         }
1592
1593         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1594                 rc = ldap_delete_s(ldap_state->ldap_struct, utf8_dn);
1595                 if (rc != LDAP_SUCCESS) {
1596                         char *ld_error = NULL;
1597                         int ld_errno;
1598
1599                         ldap_get_option(ldap_state->ldap_struct,
1600                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1601
1602                         ldap_get_option(ldap_state->ldap_struct,
1603                                         LDAP_OPT_ERROR_STRING, &ld_error);
1604                         DEBUG(10, ("Failed to delete dn: %s, error: %d (%s) "
1605                                    "(%s)\n", dn, ld_errno,
1606                                    ldap_err2string(rc),
1607                                    ld_error ? ld_error : "unknown"));
1608                         SAFE_FREE(ld_error);
1609
1610                         if (ld_errno == LDAP_SERVER_DOWN) {
1611                                 ldap_unbind(ldap_state->ldap_struct);
1612                                 ldap_state->ldap_struct = NULL;
1613                         }
1614                 }
1615         }
1616
1617         TALLOC_FREE(utf8_dn);
1618         return rc;
1619 }
1620
1621 int smbldap_extended_operation(struct smbldap_state *ldap_state,
1622                                LDAP_CONST char *reqoid, struct berval *reqdata,
1623                                LDAPControl **serverctrls, LDAPControl **clientctrls,
1624                                char **retoidp, struct berval **retdatap)
1625 {
1626         int             rc = LDAP_SERVER_DOWN;
1627         int             attempts = 0;
1628         time_t          endtime = time(NULL)+lp_ldap_timeout();
1629
1630         if (!ldap_state)
1631                 return (-1);
1632
1633         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1634                 rc = ldap_extended_operation_s(ldap_state->ldap_struct, reqoid,
1635                                                reqdata, serverctrls,
1636                                                clientctrls, retoidp, retdatap);
1637                 if (rc != LDAP_SUCCESS) {
1638                         char *ld_error = NULL;
1639                         int ld_errno;
1640
1641                         ldap_get_option(ldap_state->ldap_struct,
1642                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1643
1644                         ldap_get_option(ldap_state->ldap_struct,
1645                                         LDAP_OPT_ERROR_STRING, &ld_error);
1646                         DEBUG(10, ("Extended operation failed with error: "
1647                                    "%d (%s) (%s)\n", ld_errno,
1648                                    ldap_err2string(rc),
1649                                    ld_error ? ld_error : "unknown"));
1650                         SAFE_FREE(ld_error);
1651
1652                         if (ld_errno == LDAP_SERVER_DOWN) {
1653                                 ldap_unbind(ldap_state->ldap_struct);
1654                                 ldap_state->ldap_struct = NULL;
1655                         }
1656                 }
1657         }
1658
1659         return rc;
1660 }
1661
1662 /*******************************************************************
1663  run the search by name.
1664 ******************************************************************/
1665 int smbldap_search_suffix (struct smbldap_state *ldap_state,
1666                            const char *filter, const char **search_attr,
1667                            LDAPMessage ** result)
1668 {
1669         return smbldap_search(ldap_state, lp_ldap_suffix(), LDAP_SCOPE_SUBTREE,
1670                               filter, search_attr, 0, result);
1671 }
1672
1673 static void smbldap_idle_fn(struct event_context *event_ctx,
1674                             struct timed_event *te,
1675                             struct timeval now,
1676                             void *private_data)
1677 {
1678         struct smbldap_state *state = (struct smbldap_state *)private_data;
1679
1680         TALLOC_FREE(state->idle_event);
1681
1682         if (state->ldap_struct == NULL) {
1683                 DEBUG(10,("ldap connection not connected...\n"));
1684                 return;
1685         }
1686
1687         if ((state->last_use+SMBLDAP_IDLE_TIME) > now.tv_sec) {
1688                 DEBUG(10,("ldap connection not idle...\n"));
1689
1690                 state->idle_event = event_add_timed(
1691                         event_ctx, NULL,
1692                         timeval_add(&now, SMBLDAP_IDLE_TIME, 0),
1693                         smbldap_idle_fn,
1694                         private_data);
1695                 return;
1696         }
1697
1698         DEBUG(7,("ldap connection idle...closing connection\n"));
1699         smbldap_close(state);
1700 }
1701
1702 /**********************************************************************
1703  Housekeeping
1704  *********************************************************************/
1705
1706 void smbldap_free_struct(struct smbldap_state **ldap_state)
1707 {
1708         smbldap_close(*ldap_state);
1709
1710         if ((*ldap_state)->bind_secret) {
1711                 memset((*ldap_state)->bind_secret, '\0', strlen((*ldap_state)->bind_secret));
1712         }
1713
1714         SAFE_FREE((*ldap_state)->bind_dn);
1715         SAFE_FREE((*ldap_state)->bind_secret);
1716
1717         TALLOC_FREE((*ldap_state)->idle_event);
1718
1719         *ldap_state = NULL;
1720
1721         /* No need to free any further, as it is talloc()ed */
1722 }
1723
1724
1725 /**********************************************************************
1726  Intitalise the 'general' ldap structures, on which ldap operations may be conducted
1727  *********************************************************************/
1728
1729 NTSTATUS smbldap_init(TALLOC_CTX *mem_ctx, struct event_context *event_ctx,
1730                       const char *location,
1731                       struct smbldap_state **smbldap_state)
1732 {
1733         *smbldap_state = TALLOC_ZERO_P(mem_ctx, struct smbldap_state);
1734         if (!*smbldap_state) {
1735                 DEBUG(0, ("talloc() failed for ldapsam private_data!\n"));
1736                 return NT_STATUS_NO_MEMORY;
1737         }
1738
1739         if (location) {
1740                 (*smbldap_state)->uri = talloc_strdup(mem_ctx, location);
1741         } else {
1742                 (*smbldap_state)->uri = "ldap://localhost";
1743         }
1744
1745         (*smbldap_state)->event_context = event_ctx;
1746
1747         return NT_STATUS_OK;
1748 }
1749
1750  char *smbldap_talloc_dn(TALLOC_CTX *mem_ctx, LDAP *ld,
1751                          LDAPMessage *entry)
1752 {
1753         char *utf8_dn, *unix_dn;
1754         size_t converted_size;
1755
1756         utf8_dn = ldap_get_dn(ld, entry);
1757         if (!utf8_dn) {
1758                 DEBUG (5, ("smbldap_talloc_dn: ldap_get_dn failed\n"));
1759                 return NULL;
1760         }
1761         if (!pull_utf8_talloc(mem_ctx, &unix_dn, utf8_dn, &converted_size)) {
1762                 DEBUG (0, ("smbldap_talloc_dn: String conversion failure utf8 "
1763                            "[%s]\n", utf8_dn));
1764                 return NULL;
1765         }
1766         ldap_memfree(utf8_dn);
1767         return unix_dn;
1768 }
1769
1770 /*******************************************************************
1771  Check if root-dse has a certain Control or Extension
1772 ********************************************************************/
1773
1774 static bool smbldap_check_root_dse(LDAP *ld, const char **attrs, const char *value)
1775 {
1776         LDAPMessage *msg = NULL;
1777         LDAPMessage *entry = NULL;
1778         char **values = NULL;
1779         int rc, num_result, num_values, i;
1780         bool result = False;
1781
1782         if (!attrs[0]) {
1783                 DEBUG(3,("smbldap_check_root_dse: nothing to look for\n"));
1784                 return False;
1785         }
1786
1787         if (!strequal(attrs[0], "supportedExtension") &&
1788             !strequal(attrs[0], "supportedControl") &&
1789             !strequal(attrs[0], "namingContexts")) {
1790                 DEBUG(3,("smbldap_check_root_dse: no idea what to query root-dse for: %s ?\n", attrs[0]));
1791                 return False;
1792         }
1793
1794         rc = ldap_search_s(ld, "", LDAP_SCOPE_BASE,
1795                            "(objectclass=*)", CONST_DISCARD(char **, attrs), 0 , &msg);
1796
1797         if (rc != LDAP_SUCCESS) {
1798                 DEBUG(3,("smbldap_check_root_dse: Could not search rootDSE\n"));
1799                 return False;
1800         }
1801
1802         num_result = ldap_count_entries(ld, msg);
1803
1804         if (num_result != 1) {
1805                 DEBUG(3,("smbldap_check_root_dse: Expected one rootDSE, got %d\n", num_result));
1806                 goto done;
1807         }
1808
1809         entry = ldap_first_entry(ld, msg);
1810
1811         if (entry == NULL) {
1812                 DEBUG(3,("smbldap_check_root_dse: Could not retrieve rootDSE\n"));
1813                 goto done;
1814         }
1815
1816         values = ldap_get_values(ld, entry, attrs[0]);
1817
1818         if (values == NULL) {
1819                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not support any %s\n", attrs[0]));
1820                 goto done;
1821         }
1822
1823         num_values = ldap_count_values(values);
1824
1825         if (num_values == 0) {
1826                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not have any %s\n", attrs[0]));
1827                 goto done;
1828         }
1829
1830         for (i=0; i<num_values; i++) {
1831                 if (strcmp(values[i], value) == 0)
1832                         result = True;
1833         }
1834
1835
1836  done:
1837         if (values != NULL)
1838                 ldap_value_free(values);
1839         if (msg != NULL)
1840                 ldap_msgfree(msg);
1841
1842         return result;
1843
1844 }
1845
1846 /*******************************************************************
1847  Check if LDAP-Server supports a certain Control (OID in string format)
1848 ********************************************************************/
1849
1850 bool smbldap_has_control(LDAP *ld, const char *control)
1851 {
1852         const char *attrs[] = { "supportedControl", NULL };
1853         return smbldap_check_root_dse(ld, attrs, control);
1854 }
1855
1856 /*******************************************************************
1857  Check if LDAP-Server supports a certain Extension (OID in string format)
1858 ********************************************************************/
1859
1860 bool smbldap_has_extension(LDAP *ld, const char *extension)
1861 {
1862         const char *attrs[] = { "supportedExtension", NULL };
1863         return smbldap_check_root_dse(ld, attrs, extension);
1864 }
1865
1866 /*******************************************************************
1867  Check if LDAP-Server holds a given namingContext
1868 ********************************************************************/
1869
1870 bool smbldap_has_naming_context(LDAP *ld, const char *naming_context)
1871 {
1872         const char *attrs[] = { "namingContexts", NULL };
1873         return smbldap_check_root_dse(ld, attrs, naming_context);
1874 }
1875
1876 bool smbldap_set_creds(struct smbldap_state *ldap_state, bool anon, const char *dn, const char *secret)
1877 {
1878         ldap_state->anonymous = anon;
1879
1880         /* free any previously set credential */
1881
1882         SAFE_FREE(ldap_state->bind_dn);
1883         if (ldap_state->bind_secret) {
1884                 /* make sure secrets are zeroed out of memory */
1885                 memset(ldap_state->bind_secret, '\0', strlen(ldap_state->bind_secret));
1886                 SAFE_FREE(ldap_state->bind_secret);
1887         }
1888
1889         if ( ! anon) {
1890                 ldap_state->bind_dn = SMB_STRDUP(dn);
1891                 ldap_state->bind_secret = SMB_STRDUP(secret);
1892         }
1893
1894         return True;
1895 }