source/nsswitch/winbindd_dual.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Winbind child daemons
   5
   6    Copyright (C) Andrew Tridgell 2002
   7    Copyright (C) Volker Lendecke 2004,2005
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 /*
  25  * We fork a child per domain to be able to act non-blocking in the main
  26  * winbind daemon. A domain controller thousands of miles away being being
  27  * slow replying with a 10.000 user list should not hold up netlogon calls
  28  * that can be handled locally.
  29  */
  30
  31 #include "includes.h"
  32 #include "winbindd.h"
  33
  34 #undef DBGC_CLASS
  35 #define DBGC_CLASS DBGC_WINBIND
  36
  37 extern BOOL override_logfile;
  38
  39 /* Read some data from a client connection */
  40
  41 static void child_read_request(struct winbindd_cli_state *state)
  42 {
  43         ssize_t len;
  44
  45         /* Read data */
  46
  47         len = read_data(state->sock, (char *)&state->request,
  48                         sizeof(state->request));
  49
  50         if (len != sizeof(state->request)) {
  51                 DEBUG(len > 0 ? 0 : 3, ("Got invalid request length: %d\n", (int)len));
  52                 state->finished = True;
  53                 return;
  54         }
  55
  56         if (state->request.extra_len == 0) {
  57                 state->request.extra_data.data = NULL;
  58                 return;
  59         }
  60
  61         DEBUG(10, ("Need to read %d extra bytes\n", (int)state->request.extra_len));
  62
  63         state->request.extra_data.data =
  64                 SMB_MALLOC_ARRAY(char, state->request.extra_len + 1);
  65
  66         if (state->request.extra_data.data == NULL) {
  67                 DEBUG(0, ("malloc failed\n"));
  68                 state->finished = True;
  69                 return;
  70         }
  71
  72         /* Ensure null termination */
  73         state->request.extra_data.data[state->request.extra_len] = '\0';
  74
  75         len = read_data(state->sock, state->request.extra_data.data,
  76                         state->request.extra_len);
  77
  78         if (len != state->request.extra_len) {
  79                 DEBUG(0, ("Could not read extra data\n"));
  80                 state->finished = True;
  81                 return;
  82         }
  83 }
  84
  85 /*
  86  * Machinery for async requests sent to children. You set up a
  87  * winbindd_request, select a child to query, and issue a async_request
  88  * call. When the request is completed, the callback function you specified is
  89  * called back with the private pointer you gave to async_request.
  90  */
  91
  92 struct winbindd_async_request {
  93         struct winbindd_async_request *next, *prev;
  94         TALLOC_CTX *mem_ctx;
  95         struct winbindd_child *child;
  96         struct winbindd_request *request;
  97         struct winbindd_response *response;
  98         void (*continuation)(void *private_data, BOOL success);
  99         struct timed_event *reply_timeout_event;
 100         pid_t child_pid; /* pid of the child we're waiting on. Used to detect
 101                             a restart of the child (child->pid != child_pid). */
 102         void *private_data;
 103 };
 104
 105 static void async_main_request_sent(void *private_data, BOOL success);
 106 static void async_request_sent(void *private_data, BOOL success);
 107 static void async_reply_recv(void *private_data, BOOL success);
 108 static void schedule_async_request(struct winbindd_child *child);
 109
 110 void async_request(TALLOC_CTX *mem_ctx, struct winbindd_child *child,
 111                    struct winbindd_request *request,
 112                    struct winbindd_response *response,
 113                    void (*continuation)(void *private_data, BOOL success),
 114                    void *private_data)
 115 {
 116         struct winbindd_async_request *state;
 117
 118         SMB_ASSERT(continuation != NULL);
 119
 120         state = TALLOC_P(mem_ctx, struct winbindd_async_request);
 121
 122         if (state == NULL) {
 123                 DEBUG(0, ("talloc failed\n"));
 124                 continuation(private_data, False);
 125                 return;
 126         }
 127
 128         state->mem_ctx = mem_ctx;
 129         state->child = child;
 130         state->request = request;
 131         state->response = response;
 132         state->continuation = continuation;
 133         state->private_data = private_data;
 134
 135         DLIST_ADD_END(child->requests, state, struct winbindd_async_request *);
 136
 137         schedule_async_request(child);
 138
 139         return;
 140 }
 141
 142 static void async_main_request_sent(void *private_data, BOOL success)
 143 {
 144         struct winbindd_async_request *state =
 145                 talloc_get_type_abort(private_data, struct winbindd_async_request);
 146
 147         if (!success) {
 148                 DEBUG(5, ("Could not send async request\n"));
 149
 150                 state->response->length = sizeof(struct winbindd_response);
 151                 state->response->result = WINBINDD_ERROR;
 152                 state->continuation(state->private_data, False);
 153                 return;
 154         }
 155
 156         if (state->request->extra_len == 0) {
 157                 async_request_sent(private_data, True);
 158                 return;
 159         }
 160
 161         setup_async_write(&state->child->event, state->request->extra_data.data,
 162                           state->request->extra_len,
 163                           async_request_sent, state);
 164 }
 165
 166 /****************************************************************
 167  Handler triggered if the child winbindd doesn't respond within
 168  a given timeout.
 169 ****************************************************************/
 170
 171 static void async_request_timeout_handler(struct event_context *ctx,
 172                                         struct timed_event *te,
 173                                         const struct timeval *now,
 174                                         void *private_data)
 175 {
 176         struct winbindd_async_request *state =
 177                 talloc_get_type_abort(private_data, struct winbindd_async_request);
 178
 179         DEBUG(0,("async_request_timeout_handler: child pid %u is not responding. "
 180                 "Closing connection to it.\n",
 181                 state->child_pid ));
 182
 183         /* Deal with the reply - set to error. */
 184         async_reply_recv(private_data, False);
 185 }
 186
 187 /**************************************************************
 188  Common function called on both async send and recv fail.
 189  Cleans up the child and schedules the next request.
 190 **************************************************************/
 191
 192 static void async_request_fail(struct winbindd_async_request *state)
 193 {
 194         DLIST_REMOVE(state->child->requests, state);
 195
 196         TALLOC_FREE(state->reply_timeout_event);
 197
 198         SMB_ASSERT(state->child_pid != (pid_t)0);
 199
 200         /* If not already reaped, send kill signal to child. */
 201         if (state->child->pid == state->child_pid) {
 202                 kill(state->child_pid, SIGTERM);
 203
 204                 /*
 205                  * Close the socket to the child.
 206                  */
 207                 winbind_child_died(state->child_pid);
 208         }
 209
 210         state->response->length = sizeof(struct winbindd_response);
 211         state->response->result = WINBINDD_ERROR;
 212         state->continuation(state->private_data, False);
 213 }
 214
 215 static void async_request_sent(void *private_data_data, BOOL success)
 216 {
 217         struct winbindd_async_request *state =
 218                 talloc_get_type_abort(private_data_data, struct winbindd_async_request);
 219
 220         if (!success) {
 221                 DEBUG(5, ("Could not send async request to child pid %u\n",
 222                         (unsigned int)state->child_pid ));
 223                 async_request_fail(state);
 224                 return;
 225         }
 226
 227         /* Request successfully sent to the child, setup the wait for reply */
 228
 229         setup_async_read(&state->child->event,
 230                          &state->response->result,
 231                          sizeof(state->response->result),
 232                          async_reply_recv, state);
 233
 234         /*
 235          * Set up a timeout of 300 seconds for the response.
 236          * If we don't get it close the child socket and
 237          * report failure.
 238          */
 239
 240         state->reply_timeout_event = event_add_timed(winbind_event_context(),
 241                                                         NULL,
 242                                                         timeval_current_ofs(300,0),
 243                                                         "async_request_timeout",
 244                                                         async_request_timeout_handler,
 245                                                         state);
 246         if (!state->reply_timeout_event) {
 247                 smb_panic("async_request_sent: failed to add timeout handler.\n");
 248         }
 249 }
 250
 251 static void async_reply_recv(void *private_data, BOOL success)
 252 {
 253         struct winbindd_async_request *state =
 254                 talloc_get_type_abort(private_data, struct winbindd_async_request);
 255         struct winbindd_child *child = state->child;
 256
 257         TALLOC_FREE(state->reply_timeout_event);
 258
 259         state->response->length = sizeof(struct winbindd_response);
 260
 261         if (!success) {
 262                 DEBUG(5, ("Could not receive async reply from child pid %u\n",
 263                         (unsigned int)state->child_pid ));
 264
 265                 cache_cleanup_response(state->child_pid);
 266                 async_request_fail(state);
 267                 return;
 268         }
 269
 270         SMB_ASSERT(cache_retrieve_response(state->child_pid,
 271                                            state->response));
 272
 273         cache_cleanup_response(state->child_pid);
 274
 275         DLIST_REMOVE(child->requests, state);
 276
 277         schedule_async_request(child);
 278
 279         state->continuation(state->private_data, True);
 280 }
 281
 282 static BOOL fork_domain_child(struct winbindd_child *child);
 283
 284 static void schedule_async_request(struct winbindd_child *child)
 285 {
 286         struct winbindd_async_request *request = child->requests;
 287
 288         if (request == NULL) {
 289                 return;
 290         }
 291
 292         if (child->event.flags != 0) {
 293                 return;         /* Busy */
 294         }
 295
 296         if ((child->pid == 0) && (!fork_domain_child(child))) {
 297                 /* Cancel all outstanding requests */
 298
 299                 while (request != NULL) {
 300                         /* request might be free'd in the continuation */
 301                         struct winbindd_async_request *next = request->next;
 302                         request->continuation(request->private_data, False);
 303                         request = next;
 304                 }
 305                 return;
 306         }
 307
 308         /* Now we know who we're sending to - remember the pid. */
 309         request->child_pid = child->pid;
 310
 311         setup_async_write(&child->event, request->request,
 312                           sizeof(*request->request),
 313                           async_main_request_sent, request);
 314
 315         return;
 316 }
 317
 318 struct domain_request_state {
 319         TALLOC_CTX *mem_ctx;
 320         struct winbindd_domain *domain;
 321         struct winbindd_request *request;
 322         struct winbindd_response *response;
 323         void (*continuation)(void *private_data_data, BOOL success);
 324         void *private_data_data;
 325 };
 326
 327 static void domain_init_recv(void *private_data_data, BOOL success);
 328
 329 void async_domain_request(TALLOC_CTX *mem_ctx,
 330                           struct winbindd_domain *domain,
 331                           struct winbindd_request *request,
 332                           struct winbindd_response *response,
 333                           void (*continuation)(void *private_data_data, BOOL success),
 334                           void *private_data_data)
 335 {
 336         struct domain_request_state *state;
 337
 338         if (domain->initialized) {
 339                 async_request(mem_ctx, &domain->child, request, response,
 340                               continuation, private_data_data);
 341                 return;
 342         }
 343
 344         state = TALLOC_P(mem_ctx, struct domain_request_state);
 345         if (state == NULL) {
 346                 DEBUG(0, ("talloc failed\n"));
 347                 continuation(private_data_data, False);
 348                 return;
 349         }
 350
 351         state->mem_ctx = mem_ctx;
 352         state->domain = domain;
 353         state->request = request;
 354         state->response = response;
 355         state->continuation = continuation;
 356         state->private_data_data = private_data_data;
 357
 358         init_child_connection(domain, domain_init_recv, state);
 359 }
 360
 361 static void recvfrom_child(void *private_data_data, BOOL success)
 362 {
 363         struct winbindd_cli_state *state =
 364                 talloc_get_type_abort(private_data_data, struct winbindd_cli_state);
 365         enum winbindd_result result = state->response.result;
 366
 367         /* This is an optimization: The child has written directly to the
 368          * response buffer. The request itself is still in pending state,
 369          * state that in the result code. */
 370
 371         state->response.result = WINBINDD_PENDING;
 372
 373         if ((!success) || (result != WINBINDD_OK)) {
 374                 request_error(state);
 375                 return;
 376         }
 377
 378         request_ok(state);
 379 }
 380
 381 void sendto_child(struct winbindd_cli_state *state,
 382                   struct winbindd_child *child)
 383 {
 384         async_request(state->mem_ctx, child, &state->request,
 385                       &state->response, recvfrom_child, state);
 386 }
 387
 388 void sendto_domain(struct winbindd_cli_state *state,
 389                    struct winbindd_domain *domain)
 390 {
 391         async_domain_request(state->mem_ctx, domain,
 392                              &state->request, &state->response,
 393                              recvfrom_child, state);
 394 }
 395
 396 static void domain_init_recv(void *private_data_data, BOOL success)
 397 {
 398         struct domain_request_state *state =
 399                 talloc_get_type_abort(private_data_data, struct domain_request_state);
 400
 401         if (!success) {
 402                 DEBUG(5, ("Domain init returned an error\n"));
 403                 state->continuation(state->private_data_data, False);
 404                 return;
 405         }
 406
 407         async_request(state->mem_ctx, &state->domain->child,
 408                       state->request, state->response,
 409                       state->continuation, state->private_data_data);
 410 }
 411
 412 struct winbindd_child_dispatch_table {
 413         enum winbindd_cmd cmd;
 414         enum winbindd_result (*fn)(struct winbindd_domain *domain,
 415                                    struct winbindd_cli_state *state);
 416         const char *winbindd_cmd_name;
 417 };
 418
 419 static struct winbindd_child_dispatch_table child_dispatch_table[] = {
 420
 421         { WINBINDD_LOOKUPSID,            winbindd_dual_lookupsid,             "LOOKUPSID" },
 422         { WINBINDD_LOOKUPNAME,           winbindd_dual_lookupname,            "LOOKUPNAME" },
 423         { WINBINDD_LOOKUPRIDS,           winbindd_dual_lookuprids,            "LOOKUPRIDS" },
 424         { WINBINDD_LIST_TRUSTDOM,        winbindd_dual_list_trusted_domains,  "LIST_TRUSTDOM" },
 425         { WINBINDD_INIT_CONNECTION,      winbindd_dual_init_connection,       "INIT_CONNECTION" },
 426         { WINBINDD_GETDCNAME,            winbindd_dual_getdcname,             "GETDCNAME" },
 427         { WINBINDD_SHOW_SEQUENCE,        winbindd_dual_show_sequence,         "SHOW_SEQUENCE" },
 428         { WINBINDD_PAM_AUTH,             winbindd_dual_pam_auth,              "PAM_AUTH" },
 429         { WINBINDD_PAM_AUTH_CRAP,        winbindd_dual_pam_auth_crap,         "AUTH_CRAP" },
 430         { WINBINDD_PAM_LOGOFF,           winbindd_dual_pam_logoff,            "PAM_LOGOFF" },
 431         { WINBINDD_PAM_CHNG_PSWD_AUTH_CRAP,winbindd_dual_pam_chng_pswd_auth_crap,"CHNG_PSWD_AUTH_CRAP" },
 432         { WINBINDD_PAM_CHAUTHTOK,        winbindd_dual_pam_chauthtok,         "PAM_CHAUTHTOK" },
 433         { WINBINDD_CHECK_MACHACC,        winbindd_dual_check_machine_acct,    "CHECK_MACHACC" },
 434         { WINBINDD_DUAL_SID2UID,         winbindd_dual_sid2uid,               "DUAL_SID2UID" },
 435         { WINBINDD_DUAL_SID2GID,         winbindd_dual_sid2gid,               "DUAL_SID2GID" },
 436 #if 0   /* DISABLED until we fix the interface in Samba 3.0.26 --jerry */
 437         { WINBINDD_DUAL_SIDS2XIDS,       winbindd_dual_sids2xids,             "DUAL_SIDS2XIDS" },
 438 #endif  /* end DISABLED */
 439         { WINBINDD_DUAL_UID2SID,         winbindd_dual_uid2sid,               "DUAL_UID2SID" },
 440         { WINBINDD_DUAL_GID2SID,         winbindd_dual_gid2sid,               "DUAL_GID2SID" },
 441         { WINBINDD_DUAL_UID2NAME,        winbindd_dual_uid2name,              "DUAL_UID2NAME" },
 442         { WINBINDD_DUAL_NAME2UID,        winbindd_dual_name2uid,              "DUAL_NAME2UID" },
 443         { WINBINDD_DUAL_GID2NAME,        winbindd_dual_gid2name,              "DUAL_GID2NAME" },
 444         { WINBINDD_DUAL_NAME2GID,        winbindd_dual_name2gid,              "DUAL_NAME2GID" },
 445         { WINBINDD_DUAL_SET_MAPPING,     winbindd_dual_set_mapping,           "DUAL_SET_MAPPING" },
 446         { WINBINDD_DUAL_SET_HWM,         winbindd_dual_set_hwm,               "DUAL_SET_HWMS" },
 447         { WINBINDD_DUAL_DUMP_MAPS,       winbindd_dual_dump_maps,             "DUAL_DUMP_MAPS" },
 448         { WINBINDD_DUAL_USERINFO,        winbindd_dual_userinfo,              "DUAL_USERINFO" },
 449         { WINBINDD_ALLOCATE_UID,         winbindd_dual_allocate_uid,          "ALLOCATE_UID" },
 450         { WINBINDD_ALLOCATE_GID,         winbindd_dual_allocate_gid,          "ALLOCATE_GID" },
 451         { WINBINDD_GETUSERDOMGROUPS,     winbindd_dual_getuserdomgroups,      "GETUSERDOMGROUPS" },
 452         { WINBINDD_DUAL_GETSIDALIASES,   winbindd_dual_getsidaliases,         "GETSIDALIASES" },
 453         { WINBINDD_CCACHE_NTLMAUTH,      winbindd_dual_ccache_ntlm_auth,      "CCACHE_NTLM_AUTH" },
 454         /* End of list */
 455
 456         { WINBINDD_NUM_CMDS, NULL, "NONE" }
 457 };
 458
 459 static void child_process_request(struct winbindd_domain *domain,
 460                                   struct winbindd_cli_state *state)
 461 {
 462         struct winbindd_child_dispatch_table *table;
 463
 464         /* Free response data - we may be interrupted and receive another
 465            command before being able to send this data off. */
 466
 467         state->response.result = WINBINDD_ERROR;
 468         state->response.length = sizeof(struct winbindd_response);
 469
 470         state->mem_ctx = talloc_init("winbind request");
 471         if (state->mem_ctx == NULL)
 472                 return;
 473
 474         /* Process command */
 475
 476         for (table = child_dispatch_table; table->fn; table++) {
 477                 if (state->request.cmd == table->cmd) {
 478                         DEBUG(10,("process_request: request fn %s\n",
 479                                   table->winbindd_cmd_name ));
 480                         state->response.result = table->fn(domain, state);
 481                         break;
 482                 }
 483         }
 484
 485         if (!table->fn) {
 486                 DEBUG(10,("process_request: unknown request fn number %d\n",
 487                           (int)state->request.cmd ));
 488                 state->response.result = WINBINDD_ERROR;
 489         }
 490
 491         talloc_destroy(state->mem_ctx);
 492 }
 493
 494 void setup_domain_child(struct winbindd_domain *domain,
 495                         struct winbindd_child *child,
 496                         const char *explicit_logfile)
 497 {
 498         if (explicit_logfile != NULL) {
 499                 pstr_sprintf(child->logfilename, "%s/log.winbindd-%s",
 500                              dyn_LOGFILEBASE, explicit_logfile);
 501         } else if (domain != NULL) {
 502                 pstr_sprintf(child->logfilename, "%s/log.wb-%s",
 503                              dyn_LOGFILEBASE, domain->name);
 504         } else {
 505                 smb_panic("Internal error: domain == NULL && "
 506                           "explicit_logfile == NULL");
 507         }
 508
 509         child->domain = domain;
 510 }
 511
 512 struct winbindd_child *children = NULL;
 513
 514 void winbind_child_died(pid_t pid)
 515 {
 516         struct winbindd_child *child;
 517
 518         for (child = children; child != NULL; child = child->next) {
 519                 if (child->pid == pid) {
 520                         break;
 521                 }
 522         }
 523
 524         if (child == NULL) {
 525                 DEBUG(5, ("Already reaped child %u died\n", (unsigned int)pid));
 526                 return;
 527         }
 528
 529         remove_fd_event(&child->event);
 530         close(child->event.fd);
 531         child->event.fd = 0;
 532         child->event.flags = 0;
 533         child->pid = 0;
 534
 535         schedule_async_request(child);
 536 }
 537
 538 /* Ensure any negative cache entries with the netbios or realm names are removed. */
 539
 540 void winbindd_flush_negative_conn_cache(struct winbindd_domain *domain)
 541 {
 542         flush_negative_conn_cache_for_domain(domain->name);
 543         if (*domain->alt_name) {
 544                 flush_negative_conn_cache_for_domain(domain->alt_name);
 545         }
 546 }
 547
 548 /* Set our domains as offline and forward the offline message to our children. */
 549
 550 void winbind_msg_offline(int msg_type, struct process_id src,
 551                          void *buf, size_t len, void *private_data)
 552 {
 553         struct winbindd_child *child;
 554         struct winbindd_domain *domain;
 555
 556         DEBUG(10,("winbind_msg_offline: got offline message.\n"));
 557
 558         if (!lp_winbind_offline_logon()) {
 559                 DEBUG(10,("winbind_msg_offline: rejecting offline message.\n"));
 560                 return;
 561         }
 562
 563         /* Set our global state as offline. */
 564         if (!set_global_winbindd_state_offline()) {
 565                 DEBUG(10,("winbind_msg_offline: offline request failed.\n"));
 566                 return;
 567         }
 568
 569         /* Set all our domains as offline. */
 570         for (domain = domain_list(); domain; domain = domain->next) {
 571                 if (domain->internal) {
 572                         continue;
 573                 }
 574                 DEBUG(5,("winbind_msg_offline: marking %s offline.\n", domain->name));
 575                 set_domain_offline(domain);
 576
 577                 /* Send an offline message to the idmap child when our
 578                    primary domain goes offline */
 579
 580                 if ( domain->primary ) {
 581                         struct winbindd_child *idmap = idmap_child();
 582
 583                         if ( idmap->pid != 0 ) {
 584                                 message_send_pid(pid_to_procid(idmap->pid),
 585                                                  MSG_WINBIND_OFFLINE,
 586                                                  domain->name,
 587                                                  strlen(domain->name)+1,
 588                                                  False);
 589                         }
 590                 }
 591         }
 592
 593         for (child = children; child != NULL; child = child->next) {
 594                 /* Don't send message to idmap child.  We've already
 595                    done so above. */
 596                 if (!child->domain || (child == idmap_child())) {
 597                         continue;
 598                 }
 599
 600                 /* Or internal domains (this should not be possible....) */
 601                 if (child->domain->internal) {
 602                         continue;
 603                 }
 604
 605                 /* Each winbindd child should only process requests for one domain - make sure
 606                    we only set it online / offline for that domain. */
 607
 608                 DEBUG(10,("winbind_msg_offline: sending message to pid %u for domain %s.\n",
 609                         (unsigned int)child->pid, domain->name ));
 610
 611                 message_send_pid(pid_to_procid(child->pid), MSG_WINBIND_OFFLINE, child->domain->name,
 612                         strlen(child->domain->name)+1, False);
 613         }
 614 }
 615
 616 /* Set our domains as online and forward the online message to our children. */
 617
 618 void winbind_msg_online(int msg_type, struct process_id src,
 619                         void *buf, size_t len, void *private_data)
 620 {
 621         struct winbindd_child *child;
 622         struct winbindd_domain *domain;
 623
 624         DEBUG(10,("winbind_msg_online: got online message.\n"));
 625
 626         if (!lp_winbind_offline_logon()) {
 627                 DEBUG(10,("winbind_msg_online: rejecting online message.\n"));
 628                 return;
 629         }
 630
 631         /* Set our global state as online. */
 632         set_global_winbindd_state_online();
 633
 634         smb_nscd_flush_user_cache();
 635         smb_nscd_flush_group_cache();
 636
 637         /* Set all our domains as online. */
 638         for (domain = domain_list(); domain; domain = domain->next) {
 639                 if (domain->internal) {
 640                         continue;
 641                 }
 642                 DEBUG(5,("winbind_msg_online: requesting %s to go online.\n", domain->name));
 643
 644                 winbindd_flush_negative_conn_cache(domain);
 645                 set_domain_online_request(domain);
 646
 647                 /* Send an online message to the idmap child when our
 648                    primary domain comes back online */
 649
 650                 if ( domain->primary ) {
 651                         struct winbindd_child *idmap = idmap_child();
 652
 653                         if ( idmap->pid != 0 ) {
 654                                 message_send_pid(pid_to_procid(idmap->pid),
 655                                                  MSG_WINBIND_ONLINE,
 656                                                  domain->name,
 657                                                  strlen(domain->name)+1,
 658                                                  False);
 659                         }
 660
 661                 }
 662         }
 663
 664         for (child = children; child != NULL; child = child->next) {
 665                 /* Don't send message to idmap child. */
 666                 if (!child->domain || (child == idmap_child())) {
 667                         continue;
 668                 }
 669
 670                 /* Or internal domains (this should not be possible....) */
 671                 if (child->domain->internal) {
 672                         continue;
 673                 }
 674
 675                 /* Each winbindd child should only process requests for one domain - make sure
 676                    we only set it online / offline for that domain. */
 677
 678                 DEBUG(10,("winbind_msg_online: sending message to pid %u for domain %s.\n",
 679                         (unsigned int)child->pid, child->domain->name ));
 680
 681                 message_send_pid(pid_to_procid(child->pid), MSG_WINBIND_ONLINE, child->domain->name,
 682                         strlen(child->domain->name)+1, False);
 683         }
 684 }
 685
 686 /* Forward the online/offline messages to our children. */
 687 void winbind_msg_onlinestatus(int msg_type, struct process_id src,
 688                               void *buf, size_t len, void *private_data)
 689 {
 690         struct winbindd_child *child;
 691
 692         DEBUG(10,("winbind_msg_onlinestatus: got onlinestatus message.\n"));
 693
 694         for (child = children; child != NULL; child = child->next) {
 695                 if (child->domain && child->domain->primary) {
 696                         DEBUG(10,("winbind_msg_onlinestatus: "
 697                                   "sending message to pid %u of primary domain.\n",
 698                                   (unsigned int)child->pid));
 699                         message_send_pid(pid_to_procid(child->pid),
 700                                          MSG_WINBIND_ONLINESTATUS, buf, len, False);
 701                         break;
 702                 }
 703         }
 704 }
 705
 706
 707 static void account_lockout_policy_handler(struct event_context *ctx,
 708                                            struct timed_event *te,
 709                                            const struct timeval *now,
 710                                            void *private_data)
 711 {
 712         struct winbindd_child *child =
 713                 (struct winbindd_child *)private_data;
 714         TALLOC_CTX *mem_ctx = NULL;
 715         struct winbindd_methods *methods;
 716         SAM_UNK_INFO_12 lockout_policy;
 717         NTSTATUS result;
 718
 719         DEBUG(10,("account_lockout_policy_handler called\n"));
 720
 721         TALLOC_FREE(child->lockout_policy_event);
 722
 723         methods = child->domain->methods;
 724
 725         mem_ctx = talloc_init("account_lockout_policy_handler ctx");
 726         if (!mem_ctx) {
 727                 result = NT_STATUS_NO_MEMORY;
 728         } else {
 729                 result = methods->lockout_policy(child->domain, mem_ctx, &lockout_policy);
 730         }
 731
 732         talloc_destroy(mem_ctx);
 733
 734         if (!NT_STATUS_IS_OK(result)) {
 735                 DEBUG(10,("account_lockout_policy_handler: lockout_policy failed error %s\n",
 736                          nt_errstr(result)));
 737         }
 738
 739         child->lockout_policy_event = event_add_timed(winbind_event_context(), NULL,
 740                                                       timeval_current_ofs(3600, 0),
 741                                                       "account_lockout_policy_handler",
 742                                                       account_lockout_policy_handler,
 743                                                       child);
 744 }
 745
 746 /* Deal with a request to go offline. */
 747
 748 static void child_msg_offline(int msg_type, struct process_id src,
 749                               void *buf, size_t len, void *private_data)
 750 {
 751         struct winbindd_domain *domain;
 752         const char *domainname = (const char *)buf;
 753
 754         if (buf == NULL || len == 0) {
 755                 return;
 756         }
 757
 758         DEBUG(5,("child_msg_offline received for domain %s.\n", domainname));
 759
 760         if (!lp_winbind_offline_logon()) {
 761                 DEBUG(10,("child_msg_offline: rejecting offline message.\n"));
 762                 return;
 763         }
 764
 765         /* Set our global state as offline. */
 766         if (!set_global_winbindd_state_offline()) {
 767                 DEBUG(10,("child_msg_offline: offline request failed.\n"));
 768                 return;
 769         }
 770
 771         /* Mark the requested domain offline. */
 772
 773         for (domain = domain_list(); domain; domain = domain->next) {
 774                 if (domain->internal) {
 775                         continue;
 776                 }
 777                 if (strequal(domain->name, domainname)) {
 778                         DEBUG(5,("child_msg_offline: marking %s offline.\n", domain->name));
 779                         set_domain_offline(domain);
 780                 }
 781         }
 782 }
 783
 784 /* Deal with a request to go online. */
 785
 786 static void child_msg_online(int msg_type, struct process_id src,
 787                              void *buf, size_t len, void *private_data)
 788 {
 789         struct winbindd_domain *domain;
 790         const char *domainname = (const char *)buf;
 791
 792         if (buf == NULL || len == 0) {
 793                 return;
 794         }
 795
 796         DEBUG(5,("child_msg_online received for domain %s.\n", domainname));
 797
 798         if (!lp_winbind_offline_logon()) {
 799                 DEBUG(10,("child_msg_online: rejecting online message.\n"));
 800                 return;
 801         }
 802
 803         /* Set our global state as online. */
 804         set_global_winbindd_state_online();
 805
 806         /* Try and mark everything online - delete any negative cache entries
 807            to force a reconnect now. */
 808
 809         for (domain = domain_list(); domain; domain = domain->next) {
 810                 if (domain->internal) {
 811                         continue;
 812                 }
 813                 if (strequal(domain->name, domainname)) {
 814                         DEBUG(5,("child_msg_online: requesting %s to go online.\n", domain->name));
 815                         winbindd_flush_negative_conn_cache(domain);
 816                         set_domain_online_request(domain);
 817                 }
 818         }
 819 }
 820
 821 static const char *collect_onlinestatus(TALLOC_CTX *mem_ctx)
 822 {
 823         struct winbindd_domain *domain;
 824         char *buf = NULL;
 825
 826         if ((buf = talloc_asprintf(mem_ctx, "global:%s ",
 827                                    get_global_winbindd_state_offline() ?
 828                                    "Offline":"Online")) == NULL) {
 829                 return NULL;
 830         }
 831
 832         for (domain = domain_list(); domain; domain = domain->next) {
 833                 if ((buf = talloc_asprintf_append(buf, "%s:%s ",
 834                                                   domain->name,
 835                                                   domain->online ?
 836                                                   "Online":"Offline")) == NULL) {
 837                         return NULL;
 838                 }
 839         }
 840
 841         buf = talloc_asprintf_append(buf, "\n");
 842
 843         DEBUG(5,("collect_onlinestatus: %s", buf));
 844
 845         return buf;
 846 }
 847
 848 static void child_msg_onlinestatus(int msg_type, struct process_id src,
 849                                    void *buf, size_t len, void *private_data)
 850 {
 851         TALLOC_CTX *mem_ctx;
 852         const char *message;
 853         struct process_id *sender;
 854
 855         DEBUG(5,("winbind_msg_onlinestatus received.\n"));
 856
 857         if (!buf) {
 858                 return;
 859         }
 860
 861         sender = (struct process_id *)buf;
 862
 863         mem_ctx = talloc_init("winbind_msg_onlinestatus");
 864         if (mem_ctx == NULL) {
 865                 return;
 866         }
 867
 868         message = collect_onlinestatus(mem_ctx);
 869         if (message == NULL) {
 870                 talloc_destroy(mem_ctx);
 871                 return;
 872         }
 873
 874         message_send_pid(*sender, MSG_WINBIND_ONLINESTATUS,
 875                          message, strlen(message) + 1, True);
 876
 877         talloc_destroy(mem_ctx);
 878 }
 879
 880 static BOOL fork_domain_child(struct winbindd_child *child)
 881 {
 882         int fdpair[2];
 883         struct winbindd_cli_state state;
 884         struct winbindd_domain *domain;
 885
 886         if (socketpair(AF_UNIX, SOCK_STREAM, 0, fdpair) != 0) {
 887                 DEBUG(0, ("Could not open child pipe: %s\n",
 888                           strerror(errno)));
 889                 return False;
 890         }
 891
 892         ZERO_STRUCT(state);
 893         state.pid = sys_getpid();
 894
 895         /* Stop zombies */
 896         CatchChild();
 897
 898         /* Ensure we don't process messages whilst we're
 899            changing the disposition for the child. */
 900         message_block();
 901
 902         child->pid = sys_fork();
 903
 904         if (child->pid == -1) {
 905                 DEBUG(0, ("Could not fork: %s\n", strerror(errno)));
 906                 message_unblock();
 907                 return False;
 908         }
 909
 910         if (child->pid != 0) {
 911                 /* Parent */
 912                 close(fdpair[0]);
 913                 child->next = child->prev = NULL;
 914                 DLIST_ADD(children, child);
 915                 child->event.fd = fdpair[1];
 916                 child->event.flags = 0;
 917                 child->requests = NULL;
 918                 add_fd_event(&child->event);
 919                 /* We're ok with online/offline messages now. */
 920                 message_unblock();
 921                 return True;
 922         }
 923
 924         /* Child */
 925
 926         state.sock = fdpair[0];
 927         close(fdpair[1]);
 928
 929         /* tdb needs special fork handling */
 930         if (tdb_reopen_all(1) == -1) {
 931                 DEBUG(0,("tdb_reopen_all failed.\n"));
 932                 _exit(0);
 933         }
 934
 935         close_conns_after_fork();
 936
 937         if (!override_logfile) {
 938                 lp_set_logfile(child->logfilename);
 939                 reopen_logs();
 940         }
 941
 942         /* Don't handle the same messages as our parent. */
 943         message_deregister(MSG_SMB_CONF_UPDATED);
 944         message_deregister(MSG_SHUTDOWN);
 945         message_deregister(MSG_WINBIND_OFFLINE);
 946         message_deregister(MSG_WINBIND_ONLINE);
 947         message_deregister(MSG_WINBIND_ONLINESTATUS);
 948
 949         /* The child is ok with online/offline messages now. */
 950         message_unblock();
 951
 952         /* Handle online/offline messages. */
 953         message_register(MSG_WINBIND_OFFLINE, child_msg_offline, NULL);
 954         message_register(MSG_WINBIND_ONLINE, child_msg_online, NULL);
 955         message_register(MSG_WINBIND_ONLINESTATUS, child_msg_onlinestatus,
 956                          NULL);
 957
 958         if ( child->domain ) {
 959                 child->domain->startup = True;
 960                 child->domain->startup_time = time(NULL);
 961         }
 962
 963         /* Ensure we have no pending check_online events other
 964            than one for this domain. */
 965
 966         for (domain = domain_list(); domain; domain = domain->next) {
 967                 if (domain != child->domain) {
 968                         TALLOC_FREE(domain->check_online_event);
 969                 }
 970         }
 971
 972         /* Ensure we're not handling an event inherited from
 973            our parent. */
 974
 975         cancel_named_event(winbind_event_context(),
 976                            "krb5_ticket_refresh_handler");
 977
 978         /* We might be in the idmap child...*/
 979         if (child->domain && !(child->domain->internal) &&
 980             lp_winbind_offline_logon()) {
 981
 982                 set_domain_online_request(child->domain);
 983
 984                 child->lockout_policy_event = event_add_timed(
 985                         winbind_event_context(), NULL, timeval_zero(),
 986                         "account_lockout_policy_handler",
 987                         account_lockout_policy_handler,
 988                         child);
 989         }
 990
 991         while (1) {
 992
 993                 int ret;
 994                 fd_set read_fds;
 995                 struct timeval t;
 996                 struct timeval *tp;
 997                 struct timeval now;
 998
 999                 /* free up any talloc memory */
1000                 lp_TALLOC_FREE();
1001                 main_loop_TALLOC_FREE();
1002
1003                 run_events(winbind_event_context(), 0, NULL, NULL);
1004
1005                 GetTimeOfDay(&now);
1006
1007                 if (child->domain && child->domain->startup &&
1008                                 (now.tv_sec > child->domain->startup_time + 30)) {
1009                         /* No longer in "startup" mode. */
1010                         DEBUG(10,("fork_domain_child: domain %s no longer in 'startup' mode.\n",
1011                                 child->domain->name ));
1012                         child->domain->startup = False;
1013                 }
1014
1015                 tp = get_timed_events_timeout(winbind_event_context(), &t);
1016                 if (tp) {
1017                         DEBUG(11,("select will use timeout of %u.%u seconds\n",
1018                                 (unsigned int)tp->tv_sec, (unsigned int)tp->tv_usec ));
1019                 }
1020
1021                 /* Handle messages */
1022
1023                 message_dispatch();
1024
1025                 FD_ZERO(&read_fds);
1026                 FD_SET(state.sock, &read_fds);
1027
1028                 ret = sys_select(state.sock + 1, &read_fds, NULL, NULL, tp);
1029
1030                 if (ret == 0) {
1031                         DEBUG(11,("nothing is ready yet, continue\n"));
1032                         continue;
1033                 }
1034
1035                 if (ret == -1 && errno == EINTR) {
1036                         /* We got a signal - continue. */
1037                         continue;
1038                 }
1039
1040                 if (ret == -1 && errno != EINTR) {
1041                         DEBUG(0,("select error occured\n"));
1042                         perror("select");
1043                         return False;
1044                 }
1045
1046                 /* fetch a request from the main daemon */
1047                 child_read_request(&state);
1048
1049                 if (state.finished) {
1050                         /* we lost contact with our parent */
1051                         exit(0);
1052                 }
1053
1054                 DEBUG(4,("child daemon request %d\n", (int)state.request.cmd));
1055
1056                 ZERO_STRUCT(state.response);
1057                 state.request.null_term = '\0';
1058                 child_process_request(child->domain, &state);
1059
1060                 SAFE_FREE(state.request.extra_data.data);
1061
1062                 cache_store_response(sys_getpid(), &state.response);
1063
1064                 SAFE_FREE(state.response.extra_data.data);
1065
1066                 /* We just send the result code back, the result
1067                  * structure needs to be fetched via the
1068                  * winbindd_cache. Hmm. That needs fixing... */
1069
1070                 if (write_data(state.sock,
1071                                (const char *)&state.response.result,
1072                                sizeof(state.response.result)) !=
1073                     sizeof(state.response.result)) {
1074                         DEBUG(0, ("Could not write result\n"));
1075                         exit(1);
1076                 }
1077         }
1078 }