From 2774a02f64a57d981924e0fc65b23060803cc469 Mon Sep 17 00:00:00 2001 From: Volker Lendecke Date: Sat, 29 Aug 2009 17:17:47 +0200 Subject: s3:winbind: Convert WINBINDD_LIST_GROUPS to the new API --- source3/Makefile.in | 1 + source3/winbindd/winbindd.c | 3 +- source3/winbindd/winbindd_async.c | 180 ---------------------------- source3/winbindd/winbindd_domain.c | 8 -- source3/winbindd/winbindd_group.c | 6 - source3/winbindd/winbindd_list_groups.c | 204 ++++++++++++++++++++++++++++++++ source3/winbindd/winbindd_misc.c | 122 ------------------- source3/winbindd/winbindd_proto.h | 8 ++ 8 files changed, 215 insertions(+), 317 deletions(-) create mode 100644 source3/winbindd/winbindd_list_groups.c (limited to 'source3') diff --git a/source3/Makefile.in b/source3/Makefile.in index 91534026fa..ebf6024923 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -1210,6 +1210,7 @@ WINBINDD_OBJ1 = \ winbindd/winbindd_dsgetdcname.o \ winbindd/winbindd_getdcname.o \ winbindd/winbindd_list_users.o \ + winbindd/winbindd_list_groups.o \ auth/token_util.o \ ../nsswitch/libwbclient/wb_reqtrans.o \ smbd/connection.o diff --git a/source3/winbindd/winbindd.c b/source3/winbindd/winbindd.c index 1d3762f5b9..e583dae4a9 100644 --- a/source3/winbindd/winbindd.c +++ b/source3/winbindd/winbindd.c @@ -435,7 +435,6 @@ static struct winbindd_dispatch_table { /* Enumeration functions */ - { WINBINDD_LIST_GROUPS, winbindd_list_groups, "LIST_GROUPS" }, { WINBINDD_LIST_TRUSTDOM, winbindd_list_trusted_domains, "LIST_TRUSTDOM" }, @@ -538,6 +537,8 @@ static struct winbindd_async_dispatch_table async_nonpriv_table[] = { winbindd_endgrent_send, winbindd_endgrent_recv }, { WINBINDD_LIST_USERS, "LIST_USERS", winbindd_list_users_send, winbindd_list_users_recv }, + { WINBINDD_LIST_GROUPS, "LIST_GROUPS", + winbindd_list_groups_send, winbindd_list_groups_recv }, { 0, NULL, NULL, NULL } }; diff --git a/source3/winbindd/winbindd_async.c b/source3/winbindd/winbindd_async.c index 1e63ed1fec..6c5d92e71b 100644 --- a/source3/winbindd/winbindd_async.c +++ b/source3/winbindd/winbindd_async.c @@ -454,163 +454,6 @@ enum winbindd_result winbindd_dual_lookupname(struct winbindd_domain *domain, return WINBINDD_OK; } -/* This is the first callback after enumerating users/groups from a domain */ -static void listent_recv(TALLOC_CTX *mem_ctx, bool success, - struct winbindd_response *response, - void *c, void *private_data) -{ - void (*cont)(void *priv, bool succ, fstring dom_name, char *data) = - (void (*)(void *, bool, fstring, char*))c; - - if (!success || response->result != WINBINDD_OK) { - DEBUG(5, ("list_ent() failed!\n")); - cont(private_data, False, response->data.name.dom_name, NULL); - return; - } - - cont(private_data, True, response->data.name.dom_name, - (char *)response->extra_data.data); -} - -/* Request the name of all users/groups in a single domain */ -void winbindd_listent_async(TALLOC_CTX *mem_ctx, - struct winbindd_domain *domain, - void (*cont)(void *private_data, bool success, - fstring dom_name, char* extra_data), - void *private_data, enum ent_type type) -{ - struct winbindd_request request; - - ZERO_STRUCT(request); - if (type == LIST_USERS) - request.cmd = WINBINDD_LIST_USERS; - else if (type == LIST_GROUPS) - request.cmd = WINBINDD_LIST_GROUPS; - - do_async_domain(mem_ctx, domain, &request, listent_recv, - (void *)cont, private_data); -} - -enum winbindd_result winbindd_dual_list_users(struct winbindd_domain *domain, - struct winbindd_cli_state *state) -{ - struct wbint_userinfo *info; - NTSTATUS status; - struct winbindd_methods *methods; - uint32 num_entries = 0; - char *extra_data; - uint32_t extra_data_len = 0, i; - - /* Must copy domain into response first for debugging in parent */ - fstrcpy(state->response->data.name.dom_name, domain->name); - - /* Query user info */ - methods = domain->methods; - status = methods->query_user_list(domain, state->mem_ctx, - &num_entries, &info); - - if (!NT_STATUS_IS_OK(status)) - return WINBINDD_ERROR; - - if (num_entries == 0) - return WINBINDD_OK; - - /* Allocate some memory for extra data. Note that we limit - account names to sizeof(fstring) = 256 characters. - +1 for the ',' between group names */ - extra_data = talloc_array(state->mem_ctx, char, - (sizeof(fstring) + 1) * num_entries); - - if (!extra_data) { - DEBUG(0,("failed to enlarge buffer!\n")); - return WINBINDD_ERROR; - } - - /* Pack user list into extra data fields */ - for (i = 0; i < num_entries; i++) { - fstring acct_name, name; - - if (info[i].acct_name == NULL) - fstrcpy(acct_name, ""); - else - fstrcpy(acct_name, info[i].acct_name); - - fill_domain_username(name, domain->name, acct_name, True); - /* Append to extra data */ - memcpy(&extra_data[extra_data_len], name, strlen(name)); - extra_data_len += strlen(name); - extra_data[extra_data_len++] = ','; - } - - /* Assign extra_data fields in response structure */ - if (extra_data) { - /* remove trailing ',' */ - extra_data[extra_data_len - 1] = '\0'; - state->response->extra_data.data = extra_data; - state->response->length += extra_data_len; - } - - return WINBINDD_OK; -} - -enum winbindd_result winbindd_dual_list_groups(struct winbindd_domain *domain, - struct winbindd_cli_state *state) -{ - struct getent_state groups; - char *extra_data; - uint32_t extra_data_len = 0, i; - - ZERO_STRUCT(groups); - - /* Must copy domain into response first for debugging in parent */ - fstrcpy(state->response->data.name.dom_name, domain->name); - fstrcpy(groups.domain_name, domain->name); - - /* Get list of sam groups */ - if (!get_sam_group_entries(&groups)) { - /* this domain is empty or in an error state */ - return WINBINDD_ERROR; - } - - /* Allocate some memory for extra data. Note that we limit - account names to sizeof(fstring) = 256 characters. - +1 for the ',' between group names */ - extra_data = talloc_array( - state->mem_ctx, char, - (sizeof(fstring) + 1) * groups.num_sam_entries); - - if (!extra_data) { - DEBUG(0,("failed to enlarge buffer!\n")); - SAFE_FREE(groups.sam_entries); - return WINBINDD_ERROR; - } - - /* Pack group list into extra data fields */ - for (i = 0; i < groups.num_sam_entries; i++) { - char *group_name = ((struct acct_info *) - groups.sam_entries)[i].acct_name; - fstring name; - - fill_domain_username(name, domain->name, group_name, True); - /* Append to extra data */ - memcpy(&extra_data[extra_data_len], name, strlen(name)); - extra_data_len += strlen(name); - extra_data[extra_data_len++] = ','; - } - - SAFE_FREE(groups.sam_entries); - - /* Assign extra_data fields in response structure */ - if (extra_data) { - /* remove trailing ',' */ - extra_data[extra_data_len - 1] = '\0'; - state->response->extra_data.data = extra_data; - state->response->length += extra_data_len; - } - - return WINBINDD_OK; -} - bool print_sidlist(TALLOC_CTX *mem_ctx, const DOM_SID *sids, size_t num_sids, char **result, ssize_t *len) { @@ -671,29 +514,6 @@ bool parse_sidlist(TALLOC_CTX *mem_ctx, const char *sidstr, return True; } -static bool parse_ridlist(TALLOC_CTX *mem_ctx, char *ridstr, - uint32 **rids, size_t *num_rids) -{ - char *p; - - p = ridstr; - if (p == NULL) - return False; - - while (p[0] != '\0') { - uint32 rid; - char *q; - rid = strtoul(p, &q, 10); - if (*q != '\n') { - DEBUG(0, ("Got invalid ridstr: %s\n", p)); - return False; - } - p = q+1; - ADD_TO_ARRAY(mem_ctx, uint32, rid, rids, num_rids); - } - return True; -} - static void getsidaliases_recv(TALLOC_CTX *mem_ctx, bool success, struct winbindd_response *response, void *c, void *private_data) diff --git a/source3/winbindd/winbindd_domain.c b/source3/winbindd/winbindd_domain.c index 14376c62bf..107c83ac80 100644 --- a/source3/winbindd/winbindd_domain.c +++ b/source3/winbindd/winbindd_domain.c @@ -49,14 +49,6 @@ static const struct winbindd_child_dispatch_table domain_dispatch_table[] = { .name = "LOOKUPNAME", .struct_cmd = WINBINDD_LOOKUPNAME, .struct_fn = winbindd_dual_lookupname, - },{ - .name = "LIST_USERS", - .struct_cmd = WINBINDD_LIST_USERS, - .struct_fn = winbindd_dual_list_users, - },{ - .name = "LIST_GROUPS", - .struct_cmd = WINBINDD_LIST_GROUPS, - .struct_fn = winbindd_dual_list_groups, },{ .name = "LIST_TRUSTDOM", .struct_cmd = WINBINDD_LIST_TRUSTDOM, diff --git a/source3/winbindd/winbindd_group.c b/source3/winbindd/winbindd_group.c index 8a76071265..eab5c26df4 100644 --- a/source3/winbindd/winbindd_group.c +++ b/source3/winbindd/winbindd_group.c @@ -201,12 +201,6 @@ bool get_sam_group_entries(struct getent_state *ent) return result; } -/* List domain groups without mapping to unix ids */ -void winbindd_list_groups(struct winbindd_cli_state *state) -{ - winbindd_list_ent(state, LIST_GROUPS); -} - /* Get user supplementary groups. This is much quicker than trying to invert the groups database. We merge the groups from the gids and other_sids info3 fields as trusted domain, universal group diff --git a/source3/winbindd/winbindd_list_groups.c b/source3/winbindd/winbindd_list_groups.c new file mode 100644 index 0000000000..3795045787 --- /dev/null +++ b/source3/winbindd/winbindd_list_groups.c @@ -0,0 +1,204 @@ +/* + Unix SMB/CIFS implementation. + async implementation of WINBINDD_LIST_GROUPS + Copyright (C) Volker Lendecke 2009 + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + +#include "includes.h" +#include "winbindd.h" +#include "librpc/gen_ndr/cli_wbint.h" + +struct winbindd_list_groups_domstate { + struct tevent_req *subreq; + struct winbindd_domain *domain; + struct wbint_Principals groups; +}; + +struct winbindd_list_groups_state { + int num_received; + /* All domains */ + int num_domains; + struct winbindd_list_groups_domstate *domains; +}; + +static void winbindd_list_groups_done(struct tevent_req *subreq); + +struct tevent_req *winbindd_list_groups_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct winbindd_cli_state *cli, + struct winbindd_request *request) +{ + struct tevent_req *req; + struct winbindd_list_groups_state *state; + struct winbindd_domain *domain; + int i; + + req = tevent_req_create(mem_ctx, &state, + struct winbindd_list_groups_state); + if (req == NULL) { + return NULL; + } + + /* Ensure null termination */ + request->domain_name[sizeof(request->domain_name)-1]='\0'; + + DEBUG(3, ("list_groups %s\n", request->domain_name)); + + if (request->domain_name[0] != '\0') { + state->num_domains = 1; + } else { + state->num_domains = 0; + for (domain = domain_list(); domain; domain = domain->next) { + state->num_domains += 1; + } + } + + state->domains = talloc_array(state, + struct winbindd_list_groups_domstate, + state->num_domains); + if (tevent_req_nomem(state->domains, req)) { + return tevent_req_post(req, ev); + } + + if (request->domain_name[0] != '\0') { + state->domains[0].domain = find_domain_from_name_noinit( + request->domain_name); + if (state->domains[0].domain == NULL) { + tevent_req_nterror(req, NT_STATUS_NO_SUCH_DOMAIN); + return tevent_req_post(req, ev); + } + } else { + i = 0; + for (domain = domain_list(); domain; domain = domain->next) { + state->domains[i++].domain = domain; + } + } + + for (i=0; inum_domains; i++) { + struct winbindd_list_groups_domstate *d = &state->domains[i]; + + d->subreq = rpccli_wbint_QueryGroupList_send( + state->domains, ev, d->domain->child.rpccli, + &d->groups); + if (tevent_req_nomem(d->subreq, req)) { + TALLOC_FREE(state->domains); + return tevent_req_post(req, ev); + } + tevent_req_set_callback(d->subreq, winbindd_list_groups_done, + req); + } + state->num_received = 0; + return req; +} + +static void winbindd_list_groups_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct winbindd_list_groups_state *state = tevent_req_data( + req, struct winbindd_list_groups_state); + NTSTATUS status, result; + int i; + + status = rpccli_wbint_QueryGroupList_recv(subreq, state->domains, + &result); + + for (i=0; inum_domains; i++) { + if (subreq == state->domains[i].subreq) { + break; + } + } + if (i < state->num_domains) { + struct winbindd_list_groups_domstate *d = &state->domains[i]; + + DEBUG(10, ("Domain %s returned %d users\n", d->domain->name, + d->groups.num_principals)); + + d->subreq = NULL; + + if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(result)) { + DEBUG(10, ("list_groups for domain %s failed\n", + d->domain->name)); + d->groups.num_principals = 0; + } + } + + TALLOC_FREE(subreq); + + state->num_received += 1; + + if (state->num_received >= state->num_domains) { + tevent_req_done(req); + } +} + +NTSTATUS winbindd_list_groups_recv(struct tevent_req *req, + struct winbindd_response *response) +{ + struct winbindd_list_groups_state *state = tevent_req_data( + req, struct winbindd_list_groups_state); + NTSTATUS status; + char *result; + int i; + uint32_t j; + size_t len; + + if (tevent_req_is_nterror(req, &status)) { + return status; + } + + len = 0; + for (i=0; inum_domains; i++) { + struct winbindd_list_groups_domstate *d = &state->domains[i]; + + for (j=0; jgroups.num_principals; j++) { + fstring name; + fill_domain_username(name, d->domain->name, + d->groups.principals[j].name, + True); + len += strlen(name)+1; + } + } + + result = talloc_array(response, char, len+1); + if (result == 0) { + return NT_STATUS_NO_MEMORY; + } + + len = 0; + for (i=0; inum_domains; i++) { + struct winbindd_list_groups_domstate *d = &state->domains[i]; + + for (j=0; jgroups.num_principals; j++) { + fstring name; + size_t this_len; + fill_domain_username(name, d->domain->name, + d->groups.principals[j].name, + True); + this_len = strlen(name); + memcpy(result+len, name, this_len); + len += this_len; + result[len] = ','; + len += 1; + } + } + result[len-1] = '\0'; + + response->extra_data.data = result; + response->length += len; + + return NT_STATUS_OK; +} diff --git a/source3/winbindd/winbindd_misc.c b/source3/winbindd/winbindd_misc.c index beaf450743..b967390966 100644 --- a/source3/winbindd/winbindd_misc.c +++ b/source3/winbindd/winbindd_misc.c @@ -94,128 +94,6 @@ enum winbindd_result winbindd_dual_check_machine_acct(struct winbindd_domain *do return NT_STATUS_IS_OK(result) ? WINBINDD_OK : WINBINDD_ERROR; } -/* Helpers for listing user and group names */ - -const char *ent_type_strings[] = {"users", - "groups"}; - -static const char *get_ent_type_string(enum ent_type type) -{ - return ent_type_strings[type]; -} - -struct listent_state { - TALLOC_CTX *mem_ctx; - struct winbindd_cli_state *cli_state; - enum ent_type type; - int domain_count; - char *extra_data; - uint32_t extra_data_len; -}; - -static void listent_recv(void *private_data, bool success, fstring dom_name, - char *extra_data); - -/* List domain users/groups without mapping to unix ids */ -void winbindd_list_ent(struct winbindd_cli_state *state, enum ent_type type) -{ - struct winbindd_domain *domain; - const char *which_domain; - struct listent_state *ent_state; - - DEBUG(3, ("[%5lu]: list %s\n", (unsigned long)state->pid, - get_ent_type_string(type))); - - /* Ensure null termination */ - state->request->domain_name[sizeof(state->request->domain_name)-1]='\0'; - which_domain = state->request->domain_name; - - /* Initialize listent_state */ - ent_state = TALLOC_P(state->mem_ctx, struct listent_state); - if (ent_state == NULL) { - DEBUG(0, ("talloc failed\n")); - request_error(state); - return; - } - - ent_state->mem_ctx = state->mem_ctx; - ent_state->cli_state = state; - ent_state->type = type; - ent_state->domain_count = 0; - ent_state->extra_data = NULL; - ent_state->extra_data_len = 0; - - /* Must count the full list of expected domains before we request data - * from any of them. Otherwise it's possible for a connection to the - * first domain to fail, call listent_recv(), and return to the - * client without checking any other domains. */ - for (domain = domain_list(); domain; domain = domain->next) { - /* if we have a domain name restricting the request and this - one in the list doesn't match, then just bypass the remainder - of the loop */ - if ( *which_domain && !strequal(which_domain, domain->name) ) - continue; - - ent_state->domain_count++; - } - - /* Make sure we're enumerating at least one domain */ - if (!ent_state->domain_count) { - request_ok(state); - return; - } - - /* Enumerate list of trusted domains and request user/group list from - * each */ - for (domain = domain_list(); domain; domain = domain->next) { - if ( *which_domain && !strequal(which_domain, domain->name) ) - continue; - - winbindd_listent_async(state->mem_ctx, domain, - listent_recv, ent_state, type); - } -} - -static void listent_recv(void *private_data, bool success, fstring dom_name, - char *extra_data) -{ - /* extra_data comes to us as a '\0' terminated string of comma - separated users or groups */ - struct listent_state *state = talloc_get_type_abort( - private_data, struct listent_state); - - /* Append users/groups from one domain onto the whole list */ - if (extra_data) { - DEBUG(5, ("listent_recv: %s returned %s.\n", - dom_name, get_ent_type_string(state->type))); - if (!state->extra_data) - state->extra_data = talloc_asprintf(state->mem_ctx, - "%s", extra_data); - else - state->extra_data = talloc_asprintf_append( - state->extra_data, - ",%s", extra_data); - /* Add one for the '\0' and each additional ',' */ - state->extra_data_len += strlen(extra_data) + 1; - } - else { - DEBUG(5, ("listent_recv: %s returned no %s.\n", - dom_name, get_ent_type_string(state->type))); - } - - if (--state->domain_count) - /* Still waiting for some child domains to return */ - return; - - /* Return list of all users/groups to the client */ - if (state->extra_data) { - state->cli_state->response->extra_data.data = state->extra_data; - state->cli_state->response->length += state->extra_data_len; - } - - request_ok(state->cli_state); -} - /* Constants and helper functions for determining domain trust types */ enum trust_type { diff --git a/source3/winbindd/winbindd_proto.h b/source3/winbindd/winbindd_proto.h index 9894602523..b59d11b78f 100644 --- a/source3/winbindd/winbindd_proto.h +++ b/source3/winbindd/winbindd_proto.h @@ -950,4 +950,12 @@ struct tevent_req *winbindd_list_users_send(TALLOC_CTX *mem_ctx, NTSTATUS winbindd_list_users_recv(struct tevent_req *req, struct winbindd_response *response); +struct tevent_req *winbindd_list_groups_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct winbindd_cli_state *cli, + struct winbindd_request *request); +NTSTATUS winbindd_list_groups_recv(struct tevent_req *req, + struct winbindd_response *response); + + #endif /* _WINBINDD_PROTO_H_ */ -- cgit