source3/winbindd/idmap_autorid.c

   1 /*
   2  *  idmap_autorid: static map between Active Directory/NT RIDs
   3  *  and RFC 2307 accounts
   4  *
   5  *  based on the idmap_rid module, but this module defines the ranges
   6  *  for the domains by automatically allocating a range for each domain
   7  *
   8  *  Copyright (C) Christian Ambach, 2010-2011
   9  *
  10  *  This program is free software; you can redistribute it and/or modify
  11  *  it under the terms of the GNU General Public License as published by
  12  *  the Free Software Foundation; either version 3 of the License, or
  13  *  (at your option) any later version.
  14  *
  15  *  This program is distributed in the hope that it will be useful,
  16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  *  GNU General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  22  *
  23  */
  24
  25 #include "includes.h"
  26 #include "system/filesys.h"
  27 #include "winbindd.h"
  28 #include "dbwrap/dbwrap.h"
  29 #include "dbwrap/dbwrap_open.h"
  30 #include "idmap.h"
  31 #include "../libcli/security/dom_sid.h"
  32 #include "util_tdb.h"
  33
  34 #undef DBGC_CLASS
  35 #define DBGC_CLASS DBGC_IDMAP
  36
  37 #define HWM "NEXT RANGE"
  38 #define ALLOC_HWM "NEXT ALLOC ID"
  39 #define ALLOC_POOL_SIZE 500
  40 #define CONFIGKEY "CONFIG"
  41
  42 struct autorid_global_config {
  43         uint32_t minvalue;
  44         uint32_t rangesize;
  45         uint32_t maxranges;
  46 };
  47
  48 struct autorid_domain_config {
  49         struct dom_sid sid;
  50         uint32_t domainnum;
  51         struct autorid_global_config *globalcfg;
  52 };
  53
  54 /* handle to the tdb storing domain <-> range assignments */
  55 static struct db_context *autorid_db;
  56
  57 static NTSTATUS idmap_autorid_get_domainrange(struct db_context *db,
  58                                               void *private_data)
  59 {
  60         NTSTATUS ret;
  61         uint32_t domainnum, hwm;
  62         fstring sidstr;
  63         char *numstr;
  64         struct autorid_domain_config *cfg;
  65
  66         cfg = (struct autorid_domain_config *)private_data;
  67         dom_sid_string_buf(&(cfg->sid), sidstr, sizeof(sidstr));
  68
  69         if (!dbwrap_fetch_uint32(db, sidstr, &domainnum)) {
  70                 DEBUG(10, ("Acquiring new range for domain %s\n", sidstr));
  71
  72                 /* fetch the current HWM */
  73                 if (!dbwrap_fetch_uint32(db, HWM, &hwm)) {
  74                         DEBUG(1, ("Fatal error while fetching current "
  75                                   "HWM value!\n"));
  76                         ret = NT_STATUS_INTERNAL_ERROR;
  77                         goto error;
  78                 }
  79
  80                 /* do we have a range left? */
  81                 if (hwm >= cfg->globalcfg->maxranges) {
  82                         DEBUG(1, ("No more domain ranges available!\n"));
  83                         ret = NT_STATUS_NO_MEMORY;
  84                         goto error;
  85                 }
  86
  87                 /* increase the HWM */
  88                 ret = dbwrap_change_uint32_atomic(db, HWM, &domainnum, 1);
  89                 if (!NT_STATUS_IS_OK(ret)) {
  90                         DEBUG(1, ("Fatal error while fetching a new "
  91                                   "domain range value!\n"));
  92                         goto error;
  93                 }
  94
  95                 /* store away the new mapping in both directions */
  96                 ret = dbwrap_trans_store_uint32(db, sidstr, domainnum);
  97                 if (!NT_STATUS_IS_OK(ret)) {
  98                         DEBUG(1, ("Fatal error while storing new "
  99                                   "domain->range assignment!\n"));
 100                         goto error;
 101                 }
 102
 103                 numstr = talloc_asprintf(db, "%u", domainnum);
 104                 if (!numstr) {
 105                         ret = NT_STATUS_NO_MEMORY;
 106                         goto error;
 107                 }
 108
 109                 ret = dbwrap_trans_store_bystring(db, numstr,
 110                                                   string_term_tdb_data(sidstr),
 111                                                   TDB_INSERT);
 112                 talloc_free(numstr);
 113                 if (!NT_STATUS_IS_OK(ret)) {
 114                         DEBUG(1, ("Fatal error while storing "
 115                                   "new domain->range assignment!\n"));
 116                         goto error;
 117                 }
 118                 DEBUG(5, ("Acquired new range #%d for domain %s\n",
 119                           domainnum, sidstr));
 120         }
 121
 122         DEBUG(10, ("Using range #%d for domain %s\n", domainnum, sidstr));
 123         cfg->domainnum = domainnum;
 124
 125         return NT_STATUS_OK;
 126
 127       error:
 128         return ret;
 129
 130 }
 131
 132 static NTSTATUS idmap_autorid_id_to_sid(struct autorid_global_config *cfg,
 133                                         struct id_map *map)
 134 {
 135         uint32_t range;
 136         TDB_DATA data;
 137         char *keystr;
 138         struct dom_sid sid;
 139         NTSTATUS status;
 140
 141         /* can this be one of our ids? */
 142         if (map->xid.id < cfg->minvalue) {
 143                 DEBUG(10, ("id %d is lower than minimum value, "
 144                            "ignoring mapping request\n", map->xid.id));
 145                 map->status = ID_UNKNOWN;
 146                 return NT_STATUS_OK;
 147         }
 148
 149         if (map->xid.id > (cfg->minvalue + cfg->rangesize * cfg->maxranges)) {
 150                 DEBUG(10, ("id %d is outside of maximum id value, "
 151                            "ignoring mapping request\n", map->xid.id));
 152                 map->status = ID_UNKNOWN;
 153                 return NT_STATUS_OK;
 154         }
 155
 156         /* determine the range of this uid */
 157         range = ((map->xid.id - cfg->minvalue) / cfg->rangesize);
 158
 159         keystr = talloc_asprintf(talloc_tos(), "%u", range);
 160         if (!keystr) {
 161                 return NT_STATUS_NO_MEMORY;
 162         }
 163
 164         status = dbwrap_fetch_bystring(autorid_db, talloc_tos(), keystr, &data);
 165         TALLOC_FREE(keystr);
 166
 167         if (!NT_STATUS_IS_OK(status)) {
 168                 DEBUG(4, ("id %d belongs to range %d which does not have "
 169                           "domain mapping, ignoring mapping request\n",
 170                           map->xid.id, range));
 171                 map->status = ID_UNKNOWN;
 172                 return NT_STATUS_OK;
 173         }
 174
 175         string_to_sid(&sid, (const char *)data.dptr);
 176         TALLOC_FREE(data.dptr);
 177
 178         sid_compose(map->sid, &sid,
 179                     (map->xid.id - cfg->minvalue -
 180                      range * cfg->rangesize));
 181
 182         /* We **really** should have some way of validating
 183            the SID exists and is the correct type here.  But
 184            that is a deficiency in the idmap_rid design. */
 185
 186         map->status = ID_MAPPED;
 187         return NT_STATUS_OK;
 188 }
 189
 190 /**********************************
 191  Single sid to id lookup function.
 192 **********************************/
 193
 194 static NTSTATUS idmap_autorid_sid_to_id(struct autorid_global_config *global,
 195                                         struct autorid_domain_config *domain,
 196                                         struct id_map *map)
 197 {
 198         uint32_t rid;
 199
 200         sid_peek_rid(map->sid, &rid);
 201
 202         /* if the rid is higher than the size of the range, we cannot map it */
 203         if (rid >= global->rangesize) {
 204                 map->status = ID_UNKNOWN;
 205                 DEBUG(2, ("RID %d is larger then size of range (%d), "
 206                           "user cannot be mapped\n", rid, global->rangesize));
 207                 return NT_STATUS_UNSUCCESSFUL;
 208         }
 209         map->xid.id = global->minvalue +
 210             (global->rangesize * domain->domainnum)+rid;
 211
 212         /* We **really** should have some way of validating
 213            the SID exists and is the correct type here.  But
 214            that is a deficiency in the idmap_rid design. */
 215
 216         map->status = ID_MAPPED;
 217
 218         return NT_STATUS_OK;
 219 }
 220
 221 /**********************************
 222  lookup a set of unix ids.
 223 **********************************/
 224
 225 static NTSTATUS idmap_autorid_unixids_to_sids(struct idmap_domain *dom,
 226                                               struct id_map **ids)
 227 {
 228         struct autorid_global_config *globalcfg;
 229         NTSTATUS ret;
 230         int i;
 231
 232         /* initialize the status to avoid surprise */
 233         for (i = 0; ids[i]; i++) {
 234                 ids[i]->status = ID_UNKNOWN;
 235         }
 236
 237         globalcfg = talloc_get_type(dom->private_data,
 238                                     struct autorid_global_config);
 239
 240         for (i = 0; ids[i]; i++) {
 241
 242                 ret = idmap_autorid_id_to_sid(globalcfg, ids[i]);
 243
 244                 if ((!NT_STATUS_IS_OK(ret)) &&
 245                     (!NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED))) {
 246                         /* some fatal error occurred, log it */
 247                         DEBUG(3, ("Unexpected error resolving an ID "
 248                                   " (%d)\n", ids[i]->xid.id));
 249                         goto failure;
 250                 }
 251         }
 252         return NT_STATUS_OK;
 253
 254       failure:
 255         return ret;
 256 }
 257
 258 /**********************************
 259  lookup a set of sids.
 260 **********************************/
 261
 262 static NTSTATUS idmap_autorid_sids_to_unixids(struct idmap_domain *dom,
 263                                               struct id_map **ids)
 264 {
 265         struct autorid_global_config *global;
 266         NTSTATUS ret;
 267         int i;
 268
 269         /* initialize the status to avoid surprise */
 270         for (i = 0; ids[i]; i++) {
 271                 ids[i]->status = ID_UNKNOWN;
 272         }
 273
 274         global = talloc_get_type(dom->private_data,
 275                                  struct autorid_global_config);
 276
 277         for (i = 0; ids[i]; i++) {
 278                 struct winbindd_tdc_domain *domain;
 279                 struct autorid_domain_config domaincfg;
 280                 uint32_t rid;
 281
 282                 ZERO_STRUCT(domaincfg);
 283
 284                 sid_copy(&domaincfg.sid, ids[i]->sid);
 285                 if (!sid_split_rid(&domaincfg.sid, &rid)) {
 286                         DEBUG(4, ("Could not determine domain SID from %s, "
 287                                   "ignoring mapping request\n",
 288                                   sid_string_dbg(ids[i]->sid)));
 289                         continue;
 290                 }
 291
 292                 /*
 293                  * Check if the domain is around
 294                  */
 295                 domain = wcache_tdc_fetch_domainbysid(talloc_tos(),
 296                                                       &domaincfg.sid);
 297                 if (domain == NULL) {
 298                         DEBUG(10, ("Ignoring unknown domain sid %s\n",
 299                                    sid_string_dbg(&domaincfg.sid)));
 300                         continue;
 301                 }
 302                 TALLOC_FREE(domain);
 303
 304                 domaincfg.globalcfg = global;
 305
 306                 ret = dbwrap_trans_do(autorid_db,
 307                                       idmap_autorid_get_domainrange,
 308                                       &domaincfg);
 309
 310                 if (!NT_STATUS_IS_OK(ret)) {
 311                         DEBUG(3, ("Could not determine range for domain, "
 312                                   "check previous messages for reason\n"));
 313                         goto failure;
 314                 }
 315
 316                 ret = idmap_autorid_sid_to_id(global, &domaincfg, ids[i]);
 317
 318                 if ((!NT_STATUS_IS_OK(ret)) &&
 319                     (!NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED))) {
 320                         /* some fatal error occurred, log it */
 321                         DEBUG(3, ("Unexpected error resolving a SID (%s)\n",
 322                                   sid_string_dbg(ids[i]->sid)));
 323                         goto failure;
 324                 }
 325         }
 326         return NT_STATUS_OK;
 327
 328       failure:
 329         return ret;
 330
 331 }
 332
 333 /*
 334  * open and initialize the database which stores the ranges for the domains
 335  */
 336 static NTSTATUS idmap_autorid_db_init(void)
 337 {
 338         int32_t hwm;
 339
 340         if (autorid_db) {
 341                 /* its already open */
 342                 return NT_STATUS_OK;
 343         }
 344
 345         /* Open idmap repository */
 346         autorid_db = db_open(NULL, state_path("autorid.tdb"), 0,
 347                              TDB_DEFAULT, O_RDWR | O_CREAT, 0644);
 348
 349         if (!autorid_db) {
 350                 DEBUG(0, ("Unable to open idmap_autorid database '%s'\n",
 351                           state_path("autorid.tdb")));
 352                 return NT_STATUS_UNSUCCESSFUL;
 353         }
 354
 355         /* Initialize high water mark for the currently used range to 0 */
 356         hwm = dbwrap_fetch_int32(autorid_db, HWM);
 357         if ((hwm < 0)) {
 358                 if (!NT_STATUS_IS_OK
 359                     (dbwrap_trans_store_int32(autorid_db, HWM, 0))) {
 360                         DEBUG(0,
 361                               ("Unable to initialise HWM in autorid "
 362                                "database\n"));
 363                         return NT_STATUS_INTERNAL_DB_ERROR;
 364                 }
 365         }
 366
 367         /* Initialize high water mark for alloc pool to 0 */
 368         hwm = dbwrap_fetch_int32(autorid_db, ALLOC_HWM);
 369         if ((hwm < 0)) {
 370                 if (!NT_STATUS_IS_OK
 371                     (dbwrap_trans_store_int32(autorid_db, ALLOC_HWM, 0))) {
 372                         DEBUG(0,
 373                               ("Unable to initialise HWM in autorid "
 374                                "database\n"));
 375                         return NT_STATUS_INTERNAL_DB_ERROR;
 376                 }
 377         }
 378         return NT_STATUS_OK;
 379 }
 380
 381 static struct autorid_global_config *idmap_autorid_loadconfig(TALLOC_CTX * ctx)
 382 {
 383
 384         TDB_DATA data;
 385         struct autorid_global_config *cfg;
 386         unsigned long minvalue, rangesize, maxranges;
 387         NTSTATUS status;
 388
 389         status = dbwrap_fetch_bystring(autorid_db, ctx, CONFIGKEY, &data);
 390
 391         if (!NT_STATUS_IS_OK(status)) {
 392                 DEBUG(10, ("No saved config found\n"));
 393                 return NULL;
 394         }
 395
 396         cfg = talloc_zero(ctx, struct autorid_global_config);
 397         if (!cfg) {
 398                 return NULL;
 399         }
 400
 401         if (sscanf((char *)data.dptr,
 402                    "minvalue:%lu rangesize:%lu maxranges:%lu",
 403                    &minvalue, &rangesize, &maxranges) != 3) {
 404                 DEBUG(1,
 405                       ("Found invalid configuration data"
 406                        "creating new config\n"));
 407                 return NULL;
 408         }
 409
 410         cfg->minvalue = minvalue;
 411         cfg->rangesize = rangesize;
 412         cfg->maxranges = maxranges;
 413
 414         DEBUG(10, ("Loaded previously stored configuration "
 415                    "minvalue:%d rangesize:%d\n",
 416                    cfg->minvalue, cfg->rangesize));
 417
 418         return cfg;
 419
 420 }
 421
 422 static NTSTATUS idmap_autorid_saveconfig(struct autorid_global_config *cfg)
 423 {
 424
 425         NTSTATUS status;
 426         TDB_DATA data;
 427         char *cfgstr;
 428
 429         cfgstr =
 430             talloc_asprintf(talloc_tos(),
 431                             "minvalue:%u rangesize:%u maxranges:%u",
 432                             cfg->minvalue, cfg->rangesize, cfg->maxranges);
 433
 434         if (!cfgstr) {
 435                 return NT_STATUS_NO_MEMORY;
 436         }
 437
 438         data = string_tdb_data(cfgstr);
 439
 440         status = dbwrap_trans_store_bystring(autorid_db, CONFIGKEY,
 441                                              data, TDB_REPLACE);
 442
 443         talloc_free(cfgstr);
 444
 445         return status;
 446 }
 447
 448 static NTSTATUS idmap_autorid_initialize(struct idmap_domain *dom)
 449 {
 450         struct autorid_global_config *config;
 451         struct autorid_global_config *storedconfig = NULL;
 452         NTSTATUS status;
 453         uint32_t hwm;
 454
 455         if (!strequal(dom->name, "*")) {
 456                 DEBUG(0, ("idmap_autorid_initialize: Error: autorid configured "
 457                           "for domain '%s'. But autorid can only be used for "
 458                           "the default idmap configuration.\n", dom->name));
 459                 return NT_STATUS_INVALID_PARAMETER;
 460         }
 461
 462         config = talloc_zero(dom, struct autorid_global_config);
 463         if (!config) {
 464                 DEBUG(0, ("Out of memory!\n"));
 465                 return NT_STATUS_NO_MEMORY;
 466         }
 467
 468         status = idmap_autorid_db_init();
 469         if (!NT_STATUS_IS_OK(status)) {
 470                 goto error;
 471         }
 472
 473         config->minvalue = dom->low_id;
 474         config->rangesize = lp_parm_int(-1, "idmap config *", "rangesize", 100000);
 475
 476         if (config->rangesize < 2000) {
 477                 DEBUG(1, ("autorid rangesize must be at least 2000\n"));
 478                 status = NT_STATUS_INVALID_PARAMETER;
 479                 goto error;
 480         }
 481
 482         config->maxranges = (dom->high_id - dom->low_id + 1) /
 483             config->rangesize;
 484
 485         if (config->maxranges == 0) {
 486                 DEBUG(1, ("allowed uid range is smaller then rangesize, "
 487                           "increase uid range or decrease rangesize\n"));
 488                 status = NT_STATUS_INVALID_PARAMETER;
 489                 goto error;
 490         }
 491
 492         /* check if the high-low limit is a multiple of the rangesize */
 493         if ((dom->high_id - dom->low_id + 1) % config->rangesize != 0) {
 494                 DEBUG(5, ("High uid-low uid difference of %d "
 495                           "is not a multiple of the rangesize %d, "
 496                           "limiting ranges to lower boundary number of %d\n",
 497                           (dom->high_id - dom->low_id + 1), config->rangesize,
 498                           config->maxranges));
 499         }
 500
 501         DEBUG(10, ("Current configuration in config is "
 502                    "minvalue:%d rangesize:%d maxranges:%d\n",
 503                    config->minvalue, config->rangesize, config->maxranges));
 504
 505         /* read previously stored config and current HWM */
 506         storedconfig = idmap_autorid_loadconfig(talloc_tos());
 507
 508         if (!dbwrap_fetch_uint32(autorid_db, HWM, &hwm)) {
 509                 DEBUG(1, ("Fatal error while fetching current "
 510                           "HWM value!\n"));
 511                 status = NT_STATUS_INTERNAL_ERROR;
 512                 goto error;
 513         }
 514
 515         /* did the minimum value or rangesize change? */
 516         if (storedconfig &&
 517             ((storedconfig->minvalue != config->minvalue) ||
 518              (storedconfig->rangesize != config->rangesize))) {
 519                 DEBUG(1, ("New configuration values for rangesize or "
 520                           "minimum uid value conflict with previously "
 521                           "used values! Aborting initialization\n"));
 522                 status = NT_STATUS_INVALID_PARAMETER;
 523                 goto error;
 524         }
 525
 526         /*
 527          * has the highest uid value been reduced to setting that is not
 528          * sufficient any more for already existing ranges?
 529          */
 530         if (hwm > config->maxranges) {
 531                 DEBUG(1, ("New upper uid limit is too low to cover "
 532                           "existing mappings! Aborting initialization\n"));
 533                 status = NT_STATUS_INVALID_PARAMETER;
 534                 goto error;
 535         }
 536
 537         status = idmap_autorid_saveconfig(config);
 538
 539         if (!NT_STATUS_IS_OK(status)) {
 540                 DEBUG(1, ("Failed to store configuration data!\n"));
 541                 goto error;
 542         }
 543
 544         DEBUG(5, ("%d domain ranges with a size of %d are available\n",
 545                   config->maxranges, config->rangesize));
 546
 547         dom->private_data = config;
 548
 549         goto done;
 550
 551 error:
 552         talloc_free(config);
 553
 554 done:
 555         talloc_free(storedconfig);
 556
 557         return status;
 558 }
 559
 560 static NTSTATUS idmap_autorid_allocate_id(struct idmap_domain *dom,
 561                                           struct unixid *xid) {
 562
 563         struct autorid_global_config *globalcfg;
 564         NTSTATUS ret;
 565         uint32_t hwm;
 566
 567         if (!strequal(dom->name, "*")) {
 568                 DEBUG(3, ("idmap_autorid_allocate_id: "
 569                           "Refusing creation of mapping for domain'%s'. "
 570                           "Currently only supported for the default "
 571                           "domain \"*\".\n",
 572                            dom->name));
 573                 return NT_STATUS_NOT_IMPLEMENTED;
 574         }
 575
 576         globalcfg = talloc_get_type(dom->private_data,
 577                                     struct autorid_global_config);
 578
 579         if (!dbwrap_fetch_uint32(autorid_db, ALLOC_HWM, &hwm)) {
 580                 DEBUG(1, ("Failed to fetch current allocation HWM value!\n"));
 581                 return NT_STATUS_INTERNAL_ERROR;
 582         }
 583
 584         if (hwm > ALLOC_POOL_SIZE) {
 585                 DEBUG(1, ("allocation pool is depleted!\n"));
 586                 return NT_STATUS_NO_MEMORY;
 587         }
 588
 589         ret = dbwrap_change_uint32_atomic(autorid_db, ALLOC_HWM, &(xid->id), 1);
 590         if (!NT_STATUS_IS_OK(ret)) {
 591                 DEBUG(1, ("Fatal error while allocating new ID!\n"));
 592         }
 593         xid->id = (xid->id)+(globalcfg->minvalue);
 594
 595         return ret;
 596 }
 597
 598 /*
 599   Close the idmap tdb instance
 600 */
 601 static struct idmap_methods autorid_methods = {
 602         .init = idmap_autorid_initialize,
 603         .unixids_to_sids = idmap_autorid_unixids_to_sids,
 604         .sids_to_unixids = idmap_autorid_sids_to_unixids,
 605         .allocate_id     = idmap_autorid_allocate_id
 606 };
 607
 608 NTSTATUS idmap_autorid_init(void)
 609 {
 610         return smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION,
 611                                   "autorid", &autorid_methods);
 612 }