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