d547c3e7be1b0ad468068e4de14052dee4e8748b
[samba.git] / source3 / rpc_client / cli_lsarpc.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) Rafal Szczesniak                       2002
7    Copyright (C) Jeremy Allison                         2005.
8    Copyright (C) Michael Adam                           2007.
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25
26 /** @defgroup lsa LSA - Local Security Architecture
27  *  @ingroup rpc_client
28  *
29  * @{
30  **/
31
32 /**
33  * @file cli_lsarpc.c
34  *
35  * RPC client routines for the LSA RPC pipe.  LSA means "local
36  * security authority", which is half of a password database.
37  **/
38
39 /** Open a LSA policy handle
40  *
41  * @param cli Handle on an initialised SMB connection */
42
43 NTSTATUS rpccli_lsa_open_policy(struct rpc_pipe_client *cli,
44                                 TALLOC_CTX *mem_ctx,
45                                 bool sec_qos, uint32 des_access,
46                                 POLICY_HND *pol)
47 {
48         struct lsa_ObjectAttribute attr;
49         struct lsa_QosInfo qos;
50         uint16_t system_name = '\\';
51
52         if (sec_qos) {
53                 init_lsa_sec_qos(&qos, 0xc, 2, 1, 0);
54                 init_lsa_obj_attr(&attr,
55                                   0x18,
56                                   NULL,
57                                   NULL,
58                                   0,
59                                   NULL,
60                                   &qos);
61         } else {
62                 init_lsa_obj_attr(&attr,
63                                   0x18,
64                                   NULL,
65                                   NULL,
66                                   0,
67                                   NULL,
68                                   NULL);
69         }
70
71         return rpccli_lsa_OpenPolicy(cli, mem_ctx,
72                                      &system_name,
73                                      &attr,
74                                      des_access,
75                                      pol);
76 }
77
78 /** Open a LSA policy handle
79   *
80   * @param cli Handle on an initialised SMB connection
81   */
82
83 NTSTATUS rpccli_lsa_open_policy2(struct rpc_pipe_client *cli,
84                                  TALLOC_CTX *mem_ctx, bool sec_qos,
85                                  uint32 des_access, POLICY_HND *pol)
86 {
87         struct lsa_ObjectAttribute attr;
88         struct lsa_QosInfo qos;
89         char *srv_name_slash = talloc_asprintf(mem_ctx, "\\\\%s", cli->cli->desthost);
90
91         if (sec_qos) {
92                 init_lsa_sec_qos(&qos, 0xc, 2, 1, 0);
93                 init_lsa_obj_attr(&attr,
94                                   0x18,
95                                   NULL,
96                                   NULL,
97                                   0,
98                                   NULL,
99                                   &qos);
100         } else {
101                 init_lsa_obj_attr(&attr,
102                                   0x18,
103                                   NULL,
104                                   NULL,
105                                   0,
106                                   NULL,
107                                   NULL);
108         }
109
110         return rpccli_lsa_OpenPolicy2(cli, mem_ctx,
111                                       srv_name_slash,
112                                       &attr,
113                                       des_access,
114                                       pol);
115 }
116
117 /* Lookup a list of sids
118  *
119  * internal version withOUT memory allocation of the target arrays.
120  * this assumes suffciently sized arrays to store domains, names and types. */
121
122 static NTSTATUS rpccli_lsa_lookup_sids_noalloc(struct rpc_pipe_client *cli,
123                                                TALLOC_CTX *mem_ctx,
124                                                POLICY_HND *pol,
125                                                int num_sids,
126                                                const DOM_SID *sids,
127                                                char **domains,
128                                                char **names,
129                                                enum lsa_SidType *types)
130 {
131         prs_struct qbuf, rbuf;
132         LSA_Q_LOOKUP_SIDS q;
133         LSA_R_LOOKUP_SIDS r;
134         DOM_R_REF ref;
135         NTSTATUS result = NT_STATUS_OK;
136         TALLOC_CTX *tmp_ctx = NULL;
137         int i;
138
139         tmp_ctx = talloc_new(mem_ctx);
140         if (!tmp_ctx) {
141                 DEBUG(0, ("rpccli_lsa_lookup_sids_noalloc: out of memory!\n"));
142                 result = NT_STATUS_UNSUCCESSFUL;
143                 goto done;
144         }
145
146         ZERO_STRUCT(q);
147         ZERO_STRUCT(r);
148
149         init_q_lookup_sids(tmp_ctx, &q, pol, num_sids, sids, 1);
150
151         ZERO_STRUCT(ref);
152
153         r.dom_ref = &ref;
154
155         CLI_DO_RPC( cli, tmp_ctx, PI_LSARPC, LSA_LOOKUPSIDS,
156                         q, r,
157                         qbuf, rbuf,
158                         lsa_io_q_lookup_sids,
159                         lsa_io_r_lookup_sids,
160                         NT_STATUS_UNSUCCESSFUL );
161
162         DEBUG(10, ("LSA_LOOKUPSIDS returned '%s', mapped count = %d'\n",
163                    nt_errstr(r.status), r.mapped_count));
164
165         if (!NT_STATUS_IS_OK(r.status) &&
166             !NT_STATUS_EQUAL(r.status, NT_STATUS_NONE_MAPPED) &&
167             !NT_STATUS_EQUAL(r.status, STATUS_SOME_UNMAPPED))
168         {
169                 /* An actual error occured */
170                 result = r.status;
171                 goto done;
172         }
173
174         /* Return output parameters */
175
176         if (NT_STATUS_EQUAL(r.status, NT_STATUS_NONE_MAPPED) ||
177             (r.mapped_count == 0))
178         {
179                 for (i = 0; i < num_sids; i++) {
180                         (names)[i] = NULL;
181                         (domains)[i] = NULL;
182                         (types)[i] = SID_NAME_UNKNOWN;
183                 }
184                 result = NT_STATUS_NONE_MAPPED;
185                 goto done;
186         }
187
188         for (i = 0; i < num_sids; i++) {
189                 fstring name, dom_name;
190                 uint32 dom_idx = r.names.name[i].domain_idx;
191
192                 /* Translate optimised name through domain index array */
193
194                 if (dom_idx != 0xffffffff) {
195
196                         rpcstr_pull_unistr2_fstring(
197                                 dom_name, &ref.ref_dom[dom_idx].uni_dom_name);
198                         rpcstr_pull_unistr2_fstring(
199                                 name, &r.names.uni_name[i]);
200
201                         (names)[i] = talloc_strdup(mem_ctx, name);
202                         (domains)[i] = talloc_strdup(mem_ctx, dom_name);
203                         (types)[i] = r.names.name[i].sid_name_use;
204
205                         if (((names)[i] == NULL) || ((domains)[i] == NULL)) {
206                                 DEBUG(0, ("cli_lsa_lookup_sids_noalloc(): out of memory\n"));
207                                 result = NT_STATUS_UNSUCCESSFUL;
208                                 goto done;
209                         }
210
211                 } else {
212                         (names)[i] = NULL;
213                         (domains)[i] = NULL;
214                         (types)[i] = SID_NAME_UNKNOWN;
215                 }
216         }
217
218 done:
219         TALLOC_FREE(tmp_ctx);
220         return result;
221 }
222
223 /* Lookup a list of sids
224  *
225  * do it the right way: there is a limit (of 20480 for w2k3) entries
226  * returned by this call. when the sids list contains more entries,
227  * empty lists are returned. This version of lsa_lookup_sids passes
228  * the list of sids in hunks of LOOKUP_SIDS_HUNK_SIZE to the lsa call. */
229
230 /* This constant defines the limit of how many sids to look up
231  * in one call (maximum). the limit from the server side is
232  * at 20480 for win2k3, but we keep it at a save 1000 for now. */
233 #define LOOKUP_SIDS_HUNK_SIZE 1000
234
235 NTSTATUS rpccli_lsa_lookup_sids(struct rpc_pipe_client *cli,
236                                 TALLOC_CTX *mem_ctx,
237                                 POLICY_HND *pol,
238                                 int num_sids,
239                                 const DOM_SID *sids,
240                                 char ***domains,
241                                 char ***names,
242                                 enum lsa_SidType **types)
243 {
244         NTSTATUS result = NT_STATUS_OK;
245         int sids_left = 0;
246         int sids_processed = 0;
247         const DOM_SID *hunk_sids = sids;
248         char **hunk_domains = NULL;
249         char **hunk_names = NULL;
250         enum lsa_SidType *hunk_types = NULL;
251
252         if (num_sids) {
253                 if (!((*domains) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
254                         DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
255                         result = NT_STATUS_NO_MEMORY;
256                         goto fail;
257                 }
258
259                 if (!((*names) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
260                         DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
261                         result = NT_STATUS_NO_MEMORY;
262                         goto fail;
263                 }
264
265                 if (!((*types) = TALLOC_ARRAY(mem_ctx, enum lsa_SidType, num_sids))) {
266                         DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
267                         result = NT_STATUS_NO_MEMORY;
268                         goto fail;
269                 }
270         } else {
271                 (*domains) = NULL;
272                 (*names) = NULL;
273                 (*types) = NULL;
274         }
275
276         sids_left = num_sids;
277         hunk_domains = *domains;
278         hunk_names = *names;
279         hunk_types = *types;
280
281         while (sids_left > 0) {
282                 int hunk_num_sids;
283                 NTSTATUS hunk_result = NT_STATUS_OK;
284
285                 hunk_num_sids = ((sids_left > LOOKUP_SIDS_HUNK_SIZE)
286                                 ? LOOKUP_SIDS_HUNK_SIZE
287                                 : sids_left);
288
289                 DEBUG(10, ("rpccli_lsa_lookup_sids: processing items "
290                            "%d -- %d of %d.\n",
291                            sids_processed,
292                            sids_processed + hunk_num_sids - 1,
293                            num_sids));
294
295                 hunk_result = rpccli_lsa_lookup_sids_noalloc(cli,
296                                                              mem_ctx,
297                                                              pol,
298                                                              hunk_num_sids,
299                                                              hunk_sids,
300                                                              hunk_domains,
301                                                              hunk_names,
302                                                              hunk_types);
303
304                 if (!NT_STATUS_IS_OK(hunk_result) &&
305                     !NT_STATUS_EQUAL(hunk_result, STATUS_SOME_UNMAPPED) &&
306                     !NT_STATUS_EQUAL(hunk_result, NT_STATUS_NONE_MAPPED))
307                 {
308                         /* An actual error occured */
309                         result = hunk_result;
310                         goto fail;
311                 }
312
313                 /* adapt overall result */
314                 if (( NT_STATUS_IS_OK(result) &&
315                      !NT_STATUS_IS_OK(hunk_result))
316                     ||
317                     ( NT_STATUS_EQUAL(result, NT_STATUS_NONE_MAPPED) &&
318                      !NT_STATUS_EQUAL(hunk_result, NT_STATUS_NONE_MAPPED)))
319                 {
320                         result = STATUS_SOME_UNMAPPED;
321                 }
322
323                 sids_left -= hunk_num_sids;
324                 sids_processed += hunk_num_sids; /* only used in DEBUG */
325                 hunk_sids += hunk_num_sids;
326                 hunk_domains += hunk_num_sids;
327                 hunk_names += hunk_num_sids;
328                 hunk_types += hunk_num_sids;
329         }
330
331         return result;
332
333 fail:
334         TALLOC_FREE(*domains);
335         TALLOC_FREE(*names);
336         TALLOC_FREE(*types);
337         return result;
338 }
339
340 /** Lookup a list of names */
341
342 NTSTATUS rpccli_lsa_lookup_names(struct rpc_pipe_client *cli,
343                                  TALLOC_CTX *mem_ctx,
344                                  POLICY_HND *pol, int num_names,
345                                  const char **names,
346                                  const char ***dom_names,
347                                  int level,
348                                  DOM_SID **sids,
349                                  enum lsa_SidType **types)
350 {
351         prs_struct qbuf, rbuf;
352         LSA_Q_LOOKUP_NAMES q;
353         LSA_R_LOOKUP_NAMES r;
354         DOM_R_REF ref;
355         NTSTATUS result;
356         int i;
357
358         ZERO_STRUCT(q);
359         ZERO_STRUCT(r);
360
361         ZERO_STRUCT(ref);
362         r.dom_ref = &ref;
363
364         init_q_lookup_names(mem_ctx, &q, pol, num_names, names, level);
365
366         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_LOOKUPNAMES,
367                         q, r,
368                         qbuf, rbuf,
369                         lsa_io_q_lookup_names,
370                         lsa_io_r_lookup_names,
371                         NT_STATUS_UNSUCCESSFUL);
372
373         result = r.status;
374
375         if (!NT_STATUS_IS_OK(result) && NT_STATUS_V(result) !=
376             NT_STATUS_V(STATUS_SOME_UNMAPPED)) {
377
378                 /* An actual error occured */
379
380                 goto done;
381         }
382
383         /* Return output parameters */
384
385         if (r.mapped_count == 0) {
386                 result = NT_STATUS_NONE_MAPPED;
387                 goto done;
388         }
389
390         if (num_names) {
391                 if (!((*sids = TALLOC_ARRAY(mem_ctx, DOM_SID, num_names)))) {
392                         DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
393                         result = NT_STATUS_NO_MEMORY;
394                         goto done;
395                 }
396
397                 if (!((*types = TALLOC_ARRAY(mem_ctx, enum lsa_SidType, num_names)))) {
398                         DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
399                         result = NT_STATUS_NO_MEMORY;
400                         goto done;
401                 }
402
403                 if (dom_names != NULL) {
404                         *dom_names = TALLOC_ARRAY(mem_ctx, const char *, num_names);
405                         if (*dom_names == NULL) {
406                                 DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
407                                 result = NT_STATUS_NO_MEMORY;
408                                 goto done;
409                         }
410                 }
411         } else {
412                 *sids = NULL;
413                 *types = NULL;
414                 if (dom_names != NULL) {
415                         *dom_names = NULL;
416                 }
417         }
418
419         for (i = 0; i < num_names; i++) {
420                 DOM_RID *t_rids = r.dom_rid;
421                 uint32 dom_idx = t_rids[i].rid_idx;
422                 uint32 dom_rid = t_rids[i].rid;
423                 DOM_SID *sid = &(*sids)[i];
424
425                 /* Translate optimised sid through domain index array */
426
427                 if (dom_idx == 0xffffffff) {
428                         /* Nothing to do, this is unknown */
429                         ZERO_STRUCTP(sid);
430                         (*types)[i] = SID_NAME_UNKNOWN;
431                         continue;
432                 }
433
434                 sid_copy(sid, &ref.ref_dom[dom_idx].ref_dom.sid);
435
436                 if (dom_rid != 0xffffffff) {
437                         sid_append_rid(sid, dom_rid);
438                 }
439
440                 (*types)[i] = t_rids[i].type;
441
442                 if (dom_names == NULL) {
443                         continue;
444                 }
445
446                 (*dom_names)[i] = rpcstr_pull_unistr2_talloc(
447                         *dom_names, &ref.ref_dom[dom_idx].uni_dom_name);
448         }
449
450  done:
451
452         return result;
453 }
454
455 /**
456  * Enumerate list of trusted domains
457  *
458  * @param cli client state (cli_state) structure of the connection
459  * @param mem_ctx memory context
460  * @param pol opened lsa policy handle
461  * @param enum_ctx enumeration context ie. index of first returned domain entry
462  * @param pref_num_domains preferred max number of entries returned in one response
463  * @param num_domains total number of trusted domains returned by response
464  * @param domain_names returned trusted domain names
465  * @param domain_sids returned trusted domain sids
466  *
467  * @return nt status code of response
468  **/
469
470 NTSTATUS rpccli_lsa_enum_trust_dom(struct rpc_pipe_client *cli,
471                                    TALLOC_CTX *mem_ctx,
472                                    POLICY_HND *pol, uint32 *enum_ctx,
473                                    uint32 *num_domains,
474                                    char ***domain_names, DOM_SID **domain_sids)
475 {
476         prs_struct qbuf, rbuf;
477         LSA_Q_ENUM_TRUST_DOM in;
478         LSA_R_ENUM_TRUST_DOM out;
479         int i;
480         fstring tmp;
481
482         ZERO_STRUCT(in);
483         ZERO_STRUCT(out);
484
485         /* 64k is enough for about 2000 trusted domains */
486
487         init_q_enum_trust_dom(&in, pol, *enum_ctx, 0x10000);
488
489         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUMTRUSTDOM,
490                     in, out,
491                     qbuf, rbuf,
492                     lsa_io_q_enum_trust_dom,
493                     lsa_io_r_enum_trust_dom,
494                     NT_STATUS_UNSUCCESSFUL );
495
496
497         /* check for an actual error */
498
499         if ( !NT_STATUS_IS_OK(out.status)
500                 && !NT_STATUS_EQUAL(out.status, NT_STATUS_NO_MORE_ENTRIES)
501                 && !NT_STATUS_EQUAL(out.status, STATUS_MORE_ENTRIES) )
502         {
503                 return out.status;
504         }
505
506         /* Return output parameters */
507
508         *num_domains  = out.count;
509         *enum_ctx     = out.enum_context;
510
511         if ( out.count ) {
512
513                 /* Allocate memory for trusted domain names and sids */
514
515                 if ( !(*domain_names = TALLOC_ARRAY(mem_ctx, char *, out.count)) ) {
516                         DEBUG(0, ("cli_lsa_enum_trust_dom(): out of memory\n"));
517                         return NT_STATUS_NO_MEMORY;
518                 }
519
520                 if ( !(*domain_sids = TALLOC_ARRAY(mem_ctx, DOM_SID, out.count)) ) {
521                         DEBUG(0, ("cli_lsa_enum_trust_dom(): out of memory\n"));
522                         return NT_STATUS_NO_MEMORY;
523                 }
524
525                 /* Copy across names and sids */
526
527                 for (i = 0; i < out.count; i++) {
528
529                         rpcstr_pull( tmp, out.domlist->domains[i].name.string->buffer,
530                                 sizeof(tmp), out.domlist->domains[i].name.length, 0);
531                         (*domain_names)[i] = talloc_strdup(mem_ctx, tmp);
532
533                         sid_copy(&(*domain_sids)[i], &out.domlist->domains[i].sid->sid );
534                 }
535         }
536
537         return out.status;
538 }
539
540 /** Enumerate list of SIDs  */
541
542 NTSTATUS rpccli_lsa_enum_sids(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
543                                 POLICY_HND *pol, uint32 *enum_ctx, uint32 pref_max_length,
544                                 uint32 *num_sids, DOM_SID **sids)
545 {
546         prs_struct qbuf, rbuf;
547         LSA_Q_ENUM_ACCOUNTS q;
548         LSA_R_ENUM_ACCOUNTS r;
549         NTSTATUS result;
550         int i;
551
552         ZERO_STRUCT(q);
553         ZERO_STRUCT(r);
554
555         init_lsa_q_enum_accounts(&q, pol, *enum_ctx, pref_max_length);
556
557         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUM_ACCOUNTS,
558                 q, r,
559                 qbuf, rbuf,
560                 lsa_io_q_enum_accounts,
561                 lsa_io_r_enum_accounts,
562                 NT_STATUS_UNSUCCESSFUL);
563
564         result = r.status;
565
566         if (!NT_STATUS_IS_OK(result)) {
567                 goto done;
568         }
569
570         if (r.sids.num_entries==0)
571                 goto done;
572
573         /* Return output parameters */
574
575         *sids = TALLOC_ARRAY(mem_ctx, DOM_SID, r.sids.num_entries);
576         if (!*sids) {
577                 DEBUG(0, ("(cli_lsa_enum_sids): out of memory\n"));
578                 result = NT_STATUS_UNSUCCESSFUL;
579                 goto done;
580         }
581
582         /* Copy across names and sids */
583
584         for (i = 0; i < r.sids.num_entries; i++) {
585                 sid_copy(&(*sids)[i], &r.sids.sid[i].sid);
586         }
587
588         *num_sids= r.sids.num_entries;
589         *enum_ctx = r.enum_context;
590
591  done:
592
593         return result;
594 }
595
596 /** Enumerate user privileges
597  *
598  * @param cli Handle on an initialised SMB connection */
599
600 NTSTATUS rpccli_lsa_enum_privsaccount(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
601                              POLICY_HND *pol, uint32 *count, LUID_ATTR **set)
602 {
603         prs_struct qbuf, rbuf;
604         LSA_Q_ENUMPRIVSACCOUNT q;
605         LSA_R_ENUMPRIVSACCOUNT r;
606         NTSTATUS result;
607         int i;
608
609         ZERO_STRUCT(q);
610         ZERO_STRUCT(r);
611
612         /* Initialise input parameters */
613
614         init_lsa_q_enum_privsaccount(&q, pol);
615
616         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUMPRIVSACCOUNT,
617                 q, r,
618                 qbuf, rbuf,
619                 lsa_io_q_enum_privsaccount,
620                 lsa_io_r_enum_privsaccount,
621                 NT_STATUS_UNSUCCESSFUL);
622
623         /* Return output parameters */
624
625         result = r.status;
626
627         if (!NT_STATUS_IS_OK(result)) {
628                 goto done;
629         }
630
631         if (r.count == 0)
632                 goto done;
633
634         if (!((*set = TALLOC_ARRAY(mem_ctx, LUID_ATTR, r.count)))) {
635                 DEBUG(0, ("(cli_lsa_enum_privsaccount): out of memory\n"));
636                 result = NT_STATUS_UNSUCCESSFUL;
637                 goto done;
638         }
639
640         for (i=0; i<r.count; i++) {
641                 (*set)[i].luid.low = r.set.set[i].luid.low;
642                 (*set)[i].luid.high = r.set.set[i].luid.high;
643                 (*set)[i].attr = r.set.set[i].attr;
644         }
645
646         *count=r.count;
647  done:
648
649         return result;
650 }
651
652 /** Get a privilege value given its name */
653
654 NTSTATUS rpccli_lsa_lookup_priv_value(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
655                                  POLICY_HND *pol, const char *name, LUID *luid)
656 {
657         prs_struct qbuf, rbuf;
658         LSA_Q_LOOKUP_PRIV_VALUE q;
659         LSA_R_LOOKUP_PRIV_VALUE r;
660         NTSTATUS result;
661
662         ZERO_STRUCT(q);
663         ZERO_STRUCT(r);
664
665         /* Marshall data and send request */
666
667         init_lsa_q_lookup_priv_value(&q, pol, name);
668
669         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_LOOKUPPRIVVALUE,
670                 q, r,
671                 qbuf, rbuf,
672                 lsa_io_q_lookup_priv_value,
673                 lsa_io_r_lookup_priv_value,
674                 NT_STATUS_UNSUCCESSFUL);
675
676         result = r.status;
677
678         if (!NT_STATUS_IS_OK(result)) {
679                 goto done;
680         }
681
682         /* Return output parameters */
683
684         (*luid).low=r.luid.low;
685         (*luid).high=r.luid.high;
686
687  done:
688
689         return result;
690 }
691
692 /* Enumerate account rights This is similar to enum_privileges but
693    takes a SID directly, avoiding the open_account call.
694 */
695
696 NTSTATUS rpccli_lsa_enum_account_rights(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
697                                      POLICY_HND *pol, DOM_SID *sid,
698                                      uint32 *count, char ***priv_names)
699 {
700         prs_struct qbuf, rbuf;
701         LSA_Q_ENUM_ACCT_RIGHTS q;
702         LSA_R_ENUM_ACCT_RIGHTS r;
703         NTSTATUS result;
704         int i;
705         fstring *privileges;
706         char **names;
707
708         ZERO_STRUCT(q);
709         ZERO_STRUCT(r);
710
711         /* Marshall data and send request */
712         init_q_enum_acct_rights(&q, pol, 2, sid);
713
714         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUMACCTRIGHTS,
715                 q, r,
716                 qbuf, rbuf,
717                 lsa_io_q_enum_acct_rights,
718                 lsa_io_r_enum_acct_rights,
719                 NT_STATUS_UNSUCCESSFUL);
720
721         result = r.status;
722
723         if (!NT_STATUS_IS_OK(result)) {
724                 goto done;
725         }
726
727         *count = r.count;
728         if (! *count) {
729                 goto done;
730         }
731
732
733         privileges = TALLOC_ARRAY( mem_ctx, fstring, *count );
734         names      = TALLOC_ARRAY( mem_ctx, char *, *count );
735
736         if ((privileges == NULL) || (names == NULL)) {
737                 TALLOC_FREE(privileges);
738                 TALLOC_FREE(names);
739                 return NT_STATUS_NO_MEMORY;
740         }
741
742         for ( i=0; i<*count; i++ ) {
743                 UNISTR4 *uni_string = &r.rights->strings[i];
744
745                 if ( !uni_string->string )
746                         continue;
747
748                 rpcstr_pull( privileges[i], uni_string->string->buffer, sizeof(privileges[i]), -1, STR_TERMINATE );
749
750                 /* now copy to the return array */
751                 names[i] = talloc_strdup( mem_ctx, privileges[i] );
752         }
753
754         *priv_names = names;
755
756 done:
757
758         return result;
759 }
760
761
762
763 /* add account rights to an account. */
764
765 NTSTATUS rpccli_lsa_add_account_rights(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
766                                     POLICY_HND *pol, DOM_SID sid,
767                                         uint32 count, const char **privs_name)
768 {
769         prs_struct qbuf, rbuf;
770         LSA_Q_ADD_ACCT_RIGHTS q;
771         LSA_R_ADD_ACCT_RIGHTS r;
772         NTSTATUS result;
773
774         ZERO_STRUCT(q);
775         ZERO_STRUCT(r);
776
777         /* Marshall data and send request */
778         init_q_add_acct_rights(&q, pol, &sid, count, privs_name);
779
780         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ADDACCTRIGHTS,
781                 q, r,
782                 qbuf, rbuf,
783                 lsa_io_q_add_acct_rights,
784                 lsa_io_r_add_acct_rights,
785                 NT_STATUS_UNSUCCESSFUL);
786
787         result = r.status;
788
789         if (!NT_STATUS_IS_OK(result)) {
790                 goto done;
791         }
792 done:
793
794         return result;
795 }
796
797
798 /* remove account rights for an account. */
799
800 NTSTATUS rpccli_lsa_remove_account_rights(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
801                                        POLICY_HND *pol, DOM_SID sid, bool removeall,
802                                        uint32 count, const char **privs_name)
803 {
804         prs_struct qbuf, rbuf;
805         LSA_Q_REMOVE_ACCT_RIGHTS q;
806         LSA_R_REMOVE_ACCT_RIGHTS r;
807         NTSTATUS result;
808
809         ZERO_STRUCT(q);
810         ZERO_STRUCT(r);
811
812         /* Marshall data and send request */
813         init_q_remove_acct_rights(&q, pol, &sid, removeall?1:0, count, privs_name);
814
815         CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_REMOVEACCTRIGHTS,
816                 q, r,
817                 qbuf, rbuf,
818                 lsa_io_q_remove_acct_rights,
819                 lsa_io_r_remove_acct_rights,
820                 NT_STATUS_UNSUCCESSFUL);
821
822         result = r.status;
823
824         if (!NT_STATUS_IS_OK(result)) {
825                 goto done;
826         }
827 done:
828
829         return result;
830 }
831
832
833 #if 0
834
835 /** An example of how to use the routines in this file.  Fetch a DOMAIN
836     sid. Does complete cli setup / teardown anonymously. */
837
838 bool fetch_domain_sid( char *domain, char *remote_machine, DOM_SID *psid)
839 {
840         struct cli_state cli;
841         NTSTATUS result;
842         POLICY_HND lsa_pol;
843         bool ret = False;
844
845         ZERO_STRUCT(cli);
846         if(cli_initialise(&cli) == False) {
847                 DEBUG(0,("fetch_domain_sid: unable to initialize client connection.\n"));
848                 return False;
849         }
850
851         if(!resolve_name( remote_machine, &cli.dest_ip, 0x20)) {
852                 DEBUG(0,("fetch_domain_sid: Can't resolve address for %s\n", remote_machine));
853                 goto done;
854         }
855
856         if (!cli_connect(&cli, remote_machine, &cli.dest_ip)) {
857                 DEBUG(0,("fetch_domain_sid: unable to connect to SMB server on \
858 machine %s. Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
859                 goto done;
860         }
861
862         if (!attempt_netbios_session_request(&cli, global_myname(), remote_machine, &cli.dest_ip)) {
863                 DEBUG(0,("fetch_domain_sid: machine %s rejected the NetBIOS session request.\n",
864                         remote_machine));
865                 goto done;
866         }
867
868         cli.protocol = PROTOCOL_NT1;
869
870         if (!cli_negprot(&cli)) {
871                 DEBUG(0,("fetch_domain_sid: machine %s rejected the negotiate protocol. \
872 Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
873                 goto done;
874         }
875
876         if (cli.protocol != PROTOCOL_NT1) {
877                 DEBUG(0,("fetch_domain_sid: machine %s didn't negotiate NT protocol.\n",
878                         remote_machine));
879                 goto done;
880         }
881
882         /*
883          * Do an anonymous session setup.
884          */
885
886         if (!cli_session_setup(&cli, "", "", 0, "", 0, "")) {
887                 DEBUG(0,("fetch_domain_sid: machine %s rejected the session setup. \
888 Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
889                 goto done;
890         }
891
892         if (!(cli.sec_mode & NEGOTIATE_SECURITY_USER_LEVEL)) {
893                 DEBUG(0,("fetch_domain_sid: machine %s isn't in user level security mode\n",
894                         remote_machine));
895                 goto done;
896         }
897
898         if (!cli_send_tconX(&cli, "IPC$", "IPC", "", 1)) {
899                 DEBUG(0,("fetch_domain_sid: machine %s rejected the tconX on the IPC$ share. \
900 Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
901                 goto done;
902         }
903
904         /* Fetch domain sid */
905
906         if (!cli_nt_session_open(&cli, PI_LSARPC)) {
907                 DEBUG(0, ("fetch_domain_sid: Error connecting to SAM pipe\n"));
908                 goto done;
909         }
910
911         result = cli_lsa_open_policy(&cli, cli.mem_ctx, True, SEC_RIGHTS_QUERY_VALUE, &lsa_pol);
912         if (!NT_STATUS_IS_OK(result)) {
913                 DEBUG(0, ("fetch_domain_sid: Error opening lsa policy handle. %s\n",
914                         nt_errstr(result) ));
915                 goto done;
916         }
917
918         result = cli_lsa_query_info_policy(&cli, cli.mem_ctx, &lsa_pol, 5, domain, psid);
919         if (!NT_STATUS_IS_OK(result)) {
920                 DEBUG(0, ("fetch_domain_sid: Error querying lsa policy handle. %s\n",
921                         nt_errstr(result) ));
922                 goto done;
923         }
924
925         ret = True;
926
927   done:
928
929         cli_shutdown(&cli);
930         return ret;
931 }
932
933 #endif