diff options
Diffstat (limited to 'source3/winbindd/winbindd_getpwent.c')
| -rw-r--r-- | source3/winbindd/winbindd_getpwent.c | 162 |
1 files changed, 162 insertions, 0 deletions
diff --git a/source3/winbindd/winbindd_getpwent.c b/source3/winbindd/winbindd_getpwent.c new file mode 100644 index 0000000..5a78883 --- /dev/null +++ b/source3/winbindd/winbindd_getpwent.c @@ -0,0 +1,162 @@ +/* + Unix SMB/CIFS implementation. + async implementation of WINBINDD_GETPWENT + 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 <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" +#include "winbindd.h" + +struct winbindd_getpwent_state { + struct tevent_context *ev; + struct winbindd_cli_state *cli; + uint32_t max_users; + uint32_t num_users; + struct winbindd_pw *users; +}; + +static void winbindd_getpwent_done(struct tevent_req *subreq); + +struct tevent_req *winbindd_getpwent_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct winbindd_cli_state *cli, + struct winbindd_request *request) +{ + struct tevent_req *req, *subreq; + struct winbindd_getpwent_state *state; + + req = tevent_req_create(mem_ctx, &state, + struct winbindd_getpwent_state); + if (req == NULL) { + return NULL; + } + state->ev = ev; + state->num_users = 0; + state->cli = cli; + + D_NOTICE("[%s (%u)] Winbind external command GETPWENT start.\n" + "The caller (%s) provided room for %d entries.\n", + cli->client_name, + (unsigned int)cli->pid, + cli->client_name, + request->data.num_entries); + + if (cli->pwent_state == NULL) { + tevent_req_nterror(req, NT_STATUS_NO_MORE_ENTRIES); + return tevent_req_post(req, ev); + } + + state->max_users = MIN(500, request->data.num_entries); + if (state->max_users == 0) { + tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER); + return tevent_req_post(req, ev); + } + + state->users = talloc_zero_array(state, struct winbindd_pw, + state->max_users); + if (tevent_req_nomem(state->users, req)) { + return tevent_req_post(req, ev); + } + + subreq = wb_next_pwent_send(state, ev, cli->pwent_state, + &state->users[state->num_users]); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, winbindd_getpwent_done, req); + return req; +} + +static void winbindd_getpwent_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct winbindd_getpwent_state *state = tevent_req_data( + req, struct winbindd_getpwent_state); + NTSTATUS status; + + status = wb_next_pwent_recv(subreq); + TALLOC_FREE(subreq); + if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) { + D_DEBUG("winbindd_getpwent_done: done with %"PRIu32" users\n", + state->num_users); + TALLOC_FREE(state->cli->pwent_state); + tevent_req_done(req); + return; + } + if (tevent_req_nterror(req, status)) { + return; + } + state->num_users += 1; + if (state->num_users >= state->max_users) { + D_DEBUG("winbindd_getpwent_done: Got enough users: %"PRIu32"\n", + state->num_users); + tevent_req_done(req); + return; + } + if (state->cli->pwent_state == NULL) { + D_DEBUG("winbindd_getpwent_done: endpwent called in between\n"); + tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER); + return; + } + subreq = wb_next_pwent_send(state, state->ev, state->cli->pwent_state, + &state->users[state->num_users]); + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, winbindd_getpwent_done, req); +} + +NTSTATUS winbindd_getpwent_recv(struct tevent_req *req, + struct winbindd_response *response) +{ + struct winbindd_getpwent_state *state = tevent_req_data( + req, struct winbindd_getpwent_state); + NTSTATUS status; + uint32_t i; + + if (tevent_req_is_nterror(req, &status)) { + TALLOC_FREE(state->cli->pwent_state); + D_WARNING("getpwent failed: %s\n", nt_errstr(status)); + return status; + } + + D_NOTICE("Winbind external command GETPWENT end.\n" + "Received %"PRIu32" entries.\n" + "(name:passwd:uid:gid:gecos:dir:shell)\n", + state->num_users); + + if (state->num_users == 0) { + return NT_STATUS_NO_MORE_ENTRIES; + } + + for (i = 0; i < state->num_users; i++) { + D_NOTICE("%"PRIu32": %s:%s:%u:%u:%s:%s:%s\n", + i, + state->users[i].pw_name, + state->users[i].pw_passwd, + (unsigned int)state->users[i].pw_uid, + (unsigned int)state->users[i].pw_gid, + state->users[i].pw_gecos, + state->users[i].pw_dir, + state->users[i].pw_shell + ); + } + response->data.num_entries = state->num_users; + response->extra_data.data = talloc_move(response, &state->users); + response->length += state->num_users * sizeof(struct winbindd_pw); + return NT_STATUS_OK; +} |