pidl:Samba4/Python: add an optional 'allow_remaining' argument to __ndr_unpack__...
[Samba.git] / source3 / groupdb / mapping_tdb.c
blob1dea9e482e5826ce26bf51940abd547ef8d00b08
1 /*
2 * Unix SMB/CIFS implementation.
3 * RPC Pipe client / server routines
4 * Copyright (C) Andrew Tridgell 1992-2006,
5 * Copyright (C) Jean François Micouleau 1998-2001.
6 * Copyright (C) Volker Lendecke 2006.
7 * Copyright (C) Gerald Carter 2006.
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 3 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, see <http://www.gnu.org/licenses/>.
23 #include "includes.h"
24 #include "system/filesys.h"
25 #include "passdb.h"
26 #include "groupdb/mapping.h"
27 #include "dbwrap/dbwrap.h"
28 #include "dbwrap/dbwrap_open.h"
29 #include "util_tdb.h"
30 #include "../libcli/security/security.h"
31 #include "groupdb/mapping_tdb.h"
33 static struct db_context *db; /* used for driver files */
35 static bool enum_group_mapping(const struct dom_sid *domsid,
36 enum lsa_SidType sid_name_use,
37 GROUP_MAP ***pp_rmap,
38 size_t *p_num_entries,
39 bool unix_only);
40 static bool group_map_remove(const struct dom_sid *sid);
42 static bool mapping_switch(const char *ldb_path);
44 /****************************************************************************
45 Open the group mapping tdb.
46 ****************************************************************************/
47 static bool init_group_mapping(void)
49 const char *ldb_path;
51 if (db != NULL) {
52 return true;
55 db = db_open(NULL, state_path("group_mapping.tdb"), 0,
56 TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
57 if (db == NULL) {
58 DEBUG(0, ("Failed to open group mapping database: %s\n",
59 strerror(errno)));
60 return false;
63 ldb_path = state_path("group_mapping.ldb");
64 if (file_exist(ldb_path) && !mapping_switch(ldb_path)) {
65 unlink(state_path("group_mapping.tdb"));
66 return false;
68 } else {
69 /* handle upgrade from old versions of the database */
70 #if 0 /* -- Needs conversion to dbwrap -- */
71 const char *vstring = "INFO/version";
72 int32 vers_id;
73 GROUP_MAP *map_table = NULL;
74 size_t num_entries = 0;
76 /* handle a Samba upgrade */
77 tdb_lock_bystring(tdb, vstring);
79 /* Cope with byte-reversed older versions of the db. */
80 vers_id = tdb_fetch_int32(tdb, vstring);
81 if ((vers_id == DATABASE_VERSION_V1)
82 || (IREV(vers_id) == DATABASE_VERSION_V1)) {
84 * Written on a bigendian machine with old fetch_int
85 * code. Save as le.
87 tdb_store_int32(tdb, vstring, DATABASE_VERSION_V2);
88 vers_id = DATABASE_VERSION_V2;
91 /* if its an unknown version we remove everthing in the db */
93 if (vers_id != DATABASE_VERSION_V2) {
94 tdb_wipe_all(tdb);
95 tdb_store_int32(tdb, vstring, DATABASE_VERSION_V2);
98 tdb_unlock_bystring(tdb, vstring);
100 /* cleanup any map entries with a gid == -1 */
102 if ( enum_group_mapping( NULL, SID_NAME_UNKNOWN, &map_table,
103 &num_entries, False ) ) {
104 int i;
106 for ( i=0; i<num_entries; i++ ) {
107 if ( map_table[i].gid == -1 ) {
108 group_map_remove( &map_table[i].sid );
112 SAFE_FREE( map_table );
114 #endif
116 return true;
119 static char *group_mapping_key(TALLOC_CTX *mem_ctx, const struct dom_sid *sid)
121 char *sidstr, *result;
123 sidstr = sid_string_talloc(talloc_tos(), sid);
124 if (sidstr == NULL) {
125 return NULL;
128 result = talloc_asprintf(mem_ctx, "%s%s", GROUP_PREFIX, sidstr);
130 TALLOC_FREE(sidstr);
131 return result;
134 /****************************************************************************
135 ****************************************************************************/
136 static bool add_mapping_entry(GROUP_MAP *map, int flag)
138 char *key, *buf;
139 int len;
140 NTSTATUS status;
142 key = group_mapping_key(talloc_tos(), &map->sid);
143 if (key == NULL) {
144 return false;
147 len = tdb_pack(NULL, 0, "ddff",
148 map->gid, map->sid_name_use, map->nt_name, map->comment);
150 buf = talloc_array(key, char, len);
151 if (!buf) {
152 TALLOC_FREE(key);
153 return false;
155 len = tdb_pack((uint8 *)buf, len, "ddff", map->gid,
156 map->sid_name_use, map->nt_name, map->comment);
158 status = dbwrap_trans_store(
159 db, string_term_tdb_data(key),
160 make_tdb_data((uint8_t *)buf, len), TDB_REPLACE);
162 TALLOC_FREE(key);
164 return NT_STATUS_IS_OK(status);
168 /****************************************************************************
169 Return the sid and the type of the unix group.
170 ****************************************************************************/
172 static bool get_group_map_from_sid(struct dom_sid sid, GROUP_MAP *map)
174 TDB_DATA dbuf;
175 char *key;
176 int ret = 0;
177 NTSTATUS status;
178 fstring nt_name;
179 fstring comment;
181 /* the key is the SID, retrieving is direct */
183 key = group_mapping_key(talloc_tos(), &sid);
184 if (key == NULL) {
185 return false;
188 status = dbwrap_fetch_bystring(db, key, key, &dbuf);
189 if (!NT_STATUS_IS_OK(status)) {
190 TALLOC_FREE(key);
191 return false;
194 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "ddff",
195 &map->gid, &map->sid_name_use,
196 &nt_name, &comment);
198 TALLOC_FREE(key);
200 if ( ret == -1 ) {
201 DEBUG(3,("get_group_map_from_sid: tdb_unpack failure\n"));
202 return false;
205 sid_copy(&map->sid, &sid);
207 map->nt_name = talloc_strdup(map, nt_name);
208 if (!map->nt_name) {
209 return false;
211 map->comment = talloc_strdup(map, comment);
212 if (!map->comment) {
213 return false;
216 return true;
219 static bool dbrec2map(const struct db_record *rec, GROUP_MAP *map)
221 TDB_DATA key = dbwrap_record_get_key(rec);
222 TDB_DATA value = dbwrap_record_get_value(rec);
223 int ret = 0;
224 fstring nt_name;
225 fstring comment;
227 if ((key.dsize < strlen(GROUP_PREFIX))
228 || (strncmp((char *)key.dptr, GROUP_PREFIX,
229 GROUP_PREFIX_LEN) != 0)) {
230 return False;
233 if (!string_to_sid(&map->sid, (const char *)key.dptr
234 + GROUP_PREFIX_LEN)) {
235 return False;
238 ret = tdb_unpack(value.dptr, value.dsize, "ddff",
239 &map->gid, &map->sid_name_use,
240 &nt_name, &comment);
242 if (ret == -1) {
243 DEBUG(3, ("dbrec2map: tdb_unpack failure\n"));
244 return false;
247 map->nt_name = talloc_strdup(map, nt_name);
248 if (!map->nt_name) {
249 return false;
251 map->comment = talloc_strdup(map, comment);
252 if (!map->comment) {
253 return false;
256 return true;
259 struct find_map_state {
260 bool found;
261 const char *name; /* If != NULL, look for name */
262 gid_t gid; /* valid iff name == NULL */
263 GROUP_MAP *map;
266 static int find_map(struct db_record *rec, void *private_data)
268 struct find_map_state *state = (struct find_map_state *)private_data;
270 if (!dbrec2map(rec, state->map)) {
271 DEBUG(10, ("failed to unpack map\n"));
272 return 0;
275 if (state->name != NULL) {
276 if (strequal(state->name, state->map->nt_name)) {
277 state->found = true;
278 return 1;
281 else {
282 if (state->map->gid == state->gid) {
283 state->found = true;
284 return 1;
288 return 0;
291 /****************************************************************************
292 Return the sid and the type of the unix group.
293 ****************************************************************************/
295 static bool get_group_map_from_gid(gid_t gid, GROUP_MAP *map)
297 struct find_map_state state;
299 state.found = false;
300 state.name = NULL; /* Indicate we're looking for gid */
301 state.gid = gid;
302 state.map = map;
304 dbwrap_traverse_read(db, find_map, (void *)&state, NULL);
306 return state.found;
309 /****************************************************************************
310 Return the sid and the type of the unix group.
311 ****************************************************************************/
313 static bool get_group_map_from_ntname(const char *name, GROUP_MAP *map)
315 struct find_map_state state;
317 state.found = false;
318 state.name = name;
319 state.map = map;
321 dbwrap_traverse_read(db, find_map, (void *)&state, NULL);
323 return state.found;
326 /****************************************************************************
327 Remove a group mapping entry.
328 ****************************************************************************/
330 static bool group_map_remove(const struct dom_sid *sid)
332 char *key;
333 NTSTATUS status;
335 key = group_mapping_key(talloc_tos(), sid);
336 if (key == NULL) {
337 return false;
340 status = dbwrap_trans_delete(db, string_term_tdb_data(key));
342 TALLOC_FREE(key);
343 return NT_STATUS_IS_OK(status);
346 /****************************************************************************
347 Enumerate the group mapping.
348 ****************************************************************************/
350 struct enum_map_state {
351 const struct dom_sid *domsid;
352 enum lsa_SidType sid_name_use;
353 bool unix_only;
355 size_t num_maps;
356 GROUP_MAP **maps;
359 static int collect_map(struct db_record *rec, void *private_data)
361 struct enum_map_state *state = (struct enum_map_state *)private_data;
362 GROUP_MAP *map;
363 GROUP_MAP **tmp;
365 map = talloc_zero(NULL, GROUP_MAP);
366 if (!map) {
367 DEBUG(0, ("Unable to allocate group map!\n"));
368 return 1;
371 if (!dbrec2map(rec, map)) {
372 TALLOC_FREE(map);
373 return 0;
375 /* list only the type or everything if UNKNOWN */
376 if (state->sid_name_use != SID_NAME_UNKNOWN
377 && state->sid_name_use != map->sid_name_use) {
378 DEBUG(11,("enum_group_mapping: group %s is not of the "
379 "requested type\n", map->nt_name));
380 TALLOC_FREE(map);
381 return 0;
384 if ((state->unix_only == ENUM_ONLY_MAPPED) && (map->gid == -1)) {
385 DEBUG(11,("enum_group_mapping: group %s is non mapped\n",
386 map->nt_name));
387 TALLOC_FREE(map);
388 return 0;
391 if ((state->domsid != NULL) &&
392 (dom_sid_compare_domain(state->domsid, &map->sid) != 0)) {
393 DEBUG(11,("enum_group_mapping: group %s is not in domain\n",
394 sid_string_dbg(&map->sid)));
395 TALLOC_FREE(map);
396 return 0;
399 tmp = talloc_realloc(NULL, state->maps, GROUP_MAP *,
400 state->num_maps + 1);
401 if (!tmp) {
402 DEBUG(0,("enum_group_mapping: Unable to enlarge group "
403 "map!\n"));
404 TALLOC_FREE(map);
405 return 1;
408 state->maps = tmp;
409 state->maps[state->num_maps] = talloc_move(state->maps, &map);
410 state->num_maps++;
411 return 0;
414 static bool enum_group_mapping(const struct dom_sid *domsid,
415 enum lsa_SidType sid_name_use,
416 GROUP_MAP ***pp_rmap,
417 size_t *p_num_entries, bool unix_only)
419 struct enum_map_state state;
420 NTSTATUS status;
422 state.domsid = domsid;
423 state.sid_name_use = sid_name_use;
424 state.unix_only = unix_only;
425 state.num_maps = 0;
426 state.maps = NULL;
428 status = dbwrap_traverse_read(db, collect_map, (void *)&state, NULL);
429 if (!NT_STATUS_IS_OK(status)) {
430 TALLOC_FREE(state.maps);
431 return false;
434 *pp_rmap = state.maps;
435 *p_num_entries = state.num_maps;
437 return true;
440 /* This operation happens on session setup, so it should better be fast. We
441 * store a list of aliases a SID is member of hanging off MEMBEROF/SID. */
443 static NTSTATUS one_alias_membership(const struct dom_sid *member,
444 struct dom_sid **sids, size_t *num)
446 fstring tmp;
447 fstring key;
448 char *string_sid;
449 TDB_DATA dbuf;
450 const char *p;
451 NTSTATUS status = NT_STATUS_OK;
452 TALLOC_CTX *frame = talloc_stackframe();
454 slprintf(key, sizeof(key), "%s%s", MEMBEROF_PREFIX,
455 sid_to_fstring(tmp, member));
457 status = dbwrap_fetch_bystring(db, frame, key, &dbuf);
458 if (!NT_STATUS_IS_OK(status)) {
459 TALLOC_FREE(frame);
460 return NT_STATUS_OK;
463 p = (const char *)dbuf.dptr;
465 while (next_token_talloc(frame, &p, &string_sid, " ")) {
466 struct dom_sid alias;
467 uint32_t num_sids;
469 if (!string_to_sid(&alias, string_sid))
470 continue;
472 num_sids = *num;
473 status= add_sid_to_array_unique(NULL, &alias, sids, &num_sids);
474 if (!NT_STATUS_IS_OK(status)) {
475 goto done;
477 *num = num_sids;
480 done:
481 TALLOC_FREE(frame);
482 return status;
485 static NTSTATUS alias_memberships(const struct dom_sid *members, size_t num_members,
486 struct dom_sid **sids, size_t *num)
488 size_t i;
490 *num = 0;
491 *sids = NULL;
493 for (i=0; i<num_members; i++) {
494 NTSTATUS status = one_alias_membership(&members[i], sids, num);
495 if (!NT_STATUS_IS_OK(status))
496 return status;
498 return NT_STATUS_OK;
501 static bool is_aliasmem(const struct dom_sid *alias, const struct dom_sid *member)
503 struct dom_sid *sids;
504 size_t i;
505 size_t num;
507 /* This feels the wrong way round, but the on-disk data structure
508 * dictates it this way. */
509 if (!NT_STATUS_IS_OK(alias_memberships(member, 1, &sids, &num)))
510 return False;
512 for (i=0; i<num; i++) {
513 if (dom_sid_compare(alias, &sids[i]) == 0) {
514 TALLOC_FREE(sids);
515 return True;
518 TALLOC_FREE(sids);
519 return False;
523 static NTSTATUS add_aliasmem(const struct dom_sid *alias, const struct dom_sid *member)
525 GROUP_MAP *map;
526 char *key;
527 fstring string_sid;
528 char *new_memberstring;
529 struct db_record *rec;
530 NTSTATUS status;
531 TDB_DATA value;
533 map = talloc_zero(talloc_tos(), GROUP_MAP);
534 if (!map) {
535 return NT_STATUS_NO_MEMORY;
538 if (!get_group_map_from_sid(*alias, map)) {
539 TALLOC_FREE(map);
540 return NT_STATUS_NO_SUCH_ALIAS;
543 if ((map->sid_name_use != SID_NAME_ALIAS) &&
544 (map->sid_name_use != SID_NAME_WKN_GRP)) {
545 TALLOC_FREE(map);
546 return NT_STATUS_NO_SUCH_ALIAS;
549 TALLOC_FREE(map);
551 if (is_aliasmem(alias, member))
552 return NT_STATUS_MEMBER_IN_ALIAS;
554 sid_to_fstring(string_sid, member);
556 key = talloc_asprintf(talloc_tos(), "%s%s", MEMBEROF_PREFIX,
557 string_sid);
558 if (key == NULL) {
559 return NT_STATUS_NO_MEMORY;
562 if (dbwrap_transaction_start(db) != 0) {
563 DEBUG(0, ("transaction_start failed\n"));
564 return NT_STATUS_INTERNAL_DB_CORRUPTION;
567 rec = dbwrap_fetch_locked(db, key, string_term_tdb_data(key));
569 if (rec == NULL) {
570 DEBUG(10, ("fetch_lock failed\n"));
571 TALLOC_FREE(key);
572 status = NT_STATUS_INTERNAL_DB_CORRUPTION;
573 goto cancel;
576 value = dbwrap_record_get_value(rec);
578 sid_to_fstring(string_sid, alias);
580 if (value.dptr != NULL) {
581 new_memberstring = talloc_asprintf(
582 key, "%s %s", (char *)(value.dptr), string_sid);
583 } else {
584 new_memberstring = talloc_strdup(key, string_sid);
587 if (new_memberstring == NULL) {
588 TALLOC_FREE(key);
589 status = NT_STATUS_NO_MEMORY;
590 goto cancel;
593 status = dbwrap_record_store(rec, string_term_tdb_data(new_memberstring), 0);
595 TALLOC_FREE(key);
597 if (!NT_STATUS_IS_OK(status)) {
598 DEBUG(10, ("Could not store record: %s\n", nt_errstr(status)));
599 goto cancel;
602 if (dbwrap_transaction_commit(db) != 0) {
603 DEBUG(0, ("transaction_commit failed\n"));
604 status = NT_STATUS_INTERNAL_DB_CORRUPTION;
605 return status;
608 return NT_STATUS_OK;
610 cancel:
611 if (dbwrap_transaction_cancel(db) != 0) {
612 smb_panic("transaction_cancel failed");
615 return status;
618 struct aliasmem_state {
619 TALLOC_CTX *mem_ctx;
620 const struct dom_sid *alias;
621 struct dom_sid **sids;
622 size_t *num;
625 static int collect_aliasmem(struct db_record *rec, void *priv)
627 struct aliasmem_state *state = (struct aliasmem_state *)priv;
628 const char *p;
629 char *alias_string;
630 TALLOC_CTX *frame;
631 TDB_DATA key = dbwrap_record_get_key(rec);
632 TDB_DATA value = dbwrap_record_get_value(rec);
634 if (strncmp((const char *)key.dptr, MEMBEROF_PREFIX,
635 MEMBEROF_PREFIX_LEN) != 0)
636 return 0;
638 p = (const char *)value.dptr;
640 frame = talloc_stackframe();
642 while (next_token_talloc(frame, &p, &alias_string, " ")) {
643 struct dom_sid alias, member;
644 const char *member_string;
645 uint32_t num_sids;
647 if (!string_to_sid(&alias, alias_string))
648 continue;
650 if (dom_sid_compare(state->alias, &alias) != 0)
651 continue;
653 /* Ok, we found the alias we're looking for in the membership
654 * list currently scanned. The key represents the alias
655 * member. Add that. */
657 member_string = strchr((const char *)key.dptr, '/');
659 /* Above we tested for MEMBEROF_PREFIX which includes the
660 * slash. */
662 SMB_ASSERT(member_string != NULL);
663 member_string += 1;
665 if (!string_to_sid(&member, member_string))
666 continue;
668 num_sids = *state->num;
669 if (!NT_STATUS_IS_OK(add_sid_to_array(state->mem_ctx, &member,
670 state->sids,
671 &num_sids)))
673 /* talloc fail. */
674 break;
676 *state->num = num_sids;
679 TALLOC_FREE(frame);
680 return 0;
683 static NTSTATUS enum_aliasmem(const struct dom_sid *alias, TALLOC_CTX *mem_ctx,
684 struct dom_sid **sids, size_t *num)
686 GROUP_MAP *map;
687 struct aliasmem_state state;
689 map = talloc_zero(talloc_tos(), GROUP_MAP);
690 if (!map) {
691 return NT_STATUS_NO_MEMORY;
694 if (!get_group_map_from_sid(*alias, map)) {
695 TALLOC_FREE(map);
696 return NT_STATUS_NO_SUCH_ALIAS;
699 if ((map->sid_name_use != SID_NAME_ALIAS) &&
700 (map->sid_name_use != SID_NAME_WKN_GRP)) {
701 TALLOC_FREE(map);
702 return NT_STATUS_NO_SUCH_ALIAS;
705 TALLOC_FREE(map);
707 *sids = NULL;
708 *num = 0;
710 state.alias = alias;
711 state.sids = sids;
712 state.num = num;
713 state.mem_ctx = mem_ctx;
715 dbwrap_traverse_read(db, collect_aliasmem, &state, NULL);
716 return NT_STATUS_OK;
719 static NTSTATUS del_aliasmem(const struct dom_sid *alias, const struct dom_sid *member)
721 NTSTATUS status;
722 struct dom_sid *sids;
723 size_t i, num;
724 bool found = False;
725 char *member_string;
726 char *key;
727 fstring sid_string;
729 if (dbwrap_transaction_start(db) != 0) {
730 DEBUG(0, ("transaction_start failed\n"));
731 return NT_STATUS_INTERNAL_DB_CORRUPTION;
734 status = alias_memberships(member, 1, &sids, &num);
736 if (!NT_STATUS_IS_OK(status)) {
737 goto cancel;
740 for (i=0; i<num; i++) {
741 if (dom_sid_compare(&sids[i], alias) == 0) {
742 found = True;
743 break;
747 if (!found) {
748 TALLOC_FREE(sids);
749 status = NT_STATUS_MEMBER_NOT_IN_ALIAS;
750 goto cancel;
753 if (i < num)
754 sids[i] = sids[num-1];
756 num -= 1;
758 sid_to_fstring(sid_string, member);
760 key = talloc_asprintf(sids, "%s%s", MEMBEROF_PREFIX, sid_string);
761 if (key == NULL) {
762 TALLOC_FREE(sids);
763 status = NT_STATUS_NO_MEMORY;
764 goto cancel;
767 if (num == 0) {
768 status = dbwrap_delete_bystring(db, key);
769 goto commit;
772 member_string = talloc_strdup(sids, "");
773 if (member_string == NULL) {
774 TALLOC_FREE(sids);
775 status = NT_STATUS_NO_MEMORY;
776 goto cancel;
779 for (i=0; i<num; i++) {
781 sid_to_fstring(sid_string, &sids[i]);
783 member_string = talloc_asprintf_append_buffer(
784 member_string, " %s", sid_string);
786 if (member_string == NULL) {
787 TALLOC_FREE(sids);
788 status = NT_STATUS_NO_MEMORY;
789 goto cancel;
793 status = dbwrap_store_bystring(
794 db, key, string_term_tdb_data(member_string), 0);
795 commit:
796 TALLOC_FREE(sids);
798 if (!NT_STATUS_IS_OK(status)) {
799 DEBUG(10, ("dbwrap_store_bystring failed: %s\n",
800 nt_errstr(status)));
801 goto cancel;
804 if (dbwrap_transaction_commit(db) != 0) {
805 DEBUG(0, ("transaction_commit failed\n"));
806 status = NT_STATUS_INTERNAL_DB_CORRUPTION;
807 return status;
810 return NT_STATUS_OK;
812 cancel:
813 if (dbwrap_transaction_cancel(db) != 0) {
814 smb_panic("transaction_cancel failed");
816 return status;
820 /* -- ldb->tdb switching code -------------------------------------------- */
822 /* change this if the data format ever changes */
823 #define LTDB_PACKING_FORMAT 0x26011967
825 /* old packing formats (not supported for now,
826 * it was never used for group mapping AFAIK) */
827 #define LTDB_PACKING_FORMAT_NODN 0x26011966
829 static unsigned int pull_uint32(uint8_t *p, int ofs)
831 p += ofs;
832 return p[0] | (p[1]<<8) | (p[2]<<16) | (p[3]<<24);
836 unpack a ldb message from a linear buffer in TDB_DATA
838 static int convert_ldb_record(TDB_CONTEXT *ltdb, TDB_DATA key,
839 TDB_DATA data, void *ptr)
841 TALLOC_CTX *tmp_ctx = talloc_tos();
842 GROUP_MAP *map = NULL;
843 uint8_t *p;
844 uint32_t format;
845 uint32_t num_el;
846 unsigned int remaining;
847 unsigned int i, j;
848 size_t len;
849 char *name;
850 char *val;
851 char *q;
852 uint32_t num_mem = 0;
853 struct dom_sid *members = NULL;
855 p = (uint8_t *)data.dptr;
856 if (data.dsize < 8) {
857 errno = EIO;
858 goto failed;
861 format = pull_uint32(p, 0);
862 num_el = pull_uint32(p, 4);
863 p += 8;
865 remaining = data.dsize - 8;
867 switch (format) {
868 case LTDB_PACKING_FORMAT:
869 len = strnlen((char *)p, remaining);
870 if (len == remaining) {
871 errno = EIO;
872 goto failed;
875 if (*p == '@') {
876 /* ignore special LDB attributes */
877 return 0;
880 if (strncmp((char *)p, "rid=", 4)) {
881 /* unknown entry, ignore */
882 DEBUG(3, ("Found unknown entry in group mapping "
883 "database named [%s]\n", (char *)p));
884 return 0;
887 remaining -= len + 1;
888 p += len + 1;
889 break;
891 case LTDB_PACKING_FORMAT_NODN:
892 default:
893 errno = EIO;
894 goto failed;
897 if (num_el == 0) {
898 /* bad entry, ignore */
899 return 0;
902 if (num_el > remaining / 6) {
903 errno = EIO;
904 goto failed;
907 map = talloc_zero(NULL, GROUP_MAP);
908 if (!map) {
909 errno = ENOMEM;
910 goto failed;
913 for (i = 0; i < num_el; i++) {
914 uint32_t num_vals;
916 if (remaining < 10) {
917 errno = EIO;
918 goto failed;
920 len = strnlen((char *)p, remaining - 6);
921 if (len == remaining - 6) {
922 errno = EIO;
923 goto failed;
925 name = talloc_strndup(tmp_ctx, (char *)p, len);
926 if (name == NULL) {
927 errno = ENOMEM;
928 goto failed;
930 remaining -= len + 1;
931 p += len + 1;
933 num_vals = pull_uint32(p, 0);
934 if (strcasecmp_m(name, "member") == 0) {
935 num_mem = num_vals;
936 members = talloc_array(tmp_ctx, struct dom_sid, num_mem);
937 if (members == NULL) {
938 errno = ENOMEM;
939 goto failed;
941 } else if (num_vals != 1) {
942 errno = EIO;
943 goto failed;
946 p += 4;
947 remaining -= 4;
949 for (j = 0; j < num_vals; j++) {
950 len = pull_uint32(p, 0);
951 if (len > remaining-5) {
952 errno = EIO;
953 goto failed;
956 val = talloc_strndup(tmp_ctx, (char *)(p + 4), len);
957 if (val == NULL) {
958 errno = ENOMEM;
959 goto failed;
962 remaining -= len+4+1;
963 p += len+4+1;
965 /* we ignore unknown or uninteresting attributes
966 * (objectclass, etc.) */
967 if (strcasecmp_m(name, "gidNumber") == 0) {
968 map->gid = strtoul(val, &q, 10);
969 if (*q) {
970 errno = EIO;
971 goto failed;
973 } else if (strcasecmp_m(name, "sid") == 0) {
974 if (!string_to_sid(&map->sid, val)) {
975 errno = EIO;
976 goto failed;
978 } else if (strcasecmp_m(name, "sidNameUse") == 0) {
979 map->sid_name_use = strtoul(val, &q, 10);
980 if (*q) {
981 errno = EIO;
982 goto failed;
984 } else if (strcasecmp_m(name, "ntname") == 0) {
985 map->nt_name = talloc_strdup(map, val);
986 if (!map->nt_name) {
987 errno = ENOMEM;
988 goto failed;
990 } else if (strcasecmp_m(name, "comment") == 0) {
991 map->comment = talloc_strdup(map, val);
992 if (!map->comment) {
993 errno = ENOMEM;
994 goto failed;
996 } else if (strcasecmp_m(name, "member") == 0) {
997 if (!string_to_sid(&members[j], val)) {
998 errno = EIO;
999 goto failed;
1003 TALLOC_FREE(val);
1006 TALLOC_FREE(name);
1009 if (map->nt_name == NULL) {
1010 errno = EIO;
1011 goto failed;
1014 if (map->comment == NULL) {
1015 map->comment = talloc_strdup(map, "");
1017 if (map->comment == NULL) {
1018 errno = ENOMEM;
1019 goto failed;
1022 if (!add_mapping_entry(map, 0)) {
1023 errno = EIO;
1024 goto failed;
1027 if (num_mem) {
1028 for (j = 0; j < num_mem; j++) {
1029 NTSTATUS status;
1030 status = add_aliasmem(&map->sid, &members[j]);
1031 if (!NT_STATUS_IS_OK(status)) {
1032 errno = EIO;
1033 goto failed;
1038 if (remaining != 0) {
1039 DEBUG(0, ("Errror: %d bytes unread in ltdb_unpack_data\n",
1040 remaining));
1043 TALLOC_FREE(map);
1044 return 0;
1046 failed:
1047 TALLOC_FREE(map);
1048 return -1;
1051 static bool mapping_switch(const char *ldb_path)
1053 TDB_CONTEXT *ltdb;
1054 TALLOC_CTX *frame;
1055 char *new_path;
1056 int ret;
1058 frame = talloc_stackframe();
1060 ltdb = tdb_open_log(ldb_path, 0, TDB_DEFAULT, O_RDONLY, 0600);
1061 if (ltdb == NULL) goto failed;
1063 /* ldb is just a very fancy tdb, read out raw data and perform
1064 * conversion */
1065 ret = tdb_traverse(ltdb, convert_ldb_record, NULL);
1066 if (ret < 0) goto failed;
1068 if (ltdb) {
1069 tdb_close(ltdb);
1070 ltdb = NULL;
1073 /* now rename the old db out of the way */
1074 new_path = state_path("group_mapping.ldb.replaced");
1075 if (!new_path) {
1076 goto failed;
1078 if (rename(ldb_path, new_path) != 0) {
1079 DEBUG(0,("Failed to rename old group mapping database\n"));
1080 goto failed;
1082 TALLOC_FREE(frame);
1083 return True;
1085 failed:
1086 DEBUG(0, ("Failed to switch to tdb group mapping database\n"));
1087 if (ltdb) tdb_close(ltdb);
1088 TALLOC_FREE(frame);
1089 return False;
1092 static const struct mapping_backend tdb_backend = {
1093 .add_mapping_entry = add_mapping_entry,
1094 .get_group_map_from_sid = get_group_map_from_sid,
1095 .get_group_map_from_gid = get_group_map_from_gid,
1096 .get_group_map_from_ntname = get_group_map_from_ntname,
1097 .group_map_remove = group_map_remove,
1098 .enum_group_mapping = enum_group_mapping,
1099 .one_alias_membership = one_alias_membership,
1100 .add_aliasmem = add_aliasmem,
1101 .del_aliasmem = del_aliasmem,
1102 .enum_aliasmem = enum_aliasmem
1106 initialise the tdb mapping backend
1108 const struct mapping_backend *groupdb_tdb_init(void)
1110 if (!init_group_mapping()) {
1111 DEBUG(0,("Failed to initialise tdb mapping backend\n"));
1112 return NULL;
1115 return &tdb_backend;