?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s4-smbtorture: fix smbtorture after GetPrinterData{Ex} after IDL changes.
[Samba/cd1.git] / source4 / torture / rpc / samba3rpc.c
blob9e7b18b4d7b90da6fdcdeaf347909b1c67d60d28
1 /*
2 Unix SMB/CIFS implementation.
3
4 dcerpc torture tests, designed to walk Samba3 code paths
5
6 Copyright (C) Volker Lendecke 2006
7
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.
12
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.
17
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/>.
20 */
21
22 #include "includes.h"
23 #include "libcli/raw/libcliraw.h"
24 #include "libcli/raw/raw_proto.h"
25 #include "libcli/rap/rap.h"
26 #include "torture/util.h"
27 #include "torture/rap/proto.h"
28 #include "librpc/gen_ndr/ndr_lsa_c.h"
29 #include "librpc/gen_ndr/ndr_samr_c.h"
30 #include "librpc/gen_ndr/ndr_netlogon_c.h"
31 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
32 #include "librpc/gen_ndr/ndr_spoolss_c.h"
33 #include "librpc/gen_ndr/ndr_winreg_c.h"
34 #include "librpc/gen_ndr/ndr_wkssvc_c.h"
35 #include "lib/cmdline/popt_common.h"
36 #include "torture/rpc/rpc.h"
37 #include "libcli/libcli.h"
38 #include "libcli/smb_composite/smb_composite.h"
39 #include "libcli/auth/libcli_auth.h"
40 #include "../lib/crypto/crypto.h"
41 #include "libcli/security/security.h"
42 #include "param/param.h"
43 #include "lib/registry/registry.h"
44 #include "libcli/resolve/resolve.h"
45
46 /*
47 * This tests a RPC call using an invalid vuid
48 */
49
50 bool torture_bind_authcontext(struct torture_context *torture)
51 {
52 TALLOC_CTX *mem_ctx;
53 NTSTATUS status;
54 bool ret = false;
55 struct lsa_ObjectAttribute objectattr;
56 struct lsa_OpenPolicy2 openpolicy;
57 struct policy_handle handle;
58 struct lsa_Close close_handle;
59 struct smbcli_session *tmp;
60 struct smbcli_session *session2;
61 struct smbcli_state *cli;
62 struct dcerpc_pipe *lsa_pipe;
63 struct cli_credentials *anon_creds;
64 struct smb_composite_sesssetup setup;
65 struct smbcli_options options;
66 struct smbcli_session_options session_options;
67
68 mem_ctx = talloc_init("torture_bind_authcontext");
69
70 if (mem_ctx == NULL) {
71 d_printf("talloc_init failed\n");
72 return false;
73 }
74
75 lp_smbcli_options(torture->lp_ctx, &options);
76 lp_smbcli_session_options(torture->lp_ctx, &session_options);
77
78 status = smbcli_full_connection(mem_ctx, &cli,
79 torture_setting_string(torture, "host", NULL),
80 lp_smb_ports(torture->lp_ctx),
81 "IPC$", NULL,
82 lp_socket_options(torture->lp_ctx),
83 cmdline_credentials,
84 lp_resolve_context(torture->lp_ctx),
85 torture->ev, &options, &session_options,
86 lp_iconv_convenience(torture->lp_ctx),
87 lp_gensec_settings(torture, torture->lp_ctx));
88 if (!NT_STATUS_IS_OK(status)) {
89 d_printf("smbcli_full_connection failed: %s\n",
90 nt_errstr(status));
91 goto done;
92 }
93
94 lsa_pipe = dcerpc_pipe_init(mem_ctx, cli->transport->socket->event.ctx,
95 lp_iconv_convenience(torture->lp_ctx));
96 if (lsa_pipe == NULL) {
97 d_printf("dcerpc_pipe_init failed\n");
98 goto done;
99 }
100
101 status = dcerpc_pipe_open_smb(lsa_pipe, cli->tree, "\\lsarpc");
102 if (!NT_STATUS_IS_OK(status)) {
103 d_printf("dcerpc_pipe_open_smb failed: %s\n",
104 nt_errstr(status));
105 goto done;
106 }
107
108 status = dcerpc_bind_auth_none(lsa_pipe, &ndr_table_lsarpc);
109 if (!NT_STATUS_IS_OK(status)) {
110 d_printf("dcerpc_bind_auth_none failed: %s\n",
111 nt_errstr(status));
112 goto done;
113 }
114
115 openpolicy.in.system_name =talloc_asprintf(
116 mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
117 ZERO_STRUCT(objectattr);
118 openpolicy.in.attr = &objectattr;
119 openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
120 openpolicy.out.handle = &handle;
121
122 status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
123
124 if (!NT_STATUS_IS_OK(status)) {
125 d_printf("dcerpc_lsa_OpenPolicy2 failed: %s\n",
126 nt_errstr(status));
127 goto done;
128 }
129
130 close_handle.in.handle = &handle;
131 close_handle.out.handle = &handle;
132
133 status = dcerpc_lsa_Close(lsa_pipe, mem_ctx, &close_handle);
134 if (!NT_STATUS_IS_OK(status)) {
135 d_printf("dcerpc_lsa_Close failed: %s\n",
136 nt_errstr(status));
137 goto done;
138 }
139
140 session2 = smbcli_session_init(cli->transport, mem_ctx, false, session_options);
141 if (session2 == NULL) {
142 d_printf("smbcli_session_init failed\n");
143 goto done;
144 }
145
146 if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
147 d_printf("create_anon_creds failed\n");
148 goto done;
149 }
150
151 setup.in.sesskey = cli->transport->negotiate.sesskey;
152 setup.in.capabilities = cli->transport->negotiate.capabilities;
153 setup.in.workgroup = "";
154 setup.in.credentials = anon_creds;
155 setup.in.gensec_settings = lp_gensec_settings(torture, torture->lp_ctx);
156
157 status = smb_composite_sesssetup(session2, &setup);
158 if (!NT_STATUS_IS_OK(status)) {
159 d_printf("anon session setup failed: %s\n",
160 nt_errstr(status));
161 goto done;
162 }
163 session2->vuid = setup.out.vuid;
164
165 tmp = cli->tree->session;
166 cli->tree->session = session2;
167
168 status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
169
170 cli->tree->session = tmp;
171 talloc_free(lsa_pipe);
172 lsa_pipe = NULL;
173
174 if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
175 d_printf("dcerpc_lsa_OpenPolicy2 with wrong vuid gave %s, "
176 "expected NT_STATUS_INVALID_HANDLE\n",
177 nt_errstr(status));
178 goto done;
179 }
180
181 ret = true;
182 done:
183 talloc_free(mem_ctx);
184 return ret;
185 }
186
187 /*
188 * Bind to lsa using a specific auth method
189 */
190
191 static bool bindtest(struct smbcli_state *cli,
192 struct cli_credentials *credentials,
193 struct loadparm_context *lp_ctx,
194 uint8_t auth_type, uint8_t auth_level)
195 {
196 TALLOC_CTX *mem_ctx;
197 bool ret = false;
198 NTSTATUS status;
199
200 struct dcerpc_pipe *lsa_pipe;
201 struct lsa_ObjectAttribute objectattr;
202 struct lsa_OpenPolicy2 openpolicy;
203 struct lsa_QueryInfoPolicy query;
204 union lsa_PolicyInformation *info = NULL;
205 struct policy_handle handle;
206 struct lsa_Close close_handle;
207
208 if ((mem_ctx = talloc_init("bindtest")) == NULL) {
209 d_printf("talloc_init failed\n");
210 return false;
211 }
212
213 lsa_pipe = dcerpc_pipe_init(mem_ctx,
214 cli->transport->socket->event.ctx,
215 lp_iconv_convenience(lp_ctx));
216 if (lsa_pipe == NULL) {
217 d_printf("dcerpc_pipe_init failed\n");
218 goto done;
219 }
220
221 status = dcerpc_pipe_open_smb(lsa_pipe, cli->tree, "\\lsarpc");
222 if (!NT_STATUS_IS_OK(status)) {
223 d_printf("dcerpc_pipe_open_smb failed: %s\n",
224 nt_errstr(status));
225 goto done;
226 }
227
228 status = dcerpc_bind_auth(lsa_pipe, &ndr_table_lsarpc,
229 credentials, lp_gensec_settings(lp_ctx, lp_ctx), auth_type, auth_level,
230 NULL);
231 if (!NT_STATUS_IS_OK(status)) {
232 d_printf("dcerpc_bind_auth failed: %s\n", nt_errstr(status));
233 goto done;
234 }
235
236 openpolicy.in.system_name =talloc_asprintf(
237 mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
238 ZERO_STRUCT(objectattr);
239 openpolicy.in.attr = &objectattr;
240 openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
241 openpolicy.out.handle = &handle;
242
243 status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
244
245 if (!NT_STATUS_IS_OK(status)) {
246 d_printf("dcerpc_lsa_OpenPolicy2 failed: %s\n",
247 nt_errstr(status));
248 goto done;
249 }
250
251 query.in.handle = &handle;
252 query.in.level = LSA_POLICY_INFO_DOMAIN;
253 query.out.info = &info;
254
255 status = dcerpc_lsa_QueryInfoPolicy(lsa_pipe, mem_ctx, &query);
256 if (!NT_STATUS_IS_OK(status)) {
257 d_printf("dcerpc_lsa_QueryInfoPolicy failed: %s\n",
258 nt_errstr(status));
259 goto done;
260 }
261
262 close_handle.in.handle = &handle;
263 close_handle.out.handle = &handle;
264
265 status = dcerpc_lsa_Close(lsa_pipe, mem_ctx, &close_handle);
266 if (!NT_STATUS_IS_OK(status)) {
267 d_printf("dcerpc_lsa_Close failed: %s\n",
268 nt_errstr(status));
269 goto done;
270 }
271
272 ret = true;
273 done:
274 talloc_free(mem_ctx);
275 return ret;
276 }
277
278 /*
279 * test authenticated RPC binds with the variants Samba3 does support
280 */
281
282 bool torture_bind_samba3(struct torture_context *torture)
283 {
284 TALLOC_CTX *mem_ctx;
285 NTSTATUS status;
286 bool ret = false;
287 struct smbcli_state *cli;
288 struct smbcli_options options;
289 struct smbcli_session_options session_options;
290
291 mem_ctx = talloc_init("torture_bind_authcontext");
292
293 if (mem_ctx == NULL) {
294 d_printf("talloc_init failed\n");
295 return false;
296 }
297
298 lp_smbcli_options(torture->lp_ctx, &options);
299 lp_smbcli_session_options(torture->lp_ctx, &session_options);
300
301 status = smbcli_full_connection(mem_ctx, &cli,
302 torture_setting_string(torture, "host", NULL),
303 lp_smb_ports(torture->lp_ctx),
304 "IPC$", NULL,
305 lp_socket_options(torture->lp_ctx),
306 cmdline_credentials,
307 lp_resolve_context(torture->lp_ctx),
308 torture->ev, &options, &session_options,
309 lp_iconv_convenience(torture->lp_ctx),
310 lp_gensec_settings(torture, torture->lp_ctx));
311 if (!NT_STATUS_IS_OK(status)) {
312 d_printf("smbcli_full_connection failed: %s\n",
313 nt_errstr(status));
314 goto done;
315 }
316
317 ret = true;
318
319 ret &= bindtest(cli, cmdline_credentials, torture->lp_ctx, DCERPC_AUTH_TYPE_NTLMSSP,
320 DCERPC_AUTH_LEVEL_INTEGRITY);
321 ret &= bindtest(cli, cmdline_credentials, torture->lp_ctx, DCERPC_AUTH_TYPE_NTLMSSP,
322 DCERPC_AUTH_LEVEL_PRIVACY);
323 ret &= bindtest(cli, cmdline_credentials, torture->lp_ctx, DCERPC_AUTH_TYPE_SPNEGO,
324 DCERPC_AUTH_LEVEL_INTEGRITY);
325 ret &= bindtest(cli, cmdline_credentials, torture->lp_ctx, DCERPC_AUTH_TYPE_SPNEGO,
326 DCERPC_AUTH_LEVEL_PRIVACY);
327
328 done:
329 talloc_free(mem_ctx);
330 return ret;
331 }
332
333 /*
334 * Lookup or create a user and return all necessary info
335 */
336
337 static NTSTATUS get_usr_handle(struct smbcli_state *cli,
338 TALLOC_CTX *mem_ctx,
339 struct loadparm_context *lp_ctx,
340 struct cli_credentials *admin_creds,
341 uint8_t auth_type,
342 uint8_t auth_level,
343 const char *username,
344 char **domain,
345 struct dcerpc_pipe **result_pipe,
346 struct policy_handle **result_handle,
347 struct dom_sid **sid_p)
348 {
349 struct dcerpc_pipe *samr_pipe;
350 NTSTATUS status;
351 struct policy_handle conn_handle;
352 struct policy_handle domain_handle;
353 struct policy_handle *user_handle;
354 struct samr_Connect2 conn;
355 struct samr_EnumDomains enumdom;
356 uint32_t resume_handle = 0;
357 uint32_t num_entries = 0;
358 struct samr_SamArray *sam = NULL;
359 struct samr_LookupDomain l;
360 struct dom_sid2 *sid = NULL;
361 int dom_idx;
362 struct lsa_String domain_name;
363 struct lsa_String user_name;
364 struct samr_OpenDomain o;
365 struct samr_CreateUser2 c;
366 uint32_t user_rid,access_granted;
367
368 samr_pipe = dcerpc_pipe_init(mem_ctx,
369 cli->transport->socket->event.ctx,
370 lp_iconv_convenience(lp_ctx));
371 if (samr_pipe == NULL) {
372 d_printf("dcerpc_pipe_init failed\n");
373 status = NT_STATUS_NO_MEMORY;
374 goto fail;
375 }
376
377 status = dcerpc_pipe_open_smb(samr_pipe, cli->tree, "\\samr");
378 if (!NT_STATUS_IS_OK(status)) {
379 d_printf("dcerpc_pipe_open_smb failed: %s\n",
380 nt_errstr(status));
381 goto fail;
382 }
383
384 if (admin_creds != NULL) {
385 status = dcerpc_bind_auth(samr_pipe, &ndr_table_samr,
386 admin_creds, lp_gensec_settings(lp_ctx, lp_ctx), auth_type, auth_level,
387 NULL);
388 if (!NT_STATUS_IS_OK(status)) {
389 d_printf("dcerpc_bind_auth failed: %s\n",
390 nt_errstr(status));
391 goto fail;
392 }
393 } else {
394 /* We must have an authenticated SMB connection */
395 status = dcerpc_bind_auth_none(samr_pipe, &ndr_table_samr);
396 if (!NT_STATUS_IS_OK(status)) {
397 d_printf("dcerpc_bind_auth_none failed: %s\n",
398 nt_errstr(status));
399 goto fail;
400 }
401 }
402
403 conn.in.system_name = talloc_asprintf(
404 mem_ctx, "\\\\%s", dcerpc_server_name(samr_pipe));
405 conn.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
406 conn.out.connect_handle = &conn_handle;
407
408 status = dcerpc_samr_Connect2(samr_pipe, mem_ctx, &conn);
409 if (!NT_STATUS_IS_OK(status)) {
410 d_printf("samr_Connect2 failed: %s\n", nt_errstr(status));
411 goto fail;
412 }
413
414 enumdom.in.connect_handle = &conn_handle;
415 enumdom.in.resume_handle = &resume_handle;
416 enumdom.in.buf_size = (uint32_t)-1;
417 enumdom.out.resume_handle = &resume_handle;
418 enumdom.out.num_entries = &num_entries;
419 enumdom.out.sam = &sam;
420
421 status = dcerpc_samr_EnumDomains(samr_pipe, mem_ctx, &enumdom);
422 if (!NT_STATUS_IS_OK(status)) {
423 d_printf("samr_EnumDomains failed: %s\n", nt_errstr(status));
424 goto fail;
425 }
426
427 if (*enumdom.out.num_entries != 2) {
428 d_printf("samr_EnumDomains returned %d entries, expected 2\n",
429 *enumdom.out.num_entries);
430 status = NT_STATUS_UNSUCCESSFUL;
431 goto fail;
432 }
433
434 dom_idx = strequal(sam->entries[0].name.string,
435 "builtin") ? 1:0;
436
437 l.in.connect_handle = &conn_handle;
438 domain_name.string = sam->entries[dom_idx].name.string;
439 *domain = talloc_strdup(mem_ctx, domain_name.string);
440 l.in.domain_name = &domain_name;
441 l.out.sid = &sid;
442
443 status = dcerpc_samr_LookupDomain(samr_pipe, mem_ctx, &l);
444 if (!NT_STATUS_IS_OK(status)) {
445 d_printf("samr_LookupDomain failed: %s\n", nt_errstr(status));
446 goto fail;
447 }
448
449 o.in.connect_handle = &conn_handle;
450 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
451 o.in.sid = *l.out.sid;
452 o.out.domain_handle = &domain_handle;
453
454 status = dcerpc_samr_OpenDomain(samr_pipe, mem_ctx, &o);
455 if (!NT_STATUS_IS_OK(status)) {
456 d_printf("samr_OpenDomain failed: %s\n", nt_errstr(status));
457 goto fail;
458 }
459
460 c.in.domain_handle = &domain_handle;
461 user_name.string = username;
462 c.in.account_name = &user_name;
463 c.in.acct_flags = ACB_NORMAL;
464 c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
465 user_handle = talloc(mem_ctx, struct policy_handle);
466 c.out.user_handle = user_handle;
467 c.out.access_granted = &access_granted;
468 c.out.rid = &user_rid;
469
470 status = dcerpc_samr_CreateUser2(samr_pipe, mem_ctx, &c);
471
472 if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
473 struct samr_LookupNames ln;
474 struct samr_OpenUser ou;
475 struct samr_Ids rids, types;
476
477 ln.in.domain_handle = &domain_handle;
478 ln.in.num_names = 1;
479 ln.in.names = &user_name;
480 ln.out.rids = &rids;
481 ln.out.types = &types;
482
483 status = dcerpc_samr_LookupNames(samr_pipe, mem_ctx, &ln);
484 if (!NT_STATUS_IS_OK(status)) {
485 d_printf("samr_LookupNames failed: %s\n",
486 nt_errstr(status));
487 goto fail;
488 }
489
490 ou.in.domain_handle = &domain_handle;
491 ou.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
492 user_rid = ou.in.rid = ln.out.rids->ids[0];
493 ou.out.user_handle = user_handle;
494
495 status = dcerpc_samr_OpenUser(samr_pipe, mem_ctx, &ou);
496 if (!NT_STATUS_IS_OK(status)) {
497 d_printf("samr_OpenUser failed: %s\n",
498 nt_errstr(status));
499 goto fail;
500 }
501 }
502
503 if (!NT_STATUS_IS_OK(status)) {
504 d_printf("samr_CreateUser failed: %s\n", nt_errstr(status));
505 goto fail;
506 }
507
508 *result_pipe = samr_pipe;
509 *result_handle = user_handle;
510 if (sid_p != NULL) {
511 *sid_p = dom_sid_add_rid(mem_ctx, *l.out.sid, user_rid);
512 }
513 return NT_STATUS_OK;
514
515 fail:
516 return status;
517 }
518
519 /*
520 * Create a test user
521 */
522
523 static bool create_user(TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
524 struct loadparm_context *lp_ctx,
525 struct cli_credentials *admin_creds,
526 const char *username, const char *password,
527 char **domain_name,
528 struct dom_sid **user_sid)
529 {
530 TALLOC_CTX *tmp_ctx;
531 NTSTATUS status;
532 struct dcerpc_pipe *samr_pipe;
533 struct policy_handle *wks_handle;
534 bool ret = false;
535
536 if (!(tmp_ctx = talloc_new(mem_ctx))) {
537 d_printf("talloc_init failed\n");
538 return false;
539 }
540
541 status = get_usr_handle(cli, tmp_ctx, lp_ctx, admin_creds,
542 DCERPC_AUTH_TYPE_NTLMSSP,
543 DCERPC_AUTH_LEVEL_INTEGRITY,
544 username, domain_name, &samr_pipe, &wks_handle,
545 user_sid);
546 if (!NT_STATUS_IS_OK(status)) {
547 d_printf("get_usr_handle failed: %s\n", nt_errstr(status));
548 goto done;
549 }
550
551 {
552 struct samr_SetUserInfo2 sui2;
553 struct samr_SetUserInfo sui;
554 struct samr_QueryUserInfo qui;
555 union samr_UserInfo u_info;
556 union samr_UserInfo *info;
557 DATA_BLOB session_key;
558
559
560 ZERO_STRUCT(u_info);
561 encode_pw_buffer(u_info.info23.password.data, password,
562 STR_UNICODE);
563
564 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
565 if (!NT_STATUS_IS_OK(status)) {
566 d_printf("dcerpc_fetch_session_key failed\n");
567 goto done;
568 }
569 arcfour_crypt_blob(u_info.info23.password.data, 516,
570 &session_key);
571 u_info.info23.info.password_expired = 0;
572 u_info.info23.info.fields_present = SAMR_FIELD_NT_PASSWORD_PRESENT |
573 SAMR_FIELD_LM_PASSWORD_PRESENT |
574 SAMR_FIELD_EXPIRED_FLAG;
575 sui2.in.user_handle = wks_handle;
576 sui2.in.info = &u_info;
577 sui2.in.level = 23;
578
579 status = dcerpc_samr_SetUserInfo2(samr_pipe, tmp_ctx, &sui2);
580 if (!NT_STATUS_IS_OK(status)) {
581 d_printf("samr_SetUserInfo(23) failed: %s\n",
582 nt_errstr(status));
583 goto done;
584 }
585
586 u_info.info16.acct_flags = ACB_NORMAL;
587 sui.in.user_handle = wks_handle;
588 sui.in.info = &u_info;
589 sui.in.level = 16;
590
591 status = dcerpc_samr_SetUserInfo(samr_pipe, tmp_ctx, &sui);
592 if (!NT_STATUS_IS_OK(status)) {
593 d_printf("samr_SetUserInfo(16) failed\n");
594 goto done;
595 }
596
597 qui.in.user_handle = wks_handle;
598 qui.in.level = 21;
599 qui.out.info = &info;
600
601 status = dcerpc_samr_QueryUserInfo(samr_pipe, tmp_ctx, &qui);
602 if (!NT_STATUS_IS_OK(status)) {
603 d_printf("samr_QueryUserInfo(21) failed\n");
604 goto done;
605 }
606
607 info->info21.allow_password_change = 0;
608 info->info21.force_password_change = 0;
609 info->info21.account_name.string = NULL;
610 info->info21.rid = 0;
611 info->info21.acct_expiry = 0;
612 info->info21.fields_present = 0x81827fa; /* copy usrmgr.exe */
613
614 u_info.info21 = info->info21;
615 sui.in.user_handle = wks_handle;
616 sui.in.info = &u_info;
617 sui.in.level = 21;
618
619 status = dcerpc_samr_SetUserInfo(samr_pipe, tmp_ctx, &sui);
620 if (!NT_STATUS_IS_OK(status)) {
621 d_printf("samr_SetUserInfo(21) failed\n");
622 goto done;
623 }
624 }
625
626 *domain_name= talloc_steal(mem_ctx, *domain_name);
627 *user_sid = talloc_steal(mem_ctx, *user_sid);
628 ret = true;
629 done:
630 talloc_free(tmp_ctx);
631 return ret;
632 }
633
634 /*
635 * Delete a test user
636 */
637
638 static bool delete_user(struct smbcli_state *cli,
639 struct loadparm_context *lp_ctx,
640 struct cli_credentials *admin_creds,
641 const char *username)
642 {
643 TALLOC_CTX *mem_ctx;
644 NTSTATUS status;
645 char *dom_name;
646 struct dcerpc_pipe *samr_pipe;
647 struct policy_handle *user_handle;
648 bool ret = false;
649
650 if ((mem_ctx = talloc_init("leave")) == NULL) {
651 d_printf("talloc_init failed\n");
652 return false;
653 }
654
655 status = get_usr_handle(cli, mem_ctx, lp_ctx, admin_creds,
656 DCERPC_AUTH_TYPE_NTLMSSP,
657 DCERPC_AUTH_LEVEL_INTEGRITY,
658 username, &dom_name, &samr_pipe,
659 &user_handle, NULL);
660
661 if (!NT_STATUS_IS_OK(status)) {
662 d_printf("get_wks_handle failed: %s\n", nt_errstr(status));
663 goto done;
664 }
665
666 {
667 struct samr_DeleteUser d;
668
669 d.in.user_handle = user_handle;
670 d.out.user_handle = user_handle;
671
672 status = dcerpc_samr_DeleteUser(samr_pipe, mem_ctx, &d);
673 if (!NT_STATUS_IS_OK(status)) {
674 d_printf("samr_DeleteUser failed %s\n", nt_errstr(status));
675 goto done;
676 }
677 }
678
679 ret = true;
680
681 done:
682 talloc_free(mem_ctx);
683 return ret;
684 }
685
686 /*
687 * Do a Samba3-style join
688 */
689
690 static bool join3(struct smbcli_state *cli,
691 struct loadparm_context *lp_ctx,
692 bool use_level25,
693 struct cli_credentials *admin_creds,
694 struct cli_credentials *wks_creds)
695 {
696 TALLOC_CTX *mem_ctx;
697 NTSTATUS status;
698 char *dom_name;
699 struct dcerpc_pipe *samr_pipe;
700 struct policy_handle *wks_handle;
701 bool ret = false;
702 NTTIME last_password_change;
703
704 if ((mem_ctx = talloc_init("join3")) == NULL) {
705 d_printf("talloc_init failed\n");
706 return false;
707 }
708
709 status = get_usr_handle(
710 cli, mem_ctx, lp_ctx, admin_creds,
711 DCERPC_AUTH_TYPE_NTLMSSP,
712 DCERPC_AUTH_LEVEL_PRIVACY,
713 talloc_asprintf(mem_ctx, "%s$",
714 cli_credentials_get_workstation(wks_creds)),
715 &dom_name, &samr_pipe, &wks_handle, NULL);
716
717 if (!NT_STATUS_IS_OK(status)) {
718 d_printf("get_wks_handle failed: %s\n", nt_errstr(status));
719 goto done;
720 }
721
722 {
723 struct samr_QueryUserInfo q;
724 union samr_UserInfo *info;
725
726 q.in.user_handle = wks_handle;
727 q.in.level = 21;
728 q.out.info = &info;
729
730 status = dcerpc_samr_QueryUserInfo(samr_pipe, mem_ctx, &q);
731 if (!NT_STATUS_IS_OK(status)) {
732 d_printf("(%s) QueryUserInfo failed: %s\n",
733 __location__, nt_errstr(status));
734 goto done;
735 }
736
737 last_password_change = info->info21.last_password_change;
738 }
739
740 cli_credentials_set_domain(wks_creds, dom_name, CRED_SPECIFIED);
741
742 if (use_level25) {
743 struct samr_SetUserInfo2 sui2;
744 union samr_UserInfo u_info;
745 struct samr_UserInfo21 *i21 = &u_info.info25.info;
746 DATA_BLOB session_key;
747 DATA_BLOB confounded_session_key = data_blob_talloc(
748 mem_ctx, NULL, 16);
749 struct MD5Context ctx;
750 uint8_t confounder[16];
751
752 ZERO_STRUCT(u_info);
753
754 i21->full_name.string = talloc_asprintf(
755 mem_ctx, "%s$",
756 cli_credentials_get_workstation(wks_creds));
757 i21->acct_flags = ACB_WSTRUST;
758 i21->fields_present = SAMR_FIELD_FULL_NAME |
759 SAMR_FIELD_ACCT_FLAGS | SAMR_FIELD_NT_PASSWORD_PRESENT;
760 /* this would break the test result expectations
761 i21->fields_present |= SAMR_FIELD_EXPIRED_FLAG;
762 i21->password_expired = 1;
763 */
764
765 encode_pw_buffer(u_info.info25.password.data,
766 cli_credentials_get_password(wks_creds),
767 STR_UNICODE);
768 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
769 if (!NT_STATUS_IS_OK(status)) {
770 d_printf("dcerpc_fetch_session_key failed: %s\n",
771 nt_errstr(status));
772 goto done;
773 }
774 generate_random_buffer((uint8_t *)confounder, 16);
775
776 MD5Init(&ctx);
777 MD5Update(&ctx, confounder, 16);
778 MD5Update(&ctx, session_key.data, session_key.length);
779 MD5Final(confounded_session_key.data, &ctx);
780
781 arcfour_crypt_blob(u_info.info25.password.data, 516,
782 &confounded_session_key);
783 memcpy(&u_info.info25.password.data[516], confounder, 16);
784
785 sui2.in.user_handle = wks_handle;
786 sui2.in.level = 25;
787 sui2.in.info = &u_info;
788
789 status = dcerpc_samr_SetUserInfo2(samr_pipe, mem_ctx, &sui2);
790 if (!NT_STATUS_IS_OK(status)) {
791 d_printf("samr_SetUserInfo2(25) failed: %s\n",
792 nt_errstr(status));
793 goto done;
794 }
795 } else {
796 struct samr_SetUserInfo2 sui2;
797 struct samr_SetUserInfo sui;
798 union samr_UserInfo u_info;
799 DATA_BLOB session_key;
800
801 encode_pw_buffer(u_info.info24.password.data,
802 cli_credentials_get_password(wks_creds),
803 STR_UNICODE);
804 /* just to make this test pass */
805 u_info.info24.password_expired = 1;
806
807 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
808 if (!NT_STATUS_IS_OK(status)) {
809 d_printf("dcerpc_fetch_session_key failed\n");
810 goto done;
811 }
812 arcfour_crypt_blob(u_info.info24.password.data, 516,
813 &session_key);
814 sui2.in.user_handle = wks_handle;
815 sui2.in.info = &u_info;
816 sui2.in.level = 24;
817
818 status = dcerpc_samr_SetUserInfo2(samr_pipe, mem_ctx, &sui2);
819 if (!NT_STATUS_IS_OK(status)) {
820 d_printf("samr_SetUserInfo(24) failed: %s\n",
821 nt_errstr(status));
822 goto done;
823 }
824
825 u_info.info16.acct_flags = ACB_WSTRUST;
826 sui.in.user_handle = wks_handle;
827 sui.in.info = &u_info;
828 sui.in.level = 16;
829
830 status = dcerpc_samr_SetUserInfo(samr_pipe, mem_ctx, &sui);
831 if (!NT_STATUS_IS_OK(status)) {
832 d_printf("samr_SetUserInfo(16) failed\n");
833 goto done;
834 }
835 }
836
837 {
838 struct samr_QueryUserInfo q;
839 union samr_UserInfo *info;
840
841 q.in.user_handle = wks_handle;
842 q.in.level = 21;
843 q.out.info = &info;
844
845 status = dcerpc_samr_QueryUserInfo(samr_pipe, mem_ctx, &q);
846 if (!NT_STATUS_IS_OK(status)) {
847 d_printf("(%s) QueryUserInfo failed: %s\n",
848 __location__, nt_errstr(status));
849 goto done;
850 }
851
852 if (use_level25) {
853 if (last_password_change
854 == info->info21.last_password_change) {
855 d_printf("(%s) last_password_change unchanged "
856 "during join, level25 must change "
857 "it\n", __location__);
858 goto done;
859 }
860 }
861 else {
862 if (last_password_change
863 != info->info21.last_password_change) {
864 d_printf("(%s) last_password_change changed "
865 "during join, level24 doesn't "
866 "change it\n", __location__);
867 goto done;
868 }
869 }
870 }
871
872 ret = true;
873
874 done:
875 talloc_free(mem_ctx);
876 return ret;
877 }
878
879 /*
880 * Do a ReqChallenge/Auth2 and get the wks creds
881 */
882
883 static bool auth2(struct smbcli_state *cli,
884 struct loadparm_context *lp_ctx,
885 struct cli_credentials *wks_cred)
886 {
887 TALLOC_CTX *mem_ctx;
888 struct dcerpc_pipe *net_pipe;
889 bool result = false;
890 NTSTATUS status;
891 struct netr_ServerReqChallenge r;
892 struct netr_Credential netr_cli_creds;
893 struct netr_Credential netr_srv_creds;
894 uint32_t negotiate_flags;
895 struct netr_ServerAuthenticate2 a;
896 struct netlogon_creds_CredentialState *creds_state;
897 struct netr_Credential netr_cred;
898 struct samr_Password mach_pw;
899
900 mem_ctx = talloc_new(NULL);
901 if (mem_ctx == NULL) {
902 d_printf("talloc_new failed\n");
903 return false;
904 }
905
906 net_pipe = dcerpc_pipe_init(mem_ctx,
907 cli->transport->socket->event.ctx,
908 lp_iconv_convenience(lp_ctx));
909 if (net_pipe == NULL) {
910 d_printf("dcerpc_pipe_init failed\n");
911 goto done;
912 }
913
914 status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
915 if (!NT_STATUS_IS_OK(status)) {
916 d_printf("dcerpc_pipe_open_smb failed: %s\n",
917 nt_errstr(status));
918 goto done;
919 }
920
921 status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
922 if (!NT_STATUS_IS_OK(status)) {
923 d_printf("dcerpc_bind_auth_none failed: %s\n",
924 nt_errstr(status));
925 goto done;
926 }
927
928 r.in.computer_name = cli_credentials_get_workstation(wks_cred);
929 r.in.server_name = talloc_asprintf(
930 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
931 if (r.in.server_name == NULL) {
932 d_printf("talloc_asprintf failed\n");
933 goto done;
934 }
935 generate_random_buffer(netr_cli_creds.data,
936 sizeof(netr_cli_creds.data));
937 r.in.credentials = &netr_cli_creds;
938 r.out.return_credentials = &netr_srv_creds;
939
940 status = dcerpc_netr_ServerReqChallenge(net_pipe, mem_ctx, &r);
941 if (!NT_STATUS_IS_OK(status)) {
942 d_printf("netr_ServerReqChallenge failed: %s\n",
943 nt_errstr(status));
944 goto done;
945 }
946
947 negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
948 E_md4hash(cli_credentials_get_password(wks_cred), mach_pw.hash);
949
950 a.in.server_name = talloc_asprintf(
951 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
952 a.in.account_name = talloc_asprintf(
953 mem_ctx, "%s$", cli_credentials_get_workstation(wks_cred));
954 a.in.computer_name = cli_credentials_get_workstation(wks_cred);
955 a.in.secure_channel_type = SEC_CHAN_WKSTA;
956 a.in.negotiate_flags = &negotiate_flags;
957 a.out.negotiate_flags = &negotiate_flags;
958 a.in.credentials = &netr_cred;
959 a.out.return_credentials = &netr_cred;
960
961 creds_state = netlogon_creds_client_init(mem_ctx,
962 a.in.account_name,
963 a.in.computer_name,
964 r.in.credentials,
965 r.out.return_credentials, &mach_pw,
966 &netr_cred, negotiate_flags);
967
968 status = dcerpc_netr_ServerAuthenticate2(net_pipe, mem_ctx, &a);
969 if (!NT_STATUS_IS_OK(status)) {
970 d_printf("netr_ServerServerAuthenticate2 failed: %s\n",
971 nt_errstr(status));
972 goto done;
973 }
974
975 if (!netlogon_creds_client_check(creds_state, a.out.return_credentials)) {
976 d_printf("creds_client_check failed\n");
977 goto done;
978 }
979
980 cli_credentials_set_netlogon_creds(wks_cred, creds_state);
981
982 result = true;
983
984 done:
985 talloc_free(mem_ctx);
986 return result;
987 }
988
989 /*
990 * Do a couple of schannel protected Netlogon ops: Interactive and Network
991 * login, and change the wks password
992 */
993
994 static bool schan(struct smbcli_state *cli,
995 struct loadparm_context *lp_ctx,
996 struct cli_credentials *wks_creds,
997 struct cli_credentials *user_creds)
998 {
999 TALLOC_CTX *mem_ctx;
1000 NTSTATUS status;
1001 bool ret = false;
1002 struct dcerpc_pipe *net_pipe;
1003 int i;
1004
1005 mem_ctx = talloc_new(NULL);
1006 if (mem_ctx == NULL) {
1007 d_printf("talloc_new failed\n");
1008 return false;
1009 }
1010
1011 net_pipe = dcerpc_pipe_init(mem_ctx,
1012 cli->transport->socket->event.ctx,
1013 lp_iconv_convenience(lp_ctx));
1014 if (net_pipe == NULL) {
1015 d_printf("dcerpc_pipe_init failed\n");
1016 goto done;
1017 }
1018
1019 status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
1020 if (!NT_STATUS_IS_OK(status)) {
1021 d_printf("dcerpc_pipe_open_smb failed: %s\n",
1022 nt_errstr(status));
1023 goto done;
1024 }
1025
1026 #if 0
1027 net_pipe->conn->flags |= DCERPC_DEBUG_PRINT_IN |
1028 DCERPC_DEBUG_PRINT_OUT;
1029 #endif
1030 #if 1
1031 net_pipe->conn->flags |= (DCERPC_SIGN | DCERPC_SEAL);
1032 status = dcerpc_bind_auth(net_pipe, &ndr_table_netlogon,
1033 wks_creds, lp_gensec_settings(lp_ctx, lp_ctx), DCERPC_AUTH_TYPE_SCHANNEL,
1034 DCERPC_AUTH_LEVEL_PRIVACY,
1035 NULL);
1036 #else
1037 status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
1038 #endif
1039 if (!NT_STATUS_IS_OK(status)) {
1040 d_printf("schannel bind failed: %s\n", nt_errstr(status));
1041 goto done;
1042 }
1043
1044
1045 for (i=2; i<4; i++) {
1046 int flags;
1047 DATA_BLOB chal, nt_resp, lm_resp, names_blob, session_key;
1048 struct netlogon_creds_CredentialState *creds_state;
1049 struct netr_Authenticator netr_auth, netr_auth2;
1050 struct netr_NetworkInfo ninfo;
1051 struct netr_PasswordInfo pinfo;
1052 struct netr_LogonSamLogon r;
1053 union netr_LogonLevel logon;
1054 union netr_Validation validation;
1055 uint8_t authoritative;
1056 struct netr_Authenticator return_authenticator;
1057
1058 flags = CLI_CRED_LANMAN_AUTH | CLI_CRED_NTLM_AUTH |
1059 CLI_CRED_NTLMv2_AUTH;
1060
1061 chal = data_blob_talloc(mem_ctx, NULL, 8);
1062 if (chal.data == NULL) {
1063 d_printf("data_blob_talloc failed\n");
1064 goto done;
1065 }
1066
1067 generate_random_buffer(chal.data, chal.length);
1068 names_blob = NTLMv2_generate_names_blob(
1069 mem_ctx,
1070 cli_credentials_get_workstation(user_creds),
1071 cli_credentials_get_domain(user_creds));
1072 status = cli_credentials_get_ntlm_response(
1073 user_creds, mem_ctx, &flags, chal, names_blob,
1074 &lm_resp, &nt_resp, NULL, NULL);
1075 if (!NT_STATUS_IS_OK(status)) {
1076 d_printf("cli_credentials_get_ntlm_response failed:"
1077 " %s\n", nt_errstr(status));
1078 goto done;
1079 }
1080
1081 creds_state = cli_credentials_get_netlogon_creds(wks_creds);
1082 netlogon_creds_client_authenticator(creds_state, &netr_auth);
1083
1084 ninfo.identity_info.account_name.string =
1085 cli_credentials_get_username(user_creds);
1086 ninfo.identity_info.domain_name.string =
1087 cli_credentials_get_domain(user_creds);
1088 ninfo.identity_info.parameter_control = 0;
1089 ninfo.identity_info.logon_id_low = 0;
1090 ninfo.identity_info.logon_id_high = 0;
1091 ninfo.identity_info.workstation.string =
1092 cli_credentials_get_workstation(user_creds);
1093 memcpy(ninfo.challenge, chal.data, sizeof(ninfo.challenge));
1094 ninfo.nt.length = nt_resp.length;
1095 ninfo.nt.data = nt_resp.data;
1096 ninfo.lm.length = lm_resp.length;
1097 ninfo.lm.data = lm_resp.data;
1098
1099 logon.network = &ninfo;
1100
1101 r.in.server_name = talloc_asprintf(
1102 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
1103 ZERO_STRUCT(netr_auth2);
1104 r.in.computer_name =
1105 cli_credentials_get_workstation(wks_creds);
1106 r.in.credential = &netr_auth;
1107 r.in.return_authenticator = &netr_auth2;
1108 r.in.logon_level = 2;
1109 r.in.validation_level = i;
1110 r.in.logon = &logon;
1111 r.out.validation = &validation;
1112 r.out.authoritative = &authoritative;
1113 r.out.return_authenticator = &return_authenticator;
1114
1115 status = dcerpc_netr_LogonSamLogon(net_pipe, mem_ctx, &r);
1116 if (!NT_STATUS_IS_OK(status)) {
1117 d_printf("netr_LogonSamLogon failed: %s\n",
1118 nt_errstr(status));
1119 goto done;
1120 }
1121
1122 if ((r.out.return_authenticator == NULL) ||
1123 (!netlogon_creds_client_check(creds_state,
1124 &r.out.return_authenticator->cred))) {
1125 d_printf("Credentials check failed!\n");
1126 goto done;
1127 }
1128
1129 netlogon_creds_client_authenticator(creds_state, &netr_auth);
1130
1131 pinfo.identity_info = ninfo.identity_info;
1132 ZERO_STRUCT(pinfo.lmpassword.hash);
1133 E_md4hash(cli_credentials_get_password(user_creds),
1134 pinfo.ntpassword.hash);
1135 session_key = data_blob_talloc(mem_ctx,
1136 creds_state->session_key, 16);
1137 arcfour_crypt_blob(pinfo.ntpassword.hash,
1138 sizeof(pinfo.ntpassword.hash),
1139 &session_key);
1140
1141 logon.password = &pinfo;
1142
1143 r.in.logon_level = 1;
1144 r.in.logon = &logon;
1145 r.out.return_authenticator = &return_authenticator;
1146
1147 status = dcerpc_netr_LogonSamLogon(net_pipe, mem_ctx, &r);
1148 if (!NT_STATUS_IS_OK(status)) {
1149 d_printf("netr_LogonSamLogon failed: %s\n",
1150 nt_errstr(status));
1151 goto done;
1152 }
1153
1154 if ((r.out.return_authenticator == NULL) ||
1155 (!netlogon_creds_client_check(creds_state,
1156 &r.out.return_authenticator->cred))) {
1157 d_printf("Credentials check failed!\n");
1158 goto done;
1159 }
1160 }
1161
1162 {
1163 struct netr_ServerPasswordSet s;
1164 char *password = generate_random_password(wks_creds, 8, 255);
1165 struct netlogon_creds_CredentialState *creds_state;
1166 struct netr_Authenticator credential, return_authenticator;
1167 struct samr_Password new_password;
1168
1169 s.in.server_name = talloc_asprintf(
1170 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
1171 s.in.computer_name = cli_credentials_get_workstation(wks_creds);
1172 s.in.account_name = talloc_asprintf(
1173 mem_ctx, "%s$", s.in.computer_name);
1174 s.in.secure_channel_type = SEC_CHAN_WKSTA;
1175 s.in.credential = &credential;
1176 s.in.new_password = &new_password;
1177 s.out.return_authenticator = &return_authenticator;
1178
1179 E_md4hash(password, new_password.hash);
1180
1181 creds_state = cli_credentials_get_netlogon_creds(wks_creds);
1182 netlogon_creds_des_encrypt(creds_state, &new_password);
1183 netlogon_creds_client_authenticator(creds_state, &credential);
1184
1185 status = dcerpc_netr_ServerPasswordSet(net_pipe, mem_ctx, &s);
1186 if (!NT_STATUS_IS_OK(status)) {
1187 printf("ServerPasswordSet - %s\n", nt_errstr(status));
1188 goto done;
1189 }
1190
1191 if (!netlogon_creds_client_check(creds_state,
1192 &s.out.return_authenticator->cred)) {
1193 printf("Credential chaining failed\n");
1194 }
1195
1196 cli_credentials_set_password(wks_creds, password,
1197 CRED_SPECIFIED);
1198 }
1199
1200 ret = true;
1201 done:
1202 talloc_free(mem_ctx);
1203 return ret;
1204 }
1205
1206 /*
1207 * Delete the wks account again
1208 */
1209
1210 static bool leave(struct smbcli_state *cli,
1211 struct loadparm_context *lp_ctx,
1212 struct cli_credentials *admin_creds,
1213 struct cli_credentials *wks_creds)
1214 {
1215 char *wks_name = talloc_asprintf(
1216 NULL, "%s$", cli_credentials_get_workstation(wks_creds));
1217 bool ret;
1218
1219 ret = delete_user(cli, lp_ctx, admin_creds, wks_name);
1220 talloc_free(wks_name);
1221 return ret;
1222 }
1223
1224 /*
1225 * Test the Samba3 DC code a bit. Join, do some schan netlogon ops, leave
1226 */
1227
1228 bool torture_netlogon_samba3(struct torture_context *torture)
1229 {
1230 TALLOC_CTX *mem_ctx;
1231 NTSTATUS status;
1232 bool ret = false;
1233 struct smbcli_state *cli;
1234 struct cli_credentials *anon_creds;
1235 struct cli_credentials *wks_creds;
1236 const char *wks_name;
1237 int i;
1238 struct smbcli_options options;
1239 struct smbcli_session_options session_options;
1240
1241 wks_name = torture_setting_string(torture, "wksname", NULL);
1242 if (wks_name == NULL) {
1243 wks_name = get_myname(torture);
1244 }
1245
1246 mem_ctx = talloc_init("torture_netlogon_samba3");
1247
1248 if (mem_ctx == NULL) {
1249 d_printf("talloc_init failed\n");
1250 return false;
1251 }
1252
1253 if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
1254 d_printf("create_anon_creds failed\n");
1255 goto done;
1256 }
1257
1258 lp_smbcli_options(torture->lp_ctx, &options);
1259 lp_smbcli_session_options(torture->lp_ctx, &session_options);
1260
1261 status = smbcli_full_connection(mem_ctx, &cli,
1262 torture_setting_string(torture, "host", NULL),
1263 lp_smb_ports(torture->lp_ctx),
1264 "IPC$", NULL,
1265 lp_socket_options(torture->lp_ctx),
1266 anon_creds,
1267 lp_resolve_context(torture->lp_ctx),
1268 torture->ev, &options, &session_options,
1269 lp_iconv_convenience(torture->lp_ctx),
1270 lp_gensec_settings(torture, torture->lp_ctx));
1271 if (!NT_STATUS_IS_OK(status)) {
1272 d_printf("smbcli_full_connection failed: %s\n",
1273 nt_errstr(status));
1274 goto done;
1275 }
1276
1277 wks_creds = cli_credentials_init(mem_ctx);
1278 if (wks_creds == NULL) {
1279 d_printf("cli_credentials_init failed\n");
1280 goto done;
1281 }
1282
1283 cli_credentials_set_conf(wks_creds, torture->lp_ctx);
1284 cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
1285 cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
1286 cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
1287 cli_credentials_set_password(wks_creds,
1288 generate_random_password(wks_creds, 8, 255),
1289 CRED_SPECIFIED);
1290
1291 if (!join3(cli, torture->lp_ctx, false, cmdline_credentials, wks_creds)) {
1292 d_printf("join failed\n");
1293 goto done;
1294 }
1295
1296 cli_credentials_set_domain(
1297 cmdline_credentials, cli_credentials_get_domain(wks_creds),
1298 CRED_SPECIFIED);
1299
1300 for (i=0; i<2; i++) {
1301
1302 /* Do this more than once, the routine "schan" changes
1303 * the workstation password using the netlogon
1304 * password change routine */
1305
1306 int j;
1307
1308 if (!auth2(cli, torture->lp_ctx, wks_creds)) {
1309 d_printf("auth2 failed\n");
1310 goto done;
1311 }
1312
1313 for (j=0; j<2; j++) {
1314 if (!schan(cli, torture->lp_ctx, wks_creds, cmdline_credentials)) {
1315 d_printf("schan failed\n");
1316 goto done;
1317 }
1318 }
1319 }
1320
1321 if (!leave(cli, torture->lp_ctx, cmdline_credentials, wks_creds)) {
1322 d_printf("leave failed\n");
1323 goto done;
1324 }
1325
1326 ret = true;
1327
1328 done:
1329 talloc_free(mem_ctx);
1330 return ret;
1331 }
1332
1333 /*
1334 * Do a simple join, testjoin and leave using specified smb and samr
1335 * credentials
1336 */
1337
1338 static bool test_join3(struct torture_context *tctx,
1339 bool use_level25,
1340 struct cli_credentials *smb_creds,
1341 struct cli_credentials *samr_creds,
1342 const char *wks_name)
1343 {
1344 NTSTATUS status;
1345 bool ret = false;
1346 struct smbcli_state *cli;
1347 struct cli_credentials *wks_creds;
1348 struct smbcli_options options;
1349 struct smbcli_session_options session_options;
1350
1351 lp_smbcli_options(tctx->lp_ctx, &options);
1352 lp_smbcli_session_options(tctx->lp_ctx, &session_options);
1353
1354 status = smbcli_full_connection(tctx, &cli,
1355 torture_setting_string(tctx, "host", NULL),
1356 lp_smb_ports(tctx->lp_ctx),
1357 "IPC$", NULL, lp_socket_options(tctx->lp_ctx),
1358 smb_creds, lp_resolve_context(tctx->lp_ctx),
1359 tctx->ev, &options, &session_options,
1360 lp_iconv_convenience(tctx->lp_ctx),
1361 lp_gensec_settings(tctx, tctx->lp_ctx));
1362 if (!NT_STATUS_IS_OK(status)) {
1363 d_printf("smbcli_full_connection failed: %s\n",
1364 nt_errstr(status));
1365 goto done;
1366 }
1367
1368 wks_creds = cli_credentials_init(cli);
1369 if (wks_creds == NULL) {
1370 d_printf("cli_credentials_init failed\n");
1371 goto done;
1372 }
1373
1374 cli_credentials_set_conf(wks_creds, tctx->lp_ctx);
1375 cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
1376 cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
1377 cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
1378 cli_credentials_set_password(wks_creds,
1379 generate_random_password(wks_creds, 8, 255),
1380 CRED_SPECIFIED);
1381
1382 if (!join3(cli, tctx->lp_ctx, use_level25, samr_creds, wks_creds)) {
1383 d_printf("join failed\n");
1384 goto done;
1385 }
1386
1387 cli_credentials_set_domain(
1388 cmdline_credentials, cli_credentials_get_domain(wks_creds),
1389 CRED_SPECIFIED);
1390
1391 if (!auth2(cli, tctx->lp_ctx, wks_creds)) {
1392 d_printf("auth2 failed\n");
1393 goto done;
1394 }
1395
1396 if (!leave(cli, tctx->lp_ctx, samr_creds, wks_creds)) {
1397 d_printf("leave failed\n");
1398 goto done;
1399 }
1400
1401 talloc_free(cli);
1402
1403 ret = true;
1404
1405 done:
1406 return ret;
1407 }
1408
1409 /*
1410 * Test the different session key variants. Do it by joining, this uses the
1411 * session key in the setpassword routine. Test the join by doing the auth2.
1412 */
1413
1414 bool torture_samba3_sessionkey(struct torture_context *torture)
1415 {
1416 bool ret = false;
1417 struct cli_credentials *anon_creds;
1418 const char *wks_name;
1419
1420 wks_name = torture_setting_string(torture, "wksname", get_myname(torture));
1421
1422 if (!(anon_creds = cli_credentials_init_anon(torture))) {
1423 d_printf("create_anon_creds failed\n");
1424 goto done;
1425 }
1426
1427 cli_credentials_set_workstation(anon_creds, wks_name, CRED_SPECIFIED);
1428
1429 ret = true;
1430
1431 if (!torture_setting_bool(torture, "samba3", false)) {
1432
1433 /* Samba3 in the build farm right now does this happily. Need
1434 * to fix :-) */
1435
1436 if (test_join3(torture, false, anon_creds, NULL, wks_name)) {
1437 d_printf("join using anonymous bind on an anonymous smb "
1438 "connection succeeded -- HUH??\n");
1439 ret = false;
1440 }
1441 }
1442
1443 if (!test_join3(torture, false, anon_creds, cmdline_credentials,
1444 wks_name)) {
1445 d_printf("join using ntlmssp bind on an anonymous smb "
1446 "connection failed\n");
1447 ret = false;
1448 }
1449
1450 if (!test_join3(torture, false, cmdline_credentials, NULL, wks_name)) {
1451 d_printf("join using anonymous bind on an authenticated smb "
1452 "connection failed\n");
1453 ret = false;
1454 }
1455
1456 if (!test_join3(torture, false, cmdline_credentials,
1457 cmdline_credentials,
1458 wks_name)) {
1459 d_printf("join using ntlmssp bind on an authenticated smb "
1460 "connection failed\n");
1461 ret = false;
1462 }
1463
1464 /*
1465 * The following two are tests for setuserinfolevel 25
1466 */
1467
1468 if (!test_join3(torture, true, anon_creds, cmdline_credentials,
1469 wks_name)) {
1470 d_printf("join using ntlmssp bind on an anonymous smb "
1471 "connection failed\n");
1472 ret = false;
1473 }
1474
1475 if (!test_join3(torture, true, cmdline_credentials, NULL, wks_name)) {
1476 d_printf("join using anonymous bind on an authenticated smb "
1477 "connection failed\n");
1478 ret = false;
1479 }
1480
1481 done:
1482
1483 return ret;
1484 }
1485
1486 /*
1487 * open pipe and bind, given an IPC$ context
1488 */
1489
1490 static NTSTATUS pipe_bind_smb(TALLOC_CTX *mem_ctx,
1491 struct loadparm_context *lp_ctx,
1492 struct smbcli_tree *tree,
1493 const char *pipe_name,
1494 const struct ndr_interface_table *iface,
1495 struct dcerpc_pipe **p)
1496 {
1497 struct dcerpc_pipe *result;
1498 NTSTATUS status;
1499
1500 if (!(result = dcerpc_pipe_init(
1501 mem_ctx, tree->session->transport->socket->event.ctx,
1502 lp_iconv_convenience(lp_ctx)))) {
1503 return NT_STATUS_NO_MEMORY;
1504 }
1505
1506 status = dcerpc_pipe_open_smb(result, tree, pipe_name);
1507 if (!NT_STATUS_IS_OK(status)) {
1508 d_printf("dcerpc_pipe_open_smb failed: %s\n",
1509 nt_errstr(status));
1510 talloc_free(result);
1511 return status;
1512 }
1513
1514 status = dcerpc_bind_auth_none(result, iface);
1515 if (!NT_STATUS_IS_OK(status)) {
1516 d_printf("schannel bind failed: %s\n", nt_errstr(status));
1517 talloc_free(result);
1518 return status;
1519 }
1520
1521 *p = result;
1522 return NT_STATUS_OK;
1523 }
1524
1525 /*
1526 * Sane wrapper around lsa_LookupNames
1527 */
1528
1529 static struct dom_sid *name2sid(TALLOC_CTX *mem_ctx,
1530 struct dcerpc_pipe *p,
1531 const char *name,
1532 const char *domain)
1533 {
1534 struct lsa_ObjectAttribute attr;
1535 struct lsa_QosInfo qos;
1536 struct lsa_OpenPolicy2 r;
1537 struct lsa_Close c;
1538 NTSTATUS status;
1539 struct policy_handle handle;
1540 struct lsa_LookupNames l;
1541 struct lsa_TransSidArray sids;
1542 struct lsa_RefDomainList *domains = NULL;
1543 struct lsa_String lsa_name;
1544 uint32_t count = 0;
1545 struct dom_sid *result;
1546 TALLOC_CTX *tmp_ctx;
1547
1548 if (!(tmp_ctx = talloc_new(mem_ctx))) {
1549 return NULL;
1550 }
1551
1552 qos.len = 0;
1553 qos.impersonation_level = 2;
1554 qos.context_mode = 1;
1555 qos.effective_only = 0;
1556
1557 attr.len = 0;
1558 attr.root_dir = NULL;
1559 attr.object_name = NULL;
1560 attr.attributes = 0;
1561 attr.sec_desc = NULL;
1562 attr.sec_qos = &qos;
1563
1564 r.in.system_name = "\\";
1565 r.in.attr = &attr;
1566 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1567 r.out.handle = &handle;
1568
1569 status = dcerpc_lsa_OpenPolicy2(p, tmp_ctx, &r);
1570 if (!NT_STATUS_IS_OK(status)) {
1571 printf("OpenPolicy2 failed - %s\n", nt_errstr(status));
1572 talloc_free(tmp_ctx);
1573 return NULL;
1574 }
1575
1576 sids.count = 0;
1577 sids.sids = NULL;
1578
1579 lsa_name.string = talloc_asprintf(tmp_ctx, "%s\\%s", domain, name);
1580
1581 l.in.handle = &handle;
1582 l.in.num_names = 1;
1583 l.in.names = &lsa_name;
1584 l.in.sids = &sids;
1585 l.in.level = 1;
1586 l.in.count = &count;
1587 l.out.count = &count;
1588 l.out.sids = &sids;
1589 l.out.domains = &domains;
1590
1591 status = dcerpc_lsa_LookupNames(p, tmp_ctx, &l);
1592 if (!NT_STATUS_IS_OK(status)) {
1593 printf("LookupNames of %s failed - %s\n", lsa_name.string,
1594 nt_errstr(status));
1595 talloc_free(tmp_ctx);
1596 return NULL;
1597 }
1598
1599 result = dom_sid_add_rid(mem_ctx, domains->domains[0].sid,
1600 l.out.sids->sids[0].rid);
1601
1602 c.in.handle = &handle;
1603 c.out.handle = &handle;
1604
1605 status = dcerpc_lsa_Close(p, tmp_ctx, &c);
1606 if (!NT_STATUS_IS_OK(status)) {
1607 printf("dcerpc_lsa_Close failed - %s\n", nt_errstr(status));
1608 talloc_free(tmp_ctx);
1609 return NULL;
1610 }
1611
1612 talloc_free(tmp_ctx);
1613 return result;
1614 }
1615
1616 /*
1617 * Find out the user SID on this connection
1618 */
1619
1620 static struct dom_sid *whoami(TALLOC_CTX *mem_ctx,
1621 struct loadparm_context *lp_ctx,
1622 struct smbcli_tree *tree)
1623 {
1624 struct dcerpc_pipe *lsa;
1625 struct lsa_GetUserName r;
1626 NTSTATUS status;
1627 struct lsa_String *authority_name_p = NULL;
1628 struct lsa_String *account_name_p = NULL;
1629 struct dom_sid *result;
1630
1631 status = pipe_bind_smb(mem_ctx, lp_ctx, tree, "\\pipe\\lsarpc",
1632 &ndr_table_lsarpc, &lsa);
1633 if (!NT_STATUS_IS_OK(status)) {
1634 d_printf("(%s) Could not bind to LSA: %s\n",
1635 __location__, nt_errstr(status));
1636 return NULL;
1637 }
1638
1639 r.in.system_name = "\\";
1640 r.in.account_name = &account_name_p;
1641 r.in.authority_name = &authority_name_p;
1642 r.out.account_name = &account_name_p;
1643
1644 status = dcerpc_lsa_GetUserName(lsa, mem_ctx, &r);
1645
1646 authority_name_p = *r.out.authority_name;
1647
1648 if (!NT_STATUS_IS_OK(status)) {
1649 printf("(%s) GetUserName failed - %s\n",
1650 __location__, nt_errstr(status));
1651 talloc_free(lsa);
1652 return NULL;
1653 }
1654
1655 result = name2sid(mem_ctx, lsa, account_name_p->string,
1656 authority_name_p->string);
1657
1658 talloc_free(lsa);
1659 return result;
1660 }
1661
1662 static int destroy_tree(struct smbcli_tree *tree)
1663 {
1664 smb_tree_disconnect(tree);
1665 return 0;
1666 }
1667
1668 /*
1669 * Do a tcon, given a session
1670 */
1671
1672 NTSTATUS secondary_tcon(TALLOC_CTX *mem_ctx,
1673 struct smbcli_session *session,
1674 const char *sharename,
1675 struct smbcli_tree **res)
1676 {
1677 struct smbcli_tree *result;
1678 TALLOC_CTX *tmp_ctx;
1679 union smb_tcon tcon;
1680 NTSTATUS status;
1681
1682 if (!(tmp_ctx = talloc_new(mem_ctx))) {
1683 return NT_STATUS_NO_MEMORY;
1684 }
1685
1686 if (!(result = smbcli_tree_init(session, mem_ctx, false))) {
1687 talloc_free(tmp_ctx);
1688 return NT_STATUS_NO_MEMORY;
1689 }
1690
1691 tcon.generic.level = RAW_TCON_TCONX;
1692 tcon.tconx.in.flags = 0;
1693 tcon.tconx.in.password = data_blob(NULL, 0);
1694 tcon.tconx.in.path = sharename;
1695 tcon.tconx.in.device = "?????";
1696
1697 status = smb_raw_tcon(result, tmp_ctx, &tcon);
1698 if (!NT_STATUS_IS_OK(status)) {
1699 d_printf("(%s) smb_raw_tcon failed: %s\n", __location__,
1700 nt_errstr(status));
1701 talloc_free(tmp_ctx);
1702 return status;
1703 }
1704
1705 result->tid = tcon.tconx.out.tid;
1706 result = talloc_steal(mem_ctx, result);
1707 talloc_set_destructor(result, destroy_tree);
1708 talloc_free(tmp_ctx);
1709 *res = result;
1710 return NT_STATUS_OK;
1711 }
1712
1713 /*
1714 * Test the getusername behaviour
1715 */
1716
1717 bool torture_samba3_rpc_getusername(struct torture_context *torture)
1718 {
1719 NTSTATUS status;
1720 struct smbcli_state *cli;
1721 TALLOC_CTX *mem_ctx;
1722 bool ret = true;
1723 struct dom_sid *user_sid;
1724 struct dom_sid *created_sid;
1725 struct cli_credentials *anon_creds;
1726 struct cli_credentials *user_creds;
1727 char *domain_name;
1728 struct smbcli_options options;
1729 struct smbcli_session_options session_options;
1730
1731 if (!(mem_ctx = talloc_new(torture))) {
1732 return false;
1733 }
1734
1735 lp_smbcli_options(torture->lp_ctx, &options);
1736 lp_smbcli_session_options(torture->lp_ctx, &session_options);
1737
1738 status = smbcli_full_connection(
1739 mem_ctx, &cli, torture_setting_string(torture, "host", NULL),
1740 lp_smb_ports(torture->lp_ctx),
1741 "IPC$", NULL, lp_socket_options(torture->lp_ctx), cmdline_credentials,
1742 lp_resolve_context(torture->lp_ctx),
1743 torture->ev, &options, &session_options,
1744 lp_iconv_convenience(torture->lp_ctx),
1745 lp_gensec_settings(torture, torture->lp_ctx));
1746 if (!NT_STATUS_IS_OK(status)) {
1747 d_printf("(%s) smbcli_full_connection failed: %s\n",
1748 __location__, nt_errstr(status));
1749 ret = false;
1750 goto done;
1751 }
1752
1753 if (!(user_sid = whoami(mem_ctx, torture->lp_ctx, cli->tree))) {
1754 d_printf("(%s) whoami on auth'ed connection failed\n",
1755 __location__);
1756 ret = false;
1757 }
1758
1759 talloc_free(cli);
1760
1761 if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
1762 d_printf("(%s) create_anon_creds failed\n", __location__);
1763 ret = false;
1764 goto done;
1765 }
1766
1767 status = smbcli_full_connection(
1768 mem_ctx, &cli, torture_setting_string(torture, "host", NULL),
1769 lp_smb_ports(torture->lp_ctx), "IPC$", NULL,
1770 lp_socket_options(torture->lp_ctx), anon_creds,
1771 lp_resolve_context(torture->lp_ctx),
1772 torture->ev, &options, &session_options,
1773 lp_iconv_convenience(torture->lp_ctx),
1774 lp_gensec_settings(torture, torture->lp_ctx));
1775 if (!NT_STATUS_IS_OK(status)) {
1776 d_printf("(%s) anon smbcli_full_connection failed: %s\n",
1777 __location__, nt_errstr(status));
1778 ret = false;
1779 goto done;
1780 }
1781
1782 if (!(user_sid = whoami(mem_ctx, torture->lp_ctx, cli->tree))) {
1783 d_printf("(%s) whoami on anon connection failed\n",
1784 __location__);
1785 ret = false;
1786 goto done;
1787 }
1788
1789 if (!dom_sid_equal(user_sid,
1790 dom_sid_parse_talloc(mem_ctx, "s-1-5-7"))) {
1791 d_printf("(%s) Anon lsa_GetUserName returned %s, expected "
1792 "S-1-5-7", __location__,
1793 dom_sid_string(mem_ctx, user_sid));
1794 ret = false;
1795 }
1796
1797 if (!(user_creds = cli_credentials_init(mem_ctx))) {
1798 d_printf("(%s) cli_credentials_init failed\n", __location__);
1799 ret = false;
1800 goto done;
1801 }
1802
1803 cli_credentials_set_conf(user_creds, torture->lp_ctx);
1804 cli_credentials_set_username(user_creds, "torture_username",
1805 CRED_SPECIFIED);
1806 cli_credentials_set_password(user_creds,
1807 generate_random_password(user_creds, 8, 255),
1808 CRED_SPECIFIED);
1809
1810 if (!create_user(mem_ctx, cli, torture->lp_ctx, cmdline_credentials,
1811 cli_credentials_get_username(user_creds),
1812 cli_credentials_get_password(user_creds),
1813 &domain_name, &created_sid)) {
1814 d_printf("(%s) create_user failed\n", __location__);
1815 ret = false;
1816 goto done;
1817 }
1818
1819 cli_credentials_set_domain(user_creds, domain_name,
1820 CRED_SPECIFIED);
1821
1822 {
1823 struct smbcli_session *session2;
1824 struct smb_composite_sesssetup setup;
1825 struct smbcli_tree *tree;
1826
1827 session2 = smbcli_session_init(cli->transport, mem_ctx, false, session_options);
1828 if (session2 == NULL) {
1829 d_printf("(%s) smbcli_session_init failed\n",
1830 __location__);
1831 goto done;
1832 }
1833
1834 setup.in.sesskey = cli->transport->negotiate.sesskey;
1835 setup.in.capabilities = cli->transport->negotiate.capabilities;
1836 setup.in.workgroup = "";
1837 setup.in.credentials = user_creds;
1838 setup.in.gensec_settings = lp_gensec_settings(torture, torture->lp_ctx);
1839
1840 status = smb_composite_sesssetup(session2, &setup);
1841 if (!NT_STATUS_IS_OK(status)) {
1842 d_printf("(%s) session setup with new user failed: "
1843 "%s\n", __location__, nt_errstr(status));
1844 ret = false;
1845 goto done;
1846 }
1847 session2->vuid = setup.out.vuid;
1848
1849 if (!NT_STATUS_IS_OK(secondary_tcon(mem_ctx, session2,
1850 "IPC$", &tree))) {
1851 d_printf("(%s) secondary_tcon failed\n",
1852 __location__);
1853 ret = false;
1854 goto done;
1855 }
1856
1857 if (!(user_sid = whoami(mem_ctx, torture->lp_ctx, tree))) {
1858 d_printf("(%s) whoami on user connection failed\n",
1859 __location__);
1860 ret = false;
1861 goto del;
1862 }
1863
1864 talloc_free(tree);
1865 }
1866
1867 d_printf("Created %s, found %s\n",
1868 dom_sid_string(mem_ctx, created_sid),
1869 dom_sid_string(mem_ctx, user_sid));
1870
1871 if (!dom_sid_equal(created_sid, user_sid)) {
1872 ret = false;
1873 }
1874
1875 del:
1876 if (!delete_user(cli, torture->lp_ctx,
1877 cmdline_credentials,
1878 cli_credentials_get_username(user_creds))) {
1879 d_printf("(%s) delete_user failed\n", __location__);
1880 ret = false;
1881 }
1882
1883 done:
1884 talloc_free(mem_ctx);
1885 return ret;
1886 }
1887
1888 static bool test_NetShareGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
1889 const char *sharename)
1890 {
1891 NTSTATUS status;
1892 struct srvsvc_NetShareGetInfo r;
1893 union srvsvc_NetShareInfo info;
1894 uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007, 1501 };
1895 int i;
1896 bool ret = true;
1897
1898 r.in.server_unc = talloc_asprintf(mem_ctx, "\\\\%s",
1899 dcerpc_server_name(p));
1900 r.in.share_name = sharename;
1901 r.out.info = &info;
1902
1903 for (i=0;i<ARRAY_SIZE(levels);i++) {
1904 r.in.level = levels[i];
1905
1906 printf("testing NetShareGetInfo level %u on share '%s'\n",
1907 r.in.level, r.in.share_name);
1908
1909 status = dcerpc_srvsvc_NetShareGetInfo(p, mem_ctx, &r);
1910 if (!NT_STATUS_IS_OK(status)) {
1911 printf("NetShareGetInfo level %u on share '%s' failed"
1912 " - %s\n", r.in.level, r.in.share_name,
1913 nt_errstr(status));
1914 ret = false;
1915 continue;
1916 }
1917 if (!W_ERROR_IS_OK(r.out.result)) {
1918 printf("NetShareGetInfo level %u on share '%s' failed "
1919 "- %s\n", r.in.level, r.in.share_name,
1920 win_errstr(r.out.result));
1921 ret = false;
1922 continue;
1923 }
1924 }
1925
1926 return ret;
1927 }
1928
1929 static bool test_NetShareEnum(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
1930 const char **one_sharename)
1931 {
1932 NTSTATUS status;
1933 struct srvsvc_NetShareEnum r;
1934 struct srvsvc_NetShareInfoCtr info_ctr;
1935 struct srvsvc_NetShareCtr0 c0;
1936 struct srvsvc_NetShareCtr1 c1;
1937 struct srvsvc_NetShareCtr2 c2;
1938 struct srvsvc_NetShareCtr501 c501;
1939 struct srvsvc_NetShareCtr502 c502;
1940 struct srvsvc_NetShareCtr1004 c1004;
1941 struct srvsvc_NetShareCtr1005 c1005;
1942 struct srvsvc_NetShareCtr1006 c1006;
1943 struct srvsvc_NetShareCtr1007 c1007;
1944 uint32_t totalentries = 0;
1945 uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007 };
1946 int i;
1947 bool ret = true;
1948
1949 ZERO_STRUCT(info_ctr);
1950
1951 r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
1952 r.in.info_ctr = &info_ctr;
1953 r.in.max_buffer = (uint32_t)-1;
1954 r.in.resume_handle = NULL;
1955 r.out.totalentries = &totalentries;
1956 r.out.info_ctr = &info_ctr;
1957
1958 for (i=0;i<ARRAY_SIZE(levels);i++) {
1959 info_ctr.level = levels[i];
1960
1961 switch (info_ctr.level) {
1962 case 0:
1963 ZERO_STRUCT(c0);
1964 info_ctr.ctr.ctr0 = &c0;
1965 break;
1966 case 1:
1967 ZERO_STRUCT(c1);
1968 info_ctr.ctr.ctr1 = &c1;
1969 break;
1970 case 2:
1971 ZERO_STRUCT(c2);
1972 info_ctr.ctr.ctr2 = &c2;
1973 break;
1974 case 501:
1975 ZERO_STRUCT(c501);
1976 info_ctr.ctr.ctr501 = &c501;
1977 break;
1978 case 502:
1979 ZERO_STRUCT(c502);
1980 info_ctr.ctr.ctr502 = &c502;
1981 break;
1982 case 1004:
1983 ZERO_STRUCT(c1004);
1984 info_ctr.ctr.ctr1004 = &c1004;
1985 break;
1986 case 1005:
1987 ZERO_STRUCT(c1005);
1988 info_ctr.ctr.ctr1005 = &c1005;
1989 break;
1990 case 1006:
1991 ZERO_STRUCT(c1006);
1992 info_ctr.ctr.ctr1006 = &c1006;
1993 break;
1994 case 1007:
1995 ZERO_STRUCT(c1007);
1996 info_ctr.ctr.ctr1007 = &c1007;
1997 break;
1998 }
1999
2000 printf("testing NetShareEnum level %u\n", info_ctr.level);
2001 status = dcerpc_srvsvc_NetShareEnum(p, mem_ctx, &r);
2002 if (!NT_STATUS_IS_OK(status)) {
2003 printf("NetShareEnum level %u failed - %s\n",
2004 info_ctr.level, nt_errstr(status));
2005 ret = false;
2006 continue;
2007 }
2008 if (!W_ERROR_IS_OK(r.out.result)) {
2009 printf("NetShareEnum level %u failed - %s\n",
2010 info_ctr.level, win_errstr(r.out.result));
2011 continue;
2012 }
2013 if (info_ctr.level == 0) {
2014 struct srvsvc_NetShareCtr0 *ctr = r.out.info_ctr->ctr.ctr0;
2015 if (ctr->count > 0) {
2016 *one_sharename = ctr->array[0].name;
2017 }
2018 }
2019 }
2020
2021 return ret;
2022 }
2023
2024 bool torture_samba3_rpc_srvsvc(struct torture_context *torture)
2025 {
2026 struct dcerpc_pipe *p;
2027 TALLOC_CTX *mem_ctx;
2028 bool ret = true;
2029 const char *sharename = NULL;
2030 struct smbcli_state *cli;
2031 NTSTATUS status;
2032
2033 if (!(mem_ctx = talloc_new(torture))) {
2034 return false;
2035 }
2036
2037 if (!(torture_open_connection_share(
2038 mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
2039 "IPC$", torture->ev))) {
2040 talloc_free(mem_ctx);
2041 return false;
2042 }
2043
2044 status = pipe_bind_smb(mem_ctx, torture->lp_ctx, cli->tree,
2045 "\\pipe\\srvsvc", &ndr_table_srvsvc, &p);
2046 if (!NT_STATUS_IS_OK(status)) {
2047 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
2048 __location__, nt_errstr(status));
2049 ret = false;
2050 goto done;
2051 }
2052
2053 ret &= test_NetShareEnum(p, mem_ctx, &sharename);
2054 if (sharename == NULL) {
2055 printf("did not get sharename\n");
2056 } else {
2057 ret &= test_NetShareGetInfo(p, mem_ctx, sharename);
2058 }
2059
2060 done:
2061 talloc_free(mem_ctx);
2062 return ret;
2063 }
2064
2065 /*
2066 * Do a ReqChallenge/Auth2 with a random wks name, make sure it returns
2067 * NT_STATUS_NO_SAM_ACCOUNT
2068 */
2069
2070 bool torture_samba3_rpc_randomauth2(struct torture_context *torture)
2071 {
2072 TALLOC_CTX *mem_ctx;
2073 struct dcerpc_pipe *net_pipe;
2074 char *wksname;
2075 bool result = false;
2076 NTSTATUS status;
2077 struct netr_ServerReqChallenge r;
2078 struct netr_Credential netr_cli_creds;
2079 struct netr_Credential netr_srv_creds;
2080 uint32_t negotiate_flags;
2081 struct netr_ServerAuthenticate2 a;
2082 struct netlogon_creds_CredentialState *creds_state;
2083 struct netr_Credential netr_cred;
2084 struct samr_Password mach_pw;
2085 struct smbcli_state *cli;
2086
2087 if (!(mem_ctx = talloc_new(torture))) {
2088 d_printf("talloc_new failed\n");
2089 return false;
2090 }
2091
2092 if (!(wksname = generate_random_str_list(
2093 mem_ctx, 14, "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"))) {
2094 d_printf("generate_random_str_list failed\n");
2095 goto done;
2096 }
2097
2098 if (!(torture_open_connection_share(
2099 mem_ctx, &cli,
2100 torture, torture_setting_string(torture, "host", NULL),
2101 "IPC$", torture->ev))) {
2102 d_printf("IPC$ connection failed\n");
2103 goto done;
2104 }
2105
2106 if (!(net_pipe = dcerpc_pipe_init(
2107 mem_ctx, cli->transport->socket->event.ctx,
2108 lp_iconv_convenience(torture->lp_ctx)))) {
2109 d_printf("dcerpc_pipe_init failed\n");
2110 goto done;
2111 }
2112
2113 status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
2114 if (!NT_STATUS_IS_OK(status)) {
2115 d_printf("dcerpc_pipe_open_smb failed: %s\n",
2116 nt_errstr(status));
2117 goto done;
2118 }
2119
2120 status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
2121 if (!NT_STATUS_IS_OK(status)) {
2122 d_printf("dcerpc_bind_auth_none failed: %s\n",
2123 nt_errstr(status));
2124 goto done;
2125 }
2126
2127 r.in.computer_name = wksname;
2128 r.in.server_name = talloc_asprintf(
2129 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
2130 if (r.in.server_name == NULL) {
2131 d_printf("talloc_asprintf failed\n");
2132 goto done;
2133 }
2134 generate_random_buffer(netr_cli_creds.data,
2135 sizeof(netr_cli_creds.data));
2136 r.in.credentials = &netr_cli_creds;
2137 r.out.return_credentials = &netr_srv_creds;
2138
2139 status = dcerpc_netr_ServerReqChallenge(net_pipe, mem_ctx, &r);
2140 if (!NT_STATUS_IS_OK(status)) {
2141 d_printf("netr_ServerReqChallenge failed: %s\n",
2142 nt_errstr(status));
2143 goto done;
2144 }
2145
2146 negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
2147 E_md4hash("foobar", mach_pw.hash);
2148
2149 a.in.server_name = talloc_asprintf(
2150 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
2151 a.in.account_name = talloc_asprintf(
2152 mem_ctx, "%s$", wksname);
2153 a.in.computer_name = wksname;
2154 a.in.secure_channel_type = SEC_CHAN_WKSTA;
2155 a.in.negotiate_flags = &negotiate_flags;
2156 a.out.negotiate_flags = &negotiate_flags;
2157 a.in.credentials = &netr_cred;
2158 a.out.return_credentials = &netr_cred;
2159
2160 creds_state = netlogon_creds_client_init(mem_ctx,
2161 a.in.account_name,
2162 a.in.computer_name,
2163 r.in.credentials,
2164 r.out.return_credentials, &mach_pw,
2165 &netr_cred, negotiate_flags);
2166
2167
2168 status = dcerpc_netr_ServerAuthenticate2(net_pipe, mem_ctx, &a);
2169
2170 if (!NT_STATUS_EQUAL(status, NT_STATUS_NO_TRUST_SAM_ACCOUNT)) {
2171 d_printf("dcerpc_netr_ServerAuthenticate2 returned %s, "
2172 "expected NT_STATUS_NO_TRUST_SAM_ACCOUNT\n",
2173 nt_errstr(status));
2174 goto done;
2175 }
2176
2177 result = true;
2178 done:
2179 talloc_free(mem_ctx);
2180 return result;
2181 }
2182
2183 static struct security_descriptor *get_sharesec(TALLOC_CTX *mem_ctx,
2184 struct loadparm_context *lp_ctx,
2185 struct smbcli_session *sess,
2186 const char *sharename)
2187 {
2188 struct smbcli_tree *tree;
2189 TALLOC_CTX *tmp_ctx;
2190 struct dcerpc_pipe *p;
2191 NTSTATUS status;
2192 struct srvsvc_NetShareGetInfo r;
2193 union srvsvc_NetShareInfo info;
2194 struct security_descriptor *result;
2195
2196 if (!(tmp_ctx = talloc_new(mem_ctx))) {
2197 d_printf("talloc_new failed\n");
2198 return NULL;
2199 }
2200
2201 if (!NT_STATUS_IS_OK(secondary_tcon(tmp_ctx, sess, "IPC$", &tree))) {
2202 d_printf("secondary_tcon failed\n");
2203 talloc_free(tmp_ctx);
2204 return NULL;
2205 }
2206
2207 status = pipe_bind_smb(mem_ctx, lp_ctx, tree, "\\pipe\\srvsvc",
2208 &ndr_table_srvsvc, &p);
2209 if (!NT_STATUS_IS_OK(status)) {
2210 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
2211 __location__, nt_errstr(status));
2212 talloc_free(tmp_ctx);
2213 return NULL;
2214 }
2215
2216 #if 0
2217 p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
2218 #endif
2219
2220 r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
2221 dcerpc_server_name(p));
2222 r.in.share_name = sharename;
2223 r.in.level = 502;
2224 r.out.info = &info;
2225
2226 status = dcerpc_srvsvc_NetShareGetInfo(p, tmp_ctx, &r);
2227 if (!NT_STATUS_IS_OK(status)) {
2228 d_printf("srvsvc_NetShareGetInfo failed: %s\n",
2229 nt_errstr(status));
2230 talloc_free(tmp_ctx);
2231 return NULL;
2232 }
2233
2234 result = talloc_steal(mem_ctx, info.info502->sd_buf.sd);
2235 talloc_free(tmp_ctx);
2236 return result;
2237 }
2238
2239 static NTSTATUS set_sharesec(TALLOC_CTX *mem_ctx,
2240 struct loadparm_context *lp_ctx,
2241 struct smbcli_session *sess,
2242 const char *sharename,
2243 struct security_descriptor *sd)
2244 {
2245 struct smbcli_tree *tree;
2246 TALLOC_CTX *tmp_ctx;
2247 struct dcerpc_pipe *p;
2248 NTSTATUS status;
2249 struct sec_desc_buf i;
2250 struct srvsvc_NetShareSetInfo r;
2251 union srvsvc_NetShareInfo info;
2252 uint32_t error = 0;
2253
2254 if (!(tmp_ctx = talloc_new(mem_ctx))) {
2255 d_printf("talloc_new failed\n");
2256 return NT_STATUS_NO_MEMORY;
2257 }
2258
2259 if (!NT_STATUS_IS_OK(secondary_tcon(tmp_ctx, sess, "IPC$", &tree))) {
2260 d_printf("secondary_tcon failed\n");
2261 talloc_free(tmp_ctx);
2262 return NT_STATUS_UNSUCCESSFUL;
2263 }
2264
2265 status = pipe_bind_smb(mem_ctx, lp_ctx, tree, "\\pipe\\srvsvc",
2266 &ndr_table_srvsvc, &p);
2267 if (!NT_STATUS_IS_OK(status)) {
2268 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
2269 __location__, nt_errstr(status));
2270 talloc_free(tmp_ctx);
2271 return NT_STATUS_UNSUCCESSFUL;
2272 }
2273
2274 #if 0
2275 p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
2276 #endif
2277
2278 r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
2279 dcerpc_server_name(p));
2280 r.in.share_name = sharename;
2281 r.in.level = 1501;
2282 i.sd = sd;
2283 info.info1501 = &i;
2284 r.in.info = &info;
2285 r.in.parm_error = &error;
2286
2287 status = dcerpc_srvsvc_NetShareSetInfo(p, tmp_ctx, &r);
2288 if (!NT_STATUS_IS_OK(status)) {
2289 d_printf("srvsvc_NetShareSetInfo failed: %s\n",
2290 nt_errstr(status));
2291 }
2292
2293 talloc_free(tmp_ctx);
2294 return status;
2295 }
2296
2297 bool try_tcon(TALLOC_CTX *mem_ctx,
2298 struct loadparm_context *lp_ctx,
2299 struct security_descriptor *orig_sd,
2300 struct smbcli_session *session,
2301 const char *sharename, const struct dom_sid *user_sid,
2302 unsigned int access_mask, NTSTATUS expected_tcon,
2303 NTSTATUS expected_mkdir)
2304 {
2305 TALLOC_CTX *tmp_ctx;
2306 struct smbcli_tree *rmdir_tree, *tree;
2307 struct dom_sid *domain_sid;
2308 uint32_t rid;
2309 struct security_descriptor *sd;
2310 NTSTATUS status;
2311 bool ret = true;
2312
2313 if (!(tmp_ctx = talloc_new(mem_ctx))) {
2314 d_printf("talloc_new failed\n");
2315 return false;
2316 }
2317
2318 status = secondary_tcon(tmp_ctx, session, sharename, &rmdir_tree);
2319 if (!NT_STATUS_IS_OK(status)) {
2320 d_printf("first tcon to delete dir failed\n");
2321 talloc_free(tmp_ctx);
2322 return false;
2323 }
2324
2325 smbcli_rmdir(rmdir_tree, "sharesec_testdir");
2326
2327 if (!NT_STATUS_IS_OK(dom_sid_split_rid(tmp_ctx, user_sid,
2328 &domain_sid, &rid))) {
2329 d_printf("dom_sid_split_rid failed\n");
2330 talloc_free(tmp_ctx);
2331 return false;
2332 }
2333
2334 sd = security_descriptor_dacl_create(
2335 tmp_ctx, 0, "S-1-5-32-544",
2336 dom_sid_string(mem_ctx, dom_sid_add_rid(mem_ctx, domain_sid,
2337 DOMAIN_RID_USERS)),
2338 dom_sid_string(mem_ctx, user_sid),
2339 SEC_ACE_TYPE_ACCESS_ALLOWED, access_mask, 0, NULL);
2340 if (sd == NULL) {
2341 d_printf("security_descriptor_dacl_create failed\n");
2342 talloc_free(tmp_ctx);
2343 return false;
2344 }
2345
2346 status = set_sharesec(mem_ctx, lp_ctx, session, sharename, sd);
2347 if (!NT_STATUS_IS_OK(status)) {
2348 d_printf("custom set_sharesec failed: %s\n",
2349 nt_errstr(status));
2350 talloc_free(tmp_ctx);
2351 return false;
2352 }
2353
2354 status = secondary_tcon(tmp_ctx, session, sharename, &tree);
2355 if (!NT_STATUS_EQUAL(status, expected_tcon)) {
2356 d_printf("Expected %s, got %s\n", nt_errstr(expected_tcon),
2357 nt_errstr(status));
2358 ret = false;
2359 goto done;
2360 }
2361
2362 if (!NT_STATUS_IS_OK(status)) {
2363 /* An expected non-access, no point in trying to write */
2364 goto done;
2365 }
2366
2367 status = smbcli_mkdir(tree, "sharesec_testdir");
2368 if (!NT_STATUS_EQUAL(status, expected_mkdir)) {
2369 d_printf("(%s) Expected %s, got %s\n", __location__,
2370 nt_errstr(expected_mkdir), nt_errstr(status));
2371 ret = false;
2372 }
2373
2374 done:
2375 smbcli_rmdir(rmdir_tree, "sharesec_testdir");
2376
2377 status = set_sharesec(mem_ctx, lp_ctx, session, sharename, orig_sd);
2378 if (!NT_STATUS_IS_OK(status)) {
2379 d_printf("custom set_sharesec failed: %s\n",
2380 nt_errstr(status));
2381 talloc_free(tmp_ctx);
2382 return false;
2383 }
2384
2385 talloc_free(tmp_ctx);
2386 return ret;
2387 }
2388
2389 bool torture_samba3_rpc_sharesec(struct torture_context *torture)
2390 {
2391 TALLOC_CTX *mem_ctx;
2392 bool ret = true;
2393 struct smbcli_state *cli;
2394 struct security_descriptor *sd;
2395 struct dom_sid *user_sid;
2396
2397 if (!(mem_ctx = talloc_new(torture))) {
2398 return false;
2399 }
2400
2401 if (!(torture_open_connection_share(
2402 mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
2403 "IPC$", torture->ev))) {
2404 d_printf("IPC$ connection failed\n");
2405 talloc_free(mem_ctx);
2406 return false;
2407 }
2408
2409 if (!(user_sid = whoami(mem_ctx, torture->lp_ctx, cli->tree))) {
2410 d_printf("whoami failed\n");
2411 talloc_free(mem_ctx);
2412 return false;
2413 }
2414
2415 sd = get_sharesec(mem_ctx, torture->lp_ctx, cli->session,
2416 torture_setting_string(torture, "share", NULL));
2417
2418 ret &= try_tcon(mem_ctx, torture->lp_ctx, sd, cli->session,
2419 torture_setting_string(torture, "share", NULL),
2420 user_sid, 0, NT_STATUS_ACCESS_DENIED, NT_STATUS_OK);
2421
2422 ret &= try_tcon(mem_ctx, torture->lp_ctx, sd, cli->session,
2423 torture_setting_string(torture, "share", NULL),
2424 user_sid, SEC_FILE_READ_DATA, NT_STATUS_OK,
2425 NT_STATUS_MEDIA_WRITE_PROTECTED);
2426
2427 ret &= try_tcon(mem_ctx, torture->lp_ctx, sd, cli->session,
2428 torture_setting_string(torture, "share", NULL),
2429 user_sid, SEC_FILE_ALL, NT_STATUS_OK, NT_STATUS_OK);
2430
2431 talloc_free(mem_ctx);
2432 return ret;
2433 }
2434
2435 bool torture_samba3_rpc_lsa(struct torture_context *torture)
2436 {
2437 TALLOC_CTX *mem_ctx;
2438 bool ret = true;
2439 struct smbcli_state *cli;
2440 struct dcerpc_pipe *p;
2441 struct policy_handle lsa_handle;
2442 NTSTATUS status;
2443 struct dom_sid *domain_sid;
2444
2445 if (!(mem_ctx = talloc_new(torture))) {
2446 return false;
2447 }
2448
2449 if (!(torture_open_connection_share(
2450 mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
2451 "IPC$", torture->ev))) {
2452 d_printf("IPC$ connection failed\n");
2453 talloc_free(mem_ctx);
2454 return false;
2455 }
2456
2457 status = pipe_bind_smb(mem_ctx, torture->lp_ctx, cli->tree, "\\lsarpc",
2458 &ndr_table_lsarpc, &p);
2459 if (!NT_STATUS_IS_OK(status)) {
2460 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2461 nt_errstr(status));
2462 talloc_free(mem_ctx);
2463 return false;
2464 }
2465
2466 {
2467 struct lsa_ObjectAttribute attr;
2468 struct lsa_OpenPolicy2 o;
2469 o.in.system_name = talloc_asprintf(
2470 mem_ctx, "\\\\%s", dcerpc_server_name(p));
2471 ZERO_STRUCT(attr);
2472 o.in.attr = &attr;
2473 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2474 o.out.handle = &lsa_handle;
2475 status = dcerpc_lsa_OpenPolicy2(p, mem_ctx, &o);
2476 if (!NT_STATUS_IS_OK(status)) {
2477 d_printf("(%s) dcerpc_lsa_OpenPolicy2 failed: %s\n",
2478 __location__, nt_errstr(status));
2479 talloc_free(mem_ctx);
2480 return false;
2481 }
2482 }
2483
2484 #if 0
2485 p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
2486 #endif
2487
2488 {
2489 int i;
2490 int levels[] = { 2,3,5,6 };
2491
2492 for (i=0; i<ARRAY_SIZE(levels); i++) {
2493 struct lsa_QueryInfoPolicy r;
2494 union lsa_PolicyInformation *info = NULL;
2495 r.in.handle = &lsa_handle;
2496 r.in.level = levels[i];
2497 r.out.info = &info;
2498 status = dcerpc_lsa_QueryInfoPolicy(p, mem_ctx, &r);
2499 if (!NT_STATUS_IS_OK(status)) {
2500 d_printf("(%s) dcerpc_lsa_QueryInfoPolicy %d "
2501 "failed: %s\n", __location__,
2502 levels[i], nt_errstr(status));
2503 talloc_free(mem_ctx);
2504 return false;
2505 }
2506 if (levels[i] == 5) {
2507 domain_sid = info->account_domain.sid;
2508 }
2509 }
2510 }
2511
2512 return ret;
2513 }
2514
2515 static NTSTATUS get_servername(TALLOC_CTX *mem_ctx, struct smbcli_tree *tree,
2516 struct smb_iconv_convenience *iconv_convenience,
2517 char **name)
2518 {
2519 struct rap_WserverGetInfo r;
2520 NTSTATUS status;
2521 char servername[17];
2522
2523 r.in.level = 0;
2524 r.in.bufsize = 0xffff;
2525
2526 status = smbcli_rap_netservergetinfo(tree, iconv_convenience, mem_ctx, &r);
2527 if (!NT_STATUS_IS_OK(status)) {
2528 return status;
2529 }
2530
2531 memcpy(servername, r.out.info.info0.name, 16);
2532 servername[16] = '\0';
2533
2534 if (!pull_ascii_talloc(mem_ctx, name, servername, NULL)) {
2535 return NT_STATUS_NO_MEMORY;
2536 }
2537
2538 return NT_STATUS_OK;
2539 }
2540
2541
2542 static NTSTATUS find_printers(TALLOC_CTX *ctx, struct loadparm_context *lp_ctx,
2543 struct smbcli_tree *tree,
2544 const char ***printers, int *num_printers)
2545 {
2546 TALLOC_CTX *mem_ctx;
2547 NTSTATUS status;
2548 struct dcerpc_pipe *p;
2549 struct srvsvc_NetShareEnum r;
2550 struct srvsvc_NetShareInfoCtr info_ctr;
2551 struct srvsvc_NetShareCtr1 c1_in;
2552 struct srvsvc_NetShareCtr1 *c1;
2553 uint32_t totalentries = 0;
2554 int i;
2555
2556 mem_ctx = talloc_new(ctx);
2557 if (mem_ctx == NULL) {
2558 return NT_STATUS_NO_MEMORY;
2559 }
2560
2561 status = pipe_bind_smb(mem_ctx, lp_ctx,
2562 tree, "\\srvsvc", &ndr_table_srvsvc,
2563 &p);
2564 if (!NT_STATUS_IS_OK(status)) {
2565 d_printf("could not bind to srvsvc pipe\n");
2566 talloc_free(mem_ctx);
2567 return status;
2568 }
2569
2570 ZERO_STRUCT(c1_in);
2571 info_ctr.level = 1;
2572 info_ctr.ctr.ctr1 = &c1_in;
2573
2574 r.in.server_unc = talloc_asprintf(
2575 mem_ctx, "\\\\%s", dcerpc_server_name(p));
2576 r.in.info_ctr = &info_ctr;
2577 r.in.max_buffer = (uint32_t)-1;
2578 r.in.resume_handle = NULL;
2579 r.out.totalentries = &totalentries;
2580 r.out.info_ctr = &info_ctr;
2581
2582 status = dcerpc_srvsvc_NetShareEnum(p, mem_ctx, &r);
2583 if (!NT_STATUS_IS_OK(status)) {
2584 d_printf("NetShareEnum level %u failed - %s\n",
2585 info_ctr.level, nt_errstr(status));
2586 talloc_free(mem_ctx);
2587 return status;
2588 }
2589
2590 *printers = NULL;
2591 *num_printers = 0;
2592 c1 = r.out.info_ctr->ctr.ctr1;
2593 for (i=0; i<c1->count; i++) {
2594 if (c1->array[i].type != STYPE_PRINTQ) {
2595 continue;
2596 }
2597 if (!add_string_to_array(ctx, c1->array[i].name,
2598 printers, num_printers)) {
2599 talloc_free(ctx);
2600 return NT_STATUS_NO_MEMORY;
2601 }
2602 }
2603
2604 talloc_free(mem_ctx);
2605 return NT_STATUS_OK;
2606 }
2607
2608 static bool enumprinters(TALLOC_CTX *mem_ctx, struct dcerpc_pipe *p,
2609 const char *servername, int level, int *num_printers)
2610 {
2611 struct spoolss_EnumPrinters r;
2612 NTSTATUS status;
2613 DATA_BLOB blob;
2614 uint32_t needed;
2615 uint32_t count;
2616 union spoolss_PrinterInfo *info;
2617
2618 r.in.flags = PRINTER_ENUM_LOCAL;
2619 r.in.server = talloc_asprintf(mem_ctx, "\\\\%s", servername);
2620 r.in.level = level;
2621 r.in.buffer = NULL;
2622 r.in.offered = 0;
2623 r.out.needed = &needed;
2624 r.out.count = &count;
2625 r.out.info = &info;
2626
2627 status = dcerpc_spoolss_EnumPrinters(p, mem_ctx, &r);
2628 if (!NT_STATUS_IS_OK(status)) {
2629 d_printf("(%s) dcerpc_spoolss_EnumPrinters failed: %s\n",
2630 __location__, nt_errstr(status));
2631 return false;
2632 }
2633
2634 if (!W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2635 d_printf("(%s) EnumPrinters unexpected return code %s, should "
2636 "be WERR_INSUFFICIENT_BUFFER\n", __location__,
2637 win_errstr(r.out.result));
2638 return false;
2639 }
2640
2641 blob = data_blob_talloc_zero(mem_ctx, needed);
2642 if (blob.data == NULL) {
2643 d_printf("(%s) data_blob_talloc failed\n", __location__);
2644 return false;
2645 }
2646
2647 r.in.buffer = &blob;
2648 r.in.offered = needed;
2649
2650 status = dcerpc_spoolss_EnumPrinters(p, mem_ctx, &r);
2651 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2652 d_printf("(%s) dcerpc_spoolss_EnumPrinters failed: %s, "
2653 "%s\n", __location__, nt_errstr(status),
2654 win_errstr(r.out.result));
2655 return false;
2656 }
2657
2658 *num_printers = count;
2659
2660 return true;
2661 }
2662
2663 static NTSTATUS getprinterinfo(TALLOC_CTX *ctx, struct dcerpc_pipe *p,
2664 struct policy_handle *handle, int level,
2665 union spoolss_PrinterInfo **res)
2666 {
2667 TALLOC_CTX *mem_ctx;
2668 struct spoolss_GetPrinter r;
2669 DATA_BLOB blob;
2670 NTSTATUS status;
2671 uint32_t needed;
2672
2673 mem_ctx = talloc_new(ctx);
2674 if (mem_ctx == NULL) {
2675 return NT_STATUS_NO_MEMORY;
2676 }
2677
2678 r.in.handle = handle;
2679 r.in.level = level;
2680 r.in.buffer = NULL;
2681 r.in.offered = 0;
2682 r.out.needed = &needed;
2683
2684 status = dcerpc_spoolss_GetPrinter(p, mem_ctx, &r);
2685 if (!NT_STATUS_IS_OK(status)) {
2686 d_printf("(%s) dcerpc_spoolss_GetPrinter failed: %s\n",
2687 __location__, nt_errstr(status));
2688 talloc_free(mem_ctx);
2689 return status;
2690 }
2691
2692 if (!W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2693 printf("GetPrinter unexpected return code %s, should "
2694 "be WERR_INSUFFICIENT_BUFFER\n",
2695 win_errstr(r.out.result));
2696 talloc_free(mem_ctx);
2697 return NT_STATUS_UNSUCCESSFUL;
2698 }
2699
2700 r.in.handle = handle;
2701 r.in.level = level;
2702 blob = data_blob_talloc(mem_ctx, NULL, needed);
2703 if (blob.data == NULL) {
2704 talloc_free(mem_ctx);
2705 return NT_STATUS_NO_MEMORY;
2706 }
2707 memset(blob.data, 0, blob.length);
2708 r.in.buffer = &blob;
2709 r.in.offered = needed;
2710
2711 status = dcerpc_spoolss_GetPrinter(p, mem_ctx, &r);
2712 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2713 d_printf("(%s) dcerpc_spoolss_GetPrinter failed: %s, "
2714 "%s\n", __location__, nt_errstr(status),
2715 win_errstr(r.out.result));
2716 talloc_free(mem_ctx);
2717 return NT_STATUS_IS_OK(status) ?
2718 NT_STATUS_UNSUCCESSFUL : status;
2719 }
2720
2721 if (res != NULL) {
2722 *res = talloc_steal(ctx, r.out.info);
2723 }
2724
2725 talloc_free(mem_ctx);
2726 return NT_STATUS_OK;
2727 }
2728
2729 bool torture_samba3_rpc_spoolss(struct torture_context *torture)
2730 {
2731 TALLOC_CTX *mem_ctx;
2732 bool ret = true;
2733 struct smbcli_state *cli;
2734 struct dcerpc_pipe *p;
2735 NTSTATUS status;
2736 struct policy_handle server_handle, printer_handle;
2737 const char **printers;
2738 int num_printers;
2739 struct spoolss_UserLevel1 userlevel1;
2740 char *servername;
2741
2742 if (!(mem_ctx = talloc_new(torture))) {
2743 return false;
2744 }
2745
2746 if (!(torture_open_connection_share(
2747 mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
2748 "IPC$", torture->ev))) {
2749 d_printf("IPC$ connection failed\n");
2750 talloc_free(mem_ctx);
2751 return false;
2752 }
2753
2754 status = get_servername(mem_ctx, cli->tree, lp_iconv_convenience(torture->lp_ctx), &servername);
2755 if (!NT_STATUS_IS_OK(status)) {
2756 d_fprintf(stderr, "(%s) get_servername returned %s\n",
2757 __location__, nt_errstr(status));
2758 talloc_free(mem_ctx);
2759 return false;
2760 }
2761
2762 if (!NT_STATUS_IS_OK(find_printers(mem_ctx, torture->lp_ctx, cli->tree,
2763 &printers, &num_printers))) {
2764 talloc_free(mem_ctx);
2765 return false;
2766 }
2767
2768 if (num_printers == 0) {
2769 d_printf("Did not find printers\n");
2770 talloc_free(mem_ctx);
2771 return true;
2772 }
2773
2774 status = pipe_bind_smb(mem_ctx, torture->lp_ctx, cli->tree, "\\spoolss",
2775 &ndr_table_spoolss, &p);
2776 if (!NT_STATUS_IS_OK(status)) {
2777 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2778 nt_errstr(status));
2779 talloc_free(mem_ctx);
2780 return false;
2781 }
2782
2783 ZERO_STRUCT(userlevel1);
2784 userlevel1.client = talloc_asprintf(
2785 mem_ctx, "\\\\%s", lp_netbios_name(torture->lp_ctx));
2786 userlevel1.user = cli_credentials_get_username(cmdline_credentials);
2787 userlevel1.build = 2600;
2788 userlevel1.major = 3;
2789 userlevel1.minor = 0;
2790 userlevel1.processor = 0;
2791
2792 {
2793 struct spoolss_OpenPrinterEx r;
2794
2795 ZERO_STRUCT(r);
2796 r.in.printername = talloc_asprintf(mem_ctx, "\\\\%s",
2797 servername);
2798 r.in.datatype = NULL;
2799 r.in.access_mask = 0;
2800 r.in.level = 1;
2801 r.in.userlevel.level1 = &userlevel1;
2802 r.out.handle = &server_handle;
2803
2804 status = dcerpc_spoolss_OpenPrinterEx(p, mem_ctx, &r);
2805 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2806 d_printf("(%s) dcerpc_spoolss_OpenPrinterEx failed: "
2807 "%s, %s\n", __location__, nt_errstr(status),
2808 win_errstr(r.out.result));
2809 talloc_free(mem_ctx);
2810 return false;
2811 }
2812 }
2813
2814 {
2815 struct spoolss_ClosePrinter r;
2816
2817 r.in.handle = &server_handle;
2818 r.out.handle = &server_handle;
2819
2820 status = dcerpc_spoolss_ClosePrinter(p, mem_ctx, &r);
2821 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2822 d_printf("(%s) dcerpc_spoolss_ClosePrinter failed: "
2823 "%s, %s\n", __location__, nt_errstr(status),
2824 win_errstr(r.out.result));
2825 talloc_free(mem_ctx);
2826 return false;
2827 }
2828 }
2829
2830 {
2831 struct spoolss_OpenPrinterEx r;
2832
2833 ZERO_STRUCT(r);
2834 r.in.printername = talloc_asprintf(
2835 mem_ctx, "\\\\%s\\%s", servername, printers[0]);
2836 r.in.datatype = NULL;
2837 r.in.access_mask = 0;
2838 r.in.level = 1;
2839 r.in.userlevel.level1 = &userlevel1;
2840 r.out.handle = &printer_handle;
2841
2842 status = dcerpc_spoolss_OpenPrinterEx(p, mem_ctx, &r);
2843 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2844 d_printf("(%s) dcerpc_spoolss_OpenPrinterEx failed: "
2845 "%s, %s\n", __location__, nt_errstr(status),
2846 win_errstr(r.out.result));
2847 talloc_free(mem_ctx);
2848 return false;
2849 }
2850 }
2851
2852 {
2853 int i;
2854
2855 for (i=0; i<8; i++) {
2856 status = getprinterinfo(mem_ctx, p, &printer_handle,
2857 i, NULL);
2858 if (!NT_STATUS_IS_OK(status)) {
2859 d_printf("(%s) getprinterinfo %d failed: %s\n",
2860 __location__, i, nt_errstr(status));
2861 ret = false;
2862 }
2863 }
2864 }
2865
2866 {
2867 struct spoolss_ClosePrinter r;
2868
2869 r.in.handle = &printer_handle;
2870 r.out.handle = &printer_handle;
2871
2872 status = dcerpc_spoolss_ClosePrinter(p, mem_ctx, &r);
2873 if (!NT_STATUS_IS_OK(status)) {
2874 d_printf("(%s) dcerpc_spoolss_ClosePrinter failed: "
2875 "%s\n", __location__, nt_errstr(status));
2876 talloc_free(mem_ctx);
2877 return false;
2878 }
2879 }
2880
2881 {
2882 int num_enumerated;
2883 if (!enumprinters(mem_ctx, p, servername, 1,
2884 &num_enumerated)) {
2885 d_printf("(%s) enumprinters failed\n", __location__);
2886 talloc_free(mem_ctx);
2887 return false;
2888 }
2889 if (num_printers != num_enumerated) {
2890 d_printf("(%s) netshareenum gave %d printers, "
2891 "enumprinters lvl 1 gave %d\n", __location__,
2892 num_printers, num_enumerated);
2893 talloc_free(mem_ctx);
2894 return false;
2895 }
2896 }
2897
2898 {
2899 int num_enumerated;
2900 if (!enumprinters(mem_ctx, p, servername, 2,
2901 &num_enumerated)) {
2902 d_printf("(%s) enumprinters failed\n", __location__);
2903 talloc_free(mem_ctx);
2904 return false;
2905 }
2906 if (num_printers != num_enumerated) {
2907 d_printf("(%s) netshareenum gave %d printers, "
2908 "enumprinters lvl 2 gave %d\n", __location__,
2909 num_printers, num_enumerated);
2910 talloc_free(mem_ctx);
2911 return false;
2912 }
2913 }
2914
2915 talloc_free(mem_ctx);
2916
2917 return ret;
2918 }
2919
2920 bool torture_samba3_rpc_wkssvc(struct torture_context *torture)
2921 {
2922 TALLOC_CTX *mem_ctx;
2923 struct smbcli_state *cli;
2924 struct dcerpc_pipe *p;
2925 NTSTATUS status;
2926 char *servername;
2927
2928 if (!(mem_ctx = talloc_new(torture))) {
2929 return false;
2930 }
2931
2932 if (!(torture_open_connection_share(
2933 mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
2934 "IPC$", torture->ev))) {
2935 d_printf("IPC$ connection failed\n");
2936 talloc_free(mem_ctx);
2937 return false;
2938 }
2939
2940 status = get_servername(mem_ctx, cli->tree, lp_iconv_convenience(torture->lp_ctx), &servername);
2941 if (!NT_STATUS_IS_OK(status)) {
2942 d_fprintf(stderr, "(%s) get_servername returned %s\n",
2943 __location__, nt_errstr(status));
2944 talloc_free(mem_ctx);
2945 return false;
2946 }
2947
2948 status = pipe_bind_smb(mem_ctx, torture->lp_ctx, cli->tree, "\\wkssvc",
2949 &ndr_table_wkssvc, &p);
2950 if (!NT_STATUS_IS_OK(status)) {
2951 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2952 nt_errstr(status));
2953 talloc_free(mem_ctx);
2954 return false;
2955 }
2956
2957 {
2958 struct wkssvc_NetWkstaInfo100 wks100;
2959 union wkssvc_NetWkstaInfo info;
2960 struct wkssvc_NetWkstaGetInfo r;
2961
2962 r.in.server_name = "\\foo";
2963 r.in.level = 100;
2964 info.info100 = &wks100;
2965 r.out.info = &info;
2966
2967 status = dcerpc_wkssvc_NetWkstaGetInfo(p, mem_ctx, &r);
2968 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2969 d_printf("(%s) dcerpc_wkssvc_NetWksGetInfo failed: "
2970 "%s, %s\n", __location__, nt_errstr(status),
2971 win_errstr(r.out.result));
2972 talloc_free(mem_ctx);
2973 return false;
2974 }
2975
2976 if (strcmp(servername,
2977 r.out.info->info100->server_name) != 0) {
2978 d_printf("(%s) servername inconsistency: RAP=%s, "
2979 "dcerpc_wkssvc_NetWksGetInfo=%s",
2980 __location__, servername,
2981 r.out.info->info100->server_name);
2982 talloc_free(mem_ctx);
2983 return false;
2984 }
2985 }
2986
2987 talloc_free(mem_ctx);
2988 return true;
2989 }
2990
2991 static NTSTATUS winreg_close(struct dcerpc_pipe *p,
2992 struct policy_handle *handle)
2993 {
2994 struct winreg_CloseKey c;
2995 NTSTATUS status;
2996 TALLOC_CTX *mem_ctx;
2997
2998 c.in.handle = c.out.handle = handle;
2999
3000 if (!(mem_ctx = talloc_new(p))) {
3001 return NT_STATUS_NO_MEMORY;
3002 }
3003
3004 status = dcerpc_winreg_CloseKey(p, mem_ctx, &c);
3005 talloc_free(mem_ctx);
3006
3007 if (!NT_STATUS_IS_OK(status)) {
3008 return status;
3009 }
3010
3011 if (!W_ERROR_IS_OK(c.out.result)) {
3012 return werror_to_ntstatus(c.out.result);
3013 }
3014
3015 return NT_STATUS_OK;
3016 }
3017
3018 static NTSTATUS enumvalues(struct dcerpc_pipe *p, struct policy_handle *handle,
3019 TALLOC_CTX *mem_ctx)
3020 {
3021 uint32_t enum_index = 0;
3022
3023 while (1) {
3024 struct winreg_EnumValue r;
3025 struct winreg_ValNameBuf name;
3026 enum winreg_Type type = 0;
3027 uint8_t buf8[1024];
3028 NTSTATUS status;
3029 uint32_t size, length;
3030
3031 r.in.handle = handle;
3032 r.in.enum_index = enum_index;
3033 name.name = "";
3034 name.size = 1024;
3035 r.in.name = r.out.name = &name;
3036 size = 1024;
3037 length = 5;
3038 r.in.type = &type;
3039 r.in.value = buf8;
3040 r.in.size = &size;
3041 r.in.length = &length;
3042
3043 status = dcerpc_winreg_EnumValue(p, mem_ctx, &r);
3044 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
3045 return NT_STATUS_OK;
3046 }
3047 enum_index += 1;
3048 }
3049 }
3050
3051 static NTSTATUS enumkeys(struct dcerpc_pipe *p, struct policy_handle *handle,
3052 TALLOC_CTX *mem_ctx, int depth)
3053 {
3054 struct winreg_EnumKey r;
3055 struct winreg_StringBuf kclass, name;
3056 NTSTATUS status;
3057 NTTIME t = 0;
3058
3059 if (depth <= 0) {
3060 return NT_STATUS_OK;
3061 }
3062
3063 kclass.name = "";
3064 kclass.size = 1024;
3065
3066 r.in.handle = handle;
3067 r.in.enum_index = 0;
3068 r.in.name = &name;
3069 r.in.keyclass = &kclass;
3070 r.out.name = &name;
3071 r.in.last_changed_time = &t;
3072
3073 do {
3074 TALLOC_CTX *tmp_ctx;
3075 struct winreg_OpenKey o;
3076 struct policy_handle key_handle;
3077 int i;
3078
3079 if (!(tmp_ctx = talloc_new(mem_ctx))) {
3080 return NT_STATUS_NO_MEMORY;
3081 }
3082
3083 name.name = NULL;
3084 name.size = 1024;
3085
3086 status = dcerpc_winreg_EnumKey(p, tmp_ctx, &r);
3087 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
3088 /* We're done enumerating */
3089 talloc_free(tmp_ctx);
3090 return NT_STATUS_OK;
3091 }
3092
3093 for (i=0; i<10-depth; i++)
3094 printf(" ");
3095 printf("%s\n", r.out.name->name);
3096
3097
3098 o.in.parent_handle = handle;
3099 o.in.keyname.name = r.out.name->name;
3100 o.in.unknown = 0;
3101 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3102 o.out.handle = &key_handle;
3103
3104 status = dcerpc_winreg_OpenKey(p, tmp_ctx, &o);
3105 if (NT_STATUS_IS_OK(status) && W_ERROR_IS_OK(o.out.result)) {
3106 enumkeys(p, &key_handle, tmp_ctx, depth-1);
3107 enumvalues(p, &key_handle, tmp_ctx);
3108 status = winreg_close(p, &key_handle);
3109 if (!NT_STATUS_IS_OK(status)) {
3110 return status;
3111 }
3112 }
3113
3114 talloc_free(tmp_ctx);
3115
3116 r.in.enum_index += 1;
3117 } while(true);
3118
3119 return NT_STATUS_OK;
3120 }
3121
3122 typedef NTSTATUS (*winreg_open_fn)(struct dcerpc_pipe *, TALLOC_CTX *, void *);
3123
3124 static bool test_Open3(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
3125 const char *name, winreg_open_fn open_fn)
3126 {
3127 struct policy_handle handle;
3128 struct winreg_OpenHKLM r;
3129 NTSTATUS status;
3130
3131 r.in.system_name = 0;
3132 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3133 r.out.handle = &handle;
3134
3135 status = open_fn(p, mem_ctx, &r);
3136 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
3137 d_printf("(%s) %s failed: %s, %s\n", __location__, name,
3138 nt_errstr(status), win_errstr(r.out.result));
3139 return false;
3140 }
3141
3142 enumkeys(p, &handle, mem_ctx, 4);
3143
3144 status = winreg_close(p, &handle);
3145 if (!NT_STATUS_IS_OK(status)) {
3146 d_printf("(%s) dcerpc_CloseKey failed: %s\n",
3147 __location__, nt_errstr(status));
3148 return false;
3149 }
3150
3151 return true;
3152 }
3153
3154 bool torture_samba3_rpc_winreg(struct torture_context *torture)
3155 {
3156 NTSTATUS status;
3157 struct dcerpc_pipe *p;
3158 TALLOC_CTX *mem_ctx;
3159 bool ret = true;
3160 struct {
3161 const char *name;
3162 winreg_open_fn fn;
3163 } open_fns[] = {
3164 {"OpenHKLM", (winreg_open_fn)dcerpc_winreg_OpenHKLM },
3165 {"OpenHKU", (winreg_open_fn)dcerpc_winreg_OpenHKU },
3166 {"OpenHKPD", (winreg_open_fn)dcerpc_winreg_OpenHKPD },
3167 {"OpenHKPT", (winreg_open_fn)dcerpc_winreg_OpenHKPT },
3168 {"OpenHKCR", (winreg_open_fn)dcerpc_winreg_OpenHKCR }};
3169 #if 0
3170 int i;
3171 #endif
3172
3173 mem_ctx = talloc_init("torture_rpc_winreg");
3174
3175 status = torture_rpc_connection(torture, &p, &ndr_table_winreg);
3176
3177 if (!NT_STATUS_IS_OK(status)) {
3178 talloc_free(mem_ctx);
3179 return false;
3180 }
3181
3182 #if 1
3183 ret = test_Open3(p, mem_ctx, open_fns[0].name, open_fns[0].fn);
3184 #else
3185 for (i = 0; i < ARRAY_SIZE(open_fns); i++) {
3186 if (!test_Open3(p, mem_ctx, open_fns[i].name, open_fns[i].fn))
3187 ret = false;
3188 }
3189 #endif
3190
3191 talloc_free(mem_ctx);
3192
3193 return ret;
3194 }
3195
3196 static NTSTATUS get_shareinfo(TALLOC_CTX *mem_ctx,
3197 struct loadparm_context *lp_ctx,
3198 struct smbcli_state *cli,
3199 const char *share,
3200 struct srvsvc_NetShareInfo502 **info502)
3201 {
3202 struct smbcli_tree *ipc;
3203 struct dcerpc_pipe *p;
3204 struct srvsvc_NetShareGetInfo r;
3205 union srvsvc_NetShareInfo info;
3206 NTSTATUS status;
3207
3208 if (!(p = dcerpc_pipe_init(cli,
3209 cli->transport->socket->event.ctx,
3210 lp_iconv_convenience(lp_ctx)))) {
3211 status = NT_STATUS_NO_MEMORY;
3212 goto fail;
3213 }
3214
3215 status = secondary_tcon(p, cli->session, "IPC$", &ipc);
3216 if (!NT_STATUS_IS_OK(status)) {
3217 goto fail;
3218 }
3219
3220 status = dcerpc_pipe_open_smb(p, ipc, "\\pipe\\srvsvc");
3221 if (!NT_STATUS_IS_OK(status)) {
3222 d_printf("dcerpc_pipe_open_smb failed: %s\n",
3223 nt_errstr(status));
3224 goto fail;
3225 }
3226
3227 status = dcerpc_bind_auth_none(p, &ndr_table_srvsvc);
3228 if (!NT_STATUS_IS_OK(status)) {
3229 d_printf("dcerpc_bind_auth_none failed: %s\n",
3230 nt_errstr(status));
3231 goto fail;
3232 }
3233
3234 r.in.server_unc = talloc_asprintf(mem_ctx, "\\\\%s",
3235 dcerpc_server_name(p));
3236 r.in.share_name = share;
3237 r.in.level = 502;
3238 r.out.info = &info;
3239
3240 status = dcerpc_srvsvc_NetShareGetInfo(p, p, &r);
3241 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
3242 d_printf("(%s) srvsvc_NetShareGetInfo failed: %s, %s\n", __location__,
3243 nt_errstr(status), win_errstr(r.out.result));
3244 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(r.out.result) : status;
3245 goto fail;
3246 }
3247
3248 *info502 = talloc_move(mem_ctx, &info.info502);
3249 return NT_STATUS_OK;
3250
3251 fail:
3252 talloc_free(p);
3253 return status;
3254 }
3255
3256 /*
3257 * Get us a handle on HKLM\
3258 */
3259
3260 static NTSTATUS get_hklm_handle(TALLOC_CTX *mem_ctx,
3261 struct smbcli_state *cli,
3262 struct smb_iconv_convenience *iconv_convenience,
3263 struct dcerpc_pipe **pipe_p,
3264 struct policy_handle **handle)
3265 {
3266 struct smbcli_tree *ipc;
3267 struct dcerpc_pipe *p;
3268 struct winreg_OpenHKLM r;
3269 NTSTATUS status;
3270 struct policy_handle *result;
3271
3272 result = talloc(mem_ctx, struct policy_handle);
3273
3274 if (result == NULL) {
3275 return NT_STATUS_NO_MEMORY;
3276 }
3277
3278 if (!(p = dcerpc_pipe_init(result,
3279 cli->transport->socket->event.ctx,
3280 iconv_convenience))) {
3281 status = NT_STATUS_NO_MEMORY;
3282 goto fail;
3283 }
3284
3285 status = secondary_tcon(p, cli->session, "IPC$", &ipc);
3286 if (!NT_STATUS_IS_OK(status)) {
3287 goto fail;
3288 }
3289
3290 status = dcerpc_pipe_open_smb(p, ipc, "\\winreg");
3291 if (!NT_STATUS_IS_OK(status)) {
3292 d_printf("dcerpc_pipe_open_smb failed: %s\n",
3293 nt_errstr(status));
3294 goto fail;
3295 }
3296
3297 status = dcerpc_bind_auth_none(p, &ndr_table_winreg);
3298 if (!NT_STATUS_IS_OK(status)) {
3299 d_printf("dcerpc_bind_auth_none failed: %s\n",
3300 nt_errstr(status));
3301 goto fail;
3302 }
3303
3304 r.in.system_name = 0;
3305 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3306 r.out.handle = result;
3307
3308 status = dcerpc_winreg_OpenHKLM(p, p, &r);
3309 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
3310 d_printf("(%s) OpenHKLM failed: %s, %s\n", __location__,
3311 nt_errstr(status), win_errstr(r.out.result));
3312 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(r.out.result) : status;
3313 goto fail;
3314 }
3315
3316 *pipe_p = p;
3317 *handle = result;
3318 return NT_STATUS_OK;
3319
3320 fail:
3321 talloc_free(result);
3322 return status;
3323 }
3324
3325 static NTSTATUS torture_samba3_createshare(struct smbcli_state *cli,
3326 struct smb_iconv_convenience *iconv_convenience,
3327 const char *sharename)
3328 {
3329 struct dcerpc_pipe *p;
3330 struct policy_handle *hklm = NULL;
3331 struct policy_handle new_handle;
3332 struct winreg_CreateKey c;
3333 struct winreg_CloseKey cl;
3334 enum winreg_CreateAction action_taken;
3335 NTSTATUS status;
3336 TALLOC_CTX *mem_ctx;
3337
3338 mem_ctx = talloc_new(cli);
3339 NT_STATUS_HAVE_NO_MEMORY(mem_ctx);
3340
3341 status = get_hklm_handle(mem_ctx, cli, iconv_convenience, &p, &hklm);
3342 if (!NT_STATUS_IS_OK(status)) {
3343 d_printf("get_hklm_handle failed: %s\n", nt_errstr(status));
3344 goto fail;
3345 }
3346
3347 c.in.handle = hklm;
3348 c.in.name.name = talloc_asprintf(
3349 mem_ctx, "software\\samba\\smbconf\\%s", sharename);
3350 if (c.in.name.name == NULL) {
3351 d_printf("talloc_asprintf failed\n");
3352 status = NT_STATUS_NO_MEMORY;
3353 goto fail;
3354 }
3355 c.in.keyclass.name = "";
3356 c.in.options = 0;
3357 c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3358 c.in.secdesc = NULL;
3359 c.in.action_taken = &action_taken;
3360 c.out.new_handle = &new_handle;
3361 c.out.action_taken = &action_taken;
3362
3363 status = dcerpc_winreg_CreateKey(p, p, &c);
3364 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(c.out.result)) {
3365 d_printf("(%s) OpenKey failed: %s, %s\n", __location__,
3366 nt_errstr(status), win_errstr(c.out.result));
3367 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(c.out.result) : status;
3368 goto fail;
3369 }
3370
3371 cl.in.handle = &new_handle;
3372 cl.out.handle = &new_handle;
3373 status = dcerpc_winreg_CloseKey(p, p, &cl);
3374 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(cl.out.result)) {
3375 d_printf("(%s) OpenKey failed: %s, %s\n", __location__,
3376 nt_errstr(status), win_errstr(cl.out.result));
3377 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(cl.out.result) : status;
3378 goto fail;
3379 }
3380
3381
3382 fail:
3383 talloc_free(mem_ctx);
3384 return status;
3385 }
3386
3387 static NTSTATUS torture_samba3_deleteshare(struct torture_context *torture,
3388 struct smbcli_state *cli,
3389 const char *sharename)
3390 {
3391 struct dcerpc_pipe *p;
3392 struct policy_handle *hklm = NULL;
3393 struct winreg_DeleteKey d;
3394 NTSTATUS status;
3395 TALLOC_CTX *mem_ctx;
3396
3397 mem_ctx = talloc_new(cli);
3398 NT_STATUS_HAVE_NO_MEMORY(mem_ctx);
3399
3400 status = get_hklm_handle(cli, cli, lp_iconv_convenience(torture->lp_ctx),
3401 &p, &hklm);
3402 if (!NT_STATUS_IS_OK(status)) {
3403 d_printf("get_hklm_handle failed: %s\n", nt_errstr(status));
3404 goto fail;
3405 }
3406
3407 d.in.handle = hklm;
3408 d.in.key.name = talloc_asprintf(
3409 mem_ctx, "software\\samba\\smbconf\\%s", sharename);
3410 if (d.in.key.name == NULL) {
3411 d_printf("talloc_asprintf failed\n");
3412 status = NT_STATUS_NO_MEMORY;
3413 goto fail;
3414 }
3415 status = dcerpc_winreg_DeleteKey(p, p, &d);
3416 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(d.out.result)) {
3417 d_printf("(%s) OpenKey failed: %s, %s\n", __location__,
3418 nt_errstr(status), win_errstr(d.out.result));
3419 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(d.out.result) : status;
3420 goto fail;
3421 }
3422
3423 fail:
3424 talloc_free(mem_ctx);
3425 return status;
3426 }
3427
3428 static NTSTATUS torture_samba3_setconfig(struct smbcli_state *cli,
3429 struct loadparm_context *lp_ctx,
3430 const char *sharename,
3431 const char *parameter,
3432 const char *value)
3433 {
3434 struct dcerpc_pipe *p = NULL;
3435 struct policy_handle *hklm = NULL, key_handle;
3436 struct winreg_OpenKey o;
3437 struct winreg_SetValue s;
3438 uint32_t type;
3439 DATA_BLOB val;
3440 NTSTATUS status;
3441
3442 status = get_hklm_handle(cli, cli, lp_iconv_convenience(lp_ctx), &p, &hklm);
3443 if (!NT_STATUS_IS_OK(status)) {
3444 d_printf("get_hklm_handle failed: %s\n", nt_errstr(status));
3445 return status;;
3446 }
3447
3448 o.in.parent_handle = hklm;
3449 o.in.keyname.name = talloc_asprintf(
3450 hklm, "software\\samba\\smbconf\\%s", sharename);
3451 if (o.in.keyname.name == NULL) {
3452 d_printf("talloc_asprintf failed\n");
3453 status = NT_STATUS_NO_MEMORY;
3454 goto done;
3455 }
3456 o.in.unknown = 0;
3457 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3458 o.out.handle = &key_handle;
3459
3460 status = dcerpc_winreg_OpenKey(p, p, &o);
3461 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(o.out.result)) {
3462 d_printf("(%s) OpenKey failed: %s, %s\n", __location__,
3463 nt_errstr(status), win_errstr(o.out.result));
3464 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(o.out.result) : status;
3465 goto done;
3466 }
3467
3468 if (!reg_string_to_val(hklm, lp_iconv_convenience(lp_ctx), "REG_SZ",
3469 value, &type, &val)) {
3470 d_printf("(%s) reg_string_to_val failed\n", __location__);
3471 goto done;
3472 }
3473
3474 s.in.handle = &key_handle;
3475 s.in.name.name = parameter;
3476 s.in.type = type;
3477 s.in.data = val.data;
3478 s.in.size = val.length;
3479
3480 status = dcerpc_winreg_SetValue(p, p, &s);
3481 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(s.out.result)) {
3482 d_printf("(%s) SetValue failed: %s, %s\n", __location__,
3483 nt_errstr(status), win_errstr(s.out.result));
3484 status = NT_STATUS_IS_OK(status) ? werror_to_ntstatus(s.out.result) : status;
3485 goto done;
3486 }
3487
3488 done:
3489 talloc_free(hklm);
3490 return status;
3491 }
3492
3493 bool torture_samba3_regconfig(struct torture_context *torture)
3494 {
3495 struct smbcli_state *cli;
3496 struct srvsvc_NetShareInfo502 *i = NULL;
3497 NTSTATUS status;
3498 bool ret = false;
3499 const char *comment = "Dummer Kommentar";
3500
3501 if (!(torture_open_connection(&cli, torture, 0))) {
3502 return false;
3503 }
3504
3505 status = torture_samba3_createshare(cli, lp_iconv_convenience(torture->lp_ctx), "blubber");
3506 if (!NT_STATUS_IS_OK(status)) {
3507 torture_warning(torture, "torture_samba3_createshare failed: "
3508 "%s\n", nt_errstr(status));
3509 goto done;
3510 }
3511
3512 status = torture_samba3_setconfig(cli, torture->lp_ctx, "blubber", "comment", comment);
3513 if (!NT_STATUS_IS_OK(status)) {
3514 torture_warning(torture, "torture_samba3_setconfig failed: "
3515 "%s\n", nt_errstr(status));
3516 goto done;
3517 }
3518
3519 status = get_shareinfo(torture, torture->lp_ctx, cli, "blubber", &i);
3520 if (!NT_STATUS_IS_OK(status)) {
3521 torture_warning(torture, "get_shareinfo failed: "
3522 "%s\n", nt_errstr(status));
3523 goto done;
3524 }
3525
3526 if (strcmp(comment, i->comment) != 0) {
3527 torture_warning(torture, "Expected comment [%s], got [%s]\n",
3528 comment, i->comment);
3529 goto done;
3530 }
3531
3532 status = torture_samba3_deleteshare(torture, cli, "blubber");
3533 if (!NT_STATUS_IS_OK(status)) {
3534 torture_warning(torture, "torture_samba3_deleteshare failed: "
3535 "%s\n", nt_errstr(status));
3536 goto done;
3537 }
3538
3539 ret = true;
3540 done:
3541 talloc_free(cli);
3542 return ret;
3543 }
3544
3545 /*
3546 * Test that even with a result of 0 rids the array is returned as a
3547 * non-NULL pointer. Yes, XP does notice.
3548 */
3549
3550 bool torture_samba3_getaliasmembership_0(struct torture_context *torture)
3551 {
3552 struct dcerpc_pipe *p;
3553 NTSTATUS status;
3554 struct samr_Connect2 c;
3555 struct samr_OpenDomain o;
3556 struct dom_sid sid;
3557 struct lsa_SidPtr ptr;
3558 struct lsa_SidArray sids;
3559 struct samr_GetAliasMembership g;
3560 struct samr_Ids rids;
3561 struct policy_handle samr, domain;
3562
3563 status = torture_rpc_connection(torture, &p, &ndr_table_samr);
3564 if (!NT_STATUS_IS_OK(status)) {
3565 return false;
3566 }
3567 c.in.system_name = NULL;
3568 c.in.access_mask = SAMR_ACCESS_LOOKUP_DOMAIN;
3569 c.out.connect_handle = &samr;
3570 status = dcerpc_samr_Connect2(p, torture, &c);
3571 if (!NT_STATUS_IS_OK(status)) {
3572 return false;
3573 }
3574 dom_sid_parse("S-1-5-32", &sid);
3575 o.in.connect_handle = &samr;
3576 o.in.access_mask = SAMR_DOMAIN_ACCESS_LOOKUP_ALIAS;
3577 o.in.sid = &sid;
3578 o.out.domain_handle = &domain;
3579 status = dcerpc_samr_OpenDomain(p, torture, &o);
3580 if (!NT_STATUS_IS_OK(status)) {
3581 return false;
3582 }
3583 dom_sid_parse("S-1-2-3-4-5", &sid);
3584 ptr.sid = &sid;
3585 sids.num_sids = 1;
3586 sids.sids = &ptr;
3587 g.in.domain_handle = &domain;
3588 g.in.sids = &sids;
3589 g.out.rids = &rids;
3590 status = dcerpc_samr_GetAliasMembership(p, torture, &g);
3591 if (!NT_STATUS_IS_OK(status)) {
3592 return false;
3593 }
3594 if (rids.ids == NULL) {
3595 /* This is the piece to test here */
3596 torture_warning(torture,
3597 "torture_samba3_getaliasmembership_0: "
3598 "Server returns NULL rids array\n");
3599 return false;
3600 }
3601 return true;
3602 }
Crístian's Samba branch