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