messages_ctdb: Handle async msgs for nested event contexts
[Samba.git] / source3 / winbindd / wb_sids2xids.c
blobb8ad300ba3c80e1e82f5eccc1f87421d7ca963a7
1 /*
2 Unix SMB/CIFS implementation.
3 async sids2xids
4 Copyright (C) Volker Lendecke 2011
5 Copyright (C) Michael Adam 2012
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "includes.h"
22 #include "winbindd.h"
23 #include "../libcli/security/security.h"
24 #include "idmap_cache.h"
25 #include "librpc/gen_ndr/ndr_winbind_c.h"
26 #include "librpc/gen_ndr/ndr_netlogon.h"
27 #include "lsa.h"
29 struct wb_sids2xids_state {
30 struct tevent_context *ev;
32 struct dom_sid *sids;
33 uint32_t num_sids;
35 struct id_map *cached;
37 struct dom_sid *non_cached;
38 uint32_t num_non_cached;
41 * Domain array to use for the idmap call. The output from
42 * lookupsids cannot be used directly since for migrated
43 * objects the returned domain SID can be different than the
44 * original one. The new domain SID cannot be combined with
45 * the RID from the previous domain.
47 * The proper way would be asking for the correct RID in the
48 * new domain, but this approach avoids id mappings for
49 * invalid SIDs.
51 struct lsa_RefDomainList idmap_doms;
53 uint32_t dom_index;
54 struct wbint_TransIDArray *dom_ids;
55 struct lsa_RefDomainList idmap_dom;
56 bool tried_dclookup;
58 struct wbint_TransIDArray ids;
62 static bool wb_sids2xids_in_cache(struct dom_sid *sid, struct id_map *map);
63 static void wb_sids2xids_lookupsids_done(struct tevent_req *subreq);
64 static void wb_sids2xids_done(struct tevent_req *subreq);
65 static void wb_sids2xids_gotdc(struct tevent_req *subreq);
67 struct tevent_req *wb_sids2xids_send(TALLOC_CTX *mem_ctx,
68 struct tevent_context *ev,
69 const struct dom_sid *sids,
70 const uint32_t num_sids)
72 struct tevent_req *req, *subreq;
73 struct wb_sids2xids_state *state;
74 uint32_t i;
76 req = tevent_req_create(mem_ctx, &state,
77 struct wb_sids2xids_state);
78 if (req == NULL) {
79 return NULL;
82 state->ev = ev;
84 state->num_sids = num_sids;
86 state->sids = talloc_zero_array(state, struct dom_sid, num_sids);
87 if (tevent_req_nomem(state->sids, req)) {
88 return tevent_req_post(req, ev);
91 for (i = 0; i < num_sids; i++) {
92 sid_copy(&state->sids[i], &sids[i]);
95 state->cached = talloc_zero_array(state, struct id_map, num_sids);
96 if (tevent_req_nomem(state->cached, req)) {
97 return tevent_req_post(req, ev);
100 state->non_cached = talloc_array(state, struct dom_sid, num_sids);
101 if (tevent_req_nomem(state->non_cached, req)) {
102 return tevent_req_post(req, ev);
106 * Extract those sids that can not be resolved from cache
107 * into a separate list to be handed to id mapping, keeping
108 * the same index.
110 for (i=0; i<state->num_sids; i++) {
112 DEBUG(10, ("SID %d: %s\n", (int)i,
113 sid_string_dbg(&state->sids[i])));
115 if (wb_sids2xids_in_cache(&state->sids[i], &state->cached[i])) {
116 continue;
118 sid_copy(&state->non_cached[state->num_non_cached],
119 &state->sids[i]);
120 state->num_non_cached += 1;
123 if (state->num_non_cached == 0) {
124 tevent_req_done(req);
125 return tevent_req_post(req, ev);
128 subreq = wb_lookupsids_send(state, ev, state->non_cached,
129 state->num_non_cached);
130 if (tevent_req_nomem(subreq, req)) {
131 return tevent_req_post(req, ev);
133 tevent_req_set_callback(subreq, wb_sids2xids_lookupsids_done, req);
134 return req;
137 static bool wb_sids2xids_in_cache(struct dom_sid *sid, struct id_map *map)
139 struct unixid id;
140 bool expired;
142 if (!winbindd_use_idmap_cache()) {
143 return false;
145 if (idmap_cache_find_sid2unixid(sid, &id, &expired)) {
146 if (expired && is_domain_online(find_our_domain())) {
147 return false;
149 map->sid = sid;
150 map->xid = id;
151 map->status = ID_MAPPED;
152 return true;
154 return false;
157 static enum id_type lsa_SidType_to_id_type(const enum lsa_SidType sid_type);
158 static struct wbint_TransIDArray *wb_sids2xids_extract_for_domain_index(
159 TALLOC_CTX *mem_ctx, const struct wbint_TransIDArray *src,
160 uint32_t domain_index);
162 static void wb_sids2xids_lookupsids_done(struct tevent_req *subreq)
164 struct tevent_req *req = tevent_req_callback_data(
165 subreq, struct tevent_req);
166 struct wb_sids2xids_state *state = tevent_req_data(
167 req, struct wb_sids2xids_state);
168 struct lsa_RefDomainList *domains = NULL;
169 struct lsa_TransNameArray *names = NULL;
170 struct winbindd_child *child;
171 NTSTATUS status;
172 int i;
174 status = wb_lookupsids_recv(subreq, state, &domains, &names);
175 TALLOC_FREE(subreq);
176 if (tevent_req_nterror(req, status)) {
177 return;
180 state->ids.num_ids = state->num_non_cached;
181 state->ids.ids = talloc_array(state, struct wbint_TransID,
182 state->num_non_cached);
183 if (tevent_req_nomem(state->ids.ids, req)) {
184 return;
187 for (i=0; i<state->num_non_cached; i++) {
188 const struct dom_sid *sid = &state->non_cached[i];
189 struct dom_sid dom_sid;
190 struct lsa_TranslatedName *n = &names->names[i];
191 struct wbint_TransID *t = &state->ids.ids[i];
192 int domain_index;
193 const char *domain_name = NULL;
195 if (n->sid_index != UINT32_MAX) {
196 const struct lsa_DomainInfo *info;
197 bool match;
199 info = &domains->domains[n->sid_index];
200 match = dom_sid_in_domain(info->sid, sid);
201 if (match) {
202 domain_name = info->name.string;
205 if (domain_name == NULL) {
206 struct winbindd_domain *wb_domain = NULL;
209 * This is needed to handle Samba DCs
210 * which always return sid_index == UINT32_MAX for
211 * unknown sids.
213 wb_domain = find_domain_from_sid_noinit(sid);
214 if (wb_domain != NULL) {
215 domain_name = wb_domain->name;
218 if (domain_name == NULL) {
219 domain_name = "";
222 sid_copy(&dom_sid, sid);
223 sid_split_rid(&dom_sid, &t->rid);
224 t->type = lsa_SidType_to_id_type(n->sid_type);
225 domain_index = init_lsa_ref_domain_list(
226 state, &state->idmap_doms, domain_name, &dom_sid);
227 if (domain_index == -1) {
228 tevent_req_oom(req);
229 return;
231 t->domain_index = domain_index;
233 t->xid.id = UINT32_MAX;
234 t->xid.type = t->type;
237 TALLOC_FREE(names);
238 TALLOC_FREE(domains);
240 child = idmap_child();
242 state->dom_ids = wb_sids2xids_extract_for_domain_index(
243 state, &state->ids, state->dom_index);
244 if (tevent_req_nomem(state->dom_ids, req)) {
245 return;
248 state->idmap_dom = (struct lsa_RefDomainList) {
249 .count = 1,
250 .domains = &state->idmap_doms.domains[state->dom_index],
251 .max_size = 1
254 subreq = dcerpc_wbint_Sids2UnixIDs_send(
255 state, state->ev, child->binding_handle, &state->idmap_dom,
256 state->dom_ids);
257 if (tevent_req_nomem(subreq, req)) {
258 return;
260 tevent_req_set_callback(subreq, wb_sids2xids_done, req);
263 static enum id_type lsa_SidType_to_id_type(const enum lsa_SidType sid_type)
265 enum id_type type;
267 switch(sid_type) {
268 case SID_NAME_COMPUTER:
269 case SID_NAME_USER:
270 type = ID_TYPE_UID;
271 break;
272 case SID_NAME_DOM_GRP:
273 case SID_NAME_ALIAS:
274 case SID_NAME_WKN_GRP:
275 type = ID_TYPE_GID;
276 break;
277 default:
278 type = ID_TYPE_NOT_SPECIFIED;
279 break;
282 return type;
285 static void wb_sids2xids_done(struct tevent_req *subreq)
287 struct tevent_req *req = tevent_req_callback_data(
288 subreq, struct tevent_req);
289 struct wb_sids2xids_state *state = tevent_req_data(
290 req, struct wb_sids2xids_state);
291 NTSTATUS status, result;
292 struct winbindd_child *child;
294 struct wbint_TransIDArray *src, *dst;
295 uint32_t i, src_idx;
297 status = dcerpc_wbint_Sids2UnixIDs_recv(subreq, state, &result);
298 TALLOC_FREE(subreq);
300 if (tevent_req_nterror(req, status)) {
301 return;
304 if (NT_STATUS_EQUAL(result, NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND) &&
305 !state->tried_dclookup) {
307 struct lsa_DomainInfo *d;
309 d = &state->idmap_doms.domains[state->dom_index];
311 subreq = wb_dsgetdcname_send(
312 state, state->ev, d->name.string, NULL, NULL,
313 DS_RETURN_DNS_NAME);
314 if (tevent_req_nomem(subreq, req)) {
315 return;
317 tevent_req_set_callback(subreq, wb_sids2xids_gotdc, req);
318 return;
321 src = state->dom_ids;
322 src_idx = 0;
323 dst = &state->ids;
325 if (any_nt_status_not_ok(status, result, &status)) {
326 DBG_DEBUG("status=%s, result=%s\n", nt_errstr(status),
327 nt_errstr(result));
330 * All we can do here is to report "not mapped"
332 for (i=0; i<src->num_ids; i++) {
333 src->ids[i].xid.type = ID_TYPE_NOT_SPECIFIED;
337 for (i=0; i<dst->num_ids; i++) {
338 if (dst->ids[i].domain_index == state->dom_index) {
339 dst->ids[i].type = src->ids[src_idx].type;
340 dst->ids[i].xid = src->ids[src_idx].xid;
341 src_idx += 1;
345 TALLOC_FREE(state->dom_ids);
347 state->dom_index += 1;
348 state->tried_dclookup = false;
350 if (state->dom_index == state->idmap_doms.count) {
351 tevent_req_done(req);
352 return;
355 child = idmap_child();
357 state->dom_ids = wb_sids2xids_extract_for_domain_index(
358 state, &state->ids, state->dom_index);
359 if (tevent_req_nomem(state->dom_ids, req)) {
360 return;
363 state->idmap_dom = (struct lsa_RefDomainList) {
364 .count = 1,
365 .domains = &state->idmap_doms.domains[state->dom_index],
366 .max_size = 1
369 subreq = dcerpc_wbint_Sids2UnixIDs_send(
370 state, state->ev, child->binding_handle, &state->idmap_dom,
371 state->dom_ids);
372 if (tevent_req_nomem(subreq, req)) {
373 return;
375 tevent_req_set_callback(subreq, wb_sids2xids_done, req);
378 static void wb_sids2xids_gotdc(struct tevent_req *subreq)
380 struct tevent_req *req = tevent_req_callback_data(
381 subreq, struct tevent_req);
382 struct wb_sids2xids_state *state = tevent_req_data(
383 req, struct wb_sids2xids_state);
384 struct winbindd_child *child = idmap_child();
385 struct netr_DsRGetDCNameInfo *dcinfo;
386 NTSTATUS status;
388 status = wb_dsgetdcname_recv(subreq, state, &dcinfo);
389 TALLOC_FREE(subreq);
390 if (tevent_req_nterror(req, status)) {
391 return;
394 state->tried_dclookup = true;
397 struct lsa_DomainInfo *d =
398 &state->idmap_doms.domains[state->dom_index];
399 const char *dom_name = d->name.string;
401 status = wb_dsgetdcname_gencache_set(dom_name, dcinfo);
402 if (tevent_req_nterror(req, status)) {
403 return;
407 subreq = dcerpc_wbint_Sids2UnixIDs_send(
408 state, state->ev, child->binding_handle, &state->idmap_dom,
409 state->dom_ids);
410 if (tevent_req_nomem(subreq, req)) {
411 return;
413 tevent_req_set_callback(subreq, wb_sids2xids_done, req);
416 NTSTATUS wb_sids2xids_recv(struct tevent_req *req,
417 struct unixid xids[], uint32_t num_xids)
419 struct wb_sids2xids_state *state = tevent_req_data(
420 req, struct wb_sids2xids_state);
421 NTSTATUS status;
422 uint32_t i, num_non_cached;
424 if (tevent_req_is_nterror(req, &status)) {
425 DEBUG(5, ("wb_sids_to_xids failed: %s\n", nt_errstr(status)));
426 return status;
429 if (num_xids != state->num_sids) {
430 DEBUG(1, ("%s: Have %u xids, caller wants %u\n", __func__,
431 (unsigned)state->num_sids, num_xids));
432 return NT_STATUS_INTERNAL_ERROR;
435 num_non_cached = 0;
437 for (i=0; i<state->num_sids; i++) {
438 struct unixid xid;
440 xid.id = UINT32_MAX;
442 if (state->cached[i].sid != NULL) {
443 xid = state->cached[i].xid;
444 } else {
445 xid = state->ids.ids[num_non_cached].xid;
447 idmap_cache_set_sid2unixid(
448 &state->non_cached[num_non_cached],
449 &xid);
451 num_non_cached += 1;
454 xids[i] = xid;
457 return NT_STATUS_OK;
460 static struct wbint_TransIDArray *wb_sids2xids_extract_for_domain_index(
461 TALLOC_CTX *mem_ctx, const struct wbint_TransIDArray *src,
462 uint32_t domain_index)
464 struct wbint_TransIDArray *ret;
465 uint32_t i;
467 ret = talloc_zero(mem_ctx, struct wbint_TransIDArray);
468 if (ret == NULL) {
469 return NULL;
471 ret->ids = talloc_array(ret, struct wbint_TransID, src->num_ids);
472 if (ret->ids == NULL) {
473 TALLOC_FREE(ret);
474 return NULL;
477 for (i=0; i<src->num_ids; i++) {
478 if (src->ids[i].domain_index == domain_index) {
479 ret->ids[ret->num_ids] = src->ids[i];
480 ret->ids[ret->num_ids].domain_index = 0;
481 ret->num_ids += 1;
485 return ret;