waf: Remove lib prefix from libraries manually.
[kai/samba.git] / source4 / smbd / process_single.c
index 1cba3faee3f4de625af0eb80ec5132157fb2865f..688b46e0a01c831c03ed9655629ca9daa2335d7e 100644 (file)
@@ -1,13 +1,15 @@
 /* 
    Unix SMB/CIFS implementation.
+
    process model: process (1 process handles all client connections)
+
    Copyright (C) Andrew Tridgell 2003
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
-
+#include "smbd/process_model.h"
+#include "system/filesys.h"
+#include "cluster/cluster.h"
 
 /*
   called when the process model is selected
 */
-static void single_start_server(void)
+static void single_model_init(struct tevent_context *ev)
 {
-       smbd_process_init();
 }
 
 /*
-  called when a listening socket becomes readable
+  called when a listening socket becomes readable
 */
-static void single_accept_connection(struct event_context *ev, struct fd_event *srv_fde, time_t t, uint16_t flags)
+static void single_accept_connection(struct tevent_context *ev, 
+                                    struct loadparm_context *lp_ctx,
+                                    struct socket_context *listen_socket,
+                                    void (*new_conn)(struct tevent_context *, 
+                                                     struct loadparm_context *,
+                                                     struct socket_context *, 
+                                                     struct server_id , void *), 
+                                    void *private_data)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
-       struct fd_event fde;
-       struct timed_event idle;
-       struct server_socket *server_socket = srv_fde->private;
-       struct server_connection *conn;
-       TALLOC_CTX *mem_ctx;
+       NTSTATUS status;
+       struct socket_context *connected_socket;
 
        /* accept an incoming connection. */
-       accepted_fd = accept(srv_fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("accept_connection_single: accept: %s\n",
-                        strerror(errno)));
-               return;
-       }
-
-       mem_ctx = talloc_init("server_service_connection");
-       if (!mem_ctx) {
-               DEBUG(0,("talloc_init(server_service_connection) failed\n"));
-               return;
-       }
-
-       conn = talloc_p(mem_ctx, struct server_connection);
-       if (!conn) {
-               DEBUG(0,("talloc_p(mem_ctx, struct server_service_connection) failed\n"));
-               talloc_destroy(mem_ctx);
-               return;
-       }
-
-       ZERO_STRUCTP(conn);
-       conn->mem_ctx = mem_ctx;
-
-       fde.private     = conn;
-       fde.fd          = accepted_fd;
-       fde.flags       = EVENT_FD_READ;
-       fde.handler     = server_io_handler;
+       status = socket_accept(listen_socket, &connected_socket);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("single_accept_connection: accept: %s\n", nt_errstr(status)));
+               /* this looks strange, but is correct. 
 
-       idle.private    = conn;
-       idle.next_event = t + 300;
-       idle.handler    = server_idle_handler;
+                  We can only be here if woken up from select, due to
+                  an incoming connection.
 
-       conn->event.ctx         = ev;
-       conn->event.fde         = &fde;
-       conn->event.idle        = &idle;
-       conn->event.idle_time   = 300;
+                  We need to throttle things until the system clears
+                  enough resources to handle this new socket. 
 
-       conn->server_socket     = server_socket;
-       conn->service           = server_socket->service;
-
-       /* TODO: we need a generic socket subsystem */
-       conn->socket            = talloc_p(conn->mem_ctx, struct socket_context);
-       if (!conn->socket) {
-               DEBUG(0,("talloc_p(conn->mem_ctx, struct socket_context) failed\n"));
-               talloc_destroy(mem_ctx);
+                  If we don't then we will spin filling the log and
+                  causing more problems. We don't panic as this is
+                  probably a temporary resource constraint */
+               sleep(1);
                return;
        }
-       conn->socket->private_data      = NULL;
-       conn->socket->ops               = NULL;
-       conn->socket->client_addr       = NULL;
-       conn->socket->pkt_count         = 0;
-       conn->socket->fde               = conn->event.fde;
-
-       /* create a smb server context and add it to out event
-          handling */
-       server_socket->service->ops->accept_connection(conn);
-
-       /* accpect_connection() of the service may changed idle.next_event */
-       conn->event.fde         = event_add_fd(ev,&fde);
-       conn->event.idle        = event_add_timed(ev,&idle);
 
-       conn->socket->fde       = conn->event.fde;
+       talloc_steal(private_data, connected_socket);
 
-       DLIST_ADD(server_socket->connection_list,conn);
-
-       /* return to event handling */
-       return;
+       /* The cluster_id(0, fd) cannot collide with the incrementing
+        * task below, as the first component is 0, not 1 */
+       new_conn(ev, lp_ctx, connected_socket,
+                cluster_id(0, socket_get_fd(connected_socket)), private_data);
 }
 
-
-
-/* called when a SMB connection goes down */
-static void single_terminate_connection(struct server_connection *conn, const char *reason) 
+/*
+  called to startup a new task
+*/
+static void single_new_task(struct tevent_context *ev, 
+                           struct loadparm_context *lp_ctx,
+                           const char *service_name,
+                           void (*new_task)(struct tevent_context *, struct loadparm_context *, struct server_id, void *), 
+                           void *private_data)
 {
-       DEBUG(0,("single_terminate_connection: reason[%s]\n",reason));
-       conn->service->ops->close_connection(conn,reason);
-       close(conn->event.fde->fd);
-       event_remove_fd(conn->event.ctx, conn->event.fde);
-       event_remove_timed(conn->event.ctx, conn->event.idle);
+       /* start our taskids at 1, zero is reserved for the top
+          level samba task */
+       static uint32_t taskid = 1;
+       
+       /* We use 1 so we cannot collide in with cluster ids generated
+        * in the accept connection above, and unlikly to collide with
+        * PIDs from process modal standard (don't run samba as
+        * init) */
+       new_task(ev, lp_ctx, cluster_id(1, taskid++), private_data);
 }
 
-static int single_get_id(struct smbsrv_request *req)
+
+/* called when a task goes down */
+static void single_terminate(struct tevent_context *ev, struct loadparm_context *lp_ctx, const char *reason)
 {
-       return (int)req->smb_conn->pid;
+       DEBUG(3,("single_terminate: reason[%s]\n",reason));
 }
 
-static void single_exit_server(struct server_context *srv_ctx, const char *reason)
+/* called to set a title of a task or connection */
+static void single_set_title(struct tevent_context *ev, const char *title) 
 {
-       DEBUG(1,("single_exit_server: reason[%s]\n",reason));
 }
 
+const struct model_ops single_ops = {
+       .name                   = "single",
+       .model_init             = single_model_init,
+       .new_task               = single_new_task,
+       .accept_connection      = single_accept_connection,
+       .terminate              = single_terminate,
+       .set_title              = single_set_title,
+};
+
 /*
-  initialise the single process model, registering ourselves with the process model subsystem
+  initialise the single process model, registering ourselves with the
+  process model subsystem
  */
 NTSTATUS process_model_single_init(void)
 {
-       NTSTATUS ret;
-       struct model_ops ops;
-
-       ZERO_STRUCT(ops);
-
-       /* fill in our name */
-       ops.name = "single";
-
-       /* fill in all the operations */
-       ops.model_startup               = single_start_server;
-       ops.accept_connection           = single_accept_connection;
-       ops.terminate_connection        = single_terminate_connection;
-       ops.exit_server                 = single_exit_server;
-       ops.get_id                      = single_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 'single'!\n"));
-               return ret;
-       }
-
-       return ret;
+       return register_process_model(&single_ops);
 }