r14156: Fix coverity #114: free storage alloc'ed by sstring_sub()
[Samba/nascimento.git] / source3 / nsswitch / winbindd_dual.c
blob7864254c4656c6db77c71034c0596314928319ce
1 /*
2 Unix SMB/CIFS implementation.
4 Winbind child daemons
6 Copyright (C) Andrew Tridgell 2002
7 Copyright (C) Volker Lendecke 2004,2005
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.
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.
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.
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.
31 #include "includes.h"
32 #include "winbindd.h"
34 #undef DBGC_CLASS
35 #define DBGC_CLASS DBGC_WINBIND
37 /* Read some data from a client connection */
39 static void child_read_request(struct winbindd_cli_state *state)
41 ssize_t len;
43 /* Read data */
45 len = read_data(state->sock, (char *)&state->request,
46 sizeof(state->request));
48 if (len != sizeof(state->request)) {
49 DEBUG(0, ("Got invalid request length: %d\n", (int)len));
50 state->finished = True;
51 return;
54 if (state->request.extra_len == 0) {
55 state->request.extra_data = NULL;
56 return;
59 DEBUG(10, ("Need to read %d extra bytes\n", (int)state->request.extra_len));
61 state->request.extra_data =
62 SMB_MALLOC_ARRAY(char, state->request.extra_len + 1);
64 if (state->request.extra_data == NULL) {
65 DEBUG(0, ("malloc failed\n"));
66 state->finished = True;
67 return;
70 /* Ensure null termination */
71 state->request.extra_data[state->request.extra_len] = '\0';
73 len = read_data(state->sock, state->request.extra_data,
74 state->request.extra_len);
76 if (len != state->request.extra_len) {
77 DEBUG(0, ("Could not read extra data\n"));
78 state->finished = True;
79 return;
84 * Machinery for async requests sent to children. You set up a
85 * winbindd_request, select a child to query, and issue a async_request
86 * call. When the request is completed, the callback function you specified is
87 * called back with the private pointer you gave to async_request.
90 struct winbindd_async_request {
91 struct winbindd_async_request *next, *prev;
92 TALLOC_CTX *mem_ctx;
93 struct winbindd_child *child;
94 struct winbindd_request *request;
95 struct winbindd_response *response;
96 void (*continuation)(void *private_data, BOOL success);
97 void *private_data;
100 static void async_main_request_sent(void *private_data, BOOL success);
101 static void async_request_sent(void *private_data, BOOL success);
102 static void async_reply_recv(void *private_data, BOOL success);
103 static void schedule_async_request(struct winbindd_child *child);
105 void async_request(TALLOC_CTX *mem_ctx, struct winbindd_child *child,
106 struct winbindd_request *request,
107 struct winbindd_response *response,
108 void (*continuation)(void *private_data, BOOL success),
109 void *private_data)
111 struct winbindd_async_request *state, *tmp;
113 SMB_ASSERT(continuation != NULL);
115 state = TALLOC_P(mem_ctx, struct winbindd_async_request);
117 if (state == NULL) {
118 DEBUG(0, ("talloc failed\n"));
119 continuation(private_data, False);
120 return;
123 state->mem_ctx = mem_ctx;
124 state->child = child;
125 state->request = request;
126 state->response = response;
127 state->continuation = continuation;
128 state->private_data = private_data;
130 DLIST_ADD_END(child->requests, state, tmp);
132 schedule_async_request(child);
134 return;
137 static void async_main_request_sent(void *private_data, BOOL success)
139 struct winbindd_async_request *state =
140 talloc_get_type_abort(private_data, struct winbindd_async_request);
142 if (!success) {
143 DEBUG(5, ("Could not send async request\n"));
145 state->response->length = sizeof(struct winbindd_response);
146 state->response->result = WINBINDD_ERROR;
147 state->continuation(state->private_data, False);
148 return;
151 if (state->request->extra_len == 0) {
152 async_request_sent(private_data, True);
153 return;
156 setup_async_write(&state->child->event, state->request->extra_data,
157 state->request->extra_len,
158 async_request_sent, state);
161 static void async_request_sent(void *private_data_data, BOOL success)
163 struct winbindd_async_request *state =
164 talloc_get_type_abort(private_data_data, struct winbindd_async_request);
166 if (!success) {
167 DEBUG(5, ("Could not send async request\n"));
169 state->response->length = sizeof(struct winbindd_response);
170 state->response->result = WINBINDD_ERROR;
171 state->continuation(state->private_data, False);
172 return;
175 /* Request successfully sent to the child, setup the wait for reply */
177 setup_async_read(&state->child->event,
178 &state->response->result,
179 sizeof(state->response->result),
180 async_reply_recv, state);
183 static void async_reply_recv(void *private_data, BOOL success)
185 struct winbindd_async_request *state =
186 talloc_get_type_abort(private_data, struct winbindd_async_request);
187 struct winbindd_child *child = state->child;
189 state->response->length = sizeof(struct winbindd_response);
191 if (!success) {
192 DEBUG(5, ("Could not receive async reply\n"));
193 state->response->result = WINBINDD_ERROR;
194 return;
197 SMB_ASSERT(cache_retrieve_response(child->pid,
198 state->response));
200 cache_cleanup_response(child->pid);
202 DLIST_REMOVE(child->requests, state);
204 schedule_async_request(child);
206 state->continuation(state->private_data, True);
209 static BOOL fork_domain_child(struct winbindd_child *child);
211 static void schedule_async_request(struct winbindd_child *child)
213 struct winbindd_async_request *request = child->requests;
215 if (request == NULL) {
216 return;
219 if (child->event.flags != 0) {
220 return; /* Busy */
223 if ((child->pid == 0) && (!fork_domain_child(child))) {
224 /* Cancel all outstanding requests */
226 while (request != NULL) {
227 /* request might be free'd in the continuation */
228 struct winbindd_async_request *next = request->next;
229 request->continuation(request->private_data, False);
230 request = next;
232 return;
235 setup_async_write(&child->event, request->request,
236 sizeof(*request->request),
237 async_main_request_sent, request);
239 talloc_destroy(child->mem_ctx);
240 return;
243 struct domain_request_state {
244 TALLOC_CTX *mem_ctx;
245 struct winbindd_domain *domain;
246 struct winbindd_request *request;
247 struct winbindd_response *response;
248 void (*continuation)(void *private_data_data, BOOL success);
249 void *private_data_data;
252 static void domain_init_recv(void *private_data_data, BOOL success);
254 void async_domain_request(TALLOC_CTX *mem_ctx,
255 struct winbindd_domain *domain,
256 struct winbindd_request *request,
257 struct winbindd_response *response,
258 void (*continuation)(void *private_data_data, BOOL success),
259 void *private_data_data)
261 struct domain_request_state *state;
263 if (domain->initialized) {
264 async_request(mem_ctx, &domain->child, request, response,
265 continuation, private_data_data);
266 return;
269 state = TALLOC_P(mem_ctx, struct domain_request_state);
270 if (state == NULL) {
271 DEBUG(0, ("talloc failed\n"));
272 continuation(private_data_data, False);
273 return;
276 state->mem_ctx = mem_ctx;
277 state->domain = domain;
278 state->request = request;
279 state->response = response;
280 state->continuation = continuation;
281 state->private_data_data = private_data_data;
283 init_child_connection(domain, domain_init_recv, state);
286 static void recvfrom_child(void *private_data_data, BOOL success)
288 struct winbindd_cli_state *state =
289 talloc_get_type_abort(private_data_data, struct winbindd_cli_state);
290 enum winbindd_result result = state->response.result;
292 /* This is an optimization: The child has written directly to the
293 * response buffer. The request itself is still in pending state,
294 * state that in the result code. */
296 state->response.result = WINBINDD_PENDING;
298 if ((!success) || (result != WINBINDD_OK)) {
299 request_error(state);
300 return;
303 request_ok(state);
306 void sendto_child(struct winbindd_cli_state *state,
307 struct winbindd_child *child)
309 async_request(state->mem_ctx, child, &state->request,
310 &state->response, recvfrom_child, state);
313 void sendto_domain(struct winbindd_cli_state *state,
314 struct winbindd_domain *domain)
316 async_domain_request(state->mem_ctx, domain,
317 &state->request, &state->response,
318 recvfrom_child, state);
321 static void domain_init_recv(void *private_data_data, BOOL success)
323 struct domain_request_state *state =
324 talloc_get_type_abort(private_data_data, struct domain_request_state);
326 if (!success) {
327 DEBUG(5, ("Domain init returned an error\n"));
328 state->continuation(state->private_data_data, False);
329 return;
332 async_request(state->mem_ctx, &state->domain->child,
333 state->request, state->response,
334 state->continuation, state->private_data_data);
337 struct winbindd_child_dispatch_table {
338 enum winbindd_cmd cmd;
339 enum winbindd_result (*fn)(struct winbindd_domain *domain,
340 struct winbindd_cli_state *state);
341 const char *winbindd_cmd_name;
344 static struct winbindd_child_dispatch_table child_dispatch_table[] = {
346 { WINBINDD_LOOKUPSID, winbindd_dual_lookupsid, "LOOKUPSID" },
347 { WINBINDD_LOOKUPNAME, winbindd_dual_lookupname, "LOOKUPNAME" },
348 { WINBINDD_LIST_TRUSTDOM, winbindd_dual_list_trusted_domains, "LIST_TRUSTDOM" },
349 { WINBINDD_INIT_CONNECTION, winbindd_dual_init_connection, "INIT_CONNECTION" },
350 { WINBINDD_GETDCNAME, winbindd_dual_getdcname, "GETDCNAME" },
351 { WINBINDD_SHOW_SEQUENCE, winbindd_dual_show_sequence, "SHOW_SEQUENCE" },
352 { WINBINDD_PAM_AUTH, winbindd_dual_pam_auth, "PAM_AUTH" },
353 { WINBINDD_PAM_AUTH_CRAP, winbindd_dual_pam_auth_crap, "AUTH_CRAP" },
354 { WINBINDD_PAM_LOGOFF, winbindd_dual_pam_logoff, "PAM_LOGOFF" },
355 { WINBINDD_CHECK_MACHACC, winbindd_dual_check_machine_acct, "CHECK_MACHACC" },
356 { WINBINDD_DUAL_SID2UID, winbindd_dual_sid2uid, "DUAL_SID2UID" },
357 { WINBINDD_DUAL_SID2GID, winbindd_dual_sid2gid, "DUAL_SID2GID" },
358 { WINBINDD_DUAL_UID2NAME, winbindd_dual_uid2name, "DUAL_UID2NAME" },
359 { WINBINDD_DUAL_NAME2UID, winbindd_dual_name2uid, "DUAL_NAME2UID" },
360 { WINBINDD_DUAL_GID2NAME, winbindd_dual_gid2name, "DUAL_GID2NAME" },
361 { WINBINDD_DUAL_NAME2GID, winbindd_dual_name2gid, "DUAL_NAME2GID" },
362 { WINBINDD_DUAL_IDMAPSET, winbindd_dual_idmapset, "DUAL_IDMAPSET" },
363 { WINBINDD_DUAL_USERINFO, winbindd_dual_userinfo, "DUAL_USERINFO" },
364 { WINBINDD_ALLOCATE_UID, winbindd_dual_allocate_uid, "ALLOCATE_UID" },
365 { WINBINDD_ALLOCATE_GID, winbindd_dual_allocate_gid, "ALLOCATE_GID" },
366 { WINBINDD_GETUSERDOMGROUPS, winbindd_dual_getuserdomgroups, "GETUSERDOMGROUPS" },
367 { WINBINDD_DUAL_GETSIDALIASES, winbindd_dual_getsidaliases, "GETSIDALIASES" },
368 /* End of list */
370 { WINBINDD_NUM_CMDS, NULL, "NONE" }
373 static void child_process_request(struct winbindd_domain *domain,
374 struct winbindd_cli_state *state)
376 struct winbindd_child_dispatch_table *table;
378 /* Free response data - we may be interrupted and receive another
379 command before being able to send this data off. */
381 state->response.result = WINBINDD_ERROR;
382 state->response.length = sizeof(struct winbindd_response);
384 state->mem_ctx = talloc_init("winbind request");
385 if (state->mem_ctx == NULL)
386 return;
388 /* Process command */
390 for (table = child_dispatch_table; table->fn; table++) {
391 if (state->request.cmd == table->cmd) {
392 DEBUG(10,("process_request: request fn %s\n",
393 table->winbindd_cmd_name ));
394 state->response.result = table->fn(domain, state);
395 break;
399 if (!table->fn) {
400 DEBUG(10,("process_request: unknown request fn number %d\n",
401 (int)state->request.cmd ));
402 state->response.result = WINBINDD_ERROR;
405 talloc_destroy(state->mem_ctx);
408 void setup_domain_child(struct winbindd_domain *domain,
409 struct winbindd_child *child,
410 const char *explicit_logfile)
412 if (explicit_logfile != NULL) {
413 pstr_sprintf(child->logfilename, "%s/log.winbindd-%s",
414 dyn_LOGFILEBASE, explicit_logfile);
415 } else if (domain != NULL) {
416 pstr_sprintf(child->logfilename, "%s/log.wb-%s",
417 dyn_LOGFILEBASE, domain->name);
418 } else {
419 smb_panic("Internal error: domain == NULL && "
420 "explicit_logfile == NULL");
423 child->domain = domain;
426 struct winbindd_child *children = NULL;
428 void winbind_child_died(pid_t pid)
430 struct winbindd_child *child;
432 for (child = children; child != NULL; child = child->next) {
433 if (child->pid == pid) {
434 break;
438 if (child == NULL) {
439 DEBUG(0, ("Unknown child %d died!\n", pid));
440 return;
443 remove_fd_event(&child->event);
444 close(child->event.fd);
445 child->event.fd = 0;
446 child->event.flags = 0;
447 child->pid = 0;
449 schedule_async_request(child);
452 /* Forward the online/offline messages to our children. */
453 void winbind_msg_offline(int msg_type, struct process_id src, void *buf, size_t len)
455 struct winbindd_child *child;
457 DEBUG(10,("winbind_msg_offline: got offline message.\n"));
459 if (!lp_winbind_offline_logon()) {
460 DEBUG(10,("winbind_msg_offline: rejecting offline message.\n"));
461 return;
464 /* Set our global state as offline. */
465 if (!set_global_winbindd_state_offline()) {
466 DEBUG(10,("winbind_msg_offline: offline request failed.\n"));
467 return;
470 for (child = children; child != NULL; child = child->next) {
471 DEBUG(10,("winbind_msg_offline: sending message to pid %u.\n",
472 (unsigned int)child->pid ));
473 message_send_pid(pid_to_procid(child->pid), MSG_WINBIND_OFFLINE, NULL, 0, False);
477 /* Forward the online/offline messages to our children. */
478 void winbind_msg_online(int msg_type, struct process_id src, void *buf, size_t len)
480 struct winbindd_child *child;
482 DEBUG(10,("winbind_msg_online: got online message.\n"));
484 if (!lp_winbind_offline_logon()) {
485 DEBUG(10,("winbind_msg_online: rejecting online message.\n"));
486 return;
489 /* Set our global state as online. */
490 set_global_winbindd_state_online();
492 for (child = children; child != NULL; child = child->next) {
493 DEBUG(10,("winbind_msg_online: sending message to pid %u.\n",
494 (unsigned int)child->pid ));
495 message_send_pid(pid_to_procid(child->pid), MSG_WINBIND_ONLINE, NULL, 0, False);
499 static void account_lockout_policy_handler(struct timed_event *te,
500 const struct timeval *now,
501 void *private_data)
503 struct winbindd_child *child = private_data;
505 struct winbindd_methods *methods;
506 SAM_UNK_INFO_12 lockout_policy;
507 NTSTATUS result;
509 DEBUG(10,("account_lockout_policy_handler called\n"));
511 if (child->timed_event) {
512 TALLOC_FREE(child->timed_event);
515 methods = child->domain->methods;
517 result = methods->lockout_policy(child->domain, child->mem_ctx, &lockout_policy);
518 if (!NT_STATUS_IS_OK(result)) {
519 DEBUG(10,("account_lockout_policy_handler: failed to call lockout_policy\n"));
520 return;
523 child->timed_event = add_timed_event(child->mem_ctx,
524 timeval_current_ofs(3600, 0),
525 "account_lockout_policy_handler",
526 account_lockout_policy_handler,
527 child);
530 /* Deal with a request to go offline. */
532 static void child_msg_offline(int msg_type, struct process_id src, void *buf, size_t len)
534 struct winbindd_domain *domain;
536 DEBUG(5,("child_msg_offline received.\n"));
538 if (!lp_winbind_offline_logon()) {
539 DEBUG(10,("child_msg_offline: rejecting offline message.\n"));
540 return;
543 /* Set our global state as offline. */
544 if (!set_global_winbindd_state_offline()) {
545 DEBUG(10,("child_msg_offline: offline request failed.\n"));
546 return;
549 /* Mark all our domains as offline. */
551 for (domain = domain_list(); domain; domain = domain->next) {
552 DEBUG(5,("child_msg_offline: marking %s offline.\n", domain->name));
553 domain->online = False;
557 /* Deal with a request to go online. */
559 static void child_msg_online(int msg_type, struct process_id src, void *buf, size_t len)
561 struct winbindd_domain *domain;
563 DEBUG(5,("child_msg_online received.\n"));
565 if (!lp_winbind_offline_logon()) {
566 DEBUG(10,("child_msg_online: rejecting online message.\n"));
567 return;
570 /* Set our global state as online. */
571 set_global_winbindd_state_online();
573 /* Mark everything online - delete any negative cache entries
574 to force an immediate reconnect. */
576 for (domain = domain_list(); domain; domain = domain->next) {
577 DEBUG(5,("child_msg_online: marking %s online.\n", domain->name));
578 domain->online = True;
579 check_negative_conn_cache_timeout(domain->name, domain->dcname, 0);
583 static BOOL fork_domain_child(struct winbindd_child *child)
585 int fdpair[2];
586 struct winbindd_cli_state state;
587 extern BOOL override_logfile;
589 if (socketpair(AF_UNIX, SOCK_STREAM, 0, fdpair) != 0) {
590 DEBUG(0, ("Could not open child pipe: %s\n",
591 strerror(errno)));
592 return False;
595 ZERO_STRUCT(state);
596 state.pid = getpid();
598 /* Ensure we don't process messages whilst we're
599 changing the disposition for the child. */
600 message_block();
602 child->pid = sys_fork();
604 if (child->pid == -1) {
605 DEBUG(0, ("Could not fork: %s\n", strerror(errno)));
606 message_unblock();
607 return False;
610 if (child->pid != 0) {
611 /* Parent */
612 close(fdpair[0]);
613 child->next = child->prev = NULL;
614 DLIST_ADD(children, child);
615 child->event.fd = fdpair[1];
616 child->event.flags = 0;
617 child->requests = NULL;
618 add_fd_event(&child->event);
619 /* We're ok with online/offline messages now. */
620 message_unblock();
621 return True;
624 /* Child */
626 state.sock = fdpair[0];
627 close(fdpair[1]);
629 /* tdb needs special fork handling */
630 if (tdb_reopen_all() == -1) {
631 DEBUG(0,("tdb_reopen_all failed.\n"));
632 _exit(0);
635 close_conns_after_fork();
637 if (!override_logfile) {
638 lp_set_logfile(child->logfilename);
639 reopen_logs();
642 /* Don't handle the same messages as our parent. */
643 message_deregister(MSG_SMB_CONF_UPDATED);
644 message_deregister(MSG_SHUTDOWN);
645 message_deregister(MSG_WINBIND_OFFLINE);
646 message_deregister(MSG_WINBIND_ONLINE);
648 /* The child is ok with online/offline messages now. */
649 message_unblock();
651 child->mem_ctx = talloc_init("child_mem_ctx");
652 if (child->mem_ctx == NULL) {
653 return False;
656 if (child->domain != NULL) {
657 /* We might be in the idmap child...*/
658 child->timed_event = add_timed_event(
659 child->mem_ctx, timeval_zero(),
660 "account_lockout_policy_handler",
661 account_lockout_policy_handler,
662 child);
665 /* Handle online/offline messages. */
666 message_register(MSG_WINBIND_OFFLINE,child_msg_offline);
667 message_register(MSG_WINBIND_ONLINE,child_msg_online);
669 while (1) {
671 int ret;
672 fd_set read_fds;
673 struct timeval t;
674 struct timeval *tp;
675 struct timeval now;
677 /* free up any talloc memory */
678 lp_TALLOC_FREE();
679 main_loop_TALLOC_FREE();
681 run_events();
683 GetTimeOfDay(&now);
685 tp = get_timed_events_timeout(&t, (time_t)-1);
686 if (tp) {
687 DEBUG(11,("select will use timeout of %d seconds\n", (int)tp->tv_sec));
690 /* Handle messages */
692 message_dispatch();
694 FD_ZERO(&read_fds);
695 FD_SET(state.sock, &read_fds);
697 ret = sys_select(state.sock + 1, &read_fds, NULL, NULL, tp);
699 if (ret == 0) {
700 DEBUG(11,("nothing is ready yet, continue\n"));
701 continue;
704 if (ret == -1 && errno == EINTR) {
705 /* We got a signal - continue. */
706 continue;
709 if (ret == -1 && errno != EINTR) {
710 DEBUG(0,("select error occured\n"));
711 perror("select");
712 return False;
715 /* fetch a request from the main daemon */
716 child_read_request(&state);
718 if (state.finished) {
719 /* we lost contact with our parent */
720 exit(0);
723 DEBUG(4,("child daemon request %d\n", (int)state.request.cmd));
725 ZERO_STRUCT(state.response);
726 state.request.null_term = '\0';
727 child_process_request(child->domain, &state);
729 SAFE_FREE(state.request.extra_data);
731 cache_store_response(sys_getpid(), &state.response);
733 SAFE_FREE(state.response.extra_data);
735 /* We just send the result code back, the result
736 * structure needs to be fetched via the
737 * winbindd_cache. Hmm. That needs fixing... */
739 if (write_data(state.sock, (void *)&state.response.result,
740 sizeof(state.response.result)) !=
741 sizeof(state.response.result)) {
742 DEBUG(0, ("Could not write result\n"));
743 exit(1);