2 Unix SMB/CIFS implementation.
4 libnet_BecomeDC() tests
6 Copyright (C) Stefan Metzmacher <metze@samba.org> 2006
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "lib/cmdline/popt_common.h"
24 #include "torture/rpc/torture_rpc.h"
25 #include "libnet/libnet.h"
26 #include "dsdb/samdb/samdb.h"
27 #include "../lib/util/dlinklist.h"
28 #include "librpc/gen_ndr/ndr_drsuapi.h"
29 #include "librpc/gen_ndr/ndr_drsblobs.h"
30 #include "system/time.h"
32 #include "auth/auth.h"
33 #include "param/param.h"
34 #include "param/provision.h"
35 #include "libcli/resolve/resolve.h"
37 bool torture_net_become_dc(struct torture_context *torture)
41 struct libnet_BecomeDC b;
42 struct libnet_UnbecomeDC u;
43 struct libnet_vampire_cb_state *s;
44 struct ldb_message *msg;
50 const char *netbios_name;
51 struct cli_credentials *machine_account;
53 struct loadparm_context *lp_ctx;
54 struct ldb_context *ldb;
55 struct libnet_context *ctx;
56 struct dsdb_schema *schema;
58 char *location = NULL;
59 torture_assert_ntstatus_ok(torture, torture_temp_dir(torture, "libnet_BecomeDC", &location),
60 "torture_temp_dir should return NT_STATUS_OK" );
62 netbios_name = lpcfg_parm_string(torture->lp_ctx, NULL, "become dc", "smbtorture dc");
63 if (!netbios_name || !netbios_name[0]) {
64 netbios_name = "smbtorturedc";
67 make_nbt_name_server(&name, torture_setting_string(torture, "host", NULL));
69 /* do an initial name resolution to find its IP */
70 status = resolve_name(lpcfg_resolve_context(torture->lp_ctx),
71 &name, torture, &address, torture->ev);
72 torture_assert_ntstatus_ok(torture, status, talloc_asprintf(torture,
73 "Failed to resolve %s - %s\n",
74 name.name, nt_errstr(status)));
77 /* Join domain as a member server. */
78 tj = torture_join_domain(torture, netbios_name,
81 torture_assert(torture, tj, talloc_asprintf(torture,
82 "%s failed to join domain as workstation\n",
85 s = libnet_vampire_cb_state_init(torture, torture->lp_ctx, torture->ev,
87 torture_join_dom_netbios_name(tj),
88 torture_join_dom_dns_name(tj),
90 torture_assert(torture, s, "libnet_vampire_cb_state_init");
92 ctx = libnet_context_init(torture->ev, torture->lp_ctx);
93 ctx->cred = cmdline_credentials;
96 b.in.domain_dns_name = torture_join_dom_dns_name(tj);
97 b.in.domain_netbios_name = torture_join_dom_netbios_name(tj);
98 b.in.domain_sid = torture_join_sid(tj);
99 b.in.source_dsa_address = address;
100 b.in.dest_dsa_netbios_name = netbios_name;
102 b.in.callbacks.private_data = s;
103 b.in.callbacks.check_options = libnet_vampire_cb_check_options;
104 b.in.callbacks.prepare_db = libnet_vampire_cb_prepare_db;
105 b.in.callbacks.schema_chunk = libnet_vampire_cb_schema_chunk;
106 b.in.callbacks.config_chunk = libnet_vampire_cb_store_chunk;
107 b.in.callbacks.domain_chunk = libnet_vampire_cb_store_chunk;
109 status = libnet_BecomeDC(ctx, s, &b);
110 torture_assert_ntstatus_ok_goto(torture, status, ret, cleanup, talloc_asprintf(torture,
111 "libnet_BecomeDC() failed - %s %s\n",
112 nt_errstr(status), b.out.error_string));
113 ldb = libnet_vampire_cb_ldb(s);
115 msg = ldb_msg_new(s);
116 torture_assert_int_equal_goto(torture, (msg?1:0), 1, ret, cleanup,
117 "ldb_msg_new() failed\n");
118 msg->dn = ldb_dn_new(msg, ldb, "@ROOTDSE");
119 torture_assert_int_equal_goto(torture, (msg->dn?1:0), 1, ret, cleanup,
120 "ldb_msg_new(@ROOTDSE) failed\n");
122 ldb_ret = ldb_msg_add_string(msg, "isSynchronized", "TRUE");
123 torture_assert_int_equal_goto(torture, ldb_ret, LDB_SUCCESS, ret, cleanup,
124 "ldb_msg_add_string(msg, isSynchronized, TRUE) failed\n");
126 for (i=0; i < msg->num_elements; i++) {
127 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
130 torture_comment(torture, "mark ROOTDSE with isSynchronized=TRUE\n");
131 ldb_ret = ldb_modify(libnet_vampire_cb_ldb(s), msg);
132 torture_assert_int_equal_goto(torture, ldb_ret, LDB_SUCCESS, ret, cleanup,
133 "ldb_modify() failed\n");
135 /* commit the transaction now we know the secrets were written
138 ldb_ret = ldb_transaction_commit(ldb);
139 torture_assert_int_equal_goto(torture, ldb_ret, LDB_SUCCESS, ret, cleanup,
140 "ldb_transaction_commit() failed\n");
143 talloc_unlink(s, ldb);
145 lp_ctx = libnet_vampire_cb_lp_ctx(s);
146 sam_ldb_path = talloc_asprintf(s, "%s/%s", location, "private/sam.ldb");
147 torture_comment(torture, "Reopen the SAM LDB with system credentials and all replicated data: %s\n", sam_ldb_path);
148 ldb = ldb_wrap_connect(s, torture->ev, lp_ctx, sam_ldb_path,
149 system_session(lp_ctx),
151 torture_assert_int_equal_goto(torture, (ldb?1:0), 1, ret, cleanup,
152 talloc_asprintf(torture,
153 "Failed to open '%s'\n", sam_ldb_path));
155 schema = dsdb_get_schema(ldb, s);
156 torture_assert_int_equal_goto(torture, (schema?1:0), 1, ret, cleanup,
157 "Failed to get loaded dsdb_schema\n");
159 /* Make sure we get this from the command line */
160 if (lpcfg_parm_bool(torture->lp_ctx, NULL, "become dc", "do not unjoin", false)) {
167 u.in.domain_dns_name = torture_join_dom_dns_name(tj);
168 u.in.domain_netbios_name = torture_join_dom_netbios_name(tj);
169 u.in.source_dsa_address = address;
170 u.in.dest_dsa_netbios_name = netbios_name;
172 status = libnet_UnbecomeDC(ctx, s, &u);
173 torture_assert_ntstatus_ok(torture, status, talloc_asprintf(torture,
174 "libnet_UnbecomeDC() failed - %s %s\n",
175 nt_errstr(status), u.out.error_string));
178 torture_leave_domain(torture, tj);