smb_traffic_analyzer.c: optimize marshalling function and document
[Samba/ekacnet.git] / source4 / winbind / wb_sid2domain.c
blobbdbfc3eceb1423ad2fce4c0dde6aba9f0700c3c2
1 /*
2 Unix SMB/CIFS implementation.
4 Find and init a domain struct for a SID
6 Copyright (C) Volker Lendecke 2005
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.
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.
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/>.
22 #include "includes.h"
23 #include "libcli/composite/composite.h"
24 #include "winbind/wb_server.h"
25 #include "smbd/service_task.h"
26 #include "libcli/security/security.h"
27 #include "../lib/util/dlinklist.h"
28 #include "param/param.h"
30 static struct wbsrv_domain *find_domain_from_sid(struct wbsrv_service *service,
31 const struct dom_sid *sid)
33 struct wbsrv_domain *domain;
35 for (domain = service->domains; domain!=NULL; domain = domain->next) {
36 if (dom_sid_equal(domain->info->sid, sid)) {
37 break;
39 if (dom_sid_in_domain(domain->info->sid, sid)) {
40 break;
43 return domain;
46 struct sid2domain_state {
47 struct composite_context *ctx;
48 struct wbsrv_service *service;
49 struct dom_sid *sid;
51 struct wbsrv_domain *domain;
54 static void sid2domain_recv_dom_info(struct composite_context *ctx);
55 static void sid2domain_recv_name(struct composite_context *ctx);
56 static void sid2domain_recv_trusted_dom_info(struct composite_context *ctx);
57 static void sid2domain_recv_init(struct composite_context *ctx);
59 struct composite_context *wb_sid2domain_send(TALLOC_CTX *mem_ctx,
60 struct wbsrv_service *service,
61 const struct dom_sid *sid)
63 struct composite_context *result, *ctx;
64 struct sid2domain_state *state;
66 result = composite_create(mem_ctx, service->task->event_ctx);
67 if (result == NULL) goto failed;
69 state = talloc(result, struct sid2domain_state);
70 if (state == NULL) goto failed;
71 state->ctx = result;
72 result->private_data = state;
74 state->service = service;
75 state->sid = dom_sid_dup(state, sid);
76 if (state->sid == NULL) goto failed;
78 state->domain = find_domain_from_sid(service, sid);
79 if (state->domain != NULL) {
80 result->status = NT_STATUS_OK;
81 composite_done(result);
82 return result;
85 if (dom_sid_equal(service->primary_sid, sid) ||
86 dom_sid_in_domain(service->primary_sid, sid)) {
87 ctx = wb_get_dom_info_send(state, service, lp_workgroup(service->task->lp_ctx),
88 service->primary_sid);
89 if (ctx == NULL) goto failed;
90 ctx->async.fn = sid2domain_recv_dom_info;
91 ctx->async.private_data = state;
92 return result;
95 ctx = wb_cmd_lookupsid_send(state, service, state->sid);
96 if (ctx == NULL) goto failed;
97 composite_continue(result, ctx, sid2domain_recv_name, state);
99 return result;
101 failed:
102 talloc_free(result);
103 return NULL;
107 static void sid2domain_recv_dom_info(struct composite_context *ctx)
109 struct sid2domain_state *state =
110 talloc_get_type(ctx->async.private_data,
111 struct sid2domain_state);
112 struct wb_dom_info *info;
114 state->ctx->status = wb_get_dom_info_recv(ctx, state, &info);
115 if (!composite_is_ok(state->ctx)) return;
117 ctx = wb_init_domain_send(state, state->service, info);
119 composite_continue(state->ctx, ctx, sid2domain_recv_init, state);
122 static void sid2domain_recv_name(struct composite_context *ctx)
124 struct sid2domain_state *state =
125 talloc_get_type(ctx->async.private_data,
126 struct sid2domain_state);
127 struct wb_sid_object *name;
129 state->ctx->status = wb_cmd_lookupsid_recv(ctx, state, &name);
130 if (!composite_is_ok(state->ctx)) return;
132 if (name->type == SID_NAME_UNKNOWN) {
133 composite_error(state->ctx, NT_STATUS_NO_SUCH_DOMAIN);
134 return;
137 if (name->type != SID_NAME_DOMAIN) {
138 state->sid->num_auths -= 1;
141 ctx = wb_trusted_dom_info_send(state, state->service, name->domain,
142 state->sid);
144 composite_continue(state->ctx, ctx, sid2domain_recv_trusted_dom_info,
145 state);
148 static void sid2domain_recv_trusted_dom_info(struct composite_context *ctx)
150 struct sid2domain_state *state =
151 talloc_get_type(ctx->async.private_data,
152 struct sid2domain_state);
153 struct wb_dom_info *info;
155 state->ctx->status = wb_trusted_dom_info_recv(ctx, state, &info);
156 if (!composite_is_ok(state->ctx)) return;
158 ctx = wb_init_domain_send(state, state->service, info);
160 composite_continue(state->ctx, ctx, sid2domain_recv_init, state);
163 static void sid2domain_recv_init(struct composite_context *ctx)
165 struct sid2domain_state *state =
166 talloc_get_type(ctx->async.private_data,
167 struct sid2domain_state);
168 struct wbsrv_domain *existing;
170 state->ctx->status = wb_init_domain_recv(ctx, state,
171 &state->domain);
172 if (!composite_is_ok(state->ctx)) {
173 DEBUG(10, ("Could not init domain\n"));
174 return;
177 existing = find_domain_from_sid(state->service, state->sid);
178 if (existing != NULL) {
179 DEBUG(5, ("Initialized domain twice, dropping second one\n"));
180 talloc_free(state->domain);
181 state->domain = existing;
184 talloc_steal(state->service, state->domain);
185 DLIST_ADD(state->service->domains, state->domain);
187 composite_done(state->ctx);
190 NTSTATUS wb_sid2domain_recv(struct composite_context *ctx,
191 struct wbsrv_domain **result)
193 NTSTATUS status = composite_wait(ctx);
194 if (NT_STATUS_IS_OK(status)) {
195 struct sid2domain_state *state =
196 talloc_get_type(ctx->private_data,
197 struct sid2domain_state);
198 *result = state->domain;
200 talloc_free(ctx);
201 return status;
204 NTSTATUS wb_sid2domain(TALLOC_CTX *mem_ctx, struct wbsrv_service *service,
205 const struct dom_sid *sid,
206 struct wbsrv_domain **result)
208 struct composite_context *c = wb_sid2domain_send(mem_ctx, service,
209 sid);
210 return wb_sid2domain_recv(c, result);