From e54503e82fdbb337b4de9c5aaef252edb2e11e39 Mon Sep 17 00:00:00 2001 From: Volker Lendecke Date: Sat, 15 Aug 2009 11:21:16 +0200 Subject: [PATCH] s3:winbind: Add async wb_seqnums This is something that would have been very difficult with the old style of async requests: Send the request to all children simultaneously. --- source3/Makefile.in | 1 + source3/winbindd/wb_seqnums.c | 133 ++++++++++++++++++++++++++++++ source3/winbindd/winbindd_proto.h | 5 ++ 3 files changed, 139 insertions(+) create mode 100644 source3/winbindd/wb_seqnums.c diff --git a/source3/Makefile.in b/source3/Makefile.in index 14bc86dbf78a..61fde565c282 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -1165,6 +1165,7 @@ WINBINDD_OBJ1 = \ winbindd/wb_getpwsid.o \ winbindd/wb_gettoken.o \ winbindd/wb_seqnum.o \ + winbindd/wb_seqnums.o \ winbindd/winbindd_lookupsid.o \ winbindd/winbindd_lookupname.o \ winbindd/winbindd_sid_to_uid.o \ diff --git a/source3/winbindd/wb_seqnums.c b/source3/winbindd/wb_seqnums.c new file mode 100644 index 000000000000..f4044338a19b --- /dev/null +++ b/source3/winbindd/wb_seqnums.c @@ -0,0 +1,133 @@ +/* + Unix SMB/CIFS implementation. + async seqnums + 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 wb_seqnums_state { + int num_domains; + int num_received; + + struct tevent_req **subreqs; + struct winbindd_domain **domains; + NTSTATUS *stati; + uint32_t *seqnums; +}; + +static void wb_seqnums_done(struct tevent_req *subreq); + +struct tevent_req *wb_seqnums_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev) +{ + struct tevent_req *req; + struct wb_seqnums_state *state; + struct winbindd_domain *domain; + int i; + + req = tevent_req_create(mem_ctx, &state, struct wb_seqnums_state); + if (req == NULL) { + return NULL; + } + state->num_received = 0; + state->num_domains = 0; + + for (domain = domain_list(); domain != NULL; domain = domain->next) { + state->num_domains += 1; + } + + state->subreqs = talloc_array(state, struct tevent_req *, + state->num_domains); + state->domains = talloc_array(state, struct winbindd_domain *, + state->num_domains); + state->stati = talloc_array(state, NTSTATUS, state->num_domains); + state->seqnums = talloc_array(state, uint32_t, state->num_domains); + + if ((state->subreqs == NULL) || (state->domains == NULL) || + (state->stati == NULL) || (state->seqnums == NULL)) { + tevent_req_nterror(req, NT_STATUS_NO_MEMORY); + return tevent_req_post(req, ev); + } + + i = 0; + + for (domain = domain_list(); domain != NULL; domain = domain->next) { + state->domains[i] = domain; + state->subreqs[i] = wb_seqnum_send(state->subreqs, ev, domain); + if (tevent_req_nomem(state->subreqs[i], req)) { + /* Don't even start all the other requests */ + TALLOC_FREE(state->subreqs); + return tevent_req_post(req, ev); + } + tevent_req_set_callback(state->subreqs[i], wb_seqnums_done, + req); + i += 1; + } + return req; +} + +static void wb_seqnums_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct wb_seqnums_state *state = tevent_req_data( + req, struct wb_seqnums_state); + NTSTATUS status; + uint32_t seqnum; + int i; + + status = wb_seqnum_recv(subreq, &seqnum); + + for (i=0; inum_domains; i++) { + if (subreq == state->subreqs[i]) { + state->subreqs[i] = NULL; + state->stati[i] = status; + if (NT_STATUS_IS_OK(status)) { + state->seqnums[i] = seqnum; + state->domains[i]->sequence_number = seqnum; + } + break; + } + } + TALLOC_FREE(subreq); + + state->num_received += 1; + + if (state->num_received >= state->num_domains) { + tevent_req_done(req); + } +} + +NTSTATUS wb_seqnums_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx, + int *num_domains, struct winbindd_domain ***domains, + NTSTATUS **stati, uint32_t **seqnums) +{ + struct wb_seqnums_state *state = tevent_req_data( + req, struct wb_seqnums_state); + NTSTATUS status; + + if (tevent_req_is_nterror(req, &status)) { + return status; + } + *num_domains = state->num_domains; + *domains = talloc_move(mem_ctx, &state->domains); + *stati = talloc_move(mem_ctx, &state->stati); + *seqnums = talloc_move(mem_ctx, &state->seqnums); + return NT_STATUS_OK; +} diff --git a/source3/winbindd/winbindd_proto.h b/source3/winbindd/winbindd_proto.h index 1638b0be6297..a144c99c80f9 100644 --- a/source3/winbindd/winbindd_proto.h +++ b/source3/winbindd/winbindd_proto.h @@ -772,5 +772,10 @@ struct tevent_req *wb_seqnum_send(TALLOC_CTX *mem_ctx, struct winbindd_domain *domain); NTSTATUS wb_seqnum_recv(struct tevent_req *req, uint32_t *seqnum); +struct tevent_req *wb_seqnums_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev); +NTSTATUS wb_seqnums_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx, + int *num_domains, struct winbindd_domain ***domains, + NTSTATUS **stati, uint32_t **seqnums); #endif /* _WINBINDD_PROTO_H_ */ -- 2.34.1