0098dcf648db1134f73e8f81b5797efc56e9b10c
[samba.git] / source4 / torture / rpc / schannel.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    test suite for schannel operations
5
6    Copyright (C) Andrew Tridgell 2004
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 "librpc/gen_ndr/ndr_netlogon_c.h"
24 #include "librpc/gen_ndr/ndr_lsa_c.h"
25 #include "librpc/gen_ndr/ndr_samr_c.h"
26 #include "auth/credentials/credentials.h"
27 #include "torture/rpc/torture_rpc.h"
28 #include "lib/cmdline/popt_common.h"
29 #include "../libcli/auth/schannel.h"
30 #include "libcli/auth/libcli_auth.h"
31 #include "libcli/security/security.h"
32 #include "system/filesys.h"
33 #include "param/param.h"
34 #include "librpc/rpc/dcerpc_proto.h"
35 #include "libcli/composite/composite.h"
36 #include "lib/events/events.h"
37
38 #define TEST_MACHINE_NAME "schannel"
39
40 /*
41   try a netlogon SamLogon
42 */
43 bool test_netlogon_ex_ops(struct dcerpc_pipe *p, struct torture_context *tctx,
44                           struct cli_credentials *credentials,
45                           struct netlogon_creds_CredentialState *creds)
46 {
47         NTSTATUS status;
48         struct netr_LogonSamLogonEx r;
49         struct netr_NetworkInfo ninfo;
50         union netr_LogonLevel logon;
51         union netr_Validation validation;
52         uint8_t authoritative = 0;
53         uint32_t _flags = 0;
54         DATA_BLOB names_blob, chal, lm_resp, nt_resp;
55         int i;
56         int flags = CLI_CRED_NTLM_AUTH;
57         struct dcerpc_binding_handle *b = p->binding_handle;
58
59         struct netr_UserSessionKey key;
60         struct netr_LMSessionKey LMSessKey;
61         uint32_t validation_levels[] = { 2, 3 };
62         struct netr_SamBaseInfo *base;
63         const char *crypto_alg = "";
64         bool can_do_validation_6 = true;
65
66         if (lpcfg_client_lanman_auth(tctx->lp_ctx)) {
67                 flags |= CLI_CRED_LANMAN_AUTH;
68         }
69
70         if (lpcfg_client_ntlmv2_auth(tctx->lp_ctx)) {
71                 flags |= CLI_CRED_NTLMv2_AUTH;
72         }
73
74         cli_credentials_get_ntlm_username_domain(cmdline_credentials, tctx,
75                                                  &ninfo.identity_info.account_name.string,
76                                                  &ninfo.identity_info.domain_name.string);
77
78         generate_random_buffer(ninfo.challenge,
79                                sizeof(ninfo.challenge));
80         chal = data_blob_const(ninfo.challenge,
81                                sizeof(ninfo.challenge));
82
83         names_blob = NTLMv2_generate_names_blob(tctx, cli_credentials_get_workstation(credentials),
84                                                 cli_credentials_get_domain(credentials));
85
86         status = cli_credentials_get_ntlm_response(cmdline_credentials, tctx,
87                                                    &flags,
88                                                    chal,
89                                                    names_blob,
90                                                    &lm_resp, &nt_resp,
91                                                    NULL, NULL);
92         torture_assert_ntstatus_ok(tctx, status,
93                                    "cli_credentials_get_ntlm_response failed");
94
95         ninfo.lm.data = lm_resp.data;
96         ninfo.lm.length = lm_resp.length;
97
98         ninfo.nt.data = nt_resp.data;
99         ninfo.nt.length = nt_resp.length;
100
101         ninfo.identity_info.parameter_control = 0;
102         ninfo.identity_info.logon_id_low = 0;
103         ninfo.identity_info.logon_id_high = 0;
104         ninfo.identity_info.workstation.string = cli_credentials_get_workstation(credentials);
105
106         logon.network = &ninfo;
107
108         r.in.server_name = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
109         r.in.computer_name = cli_credentials_get_workstation(credentials);
110         r.in.logon_level = NetlogonNetworkInformation;
111         r.in.logon= &logon;
112         r.in.flags = &_flags;
113         r.out.validation = &validation;
114         r.out.authoritative = &authoritative;
115         r.out.flags = &_flags;
116
117         /*
118         - retrieve level6
119         - save usrsession and lmsession key
120         - retrieve level 2
121         - calculate, compare
122         - retrieve level 3
123         - calculate, compare
124         */
125
126         if (creds) {
127                 if (creds->negotiate_flags & NETLOGON_NEG_SUPPORTS_AES) {
128                         crypto_alg = "AES";
129                 } else if (creds->negotiate_flags & NETLOGON_NEG_ARCFOUR) {
130                         crypto_alg = "ARCFOUR";
131                 }
132         }
133
134         r.in.validation_level = 6;
135
136         torture_comment(tctx,
137                         "Testing LogonSamLogonEx with name %s using %s and validation_level: %d\n",
138                         ninfo.identity_info.account_name.string, crypto_alg,
139                         r.in.validation_level);
140
141         torture_assert_ntstatus_ok(tctx,
142                 dcerpc_netr_LogonSamLogonEx_r(b, tctx, &r),
143                 "LogonSamLogonEx failed");
144         if (NT_STATUS_EQUAL(r.out.result, NT_STATUS_INVALID_INFO_CLASS)) {
145                 can_do_validation_6 = false;
146         } else {
147                 torture_assert_ntstatus_ok(tctx, r.out.result,
148                         "LogonSamLogonEx failed");
149
150                 key = r.out.validation->sam6->base.key;
151                 LMSessKey = r.out.validation->sam6->base.LMSessKey;
152
153                 DEBUG(1,("unencrypted session keys from validation_level 6:\n"));
154                 dump_data(1, r.out.validation->sam6->base.key.key, 16);
155                 dump_data(1, r.out.validation->sam6->base.LMSessKey.key, 8);
156         }
157
158         for (i=0; i < ARRAY_SIZE(validation_levels); i++) {
159
160                 r.in.validation_level = validation_levels[i];
161
162                 torture_comment(tctx,
163                         "Testing LogonSamLogonEx with name %s using %s and validation_level: %d\n",
164                         ninfo.identity_info.account_name.string, crypto_alg,
165                         r.in.validation_level);
166
167                 torture_assert_ntstatus_ok(tctx,
168                         dcerpc_netr_LogonSamLogonEx_r(b, tctx, &r),
169                         "LogonSamLogonEx failed");
170                 torture_assert_ntstatus_ok(tctx, r.out.result,
171                         "LogonSamLogonEx failed");
172
173                 if (creds == NULL) {
174                         /* when this test is called without creds no point in
175                          * testing the session keys */
176                         continue;
177                 }
178
179                 switch (validation_levels[i]) {
180                 case 2:
181                         base = &r.out.validation->sam2->base;
182                         break;
183                 case 3:
184                         base = &r.out.validation->sam3->base;
185                         break;
186                 default:
187                         break;
188                 }
189
190                 DEBUG(1,("encrypted keys validation_level %d:\n",
191                         validation_levels[i]));
192                 dump_data(1, base->key.key, 16);
193                 dump_data(1, base->LMSessKey.key, 8);
194
195                 if (creds->negotiate_flags & NETLOGON_NEG_SUPPORTS_AES) {
196                         netlogon_creds_aes_decrypt(creds, base->key.key, 16);
197                         netlogon_creds_aes_decrypt(creds, base->LMSessKey.key, 8);
198                 } else if (creds->negotiate_flags & NETLOGON_NEG_ARCFOUR) {
199                         netlogon_creds_arcfour_crypt(creds, base->key.key, 16);
200                         netlogon_creds_arcfour_crypt(creds, base->LMSessKey.key, 8);
201                 }
202
203                 DEBUG(1,("decryped keys validation_level %d\n",
204                         validation_levels[i]));
205
206                 dump_data(1, base->key.key, 16);
207                 dump_data(1, base->LMSessKey.key, 8);
208
209                 if (!can_do_validation_6) {
210                         /* we cant compare against unencrypted keys */
211                         continue;
212                 }
213
214                 torture_assert_mem_equal(tctx,
215                                          base->key.key,
216                                          key.key,
217                                          16,
218                                          "unexpected user session key\n");
219                 torture_assert_mem_equal(tctx,
220                                          base->LMSessKey.key,
221                                          LMSessKey.key,
222                                          8,
223                                          "unexpected LM session key\n");
224         }
225
226         return true;
227 }
228
229 /*
230   do some samr ops using the schannel connection
231  */
232 static bool test_samr_ops(struct torture_context *tctx,
233                           struct dcerpc_binding_handle *b)
234 {
235         struct samr_GetDomPwInfo r;
236         struct samr_PwInfo info;
237         struct samr_Connect connect_r;
238         struct samr_OpenDomain opendom;
239         int i;
240         struct lsa_String name;
241         struct policy_handle handle;
242         struct policy_handle domain_handle;
243
244         name.string = lpcfg_workgroup(tctx->lp_ctx);
245         r.in.domain_name = &name;
246         r.out.info = &info;
247
248         connect_r.in.system_name = 0;
249         connect_r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
250         connect_r.out.connect_handle = &handle;
251
252         torture_comment(tctx, "Testing Connect and OpenDomain on BUILTIN\n");
253
254         torture_assert_ntstatus_ok(tctx, dcerpc_samr_Connect_r(b, tctx, &connect_r),
255                 "Connect failed");
256         if (!NT_STATUS_IS_OK(connect_r.out.result)) {
257                 if (NT_STATUS_EQUAL(connect_r.out.result, NT_STATUS_ACCESS_DENIED)) {
258                         torture_comment(tctx, "Connect failed (expected, schannel mapped to anonymous): %s\n",
259                                nt_errstr(connect_r.out.result));
260                 } else {
261                         torture_comment(tctx, "Connect failed - %s\n", nt_errstr(connect_r.out.result));
262                         return false;
263                 }
264         } else {
265                 opendom.in.connect_handle = &handle;
266                 opendom.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
267                 opendom.in.sid = dom_sid_parse_talloc(tctx, "S-1-5-32");
268                 opendom.out.domain_handle = &domain_handle;
269
270                 torture_assert_ntstatus_ok(tctx, dcerpc_samr_OpenDomain_r(b, tctx, &opendom),
271                         "OpenDomain failed");
272                 if (!NT_STATUS_IS_OK(opendom.out.result)) {
273                         torture_comment(tctx, "OpenDomain failed - %s\n", nt_errstr(opendom.out.result));
274                         return false;
275                 }
276         }
277
278         torture_comment(tctx, "Testing GetDomPwInfo with name %s\n", r.in.domain_name->string);
279
280         /* do several ops to test credential chaining */
281         for (i=0;i<5;i++) {
282                 torture_assert_ntstatus_ok(tctx, dcerpc_samr_GetDomPwInfo_r(b, tctx, &r),
283                         "GetDomPwInfo failed");
284                 if (!NT_STATUS_IS_OK(r.out.result)) {
285                         if (!NT_STATUS_EQUAL(r.out.result, NT_STATUS_ACCESS_DENIED)) {
286                                 torture_comment(tctx, "GetDomPwInfo op %d failed - %s\n", i, nt_errstr(r.out.result));
287                                 return false;
288                         }
289                 }
290         }
291
292         return true;
293 }
294
295
296 /*
297   do some lsa ops using the schannel connection
298  */
299 static bool test_lsa_ops(struct torture_context *tctx, struct dcerpc_pipe *p)
300 {
301         struct lsa_GetUserName r;
302         bool ret = true;
303         struct lsa_String *account_name_p = NULL;
304         struct lsa_String *authority_name_p = NULL;
305         struct dcerpc_binding_handle *b = p->binding_handle;
306
307         torture_comment(tctx, "\nTesting GetUserName\n");
308
309         r.in.system_name = "\\";
310         r.in.account_name = &account_name_p;
311         r.in.authority_name = &authority_name_p;
312         r.out.account_name = &account_name_p;
313
314         /* do several ops to test credential chaining and various operations */
315         torture_assert_ntstatus_ok(tctx, dcerpc_lsa_GetUserName_r(b, tctx, &r),
316                 "lsa_GetUserName failed");
317
318         authority_name_p = *r.out.authority_name;
319
320         if (!NT_STATUS_IS_OK(r.out.result)) {
321                 torture_comment(tctx, "GetUserName failed - %s\n", nt_errstr(r.out.result));
322                 return false;
323         } else {
324                 if (!r.out.account_name) {
325                         return false;
326                 }
327
328                 if (strcmp(account_name_p->string, "ANONYMOUS LOGON") != 0) {
329                         torture_comment(tctx, "GetUserName returned wrong user: %s, expected %s\n",
330                                account_name_p->string, "ANONYMOUS LOGON");
331                         /* FIXME: gd */
332                         if (!torture_setting_bool(tctx, "samba3", false)) {
333                                 return false;
334                         }
335                 }
336                 if (!authority_name_p || !authority_name_p->string) {
337                         return false;
338                 }
339
340                 if (strcmp(authority_name_p->string, "NT AUTHORITY") != 0) {
341                         torture_comment(tctx, "GetUserName returned wrong user: %s, expected %s\n",
342                                authority_name_p->string, "NT AUTHORITY");
343                         /* FIXME: gd */
344                         if (!torture_setting_bool(tctx, "samba3", false)) {
345                                 return false;
346                         }
347                 }
348         }
349
350         return ret;
351 }
352
353
354 /*
355   test a schannel connection with the given flags
356  */
357 static bool test_schannel(struct torture_context *tctx,
358                           uint16_t acct_flags, uint32_t dcerpc_flags,
359                           int i)
360 {
361         struct test_join *join_ctx;
362         NTSTATUS status;
363         const char *binding = torture_setting_string(tctx, "binding", NULL);
364         struct dcerpc_binding *b;
365         struct dcerpc_pipe *p = NULL;
366         struct dcerpc_pipe *p_netlogon = NULL;
367         struct dcerpc_pipe *p_netlogon2 = NULL;
368         struct dcerpc_pipe *p_netlogon3 = NULL;
369         struct dcerpc_pipe *p_samr2 = NULL;
370         struct dcerpc_pipe *p_lsa = NULL;
371         struct netlogon_creds_CredentialState *creds;
372         struct cli_credentials *credentials;
373         enum dcerpc_transport_t transport;
374
375         join_ctx = torture_join_domain(tctx,
376                                        talloc_asprintf(tctx, "%s%d", TEST_MACHINE_NAME, i),
377                                        acct_flags, &credentials);
378         torture_assert(tctx, join_ctx != NULL, "Failed to join domain");
379
380         status = dcerpc_parse_binding(tctx, binding, &b);
381         torture_assert_ntstatus_ok(tctx, status, "Bad binding string");
382
383         b->flags &= ~DCERPC_AUTH_OPTIONS;
384         b->flags |= dcerpc_flags;
385
386         status = dcerpc_pipe_connect_b(tctx, &p, b, &ndr_table_samr,
387                                        credentials, tctx->ev, tctx->lp_ctx);
388         torture_assert_ntstatus_ok(tctx, status,
389                 "Failed to connect to samr with schannel");
390
391         torture_assert(tctx, test_samr_ops(tctx, p->binding_handle),
392                        "Failed to process schannel secured SAMR ops");
393
394         /* Also test that when we connect to the netlogon pipe, that
395          * the credentials we setup on the first pipe are valid for
396          * the second */
397
398         /* Swap the binding details from SAMR to NETLOGON */
399         status = dcerpc_epm_map_binding(tctx, b, &ndr_table_netlogon, tctx->ev, tctx->lp_ctx);
400         torture_assert_ntstatus_ok(tctx, status, "epm map");
401
402         status = dcerpc_secondary_connection(p, &p_netlogon,
403                                              b);
404         torture_assert_ntstatus_ok(tctx, status, "secondary connection");
405
406         status = dcerpc_bind_auth(p_netlogon, &ndr_table_netlogon,
407                                   credentials, lpcfg_gensec_settings(tctx, tctx->lp_ctx),
408                                   DCERPC_AUTH_TYPE_SCHANNEL,
409                                   dcerpc_auth_level(p->conn),
410                                   NULL);
411
412         torture_assert_ntstatus_ok(tctx, status, "bind auth");
413
414         creds = cli_credentials_get_netlogon_creds(credentials);
415         torture_assert(tctx, (creds != NULL), "schannel creds");
416
417         /* checks the capabilities */
418         torture_assert(tctx, test_netlogon_capabilities(p_netlogon, tctx, credentials, creds),
419                        "Failed to process schannel secured capability ops (on fresh connection)");
420
421         /* do a couple of logins */
422         torture_assert(tctx, test_netlogon_ops(p_netlogon, tctx, credentials, creds),
423                 "Failed to process schannel secured NETLOGON ops");
424
425         torture_assert(tctx, test_netlogon_ex_ops(p_netlogon, tctx, credentials, creds),
426                 "Failed to process schannel secured NETLOGON EX ops");
427
428         /* we *MUST* use ncacn_np for openpolicy etc. */
429         transport = b->transport;
430         b->transport = NCACN_NP;
431
432         /* Swap the binding details from SAMR to LSARPC */
433         status = dcerpc_epm_map_binding(tctx, b, &ndr_table_lsarpc, tctx->ev, tctx->lp_ctx);
434         torture_assert_ntstatus_ok(tctx, status, "epm map");
435
436         torture_assert_ntstatus_ok(tctx,
437                 dcerpc_pipe_connect_b(tctx, &p_lsa, b, &ndr_table_lsarpc,
438                                       credentials, tctx->ev, tctx->lp_ctx),
439                 "failed to connect lsarpc with schannel");
440
441         torture_assert(tctx, test_lsa_ops(tctx, p_lsa),
442                 "Failed to process schannel secured LSA ops");
443
444         talloc_free(p_lsa);
445         p_lsa = NULL;
446
447         b->transport = transport;
448
449         /* we *MUST* use ncacn_ip_tcp for lookupsids3/lookupnames4 */
450         transport = b->transport;
451         b->transport = NCACN_IP_TCP;
452
453         torture_assert_ntstatus_ok(tctx,
454                 dcerpc_epm_map_binding(tctx, b, &ndr_table_lsarpc, tctx->ev, tctx->lp_ctx),
455                 "failed to call epm map");
456
457         torture_assert_ntstatus_ok(tctx,
458                 dcerpc_pipe_connect_b(tctx, &p_lsa, b, &ndr_table_lsarpc,
459                                       credentials, tctx->ev, tctx->lp_ctx),
460                 "failed to connect lsarpc with schannel");
461
462         torture_assert(tctx,
463                 test_many_LookupSids(p_lsa, tctx, NULL),
464                 "LsaLookupSids3 failed!\n");
465
466         b->transport = transport;
467
468         /* Drop the socket, we want to start from scratch */
469         talloc_free(p);
470         p = NULL;
471
472         /* Now see what we are still allowed to do */
473
474         status = dcerpc_parse_binding(tctx, binding, &b);
475         torture_assert_ntstatus_ok(tctx, status, "Bad binding string");
476
477         b->flags &= ~DCERPC_AUTH_OPTIONS;
478         b->flags |= dcerpc_flags;
479
480         status = dcerpc_pipe_connect_b(tctx, &p_samr2, b, &ndr_table_samr,
481                                        credentials, tctx->ev, tctx->lp_ctx);
482         torture_assert_ntstatus_ok(tctx, status,
483                 "Failed to connect with schannel");
484
485         /* do a some SAMR operations.  We have *not* done a new serverauthenticate */
486         torture_assert (tctx, test_samr_ops(tctx, p_samr2->binding_handle),
487                         "Failed to process schannel secured SAMR ops (on fresh connection)");
488
489         /* Swap the binding details from SAMR to NETLOGON */
490         status = dcerpc_epm_map_binding(tctx, b, &ndr_table_netlogon, tctx->ev, tctx->lp_ctx);
491         torture_assert_ntstatus_ok(tctx, status, "epm");
492
493         status = dcerpc_secondary_connection(p_samr2, &p_netlogon2,
494                                              b);
495         torture_assert_ntstatus_ok(tctx, status, "secondary connection");
496
497         /* and now setup an SCHANNEL bind on netlogon */
498         status = dcerpc_bind_auth(p_netlogon2, &ndr_table_netlogon,
499                                   credentials, lpcfg_gensec_settings(tctx, tctx->lp_ctx),
500                                   DCERPC_AUTH_TYPE_SCHANNEL,
501                                   dcerpc_auth_level(p_samr2->conn),
502                                   NULL);
503
504         torture_assert_ntstatus_ok(tctx, status, "auth failed");
505
506         /* checks the capabilities */
507         torture_assert(tctx, test_netlogon_capabilities(p_netlogon2, tctx, credentials, creds),
508                        "Failed to process schannel secured capability ops (on fresh connection)");
509
510         /* Try the schannel-only SamLogonEx operation */
511         torture_assert(tctx, test_netlogon_ex_ops(p_netlogon2, tctx, credentials, creds),
512                        "Failed to process schannel secured NETLOGON EX ops (on fresh connection)");
513
514
515         /* And the more traditional style, proving that the
516          * credentials chaining state is fully present */
517         torture_assert(tctx, test_netlogon_ops(p_netlogon2, tctx, credentials, creds),
518                              "Failed to process schannel secured NETLOGON ops (on fresh connection)");
519
520         /* Drop the socket, we want to start from scratch (again) */
521         talloc_free(p_samr2);
522
523         /* We don't want schannel for this test */
524         b->flags &= ~DCERPC_AUTH_OPTIONS;
525
526         status = dcerpc_pipe_connect_b(tctx, &p_netlogon3, b, &ndr_table_netlogon,
527                                        credentials, tctx->ev, tctx->lp_ctx);
528         torture_assert_ntstatus_ok(tctx, status, "Failed to connect without schannel");
529
530         torture_assert(tctx, !test_netlogon_ex_ops(p_netlogon3, tctx, credentials, creds),
531                         "Processed NOT schannel secured NETLOGON EX ops without SCHANNEL (unsafe)");
532
533         /* Required because the previous call will mark the current context as having failed */
534         tctx->last_result = TORTURE_OK;
535         tctx->last_reason = NULL;
536
537         torture_assert(tctx, test_netlogon_ops(p_netlogon3, tctx, credentials, creds),
538                         "Failed to processed NOT schannel secured NETLOGON ops without new ServerAuth");
539
540         torture_leave_domain(tctx, join_ctx);
541         return true;
542 }
543
544
545
546 /*
547   a schannel test suite
548  */
549 bool torture_rpc_schannel(struct torture_context *torture)
550 {
551         bool ret = true;
552         struct {
553                 uint16_t acct_flags;
554                 uint32_t dcerpc_flags;
555         } tests[] = {
556                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_AUTO},
557                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_AUTO},
558                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_128},
559                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_128 },
560                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_AES},
561                 { ACB_WSTRUST,   DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_AES },
562                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_AUTO},
563                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_AUTO},
564                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_128 },
565                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_128 },
566                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SIGN | DCERPC_SCHANNEL_AES },
567                 { ACB_SVRTRUST,  DCERPC_SCHANNEL | DCERPC_SEAL | DCERPC_SCHANNEL_AES }
568         };
569         int i;
570
571         for (i=0;i<ARRAY_SIZE(tests);i++) {
572                 torture_comment(torture, "Testing with acct_flags=0x%x dcerpc_flags=0x%x \n",
573                        tests[i].acct_flags, tests[i].dcerpc_flags);
574
575                 if (!test_schannel(torture,
576                                    tests[i].acct_flags, tests[i].dcerpc_flags,
577                                    i)) {
578                         torture_comment(torture, "Failed with acct_flags=0x%x dcerpc_flags=0x%x \n",
579                                tests[i].acct_flags, tests[i].dcerpc_flags);
580                         ret = false;
581                 }
582         }
583
584         return ret;
585 }
586
587 /*
588   test two schannel connections
589  */
590 bool torture_rpc_schannel2(struct torture_context *torture)
591 {
592         struct test_join *join_ctx;
593         NTSTATUS status;
594         const char *binding = torture_setting_string(torture, "binding", NULL);
595         struct dcerpc_binding *b;
596         struct dcerpc_pipe *p1 = NULL, *p2 = NULL;
597         struct cli_credentials *credentials1, *credentials2;
598         uint32_t dcerpc_flags = DCERPC_SCHANNEL | DCERPC_SIGN;
599
600         join_ctx = torture_join_domain(torture, talloc_asprintf(torture, "%s2", TEST_MACHINE_NAME),
601                                        ACB_WSTRUST, &credentials1);
602         torture_assert(torture, join_ctx != NULL,
603                        "Failed to join domain with acct_flags=ACB_WSTRUST");
604
605         credentials2 = cli_credentials_shallow_copy(torture, credentials1);
606         cli_credentials_set_netlogon_creds(credentials1, NULL);
607         cli_credentials_set_netlogon_creds(credentials2, NULL);
608
609         status = dcerpc_parse_binding(torture, binding, &b);
610         torture_assert_ntstatus_ok(torture, status, "Bad binding string");
611
612         b->flags &= ~DCERPC_AUTH_OPTIONS;
613         b->flags |= dcerpc_flags;
614
615         torture_comment(torture, "Opening first connection\n");
616         status = dcerpc_pipe_connect_b(torture, &p1, b, &ndr_table_netlogon,
617                                        credentials1, torture->ev, torture->lp_ctx);
618         torture_assert_ntstatus_ok(torture, status, "Failed to connect with schannel");
619
620         torture_comment(torture, "Opening second connection\n");
621         status = dcerpc_pipe_connect_b(torture, &p2, b, &ndr_table_netlogon,
622                                        credentials2, torture->ev, torture->lp_ctx);
623         torture_assert_ntstatus_ok(torture, status, "Failed to connect with schannel");
624
625         cli_credentials_set_netlogon_creds(credentials1, NULL);
626         cli_credentials_set_netlogon_creds(credentials2, NULL);
627
628         torture_comment(torture, "Testing logon on pipe1\n");
629         if (!test_netlogon_ex_ops(p1, torture, credentials1, NULL))
630                 return false;
631
632         torture_comment(torture, "Testing logon on pipe2\n");
633         if (!test_netlogon_ex_ops(p2, torture, credentials2, NULL))
634                 return false;
635
636         torture_comment(torture, "Again on pipe1\n");
637         if (!test_netlogon_ex_ops(p1, torture, credentials1, NULL))
638                 return false;
639
640         torture_comment(torture, "Again on pipe2\n");
641         if (!test_netlogon_ex_ops(p2, torture, credentials2, NULL))
642                 return false;
643
644         torture_leave_domain(torture, join_ctx);
645         return true;
646 }
647
648 struct torture_schannel_bench;
649
650 struct torture_schannel_bench_conn {
651         struct torture_schannel_bench *s;
652         int index;
653         struct cli_credentials *wks_creds;
654         struct dcerpc_pipe *pipe;
655         struct netr_LogonSamLogonEx r;
656         struct netr_NetworkInfo ninfo;
657         TALLOC_CTX *tmp;
658         uint64_t total;
659         uint32_t count;
660 };
661
662 struct torture_schannel_bench {
663         struct torture_context *tctx;
664         bool progress;
665         int timelimit;
666         int nprocs;
667         int nconns;
668         struct torture_schannel_bench_conn *conns;
669         struct test_join *join_ctx1;
670         struct cli_credentials *wks_creds1;
671         struct test_join *join_ctx2;
672         struct cli_credentials *wks_creds2;
673         struct cli_credentials *user1_creds;
674         struct cli_credentials *user2_creds;
675         struct dcerpc_binding *b;
676         NTSTATUS error;
677         uint64_t total;
678         uint32_t count;
679         bool stopped;
680 };
681
682 static void torture_schannel_bench_connected(struct composite_context *c)
683 {
684         struct torture_schannel_bench_conn *conn =
685                 (struct torture_schannel_bench_conn *)c->async.private_data;
686         struct torture_schannel_bench *s = talloc_get_type(conn->s,
687                                            struct torture_schannel_bench);
688
689         s->error = dcerpc_pipe_connect_b_recv(c, s->conns, &conn->pipe);
690         torture_comment(s->tctx, "conn[%u]: %s\n", conn->index, nt_errstr(s->error));
691         if (NT_STATUS_IS_OK(s->error)) {
692                 s->nconns++;
693         }
694 }
695
696 static void torture_schannel_bench_recv(struct tevent_req *subreq);
697
698 static bool torture_schannel_bench_start(struct torture_schannel_bench_conn *conn)
699 {
700         struct torture_schannel_bench *s = conn->s;
701         NTSTATUS status;
702         DATA_BLOB names_blob, chal, lm_resp, nt_resp;
703         int flags = CLI_CRED_NTLM_AUTH;
704         struct tevent_req *subreq;
705         struct cli_credentials *user_creds;
706
707         if (conn->total % 2) {
708                 user_creds = s->user1_creds;
709         } else {
710                 user_creds = s->user2_creds;
711         }
712
713         if (lpcfg_client_lanman_auth(s->tctx->lp_ctx)) {
714                 flags |= CLI_CRED_LANMAN_AUTH;
715         }
716
717         if (lpcfg_client_ntlmv2_auth(s->tctx->lp_ctx)) {
718                 flags |= CLI_CRED_NTLMv2_AUTH;
719         }
720
721         talloc_free(conn->tmp);
722         conn->tmp = talloc_new(s);
723         ZERO_STRUCT(conn->ninfo);
724         ZERO_STRUCT(conn->r);
725
726         cli_credentials_get_ntlm_username_domain(user_creds, conn->tmp,
727                                                  &conn->ninfo.identity_info.account_name.string,
728                                                  &conn->ninfo.identity_info.domain_name.string);
729
730         generate_random_buffer(conn->ninfo.challenge,
731                                sizeof(conn->ninfo.challenge));
732         chal = data_blob_const(conn->ninfo.challenge,
733                                sizeof(conn->ninfo.challenge));
734
735         names_blob = NTLMv2_generate_names_blob(conn->tmp,
736                                                 cli_credentials_get_workstation(conn->wks_creds),
737                                                 cli_credentials_get_domain(conn->wks_creds));
738
739         status = cli_credentials_get_ntlm_response(user_creds, conn->tmp,
740                                                    &flags,
741                                                    chal,
742                                                    names_blob,
743                                                    &lm_resp, &nt_resp,
744                                                    NULL, NULL);
745         torture_assert_ntstatus_ok(s->tctx, status,
746                                    "cli_credentials_get_ntlm_response failed");
747
748         conn->ninfo.lm.data = lm_resp.data;
749         conn->ninfo.lm.length = lm_resp.length;
750
751         conn->ninfo.nt.data = nt_resp.data;
752         conn->ninfo.nt.length = nt_resp.length;
753
754         conn->ninfo.identity_info.parameter_control = 0;
755         conn->ninfo.identity_info.logon_id_low = 0;
756         conn->ninfo.identity_info.logon_id_high = 0;
757         conn->ninfo.identity_info.workstation.string = cli_credentials_get_workstation(conn->wks_creds);
758
759         conn->r.in.server_name = talloc_asprintf(conn->tmp, "\\\\%s", dcerpc_server_name(conn->pipe));
760         conn->r.in.computer_name = cli_credentials_get_workstation(conn->wks_creds);
761         conn->r.in.logon_level = NetlogonNetworkInformation;
762         conn->r.in.logon = talloc(conn->tmp, union netr_LogonLevel);
763         conn->r.in.logon->network = &conn->ninfo;
764         conn->r.in.flags = talloc(conn->tmp, uint32_t);
765         conn->r.in.validation_level = 2;
766         conn->r.out.validation = talloc(conn->tmp, union netr_Validation);
767         conn->r.out.authoritative = talloc(conn->tmp, uint8_t);
768         conn->r.out.flags = conn->r.in.flags;
769
770         subreq = dcerpc_netr_LogonSamLogonEx_r_send(s, s->tctx->ev,
771                                                     conn->pipe->binding_handle,
772                                                     &conn->r);
773         torture_assert(s->tctx, subreq, "Failed to setup LogonSamLogonEx request");
774
775         tevent_req_set_callback(subreq, torture_schannel_bench_recv, conn);
776
777         return true;
778 }
779
780 static void torture_schannel_bench_recv(struct tevent_req *subreq)
781 {
782         bool ret;
783         struct torture_schannel_bench_conn *conn =
784                 (struct torture_schannel_bench_conn *)tevent_req_callback_data_void(subreq);
785         struct torture_schannel_bench *s = talloc_get_type(conn->s,
786                                            struct torture_schannel_bench);
787
788         s->error = dcerpc_netr_LogonSamLogonEx_r_recv(subreq, subreq);
789         TALLOC_FREE(subreq);
790         if (!NT_STATUS_IS_OK(s->error)) {
791                 return;
792         }
793
794         conn->total++;
795         conn->count++;
796
797         if (s->stopped) {
798                 return;
799         }
800
801         ret = torture_schannel_bench_start(conn);
802         if (!ret) {
803                 s->error = NT_STATUS_INTERNAL_ERROR;
804         }
805 }
806
807 /*
808   test multiple schannel connection in parallel
809  */
810 bool torture_rpc_schannel_bench1(struct torture_context *torture)
811 {
812         bool ret = true;
813         NTSTATUS status;
814         const char *binding = torture_setting_string(torture, "binding", NULL);
815         struct torture_schannel_bench *s;
816         struct timeval start;
817         struct timeval end;
818         int i;
819         const char *tmp;
820
821         s = talloc_zero(torture, struct torture_schannel_bench);
822         s->tctx = torture;
823         s->progress = torture_setting_bool(torture, "progress", true);
824         s->timelimit = torture_setting_int(torture, "timelimit", 10);
825         s->nprocs = torture_setting_int(torture, "nprocs", 4);
826         s->conns = talloc_zero_array(s, struct torture_schannel_bench_conn, s->nprocs);
827
828         s->user1_creds = cli_credentials_shallow_copy(s, cmdline_credentials);
829         tmp = torture_setting_string(s->tctx, "extra_user1", NULL);
830         if (tmp) {
831                 cli_credentials_parse_string(s->user1_creds, tmp, CRED_SPECIFIED);
832         }
833         s->user2_creds = cli_credentials_shallow_copy(s, cmdline_credentials);
834         tmp = torture_setting_string(s->tctx, "extra_user2", NULL);
835         if (tmp) {
836                 cli_credentials_parse_string(s->user1_creds, tmp, CRED_SPECIFIED);
837         }
838
839         s->join_ctx1 = torture_join_domain(s->tctx, talloc_asprintf(s, "%sb", TEST_MACHINE_NAME),
840                                            ACB_WSTRUST, &s->wks_creds1);
841         torture_assert(torture, s->join_ctx1 != NULL,
842                        "Failed to join domain with acct_flags=ACB_WSTRUST");
843         s->join_ctx2 = torture_join_domain(s->tctx, talloc_asprintf(s, "%sc", TEST_MACHINE_NAME),
844                                            ACB_WSTRUST, &s->wks_creds2);
845         torture_assert(torture, s->join_ctx2 != NULL,
846                        "Failed to join domain with acct_flags=ACB_WSTRUST");
847
848         cli_credentials_set_kerberos_state(s->wks_creds1, CRED_DONT_USE_KERBEROS);
849         cli_credentials_set_kerberos_state(s->wks_creds2, CRED_DONT_USE_KERBEROS);
850
851         for (i=0; i < s->nprocs; i++) {
852                 struct cli_credentials *wks = s->wks_creds1;
853
854                 if ((i % 2) && (torture_setting_bool(torture, "multijoin", false))) {
855                         wks = s->wks_creds2;
856                 }
857
858                 s->conns[i].s = s;
859                 s->conns[i].index = i;
860                 s->conns[i].wks_creds = cli_credentials_shallow_copy(s->conns, wks);
861                 cli_credentials_set_netlogon_creds(s->conns[i].wks_creds, NULL);
862         }
863
864         status = dcerpc_parse_binding(s, binding, &s->b);
865         torture_assert_ntstatus_ok(torture, status, "Bad binding string");
866         s->b->flags &= ~DCERPC_AUTH_OPTIONS;
867         s->b->flags |= DCERPC_SCHANNEL | DCERPC_SIGN;
868
869         torture_comment(torture, "Opening %d connections in parallel\n", s->nprocs);
870         for (i=0; i < s->nprocs; i++) {
871 #if 1
872                 s->error = dcerpc_pipe_connect_b(s->conns, &s->conns[i].pipe, s->b,
873                                                  &ndr_table_netlogon,
874                                                  s->conns[i].wks_creds,
875                                                  torture->ev, torture->lp_ctx);
876                 torture_assert_ntstatus_ok(torture, s->error, "Failed to connect with schannel");
877 #else
878                 /*
879                  * This path doesn't work against windows,
880                  * because of windows drops the connections
881                  * which haven't reached a session setup yet
882                  *
883                  * The same as the reset on zero vc stuff.
884                  */
885                 struct composite_context *c;
886                 c = dcerpc_pipe_connect_b_send(s->conns, s->b,
887                                                &ndr_table_netlogon,
888                                                s->conns[i].wks_creds,
889                                                torture->ev,
890                                                torture->lp_ctx);
891                 torture_assert(torture, c != NULL, "Failed to setup connect");
892                 c->async.fn = torture_schannel_bench_connected;
893                 c->async.private_data = &s->conns[i];
894         }
895
896         while (NT_STATUS_IS_OK(s->error) && s->nprocs != s->nconns) {
897                 int ev_ret = tevent_loop_once(torture->ev);
898                 torture_assert(torture, ev_ret == 0, "tevent_loop_once failed");
899 #endif
900         }
901         torture_assert_ntstatus_ok(torture, s->error, "Failed establish a connect");
902
903         /*
904          * Change the workstation password after establishing the netlogon
905          * schannel connections to prove that existing connections are not
906          * affected by a wks pwchange.
907          */
908
909         {
910                 struct netr_ServerPasswordSet pwset;
911                 char *password = generate_random_password(s->join_ctx1, 8, 255);
912                 struct netlogon_creds_CredentialState *creds_state;
913                 struct dcerpc_pipe *net_pipe;
914                 struct netr_Authenticator credential, return_authenticator;
915                 struct samr_Password new_password;
916
917                 status = dcerpc_pipe_connect_b(s, &net_pipe, s->b,
918                                                &ndr_table_netlogon,
919                                                s->wks_creds1,
920                                                torture->ev, torture->lp_ctx);
921
922                 torture_assert_ntstatus_ok(torture, status,
923                                            "dcerpc_pipe_connect_b failed");
924
925                 pwset.in.server_name = talloc_asprintf(
926                         net_pipe, "\\\\%s", dcerpc_server_name(net_pipe));
927                 pwset.in.computer_name =
928                         cli_credentials_get_workstation(s->wks_creds1);
929                 pwset.in.account_name = talloc_asprintf(
930                         net_pipe, "%s$", pwset.in.computer_name);
931                 pwset.in.secure_channel_type = SEC_CHAN_WKSTA;
932                 pwset.in.credential = &credential;
933                 pwset.in.new_password = &new_password;
934                 pwset.out.return_authenticator = &return_authenticator;
935
936                 E_md4hash(password, new_password.hash);
937
938                 creds_state = cli_credentials_get_netlogon_creds(
939                         s->wks_creds1);
940                 netlogon_creds_des_encrypt(creds_state, &new_password);
941                 netlogon_creds_client_authenticator(creds_state, &credential);
942
943                 torture_assert_ntstatus_ok(torture, dcerpc_netr_ServerPasswordSet_r(net_pipe->binding_handle, torture, &pwset),
944                         "ServerPasswordSet failed");
945                 torture_assert_ntstatus_ok(torture, pwset.out.result,
946                                            "ServerPasswordSet failed");
947
948                 if (!netlogon_creds_client_check(creds_state,
949                                         &pwset.out.return_authenticator->cred)) {
950                         torture_comment(torture, "Credential chaining failed\n");
951                 }
952
953                 cli_credentials_set_password(s->wks_creds1, password,
954                                              CRED_SPECIFIED);
955
956                 talloc_free(net_pipe);
957
958                 /* Just as a test, connect with the new creds */
959
960                 cli_credentials_set_netlogon_creds(s->wks_creds1, NULL);
961
962                 status = dcerpc_pipe_connect_b(s, &net_pipe, s->b,
963                                                &ndr_table_netlogon,
964                                                s->wks_creds1,
965                                                torture->ev, torture->lp_ctx);
966
967                 torture_assert_ntstatus_ok(torture, status,
968                                            "dcerpc_pipe_connect_b failed");
969
970                 talloc_free(net_pipe);
971         }
972
973         torture_comment(torture, "Start looping LogonSamLogonEx on %d connections for %d secs\n",
974                         s->nprocs, s->timelimit);
975         for (i=0; i < s->nprocs; i++) {
976                 ret = torture_schannel_bench_start(&s->conns[i]);
977                 torture_assert(torture, ret, "Failed to setup LogonSamLogonEx");
978         }
979
980         start = timeval_current();
981         end = timeval_add(&start, s->timelimit, 0);
982
983         while (NT_STATUS_IS_OK(s->error) && !timeval_expired(&end)) {
984                 int ev_ret = tevent_loop_once(torture->ev);
985                 torture_assert(torture, ev_ret == 0, "tevent_loop_once failed");
986         }
987         torture_assert_ntstatus_ok(torture, s->error, "Failed some request");
988         s->stopped = true;
989         talloc_free(s->conns);
990
991         for (i=0; i < s->nprocs; i++) {
992                 s->total += s->conns[i].total;
993         }
994
995         torture_comment(torture,
996                         "Total ops[%llu] (%u ops/s)\n",
997                         (unsigned long long)s->total,
998                         (unsigned)s->total/s->timelimit);
999
1000         torture_leave_domain(torture, s->join_ctx1);
1001         torture_leave_domain(torture, s->join_ctx2);
1002         return true;
1003 }