source: trunk/samba/source/nsswitch/winbindd.c @ 30

Last change on this file since 30 was 30, checked in by Paul Smedley, 14 years ago

Code updated to Samba 3.0.25rc2 level

File size: 27.7 KB
Line 
1/*
2   Unix SMB/CIFS implementation.
3
4   Winbind daemon for ntdom nss module
5
6   Copyright (C) by Tim Potter 2000-2002
7   Copyright (C) Andrew Tridgell 2002
8   Copyright (C) Jelmer Vernooij 2003
9   Copyright (C) Volker Lendecke 2004
10   
11   This program is free software; you can redistribute it and/or modify
12   it under the terms of the GNU General Public License as published by
13   the Free Software Foundation; either version 2 of the License, or
14   (at your option) any later version.
15   
16   This program is distributed in the hope that it will be useful,
17   but WITHOUT ANY WARRANTY; without even the implied warranty of
18   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19   GNU General Public License for more details.
20   
21   You should have received a copy of the GNU General Public License
22   along with this program; if not, write to the Free Software
23   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24*/
25
26#include "includes.h"
27#include "winbindd.h"
28
29#undef DBGC_CLASS
30#define DBGC_CLASS DBGC_WINBIND
31
32BOOL opt_nocache = False;
33static BOOL interactive = False;
34
35extern BOOL override_logfile;
36
37struct event_context *winbind_event_context(void)
38{
39        static struct event_context *ctx;
40
41        if (!ctx && !(ctx = event_context_init(NULL))) {
42                smb_panic("Could not init winbind event context\n");
43        }
44        return ctx;
45}
46
47/* Reload configuration */
48
49static BOOL reload_services_file(void)
50{
51        BOOL ret;
52
53        if (lp_loaded()) {
54                pstring fname;
55
56                pstrcpy(fname,lp_configfile());
57                if (file_exist(fname,NULL) && !strcsequal(fname,dyn_CONFIGFILE)) {
58                        pstrcpy(dyn_CONFIGFILE,fname);
59                }
60        }
61
62        reopen_logs();
63        ret = lp_load(dyn_CONFIGFILE,False,False,True,True);
64
65        reopen_logs();
66        load_interfaces();
67
68        return(ret);
69}
70
71
72/**************************************************************************** **
73 Handle a fault..
74 **************************************************************************** */
75
76static void fault_quit(void)
77{
78        dump_core();
79}
80
81static void winbindd_status(void)
82{
83        struct winbindd_cli_state *tmp;
84
85        DEBUG(0, ("winbindd status:\n"));
86
87        /* Print client state information */
88       
89        DEBUG(0, ("\t%d clients currently active\n", winbindd_num_clients()));
90       
91        if (DEBUGLEVEL >= 2 && winbindd_num_clients()) {
92                DEBUG(2, ("\tclient list:\n"));
93                for(tmp = winbindd_client_list(); tmp; tmp = tmp->next) {
94                        DEBUGADD(2, ("\t\tpid %lu, sock %d\n",
95                                  (unsigned long)tmp->pid, tmp->sock));
96                }
97        }
98}
99
100/* Print winbindd status to log file */
101
102static void print_winbindd_status(void)
103{
104        winbindd_status();
105}
106
107/* Flush client cache */
108
109static void flush_caches(void)
110{
111        /* We need to invalidate cached user list entries on a SIGHUP
112           otherwise cached access denied errors due to restrict anonymous
113           hang around until the sequence number changes. */
114
115        wcache_invalidate_cache();
116}
117
118/* Handle the signal by unlinking socket and exiting */
119
120static void terminate(void)
121{
122        pstring path;
123
124        /* Remove socket file */
125        pstr_sprintf(path, "%s/%s", 
126                 WINBINDD_SOCKET_DIR, WINBINDD_SOCKET_NAME);
127        unlink(path);
128
129        idmap_close();
130       
131        trustdom_cache_shutdown();
132
133#if 0
134        if (interactive) {
135                TALLOC_CTX *mem_ctx = talloc_init("end_description");
136                char *description = talloc_describe_all(mem_ctx);
137
138                DEBUG(3, ("tallocs left:\n%s\n", description));
139                talloc_destroy(mem_ctx);
140        }
141#endif
142
143        exit(0);
144}
145
146static BOOL do_sigterm;
147
148static void termination_handler(int signum)
149{
150        do_sigterm = True;
151        sys_select_signal(signum);
152}
153
154static BOOL do_sigusr2;
155
156static void sigusr2_handler(int signum)
157{
158        do_sigusr2 = True;
159        sys_select_signal(SIGUSR2);
160}
161
162static BOOL do_sighup;
163
164static void sighup_handler(int signum)
165{
166        do_sighup = True;
167        sys_select_signal(SIGHUP);
168}
169
170static BOOL do_sigchld;
171
172static void sigchld_handler(int signum)
173{
174        do_sigchld = True;
175        sys_select_signal(SIGCHLD);
176}
177
178/* React on 'smbcontrol winbindd reload-config' in the same way as on SIGHUP*/
179static void msg_reload_services(int msg_type, struct process_id src,
180                                void *buf, size_t len, void *private_data)
181{
182        /* Flush various caches */
183        flush_caches();
184        reload_services_file();
185}
186
187/* React on 'smbcontrol winbindd shutdown' in the same way as on SIGTERM*/
188static void msg_shutdown(int msg_type, struct process_id src,
189                         void *buf, size_t len, void *private_data)
190{
191        do_sigterm = True;
192}
193
194static struct winbindd_dispatch_table {
195        enum winbindd_cmd cmd;
196        void (*fn)(struct winbindd_cli_state *state);
197        const char *winbindd_cmd_name;
198} dispatch_table[] = {
199       
200        /* User functions */
201
202        { WINBINDD_GETPWNAM, winbindd_getpwnam, "GETPWNAM" },
203        { WINBINDD_GETPWUID, winbindd_getpwuid, "GETPWUID" },
204
205        { WINBINDD_SETPWENT, winbindd_setpwent, "SETPWENT" },
206        { WINBINDD_ENDPWENT, winbindd_endpwent, "ENDPWENT" },
207        { WINBINDD_GETPWENT, winbindd_getpwent, "GETPWENT" },
208
209        { WINBINDD_GETGROUPS, winbindd_getgroups, "GETGROUPS" },
210        { WINBINDD_GETUSERSIDS, winbindd_getusersids, "GETUSERSIDS" },
211        { WINBINDD_GETUSERDOMGROUPS, winbindd_getuserdomgroups,
212          "GETUSERDOMGROUPS" },
213
214        /* Group functions */
215
216        { WINBINDD_GETGRNAM, winbindd_getgrnam, "GETGRNAM" },
217        { WINBINDD_GETGRGID, winbindd_getgrgid, "GETGRGID" },
218        { WINBINDD_SETGRENT, winbindd_setgrent, "SETGRENT" },
219        { WINBINDD_ENDGRENT, winbindd_endgrent, "ENDGRENT" },
220        { WINBINDD_GETGRENT, winbindd_getgrent, "GETGRENT" },
221        { WINBINDD_GETGRLST, winbindd_getgrent, "GETGRLST" },
222
223        /* PAM auth functions */
224
225        { WINBINDD_PAM_AUTH, winbindd_pam_auth, "PAM_AUTH" },
226        { WINBINDD_PAM_AUTH_CRAP, winbindd_pam_auth_crap, "AUTH_CRAP" },
227        { WINBINDD_PAM_CHAUTHTOK, winbindd_pam_chauthtok, "CHAUTHTOK" },
228        { WINBINDD_PAM_LOGOFF, winbindd_pam_logoff, "PAM_LOGOFF" },
229        { WINBINDD_PAM_CHNG_PSWD_AUTH_CRAP, winbindd_pam_chng_pswd_auth_crap, "CHNG_PSWD_AUTH_CRAP" },
230
231        /* Enumeration functions */
232
233        { WINBINDD_LIST_USERS, winbindd_list_users, "LIST_USERS" },
234        { WINBINDD_LIST_GROUPS, winbindd_list_groups, "LIST_GROUPS" },
235        { WINBINDD_LIST_TRUSTDOM, winbindd_list_trusted_domains,
236          "LIST_TRUSTDOM" },
237        { WINBINDD_SHOW_SEQUENCE, winbindd_show_sequence, "SHOW_SEQUENCE" },
238
239        /* SID related functions */
240
241        { WINBINDD_LOOKUPSID, winbindd_lookupsid, "LOOKUPSID" },
242        { WINBINDD_LOOKUPNAME, winbindd_lookupname, "LOOKUPNAME" },
243        { WINBINDD_LOOKUPRIDS, winbindd_lookuprids, "LOOKUPRIDS" },
244
245        /* Lookup related functions */
246
247        { WINBINDD_SID_TO_UID, winbindd_sid_to_uid, "SID_TO_UID" },
248        { WINBINDD_SID_TO_GID, winbindd_sid_to_gid, "SID_TO_GID" },
249        { WINBINDD_UID_TO_SID, winbindd_uid_to_sid, "UID_TO_SID" },
250        { WINBINDD_GID_TO_SID, winbindd_gid_to_sid, "GID_TO_SID" },
251#if 0   /* DISABLED until we fix the interface in Samba 3.0.26 --jerry */
252        { WINBINDD_SIDS_TO_XIDS, winbindd_sids_to_unixids, "SIDS_TO_XIDS" },
253#endif  /* end DISABLED */
254        { WINBINDD_ALLOCATE_UID, winbindd_allocate_uid, "ALLOCATE_UID" },
255        { WINBINDD_ALLOCATE_GID, winbindd_allocate_gid, "ALLOCATE_GID" },
256        { WINBINDD_SET_MAPPING, winbindd_set_mapping, "SET_MAPPING" },
257        { WINBINDD_SET_HWM, winbindd_set_hwm, "SET_HWMS" },
258
259        /* Miscellaneous */
260
261        { WINBINDD_DUMP_MAPS, winbindd_dump_maps, "DUMP_MAPS" },
262
263        { WINBINDD_CHECK_MACHACC, winbindd_check_machine_acct, "CHECK_MACHACC" },
264        { WINBINDD_PING, winbindd_ping, "PING" },
265        { WINBINDD_INFO, winbindd_info, "INFO" },
266        { WINBINDD_INTERFACE_VERSION, winbindd_interface_version,
267          "INTERFACE_VERSION" },
268        { WINBINDD_DOMAIN_NAME, winbindd_domain_name, "DOMAIN_NAME" },
269        { WINBINDD_DOMAIN_INFO, winbindd_domain_info, "DOMAIN_INFO" },
270        { WINBINDD_NETBIOS_NAME, winbindd_netbios_name, "NETBIOS_NAME" },
271        { WINBINDD_PRIV_PIPE_DIR, winbindd_priv_pipe_dir,
272          "WINBINDD_PRIV_PIPE_DIR" },
273        { WINBINDD_GETDCNAME, winbindd_getdcname, "GETDCNAME" },
274
275        /* Credential cache access */
276        { WINBINDD_CCACHE_NTLMAUTH, winbindd_ccache_ntlm_auth, "NTLMAUTH" },
277
278        /* WINS functions */
279
280        { WINBINDD_WINS_BYNAME, winbindd_wins_byname, "WINS_BYNAME" },
281        { WINBINDD_WINS_BYIP, winbindd_wins_byip, "WINS_BYIP" },
282       
283        /* End of list */
284
285        { WINBINDD_NUM_CMDS, NULL, "NONE" }
286};
287
288static void process_request(struct winbindd_cli_state *state)
289{
290        struct winbindd_dispatch_table *table = dispatch_table;
291
292        /* Free response data - we may be interrupted and receive another
293           command before being able to send this data off. */
294
295        SAFE_FREE(state->response.extra_data.data); 
296
297        ZERO_STRUCT(state->response);
298
299        state->response.result = WINBINDD_PENDING;
300        state->response.length = sizeof(struct winbindd_response);
301
302        state->mem_ctx = talloc_init("winbind request");
303        if (state->mem_ctx == NULL)
304                return;
305
306        /* Process command */
307
308        for (table = dispatch_table; table->fn; table++) {
309                if (state->request.cmd == table->cmd) {
310                        DEBUG(10,("process_request: request fn %s\n",
311                                  table->winbindd_cmd_name ));
312                        table->fn(state);
313                        break;
314                }
315        }
316
317        if (!table->fn) {
318                DEBUG(10,("process_request: unknown request fn number %d\n",
319                          (int)state->request.cmd ));
320                request_error(state);
321        }
322}
323
324/*
325 * A list of file descriptors being monitored by select in the main processing
326 * loop. fd_event->handler is called whenever the socket is readable/writable.
327 */
328
329static struct fd_event *fd_events = NULL;
330
331void add_fd_event(struct fd_event *ev)
332{
333        struct fd_event *match;
334
335        /* only add unique fd_event structs */
336
337        for (match=fd_events; match; match=match->next ) {
338#ifdef DEVELOPER
339                SMB_ASSERT( match != ev );
340#else
341                if ( match == ev )
342                        return;
343#endif
344        }
345
346        DLIST_ADD(fd_events, ev);
347}
348
349void remove_fd_event(struct fd_event *ev)
350{
351        DLIST_REMOVE(fd_events, ev);
352}
353
354/*
355 * Handler for fd_events to complete a read/write request, set up by
356 * setup_async_read/setup_async_write.
357 */
358
359static void rw_callback(struct fd_event *event, int flags)
360{
361        size_t todo;
362        ssize_t done = 0;
363
364        todo = event->length - event->done;
365
366        if (event->flags & EVENT_FD_WRITE) {
367                SMB_ASSERT(flags == EVENT_FD_WRITE);
368                done = sys_write(event->fd,
369                                 &((char *)event->data)[event->done],
370                                 todo);
371
372                if (done <= 0) {
373                        event->flags = 0;
374                        event->finished(event->private_data, False);
375                        return;
376                }
377        }
378
379        if (event->flags & EVENT_FD_READ) {
380                SMB_ASSERT(flags == EVENT_FD_READ);
381                done = sys_read(event->fd, &((char *)event->data)[event->done],
382                                todo);
383
384                if (done <= 0) {
385                        event->flags = 0;
386                        event->finished(event->private_data, False);
387                        return;
388                }
389        }
390
391        event->done += done;
392
393        if (event->done == event->length) {
394                event->flags = 0;
395                event->finished(event->private_data, True);
396        }
397}
398
399/*
400 * Request an async read/write on a fd_event structure. (*finished) is called
401 * when the request is completed or an error had occurred.
402 */
403
404void setup_async_read(struct fd_event *event, void *data, size_t length,
405                      void (*finished)(void *private_data, BOOL success),
406                      void *private_data)
407{
408        SMB_ASSERT(event->flags == 0);
409        event->data = data;
410        event->length = length;
411        event->done = 0;
412        event->handler = rw_callback;
413        event->finished = finished;
414        event->private_data = private_data;
415        event->flags = EVENT_FD_READ;
416}
417
418void setup_async_write(struct fd_event *event, void *data, size_t length,
419                       void (*finished)(void *private_data, BOOL success),
420                       void *private_data)
421{
422        SMB_ASSERT(event->flags == 0);
423        event->data = data;
424        event->length = length;
425        event->done = 0;
426        event->handler = rw_callback;
427        event->finished = finished;
428        event->private_data = private_data;
429        event->flags = EVENT_FD_WRITE;
430}
431
432/*
433 * This is the main event loop of winbind requests. It goes through a
434 * state-machine of 3 read/write requests, 4 if you have extra data to send.
435 *
436 * An idle winbind client has a read request of 4 bytes outstanding,
437 * finalizing function is request_len_recv, checking the length. request_recv
438 * then processes the packet. The processing function then at some point has
439 * to call request_finished which schedules sending the response.
440 */
441
442static void request_len_recv(void *private_data, BOOL success);
443static void request_recv(void *private_data, BOOL success);
444static void request_main_recv(void *private_data, BOOL success);
445static void request_finished(struct winbindd_cli_state *state);
446void request_finished_cont(void *private_data, BOOL success);
447static void response_main_sent(void *private_data, BOOL success);
448static void response_extra_sent(void *private_data, BOOL success);
449
450static void response_extra_sent(void *private_data, BOOL success)
451{
452        struct winbindd_cli_state *state =
453                talloc_get_type_abort(private_data, struct winbindd_cli_state);
454
455        if (state->mem_ctx != NULL) {
456                talloc_destroy(state->mem_ctx);
457                state->mem_ctx = NULL;
458        }
459
460        if (!success) {
461                state->finished = True;
462                return;
463        }
464
465        SAFE_FREE(state->request.extra_data.data);
466        SAFE_FREE(state->response.extra_data.data);
467
468        setup_async_read(&state->fd_event, &state->request, sizeof(uint32),
469                         request_len_recv, state);
470}
471
472static void response_main_sent(void *private_data, BOOL success)
473{
474        struct winbindd_cli_state *state =
475                talloc_get_type_abort(private_data, struct winbindd_cli_state);
476
477        if (!success) {
478                state->finished = True;
479                return;
480        }
481
482        if (state->response.length == sizeof(state->response)) {
483                if (state->mem_ctx != NULL) {
484                        talloc_destroy(state->mem_ctx);
485                        state->mem_ctx = NULL;
486                }
487
488                setup_async_read(&state->fd_event, &state->request,
489                                 sizeof(uint32), request_len_recv, state);
490                return;
491        }
492
493        setup_async_write(&state->fd_event, state->response.extra_data.data,
494                          state->response.length - sizeof(state->response),
495                          response_extra_sent, state);
496}
497
498static void request_finished(struct winbindd_cli_state *state)
499{
500        setup_async_write(&state->fd_event, &state->response,
501                          sizeof(state->response), response_main_sent, state);
502}
503
504void request_error(struct winbindd_cli_state *state)
505{
506        SMB_ASSERT(state->response.result == WINBINDD_PENDING);
507        state->response.result = WINBINDD_ERROR;
508        request_finished(state);
509}
510
511void request_ok(struct winbindd_cli_state *state)
512{
513        SMB_ASSERT(state->response.result == WINBINDD_PENDING);
514        state->response.result = WINBINDD_OK;
515        request_finished(state);
516}
517
518void request_finished_cont(void *private_data, BOOL success)
519{
520        struct winbindd_cli_state *state =
521                talloc_get_type_abort(private_data, struct winbindd_cli_state);
522
523        if (success)
524                request_ok(state);
525        else
526                request_error(state);
527}
528
529static void request_len_recv(void *private_data, BOOL success)
530{
531        struct winbindd_cli_state *state =
532                talloc_get_type_abort(private_data, struct winbindd_cli_state);
533
534        if (!success) {
535                state->finished = True;
536                return;
537        }
538
539        if (*(uint32 *)(&state->request) != sizeof(state->request)) {
540                DEBUG(0,("request_len_recv: Invalid request size received: %d\n",
541                         *(uint32 *)(&state->request)));
542                state->finished = True;
543                return;
544        }
545
546        setup_async_read(&state->fd_event, (uint32 *)(&state->request)+1,
547                         sizeof(state->request) - sizeof(uint32),
548                         request_main_recv, state);
549}
550
551static void request_main_recv(void *private_data, BOOL success)
552{
553        struct winbindd_cli_state *state =
554                talloc_get_type_abort(private_data, struct winbindd_cli_state);
555
556        if (!success) {
557                state->finished = True;
558                return;
559        }
560
561        if (state->request.extra_len == 0) {
562                state->request.extra_data.data = NULL;
563                request_recv(state, True);
564                return;
565        }
566
567        if ((!state->privileged) &&
568            (state->request.extra_len > WINBINDD_MAX_EXTRA_DATA)) {
569                DEBUG(3, ("Got request with %d bytes extra data on "
570                          "unprivileged socket\n", (int)state->request.extra_len));
571                state->request.extra_data.data = NULL;
572                state->finished = True;
573                return;
574        }
575
576        state->request.extra_data.data =
577                SMB_MALLOC_ARRAY(char, state->request.extra_len + 1);
578
579        if (state->request.extra_data.data == NULL) {
580                DEBUG(0, ("malloc failed\n"));
581                state->finished = True;
582                return;
583        }
584
585        /* Ensure null termination */
586        state->request.extra_data.data[state->request.extra_len] = '\0';
587
588        setup_async_read(&state->fd_event, state->request.extra_data.data,
589                         state->request.extra_len, request_recv, state);
590}
591
592static void request_recv(void *private_data, BOOL success)
593{
594        struct winbindd_cli_state *state =
595                talloc_get_type_abort(private_data, struct winbindd_cli_state);
596
597        if (!success) {
598                state->finished = True;
599                return;
600        }
601
602        process_request(state);
603}
604
605/* Process a new connection by adding it to the client connection list */
606
607static void new_connection(int listen_sock, BOOL privileged)
608{
609        struct sockaddr_un sunaddr;
610        struct winbindd_cli_state *state;
611        socklen_t len;
612        int sock;
613       
614        /* Accept connection */
615       
616        len = sizeof(sunaddr);
617
618        do {
619                sock = accept(listen_sock, (struct sockaddr *)&sunaddr, &len);
620        } while (sock == -1 && errno == EINTR);
621
622        if (sock == -1)
623                return;
624       
625        DEBUG(6,("accepted socket %d\n", sock));
626       
627        /* Create new connection structure */
628       
629        if ((state = TALLOC_ZERO_P(NULL, struct winbindd_cli_state)) == NULL) {
630                close(sock);
631                return;
632        }
633       
634        state->sock = sock;
635
636        state->last_access = time(NULL);       
637
638        state->privileged = privileged;
639
640        state->fd_event.fd = state->sock;
641        state->fd_event.flags = 0;
642        add_fd_event(&state->fd_event);
643
644        setup_async_read(&state->fd_event, &state->request, sizeof(uint32),
645                         request_len_recv, state);
646
647        /* Add to connection list */
648       
649        winbindd_add_client(state);
650}
651
652/* Remove a client connection from client connection list */
653
654static void remove_client(struct winbindd_cli_state *state)
655{
656        /* It's a dead client - hold a funeral */
657       
658        if (state == NULL) {
659                return;
660        }
661               
662        /* Close socket */
663               
664        close(state->sock);
665               
666        /* Free any getent state */
667               
668        free_getent_state(state->getpwent_state);
669        free_getent_state(state->getgrent_state);
670               
671        /* We may have some extra data that was not freed if the client was
672           killed unexpectedly */
673
674        SAFE_FREE(state->response.extra_data.data);
675
676        if (state->mem_ctx != NULL) {
677                talloc_destroy(state->mem_ctx);
678                state->mem_ctx = NULL;
679        }
680
681        remove_fd_event(&state->fd_event);
682               
683        /* Remove from list and free */
684               
685        winbindd_remove_client(state);
686        TALLOC_FREE(state);
687}
688
689/* Shutdown client connection which has been idle for the longest time */
690
691static BOOL remove_idle_client(void)
692{
693        struct winbindd_cli_state *state, *remove_state = NULL;
694        time_t last_access = 0;
695        int nidle = 0;
696
697        for (state = winbindd_client_list(); state; state = state->next) {
698                if (state->response.result != WINBINDD_PENDING &&
699                    !state->getpwent_state && !state->getgrent_state) {
700                        nidle++;
701                        if (!last_access || state->last_access < last_access) {
702                                last_access = state->last_access;
703                                remove_state = state;
704                        }
705                }
706        }
707
708        if (remove_state) {
709                DEBUG(5,("Found %d idle client connections, shutting down sock %d, pid %u\n",
710                        nidle, remove_state->sock, (unsigned int)remove_state->pid));
711                remove_client(remove_state);
712                return True;
713        }
714
715        return False;
716}
717
718/* Process incoming clients on listen_sock.  We use a tricky non-blocking,
719   non-forking, non-threaded model which allows us to handle many
720   simultaneous connections while remaining impervious to many denial of
721   service attacks. */
722
723static void process_loop(void)
724{
725        struct winbindd_cli_state *state;
726        struct fd_event *ev;
727        fd_set r_fds, w_fds;
728        int maxfd, listen_sock, listen_priv_sock, selret;
729        struct timeval timeout, ev_timeout;
730
731        /* Open Sockets here to get stuff going ASAP */
732        listen_sock = open_winbindd_socket();
733        listen_priv_sock = open_winbindd_priv_socket();
734
735        if (listen_sock == -1 || listen_priv_sock == -1) {
736                perror("open_winbind_socket");
737                exit(1);
738        }
739
740        /* We'll be doing this a lot */
741
742        /* Handle messages */
743
744        message_dispatch();
745
746        run_events(winbind_event_context(), 0, NULL, NULL);
747
748        /* refresh the trusted domain cache */
749
750        rescan_trusted_domains();
751
752        /* Free up temporary memory */
753
754        lp_TALLOC_FREE();
755        main_loop_TALLOC_FREE();
756
757        /* Initialise fd lists for select() */
758
759        maxfd = MAX(listen_sock, listen_priv_sock);
760
761        FD_ZERO(&r_fds);
762        FD_ZERO(&w_fds);
763        FD_SET(listen_sock, &r_fds);
764        FD_SET(listen_priv_sock, &r_fds);
765
766        timeout.tv_sec = WINBINDD_ESTABLISH_LOOP;
767        timeout.tv_usec = 0;
768
769        /* Check for any event timeouts. */
770        if (get_timed_events_timeout(winbind_event_context(), &ev_timeout)) {
771                timeout = timeval_min(&timeout, &ev_timeout);
772        }
773
774        /* Set up client readers and writers */
775
776        state = winbindd_client_list();
777
778        while (state) {
779
780                struct winbindd_cli_state *next = state->next;
781
782                /* Dispose of client connection if it is marked as
783                   finished */ 
784
785                if (state->finished)
786                        remove_client(state);
787
788                state = next;
789        }
790
791        for (ev = fd_events; ev; ev = ev->next) {
792                if (ev->flags & EVENT_FD_READ) {
793                        FD_SET(ev->fd, &r_fds);
794                        maxfd = MAX(ev->fd, maxfd);
795                }
796                if (ev->flags & EVENT_FD_WRITE) {
797                        FD_SET(ev->fd, &w_fds);
798                        maxfd = MAX(ev->fd, maxfd);
799                }
800        }
801
802        /* Call select */
803       
804        selret = sys_select(maxfd + 1, &r_fds, &w_fds, NULL, &timeout);
805
806        if (selret == 0) {
807                goto no_fds_ready;
808        }
809
810        if (selret == -1) {
811                if (errno == EINTR) {
812                        goto no_fds_ready;
813                }
814
815                /* Select error, something is badly wrong */
816
817                perror("select");
818                exit(1);
819        }
820
821        /* selret > 0 */
822
823        ev = fd_events;
824        while (ev != NULL) {
825                struct fd_event *next = ev->next;
826                int flags = 0;
827                if (FD_ISSET(ev->fd, &r_fds))
828                        flags |= EVENT_FD_READ;
829                if (FD_ISSET(ev->fd, &w_fds))
830                        flags |= EVENT_FD_WRITE;
831                if (flags)
832                        ev->handler(ev, flags);
833                ev = next;
834        }
835
836        if (FD_ISSET(listen_sock, &r_fds)) {
837                while (winbindd_num_clients() >
838                       WINBINDD_MAX_SIMULTANEOUS_CLIENTS - 1) {
839                        DEBUG(5,("winbindd: Exceeding %d client "
840                                 "connections, removing idle "
841                                 "connection.\n",
842                                 WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
843                        if (!remove_idle_client()) {
844                                DEBUG(0,("winbindd: Exceeding %d "
845                                         "client connections, no idle "
846                                         "connection found\n",
847                                         WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
848                                break;
849                        }
850                }
851                /* new, non-privileged connection */
852                new_connection(listen_sock, False);
853        }
854           
855        if (FD_ISSET(listen_priv_sock, &r_fds)) {
856                while (winbindd_num_clients() >
857                       WINBINDD_MAX_SIMULTANEOUS_CLIENTS - 1) {
858                        DEBUG(5,("winbindd: Exceeding %d client "
859                                 "connections, removing idle "
860                                 "connection.\n",
861                                 WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
862                        if (!remove_idle_client()) {
863                                DEBUG(0,("winbindd: Exceeding %d "
864                                         "client connections, no idle "
865                                         "connection found\n",
866                                         WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
867                                break;
868                        }
869                }
870                /* new, privileged connection */
871                new_connection(listen_priv_sock, True);
872        }
873
874 no_fds_ready:
875
876#if 0
877        winbindd_check_cache_size(time(NULL));
878#endif
879
880        /* Check signal handling things */
881
882        if (do_sigterm)
883                terminate();
884
885        if (do_sighup) {
886
887                DEBUG(3, ("got SIGHUP\n"));
888
889                msg_reload_services(MSG_SMB_CONF_UPDATED, pid_to_procid(0), NULL, 0, NULL);
890                do_sighup = False;
891        }
892
893        if (do_sigusr2) {
894                print_winbindd_status();
895                do_sigusr2 = False;
896        }
897
898        if (do_sigchld) {
899                pid_t pid;
900
901                do_sigchld = False;
902
903                while ((pid = sys_waitpid(-1, NULL, WNOHANG)) > 0) {
904                        winbind_child_died(pid);
905                }
906        }
907}
908
909/* Main function */
910
911int main(int argc, char **argv, char **envp)
912{
913        pstring logfile;
914        static BOOL Fork = True;
915        static BOOL log_stdout = False;
916        static BOOL no_process_group = False;
917        struct poptOption long_options[] = {
918                POPT_AUTOHELP
919                { "stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
920                { "foreground", 'F', POPT_ARG_VAL, &Fork, False, "Daemon in foreground mode" },
921                { "no-process-group", 0, POPT_ARG_VAL, &no_process_group, True, "Don't create a new process group" },
922                { "interactive", 'i', POPT_ARG_NONE, NULL, 'i', "Interactive mode" },
923                { "no-caching", 'n', POPT_ARG_VAL, &opt_nocache, True, "Disable caching" },
924                POPT_COMMON_SAMBA
925                POPT_TABLEEND
926        };
927        poptContext pc;
928        int opt;
929
930        /* glibc (?) likes to print "User defined signal 1" and exit if a
931           SIGUSR[12] is received before a handler is installed */
932
933        CatchSignal(SIGUSR1, SIG_IGN);
934        CatchSignal(SIGUSR2, SIG_IGN);
935
936        fault_setup((void (*)(void *))fault_quit );
937        dump_core_setup("winbindd");
938
939        load_case_tables();
940
941        /* Initialise for running in non-root mode */
942
943        sec_init();
944
945        set_remote_machine_name("winbindd", False);
946
947        /* Set environment variable so we don't recursively call ourselves.
948           This may also be useful interactively. */
949
950        if ( !winbind_off() ) {
951                DEBUG(0,("Failed to disable recusive winbindd calls.  Exiting.\n"));
952                exit(1);
953        }
954
955        /* Initialise samba/rpc client stuff */
956
957        pc = poptGetContext("winbindd", argc, (const char **)argv, long_options,
958                                                POPT_CONTEXT_KEEP_FIRST);
959
960        while ((opt = poptGetNextOpt(pc)) != -1) {
961                switch (opt) {
962                        /* Don't become a daemon */
963                case 'i':
964                        interactive = True;
965                        log_stdout = True;
966                        Fork = False;
967                        break;
968                }
969        }
970
971
972        if (log_stdout && Fork) {
973                printf("Can't log to stdout (-S) unless daemon is in foreground +(-F) or interactive (-i)\n");
974                poptPrintUsage(pc, stderr, 0);
975                exit(1);
976        }
977
978        if (!override_logfile) {
979                pstr_sprintf(logfile, "%s/log.winbindd", dyn_LOGFILEBASE);
980                lp_set_logfile(logfile);
981        }
982        setup_logging("winbindd", log_stdout);
983        reopen_logs();
984
985        DEBUG(1, ("winbindd version %s started.\n%s\n", 
986                  SAMBA_VERSION_STRING, 
987                  COPYRIGHT_STARTUP_MESSAGE) );
988
989        if (!reload_services_file()) {
990                DEBUG(0, ("error opening config file\n"));
991                exit(1);
992        }
993
994        if (!directory_exist(lp_lockdir(), NULL)) {
995                mkdir(lp_lockdir(), 0755);
996        }
997
998        /* Setup names. */
999
1000        if (!init_names())
1001                exit(1);
1002
1003        load_interfaces();
1004
1005        if (!secrets_init()) {
1006
1007                DEBUG(0,("Could not initialize domain trust account secrets. Giving up\n"));
1008                return False;
1009        }
1010
1011        /* Enable netbios namecache */
1012
1013        namecache_enable();
1014
1015        /* Winbind daemon initialisation */
1016
1017        if ( ! NT_STATUS_IS_OK(idmap_init_cache()) ) {
1018                DEBUG(1, ("Could not init idmap cache!\n"));           
1019        }
1020
1021        /* Unblock all signals we are interested in as they may have been
1022           blocked by the parent process. */
1023
1024        BlockSignals(False, SIGINT);
1025        BlockSignals(False, SIGQUIT);
1026        BlockSignals(False, SIGTERM);
1027        BlockSignals(False, SIGUSR1);
1028        BlockSignals(False, SIGUSR2);
1029        BlockSignals(False, SIGHUP);
1030        BlockSignals(False, SIGCHLD);
1031
1032        /* Setup signal handlers */
1033       
1034        CatchSignal(SIGINT, termination_handler);      /* Exit on these sigs */
1035        CatchSignal(SIGQUIT, termination_handler);
1036        CatchSignal(SIGTERM, termination_handler);
1037        CatchSignal(SIGCHLD, sigchld_handler);
1038
1039        CatchSignal(SIGPIPE, SIG_IGN);                 /* Ignore sigpipe */
1040
1041        CatchSignal(SIGUSR2, sigusr2_handler);         /* Debugging sigs */
1042        CatchSignal(SIGHUP, sighup_handler);
1043
1044        if (!interactive)
1045                become_daemon(Fork, no_process_group);
1046
1047        pidfile_create("winbindd");
1048
1049#if HAVE_SETPGID
1050        /*
1051         * If we're interactive we want to set our own process group for
1052         * signal management.
1053         */
1054        if (interactive && !no_process_group)
1055                setpgid( (pid_t)0, (pid_t)0);
1056#endif
1057
1058        TimeInit();
1059
1060        /* Initialise messaging system */
1061
1062        if (!message_init()) {
1063                DEBUG(0, ("unable to initialize messaging system\n"));
1064                exit(1);
1065        }
1066       
1067        /* Initialize cache (ensure version is correct). */
1068        if (!initialize_winbindd_cache()) {
1069                exit(1);
1070        }
1071
1072        /* React on 'smbcontrol winbindd reload-config' in the same way
1073           as to SIGHUP signal */
1074        message_register(MSG_SMB_CONF_UPDATED, msg_reload_services, NULL);
1075        message_register(MSG_SHUTDOWN, msg_shutdown, NULL);
1076
1077        /* Handle online/offline messages. */
1078        message_register(MSG_WINBIND_OFFLINE, winbind_msg_offline, NULL);
1079        message_register(MSG_WINBIND_ONLINE, winbind_msg_online, NULL);
1080        message_register(MSG_WINBIND_ONLINESTATUS, winbind_msg_onlinestatus,
1081                         NULL);
1082
1083        poptFreeContext(pc);
1084
1085        netsamlogon_cache_init(); /* Non-critical */
1086       
1087        if (!init_domain_list()) {
1088                DEBUG(0,("unable to initalize domain list\n"));
1089                exit(1);
1090        }
1091
1092        init_idmap_child();
1093
1094        smb_nscd_flush_user_cache();
1095        smb_nscd_flush_group_cache();
1096
1097        /* Loop waiting for requests */
1098
1099        while (1)
1100                process_loop();
1101
1102        return 0;
1103}
Note: See TracBrowser for help on using the repository browser.