Always initialise this, to assist callers doing loops over this call.
[samba.git] / source / rpc_client / cli_samr.c
1 /* 
2    Unix SMB/CIFS implementation.
3    RPC pipe client
4    Copyright (C) Tim Potter                        2000-2001,
5    Copyright (C) Andrew Tridgell              1992-1997,2000,
6    Copyright (C) Luke Kenneth Casson Leighton 1996-1997,2000,
7    Copyright (C) Paul Ashton                       1997,2000,
8    Copyright (C) Elrond                                 2000,
9    Copyright (C) Rafal Szczesniak                       2002.
10    
11    This program is free software; you can redistribute it and/or modify
12    it under the terms of the GNU General Public License as published by
13    the Free Software Foundation; either version 2 of the License, or
14    (at your option) any later version.
15    
16    This program is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19    GNU General Public License for more details.
20    
21    You should have received a copy of the GNU General Public License
22    along with this program; if not, write to the Free Software
23    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 */
25
26 #include "includes.h"
27
28 /* Connect to SAMR database */
29
30 NTSTATUS cli_samr_connect(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
31                           uint32 access_mask, POLICY_HND *connect_pol)
32 {
33         prs_struct qbuf, rbuf;
34         SAMR_Q_CONNECT q;
35         SAMR_R_CONNECT r;
36         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
37
38         ZERO_STRUCT(q);
39         ZERO_STRUCT(r);
40
41         /* Initialise parse structures */
42
43         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
44         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
45
46         /* Marshall data and send request */
47
48         init_samr_q_connect(&q, cli->desthost, access_mask);
49
50         if (!samr_io_q_connect("", &q, &qbuf, 0) ||
51             !rpc_api_pipe_req(cli, SAMR_CONNECT, &qbuf, &rbuf))
52                 goto done;
53
54         /* Unmarshall response */
55
56         if (!samr_io_r_connect("", &r, &rbuf, 0))
57                 goto done;
58
59         /* Return output parameters */
60
61         if (NT_STATUS_IS_OK(result = r.status)) {
62                 *connect_pol = r.connect_pol;
63 #ifdef __INSURE__
64                 connect_pol->marker = malloc(1);
65 #endif
66         }
67
68  done:
69         prs_mem_free(&qbuf);
70         prs_mem_free(&rbuf);
71
72         return result;
73 }
74
75 /* Connect to SAMR database */
76
77 NTSTATUS cli_samr_connect4(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
78                            uint32 access_mask, POLICY_HND *connect_pol)
79 {
80         prs_struct qbuf, rbuf;
81         SAMR_Q_CONNECT4 q;
82         SAMR_R_CONNECT4 r;
83         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
84
85         ZERO_STRUCT(q);
86         ZERO_STRUCT(r);
87
88         /* Initialise parse structures */
89
90         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
91         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
92
93         /* Marshall data and send request */
94
95         init_samr_q_connect4(&q, cli->desthost, access_mask);
96
97         if (!samr_io_q_connect4("", &q, &qbuf, 0) ||
98             !rpc_api_pipe_req(cli, SAMR_CONNECT4, &qbuf, &rbuf))
99                 goto done;
100
101         /* Unmarshall response */
102
103         if (!samr_io_r_connect4("", &r, &rbuf, 0))
104                 goto done;
105
106         /* Return output parameters */
107
108         if (NT_STATUS_IS_OK(result = r.status)) {
109                 *connect_pol = r.connect_pol;
110 #ifdef __INSURE__
111                 connect_pol->marker = malloc(1);
112 #endif
113         }
114
115  done:
116         prs_mem_free(&qbuf);
117         prs_mem_free(&rbuf);
118
119         return result;
120 }
121
122 /* Close SAMR handle */
123
124 NTSTATUS cli_samr_close(struct cli_state *cli, TALLOC_CTX *mem_ctx,
125                         POLICY_HND *connect_pol)
126 {
127         prs_struct qbuf, rbuf;
128         SAMR_Q_CLOSE_HND q;
129         SAMR_R_CLOSE_HND r;
130         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
131
132         ZERO_STRUCT(q);
133         ZERO_STRUCT(r);
134
135         /* Initialise parse structures */
136
137         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
138         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
139
140         /* Marshall data and send request */
141
142         init_samr_q_close_hnd(&q, connect_pol);
143
144         if (!samr_io_q_close_hnd("", &q, &qbuf, 0) ||
145             !rpc_api_pipe_req(cli, SAMR_CLOSE_HND, &qbuf, &rbuf))
146                 goto done;
147
148         /* Unmarshall response */
149
150         if (!samr_io_r_close_hnd("", &r, &rbuf, 0))
151                 goto done;
152
153         /* Return output parameters */
154
155         if (NT_STATUS_IS_OK(result = r.status)) {
156 #ifdef __INSURE__
157                 SAFE_FREE(connect_pol->marker);
158 #endif
159                 *connect_pol = r.pol;
160         }
161
162  done:
163         prs_mem_free(&qbuf);
164         prs_mem_free(&rbuf);
165
166         return result;
167 }
168
169 /* Open handle on a domain */
170
171 NTSTATUS cli_samr_open_domain(struct cli_state *cli, TALLOC_CTX *mem_ctx,
172                               POLICY_HND *connect_pol, uint32 access_mask, 
173                               const DOM_SID *domain_sid, POLICY_HND *domain_pol)
174 {
175         prs_struct qbuf, rbuf;
176         SAMR_Q_OPEN_DOMAIN q;
177         SAMR_R_OPEN_DOMAIN r;
178         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
179
180         ZERO_STRUCT(q);
181         ZERO_STRUCT(r);
182
183         /* Initialise parse structures */
184
185         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
186         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
187
188         /* Marshall data and send request */
189
190         init_samr_q_open_domain(&q, connect_pol, access_mask, domain_sid);
191
192         if (!samr_io_q_open_domain("", &q, &qbuf, 0) ||
193             !rpc_api_pipe_req(cli, SAMR_OPEN_DOMAIN, &qbuf, &rbuf))
194                 goto done;
195
196         /* Unmarshall response */
197
198         if (!samr_io_r_open_domain("", &r, &rbuf, 0))
199                 goto done;
200
201         /* Return output parameters */
202
203         if (NT_STATUS_IS_OK(result = r.status)) {
204                 *domain_pol = r.domain_pol;
205 #ifdef __INSURE__
206                 domain_pol->marker = malloc(1);
207 #endif
208         }
209
210  done:
211         prs_mem_free(&qbuf);
212         prs_mem_free(&rbuf);
213
214         return result;
215 }
216
217 /* Open handle on a user */
218
219 NTSTATUS cli_samr_open_user(struct cli_state *cli, TALLOC_CTX *mem_ctx,
220                             POLICY_HND *domain_pol, uint32 access_mask, 
221                             uint32 user_rid, POLICY_HND *user_pol)
222 {
223         prs_struct qbuf, rbuf;
224         SAMR_Q_OPEN_USER q;
225         SAMR_R_OPEN_USER r;
226         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
227
228         ZERO_STRUCT(q);
229         ZERO_STRUCT(r);
230
231         /* Initialise parse structures */
232
233         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
234         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
235
236         /* Marshall data and send request */
237
238         init_samr_q_open_user(&q, domain_pol, access_mask, user_rid);
239
240         if (!samr_io_q_open_user("", &q, &qbuf, 0) ||
241             !rpc_api_pipe_req(cli, SAMR_OPEN_USER, &qbuf, &rbuf))
242                 goto done;
243
244         /* Unmarshall response */
245
246         if (!samr_io_r_open_user("", &r, &rbuf, 0))
247                 goto done;
248
249         /* Return output parameters */
250
251         if (NT_STATUS_IS_OK(result = r.status)) {
252                 *user_pol = r.user_pol;
253 #ifdef __INSURE__
254                 user_pol->marker = malloc(1);
255 #endif
256         }
257
258  done:
259         prs_mem_free(&qbuf);
260         prs_mem_free(&rbuf);
261
262         return result;
263 }
264
265 /* Open handle on a group */
266
267 NTSTATUS cli_samr_open_group(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
268                              POLICY_HND *domain_pol, uint32 access_mask, 
269                              uint32 group_rid, POLICY_HND *group_pol)
270 {
271         prs_struct qbuf, rbuf;
272         SAMR_Q_OPEN_GROUP q;
273         SAMR_R_OPEN_GROUP r;
274         NTSTATUS result =  NT_STATUS_UNSUCCESSFUL;
275
276         ZERO_STRUCT(q);
277         ZERO_STRUCT(r);
278
279         /* Initialise parse structures */
280
281         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
282         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
283
284         /* Marshall data and send request */
285
286         init_samr_q_open_group(&q, domain_pol, access_mask, group_rid);
287
288         if (!samr_io_q_open_group("", &q, &qbuf, 0) ||
289             !rpc_api_pipe_req(cli, SAMR_OPEN_GROUP, &qbuf, &rbuf))
290                 goto done;
291
292         /* Unmarshall response */
293
294         if (!samr_io_r_open_group("", &r, &rbuf, 0))
295                 goto done;
296
297         /* Return output parameters */
298
299         if (NT_STATUS_IS_OK(result = r.status)) {
300                 *group_pol = r.pol;
301 #ifdef __INSURE__
302                 group_pol->marker = malloc(1);
303 #endif
304         }
305
306  done:
307         prs_mem_free(&qbuf);
308         prs_mem_free(&rbuf);
309
310         return result;
311 }
312
313 /* Query user info */
314
315 NTSTATUS cli_samr_query_userinfo(struct cli_state *cli, TALLOC_CTX *mem_ctx,
316                                  POLICY_HND *user_pol, uint16 switch_value, 
317                                  SAM_USERINFO_CTR **ctr)
318 {
319         prs_struct qbuf, rbuf;
320         SAMR_Q_QUERY_USERINFO q;
321         SAMR_R_QUERY_USERINFO r;
322         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
323
324         ZERO_STRUCT(q);
325         ZERO_STRUCT(r);
326
327         /* Initialise parse structures */
328
329         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
330         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
331
332         /* Marshall data and send request */
333
334         init_samr_q_query_userinfo(&q, user_pol, switch_value);
335
336         if (!samr_io_q_query_userinfo("", &q, &qbuf, 0) ||
337             !rpc_api_pipe_req(cli, SAMR_QUERY_USERINFO, &qbuf, &rbuf))
338                 goto done;
339
340         /* Unmarshall response */
341
342         if (!samr_io_r_query_userinfo("", &r, &rbuf, 0))
343                 goto done;
344
345         /* Return output parameters */
346
347         result = r.status;
348         *ctr = r.ctr;
349
350  done:
351         prs_mem_free(&qbuf);
352         prs_mem_free(&rbuf);
353
354         return result;
355 }
356
357 /* Query group info */
358
359 NTSTATUS cli_samr_query_groupinfo(struct cli_state *cli, TALLOC_CTX *mem_ctx,
360                                   POLICY_HND *group_pol, uint32 info_level, 
361                                   GROUP_INFO_CTR **ctr)
362 {
363         prs_struct qbuf, rbuf;
364         SAMR_Q_QUERY_GROUPINFO q;
365         SAMR_R_QUERY_GROUPINFO r;
366         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
367
368         ZERO_STRUCT(q);
369         ZERO_STRUCT(r);
370
371         /* Initialise parse structures */
372
373         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
374         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
375
376         /* Marshall data and send request */
377
378         init_samr_q_query_groupinfo(&q, group_pol, info_level);
379
380         if (!samr_io_q_query_groupinfo("", &q, &qbuf, 0) ||
381             !rpc_api_pipe_req(cli, SAMR_QUERY_GROUPINFO, &qbuf, &rbuf))
382                 goto done;
383
384         /* Unmarshall response */
385
386         if (!samr_io_r_query_groupinfo("", &r, &rbuf, 0))
387                 goto done;
388
389         *ctr = r.ctr;
390
391         /* Return output parameters */
392
393         result = r.status;
394
395  done:
396         prs_mem_free(&qbuf);
397         prs_mem_free(&rbuf);
398
399         return result;
400 }
401
402 /* Query user groups */
403
404 NTSTATUS cli_samr_query_usergroups(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
405                                    POLICY_HND *user_pol, uint32 *num_groups, 
406                                    DOM_GID **gid)
407 {
408         prs_struct qbuf, rbuf;
409         SAMR_Q_QUERY_USERGROUPS q;
410         SAMR_R_QUERY_USERGROUPS r;
411         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
412
413         ZERO_STRUCT(q);
414         ZERO_STRUCT(r);
415
416         /* Initialise parse structures */
417
418         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
419         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
420
421         /* Marshall data and send request */
422
423         init_samr_q_query_usergroups(&q, user_pol);
424
425         if (!samr_io_q_query_usergroups("", &q, &qbuf, 0) ||
426             !rpc_api_pipe_req(cli, SAMR_QUERY_USERGROUPS, &qbuf, &rbuf))
427                 goto done;
428
429         /* Unmarshall response */
430
431         if (!samr_io_r_query_usergroups("", &r, &rbuf, 0))
432                 goto done;
433
434         /* Return output parameters */
435
436         if (NT_STATUS_IS_OK(result = r.status)) {
437                 *num_groups = r.num_entries;
438                 *gid = r.gid;
439         }
440
441  done:
442         prs_mem_free(&qbuf);
443         prs_mem_free(&rbuf);
444
445         return result;
446 }
447
448 /* Query user aliases */
449
450 NTSTATUS cli_samr_query_useraliases(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
451                                    POLICY_HND *user_pol, uint32 num_sids, DOM_SID2 *sid,
452                                    uint32 *num_aliases, uint32 **als_rids)
453 {
454         prs_struct qbuf, rbuf;
455         SAMR_Q_QUERY_USERALIASES q;
456         SAMR_R_QUERY_USERALIASES r;
457         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
458         unsigned int ptr=1;
459         
460         ZERO_STRUCT(q);
461         ZERO_STRUCT(r);
462
463         /* Initialise parse structures */
464
465         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
466         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
467
468         /* Marshall data and send request */
469
470         init_samr_q_query_useraliases(&q, user_pol, num_sids, &ptr, sid);
471
472         if (!samr_io_q_query_useraliases("", &q, &qbuf, 0) ||
473             !rpc_api_pipe_req(cli, SAMR_QUERY_USERALIASES, &qbuf, &rbuf))
474                 goto done;
475
476         /* Unmarshall response */
477
478         if (!samr_io_r_query_useraliases("", &r, &rbuf, 0))
479                 goto done;
480
481         /* Return output parameters */
482
483         if (NT_STATUS_IS_OK(result = r.status)) {
484                 *num_aliases = r.num_entries;
485                 *als_rids = r.rid;
486         }
487
488  done:
489         prs_mem_free(&qbuf);
490         prs_mem_free(&rbuf);
491
492         return result;
493 }
494
495 /* Query user groups */
496
497 NTSTATUS cli_samr_query_groupmem(struct cli_state *cli, TALLOC_CTX *mem_ctx,
498                                  POLICY_HND *group_pol, uint32 *num_mem, 
499                                  uint32 **rid, uint32 **attr)
500 {
501         prs_struct qbuf, rbuf;
502         SAMR_Q_QUERY_GROUPMEM q;
503         SAMR_R_QUERY_GROUPMEM r;
504         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
505
506         ZERO_STRUCT(q);
507         ZERO_STRUCT(r);
508
509         /* Initialise parse structures */
510
511         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
512         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
513
514         /* Marshall data and send request */
515
516         init_samr_q_query_groupmem(&q, group_pol);
517
518         if (!samr_io_q_query_groupmem("", &q, &qbuf, 0) ||
519             !rpc_api_pipe_req(cli, SAMR_QUERY_GROUPMEM, &qbuf, &rbuf))
520                 goto done;
521
522         /* Unmarshall response */
523
524         if (!samr_io_r_query_groupmem("", &r, &rbuf, 0))
525                 goto done;
526
527         /* Return output parameters */
528
529         if (NT_STATUS_IS_OK(result = r.status)) {
530                 *num_mem = r.num_entries;
531                 *rid = r.rid;
532                 *attr = r.attr;
533         }
534
535  done:
536         prs_mem_free(&qbuf);
537         prs_mem_free(&rbuf);
538
539         return result;
540 }
541
542 /**
543  * Enumerate domain users
544  *
545  * @param cli client state structure
546  * @param mem_ctx talloc context
547  * @param pol opened domain policy handle
548  * @param start_idx starting index of enumeration, returns context for
549                     next enumeration
550  * @param acb_mask account control bit mask (to enumerate some particular
551  *                 kind of accounts)
552  * @param size max acceptable size of response
553  * @param dom_users returned array of domain user names
554  * @param rids returned array of domain user RIDs
555  * @param num_dom_users numer returned entries
556  * 
557  * @return NTSTATUS returned in rpc response
558  **/
559 NTSTATUS cli_samr_enum_dom_users(struct cli_state *cli, TALLOC_CTX *mem_ctx,
560                                  POLICY_HND *pol, uint32 *start_idx, uint16 acb_mask,
561                                  uint32 size, char ***dom_users, uint32 **rids,
562                                  uint32 *num_dom_users)
563 {
564         prs_struct qbuf;
565         prs_struct rbuf;
566         SAMR_Q_ENUM_DOM_USERS q;
567         SAMR_R_ENUM_DOM_USERS r;
568         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
569         int i;
570         
571         ZERO_STRUCT(q);
572         ZERO_STRUCT(r);
573         
574         /* always init this */
575         *num_dom_users = 0;
576         
577         /* Initialise parse structures */
578
579         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
580         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
581         
582         /* Fill query structure with parameters */
583
584         init_samr_q_enum_dom_users(&q, pol, *start_idx, acb_mask, 0, size);
585         
586         if (!samr_io_q_enum_dom_users("", &q, &qbuf, 0) ||
587             !rpc_api_pipe_req(cli, SAMR_ENUM_DOM_USERS, &qbuf, &rbuf)) {
588                 goto done;
589         }
590
591         /* unpack received stream */
592
593         if(!samr_io_r_enum_dom_users("", &r, &rbuf, 0))
594                 goto done;
595         
596         result = r.status;
597
598         if (!NT_STATUS_IS_OK(result) &&
599             NT_STATUS_V(result) != NT_STATUS_V(STATUS_MORE_ENTRIES))
600                 goto done;
601         
602         *start_idx = r.next_idx;
603         *num_dom_users = r.num_entries2;
604
605         if (r.num_entries2) {
606                 /* allocate memory needed to return received data */    
607                 *rids = (uint32*)talloc(mem_ctx, sizeof(uint32) * r.num_entries2);
608                 if (!*rids) {
609                         DEBUG(0, ("Error in cli_samr_enum_dom_users(): out of memory\n"));
610                         return NT_STATUS_NO_MEMORY;
611                 }
612                 
613                 *dom_users = (char**)talloc(mem_ctx, sizeof(char*) * r.num_entries2);
614                 if (!*dom_users) {
615                         DEBUG(0, ("Error in cli_samr_enum_dom_users(): out of memory\n"));
616                         return NT_STATUS_NO_MEMORY;
617                 }
618                 
619                 /* fill output buffers with rpc response */
620                 for (i = 0; i < r.num_entries2; i++) {
621                         fstring conv_buf;
622                         
623                         (*rids)[i] = r.sam[i].rid;
624                         unistr2_to_ascii(conv_buf, &(r.uni_acct_name[i]), sizeof(conv_buf) - 1);
625                         (*dom_users)[i] = talloc_strdup(mem_ctx, conv_buf);
626                 }
627         }
628         
629 done:
630         prs_mem_free(&qbuf);
631         prs_mem_free(&rbuf);
632         
633         return result;
634 };
635
636 /* Enumerate domain groups */
637
638 NTSTATUS cli_samr_enum_dom_groups(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
639                                   POLICY_HND *pol, uint32 *start_idx, 
640                                   uint32 size, struct acct_info **dom_groups,
641                                   uint32 *num_dom_groups)
642 {
643         prs_struct qbuf, rbuf;
644         SAMR_Q_ENUM_DOM_GROUPS q;
645         SAMR_R_ENUM_DOM_GROUPS r;
646         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
647         uint32 name_idx, i;
648
649         ZERO_STRUCT(q);
650         ZERO_STRUCT(r);
651
652         /* Initialise parse structures */
653
654         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
655         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
656
657         /* Marshall data and send request */
658
659         init_samr_q_enum_dom_groups(&q, pol, *start_idx, size);
660
661         if (!samr_io_q_enum_dom_groups("", &q, &qbuf, 0) ||
662             !rpc_api_pipe_req(cli, SAMR_ENUM_DOM_GROUPS, &qbuf, &rbuf))
663                 goto done;
664
665         /* Unmarshall response */
666
667         if (!samr_io_r_enum_dom_groups("", &r, &rbuf, 0))
668                 goto done;
669
670         /* Return output parameters */
671
672         result = r.status;
673
674         if (!NT_STATUS_IS_OK(result) &&
675             NT_STATUS_V(result) != NT_STATUS_V(STATUS_MORE_ENTRIES))
676                 goto done;
677
678         *num_dom_groups = r.num_entries2;
679
680         if (!((*dom_groups) = (struct acct_info *)
681               talloc(mem_ctx, sizeof(struct acct_info) * *num_dom_groups))) {
682                 result = NT_STATUS_UNSUCCESSFUL;
683                 goto done;
684         }
685
686         memset(*dom_groups, 0, sizeof(struct acct_info) * *num_dom_groups);
687
688         name_idx = 0;
689
690         for (i = 0; i < *num_dom_groups; i++) {
691
692                 (*dom_groups)[i].rid = r.sam[i].rid;
693
694                 if (r.sam[i].hdr_name.buffer) {
695                         unistr2_to_ascii((*dom_groups)[i].acct_name,
696                                          &r.uni_grp_name[name_idx],
697                                          sizeof(fstring) - 1);
698                         name_idx++;
699                 }
700
701                 *start_idx = r.next_idx;
702         }
703
704  done:
705         prs_mem_free(&qbuf);
706         prs_mem_free(&rbuf);
707
708         return result;
709 }
710
711 /* Enumerate domain groups */
712
713 NTSTATUS cli_samr_enum_als_groups(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
714                                   POLICY_HND *pol, uint32 *start_idx, 
715                                   uint32 size, struct acct_info **dom_groups,
716                                   uint32 *num_dom_groups)
717 {
718         prs_struct qbuf, rbuf;
719         SAMR_Q_ENUM_DOM_ALIASES q;
720         SAMR_R_ENUM_DOM_ALIASES r;
721         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
722         uint32 name_idx, i;
723
724         ZERO_STRUCT(q);
725         ZERO_STRUCT(r);
726
727         /* Initialise parse structures */
728
729         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
730         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
731
732         /* Marshall data and send request */
733
734         init_samr_q_enum_dom_aliases(&q, pol, *start_idx, size);
735
736         if (!samr_io_q_enum_dom_aliases("", &q, &qbuf, 0) ||
737             !rpc_api_pipe_req(cli, SAMR_ENUM_DOM_ALIASES, &qbuf, &rbuf)) {
738                 goto done;
739         }
740
741         /* Unmarshall response */
742
743         if (!samr_io_r_enum_dom_aliases("", &r, &rbuf, 0)) {
744                 goto done;
745         }
746
747         /* Return output parameters */
748
749         result = r.status;
750
751         if (!NT_STATUS_IS_OK(result) &&
752             NT_STATUS_V(result) != NT_STATUS_V(STATUS_MORE_ENTRIES)) {
753                 goto done;
754         }
755
756         *num_dom_groups = r.num_entries2;
757
758         if (!((*dom_groups) = (struct acct_info *)
759               talloc(mem_ctx, sizeof(struct acct_info) * *num_dom_groups))) {
760                 result = NT_STATUS_UNSUCCESSFUL;
761                 goto done;
762         }
763
764         memset(*dom_groups, 0, sizeof(struct acct_info) * *num_dom_groups);
765
766         name_idx = 0;
767
768         for (i = 0; i < *num_dom_groups; i++) {
769
770                 (*dom_groups)[i].rid = r.sam[i].rid;
771
772                 if (r.sam[i].hdr_name.buffer) {
773                         unistr2_to_ascii((*dom_groups)[i].acct_name,
774                                          &r.uni_grp_name[name_idx],
775                                          sizeof(fstring) - 1);
776                         name_idx++;
777                 }
778
779                 *start_idx = r.next_idx;
780         }
781
782  done:
783         prs_mem_free(&qbuf);
784         prs_mem_free(&rbuf);
785
786         return result;
787 }
788
789 /* Query alias members */
790
791 NTSTATUS cli_samr_query_aliasmem(struct cli_state *cli, TALLOC_CTX *mem_ctx,
792                                  POLICY_HND *alias_pol, uint32 *num_mem, 
793                                  DOM_SID **sids)
794 {
795         prs_struct qbuf, rbuf;
796         SAMR_Q_QUERY_ALIASMEM q;
797         SAMR_R_QUERY_ALIASMEM r;
798         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
799         uint32 i;
800
801         ZERO_STRUCT(q);
802         ZERO_STRUCT(r);
803
804         /* Initialise parse structures */
805
806         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
807         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
808
809         /* Marshall data and send request */
810
811         init_samr_q_query_aliasmem(&q, alias_pol);
812
813         if (!samr_io_q_query_aliasmem("", &q, &qbuf, 0) ||
814             !rpc_api_pipe_req(cli, SAMR_QUERY_ALIASMEM, &qbuf, &rbuf)) {
815                 goto done;
816         }
817
818         /* Unmarshall response */
819
820         if (!samr_io_r_query_aliasmem("", &r, &rbuf, 0)) {
821                 goto done;
822         }
823
824         /* Return output parameters */
825
826         if (!NT_STATUS_IS_OK(result = r.status)) {
827                 goto done;
828         }
829
830         *num_mem = r.num_sids;
831
832         if (!(*sids = talloc(mem_ctx, sizeof(DOM_SID) * *num_mem))) {
833                 result = NT_STATUS_UNSUCCESSFUL;
834                 goto done;
835         }
836
837         for (i = 0; i < *num_mem; i++) {
838                 (*sids)[i] = r.sid[i].sid;
839         }
840
841  done:
842         prs_mem_free(&qbuf);
843         prs_mem_free(&rbuf);
844
845         return result;
846 }
847
848 /* Open handle on an alias */
849
850 NTSTATUS cli_samr_open_alias(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
851                              POLICY_HND *domain_pol, uint32 access_mask, 
852                              uint32 alias_rid, POLICY_HND *alias_pol)
853 {
854         prs_struct qbuf, rbuf;
855         SAMR_Q_OPEN_ALIAS q;
856         SAMR_R_OPEN_ALIAS r;
857         NTSTATUS result;
858
859         ZERO_STRUCT(q);
860         ZERO_STRUCT(r);
861
862         /* Initialise parse structures */
863
864         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
865         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
866
867         /* Marshall data and send request */
868
869         init_samr_q_open_alias(&q, domain_pol, access_mask, alias_rid);
870
871         if (!samr_io_q_open_alias("", &q, &qbuf, 0) ||
872             !rpc_api_pipe_req(cli, SAMR_OPEN_ALIAS, &qbuf, &rbuf)) {
873                 result = NT_STATUS_UNSUCCESSFUL;
874                 goto done;
875         }
876
877         /* Unmarshall response */
878
879         if (!samr_io_r_open_alias("", &r, &rbuf, 0)) {
880                 result = NT_STATUS_UNSUCCESSFUL;
881                 goto done;
882         }
883
884         /* Return output parameters */
885
886         if (NT_STATUS_IS_OK(result = r.status)) {
887                 *alias_pol = r.pol;
888 #ifdef __INSURE__
889                 alias_pol->marker = malloc(1);
890 #endif
891         }
892
893  done:
894         prs_mem_free(&qbuf);
895         prs_mem_free(&rbuf);
896
897         return result;
898 }
899
900 /* Query domain info */
901
902 NTSTATUS cli_samr_query_dom_info(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
903                                  POLICY_HND *domain_pol, uint16 switch_value,
904                                  SAM_UNK_CTR *ctr)
905 {
906         prs_struct qbuf, rbuf;
907         SAMR_Q_QUERY_DOMAIN_INFO q;
908         SAMR_R_QUERY_DOMAIN_INFO r;
909         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
910
911         ZERO_STRUCT(q);
912         ZERO_STRUCT(r);
913
914         /* Initialise parse structures */
915
916         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
917         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
918
919         /* Marshall data and send request */
920
921         init_samr_q_query_dom_info(&q, domain_pol, switch_value);
922
923         if (!samr_io_q_query_dom_info("", &q, &qbuf, 0) ||
924             !rpc_api_pipe_req(cli, SAMR_QUERY_DOMAIN_INFO, &qbuf, &rbuf)) {
925                 goto done;
926         }
927
928         /* Unmarshall response */
929
930         r.ctr = ctr;
931
932         if (!samr_io_r_query_dom_info("", &r, &rbuf, 0)) {
933                 goto done;
934         }
935
936         /* Return output parameters */
937
938         if (!NT_STATUS_IS_OK(result = r.status)) {
939                 goto done;
940         }
941
942  done:
943         prs_mem_free(&qbuf);
944         prs_mem_free(&rbuf);
945
946         return result;
947 }
948
949 /* This function returns the bizzare set of (max_entries, max_size) required
950    for the QueryDisplayInfo RPC to actually work against a domain controller
951    with large (10k and higher) numbers of users.  These values were 
952    obtained by inspection using ethereal and NT4 running User Manager. */
953
954 void get_query_dispinfo_params(int loop_count, uint32 *max_entries,
955                                uint32 *max_size)
956 {
957         switch(loop_count) {
958         case 0:
959                 *max_entries = 512;
960                 *max_size = 16383;
961                 break;
962         case 1:
963                 *max_entries = 1024;
964                 *max_size = 32766;
965                 break;
966         case 2:
967                 *max_entries = 2048;
968                 *max_size = 65532;
969                 break;
970         case 3:
971                 *max_entries = 4096;
972                 *max_size = 131064;
973                 break;
974         default:              /* loop_count >= 4 */
975                 *max_entries = 4096;
976                 *max_size = 131071;
977                 break;
978         }
979 }                    
980
981 /* Query display info */
982
983 NTSTATUS cli_samr_query_dispinfo(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
984                                  POLICY_HND *domain_pol, uint32 *start_idx,
985                                  uint16 switch_value, uint32 *num_entries,
986                                  uint32 max_entries, uint32 max_size,
987                                  SAM_DISPINFO_CTR *ctr)
988 {
989         prs_struct qbuf, rbuf;
990         SAMR_Q_QUERY_DISPINFO q;
991         SAMR_R_QUERY_DISPINFO r;
992         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
993
994         ZERO_STRUCT(q);
995         ZERO_STRUCT(r);
996
997         /* Initialise parse structures */
998
999         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1000         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1001
1002         /* Marshall data and send request */
1003
1004         init_samr_q_query_dispinfo(&q, domain_pol, switch_value,
1005                                    *start_idx, max_entries, max_size);
1006
1007         if (!samr_io_q_query_dispinfo("", &q, &qbuf, 0) ||
1008             !rpc_api_pipe_req(cli, SAMR_QUERY_DISPINFO, &qbuf, &rbuf)) {
1009                 goto done;
1010         }
1011
1012         /* Unmarshall response */
1013
1014         r.ctr = ctr;
1015
1016         if (!samr_io_r_query_dispinfo("", &r, &rbuf, 0)) {
1017                 goto done;
1018         }
1019
1020         /* Return output parameters */
1021
1022         result = r.status;
1023
1024         if (!NT_STATUS_IS_OK(result) &&
1025             NT_STATUS_V(result) != NT_STATUS_V(STATUS_MORE_ENTRIES)) {
1026                 goto done;
1027         }
1028
1029         *num_entries = r.num_entries;
1030         *start_idx += r.num_entries;  /* No next_idx in this structure! */
1031
1032  done:
1033         prs_mem_free(&qbuf);
1034         prs_mem_free(&rbuf);
1035
1036         return result;
1037 }
1038
1039 /* Lookup rids.  Note that NT4 seems to crash if more than ~1000 rids are
1040    looked up in one packet. */
1041
1042 NTSTATUS cli_samr_lookup_rids(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1043                               POLICY_HND *domain_pol, uint32 flags,
1044                               uint32 num_rids, uint32 *rids, 
1045                               uint32 *num_names, char ***names,
1046                               uint32 **name_types)
1047 {
1048         prs_struct qbuf, rbuf;
1049         SAMR_Q_LOOKUP_RIDS q;
1050         SAMR_R_LOOKUP_RIDS r;
1051         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1052         uint32 i;
1053
1054         if (num_rids > 1000) {
1055                 DEBUG(2, ("cli_samr_lookup_rids: warning: NT4 can crash if "
1056                           "more than ~1000 rids are looked up at once.\n"));
1057         }
1058
1059         ZERO_STRUCT(q);
1060         ZERO_STRUCT(r);
1061
1062         /* Initialise parse structures */
1063
1064         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1065         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1066
1067         /* Marshall data and send request */
1068
1069         init_samr_q_lookup_rids(mem_ctx, &q, domain_pol, flags,
1070                                 num_rids, rids);
1071
1072         if (!samr_io_q_lookup_rids("", &q, &qbuf, 0) ||
1073             !rpc_api_pipe_req(cli, SAMR_LOOKUP_RIDS, &qbuf, &rbuf)) {
1074                 goto done;
1075         }
1076
1077         /* Unmarshall response */
1078
1079         if (!samr_io_r_lookup_rids("", &r, &rbuf, 0)) {
1080                 goto done;
1081         }
1082
1083         /* Return output parameters */
1084
1085         if (!NT_STATUS_IS_OK(result = r.status)) {
1086                 goto done;
1087         }
1088
1089         if (r.num_names1 == 0) {
1090                 *num_names = 0;
1091                 *names = NULL;
1092                 goto done;
1093         }
1094
1095         *num_names = r.num_names1;
1096         *names = talloc(mem_ctx, sizeof(char *) * r.num_names1);
1097         *name_types = talloc(mem_ctx, sizeof(uint32) * r.num_names1);
1098
1099         for (i = 0; i < r.num_names1; i++) {
1100                 fstring tmp;
1101
1102                 unistr2_to_ascii(tmp, &r.uni_name[i], sizeof(tmp) - 1);
1103                 (*names)[i] = talloc_strdup(mem_ctx, tmp);
1104                 (*name_types)[i] = r.type[i];
1105         }
1106
1107  done:
1108         prs_mem_free(&qbuf);
1109         prs_mem_free(&rbuf);
1110
1111         return result;
1112 }
1113
1114 /* Lookup names */
1115
1116 NTSTATUS cli_samr_lookup_names(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1117                                POLICY_HND *domain_pol, uint32 flags,
1118                                uint32 num_names, const char **names,
1119                                uint32 *num_rids, uint32 **rids,
1120                                uint32 **rid_types)
1121 {
1122         prs_struct qbuf, rbuf;
1123         SAMR_Q_LOOKUP_NAMES q;
1124         SAMR_R_LOOKUP_NAMES r;
1125         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1126         uint32 i;
1127
1128         ZERO_STRUCT(q);
1129         ZERO_STRUCT(r);
1130
1131         /* Initialise parse structures */
1132
1133         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1134         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1135
1136         /* Marshall data and send request */
1137
1138         init_samr_q_lookup_names(mem_ctx, &q, domain_pol, flags,
1139                                  num_names, names);
1140
1141         if (!samr_io_q_lookup_names("", &q, &qbuf, 0) ||
1142             !rpc_api_pipe_req(cli, SAMR_LOOKUP_NAMES, &qbuf, &rbuf)) {
1143                 goto done;
1144         }
1145
1146         /* Unmarshall response */
1147
1148         if (!samr_io_r_lookup_names("", &r, &rbuf, 0)) {
1149                 goto done;
1150         }
1151
1152         /* Return output parameters */
1153
1154         if (!NT_STATUS_IS_OK(result = r.status)) {
1155                 goto done;
1156         }
1157
1158         if (r.num_rids1 == 0) {
1159                 *num_rids = 0;
1160                 goto done;
1161         }
1162
1163         *num_rids = r.num_rids1;
1164         *rids = talloc(mem_ctx, sizeof(uint32) * r.num_rids1);
1165         *rid_types = talloc(mem_ctx, sizeof(uint32) * r.num_rids1);
1166
1167         for (i = 0; i < r.num_rids1; i++) {
1168                 (*rids)[i] = r.rids[i];
1169                 (*rid_types)[i] = r.types[i];
1170         }
1171
1172  done:
1173         prs_mem_free(&qbuf);
1174         prs_mem_free(&rbuf);
1175
1176         return result;
1177 }
1178
1179 /* Create a domain user */
1180
1181 NTSTATUS cli_samr_create_dom_user(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1182                                   POLICY_HND *domain_pol, const char *acct_name,
1183                                   uint32 acb_info, uint32 unknown, 
1184                                   POLICY_HND *user_pol, uint32 *rid)
1185 {
1186         prs_struct qbuf, rbuf;
1187         SAMR_Q_CREATE_USER q;
1188         SAMR_R_CREATE_USER r;
1189         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1190
1191         ZERO_STRUCT(q);
1192         ZERO_STRUCT(r);
1193
1194         /* Initialise parse structures */
1195
1196         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1197         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1198
1199         /* Marshall data and send request */
1200
1201         init_samr_q_create_user(&q, domain_pol, acct_name, acb_info, unknown);
1202
1203         if (!samr_io_q_create_user("", &q, &qbuf, 0) ||
1204             !rpc_api_pipe_req(cli, SAMR_CREATE_USER, &qbuf, &rbuf)) {
1205                 goto done;
1206         }
1207
1208         /* Unmarshall response */
1209
1210         if (!samr_io_r_create_user("", &r, &rbuf, 0)) {
1211                 goto done;
1212         }
1213
1214         /* Return output parameters */
1215
1216         if (!NT_STATUS_IS_OK(result = r.status)) {
1217                 goto done;
1218         }
1219
1220         if (user_pol)
1221                 *user_pol = r.user_pol;
1222
1223         if (rid)
1224                 *rid = r.user_rid;
1225
1226  done:
1227         prs_mem_free(&qbuf);
1228         prs_mem_free(&rbuf);
1229
1230         return result;
1231 }
1232
1233 /* Set userinfo */
1234
1235 NTSTATUS cli_samr_set_userinfo(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1236                                POLICY_HND *user_pol, uint16 switch_value,
1237                                uchar sess_key[16], SAM_USERINFO_CTR *ctr)
1238 {
1239         prs_struct qbuf, rbuf;
1240         SAMR_Q_SET_USERINFO q;
1241         SAMR_R_SET_USERINFO r;
1242         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1243
1244         ZERO_STRUCT(q);
1245         ZERO_STRUCT(r);
1246
1247         /* Initialise parse structures */
1248
1249         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1250         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1251
1252         /* Marshall data and send request */
1253
1254         q.ctr = ctr;
1255
1256         init_samr_q_set_userinfo(&q, user_pol, sess_key, switch_value, 
1257                                  ctr->info.id);
1258
1259         if (!samr_io_q_set_userinfo("", &q, &qbuf, 0) ||
1260             !rpc_api_pipe_req(cli, SAMR_SET_USERINFO, &qbuf, &rbuf)) {
1261                 goto done;
1262         }
1263
1264         /* Unmarshall response */
1265
1266         if (!samr_io_r_set_userinfo("", &r, &rbuf, 0)) {
1267                 goto done;
1268         }
1269
1270         /* Return output parameters */
1271
1272         if (!NT_STATUS_IS_OK(result = r.status)) {
1273                 goto done;
1274         }
1275
1276  done:
1277         prs_mem_free(&qbuf);
1278         prs_mem_free(&rbuf);
1279
1280         return result;
1281 }
1282
1283 /* Set userinfo2 */
1284
1285 NTSTATUS cli_samr_set_userinfo2(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1286                                 POLICY_HND *user_pol, uint16 switch_value,
1287                                 uchar sess_key[16], SAM_USERINFO_CTR *ctr)
1288 {
1289         prs_struct qbuf, rbuf;
1290         SAMR_Q_SET_USERINFO2 q;
1291         SAMR_R_SET_USERINFO2 r;
1292         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1293
1294         ZERO_STRUCT(q);
1295         ZERO_STRUCT(r);
1296
1297         /* Initialise parse structures */
1298
1299         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1300         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1301
1302         /* Marshall data and send request */
1303
1304         init_samr_q_set_userinfo2(&q, user_pol, sess_key, switch_value, ctr);
1305
1306         if (!samr_io_q_set_userinfo2("", &q, &qbuf, 0) ||
1307             !rpc_api_pipe_req(cli, SAMR_SET_USERINFO2, &qbuf, &rbuf)) {
1308                 goto done;
1309         }
1310
1311         /* Unmarshall response */
1312
1313         if (!samr_io_r_set_userinfo2("", &r, &rbuf, 0)) {
1314                 goto done;
1315         }
1316
1317         /* Return output parameters */
1318
1319         if (!NT_STATUS_IS_OK(result = r.status)) {
1320                 goto done;
1321         }
1322
1323  done:
1324         prs_mem_free(&qbuf);
1325         prs_mem_free(&rbuf);
1326
1327         return result;
1328 }
1329
1330 /* Delete domain user */
1331
1332 NTSTATUS cli_samr_delete_dom_user(struct cli_state *cli, TALLOC_CTX *mem_ctx, 
1333                                   POLICY_HND *user_pol)
1334 {
1335         prs_struct qbuf, rbuf;
1336         SAMR_Q_DELETE_DOM_USER q;
1337         SAMR_R_DELETE_DOM_USER r;
1338         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1339
1340         ZERO_STRUCT(q);
1341         ZERO_STRUCT(r);
1342
1343         /* Initialise parse structures */
1344
1345         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1346         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1347
1348         /* Marshall data and send request */
1349
1350         init_samr_q_delete_dom_user(&q, user_pol);
1351
1352         if (!samr_io_q_delete_dom_user("", &q, &qbuf, 0) ||
1353             !rpc_api_pipe_req(cli, SAMR_DELETE_DOM_USER, &qbuf, &rbuf)) {
1354                 goto done;
1355         }
1356
1357         /* Unmarshall response */
1358
1359         if (!samr_io_r_delete_dom_user("", &r, &rbuf, 0)) {
1360                 goto done;
1361         }
1362
1363         /* Return output parameters */
1364
1365         result = r.status;
1366
1367  done:
1368         prs_mem_free(&qbuf);
1369         prs_mem_free(&rbuf);
1370
1371         return result;
1372 }
1373
1374 /* Query user security object */
1375
1376 NTSTATUS cli_samr_query_sec_obj(struct cli_state *cli, TALLOC_CTX *mem_ctx,
1377                                  POLICY_HND *user_pol, uint16 switch_value, 
1378                                  TALLOC_CTX *ctx, SEC_DESC_BUF **sec_desc_buf)
1379 {
1380         prs_struct qbuf, rbuf;
1381         SAMR_Q_QUERY_SEC_OBJ q;
1382         SAMR_R_QUERY_SEC_OBJ r;
1383         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1384
1385         ZERO_STRUCT(q);
1386         ZERO_STRUCT(r);
1387
1388         /* Initialise parse structures */
1389
1390         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1391         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1392
1393         /* Marshall data and send request */
1394
1395         init_samr_q_query_sec_obj(&q, user_pol, switch_value);
1396
1397         if (!samr_io_q_query_sec_obj("", &q, &qbuf, 0) ||
1398             !rpc_api_pipe_req(cli, SAMR_QUERY_SEC_OBJECT, &qbuf, &rbuf)) {
1399                 goto done;
1400         }
1401
1402         /* Unmarshall response */
1403
1404         if (!samr_io_r_query_sec_obj("", &r, &rbuf, 0)) {
1405                 goto done;
1406         }
1407
1408         /* Return output parameters */
1409
1410         result = r.status;
1411         *sec_desc_buf=dup_sec_desc_buf(ctx, r.buf);
1412
1413  done:
1414         prs_mem_free(&qbuf);
1415         prs_mem_free(&rbuf);
1416
1417         return result;
1418 }
1419
1420 /* Get domain password info */
1421
1422 NTSTATUS cli_samr_get_dom_pwinfo(struct cli_state *cli, TALLOC_CTX *mem_ctx,
1423                                  uint16 *unk_0, uint16 *unk_1, uint16 *unk_2)
1424 {
1425         prs_struct qbuf, rbuf;
1426         SAMR_Q_GET_DOM_PWINFO q;
1427         SAMR_R_GET_DOM_PWINFO r;
1428         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1429
1430         ZERO_STRUCT(q);
1431         ZERO_STRUCT(r);
1432
1433         /* Initialise parse structures */
1434
1435         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1436         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1437
1438         /* Marshall data and send request */
1439
1440         init_samr_q_get_dom_pwinfo(&q, cli->desthost);
1441
1442         if (!samr_io_q_get_dom_pwinfo("", &q, &qbuf, 0) ||
1443             !rpc_api_pipe_req(cli, SAMR_GET_DOM_PWINFO, &qbuf, &rbuf))
1444                 goto done;
1445
1446         /* Unmarshall response */
1447
1448         if (!samr_io_r_get_dom_pwinfo("", &r, &rbuf, 0))
1449                 goto done;
1450
1451         /* Return output parameters */
1452
1453         result = r.status;
1454
1455         if (NT_STATUS_IS_OK(result)) {
1456                 if (unk_0)
1457                         *unk_0 = r.unk_0;
1458                 if (unk_1)
1459                         *unk_1 = r.unk_1;
1460                 if (unk_2)
1461                         *unk_2 = r.unk_2;
1462         }
1463
1464  done:
1465         prs_mem_free(&qbuf);
1466         prs_mem_free(&rbuf);
1467
1468         return result;
1469 }
1470
1471 /* Lookup Domain Name */
1472
1473 NTSTATUS cli_samr_lookup_domain(struct cli_state *cli, TALLOC_CTX *mem_ctx,
1474                                 POLICY_HND *user_pol, char *domain_name, 
1475                                 DOM_SID *sid)
1476 {
1477         prs_struct qbuf, rbuf;
1478         SAMR_Q_LOOKUP_DOMAIN q;
1479         SAMR_R_LOOKUP_DOMAIN r;
1480         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1481
1482         ZERO_STRUCT(q);
1483         ZERO_STRUCT(r);
1484
1485         /* Initialise parse structures */
1486
1487         prs_init(&qbuf, MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL);
1488         prs_init(&rbuf, 0, mem_ctx, UNMARSHALL);
1489
1490         /* Marshall data and send request */
1491
1492         init_samr_q_lookup_domain(&q, user_pol, domain_name);
1493
1494         if (!samr_io_q_lookup_domain("", &q, &qbuf, 0) ||
1495             !rpc_api_pipe_req(cli, SAMR_LOOKUP_DOMAIN, &qbuf, &rbuf))
1496                 goto done;
1497
1498         /* Unmarshall response */
1499
1500         if (!samr_io_r_lookup_domain("", &r, &rbuf, 0))
1501                 goto done;
1502
1503         /* Return output parameters */
1504
1505         result = r.status;
1506
1507         if (NT_STATUS_IS_OK(result))
1508                 sid_copy(sid, &r.dom_sid.sid);
1509
1510  done:
1511         prs_mem_free(&qbuf);
1512         prs_mem_free(&rbuf);
1513
1514         return result;
1515 }