Deal with c_req send errors asynchronously
[Samba/vfs_proxy.git] / source4 / libnet / libnet_group.c
blobaf5fe4d5d375ed18d9f16ad2746e73766f002f7b
1 /*
2 Unix SMB/CIFS implementation.
4 Copyright (C) Rafal Szczesniak 2007
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "includes.h"
22 #include "libnet/libnet.h"
23 #include "libcli/composite/composite.h"
24 #include "librpc/gen_ndr/lsa.h"
25 #include "librpc/gen_ndr/ndr_lsa_c.h"
26 #include "librpc/gen_ndr/samr.h"
27 #include "librpc/gen_ndr/ndr_samr_c.h"
28 #include "libcli/security/security.h"
31 struct create_group_state {
32 struct libnet_context *ctx;
33 struct libnet_CreateGroup r;
34 struct libnet_DomainOpen domain_open;
35 struct libnet_rpc_groupadd group_add;
37 /* information about the progress */
38 void (*monitor_fn)(struct monitor_msg *);
42 static void continue_domain_opened(struct composite_context *ctx);
43 static void continue_rpc_group_added(struct composite_context *ctx);
46 struct composite_context* libnet_CreateGroup_send(struct libnet_context *ctx,
47 TALLOC_CTX *mem_ctx,
48 struct libnet_CreateGroup *r,
49 void (*monitor)(struct monitor_msg*))
51 struct composite_context *c;
52 struct create_group_state *s;
53 struct composite_context *create_req;
54 bool prereq_met = false;
56 /* composite context allocation and setup */
57 c = composite_create(mem_ctx, ctx->event_ctx);
58 if (c == NULL) return NULL;
60 s = talloc_zero(c, struct create_group_state);
61 if (composite_nomem(s, c)) return c;
63 c->private_data = s;
65 s->ctx = ctx;
66 s->r = *r;
67 ZERO_STRUCT(s->r.out);
69 /* prerequisite: make sure we have a valid samr domain handle */
70 prereq_met = samr_domain_opened(ctx, s->r.in.domain_name, &c, &s->domain_open,
71 continue_domain_opened, monitor);
72 if (!prereq_met) return c;
74 /* prepare arguments of rpc group add call */
75 s->group_add.in.groupname = r->in.group_name;
76 s->group_add.in.domain_handle = ctx->samr.handle;
78 /* send the request */
79 create_req = libnet_rpc_groupadd_send(ctx->samr.pipe, &s->group_add, monitor);
80 if (composite_nomem(create_req, c)) return c;
82 composite_continue(c, create_req, continue_rpc_group_added, c);
83 return c;
87 static void continue_domain_opened(struct composite_context *ctx)
89 struct composite_context *c;
90 struct create_group_state *s;
91 struct composite_context *create_req;
93 c = talloc_get_type(ctx->async.private_data, struct composite_context);
94 s = talloc_get_type(c->private_data, struct create_group_state);
96 c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
97 if (!composite_is_ok(c)) return;
99 /* prepare arguments of groupadd call */
100 s->group_add.in.groupname = s->r.in.group_name;
101 s->group_add.in.domain_handle = s->ctx->samr.handle;
103 /* send the request */
104 create_req = libnet_rpc_groupadd_send(s->ctx->samr.pipe, &s->group_add,
105 s->monitor_fn);
106 if (composite_nomem(create_req, c)) return;
108 composite_continue(c, create_req, continue_rpc_group_added, c);
112 static void continue_rpc_group_added(struct composite_context *ctx)
114 struct composite_context *c;
115 struct create_group_state *s;
117 c = talloc_get_type(ctx->async.private_data, struct composite_context);
118 s = talloc_get_type(c->private_data, struct create_group_state);
120 /* receive result of group add call */
121 c->status = libnet_rpc_groupadd_recv(ctx, c, &s->group_add);
122 if (!composite_is_ok(c)) return;
124 /* we're done */
125 composite_done(c);
130 * Receive result of CreateGroup call
132 * @param c composite context returned by send request routine
133 * @param mem_ctx memory context of this call
134 * @param r pointer to a structure containing arguments and result of this call
135 * @return nt status
137 NTSTATUS libnet_CreateGroup_recv(struct composite_context *c,
138 TALLOC_CTX *mem_ctx,
139 struct libnet_CreateGroup *r)
141 NTSTATUS status;
142 struct create_group_state *s;
144 status = composite_wait(c);
145 if (!NT_STATUS_IS_OK(status)) {
146 s = talloc_get_type(c->private_data, struct create_group_state);
147 r->out.error_string = talloc_strdup(mem_ctx, nt_errstr(status));
150 return status;
155 * Create domain group
157 * @param ctx initialised libnet context
158 * @param mem_ctx memory context of this call
159 * @param io pointer to structure containing arguments and result of this call
160 * @return nt status
162 NTSTATUS libnet_CreateGroup(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
163 struct libnet_CreateGroup *io)
165 struct composite_context *c;
167 c = libnet_CreateGroup_send(ctx, mem_ctx, io, NULL);
168 return libnet_CreateGroup_recv(c, mem_ctx, io);
172 struct group_info_state {
173 struct libnet_context *ctx;
174 const char *domain_name;
175 const char *group_name;
176 struct libnet_LookupName lookup;
177 struct libnet_DomainOpen domopen;
178 struct libnet_rpc_groupinfo info;
180 /* information about the progress */
181 void (*monitor_fn)(struct monitor_msg *);
185 static void continue_domain_open_info(struct composite_context *ctx);
186 static void continue_name_found(struct composite_context *ctx);
187 static void continue_group_info(struct composite_context *ctx);
190 * Sends request to get group information
192 * @param ctx initialised libnet context
193 * @param mem_ctx memory context of this call
194 * @param io pointer to structure containing arguments the call
195 * @param monitor function pointer for receiving monitor messages
196 * @return composite context of this request
198 struct composite_context* libnet_GroupInfo_send(struct libnet_context *ctx,
199 TALLOC_CTX *mem_ctx,
200 struct libnet_GroupInfo *io,
201 void (*monitor)(struct monitor_msg*))
203 struct composite_context *c;
204 struct group_info_state *s;
205 bool prereq_met = false;
206 struct composite_context *lookup_req;
208 /* composite context allocation and setup */
209 c = composite_create(mem_ctx, ctx->event_ctx);
210 if (c == NULL) return NULL;
212 s = talloc_zero(c, struct group_info_state);
213 if (composite_nomem(s, c)) return c;
215 c->private_data = s;
217 /* store arguments in the state structure */
218 s->monitor_fn = monitor;
219 s->ctx = ctx;
220 s->domain_name = talloc_strdup(c, io->in.domain_name);
221 s->group_name = talloc_strdup(c, io->in.group_name);
223 /* prerequisite: make sure the domain is opened */
224 prereq_met = samr_domain_opened(ctx, s->domain_name, &c, &s->domopen,
225 continue_domain_open_info, monitor);
226 if (!prereq_met) return c;
228 /* prepare arguments for LookupName call */
229 s->lookup.in.name = s->group_name;
230 s->lookup.in.domain_name = s->domain_name;
232 /* send the request */
233 lookup_req = libnet_LookupName_send(s->ctx, c, &s->lookup, s->monitor_fn);
234 if (composite_nomem(lookup_req, c)) return c;
236 /* set the next stage */
237 composite_continue(c, lookup_req, continue_name_found, c);
238 return c;
243 * Stage 0.5 (optional): receive opened domain and send lookup name request
245 static void continue_domain_open_info(struct composite_context *ctx)
247 struct composite_context *c;
248 struct group_info_state *s;
249 struct composite_context *lookup_req;
251 c = talloc_get_type(ctx->async.private_data, struct composite_context);
252 s = talloc_get_type(c->private_data, struct group_info_state);
254 /* receive domain handle */
255 c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domopen);
256 if (!composite_is_ok(c)) return;
258 /* prepare arguments for LookupName call */
259 s->lookup.in.name = s->group_name;
260 s->lookup.in.domain_name = s->domain_name;
262 /* send the request */
263 lookup_req = libnet_LookupName_send(s->ctx, c, &s->lookup, s->monitor_fn);
264 if (composite_nomem(lookup_req, c)) return;
266 /* set the next stage */
267 composite_continue(c, lookup_req, continue_name_found, c);
272 * Stage 1: Receive SID found and send request for group info
274 static void continue_name_found(struct composite_context *ctx)
276 struct composite_context *c;
277 struct group_info_state *s;
278 struct composite_context *info_req;
280 c = talloc_get_type(ctx->async.private_data, struct composite_context);
281 s = talloc_get_type(c->private_data, struct group_info_state);
283 /* receive SID assiociated with name found */
284 c->status = libnet_LookupName_recv(ctx, c, &s->lookup);
285 if (!composite_is_ok(c)) return;
287 /* Is is a group SID actually ? */
288 if (s->lookup.out.sid_type != SID_NAME_DOM_GRP &&
289 s->lookup.out.sid_type != SID_NAME_ALIAS) {
290 composite_error(c, NT_STATUS_NO_SUCH_GROUP);
293 /* prepare arguments for groupinfo call */
294 s->info.in.domain_handle = s->ctx->samr.handle;
295 s->info.in.groupname = s->group_name;
296 s->info.in.sid = s->lookup.out.sidstr;
297 /* we're looking for all information available */
298 s->info.in.level = GROUPINFOALL;
300 /* send the request */
301 info_req = libnet_rpc_groupinfo_send(s->ctx->samr.pipe, &s->info, s->monitor_fn);
302 if (composite_nomem(info_req, c)) return;
304 /* set the next stage */
305 composite_continue(c, info_req, continue_group_info, c);
310 * Stage 2: Receive group information
312 static void continue_group_info(struct composite_context *ctx)
314 struct composite_context *c;
315 struct group_info_state *s;
317 c = talloc_get_type(ctx->async.private_data, struct composite_context);
318 s = talloc_get_type(c->private_data, struct group_info_state);
320 /* receive group information */
321 c->status = libnet_rpc_groupinfo_recv(ctx, c, &s->info);
322 if (!composite_is_ok(c)) return;
324 /* we're done */
325 composite_done(c);
330 * Receive group information
332 * @param c composite context returned by libnet_GroupInfo_send
333 * @param mem_ctx memory context of this call
334 * @param io pointer to structure receiving results of the call
335 * @result nt status
337 NTSTATUS libnet_GroupInfo_recv(struct composite_context* c, TALLOC_CTX *mem_ctx,
338 struct libnet_GroupInfo *io)
340 NTSTATUS status;
341 struct group_info_state *s;
343 status = composite_wait(c);
344 if (NT_STATUS_IS_OK(status)) {
345 /* put the results into io structure if everything went fine */
346 s = talloc_get_type(c->private_data, struct group_info_state);
348 io->out.group_sid = talloc_steal(mem_ctx, s->lookup.out.sid);
349 io->out.num_members = s->info.out.info.all.num_members;
350 io->out.description = talloc_steal(mem_ctx, s->info.out.info.all.description.string);
352 io->out.error_string = talloc_strdup(mem_ctx, "Success");
354 } else {
355 io->out.error_string = talloc_asprintf(mem_ctx, "Error: %s", nt_errstr(status));
358 talloc_free(c);
360 return status;
365 * Obtains specified group information
367 * @param ctx initialised libnet context
368 * @param mem_ctx memory context of the call
369 * @param io pointer to a structure containing arguments and results of the call
371 NTSTATUS libnet_GroupInfo(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
372 struct libnet_GroupInfo *io)
374 struct composite_context *c = libnet_GroupInfo_send(ctx, mem_ctx,
375 io, NULL);
376 return libnet_GroupInfo_recv(c, mem_ctx, io);
380 struct grouplist_state {
381 struct libnet_context *ctx;
382 const char *domain_name;
383 struct lsa_DomainInfo dominfo;
384 int page_size;
385 uint32_t resume_index;
386 struct grouplist *groups;
387 uint32_t count;
389 struct libnet_DomainOpen domain_open;
390 struct lsa_QueryInfoPolicy query_domain;
391 struct samr_EnumDomainGroups group_list;
393 void (*monitor_fn)(struct monitor_msg*);
397 static void continue_lsa_domain_opened(struct composite_context *ctx);
398 static void continue_domain_queried(struct rpc_request *req);
399 static void continue_samr_domain_opened(struct composite_context *ctx);
400 static void continue_domain_queried(struct rpc_request *req);
401 static void continue_groups_enumerated(struct rpc_request *req);
405 * Sends request to list (enumerate) group accounts
407 * @param ctx initialised libnet context
408 * @param mem_ctx memory context of this call
409 * @param io pointer to structure containing arguments and results of this call
410 * @param monitor function pointer for receiving monitor messages
411 * @return compostite context of this request
413 struct composite_context *libnet_GroupList_send(struct libnet_context *ctx,
414 TALLOC_CTX *mem_ctx,
415 struct libnet_GroupList *io,
416 void (*monitor)(struct monitor_msg*))
418 struct composite_context *c;
419 struct grouplist_state *s;
420 struct rpc_request *query_req;
421 bool prereq_met = false;
423 /* composite context allocation and setup */
424 c = composite_create(mem_ctx, ctx->event_ctx);
425 if (c == NULL) return NULL;
427 s = talloc_zero(c, struct grouplist_state);
428 if (composite_nomem(s, c)) return c;
430 c->private_data = s;
432 /* store the arguments in the state structure */
433 s->ctx = ctx;
434 s->page_size = io->in.page_size;
435 s->resume_index = (uint32_t)io->in.resume_index;
436 s->domain_name = talloc_strdup(c, io->in.domain_name);
437 s->monitor_fn = monitor;
439 /* make sure we have lsa domain handle before doing anything */
440 prereq_met = lsa_domain_opened(ctx, s->domain_name, &c, &s->domain_open,
441 continue_lsa_domain_opened, monitor);
442 if (!prereq_met) return c;
444 /* prepare arguments of QueryDomainInfo call */
445 s->query_domain.in.handle = &ctx->lsa.handle;
446 s->query_domain.in.level = LSA_POLICY_INFO_DOMAIN;
447 s->query_domain.out.info = talloc_zero(c, union lsa_PolicyInformation *);
448 if (composite_nomem(s->query_domain.out.info, c)) return c;
450 /* send the request */
451 query_req = dcerpc_lsa_QueryInfoPolicy_send(ctx->lsa.pipe, c, &s->query_domain);
452 if (composite_nomem(query_req, c)) return c;
454 composite_continue_rpc(c, query_req, continue_domain_queried, c);
455 return c;
460 * Stage 0.5 (optional): receive lsa domain handle and send
461 * request to query domain info
463 static void continue_lsa_domain_opened(struct composite_context *ctx)
465 struct composite_context *c;
466 struct grouplist_state *s;
467 struct rpc_request *query_req;
469 c = talloc_get_type(ctx->async.private_data, struct composite_context);
470 s = talloc_get_type(c->private_data, struct grouplist_state);
472 /* receive lsa domain handle */
473 c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
474 if (!composite_is_ok(c)) return;
476 /* prepare arguments of QueryDomainInfo call */
477 s->query_domain.in.handle = &s->ctx->lsa.handle;
478 s->query_domain.in.level = LSA_POLICY_INFO_DOMAIN;
479 s->query_domain.out.info = talloc_zero(c, union lsa_PolicyInformation *);
480 if (composite_nomem(s->query_domain.out.info, c)) return;
482 /* send the request */
483 query_req = dcerpc_lsa_QueryInfoPolicy_send(s->ctx->lsa.pipe, c, &s->query_domain);
484 if (composite_nomem(query_req, c)) return;
486 composite_continue_rpc(c, query_req, continue_domain_queried, c);
491 * Stage 1: receive domain info and request to enum groups
492 * provided a valid samr handle is opened
494 static void continue_domain_queried(struct rpc_request *req)
496 struct composite_context *c;
497 struct grouplist_state *s;
498 struct rpc_request *enum_req;
499 bool prereq_met = false;
501 c = talloc_get_type(req->async.private_data, struct composite_context);
502 s = talloc_get_type(c->private_data, struct grouplist_state);
504 /* receive result of rpc request */
505 c->status = dcerpc_ndr_request_recv(req);
506 if (!composite_is_ok(c)) return;
508 /* get the returned domain info */
509 s->dominfo = (*s->query_domain.out.info)->domain;
511 /* make sure we have samr domain handle before continuing */
512 prereq_met = samr_domain_opened(s->ctx, s->domain_name, &c, &s->domain_open,
513 continue_samr_domain_opened, s->monitor_fn);
514 if (!prereq_met) return;
516 /* prepare arguments od EnumDomainGroups call */
517 s->group_list.in.domain_handle = &s->ctx->samr.handle;
518 s->group_list.in.max_size = s->page_size;
519 s->group_list.in.resume_handle = &s->resume_index;
520 s->group_list.out.resume_handle = &s->resume_index;
521 s->group_list.out.num_entries = talloc(s, uint32_t);
522 if (composite_nomem(s->group_list.out.num_entries, c)) return;
523 s->group_list.out.sam = talloc(s, struct samr_SamArray *);
524 if (composite_nomem(s->group_list.out.sam, c)) return;
526 /* send the request */
527 enum_req = dcerpc_samr_EnumDomainGroups_send(s->ctx->samr.pipe, c, &s->group_list);
528 if (composite_nomem(enum_req, c)) return;
530 composite_continue_rpc(c, enum_req, continue_groups_enumerated, c);
535 * Stage 1.5 (optional): receive samr domain handle
536 * and request to enumerate accounts
538 static void continue_samr_domain_opened(struct composite_context *ctx)
540 struct composite_context *c;
541 struct grouplist_state *s;
542 struct rpc_request *enum_req;
544 c = talloc_get_type(ctx->async.private_data, struct composite_context);
545 s = talloc_get_type(c->private_data, struct grouplist_state);
547 /* receive samr domain handle */
548 c->status = libnet_DomainOpen_recv(ctx, s->ctx, c, &s->domain_open);
549 if (!composite_is_ok(c)) return;
551 /* prepare arguments of EnumDomainGroups call */
552 s->group_list.in.domain_handle = &s->ctx->samr.handle;
553 s->group_list.in.max_size = s->page_size;
554 s->group_list.in.resume_handle = &s->resume_index;
555 s->group_list.out.resume_handle = &s->resume_index;
556 s->group_list.out.num_entries = talloc(s, uint32_t);
557 if (composite_nomem(s->group_list.out.num_entries, c)) return;
558 s->group_list.out.sam = talloc(s, struct samr_SamArray *);
559 if (composite_nomem(s->group_list.out.sam, c)) return;
561 /* send the request */
562 enum_req = dcerpc_samr_EnumDomainGroups_send(s->ctx->samr.pipe, c, &s->group_list);
563 if (composite_nomem(enum_req, c)) return;
565 composite_continue_rpc(c, enum_req, continue_groups_enumerated, c);
570 * Stage 2: receive enumerated groups and their rids
572 static void continue_groups_enumerated(struct rpc_request *req)
574 struct composite_context *c;
575 struct grouplist_state *s;
576 int i;
578 c = talloc_get_type(req->async.private_data, struct composite_context);
579 s = talloc_get_type(c->private_data, struct grouplist_state);
581 /* receive result of rpc request */
582 c->status = dcerpc_ndr_request_recv(req);
583 if (!composite_is_ok(c)) return;
585 /* get the actual status of the rpc call result
586 (instead of rpc layer) */
587 c->status = s->group_list.out.result;
589 /* we're interested in status "ok" as well as two
590 enum-specific status codes */
591 if (NT_STATUS_IS_OK(c->status) ||
592 NT_STATUS_EQUAL(c->status, STATUS_MORE_ENTRIES) ||
593 NT_STATUS_EQUAL(c->status, NT_STATUS_NO_MORE_ENTRIES)) {
595 /* get enumerated accounts counter and resume handle (the latter allows
596 making subsequent call to continue enumeration) */
597 s->resume_index = *s->group_list.out.resume_handle;
598 s->count = *s->group_list.out.num_entries;
600 /* prepare returned group accounts array */
601 s->groups = talloc_array(c, struct grouplist, (*s->group_list.out.sam)->count);
602 if (composite_nomem(s->groups, c)) return;
604 for (i = 0; i < (*s->group_list.out.sam)->count; i++) {
605 struct dom_sid *group_sid;
606 struct samr_SamEntry *entry = &(*s->group_list.out.sam)->entries[i];
607 struct dom_sid *domain_sid = (*s->query_domain.out.info)->domain.sid;
609 /* construct group sid from returned rid and queried domain sid */
610 group_sid = dom_sid_add_rid(c, domain_sid, entry->idx);
611 if (composite_nomem(group_sid, c)) return;
613 /* groupname */
614 s->groups[i].groupname = talloc_strdup(c, entry->name.string);
615 if (composite_nomem(s->groups[i].groupname, c)) return;
617 /* sid string */
618 s->groups[i].sid = dom_sid_string(c, group_sid);
619 if (composite_nomem(s->groups[i].sid, c)) return;
622 /* that's it */
623 composite_done(c);
625 } else {
626 /* something went wrong */
627 composite_error(c, c->status);
633 * Receive result of GroupList call
635 * @param c composite context returned by send request routine
636 * @param mem_ctx memory context of this call
637 * @param io pointer to structure containing arguments and result of this call
638 * @param nt status
640 NTSTATUS libnet_GroupList_recv(struct composite_context *c, TALLOC_CTX *mem_ctx,
641 struct libnet_GroupList *io)
643 NTSTATUS status;
644 struct grouplist_state *s;
646 if (c == NULL || mem_ctx == NULL || io == NULL) {
647 return NT_STATUS_INVALID_PARAMETER;
650 status = composite_wait(c);
651 if (NT_STATUS_IS_OK(status) ||
652 NT_STATUS_EQUAL(status, STATUS_MORE_ENTRIES) ||
653 NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
655 s = talloc_get_type(c->private_data, struct grouplist_state);
657 /* get results from composite context */
658 io->out.count = s->count;
659 io->out.resume_index = s->resume_index;
660 io->out.groups = talloc_steal(mem_ctx, s->groups);
662 if (NT_STATUS_IS_OK(status)) {
663 io->out.error_string = talloc_asprintf(mem_ctx, "Success");
664 } else {
665 /* success, but we're not done yet */
666 io->out.error_string = talloc_asprintf(mem_ctx, "Success (status: %s)",
667 nt_errstr(status));
670 } else {
671 io->out.error_string = talloc_asprintf(mem_ctx, "Error: %s", nt_errstr(status));
674 return status;
679 * Enumerate domain groups
681 * @param ctx initialised libnet context
682 * @param mem_ctx memory context of this call
683 * @param io pointer to structure containing arguments and result of this call
684 * @return nt status
686 NTSTATUS libnet_GroupList(struct libnet_context *ctx, TALLOC_CTX *mem_ctx,
687 struct libnet_GroupList *io)
689 struct composite_context *c;
691 c = libnet_GroupList_send(ctx, mem_ctx, io, NULL);
692 return libnet_GroupList_recv(c, mem_ctx, io);