autorid: make the checks for bumping num_mapped identical for alloc and rid case
[Samba.git] / source3 / winbindd / idmap_autorid.c
blobb66caec90226175e3073875f5f344c43824fcc98
1 /*
2 * idmap_autorid: static map between Active Directory/NT RIDs
3 * and RFC 2307 accounts
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
8 * Copyright (C) Christian Ambach, 2010-2012
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.
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.
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/>.
26 * This module allocates ranges for domains to be used in a
27 * algorithmic mode like idmap_rid. Multiple ranges are supported
28 * for a single domain: If a rid exceeds the range size, a matching
29 * range is allocated to hold the rid's id.
31 * Here are the formulas applied:
34 * For a sid of the form domain_sid-rid, we have
36 * rid = reduced_rid + domain_range_index * range_size
38 * with
39 * reduced_rid := rid % range_size
40 * domain_range_index := rid / range_size
42 * And reduced_rid fits into a range.
44 * In the database, we associate a range_number to
45 * the pair domain_sid,domain_range_index.
47 * Now the unix id for the given sid calculates as:
49 * id = reduced_rid + range_low_id
51 * with
53 * range_low_id = low_id + range_number * range_size
56 * The inverse calculation goes like this:
58 * Given a unix id, let
60 * normalized_id := id - low_id
61 * reduced_rid := normalized_id % range_size
62 * range_number = normalized_id / range_size
64 * Then we have
66 * id = reduced_rid + low_id + range_number * range_size
68 * From the database, get the domain_sid,domain_range_index pair
69 * belonging to the range_number (if there is already one).
71 * Then the rid for the unix id calculates as:
73 * rid = reduced_rid + domain_range_index * range_size
76 #include "idmap_autorid_tdb.h"
77 #include "winbindd.h"
78 #include "idmap.h"
79 #include "idmap_rw.h"
80 #include "../libcli/security/dom_sid.h"
82 #undef DBGC_CLASS
83 #define DBGC_CLASS DBGC_IDMAP
85 /* handle to the tdb storing domain <-> range assignments */
86 static struct db_context *autorid_db;
88 static bool ignore_builtin = false;
90 static NTSTATUS idmap_autorid_get_alloc_range(struct idmap_domain *dom,
91 struct autorid_range_config *range)
93 NTSTATUS status;
95 ZERO_STRUCT(*range);
97 fstrcpy(range->domsid, ALLOC_RANGE);
99 status = idmap_autorid_get_domainrange(autorid_db,
100 range,
101 dom->read_only);
103 return status;
106 static NTSTATUS idmap_autorid_allocate_id(struct idmap_domain *dom,
107 struct unixid *xid) {
109 NTSTATUS ret;
110 struct autorid_range_config range;
112 if (dom->read_only) {
113 DEBUG(3, ("Backend is read-only, refusing "
114 "new allocation request\n"));
115 return NT_STATUS_UNSUCCESSFUL;
118 /* fetch the range for the allocation pool */
120 ret = idmap_autorid_get_alloc_range(dom, &range);
121 if (!NT_STATUS_IS_OK(ret)) {
122 DEBUG(3, ("Could not determine range for allocation pool, "
123 "check previous messages for reason\n"));
124 return ret;
127 ret = idmap_tdb_common_get_new_id(dom, xid);
129 if (!NT_STATUS_IS_OK(ret)) {
130 DEBUG(1, ("Fatal error while allocating new ID!\n"));
131 return ret;
134 xid->id = xid->id + range.low_id;
136 DEBUG(10, ("Returned new %s %d from allocation range\n",
137 (xid->type==ID_TYPE_UID)?"uid":"gid", xid->id));
139 return ret;
143 * map a SID to xid using the idmap_tdb like pool
145 static NTSTATUS idmap_autorid_id_to_sid_alloc(struct idmap_domain *dom,
146 struct id_map *map)
148 NTSTATUS ret;
150 /* look out for the mapping */
151 ret = idmap_tdb_common_unixid_to_sid(dom, map);
153 if (NT_STATUS_IS_OK(ret)) {
154 map->status = ID_MAPPED;
155 return ret;
158 map->status = ID_UNKNOWN;
160 DEBUG(10, ("no ID->SID mapping for %d could be found\n", map->xid.id));
162 return ret;
165 static NTSTATUS idmap_autorid_id_to_sid(struct autorid_global_config *cfg,
166 struct idmap_domain *dom,
167 struct id_map *map)
169 uint32_t range_number;
170 uint32_t domain_range_index = 0;
171 uint32_t normalized_id;
172 uint32_t reduced_rid;
173 uint32_t rid;
174 TDB_DATA data = tdb_null;
175 char *keystr;
176 struct dom_sid domsid;
177 NTSTATUS status;
178 bool ok;
179 const char *q = NULL;
181 /* can this be one of our ids? */
182 if (map->xid.id < cfg->minvalue) {
183 DEBUG(10, ("id %d is lower than minimum value, "
184 "ignoring mapping request\n", map->xid.id));
185 map->status = ID_UNKNOWN;
186 return NT_STATUS_OK;
189 if (map->xid.id > (cfg->minvalue + cfg->rangesize * cfg->maxranges)) {
190 DEBUG(10, ("id %d is outside of maximum id value, "
191 "ignoring mapping request\n", map->xid.id));
192 map->status = ID_UNKNOWN;
193 return NT_STATUS_OK;
196 /* determine the range of this uid */
198 normalized_id = map->xid.id - cfg->minvalue;
199 range_number = normalized_id / cfg->rangesize;
201 keystr = talloc_asprintf(talloc_tos(), "%u", range_number);
202 if (!keystr) {
203 return NT_STATUS_NO_MEMORY;
206 status = dbwrap_fetch_bystring(autorid_db, talloc_tos(), keystr, &data);
207 TALLOC_FREE(keystr);
209 if (!NT_STATUS_IS_OK(status)) {
210 DEBUG(4, ("id %d belongs to range %d which does not have "
211 "domain mapping, ignoring mapping request\n",
212 map->xid.id, range_number));
213 TALLOC_FREE(data.dptr);
214 map->status = ID_UNKNOWN;
215 return NT_STATUS_OK;
218 if (strncmp((const char *)data.dptr,
219 ALLOC_RANGE,
220 strlen(ALLOC_RANGE)) == 0) {
222 * this is from the alloc range, check if there is a mapping
224 DEBUG(5, ("id %d belongs to allocation range, "
225 "checking for mapping\n",
226 map->xid.id));
227 TALLOC_FREE(data.dptr);
228 return idmap_autorid_id_to_sid_alloc(dom, map);
231 ok = dom_sid_parse_endp((const char *)data.dptr, &domsid, &q);
232 TALLOC_FREE(data.dptr);
233 if (!ok) {
234 map->status = ID_UNKNOWN;
235 return NT_STATUS_OK;
237 if ((q != NULL) && (*q != '\0'))
238 if (sscanf(q+1, "%"SCNu32, &domain_range_index) != 1) {
239 DEBUG(10, ("Domain range index not found, "
240 "ignoring mapping request\n"));
241 map->status = ID_UNKNOWN;
242 return NT_STATUS_OK;
245 reduced_rid = normalized_id % cfg->rangesize;
246 rid = reduced_rid + domain_range_index * cfg->rangesize;
248 sid_compose(map->sid, &domsid, rid);
250 /* We **really** should have some way of validating
251 the SID exists and is the correct type here. But
252 that is a deficiency in the idmap_rid design. */
254 map->status = ID_MAPPED;
255 map->xid.type = ID_TYPE_BOTH;
257 return NT_STATUS_OK;
260 /**********************************
261 Single sid to id lookup function.
262 **********************************/
264 static NTSTATUS idmap_autorid_sid_to_id_rid(
265 struct autorid_global_config *global,
266 struct autorid_range_config *range,
267 struct id_map *map)
269 uint32_t rid;
270 uint32_t reduced_rid;
272 sid_peek_rid(map->sid, &rid);
274 reduced_rid = rid % global->rangesize;
276 map->xid.id = reduced_rid + range->low_id;
277 map->xid.type = ID_TYPE_BOTH;
278 map->status = ID_MAPPED;
280 return NT_STATUS_OK;
283 /**********************************
284 lookup a set of unix ids.
285 **********************************/
287 static NTSTATUS idmap_autorid_unixids_to_sids(struct idmap_domain *dom,
288 struct id_map **ids)
290 struct idmap_tdb_common_context *commoncfg;
291 struct autorid_global_config *globalcfg;
292 NTSTATUS ret;
293 int i;
294 int num_tomap = 0;
295 int num_mapped = 0;
297 /* initialize the status to avoid surprise */
298 for (i = 0; ids[i]; i++) {
299 ids[i]->status = ID_UNKNOWN;
300 num_tomap++;
303 commoncfg =
304 talloc_get_type_abort(dom->private_data,
305 struct idmap_tdb_common_context);
307 globalcfg = talloc_get_type(commoncfg->private_data,
308 struct autorid_global_config);
310 for (i = 0; ids[i]; i++) {
312 ret = idmap_autorid_id_to_sid(globalcfg, dom, ids[i]);
314 if ((!NT_STATUS_IS_OK(ret)) &&
315 (!NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED))) {
316 /* some fatal error occurred, log it */
317 DEBUG(3, ("Unexpected error resolving an ID "
318 " (%d)\n", ids[i]->xid.id));
319 goto failure;
322 if (NT_STATUS_IS_OK(ret) && ids[i]->status == ID_MAPPED) {
323 num_mapped++;
328 if (num_tomap == num_mapped) {
329 return NT_STATUS_OK;
330 } else if (num_mapped == 0) {
331 return NT_STATUS_NONE_MAPPED;
334 return STATUS_SOME_UNMAPPED;
337 failure:
338 return ret;
342 * map a SID to xid using the idmap_tdb like pool
344 static NTSTATUS idmap_autorid_sid_to_id_alloc(struct idmap_domain *dom,
345 struct id_map *map,
346 struct idmap_tdb_common_context *ctx)
348 NTSTATUS ret;
349 int res;
351 map->status = ID_UNKNOWN;
353 /* see if we already have a mapping */
354 ret = idmap_tdb_common_sid_to_unixid(dom, map);
356 if (NT_STATUS_IS_OK(ret)) {
357 map->status = ID_MAPPED;
358 return ret;
361 /* bad things happened */
362 if (!NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED)) {
363 DEBUG(1, ("Looking up SID->ID mapping for %s failed: %s\n",
364 sid_string_dbg(map->sid), nt_errstr(ret)));
365 return ret;
368 if (dom->read_only) {
369 DEBUG(3, ("Not allocating new mapping for %s, because backend "
370 "is read-only\n", sid_string_dbg(map->sid)));
371 map->status = ID_UNMAPPED;
372 return NT_STATUS_NONE_MAPPED;
375 DEBUG(10, ("Creating new mapping in pool for %s\n",
376 sid_string_dbg(map->sid)));
378 /* create new mapping */
379 res = dbwrap_transaction_start(ctx->db);
380 if (res != 0) {
381 DEBUG(2, ("transaction_start failed\n"));
382 return NT_STATUS_INTERNAL_DB_CORRUPTION;
385 ret = idmap_tdb_common_new_mapping(dom, map);
386 if (!NT_STATUS_IS_OK(ret)) {
387 if (dbwrap_transaction_cancel(ctx->db) != 0) {
388 smb_panic("Cancelling transaction failed");
390 map->status = ID_UNMAPPED;
391 return ret;
394 res = dbwrap_transaction_commit(ctx->db);
395 if (res == 0) {
396 map->status = ID_MAPPED;
397 return NT_STATUS_OK;
400 DEBUG(2, ("transaction_commit failed\n"));
401 return NT_STATUS_INTERNAL_DB_CORRUPTION;
405 /**********************************
406 lookup a set of sids.
407 **********************************/
409 static NTSTATUS idmap_autorid_sids_to_unixids(struct idmap_domain *dom,
410 struct id_map **ids)
412 struct idmap_tdb_common_context *commoncfg;
413 struct autorid_global_config *global;
414 NTSTATUS ret;
415 int i;
416 int num_tomap = 0;
417 int num_mapped = 0;
419 /* initialize the status to avoid surprise */
420 for (i = 0; ids[i]; i++) {
421 ids[i]->status = ID_UNKNOWN;
422 num_tomap++;
425 commoncfg =
426 talloc_get_type_abort(dom->private_data,
427 struct idmap_tdb_common_context);
429 global = talloc_get_type(commoncfg->private_data,
430 struct autorid_global_config);
432 for (i = 0; ids[i]; i++) {
433 struct winbindd_tdc_domain *domain;
434 struct autorid_range_config range;
435 uint32_t rid;
436 struct dom_sid domainsid;
438 ZERO_STRUCT(range);
440 DEBUG(10, ("Trying to map %s\n", sid_string_dbg(ids[i]->sid)));
442 sid_copy(&domainsid, ids[i]->sid);
443 if (!sid_split_rid(&domainsid, &rid)) {
444 DEBUG(4, ("Could not determine domain SID from %s, "
445 "ignoring mapping request\n",
446 sid_string_dbg(ids[i]->sid)));
447 continue;
450 /* is this a well-known SID? */
452 if (sid_check_is_wellknown_domain(&domainsid, NULL)) {
454 DEBUG(10, ("SID %s is well-known, using pool\n",
455 sid_string_dbg(ids[i]->sid)));
457 ret = idmap_autorid_sid_to_id_alloc(dom, ids[i],
458 commoncfg);
460 if (!NT_STATUS_IS_OK(ret) &&
461 !NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED)) {
462 DEBUG(3, ("Unexpected error resolving "
463 "SID (%s)\n",
464 sid_string_dbg(ids[i]->sid)));
465 goto failure;
468 if (NT_STATUS_IS_OK(ret) && ids[i]->status == ID_MAPPED) {
469 num_mapped++;
472 continue;
475 /* BUILTIN is passdb's job */
476 if (dom_sid_equal(&domainsid, &global_sid_Builtin) &&
477 ignore_builtin) {
478 DEBUG(10, ("Ignoring request for BUILTIN domain\n"));
479 continue;
483 * Check if the domain is around
485 domain = wcache_tdc_fetch_domainbysid(talloc_tos(),
486 &domainsid);
487 if (domain == NULL) {
488 DEBUG(10, ("Ignoring unknown domain sid %s\n",
489 sid_string_dbg(&domainsid)));
490 continue;
492 TALLOC_FREE(domain);
494 sid_to_fstring(range.domsid, &domainsid);
496 /* Calculate domain_range_index for multi-range support */
497 range.domain_range_index = rid / (global->rangesize);
499 ret = idmap_autorid_get_domainrange(autorid_db, &range,
500 dom->read_only);
502 /* read-only mode and a new domain range would be required? */
503 if (NT_STATUS_EQUAL(ret, NT_STATUS_NOT_FOUND) &&
504 dom->read_only) {
505 DEBUG(10, ("read-only is enabled, did not allocate "
506 "new range for domain %s\n",
507 sid_string_dbg(&domainsid)));
508 continue;
511 if (!NT_STATUS_IS_OK(ret)) {
512 DEBUG(3, ("Could not determine range for domain, "
513 "check previous messages for reason\n"));
514 goto failure;
517 ret = idmap_autorid_sid_to_id_rid(global, &range, ids[i]);
519 if ((!NT_STATUS_IS_OK(ret)) &&
520 (!NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED))) {
521 /* some fatal error occurred, log it */
522 DEBUG(3, ("Unexpected error resolving a SID (%s)\n",
523 sid_string_dbg(ids[i]->sid)));
524 goto failure;
527 if (NT_STATUS_IS_OK(ret) && ids[i]->status == ID_MAPPED) {
528 num_mapped++;
532 if (num_tomap == num_mapped) {
533 return NT_STATUS_OK;
534 } else if (num_mapped == 0) {
535 return NT_STATUS_NONE_MAPPED;
538 return STATUS_SOME_UNMAPPED;
540 failure:
541 return ret;
545 static NTSTATUS idmap_autorid_preallocate_wellknown(struct idmap_domain *dom)
547 const char *groups[] = { "S-1-1-0", "S-1-2-0", "S-1-2-1",
548 "S-1-3-0", "S-1-3-1", "S-1-3-2", "S-1-3-3", "S-1-3-4",
549 "S-1-5-1", "S-1-5-2", "S-1-5-3", "S-1-5-4", "S-1-5-6",
550 "S-1-5-7", "S-1-5-8", "S-1-5-9", "S-1-5-10", "S-1-5-11",
551 "S-1-5-12", "S-1-5-13", "S-1-5-14", "S-1-5-15",
552 "S-1-5-17", "S-1-5-18", "S-1-5-19", "S-1-5-20"
555 struct id_map **maps;
556 int i, num;
557 NTSTATUS status;
559 if (dom->read_only) {
560 return NT_STATUS_OK;
563 num = ARRAY_SIZE(groups);
565 maps = talloc_array(talloc_tos(), struct id_map*, num+1);
566 if (!maps) {
567 return NT_STATUS_NO_MEMORY;
570 for (i = 0; i < num; i++) {
571 maps[i] = talloc(maps, struct id_map);
572 if (maps[i] == NULL) {
573 talloc_free(maps);
574 return NT_STATUS_NO_MEMORY;
576 maps[i]->xid.type = ID_TYPE_GID;
577 maps[i]->sid = dom_sid_parse_talloc(maps, groups[i]);
580 maps[num] = NULL;
582 status = idmap_autorid_sids_to_unixids(dom, maps);
584 DEBUG(10,("Preallocation run finished with status %s\n",
585 nt_errstr(status)));
587 talloc_free(maps);
589 return NT_STATUS_IS_OK(status)?NT_STATUS_OK:NT_STATUS_UNSUCCESSFUL;
592 static NTSTATUS idmap_autorid_initialize_action(struct db_context *db,
593 void *private_data)
595 struct idmap_domain *dom;
596 struct idmap_tdb_common_context *common;
597 struct autorid_global_config *config;
598 NTSTATUS status;
600 dom = (struct idmap_domain *)private_data;
601 common = (struct idmap_tdb_common_context *)dom->private_data;
602 config = (struct autorid_global_config *)common->private_data;
604 status = idmap_autorid_init_hwms(db);
605 if (!NT_STATUS_IS_OK(status)) {
606 return status;
609 status = idmap_autorid_saveconfig(db, config);
610 if (!NT_STATUS_IS_OK(status)) {
611 DEBUG(1, ("Failed to store configuration data!\n"));
612 return status;
615 status = idmap_autorid_preallocate_wellknown(dom);
616 if (!NT_STATUS_IS_OK(status)) {
617 DEBUG(1, ("Failed to preallocate wellknown sids: %s\n",
618 nt_errstr(status)));
619 return status;
622 return NT_STATUS_OK;
625 static NTSTATUS idmap_autorid_initialize(struct idmap_domain *dom)
627 struct idmap_tdb_common_context *commonconfig;
628 struct autorid_global_config *config;
629 NTSTATUS status;
631 if (!strequal(dom->name, "*")) {
632 DEBUG(0, ("idmap_autorid_initialize: Error: autorid configured "
633 "for domain '%s'. But autorid can only be used for "
634 "the default idmap configuration.\n", dom->name));
635 return NT_STATUS_INVALID_PARAMETER;
638 commonconfig = talloc_zero(dom, struct idmap_tdb_common_context);
639 if (!commonconfig) {
640 DEBUG(0, ("Out of memory!\n"));
641 return NT_STATUS_NO_MEMORY;
643 dom->private_data = commonconfig;
645 commonconfig->rw_ops = talloc_zero(commonconfig, struct idmap_rw_ops);
646 if (commonconfig->rw_ops == NULL) {
647 DEBUG(0, ("Out of memory!\n"));
648 return NT_STATUS_NO_MEMORY;
651 config = talloc_zero(commonconfig, struct autorid_global_config);
652 if (!config) {
653 DEBUG(0, ("Out of memory!\n"));
654 return NT_STATUS_NO_MEMORY;
656 commonconfig->private_data = config;
658 config->minvalue = dom->low_id;
659 config->rangesize = lp_parm_int(-1, "idmap config *",
660 "rangesize", 100000);
662 config->maxranges = (dom->high_id - dom->low_id + 1) /
663 config->rangesize;
665 if (config->maxranges == 0) {
666 DEBUG(1, ("Allowed uid range is smaller than rangesize. "
667 "Increase uid range or decrease rangesize.\n"));
668 status = NT_STATUS_INVALID_PARAMETER;
669 goto error;
672 /* check if the high-low limit is a multiple of the rangesize */
673 if ((dom->high_id - dom->low_id + 1) % config->rangesize != 0) {
674 DEBUG(5, ("High uid-low uid difference of %d "
675 "is not a multiple of the rangesize %d, "
676 "limiting ranges to lower boundary number of %d\n",
677 (dom->high_id - dom->low_id + 1), config->rangesize,
678 config->maxranges));
681 DEBUG(5, ("%d domain ranges with a size of %d are available\n",
682 config->maxranges, config->rangesize));
684 ignore_builtin = lp_parm_bool(-1, "idmap config *",
685 "ignore builtin", false);
687 /* fill the TDB common configuration */
689 commonconfig->max_id = config->rangesize -1;
690 commonconfig->hwmkey_uid = ALLOC_HWM_UID;
691 commonconfig->hwmkey_gid = ALLOC_HWM_GID;
692 commonconfig->rw_ops->get_new_id = idmap_autorid_allocate_id;
693 commonconfig->rw_ops->set_mapping = idmap_tdb_common_set_mapping;
695 status = idmap_autorid_db_open(state_path("autorid.tdb"),
696 NULL, /* TALLOC_CTX */
697 &autorid_db);
698 if (!NT_STATUS_IS_OK(status)) {
699 goto error;
702 commonconfig->db = autorid_db;
704 status = dbwrap_trans_do(autorid_db,
705 idmap_autorid_initialize_action,
706 dom);
707 if (!NT_STATUS_IS_OK(status)) {
708 DEBUG(1, ("Failed to init the idmap database: %s\n",
709 nt_errstr(status)));
710 goto error;
713 goto done;
715 error:
716 talloc_free(config);
718 done:
719 return status;
723 Close the idmap tdb instance
725 static struct idmap_methods autorid_methods = {
726 .init = idmap_autorid_initialize,
727 .unixids_to_sids = idmap_autorid_unixids_to_sids,
728 .sids_to_unixids = idmap_autorid_sids_to_unixids,
729 .allocate_id = idmap_autorid_allocate_id
732 NTSTATUS idmap_autorid_init(void)
734 return smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION,
735 "autorid", &autorid_methods);