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
1058                 /* get the default dn and password only if they are not set already */
1059                 if (!fetch_ldap_pw(&ldap_state->bind_dn, &ldap_state->bind_secret)) {
1060                         DEBUG(0, ("ldap_connect_system: Failed to retrieve password from secrets.tdb\n"));
1061                         return LDAP_INVALID_CREDENTIALS;
1062                 }
1063         }
1064
1065         /* removed the sasl_bind_s "EXTERNAL" stuff, as my testsuite
1066            (OpenLDAP) doesnt' seem to support it */
1067
1068         DEBUG(10,("ldap_connect_system: Binding to ldap server %s as \"%s\"\n",
1069                   ldap_state->uri, ldap_state->bind_dn));
1070
1071 #ifdef HAVE_LDAP_SET_REBIND_PROC
1072 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
1073 # if LDAP_SET_REBIND_PROC_ARGS == 2
1074         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect);
1075 # endif
1076 # if LDAP_SET_REBIND_PROC_ARGS == 3
1077         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect_with_state, (void *)ldap_state);
1078 # endif
1079 #else /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1080 # if LDAP_SET_REBIND_PROC_ARGS == 2
1081         ldap_set_rebind_proc(ldap_struct, &rebindproc);
1082 # endif
1083 # if LDAP_SET_REBIND_PROC_ARGS == 3
1084         ldap_set_rebind_proc(ldap_struct, &rebindproc_with_state, (void *)ldap_state);
1085 # endif
1086 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
1087 #endif
1088
1089         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
1090
1091         if (rc != LDAP_SUCCESS) {
1092                 char *ld_error = NULL;
1093                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING,
1094                                 &ld_error);
1095                 DEBUG(ldap_state->num_failures ? 2 : 0,
1096                       ("failed to bind to server %s with dn=\"%s\" Error: %s\n\t%s\n",
1097                                ldap_state->uri,
1098                                ldap_state->bind_dn ? ldap_state->bind_dn : "[Anonymous bind]",
1099                                ldap_err2string(rc),
1100                                ld_error ? ld_error : "(unknown)"));
1101                 SAFE_FREE(ld_error);
1102                 ldap_state->num_failures++;
1103                 return rc;
1104         }
1105
1106         ldap_state->num_failures = 0;
1107         ldap_state->paged_results = False;
1108
1109         ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version);
1110
1111         if (smbldap_has_control(ldap_state->ldap_struct, ADS_PAGE_CTL_OID) && version == 3) {
1112                 ldap_state->paged_results = True;
1113         }
1114
1115         DEBUG(3, ("ldap_connect_system: successful connection to the LDAP server\n"));
1116         DEBUGADD(10, ("ldap_connect_system: LDAP server %s support paged results\n",
1117                 ldap_state->paged_results ? "does" : "does not"));
1118         return rc;
1119 }
1120
1121 static void smbldap_idle_fn(struct event_context *event_ctx,
1122                             struct timed_event *te,
1123                             struct timeval now,
1124                             void *private_data);
1125
1126 /**********************************************************************
1127  Connect to LDAP server (called before every ldap operation)
1128 *********************************************************************/
1129 static int smbldap_open(struct smbldap_state *ldap_state)
1130 {
1131         int rc, opt_rc;
1132         bool reopen = False;
1133         SMB_ASSERT(ldap_state);
1134
1135         if ((ldap_state->ldap_struct != NULL) && ((ldap_state->last_ping + SMBLDAP_DONT_PING_TIME) < time(NULL))) {
1136
1137 #ifdef HAVE_UNIXSOCKET
1138                 struct sockaddr_un addr;
1139 #else
1140                 struct sockaddr addr;
1141 #endif
1142                 socklen_t len = sizeof(addr);
1143                 int sd;
1144
1145                 opt_rc = ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_DESC, &sd);
1146                 if (opt_rc == 0 && (getpeername(sd, (struct sockaddr *) &addr, &len)) < 0 )
1147                         reopen = True;
1148
1149 #ifdef HAVE_UNIXSOCKET
1150                 if (opt_rc == 0 && addr.sun_family == AF_UNIX)
1151                         reopen = True;
1152 #endif
1153                 if (reopen) {
1154                         /* the other end has died. reopen. */
1155                         ldap_unbind(ldap_state->ldap_struct);
1156                         ldap_state->ldap_struct = NULL;
1157                         ldap_state->last_ping = (time_t)0;
1158                 } else {
1159                         ldap_state->last_ping = time(NULL);
1160                 }
1161         }
1162
1163         if (ldap_state->ldap_struct != NULL) {
1164                 DEBUG(11,("smbldap_open: already connected to the LDAP server\n"));
1165                 return LDAP_SUCCESS;
1166         }
1167
1168         if ((rc = smbldap_open_connection(ldap_state))) {
1169                 return rc;
1170         }
1171
1172         if ((rc = smbldap_connect_system(ldap_state, ldap_state->ldap_struct))) {
1173                 ldap_unbind(ldap_state->ldap_struct);
1174                 ldap_state->ldap_struct = NULL;
1175                 return rc;
1176         }
1177
1178
1179         ldap_state->last_ping = time(NULL);
1180         ldap_state->pid = sys_getpid();
1181
1182         TALLOC_FREE(ldap_state->idle_event);
1183
1184         if (ldap_state->event_context != NULL) {
1185                 ldap_state->idle_event = event_add_timed(
1186                         ldap_state->event_context, NULL,
1187                         timeval_current_ofs(SMBLDAP_IDLE_TIME, 0),
1188                         smbldap_idle_fn, ldap_state);
1189         }
1190
1191         DEBUG(4,("The LDAP server is successfully connected\n"));
1192
1193         return LDAP_SUCCESS;
1194 }
1195
1196 /**********************************************************************
1197 Disconnect from LDAP server
1198 *********************************************************************/
1199 static NTSTATUS smbldap_close(struct smbldap_state *ldap_state)
1200 {
1201         if (!ldap_state)
1202                 return NT_STATUS_INVALID_PARAMETER;
1203
1204         if (ldap_state->ldap_struct != NULL) {
1205                 ldap_unbind(ldap_state->ldap_struct);
1206                 ldap_state->ldap_struct = NULL;
1207         }
1208
1209         smbldap_delete_state(ldap_state);
1210
1211         DEBUG(5,("The connection to the LDAP server was closed\n"));
1212         /* maybe free the results here --metze */
1213
1214         return NT_STATUS_OK;
1215 }
1216
1217 static bool got_alarm;
1218
1219 static void (*old_handler)(int);
1220
1221 static void gotalarm_sig(int dummy)
1222 {
1223         got_alarm = True;
1224 }
1225
1226 static int another_ldap_try(struct smbldap_state *ldap_state, int *rc,
1227                             int *attempts, time_t endtime)
1228 {
1229         time_t now = time(NULL);
1230         int open_rc = LDAP_SERVER_DOWN;
1231
1232         if (*rc != LDAP_SERVER_DOWN)
1233                 goto no_next;
1234
1235         if (now >= endtime) {
1236                 smbldap_close(ldap_state);
1237                 *rc = LDAP_TIMEOUT;
1238                 goto no_next;
1239         }
1240
1241         if (*attempts == 0) {
1242                 got_alarm = False;
1243                 old_handler = CatchSignal(SIGALRM, gotalarm_sig);
1244                 alarm(endtime - now);
1245
1246                 if (ldap_state->pid != sys_getpid())
1247                         smbldap_close(ldap_state);
1248         }
1249
1250         while (1) {
1251
1252                 if (*attempts != 0)
1253                         smb_msleep(1000);
1254
1255                 *attempts += 1;
1256
1257                 open_rc = smbldap_open(ldap_state);
1258
1259                 if (open_rc == LDAP_SUCCESS) {
1260                         ldap_state->last_use = now;
1261                         return True;
1262                 }
1263
1264                 if (open_rc == LDAP_INSUFFICIENT_ACCESS) {
1265                         /* The fact that we are non-root or any other
1266                          * access-denied condition will not change in the next
1267                          * round of trying */
1268                         *rc = open_rc;
1269                         break;
1270                 }
1271
1272                 if (got_alarm) {
1273                         *rc = LDAP_TIMEOUT;
1274                         break;
1275                 }
1276
1277                 if (open_rc != LDAP_SUCCESS) {
1278                         DEBUG(1, ("Connection to LDAP server failed for the "
1279                                   "%d try!\n", *attempts));
1280                 }
1281         }
1282
1283  no_next:
1284         CatchSignal(SIGALRM, old_handler);
1285         alarm(0);
1286         ldap_state->last_use = now;
1287         return False;
1288 }
1289
1290 /*********************************************************************
1291  ********************************************************************/
1292
1293 static int smbldap_search_ext(struct smbldap_state *ldap_state,
1294                               const char *base, int scope, const char *filter,
1295                               const char *attrs[], int attrsonly,
1296                               LDAPControl **sctrls, LDAPControl **cctrls,
1297                               int sizelimit, LDAPMessage **res)
1298 {
1299         int             rc = LDAP_SERVER_DOWN;
1300         int             attempts = 0;
1301         char           *utf8_filter;
1302         time_t          endtime = time(NULL)+lp_ldap_timeout();
1303         struct          timeval timeout;
1304         size_t          converted_size;
1305
1306         SMB_ASSERT(ldap_state);
1307
1308         DEBUG(5,("smbldap_search_ext: base => [%s], filter => [%s], "
1309                  "scope => [%d]\n", base, filter, scope));
1310
1311         if (ldap_state->last_rebind.tv_sec > 0) {
1312                 struct timeval  tval;
1313                 int64_t tdiff = 0;
1314                 int             sleep_time = 0;
1315
1316                 ZERO_STRUCT(tval);
1317                 GetTimeOfDay(&tval);
1318
1319                 tdiff = usec_time_diff(&tval, &ldap_state->last_rebind);
1320                 tdiff /= 1000; /* Convert to milliseconds. */
1321
1322                 sleep_time = lp_ldap_replication_sleep()-(int)tdiff;
1323                 sleep_time = MIN(sleep_time, MAX_LDAP_REPLICATION_SLEEP_TIME);
1324
1325                 if (sleep_time > 0) {
1326                         /* we wait for the LDAP replication */
1327                         DEBUG(5,("smbldap_search_ext: waiting %d milliseconds "
1328                                  "for LDAP replication.\n",sleep_time));
1329                         smb_msleep(sleep_time);
1330                         DEBUG(5,("smbldap_search_ext: go on!\n"));
1331                 }
1332                 ZERO_STRUCT(ldap_state->last_rebind);
1333         }
1334
1335         if (!push_utf8_talloc(talloc_tos(), &utf8_filter, filter, &converted_size)) {
1336                 return LDAP_NO_MEMORY;
1337         }
1338
1339         /* Setup timeout for the ldap_search_ext_s call - local and remote. */
1340         timeout.tv_sec = lp_ldap_timeout();
1341         timeout.tv_usec = 0;
1342
1343         /* Setup alarm timeout.... Do we need both of these ? JRA.
1344          * Yes, I think we do need both of these. The server timeout only
1345          * covers the case where the server's operation takes too long. It
1346          * does not cover the case where the request hangs on its way to the
1347          * server. The server side timeout is not strictly necessary, it's
1348          * just a bit more kind to the server. VL. */
1349
1350         got_alarm = 0;
1351         CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
1352         alarm(lp_ldap_timeout());
1353         /* End setup timeout. */
1354
1355         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1356                 rc = ldap_search_ext_s(ldap_state->ldap_struct, base, scope,
1357                                        utf8_filter,
1358                                        CONST_DISCARD(char **, attrs),
1359                                        attrsonly, sctrls, cctrls, &timeout,
1360                                        sizelimit, res);
1361                 if (rc != LDAP_SUCCESS) {
1362                         char *ld_error = NULL;
1363                         int ld_errno;
1364
1365                         ldap_get_option(ldap_state->ldap_struct,
1366                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1367
1368                         ldap_get_option(ldap_state->ldap_struct,
1369                                         LDAP_OPT_ERROR_STRING, &ld_error);
1370                         DEBUG(10, ("Failed search for base: %s, error: %d (%s) "
1371                                    "(%s)\n", base, ld_errno,
1372                                    ldap_err2string(rc),
1373                                    ld_error ? ld_error : "unknown"));
1374                         SAFE_FREE(ld_error);
1375
1376                         if (ld_errno == LDAP_SERVER_DOWN) {
1377                                 ldap_unbind(ldap_state->ldap_struct);
1378                                 ldap_state->ldap_struct = NULL;
1379                         }
1380                 }
1381         }
1382
1383         TALLOC_FREE(utf8_filter);
1384
1385         /* Teardown timeout. */
1386         CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
1387         alarm(0);
1388
1389         if (got_alarm != 0)
1390                 return LDAP_TIMELIMIT_EXCEEDED;
1391
1392         return rc;
1393 }
1394
1395 int smbldap_search(struct smbldap_state *ldap_state,
1396                    const char *base, int scope, const char *filter,
1397                    const char *attrs[], int attrsonly,
1398                    LDAPMessage **res)
1399 {
1400         return smbldap_search_ext(ldap_state, base, scope, filter, attrs,
1401                                   attrsonly, NULL, NULL, LDAP_NO_LIMIT, res);
1402 }
1403
1404 int smbldap_search_paged(struct smbldap_state *ldap_state,
1405                          const char *base, int scope, const char *filter,
1406                          const char **attrs, int attrsonly, int pagesize,
1407                          LDAPMessage **res, void **cookie)
1408 {
1409         LDAPControl     pr;
1410         LDAPControl     **rcontrols;
1411         LDAPControl     *controls[2] = { NULL, NULL};
1412         BerElement      *cookie_be = NULL;
1413         struct berval   *cookie_bv = NULL;
1414         int             tmp = 0, i, rc;
1415         bool            critical = True;
1416
1417         *res = NULL;
1418
1419         DEBUG(3,("smbldap_search_paged: base => [%s], filter => [%s],"
1420                  "scope => [%d], pagesize => [%d]\n",
1421                  base, filter, scope, pagesize));
1422
1423         cookie_be = ber_alloc_t(LBER_USE_DER);
1424         if (cookie_be == NULL) {
1425                 DEBUG(0,("smbldap_create_page_control: ber_alloc_t returns "
1426                          "NULL\n"));
1427                 return LDAP_NO_MEMORY;
1428         }
1429
1430         /* construct cookie */
1431         if (*cookie != NULL) {
1432                 ber_printf(cookie_be, "{iO}", (ber_int_t) pagesize, *cookie);
1433                 ber_bvfree((struct berval *)*cookie); /* don't need it from last time */
1434                 *cookie = NULL;
1435         } else {
1436                 ber_printf(cookie_be, "{io}", (ber_int_t) pagesize, "", 0);
1437         }
1438         ber_flatten(cookie_be, &cookie_bv);
1439
1440         pr.ldctl_oid = CONST_DISCARD(char *, ADS_PAGE_CTL_OID);
1441         pr.ldctl_iscritical = (char) critical;
1442         pr.ldctl_value.bv_len = cookie_bv->bv_len;
1443         pr.ldctl_value.bv_val = cookie_bv->bv_val;
1444
1445         controls[0] = &pr;
1446         controls[1] = NULL;
1447
1448         rc = smbldap_search_ext(ldap_state, base, scope, filter, attrs,
1449                                  0, controls, NULL, LDAP_NO_LIMIT, res);
1450
1451         ber_free(cookie_be, 1);
1452         ber_bvfree(cookie_bv);
1453
1454         if (rc != 0) {
1455                 DEBUG(3,("smbldap_search_paged: smbldap_search_ext(%s) "
1456                          "failed with [%s]\n", filter, ldap_err2string(rc)));
1457                 goto done;
1458         }
1459
1460         DEBUG(3,("smbldap_search_paged: search was successful\n"));
1461
1462         rc = ldap_parse_result(ldap_state->ldap_struct, *res, NULL, NULL,
1463                                NULL, NULL, &rcontrols,  0);
1464         if (rc != 0) {
1465                 DEBUG(3,("smbldap_search_paged: ldap_parse_result failed " \
1466                          "with [%s]\n", ldap_err2string(rc)));
1467                 goto done;
1468         }
1469
1470         if (rcontrols == NULL)
1471                 goto done;
1472
1473         for (i=0; rcontrols[i]; i++) {
1474
1475                 if (strcmp(ADS_PAGE_CTL_OID, rcontrols[i]->ldctl_oid) != 0)
1476                         continue;
1477
1478                 cookie_be = ber_init(&rcontrols[i]->ldctl_value);
1479                 ber_scanf(cookie_be,"{iO}", &tmp, &cookie_bv);
1480                 /* the berval is the cookie, but must be freed when it is all
1481                    done */
1482                 if (cookie_bv->bv_len)
1483                         *cookie=ber_bvdup(cookie_bv);
1484                 else
1485                         *cookie=NULL;
1486                 ber_bvfree(cookie_bv);
1487                 ber_free(cookie_be, 1);
1488                 break;
1489         }
1490         ldap_controls_free(rcontrols);
1491 done:
1492         return rc;
1493 }
1494
1495 int smbldap_modify(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1496 {
1497         int             rc = LDAP_SERVER_DOWN;
1498         int             attempts = 0;
1499         char           *utf8_dn;
1500         time_t          endtime = time(NULL)+lp_ldap_timeout();
1501         size_t          converted_size;
1502
1503         SMB_ASSERT(ldap_state);
1504
1505         DEBUG(5,("smbldap_modify: dn => [%s]\n", dn ));
1506
1507         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1508                 return LDAP_NO_MEMORY;
1509         }
1510
1511         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1512                 rc = ldap_modify_s(ldap_state->ldap_struct, utf8_dn, attrs);
1513                 if (rc != LDAP_SUCCESS) {
1514                         char *ld_error = NULL;
1515                         int ld_errno;
1516
1517                         ldap_get_option(ldap_state->ldap_struct,
1518                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1519
1520                         ldap_get_option(ldap_state->ldap_struct,
1521                                         LDAP_OPT_ERROR_STRING, &ld_error);
1522                         DEBUG(10, ("Failed to modify dn: %s, error: %d (%s) "
1523                                    "(%s)\n", dn, ld_errno,
1524                                    ldap_err2string(rc),
1525                                    ld_error ? ld_error : "unknown"));
1526                         SAFE_FREE(ld_error);
1527
1528                         if (ld_errno == LDAP_SERVER_DOWN) {
1529                                 ldap_unbind(ldap_state->ldap_struct);
1530                                 ldap_state->ldap_struct = NULL;
1531                         }
1532                 }
1533         }
1534
1535         TALLOC_FREE(utf8_dn);
1536         return rc;
1537 }
1538
1539 int smbldap_add(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1540 {
1541         int             rc = LDAP_SERVER_DOWN;
1542         int             attempts = 0;
1543         char           *utf8_dn;
1544         time_t          endtime = time(NULL)+lp_ldap_timeout();
1545         size_t          converted_size;
1546
1547         SMB_ASSERT(ldap_state);
1548
1549         DEBUG(5,("smbldap_add: dn => [%s]\n", dn ));
1550
1551         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1552                 return LDAP_NO_MEMORY;
1553         }
1554
1555         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1556                 rc = ldap_add_s(ldap_state->ldap_struct, utf8_dn, attrs);
1557                 if (rc != LDAP_SUCCESS) {
1558                         char *ld_error = NULL;
1559                         int ld_errno;
1560
1561                         ldap_get_option(ldap_state->ldap_struct,
1562                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1563
1564                         ldap_get_option(ldap_state->ldap_struct,
1565                                         LDAP_OPT_ERROR_STRING, &ld_error);
1566                         DEBUG(10, ("Failed to add dn: %s, error: %d (%s) "
1567                                    "(%s)\n", dn, ld_errno,
1568                                    ldap_err2string(rc),
1569                                    ld_error ? ld_error : "unknown"));
1570                         SAFE_FREE(ld_error);
1571
1572                         if (ld_errno == LDAP_SERVER_DOWN) {
1573                                 ldap_unbind(ldap_state->ldap_struct);
1574                                 ldap_state->ldap_struct = NULL;
1575                         }
1576                 }
1577         }
1578
1579         TALLOC_FREE(utf8_dn);
1580         return rc;
1581 }
1582
1583 int smbldap_delete(struct smbldap_state *ldap_state, const char *dn)
1584 {
1585         int             rc = LDAP_SERVER_DOWN;
1586         int             attempts = 0;
1587         char           *utf8_dn;
1588         time_t          endtime = time(NULL)+lp_ldap_timeout();
1589         size_t          converted_size;
1590
1591         SMB_ASSERT(ldap_state);
1592
1593         DEBUG(5,("smbldap_delete: dn => [%s]\n", dn ));
1594
1595         if (!push_utf8_talloc(talloc_tos(), &utf8_dn, dn, &converted_size)) {
1596                 return LDAP_NO_MEMORY;
1597         }
1598
1599         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1600                 rc = ldap_delete_s(ldap_state->ldap_struct, utf8_dn);
1601                 if (rc != LDAP_SUCCESS) {
1602                         char *ld_error = NULL;
1603                         int ld_errno;
1604
1605                         ldap_get_option(ldap_state->ldap_struct,
1606                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1607
1608                         ldap_get_option(ldap_state->ldap_struct,
1609                                         LDAP_OPT_ERROR_STRING, &ld_error);
1610                         DEBUG(10, ("Failed to delete dn: %s, error: %d (%s) "
1611                                    "(%s)\n", dn, ld_errno,
1612                                    ldap_err2string(rc),
1613                                    ld_error ? ld_error : "unknown"));
1614                         SAFE_FREE(ld_error);
1615
1616                         if (ld_errno == LDAP_SERVER_DOWN) {
1617                                 ldap_unbind(ldap_state->ldap_struct);
1618                                 ldap_state->ldap_struct = NULL;
1619                         }
1620                 }
1621         }
1622
1623         TALLOC_FREE(utf8_dn);
1624         return rc;
1625 }
1626
1627 int smbldap_extended_operation(struct smbldap_state *ldap_state,
1628                                LDAP_CONST char *reqoid, struct berval *reqdata,
1629                                LDAPControl **serverctrls, LDAPControl **clientctrls,
1630                                char **retoidp, struct berval **retdatap)
1631 {
1632         int             rc = LDAP_SERVER_DOWN;
1633         int             attempts = 0;
1634         time_t          endtime = time(NULL)+lp_ldap_timeout();
1635
1636         if (!ldap_state)
1637                 return (-1);
1638
1639         while (another_ldap_try(ldap_state, &rc, &attempts, endtime)) {
1640                 rc = ldap_extended_operation_s(ldap_state->ldap_struct, reqoid,
1641                                                reqdata, serverctrls,
1642                                                clientctrls, retoidp, retdatap);
1643                 if (rc != LDAP_SUCCESS) {
1644                         char *ld_error = NULL;
1645                         int ld_errno;
1646
1647                         ldap_get_option(ldap_state->ldap_struct,
1648                                         LDAP_OPT_ERROR_NUMBER, &ld_errno);
1649
1650                         ldap_get_option(ldap_state->ldap_struct,
1651                                         LDAP_OPT_ERROR_STRING, &ld_error);
1652                         DEBUG(10, ("Extended operation failed with error: "
1653                                    "%d (%s) (%s)\n", ld_errno,
1654                                    ldap_err2string(rc),
1655                                    ld_error ? ld_error : "unknown"));
1656                         SAFE_FREE(ld_error);
1657
1658                         if (ld_errno == LDAP_SERVER_DOWN) {
1659                                 ldap_unbind(ldap_state->ldap_struct);
1660                                 ldap_state->ldap_struct = NULL;
1661                         }
1662                 }
1663         }
1664
1665         return rc;
1666 }
1667
1668 /*******************************************************************
1669  run the search by name.
1670 ******************************************************************/
1671 int smbldap_search_suffix (struct smbldap_state *ldap_state,
1672                            const char *filter, const char **search_attr,
1673                            LDAPMessage ** result)
1674 {
1675         return smbldap_search(ldap_state, lp_ldap_suffix(), LDAP_SCOPE_SUBTREE,
1676                               filter, search_attr, 0, result);
1677 }
1678
1679 static void smbldap_idle_fn(struct event_context *event_ctx,
1680                             struct timed_event *te,
1681                             struct timeval now,
1682                             void *private_data)
1683 {
1684         struct smbldap_state *state = (struct smbldap_state *)private_data;
1685
1686         TALLOC_FREE(state->idle_event);
1687
1688         if (state->ldap_struct == NULL) {
1689                 DEBUG(10,("ldap connection not connected...\n"));
1690                 return;
1691         }
1692
1693         if ((state->last_use+SMBLDAP_IDLE_TIME) > now.tv_sec) {
1694                 DEBUG(10,("ldap connection not idle...\n"));
1695
1696                 state->idle_event = event_add_timed(
1697                         event_ctx, NULL,
1698                         timeval_add(&now, SMBLDAP_IDLE_TIME, 0),
1699                         smbldap_idle_fn,
1700                         private_data);
1701                 return;
1702         }
1703
1704         DEBUG(7,("ldap connection idle...closing connection\n"));
1705         smbldap_close(state);
1706 }
1707
1708 /**********************************************************************
1709  Housekeeping
1710  *********************************************************************/
1711
1712 void smbldap_free_struct(struct smbldap_state **ldap_state)
1713 {
1714         smbldap_close(*ldap_state);
1715
1716         if ((*ldap_state)->bind_secret) {
1717                 memset((*ldap_state)->bind_secret, '\0', strlen((*ldap_state)->bind_secret));
1718         }
1719
1720         SAFE_FREE((*ldap_state)->bind_dn);
1721         SAFE_FREE((*ldap_state)->bind_secret);
1722
1723         TALLOC_FREE((*ldap_state)->idle_event);
1724
1725         *ldap_state = NULL;
1726
1727         /* No need to free any further, as it is talloc()ed */
1728 }
1729
1730
1731 /**********************************************************************
1732  Intitalise the 'general' ldap structures, on which ldap operations may be conducted
1733  *********************************************************************/
1734
1735 NTSTATUS smbldap_init(TALLOC_CTX *mem_ctx, struct event_context *event_ctx,
1736                       const char *location,
1737                       struct smbldap_state **smbldap_state)
1738 {
1739         *smbldap_state = TALLOC_ZERO_P(mem_ctx, struct smbldap_state);
1740         if (!*smbldap_state) {
1741                 DEBUG(0, ("talloc() failed for ldapsam private_data!\n"));
1742                 return NT_STATUS_NO_MEMORY;
1743         }
1744
1745         if (location) {
1746                 (*smbldap_state)->uri = talloc_strdup(mem_ctx, location);
1747         } else {
1748                 (*smbldap_state)->uri = "ldap://localhost";
1749         }
1750
1751         (*smbldap_state)->event_context = event_ctx;
1752
1753         return NT_STATUS_OK;
1754 }
1755
1756  char *smbldap_talloc_dn(TALLOC_CTX *mem_ctx, LDAP *ld,
1757                          LDAPMessage *entry)
1758 {
1759         char *utf8_dn, *unix_dn;
1760         size_t converted_size;
1761
1762         utf8_dn = ldap_get_dn(ld, entry);
1763         if (!utf8_dn) {
1764                 DEBUG (5, ("smbldap_talloc_dn: ldap_get_dn failed\n"));
1765                 return NULL;
1766         }
1767         if (!pull_utf8_talloc(mem_ctx, &unix_dn, utf8_dn, &converted_size)) {
1768                 DEBUG (0, ("smbldap_talloc_dn: String conversion failure utf8 "
1769                            "[%s]\n", utf8_dn));
1770                 return NULL;
1771         }
1772         ldap_memfree(utf8_dn);
1773         return unix_dn;
1774 }
1775
1776 /*******************************************************************
1777  Check if root-dse has a certain Control or Extension
1778 ********************************************************************/
1779
1780 static bool smbldap_check_root_dse(LDAP *ld, const char **attrs, const char *value)
1781 {
1782         LDAPMessage *msg = NULL;
1783         LDAPMessage *entry = NULL;
1784         char **values = NULL;
1785         int rc, num_result, num_values, i;
1786         bool result = False;
1787
1788         if (!attrs[0]) {
1789                 DEBUG(3,("smbldap_check_root_dse: nothing to look for\n"));
1790                 return False;
1791         }
1792
1793         if (!strequal(attrs[0], "supportedExtension") &&
1794             !strequal(attrs[0], "supportedControl") &&
1795             !strequal(attrs[0], "namingContexts")) {
1796                 DEBUG(3,("smbldap_check_root_dse: no idea what to query root-dse for: %s ?\n", attrs[0]));
1797                 return False;
1798         }
1799
1800         rc = ldap_search_s(ld, "", LDAP_SCOPE_BASE,
1801                            "(objectclass=*)", CONST_DISCARD(char **, attrs), 0 , &msg);
1802
1803         if (rc != LDAP_SUCCESS) {
1804                 DEBUG(3,("smbldap_check_root_dse: Could not search rootDSE\n"));
1805                 return False;
1806         }
1807
1808         num_result = ldap_count_entries(ld, msg);
1809
1810         if (num_result != 1) {
1811                 DEBUG(3,("smbldap_check_root_dse: Expected one rootDSE, got %d\n", num_result));
1812                 goto done;
1813         }
1814
1815         entry = ldap_first_entry(ld, msg);
1816
1817         if (entry == NULL) {
1818                 DEBUG(3,("smbldap_check_root_dse: Could not retrieve rootDSE\n"));
1819                 goto done;
1820         }
1821
1822         values = ldap_get_values(ld, entry, attrs[0]);
1823
1824         if (values == NULL) {
1825                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not support any %s\n", attrs[0]));
1826                 goto done;
1827         }
1828
1829         num_values = ldap_count_values(values);
1830
1831         if (num_values == 0) {
1832                 DEBUG(5,("smbldap_check_root_dse: LDAP Server does not have any %s\n", attrs[0]));
1833                 goto done;
1834         }
1835
1836         for (i=0; i<num_values; i++) {
1837                 if (strcmp(values[i], value) == 0)
1838                         result = True;
1839         }
1840
1841
1842  done:
1843         if (values != NULL)
1844                 ldap_value_free(values);
1845         if (msg != NULL)
1846                 ldap_msgfree(msg);
1847
1848         return result;
1849
1850 }
1851
1852 /*******************************************************************
1853  Check if LDAP-Server supports a certain Control (OID in string format)
1854 ********************************************************************/
1855
1856 bool smbldap_has_control(LDAP *ld, const char *control)
1857 {
1858         const char *attrs[] = { "supportedControl", NULL };
1859         return smbldap_check_root_dse(ld, attrs, control);
1860 }
1861
1862 /*******************************************************************
1863  Check if LDAP-Server supports a certain Extension (OID in string format)
1864 ********************************************************************/
1865
1866 bool smbldap_has_extension(LDAP *ld, const char *extension)
1867 {
1868         const char *attrs[] = { "supportedExtension", NULL };
1869         return smbldap_check_root_dse(ld, attrs, extension);
1870 }
1871
1872 /*******************************************************************
1873  Check if LDAP-Server holds a given namingContext
1874 ********************************************************************/
1875
1876 bool smbldap_has_naming_context(LDAP *ld, const char *naming_context)
1877 {
1878         const char *attrs[] = { "namingContexts", NULL };
1879         return smbldap_check_root_dse(ld, attrs, naming_context);
1880 }
1881
1882 bool smbldap_set_creds(struct smbldap_state *ldap_state, bool anon, const char *dn, const char *secret)
1883 {
1884         ldap_state->anonymous = anon;
1885
1886         /* free any previously set credential */
1887
1888         SAFE_FREE(ldap_state->bind_dn);
1889         if (ldap_state->bind_secret) {
1890                 /* make sure secrets are zeroed out of memory */
1891                 memset(ldap_state->bind_secret, '\0', strlen(ldap_state->bind_secret));
1892                 SAFE_FREE(ldap_state->bind_secret);
1893         }
1894
1895         if ( ! anon) {
1896                 ldap_state->bind_dn = SMB_STRDUP(dn);
1897                 ldap_state->bind_secret = SMB_STRDUP(secret);
1898         }
1899
1900         return True;
1901 }