r5296: - only include the tdb headers where they are needed
[kai/samba.git] / source / smbd / process_standard.c
index 84166196d5d4b99233b980392d4a73f511a06fa7..5439e5235a52d1ae6f798ee97b5bd0f272fb7450 100644 (file)
@@ -1,8 +1,11 @@
 /* 
    Unix SMB/CIFS implementation.
+
    process model: standard (1 process per client connection)
-   Copyright (C) Andrew Tridgell 1992-2003
+
+   Copyright (C) Andrew Tridgell 1992-2005
    Copyright (C) James J Myers 2003 <myersjj@samba.org>
+   Copyright (C) Stefan (metze) Metzmacher 2004
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 */
 
 #include "includes.h"
+#include "lib/events/events.h"
+#include "lib/tdb/include/tdb.h"
+#include "dlinklist.h"
+#include "smb_server/smb_server.h"
 
 /*
   called when the process model is selected
 */
-static void model_startup(void)
+static void standard_model_init(struct event_context *ev)
 {
        signal(SIGCHLD, SIG_IGN);
 }
 
 /*
-  called when a listening socket becomes readable
+  called when a listening socket becomes readable
 */
-static void accept_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16_t flags)
+static void standard_accept_connection(struct event_context *ev, 
+                                      struct socket_context *sock, 
+                                      void (*new_conn)(struct event_context *, struct socket_context *, 
+                                                       uint32_t , void *), 
+                                      void *private)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
+       NTSTATUS status;
+       struct socket_context *sock2;
        pid_t pid;
-       struct model_ops *model_ops = fde->private;
+       struct event_context *ev2;
 
-       accepted_fd = accept(fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("accept_connection_standard: accept: %s\n",
-                        strerror(errno)));
+       /* accept an incoming connection. */
+       status = socket_accept(sock, &sock2);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("standard_accept_connection: accept: %s\n",
+                        nt_errstr(status)));
                return;
        }
 
@@ -51,123 +62,130 @@ static void accept_connection(struct event_context *ev, struct fd_event *fde, ti
 
        if (pid != 0) {
                /* parent or error code ... */
-
-               close(accepted_fd);
+               talloc_free(sock2);
                /* go back to the event loop */
                return;
        }
 
-       /* Child code ... */
+       /* This is now the child code. We need a completely new event_context to work with */
+       ev2 = event_context_init(NULL);
+
+       /* the service has given us a private pointer that
+          encapsulates the context it needs for this new connection -
+          everything else will be freed */
+       talloc_steal(ev2, private);
+       talloc_steal(private, sock2);
+
+       /* this will free all the listening sockets and all state that
+          is not associated with this new connection */
+       talloc_free(sock);
+       talloc_free(ev);
 
-       /* close all the listening sockets */
-       event_remove_fd_all_handler(ev, model_ops->accept_connection);
-       event_remove_fd_all_handler(ev, model_ops->accept_rpc_connection);
+       /* we don't care if the dup fails, as its only a select()
+          speed optimisation */
+       socket_dup(sock2);
                        
        /* tdb needs special fork handling */
        if (tdb_reopen_all() == -1) {
-               DEBUG(0,("accept_connection_standard: tdb_reopen_all failed.\n"));
+               DEBUG(0,("standard_accept_connection: tdb_reopen_all failed.\n"));
        }
 
-       /* Load DSO's */
-       init_modules();
-               
-       /* initialize new process */
-       smbd_process_init();
-               
-       init_smbsession(ev, model_ops, accepted_fd, smbd_read_handler);
+       /* Ensure that the forked children do not expose identical random streams */
+       set_need_random_reseed();
 
-       /* return to the event loop */
+       /* setup this new connection */
+       new_conn(ev2, sock2, getpid(), private);
+
+       /* we can't return to the top level here, as that event context is gone,
+          so we now process events in the new event context until there are no
+          more to process */      
+       event_loop_wait(ev2);
+
+       talloc_free(ev2);
+       exit(0);
 }
 
 /*
-  called when a rpc listening socket becomes readable
+  called to create a new server task
 */
-static void accept_rpc_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16_t flags)
+static void standard_new_task(struct event_context *ev, 
+                             void (*new_task)(struct event_context *, uint32_t , void *), 
+                             void *private)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
        pid_t pid;
-
-       accepted_fd = accept(fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("accept_connection_standard: accept: %s\n",
-                        strerror(errno)));
-               return;
-       }
+       struct event_context *ev2;
 
        pid = fork();
 
        if (pid != 0) {
-               /* parent or error code ... */
-               close(accepted_fd);
-               /* go back to the event loop */
+               /* parent or error code ... go back to the event loop */
                return;
        }
 
-       /* Child code ... */
+       /* This is now the child code. We need a completely new event_context to work with */
+       ev2 = event_context_init(NULL);
 
-       /* close all the listening sockets */
-       event_remove_fd_all_handler(ev, accept_connection);
-       event_remove_fd_all_handler(ev, accept_rpc_connection);
-                       
-       init_rpc_session(ev, fde->private, accepted_fd); 
-}
+       /* the service has given us a private pointer that
+          encapsulates the context it needs for this new connection -
+          everything else will be freed */
+       talloc_steal(ev2, private);
 
-/* called when a SMB connection goes down */
-static void terminate_connection(struct smbsrv_context *server, const char *reason) 
-{
-       server_terminate(server);
-       /* terminate this process */
+       /* this will free all the listening sockets and all state that
+          is not associated with this new connection */
+       talloc_free(ev);
+
+       /* tdb needs special fork handling */
+       if (tdb_reopen_all() == -1) {
+               DEBUG(0,("standard_accept_connection: tdb_reopen_all failed.\n"));
+       }
+
+       /* Ensure that the forked children do not expose identical random streams */
+       set_need_random_reseed();
+
+       /* setup this new connection */
+       new_task(ev2, getpid(), private);
+
+       /* we can't return to the top level here, as that event context is gone,
+          so we now process events in the new event context until there are no
+          more to process */      
+       event_loop_wait(ev2);
+
+       talloc_free(ev2);
        exit(0);
 }
 
-/* called when a rpc connection goes down */
-static void terminate_rpc_connection(void *r, const char *reason) 
+
+/* called when a task goes down */
+static void standard_terminate(struct event_context *ev, const char *reason) 
 {
-       rpc_server_terminate(r);
+       DEBUG(2,("standard_terminate: reason[%s]\n",reason));
+
+       /* this init_iconv() has the effect of freeing the iconv context memory,
+          which makes leak checking easier */
+       init_iconv();
+
+       /* the secrets db should really hang off the connection structure */
+       secrets_shutdown();
+
+       talloc_free(ev);
+
        /* terminate this process */
        exit(0);
 }
 
-static int get_id(struct smbsrv_request *req)
-{
-       return (int)req->smb_ctx->pid;
-}
 
-static void standard_exit_server(struct smbsrv_context *smb, const char *reason)
-{
-       DEBUG(1,("standard_exit_server: reason[%s]\n",reason));
-}
+static const struct model_ops standard_ops = {
+       .name                   = "standard",
+       .model_init             = standard_model_init,
+       .accept_connection      = standard_accept_connection,
+       .new_task               = standard_new_task,
+       .terminate              = standard_terminate,
+};
 
 /*
   initialise the standard process model, registering ourselves with the process model subsystem
  */
 NTSTATUS process_model_standard_init(void)
 {
-       NTSTATUS ret;
-       struct model_ops ops;
-
-       ZERO_STRUCT(ops);
-
-       /* fill in our name */
-       ops.name = "standard";
-
-       /* fill in all the operations */
-       ops.model_startup = model_startup;
-       ops.accept_connection = accept_connection;
-       ops.accept_rpc_connection = accept_rpc_connection;
-       ops.terminate_connection = terminate_connection;
-       ops.terminate_rpc_connection = terminate_rpc_connection;
-       ops.exit_server = standard_exit_server;
-       ops.get_id = get_id;
-
-       /* register ourselves with the PROCESS_MODEL subsystem. */
-       ret = register_backend("process_model", &ops);
-       if (!NT_STATUS_IS_OK(ret)) {
-               DEBUG(0,("Failed to register process_model 'standard'!\n"));
-               return ret;
-       }
-
-       return ret;
+       return register_process_model(&standard_ops);
 }