source3/passdb/pdb_tdb.c

   1 /*
   2  * Unix SMB/CIFS implementation.
   3  * SMB parameters and setup
   4  * Copyright (C) Andrew Tridgell 1992-1998
   5  * Copyright (C) Simo Sorce 2000-2002
   6  * Copyright (C) Gerald Carter 2000
   7  * Copyright (C) Jeremy Allison 2001
   8  * Copyright (C) Andrew Bartlett 2002
   9  *
  10  * This program is free software; you can redistribute it and/or modify it under
  11  * the terms of the GNU General Public License as published by the Free
  12  * Software Foundation; either version 2 of the License, or (at your option)
  13  * any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful, but WITHOUT
  16  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  17  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  18  * more details.
  19  *
  20  * You should have received a copy of the GNU General Public License along with
  21  * this program; if not, write to the Free Software Foundation, Inc., 675
  22  * Mass Ave, Cambridge, MA 02139, USA.
  23  */
  24
  25 #include "includes.h"
  26
  27 #if 0 /* when made a module use this */
  28
  29 static int tdbsam_debug_level = DBGC_ALL;
  30 #undef DBGC_CLASS
  31 #define DBGC_CLASS tdbsam_debug_level
  32
  33 #else
  34
  35 #undef DBGC_CLASS
  36 #define DBGC_CLASS DBGC_PASSDB
  37
  38 #endif
  39
  40 #ifdef WITH_TDB_SAM
  41
  42 #define PDB_VERSION             "20010830"
  43 #define PASSDB_FILE_NAME        "passdb.tdb"
  44 #define TDB_FORMAT_STRING       "ddddddBBBBBBBBBBBBddBBwdwdBdd"
  45 #define USERPREFIX              "USER_"
  46 #define RIDPREFIX               "RID_"
  47
  48 struct tdbsam_privates {
  49         TDB_CONTEXT     *passwd_tdb;
  50         TDB_DATA        key;
  51
  52         /* retrive-once info */
  53         const char *tdbsam_location;
  54
  55         BOOL permit_non_unix_accounts;
  56
  57         BOOL algorithmic_rids;
  58
  59         uint32 low_nua_rid;
  60         uint32 high_nua_rid;
  61 };
  62
  63 /**********************************************************************
  64  Intialize a SAM_ACCOUNT struct from a BYTE buffer of size len
  65  *********************************************************************/
  66
  67 static BOOL init_sam_from_buffer (struct tdbsam_privates *tdb_state,
  68                                   SAM_ACCOUNT *sampass, uint8 *buf, uint32 buflen)
  69 {
  70
  71         /* times are stored as 32bit integer
  72            take care on system with 64bit wide time_t
  73            --SSS */
  74         uint32  logon_time,
  75                 logoff_time,
  76                 kickoff_time,
  77                 pass_last_set_time,
  78                 pass_can_change_time,
  79                 pass_must_change_time;
  80         char *username;
  81         char *domain;
  82         char *nt_username;
  83         char *dir_drive;
  84         char *unknown_str;
  85         char *munged_dial;
  86         char *fullname;
  87         char *homedir;
  88         char *logon_script;
  89         char *profile_path;
  90         char *acct_desc;
  91         char *workstations;
  92         uint32  username_len, domain_len, nt_username_len,
  93                 dir_drive_len, unknown_str_len, munged_dial_len,
  94                 fullname_len, homedir_len, logon_script_len,
  95                 profile_path_len, acct_desc_len, workstations_len;
  96
  97         uint32  user_rid, group_rid, unknown_3, hours_len, unknown_5, unknown_6;
  98         uint16  acct_ctrl, logon_divs;
  99         uint8   *hours;
 100         static uint8    *lm_pw_ptr, *nt_pw_ptr;
 101         uint32          len = 0;
 102         uint32          lm_pw_len, nt_pw_len, hourslen;
 103         BOOL ret = True;
 104         struct passwd *pw;
 105         uid_t uid = -1;
 106         gid_t gid = -1; /* This is what standard sub advanced expects if no gid is known */
 107
 108         if(sampass == NULL || buf == NULL) {
 109                 DEBUG(0, ("init_sam_from_buffer: NULL parameters found!\n"));
 110                 return False;
 111         }
 112
 113         /* unpack the buffer into variables */
 114         len = tdb_unpack (buf, buflen, TDB_FORMAT_STRING,
 115                 &logon_time,
 116                 &logoff_time,
 117                 &kickoff_time,
 118                 &pass_last_set_time,
 119                 &pass_can_change_time,
 120                 &pass_must_change_time,
 121                 &username_len, &username,
 122                 &domain_len, &domain,
 123                 &nt_username_len, &nt_username,
 124                 &fullname_len, &fullname,
 125                 &homedir_len, &homedir,
 126                 &dir_drive_len, &dir_drive,
 127                 &logon_script_len, &logon_script,
 128                 &profile_path_len, &profile_path,
 129                 &acct_desc_len, &acct_desc,
 130                 &workstations_len, &workstations,
 131                 &unknown_str_len, &unknown_str,
 132                 &munged_dial_len, &munged_dial,
 133                 &user_rid,
 134                 &group_rid,
 135                 &lm_pw_len, &lm_pw_ptr,
 136                 &nt_pw_len, &nt_pw_ptr,
 137                 &acct_ctrl,
 138                 &unknown_3,
 139                 &logon_divs,
 140                 &hours_len,
 141                 &hourslen, &hours,
 142                 &unknown_5,
 143                 &unknown_6);
 144
 145         if (len == -1)  {
 146                 ret = False;
 147                 goto done;
 148         }
 149
 150         /* validate the account and fill in UNIX uid and gid. Standard
 151          * getpwnam() is used instead of Get_Pwnam() as we do not need
 152          * to try case permutations
 153          */
 154         if (!username || !(pw = getpwnam_alloc(username))) {
 155                 if (!(tdb_state->permit_non_unix_accounts)) {
 156                         DEBUG(0,("tdbsam: getpwnam_alloc(%s) return NULL.  User does not exist!\n", username));
 157                         ret = False;
 158                         goto done;
 159                 }
 160         }
 161
 162         if (pw) {
 163                 uid = pw->pw_uid;
 164                 gid = pw->pw_gid;
 165
 166                 pdb_set_unix_homedir(sampass, pw->pw_dir);
 167
 168                 passwd_free(&pw);
 169
 170                 pdb_set_uid(sampass, uid);
 171                 pdb_set_gid(sampass, gid);
 172         }
 173
 174         pdb_set_logon_time(sampass, logon_time, True);
 175         pdb_set_logoff_time(sampass, logoff_time, True);
 176         pdb_set_kickoff_time(sampass, kickoff_time, True);
 177         pdb_set_pass_can_change_time(sampass, pass_can_change_time, True);
 178         pdb_set_pass_must_change_time(sampass, pass_must_change_time, True);
 179         pdb_set_pass_last_set_time(sampass, pass_last_set_time);
 180
 181         pdb_set_username     (sampass, username);
 182         pdb_set_domain       (sampass, domain);
 183         pdb_set_nt_username  (sampass, nt_username);
 184         pdb_set_fullname     (sampass, fullname);
 185
 186         if (homedir) {
 187                 pdb_set_homedir(sampass, homedir, True);
 188         }
 189         else {
 190                 pdb_set_homedir(sampass,
 191                                 talloc_sub_specified(sampass->mem_ctx,
 192                                                        lp_logon_home(),
 193                                                        username, domain,
 194                                                        uid, gid),
 195                                 False);
 196         }
 197
 198         if (dir_drive)
 199                 pdb_set_dir_drive(sampass, dir_drive, True);
 200         else {
 201                 pdb_set_dir_drive(sampass,
 202                                   talloc_sub_specified(sampass->mem_ctx,
 203                                                          lp_logon_drive(),
 204                                                          username, domain,
 205                                                          uid, gid),
 206                                   False);
 207         }
 208
 209         if (logon_script)
 210                 pdb_set_logon_script(sampass, logon_script, True);
 211         else {
 212                 pdb_set_logon_script(sampass,
 213                                      talloc_sub_specified(sampass->mem_ctx,
 214                                                             lp_logon_script(),
 215                                                             username, domain,
 216                                                             uid, gid),
 217                                   False);
 218         }
 219
 220         if (profile_path) {
 221                 pdb_set_profile_path(sampass, profile_path, True);
 222         } else {
 223                 pdb_set_profile_path(sampass,
 224                                      talloc_sub_specified(sampass->mem_ctx,
 225                                                             lp_logon_path(),
 226                                                             username, domain,
 227                                                             uid, gid),
 228                                      False);
 229         }
 230
 231         pdb_set_acct_desc    (sampass, acct_desc);
 232         pdb_set_workstations (sampass, workstations);
 233         pdb_set_munged_dial  (sampass, munged_dial);
 234
 235         if (lm_pw_ptr && lm_pw_len == LM_HASH_LEN) {
 236                 if (!pdb_set_lanman_passwd(sampass, lm_pw_ptr)) {
 237                         ret = False;
 238                         goto done;
 239                 }
 240         }
 241
 242         if (nt_pw_ptr && nt_pw_len == NT_HASH_LEN) {
 243                 if (!pdb_set_nt_passwd(sampass, nt_pw_ptr)) {
 244                         ret = False;
 245                         goto done;
 246                 }
 247         }
 248
 249         pdb_set_user_sid_from_rid(sampass, user_rid);
 250         pdb_set_group_sid_from_rid(sampass, group_rid);
 251         pdb_set_unknown_3(sampass, unknown_3);
 252         pdb_set_hours_len(sampass, hours_len);
 253         pdb_set_unknown_5(sampass, unknown_5);
 254         pdb_set_unknown_6(sampass, unknown_6);
 255         pdb_set_acct_ctrl(sampass, acct_ctrl);
 256         pdb_set_logon_divs(sampass, logon_divs);
 257         pdb_set_hours(sampass, hours);
 258
 259 done:
 260
 261         SAFE_FREE(username);
 262         SAFE_FREE(domain);
 263         SAFE_FREE(nt_username);
 264         SAFE_FREE(fullname);
 265         SAFE_FREE(homedir);
 266         SAFE_FREE(dir_drive);
 267         SAFE_FREE(logon_script);
 268         SAFE_FREE(profile_path);
 269         SAFE_FREE(acct_desc);
 270         SAFE_FREE(workstations);
 271         SAFE_FREE(munged_dial);
 272
 273         return ret;
 274 }
 275
 276 /**********************************************************************
 277  Intialize a BYTE buffer from a SAM_ACCOUNT struct
 278  *********************************************************************/
 279 static uint32 init_buffer_from_sam (struct tdbsam_privates *tdb_state,
 280                                     uint8 **buf, const SAM_ACCOUNT *sampass)
 281 {
 282         size_t          len, buflen;
 283
 284         /* times are stored as 32bit integer
 285            take care on system with 64bit wide time_t
 286            --SSS */
 287         uint32  logon_time,
 288                 logoff_time,
 289                 kickoff_time,
 290                 pass_last_set_time,
 291                 pass_can_change_time,
 292                 pass_must_change_time;
 293
 294         uint32  user_rid, group_rid;
 295
 296         const char *username;
 297         const char *domain;
 298         const char *nt_username;
 299         const char *dir_drive;
 300         const char *unknown_str;
 301         const char *munged_dial;
 302         const char *fullname;
 303         const char *homedir;
 304         const char *logon_script;
 305         const char *profile_path;
 306         const char *acct_desc;
 307         const char *workstations;
 308         uint32  username_len, domain_len, nt_username_len,
 309                 dir_drive_len, unknown_str_len, munged_dial_len,
 310                 fullname_len, homedir_len, logon_script_len,
 311                 profile_path_len, acct_desc_len, workstations_len;
 312
 313         const uint8 *lm_pw;
 314         const uint8 *nt_pw;
 315         uint32  lm_pw_len = 16;
 316         uint32  nt_pw_len = 16;
 317
 318         /* do we have a valid SAM_ACCOUNT pointer? */
 319         if (sampass == NULL) {
 320                 DEBUG(0, ("init_buffer_from_sam: SAM_ACCOUNT is NULL!\n"));
 321                 return -1;
 322         }
 323
 324         *buf = NULL;
 325         buflen = 0;
 326
 327         logon_time = (uint32)pdb_get_logon_time(sampass);
 328         logoff_time = (uint32)pdb_get_logoff_time(sampass);
 329         kickoff_time = (uint32)pdb_get_kickoff_time(sampass);
 330         pass_can_change_time = (uint32)pdb_get_pass_can_change_time(sampass);
 331         pass_must_change_time = (uint32)pdb_get_pass_must_change_time(sampass);
 332         pass_last_set_time = (uint32)pdb_get_pass_last_set_time(sampass);
 333
 334         user_rid = pdb_get_user_rid(sampass);
 335         group_rid = pdb_get_group_rid(sampass);
 336
 337         username = pdb_get_username(sampass);
 338         if (username) username_len = strlen(username) +1;
 339         else username_len = 0;
 340
 341         domain = pdb_get_domain(sampass);
 342         if (domain) domain_len = strlen(domain) +1;
 343         else domain_len = 0;
 344
 345         nt_username = pdb_get_nt_username(sampass);
 346         if (nt_username) nt_username_len = strlen(nt_username) +1;
 347         else nt_username_len = 0;
 348
 349         fullname = pdb_get_fullname(sampass);
 350         if (fullname) fullname_len = strlen(fullname) +1;
 351         else fullname_len = 0;
 352
 353         /*
 354          * Only updates fields which have been set (not defaults from smb.conf)
 355          */
 356
 357         if (IS_SAM_SET(sampass, FLAG_SAM_DRIVE))
 358           dir_drive = pdb_get_dir_drive(sampass);
 359         else dir_drive = NULL;
 360         if (dir_drive) dir_drive_len = strlen(dir_drive) +1;
 361         else dir_drive_len = 0;
 362
 363         if (IS_SAM_SET(sampass, FLAG_SAM_SMBHOME)) homedir = pdb_get_homedir(sampass);
 364         else homedir = NULL;
 365         if (homedir) homedir_len = strlen(homedir) +1;
 366         else homedir_len = 0;
 367
 368         if (IS_SAM_SET(sampass, FLAG_SAM_LOGONSCRIPT)) logon_script = pdb_get_logon_script(sampass);
 369         else logon_script = NULL;
 370         if (logon_script) logon_script_len = strlen(logon_script) +1;
 371         else logon_script_len = 0;
 372
 373         if (IS_SAM_SET(sampass, FLAG_SAM_PROFILE)) profile_path = pdb_get_profile_path(sampass);
 374         else profile_path = NULL;
 375         if (profile_path) profile_path_len = strlen(profile_path) +1;
 376         else profile_path_len = 0;
 377
 378         lm_pw = pdb_get_lanman_passwd(sampass);
 379         if (!lm_pw) lm_pw_len = 0;
 380
 381         nt_pw = pdb_get_nt_passwd(sampass);
 382         if (!nt_pw) nt_pw_len = 0;
 383
 384         acct_desc = pdb_get_acct_desc(sampass);
 385         if (acct_desc) acct_desc_len = strlen(acct_desc) +1;
 386         else acct_desc_len = 0;
 387
 388         workstations = pdb_get_workstations(sampass);
 389         if (workstations) workstations_len = strlen(workstations) +1;
 390         else workstations_len = 0;
 391
 392         unknown_str = NULL;
 393         unknown_str_len = 0;
 394
 395         munged_dial = pdb_get_munged_dial(sampass);
 396         if (munged_dial) munged_dial_len = strlen(munged_dial) +1;
 397         else munged_dial_len = 0;
 398
 399         /* one time to get the size needed */
 400         len = tdb_pack(NULL, 0,  TDB_FORMAT_STRING,
 401                 logon_time,
 402                 logoff_time,
 403                 kickoff_time,
 404                 pass_last_set_time,
 405                 pass_can_change_time,
 406                 pass_must_change_time,
 407                 username_len, username,
 408                 domain_len, domain,
 409                 nt_username_len, nt_username,
 410                 fullname_len, fullname,
 411                 homedir_len, homedir,
 412                 dir_drive_len, dir_drive,
 413                 logon_script_len, logon_script,
 414                 profile_path_len, profile_path,
 415                 acct_desc_len, acct_desc,
 416                 workstations_len, workstations,
 417                 unknown_str_len, unknown_str,
 418                 munged_dial_len, munged_dial,
 419                 user_rid,
 420                 group_rid,
 421                 lm_pw_len, lm_pw,
 422                 nt_pw_len, nt_pw,
 423                 pdb_get_acct_ctrl(sampass),
 424                 pdb_get_unknown3(sampass),
 425                 pdb_get_logon_divs(sampass),
 426                 pdb_get_hours_len(sampass),
 427                 MAX_HOURS_LEN, pdb_get_hours(sampass),
 428                 pdb_get_unknown5(sampass),
 429                 pdb_get_unknown6(sampass));
 430
 431
 432         /* malloc the space needed */
 433         if ( (*buf=(uint8*)malloc(len)) == NULL) {
 434                 DEBUG(0,("init_buffer_from_sam: Unable to malloc() memory for buffer!\n"));
 435                 return (-1);
 436         }
 437
 438         /* now for the real call to tdb_pack() */
 439         buflen = tdb_pack(*buf, len,  TDB_FORMAT_STRING,
 440                 logon_time,
 441                 logoff_time,
 442                 kickoff_time,
 443                 pass_last_set_time,
 444                 pass_can_change_time,
 445                 pass_must_change_time,
 446                 username_len, username,
 447                 domain_len, domain,
 448                 nt_username_len, nt_username,
 449                 fullname_len, fullname,
 450                 homedir_len, homedir,
 451                 dir_drive_len, dir_drive,
 452                 logon_script_len, logon_script,
 453                 profile_path_len, profile_path,
 454                 acct_desc_len, acct_desc,
 455                 workstations_len, workstations,
 456                 unknown_str_len, unknown_str,
 457                 munged_dial_len, munged_dial,
 458                 user_rid,
 459                 group_rid,
 460                 lm_pw_len, lm_pw,
 461                 nt_pw_len, nt_pw,
 462                 pdb_get_acct_ctrl(sampass),
 463                 pdb_get_unknown3(sampass),
 464                 pdb_get_logon_divs(sampass),
 465                 pdb_get_hours_len(sampass),
 466                 MAX_HOURS_LEN, pdb_get_hours(sampass),
 467                 pdb_get_unknown5(sampass),
 468                 pdb_get_unknown6(sampass));
 469
 470
 471         /* check to make sure we got it correct */
 472         if (buflen != len) {
 473                 DEBUG(0, ("init_buffer_from_sam: somthing odd is going on here: bufflen (%d) != len (%d) in tdb_pack operations!\n",
 474                           buflen, len));
 475                 /* error */
 476                 SAFE_FREE (*buf);
 477                 return (-1);
 478         }
 479
 480         return (buflen);
 481 }
 482
 483 /***************************************************************
 484  Open the TDB passwd database for SAM account enumeration.
 485 ****************************************************************/
 486
 487 static BOOL tdbsam_setsampwent(struct pdb_methods *my_methods, BOOL update)
 488 {
 489         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 490
 491         /* Open tdb passwd */
 492         if (!(tdb_state->passwd_tdb = tdb_open_log(tdb_state->tdbsam_location, 0, TDB_DEFAULT, update?(O_RDWR|O_CREAT):O_RDONLY, 0600)))
 493         {
 494                 DEBUG(0, ("Unable to open/create TDB passwd\n"));
 495                 return False;
 496         }
 497
 498         tdb_state->key = tdb_firstkey(tdb_state->passwd_tdb);
 499
 500         return True;
 501 }
 502
 503 static void close_tdb(struct tdbsam_privates *tdb_state)
 504 {
 505         if (tdb_state->passwd_tdb) {
 506                 tdb_close(tdb_state->passwd_tdb);
 507                 tdb_state->passwd_tdb = NULL;
 508         }
 509 }
 510
 511 /***************************************************************
 512  End enumeration of the TDB passwd list.
 513 ****************************************************************/
 514
 515 static void tdbsam_endsampwent(struct pdb_methods *my_methods)
 516 {
 517         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 518         close_tdb(tdb_state);
 519
 520         DEBUG(7, ("endtdbpwent: closed sam database.\n"));
 521 }
 522
 523 /*****************************************************************
 524  Get one SAM_ACCOUNT from the TDB (next in line)
 525 *****************************************************************/
 526
 527 static BOOL tdbsam_getsampwent(struct pdb_methods *my_methods, SAM_ACCOUNT *user)
 528 {
 529         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 530         TDB_DATA        data;
 531         char *prefix = USERPREFIX;
 532         int  prefixlen = strlen (prefix);
 533
 534
 535         if (user==NULL) {
 536                 DEBUG(0,("pdb_get_sampwent: SAM_ACCOUNT is NULL.\n"));
 537                 return False;
 538         }
 539
 540         /* skip all non-USER entries (eg. RIDs) */
 541         while ((tdb_state->key.dsize != 0) && (strncmp(tdb_state->key.dptr, prefix, prefixlen)))
 542                 /* increment to next in line */
 543                 tdb_state->key = tdb_nextkey(tdb_state->passwd_tdb, tdb_state->key);
 544
 545         /* do we have an valid interation pointer? */
 546         if(tdb_state->passwd_tdb == NULL) {
 547                 DEBUG(0,("pdb_get_sampwent: Bad TDB Context pointer.\n"));
 548                 return False;
 549         }
 550
 551         data = tdb_fetch(tdb_state->passwd_tdb, tdb_state->key);
 552         if (!data.dptr) {
 553                 DEBUG(5,("pdb_getsampwent: database entry not found.\n"));
 554                 return False;
 555         }
 556
 557         /* unpack the buffer */
 558         if (!init_sam_from_buffer(tdb_state, user, data.dptr, data.dsize)) {
 559                 DEBUG(0,("pdb_getsampwent: Bad SAM_ACCOUNT entry returned from TDB!\n"));
 560                 SAFE_FREE(data.dptr);
 561                 return False;
 562         }
 563         SAFE_FREE(data.dptr);
 564
 565         /* increment to next in line */
 566         tdb_state->key = tdb_nextkey(tdb_state->passwd_tdb, tdb_state->key);
 567
 568         return True;
 569 }
 570
 571 /******************************************************************
 572  Lookup a name in the SAM TDB
 573 ******************************************************************/
 574
 575 static BOOL tdbsam_getsampwnam (struct pdb_methods *my_methods, SAM_ACCOUNT *user, const char *sname)
 576 {
 577         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 578         TDB_CONTEXT     *pwd_tdb;
 579         TDB_DATA        data, key;
 580         fstring         keystr;
 581         fstring         name;
 582
 583         if (user==NULL) {
 584                 DEBUG(0,("pdb_getsampwnam: SAM_ACCOUNT is NULL.\n"));
 585                 return False;
 586         }
 587
 588         /* Data is stored in all lower-case */
 589         unix_strlower(sname, -1, name, sizeof(name));
 590
 591         /* set search key */
 592         slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
 593         key.dptr = keystr;
 594         key.dsize = strlen(keystr) + 1;
 595
 596         /* open the accounts TDB */
 597         if (!(pwd_tdb = tdb_open_log(tdb_state->tdbsam_location, 0, TDB_DEFAULT, O_RDONLY, 0600))) {
 598                 DEBUG(0, ("pdb_getsampwnam: Unable to open TDB passwd (%s)!\n", tdb_state->tdbsam_location));
 599                 return False;
 600         }
 601
 602         /* get the record */
 603         data = tdb_fetch(pwd_tdb, key);
 604         if (!data.dptr) {
 605                 DEBUG(5,("pdb_getsampwnam (TDB): error fetching database.\n"));
 606                 DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
 607                 DEBUGADD(5, (" Key: %s\n", keystr));
 608                 tdb_close(pwd_tdb);
 609                 return False;
 610         }
 611
 612         /* unpack the buffer */
 613         if (!init_sam_from_buffer(tdb_state, user, data.dptr, data.dsize)) {
 614                 DEBUG(0,("pdb_getsampwent: Bad SAM_ACCOUNT entry returned from TDB!\n"));
 615                 SAFE_FREE(data.dptr);
 616                 tdb_close(pwd_tdb);
 617                 return False;
 618         }
 619         SAFE_FREE(data.dptr);
 620
 621         /* no further use for database, close it now */
 622         tdb_close(pwd_tdb);
 623
 624         return True;
 625 }
 626
 627 /***************************************************************************
 628  Search by rid
 629  **************************************************************************/
 630
 631 static BOOL tdbsam_getsampwrid (struct pdb_methods *my_methods, SAM_ACCOUNT *user, uint32 rid)
 632 {
 633         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 634         TDB_CONTEXT             *pwd_tdb;
 635         TDB_DATA                data, key;
 636         fstring                 keystr;
 637         fstring                 name;
 638
 639         if (user==NULL) {
 640                 DEBUG(0,("pdb_getsampwrid: SAM_ACCOUNT is NULL.\n"));
 641                 return False;
 642         }
 643
 644         /* set search key */
 645         slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, rid);
 646         key.dptr = keystr;
 647         key.dsize = strlen (keystr) + 1;
 648
 649         /* open the accounts TDB */
 650         if (!(pwd_tdb = tdb_open_log(tdb_state->tdbsam_location, 0, TDB_DEFAULT, O_RDONLY, 0600))) {
 651                 DEBUG(0, ("pdb_getsampwrid: Unable to open TDB rid database!\n"));
 652                 return False;
 653         }
 654
 655         /* get the record */
 656         data = tdb_fetch (pwd_tdb, key);
 657         if (!data.dptr) {
 658                 DEBUG(5,("pdb_getsampwrid (TDB): error looking up RID %d by key %s.\n", rid, keystr));
 659                 DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
 660                 tdb_close (pwd_tdb);
 661                 return False;
 662         }
 663
 664         fstrcpy (name, data.dptr);
 665         SAFE_FREE(data.dptr);
 666
 667         tdb_close (pwd_tdb);
 668
 669         return tdbsam_getsampwnam (my_methods, user, name);
 670 }
 671
 672 static BOOL tdbsam_getsampwsid(struct pdb_methods *my_methods, SAM_ACCOUNT * user, DOM_SID *sid)
 673 {
 674         uint32 rid;
 675         if (!sid_peek_check_rid(get_global_sam_sid(), sid, &rid))
 676                 return False;
 677         return tdbsam_getsampwrid(my_methods, user, rid);
 678 }
 679
 680 /***************************************************************************
 681  Delete a SAM_ACCOUNT
 682 ****************************************************************************/
 683
 684 static BOOL tdbsam_delete_sam_account(struct pdb_methods *my_methods, SAM_ACCOUNT *sam_pass)
 685 {
 686         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 687         TDB_CONTEXT     *pwd_tdb;
 688         TDB_DATA        key;
 689         fstring         keystr;
 690         uint32          rid;
 691         fstring         name;
 692
 693         unix_strlower(pdb_get_username(sam_pass), -1, name, sizeof(name));
 694
 695         /* open the TDB */
 696         if (!(pwd_tdb = tdb_open_log(tdb_state->tdbsam_location, 0, TDB_DEFAULT, O_RDWR, 0600))) {
 697                 DEBUG(0, ("Unable to open TDB passwd!"));
 698                 return False;
 699         }
 700
 701         /* set the search key */
 702         slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
 703         key.dptr = keystr;
 704         key.dsize = strlen (keystr) + 1;
 705
 706         rid = pdb_get_user_rid(sam_pass);
 707
 708         /* it's outaa here!  8^) */
 709         if (tdb_delete(pwd_tdb, key) != TDB_SUCCESS) {
 710                 DEBUG(5, ("Error deleting entry from tdb passwd database!\n"));
 711                 DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
 712                 tdb_close(pwd_tdb);
 713                 return False;
 714         }
 715
 716         /* delete also the RID key */
 717
 718         /* set the search key */
 719         slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, rid);
 720         key.dptr = keystr;
 721         key.dsize = strlen (keystr) + 1;
 722
 723         /* it's outaa here!  8^) */
 724         if (tdb_delete(pwd_tdb, key) != TDB_SUCCESS) {
 725                 DEBUG(5, ("Error deleting entry from tdb rid database!\n"));
 726                 DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
 727                 tdb_close(pwd_tdb);
 728                 return False;
 729         }
 730
 731         tdb_close(pwd_tdb);
 732
 733         return True;
 734 }
 735
 736 /***************************************************************************
 737  Update the TDB SAM
 738 ****************************************************************************/
 739
 740 static BOOL tdb_update_sam(struct pdb_methods *my_methods, SAM_ACCOUNT* newpwd, int flag)
 741 {
 742         struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
 743         TDB_CONTEXT     *pwd_tdb = NULL;
 744         TDB_DATA        key, data;
 745         uint8           *buf = NULL;
 746         fstring         keystr;
 747         fstring         name;
 748         BOOL            ret = True;
 749         uint32          user_rid;
 750         BOOL            tdb_ret;
 751
 752         /* invalidate the existing TDB iterator if it is open */
 753         if (tdb_state->passwd_tdb) {
 754                 tdb_close(tdb_state->passwd_tdb);
 755                 tdb_state->passwd_tdb = NULL;
 756         }
 757
 758         /* open the account TDB passwd*/
 759         pwd_tdb = tdb_open_log(tdb_state->tdbsam_location, 0, TDB_DEFAULT, O_RDWR | O_CREAT, 0600);
 760         if (!pwd_tdb)
 761         {
 762                 DEBUG(0, ("tdb_update_sam: Unable to open TDB passwd (%s)!\n", tdb_state->tdbsam_location));
 763                 return False;
 764         }
 765
 766         /* if flag == TDB_INSERT then make up a new RID else throw an error. */
 767         if (!(user_rid = pdb_get_user_rid(newpwd))) {
 768                 if (flag & TDB_INSERT) {
 769                         if (IS_SAM_UNIX_USER(newpwd)) {
 770                                 if (tdb_state->algorithmic_rids) {
 771                                         user_rid = fallback_pdb_uid_to_user_rid(pdb_get_uid(newpwd));
 772                                 } else {
 773                                         user_rid = BASE_RID;
 774                                         tdb_ret = tdb_change_uint32_atomic(pwd_tdb, "RID_COUNTER", &user_rid, RID_MULTIPLIER);
 775                                         if (!tdb_ret) {
 776                                                 ret = False;
 777                                                 goto done;
 778                                         }
 779                                 }
 780                                 pdb_set_user_sid_from_rid(newpwd, user_rid);
 781                         } else {
 782                                 user_rid = tdb_state->low_nua_rid;
 783                                 tdb_ret = tdb_change_uint32_atomic(pwd_tdb, "NUA_RID_COUNTER", &user_rid, RID_MULTIPLIER);
 784                                 if (!tdb_ret) {
 785                                         ret = False;
 786                                         goto done;
 787                                 }
 788                                 if (user_rid > tdb_state->high_nua_rid) {
 789                                         DEBUG(0, ("tdbsam: no NUA rids available, cannot add user %s!\n", pdb_get_username(newpwd)));
 790                                         ret = False;
 791                                         goto done;
 792                                 }
 793                                 pdb_set_user_sid_from_rid(newpwd, user_rid);
 794                         }
 795                 } else {
 796                         DEBUG (0,("tdb_update_sam: Failing to store a SAM_ACCOUNT for [%s] without a RID\n",pdb_get_username(newpwd)));
 797                         ret = False;
 798                         goto done;
 799                 }
 800         }
 801
 802         if (!pdb_get_group_rid(newpwd)) {
 803                 if (flag & TDB_INSERT) {
 804                         if (!tdb_state->permit_non_unix_accounts) {
 805                                 DEBUG (0,("tdb_update_sam: Failing to store a SAM_ACCOUNT for [%s] without a primary group RID\n",pdb_get_username(newpwd)));
 806                                 ret = False;
 807                                 goto done;
 808                         } else {
 809                                 /* This seems like a good default choice for non-unix users */
 810                                 pdb_set_group_sid_from_rid(newpwd, DOMAIN_GROUP_RID_USERS);
 811                         }
 812                 } else {
 813                         DEBUG (0,("tdb_update_sam: Failing to store a SAM_ACCOUNT for [%s] without a primary group RID\n",pdb_get_username(newpwd)));
 814                         ret = False;
 815                         goto done;
 816                 }
 817         }
 818
 819         /* copy the SAM_ACCOUNT struct into a BYTE buffer for storage */
 820         if ((data.dsize=init_buffer_from_sam (tdb_state, &buf, newpwd)) == -1) {
 821                 DEBUG(0,("tdb_update_sam: ERROR - Unable to copy SAM_ACCOUNT info BYTE buffer!\n"));
 822                 ret = False;
 823                 goto done;
 824         }
 825         data.dptr = buf;
 826
 827         unix_strlower(pdb_get_username(newpwd), -1, name, sizeof(name));
 828
 829         DEBUG(5, ("Storing %saccount %s with RID %d\n", flag == TDB_INSERT ? "(new) " : "", name, user_rid));
 830
 831         /* setup the USER index key */
 832         slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
 833         key.dptr = keystr;
 834         key.dsize = strlen (keystr) + 1;
 835
 836         /* add the account */
 837         if (tdb_store(pwd_tdb, key, data, flag) != TDB_SUCCESS) {
 838                 DEBUG(0, ("Unable to modify passwd TDB!"));
 839                 DEBUGADD(0, (" Error: %s", tdb_errorstr(pwd_tdb)));
 840                 DEBUGADD(0, (" occured while storing the main record (%s)\n", keystr));
 841                 ret = False;
 842                 goto done;
 843         }
 844
 845         /* setup RID data */
 846         data.dsize = sizeof(fstring);
 847         data.dptr = name;
 848
 849         /* setup the RID index key */
 850         slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, user_rid);
 851         key.dptr = keystr;
 852         key.dsize = strlen (keystr) + 1;
 853
 854         /* add the reference */
 855         if (tdb_store(pwd_tdb, key, data, flag) != TDB_SUCCESS) {
 856                 DEBUG(0, ("Unable to modify TDB passwd !"));
 857                 DEBUGADD(0, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
 858                 DEBUGADD(0, (" occured while storing the RID index (%s)\n", keystr));
 859                 ret = False;
 860                 goto done;
 861         }
 862
 863 done:
 864         /* cleanup */
 865         tdb_close (pwd_tdb);
 866         SAFE_FREE(buf);
 867
 868         return (ret);
 869 }
 870
 871 /***************************************************************************
 872  Modifies an existing SAM_ACCOUNT
 873 ****************************************************************************/
 874
 875 static BOOL tdbsam_update_sam_account (struct pdb_methods *my_methods, SAM_ACCOUNT *newpwd)
 876 {
 877         return (tdb_update_sam(my_methods, newpwd, TDB_MODIFY));
 878 }
 879
 880 /***************************************************************************
 881  Adds an existing SAM_ACCOUNT
 882 ****************************************************************************/
 883
 884 static BOOL tdbsam_add_sam_account (struct pdb_methods *my_methods, SAM_ACCOUNT *newpwd)
 885 {
 886         return (tdb_update_sam(my_methods, newpwd, TDB_INSERT));
 887 }
 888
 889 static void free_private_data(void **vp)
 890 {
 891         struct tdbsam_privates **tdb_state = (struct tdbsam_privates **)vp;
 892         close_tdb(*tdb_state);
 893         *tdb_state = NULL;
 894
 895         /* No need to free any further, as it is talloc()ed */
 896 }
 897
 898
 899 NTSTATUS pdb_init_tdbsam(PDB_CONTEXT *pdb_context, PDB_METHODS **pdb_method, const char *location)
 900 {
 901         NTSTATUS nt_status;
 902         struct tdbsam_privates *tdb_state;
 903
 904 #if 0 /* when made a module use this */
 905         tdbsam_debug_level = debug_add_class("tdbsam");
 906         if(tdbsam_debug_level == -1) {
 907                 tdbsam_debug_level = DBGC_ALL;
 908                 DEBUG(0, ("tdbsam: Couldn't register custom debugging class!\n"));
 909         }
 910 #endif
 911
 912         if (!NT_STATUS_IS_OK(nt_status = make_pdb_methods(pdb_context->mem_ctx, pdb_method))) {
 913                 return nt_status;
 914         }
 915
 916         (*pdb_method)->name = "tdbsam";
 917
 918         (*pdb_method)->setsampwent = tdbsam_setsampwent;
 919         (*pdb_method)->endsampwent = tdbsam_endsampwent;
 920         (*pdb_method)->getsampwent = tdbsam_getsampwent;
 921         (*pdb_method)->getsampwnam = tdbsam_getsampwnam;
 922         (*pdb_method)->getsampwsid = tdbsam_getsampwsid;
 923         (*pdb_method)->add_sam_account = tdbsam_add_sam_account;
 924         (*pdb_method)->update_sam_account = tdbsam_update_sam_account;
 925         (*pdb_method)->delete_sam_account = tdbsam_delete_sam_account;
 926
 927         tdb_state = talloc_zero(pdb_context->mem_ctx, sizeof(struct tdbsam_privates));
 928
 929         if (!tdb_state) {
 930                 DEBUG(0, ("talloc() failed for tdbsam private_data!\n"));
 931                 return NT_STATUS_NO_MEMORY;
 932         }
 933
 934         if (location) {
 935                 tdb_state->tdbsam_location = talloc_strdup(pdb_context->mem_ctx, location);
 936         } else {
 937                 pstring tdbfile;
 938                 get_private_directory(tdbfile);
 939                 pstrcat(tdbfile, "/");
 940                 pstrcat(tdbfile, PASSDB_FILE_NAME);
 941                 tdb_state->tdbsam_location = talloc_strdup(pdb_context->mem_ctx, tdbfile);
 942         }
 943
 944         tdb_state->algorithmic_rids = True;
 945
 946         (*pdb_method)->private_data = tdb_state;
 947
 948         (*pdb_method)->free_private_data = free_private_data;
 949
 950         return NT_STATUS_OK;
 951 }
 952
 953 NTSTATUS pdb_init_tdbsam_nua(PDB_CONTEXT *pdb_context, PDB_METHODS **pdb_method, const char *location)
 954 {
 955         NTSTATUS nt_status;
 956         struct tdbsam_privates *tdb_state;
 957         uint32 low_nua_uid, high_nua_uid;
 958
 959         if (!NT_STATUS_IS_OK(nt_status = pdb_init_tdbsam(pdb_context, pdb_method, location))) {
 960                 return nt_status;
 961         }
 962
 963         (*pdb_method)->name = "tdbsam_nua";
 964
 965         tdb_state = (*pdb_method)->private_data;
 966
 967         tdb_state->permit_non_unix_accounts = True;
 968
 969         if (!lp_non_unix_account_range(&low_nua_uid, &high_nua_uid)) {
 970                 DEBUG(0, ("cannot use tdbsam_nua without 'non unix account range' in smb.conf!\n"));
 971                 return NT_STATUS_UNSUCCESSFUL;
 972         }
 973
 974         tdb_state->low_nua_rid=fallback_pdb_uid_to_user_rid(low_nua_uid);
 975
 976         tdb_state->high_nua_rid=fallback_pdb_uid_to_user_rid(high_nua_uid);
 977
 978         return NT_STATUS_OK;
 979 }
 980
 981
 982 #else
 983
 984 NTSTATUS pdb_init_tdbsam(PDB_CONTEXT *pdb_context, PDB_METHODS **pdb_method, const char *location)
 985 {
 986         DEBUG(0, ("tdbsam not compiled in!\n"));
 987         return NT_STATUS_UNSUCCESSFUL;
 988 }
 989
 990 NTSTATUS pdb_init_tdbsam_nua(PDB_CONTEXT *pdb_context, PDB_METHODS **pdb_method, const char *location)
 991 {
 992         DEBUG(0, ("tdbsam_nua not compiled in!\n"));
 993         return NT_STATUS_UNSUCCESSFUL;
 994 }
 995
 996
 997 #endif