s3:winbind: Convert WINBINDD_LIST_GROUPS to the new API
[sfrench/samba-autobuild/.git] / source3 / winbindd / winbindd_async.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Async helpers for blocking functions
5
6    Copyright (C) Volker Lendecke 2005
7    Copyright (C) Gerald Carter 2006
8
9    The helpers always consist of three functions: 
10
11    * A request setup function that takes the necessary parameters together
12      with a continuation function that is to be called upon completion
13
14    * A private continuation function that is internal only. This is to be
15      called by the lower-level functions in do_async(). Its only task is to
16      properly call the continuation function named above.
17
18    * A worker function that is called inside the appropriate child process.
19
20    This program is free software; you can redistribute it and/or modify
21    it under the terms of the GNU General Public License as published by
22    the Free Software Foundation; either version 3 of the License, or
23    (at your option) any later version.
24
25    This program is distributed in the hope that it will be useful,
26    but WITHOUT ANY WARRANTY; without even the implied warranty of
27    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
28    GNU General Public License for more details.
29
30    You should have received a copy of the GNU General Public License
31    along with this program.  If not, see <http://www.gnu.org/licenses/>.
32 */
33
34 #include "includes.h"
35 #include "winbindd.h"
36
37 #undef DBGC_CLASS
38 #define DBGC_CLASS DBGC_WINBIND
39
40 struct do_async_state {
41         TALLOC_CTX *mem_ctx;
42         struct winbindd_request request;
43         struct winbindd_response response;
44         void (*cont)(TALLOC_CTX *mem_ctx,
45                      bool success,
46                      struct winbindd_response *response,
47                      void *c, void *private_data);
48         void *c, *private_data;
49 };
50
51 static void do_async_recv(void *private_data, bool success)
52 {
53         struct do_async_state *state =
54                 talloc_get_type_abort(private_data, struct do_async_state);
55
56         state->cont(state->mem_ctx, success, &state->response,
57                     state->c, state->private_data);
58 }
59
60 void do_async(TALLOC_CTX *mem_ctx, struct winbindd_child *child,
61               const struct winbindd_request *request,
62               void (*cont)(TALLOC_CTX *mem_ctx, bool success,
63                            struct winbindd_response *response,
64                            void *c, void *private_data),
65               void *c, void *private_data)
66 {
67         struct do_async_state *state;
68
69         state = TALLOC_ZERO_P(mem_ctx, struct do_async_state);
70         if (state == NULL) {
71                 DEBUG(0, ("talloc failed\n"));
72                 cont(mem_ctx, False, NULL, c, private_data);
73                 return;
74         }
75
76         state->mem_ctx = mem_ctx;
77         state->request = *request;
78         state->request.length = sizeof(state->request);
79         state->cont = cont;
80         state->c = c;
81         state->private_data = private_data;
82
83         async_request(mem_ctx, child, &state->request,
84                       &state->response, do_async_recv, state);
85 }
86
87 static void do_async_domain(TALLOC_CTX *mem_ctx, struct winbindd_domain *domain,
88                             const struct winbindd_request *request,
89                             void (*cont)(TALLOC_CTX *mem_ctx, bool success,
90                                          struct winbindd_response *response,
91                                          void *c, void *private_data),
92                             void *c, void *private_data)
93 {
94         struct do_async_state *state;
95
96         state = TALLOC_ZERO_P(mem_ctx, struct do_async_state);
97         if (state == NULL) {
98                 DEBUG(0, ("talloc failed\n"));
99                 cont(mem_ctx, False, NULL, c, private_data);
100                 return;
101         }
102
103         state->mem_ctx = mem_ctx;
104         state->request = *request;
105         state->request.length = sizeof(state->request);
106         state->cont = cont;
107         state->c = c;
108         state->private_data = private_data;
109
110         async_domain_request(mem_ctx, domain, &state->request,
111                              &state->response, do_async_recv, state);
112 }
113
114 struct lookupsid_state {
115         DOM_SID sid;    
116         void *caller_private_data;
117 };
118
119
120 static void lookupsid_recv2(TALLOC_CTX *mem_ctx, bool success,
121                            struct winbindd_response *response,
122                            void *c, void *private_data)
123 {
124         void (*cont)(void *priv, bool succ, const char *dom_name,
125                      const char *name, enum lsa_SidType type) =
126                 (void (*)(void *, bool, const char *, const char *,
127                           enum lsa_SidType))c;
128         struct lookupsid_state *s = talloc_get_type_abort(private_data, 
129                                                           struct lookupsid_state);
130
131         if (!success) {
132                 DEBUG(5, ("Could not trigger lookupsid\n"));
133                 cont(s->caller_private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
134                 return;
135         }
136
137         if (response->result != WINBINDD_OK) {
138                 DEBUG(5, ("lookupsid (forest root) returned an error\n"));              
139                 cont(s->caller_private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
140                 return;
141         }
142
143         cont(s->caller_private_data, True, response->data.name.dom_name,
144              response->data.name.name,
145              (enum lsa_SidType)response->data.name.type);
146 }
147
148 static void lookupsid_recv(TALLOC_CTX *mem_ctx, bool success,
149                            struct winbindd_response *response,
150                            void *c, void *private_data)
151 {
152         void (*cont)(void *priv, bool succ, const char *dom_name,
153                      const char *name, enum lsa_SidType type) =
154                 (void (*)(void *, bool, const char *, const char *,
155                           enum lsa_SidType))c;
156         struct lookupsid_state *s = talloc_get_type_abort(private_data, 
157                                                           struct lookupsid_state);
158
159         if (!success) {
160                 DEBUG(5, ("Could not trigger lookupsid\n"));
161                 cont(s->caller_private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
162                 return;
163         }
164
165         if (response->result != WINBINDD_OK) {
166                 /* Try again using the forest root */
167                 struct winbindd_domain *root_domain = find_root_domain();
168                 struct winbindd_request request;
169
170                 if ( !root_domain ) {
171                         DEBUG(5,("lookupsid_recv: unable to determine forest root\n"));
172                         cont(s->caller_private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
173                         return;
174                 }
175
176                 ZERO_STRUCT(request);
177                 request.cmd = WINBINDD_LOOKUPSID;
178                 sid_to_fstring(request.data.sid, &s->sid);
179
180                 do_async_domain(mem_ctx, root_domain, &request, lookupsid_recv2,
181                                 (void *)cont, s);
182
183                 return;
184         }
185
186         cont(s->caller_private_data, True, response->data.name.dom_name,
187              response->data.name.name,
188              (enum lsa_SidType)response->data.name.type);
189 }
190
191 void winbindd_lookupsid_async(TALLOC_CTX *mem_ctx, const DOM_SID *sid,
192                               void (*cont)(void *private_data, bool success,
193                                            const char *dom_name,
194                                            const char *name,
195                                            enum lsa_SidType type),
196                               void *private_data)
197 {
198         struct winbindd_domain *domain;
199         struct winbindd_request request;
200         struct lookupsid_state *s;      
201
202         domain = find_lookup_domain_from_sid(sid);
203         if (domain == NULL) {
204                 DEBUG(5, ("Could not find domain for sid %s\n",
205                           sid_string_dbg(sid)));
206                 cont(private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
207                 return;
208         }
209
210         ZERO_STRUCT(request);
211         request.cmd = WINBINDD_LOOKUPSID;
212         sid_to_fstring(request.data.sid, sid);
213
214         if ( (s = TALLOC_ZERO_P(mem_ctx, struct lookupsid_state)) == NULL ) {
215                 DEBUG(0, ("winbindd_lookupsid_async: talloc failed\n"));
216                 cont(private_data, False, NULL, NULL, SID_NAME_UNKNOWN);
217                 return;
218         }
219
220         sid_copy( &s->sid, sid );       
221         s->caller_private_data = private_data;  
222
223         do_async_domain(mem_ctx, domain, &request, lookupsid_recv,
224                         (void *)cont, s);
225 }
226
227 enum winbindd_result winbindd_dual_lookupsid(struct winbindd_domain *domain,
228                                              struct winbindd_cli_state *state)
229 {
230         enum lsa_SidType type;
231         DOM_SID sid;
232         char *name;
233         char *dom_name;
234
235         /* Ensure null termination */
236         state->request->data.sid[sizeof(state->request->data.sid)-1]='\0';
237
238         DEBUG(3, ("[%5lu]: lookupsid %s\n", (unsigned long)state->pid, 
239                   state->request->data.sid));
240
241         /* Lookup sid from PDC using lsa_lookup_sids() */
242
243         if (!string_to_sid(&sid, state->request->data.sid)) {
244                 DEBUG(5, ("%s not a SID\n", state->request->data.sid));
245                 return WINBINDD_ERROR;
246         }
247
248         /* Lookup the sid */
249
250         if (!winbindd_lookup_name_by_sid(state->mem_ctx, domain, &sid, 
251                                          &dom_name, &name, &type)) 
252         {
253                 TALLOC_FREE(dom_name);
254                 TALLOC_FREE(name);
255                 return WINBINDD_ERROR;
256         }
257
258         fstrcpy(state->response->data.name.dom_name, dom_name);
259         fstrcpy(state->response->data.name.name, name);
260         state->response->data.name.type = type;
261
262         TALLOC_FREE(dom_name);
263         TALLOC_FREE(name);
264         return WINBINDD_OK;
265 }
266
267 /********************************************************************
268  This is the second callback after contacting the forest root
269 ********************************************************************/
270
271 struct lookupname_state {
272         char *dom_name;
273         char *name;
274         void *caller_private_data;
275 };
276
277
278 static void lookupname_recv2(TALLOC_CTX *mem_ctx, bool success,
279                             struct winbindd_response *response,
280                             void *c, void *private_data)
281 {
282         void (*cont)(void *priv, bool succ, const DOM_SID *sid,
283                      enum lsa_SidType type) =
284                 (void (*)(void *, bool, const DOM_SID *, enum lsa_SidType))c;
285         DOM_SID sid;
286         struct lookupname_state *s = talloc_get_type_abort( private_data,
287                                                             struct lookupname_state );
288
289         if (!success) {
290                 DEBUG(5, ("Could not trigger lookup_name\n"));
291                 cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
292                 return;
293         }
294
295         if (response->result != WINBINDD_OK) {
296                 DEBUG(5, ("lookup_name returned an error\n"));
297                 cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
298                 return;
299         }
300
301         if (!string_to_sid(&sid, response->data.sid.sid)) {
302                 DEBUG(0, ("Could not convert string %s to sid\n",
303                           response->data.sid.sid));
304                 cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
305                 return;
306         }
307
308         cont(s->caller_private_data, True, &sid,
309              (enum lsa_SidType)response->data.sid.type);
310 }
311
312 /********************************************************************
313  This is the first callback after contacting our own domain
314 ********************************************************************/
315
316 static void lookupname_recv(TALLOC_CTX *mem_ctx, bool success,
317                             struct winbindd_response *response,
318                             void *c, void *private_data)
319 {
320         void (*cont)(void *priv, bool succ, const DOM_SID *sid,
321                      enum lsa_SidType type) =
322                 (void (*)(void *, bool, const DOM_SID *, enum lsa_SidType))c;
323         DOM_SID sid;
324         struct lookupname_state *s = talloc_get_type_abort( private_data,
325                                                             struct lookupname_state );  
326
327         if (!success) {
328                 DEBUG(5, ("lookupname_recv: lookup_name() failed!\n"));
329                 cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
330                 return;
331         }
332
333         if (response->result != WINBINDD_OK) {
334                 /* Try again using the forest root */
335                 struct winbindd_domain *root_domain = find_root_domain();
336                 struct winbindd_request request;
337
338                 if ( !root_domain ) {
339                         DEBUG(5,("lookupname_recv: unable to determine forest root\n"));
340                         cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
341                         return;
342                 }
343
344                 ZERO_STRUCT(request);
345                 request.cmd = WINBINDD_LOOKUPNAME;
346
347                 fstrcpy( request.data.name.dom_name, s->dom_name );
348                 fstrcpy( request.data.name.name, s->name );
349
350                 do_async_domain(mem_ctx, root_domain, &request, lookupname_recv2,
351                                 (void *)cont, s);
352
353                 return;
354         }
355
356         if (!string_to_sid(&sid, response->data.sid.sid)) {
357                 DEBUG(0, ("Could not convert string %s to sid\n",
358                           response->data.sid.sid));
359                 cont(s->caller_private_data, False, NULL, SID_NAME_UNKNOWN);
360                 return;
361         }
362
363         cont(s->caller_private_data, True, &sid,
364              (enum lsa_SidType)response->data.sid.type);
365 }
366
367 /********************************************************************
368  The lookup name call first contacts a DC in its own domain
369  and fallbacks to contact a DC if the forest in our domain doesn't
370  know the name.
371 ********************************************************************/
372
373 void winbindd_lookupname_async(TALLOC_CTX *mem_ctx,
374                                const char *dom_name, const char *name,
375                                void (*cont)(void *private_data, bool success,
376                                             const DOM_SID *sid,
377                                             enum lsa_SidType type),
378                                enum winbindd_cmd orig_cmd,
379                                void *private_data)
380 {
381         struct winbindd_request request;
382         struct winbindd_domain *domain;
383         struct lookupname_state *s;
384
385         domain = find_lookup_domain_from_name(dom_name);
386         if (domain == NULL) {
387                 DEBUG(5, ("Could not find domain for name '%s'\n", dom_name));
388                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
389                 return;
390         }
391
392         ZERO_STRUCT(request);
393         request.cmd = WINBINDD_LOOKUPNAME;
394         request.original_cmd = orig_cmd;
395         fstrcpy(request.data.name.dom_name, dom_name);
396         fstrcpy(request.data.name.name, name);
397
398         if ( (s = TALLOC_ZERO_P(mem_ctx, struct lookupname_state)) == NULL ) {
399                 DEBUG(0, ("winbindd_lookupname_async: talloc failed\n"));
400                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
401                 return;
402         }
403
404         s->dom_name = talloc_strdup( s, dom_name );
405         s->name     = talloc_strdup( s, name );
406         if (!s->dom_name || !s->name) {
407                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
408                 return;
409         }
410
411         s->caller_private_data = private_data;
412
413         do_async_domain(mem_ctx, domain, &request, lookupname_recv,
414                         (void *)cont, s);
415 }
416
417 enum winbindd_result winbindd_dual_lookupname(struct winbindd_domain *domain,
418                                               struct winbindd_cli_state *state)
419 {
420         enum lsa_SidType type;
421         char *name_domain, *name_user;
422         DOM_SID sid;
423         char *p;
424
425         /* Ensure null termination */
426         state->request->data.name.dom_name[sizeof(state->request->data.name.dom_name)-1]='\0';
427
428         /* Ensure null termination */
429         state->request->data.name.name[sizeof(state->request->data.name.name)-1]='\0';
430
431         /* cope with the name being a fully qualified name */
432         p = strstr(state->request->data.name.name, lp_winbind_separator());
433         if (p) {
434                 *p = 0;
435                 name_domain = state->request->data.name.name;
436                 name_user = p+1;
437         } else {
438                 name_domain = state->request->data.name.dom_name;
439                 name_user = state->request->data.name.name;
440         }
441
442         DEBUG(3, ("[%5lu]: lookupname %s%s%s\n", (unsigned long)state->pid,
443                   name_domain, lp_winbind_separator(), name_user));
444
445         /* Lookup name from DC using lsa_lookup_names() */
446         if (!winbindd_lookup_sid_by_name(state->mem_ctx, state->request->original_cmd, domain, name_domain,
447                                          name_user, &sid, &type)) {
448                 return WINBINDD_ERROR;
449         }
450
451         sid_to_fstring(state->response->data.sid.sid, &sid);
452         state->response->data.sid.type = type;
453
454         return WINBINDD_OK;
455 }
456
457 bool print_sidlist(TALLOC_CTX *mem_ctx, const DOM_SID *sids,
458                    size_t num_sids, char **result, ssize_t *len)
459 {
460         size_t i;
461         size_t buflen = 0;
462
463         *len = 0;
464         *result = NULL;
465         for (i=0; i<num_sids; i++) {
466                 fstring tmp;
467                 sprintf_append(mem_ctx, result, len, &buflen,
468                                "%s\n", sid_to_fstring(tmp, &sids[i]));
469         }
470
471         if ((num_sids != 0) && (*result == NULL)) {
472                 return False;
473         }
474
475         return True;
476 }
477
478 bool parse_sidlist(TALLOC_CTX *mem_ctx, const char *sidstr,
479                    DOM_SID **sids, size_t *num_sids)
480 {
481         const char *p, *q;
482
483         p = sidstr;
484         if (p == NULL)
485                 return False;
486
487         while (p[0] != '\0') {
488                 fstring tmp;
489                 size_t sidlen;
490                 DOM_SID sid;
491                 q = strchr(p, '\n');
492                 if (q == NULL) {
493                         DEBUG(0, ("Got invalid sidstr: %s\n", p));
494                         return False;
495                 }
496                 sidlen = PTR_DIFF(q, p);
497                 if (sidlen >= sizeof(tmp)-1) {
498                         return false;
499                 }
500                 memcpy(tmp, p, sidlen);
501                 tmp[sidlen] = '\0';
502                 q += 1;
503                 if (!string_to_sid(&sid, tmp)) {
504                         DEBUG(0, ("Could not parse sid %s\n", p));
505                         return False;
506                 }
507                 if (!NT_STATUS_IS_OK(add_sid_to_array(mem_ctx, &sid, sids,
508                                                       num_sids)))
509                 {
510                         return False;
511                 }
512                 p = q;
513         }
514         return True;
515 }
516
517 static void getsidaliases_recv(TALLOC_CTX *mem_ctx, bool success,
518                                struct winbindd_response *response,
519                                void *c, void *private_data)
520 {
521         void (*cont)(void *priv, bool succ,
522                      DOM_SID *aliases, size_t num_aliases) =
523                 (void (*)(void *, bool, DOM_SID *, size_t))c;
524         char *aliases_str;
525         DOM_SID *sids = NULL;
526         size_t num_sids = 0;
527
528         if (!success) {
529                 DEBUG(5, ("Could not trigger getsidaliases\n"));
530                 cont(private_data, success, NULL, 0);
531                 return;
532         }
533
534         if (response->result != WINBINDD_OK) {
535                 DEBUG(5, ("getsidaliases returned an error\n"));
536                 cont(private_data, False, NULL, 0);
537                 return;
538         }
539
540         aliases_str = (char *)response->extra_data.data;
541
542         if (aliases_str == NULL) {
543                 DEBUG(10, ("getsidaliases return 0 SIDs\n"));
544                 cont(private_data, True, NULL, 0);
545                 return;
546         }
547
548         if (!parse_sidlist(mem_ctx, aliases_str, &sids, &num_sids)) {
549                 DEBUG(0, ("Could not parse sids\n"));
550                 cont(private_data, False, NULL, 0);
551                 return;
552         }
553
554         cont(private_data, True, sids, num_sids);
555 }
556
557 void winbindd_getsidaliases_async(struct winbindd_domain *domain,
558                                   TALLOC_CTX *mem_ctx,
559                                   const DOM_SID *sids, size_t num_sids,
560                                   void (*cont)(void *private_data,
561                                                bool success,
562                                                const DOM_SID *aliases,
563                                                size_t num_aliases),
564                                   void *private_data)
565 {
566         struct winbindd_request request;
567         char *sidstr = NULL;
568         ssize_t len;
569
570         if (num_sids == 0) {
571                 cont(private_data, True, NULL, 0);
572                 return;
573         }
574
575         if (!print_sidlist(mem_ctx, sids, num_sids, &sidstr, &len)) {
576                 cont(private_data, False, NULL, 0);
577                 return;
578         }
579
580         ZERO_STRUCT(request);
581         request.cmd = WINBINDD_DUAL_GETSIDALIASES;
582         request.extra_len = len;
583         request.extra_data.data = sidstr;
584
585         do_async_domain(mem_ctx, domain, &request, getsidaliases_recv,
586                         (void *)cont, private_data);
587 }
588
589 static void query_user_recv(TALLOC_CTX *mem_ctx, bool success,
590                             struct winbindd_response *response,
591                             void *c, void *private_data)
592 {
593         void (*cont)(void *priv, bool succ, const char *acct_name,
594                      const char *full_name, const char *homedir, 
595                      const char *shell, uint32 gid, uint32 group_rid) =
596                 (void (*)(void *, bool, const char *, const char *,
597                           const char *, const char *, uint32, uint32))c;
598
599         if (!success) {
600                 DEBUG(5, ("Could not trigger query_user\n"));
601                 cont(private_data, False, NULL, NULL, NULL, NULL, -1, -1);
602                 return;
603         }
604
605         if (response->result != WINBINDD_OK) {
606                 DEBUG(5, ("query_user returned an error\n"));
607                 cont(private_data, False, NULL, NULL, NULL, NULL, -1, -1);
608                 return;
609         }
610
611         cont(private_data, True, response->data.user_info.acct_name,
612              response->data.user_info.full_name,
613              response->data.user_info.homedir,
614              response->data.user_info.shell,
615              response->data.user_info.primary_gid,
616              response->data.user_info.group_rid);
617 }
618
619 void query_user_async(TALLOC_CTX *mem_ctx, struct winbindd_domain *domain,
620                       const DOM_SID *sid,
621                       void (*cont)(void *private_data, bool success,
622                                    const char *acct_name,
623                                    const char *full_name,
624                                    const char *homedir,
625                                    const char *shell,
626                                    gid_t gid,
627                                    uint32 group_rid),
628                       void *private_data)
629 {
630         struct winbindd_request request;
631         ZERO_STRUCT(request);
632         request.cmd = WINBINDD_DUAL_USERINFO;
633         sid_to_fstring(request.data.sid, sid);
634         do_async_domain(mem_ctx, domain, &request, query_user_recv,
635                         (void *)cont, private_data);
636 }
637
638 enum winbindd_result winbindd_dual_ping(struct winbindd_domain *domain,
639                                         struct winbindd_cli_state *state)
640 {
641         return WINBINDD_OK;
642 }