source3/lib/smbldap.c

   1 /*
   2    Unix SMB/CIFS mplementation.
   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 2 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, write to the Free Software
  22    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  23
  24 */
  25
  26 #include "includes.h"
  27 #include "smbldap.h"
  28
  29 #ifndef LDAP_OPT_SUCCESS
  30 #define LDAP_OPT_SUCCESS 0
  31 #endif
  32
  33 /* Try not to hit the up or down server forever */
  34
  35 #define SMBLDAP_DONT_PING_TIME 10       /* ping only all 10 seconds */
  36 #define SMBLDAP_NUM_RETRIES 8           /* retry only 8 times */
  37
  38 #define SMBLDAP_IDLE_TIME 150           /* After 2.5 minutes disconnect */
  39
  40
  41 /* attributes used by Samba 2.2 */
  42
  43 ATTRIB_MAP_ENTRY attrib_map_v22[] = {
  44         { LDAP_ATTR_UID,                "uid"           },
  45         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
  46         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
  47         { LDAP_ATTR_UNIX_HOME,          "homeDirectory" },
  48         { LDAP_ATTR_PWD_LAST_SET,       "pwdLastSet"    },
  49         { LDAP_ATTR_PWD_CAN_CHANGE,     "pwdCanChange"  },
  50         { LDAP_ATTR_PWD_MUST_CHANGE,    "pwdMustChange" },
  51         { LDAP_ATTR_LOGON_TIME,         "logonTime"     },
  52         { LDAP_ATTR_LOGOFF_TIME,        "logoffTime"    },
  53         { LDAP_ATTR_KICKOFF_TIME,       "kickoffTime"   },
  54         { LDAP_ATTR_CN,                 "cn"            },
  55         { LDAP_ATTR_DISPLAY_NAME,       "displayName"   },
  56         { LDAP_ATTR_HOME_PATH,          "smbHome"       },
  57         { LDAP_ATTR_HOME_DRIVE,         "homeDrives"    },
  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_LIST_END,           NULL            }
  70 };
  71
  72 /* attributes used by Samba 3.0's sambaSamAccount */
  73
  74 ATTRIB_MAP_ENTRY attrib_map_v30[] = {
  75         { LDAP_ATTR_UID,                "uid"                   },
  76         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
  77         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
  78         { LDAP_ATTR_UNIX_HOME,          "homeDirectory"         },
  79         { LDAP_ATTR_PWD_LAST_SET,       "sambaPwdLastSet"       },
  80         { LDAP_ATTR_PWD_CAN_CHANGE,     "sambaPwdCanChange"     },
  81         { LDAP_ATTR_PWD_MUST_CHANGE,    "sambaPwdMustChange"    },
  82         { LDAP_ATTR_LOGON_TIME,         "sambaLogonTime"        },
  83         { LDAP_ATTR_LOGOFF_TIME,        "sambaLogoffTime"       },
  84         { LDAP_ATTR_KICKOFF_TIME,       "sambaKickoffTime"      },
  85         { LDAP_ATTR_CN,                 "cn"                    },
  86         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
  87         { LDAP_ATTR_HOME_DRIVE,         "sambaHomeDrive"        },
  88         { LDAP_ATTR_HOME_PATH,          "sambaHomePath"         },
  89         { LDAP_ATTR_LOGON_SCRIPT,       "sambaLogonScript"      },
  90         { LDAP_ATTR_PROFILE_PATH,       "sambaProfilePath"      },
  91         { LDAP_ATTR_DESC,               "description"           },
  92         { LDAP_ATTR_USER_WKS,           "sambaUserWorkstations" },
  93         { LDAP_ATTR_USER_SID,           LDAP_ATTRIBUTE_SID      },
  94         { LDAP_ATTR_PRIMARY_GROUP_SID,  "sambaPrimaryGroupSID"  },
  95         { LDAP_ATTR_LMPW,               "sambaLMPassword"       },
  96         { LDAP_ATTR_NTPW,               "sambaNTPassword"       },
  97         { LDAP_ATTR_DOMAIN,             "sambaDomainName"       },
  98         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
  99         { LDAP_ATTR_ACB_INFO,           "sambaAcctFlags"        },
 100         { LDAP_ATTR_MUNGED_DIAL,        "sambaMungedDial"       },
 101         { LDAP_ATTR_BAD_PASSWORD_COUNT, "sambaBadPasswordCount" },
 102         { LDAP_ATTR_BAD_PASSWORD_TIME,  "sambaBadPasswordTime"  },
 103         { LDAP_ATTR_LIST_END,           NULL                    }
 104 };
 105
 106 /* attributes used for allocating RIDs */
 107
 108 ATTRIB_MAP_ENTRY dominfo_attr_list[] = {
 109         { LDAP_ATTR_DOMAIN,             "sambaDomainName"       },
 110         { LDAP_ATTR_NEXT_RID,           "sambaNextRid"          },
 111         { LDAP_ATTR_NEXT_USERRID,       "sambaNextUserRid"      },
 112         { LDAP_ATTR_NEXT_GROUPRID,      "sambaNextGroupRid"     },
 113         { LDAP_ATTR_DOM_SID,            LDAP_ATTRIBUTE_SID      },
 114         { LDAP_ATTR_ALGORITHMIC_RID_BASE,"sambaAlgorithmicRidBase"},
 115         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
 116         { LDAP_ATTR_LIST_END,           NULL                    },
 117 };
 118
 119 /* Samba 3.0 group mapping attributes */
 120
 121 ATTRIB_MAP_ENTRY groupmap_attr_list[] = {
 122         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
 123         { LDAP_ATTR_GROUP_SID,          LDAP_ATTRIBUTE_SID      },
 124         { LDAP_ATTR_GROUP_TYPE,         "sambaGroupType"        },
 125         { LDAP_ATTR_SID_LIST,           "sambaSIDList"          },
 126         { LDAP_ATTR_DESC,               "description"           },
 127         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
 128         { LDAP_ATTR_CN,                 "cn"                    },
 129         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
 130         { LDAP_ATTR_LIST_END,           NULL                    }
 131 };
 132
 133 ATTRIB_MAP_ENTRY groupmap_attr_list_to_delete[] = {
 134         { LDAP_ATTR_GROUP_SID,          LDAP_ATTRIBUTE_SID      },
 135         { LDAP_ATTR_GROUP_TYPE,         "sambaGroupType"        },
 136         { LDAP_ATTR_DESC,               "description"           },
 137         { LDAP_ATTR_DISPLAY_NAME,       "displayName"           },
 138         { LDAP_ATTR_SID_LIST,           "sambaSIDList"          },
 139         { LDAP_ATTR_LIST_END,           NULL                    }
 140 };
 141
 142 /* idmap_ldap sambaUnixIdPool */
 143
 144 ATTRIB_MAP_ENTRY idpool_attr_list[] = {
 145         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
 146         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
 147         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
 148         { LDAP_ATTR_LIST_END,           NULL                    }
 149 };
 150
 151 ATTRIB_MAP_ENTRY sidmap_attr_list[] = {
 152         { LDAP_ATTR_SID,                LDAP_ATTRIBUTE_SID      },
 153         { LDAP_ATTR_UIDNUMBER,          LDAP_ATTRIBUTE_UIDNUMBER},
 154         { LDAP_ATTR_GIDNUMBER,          LDAP_ATTRIBUTE_GIDNUMBER},
 155         { LDAP_ATTR_OBJCLASS,           "objectClass"           },
 156         { LDAP_ATTR_LIST_END,           NULL                    }
 157 };
 158
 159 /**********************************************************************
 160  perform a simple table lookup and return the attribute name
 161  **********************************************************************/
 162
 163  const char* get_attr_key2string( ATTRIB_MAP_ENTRY table[], int key )
 164 {
 165         int i = 0;
 166
 167         while ( table[i].attrib != LDAP_ATTR_LIST_END ) {
 168                 if ( table[i].attrib == key )
 169                         return table[i].name;
 170                 i++;
 171         }
 172
 173         return NULL;
 174 }
 175
 176
 177 /**********************************************************************
 178  Return the list of attribute names from a mapping table
 179  **********************************************************************/
 180
 181  char** get_attr_list( ATTRIB_MAP_ENTRY table[] )
 182 {
 183         char **names;
 184         int i = 0;
 185
 186         while ( table[i].attrib != LDAP_ATTR_LIST_END )
 187                 i++;
 188         i++;
 189
 190         names = (char**)malloc( sizeof(char*)*i );
 191         if ( !names ) {
 192                 DEBUG(0,("get_attr_list: out of memory\n"));
 193                 return NULL;
 194         }
 195
 196         i = 0;
 197         while ( table[i].attrib != LDAP_ATTR_LIST_END ) {
 198                 names[i] = strdup( table[i].name );
 199                 i++;
 200         }
 201         names[i] = NULL;
 202
 203         return names;
 204 }
 205
 206 /*********************************************************************
 207  Cleanup
 208  ********************************************************************/
 209
 210  void free_attr_list( char **list )
 211 {
 212         int i = 0;
 213
 214         if ( !list )
 215                 return;
 216
 217         while ( list[i] ) {
 218                 SAFE_FREE( list[i] );
 219                 i+=1;
 220         }
 221
 222         SAFE_FREE( list );
 223 }
 224
 225 /*******************************************************************
 226  find the ldap password
 227 ******************************************************************/
 228 static BOOL fetch_ldap_pw(char **dn, char** pw)
 229 {
 230         char *key = NULL;
 231         size_t size;
 232
 233         *dn = smb_xstrdup(lp_ldap_admin_dn());
 234
 235         if (asprintf(&key, "%s/%s", SECRETS_LDAP_BIND_PW, *dn) < 0) {
 236                 SAFE_FREE(*dn);
 237                 DEBUG(0, ("fetch_ldap_pw: asprintf failed!\n"));
 238         }
 239
 240         *pw=secrets_fetch(key, &size);
 241         SAFE_FREE(key);
 242
 243         if (!size) {
 244                 /* Upgrade 2.2 style entry */
 245                 char *p;
 246                 char* old_style_key = strdup(*dn);
 247                 char *data;
 248                 fstring old_style_pw;
 249
 250                 if (!old_style_key) {
 251                         DEBUG(0, ("fetch_ldap_pw: strdup failed!\n"));
 252                         return False;
 253                 }
 254
 255                 for (p=old_style_key; *p; p++)
 256                         if (*p == ',') *p = '/';
 257
 258                 data=secrets_fetch(old_style_key, &size);
 259                 if (!size && size < sizeof(old_style_pw)) {
 260                         DEBUG(0,("fetch_ldap_pw: neither ldap secret retrieved!\n"));
 261                         SAFE_FREE(old_style_key);
 262                         SAFE_FREE(*dn);
 263                         return False;
 264                 }
 265
 266                 size = MIN(size, sizeof(fstring)-1);
 267                 strncpy(old_style_pw, data, size);
 268                 old_style_pw[size] = 0;
 269
 270                 SAFE_FREE(data);
 271
 272                 if (!secrets_store_ldap_pw(*dn, old_style_pw)) {
 273                         DEBUG(0,("fetch_ldap_pw: ldap secret could not be upgraded!\n"));
 274                         SAFE_FREE(old_style_key);
 275                         SAFE_FREE(*dn);
 276                         return False;
 277                 }
 278                 if (!secrets_delete(old_style_key)) {
 279                         DEBUG(0,("fetch_ldap_pw: old ldap secret could not be deleted!\n"));
 280                 }
 281
 282                 SAFE_FREE(old_style_key);
 283
 284                 *pw = smb_xstrdup(old_style_pw);
 285         }
 286
 287         return True;
 288 }
 289
 290 /*******************************************************************
 291  Search an attribute and return the first value found.
 292 ******************************************************************/
 293
 294  BOOL smbldap_get_single_attribute (LDAP * ldap_struct, LDAPMessage * entry,
 295                                     const char *attribute, char *value,
 296                                     int max_len)
 297 {
 298         char **values;
 299
 300         if ( !attribute )
 301                 return False;
 302
 303         value[0] = '\0';
 304
 305         if ((values = ldap_get_values (ldap_struct, entry, attribute)) == NULL) {
 306                 DEBUG (10, ("smbldap_get_single_attribute: [%s] = [<does not exist>]\n", attribute));
 307
 308                 return False;
 309         }
 310
 311         if (convert_string(CH_UTF8, CH_UNIX,values[0], -1, value, max_len, False) == (size_t)-1) {
 312                 DEBUG(1, ("smbldap_get_single_attribute: string conversion of [%s] = [%s] failed!\n",
 313                           attribute, values[0]));
 314                 ldap_value_free(values);
 315                 return False;
 316         }
 317
 318         ldap_value_free(values);
 319 #ifdef DEBUG_PASSWORDS
 320         DEBUG (100, ("smbldap_get_single_attribute: [%s] = [%s]\n", attribute, value));
 321 #endif
 322         return True;
 323 }
 324
 325  BOOL smbldap_get_single_pstring (LDAP * ldap_struct, LDAPMessage * entry,
 326                                   const char *attribute, pstring value)
 327 {
 328         return smbldap_get_single_attribute(ldap_struct, entry,
 329                                             attribute, value,
 330                                             sizeof(pstring));
 331 }
 332
 333 /************************************************************************
 334  Routine to manage the LDAPMod structure array
 335  manage memory used by the array, by each struct, and values
 336  ***********************************************************************/
 337
 338  void smbldap_set_mod (LDAPMod *** modlist, int modop, const char *attribute, const char *value)
 339 {
 340         LDAPMod **mods;
 341         int i;
 342         int j;
 343
 344         mods = *modlist;
 345
 346         /* sanity checks on the mod values */
 347
 348         if (attribute == NULL || *attribute == '\0')
 349                 return;
 350 #if 0   /* commented out after discussion with abartlet.  Do not reenable.
 351            left here so other so re-add similar code   --jerry */
 352         if (value == NULL || *value == '\0')
 353                 return;
 354 #endif
 355
 356         if (mods == NULL)
 357         {
 358                 mods = (LDAPMod **) malloc(sizeof(LDAPMod *));
 359                 if (mods == NULL)
 360                 {
 361                         DEBUG(0, ("make_a_mod: out of memory!\n"));
 362                         return;
 363                 }
 364                 mods[0] = NULL;
 365         }
 366
 367         for (i = 0; mods[i] != NULL; ++i) {
 368                 if (mods[i]->mod_op == modop && strequal(mods[i]->mod_type, attribute))
 369                         break;
 370         }
 371
 372         if (mods[i] == NULL)
 373         {
 374                 mods = (LDAPMod **) Realloc (mods, (i + 2) * sizeof (LDAPMod *));
 375                 if (mods == NULL)
 376                 {
 377                         DEBUG(0, ("make_a_mod: out of memory!\n"));
 378                         return;
 379                 }
 380                 mods[i] = (LDAPMod *) malloc(sizeof(LDAPMod));
 381                 if (mods[i] == NULL)
 382                 {
 383                         DEBUG(0, ("make_a_mod: out of memory!\n"));
 384                         return;
 385                 }
 386                 mods[i]->mod_op = modop;
 387                 mods[i]->mod_values = NULL;
 388                 mods[i]->mod_type = strdup(attribute);
 389                 mods[i + 1] = NULL;
 390         }
 391
 392         if (value != NULL)
 393         {
 394                 char *utf8_value = NULL;
 395
 396                 j = 0;
 397                 if (mods[i]->mod_values != NULL) {
 398                         for (; mods[i]->mod_values[j] != NULL; j++);
 399                 }
 400                 mods[i]->mod_values = (char **)Realloc(mods[i]->mod_values,
 401                                                (j + 2) * sizeof (char *));
 402
 403                 if (mods[i]->mod_values == NULL) {
 404                         DEBUG (0, ("make_a_mod: Memory allocation failure!\n"));
 405                         return;
 406                 }
 407
 408                 if (push_utf8_allocate(&utf8_value, value) == (size_t)-1) {
 409                         DEBUG (0, ("make_a_mod: String conversion failure!\n"));
 410                         return;
 411                 }
 412
 413                 mods[i]->mod_values[j] = utf8_value;
 414
 415                 mods[i]->mod_values[j + 1] = NULL;
 416         }
 417         *modlist = mods;
 418 }
 419
 420 /**********************************************************************
 421   Set attribute to newval in LDAP, regardless of what value the
 422   attribute had in LDAP before.
 423 *********************************************************************/
 424
 425  void smbldap_make_mod(LDAP *ldap_struct, LDAPMessage *existing,
 426                       LDAPMod ***mods,
 427                       const char *attribute, const char *newval)
 428 {
 429         char oldval[2048]; /* current largest allowed value is mungeddial */
 430         BOOL existed;
 431
 432         if (existing != NULL) {
 433                 existed = smbldap_get_single_attribute(ldap_struct, existing, attribute, oldval, sizeof(oldval));
 434         } else {
 435                 existed = False;
 436                 *oldval = '\0';
 437         }
 438
 439         /* all of our string attributes are case insensitive */
 440
 441         if (existed && newval && (StrCaseCmp(oldval, newval) == 0)) {
 442
 443                 /* Believe it or not, but LDAP will deny a delete and
 444                    an add at the same time if the values are the
 445                    same... */
 446                 return;
 447         }
 448
 449         if (existed) {
 450                 /* There has been no value before, so don't delete it.
 451                  * Here's a possible race: We might end up with
 452                  * duplicate attributes */
 453                 /* By deleting exactly the value we found in the entry this
 454                  * should be race-free in the sense that the LDAP-Server will
 455                  * deny the complete operation if somebody changed the
 456                  * attribute behind our back. */
 457                 /* This will also allow modifying single valued attributes
 458                  * in Novell NDS. In NDS you have to first remove attribute and then
 459                  * you could add new value */
 460
 461                 smbldap_set_mod(mods, LDAP_MOD_DELETE, attribute, oldval);
 462         }
 463
 464         /* Regardless of the real operation (add or modify)
 465            we add the new value here. We rely on deleting
 466            the old value, should it exist. */
 467
 468         if ((newval != NULL) && (strlen(newval) > 0)) {
 469                 smbldap_set_mod(mods, LDAP_MOD_ADD, attribute, newval);
 470         }
 471 }
 472
 473 /**********************************************************************
 474  Some varients of the LDAP rebind code do not pass in the third 'arg'
 475  pointer to a void*, so we try and work around it by assuming that the
 476  value of the 'LDAP *' pointer is the same as the one we had passed in
 477  **********************************************************************/
 478
 479 struct smbldap_state_lookup {
 480         LDAP *ld;
 481         struct smbldap_state *smbldap_state;
 482         struct smbldap_state_lookup *prev, *next;
 483 };
 484
 485 static struct smbldap_state_lookup *smbldap_state_lookup_list;
 486
 487 static struct smbldap_state *smbldap_find_state(LDAP *ld)
 488 {
 489         struct smbldap_state_lookup *t;
 490
 491         for (t = smbldap_state_lookup_list; t; t = t->next) {
 492                 if (t->ld == ld) {
 493                         return t->smbldap_state;
 494                 }
 495         }
 496         return NULL;
 497 }
 498
 499 static void smbldap_delete_state(struct smbldap_state *smbldap_state)
 500 {
 501         struct smbldap_state_lookup *t;
 502
 503         for (t = smbldap_state_lookup_list; t; t = t->next) {
 504                 if (t->smbldap_state == smbldap_state) {
 505                         DLIST_REMOVE(smbldap_state_lookup_list, t);
 506                         SAFE_FREE(t);
 507                         return;
 508                 }
 509         }
 510 }
 511
 512 static void smbldap_store_state(LDAP *ld, struct smbldap_state *smbldap_state)
 513 {
 514         struct smbldap_state *tmp_ldap_state;
 515         struct smbldap_state_lookup *t;
 516         struct smbldap_state_lookup *tmp;
 517
 518         if ((tmp_ldap_state = smbldap_find_state(ld))) {
 519                 SMB_ASSERT(tmp_ldap_state == smbldap_state);
 520                 return;
 521         }
 522
 523         t = smb_xmalloc(sizeof(*t));
 524         ZERO_STRUCTP(t);
 525
 526         DLIST_ADD_END(smbldap_state_lookup_list, t, tmp);
 527         t->ld = ld;
 528         t->smbldap_state = smbldap_state;
 529 }
 530
 531 /*******************************************************************
 532  open a connection to the ldap server.
 533 ******************************************************************/
 534 static int smbldap_open_connection (struct smbldap_state *ldap_state)
 535
 536 {
 537         int rc = LDAP_SUCCESS;
 538         int version;
 539         BOOL ldap_v3 = False;
 540         LDAP **ldap_struct = &ldap_state->ldap_struct;
 541
 542 #ifdef HAVE_LDAP_INITIALIZE
 543         DEBUG(10, ("smbldap_open_connection: %s\n", ldap_state->uri));
 544
 545         if ((rc = ldap_initialize(ldap_struct, ldap_state->uri)) != LDAP_SUCCESS) {
 546                 DEBUG(0, ("ldap_initialize: %s\n", ldap_err2string(rc)));
 547                 return rc;
 548         }
 549 #else
 550
 551         /* Parse the string manually */
 552
 553         {
 554                 int port = 0;
 555                 fstring protocol;
 556                 fstring host;
 557                 const char *p = ldap_state->uri;
 558                 SMB_ASSERT(sizeof(protocol)>10 && sizeof(host)>254);
 559
 560                 /* skip leading "URL:" (if any) */
 561                 if ( strnequal( p, "URL:", 4 ) ) {
 562                         p += 4;
 563                 }
 564
 565                 sscanf(p, "%10[^:]://%254[^:/]:%d", protocol, host, &port);
 566
 567                 if (port == 0) {
 568                         if (strequal(protocol, "ldap")) {
 569                                 port = LDAP_PORT;
 570                         } else if (strequal(protocol, "ldaps")) {
 571                                 port = LDAPS_PORT;
 572                         } else {
 573                                 DEBUG(0, ("unrecognised protocol (%s)!\n", protocol));
 574                         }
 575                 }
 576
 577                 if ((*ldap_struct = ldap_init(host, port)) == NULL)     {
 578                         DEBUG(0, ("ldap_init failed !\n"));
 579                         return LDAP_OPERATIONS_ERROR;
 580                 }
 581
 582                 if (strequal(protocol, "ldaps")) {
 583 #ifdef LDAP_OPT_X_TLS
 584                         int tls = LDAP_OPT_X_TLS_HARD;
 585                         if (ldap_set_option (*ldap_struct, LDAP_OPT_X_TLS, &tls) != LDAP_SUCCESS)
 586                         {
 587                                 DEBUG(0, ("Failed to setup a TLS session\n"));
 588                         }
 589
 590                         DEBUG(3,("LDAPS option set...!\n"));
 591 #else
 592                         DEBUG(0,("smbldap_open_connection: Secure connection not supported by LDAP client libraries!\n"));
 593                         return LDAP_OPERATIONS_ERROR;
 594 #endif
 595                 }
 596         }
 597 #endif
 598
 599         /* Store the LDAP pointer in a lookup list */
 600
 601         smbldap_store_state(*ldap_struct, ldap_state);
 602
 603         /* Upgrade to LDAPv3 if possible */
 604
 605         if (ldap_get_option(*ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version) == LDAP_OPT_SUCCESS)
 606         {
 607                 if (version != LDAP_VERSION3)
 608                 {
 609                         version = LDAP_VERSION3;
 610                         if (ldap_set_option (*ldap_struct, LDAP_OPT_PROTOCOL_VERSION, &version) == LDAP_OPT_SUCCESS) {
 611                                 ldap_v3 = True;
 612                         }
 613                 } else {
 614                         ldap_v3 = True;
 615                 }
 616         }
 617
 618         if (lp_ldap_ssl() == LDAP_SSL_START_TLS) {
 619 #ifdef LDAP_OPT_X_TLS
 620                 if (ldap_v3) {
 621                         if ((rc = ldap_start_tls_s (*ldap_struct, NULL, NULL)) != LDAP_SUCCESS)
 622                         {
 623                                 DEBUG(0,("Failed to issue the StartTLS instruction: %s\n",
 624                                          ldap_err2string(rc)));
 625                                 return rc;
 626                         }
 627                         DEBUG (3, ("StartTLS issued: using a TLS connection\n"));
 628                 } else {
 629
 630                         DEBUG(0, ("Need LDAPv3 for Start TLS\n"));
 631                         return LDAP_OPERATIONS_ERROR;
 632                 }
 633 #else
 634                 DEBUG(0,("smbldap_open_connection: StartTLS not supported by LDAP client libraries!\n"));
 635                 return LDAP_OPERATIONS_ERROR;
 636 #endif
 637         }
 638
 639         DEBUG(2, ("smbldap_open_connection: connection opened\n"));
 640         return rc;
 641 }
 642
 643
 644 /*******************************************************************
 645  a rebind function for authenticated referrals
 646  This version takes a void* that we can shove useful stuff in :-)
 647 ******************************************************************/
 648 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 649 #else
 650 static int rebindproc_with_state  (LDAP * ld, char **whop, char **credp,
 651                                    int *methodp, int freeit, void *arg)
 652 {
 653         struct smbldap_state *ldap_state = arg;
 654
 655         /** @TODO Should we be doing something to check what servers we rebind to?
 656             Could we get a referral to a machine that we don't want to give our
 657             username and password to? */
 658
 659         if (freeit) {
 660                 SAFE_FREE(*whop);
 661                 memset(*credp, '\0', strlen(*credp));
 662                 SAFE_FREE(*credp);
 663         } else {
 664                 DEBUG(5,("rebind_proc_with_state: Rebinding as \"%s\"\n",
 665                           ldap_state->bind_dn));
 666
 667                 *whop = strdup(ldap_state->bind_dn);
 668                 if (!*whop) {
 669                         return LDAP_NO_MEMORY;
 670                 }
 671                 *credp = strdup(ldap_state->bind_secret);
 672                 if (!*credp) {
 673                         SAFE_FREE(*whop);
 674                         return LDAP_NO_MEMORY;
 675                 }
 676                 *methodp = LDAP_AUTH_SIMPLE;
 677         }
 678
 679         GetTimeOfDay(&ldap_state->last_rebind);
 680
 681         return 0;
 682 }
 683 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 684
 685 /*******************************************************************
 686  a rebind function for authenticated referrals
 687  This version takes a void* that we can shove useful stuff in :-)
 688  and actually does the connection.
 689 ******************************************************************/
 690 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 691 static int rebindproc_connect_with_state (LDAP *ldap_struct,
 692                                           LDAP_CONST char *url,
 693                                           ber_tag_t request,
 694                                           ber_int_t msgid, void *arg)
 695 {
 696         struct smbldap_state *ldap_state = arg;
 697         int rc;
 698         DEBUG(5,("rebindproc_connect_with_state: Rebinding as \"%s\"\n",
 699                  ldap_state->bind_dn));
 700
 701         /** @TODO Should we be doing something to check what servers we rebind to?
 702             Could we get a referral to a machine that we don't want to give our
 703             username and password to? */
 704
 705         rc = ldap_simple_bind_s(ldap_struct, ldap_state->bind_dn, ldap_state->bind_secret);
 706
 707         GetTimeOfDay(&ldap_state->last_rebind);
 708
 709         return rc;
 710 }
 711 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 712
 713 /*******************************************************************
 714  Add a rebind function for authenticated referrals
 715 ******************************************************************/
 716 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 717 #else
 718 # if LDAP_SET_REBIND_PROC_ARGS == 2
 719 static int rebindproc (LDAP *ldap_struct, char **whop, char **credp,
 720                        int *method, int freeit )
 721 {
 722         struct smbldap_state *ldap_state = smbldap_find_state(ldap_struct);
 723
 724         return rebindproc_with_state(ldap_struct, whop, credp,
 725                                      method, freeit, ldap_state);
 726
 727 }
 728 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
 729 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 730
 731 /*******************************************************************
 732  a rebind function for authenticated referrals
 733  this also does the connection, but no void*.
 734 ******************************************************************/
 735 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 736 # if LDAP_SET_REBIND_PROC_ARGS == 2
 737 static int rebindproc_connect (LDAP * ld, LDAP_CONST char *url, int request,
 738                                ber_int_t msgid)
 739 {
 740         struct smbldap_state *ldap_state = smbldap_find_state(ld);
 741
 742         return rebindproc_connect_with_state(ld, url, (ber_tag_t)request, msgid,
 743                                              ldap_state);
 744 }
 745 # endif /*LDAP_SET_REBIND_PROC_ARGS == 2*/
 746 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 747
 748 /*******************************************************************
 749  connect to the ldap server under system privilege.
 750 ******************************************************************/
 751 static int smbldap_connect_system(struct smbldap_state *ldap_state, LDAP * ldap_struct)
 752 {
 753         int rc;
 754         char *ldap_dn;
 755         char *ldap_secret;
 756
 757         /* get the password */
 758         if (!fetch_ldap_pw(&ldap_dn, &ldap_secret)) {
 759                 DEBUG(0, ("ldap_connect_system: Failed to retrieve password from secrets.tdb\n"));
 760                 return LDAP_INVALID_CREDENTIALS;
 761         }
 762
 763         ldap_state->bind_dn = ldap_dn;
 764         ldap_state->bind_secret = ldap_secret;
 765
 766         /* removed the sasl_bind_s "EXTERNAL" stuff, as my testsuite
 767            (OpenLDAP) doesnt' seem to support it */
 768
 769         DEBUG(10,("ldap_connect_system: Binding to ldap server %s as \"%s\"\n",
 770                   ldap_state->uri, ldap_dn));
 771
 772 #if defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)
 773 # if LDAP_SET_REBIND_PROC_ARGS == 2
 774         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect);
 775 # endif
 776 # if LDAP_SET_REBIND_PROC_ARGS == 3
 777         ldap_set_rebind_proc(ldap_struct, &rebindproc_connect_with_state, (void *)ldap_state);
 778 # endif
 779 #else /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 780 # if LDAP_SET_REBIND_PROC_ARGS == 2
 781         ldap_set_rebind_proc(ldap_struct, &rebindproc);
 782 # endif
 783 # if LDAP_SET_REBIND_PROC_ARGS == 3
 784         ldap_set_rebind_proc(ldap_struct, &rebindproc_with_state, (void *)ldap_state);
 785 # endif
 786 #endif /*defined(LDAP_API_FEATURE_X_OPENLDAP) && (LDAP_API_VERSION > 2000)*/
 787
 788         rc = ldap_simple_bind_s(ldap_struct, ldap_dn, ldap_secret);
 789
 790         if (rc != LDAP_SUCCESS) {
 791                 char *ld_error = NULL;
 792                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING,
 793                                 &ld_error);
 794                 DEBUG(ldap_state->num_failures ? 2 : 0,
 795                       ("failed to bind to server with dn= %s Error: %s\n\t%s\n",
 796                                ldap_dn ? ldap_dn : "(unknown)", ldap_err2string(rc),
 797                                ld_error ? ld_error : "(unknown)"));
 798                 SAFE_FREE(ld_error);
 799                 ldap_state->num_failures++;
 800                 return rc;
 801         }
 802
 803         ldap_state->num_failures = 0;
 804
 805         DEBUG(3, ("ldap_connect_system: succesful connection to the LDAP server\n"));
 806         return rc;
 807 }
 808
 809 /**********************************************************************
 810  Connect to LDAP server (called before every ldap operation)
 811 *********************************************************************/
 812 static int smbldap_open(struct smbldap_state *ldap_state)
 813 {
 814         int rc;
 815         SMB_ASSERT(ldap_state);
 816
 817 #ifndef NO_LDAP_SECURITY
 818         if (geteuid() != 0) {
 819                 DEBUG(0, ("smbldap_open: cannot access LDAP when not root..\n"));
 820                 return  LDAP_INSUFFICIENT_ACCESS;
 821         }
 822 #endif
 823
 824         if ((ldap_state->ldap_struct != NULL) && ((ldap_state->last_ping + SMBLDAP_DONT_PING_TIME) < time(NULL))) {
 825                 struct sockaddr_un addr;
 826                 socklen_t len = sizeof(addr);
 827                 int sd;
 828                 if (ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_DESC, &sd) == 0 &&
 829                     getpeername(sd, (struct sockaddr *) &addr, &len) < 0) {
 830                         /* the other end has died. reopen. */
 831                         ldap_unbind_ext(ldap_state->ldap_struct, NULL, NULL);
 832                         ldap_state->ldap_struct = NULL;
 833                         ldap_state->last_ping = (time_t)0;
 834                 } else {
 835                         ldap_state->last_ping = time(NULL);
 836                 }
 837         }
 838
 839         if (ldap_state->ldap_struct != NULL) {
 840                 DEBUG(11,("smbldap_open: already connected to the LDAP server\n"));
 841                 return LDAP_SUCCESS;
 842         }
 843
 844         if ((rc = smbldap_open_connection(ldap_state))) {
 845                 return rc;
 846         }
 847
 848         if ((rc = smbldap_connect_system(ldap_state, ldap_state->ldap_struct))) {
 849                 ldap_unbind_ext(ldap_state->ldap_struct, NULL, NULL);
 850                 ldap_state->ldap_struct = NULL;
 851                 return rc;
 852         }
 853
 854
 855         ldap_state->last_ping = time(NULL);
 856         DEBUG(4,("The LDAP server is succesfully connected\n"));
 857
 858         return LDAP_SUCCESS;
 859 }
 860
 861 /**********************************************************************
 862 Disconnect from LDAP server
 863 *********************************************************************/
 864 static NTSTATUS smbldap_close(struct smbldap_state *ldap_state)
 865 {
 866         if (!ldap_state)
 867                 return NT_STATUS_INVALID_PARAMETER;
 868
 869         if (ldap_state->ldap_struct != NULL) {
 870                 ldap_unbind_ext(ldap_state->ldap_struct, NULL, NULL);
 871                 ldap_state->ldap_struct = NULL;
 872         }
 873
 874         smbldap_delete_state(ldap_state);
 875
 876         DEBUG(5,("The connection to the LDAP server was closed\n"));
 877         /* maybe free the results here --metze */
 878
 879
 880
 881         return NT_STATUS_OK;
 882 }
 883
 884 int smbldap_retry_open(struct smbldap_state *ldap_state, int *attempts)
 885 {
 886         int rc;
 887
 888         SMB_ASSERT(ldap_state && attempts);
 889
 890         if (*attempts != 0) {
 891                 unsigned int sleep_time;
 892                 uint8 rand_byte;
 893
 894                 /* Sleep for a random timeout */
 895                 rand_byte = (char)(sys_random());
 896
 897                 sleep_time = (((*attempts)*(*attempts))/2)*rand_byte*2;
 898                 /* we retry after (0.5, 1, 2, 3, 4.5, 6) seconds
 899                    on average.
 900                  */
 901                 DEBUG(3, ("Sleeping for %u milliseconds before reconnecting\n",
 902                           sleep_time));
 903                 smb_msleep(sleep_time);
 904         }
 905         (*attempts)++;
 906
 907         if ((rc = smbldap_open(ldap_state))) {
 908                 DEBUG(1,("Connection to LDAP Server failed for the %d try!\n",*attempts));
 909                 return rc;
 910         }
 911
 912         return LDAP_SUCCESS;
 913 }
 914
 915
 916 /*********************************************************************
 917  ********************************************************************/
 918
 919 int smbldap_search(struct smbldap_state *ldap_state,
 920                    const char *base, int scope, const char *filter,
 921                    char *attrs[], int attrsonly,
 922                    LDAPMessage **res)
 923 {
 924         int             rc = LDAP_SERVER_DOWN;
 925         int             attempts = 0;
 926         char           *utf8_filter;
 927
 928         SMB_ASSERT(ldap_state);
 929
 930         DEBUG(5,("smbldap_search: base => [%s], filter => [%s], scope => [%d]\n",
 931                 base, filter, scope));
 932
 933         if (ldap_state->last_rebind.tv_sec > 0) {
 934                 struct timeval  tval;
 935                 SMB_BIG_INT     tdiff = 0;
 936                 int             sleep_time = 0;
 937
 938                 ZERO_STRUCT(tval);
 939                 GetTimeOfDay(&tval);
 940
 941                 tdiff = usec_time_diff(&tval, &ldap_state->last_rebind.tv_sec);
 942                 tdiff /= 1000; /* Convert to milliseconds. */
 943
 944                 sleep_time = lp_ldap_replication_sleep()-(int)tdiff;
 945                 sleep_time = MIN(sleep_time, MAX_LDAP_REPLICATION_SLEEP_TIME);
 946
 947                 if (sleep_time > 0) {
 948                         /* we wait for the LDAP replication */
 949                         DEBUG(5,("smbldap_search: waiting %d milliseconds for LDAP replication.\n",sleep_time));
 950                         smb_msleep(sleep_time);
 951                         DEBUG(5,("smbldap_search: go on!\n"));
 952                 }
 953                 ZERO_STRUCT(ldap_state->last_rebind);
 954         }
 955
 956         if (push_utf8_allocate(&utf8_filter, filter) == (size_t)-1) {
 957                 return LDAP_NO_MEMORY;
 958         }
 959
 960         while ((rc == LDAP_SERVER_DOWN) && (attempts < SMBLDAP_NUM_RETRIES)) {
 961
 962                 if ((rc = smbldap_retry_open(ldap_state,&attempts)) != LDAP_SUCCESS)
 963                         continue;
 964
 965                 rc = ldap_search_s(ldap_state->ldap_struct, base, scope,
 966                                    utf8_filter, attrs, attrsonly, res);
 967         }
 968
 969         if (rc == LDAP_SERVER_DOWN) {
 970                 DEBUG(0,("%s: LDAP server is down!\n",FUNCTION_MACRO));
 971                 smbldap_close(ldap_state);
 972         }
 973
 974         ldap_state->last_use = time(NULL);
 975
 976         SAFE_FREE(utf8_filter);
 977         return rc;
 978 }
 979
 980 int smbldap_modify(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
 981 {
 982         int             rc = LDAP_SERVER_DOWN;
 983         int             attempts = 0;
 984         char           *utf8_dn;
 985
 986         SMB_ASSERT(ldap_state);
 987
 988         DEBUG(5,("smbldap_modify: dn => [%s]\n", dn ));
 989
 990         if (push_utf8_allocate(&utf8_dn, dn) == (size_t)-1) {
 991                 return LDAP_NO_MEMORY;
 992         }
 993
 994         while ((rc == LDAP_SERVER_DOWN) && (attempts < SMBLDAP_NUM_RETRIES)) {
 995
 996                 if ((rc = smbldap_retry_open(ldap_state,&attempts)) != LDAP_SUCCESS)
 997                         continue;
 998
 999                 rc = ldap_modify_s(ldap_state->ldap_struct, utf8_dn, attrs);
1000         }
1001
1002         if (rc == LDAP_SERVER_DOWN) {
1003                 DEBUG(0,("%s: LDAP server is down!\n",FUNCTION_MACRO));
1004                 smbldap_close(ldap_state);
1005         }
1006
1007         ldap_state->last_use = time(NULL);
1008
1009         SAFE_FREE(utf8_dn);
1010         return rc;
1011 }
1012
1013 int smbldap_add(struct smbldap_state *ldap_state, const char *dn, LDAPMod *attrs[])
1014 {
1015         int             rc = LDAP_SERVER_DOWN;
1016         int             attempts = 0;
1017         char           *utf8_dn;
1018
1019         SMB_ASSERT(ldap_state);
1020
1021         DEBUG(5,("smbldap_add: dn => [%s]\n", dn ));
1022
1023         if (push_utf8_allocate(&utf8_dn, dn) == (size_t)-1) {
1024                 return LDAP_NO_MEMORY;
1025         }
1026
1027         while ((rc == LDAP_SERVER_DOWN) && (attempts < SMBLDAP_NUM_RETRIES)) {
1028
1029                 if ((rc = smbldap_retry_open(ldap_state,&attempts)) != LDAP_SUCCESS)
1030                         continue;
1031
1032                 rc = ldap_add_s(ldap_state->ldap_struct, utf8_dn, attrs);
1033         }
1034
1035         if (rc == LDAP_SERVER_DOWN) {
1036                 DEBUG(0,("%s: LDAP server is down!\n",FUNCTION_MACRO));
1037                 smbldap_close(ldap_state);
1038         }
1039
1040         ldap_state->last_use = time(NULL);
1041
1042         SAFE_FREE(utf8_dn);
1043         return rc;
1044 }
1045
1046 int smbldap_delete(struct smbldap_state *ldap_state, const char *dn)
1047 {
1048         int             rc = LDAP_SERVER_DOWN;
1049         int             attempts = 0;
1050         char           *utf8_dn;
1051
1052         SMB_ASSERT(ldap_state);
1053
1054         DEBUG(5,("smbldap_delete: dn => [%s]\n", dn ));
1055
1056         if (push_utf8_allocate(&utf8_dn, dn) == (size_t)-1) {
1057                 return LDAP_NO_MEMORY;
1058         }
1059
1060         while ((rc == LDAP_SERVER_DOWN) && (attempts < SMBLDAP_NUM_RETRIES)) {
1061
1062                 if ((rc = smbldap_retry_open(ldap_state,&attempts)) != LDAP_SUCCESS)
1063                         continue;
1064
1065                 rc = ldap_delete_s(ldap_state->ldap_struct, utf8_dn);
1066         }
1067
1068         if (rc == LDAP_SERVER_DOWN) {
1069                 DEBUG(0,("%s: LDAP server is down!\n",FUNCTION_MACRO));
1070                 smbldap_close(ldap_state);
1071         }
1072
1073         ldap_state->last_use = time(NULL);
1074
1075         SAFE_FREE(utf8_dn);
1076         return rc;
1077 }
1078
1079 int smbldap_extended_operation(struct smbldap_state *ldap_state,
1080                                LDAP_CONST char *reqoid, struct berval *reqdata,
1081                                LDAPControl **serverctrls, LDAPControl **clientctrls,
1082                                char **retoidp, struct berval **retdatap)
1083 {
1084         int             rc = LDAP_SERVER_DOWN;
1085         int             attempts = 0;
1086
1087         if (!ldap_state)
1088                 return (-1);
1089
1090         while ((rc == LDAP_SERVER_DOWN) && (attempts < SMBLDAP_NUM_RETRIES)) {
1091
1092                 if ((rc = smbldap_retry_open(ldap_state,&attempts)) != LDAP_SUCCESS)
1093                         continue;
1094
1095                 rc = ldap_extended_operation_s(ldap_state->ldap_struct, reqoid, reqdata,
1096                                                serverctrls, clientctrls, retoidp, retdatap);
1097         }
1098
1099         if (rc == LDAP_SERVER_DOWN) {
1100                 DEBUG(0,("%s: LDAP server is down!\n",FUNCTION_MACRO));
1101                 smbldap_close(ldap_state);
1102         }
1103
1104         ldap_state->last_use = time(NULL);
1105
1106         return rc;
1107 }
1108
1109 /*******************************************************************
1110  run the search by name.
1111 ******************************************************************/
1112 int smbldap_search_suffix (struct smbldap_state *ldap_state, const char *filter,
1113                            char **search_attr, LDAPMessage ** result)
1114 {
1115         int scope = LDAP_SCOPE_SUBTREE;
1116         int rc;
1117
1118         rc = smbldap_search(ldap_state, lp_ldap_suffix(), scope, filter, search_attr, 0, result);
1119
1120         if (rc != LDAP_SUCCESS) {
1121                 char *ld_error = NULL;
1122                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING,
1123                                 &ld_error);
1124                 DEBUG(0,("smbldap_search_suffix: Problem during the LDAP search: %s (%s)\n",
1125                         ld_error?ld_error:"(unknown)", ldap_err2string (rc)));
1126                 SAFE_FREE(ld_error);
1127         }
1128
1129         return rc;
1130 }
1131
1132 static void smbldap_idle_fn(void **data, time_t *interval, time_t now)
1133 {
1134         struct smbldap_state *state = (struct smbldap_state *)(*data);
1135
1136         if (state->ldap_struct == NULL) {
1137                 DEBUG(10,("ldap connection not connected...\n"));
1138                 return;
1139         }
1140
1141         if ((state->last_use+SMBLDAP_IDLE_TIME) > now) {
1142                 DEBUG(10,("ldap connection not idle...\n"));
1143                 return;
1144         }
1145
1146         DEBUG(7,("ldap connection idle...closing connection\n"));
1147         smbldap_close(state);
1148 }
1149
1150 /**********************************************************************
1151  Housekeeping
1152  *********************************************************************/
1153
1154 void smbldap_free_struct(struct smbldap_state **ldap_state)
1155 {
1156         smbldap_close(*ldap_state);
1157
1158         if ((*ldap_state)->bind_secret) {
1159                 memset((*ldap_state)->bind_secret, '\0', strlen((*ldap_state)->bind_secret));
1160         }
1161
1162         SAFE_FREE((*ldap_state)->bind_dn);
1163         SAFE_FREE((*ldap_state)->bind_secret);
1164
1165         smb_unregister_idle_event((*ldap_state)->event_id);
1166
1167         *ldap_state = NULL;
1168
1169         /* No need to free any further, as it is talloc()ed */
1170 }
1171
1172
1173 /**********************************************************************
1174  Intitalise the 'general' ldap structures, on which ldap operations may be conducted
1175  *********************************************************************/
1176
1177 NTSTATUS smbldap_init(TALLOC_CTX *mem_ctx, const char *location, struct smbldap_state **smbldap_state)
1178 {
1179         *smbldap_state = talloc_zero(mem_ctx, sizeof(**smbldap_state));
1180         if (!*smbldap_state) {
1181                 DEBUG(0, ("talloc() failed for ldapsam private_data!\n"));
1182                 return NT_STATUS_NO_MEMORY;
1183         }
1184
1185         if (location) {
1186                 (*smbldap_state)->uri = talloc_strdup(mem_ctx, location);
1187         } else {
1188                 (*smbldap_state)->uri = "ldap://localhost";
1189         }
1190
1191         (*smbldap_state)->event_id =
1192                 smb_register_idle_event(smbldap_idle_fn, (void *)(*smbldap_state),
1193                                         SMBLDAP_IDLE_TIME);
1194
1195         if ((*smbldap_state)->event_id == SMB_EVENT_ID_INVALID) {
1196                 DEBUG(0,("Failed to register LDAP idle event!\n"));
1197                 return NT_STATUS_INVALID_HANDLE;
1198         }
1199
1200         return NT_STATUS_OK;
1201 }
1202
1203 /**********************************************************************
1204  Add the sambaDomain to LDAP, so we don't have to search for this stuff
1205  again.  This is a once-add operation for now.
1206
1207  TODO:  Add other attributes, and allow modification.
1208 *********************************************************************/
1209 static NTSTATUS add_new_domain_info(struct smbldap_state *ldap_state,
1210                                     const char *domain_name)
1211 {
1212         fstring sid_string;
1213         fstring algorithmic_rid_base_string;
1214         pstring filter, dn;
1215         LDAPMod **mods = NULL;
1216         int rc;
1217         int ldap_op;
1218         LDAPMessage *result = NULL;
1219         int num_result;
1220         char **attr_list;
1221         uid_t u_low, u_high;
1222         gid_t g_low, g_high;
1223         uint32 rid_low, rid_high;
1224
1225         slprintf (filter, sizeof (filter) - 1, "(&(%s=%s)(objectclass=%s))",
1226                   get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
1227                   domain_name, LDAP_OBJ_DOMINFO);
1228
1229         attr_list = get_attr_list( dominfo_attr_list );
1230         rc = smbldap_search_suffix(ldap_state, filter, attr_list, &result);
1231         free_attr_list( attr_list );
1232
1233         if (rc != LDAP_SUCCESS) {
1234                 return NT_STATUS_UNSUCCESSFUL;
1235         }
1236
1237         num_result = ldap_count_entries(ldap_state->ldap_struct, result);
1238
1239         if (num_result > 1) {
1240                 DEBUG (0, ("More than domain with that name exists: bailing out!\n"));
1241                 ldap_msgfree(result);
1242                 return NT_STATUS_UNSUCCESSFUL;
1243         }
1244
1245         /* Check if we need to add an entry */
1246         DEBUG(3,("Adding new domain\n"));
1247         ldap_op = LDAP_MOD_ADD;
1248
1249         pstr_sprintf(dn, "%s=%s,%s", get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
1250                 domain_name, lp_ldap_suffix());
1251
1252         /* Free original search */
1253         ldap_msgfree(result);
1254
1255         /* make the changes - the entry *must* not already have samba attributes */
1256         smbldap_set_mod(&mods, LDAP_MOD_ADD, get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
1257                 domain_name);
1258
1259         /* If we don't have an entry, then ask secrets.tdb for what it thinks.
1260            It may choose to make it up */
1261
1262         sid_to_string(sid_string, get_global_sam_sid());
1263         smbldap_set_mod(&mods, LDAP_MOD_ADD, get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOM_SID), sid_string);
1264
1265         slprintf(algorithmic_rid_base_string, sizeof(algorithmic_rid_base_string) - 1, "%i", algorithmic_rid_base());
1266         smbldap_set_mod(&mods, LDAP_MOD_ADD, get_attr_key2string(dominfo_attr_list, LDAP_ATTR_ALGORITHMIC_RID_BASE),
1267                         algorithmic_rid_base_string);
1268         smbldap_set_mod(&mods, LDAP_MOD_ADD, "objectclass", LDAP_OBJ_DOMINFO);
1269
1270         /* add the sambaNext[User|Group]Rid attributes if the idmap ranges are set.
1271            TODO: fix all the places where the line between idmap and normal operations
1272            needed by smbd gets fuzzy   --jerry 2003-08-11                              */
1273
1274         if ( lp_idmap_uid(&u_low, &u_high) && lp_idmap_gid(&g_low, &g_high)
1275                 && get_free_rid_range(&rid_low, &rid_high) )
1276         {
1277                 fstring rid_str;
1278
1279                 fstr_sprintf( rid_str, "%i", rid_high|USER_RID_TYPE );
1280                 DEBUG(10,("setting next available user rid [%s]\n", rid_str));
1281                 smbldap_set_mod(&mods, LDAP_MOD_ADD,
1282                         get_attr_key2string(dominfo_attr_list, LDAP_ATTR_NEXT_USERRID),
1283                         rid_str);
1284
1285                 fstr_sprintf( rid_str, "%i", rid_high|GROUP_RID_TYPE );
1286                 DEBUG(10,("setting next available group rid [%s]\n", rid_str));
1287                 smbldap_set_mod(&mods, LDAP_MOD_ADD,
1288                         get_attr_key2string(dominfo_attr_list, LDAP_ATTR_NEXT_GROUPRID),
1289                         rid_str);
1290
1291         }
1292
1293
1294         switch(ldap_op)
1295         {
1296         case LDAP_MOD_ADD:
1297                 rc = smbldap_add(ldap_state, dn, mods);
1298                 break;
1299         case LDAP_MOD_REPLACE:
1300                 rc = smbldap_modify(ldap_state, dn, mods);
1301                 break;
1302         default:
1303                 DEBUG(0,("Wrong LDAP operation type: %d!\n", ldap_op));
1304                 return NT_STATUS_INVALID_PARAMETER;
1305         }
1306
1307         if (rc!=LDAP_SUCCESS) {
1308                 char *ld_error = NULL;
1309                 ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING, &ld_error);
1310                 DEBUG(1,("failed to %s domain dn= %s with: %s\n\t%s\n",
1311                        ldap_op == LDAP_MOD_ADD ? "add" : "modify",
1312                        dn, ldap_err2string(rc),
1313                        ld_error?ld_error:"unknown"));
1314                 SAFE_FREE(ld_error);
1315
1316                 ldap_mods_free(mods, True);
1317                 return NT_STATUS_UNSUCCESSFUL;
1318         }
1319
1320         DEBUG(2,("added: domain = %s in the LDAP database\n", domain_name));
1321         ldap_mods_free(mods, True);
1322         return NT_STATUS_OK;
1323 }
1324
1325 /**********************************************************************
1326 Search for the domain info entry
1327 *********************************************************************/
1328 NTSTATUS smbldap_search_domain_info(struct smbldap_state *ldap_state,
1329                                     LDAPMessage ** result, const char *domain_name,
1330                                     BOOL try_add)
1331 {
1332         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
1333         pstring filter;
1334         int rc;
1335         char **attr_list;
1336         int count;
1337
1338         pstr_sprintf(filter, "(&(objectClass=%s)(%s=%s))",
1339                 LDAP_OBJ_DOMINFO,
1340                 get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
1341                 domain_name);
1342
1343         DEBUG(2, ("Searching for:[%s]\n", filter));
1344
1345
1346         attr_list = get_attr_list( dominfo_attr_list );
1347         rc = smbldap_search_suffix(ldap_state, filter, attr_list , result);
1348         free_attr_list( attr_list );
1349
1350         if (rc != LDAP_SUCCESS) {
1351                 DEBUG(2,("Problem during LDAPsearch: %s\n", ldap_err2string (rc)));
1352                 DEBUG(2,("Query was: %s, %s\n", lp_ldap_suffix(), filter));
1353         } else if (ldap_count_entries(ldap_state->ldap_struct, *result) < 1) {
1354                 DEBUG(3, ("Got no domain info entries for domain\n"));
1355                 ldap_msgfree(*result);
1356                 *result = NULL;
1357                 if (try_add && NT_STATUS_IS_OK(ret = add_new_domain_info(ldap_state, domain_name))) {
1358                         return smbldap_search_domain_info(ldap_state, result, domain_name, False);
1359                 }
1360                 else {
1361                         DEBUG(0, ("Adding domain info for %s failed with %s\n",
1362                                 domain_name, nt_errstr(ret)));
1363                         return ret;
1364                 }
1365         } else if ((count = ldap_count_entries(ldap_state->ldap_struct, *result)) > 1) {
1366                 DEBUG(0, ("Got too many (%d) domain info entries for domain %s\n",
1367                           count, domain_name));
1368                 ldap_msgfree(*result);
1369                 *result = NULL;
1370                 return ret;
1371         } else {
1372                 return NT_STATUS_OK;
1373         }
1374
1375         return ret;
1376 }
1377
1378 /*******************************************************************
1379  Return a copy of the DN for a LDAPMessage. Convert from utf8 to CH_UNIX.
1380 ********************************************************************/
1381
1382 char *smbldap_get_dn(LDAP *ld, LDAPMessage *entry)
1383 {
1384         char *utf8_dn, *unix_dn;
1385
1386         utf8_dn = ldap_get_dn(ld, entry);
1387         if (!utf8_dn) {
1388                 DEBUG (5, ("smbldap_get_dn: ldap_get_dn failed\n"));
1389                 return NULL;
1390         }
1391         if (pull_utf8_allocate(&unix_dn, utf8_dn) == (size_t)-1) {
1392                 DEBUG (0, ("smbldap_get_dn: String conversion failure utf8 [%s]\n", utf8_dn));
1393                 return NULL;
1394         }
1395         ldap_memfree(utf8_dn);
1396         return unix_dn;
1397 }
1398