s3-registry: Check return code of push_reg_sz().
[kai/samba.git] / source3 / winbindd / winbindd_show_sequence.c
1 /*
2    Unix SMB/CIFS implementation.
3    async implementation of WINBINDD_SHOW_SEQUENCE
4    Copyright (C) Volker Lendecke 2009
5
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.
10
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.
15
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/>.
18 */
19
20 #include "includes.h"
21 #include "winbindd.h"
22
23 struct winbindd_show_sequence_state {
24         bool one_domain;
25         /* One domain */
26         uint32_t seqnum;
27
28         /* All domains */
29         int num_domains;
30         NTSTATUS *stati;
31         struct winbindd_domain **domains;
32         uint32_t *seqnums;
33 };
34
35 static void winbindd_show_sequence_done_one(struct tevent_req *subreq);
36 static void winbindd_show_sequence_done_all(struct tevent_req *subreq);
37
38 struct tevent_req *winbindd_show_sequence_send(TALLOC_CTX *mem_ctx,
39                                                struct tevent_context *ev,
40                                                struct winbindd_cli_state *cli,
41                                                struct winbindd_request *request)
42 {
43         struct tevent_req *req, *subreq;
44         struct winbindd_show_sequence_state *state;
45
46         req = tevent_req_create(mem_ctx, &state,
47                                 struct winbindd_show_sequence_state);
48         if (req == NULL) {
49                 return NULL;
50         }
51         state->one_domain = false;
52         state->domains = NULL;
53         state->stati = NULL;
54         state->seqnums = NULL;
55
56         /* Ensure null termination */
57         request->domain_name[sizeof(request->domain_name)-1]='\0';
58
59         DEBUG(3, ("show_sequence %s\n", request->domain_name));
60
61         if (request->domain_name[0] != '\0') {
62                 struct winbindd_domain *domain;
63
64                 state->one_domain = true;
65
66                 domain = find_domain_from_name_noinit(
67                         request->domain_name);
68                 if (domain == NULL) {
69                         tevent_req_nterror(req, NT_STATUS_NO_SUCH_DOMAIN);
70                         return tevent_req_post(req, ev);
71                 }
72
73                 subreq = wb_seqnum_send(state, ev, domain);
74                 if (tevent_req_nomem(subreq, req)) {
75                         return tevent_req_post(req, ev);
76                 }
77                 tevent_req_set_callback(
78                         subreq, winbindd_show_sequence_done_one, req);
79                 return req;
80         }
81
82         subreq = wb_seqnums_send(state, ev);
83         if (tevent_req_nomem(subreq, req)) {
84                 return tevent_req_post(req, ev);
85         }
86         tevent_req_set_callback(subreq, winbindd_show_sequence_done_all, req);
87         return req;
88 }
89
90 static void winbindd_show_sequence_done_one(struct tevent_req *subreq)
91 {
92         struct tevent_req *req = tevent_req_callback_data(
93                 subreq, struct tevent_req);
94         struct winbindd_show_sequence_state *state = tevent_req_data(
95                 req, struct winbindd_show_sequence_state);
96         NTSTATUS status;
97
98         status = wb_seqnum_recv(subreq, &state->seqnum);
99         TALLOC_FREE(subreq);
100         if (tevent_req_nterror(req, status)) {
101                 return;
102         }
103         tevent_req_done(req);
104 }
105
106 static void winbindd_show_sequence_done_all(struct tevent_req *subreq)
107 {
108         struct tevent_req *req = tevent_req_callback_data(
109                 subreq, struct tevent_req);
110         struct winbindd_show_sequence_state *state = tevent_req_data(
111                 req, struct winbindd_show_sequence_state);
112         NTSTATUS status;
113
114         status = wb_seqnums_recv(subreq, state, &state->num_domains,
115                                  &state->domains, &state->stati,
116                                  &state->seqnums);
117         TALLOC_FREE(subreq);
118         if (tevent_req_nterror(req, status)) {
119                 return;
120         }
121         tevent_req_done(req);
122 }
123
124 NTSTATUS winbindd_show_sequence_recv(struct tevent_req *req,
125                                      struct winbindd_response *response)
126 {
127         struct winbindd_show_sequence_state *state = tevent_req_data(
128                 req, struct winbindd_show_sequence_state);
129         NTSTATUS status;
130         char *extra_data;
131         int i;
132
133         if (tevent_req_is_nterror(req, &status)) {
134                 return status;
135         }
136
137         if (state->one_domain) {
138                 response->data.sequence_number = state->seqnum;
139                 return NT_STATUS_OK;
140         }
141
142         extra_data = talloc_strdup(response, "");
143         if (extra_data == NULL) {
144                 return NT_STATUS_NO_MEMORY;
145         }
146
147         for (i=0; i<state->num_domains; i++) {
148                 if (!NT_STATUS_IS_OK(state->stati[i])
149                     || (state->seqnums[i] == DOM_SEQUENCE_NONE)) {
150                         extra_data = talloc_asprintf_append_buffer(
151                                 extra_data, "%s : DISCONNECTED\n",
152                                 state->domains[i]->name);
153                 } else {
154                         extra_data = talloc_asprintf_append_buffer(
155                                 extra_data, "%s : %d\n",
156                                 state->domains[i]->name,
157                                 (int)state->seqnums[i]);
158                 }
159                 if (extra_data == NULL) {
160                         return NT_STATUS_NO_MEMORY;
161                 }
162         }
163
164         response->extra_data.data = extra_data;
165         response->length += talloc_get_size(extra_data);
166         return NT_STATUS_OK;
167 }