r2447: let the server code use the new lib/socket/ stuff
authorStefan Metzmacher <metze@samba.org>
Mon, 20 Sep 2004 12:31:07 +0000 (12:31 +0000)
committerGerald (Jerry) Carter <jerry@samba.org>
Wed, 10 Oct 2007 17:58:54 +0000 (12:58 -0500)
metze
(This used to be commit 2fd577d2417e117a7e8c1a56feb147eae805df34)

source4/ldap_server/ldap_server.c
source4/ntvfs/cifs/vfs_cifs.c
source4/rpc_server/dcerpc_server.c
source4/rpc_server/dcerpc_tcp.c
source4/smb_server/request.c
source4/smb_server/service.c
source4/smb_server/smb_server.c
source4/smbd/process_single.c
source4/smbd/process_standard.c
source4/smbd/process_thread.c
source4/smbd/service.c

index 09df833722368fdd5c17eb1c12abeea2a529faa4..59065e142d8d77d25ba7f138986ccdfaf288174f 100644 (file)
@@ -33,13 +33,15 @@ static void ldapsrv_terminate_connection(struct ldapsrv_connection *ldap_conn, c
   add a socket address to the list of events, one event per port
 */
 static void add_socket(struct server_service *service, 
-                      const struct model_ops *model_ops,
-                      struct socket_context *socket_ctx, 
+                      const struct model_ops *model_ops, 
                       struct in_addr *ifip)
 {
        uint16_t port = 389;
+       char *ip_str = talloc_strdup(service->mem_ctx, inet_ntoa(*ifip));
 
-       service_setup_socket(service, model_ops, socket_ctx, ifip, &port);
+       service_setup_socket(service, model_ops, ip_str, &port);
+
+       talloc_free(ip_str);
 }
 
 /****************************************************************************
@@ -67,7 +69,7 @@ static void ldapsrv_init(struct server_service *service,
                                continue;
                        }
 
-                       add_socket(service, model_ops, NULL, ifip);
+                       add_socket(service, model_ops, ifip);
                }
        } else {
                struct in_addr *ifip;
@@ -79,7 +81,7 @@ static void ldapsrv_init(struct server_service *service,
 
                /* Just bind to lp_socket_address() (usually 0.0.0.0) */
                ifip = interpret_addr2(mem_ctx, lp_socket_address());
-               add_socket(service, model_ops, NULL, ifip);
+               add_socket(service, model_ops, ifip);
 
                talloc_destroy(mem_ctx);
        }
@@ -103,25 +105,42 @@ static BOOL append_to_buf(struct rw_buffer *buf, uint8_t *data, size_t length)
        return True;
 }
 
-static BOOL read_into_buf(int fd, struct rw_buffer *buf)
+static BOOL read_into_buf(struct socket_context *sock, struct rw_buffer *buf)
 {
-       char tmp_buf[1024];
-       int len;
+       NTSTATUS status;
+       DATA_BLOB tmp_blob;
+       BOOL ret;
 
-       len = read(fd, tmp_buf, sizeof(tmp_buf));
-       if (len == 0)
+       status = socket_recv(sock, sock, &tmp_blob, 1024, 0);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("socket_recv: %s\n",nt_errstr(status)));
                return False;
+       }
+
+       ret = append_to_buf(buf, tmp_blob.data, tmp_blob.length);
 
-       return append_to_buf(buf, tmp_buf, len);
+       talloc_free(tmp_blob.data);
+
+       return ret;
 }
 
-static BOOL write_from_buf(int fd, struct rw_buffer *buf)
+static BOOL write_from_buf(struct socket_context *sock, struct rw_buffer *buf)
 {
-       int len;
+       NTSTATUS status;
+       DATA_BLOB tmp_blob;
+       size_t sendlen;
+
+       tmp_blob.data = buf->data;
+       tmp_blob.length = buf->length;
 
-       len = write(fd, buf->data, buf->length);
-       if (len != buf->length)
+       status = socket_send(sock, sock, &tmp_blob, &sendlen, 0);
+       if (!NT_STATUS_IS_OK(status)) {
                return False;
+       }
+
+       if (buf->length != sendlen) {
+               return False;
+       }
 
        return True;
 }
@@ -495,7 +514,7 @@ static void ldapsrv_recv(struct server_connection *conn, time_t t,
 
        DEBUG(10,("ldapsrv_recv\n"));
 
-       if (!read_into_buf(conn->event.fde->fd, &ldap_conn->in_buffer)) {
+       if (!read_into_buf(conn->socket, &ldap_conn->in_buffer)) {
                ldapsrv_terminate_connection(ldap_conn, "read_into_buf() failed");
                return;
        }
@@ -570,7 +589,7 @@ static void ldapsrv_send(struct server_connection *conn, time_t t,
 
        DEBUG(10,("ldapsrv_send\n"));
 
-       if (!write_from_buf(conn->event.fde->fd, &ldap_conn->out_buffer)) {
+       if (!write_from_buf(conn->socket, &ldap_conn->out_buffer)) {
                ldapsrv_terminate_connection(ldap_conn, "write_from_buf() failed");
                return;
        }
@@ -617,7 +636,7 @@ static void ldapsrv_accept(struct server_connection *conn)
        ldap_conn->connection = conn;
 
        conn->private_data = ldap_conn;
-       
+
        return;
 }
 
index ecc2c7ebcb9822992c6da1c4729ffc0dc560ba74..f0fa5dc85b979fd8f33584377123cbf965b3f721 100644 (file)
@@ -51,7 +51,9 @@ struct async_info {
 static void idle_func(struct smbcli_transport *transport, void *p_private)
 {
        struct cvfs_private *private = p_private;
-       if (socket_pending(private->tcon->smb_conn->connection->event.fde->fd)) {
+       int fd = socket_get_fd(private->tcon->smb_conn->connection->socket);
+
+       if (socket_pending(fd)) {
                smbd_process_async(private->tcon->smb_conn);
        }
 }
index e619053f6a4f8009f17e83c6aaa31f0ac1bdb5c2..14b7c14d2a695b5677cac524bd220332b2970226 100644 (file)
@@ -929,7 +929,7 @@ NTSTATUS dcesrv_input(struct dcesrv_connection *dce_conn, const DATA_BLOB *data)
 */
 NTSTATUS dcesrv_output(struct dcesrv_connection *dce_conn, 
                       void *private,
-                      ssize_t (*write_fn)(void *, const void *, size_t))
+                      ssize_t (*write_fn)(void *, DATA_BLOB *))
 {
        struct dcesrv_call_state *call;
        struct dcesrv_call_reply *rep;
@@ -942,7 +942,7 @@ NTSTATUS dcesrv_output(struct dcesrv_connection *dce_conn,
        }
        rep = call->replies;
 
-       nwritten = write_fn(private, rep->data.data, rep->data.length);
+       nwritten = write_fn(private, &rep->data);
        if (nwritten == -1) {
                /* TODO: hmm, how do we cope with this? destroy the
                   connection perhaps? */
@@ -972,13 +972,13 @@ NTSTATUS dcesrv_output(struct dcesrv_connection *dce_conn,
 /*
   write_fn() for dcesrv_output_blob()
 */
-static ssize_t dcesrv_output_blob_write_fn(void *private, const void *buf, size_t count)
+static ssize_t dcesrv_output_blob_write_fn(void *private, DATA_BLOB *out)
 {
        DATA_BLOB *blob = private;
-       if (count < blob->length) {
-               blob->length = count;
+       if (out->length < blob->length) {
+               blob->length = out->length;
        }
-       memcpy(blob->data, buf, blob->length);
+       memcpy(blob->data, out->data, blob->length);
        return blob->length;
 }
 
index 26ced8c0874c1b4869b9165e0cf2d61dc2de882c..c5f88d467f6af1942807c166f33cdd5ae71a2815 100644 (file)
@@ -31,15 +31,18 @@ struct dcesrv_socket_context {
 /*
   write_fn callback for dcesrv_output()
 */
-static ssize_t dcerpc_write_fn(void *private, const void *buf, size_t count)
+static ssize_t dcerpc_write_fn(void *private, DATA_BLOB *out)
 {
-       struct fd_event *fde = private;
-       ssize_t ret;
-       ret = write(fde->fd, buf, count);
-       if (ret == -1 && errno == EINTR) {
-               return 0;
+       NTSTATUS status;
+       struct socket_context *sock = private;
+       size_t sendlen;
+
+       status = socket_send(sock, sock, out, &sendlen, 0);
+       if (!NT_STATUS_IS_OK(status)) {
+               return -1;
        }
-       return ret;
+
+       return sendlen;
 }
 
 void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char *reason)
@@ -52,24 +55,24 @@ void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char
 */
 static void add_socket_rpc(struct server_service *service, 
                       const struct model_ops *model_ops,
-                      struct socket_context *socket_ctx,
                       struct dcesrv_context *dce_ctx, 
                       struct in_addr *ifip)
 {
        struct dcesrv_endpoint *e;
+       char *ip_str = talloc_strdup(service->mem_ctx, inet_ntoa(*ifip));
 
        for (e=dce_ctx->endpoint_list;e;e=e->next) {
                if (e->ep_description.type == ENDPOINT_TCP) {
                        struct server_socket *sock;
                        struct dcesrv_socket_context *dcesrv_sock;
 
-                       sock = service_setup_socket(service,model_ops,socket_ctx,ifip, &e->ep_description.info.tcp_port);
+                       sock = service_setup_socket(service,model_ops, ip_str, &e->ep_description.info.tcp_port);
                        if (!sock) {
                                DEBUG(0,("service_setup_socket(port=%u) failed\n",e->ep_description.info.tcp_port));
                                continue;
                        }
 
-                       dcesrv_sock = talloc_p(sock->mem_ctx, struct dcesrv_socket_context);
+                       dcesrv_sock = talloc_p(sock, struct dcesrv_socket_context);
                        if (!dcesrv_sock) {
                                DEBUG(0,("talloc_p(sock->mem_ctx, struct dcesrv_socket_context) failed\n"));
                                continue;
@@ -78,10 +81,12 @@ static void add_socket_rpc(struct server_service *service,
                        /* remeber the enpoint of this socket */
                        dcesrv_sock->endpoint           = e;
                        dcesrv_sock->dcesrv_ctx         = dce_ctx;
-                       
+
                        sock->private_data = dcesrv_sock;
                }
        }
+
+       talloc_free(ip_str);
 }
 
 /****************************************************************************
@@ -99,7 +104,7 @@ void dcesrv_tcp_init(struct server_service *service, const struct model_ops *mod
                        if (ifip == NULL) {
                                continue;
                        }
-                       add_socket_rpc(service, model_ops, NULL, dce_ctx,  ifip);
+                       add_socket_rpc(service, model_ops, dce_ctx,  ifip);
                }
        } else {
                struct in_addr *ifip;
@@ -109,7 +114,7 @@ void dcesrv_tcp_init(struct server_service *service, const struct model_ops *mod
                }
 
                ifip = interpret_addr2(mem_ctx, lp_socket_address());
-               add_socket_rpc(service, model_ops, NULL, dce_ctx,  ifip);
+               add_socket_rpc(service, model_ops, dce_ctx,  ifip);
                talloc_destroy(mem_ctx);
        }
 
@@ -135,45 +140,32 @@ void dcesrv_tcp_accept(struct server_connection *conn)
 
        conn->private_data = dcesrv_conn;
 
-       /* TODO: this should to the generic code
-        *       but the smb server can't handle it yet
-        *       --metze
-        */ 
-       set_blocking(conn->event.fde->fd, False);
-
        return; 
 }
 
 void dcesrv_tcp_recv(struct server_connection *conn, time_t t, uint16_t flags)
 {
+       NTSTATUS status;
        struct dcesrv_connection *dce_conn = conn->private_data;
-       DATA_BLOB blob;
-       ssize_t ret;
+       DATA_BLOB tmp_blob;
 
        DEBUG(10,("dcesrv_tcp_recv\n"));
 
-       blob = data_blob(NULL, 0x4000);
-       if (!blob.data) {
-               dcesrv_terminate_connection(dce_conn, "eof on socket");
+       status = socket_recv(conn->socket, conn->socket, &tmp_blob, 0x4000, 0);
+       if (!NT_STATUS_IS_OK(status)) {
+               if (NT_STATUS_IS_ERR(status)) {
+                       dcesrv_terminate_connection(dce_conn, "eof on socket");
+                       return;
+               }
                return;
        }
 
-       ret = read(conn->event.fde->fd, blob.data, blob.length);
-       if (ret == 0 || (ret == -1 && errno != EINTR)) {
-               data_blob_free(&blob);
+       status = dcesrv_input(dce_conn, &tmp_blob);
+       talloc_free(tmp_blob.data);
+       if (!NT_STATUS_IS_OK(status)) {
                dcesrv_terminate_connection(dce_conn, "eof on socket");
                return;
        }
-       if (ret == -1) {
-               data_blob_free(&blob);
-               return;
-       }
-
-       blob.length = ret;
-
-       dcesrv_input(dce_conn, &blob);
-
-       data_blob_free(&blob);
 
        if (dce_conn->call_list && dce_conn->call_list->replies) {
                conn->event.fde->flags |= EVENT_FD_WRITE;
@@ -189,16 +181,17 @@ void dcesrv_tcp_send(struct server_connection *conn, time_t t, uint16_t flags)
 
        DEBUG(10,("dcesrv_tcp_send\n"));
 
-       status = dcesrv_output(dce_conn, conn->event.fde, dcerpc_write_fn);
-       if (NT_STATUS_IS_ERR(status)) {
-               /* TODO: destroy fd_event? */
+       status = dcesrv_output(dce_conn, conn->socket, dcerpc_write_fn);
+       if (!NT_STATUS_IS_OK(status)) {
+               dcesrv_terminate_connection(dce_conn, "eof on socket");
+               return;
        }
 
        if (!dce_conn->call_list || !dce_conn->call_list->replies) {
                conn->event.fde->flags &= ~EVENT_FD_WRITE;
        }
 
-       return; 
+       return;
 }
 
 void dcesrv_tcp_idle(struct server_connection *conn, time_t t)
index a5ca789a396c919abd79ef0859b87bdbf387582e..80663b9ccc8b99980ae13c9f43b20caffb9c3ce7 100644 (file)
@@ -247,11 +247,19 @@ void req_grow_data(struct smbsrv_request *req, uint_t new_size)
 */
 void req_send_reply_nosign(struct smbsrv_request *req)
 {
+       NTSTATUS status;
+       DATA_BLOB tmp_blob;
+       size_t sendlen;
+
        if (req->out.size > NBT_HDR_SIZE) {
                _smb_setlen(req->out.buffer, req->out.size - NBT_HDR_SIZE);
        }
 
-       if (write_data(req->smb_conn->connection->event.fde->fd, req->out.buffer, req->out.size) != req->out.size) {
+       tmp_blob.data = req->out.buffer;
+       tmp_blob.length = req->out.size;
+
+       status = socket_send(req->smb_conn->connection->socket, req, &tmp_blob, &sendlen, SOCKET_FLAG_BLOCK);
+       if (!NT_STATUS_IS_OK(status) || (req->out.size != sendlen)) {
                smbsrv_terminate_connection(req->smb_conn, "failed to send reply\n");
        }
 
index 54e96476fe0c4a83f1395d31a820abe823272961..843b0a3872528b2cbff2e1f857bd750c3fdd0cbd 100644 (file)
@@ -249,7 +249,7 @@ close a cnum
 void close_cnum(struct smbsrv_tcon *tcon)
 {
        DEBUG(3,("%s closed connection to service %s\n",
-                get_socket_addr(tcon, tcon->smb_conn->connection->event.fde->fd),
+                socket_get_peer_addr(tcon->smb_conn->connection->socket, tcon),
                 lp_servicename(SNUM(tcon))));
 
        /* tell the ntvfs backend that we are disconnecting */
index 228ff4f4a3a8971e7032fc77a5ca688655320ce6..516babf4b72fc27dfb068183138730ae7c52ae4d 100644 (file)
@@ -60,35 +60,40 @@ receive a SMB request from the wire, forming a request_context from the result
 ****************************************************************************/
 static struct smbsrv_request *receive_smb_request(struct smbsrv_connection *smb_conn)
 {
+       NTSTATUS status;
        ssize_t len, len2;
-       char header[4];
+       DATA_BLOB tmp_blob;
        struct smbsrv_request *req;
 
-       len = read_data(smb_conn->connection->event.fde->fd, header, 4);
-       if (len != 4) {
+       status = socket_recv(smb_conn->connection->socket, smb_conn, &tmp_blob, 4, SOCKET_FLAG_BLOCK|SOCKET_FLAG_PEEK);
+       if (!NT_STATUS_IS_OK(status)) {
+               return NULL;
+       }
+       if (tmp_blob.length != 4) {
                return NULL;
        }
 
-       len = smb_len(header);
+       len = smb_len(tmp_blob.data);
+       talloc_free(tmp_blob.data);
 
        req = init_smb_request(smb_conn);
 
        GetTimeOfDay(&req->request_time);
        req->chained_fnum = -1;
-       
-       /* allocate the incoming buffer at the right size */
-       req->in.buffer = talloc(req, len + NBT_HDR_SIZE);
 
-       /* fill in the already received header */
-       memcpy(req->in.buffer, header, 4);
+       len2 = len + NBT_HDR_SIZE;
 
-       len2 = read_data(smb_conn->connection->event.fde->fd, req->in.buffer + NBT_HDR_SIZE, len);
-       if (len2 != len) {
+       status = socket_recv(smb_conn->connection->socket, req, &tmp_blob, len2, SOCKET_FLAG_BLOCK);
+       if (!NT_STATUS_IS_OK(status)) {
+               return NULL;
+       }
+       if (tmp_blob.length != len2) {
                return NULL;
        }
 
        /* fill in the rest of the req->in structure */
-       req->in.size = len + NBT_HDR_SIZE;
+       req->in.buffer = tmp_blob.data;
+       req->in.size = len2;
        req->in.allocated = req->in.size;
        req->in.hdr = req->in.buffer + NBT_HDR_SIZE;
        req->in.vwv = req->in.hdr + HDR_VWV;
@@ -683,12 +688,15 @@ static void add_socket(struct server_service *service,
 {
        const char **ports = lp_smb_ports();
        int i;
+       char *ip_str = talloc_strdup(service->mem_ctx, inet_ntoa(*ifip));
 
        for (i=0;ports[i];i++) {
                uint16_t port = atoi(ports[i]);
                if (port == 0) continue;
-               service_setup_socket(service, model_ops, socket_ctx, ifip, &port);
+               service_setup_socket(service, model_ops, ip_str, &port);
        }
+
+       talloc_free(ip_str);
 }
 
 /****************************************************************************
@@ -831,8 +839,10 @@ void smbsrv_accept(struct server_connection *conn)
 
        /* set an initial client name based on its IP address. This will be replaced with
           the netbios name later if it gives us one */
-       socket_addr = get_socket_addr(smb_conn, conn->event.fde->fd);
-       sub_set_remote_machine(socket_addr);
+       socket_addr = socket_get_peer_addr(conn->socket, smb_conn);
+       if (socket_addr) {
+               sub_set_remote_machine(socket_addr);
+       }
 
        /* now initialise a few default values associated with this smb socket */
        smb_conn->negotiate.max_send = 0xFFFF;
index 8bd00c53b95e2c66bb494a82fe9881afbcd6f7d8..f3b69eae28b96776883e0fcf2e7fb80aac84403b 100644 (file)
@@ -36,23 +36,22 @@ static void single_start_server(void)
 */
 static void single_accept_connection(struct event_context *ev, struct fd_event *srv_fde, time_t t, uint16_t flags)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
+       NTSTATUS status;
+       struct socket_context *sock;
        struct server_socket *server_socket = srv_fde->private;
        struct server_connection *conn;
 
        /* accept an incoming connection. */
-       accepted_fd = accept(srv_fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
+       status = socket_accept(server_socket->socket, &sock, 0);
+       if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0,("accept_connection_single: accept: %s\n",
-                        strerror(errno)));
+                        nt_errstr(status)));
                return;
        }
 
-       conn = server_setup_connection(ev, server_socket, accepted_fd, t);
+       conn = server_setup_connection(ev, server_socket, sock, t);
        if (!conn) {
-               DEBUG(0,("server_setup_connection(ev, server_socket, accepted_fd) failed\n"));
+               DEBUG(0,("server_setup_connection(ev, server_socket, sock, t) failed\n"));
                return;
        }
 
index c12a1f4305edfd9b8d9f16853104a01466b5a533..9dfda41db3b98117bef28eeea2dc3a8ae29e7800 100644 (file)
@@ -36,18 +36,17 @@ static void standard_model_startup(void)
 */
 static void standard_accept_connection(struct event_context *ev, struct fd_event *srv_fde, time_t t, uint16_t flags)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
-       pid_t pid;
+       NTSTATUS status;
+       struct socket_context *sock;
        struct server_socket *server_socket = srv_fde->private;
        struct server_connection *conn;
+       pid_t pid;
 
        /* accept an incoming connection. */
-       accepted_fd = accept(srv_fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("standard_accept_connection: accept: %s\n",
-                        strerror(errno)));
+       status = socket_accept(server_socket->socket, &sock, 0);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("accept_connection_single: accept: %s\n",
+                        nt_errstr(status)));
                return;
        }
 
@@ -56,7 +55,7 @@ static void standard_accept_connection(struct event_context *ev, struct fd_event
        if (pid != 0) {
                /* parent or error code ... */
 
-               close(accepted_fd);
+               socket_destroy(sock);
                /* go back to the event loop */
                return;
        }
@@ -75,9 +74,9 @@ static void standard_accept_connection(struct event_context *ev, struct fd_event
 
        set_need_random_reseed();
 
-       conn = server_setup_connection(ev, server_socket, accepted_fd, t);
+       conn = server_setup_connection(ev, server_socket, sock, t);
        if (!conn) {
-               DEBUG(0,("server_setup_connection(ev, server_socket, accepted_fd) failed\n"));
+               DEBUG(0,("server_setup_connection(ev, server_socket, sock, t) failed\n"));
                return;
        }
 
index ced07d5d76ce925f2e43c4c3b12402f0aa2ac409..ef4d53e1880444e3a26c7f603b670ae3c7793913 100644 (file)
@@ -49,19 +49,19 @@ static int thread_get_id(struct smbsrv_request *req)
 static void thread_accept_connection(struct event_context *ev, struct fd_event *srv_fde, 
                              time_t t, uint16_t flags)
 {              
-       int accepted_fd, rc;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
+       NTSTATUS status;
+       struct socket_context *sock;
+       int rc;
        pthread_t thread_id;
        pthread_attr_t thread_attr;
        struct server_socket *server_socket = srv_fde->private;
        struct server_connection *conn;
 
        /* accept an incoming connection. */
-       accepted_fd = accept(srv_fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("standard_accept_connection: accept: %s\n",
-                        strerror(errno)));
+       status = socket_accept(server_socket->socket, &sock, 0);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("accept_connection_single: accept: %s\n",
+                        nt_errstr(status)));
                return;
        }
        
@@ -72,16 +72,18 @@ static void thread_accept_connection(struct event_context *ev, struct fd_event *
           main event_context is continued.
        */
 
-
        ev = event_context_init();
        if (!ev) {
                DEBUG(0,("thread_accept_connection: failed to create event_context!\n"));
+               socket_destroy(sock);
                return; 
        }
 
-       conn = server_setup_connection(ev, server_socket, accepted_fd, t);
+       conn = server_setup_connection(ev, server_socket, sock, t);
        if (!conn) {
-               DEBUG(0,("server_setup_connection(ev, server_socket, accepted_fd) failed\n"));
+               DEBUG(0,("server_setup_connection(ev, server_socket, sock, t) failed\n"));
+               event_context_destroy(ev);
+               socket_destroy(sock);
                return;
        }
 
@@ -98,9 +100,12 @@ static void thread_accept_connection(struct event_context *ev, struct fd_event *
        pthread_attr_destroy(&thread_attr);
        if (rc == 0) {
                DEBUG(4,("accept_connection_thread: created thread_id=%lu for fd=%d\n", 
-                       (unsigned long int)thread_id, accepted_fd));
+                       (unsigned long int)thread_id, socket_get_fd(sock)));
        } else {
-               DEBUG(0,("accept_connection_thread: thread create failed for fd=%d, rc=%d\n", accepted_fd, rc));
+               DEBUG(0,("accept_connection_thread: thread create failed for fd=%d, rc=%d\n", socket_get_fd(sock), rc));
+               event_context_destroy(ev);
+               socket_destroy(sock);
+               return;
        }
 }
 
index bcdb953e56672f72273db0ea6c089c6b52616824..22d8f237654b91fb90271ab38d05884751a7a63e 100644 (file)
@@ -97,103 +97,111 @@ struct server_context *server_service_startup(const char *model)
 }
 
 /*
-  setup a single listener of any type
-  if you pass *port == 0, then a port < 1024 is used
+  setup a listen stream socket
+  if you pass *port == 0, then a port > 1024 is used
  */
 struct server_socket *service_setup_socket(struct server_service *service,
-                        const struct model_ops *model_ops,
-                        struct socket_context *socket_ctx,
-                        struct in_addr *ifip, uint16_t *port)
+                        const struct model_ops *model_ops,
+                        const char *sock_addr,
+                        uint16_t *port)
 {
-       TALLOC_CTX *mem_ctx;
-       struct server_socket *sock;
+       NTSTATUS status;
+       struct server_socket *srv_sock;
+       struct socket_context *socket_ctx;
        struct fd_event fde;
        int i;
 
-       mem_ctx = talloc_init("struct server_socket");
-
-       sock = talloc_p(mem_ctx, struct server_socket);
-       if (!sock) {
-               DEBUG(0,("talloc_p(mem_ctx, struct server_socket) failed\n"));
-               return NULL;    
-       }
-
        if (*port == 0) {
-               fde.fd = -1;
                for (i=SERVER_TCP_LOW_PORT;i<= SERVER_TCP_HIGH_PORT;i++) {
-                       fde.fd = open_socket_in(SOCK_STREAM, i, 0, ifip->s_addr, True);                 
-                       if (fde.fd != -1) break;
-               }
-               if (fde.fd != -1) {
-                       *port = i;
+                       status = socket_create("ipv4", SOCKET_TYPE_STREAM, &socket_ctx, 0);
+                       if (NT_STATUS_IS_OK(status)) {
+                               *port = i;
+                               break;
+                       }
                }
        } else {
-               fde.fd = open_socket_in(SOCK_STREAM, *port, 0, ifip->s_addr, True);
+               status = socket_create("ipv4", SOCKET_TYPE_STREAM, &socket_ctx, 0);
        }
 
-       if (fde.fd == -1) {
+       if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0,("Failed to open socket on %s:%u - %s\n",
-                        inet_ntoa(*ifip), *port, strerror(errno)));
+                       sock_addr, *port, nt_errstr(status)));
                return NULL;
        }
 
        /* ready to listen */
-       set_socket_options(fde.fd, "SO_KEEPALIVE"); 
-       set_socket_options(fde.fd, lp_socket_options());
-      
-       if (listen(fde.fd, SERVER_LISTEN_BACKLOG) == -1) {
+       status = socket_set_option(socket_ctx, "SO_KEEPALIVE", NULL);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("socket_set_option(socket_ctx, SO_KEEPALIVE, NULL): %s\n",
+                       nt_errstr(status)));
+               socket_destroy(socket_ctx);
+               return NULL;
+       }
+       status = socket_set_option(socket_ctx, lp_socket_options(), NULL);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("socket_set_option(socket_ctx, lp_socket_options(), NULL): %s\n",
+                       nt_errstr(status)));
+               socket_destroy(socket_ctx);
+               return NULL;
+       }
+
+       /* TODO: set socket ACL's here when they're implemented */
+
+       status = socket_listen(socket_ctx, sock_addr, *port, SERVER_LISTEN_BACKLOG, 0);
+       if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0,("Failed to listen on %s:%u - %s\n",
-                        inet_ntoa(*ifip), *port, strerror(errno)));
-               close(fde.fd);
+                       sock_addr, *port, nt_errstr(status)));
+               socket_destroy(socket_ctx);
+               return NULL;
+       }
+
+       srv_sock = talloc_p(NULL, struct server_socket);
+       if (!srv_sock) {
+               DEBUG(0,("talloc_p(mem_ctx, struct server_socket) failed\n"));
+               socket_destroy(socket_ctx);
                return NULL;
        }
 
        /* we are only interested in read events on the listen socket */
-       fde.flags = EVENT_FD_READ;
-       fde.private = sock;
-       fde.handler = model_ops->accept_connection;
-
-       ZERO_STRUCTP(sock);
-       sock->mem_ctx   = mem_ctx;
-       sock->service   = service;
-       sock->socket    = socket_ctx;
-       sock->event.ctx = service->srv_ctx->events;
-       sock->event.fde = event_add_fd(sock->event.ctx, &fde);
-       if (!sock->event.fde) {
-               DEBUG(0,("event_add_fd(sock->event.ctx, &fde) failed\n"));
+       fde.fd          = socket_get_fd(socket_ctx);
+       fde.flags       = EVENT_FD_READ;
+       fde.private     = srv_sock;
+       fde.handler     = model_ops->accept_connection;
+
+       ZERO_STRUCTP(srv_sock);
+       srv_sock->mem_ctx       = srv_sock;
+       srv_sock->service       = service;
+       srv_sock->socket        = socket_ctx;
+       srv_sock->event.ctx     = service->srv_ctx->events;
+       srv_sock->event.fde     = event_add_fd(srv_sock->event.ctx, &fde);
+       if (!srv_sock->event.fde) {
+               DEBUG(0,("event_add_fd(srv_sock->event.ctx, &fde) failed\n"));
+               socket_destroy(socket_ctx);
                return NULL;
        }
 
-       DLIST_ADD(service->socket_list, sock);
+       DLIST_ADD(service->socket_list, srv_sock);
 
-       return sock;
+       return srv_sock;
 }
 
-struct server_connection *server_setup_connection(struct event_context *ev, struct server_socket *server_socket, int accepted_fd, time_t t)
+struct server_connection *server_setup_connection(struct event_context *ev, struct server_socket *server_socket, struct socket_context *sock, time_t t)
 {
        struct fd_event fde;
        struct timed_event idle;
        struct server_connection *srv_conn;
-       TALLOC_CTX *mem_ctx;
 
-       mem_ctx = talloc_init("server_service_connection");
-       if (!mem_ctx) {
-               DEBUG(0,("talloc_init(server_service_connection) failed\n"));
-               return NULL;
-       }
-
-       srv_conn = talloc_p(mem_ctx, struct server_connection);
+       srv_conn = talloc_p(NULL, struct server_connection);
        if (!srv_conn) {
                DEBUG(0,("talloc_p(mem_ctx, struct server_service_connection) failed\n"));
-               talloc_destroy(mem_ctx);
                return NULL;
        }
 
        ZERO_STRUCTP(srv_conn);
-       srv_conn->mem_ctx = mem_ctx;
+       srv_conn->mem_ctx = srv_conn;
 
        fde.private     = srv_conn;
-       fde.fd          = accepted_fd;
+       fde.fd          = socket_get_fd(sock);
        fde.flags       = EVENT_FD_READ;
        fde.handler     = server_io_handler;
 
@@ -208,9 +216,7 @@ struct server_connection *server_setup_connection(struct event_context *ev, stru
 
        srv_conn->server_socket         = server_socket;
        srv_conn->service               = server_socket->service;
-
-       /* TODO: we need a generic socket subsystem */
-       srv_conn->socket                = NULL;
+       srv_conn->socket                = sock;
 
        /* create a smb server context and add it to out event
           handling */
@@ -234,7 +240,7 @@ void server_terminate_connection(struct server_connection *srv_conn, const char
 
 void server_destroy_connection(struct server_connection *srv_conn)
 {
-       close(srv_conn->event.fde->fd);
+       socket_destroy(srv_conn->socket);
 
        event_remove_fd(srv_conn->event.ctx, srv_conn->event.fde);
        srv_conn->event.fde = NULL;