Line data Source code
1 : /*
2 : Unix SMB/CIFS implementation.
3 :
4 : async seqnums, update the seqnums in winbindd_cache.c
5 :
6 : Copyright (C) Volker Lendecke 2009
7 :
8 : This program is free software; you can redistribute it and/or modify
9 : it under the terms of the GNU General Public License as published by
10 : the Free Software Foundation; either version 3 of the License, or
11 : (at your option) any later version.
12 :
13 : This program is distributed in the hope that it will be useful,
14 : but WITHOUT ANY WARRANTY; without even the implied warranty of
15 : MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 : GNU General Public License for more details.
17 :
18 : You should have received a copy of the GNU General Public License
19 : along with this program. If not, see <http://www.gnu.org/licenses/>.
20 : */
21 :
22 : #include "includes.h"
23 : #include "winbindd.h"
24 : #include "librpc/gen_ndr/ndr_winbind_c.h"
25 :
26 : struct wb_seqnums_state {
27 : int num_domains;
28 : int num_received;
29 :
30 : struct tevent_req **subreqs;
31 : struct winbindd_domain **domains;
32 : NTSTATUS *stati;
33 : uint32_t *seqnums;
34 : };
35 :
36 : static void wb_seqnums_done(struct tevent_req *subreq);
37 :
38 2 : struct tevent_req *wb_seqnums_send(TALLOC_CTX *mem_ctx,
39 : struct tevent_context *ev)
40 : {
41 : struct tevent_req *req;
42 : struct wb_seqnums_state *state;
43 : struct winbindd_domain *domain;
44 : int i;
45 :
46 2 : req = tevent_req_create(mem_ctx, &state, struct wb_seqnums_state);
47 2 : if (req == NULL) {
48 0 : return NULL;
49 : }
50 2 : state->num_received = 0;
51 2 : state->num_domains = 0;
52 :
53 12 : for (domain = domain_list(); domain != NULL; domain = domain->next) {
54 10 : state->num_domains += 1;
55 : }
56 :
57 2 : state->subreqs = talloc_array(state, struct tevent_req *,
58 : state->num_domains);
59 2 : state->domains = talloc_zero_array(state, struct winbindd_domain *,
60 : state->num_domains);
61 2 : state->stati = talloc_array(state, NTSTATUS, state->num_domains);
62 2 : state->seqnums = talloc_array(state, uint32_t, state->num_domains);
63 :
64 3 : if ((state->subreqs == NULL) || (state->domains == NULL) ||
65 3 : (state->stati == NULL) || (state->seqnums == NULL)) {
66 0 : tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
67 0 : return tevent_req_post(req, ev);
68 : }
69 :
70 2 : i = 0;
71 :
72 12 : for (domain = domain_list(); domain != NULL; domain = domain->next) {
73 10 : state->domains[i] = domain;
74 10 : state->subreqs[i] = wb_seqnum_send(state->subreqs, ev, domain);
75 10 : if (tevent_req_nomem(state->subreqs[i], req)) {
76 : /* Don't even start all the other requests */
77 0 : TALLOC_FREE(state->subreqs);
78 0 : return tevent_req_post(req, ev);
79 : }
80 10 : tevent_req_set_callback(state->subreqs[i], wb_seqnums_done,
81 : req);
82 10 : i += 1;
83 : }
84 2 : return req;
85 : }
86 :
87 10 : static void wb_seqnums_done(struct tevent_req *subreq)
88 : {
89 10 : struct tevent_req *req = tevent_req_callback_data(
90 : subreq, struct tevent_req);
91 10 : struct wb_seqnums_state *state = tevent_req_data(
92 : req, struct wb_seqnums_state);
93 : NTSTATUS status;
94 : uint32_t seqnum;
95 : int i;
96 :
97 10 : status = wb_seqnum_recv(subreq, &seqnum);
98 :
99 30 : for (i=0; i<state->num_domains; i++) {
100 30 : if (subreq == state->subreqs[i]) {
101 10 : break;
102 : }
103 : }
104 10 : if (i < state->num_domains) {
105 : /* found one */
106 :
107 10 : state->subreqs[i] = NULL;
108 10 : state->stati[i] = status;
109 10 : if (NT_STATUS_IS_OK(status)) {
110 10 : state->seqnums[i] = seqnum;
111 :
112 : /*
113 : * This first assignment might be removed
114 : * later
115 : */
116 10 : state->domains[i]->sequence_number = seqnum;
117 :
118 20 : if (!wcache_store_seqnum(state->domains[i]->name,
119 10 : state->seqnums[i],
120 : time(NULL))) {
121 0 : DEBUG(1, ("wcache_store_seqnum failed for "
122 : "domain %s\n",
123 : state->domains[i]->name));
124 : }
125 : }
126 : }
127 :
128 10 : TALLOC_FREE(subreq);
129 :
130 10 : state->num_received += 1;
131 :
132 10 : if (state->num_received >= state->num_domains) {
133 2 : tevent_req_done(req);
134 : }
135 10 : }
136 :
137 2 : NTSTATUS wb_seqnums_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
138 : int *num_domains, struct winbindd_domain ***domains,
139 : NTSTATUS **stati, uint32_t **seqnums)
140 : {
141 2 : struct wb_seqnums_state *state = tevent_req_data(
142 : req, struct wb_seqnums_state);
143 : NTSTATUS status;
144 :
145 2 : if (tevent_req_is_nterror(req, &status)) {
146 0 : return status;
147 : }
148 2 : *num_domains = state->num_domains;
149 2 : *domains = talloc_move(mem_ctx, &state->domains);
150 2 : *stati = talloc_move(mem_ctx, &state->stati);
151 2 : *seqnums = talloc_move(mem_ctx, &state->seqnums);
152 2 : return NT_STATUS_OK;
153 : }
|
