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 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 in 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         if ( (domain = find_lookup_domain_from_name(dom_name)) == NULL ) {
 386                 DEBUG(5, ("Could not find domain for name '%s'\n", dom_name));
 387                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
 388                 return;
 389         }
 390
 391         ZERO_STRUCT(request);
 392         request.cmd = WINBINDD_LOOKUPNAME;
 393         request.original_cmd = orig_cmd;
 394         fstrcpy(request.data.name.dom_name, dom_name);
 395         fstrcpy(request.data.name.name, name);
 396
 397         if ( (s = TALLOC_ZERO_P(mem_ctx, struct lookupname_state)) == NULL ) {
 398                 DEBUG(0, ("winbindd_lookupname_async: talloc failed\n"));
 399                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
 400                 return;
 401         }
 402
 403         s->dom_name = talloc_strdup( s, dom_name );
 404         s->name     = talloc_strdup( s, name );
 405         if (!s->dom_name || !s->name) {
 406                 cont(private_data, False, NULL, SID_NAME_UNKNOWN);
 407                 return;
 408         }
 409
 410         s->caller_private_data = private_data;
 411
 412         do_async_domain(mem_ctx, domain, &request, lookupname_recv,
 413                         (void *)cont, s);
 414 }
 415
 416 enum winbindd_result winbindd_dual_lookupname(struct winbindd_domain *domain,
 417                                               struct winbindd_cli_state *state)
 418 {
 419         enum lsa_SidType type;
 420         char *name_domain, *name_user;
 421         DOM_SID sid;
 422         char *p;
 423
 424         /* Ensure null termination */
 425         state->request.data.name.dom_name[sizeof(state->request.data.name.dom_name)-1]='\0';
 426
 427         /* Ensure null termination */
 428         state->request.data.name.name[sizeof(state->request.data.name.name)-1]='\0';
 429
 430         /* cope with the name being a fully qualified name */
 431         p = strstr(state->request.data.name.name, lp_winbind_separator());
 432         if (p) {
 433                 *p = 0;
 434                 name_domain = state->request.data.name.name;
 435                 name_user = p+1;
 436         } else {
 437                 name_domain = state->request.data.name.dom_name;
 438                 name_user = state->request.data.name.name;
 439         }
 440
 441         DEBUG(3, ("[%5lu]: lookupname %s%s%s\n", (unsigned long)state->pid,
 442                   name_domain, lp_winbind_separator(), name_user));
 443
 444         /* Lookup name from DC using lsa_lookup_names() */
 445         if (!winbindd_lookup_sid_by_name(state->mem_ctx, state->request.original_cmd, domain, name_domain,
 446                                          name_user, &sid, &type)) {
 447                 return WINBINDD_ERROR;
 448         }
 449
 450         sid_to_fstring(state->response.data.sid.sid, &sid);
 451         state->response.data.sid.type = type;
 452
 453         return WINBINDD_OK;
 454 }
 455
 456 /* This is the first callback after enumerating users/groups from a domain */
 457 static void listent_recv(TALLOC_CTX *mem_ctx, bool success,
 458                             struct winbindd_response *response,
 459                             void *c, void *private_data)
 460 {
 461         void (*cont)(void *priv, bool succ, fstring dom_name, char *data) =
 462                 (void (*)(void *, bool, fstring, char*))c;
 463
 464         if (!success || response->result != WINBINDD_OK) {
 465                 DEBUG(5, ("list_ent() failed!\n"));
 466                 cont(private_data, False, response->data.name.dom_name, NULL);
 467                 return;
 468         }
 469
 470         cont(private_data, True, response->data.name.dom_name,
 471              (char *)response->extra_data.data);
 472
 473         SAFE_FREE(response->extra_data.data);
 474 }
 475
 476 /* Request the name of all users/groups in a single domain */
 477 void winbindd_listent_async(TALLOC_CTX *mem_ctx,
 478                                struct winbindd_domain *domain,
 479                                void (*cont)(void *private_data, bool success,
 480                                      fstring dom_name, char* extra_data),
 481                                void *private_data, enum ent_type type)
 482 {
 483         struct winbindd_request request;
 484
 485         ZERO_STRUCT(request);
 486         if (type == LIST_USERS)
 487                 request.cmd = WINBINDD_LIST_USERS;
 488         else if (type == LIST_GROUPS)
 489                 request.cmd = WINBINDD_LIST_GROUPS;
 490
 491         do_async_domain(mem_ctx, domain, &request, listent_recv,
 492                         (void *)cont, private_data);
 493 }
 494
 495 enum winbindd_result winbindd_dual_list_users(struct winbindd_domain *domain,
 496                                               struct winbindd_cli_state *state)
 497 {
 498         WINBIND_USERINFO *info;
 499         NTSTATUS status;
 500         struct winbindd_methods *methods;
 501         uint32 num_entries = 0;
 502         char *extra_data = NULL;
 503         uint32_t extra_data_len = 0, i;
 504
 505         /* Must copy domain into response first for debugging in parent */
 506         fstrcpy(state->response.data.name.dom_name, domain->name);
 507
 508         /* Query user info */
 509         methods = domain->methods;
 510         status = methods->query_user_list(domain, state->mem_ctx,
 511                                           &num_entries, &info);
 512
 513         if (!NT_STATUS_IS_OK(status))
 514                 return WINBINDD_ERROR;
 515
 516         if (num_entries == 0)
 517                 return WINBINDD_OK;
 518
 519         /* Allocate some memory for extra data.  Note that we limit
 520            account names to sizeof(fstring) = 256 characters.
 521            +1 for the ',' between group names */
 522         extra_data = (char *)SMB_REALLOC(extra_data,
 523                 (sizeof(fstring) + 1) * num_entries);
 524
 525         if (!extra_data) {
 526                 DEBUG(0,("failed to enlarge buffer!\n"));
 527                 return WINBINDD_ERROR;
 528         }
 529
 530         /* Pack user list into extra data fields */
 531         for (i = 0; i < num_entries; i++) {
 532                 fstring acct_name, name;
 533
 534                 if (info[i].acct_name == NULL)
 535                         fstrcpy(acct_name, "");
 536                 else
 537                         fstrcpy(acct_name, info[i].acct_name);
 538
 539                 fill_domain_username(name, domain->name, acct_name, True);
 540                 /* Append to extra data */
 541                 memcpy(&extra_data[extra_data_len], name, strlen(name));
 542                 extra_data_len += strlen(name);
 543                 extra_data[extra_data_len++] = ',';
 544         }
 545
 546         /* Assign extra_data fields in response structure */
 547         if (extra_data) {
 548                 /* remove trailing ',' */
 549                 extra_data[extra_data_len - 1] = '\0';
 550                 state->response.extra_data.data = extra_data;
 551                 state->response.length += extra_data_len;
 552         }
 553
 554         return WINBINDD_OK;
 555 }
 556
 557 enum winbindd_result winbindd_dual_list_groups(struct winbindd_domain *domain,
 558                                                struct winbindd_cli_state *state)
 559 {
 560         struct getent_state groups = {};
 561         char *extra_data = NULL;
 562         uint32_t extra_data_len = 0, i;
 563
 564         /* Must copy domain into response first for debugging in parent */
 565         fstrcpy(state->response.data.name.dom_name, domain->name);
 566         fstrcpy(groups.domain_name, domain->name);
 567
 568         /* Get list of sam groups */
 569         if (!get_sam_group_entries(&groups)) {
 570                 /* this domain is empty or in an error state */
 571                 return WINBINDD_ERROR;
 572         }
 573
 574         /* Allocate some memory for extra data.  Note that we limit
 575            account names to sizeof(fstring) = 256 characters.
 576            +1 for the ',' between group names */
 577         extra_data = (char *)SMB_REALLOC(extra_data,
 578                 (sizeof(fstring) + 1) * groups.num_sam_entries);
 579
 580         if (!extra_data) {
 581                 DEBUG(0,("failed to enlarge buffer!\n"));
 582                 SAFE_FREE(groups.sam_entries);
 583                 return WINBINDD_ERROR;
 584         }
 585
 586         /* Pack group list into extra data fields */
 587         for (i = 0; i < groups.num_sam_entries; i++) {
 588                 char *group_name = ((struct acct_info *)
 589                                     groups.sam_entries)[i].acct_name;
 590                 fstring name;
 591
 592                 fill_domain_username(name, domain->name, group_name, True);
 593                 /* Append to extra data */
 594                 memcpy(&extra_data[extra_data_len], name, strlen(name));
 595                 extra_data_len += strlen(name);
 596                 extra_data[extra_data_len++] = ',';
 597         }
 598
 599         SAFE_FREE(groups.sam_entries);
 600
 601         /* Assign extra_data fields in response structure */
 602         if (extra_data) {
 603                 /* remove trailing ',' */
 604                 extra_data[extra_data_len - 1] = '\0';
 605                 state->response.extra_data.data = extra_data;
 606                 state->response.length += extra_data_len;
 607         }
 608
 609         return WINBINDD_OK;
 610 }
 611
 612 bool print_sidlist(TALLOC_CTX *mem_ctx, const DOM_SID *sids,
 613                    size_t num_sids, char **result, ssize_t *len)
 614 {
 615         size_t i;
 616         size_t buflen = 0;
 617
 618         *len = 0;
 619         *result = NULL;
 620         for (i=0; i<num_sids; i++) {
 621                 fstring tmp;
 622                 sprintf_append(mem_ctx, result, len, &buflen,
 623                                "%s\n", sid_to_fstring(tmp, &sids[i]));
 624         }
 625
 626         if ((num_sids != 0) && (*result == NULL)) {
 627                 return False;
 628         }
 629
 630         return True;
 631 }
 632
 633 static bool parse_sidlist(TALLOC_CTX *mem_ctx, char *sidstr,
 634                           DOM_SID **sids, size_t *num_sids)
 635 {
 636         char *p, *q;
 637
 638         p = sidstr;
 639         if (p == NULL)
 640                 return False;
 641
 642         while (p[0] != '\0') {
 643                 DOM_SID sid;
 644                 q = strchr(p, '\n');
 645                 if (q == NULL) {
 646                         DEBUG(0, ("Got invalid sidstr: %s\n", p));
 647                         return False;
 648                 }
 649                 *q = '\0';
 650                 q += 1;
 651                 if (!string_to_sid(&sid, p)) {
 652                         DEBUG(0, ("Could not parse sid %s\n", p));
 653                         return False;
 654                 }
 655                 if (!NT_STATUS_IS_OK(add_sid_to_array(mem_ctx, &sid, sids,
 656                                                       num_sids)))
 657                 {
 658                         return False;
 659                 }
 660                 p = q;
 661         }
 662         return True;
 663 }
 664
 665 static bool parse_ridlist(TALLOC_CTX *mem_ctx, char *ridstr,
 666                           uint32 **rids, size_t *num_rids)
 667 {
 668         char *p;
 669
 670         p = ridstr;
 671         if (p == NULL)
 672                 return False;
 673
 674         while (p[0] != '\0') {
 675                 uint32 rid;
 676                 char *q;
 677                 rid = strtoul(p, &q, 10);
 678                 if (*q != '\n') {
 679                         DEBUG(0, ("Got invalid ridstr: %s\n", p));
 680                         return False;
 681                 }
 682                 p = q+1;
 683                 ADD_TO_ARRAY(mem_ctx, uint32, rid, rids, num_rids);
 684         }
 685         return True;
 686 }
 687
 688 enum winbindd_result winbindd_dual_lookuprids(struct winbindd_domain *domain,
 689                                               struct winbindd_cli_state *state)
 690 {
 691         uint32 *rids = NULL;
 692         size_t i, buflen, num_rids = 0;
 693         ssize_t len;
 694         DOM_SID domain_sid;
 695         char *domain_name;
 696         char **names;
 697         enum lsa_SidType *types;
 698         NTSTATUS status;
 699         char *result;
 700
 701         DEBUG(10, ("Looking up RIDs for domain %s (%s)\n",
 702                    state->request.domain_name,
 703                    state->request.data.sid));
 704
 705         if (!parse_ridlist(state->mem_ctx, state->request.extra_data.data,
 706                            &rids, &num_rids)) {
 707                 DEBUG(5, ("Could not parse ridlist\n"));
 708                 return WINBINDD_ERROR;
 709         }
 710
 711         if (!string_to_sid(&domain_sid, state->request.data.sid)) {
 712                 DEBUG(5, ("Could not parse domain sid %s\n",
 713                           state->request.data.sid));
 714                 return WINBINDD_ERROR;
 715         }
 716
 717         status = domain->methods->rids_to_names(domain, state->mem_ctx,
 718                                                 &domain_sid, rids, num_rids,
 719                                                 &domain_name,
 720                                                 &names, &types);
 721
 722         if (!NT_STATUS_IS_OK(status) &&
 723             !NT_STATUS_EQUAL(status, STATUS_SOME_UNMAPPED)) {
 724                 return WINBINDD_ERROR;
 725         }
 726
 727         len = 0;
 728         buflen = 0;
 729         result = NULL;
 730
 731         for (i=0; i<num_rids; i++) {
 732                 sprintf_append(state->mem_ctx, &result, &len, &buflen,
 733                                "%d %s\n", types[i], names[i]);
 734         }
 735
 736         fstrcpy(state->response.data.domain_name, domain_name);
 737
 738         if (result != NULL) {
 739                 state->response.extra_data.data = SMB_STRDUP(result);
 740                 if (!state->response.extra_data.data) {
 741                         return WINBINDD_ERROR;
 742                 }
 743                 state->response.length += len+1;
 744         }
 745
 746         return WINBINDD_OK;
 747 }
 748
 749 static void getsidaliases_recv(TALLOC_CTX *mem_ctx, bool success,
 750                                struct winbindd_response *response,
 751                                void *c, void *private_data)
 752 {
 753         void (*cont)(void *priv, bool succ,
 754                      DOM_SID *aliases, size_t num_aliases) =
 755                 (void (*)(void *, bool, DOM_SID *, size_t))c;
 756         char *aliases_str;
 757         DOM_SID *sids = NULL;
 758         size_t num_sids = 0;
 759
 760         if (!success) {
 761                 DEBUG(5, ("Could not trigger getsidaliases\n"));
 762                 cont(private_data, success, NULL, 0);
 763                 return;
 764         }
 765
 766         if (response->result != WINBINDD_OK) {
 767                 DEBUG(5, ("getsidaliases returned an error\n"));
 768                 cont(private_data, False, NULL, 0);
 769                 return;
 770         }
 771
 772         aliases_str = (char *)response->extra_data.data;
 773
 774         if (aliases_str == NULL) {
 775                 DEBUG(10, ("getsidaliases return 0 SIDs\n"));
 776                 cont(private_data, True, NULL, 0);
 777                 return;
 778         }
 779
 780         if (!parse_sidlist(mem_ctx, aliases_str, &sids, &num_sids)) {
 781                 DEBUG(0, ("Could not parse sids\n"));
 782                 cont(private_data, False, NULL, 0);
 783                 return;
 784         }
 785
 786         SAFE_FREE(response->extra_data.data);
 787
 788         cont(private_data, True, sids, num_sids);
 789 }
 790
 791 void winbindd_getsidaliases_async(struct winbindd_domain *domain,
 792                                   TALLOC_CTX *mem_ctx,
 793                                   const DOM_SID *sids, size_t num_sids,
 794                                   void (*cont)(void *private_data,
 795                                                bool success,
 796                                                const DOM_SID *aliases,
 797                                                size_t num_aliases),
 798                                   void *private_data)
 799 {
 800         struct winbindd_request request;
 801         char *sidstr = NULL;
 802         ssize_t len;
 803
 804         if (num_sids == 0) {
 805                 cont(private_data, True, NULL, 0);
 806                 return;
 807         }
 808
 809         if (!print_sidlist(mem_ctx, sids, num_sids, &sidstr, &len)) {
 810                 cont(private_data, False, NULL, 0);
 811                 return;
 812         }
 813
 814         ZERO_STRUCT(request);
 815         request.cmd = WINBINDD_DUAL_GETSIDALIASES;
 816         request.extra_len = len;
 817         request.extra_data.data = sidstr;
 818
 819         do_async_domain(mem_ctx, domain, &request, getsidaliases_recv,
 820                         (void *)cont, private_data);
 821 }
 822
 823 enum winbindd_result winbindd_dual_getsidaliases(struct winbindd_domain *domain,
 824                                                  struct winbindd_cli_state *state)
 825 {
 826         DOM_SID *sids = NULL;
 827         size_t num_sids = 0;
 828         char *sidstr = NULL;
 829         ssize_t len;
 830         size_t i;
 831         uint32 num_aliases;
 832         uint32 *alias_rids;
 833         NTSTATUS result;
 834
 835         DEBUG(3, ("[%5lu]: getsidaliases\n", (unsigned long)state->pid));
 836
 837         sidstr = state->request.extra_data.data;
 838         if (sidstr == NULL) {
 839                 sidstr = talloc_strdup(state->mem_ctx, "\n"); /* No SID */
 840                 if (!sidstr) {
 841                         DEBUG(0, ("Out of memory\n"));
 842                         return WINBINDD_ERROR;
 843                 }
 844         }
 845
 846         DEBUG(10, ("Sidlist: %s\n", sidstr));
 847
 848         if (!parse_sidlist(state->mem_ctx, sidstr, &sids, &num_sids)) {
 849                 DEBUG(0, ("Could not parse SID list: %s\n", sidstr));
 850                 return WINBINDD_ERROR;
 851         }
 852
 853         num_aliases = 0;
 854         alias_rids = NULL;
 855
 856         result = domain->methods->lookup_useraliases(domain,
 857                                                      state->mem_ctx,
 858                                                      num_sids, sids,
 859                                                      &num_aliases,
 860                                                      &alias_rids);
 861
 862         if (!NT_STATUS_IS_OK(result)) {
 863                 DEBUG(3, ("Could not lookup_useraliases: %s\n",
 864                           nt_errstr(result)));
 865                 return WINBINDD_ERROR;
 866         }
 867
 868         num_sids = 0;
 869         sids = NULL;
 870         sidstr = NULL;
 871
 872         DEBUG(10, ("Got %d aliases\n", num_aliases));
 873
 874         for (i=0; i<num_aliases; i++) {
 875                 DOM_SID sid;
 876                 DEBUGADD(10, (" rid %d\n", alias_rids[i]));
 877                 sid_copy(&sid, &domain->sid);
 878                 sid_append_rid(&sid, alias_rids[i]);
 879                 result = add_sid_to_array(state->mem_ctx, &sid, &sids,
 880                                           &num_sids);
 881                 if (!NT_STATUS_IS_OK(result)) {
 882                         return WINBINDD_ERROR;
 883                 }
 884         }
 885
 886
 887         if (!print_sidlist(state->mem_ctx, sids, num_sids, &sidstr, &len)) {
 888                 DEBUG(0, ("Could not print_sidlist\n"));
 889                 state->response.extra_data.data = NULL;
 890                 return WINBINDD_ERROR;
 891         }
 892
 893         state->response.extra_data.data = NULL;
 894
 895         if (sidstr) {
 896                 state->response.extra_data.data = SMB_STRDUP(sidstr);
 897                 if (!state->response.extra_data.data) {
 898                         DEBUG(0, ("Out of memory\n"));
 899                         return WINBINDD_ERROR;
 900                 }
 901                 DEBUG(10, ("aliases_list: %s\n",
 902                            (char *)state->response.extra_data.data));
 903                 state->response.length += len+1;
 904         }
 905
 906         return WINBINDD_OK;
 907 }
 908
 909 struct gettoken_state {
 910         TALLOC_CTX *mem_ctx;
 911         DOM_SID user_sid;
 912         struct winbindd_domain *alias_domain;
 913         struct winbindd_domain *local_alias_domain;
 914         struct winbindd_domain *builtin_domain;
 915         DOM_SID *sids;
 916         size_t num_sids;
 917         void (*cont)(void *private_data, bool success, DOM_SID *sids, size_t num_sids);
 918         void *private_data;
 919 };
 920
 921 static void gettoken_recvdomgroups(TALLOC_CTX *mem_ctx, bool success,
 922                                    struct winbindd_response *response,
 923                                    void *c, void *private_data);
 924 static void gettoken_recvaliases(void *private_data, bool success,
 925                                  const DOM_SID *aliases,
 926                                  size_t num_aliases);
 927
 928
 929 void winbindd_gettoken_async(TALLOC_CTX *mem_ctx, const DOM_SID *user_sid,
 930                              void (*cont)(void *private_data, bool success,
 931                                           DOM_SID *sids, size_t num_sids),
 932                              void *private_data)
 933 {
 934         struct winbindd_domain *domain;
 935         struct winbindd_request request;
 936         struct gettoken_state *state;
 937
 938         state = TALLOC_ZERO_P(mem_ctx, struct gettoken_state);
 939         if (state == NULL) {
 940                 DEBUG(0, ("talloc failed\n"));
 941                 cont(private_data, False, NULL, 0);
 942                 return;
 943         }
 944
 945         state->mem_ctx = mem_ctx;
 946         sid_copy(&state->user_sid, user_sid);
 947         state->alias_domain = find_our_domain();
 948         state->local_alias_domain = find_domain_from_name( get_global_sam_name() );
 949         state->builtin_domain = find_builtin_domain();
 950         state->cont = cont;
 951         state->private_data = private_data;
 952
 953         domain = find_domain_from_sid_noinit(user_sid);
 954         if (domain == NULL) {
 955                 DEBUG(5, ("Could not find domain from SID %s\n",
 956                           sid_string_dbg(user_sid)));
 957                 cont(private_data, False, NULL, 0);
 958                 return;
 959         }
 960
 961         ZERO_STRUCT(request);
 962         request.cmd = WINBINDD_GETUSERDOMGROUPS;
 963         sid_to_fstring(request.data.sid, user_sid);
 964
 965         do_async_domain(mem_ctx, domain, &request, gettoken_recvdomgroups,
 966                         NULL, state);
 967 }
 968
 969 static void gettoken_recvdomgroups(TALLOC_CTX *mem_ctx, bool success,
 970                                    struct winbindd_response *response,
 971                                    void *c, void *private_data)
 972 {
 973         struct gettoken_state *state =
 974                 talloc_get_type_abort(private_data, struct gettoken_state);
 975         char *sids_str;
 976
 977         if (!success) {
 978                 DEBUG(10, ("Could not get domain groups\n"));
 979                 state->cont(state->private_data, False, NULL, 0);
 980                 return;
 981         }
 982
 983         sids_str = (char *)response->extra_data.data;
 984
 985         if (sids_str == NULL) {
 986                 /* This could be normal if we are dealing with a
 987                    local user and local groups */
 988
 989                 if ( !sid_check_is_in_our_domain( &state->user_sid ) ) {
 990                         DEBUG(10, ("Received no domain groups\n"));
 991                         state->cont(state->private_data, True, NULL, 0);
 992                         return;
 993                 }
 994         }
 995
 996         state->sids = NULL;
 997         state->num_sids = 0;
 998
 999         if (!NT_STATUS_IS_OK(add_sid_to_array(mem_ctx, &state->user_sid,
1000                                               &state->sids, &state->num_sids)))
1001         {
1002                 DEBUG(0, ("Out of memory\n"));
1003                 state->cont(state->private_data, False, NULL, 0);
1004                 return;
1005         }
1006
1007         if (sids_str && !parse_sidlist(mem_ctx, sids_str, &state->sids,
1008                            &state->num_sids)) {
1009                 DEBUG(0, ("Could not parse sids\n"));
1010                 state->cont(state->private_data, False, NULL, 0);
1011                 return;
1012         }
1013
1014         SAFE_FREE(response->extra_data.data);
1015
1016         if (state->alias_domain == NULL) {
1017                 DEBUG(10, ("Don't expand domain local groups\n"));
1018                 state->cont(state->private_data, True, state->sids,
1019                             state->num_sids);
1020                 return;
1021         }
1022
1023         winbindd_getsidaliases_async(state->alias_domain, mem_ctx,
1024                                      state->sids, state->num_sids,
1025                                      gettoken_recvaliases, state);
1026 }
1027
1028 static void gettoken_recvaliases(void *private_data, bool success,
1029                                  const DOM_SID *aliases,
1030                                  size_t num_aliases)
1031 {
1032         struct gettoken_state *state = (struct gettoken_state *)private_data;
1033         size_t i;
1034
1035         if (!success) {
1036                 DEBUG(10, ("Could not receive domain local groups\n"));
1037                 state->cont(state->private_data, False, NULL, 0);
1038                 return;
1039         }
1040
1041         for (i=0; i<num_aliases; i++) {
1042                 if (!NT_STATUS_IS_OK(add_sid_to_array(state->mem_ctx,
1043                                                       &aliases[i],
1044                                                       &state->sids,
1045                                                       &state->num_sids)))
1046                 {
1047                         DEBUG(0, ("Out of memory\n"));
1048                         state->cont(state->private_data, False, NULL, 0);
1049                         return;
1050                 }
1051         }
1052
1053         if (state->local_alias_domain != NULL) {
1054                 struct winbindd_domain *local_domain = state->local_alias_domain;
1055                 DEBUG(10, ("Expanding our own local groups\n"));
1056                 state->local_alias_domain = NULL;
1057                 winbindd_getsidaliases_async(local_domain, state->mem_ctx,
1058                                              state->sids, state->num_sids,
1059                                              gettoken_recvaliases, state);
1060                 return;
1061         }
1062
1063         if (state->builtin_domain != NULL) {
1064                 struct winbindd_domain *builtin_domain = state->builtin_domain;
1065                 DEBUG(10, ("Expanding our own BUILTIN groups\n"));
1066                 state->builtin_domain = NULL;
1067                 winbindd_getsidaliases_async(builtin_domain, state->mem_ctx,
1068                                              state->sids, state->num_sids,
1069                                              gettoken_recvaliases, state);
1070                 return;
1071         }
1072
1073         state->cont(state->private_data, True, state->sids, state->num_sids);
1074 }
1075
1076 static void query_user_recv(TALLOC_CTX *mem_ctx, bool success,
1077                             struct winbindd_response *response,
1078                             void *c, void *private_data)
1079 {
1080         void (*cont)(void *priv, bool succ, const char *acct_name,
1081                      const char *full_name, const char *homedir,
1082                      const char *shell, uint32 gid, uint32 group_rid) =
1083                 (void (*)(void *, bool, const char *, const char *,
1084                           const char *, const char *, uint32, uint32))c;
1085
1086         if (!success) {
1087                 DEBUG(5, ("Could not trigger query_user\n"));
1088                 cont(private_data, False, NULL, NULL, NULL, NULL, -1, -1);
1089                 return;
1090         }
1091
1092         if (response->result != WINBINDD_OK) {
1093                 DEBUG(5, ("query_user returned an error\n"));
1094                 cont(private_data, False, NULL, NULL, NULL, NULL, -1, -1);
1095                 return;
1096         }
1097
1098         cont(private_data, True, response->data.user_info.acct_name,
1099              response->data.user_info.full_name,
1100              response->data.user_info.homedir,
1101              response->data.user_info.shell,
1102              response->data.user_info.primary_gid,
1103              response->data.user_info.group_rid);
1104 }
1105
1106 void query_user_async(TALLOC_CTX *mem_ctx, struct winbindd_domain *domain,
1107                       const DOM_SID *sid,
1108                       void (*cont)(void *private_data, bool success,
1109                                    const char *acct_name,
1110                                    const char *full_name,
1111                                    const char *homedir,
1112                                    const char *shell,
1113                                    gid_t gid,
1114                                    uint32 group_rid),
1115                       void *private_data)
1116 {
1117         struct winbindd_request request;
1118         ZERO_STRUCT(request);
1119         request.cmd = WINBINDD_DUAL_USERINFO;
1120         sid_to_fstring(request.data.sid, sid);
1121         do_async_domain(mem_ctx, domain, &request, query_user_recv,
1122                         (void *)cont, private_data);
1123 }