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