Fix bug #7669.
[Samba.git] / source / groupdb / mapping_ldb.c
blob658009af2ead197bcdd8545df3f0da853c7340b7
1 /*
2 * Unix SMB/CIFS implementation.
4 * group mapping code on top of ldb
6 * Copyright (C) Andrew Tridgell 2006
8 * based on tdb group mapping code from groupdb/mapping.c
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.
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/>.
24 #include "includes.h"
25 #include "groupdb/mapping.h"
26 #include "lib/ldb/include/includes.h"
27 #include "lib/ldb/include/ldb_errors.h"
29 static struct ldb_context *ldb;
31 static bool mapping_upgrade(const char *tdb_path);
34 connect to the group mapping ldb
36 static bool init_group_mapping(void)
38 bool existed;
39 const char *init_ldif[] =
40 { "dn: @ATTRIBUTES\n" \
41 "ntName: CASE_INSENSITIVE\n" \
42 "\n",
43 "dn: @INDEXLIST\n" \
44 "@IDXATTR: gidNumber\n" \
45 "@IDXATTR: ntName\n" \
46 "@IDXATTR: member\n" };
47 const char *db_path, *tdb_path;
48 int ret;
49 int flags = 0;
51 if (ldb != NULL) {
52 return True;
55 /* this is needed as Samba3 doesn't have this globally yet */
56 ldb_global_init();
58 db_path = state_path("group_mapping.ldb");
60 ldb = ldb_init(NULL);
61 if (ldb == NULL) goto failed;
63 /* Ensure this db is created read/write for root only. */
64 ldb_set_create_perms(ldb, 0600);
66 existed = file_exist(db_path, NULL);
68 if (lp_parm_bool(-1, "groupmap", "nosync", False)) {
69 flags |= LDB_FLG_NOSYNC;
72 if (!lp_use_mmap()) {
73 flags |= LDB_FLG_NOMMAP;
76 ret = ldb_connect(ldb, db_path, flags, NULL);
77 if (ret != LDB_SUCCESS) {
78 goto failed;
81 /* force the permissions on the ldb to 0600 - this will fix
82 existing databases as well as new ones */
83 if (chmod(db_path, 0600) != 0) {
84 goto failed;
87 if (!existed) {
88 /* initialise the ldb with an index */
89 struct ldb_ldif *ldif;
90 int i;
91 for (i=0;i<ARRAY_SIZE(init_ldif);i++) {
92 ldif = ldb_ldif_read_string(ldb, &init_ldif[i]);
93 if (ldif == NULL) goto failed;
94 ret = ldb_add(ldb, ldif->msg);
95 talloc_free(ldif);
96 if (ret == -1) goto failed;
100 /* possibly upgrade */
101 tdb_path = state_path("group_mapping.tdb");
102 if (file_exist(tdb_path, NULL) && !mapping_upgrade(tdb_path)) {
103 unlink(state_path("group_mapping.ldb"));
104 goto failed;
107 return True;
109 failed:
110 DEBUG(0,("Failed to open group mapping ldb '%s' - '%s'\n",
111 db_path, ldb?ldb_errstring(ldb):strerror(errno)));
112 talloc_free(ldb);
113 ldb = NULL;
114 return False;
119 form the DN for a mapping entry from a SID
121 static struct ldb_dn *mapping_dn(TALLOC_CTX *mem_ctx, const DOM_SID *sid)
123 fstring string_sid;
124 uint32_t rid;
125 DOM_SID domsid;
127 sid_copy(&domsid, sid);
128 if (!sid_split_rid(&domsid, &rid)) {
129 return NULL;
131 if (!sid_to_fstring(string_sid, &domsid)) {
132 return NULL;
134 /* we split by domain and rid so we can do a subtree search
135 when we only want one domain */
136 return ldb_dn_string_compose(mem_ctx, NULL, "rid=%u,domain=%s",
137 rid, string_sid);
141 add a group mapping entry
143 static bool add_mapping_entry(GROUP_MAP *map, int flag)
145 struct ldb_message *msg;
146 int ret, i;
147 fstring string_sid;
149 msg = ldb_msg_new(ldb);
150 if (msg == NULL) {
151 return False;
154 msg->dn = mapping_dn(msg, &map->sid);
155 if (msg->dn == NULL) {
156 goto failed;
159 if (ldb_msg_add_string(msg, "objectClass", "groupMap") != LDB_SUCCESS ||
160 ldb_msg_add_string(msg, "sid",
161 sid_to_fstring(string_sid, &map->sid)) != LDB_SUCCESS ||
162 ldb_msg_add_fmt(msg, "gidNumber", "%u", (unsigned)map->gid) != LDB_SUCCESS ||
163 ldb_msg_add_fmt(msg, "sidNameUse", "%u", (unsigned)map->sid_name_use) != LDB_SUCCESS ||
164 ldb_msg_add_string(msg, "comment", map->comment) != LDB_SUCCESS ||
165 ldb_msg_add_string(msg, "ntName", map->nt_name) != LDB_SUCCESS) {
166 goto failed;
169 ret = ldb_add(ldb, msg);
171 /* if it exists we update it. This is a hangover from the semantics the
172 tdb backend had */
173 if (ret == LDB_ERR_ENTRY_ALREADY_EXISTS) {
174 for (i=0;i<msg->num_elements;i++) {
175 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
177 ret = ldb_modify(ldb, msg);
180 talloc_free(msg);
182 return ret == LDB_SUCCESS;
184 failed:
185 talloc_free(msg);
186 return False;
190 unpack a ldb message into a GROUP_MAP structure
192 static bool msg_to_group_map(struct ldb_message *msg, GROUP_MAP *map)
194 const char *sidstr;
196 map->gid = ldb_msg_find_attr_as_int(msg, "gidNumber", -1);
197 map->sid_name_use = ldb_msg_find_attr_as_int(msg, "sidNameUse", -1);
198 fstrcpy(map->nt_name, ldb_msg_find_attr_as_string(msg, "ntName", NULL));
199 fstrcpy(map->comment, ldb_msg_find_attr_as_string(msg, "comment", NULL));
200 sidstr = ldb_msg_find_attr_as_string(msg, "sid", NULL);
202 if (!string_to_sid(&map->sid, sidstr) ||
203 map->gid == (gid_t)-1 ||
204 map->sid_name_use == (enum lsa_SidType)-1) {
205 DEBUG(0,("Unable to unpack group mapping\n"));
206 return False;
209 return True;
213 return a group map entry for a given sid
215 static bool get_group_map_from_sid(DOM_SID sid, GROUP_MAP *map)
217 int ret;
218 struct ldb_dn *dn;
219 struct ldb_result *res=NULL;
221 dn = mapping_dn(ldb, &sid);
222 if (dn == NULL) goto failed;
224 ret = ldb_search(ldb, dn, LDB_SCOPE_BASE, NULL, NULL, &res);
225 talloc_steal(dn, res);
226 if (ret != LDB_SUCCESS || res->count != 1) {
227 goto failed;
230 if (!msg_to_group_map(res->msgs[0], map)) goto failed;
232 talloc_free(dn);
233 return True;
235 failed:
236 talloc_free(dn);
237 return False;
241 return a group map entry for a given gid
243 static bool get_group_map_from_gid(gid_t gid, GROUP_MAP *map)
245 int ret;
246 char *expr;
247 struct ldb_result *res=NULL;
249 expr = talloc_asprintf(ldb, "(&(gidNumber=%u)(objectClass=groupMap))",
250 (unsigned)gid);
251 if (expr == NULL) goto failed;
253 ret = ldb_search(ldb, NULL, LDB_SCOPE_SUBTREE, expr, NULL, &res);
254 talloc_steal(expr, res);
255 if (ret != LDB_SUCCESS || res->count != 1) goto failed;
257 if (!msg_to_group_map(res->msgs[0], map)) goto failed;
259 talloc_free(expr);
260 return True;
262 failed:
263 talloc_free(expr);
264 return False;
268 Return the sid and the type of the unix group.
270 static bool get_group_map_from_ntname(const char *name, GROUP_MAP *map)
272 int ret;
273 char *expr;
274 struct ldb_result *res=NULL;
276 expr = talloc_asprintf(ldb, "(&(ntName=%s)(objectClass=groupMap))", name);
277 if (expr == NULL) goto failed;
279 ret = ldb_search(ldb, NULL, LDB_SCOPE_SUBTREE, expr, NULL, &res);
280 talloc_steal(expr, res);
281 if (ret != LDB_SUCCESS || res->count != 1) goto failed;
283 if (!msg_to_group_map(res->msgs[0], map)) goto failed;
285 talloc_free(expr);
286 return True;
288 failed:
289 talloc_free(expr);
290 return False;
294 Remove a group mapping entry.
296 static bool group_map_remove(const DOM_SID *sid)
298 struct ldb_dn *dn;
299 int ret;
301 dn = mapping_dn(ldb, sid);
302 if (dn == NULL) {
303 return False;
305 ret = ldb_delete(ldb, dn);
306 talloc_free(dn);
308 return ret == LDB_SUCCESS;
313 Enumerate the group mappings for a domain
315 static bool enum_group_mapping(const DOM_SID *domsid, enum lsa_SidType sid_name_use,
316 GROUP_MAP **pp_rmap,
317 size_t *p_num_entries, bool unix_only)
319 int i, ret;
320 char *expr;
321 fstring name;
322 struct ldb_result *res = NULL;
323 struct ldb_dn *basedn=NULL;
324 TALLOC_CTX *tmp_ctx;
326 tmp_ctx = talloc_new(ldb);
327 if (tmp_ctx == NULL) goto failed;
329 if (sid_name_use == SID_NAME_UNKNOWN) {
330 expr = talloc_asprintf(tmp_ctx, "(&(objectClass=groupMap))");
331 } else {
332 expr = talloc_asprintf(tmp_ctx, "(&(sidNameUse=%u)(objectClass=groupMap))",
333 sid_name_use);
335 if (expr == NULL) goto failed;
337 /* we do a subtree search on the domain */
338 if (domsid != NULL) {
339 sid_to_fstring(name, domsid);
340 basedn = ldb_dn_string_compose(tmp_ctx, NULL, "domain=%s", name);
341 if (basedn == NULL) goto failed;
344 ret = ldb_search(ldb, basedn, LDB_SCOPE_SUBTREE, expr, NULL, &res);
345 talloc_steal(tmp_ctx, res);
346 if (ret != LDB_SUCCESS) goto failed;
348 (*pp_rmap) = NULL;
349 *p_num_entries = 0;
351 for (i=0;i<res->count;i++) {
352 (*pp_rmap) = SMB_REALLOC_ARRAY((*pp_rmap), GROUP_MAP,
353 (*p_num_entries)+1);
354 if (!(*pp_rmap)) goto failed;
356 if (!msg_to_group_map(res->msgs[i], &(*pp_rmap)[*p_num_entries])) {
357 goto failed;
360 (*p_num_entries)++;
363 talloc_free(tmp_ctx);
364 return True;
366 failed:
367 talloc_free(tmp_ctx);
368 return False;
372 This operation happens on session setup, so it should better be fast. We
373 store a list of aliases a SID is member of hanging off MEMBEROF/SID.
375 static NTSTATUS one_alias_membership(const DOM_SID *member,
376 DOM_SID **sids, size_t *num)
378 const char *attrs[] = {
379 "sid",
380 NULL
382 DOM_SID alias;
383 char *expr;
384 int ret, i;
385 struct ldb_result *res=NULL;
386 fstring string_sid;
387 NTSTATUS status = NT_STATUS_INTERNAL_DB_CORRUPTION;
389 if (!sid_to_fstring(string_sid, member)) {
390 return NT_STATUS_INVALID_PARAMETER;
393 expr = talloc_asprintf(ldb, "(&(member=%s)(objectClass=groupMap))",
394 string_sid);
395 if (expr == NULL) goto failed;
397 ret = ldb_search(ldb, NULL, LDB_SCOPE_SUBTREE, expr, attrs, &res);
398 talloc_steal(expr, res);
399 if (ret != LDB_SUCCESS) {
400 goto failed;
403 for (i=0;i<res->count;i++) {
404 struct ldb_message_element *el;
405 el = ldb_msg_find_element(res->msgs[i], "sid");
406 if (el == NULL || el->num_values != 1) {
407 status = NT_STATUS_INTERNAL_DB_CORRUPTION;
408 goto failed;
410 string_to_sid(&alias, (char *)el->values[0].data);
411 status = add_sid_to_array_unique(NULL, &alias, sids, num);
412 if (!NT_STATUS_IS_OK(status)) {
413 goto failed;
417 talloc_free(expr);
418 return NT_STATUS_OK;
420 failed:
421 talloc_free(expr);
422 return status;
426 add/remove a member field
428 static NTSTATUS modify_aliasmem(const DOM_SID *alias, const DOM_SID *member,
429 int operation)
431 fstring string_sid;
432 int ret;
433 struct ldb_message msg;
434 struct ldb_message_element el;
435 struct ldb_val val;
436 TALLOC_CTX *tmp_ctx;
437 GROUP_MAP map;
439 if (!get_group_map_from_sid(*alias, &map)) {
440 sid_to_fstring(string_sid, alias);
441 return NT_STATUS_NO_SUCH_ALIAS;
444 if ((map.sid_name_use != SID_NAME_ALIAS) &&
445 (map.sid_name_use != SID_NAME_WKN_GRP)) {
446 DEBUG(0,("sid_name_use=%d\n", map.sid_name_use));
447 return NT_STATUS_NO_SUCH_ALIAS;
450 tmp_ctx = talloc_new(NULL);
451 if (tmp_ctx == NULL) {
452 return NT_STATUS_NO_MEMORY;
455 msg.dn = mapping_dn(tmp_ctx, alias);
456 if (msg.dn == NULL) {
457 return NT_STATUS_NO_MEMORY;
459 msg.num_elements = 1;
460 msg.elements = &el;
461 el.flags = operation;
462 el.name = talloc_strdup(tmp_ctx, "member");
463 el.num_values = 1;
464 el.values = &val;
465 sid_to_fstring(string_sid, member);
466 val.data = (uint8_t *)string_sid;
467 val.length = strlen(string_sid);
469 ret = ldb_modify(ldb, &msg);
470 talloc_free(tmp_ctx);
472 if (ret == LDB_ERR_NO_SUCH_OBJECT) {
473 return NT_STATUS_NO_SUCH_ALIAS;
476 if (operation == LDB_FLAG_MOD_ADD &&
477 ret == LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS) {
478 return NT_STATUS_MEMBER_IN_ALIAS;
481 return (ret == LDB_SUCCESS ? NT_STATUS_OK : NT_STATUS_ACCESS_DENIED);
484 static NTSTATUS add_aliasmem(const DOM_SID *alias, const DOM_SID *member)
486 return modify_aliasmem(alias, member, LDB_FLAG_MOD_ADD);
489 static NTSTATUS del_aliasmem(const DOM_SID *alias, const DOM_SID *member)
491 return modify_aliasmem(alias, member, LDB_FLAG_MOD_DELETE);
496 enumerate sids that have the given alias set in member
498 static NTSTATUS enum_aliasmem(const DOM_SID *alias, DOM_SID **sids, size_t *num)
500 const char *attrs[] = {
501 "member",
502 NULL
504 int ret, i;
505 NTSTATUS status = NT_STATUS_OK;
506 struct ldb_result *res=NULL;
507 struct ldb_dn *dn;
508 struct ldb_message_element *el;
510 *sids = NULL;
511 *num = 0;
513 dn = mapping_dn(ldb, alias);
514 if (dn == NULL) {
515 return NT_STATUS_NO_MEMORY;
518 ret = ldb_search(ldb, dn, LDB_SCOPE_BASE, NULL, attrs, &res);
519 talloc_steal(dn, res);
520 if (ret == LDB_SUCCESS && res->count == 0) {
521 talloc_free(dn);
522 return NT_STATUS_OK;
524 if (ret != LDB_SUCCESS) {
525 talloc_free(dn);
526 return NT_STATUS_INTERNAL_DB_CORRUPTION;
529 el = ldb_msg_find_element(res->msgs[0], "member");
530 if (el == NULL) {
531 talloc_free(dn);
532 return NT_STATUS_OK;
535 for (i=0;i<el->num_values;i++) {
536 DOM_SID sid;
537 string_to_sid(&sid, (const char *)el->values[i].data);
538 status = add_sid_to_array_unique(NULL, &sid, sids, num);
539 if (!NT_STATUS_IS_OK(status)) {
540 goto done;
544 done:
545 talloc_free(dn);
546 return status;
550 upgrade one group mapping record from the old tdb format
552 static int upgrade_map_record(TDB_CONTEXT *tdb_ctx, TDB_DATA key,
553 TDB_DATA data, void *state)
555 int ret;
556 GROUP_MAP map;
558 if (strncmp((char *)key.dptr, GROUP_PREFIX,
559 MIN(key.dsize, strlen(GROUP_PREFIX))) != 0) {
560 return 0;
563 if (!string_to_sid(&map.sid, strlen(GROUP_PREFIX) + (const char *)key.dptr)) {
564 DEBUG(0,("Bad sid key '%s' during upgrade\n", (const char *)key.dptr));
565 *(int *)state = -1;
566 return -1;
569 ret = tdb_unpack(data.dptr, data.dsize, "ddff",
570 &map.gid, &map.sid_name_use, &map.nt_name, &map.comment);
571 if (ret == -1) {
572 DEBUG(0,("Failed to unpack group map record during upgrade\n"));
573 *(int *)state = -1;
574 return -1;
577 if ((int)map.gid == -1) {
579 * Ignore old invalid mappings
581 return 0;
584 if (!add_mapping_entry(&map, 0)) {
585 DEBUG(0,("Failed to add mapping entry during upgrade\n"));
586 *(int *)state = -1;
587 return -1;
590 return 0;
594 upgrade one alias record from the old tdb format
596 static int upgrade_alias_record(TDB_CONTEXT *tdb_ctx, TDB_DATA key,
597 TDB_DATA data, void *state)
599 const char *p = (const char *)data.dptr;
600 char *string_sid;
601 DOM_SID member;
602 TALLOC_CTX *frame;
604 if (strncmp((char *)key.dptr, MEMBEROF_PREFIX,
605 MIN(key.dsize, strlen(MEMBEROF_PREFIX))) != 0) {
606 return 0;
609 if (!string_to_sid(&member, strlen(MEMBEROF_PREFIX) + (const char *)key.dptr)) {
610 DEBUG(0,("Bad alias key %s during upgrade\n",
611 (const char *)key.dptr));
612 *(int *)state = -1;
615 frame = talloc_stackframe();
616 while (next_token_talloc(frame,&p, &string_sid, " ")) {
617 DOM_SID alias;
618 NTSTATUS status;
619 string_to_sid(&alias, string_sid);
620 status = add_aliasmem(&alias, &member);
621 if (NT_STATUS_EQUAL(status, NT_STATUS_NO_SUCH_ALIAS)) {
622 DEBUG(0,("Ignoring orphaned alias record '%s'\n",
623 string_sid));
624 } else if (!NT_STATUS_IS_OK(status)) {
625 DEBUG(0,("Failed to add alias member during upgrade - %s\n",
626 nt_errstr(status)));
627 *(int *)state = -1;
628 TALLOC_FREE(frame);
629 return -1;
632 TALLOC_FREE(frame);
633 return 0;
637 upgrade from a old style tdb
639 static bool mapping_upgrade(const char *tdb_path)
641 static TDB_CONTEXT *tdb;
642 int ret, status=0;
644 tdb = tdb_open_log(tdb_path, 0, TDB_DEFAULT, O_RDWR, 0600);
645 if (tdb == NULL) goto failed;
647 /* we have to do the map records first, as alias records may
648 reference them */
649 ret = tdb_traverse(tdb, upgrade_map_record, &status);
650 if (ret == -1 || status == -1) goto failed;
652 ret = tdb_traverse(tdb, upgrade_alias_record, &status);
653 if (ret == -1 || status == -1) goto failed;
655 if (tdb) {
656 tdb_close(tdb);
657 tdb = NULL;
661 const char *old_path = tdb_path;
662 char *new_path = state_path("group_mapping.tdb.upgraded");
664 if (!new_path) {
665 goto failed;
667 if (rename(old_path, new_path) != 0) {
668 DEBUG(0,("Failed to rename old group mapping database\n"));
669 goto failed;
672 return True;
674 failed:
675 DEBUG(0,("Failed to upgrade group mapping database\n"));
676 if (tdb) tdb_close(tdb);
677 return False;
682 static const struct mapping_backend ldb_backend = {
683 .add_mapping_entry = add_mapping_entry,
684 .get_group_map_from_sid = get_group_map_from_sid,
685 .get_group_map_from_gid = get_group_map_from_gid,
686 .get_group_map_from_ntname = get_group_map_from_ntname,
687 .group_map_remove = group_map_remove,
688 .enum_group_mapping = enum_group_mapping,
689 .one_alias_membership = one_alias_membership,
690 .add_aliasmem = add_aliasmem,
691 .del_aliasmem = del_aliasmem,
692 .enum_aliasmem = enum_aliasmem
696 initialise the ldb mapping backend
698 const struct mapping_backend *groupdb_ldb_init(void)
700 if (!init_group_mapping()) {
701 DEBUG(0,("Failed to initialise ldb mapping backend\n"));
702 return NULL;
705 return &ldb_backend;