s3: ?true:false is a *bit* pointless :-)
[amitay/samba.git] / source3 / smbd / process.c
index f032e6ab785c2bca739abc6d77d853e71a31b739..2e319135c7bfe9a5aa0eca5850fb4dcc9417b8c3 100644 (file)
@@ -35,6 +35,7 @@
 #include "../librpc/gen_ndr/srv_winreg.h"
 #include "../librpc/gen_ndr/srv_wkssvc.h"
 #include "librpc/gen_ndr/messaging.h"
+#include "printing/nt_printing_migrate.h"
 
 extern bool global_machine_password_needs_changing;
 
@@ -46,19 +47,19 @@ static bool smbd_lock_socket_internal(struct smbd_server_connection *sconn)
 {
        bool ok;
 
-       if (smbd_server_conn->smb1.echo_handler.socket_lock_fd == -1) {
+       if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
                return true;
        }
 
-       smbd_server_conn->smb1.echo_handler.ref_count++;
+       sconn->smb1.echo_handler.ref_count++;
 
-       if (smbd_server_conn->smb1.echo_handler.ref_count > 1) {
+       if (sconn->smb1.echo_handler.ref_count > 1) {
                return true;
        }
 
        DEBUG(10,("pid[%d] wait for socket lock\n", (int)sys_getpid()));
 
-       ok = fcntl_lock(smbd_server_conn->smb1.echo_handler.socket_lock_fd,
+       ok = fcntl_lock(sconn->smb1.echo_handler.socket_lock_fd,
                        SMB_F_SETLKW, 0, 0, F_WRLCK);
        if (!ok) {
                return false;
@@ -80,17 +81,17 @@ static bool smbd_unlock_socket_internal(struct smbd_server_connection *sconn)
 {
        bool ok;
 
-       if (smbd_server_conn->smb1.echo_handler.socket_lock_fd == -1) {
+       if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
                return true;
        }
 
-       smbd_server_conn->smb1.echo_handler.ref_count--;
+       sconn->smb1.echo_handler.ref_count--;
 
-       if (smbd_server_conn->smb1.echo_handler.ref_count > 0) {
+       if (sconn->smb1.echo_handler.ref_count > 0) {
                return true;
        }
 
-       ok = fcntl_lock(smbd_server_conn->smb1.echo_handler.socket_lock_fd,
+       ok = fcntl_lock(sconn->smb1.echo_handler.socket_lock_fd,
                        SMB_F_SETLKW, 0, 0, F_UNLCK);
        if (!ok) {
                return false;
@@ -263,7 +264,8 @@ static NTSTATUS receive_smb_raw_talloc_partial_read(TALLOC_CTX *mem_ctx,
         * valid writeX call.
         */
 
-       if (is_valid_writeX_buffer((uint8_t *)writeX_header)) {
+       if (is_valid_writeX_buffer(smbd_server_conn,
+                                  (uint8_t *)writeX_header)) {
                /*
                 * If the data offset is beyond what
                 * we've read, drain the extra bytes.
@@ -440,11 +442,12 @@ static NTSTATUS receive_smb_talloc(TALLOC_CTX *mem_ctx,   int fd,
  * Initialize a struct smb_request from an inbuf
  */
 
-static bool init_smb_request(struct smb_request *req, const uint8 *inbuf,
+static bool init_smb_request(struct smb_request *req,
+                            struct smbd_server_connection *sconn,
+                            const uint8 *inbuf,
                             size_t unread_bytes, bool encrypted,
                             uint32_t seqnum)
 {
-       struct smbd_server_connection *sconn = smbd_server_conn;
        size_t req_size = smb_len(inbuf) + 4;
        /* Ensure we have at least smb_size bytes. */
        if (req_size < smb_size) {
@@ -465,6 +468,7 @@ static bool init_smb_request(struct smb_request *req, const uint8 *inbuf,
        req->buf    = (const uint8_t *)smb_buf(inbuf);
        req->unread_bytes = unread_bytes;
        req->encrypted = encrypted;
+       req->sconn = sconn;
        req->conn = conn_find(sconn,req->tid);
        req->chain_fsp = NULL;
        req->chain_outbuf = NULL;
@@ -606,8 +610,8 @@ void remove_deferred_open_message_smb(uint64_t mid)
 {
        struct pending_message_list *pml;
 
-       if (smbd_server_conn->allow_smb2) {
-               remove_deferred_open_message_smb2(mid);
+       if (smbd_server_conn->using_smb2) {
+               remove_deferred_open_message_smb2(smbd_server_conn, mid);
                return;
        }
 
@@ -634,8 +638,8 @@ void schedule_deferred_open_message_smb(uint64_t mid)
        struct pending_message_list *pml;
        int i = 0;
 
-       if (smbd_server_conn->allow_smb2) {
-               schedule_deferred_open_message_smb2(mid);
+       if (smbd_server_conn->using_smb2) {
+               schedule_deferred_open_message_smb2(smbd_server_conn, mid);
                return;
        }
 
@@ -695,8 +699,8 @@ bool open_was_deferred(uint64_t mid)
 {
        struct pending_message_list *pml;
 
-       if (smbd_server_conn->allow_smb2) {
-               return open_was_deferred_smb2(mid);
+       if (smbd_server_conn->using_smb2) {
+               return open_was_deferred_smb2(smbd_server_conn, mid);
        }
 
        for (pml = deferred_open_queue; pml; pml = pml->next) {
@@ -733,7 +737,7 @@ bool get_deferred_open_message_state(struct smb_request *smbreq,
 {
        struct pending_message_list *pml;
 
-       if (smbd_server_conn->allow_smb2) {
+       if (smbd_server_conn->using_smb2) {
                return get_deferred_open_message_state_smb2(smbreq->smb2req,
                                        p_request_time,
                                        pp_state);
@@ -906,20 +910,20 @@ static void smbd_sig_hup_handler(struct tevent_context *ev,
                                  void *siginfo,
                                  void *private_data)
 {
+       struct messaging_context *msg_ctx = talloc_get_type_abort(
+               private_data, struct messaging_context);
        change_to_root_user();
        DEBUG(1,("Reloading services after SIGHUP\n"));
-       reload_services(False);
+       reload_services(msg_ctx, False);
 }
 
-void smbd_setup_sig_hup_handler(void)
+void smbd_setup_sig_hup_handler(struct tevent_context *ev,
+                               struct messaging_context *msg_ctx)
 {
        struct tevent_signal *se;
 
-       se = tevent_add_signal(smbd_event_context(),
-                              smbd_event_context(),
-                              SIGHUP, 0,
-                              smbd_sig_hup_handler,
-                              NULL);
+       se = tevent_add_signal(ev, ev, SIGHUP, 0, smbd_sig_hup_handler,
+                              msg_ctx);
        if (!se) {
                exit_server("failed to setup SIGHUP handler");
        }
@@ -971,6 +975,17 @@ static NTSTATUS smbd_server_connection_loop_once(struct smbd_server_connection *
                errno = sav;
        }
 
+        if ((conn->smb1.echo_handler.trusted_fd != -1)
+           && FD_ISSET(smbd_server_fd(), &r_fds)
+           && FD_ISSET(conn->smb1.echo_handler.trusted_fd, &r_fds)) {
+               /*
+                * Prefer to read pending requests from the echo handler. To
+                * quote Jeremy (da70f8ab1): This is a hack of monstrous
+                * proportions...
+                */
+               FD_CLR(smbd_server_fd(), &r_fds);
+        }
+
        if (run_events(smbd_event_context(), selrtn, &r_fds, &w_fds)) {
                return NT_STATUS_RETRY;
        }
@@ -1397,7 +1412,7 @@ static connection_struct *switch_message(uint8 type, struct smb_request *req, in
        int flags;
        uint16 session_tag;
        connection_struct *conn = NULL;
-       struct smbd_server_connection *sconn = smbd_server_conn;
+       struct smbd_server_connection *sconn = req->sconn;
 
        errno = 0;
 
@@ -1550,8 +1565,8 @@ static void construct_reply(char *inbuf, int size, size_t unread_bytes,
                smb_panic("could not allocate smb_request");
        }
 
-       if (!init_smb_request(req, (uint8 *)inbuf, unread_bytes, encrypted,
-                             seqnum)) {
+       if (!init_smb_request(req, smbd_server_conn, (uint8 *)inbuf,
+                             unread_bytes, encrypted, seqnum)) {
                exit_server_cleanly("Invalid SMB request");
        }
 
@@ -1625,11 +1640,13 @@ static void process_smb(struct smbd_server_connection *conn,
                /*
                 * NetBIOS session request, keepalive, etc.
                 */
-               reply_special((char *)inbuf);
+               reply_special(conn, (char *)inbuf);
                goto done;
        }
 
-       if (smbd_server_conn->allow_smb2) {
+       if (smbd_server_conn->using_smb2) {
+               /* At this point we're not really using smb2,
+                * we make the decision here.. */
                if (smbd_is_smb2_header(inbuf, nread)) {
                        smbd_smb2_first_negprot(smbd_server_conn, inbuf, nread);
                        return;
@@ -1637,7 +1654,7 @@ static void process_smb(struct smbd_server_connection *conn,
                                && CVAL(inbuf, smb_com) != 0x72) {
                        /* This is a non-negprot SMB1 packet.
                           Disable SMB2 from now on. */
-                       smbd_server_conn->allow_smb2 = false;
+                       smbd_server_conn->using_smb2 = false;
                }
        }
 
@@ -2181,7 +2198,7 @@ void chain_reply(struct smb_request *req)
  Check if services need reloading.
 ****************************************************************************/
 
-void check_reload(time_t t)
+static void check_reload(struct messaging_context *msg_ctx, time_t t)
 {
        time_t printcap_cache_time = (time_t)lp_printcap_cache_time();
 
@@ -2206,7 +2223,7 @@ void check_reload(time_t t)
        }
 
        if (t >= last_smb_conf_reload_time+SMBD_RELOAD_CHECK) {
-               reload_services(True);
+               reload_services(msg_ctx, True);
                last_smb_conf_reload_time = t;
        }
 
@@ -2220,7 +2237,7 @@ void check_reload(time_t t)
                        || (t-last_printer_reload_time  < 0) ) 
                {
                        DEBUG( 3,( "Printcap cache time expired.\n"));
-                       reload_printers();
+                       reload_printers(msg_ctx);
                        last_printer_reload_time = t;
                }
        }
@@ -2258,7 +2275,7 @@ static void smbd_server_connection_read_handler(
        NTSTATUS status;
        uint32_t seqnum;
 
-       bool from_client = (smbd_server_fd() == fd)?true:false;
+       bool from_client = (smbd_server_fd() == fd);
 
        if (from_client) {
                smbd_lock_socket(conn);
@@ -2397,7 +2414,7 @@ static bool keepalive_fn(const struct timeval *now, void *private_data)
        struct smbd_server_connection *sconn = smbd_server_conn;
        bool ret;
 
-       if (sconn->allow_smb2) {
+       if (sconn->using_smb2) {
                /* Don't do keepalives on an SMB2 connection. */
                return false;
        }
@@ -2418,15 +2435,16 @@ static bool keepalive_fn(const struct timeval *now, void *private_data)
  */
 static bool deadtime_fn(const struct timeval *now, void *private_data)
 {
-       struct smbd_server_connection *sconn = smbd_server_conn;
+       struct smbd_server_connection *sconn =
+               (struct smbd_server_connection *)private_data;
 
-       if (sconn->allow_smb2) {
+       if (sconn->using_smb2) {
                /* TODO: implement real idle check */
                if (sconn->smb2.sessions.list) {
                        return true;
                }
                DEBUG( 2, ( "Closing idle SMB2 connection\n" ) );
-               messaging_send(smbd_messaging_context(), procid_self(),
+               messaging_send(sconn->msg_ctx, procid_self(),
                               MSG_SHUTDOWN, &data_blob_null);
                return false;
        }
@@ -2434,7 +2452,7 @@ static bool deadtime_fn(const struct timeval *now, void *private_data)
        if ((conn_num_open(sconn) == 0)
            || (conn_idle_all(sconn, now->tv_sec))) {
                DEBUG( 2, ( "Closing idle SMB1 connection\n" ) );
-               messaging_send(smbd_messaging_context(), procid_self(),
+               messaging_send(sconn->msg_ctx, procid_self(),
                               MSG_SHUTDOWN, &data_blob_null);
                return False;
        }
@@ -2448,13 +2466,15 @@ static bool deadtime_fn(const struct timeval *now, void *private_data)
 
 static bool housekeeping_fn(const struct timeval *now, void *private_data)
 {
+       struct messaging_context *msg_ctx = talloc_get_type_abort(
+               private_data, struct messaging_context);
        change_to_root_user();
 
        /* update printer queue caches if necessary */
-       update_monitored_printq_cache();
+       update_monitored_printq_cache(msg_ctx);
 
        /* check if we need to reload services */
-       check_reload(time(NULL));
+       check_reload(msg_ctx, time(NULL));
 
        /* Change machine password if neccessary. */
        attempt_machine_password_change();
@@ -2572,7 +2592,8 @@ static bool smbd_echo_reply(int fd,
                return false;
        }
 
-       if (!init_smb_request(&req, inbuf, 0, false, seqnum)) {
+       if (!init_smb_request(&req, smbd_server_conn, inbuf, 0, false,
+                             seqnum)) {
                return false;
        }
        req.inbuf = inbuf;
@@ -2789,8 +2810,9 @@ static bool fork_echo_handler(struct smbd_server_connection *sconn)
 
                close(listener_pipe[0]);
 
-               status = reinit_after_fork(smbd_messaging_context(),
-                                          smbd_event_context(), false);
+               status = reinit_after_fork(sconn->msg_ctx,
+                                          smbd_event_context(),
+                                          procid_self(), false);
                if (!NT_STATUS_IS_OK(status)) {
                        DEBUG(1, ("reinit_after_fork failed: %s\n",
                                  nt_errstr(status)));
@@ -2838,11 +2860,18 @@ fail:
        return false;
 }
 
+static bool spoolss_init_cb(void *ptr)
+{
+       struct messaging_context *msg_ctx = talloc_get_type_abort(
+               ptr, struct messaging_context);
+       return nt_printing_tdb_migrate(msg_ctx);
+}
+
 /****************************************************************************
  Process commands from the client
 ****************************************************************************/
 
-void smbd_process(void)
+void smbd_process(struct smbd_server_connection *sconn)
 {
        TALLOC_CTX *frame = talloc_stackframe();
        struct sockaddr_storage ss;
@@ -2852,11 +2881,19 @@ void smbd_process(void)
        struct tsocket_address *remote_address = NULL;
        const char *remaddr = NULL;
        int ret;
+       struct rpc_srv_callbacks spoolss_cb;
 
        if (lp_maxprotocol() == PROTOCOL_SMB2 &&
            lp_security() != SEC_SHARE &&
            !lp_async_smb_echo_handler()) {
-               smbd_server_conn->allow_smb2 = true;
+               /*
+                * We're not making the desion here,
+                * we're just allowing the client
+                * to decide between SMB1 and SMB2
+                * with the first negprot
+                * packet.
+                */
+               sconn->using_smb2 = true;
        }
 
        /* Ensure child is set to blocking mode */
@@ -2871,15 +2908,15 @@ void smbd_process(void)
        if (ret != 0) {
                int level = (errno == ENOTCONN)?2:0;
                DEBUG(level,("getpeername() failed - %s\n", strerror(errno)));
-               exit_server("getpeername() failed.\n");
+               exit_server_cleanly("getpeername() failed.\n");
        }
-       ret = tsocket_address_bsd_from_sockaddr(smbd_server_conn,
+       ret = tsocket_address_bsd_from_sockaddr(sconn,
                                                sa, sa_len,
                                                &remote_address);
        if (ret != 0) {
                DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
                        __location__, strerror(errno)));
-               exit_server("tsocket_address_bsd_from_sockaddr remote failed.\n");
+               exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
        }
 
        sa = (struct sockaddr *)(void *)&ss;
@@ -2888,23 +2925,23 @@ void smbd_process(void)
        if (ret != 0) {
                int level = (errno == ENOTCONN)?2:0;
                DEBUG(level,("getsockname() failed - %s\n", strerror(errno)));
-               exit_server("getsockname() failed.\n");
+               exit_server_cleanly("getsockname() failed.\n");
        }
-       ret = tsocket_address_bsd_from_sockaddr(smbd_server_conn,
+       ret = tsocket_address_bsd_from_sockaddr(sconn,
                                                sa, sa_len,
                                                &local_address);
        if (ret != 0) {
                DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
                        __location__, strerror(errno)));
-               exit_server("tsocket_address_bsd_from_sockaddr remote failed.\n");
+               exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
        }
 
-       smbd_server_conn->local_address = local_address;
-       smbd_server_conn->remote_address = remote_address;
+       sconn->local_address = local_address;
+       sconn->remote_address = remote_address;
 
        if (tsocket_address_is_inet(remote_address, "ip")) {
                remaddr = tsocket_address_inet_addr_string(
-                               smbd_server_conn->remote_address,
+                               sconn->remote_address,
                                talloc_tos());
                if (remaddr == NULL) {
 
@@ -2916,7 +2953,7 @@ void smbd_process(void)
        /* this is needed so that we get decent entries
           in smbstatus for port 445 connects */
        set_remote_machine_name(remaddr, false);
-       reload_services(true);
+       reload_services(sconn->msg_ctx, true);
 
        /*
         * Before the first packet, check the global hosts allow/ hosts deny
@@ -2944,8 +2981,6 @@ void smbd_process(void)
                   tsocket_address_string(remote_address, talloc_tos()),
                   tsocket_address_string(local_address, talloc_tos())));
 
-       static_init_rpc;
-
        init_modules();
 
        smb_perfcount_init();
@@ -2966,33 +3001,33 @@ void smbd_process(void)
                DEBUG(0,("Changed root to %s\n", lp_rootdir()));
        }
 
-       if (!srv_init_signing(smbd_server_conn)) {
+       if (!srv_init_signing(sconn)) {
                exit_server("Failed to init smb_signing");
        }
 
-       if (lp_async_smb_echo_handler() && !fork_echo_handler(smbd_server_conn)) {
+       if (lp_async_smb_echo_handler() && !fork_echo_handler(sconn)) {
                exit_server("Failed to fork echo handler");
        }
 
        /* Setup oplocks */
-       if (!init_oplocks(smbd_messaging_context()))
+       if (!init_oplocks(sconn->msg_ctx))
                exit_server("Failed to init oplocks");
 
        /* register our message handlers */
-       messaging_register(smbd_messaging_context(), NULL,
+       messaging_register(sconn->msg_ctx, NULL,
                           MSG_SMB_FORCE_TDIS, msg_force_tdis);
-       messaging_register(smbd_messaging_context(), NULL,
+       messaging_register(sconn->msg_ctx, NULL,
                           MSG_SMB_RELEASE_IP, msg_release_ip);
-       messaging_register(smbd_messaging_context(), NULL,
+       messaging_register(sconn->msg_ctx, NULL,
                           MSG_SMB_CLOSE_FILE, msg_close_file);
 
        /*
         * Use the default MSG_DEBUG handler to avoid rebroadcasting
         * MSGs to all child processes
         */
-       messaging_deregister(smbd_messaging_context(),
+       messaging_deregister(sconn->msg_ctx,
                             MSG_DEBUG, NULL);
-       messaging_register(smbd_messaging_context(), NULL,
+       messaging_register(sconn->msg_ctx, NULL,
                           MSG_DEBUG, debug_message);
 
        if ((lp_keepalive() != 0)
@@ -3006,14 +3041,15 @@ void smbd_process(void)
 
        if (!(event_add_idle(smbd_event_context(), NULL,
                             timeval_set(IDLE_CLOSED_TIMEOUT, 0),
-                            "deadtime", deadtime_fn, NULL))) {
+                            "deadtime", deadtime_fn, sconn))) {
                DEBUG(0, ("Could not add deadtime event\n"));
                exit(1);
        }
 
        if (!(event_add_idle(smbd_event_context(), NULL,
                             timeval_set(SMBD_SELECT_TIMEOUT, 0),
-                            "housekeeping", housekeeping_fn, NULL))) {
+                            "housekeeping", housekeeping_fn,
+                            sconn->msg_ctx))) {
                DEBUG(0, ("Could not add housekeeping event\n"));
                exit(1);
        }
@@ -3035,7 +3071,7 @@ void smbd_process(void)
                        NTSTATUS status;
 
                        status = ctdbd_register_ips(
-                               messaging_ctdbd_connection(),
+                               messaging_ctdbd_connection(procid_self()),
                                &srv, &clnt, release_ip, NULL);
 
                        if (!NT_STATUS_IS_OK(status)) {
@@ -3052,37 +3088,56 @@ void smbd_process(void)
 
 #endif
 
-       smbd_server_conn->nbt.got_session = false;
+       sconn->nbt.got_session = false;
 
-       smbd_server_conn->smb1.negprot.max_recv = MIN(lp_maxxmit(),BUFFER_SIZE);
+       sconn->smb1.negprot.max_recv = MIN(lp_maxxmit(),BUFFER_SIZE);
 
-       smbd_server_conn->smb1.sessions.done_sesssetup = false;
-       smbd_server_conn->smb1.sessions.max_send = BUFFER_SIZE;
-       smbd_server_conn->smb1.sessions.last_session_tag = UID_FIELD_INVALID;
+       sconn->smb1.sessions.done_sesssetup = false;
+       sconn->smb1.sessions.max_send = BUFFER_SIZE;
+       sconn->smb1.sessions.last_session_tag = UID_FIELD_INVALID;
        /* users from session setup */
-       smbd_server_conn->smb1.sessions.session_userlist = NULL;
+       sconn->smb1.sessions.session_userlist = NULL;
        /* workgroup from session setup. */
-       smbd_server_conn->smb1.sessions.session_workgroup = NULL;
+       sconn->smb1.sessions.session_workgroup = NULL;
        /* this holds info on user ids that are already validated for this VC */
-       smbd_server_conn->smb1.sessions.validated_users = NULL;
-       smbd_server_conn->smb1.sessions.next_vuid = VUID_OFFSET;
-       smbd_server_conn->smb1.sessions.num_validated_vuids = 0;
+       sconn->smb1.sessions.validated_users = NULL;
+       sconn->smb1.sessions.next_vuid = VUID_OFFSET;
+       sconn->smb1.sessions.num_validated_vuids = 0;
 
-       conn_init(smbd_server_conn);
-       if (!init_dptrs(smbd_server_conn)) {
+       conn_init(sconn);
+       if (!init_dptrs(sconn)) {
                exit_server("init_dptrs() failed");
        }
 
-       smbd_server_conn->smb1.fde = event_add_fd(smbd_event_context(),
-                                                 smbd_server_conn,
+       sconn->smb1.fde = event_add_fd(smbd_event_context(),
+                                                 sconn,
                                                  smbd_server_fd(),
                                                  EVENT_FD_READ,
                                                  smbd_server_connection_handler,
-                                                 smbd_server_conn);
-       if (!smbd_server_conn->smb1.fde) {
+                                                 sconn);
+       if (!sconn->smb1.fde) {
                exit_server("failed to create smbd_server_connection fde");
        }
 
+       /*
+        * Initialize spoolss with an init function to convert printers first.
+        * static_init_rpc will try to initialize the spoolss server too but you
+        * can't register it twice.
+        */
+       spoolss_cb.init = spoolss_init_cb;
+       spoolss_cb.shutdown = NULL;
+       spoolss_cb.private_data = sconn->msg_ctx;
+
+       if (!NT_STATUS_IS_OK(rpc_winreg_init(NULL))) {
+               exit(1);
+       }
+
+       if (!NT_STATUS_IS_OK(rpc_spoolss_init(&spoolss_cb))) {
+               exit(1);
+       }
+
+       static_init_rpc;
+
        TALLOC_FREE(frame);
 
        while (True) {
@@ -3092,7 +3147,7 @@ void smbd_process(void)
 
                errno = 0;
 
-               status = smbd_server_connection_loop_once(smbd_server_conn);
+               status = smbd_server_connection_loop_once(sconn);
                if (!NT_STATUS_EQUAL(status, NT_STATUS_RETRY) &&
                    !NT_STATUS_IS_OK(status)) {
                        DEBUG(3, ("smbd_server_connection_loop_once failed: %s,"