r22895: Convert some more calls from message_send_buf to messaging_send_buf
[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                 messaging_send_buf(winbind_messaging_context(),
528                                    pid_to_procid(child->pid),
529                                    MSG_WINBIND_OFFLINE,
530                                    (uint8 *)child->domain->name,
531                                    strlen(child->domain->name)+1);
532         }
533 }
534
535 /* Set our domains as online and forward the online message to our children. */
536
537 void winbind_msg_online(int msg_type, struct server_id src,
538                         void *buf, size_t len, void *private_data)
539 {
540         struct winbindd_child *child;
541         struct winbindd_domain *domain;
542
543         DEBUG(10,("winbind_msg_online: got online message.\n"));
544
545         if (!lp_winbind_offline_logon()) {
546                 DEBUG(10,("winbind_msg_online: rejecting online message.\n"));
547                 return;
548         }
549
550         /* Set our global state as online. */
551         set_global_winbindd_state_online();
552
553         smb_nscd_flush_user_cache();
554         smb_nscd_flush_group_cache();
555
556         /* Set all our domains as online. */
557         for (domain = domain_list(); domain; domain = domain->next) {
558                 if (domain->internal) {
559                         continue;
560                 }
561                 DEBUG(5,("winbind_msg_online: requesting %s to go online.\n", domain->name));
562
563                 winbindd_flush_negative_conn_cache(domain);
564                 set_domain_online_request(domain);
565
566                 /* Send an online message to the idmap child when our
567                    primary domain comes back online */
568
569                 if ( domain->primary ) {
570                         struct winbindd_child *idmap = idmap_child();
571                         
572                         if ( idmap->pid != 0 ) {
573                                 messaging_send_buf(winbind_messaging_context(),
574                                                    pid_to_procid(idmap->pid), 
575                                                    MSG_WINBIND_ONLINE,
576                                                    (uint8 *)domain->name,
577                                                    strlen(domain->name)+1);
578                         }
579                         
580                 }
581         }
582
583         for (child = children; child != NULL; child = child->next) {
584                 /* Don't send message to idmap child. */
585                 if (!child->domain || (child == idmap_child())) {
586                         continue;
587                 }
588
589                 /* Or internal domains (this should not be possible....) */
590                 if (child->domain->internal) {
591                         continue;
592                 }
593
594                 /* Each winbindd child should only process requests for one domain - make sure
595                    we only set it online / offline for that domain. */
596
597                 DEBUG(10,("winbind_msg_online: sending message to pid %u for domain %s.\n",
598                         (unsigned int)child->pid, child->domain->name ));
599
600                 messaging_send_buf(winbind_messaging_context(),
601                                    pid_to_procid(child->pid),
602                                    MSG_WINBIND_ONLINE,
603                                    (uint8 *)child->domain->name,
604                                    strlen(child->domain->name)+1);
605         }
606 }
607
608 /* Forward the online/offline messages to our children. */
609 void winbind_msg_onlinestatus(int msg_type, struct server_id src,
610                               void *buf, size_t len, void *private_data)
611 {
612         struct winbindd_child *child;
613
614         DEBUG(10,("winbind_msg_onlinestatus: got onlinestatus message.\n"));
615
616         for (child = children; child != NULL; child = child->next) {
617                 if (child->domain && child->domain->primary) {
618                         DEBUG(10,("winbind_msg_onlinestatus: "
619                                   "sending message to pid %u of primary domain.\n",
620                                   (unsigned int)child->pid));
621                         messaging_send_buf(winbind_messaging_context(),
622                                            pid_to_procid(child->pid), 
623                                            MSG_WINBIND_ONLINESTATUS,
624                                            (uint8 *)buf, len);
625                         break;
626                 }
627         }
628 }
629
630
631 static void account_lockout_policy_handler(struct event_context *ctx,
632                                            struct timed_event *te,
633                                            const struct timeval *now,
634                                            void *private_data)
635 {
636         struct winbindd_child *child =
637                 (struct winbindd_child *)private_data;
638         TALLOC_CTX *mem_ctx = NULL;
639         struct winbindd_methods *methods;
640         SAM_UNK_INFO_12 lockout_policy;
641         NTSTATUS result;
642
643         DEBUG(10,("account_lockout_policy_handler called\n"));
644
645         if (child->lockout_policy_event) {
646                 TALLOC_FREE(child->lockout_policy_event);
647         }
648
649         if ( !winbindd_can_contact_domain( child->domain ) ) {
650                 DEBUG(10,("account_lockout_policy_handler: Removing myself since I "
651                           "do not have an incoming trust to domain %s\n", 
652                           child->domain->name));
653
654                 return;         
655         }
656
657         methods = child->domain->methods;
658
659         mem_ctx = talloc_init("account_lockout_policy_handler ctx");
660         if (!mem_ctx) {
661                 result = NT_STATUS_NO_MEMORY;
662         } else {
663                 result = methods->lockout_policy(child->domain, mem_ctx, &lockout_policy);
664         }
665
666         talloc_destroy(mem_ctx);
667
668         if (!NT_STATUS_IS_OK(result)) {
669                 DEBUG(10,("account_lockout_policy_handler: lockout_policy failed error %s\n",
670                          nt_errstr(result)));
671         }
672
673         child->lockout_policy_event = event_add_timed(winbind_event_context(), NULL,
674                                                       timeval_current_ofs(3600, 0),
675                                                       "account_lockout_policy_handler",
676                                                       account_lockout_policy_handler,
677                                                       child);
678 }
679
680 /* Deal with a request to go offline. */
681
682 static void child_msg_offline(int msg_type, struct server_id src,
683                               void *buf, size_t len, void *private_data)
684 {
685         struct winbindd_domain *domain;
686         const char *domainname = (const char *)buf;
687
688         if (buf == NULL || len == 0) {
689                 return;
690         }
691
692         DEBUG(5,("child_msg_offline received for domain %s.\n", domainname));
693
694         if (!lp_winbind_offline_logon()) {
695                 DEBUG(10,("child_msg_offline: rejecting offline message.\n"));
696                 return;
697         }
698
699         /* Mark the requested domain offline. */
700
701         for (domain = domain_list(); domain; domain = domain->next) {
702                 if (domain->internal) {
703                         continue;
704                 }
705                 if (strequal(domain->name, domainname)) {
706                         DEBUG(5,("child_msg_offline: marking %s offline.\n", domain->name));
707                         set_domain_offline(domain);
708                 }
709         }
710 }
711
712 /* Deal with a request to go online. */
713
714 static void child_msg_online(int msg_type, struct server_id src,
715                              void *buf, size_t len, void *private_data)
716 {
717         struct winbindd_domain *domain;
718         const char *domainname = (const char *)buf;
719
720         if (buf == NULL || len == 0) {
721                 return;
722         }
723
724         DEBUG(5,("child_msg_online received for domain %s.\n", domainname));
725
726         if (!lp_winbind_offline_logon()) {
727                 DEBUG(10,("child_msg_online: rejecting online message.\n"));
728                 return;
729         }
730
731         /* Set our global state as online. */
732         set_global_winbindd_state_online();
733
734         /* Try and mark everything online - delete any negative cache entries
735            to force a reconnect now. */
736
737         for (domain = domain_list(); domain; domain = domain->next) {
738                 if (domain->internal) {
739                         continue;
740                 }
741                 if (strequal(domain->name, domainname)) {
742                         DEBUG(5,("child_msg_online: requesting %s to go online.\n", domain->name));
743                         winbindd_flush_negative_conn_cache(domain);
744                         set_domain_online_request(domain);
745                 }
746         }
747 }
748
749 static const char *collect_onlinestatus(TALLOC_CTX *mem_ctx)
750 {
751         struct winbindd_domain *domain;
752         char *buf = NULL;
753
754         if ((buf = talloc_asprintf(mem_ctx, "global:%s ", 
755                                    get_global_winbindd_state_offline() ? 
756                                    "Offline":"Online")) == NULL) {
757                 return NULL;
758         }
759
760         for (domain = domain_list(); domain; domain = domain->next) {
761                 if ((buf = talloc_asprintf_append(buf, "%s:%s ", 
762                                                   domain->name, 
763                                                   domain->online ?
764                                                   "Online":"Offline")) == NULL) {
765                         return NULL;
766                 }
767         }
768
769         buf = talloc_asprintf_append(buf, "\n");
770
771         DEBUG(5,("collect_onlinestatus: %s", buf));
772
773         return buf;
774 }
775
776 static void child_msg_onlinestatus(int msg_type, struct server_id src,
777                                    void *buf, size_t len, void *private_data)
778 {
779         TALLOC_CTX *mem_ctx;
780         const char *message;
781         struct server_id *sender;
782         
783         DEBUG(5,("winbind_msg_onlinestatus received.\n"));
784
785         if (!buf) {
786                 return;
787         }
788
789         sender = (struct server_id *)buf;
790
791         mem_ctx = talloc_init("winbind_msg_onlinestatus");
792         if (mem_ctx == NULL) {
793                 return;
794         }
795         
796         message = collect_onlinestatus(mem_ctx);
797         if (message == NULL) {
798                 talloc_destroy(mem_ctx);
799                 return;
800         }
801
802         messaging_send_buf(winbind_messaging_context(),
803                            *sender, MSG_WINBIND_ONLINESTATUS, 
804                            (uint8 *)message, strlen(message) + 1);
805
806         talloc_destroy(mem_ctx);
807 }
808
809 static BOOL fork_domain_child(struct winbindd_child *child)
810 {
811         int fdpair[2];
812         struct winbindd_cli_state state;
813         struct winbindd_domain *domain;
814
815         if (socketpair(AF_UNIX, SOCK_STREAM, 0, fdpair) != 0) {
816                 DEBUG(0, ("Could not open child pipe: %s\n",
817                           strerror(errno)));
818                 return False;
819         }
820
821         ZERO_STRUCT(state);
822         state.pid = sys_getpid();
823
824         /* Stop zombies */
825         CatchChild();
826
827         /* Ensure we don't process messages whilst we're
828            changing the disposition for the child. */
829         message_block();
830
831         child->pid = sys_fork();
832
833         if (child->pid == -1) {
834                 DEBUG(0, ("Could not fork: %s\n", strerror(errno)));
835                 message_unblock();
836                 return False;
837         }
838
839         if (child->pid != 0) {
840                 /* Parent */
841                 close(fdpair[0]);
842                 child->next = child->prev = NULL;
843                 DLIST_ADD(children, child);
844                 child->event.fd = fdpair[1];
845                 child->event.flags = 0;
846                 child->requests = NULL;
847                 add_fd_event(&child->event);
848                 /* We're ok with online/offline messages now. */
849                 message_unblock();
850                 return True;
851         }
852
853         /* Child */
854
855         state.sock = fdpair[0];
856         close(fdpair[1]);
857
858         /* tdb needs special fork handling */
859         if (tdb_reopen_all(1) == -1) {
860                 DEBUG(0,("tdb_reopen_all failed.\n"));
861                 _exit(0);
862         }
863
864         close_conns_after_fork();
865
866         if (!override_logfile) {
867                 lp_set_logfile(child->logfilename);
868                 reopen_logs();
869         }
870
871         /* Don't handle the same messages as our parent. */
872         message_deregister(MSG_SMB_CONF_UPDATED);
873         message_deregister(MSG_SHUTDOWN);
874         message_deregister(MSG_WINBIND_OFFLINE);
875         message_deregister(MSG_WINBIND_ONLINE);
876         message_deregister(MSG_WINBIND_ONLINESTATUS);
877
878         /* The child is ok with online/offline messages now. */
879         message_unblock();
880
881         /* Handle online/offline messages. */
882         message_register(MSG_WINBIND_OFFLINE, child_msg_offline, NULL);
883         message_register(MSG_WINBIND_ONLINE, child_msg_online, NULL);
884         message_register(MSG_WINBIND_ONLINESTATUS, child_msg_onlinestatus,
885                          NULL);
886
887         if ( child->domain ) {
888                 child->domain->startup = True;
889                 child->domain->startup_time = time(NULL);
890         }
891
892         /* Ensure we have no pending check_online events other
893            than one for this domain. */
894
895         for (domain = domain_list(); domain; domain = domain->next) {
896                 if (domain != child->domain) {
897                         if (domain->check_online_event) {
898                                 TALLOC_FREE(domain->check_online_event);
899                         }
900                 }
901         }
902
903         /* Ensure we're not handling an event inherited from
904            our parent. */
905
906         cancel_named_event(winbind_event_context(),
907                            "krb5_ticket_refresh_handler");
908
909         /* We might be in the idmap child...*/
910         if (child->domain && !(child->domain->internal) &&
911             lp_winbind_offline_logon()) {
912
913                 set_domain_online_request(child->domain);
914
915                 child->lockout_policy_event = event_add_timed(
916                         winbind_event_context(), NULL, timeval_zero(),
917                         "account_lockout_policy_handler",
918                         account_lockout_policy_handler,
919                         child);
920         }
921
922         while (1) {
923
924                 int ret;
925                 fd_set read_fds;
926                 struct timeval t;
927                 struct timeval *tp;
928                 struct timeval now;
929
930                 /* free up any talloc memory */
931                 lp_TALLOC_FREE();
932                 main_loop_TALLOC_FREE();
933
934                 run_events(winbind_event_context(), 0, NULL, NULL);
935
936                 GetTimeOfDay(&now);
937
938                 if (child->domain && child->domain->startup &&
939                                 (now.tv_sec > child->domain->startup_time + 30)) {
940                         /* No longer in "startup" mode. */
941                         DEBUG(10,("fork_domain_child: domain %s no longer in 'startup' mode.\n",
942                                 child->domain->name ));
943                         child->domain->startup = False;
944                 }
945
946                 tp = get_timed_events_timeout(winbind_event_context(), &t);
947                 if (tp) {
948                         DEBUG(11,("select will use timeout of %u.%u seconds\n",
949                                 (unsigned int)tp->tv_sec, (unsigned int)tp->tv_usec ));
950                 }
951
952                 /* Handle messages */
953
954                 message_dispatch();
955
956                 FD_ZERO(&read_fds);
957                 FD_SET(state.sock, &read_fds);
958
959                 ret = sys_select(state.sock + 1, &read_fds, NULL, NULL, tp);
960
961                 if (ret == 0) {
962                         DEBUG(11,("nothing is ready yet, continue\n"));
963                         continue;
964                 }
965
966                 if (ret == -1 && errno == EINTR) {
967                         /* We got a signal - continue. */
968                         continue;
969                 }
970
971                 if (ret == -1 && errno != EINTR) {
972                         DEBUG(0,("select error occured\n"));
973                         perror("select");
974                         return False;
975                 }
976
977                 /* fetch a request from the main daemon */
978                 child_read_request(&state);
979
980                 if (state.finished) {
981                         /* we lost contact with our parent */
982                         exit(0);
983                 }
984
985                 DEBUG(4,("child daemon request %d\n", (int)state.request.cmd));
986
987                 ZERO_STRUCT(state.response);
988                 state.request.null_term = '\0';
989                 child_process_request(child->domain, &state);
990
991                 SAFE_FREE(state.request.extra_data.data);
992
993                 cache_store_response(sys_getpid(), &state.response);
994
995                 SAFE_FREE(state.response.extra_data.data);
996
997                 /* We just send the result code back, the result
998                  * structure needs to be fetched via the
999                  * winbindd_cache. Hmm. That needs fixing... */
1000
1001                 if (write_data(state.sock,
1002                                (const char *)&state.response.result,
1003                                sizeof(state.response.result)) !=
1004                     sizeof(state.response.result)) {
1005                         DEBUG(0, ("Could not write result\n"));
1006                         exit(1);
1007                 }
1008         }
1009 }