2 * Samba Unix/Linux SMB client library
4 * Copyright (C) Gregor Beck 2011
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 * @brief Check the idmap database.
22 * @author Gregor Beck <gb@sernet.de>
26 #include "net_idmap_check.h"
28 #include "system/filesys.h"
29 #include "dbwrap/dbwrap.h"
30 #include "dbwrap/dbwrap_open.h"
31 #include "dbwrap/dbwrap_rbt.h"
33 #include "../libcli/security/dom_sid.h"
39 static int traverse_commit(struct db_record
*diff_rec
, void* data
);
40 static int traverse_check(struct db_record
*rec
, void* data
);
42 /* TDB_DATA *******************************************************************/
43 static char* print_data(TALLOC_CTX
* mem_ctx
, TDB_DATA d
);
44 static TDB_DATA
parse_data(TALLOC_CTX
* mem_ctx
, const char** ptr
);
45 static TDB_DATA
talloc_copy(TALLOC_CTX
* mem_ctx
, TDB_DATA data
);
46 static bool is_empty(TDB_DATA data
) {
47 return (data
.dsize
== 0) || (data
.dptr
== NULL
);
50 /* record *********************************************************************/
54 DT_SID
, DT_UID
, DT_GID
,
55 DT_HWM
, DT_VER
, DT_SEQ
,
59 enum DT key_type
, val_type
;
65 static struct record
* parse_record(TALLOC_CTX
* ctx
, TDB_DATA key
, TDB_DATA val
);
66 static struct record
* reverse_record(struct record
* rec
);
68 static bool is_invalid(const struct record
* r
) {
69 return (r
->key_type
== DT_INV
) || (r
->val_type
== DT_INV
);
72 static bool is_map(const struct record
* r
) {
73 return (r
->key_type
== DT_SID
)
74 || (r
->key_type
== DT_UID
) || (r
->key_type
== DT_GID
);
77 /* action *********************************************************************/
79 typedef struct check_action
{
89 struct check_actions
{
90 check_action invalid_record
;
91 check_action missing_reverse
;
92 check_action invalid_mapping
;
93 check_action invalid_edit
;
94 check_action record_exists
;
95 check_action no_version
;
96 check_action wrong_version
;
97 check_action invalid_hwm
;
99 check_action valid_mapping
;
100 check_action valid_other
;
101 check_action invalid_diff
;
104 static struct check_actions
105 check_actions_init(const struct check_options
* opts
) {
106 struct check_actions ret
= {
107 .invalid_record
= (check_action
) {
108 .name
= "Invalid record",
109 .prompt
= "[e]dit/[d]elete/[D]elete all"
110 "/[s]kip/[S]kip all",
112 .default_action
= 'e',
115 .missing_reverse
= (check_action
) {
116 .name
= "Missing reverse mapping for",
117 .prompt
= "[f]ix/[F]ix all/[e]dit/[d]elete/[D]elete all"
118 "/[s]kip/[S]kip all",
120 .default_action
= 'f',
123 .invalid_mapping
= (check_action
) {
124 .fmt
= "%1$s: %2$s -> %3$s\n(%4$s <- %3$s)\n",
125 .name
= "Invalid mapping",
126 .prompt
= "[e]dit/[d]elete/[D]elete all"
127 "/[s]kip/[S]kip all",
129 .default_action
= 'd',
132 .invalid_edit
= (check_action
) {
133 .name
= "Invalid record",
134 .prompt
= "[e]dit/[d]elete/[D]elete all"
135 "/[s]kip/[S]kip all",
137 .default_action
= 'e',
140 .record_exists
= (check_action
) {
141 .fmt
= "%1$s: %2$s\n-%4$s\n+%3$s\n",
142 .name
= "Record exists",
143 .prompt
= "[o]verwrite/[O]verwrite all/[e]dit"
144 "/[d]elete/[D]elete all/[s]kip/[S]kip all",
146 .default_action
= 'o',
149 .no_version
= (check_action
) {
150 .prompt
= "[f]ix/[s]kip/[a]bort",
152 .default_action
= 'f',
154 .wrong_version
= (check_action
) {
155 .prompt
= "[f]ix/[s]kip/[a]bort",
157 .default_action
= 'a',
159 .invalid_hwm
= (check_action
) {
160 .prompt
= "[f]ix/[s]kip",
162 .default_action
= 'f',
164 .commit
= (check_action
) {
165 .prompt
= "[c]ommit/[l]ist/[s]kip",
167 .default_action
= 'l',
170 .valid_mapping
= (check_action
) {
171 .fmt
= "%1$s: %2$s <-> %3$s\n",
174 .verbose
= opts
->verbose
,
176 .valid_other
= (check_action
) {
179 .verbose
= opts
->verbose
,
181 .invalid_diff
= (check_action
) {
182 .prompt
= "[s]kip/[S]kip all/[c]ommit/[C]ommit all"
185 .default_action
= 's',
190 ret
.invalid_record
.auto_action
= 's';
191 ret
.missing_reverse
.auto_action
= 's';
192 ret
.invalid_mapping
.auto_action
= 's';
193 ret
.no_version
.auto_action
= 's';
194 ret
.wrong_version
.auto_action
= 's';
195 ret
.invalid_hwm
.auto_action
= 's';
196 ret
.commit
.auto_action
= 's';
199 if (opts
->automatic
) {
200 ret
.invalid_record
.auto_action
= 'd'; /* delete */
201 ret
.missing_reverse
.auto_action
= 'f'; /* fix */
202 ret
.invalid_mapping
.auto_action
= 'd'; /* delete */
203 ret
.no_version
.auto_action
= 'f'; /* fix */
204 ret
.wrong_version
.auto_action
= 'a'; /* abort */
205 ret
.invalid_hwm
.auto_action
= 'f'; /* fix */
206 ret
.commit
.auto_action
= 'c'; /* commit */
207 ret
.invalid_diff
.auto_action
= 'a'; /* abort */
209 ret
.wrong_version
.auto_action
= 'f'; /* fix */
210 ret
.invalid_diff
.auto_action
= 'c'; /* commit */
214 ret
.invalid_diff
.auto_action
= 'c'; /* commit */
215 /* ret.commit.auto_action = 'c';*/ /* commit */
221 static char get_action(struct check_action
* a
, struct record
* r
, TDB_DATA
* v
) {
223 if (a
->verbose
&& (r
!= NULL
)) {
225 d_printf("%s: %s ", a
->name
, print_data(r
, r
->key
));
227 d_printf("-> %s\n", print_data(r
, r
->val
));
228 } else if (r
->key_type
== DT_HWM
||
229 r
->key_type
== DT_VER
||
230 r
->key_type
== DT_SEQ
)
232 d_printf(": %ld\n", r
->id
);
237 d_printf(a
->fmt
, a
->name
,
238 print_data(r
, r
->key
),
239 print_data(r
, r
->val
),
240 (v
? print_data(r
, *v
) : ""));
244 if (a
->auto_action
!= '\0') {
245 return a
->auto_action
;
248 ret
= interact_prompt(a
->prompt
, a
->answers
, a
->default_action
);
252 a
->auto_action
= ret
;
254 a
->default_action
= ret
;
258 /* *************************************************************************/
264 static TDB_DATA
pack_diff(TDB_DATA_diff
* diff
) {
266 .dptr
= (uint8_t *)diff
,
267 .dsize
= sizeof(TDB_DATA_diff
),
271 static TDB_DATA_diff
unpack_diff(TDB_DATA data
) {
272 assert(data
.dsize
== sizeof(TDB_DATA_diff
));
273 return *(TDB_DATA_diff
*)data
.dptr
;
276 #define DEBUG_DIFF(LEV,MEM,MSG,KEY,OLD,NEW) \
277 DEBUG(LEV, ("%s: %s\n", MSG, print_data(MEM, KEY))); \
278 if (!is_empty(OLD)) { \
279 DEBUGADD(LEV, ("-%s\n", print_data(MEM, OLD))); \
281 if (!is_empty(NEW)) { \
282 DEBUGADD(LEV, ("+%s\n", print_data(MEM, NEW))); \
289 struct db_context
*db
;
290 struct db_context
*diff
;
291 struct check_actions action
;
296 unsigned n_invalid_record
;
297 unsigned n_missing_reverse
;
298 unsigned n_invalid_mappping
;
302 struct check_options opts
;
306 static void adjust_hwm(struct check_ctx
* ctx
, const struct record
* r
);
308 static int add_record(struct check_ctx
* ctx
, TDB_DATA key
, TDB_DATA value
)
312 TALLOC_CTX
* mem
= talloc_new(ctx
->diff
);
314 struct db_record
*rec
= dbwrap_fetch_locked(ctx
->diff
, mem
, key
);
320 recvalue
= dbwrap_record_get_value(rec
);
322 if (recvalue
.dptr
== 0) { /* first entry */
323 status
= dbwrap_fetch(ctx
->db
, ctx
->diff
, key
, &diff
.oval
);
324 if (!NT_STATUS_IS_OK(status
)) {
325 diff
.oval
= tdb_null
;
328 diff
= unpack_diff(recvalue
);
329 talloc_free(diff
.nval
.dptr
);
331 diff
.nval
= talloc_copy(ctx
->diff
, value
);
333 DEBUG_DIFF(2, mem
, "TDB DIFF", key
, diff
.oval
, diff
.nval
);
335 status
= dbwrap_record_store(rec
, pack_diff(&diff
), 0);
339 if (!NT_STATUS_IS_OK(status
)) {
340 DEBUG(0, ("could not store record %s\n", nt_errstr(status
)));
347 static int del_record(struct check_ctx
* ctx
, TDB_DATA key
) {
348 return add_record(ctx
, key
, tdb_null
);
352 fetch_record(struct check_ctx
* ctx
, TALLOC_CTX
* mem_ctx
, TDB_DATA key
)
357 status
= dbwrap_fetch(ctx
->diff
, mem_ctx
, key
, &tmp
);
359 if (NT_STATUS_IS_OK(status
)) {
360 TDB_DATA_diff diff
= unpack_diff(tmp
);
361 TDB_DATA ret
= talloc_copy(mem_ctx
, diff
.nval
);
362 talloc_free(tmp
.dptr
);
366 status
= dbwrap_fetch(ctx
->db
, mem_ctx
, key
, &tmp
);
367 if (!NT_STATUS_IS_OK(status
)) {
374 static void edit_record(struct record
* r
) {
375 TALLOC_CTX
* mem
= talloc_new(r
);
376 cbuf
* ost
= cbuf_new(mem
);
381 cbuf_printf(ost
, "%s %s\n",
382 print_data(mem
, r
->key
), print_data(mem
, r
->val
));
383 str
= interact_edit(mem
, cbuf_gets(ost
, 0));
384 key
= parse_data(mem
, &str
);
385 val
= parse_data(mem
, &str
);
386 nr
= parse_record(talloc_parent(r
), key
, val
);
393 static bool check_version(struct check_ctx
* ctx
) {
394 static const char* key
= "IDMAP_VERSION";
398 struct check_actions
* act
= &ctx
->action
;
400 status
= dbwrap_fetch_uint32(ctx
->db
, key
, &version
);
401 if (!NT_STATUS_IS_OK(status
)) {
402 d_printf("No version number, assume 2\n");
403 action
= get_action(&act
->no_version
, NULL
, NULL
);
404 } else if (version
!= 2) {
405 d_printf("Wrong version number %d, should be 2\n", version
);
406 action
= get_action(&act
->wrong_version
, NULL
, NULL
);
412 SIVAL(&version
, 0, 2);
413 add_record(ctx
, string_term_tdb_data(key
),
414 make_tdb_data((uint8_t *)&version
, sizeof(uint32_t)));
424 static void check_hwm(struct check_ctx
* ctx
, const char* key
, uint32_t target
) {
428 struct check_actions
* act
= &ctx
->action
;
430 status
= dbwrap_fetch_uint32(ctx
->db
, key
, &hwm
);
431 if (!NT_STATUS_IS_OK(status
)) {
432 d_printf("No %s should be %d\n", key
, target
);
433 action
= get_action(&act
->invalid_hwm
, NULL
, NULL
);
434 } else if (target
< hwm
) {
435 d_printf("Invalid %s %d: should be %d\n", key
, hwm
, target
);
436 action
= get_action(&act
->invalid_hwm
, NULL
, NULL
);
439 SIVAL(&hwm
, 0, target
);
440 add_record(ctx
, string_term_tdb_data(key
),
441 make_tdb_data((uint8_t *)&hwm
, sizeof(uint32_t)));
445 int traverse_check(struct db_record
*rec
, void* data
) {
446 struct check_ctx
* ctx
= (struct check_ctx
*)data
;
447 struct check_actions
* act
= &ctx
->action
;
448 TALLOC_CTX
* mem
= talloc_new(ctx
->diff
);
454 key
= dbwrap_record_get_key(rec
);
455 value
= dbwrap_record_get_value(rec
);
457 r
= parse_record(mem
, key
, value
);
460 action
= get_action(&act
->invalid_record
, r
, NULL
);
461 ctx
->n_invalid_record
++;
462 } else if (is_map(r
)) {
463 TDB_DATA back
= fetch_record(ctx
, mem
, r
->val
);
464 if (back
.dptr
== NULL
) {
465 action
= get_action(&act
->missing_reverse
, r
, NULL
);
466 ctx
->n_missing_reverse
++;
467 } else if (!tdb_data_equal(r
->key
, back
)) {
468 action
= get_action(&act
->invalid_mapping
, r
, &back
);
469 ctx
->n_invalid_mappping
++;
471 if (r
->key_type
== DT_SID
) {
472 action
= get_action(&act
->valid_mapping
, r
, NULL
);
475 action
= get_action(&act
->valid_mapping
, NULL
,
481 action
= get_action(&act
->valid_other
, r
, NULL
);
489 case 'd': /* delete */
490 del_record(ctx
, key
);
492 case 'f': /* add reverse mapping */
493 add_record(ctx
, value
, key
);
499 action
= get_action(&act
->invalid_edit
, r
,NULL
);
502 if (!tdb_data_equal(key
, r
->key
)) {
503 TDB_DATA oval
= fetch_record(ctx
, mem
, r
->key
);
504 if (!is_empty(oval
) &&
505 !tdb_data_equal(oval
, r
->val
))
507 action
= get_action(&act
->record_exists
,
515 TDB_DATA okey
= fetch_record(ctx
, mem
, r
->val
);
516 if (!is_empty(okey
) &&
517 !tdb_data_equal(okey
, r
->key
))
519 action
= get_action(&act
->record_exists
,
525 case 'o': /* overwrite */
527 if (!tdb_data_equal(key
, r
->key
)) {
528 del_record(ctx
, key
);
530 add_record(ctx
, r
->key
, r
->val
);
532 add_record(ctx
, r
->val
, r
->key
);
543 /******************************************************************************/
545 void adjust_hwm(struct check_ctx
* ctx
, const struct record
* r
) {
546 enum DT type
= (r
->key_type
== DT_SID
) ? r
->val_type
: r
->key_type
;
547 if (type
== DT_UID
) {
548 ctx
->uid_hwm
= MAX(ctx
->uid_hwm
, r
->id
);
549 } else if (type
== DT_GID
) {
550 ctx
->gid_hwm
= MAX(ctx
->gid_hwm
, r
->id
);
554 TDB_DATA
talloc_copy(TALLOC_CTX
* mem_ctx
, TDB_DATA data
) {
556 .dptr
= (uint8_t *)talloc_size(mem_ctx
, data
.dsize
+1),
559 if (ret
.dptr
== NULL
) {
562 memcpy(ret
.dptr
, data
.dptr
, data
.dsize
);
563 ret
.dptr
[ret
.dsize
] = '\0';
568 static bool is_cstr(TDB_DATA str
) {
569 return !is_empty(str
) && str
.dptr
[str
.dsize
-1] == '\0';
572 static bool parse_sid (TDB_DATA str
, enum DT
* type
, struct dom_sid
* sid
) {
574 const char* s
= (const char*)str
.dptr
;
575 if ((s
[0] == 'S') && string_to_sid(&tmp
, s
)) {
583 static bool parse_xid(TDB_DATA str
, enum DT
* type
, unsigned long* id
) {
586 if (sscanf((const char*)str
.dptr
, "%cID %lu%c", &c
, &tmp
, &t
) == 2) {
591 } else if (c
== 'G') {
602 parse_record(TALLOC_CTX
* mem_ctx
, TDB_DATA key
, TDB_DATA val
)
604 struct record
* ret
= talloc_zero(mem_ctx
, struct record
);
606 DEBUG(0, ("Out of memory.\n"));
609 ret
->key
= talloc_copy(ret
, key
);
610 ret
->val
= talloc_copy(ret
, val
);
611 if ((ret
->key
.dptr
== NULL
&& key
.dptr
!= NULL
) ||
612 (ret
->val
.dptr
== NULL
&& val
.dptr
!= NULL
))
615 DEBUG(0, ("Out of memory.\n"));
618 assert((ret
->key_type
== DT_INV
) && (ret
->val_type
== DT_INV
));
623 if (parse_sid(key
, &ret
->key_type
, &ret
->sid
)) {
624 parse_xid(val
, &ret
->val_type
, &ret
->id
);
625 } else if (parse_xid(key
, &ret
->key_type
, &ret
->id
)) {
627 parse_sid(val
, &ret
->val_type
, &ret
->sid
);
629 } else if (strcmp((const char*)key
.dptr
, "USER HWM") == 0) {
630 ret
->key_type
= DT_HWM
;
631 if (val
.dsize
== 4) {
632 ret
->id
= IVAL(val
.dptr
,0);
633 ret
->val_type
= DT_UID
;
635 } else if (strcmp((const char*)key
.dptr
, "GROUP HWM") == 0) {
636 ret
->key_type
= DT_HWM
;
637 if (val
.dsize
== 4) {
638 ret
->id
= IVAL(val
.dptr
,0);
639 ret
->val_type
= DT_GID
;
641 } else if (strcmp((const char*)key
.dptr
, "IDMAP_VERSION") == 0) {
642 ret
->key_type
= DT_VER
;
643 if (val
.dsize
== 4) {
644 ret
->id
= IVAL(val
.dptr
,0);
645 ret
->val_type
= DT_VER
;
647 } else if (strcmp((const char*)key
.dptr
, "__db_sequence_number__") == 0) {
648 ret
->key_type
= DT_SEQ
;
649 if (val
.dsize
== 8) {
650 ret
->id
= *(uint64_t*)val
.dptr
;
651 ret
->val_type
= DT_SEQ
;
658 struct record
* reverse_record(struct record
* in
)
660 return parse_record(talloc_parent(in
), in
->val
, in
->key
);
664 /******************************************************************************/
667 char* print_data(TALLOC_CTX
* mem_ctx
, TDB_DATA d
)
671 cbuf
* ost
= cbuf_new(mem_ctx
);
672 int len
= cbuf_print_quoted(ost
, (const char*)d
.dptr
, d
.dsize
);
674 cbuf_swapptr(ost
, &ret
, 0);
675 talloc_steal(mem_ctx
, ret
);
680 return talloc_strdup(mem_ctx
, "<NULL>");
684 TDB_DATA
parse_data(TALLOC_CTX
* mem_ctx
, const char** ptr
) {
685 cbuf
* ost
= cbuf_new(mem_ctx
);
686 TDB_DATA ret
= tdb_null
;
688 if (srprs_quoted(ptr
, ost
)) {
689 ret
.dsize
= cbuf_getpos(ost
);
690 ret
.dptr
= (uint8_t *)talloc_steal(mem_ctx
, cbuf_gets(ost
,0));
696 static int traverse_print_diff(struct db_record
*rec
, void* data
) {
697 struct check_ctx
* ctx
= (struct check_ctx
*)data
;
701 TALLOC_CTX
* mem
= talloc_new(ctx
->diff
);
703 key
= dbwrap_record_get_key(rec
);
704 value
= dbwrap_record_get_value(rec
);
705 diff
= unpack_diff(value
);
707 DEBUG_DIFF(0, mem
, "DIFF", key
, diff
.oval
, diff
.nval
);
714 static int traverse_commit(struct db_record
*diff_rec
, void* data
) {
715 struct check_ctx
* ctx
= (struct check_ctx
*)data
;
720 TALLOC_CTX
* mem
= talloc_new(ctx
->diff
);
723 struct check_actions
* act
= &ctx
->action
;
724 struct db_record
* rec
;
726 key
= dbwrap_record_get_key(diff_rec
);
727 diff_value
= dbwrap_record_get_value(diff_rec
);
728 diff
= unpack_diff(diff_value
);
730 rec
= dbwrap_fetch_locked(ctx
->db
, mem
, key
);
735 value
= dbwrap_record_get_value(rec
);
737 if (!tdb_data_equal(value
, diff
.oval
)) {
740 d_printf("Warning: record has changed: %s\n"
741 "expected: %s got %s\n", print_data(mem
, key
),
742 print_data(mem
, diff
.oval
),
743 print_data(mem
, value
));
745 action
= get_action(&act
->invalid_diff
, NULL
, NULL
);
749 } else if (action
== 'a') {
754 DEBUG_DIFF(0, mem
, "Commit", key
, diff
.oval
, diff
.nval
);
756 if (is_empty(diff
.nval
)) {
757 status
= dbwrap_record_delete(rec
);
759 status
= dbwrap_record_store(rec
, diff
.nval
, 0);
762 if (!NT_STATUS_IS_OK(status
)) {
763 DEBUG(0, ("could not store record %s\n", nt_errstr(status
)));
764 if (!ctx
->opts
.force
) {
774 static struct check_ctx
*
775 check_init(TALLOC_CTX
* mem_ctx
, const struct check_options
* o
)
777 struct check_ctx
* ctx
= talloc_zero(mem_ctx
, struct check_ctx
);
779 DEBUG(0, (_("No memory\n")));
783 ctx
->diff
= db_open_rbt(ctx
);
784 if (ctx
->diff
== NULL
) {
786 DEBUG(0, (_("No memory\n")));
790 ctx
->action
= check_actions_init(o
);
795 static bool check_open_db(struct check_ctx
* ctx
, const char* name
, int oflags
)
798 d_fprintf(stderr
, _("Error: name == NULL in check_open_db().\n"));
802 if (ctx
->db
!= NULL
) {
803 if ((ctx
->oflags
== oflags
) && (strcmp(ctx
->name
, name
))) {
806 TALLOC_FREE(ctx
->db
);
810 ctx
->db
= db_open(ctx
, name
, 0, TDB_DEFAULT
, oflags
, 0);
811 if (ctx
->db
== NULL
) {
813 _("Could not open idmap db (%s) for writing: %s\n"),
814 name
, strerror(errno
));
818 if (ctx
->name
!= name
) {
819 TALLOC_FREE(ctx
->name
);
820 ctx
->name
= talloc_strdup(ctx
, name
);
823 ctx
->oflags
= oflags
;
827 static bool check_do_checks(struct check_ctx
* ctx
)
831 if (!check_version(ctx
)) {
835 status
= dbwrap_traverse(ctx
->db
, traverse_check
, ctx
, NULL
);
837 if (!NT_STATUS_IS_OK(status
)) {
838 DEBUG(0, ("failed to traverse %s\n", ctx
->name
));
842 check_hwm(ctx
, "USER HWM", ctx
->uid_hwm
+ 1);
843 check_hwm(ctx
, "GROUP HWM", ctx
->gid_hwm
+ 1);
848 static void check_summary(const struct check_ctx
* ctx
)
850 d_printf("uid hwm: %d\ngid hwm: %d\n", ctx
->uid_hwm
, ctx
->gid_hwm
);
851 d_printf("mappings: %d\nother: %d\n", ctx
->n_map
, ctx
->n_other
);
852 d_printf("invalid records: %d\nmissing links: %d\ninvalid links: %d\n",
853 ctx
->n_invalid_record
, ctx
->n_missing_reverse
,
854 ctx
->n_invalid_mappping
);
855 d_printf("%u changes:\n", ctx
->n_diff
);
858 static bool check_transaction_start(struct check_ctx
* ctx
) {
859 return (dbwrap_transaction_start(ctx
->db
) == 0);
862 static bool check_transaction_commit(struct check_ctx
* ctx
) {
863 return (dbwrap_transaction_commit(ctx
->db
) == 0);
866 static bool check_transaction_cancel(struct check_ctx
* ctx
) {
867 return (dbwrap_transaction_cancel(ctx
->db
) == 0);
871 static void check_diff_list(struct check_ctx
* ctx
) {
872 NTSTATUS status
= dbwrap_traverse(ctx
->diff
, traverse_print_diff
, ctx
, NULL
);
874 if (!NT_STATUS_IS_OK(status
)) {
875 DEBUG(0, ("failed to traverse diff\n"));
880 static bool check_commit(struct check_ctx
* ctx
)
882 struct check_actions
* act
= &ctx
->action
;
884 NTSTATUS status
= NT_STATUS_OK
;
888 if (ctx
->n_diff
== 0) {
892 while ((action
= get_action(&act
->commit
, NULL
, NULL
)) == 'l') {
893 check_diff_list(ctx
);
898 assert(action
== 'c');
900 if (!check_open_db(ctx
, ctx
->name
, O_RDWR
)) {
904 if (!check_transaction_start(ctx
)) {
908 status
= dbwrap_traverse(ctx
->diff
, traverse_commit
, ctx
, NULL
);
910 if (!NT_STATUS_IS_OK(status
)) {
911 check_transaction_cancel(ctx
);
914 if (ctx
->opts
.test
) { /*get_action? */
915 return check_transaction_cancel(ctx
);
917 return check_transaction_commit(ctx
);
921 int net_idmap_check_db(const char* db
, const struct check_options
* o
)
924 TALLOC_CTX
* mem_ctx
= talloc_stackframe();
925 struct check_ctx
* ctx
= check_init(mem_ctx
, o
);
927 if (!o
->automatic
&& !isatty(STDIN_FILENO
)) {
928 DEBUG(0, ("Interactive use needs tty, use --auto\n"));
932 if (check_open_db(ctx
, db
, O_RDWR
)
933 && check_transaction_start(ctx
))
935 if ( check_do_checks(ctx
)
937 && check_transaction_commit(ctx
))
941 check_transaction_cancel(ctx
);
945 if (check_open_db(ctx
, db
, O_RDONLY
)
946 && check_do_checks(ctx
)
947 && check_commit(ctx
))
953 talloc_free(mem_ctx
);