add a test for a LSA lookupnames with a NULL string
[kai/samba.git] / source4 / torture / rpc / lsa.c
1 /* 
2    Unix SMB/CIFS implementation.
3    test suite for lsa rpc operations
4
5    Copyright (C) Andrew Tridgell 2003
6    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
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 "torture/torture.h"
24 #include "librpc/gen_ndr/ndr_lsa_c.h"
25 #include "librpc/gen_ndr/netlogon.h"
26 #include "librpc/gen_ndr/ndr_drsblobs.h"
27 #include "lib/events/events.h"
28 #include "libcli/security/security.h"
29 #include "libcli/auth/libcli_auth.h"
30 #include "torture/rpc/rpc.h"
31 #include "param/param.h"
32 #include "../lib/crypto/crypto.h"
33 #define TEST_MACHINENAME "lsatestmach"
34
35 static void init_lsa_String(struct lsa_String *name, const char *s)
36 {
37         name->string = s;
38 }
39
40 static bool test_OpenPolicy(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
41 {
42         struct lsa_ObjectAttribute attr;
43         struct policy_handle handle;
44         struct lsa_QosInfo qos;
45         struct lsa_OpenPolicy r;
46         NTSTATUS status;
47         uint16_t system_name = '\\';
48
49         printf("\ntesting OpenPolicy\n");
50
51         qos.len = 0;
52         qos.impersonation_level = 2;
53         qos.context_mode = 1;
54         qos.effective_only = 0;
55
56         attr.len = 0;
57         attr.root_dir = NULL;
58         attr.object_name = NULL;
59         attr.attributes = 0;
60         attr.sec_desc = NULL;
61         attr.sec_qos = &qos;
62
63         r.in.system_name = &system_name;
64         r.in.attr = &attr;
65         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
66         r.out.handle = &handle;
67
68         status = dcerpc_lsa_OpenPolicy(p, mem_ctx, &r);
69         if (!NT_STATUS_IS_OK(status)) {
70                 if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
71                     NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
72                         printf("not considering %s to be an error\n", nt_errstr(status));
73                         return true;
74                 }
75                 printf("OpenPolicy failed - %s\n", nt_errstr(status));
76                 return false;
77         }
78
79         return true;
80 }
81
82
83 bool test_lsa_OpenPolicy2(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, 
84                           struct policy_handle **handle)
85 {
86         struct lsa_ObjectAttribute attr;
87         struct lsa_QosInfo qos;
88         struct lsa_OpenPolicy2 r;
89         NTSTATUS status;
90
91         printf("\ntesting OpenPolicy2\n");
92
93         *handle = talloc(mem_ctx, struct policy_handle);
94         if (!*handle) {
95                 return false;
96         }
97
98         qos.len = 0;
99         qos.impersonation_level = 2;
100         qos.context_mode = 1;
101         qos.effective_only = 0;
102
103         attr.len = 0;
104         attr.root_dir = NULL;
105         attr.object_name = NULL;
106         attr.attributes = 0;
107         attr.sec_desc = NULL;
108         attr.sec_qos = &qos;
109
110         r.in.system_name = "\\";
111         r.in.attr = &attr;
112         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
113         r.out.handle = *handle;
114
115         status = dcerpc_lsa_OpenPolicy2(p, mem_ctx, &r);
116         if (!NT_STATUS_IS_OK(status)) {
117                 if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
118                     NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
119                         printf("not considering %s to be an error\n", nt_errstr(status));
120                         talloc_free(*handle);
121                         *handle = NULL;
122                         return true;
123                 }
124                 printf("OpenPolicy2 failed - %s\n", nt_errstr(status));
125                 return false;
126         }
127
128         return true;
129 }
130
131
132 static const char *sid_type_lookup(enum lsa_SidType r)
133 {
134         switch (r) {
135                 case SID_NAME_USE_NONE: return "SID_NAME_USE_NONE"; break;
136                 case SID_NAME_USER: return "SID_NAME_USER"; break;
137                 case SID_NAME_DOM_GRP: return "SID_NAME_DOM_GRP"; break;
138                 case SID_NAME_DOMAIN: return "SID_NAME_DOMAIN"; break;
139                 case SID_NAME_ALIAS: return "SID_NAME_ALIAS"; break;
140                 case SID_NAME_WKN_GRP: return "SID_NAME_WKN_GRP"; break;
141                 case SID_NAME_DELETED: return "SID_NAME_DELETED"; break;
142                 case SID_NAME_INVALID: return "SID_NAME_INVALID"; break;
143                 case SID_NAME_UNKNOWN: return "SID_NAME_UNKNOWN"; break;
144                 case SID_NAME_COMPUTER: return "SID_NAME_COMPUTER"; break;
145         }
146         return "Invalid sid type\n";
147 }
148
149 static bool test_LookupNames(struct dcerpc_pipe *p, 
150                              TALLOC_CTX *mem_ctx, 
151                              struct policy_handle *handle,
152                              struct lsa_TransNameArray *tnames)
153 {
154         struct lsa_LookupNames r;
155         struct lsa_TransSidArray sids;
156         struct lsa_String *names;
157         uint32_t count = 0;
158         NTSTATUS status;
159         int i;
160
161         printf("\nTesting LookupNames with %d names\n", tnames->count);
162
163         sids.count = 0;
164         sids.sids = NULL;
165
166         names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
167         for (i=0;i<tnames->count;i++) {
168                 init_lsa_String(&names[i], tnames->names[i].name.string);
169         }
170
171         r.in.handle = handle;
172         r.in.num_names = tnames->count;
173         r.in.names = names;
174         r.in.sids = &sids;
175         r.in.level = 1;
176         r.in.count = &count;
177         r.out.count = &count;
178         r.out.sids = &sids;
179
180         status = dcerpc_lsa_LookupNames(p, mem_ctx, &r);
181
182         if (NT_STATUS_EQUAL(status, STATUS_SOME_UNMAPPED) || 
183             NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED)) {
184                 for (i=0;i< tnames->count;i++) {
185                         if (i < count && sids.sids[i].sid_type == SID_NAME_UNKNOWN) {
186                                 printf("LookupName of %s was unmapped\n", 
187                                        tnames->names[i].name.string);   
188                         } else if (i >=count) {
189                                 printf("LookupName of %s failed to return a result\n",
190                                        tnames->names[i].name.string);
191                         }
192                 }
193                 printf("LookupNames failed - %s\n", nt_errstr(status));
194                 return false;
195         } else if (!NT_STATUS_IS_OK(status)) {
196                 printf("LookupNames failed - %s\n", nt_errstr(status));
197                 return false;
198         }
199         
200         for (i=0;i< tnames->count;i++) {
201                 if (i < count && sids.sids[i].sid_type != tnames->names[i].sid_type) {
202                         printf("LookupName of %s got unexpected name type: %s\n", 
203                                tnames->names[i].name.string, sid_type_lookup(sids.sids[i].sid_type));
204                 } else if (i >=count) {
205                         printf("LookupName of %s failed to return a result\n",
206                                tnames->names[i].name.string);
207                 }
208         }
209         printf("\n");
210
211         return true;
212 }
213
214 static bool test_LookupNames_bogus(struct dcerpc_pipe *p, 
215                             TALLOC_CTX *mem_ctx, 
216                             struct policy_handle *handle)
217 {
218         struct lsa_LookupNames r;
219         struct lsa_TransSidArray sids;
220         struct lsa_String *names;
221         uint32_t count = 0;
222         NTSTATUS status;
223         int i;
224
225         struct lsa_TranslatedName name[2];
226         struct lsa_TransNameArray tnames;
227
228         tnames.names = name;
229         tnames.count = 2;
230         name[0].name.string = "NT AUTHORITY\\BOGUS";
231         name[1].name.string = NULL;
232
233         printf("\nTesting LookupNames with bogus names\n");
234
235         sids.count = 0;
236         sids.sids = NULL;
237
238         names = talloc_array(mem_ctx, struct lsa_String, tnames.count);
239         for (i=0;i<tnames.count;i++) {
240                 init_lsa_String(&names[i], tnames.names[i].name.string);
241         }
242
243         r.in.handle = handle;
244         r.in.num_names = tnames.count;
245         r.in.names = names;
246         r.in.sids = &sids;
247         r.in.level = 1;
248         r.in.count = &count;
249         r.out.count = &count;
250         r.out.sids = &sids;
251
252         status = dcerpc_lsa_LookupNames(p, mem_ctx, &r);
253         if (!NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED)) {
254                 printf("LookupNames failed - %s\n", nt_errstr(status));
255                 return false;
256         }
257
258         printf("\n");
259
260         return true;
261 }
262
263 static bool test_LookupNames_wellknown(struct dcerpc_pipe *p, 
264                                        TALLOC_CTX *mem_ctx, 
265                                        struct policy_handle *handle)
266 {
267         struct lsa_TranslatedName name;
268         struct lsa_TransNameArray tnames;
269         bool ret = true;
270
271         printf("Testing LookupNames with well known names\n");
272
273         tnames.names = &name;
274         tnames.count = 1;
275         name.name.string = "NT AUTHORITY\\SYSTEM";
276         name.sid_type = SID_NAME_WKN_GRP;
277         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
278
279         name.name.string = "NT AUTHORITY\\ANONYMOUS LOGON";
280         name.sid_type = SID_NAME_WKN_GRP;
281         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
282
283         name.name.string = "NT AUTHORITY\\Authenticated Users";
284         name.sid_type = SID_NAME_WKN_GRP;
285         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
286
287 #if 0
288         name.name.string = "NT AUTHORITY";
289         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
290
291         name.name.string = "NT AUTHORITY\\";
292         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
293 #endif
294
295         name.name.string = "BUILTIN\\";
296         name.sid_type = SID_NAME_DOMAIN;
297         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
298
299         name.name.string = "BUILTIN\\Administrators";
300         name.sid_type = SID_NAME_ALIAS;
301         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
302
303         name.name.string = "SYSTEM";
304         name.sid_type = SID_NAME_WKN_GRP;
305         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
306
307         name.name.string = "Everyone";
308         name.sid_type = SID_NAME_WKN_GRP;
309         ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
310         return ret;
311 }
312
313 static bool test_LookupNames2(struct dcerpc_pipe *p, 
314                               TALLOC_CTX *mem_ctx, 
315                               struct policy_handle *handle,
316                               struct lsa_TransNameArray2 *tnames)
317 {
318         struct lsa_LookupNames2 r;
319         struct lsa_TransSidArray2 sids;
320         struct lsa_String *names;
321         uint32_t count = 0;
322         NTSTATUS status;
323         int i;
324
325         printf("\nTesting LookupNames2 with %d names\n", tnames->count);
326
327         sids.count = 0;
328         sids.sids = NULL;
329
330         names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
331         for (i=0;i<tnames->count;i++) {
332                 init_lsa_String(&names[i], tnames->names[i].name.string);
333         }
334
335         r.in.handle = handle;
336         r.in.num_names = tnames->count;
337         r.in.names = names;
338         r.in.sids = &sids;
339         r.in.level = 1;
340         r.in.count = &count;
341         r.in.unknown1 = 0;
342         r.in.unknown2 = 0;
343         r.out.count = &count;
344         r.out.sids = &sids;
345
346         status = dcerpc_lsa_LookupNames2(p, mem_ctx, &r);
347         if (!NT_STATUS_IS_OK(status)) {
348                 printf("LookupNames2 failed - %s\n", nt_errstr(status));
349                 return false;
350         }
351
352         printf("\n");
353
354         return true;
355 }
356
357
358 static bool test_LookupNames3(struct dcerpc_pipe *p, 
359                               TALLOC_CTX *mem_ctx, 
360                               struct policy_handle *handle,
361                               struct lsa_TransNameArray2 *tnames)
362 {
363         struct lsa_LookupNames3 r;
364         struct lsa_TransSidArray3 sids;
365         struct lsa_String *names;
366         uint32_t count = 0;
367         NTSTATUS status;
368         int i;
369
370         printf("\nTesting LookupNames3 with %d names\n", tnames->count);
371
372         sids.count = 0;
373         sids.sids = NULL;
374
375         names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
376         for (i=0;i<tnames->count;i++) {
377                 init_lsa_String(&names[i], tnames->names[i].name.string);
378         }
379
380         r.in.handle = handle;
381         r.in.num_names = tnames->count;
382         r.in.names = names;
383         r.in.sids = &sids;
384         r.in.level = 1;
385         r.in.count = &count;
386         r.in.unknown1 = 0;
387         r.in.unknown2 = 0;
388         r.out.count = &count;
389         r.out.sids = &sids;
390
391         status = dcerpc_lsa_LookupNames3(p, mem_ctx, &r);
392         if (!NT_STATUS_IS_OK(status)) {
393                 printf("LookupNames3 failed - %s\n", nt_errstr(status));
394                 return false;
395         }
396
397         printf("\n");
398
399         return true;
400 }
401
402 static bool test_LookupNames4(struct dcerpc_pipe *p, 
403                               TALLOC_CTX *mem_ctx, 
404                               struct lsa_TransNameArray2 *tnames)
405 {
406         struct lsa_LookupNames4 r;
407         struct lsa_TransSidArray3 sids;
408         struct lsa_String *names;
409         uint32_t count = 0;
410         NTSTATUS status;
411         int i;
412
413         printf("\nTesting LookupNames4 with %d names\n", tnames->count);
414
415         sids.count = 0;
416         sids.sids = NULL;
417
418         names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
419         for (i=0;i<tnames->count;i++) {
420                 init_lsa_String(&names[i], tnames->names[i].name.string);
421         }
422
423         r.in.num_names = tnames->count;
424         r.in.names = names;
425         r.in.sids = &sids;
426         r.in.level = 1;
427         r.in.count = &count;
428         r.in.unknown1 = 0;
429         r.in.unknown2 = 0;
430         r.out.count = &count;
431         r.out.sids = &sids;
432
433         status = dcerpc_lsa_LookupNames4(p, mem_ctx, &r);
434         if (!NT_STATUS_IS_OK(status)) {
435                 printf("LookupNames4 failed - %s\n", nt_errstr(status));
436                 return false;
437         }
438
439         printf("\n");
440
441         return true;
442 }
443
444
445 static bool test_LookupSids(struct dcerpc_pipe *p, 
446                             TALLOC_CTX *mem_ctx, 
447                             struct policy_handle *handle,
448                             struct lsa_SidArray *sids)
449 {
450         struct lsa_LookupSids r;
451         struct lsa_TransNameArray names;
452         uint32_t count = sids->num_sids;
453         NTSTATUS status;
454
455         printf("\nTesting LookupSids\n");
456
457         names.count = 0;
458         names.names = NULL;
459
460         r.in.handle = handle;
461         r.in.sids = sids;
462         r.in.names = &names;
463         r.in.level = 1;
464         r.in.count = &count;
465         r.out.count = &count;
466         r.out.names = &names;
467
468         status = dcerpc_lsa_LookupSids(p, mem_ctx, &r);
469         if (!NT_STATUS_IS_OK(status)) {
470                 printf("LookupSids failed - %s\n", nt_errstr(status));
471                 return false;
472         }
473
474         printf("\n");
475
476         if (!test_LookupNames(p, mem_ctx, handle, &names)) {
477                 return false;
478         }
479
480         return true;
481 }
482
483
484 static bool test_LookupSids2(struct dcerpc_pipe *p, 
485                             TALLOC_CTX *mem_ctx, 
486                             struct policy_handle *handle,
487                             struct lsa_SidArray *sids)
488 {
489         struct lsa_LookupSids2 r;
490         struct lsa_TransNameArray2 names;
491         uint32_t count = sids->num_sids;
492         NTSTATUS status;
493
494         printf("\nTesting LookupSids2\n");
495
496         names.count = 0;
497         names.names = NULL;
498
499         r.in.handle = handle;
500         r.in.sids = sids;
501         r.in.names = &names;
502         r.in.level = 1;
503         r.in.count = &count;
504         r.in.unknown1 = 0;
505         r.in.unknown2 = 0;
506         r.out.count = &count;
507         r.out.names = &names;
508
509         status = dcerpc_lsa_LookupSids2(p, mem_ctx, &r);
510         if (!NT_STATUS_IS_OK(status)) {
511                 printf("LookupSids2 failed - %s\n", nt_errstr(status));
512                 return false;
513         }
514
515         printf("\n");
516
517         if (!test_LookupNames2(p, mem_ctx, handle, &names)) {
518                 return false;
519         }
520
521         if (!test_LookupNames3(p, mem_ctx, handle, &names)) {
522                 return false;
523         }
524
525         return true;
526 }
527
528 static bool test_LookupSids3(struct dcerpc_pipe *p, 
529                             TALLOC_CTX *mem_ctx, 
530                             struct lsa_SidArray *sids)
531 {
532         struct lsa_LookupSids3 r;
533         struct lsa_TransNameArray2 names;
534         uint32_t count = sids->num_sids;
535         NTSTATUS status;
536
537         printf("\nTesting LookupSids3\n");
538
539         names.count = 0;
540         names.names = NULL;
541
542         r.in.sids = sids;
543         r.in.names = &names;
544         r.in.level = 1;
545         r.in.count = &count;
546         r.in.unknown1 = 0;
547         r.in.unknown2 = 0;
548         r.out.count = &count;
549         r.out.names = &names;
550
551         status = dcerpc_lsa_LookupSids3(p, mem_ctx, &r);
552         if (!NT_STATUS_IS_OK(status)) {
553                 if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
554                     NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
555                         printf("not considering %s to be an error\n", nt_errstr(status));
556                         return true;
557                 }
558                 printf("LookupSids3 failed - %s - not considered an error\n", 
559                        nt_errstr(status));
560                 return false;
561         }
562
563         printf("\n");
564
565         if (!test_LookupNames4(p, mem_ctx, &names)) {
566                 return false;
567         }
568
569         return true;
570 }
571
572 bool test_many_LookupSids(struct dcerpc_pipe *p, 
573                           TALLOC_CTX *mem_ctx, 
574                           struct policy_handle *handle)
575 {
576         uint32_t count;
577         NTSTATUS status;
578         struct lsa_SidArray sids;
579         int i;
580
581         printf("\nTesting LookupSids with lots of SIDs\n");
582
583         sids.num_sids = 100;
584
585         sids.sids = talloc_array(mem_ctx, struct lsa_SidPtr, sids.num_sids);
586
587         for (i=0; i<sids.num_sids; i++) {
588                 const char *sidstr = "S-1-5-32-545";
589                 sids.sids[i].sid = dom_sid_parse_talloc(mem_ctx, sidstr);
590         }
591
592         count = sids.num_sids;
593
594         if (handle) {
595                 struct lsa_LookupSids r;
596                 struct lsa_TransNameArray names;
597                 names.count = 0;
598                 names.names = NULL;
599
600                 r.in.handle = handle;
601                 r.in.sids = &sids;
602                 r.in.names = &names;
603                 r.in.level = 1;
604                 r.in.count = &names.count;
605                 r.out.count = &count;
606                 r.out.names = &names;
607                 
608                 status = dcerpc_lsa_LookupSids(p, mem_ctx, &r);
609                 if (!NT_STATUS_IS_OK(status)) {
610                         printf("LookupSids failed - %s\n", nt_errstr(status));
611                         return false;
612                 }
613                 
614                 printf("\n");
615                 
616                 if (!test_LookupNames(p, mem_ctx, handle, &names)) {
617                         return false;
618                 }
619         } else if (p->conn->security_state.auth_info->auth_type == DCERPC_AUTH_TYPE_SCHANNEL &&
620                    p->conn->security_state.auth_info->auth_level >= DCERPC_AUTH_LEVEL_INTEGRITY) {
621                 struct lsa_LookupSids3 r;
622                 struct lsa_TransNameArray2 names;
623
624                 names.count = 0;
625                 names.names = NULL;
626
627                 printf("\nTesting LookupSids3\n");
628                 
629                 r.in.sids = &sids;
630                 r.in.names = &names;
631                 r.in.level = 1;
632                 r.in.count = &count;
633                 r.in.unknown1 = 0;
634                 r.in.unknown2 = 0;
635                 r.out.count = &count;
636                 r.out.names = &names;
637                 
638                 status = dcerpc_lsa_LookupSids3(p, mem_ctx, &r);
639                 if (!NT_STATUS_IS_OK(status)) {
640                         if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
641                             NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
642                                 printf("not considering %s to be an error\n", nt_errstr(status));
643                                 return true;
644                         }
645                         printf("LookupSids3 failed - %s\n", 
646                                nt_errstr(status));
647                         return false;
648                 }
649                 if (!test_LookupNames4(p, mem_ctx, &names)) {
650                         return false;
651                 }
652         }
653
654         printf("\n");
655
656
657
658         return true;
659 }
660
661 static void lookupsids_cb(struct rpc_request *req)
662 {
663         int *replies = (int *)req->async.private_data;
664         NTSTATUS status;
665
666         status = dcerpc_ndr_request_recv(req);
667         if (!NT_STATUS_IS_OK(status)) {
668                 printf("lookupsids returned %s\n", nt_errstr(status));
669                 *replies = -1;
670         }
671
672         if (*replies >= 0) {
673                 *replies += 1;
674         }
675 }
676
677 static bool test_LookupSids_async(struct dcerpc_pipe *p, 
678                                   TALLOC_CTX *mem_ctx, 
679                                   struct policy_handle *handle)
680 {
681         struct lsa_SidArray sids;
682         struct lsa_SidPtr sidptr;
683         uint32_t *count;
684         struct lsa_TransNameArray *names;
685         struct lsa_LookupSids *r;
686         struct rpc_request **req;
687         int i, replies;
688         bool ret = true;
689         const int num_async_requests = 50;
690
691         count = talloc_array(mem_ctx, uint32_t, num_async_requests);
692         names = talloc_array(mem_ctx, struct lsa_TransNameArray, num_async_requests);
693         r = talloc_array(mem_ctx, struct lsa_LookupSids, num_async_requests);
694
695         printf("\nTesting %d async lookupsids request\n", num_async_requests);
696
697         req = talloc_array(mem_ctx, struct rpc_request *, num_async_requests);
698
699         sids.num_sids = 1;
700         sids.sids = &sidptr;
701         sidptr.sid = dom_sid_parse_talloc(mem_ctx, "S-1-5-32-545");
702
703         replies = 0;
704
705         for (i=0; i<num_async_requests; i++) {
706                 count[i] = 0;
707                 names[i].count = 0;
708                 names[i].names = NULL;
709
710                 r[i].in.handle = handle;
711                 r[i].in.sids = &sids;
712                 r[i].in.names = &names[i];
713                 r[i].in.level = 1;
714                 r[i].in.count = &names[i].count;
715                 r[i].out.count = &count[i];
716                 r[i].out.names = &names[i];
717                 
718                 req[i] = dcerpc_lsa_LookupSids_send(p, req, &r[i]);
719                 if (req[i] == NULL) {
720                         ret = false;
721                         break;
722                 }
723
724                 req[i]->async.callback = lookupsids_cb;
725                 req[i]->async.private_data = &replies;
726         }
727
728         while (replies >= 0 && replies < num_async_requests) {
729                 event_loop_once(p->conn->event_ctx);
730         }
731
732         talloc_free(req);
733
734         if (replies < 0) {
735                 ret = false;
736         }
737
738         return ret;
739 }
740
741 static bool test_LookupPrivValue(struct dcerpc_pipe *p, 
742                                  TALLOC_CTX *mem_ctx, 
743                                  struct policy_handle *handle,
744                                  struct lsa_String *name)
745 {
746         NTSTATUS status;
747         struct lsa_LookupPrivValue r;
748         struct lsa_LUID luid;
749
750         r.in.handle = handle;
751         r.in.name = name;
752         r.out.luid = &luid;
753
754         status = dcerpc_lsa_LookupPrivValue(p, mem_ctx, &r);
755         if (!NT_STATUS_IS_OK(status)) {
756                 printf("\nLookupPrivValue failed - %s\n", nt_errstr(status));
757                 return false;
758         }
759
760         return true;
761 }
762
763 static bool test_LookupPrivName(struct dcerpc_pipe *p, 
764                                 TALLOC_CTX *mem_ctx, 
765                                 struct policy_handle *handle,
766                                 struct lsa_LUID *luid)
767 {
768         NTSTATUS status;
769         struct lsa_LookupPrivName r;
770
771         r.in.handle = handle;
772         r.in.luid = luid;
773
774         status = dcerpc_lsa_LookupPrivName(p, mem_ctx, &r);
775         if (!NT_STATUS_IS_OK(status)) {
776                 printf("\nLookupPrivName failed - %s\n", nt_errstr(status));
777                 return false;
778         }
779
780         return true;
781 }
782
783 static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p, 
784                                              TALLOC_CTX *mem_ctx,                                 
785                                              struct policy_handle *handle,
786                                              struct policy_handle *acct_handle,
787                                              struct lsa_LUID *luid)
788 {
789         NTSTATUS status;
790         struct lsa_RemovePrivilegesFromAccount r;
791         struct lsa_PrivilegeSet privs;
792         bool ret = true;
793
794         printf("Testing RemovePrivilegesFromAccount\n");
795
796         r.in.handle = acct_handle;
797         r.in.remove_all = 0;
798         r.in.privs = &privs;
799
800         privs.count = 1;
801         privs.unknown = 0;
802         privs.set = talloc_array(mem_ctx, struct lsa_LUIDAttribute, 1);
803         privs.set[0].luid = *luid;
804         privs.set[0].attribute = 0;
805
806         status = dcerpc_lsa_RemovePrivilegesFromAccount(p, mem_ctx, &r);
807         if (!NT_STATUS_IS_OK(status)) {
808                 
809                 struct lsa_LookupPrivName r_name;
810                 
811                 r_name.in.handle = handle;
812                 r_name.in.luid = luid;
813                 
814                 status = dcerpc_lsa_LookupPrivName(p, mem_ctx, &r_name);
815                 if (!NT_STATUS_IS_OK(status)) {
816                         printf("\nLookupPrivName failed - %s\n", nt_errstr(status));
817                         return false;
818                 }
819                 /* Windows 2008 does not allow this to be removed */
820                 if (strcmp("SeAuditPrivilege", r_name.out.name->string) == 0) {
821                         return ret;
822                 }
823
824                 printf("RemovePrivilegesFromAccount failed to remove %s - %s\n", 
825                        r_name.out.name->string, 
826                        nt_errstr(status));
827                 return false;
828         }
829
830         return ret;
831 }
832
833 static bool test_AddPrivilegesToAccount(struct dcerpc_pipe *p, 
834                                         TALLOC_CTX *mem_ctx,                              
835                                         struct policy_handle *acct_handle,
836                                         struct lsa_LUID *luid)
837 {
838         NTSTATUS status;
839         struct lsa_AddPrivilegesToAccount r;
840         struct lsa_PrivilegeSet privs;
841         bool ret = true;
842
843         printf("Testing AddPrivilegesToAccount\n");
844
845         r.in.handle = acct_handle;
846         r.in.privs = &privs;
847
848         privs.count = 1;
849         privs.unknown = 0;
850         privs.set = talloc_array(mem_ctx, struct lsa_LUIDAttribute, 1);
851         privs.set[0].luid = *luid;
852         privs.set[0].attribute = 0;
853
854         status = dcerpc_lsa_AddPrivilegesToAccount(p, mem_ctx, &r);
855         if (!NT_STATUS_IS_OK(status)) {
856                 printf("AddPrivilegesToAccount failed - %s\n", nt_errstr(status));
857                 return false;
858         }
859
860         return ret;
861 }
862
863 static bool test_EnumPrivsAccount(struct dcerpc_pipe *p, 
864                                   TALLOC_CTX *mem_ctx,                            
865                                   struct policy_handle *handle,
866                                   struct policy_handle *acct_handle)
867 {
868         NTSTATUS status;
869         struct lsa_EnumPrivsAccount r;
870         bool ret = true;
871
872         printf("Testing EnumPrivsAccount\n");
873
874         r.in.handle = acct_handle;
875
876         status = dcerpc_lsa_EnumPrivsAccount(p, mem_ctx, &r);
877         if (!NT_STATUS_IS_OK(status)) {
878                 printf("EnumPrivsAccount failed - %s\n", nt_errstr(status));
879                 return false;
880         }
881
882         if (r.out.privs && r.out.privs->count > 0) {
883                 int i;
884                 for (i=0;i<r.out.privs->count;i++) {
885                         test_LookupPrivName(p, mem_ctx, handle, 
886                                             &r.out.privs->set[i].luid);
887                 }
888
889                 ret &= test_RemovePrivilegesFromAccount(p, mem_ctx, handle, acct_handle, 
890                                                         &r.out.privs->set[0].luid);
891                 ret &= test_AddPrivilegesToAccount(p, mem_ctx, acct_handle, 
892                                                    &r.out.privs->set[0].luid);
893         }
894
895         return ret;
896 }
897
898 static bool test_Delete(struct dcerpc_pipe *p, 
899                        TALLOC_CTX *mem_ctx, 
900                        struct policy_handle *handle)
901 {
902         NTSTATUS status;
903         struct lsa_Delete r;
904
905         printf("testing Delete\n");
906
907         r.in.handle = handle;
908         status = dcerpc_lsa_Delete(p, mem_ctx, &r);
909         if (!NT_STATUS_EQUAL(status, NT_STATUS_NOT_SUPPORTED)) {
910                 printf("Delete should have failed NT_STATUS_NOT_SUPPORTED - %s\n", nt_errstr(status));
911                 return false;
912         }
913
914         return true;
915 }
916
917 static bool test_DeleteObject(struct dcerpc_pipe *p, 
918                               TALLOC_CTX *mem_ctx, 
919                               struct policy_handle *handle)
920 {
921         NTSTATUS status;
922         struct lsa_DeleteObject r;
923
924         printf("testing DeleteObject\n");
925
926         r.in.handle = handle;
927         r.out.handle = handle;
928         status = dcerpc_lsa_DeleteObject(p, mem_ctx, &r);
929         if (!NT_STATUS_IS_OK(status)) {
930                 printf("Delete failed - %s\n", nt_errstr(status));
931                 return false;
932         }
933
934         return true;
935 }
936
937
938 static bool test_CreateAccount(struct dcerpc_pipe *p, 
939                                TALLOC_CTX *mem_ctx, 
940                                struct policy_handle *handle)
941 {
942         NTSTATUS status;
943         struct lsa_CreateAccount r;
944         struct dom_sid2 *newsid;
945         struct policy_handle acct_handle;
946
947         newsid = dom_sid_parse_talloc(mem_ctx, "S-1-5-12349876-4321-2854");
948
949         printf("Testing CreateAccount\n");
950
951         r.in.handle = handle;
952         r.in.sid = newsid;
953         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
954         r.out.acct_handle = &acct_handle;
955
956         status = dcerpc_lsa_CreateAccount(p, mem_ctx, &r);
957         if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
958                 struct lsa_OpenAccount r_o;
959                 r_o.in.handle = handle;
960                 r_o.in.sid = newsid;
961                 r_o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
962                 r_o.out.acct_handle = &acct_handle;
963                 
964                 status = dcerpc_lsa_OpenAccount(p, mem_ctx, &r_o);
965                 if (!NT_STATUS_IS_OK(status)) {
966                         printf("OpenAccount failed - %s\n", nt_errstr(status));
967                         return false;
968                 }
969         } else if (!NT_STATUS_IS_OK(status)) {
970                 printf("CreateAccount failed - %s\n", nt_errstr(status));
971                 return false;
972         }
973
974         if (!test_Delete(p, mem_ctx, &acct_handle)) {
975                 return false;
976         }
977
978         if (!test_DeleteObject(p, mem_ctx, &acct_handle)) {
979                 return false;
980         }
981
982         return true;
983 }
984
985 static bool test_DeleteTrustedDomain(struct dcerpc_pipe *p, 
986                                      TALLOC_CTX *mem_ctx, 
987                                      struct policy_handle *handle,
988                                      struct lsa_StringLarge name)
989 {
990         NTSTATUS status;
991         struct lsa_OpenTrustedDomainByName r;
992         struct policy_handle trustdom_handle;
993
994         r.in.handle = handle;
995         r.in.name.string = name.string;
996         r.in.access_mask = SEC_STD_DELETE;
997         r.out.trustdom_handle = &trustdom_handle;
998
999         status = dcerpc_lsa_OpenTrustedDomainByName(p, mem_ctx, &r);
1000         if (!NT_STATUS_IS_OK(status)) {
1001                 printf("lsa_OpenTrustedDomainByName failed - %s\n", nt_errstr(status));
1002                 return false;
1003         }
1004
1005         if (!test_Delete(p, mem_ctx, &trustdom_handle)) {
1006                 return false;
1007         }
1008
1009         if (!test_DeleteObject(p, mem_ctx, &trustdom_handle)) {
1010                 return false;
1011         }
1012
1013         return true;
1014 }
1015
1016 static bool test_DeleteTrustedDomainBySid(struct dcerpc_pipe *p, 
1017                                           TALLOC_CTX *mem_ctx, 
1018                                           struct policy_handle *handle,
1019                                           struct dom_sid *sid)
1020 {
1021         NTSTATUS status;
1022         struct lsa_DeleteTrustedDomain r;
1023
1024         r.in.handle = handle;
1025         r.in.dom_sid = sid;
1026
1027         status = dcerpc_lsa_DeleteTrustedDomain(p, mem_ctx, &r);
1028         if (!NT_STATUS_IS_OK(status)) {
1029                 printf("lsa_DeleteTrustedDomain failed - %s\n", nt_errstr(status));
1030                 return false;
1031         }
1032
1033         return true;
1034 }
1035
1036
1037 static bool test_CreateSecret(struct dcerpc_pipe *p, 
1038                               TALLOC_CTX *mem_ctx, 
1039                               struct policy_handle *handle)
1040 {
1041         NTSTATUS status;
1042         struct lsa_CreateSecret r;
1043         struct lsa_OpenSecret r2;
1044         struct lsa_SetSecret r3;
1045         struct lsa_QuerySecret r4;
1046         struct lsa_SetSecret r5;
1047         struct lsa_QuerySecret r6;
1048         struct lsa_SetSecret r7;
1049         struct lsa_QuerySecret r8;
1050         struct policy_handle sec_handle, sec_handle2, sec_handle3;
1051         struct lsa_DeleteObject d_o;
1052         struct lsa_DATA_BUF buf1;
1053         struct lsa_DATA_BUF_PTR bufp1;
1054         struct lsa_DATA_BUF_PTR bufp2;
1055         DATA_BLOB enc_key;
1056         bool ret = true;
1057         DATA_BLOB session_key;
1058         NTTIME old_mtime, new_mtime;
1059         DATA_BLOB blob1, blob2;
1060         const char *secret1 = "abcdef12345699qwerty";
1061         char *secret2;
1062         const char *secret3 = "ABCDEF12345699QWERTY";
1063         char *secret4;
1064         const char *secret5 = "NEW-SAMBA4-SECRET";
1065         char *secret6;
1066         char *secname[2];
1067         int i;
1068         const int LOCAL = 0;
1069         const int GLOBAL = 1;
1070
1071         secname[LOCAL] = talloc_asprintf(mem_ctx, "torturesecret-%u", (uint_t)random());
1072         secname[GLOBAL] = talloc_asprintf(mem_ctx, "G$torturesecret-%u", (uint_t)random());
1073
1074         for (i=0; i< 2; i++) {
1075                 printf("Testing CreateSecret of %s\n", secname[i]);
1076                 
1077                 init_lsa_String(&r.in.name, secname[i]);
1078                 
1079                 r.in.handle = handle;
1080                 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1081                 r.out.sec_handle = &sec_handle;
1082                 
1083                 status = dcerpc_lsa_CreateSecret(p, mem_ctx, &r);
1084                 if (!NT_STATUS_IS_OK(status)) {
1085                         printf("CreateSecret failed - %s\n", nt_errstr(status));
1086                         return false;
1087                 }
1088                 
1089                 r.in.handle = handle;
1090                 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1091                 r.out.sec_handle = &sec_handle3;
1092                 
1093                 status = dcerpc_lsa_CreateSecret(p, mem_ctx, &r);
1094                 if (!NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
1095                         printf("CreateSecret should have failed OBJECT_NAME_COLLISION - %s\n", nt_errstr(status));
1096                         return false;
1097                 }
1098                 
1099                 r2.in.handle = handle;
1100                 r2.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1101                 r2.in.name = r.in.name;
1102                 r2.out.sec_handle = &sec_handle2;
1103                 
1104                 printf("Testing OpenSecret\n");
1105                 
1106                 status = dcerpc_lsa_OpenSecret(p, mem_ctx, &r2);
1107                 if (!NT_STATUS_IS_OK(status)) {
1108                         printf("OpenSecret failed - %s\n", nt_errstr(status));
1109                         return false;
1110                 }
1111                 
1112                 status = dcerpc_fetch_session_key(p, &session_key);
1113                 if (!NT_STATUS_IS_OK(status)) {
1114                         printf("dcerpc_fetch_session_key failed - %s\n", nt_errstr(status));
1115                         return false;
1116                 }
1117                 
1118                 enc_key = sess_encrypt_string(secret1, &session_key);
1119                 
1120                 r3.in.sec_handle = &sec_handle;
1121                 r3.in.new_val = &buf1;
1122                 r3.in.old_val = NULL;
1123                 r3.in.new_val->data = enc_key.data;
1124                 r3.in.new_val->length = enc_key.length;
1125                 r3.in.new_val->size = enc_key.length;
1126                 
1127                 printf("Testing SetSecret\n");
1128                 
1129                 status = dcerpc_lsa_SetSecret(p, mem_ctx, &r3);
1130                 if (!NT_STATUS_IS_OK(status)) {
1131                         printf("SetSecret failed - %s\n", nt_errstr(status));
1132                         return false;
1133                 }
1134                 
1135                 r3.in.sec_handle = &sec_handle;
1136                 r3.in.new_val = &buf1;
1137                 r3.in.old_val = NULL;
1138                 r3.in.new_val->data = enc_key.data;
1139                 r3.in.new_val->length = enc_key.length;
1140                 r3.in.new_val->size = enc_key.length;
1141                 
1142                 /* break the encrypted data */
1143                 enc_key.data[0]++;
1144
1145                 printf("Testing SetSecret with broken key\n");
1146                 
1147                 status = dcerpc_lsa_SetSecret(p, mem_ctx, &r3);
1148                 if (!NT_STATUS_EQUAL(status, NT_STATUS_UNKNOWN_REVISION)) {
1149                         printf("SetSecret should have failed UNKNOWN_REVISION - %s\n", nt_errstr(status));
1150                         ret = false;
1151                 }
1152                 
1153                 data_blob_free(&enc_key);
1154                 
1155                 ZERO_STRUCT(new_mtime);
1156                 ZERO_STRUCT(old_mtime);
1157                 
1158                 /* fetch the secret back again */
1159                 r4.in.sec_handle = &sec_handle;
1160                 r4.in.new_val = &bufp1;
1161                 r4.in.new_mtime = &new_mtime;
1162                 r4.in.old_val = NULL;
1163                 r4.in.old_mtime = NULL;
1164                 
1165                 bufp1.buf = NULL;
1166                 
1167                 printf("Testing QuerySecret\n");
1168                 status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r4);
1169                 if (!NT_STATUS_IS_OK(status)) {
1170                         printf("QuerySecret failed - %s\n", nt_errstr(status));
1171                         ret = false;
1172                 } else {
1173                         if (r4.out.new_val == NULL || r4.out.new_val->buf == NULL) {
1174                                 printf("No secret buffer returned\n");
1175                                 ret = false;
1176                         } else {
1177                                 blob1.data = r4.out.new_val->buf->data;
1178                                 blob1.length = r4.out.new_val->buf->size;
1179                                 
1180                                 blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
1181                                 
1182                                 secret2 = sess_decrypt_string(mem_ctx, 
1183                                                               &blob1, &session_key);
1184                                 
1185                                 if (strcmp(secret1, secret2) != 0) {
1186                                         printf("Returned secret (r4) '%s' doesn't match '%s'\n", 
1187                                                secret2, secret1);
1188                                         ret = false;
1189                                 }
1190                         }
1191                 }
1192                 
1193                 enc_key = sess_encrypt_string(secret3, &session_key);
1194                 
1195                 r5.in.sec_handle = &sec_handle;
1196                 r5.in.new_val = &buf1;
1197                 r5.in.old_val = NULL;
1198                 r5.in.new_val->data = enc_key.data;
1199                 r5.in.new_val->length = enc_key.length;
1200                 r5.in.new_val->size = enc_key.length;
1201
1202
1203                 msleep(200);
1204                 printf("Testing SetSecret (existing value should move to old)\n");
1205                 
1206                 status = dcerpc_lsa_SetSecret(p, mem_ctx, &r5);
1207                 if (!NT_STATUS_IS_OK(status)) {
1208                         printf("SetSecret failed - %s\n", nt_errstr(status));
1209                         ret = false;
1210                 }
1211                 
1212                 data_blob_free(&enc_key);
1213                 
1214                 ZERO_STRUCT(new_mtime);
1215                 ZERO_STRUCT(old_mtime);
1216                 
1217                 /* fetch the secret back again */
1218                 r6.in.sec_handle = &sec_handle;
1219                 r6.in.new_val = &bufp1;
1220                 r6.in.new_mtime = &new_mtime;
1221                 r6.in.old_val = &bufp2;
1222                 r6.in.old_mtime = &old_mtime;
1223                 
1224                 bufp1.buf = NULL;
1225                 bufp2.buf = NULL;
1226                 
1227                 status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r6);
1228                 if (!NT_STATUS_IS_OK(status)) {
1229                         printf("QuerySecret failed - %s\n", nt_errstr(status));
1230                         ret = false;
1231                         secret4 = NULL;
1232                 } else {
1233
1234                         if (r6.out.new_val->buf == NULL || r6.out.old_val->buf == NULL 
1235                                 || r6.out.new_mtime == NULL || r6.out.old_mtime == NULL) {
1236                                 printf("Both secret buffers and both times not returned\n");
1237                                 ret = false;
1238                                 secret4 = NULL;
1239                         } else {
1240                                 blob1.data = r6.out.new_val->buf->data;
1241                                 blob1.length = r6.out.new_val->buf->size;
1242                                 
1243                                 blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
1244                                 
1245                                 secret4 = sess_decrypt_string(mem_ctx, 
1246                                                               &blob1, &session_key);
1247                                 
1248                                 if (strcmp(secret3, secret4) != 0) {
1249                                         printf("Returned NEW secret %s doesn't match %s\n", secret4, secret3);
1250                                         ret = false;
1251                                 }
1252
1253                                 blob1.data = r6.out.old_val->buf->data;
1254                                 blob1.length = r6.out.old_val->buf->length;
1255                                 
1256                                 blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
1257                                 
1258                                 secret2 = sess_decrypt_string(mem_ctx, 
1259                                                               &blob1, &session_key);
1260                                 
1261                                 if (strcmp(secret1, secret2) != 0) {
1262                                         printf("Returned OLD secret %s doesn't match %s\n", secret2, secret1);
1263                                         ret = false;
1264                                 }
1265                                 
1266                                 if (*r6.out.new_mtime == *r6.out.old_mtime) {
1267                                         printf("Returned secret (r6-%d) %s must not have same mtime for both secrets: %s != %s\n", 
1268                                                i,
1269                                                secname[i],
1270                                                nt_time_string(mem_ctx, *r6.out.old_mtime), 
1271                                                nt_time_string(mem_ctx, *r6.out.new_mtime));
1272                                         ret = false;
1273                                 }
1274                         }
1275                 }
1276
1277                 enc_key = sess_encrypt_string(secret5, &session_key);
1278                 
1279                 r7.in.sec_handle = &sec_handle;
1280                 r7.in.old_val = &buf1;
1281                 r7.in.old_val->data = enc_key.data;
1282                 r7.in.old_val->length = enc_key.length;
1283                 r7.in.old_val->size = enc_key.length;
1284                 r7.in.new_val = NULL;
1285                 
1286                 printf("Testing SetSecret of old Secret only\n");
1287                 
1288                 status = dcerpc_lsa_SetSecret(p, mem_ctx, &r7);
1289                 if (!NT_STATUS_IS_OK(status)) {
1290                         printf("SetSecret failed - %s\n", nt_errstr(status));
1291                         ret = false;
1292                 }
1293                 
1294                 data_blob_free(&enc_key);
1295                 
1296                 /* fetch the secret back again */
1297                 r8.in.sec_handle = &sec_handle;
1298                 r8.in.new_val = &bufp1;
1299                 r8.in.new_mtime = &new_mtime;
1300                 r8.in.old_val = &bufp2;
1301                 r8.in.old_mtime = &old_mtime;
1302                 
1303                 bufp1.buf = NULL;
1304                 bufp2.buf = NULL;
1305                 
1306                 status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r8);
1307                 if (!NT_STATUS_IS_OK(status)) {
1308                         printf("QuerySecret failed - %s\n", nt_errstr(status));
1309                         ret = false;
1310                 } else {
1311                         if (!r8.out.new_val || !r8.out.old_val) {
1312                                 printf("in/out pointers not returned, despite being set on in for QuerySecret\n");
1313                                 ret = false;
1314                         } else if (r8.out.new_val->buf != NULL) {
1315                                 printf("NEW secret buffer must not be returned after OLD set\n");
1316                                 ret = false;
1317                         } else if (r8.out.old_val->buf == NULL) {
1318                                 printf("OLD secret buffer was not returned after OLD set\n");
1319                                 ret = false;
1320                         } else if (r8.out.new_mtime == NULL || r8.out.old_mtime == NULL) {
1321                                 printf("Both times not returned after OLD set\n");
1322                                 ret = false;
1323                         } else {
1324                                 blob1.data = r8.out.old_val->buf->data;
1325                                 blob1.length = r8.out.old_val->buf->size;
1326                                 
1327                                 blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
1328                                 
1329                                 secret6 = sess_decrypt_string(mem_ctx,
1330                                                               &blob1, &session_key);
1331                                 
1332                                 if (strcmp(secret5, secret6) != 0) {
1333                                         printf("Returned OLD secret %s doesn't match %s\n", secret5, secret6);
1334                                         ret = false;
1335                                 }
1336                                 
1337                                 if (*r8.out.new_mtime != *r8.out.old_mtime) {
1338                                         printf("Returned secret (r8) %s did not had same mtime for both secrets: %s != %s\n", 
1339                                                secname[i],
1340                                                nt_time_string(mem_ctx, *r8.out.old_mtime),
1341                                                nt_time_string(mem_ctx, *r8.out.new_mtime));
1342                                         ret = false;
1343                                 }
1344                         }
1345                 }
1346
1347                 if (!test_Delete(p, mem_ctx, &sec_handle)) {
1348                         ret = false;
1349                 }
1350                 
1351                 if (!test_DeleteObject(p, mem_ctx, &sec_handle)) {
1352                         return false;
1353                 }
1354
1355                 d_o.in.handle = &sec_handle2;
1356                 d_o.out.handle = &sec_handle2;
1357                 status = dcerpc_lsa_DeleteObject(p, mem_ctx, &d_o);
1358                 if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
1359                         printf("Second delete expected INVALID_HANDLE - %s\n", nt_errstr(status));
1360                         ret = false;
1361                 } else {
1362
1363                         printf("Testing OpenSecret of just-deleted secret\n");
1364                         
1365                         status = dcerpc_lsa_OpenSecret(p, mem_ctx, &r2);
1366                         if (!NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1367                                 printf("OpenSecret expected OBJECT_NAME_NOT_FOUND - %s\n", nt_errstr(status));
1368                                 ret = false;
1369                         }
1370                 }
1371                 
1372         }
1373
1374         return ret;
1375 }
1376
1377
1378 static bool test_EnumAccountRights(struct dcerpc_pipe *p, 
1379                                    TALLOC_CTX *mem_ctx, 
1380                                    struct policy_handle *acct_handle,
1381                                    struct dom_sid *sid)
1382 {
1383         NTSTATUS status;
1384         struct lsa_EnumAccountRights r;
1385         struct lsa_RightSet rights;
1386
1387         printf("Testing EnumAccountRights\n");
1388
1389         r.in.handle = acct_handle;
1390         r.in.sid = sid;
1391         r.out.rights = &rights;
1392
1393         status = dcerpc_lsa_EnumAccountRights(p, mem_ctx, &r);
1394         if (!NT_STATUS_IS_OK(status)) {
1395                 printf("EnumAccountRights of %s failed - %s\n", 
1396                        dom_sid_string(mem_ctx, sid), nt_errstr(status));
1397                 return false;
1398         }
1399
1400         return true;
1401 }
1402
1403
1404 static bool test_QuerySecurity(struct dcerpc_pipe *p, 
1405                              struct torture_context *tctx, 
1406                              struct policy_handle *handle,
1407                              struct policy_handle *acct_handle)
1408 {
1409         NTSTATUS status;
1410         struct lsa_QuerySecurity r;
1411
1412         if (torture_setting_bool(tctx, "samba4", false)) {
1413                 printf("skipping QuerySecurity test against Samba4\n");
1414                 return true;
1415         }
1416
1417         printf("Testing QuerySecurity\n");
1418
1419         r.in.handle = acct_handle;
1420         r.in.sec_info = 7;
1421
1422         status = dcerpc_lsa_QuerySecurity(p, tctx, &r);
1423         if (!NT_STATUS_IS_OK(status)) {
1424                 printf("QuerySecurity failed - %s\n", nt_errstr(status));
1425                 return false;
1426         }
1427
1428         return true;
1429 }
1430
1431 static bool test_OpenAccount(struct dcerpc_pipe *p, 
1432                              TALLOC_CTX *mem_ctx, 
1433                              struct policy_handle *handle,
1434                              struct dom_sid *sid)
1435 {
1436         NTSTATUS status;
1437         struct lsa_OpenAccount r;
1438         struct policy_handle acct_handle;
1439
1440         printf("Testing OpenAccount\n");
1441
1442         r.in.handle = handle;
1443         r.in.sid = sid;
1444         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1445         r.out.acct_handle = &acct_handle;
1446
1447         status = dcerpc_lsa_OpenAccount(p, mem_ctx, &r);
1448         if (!NT_STATUS_IS_OK(status)) {
1449                 printf("OpenAccount failed - %s\n", nt_errstr(status));
1450                 return false;
1451         }
1452
1453         if (!test_EnumPrivsAccount(p, mem_ctx, handle, &acct_handle)) {
1454                 return false;
1455         }
1456
1457         if (!test_QuerySecurity(p, mem_ctx, handle, &acct_handle)) {
1458                 return false;
1459         }
1460
1461         return true;
1462 }
1463
1464 static bool test_EnumAccounts(struct dcerpc_pipe *p, 
1465                           TALLOC_CTX *mem_ctx, 
1466                           struct policy_handle *handle)
1467 {
1468         NTSTATUS status;
1469         struct lsa_EnumAccounts r;
1470         struct lsa_SidArray sids1, sids2;
1471         uint32_t resume_handle = 0;
1472         int i;
1473         bool ret = true;
1474
1475         printf("\ntesting EnumAccounts\n");
1476
1477         r.in.handle = handle;
1478         r.in.resume_handle = &resume_handle;
1479         r.in.num_entries = 100;
1480         r.out.resume_handle = &resume_handle;
1481         r.out.sids = &sids1;
1482
1483         resume_handle = 0;
1484         while (true) {
1485                 status = dcerpc_lsa_EnumAccounts(p, mem_ctx, &r);
1486                 if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
1487                         break;
1488                 }
1489                 if (!NT_STATUS_IS_OK(status)) {
1490                         printf("EnumAccounts failed - %s\n", nt_errstr(status));
1491                         return false;
1492                 }
1493
1494                 if (!test_LookupSids(p, mem_ctx, handle, &sids1)) {
1495                         return false;
1496                 }
1497
1498                 if (!test_LookupSids2(p, mem_ctx, handle, &sids1)) {
1499                         return false;
1500                 }
1501
1502                 /* Can't test lookupSids3 here, as clearly we must not
1503                  * be on schannel, or we would not be able to do the
1504                  * rest */
1505
1506                 printf("testing all accounts\n");
1507                 for (i=0;i<sids1.num_sids;i++) {
1508                         ret &= test_OpenAccount(p, mem_ctx, handle, sids1.sids[i].sid);
1509                         ret &= test_EnumAccountRights(p, mem_ctx, handle, sids1.sids[i].sid);
1510                 }
1511                 printf("\n");
1512         }
1513
1514         if (sids1.num_sids < 3) {
1515                 return ret;
1516         }
1517         
1518         printf("trying EnumAccounts partial listing (asking for 1 at 2)\n");
1519         resume_handle = 2;
1520         r.in.num_entries = 1;
1521         r.out.sids = &sids2;
1522
1523         status = dcerpc_lsa_EnumAccounts(p, mem_ctx, &r);
1524         if (!NT_STATUS_IS_OK(status)) {
1525                 printf("EnumAccounts failed - %s\n", nt_errstr(status));
1526                 return false;
1527         }
1528
1529         if (sids2.num_sids != 1) {
1530                 printf("Returned wrong number of entries (%d)\n", sids2.num_sids);
1531                 return false;
1532         }
1533
1534         return true;
1535 }
1536
1537 static bool test_LookupPrivDisplayName(struct dcerpc_pipe *p,
1538                                 TALLOC_CTX *mem_ctx,
1539                                 struct policy_handle *handle,
1540                                 struct lsa_String *priv_name)
1541 {
1542         struct lsa_LookupPrivDisplayName r;
1543         NTSTATUS status;
1544         /* produce a reasonable range of language output without screwing up
1545            terminals */
1546         uint16_t language_id = (random() % 4) + 0x409;
1547
1548         printf("testing LookupPrivDisplayName(%s)\n", priv_name->string);
1549         
1550         r.in.handle = handle;
1551         r.in.name = priv_name;
1552         r.in.language_id = &language_id;
1553         r.out.language_id = &language_id;
1554         r.in.unknown = 0;
1555
1556         status = dcerpc_lsa_LookupPrivDisplayName(p, mem_ctx, &r);
1557         if (!NT_STATUS_IS_OK(status)) {
1558                 printf("LookupPrivDisplayName failed - %s\n", nt_errstr(status));
1559                 return false;
1560         }
1561         printf("%s -> \"%s\"  (language 0x%x/0x%x)\n", 
1562                priv_name->string, r.out.disp_name->string, 
1563                *r.in.language_id, *r.out.language_id);
1564
1565         return true;
1566 }
1567
1568 static bool test_EnumAccountsWithUserRight(struct dcerpc_pipe *p, 
1569                                 TALLOC_CTX *mem_ctx,
1570                                 struct policy_handle *handle,
1571                                 struct lsa_String *priv_name)
1572 {
1573         struct lsa_EnumAccountsWithUserRight r;
1574         struct lsa_SidArray sids;
1575         NTSTATUS status;
1576
1577         ZERO_STRUCT(sids);
1578         
1579         printf("testing EnumAccountsWithUserRight(%s)\n", priv_name->string);
1580         
1581         r.in.handle = handle;
1582         r.in.name = priv_name;
1583         r.out.sids = &sids;
1584
1585         status = dcerpc_lsa_EnumAccountsWithUserRight(p, mem_ctx, &r);
1586
1587         /* NT_STATUS_NO_MORE_ENTRIES means noone has this privilege */
1588         if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
1589                 return true;
1590         }
1591
1592         if (!NT_STATUS_IS_OK(status)) {
1593                 printf("EnumAccountsWithUserRight failed - %s\n", nt_errstr(status));
1594                 return false;
1595         }
1596         
1597         return true;
1598 }
1599
1600
1601 static bool test_EnumPrivs(struct dcerpc_pipe *p, 
1602                            TALLOC_CTX *mem_ctx, 
1603                            struct policy_handle *handle)
1604 {
1605         NTSTATUS status;
1606         struct lsa_EnumPrivs r;
1607         struct lsa_PrivArray privs1;
1608         uint32_t resume_handle = 0;
1609         int i;
1610         bool ret = true;
1611
1612         printf("\ntesting EnumPrivs\n");
1613
1614         r.in.handle = handle;
1615         r.in.resume_handle = &resume_handle;
1616         r.in.max_count = 100;
1617         r.out.resume_handle = &resume_handle;
1618         r.out.privs = &privs1;
1619
1620         resume_handle = 0;
1621         status = dcerpc_lsa_EnumPrivs(p, mem_ctx, &r);
1622         if (!NT_STATUS_IS_OK(status)) {
1623                 printf("EnumPrivs failed - %s\n", nt_errstr(status));
1624                 return false;
1625         }
1626
1627         for (i = 0; i< privs1.count; i++) {
1628                 test_LookupPrivDisplayName(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name);
1629                 test_LookupPrivValue(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name);
1630                 if (!test_EnumAccountsWithUserRight(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name)) {
1631                         ret = false;
1632                 }
1633         }
1634
1635         return ret;
1636 }
1637
1638 static bool test_QueryForestTrustInformation(struct dcerpc_pipe *p, 
1639                                              struct torture_context *tctx, 
1640                                              struct policy_handle *handle,
1641                                              const char *trusted_domain_name)
1642 {
1643         bool ret = true;
1644         struct lsa_lsaRQueryForestTrustInformation r;
1645         NTSTATUS status;
1646         struct lsa_String string;
1647         struct lsa_ForestTrustInformation info, *info_ptr;
1648
1649         printf("\nTesting lsaRQueryForestTrustInformation\n");
1650
1651         if (torture_setting_bool(tctx, "samba4", false)) {
1652                 printf("skipping QueryForestTrustInformation against Samba4\n");
1653                 return true;
1654         }
1655
1656         ZERO_STRUCT(string);
1657
1658         if (trusted_domain_name) {
1659                 init_lsa_String(&string, trusted_domain_name);
1660         }
1661
1662         info_ptr = &info;
1663
1664         r.in.handle = handle;
1665         r.in.trusted_domain_name = &string;
1666         r.in.unknown = 0;
1667         r.out.forest_trust_info = &info_ptr;
1668
1669         status = dcerpc_lsa_lsaRQueryForestTrustInformation(p, tctx, &r);
1670
1671         if (!NT_STATUS_IS_OK(status)) {
1672                 printf("lsaRQueryForestTrustInformation failed - %s\n", nt_errstr(status));
1673                 ret = false;
1674         }
1675
1676         return ret;
1677 }
1678
1679 static bool test_query_each_TrustDomEx(struct dcerpc_pipe *p, 
1680                                        TALLOC_CTX *mem_ctx, 
1681                                        struct policy_handle *handle, 
1682                                        struct lsa_DomainListEx *domains) 
1683 {
1684         int i;
1685         bool ret = true;
1686
1687         for (i=0; i< domains->count; i++) {
1688
1689                 if (domains->domains[i].trust_attributes & NETR_TRUST_ATTRIBUTE_FOREST_TRANSITIVE) {
1690                         ret &= test_QueryForestTrustInformation(p, mem_ctx, handle, 
1691                                                                 domains->domains[i].domain_name.string);
1692                 }
1693         }
1694
1695         return ret;
1696 }
1697
1698 static bool test_query_each_TrustDom(struct dcerpc_pipe *p, 
1699                                      TALLOC_CTX *mem_ctx, 
1700                                      struct policy_handle *handle, 
1701                                      struct lsa_DomainList *domains) 
1702 {
1703         NTSTATUS status;
1704         int i,j;
1705         bool ret = true;
1706                 
1707         printf("\nTesting OpenTrustedDomain, OpenTrustedDomainByName and QueryInfoTrustedDomain\n");
1708         for (i=0; i< domains->count; i++) {
1709                 struct lsa_OpenTrustedDomain trust;
1710                 struct lsa_OpenTrustedDomainByName trust_by_name;
1711                 struct policy_handle trustdom_handle;
1712                 struct policy_handle handle2;
1713                 struct lsa_Close c;
1714                 struct lsa_CloseTrustedDomainEx c_trust;
1715                 int levels [] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13};
1716                 int ok[]      = {1, 0, 1, 0, 0, 1, 0, 1, 0,  0,  0,  1, 1};
1717
1718                 if (domains->domains[i].sid) {
1719                         trust.in.handle = handle;
1720                         trust.in.sid = domains->domains[i].sid;
1721                         trust.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1722                         trust.out.trustdom_handle = &trustdom_handle;
1723                         
1724                         status = dcerpc_lsa_OpenTrustedDomain(p, mem_ctx, &trust);
1725                         
1726                         if (!NT_STATUS_IS_OK(status)) {
1727                                 printf("OpenTrustedDomain failed - %s\n", nt_errstr(status));
1728                                 return false;
1729                         }
1730                         
1731                         c.in.handle = &trustdom_handle;
1732                         c.out.handle = &handle2;
1733                         
1734                         c_trust.in.handle = &trustdom_handle;
1735                         c_trust.out.handle = &handle2;
1736                         
1737                         for (j=0; j < ARRAY_SIZE(levels); j++) {
1738                                 struct lsa_QueryTrustedDomainInfo q;
1739                                 union lsa_TrustedDomainInfo info;
1740                                 q.in.trustdom_handle = &trustdom_handle;
1741                                 q.in.level = levels[j];
1742                                 q.out.info = &info;
1743                                 status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
1744                                 if (!NT_STATUS_IS_OK(status) && ok[j]) {
1745                                         printf("QueryTrustedDomainInfo level %d failed - %s\n", 
1746                                                levels[j], nt_errstr(status));
1747                                         ret = false;
1748                                 } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
1749                                         printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n", 
1750                                                levels[j], nt_errstr(status));
1751                                         ret = false;
1752                                 }
1753                         }
1754                         
1755                         status = dcerpc_lsa_CloseTrustedDomainEx(p, mem_ctx, &c_trust);
1756                         if (!NT_STATUS_EQUAL(status, NT_STATUS_NOT_IMPLEMENTED)) {
1757                                 printf("Expected CloseTrustedDomainEx to return NT_STATUS_NOT_IMPLEMENTED, instead - %s\n", nt_errstr(status));
1758                                 return false;
1759                         }
1760                         
1761                         c.in.handle = &trustdom_handle;
1762                         c.out.handle = &handle2;
1763                         
1764                         status = dcerpc_lsa_Close(p, mem_ctx, &c);
1765                         if (!NT_STATUS_IS_OK(status)) {
1766                                 printf("Close of trusted domain failed - %s\n", nt_errstr(status));
1767                                 return false;
1768                         }
1769
1770                         for (j=0; j < ARRAY_SIZE(levels); j++) {
1771                                 struct lsa_QueryTrustedDomainInfoBySid q;
1772                                 union lsa_TrustedDomainInfo info;
1773                                 
1774                                 if (!domains->domains[i].sid) {
1775                                         continue;
1776                                 }
1777                                 
1778                                 q.in.handle  = handle;
1779                                 q.in.dom_sid = domains->domains[i].sid;
1780                                 q.in.level   = levels[j];
1781                                 q.out.info   = &info;
1782                                 status = dcerpc_lsa_QueryTrustedDomainInfoBySid(p, mem_ctx, &q);
1783                                 if (!NT_STATUS_IS_OK(status) && ok[j]) {
1784                                         printf("QueryTrustedDomainInfoBySid level %d failed - %s\n", 
1785                                                levels[j], nt_errstr(status));
1786                                         ret = false;
1787                                 } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
1788                                         printf("QueryTrustedDomainInfoBySid level %d unexpectedly succeeded - %s\n", 
1789                                                levels[j], nt_errstr(status));
1790                                         ret = false;
1791                                 }
1792                         }
1793                 }
1794
1795                 trust_by_name.in.handle = handle;
1796                 trust_by_name.in.name.string = domains->domains[i].name.string;
1797                 trust_by_name.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1798                 trust_by_name.out.trustdom_handle = &trustdom_handle;
1799                         
1800                 status = dcerpc_lsa_OpenTrustedDomainByName(p, mem_ctx, &trust_by_name);
1801                         
1802                 if (!NT_STATUS_IS_OK(status)) {
1803                         printf("OpenTrustedDomainByName failed - %s\n", nt_errstr(status));
1804                         return false;
1805                 }
1806
1807                 for (j=0; j < ARRAY_SIZE(levels); j++) {
1808                         struct lsa_QueryTrustedDomainInfo q;
1809                         union lsa_TrustedDomainInfo info;
1810                         q.in.trustdom_handle = &trustdom_handle;
1811                         q.in.level = levels[j];
1812                         q.out.info = &info;
1813                         status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
1814                         if (!NT_STATUS_IS_OK(status) && ok[j]) {
1815                                 printf("QueryTrustedDomainInfo level %d failed - %s\n", 
1816                                        levels[j], nt_errstr(status));
1817                                 ret = false;
1818                         } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
1819                                 printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n", 
1820                                        levels[j], nt_errstr(status));
1821                                 ret = false;
1822                         }
1823                 }
1824                 
1825                 c.in.handle = &trustdom_handle;
1826                 c.out.handle = &handle2;
1827
1828                 status = dcerpc_lsa_Close(p, mem_ctx, &c);
1829                 if (!NT_STATUS_IS_OK(status)) {
1830                         printf("Close of trusted domain failed - %s\n", nt_errstr(status));
1831                         return false;
1832                 }
1833
1834                 for (j=0; j < ARRAY_SIZE(levels); j++) {
1835                         struct lsa_QueryTrustedDomainInfoByName q;
1836                         union lsa_TrustedDomainInfo info;
1837                         q.in.handle         = handle;
1838                         q.in.trusted_domain.string = domains->domains[i].name.string;
1839                         q.in.level          = levels[j];
1840                         q.out.info          = &info;
1841                         status = dcerpc_lsa_QueryTrustedDomainInfoByName(p, mem_ctx, &q);
1842                         if (!NT_STATUS_IS_OK(status) && ok[j]) {
1843                                 printf("QueryTrustedDomainInfoByName level %d failed - %s\n", 
1844                                        levels[j], nt_errstr(status));
1845                                 ret = false;
1846                         } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
1847                                 printf("QueryTrustedDomainInfoByName level %d unexpectedly succeeded - %s\n", 
1848                                        levels[j], nt_errstr(status));
1849                                 ret = false;
1850                         }
1851                 }
1852         }
1853         return ret;
1854 }
1855
1856 static bool test_EnumTrustDom(struct dcerpc_pipe *p, 
1857                               TALLOC_CTX *mem_ctx, 
1858                               struct policy_handle *handle)
1859 {
1860         struct lsa_EnumTrustDom r;
1861         struct lsa_EnumTrustedDomainsEx r_ex;
1862         NTSTATUS enum_status;
1863         uint32_t resume_handle = 0;
1864         struct lsa_DomainList domains;
1865         struct lsa_DomainListEx domains_ex;
1866         bool ret = true;
1867
1868         printf("\nTesting EnumTrustDom\n");
1869
1870         r.in.handle = handle;
1871         r.in.resume_handle = &resume_handle;
1872         r.in.max_size = 0;
1873         r.out.domains = &domains;
1874         r.out.resume_handle = &resume_handle;
1875         
1876         enum_status = dcerpc_lsa_EnumTrustDom(p, mem_ctx, &r);
1877         
1878         if (NT_STATUS_IS_OK(enum_status)) {
1879                 if (domains.count == 0) {
1880                         printf("EnumTrustDom failed - should have returned 'NT_STATUS_NO_MORE_ENTRIES' for 0 trusted domains\n");
1881                         return false;
1882                 }
1883         } else if (!(NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES) || NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES))) {
1884                 printf("EnumTrustDom of zero size failed - %s\n", nt_errstr(enum_status));
1885                 return false;
1886         }
1887                 
1888         /* Start from the bottom again */
1889         resume_handle = 0;
1890
1891         do {
1892                 r.in.handle = handle;
1893                 r.in.resume_handle = &resume_handle;
1894                 r.in.max_size = LSA_ENUM_TRUST_DOMAIN_MULTIPLIER * 3;
1895                 r.out.domains = &domains;
1896                 r.out.resume_handle = &resume_handle;
1897                 
1898                 enum_status = dcerpc_lsa_EnumTrustDom(p, mem_ctx, &r);
1899                 
1900                 /* NO_MORE_ENTRIES is allowed */
1901                 if (NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES)) {
1902                         return true;
1903                 } else if (NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)) {
1904                         /* Windows 2003 gets this off by one on the first run */
1905                         if (r.out.domains->count < 3 || r.out.domains->count > 4) {
1906                                 printf("EnumTrustDom didn't fill the buffer we "
1907                                        "asked it to (got %d, expected %d / %d == %d entries)\n",
1908                                        r.out.domains->count, LSA_ENUM_TRUST_DOMAIN_MULTIPLIER * 3, 
1909                                        LSA_ENUM_TRUST_DOMAIN_MULTIPLIER, r.in.max_size);
1910                                 ret = false;
1911                         }
1912                 } else if (!NT_STATUS_IS_OK(enum_status)) {
1913                         printf("EnumTrustDom failed - %s\n", nt_errstr(enum_status));
1914                         return false;
1915                 }
1916                 
1917                 if (domains.count == 0) {
1918                         printf("EnumTrustDom failed - should have returned 'NT_STATUS_NO_MORE_ENTRIES' for 0 trusted domains\n");
1919                         return false;
1920                 }
1921
1922                 ret &= test_query_each_TrustDom(p, mem_ctx, handle, &domains);
1923                 
1924         } while ((NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)));
1925
1926         printf("\nTesting EnumTrustedDomainsEx\n");
1927
1928         r_ex.in.handle = handle;
1929         r_ex.in.resume_handle = &resume_handle;
1930         r_ex.in.max_size = LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER * 3;
1931         r_ex.out.domains = &domains_ex;
1932         r_ex.out.resume_handle = &resume_handle;
1933         
1934         enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, mem_ctx, &r_ex);
1935         
1936         if (!(NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES) || NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES))) {
1937                 printf("EnumTrustedDomainEx of zero size failed - %s\n", nt_errstr(enum_status));
1938                 return false;
1939         }
1940                 
1941         resume_handle = 0;
1942         do {
1943                 r_ex.in.handle = handle;
1944                 r_ex.in.resume_handle = &resume_handle;
1945                 r_ex.in.max_size = LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER * 3;
1946                 r_ex.out.domains = &domains_ex;
1947                 r_ex.out.resume_handle = &resume_handle;
1948                 
1949                 enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, mem_ctx, &r_ex);
1950                 
1951                 /* NO_MORE_ENTRIES is allowed */
1952                 if (NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES)) {
1953                         return true;
1954                 } else if (NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)) {
1955                         /* Windows 2003 gets this off by one on the first run */
1956                         if (r_ex.out.domains->count < 3 || r_ex.out.domains->count > 4) {
1957                                 printf("EnumTrustDom didn't fill the buffer we "
1958                                        "asked it to (got %d, expected %d / %d == %d entries)\n",
1959                                        r_ex.out.domains->count, 
1960                                        r_ex.in.max_size,
1961                                        LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER, 
1962                                        r_ex.in.max_size / LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER);
1963                         }
1964                 } else if (!NT_STATUS_IS_OK(enum_status)) {
1965                         printf("EnumTrustedDomainEx failed - %s\n", nt_errstr(enum_status));
1966                         return false;
1967                 }
1968
1969                 if (domains_ex.count == 0) {
1970                         printf("EnumTrustDomainEx failed - should have returned 'NT_STATUS_NO_MORE_ENTRIES' for 0 trusted domains\n");
1971                         return false;
1972                 }
1973
1974                 ret &= test_query_each_TrustDomEx(p, mem_ctx, handle, &domains_ex);
1975                 
1976         } while ((NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)));
1977
1978         return ret;
1979 }
1980
1981 static bool test_CreateTrustedDomain(struct dcerpc_pipe *p, 
1982                                      TALLOC_CTX *mem_ctx, 
1983                                      struct policy_handle *handle)
1984 {
1985         NTSTATUS status;
1986         bool ret = true;
1987         struct lsa_CreateTrustedDomain r;
1988         struct lsa_DomainInfo trustinfo;
1989         struct dom_sid *domsid[12];
1990         struct policy_handle trustdom_handle[12];
1991         struct lsa_QueryTrustedDomainInfo q;
1992         int i;
1993
1994         printf("Testing CreateTrustedDomain for 12 domains\n");
1995
1996         if (!test_EnumTrustDom(p, mem_ctx, handle)) {
1997                 ret = false;
1998         }
1999         
2000         for (i=0; i< 12; i++) {
2001                 char *trust_name = talloc_asprintf(mem_ctx, "torturedom%02d", i);
2002                 char *trust_sid = talloc_asprintf(mem_ctx, "S-1-5-21-97398-379795-100%02d", i);
2003                 
2004                 domsid[i] = dom_sid_parse_talloc(mem_ctx, trust_sid);
2005
2006                 trustinfo.sid = domsid[i];
2007                 init_lsa_String((struct lsa_String *)&trustinfo.name, trust_name);
2008
2009                 r.in.policy_handle = handle;
2010                 r.in.info = &trustinfo;
2011                 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2012                 r.out.trustdom_handle = &trustdom_handle[i];
2013                 
2014                 status = dcerpc_lsa_CreateTrustedDomain(p, mem_ctx, &r);
2015                 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
2016                         test_DeleteTrustedDomain(p, mem_ctx, handle, trustinfo.name);
2017                         status = dcerpc_lsa_CreateTrustedDomain(p, mem_ctx, &r);
2018                 }
2019                 if (!NT_STATUS_IS_OK(status)) {
2020                         printf("CreateTrustedDomain failed - %s\n", nt_errstr(status));
2021                         ret = false;
2022                 } else {
2023                 
2024                         q.in.trustdom_handle = &trustdom_handle[i];
2025                         q.in.level = LSA_TRUSTED_DOMAIN_INFO_INFO_EX;
2026                         status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
2027                         if (!NT_STATUS_IS_OK(status)) {
2028                                 printf("QueryTrustedDomainInfo level 1 failed - %s\n", nt_errstr(status));
2029                                 ret = false;
2030                         } else if (!q.out.info) {
2031                                 ret = false;
2032                         } else {
2033                                 if (strcmp(q.out.info->info_ex.netbios_name.string, trustinfo.name.string) != 0) {
2034                                         printf("QueryTrustedDomainInfo returned inconsistant short name: %s != %s\n",
2035                                                q.out.info->info_ex.netbios_name.string, trustinfo.name.string);
2036                                         ret = false;
2037                                 }
2038                                 if (q.out.info->info_ex.trust_type != LSA_TRUST_TYPE_DOWNLEVEL) {
2039                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n", 
2040                                                trust_name, q.out.info->info_ex.trust_type, LSA_TRUST_TYPE_DOWNLEVEL);
2041                                         ret = false;
2042                                 }
2043                                 if (q.out.info->info_ex.trust_attributes != 0) {
2044                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n", 
2045                                                trust_name, q.out.info->info_ex.trust_attributes, 0);
2046                                         ret = false;
2047                                 }
2048                                 if (q.out.info->info_ex.trust_direction != LSA_TRUST_DIRECTION_OUTBOUND) {
2049                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n", 
2050                                                trust_name, q.out.info->info_ex.trust_direction, LSA_TRUST_DIRECTION_OUTBOUND);
2051                                         ret = false;
2052                                 }
2053                         }
2054                 }
2055         }
2056
2057         /* now that we have some domains to look over, we can test the enum calls */
2058         if (!test_EnumTrustDom(p, mem_ctx, handle)) {
2059                 ret = false;
2060         }
2061         
2062         for (i=0; i<12; i++) {
2063                 if (!test_DeleteTrustedDomainBySid(p, mem_ctx, handle, domsid[i])) {
2064                         ret = false;
2065                 }
2066         }
2067
2068         return ret;
2069 }
2070
2071 static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p, 
2072                                         struct torture_context *tctx, 
2073                                         TALLOC_CTX *mem_ctx, 
2074                                         struct policy_handle *handle)
2075 {
2076         NTSTATUS status;
2077         bool ret = true;
2078         struct lsa_CreateTrustedDomainEx2 r;
2079         struct lsa_TrustDomainInfoInfoEx trustinfo;
2080         struct lsa_TrustDomainInfoAuthInfoInternal authinfo;
2081         struct trustDomainPasswords auth_struct;
2082         DATA_BLOB auth_blob;
2083         struct dom_sid *domsid[12];
2084         struct policy_handle trustdom_handle[12];
2085         struct lsa_QueryTrustedDomainInfo q;
2086         DATA_BLOB session_key;
2087         enum ndr_err_code ndr_err;
2088         int i;
2089
2090         printf("Testing CreateTrustedDomainEx2 for 12 domains\n");
2091
2092         status = dcerpc_fetch_session_key(p, &session_key);
2093         if (!NT_STATUS_IS_OK(status)) {
2094                 printf("dcerpc_fetch_session_key failed - %s\n", nt_errstr(status));
2095                 return false;
2096         }
2097                 
2098         for (i=0; i< 12; i++) {
2099                 char *trust_name = talloc_asprintf(mem_ctx, "torturedom%02d", i);
2100                 char *trust_name_dns = talloc_asprintf(mem_ctx, "torturedom%02d.samba.example.com", i);
2101                 char *trust_sid = talloc_asprintf(mem_ctx, "S-1-5-21-97398-379795-100%02d", i);
2102                 
2103                 domsid[i] = dom_sid_parse_talloc(mem_ctx, trust_sid);
2104
2105                 trustinfo.sid = domsid[i];
2106                 trustinfo.netbios_name.string = trust_name;
2107                 trustinfo.domain_name.string = trust_name_dns;
2108
2109                 /* Create inbound, some outbound, and some
2110                  * bi-directional trusts in a repeating pattern based
2111                  * on i */
2112
2113                 /* 1 == inbound, 2 == outbound, 3 == both */
2114                 trustinfo.trust_direction = (i % 3) + 1;
2115
2116                 /* Try different trust types too */
2117
2118                 /* 1 == downleven (NT4), 2 == uplevel (ADS), 3 == MIT (kerberos but not AD) */
2119                 trustinfo.trust_type = (((i / 3) + 1) % 3) + 1;
2120
2121                 trustinfo.trust_attributes = LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION;
2122
2123                 generate_random_buffer(auth_struct.confounder, sizeof(auth_struct.confounder));
2124
2125                 auth_struct.outgoing.count = 0;
2126                 auth_struct.incoming.count = 0;
2127
2128                 ndr_err = ndr_push_struct_blob(&auth_blob, mem_ctx, lp_iconv_convenience(tctx->lp_ctx), &auth_struct,
2129                                                (ndr_push_flags_fn_t)ndr_push_trustDomainPasswords);
2130                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2131                         printf("ndr_push_struct_blob of trustDomainPasswords structure failed");
2132                         ret = false;
2133                 }
2134
2135                 arcfour_crypt_blob(auth_blob.data, auth_blob.length, &session_key);
2136
2137                 authinfo.auth_blob.size = auth_blob.length;
2138                 authinfo.auth_blob.data = auth_blob.data;
2139
2140                 r.in.policy_handle = handle;
2141                 r.in.info = &trustinfo;
2142                 r.in.auth_info = &authinfo;
2143                 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2144                 r.out.trustdom_handle = &trustdom_handle[i];
2145                 
2146                 status = dcerpc_lsa_CreateTrustedDomainEx2(p, mem_ctx, &r);
2147                 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
2148                         test_DeleteTrustedDomain(p, mem_ctx, handle, trustinfo.netbios_name);
2149                         status = dcerpc_lsa_CreateTrustedDomainEx2(p, mem_ctx, &r);
2150                 }
2151                 if (!NT_STATUS_IS_OK(status)) {
2152                         printf("CreateTrustedDomainEx failed2 - %s\n", nt_errstr(status));
2153                         ret = false;
2154                 } else {
2155                 
2156                         q.in.trustdom_handle = &trustdom_handle[i];
2157                         q.in.level = LSA_TRUSTED_DOMAIN_INFO_INFO_EX;
2158                         status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
2159                         if (!NT_STATUS_IS_OK(status)) {
2160                                 printf("QueryTrustedDomainInfo level 1 failed - %s\n", nt_errstr(status));
2161                                 ret = false;
2162                         } else if (!q.out.info) {
2163                                 ret = false;
2164                         } else {
2165                                 if (strcmp(q.out.info->info_ex.netbios_name.string, trustinfo.netbios_name.string) != 0) {
2166                                         printf("QueryTrustedDomainInfo returned inconsistant short name: %s != %s\n",
2167                                                q.out.info->info_ex.netbios_name.string, trustinfo.netbios_name.string);
2168                                         ret = false;
2169                                 }
2170                                 if (q.out.info->info_ex.trust_type != trustinfo.trust_type) {
2171                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n", 
2172                                                trust_name, q.out.info->info_ex.trust_type, trustinfo.trust_type);
2173                                         ret = false;
2174                                 }
2175                                 if (q.out.info->info_ex.trust_attributes != LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION) {
2176                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n", 
2177                                                trust_name, q.out.info->info_ex.trust_attributes, LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION);
2178                                         ret = false;
2179                                 }
2180                                 if (q.out.info->info_ex.trust_direction != trustinfo.trust_direction) {
2181                                         printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n", 
2182                                                trust_name, q.out.info->info_ex.trust_direction, trustinfo.trust_direction);
2183                                         ret = false;
2184                                 }
2185                         }
2186                 }
2187         }
2188
2189         /* now that we have some domains to look over, we can test the enum calls */
2190         if (!test_EnumTrustDom(p, mem_ctx, handle)) {
2191                 ret = false;
2192         }
2193         
2194         for (i=0; i<12; i++) {
2195                 if (!test_DeleteTrustedDomainBySid(p, mem_ctx, handle, domsid[i])) {
2196                         ret = false;
2197                 }
2198         }
2199
2200         return ret;
2201 }
2202
2203 static bool test_QueryDomainInfoPolicy(struct dcerpc_pipe *p, 
2204                                  struct torture_context *tctx, 
2205                                  struct policy_handle *handle)
2206 {
2207         struct lsa_QueryDomainInformationPolicy r;
2208         NTSTATUS status;
2209         int i;
2210         bool ret = true;
2211
2212         printf("\nTesting QueryDomainInformationPolicy\n");
2213
2214         for (i=2;i<4;i++) {
2215                 r.in.handle = handle;
2216                 r.in.level = i;
2217
2218                 printf("\ntrying QueryDomainInformationPolicy level %d\n", i);
2219
2220                 status = dcerpc_lsa_QueryDomainInformationPolicy(p, tctx, &r);
2221
2222                 /* If the server does not support EFS, then this is the correct return */
2223                 if (i == LSA_DOMAIN_INFO_POLICY_EFS && NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
2224                         continue;
2225                 } else if (!NT_STATUS_IS_OK(status)) {
2226                         printf("QueryDomainInformationPolicy failed - %s\n", nt_errstr(status));
2227                         ret = false;
2228                         continue;
2229                 }
2230         }
2231
2232         return ret;
2233 }
2234
2235
2236 static bool test_QueryInfoPolicy(struct dcerpc_pipe *p, 
2237                                  struct torture_context *tctx, 
2238                                  struct policy_handle *handle)
2239 {
2240         struct lsa_QueryInfoPolicy r;
2241         NTSTATUS status;
2242         int i;
2243         bool ret = true;
2244         printf("\nTesting QueryInfoPolicy\n");
2245
2246         for (i=1;i<13;i++) {
2247                 r.in.handle = handle;
2248                 r.in.level = i;
2249
2250                 printf("\ntrying QueryInfoPolicy level %d\n", i);
2251
2252                 status = dcerpc_lsa_QueryInfoPolicy(p, tctx, &r);
2253
2254                 switch (i) {
2255                 case LSA_POLICY_INFO_DB:
2256                 case LSA_POLICY_INFO_AUDIT_FULL_SET:
2257                 case LSA_POLICY_INFO_AUDIT_FULL_QUERY:
2258                         if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_PARAMETER)) {
2259                                 printf("server should have failed level %u: %s\n", i, nt_errstr(status));
2260                                 ret = false;
2261                         }
2262                         break;
2263                 case LSA_POLICY_INFO_DOMAIN:
2264                 case LSA_POLICY_INFO_ACCOUNT_DOMAIN:
2265                 case LSA_POLICY_INFO_DNS:
2266                         if (!NT_STATUS_IS_OK(status)) {
2267                                 printf("QueryInfoPolicy failed - %s\n", nt_errstr(status));
2268                                 ret = false;
2269                         }
2270                         break;
2271                 default:
2272                         if (torture_setting_bool(tctx, "samba4", false)) {
2273                                 /* Other levels not implemented yet */
2274                                 if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_INFO_CLASS)) {
2275                                         printf("QueryInfoPolicy failed - %s\n", nt_errstr(status));
2276                                         ret = false;
2277                                 }
2278                         } else if (!NT_STATUS_IS_OK(status)) {
2279                                 printf("QueryInfoPolicy failed - %s\n", nt_errstr(status));
2280                                 ret = false;
2281                         }
2282                         break;
2283                 }
2284
2285                 if (NT_STATUS_IS_OK(status) && i == LSA_POLICY_INFO_DNS) {
2286                         /* Let's look up some of these names */
2287
2288                         struct lsa_TransNameArray tnames;
2289                         tnames.count = 14;
2290                         tnames.names = talloc_zero_array(tctx, struct lsa_TranslatedName, tnames.count);
2291                         tnames.names[0].name.string = r.out.info->dns.name.string;
2292                         tnames.names[0].sid_type = SID_NAME_DOMAIN;
2293                         tnames.names[1].name.string = r.out.info->dns.dns_domain.string;
2294                         tnames.names[1].sid_type = SID_NAME_DOMAIN;
2295                         tnames.names[2].name.string = talloc_asprintf(tctx, "%s\\", r.out.info->dns.name.string);
2296                         tnames.names[2].sid_type = SID_NAME_DOMAIN;
2297                         tnames.names[3].name.string = talloc_asprintf(tctx, "%s\\", r.out.info->dns.dns_domain.string);
2298                         tnames.names[3].sid_type = SID_NAME_DOMAIN;
2299                         tnames.names[4].name.string = talloc_asprintf(tctx, "%s\\guest", r.out.info->dns.name.string);
2300                         tnames.names[4].sid_type = SID_NAME_USER;
2301                         tnames.names[5].name.string = talloc_asprintf(tctx, "%s\\krbtgt", r.out.info->dns.name.string);
2302                         tnames.names[5].sid_type = SID_NAME_USER;
2303                         tnames.names[6].name.string = talloc_asprintf(tctx, "%s\\guest", r.out.info->dns.dns_domain.string);
2304                         tnames.names[6].sid_type = SID_NAME_USER;
2305                         tnames.names[7].name.string = talloc_asprintf(tctx, "%s\\krbtgt", r.out.info->dns.dns_domain.string);
2306                         tnames.names[7].sid_type = SID_NAME_USER;
2307                         tnames.names[8].name.string = talloc_asprintf(tctx, "krbtgt@%s", r.out.info->dns.name.string);
2308                         tnames.names[8].sid_type = SID_NAME_USER;
2309                         tnames.names[9].name.string = talloc_asprintf(tctx, "krbtgt@%s", r.out.info->dns.dns_domain.string);
2310                         tnames.names[9].sid_type = SID_NAME_USER;
2311                         tnames.names[10].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", r.out.info->dns.name.string);
2312                         tnames.names[10].sid_type = SID_NAME_USER;
2313                         tnames.names[11].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", r.out.info->dns.dns_domain.string);
2314                         tnames.names[11].sid_type = SID_NAME_USER;
2315                         tnames.names[12].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", r.out.info->dns.name.string);
2316                         tnames.names[12].sid_type = SID_NAME_USER;
2317                         tnames.names[13].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", r.out.info->dns.dns_domain.string);
2318                         tnames.names[13].sid_type = SID_NAME_USER;
2319                         ret &= test_LookupNames(p, tctx, handle, &tnames);
2320
2321                 }
2322         }
2323
2324         return ret;
2325 }
2326
2327 static bool test_QueryInfoPolicy2(struct dcerpc_pipe *p, 
2328                                   struct torture_context *tctx, 
2329                                   struct policy_handle *handle)
2330 {
2331         struct lsa_QueryInfoPolicy2 r;
2332         NTSTATUS status;
2333         int i;
2334         bool ret = true;
2335         printf("\nTesting QueryInfoPolicy2\n");
2336         for (i=1;i<13;i++) {
2337                 r.in.handle = handle;
2338                 r.in.level = i;
2339
2340                 printf("\ntrying QueryInfoPolicy2 level %d\n", i);
2341
2342                 status = dcerpc_lsa_QueryInfoPolicy2(p, tctx, &r);
2343                 
2344                 switch (i) {
2345                 case LSA_POLICY_INFO_DB:
2346                 case LSA_POLICY_INFO_AUDIT_FULL_SET:
2347                 case LSA_POLICY_INFO_AUDIT_FULL_QUERY:
2348                         if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_PARAMETER)) {
2349                                 printf("server should have failed level %u: %s\n", i, nt_errstr(status));
2350                                 ret = false;
2351                         }
2352                         break;
2353                 case LSA_POLICY_INFO_DOMAIN:
2354                 case LSA_POLICY_INFO_ACCOUNT_DOMAIN:
2355                 case LSA_POLICY_INFO_DNS:
2356                         if (!NT_STATUS_IS_OK(status)) {
2357                                 printf("QueryInfoPolicy2 failed - %s\n", nt_errstr(status));
2358                                 ret = false;
2359                         }
2360                         break;
2361                 default:
2362                         if (torture_setting_bool(tctx, "samba4", false)) {
2363                                 /* Other levels not implemented yet */
2364                                 if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_INFO_CLASS)) {
2365                                         printf("QueryInfoPolicy2 failed - %s\n", nt_errstr(status));
2366                                         ret = false;
2367                                 }
2368                         } else if (!NT_STATUS_IS_OK(status)) {
2369                                 printf("QueryInfoPolicy2 failed - %s\n", nt_errstr(status));
2370                                 ret = false;
2371                         }
2372                         break;
2373                 }
2374         }
2375
2376         return ret;
2377 }
2378
2379 static bool test_GetUserName(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
2380 {
2381         struct lsa_GetUserName r;
2382         NTSTATUS status;
2383         bool ret = true;
2384         struct lsa_StringPointer authority_name_p;
2385
2386         printf("\nTesting GetUserName\n");
2387
2388         r.in.system_name = "\\";
2389         r.in.account_name = NULL;
2390         r.in.authority_name = &authority_name_p;
2391         authority_name_p.string = NULL;
2392
2393         status = dcerpc_lsa_GetUserName(p, mem_ctx, &r);
2394
2395         if (!NT_STATUS_IS_OK(status)) {
2396                 printf("GetUserName failed - %s\n", nt_errstr(status));
2397                 ret = false;
2398         }
2399
2400         return ret;
2401 }
2402
2403 bool test_lsa_Close(struct dcerpc_pipe *p, 
2404                     TALLOC_CTX *mem_ctx, 
2405                     struct policy_handle *handle)
2406 {
2407         NTSTATUS status;
2408         struct lsa_Close r;
2409         struct policy_handle handle2;
2410
2411         printf("\ntesting Close\n");
2412
2413         r.in.handle = handle;
2414         r.out.handle = &handle2;
2415
2416         status = dcerpc_lsa_Close(p, mem_ctx, &r);
2417         if (!NT_STATUS_IS_OK(status)) {
2418                 printf("Close failed - %s\n", nt_errstr(status));
2419                 return false;
2420         }
2421
2422         status = dcerpc_lsa_Close(p, mem_ctx, &r);
2423         /* its really a fault - we need a status code for rpc fault */
2424         if (!NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
2425                 printf("Close failed - %s\n", nt_errstr(status));
2426                 return false;
2427         }
2428
2429         printf("\n");
2430
2431         return true;
2432 }
2433
2434 bool torture_rpc_lsa(struct torture_context *tctx)
2435 {
2436         NTSTATUS status;
2437         struct dcerpc_pipe *p;
2438         bool ret = true;
2439         struct policy_handle *handle;
2440         struct test_join *join = NULL;
2441         struct cli_credentials *machine_creds;
2442
2443         status = torture_rpc_connection(tctx, &p, &ndr_table_lsarpc);
2444         if (!NT_STATUS_IS_OK(status)) {
2445                 return false;
2446         }
2447
2448         if (!test_OpenPolicy(p, tctx)) {
2449                 ret = false;
2450         }
2451
2452         if (!test_lsa_OpenPolicy2(p, tctx, &handle)) {
2453                 ret = false;
2454         }
2455
2456         if (handle) {
2457                 join = torture_join_domain(tctx, TEST_MACHINENAME, ACB_WSTRUST, &machine_creds);
2458                 if (!join) {
2459                         ret = false;
2460                 }
2461
2462                 if (!test_LookupNames_wellknown(p, tctx, handle)) {
2463                         ret = false;
2464                 }               
2465
2466                 if (!test_LookupNames_bogus(p, tctx, handle)) {
2467                         ret = false;
2468                 }               
2469
2470                 if (!test_LookupSids_async(p, tctx, handle)) {
2471                         ret = false;
2472                 }
2473
2474                 if (!test_QueryDomainInfoPolicy(p, tctx, handle)) {
2475                         ret = false;
2476                 }
2477                 
2478                 if (!test_CreateAccount(p, tctx, handle)) {
2479                         ret = false;
2480                 }
2481                 
2482                 if (!test_CreateSecret(p, tctx, handle)) {
2483                         ret = false;
2484                 }
2485                 
2486                 if (!test_CreateTrustedDomain(p, tctx, handle)) {
2487                         ret = false;
2488                 }
2489
2490                 if (!test_CreateTrustedDomainEx2(p, tctx, tctx, handle)) {
2491                         ret = false;
2492                 }
2493
2494                 if (!test_EnumAccounts(p, tctx, handle)) {
2495                         ret = false;
2496                 }
2497                 
2498                 if (!test_EnumPrivs(p, tctx, handle)) {
2499                         ret = false;
2500                 }
2501                 
2502                 if (!test_QueryInfoPolicy(p, tctx, handle)) {
2503                         ret = false;
2504                 }
2505                 
2506                 if (!test_QueryInfoPolicy2(p, tctx, handle)) {
2507                         ret = false;
2508                 }
2509                 
2510                 if (!test_Delete(p, tctx, handle)) {
2511                         ret = false;
2512                 }
2513                 
2514                 if (!test_many_LookupSids(p, tctx, handle)) {
2515                         ret = false;
2516                 }
2517                 
2518                 if (!test_lsa_Close(p, tctx, handle)) {
2519                         ret = false;
2520                 }
2521                 
2522                 torture_leave_domain(tctx, join);
2523
2524         } else {
2525                 if (!test_many_LookupSids(p, tctx, handle)) {
2526                         ret = false;
2527                 }
2528         }
2529
2530         if (!test_GetUserName(p, tctx)) {
2531                 ret = false;
2532         }
2533
2534         return ret;
2535 }
2536
2537 bool torture_rpc_lsa_get_user(struct torture_context *torture)
2538 {
2539         NTSTATUS status;
2540         struct dcerpc_pipe *p;
2541         TALLOC_CTX *mem_ctx;
2542         bool ret = true;
2543
2544         mem_ctx = talloc_init("torture_rpc_lsa_get_user");
2545
2546         status = torture_rpc_connection(torture, &p, &ndr_table_lsarpc);
2547         if (!NT_STATUS_IS_OK(status)) {
2548                 talloc_free(mem_ctx);
2549                 return false;
2550         }
2551
2552         if (!test_GetUserName(p, mem_ctx)) {
2553                 ret = false;
2554         }
2555                 
2556         talloc_free(mem_ctx);
2557
2558         return ret;
2559 }