s3: Simplify smb_splice_chain
[Samba/gebeck_regimport.git] / source3 / smbd / process.c
blob391ddc70bafbe808130b686b2a98e91374c1fb6e
1 /*
2 Unix SMB/CIFS implementation.
3 process incoming packets - main loop
4 Copyright (C) Andrew Tridgell 1992-1998
5 Copyright (C) Volker Lendecke 2005-2007
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "includes.h"
22 #include "../lib/tsocket/tsocket.h"
23 #include "system/filesys.h"
24 #include "smbd/smbd.h"
25 #include "smbd/globals.h"
26 #include "librpc/gen_ndr/netlogon.h"
27 #include "../lib/async_req/async_sock.h"
28 #include "ctdbd_conn.h"
29 #include "../lib/util/select.h"
30 #include "printing/queue_process.h"
31 #include "system/select.h"
32 #include "passdb.h"
33 #include "auth.h"
34 #include "messages.h"
35 #include "smbprofile.h"
36 #include "rpc_server/spoolss/srv_spoolss_nt.h"
37 #include "libsmb/libsmb.h"
38 #include "../lib/util/tevent_ntstatus.h"
39 #include "../libcli/security/dom_sid.h"
40 #include "../libcli/security/security_token.h"
41 #include "lib/id_cache.h"
43 extern bool global_machine_password_needs_changing;
45 /* Internal message queue for deferred opens. */
46 struct pending_message_list {
47 struct pending_message_list *next, *prev;
48 struct timeval request_time; /* When was this first issued? */
49 struct smbd_server_connection *sconn;
50 struct timed_event *te;
51 struct smb_perfcount_data pcd;
52 uint32_t seqnum;
53 bool encrypted;
54 bool processed;
55 DATA_BLOB buf;
56 DATA_BLOB private_data;
59 static void construct_reply_common(struct smb_request *req, const char *inbuf,
60 char *outbuf);
61 static struct pending_message_list *get_deferred_open_message_smb(
62 struct smbd_server_connection *sconn, uint64_t mid);
64 static bool smbd_lock_socket_internal(struct smbd_server_connection *sconn)
66 bool ok;
68 if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
69 return true;
72 sconn->smb1.echo_handler.ref_count++;
74 if (sconn->smb1.echo_handler.ref_count > 1) {
75 return true;
78 DEBUG(10,("pid[%d] wait for socket lock\n", (int)sys_getpid()));
80 do {
81 ok = fcntl_lock(
82 sconn->smb1.echo_handler.socket_lock_fd,
83 SMB_F_SETLKW, 0, 0, F_WRLCK);
84 } while (!ok && (errno == EINTR));
86 if (!ok) {
87 DEBUG(1, ("fcntl_lock failed: %s\n", strerror(errno)));
88 return false;
91 DEBUG(10,("pid[%d] got for socket lock\n", (int)sys_getpid()));
93 return true;
96 void smbd_lock_socket(struct smbd_server_connection *sconn)
98 if (!smbd_lock_socket_internal(sconn)) {
99 exit_server_cleanly("failed to lock socket");
103 static bool smbd_unlock_socket_internal(struct smbd_server_connection *sconn)
105 bool ok;
107 if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
108 return true;
111 sconn->smb1.echo_handler.ref_count--;
113 if (sconn->smb1.echo_handler.ref_count > 0) {
114 return true;
117 do {
118 ok = fcntl_lock(
119 sconn->smb1.echo_handler.socket_lock_fd,
120 SMB_F_SETLKW, 0, 0, F_UNLCK);
121 } while (!ok && (errno == EINTR));
123 if (!ok) {
124 DEBUG(1, ("fcntl_lock failed: %s\n", strerror(errno)));
125 return false;
128 DEBUG(10,("pid[%d] unlocked socket\n", (int)sys_getpid()));
130 return true;
133 void smbd_unlock_socket(struct smbd_server_connection *sconn)
135 if (!smbd_unlock_socket_internal(sconn)) {
136 exit_server_cleanly("failed to unlock socket");
140 /* Accessor function for smb_read_error for smbd functions. */
142 /****************************************************************************
143 Send an smb to a fd.
144 ****************************************************************************/
146 bool srv_send_smb(struct smbd_server_connection *sconn, char *buffer,
147 bool do_signing, uint32_t seqnum,
148 bool do_encrypt,
149 struct smb_perfcount_data *pcd)
151 size_t len = 0;
152 size_t nwritten=0;
153 ssize_t ret;
154 char *buf_out = buffer;
156 smbd_lock_socket(sconn);
158 if (do_signing) {
159 /* Sign the outgoing packet if required. */
160 srv_calculate_sign_mac(sconn, buf_out, seqnum);
163 if (do_encrypt) {
164 NTSTATUS status = srv_encrypt_buffer(sconn, buffer, &buf_out);
165 if (!NT_STATUS_IS_OK(status)) {
166 DEBUG(0, ("send_smb: SMB encryption failed "
167 "on outgoing packet! Error %s\n",
168 nt_errstr(status) ));
169 goto out;
173 len = smb_len(buf_out) + 4;
175 ret = write_data(sconn->sock, buf_out+nwritten, len - nwritten);
176 if (ret <= 0) {
178 char addr[INET6_ADDRSTRLEN];
180 * Try and give an error message saying what
181 * client failed.
183 DEBUG(1,("pid[%d] Error writing %d bytes to client %s. %d. (%s)\n",
184 (int)sys_getpid(), (int)len,
185 get_peer_addr(sconn->sock, addr, sizeof(addr)),
186 (int)ret, strerror(errno) ));
188 srv_free_enc_buffer(sconn, buf_out);
189 goto out;
192 SMB_PERFCOUNT_SET_MSGLEN_OUT(pcd, len);
193 srv_free_enc_buffer(sconn, buf_out);
194 out:
195 SMB_PERFCOUNT_END(pcd);
197 smbd_unlock_socket(sconn);
198 return true;
201 /*******************************************************************
202 Setup the word count and byte count for a smb message.
203 ********************************************************************/
205 int srv_set_message(char *buf,
206 int num_words,
207 int num_bytes,
208 bool zero)
210 if (zero && (num_words || num_bytes)) {
211 memset(buf + smb_size,'\0',num_words*2 + num_bytes);
213 SCVAL(buf,smb_wct,num_words);
214 SSVAL(buf,smb_vwv + num_words*SIZEOFWORD,num_bytes);
215 smb_setlen(buf,(smb_size + num_words*2 + num_bytes - 4));
216 return (smb_size + num_words*2 + num_bytes);
219 static bool valid_smb_header(struct smbd_server_connection *sconn,
220 const uint8_t *inbuf)
222 if (is_encrypted_packet(sconn, inbuf)) {
223 return true;
226 * This used to be (strncmp(smb_base(inbuf),"\377SMB",4) == 0)
227 * but it just looks weird to call strncmp for this one.
229 return (IVAL(smb_base(inbuf), 0) == 0x424D53FF);
232 /* Socket functions for smbd packet processing. */
234 static bool valid_packet_size(size_t len)
237 * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
238 * of header. Don't print the error if this fits.... JRA.
241 if (len > (BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
242 DEBUG(0,("Invalid packet length! (%lu bytes).\n",
243 (unsigned long)len));
244 return false;
246 return true;
249 static NTSTATUS read_packet_remainder(int fd, char *buffer,
250 unsigned int timeout, ssize_t len)
252 NTSTATUS status;
254 if (len <= 0) {
255 return NT_STATUS_OK;
258 status = read_fd_with_timeout(fd, buffer, len, len, timeout, NULL);
259 if (!NT_STATUS_IS_OK(status)) {
260 char addr[INET6_ADDRSTRLEN];
261 DEBUG(0, ("read_fd_with_timeout failed for client %s read "
262 "error = %s.\n",
263 get_peer_addr(fd, addr, sizeof(addr)),
264 nt_errstr(status)));
266 return status;
269 /****************************************************************************
270 Attempt a zerocopy writeX read. We know here that len > smb_size-4
271 ****************************************************************************/
274 * Unfortunately, earlier versions of smbclient/libsmbclient
275 * don't send this "standard" writeX header. I've fixed this
276 * for 3.2 but we'll use the old method with earlier versions.
277 * Windows and CIFSFS at least use this standard size. Not
278 * sure about MacOSX.
281 #define STANDARD_WRITE_AND_X_HEADER_SIZE (smb_size - 4 + /* basic header */ \
282 (2*14) + /* word count (including bcc) */ \
283 1 /* pad byte */)
285 static NTSTATUS receive_smb_raw_talloc_partial_read(TALLOC_CTX *mem_ctx,
286 const char lenbuf[4],
287 struct smbd_server_connection *sconn,
288 int sock,
289 char **buffer,
290 unsigned int timeout,
291 size_t *p_unread,
292 size_t *len_ret)
294 /* Size of a WRITEX call (+4 byte len). */
295 char writeX_header[4 + STANDARD_WRITE_AND_X_HEADER_SIZE];
296 ssize_t len = smb_len_large(lenbuf); /* Could be a UNIX large writeX. */
297 ssize_t toread;
298 NTSTATUS status;
300 memcpy(writeX_header, lenbuf, 4);
302 status = read_fd_with_timeout(
303 sock, writeX_header + 4,
304 STANDARD_WRITE_AND_X_HEADER_SIZE,
305 STANDARD_WRITE_AND_X_HEADER_SIZE,
306 timeout, NULL);
308 if (!NT_STATUS_IS_OK(status)) {
309 DEBUG(0, ("read_fd_with_timeout failed for client %s read "
310 "error = %s.\n",
311 tsocket_address_string(sconn->remote_address,
312 talloc_tos()),
313 nt_errstr(status)));
314 return status;
318 * Ok - now try and see if this is a possible
319 * valid writeX call.
322 if (is_valid_writeX_buffer(sconn, (uint8_t *)writeX_header)) {
324 * If the data offset is beyond what
325 * we've read, drain the extra bytes.
327 uint16_t doff = SVAL(writeX_header,smb_vwv11);
328 ssize_t newlen;
330 if (doff > STANDARD_WRITE_AND_X_HEADER_SIZE) {
331 size_t drain = doff - STANDARD_WRITE_AND_X_HEADER_SIZE;
332 if (drain_socket(sock, drain) != drain) {
333 smb_panic("receive_smb_raw_talloc_partial_read:"
334 " failed to drain pending bytes");
336 } else {
337 doff = STANDARD_WRITE_AND_X_HEADER_SIZE;
340 /* Spoof down the length and null out the bcc. */
341 set_message_bcc(writeX_header, 0);
342 newlen = smb_len(writeX_header);
344 /* Copy the header we've written. */
346 *buffer = (char *)talloc_memdup(mem_ctx,
347 writeX_header,
348 sizeof(writeX_header));
350 if (*buffer == NULL) {
351 DEBUG(0, ("Could not allocate inbuf of length %d\n",
352 (int)sizeof(writeX_header)));
353 return NT_STATUS_NO_MEMORY;
356 /* Work out the remaining bytes. */
357 *p_unread = len - STANDARD_WRITE_AND_X_HEADER_SIZE;
358 *len_ret = newlen + 4;
359 return NT_STATUS_OK;
362 if (!valid_packet_size(len)) {
363 return NT_STATUS_INVALID_PARAMETER;
367 * Not a valid writeX call. Just do the standard
368 * talloc and return.
371 *buffer = talloc_array(mem_ctx, char, len+4);
373 if (*buffer == NULL) {
374 DEBUG(0, ("Could not allocate inbuf of length %d\n",
375 (int)len+4));
376 return NT_STATUS_NO_MEMORY;
379 /* Copy in what we already read. */
380 memcpy(*buffer,
381 writeX_header,
382 4 + STANDARD_WRITE_AND_X_HEADER_SIZE);
383 toread = len - STANDARD_WRITE_AND_X_HEADER_SIZE;
385 if(toread > 0) {
386 status = read_packet_remainder(
387 sock,
388 (*buffer) + 4 + STANDARD_WRITE_AND_X_HEADER_SIZE,
389 timeout, toread);
391 if (!NT_STATUS_IS_OK(status)) {
392 DEBUG(10, ("receive_smb_raw_talloc_partial_read: %s\n",
393 nt_errstr(status)));
394 return status;
398 *len_ret = len + 4;
399 return NT_STATUS_OK;
402 static NTSTATUS receive_smb_raw_talloc(TALLOC_CTX *mem_ctx,
403 struct smbd_server_connection *sconn,
404 int sock,
405 char **buffer, unsigned int timeout,
406 size_t *p_unread, size_t *plen)
408 char lenbuf[4];
409 size_t len;
410 int min_recv_size = lp_min_receive_file_size();
411 NTSTATUS status;
413 *p_unread = 0;
415 status = read_smb_length_return_keepalive(sock, lenbuf, timeout,
416 &len);
417 if (!NT_STATUS_IS_OK(status)) {
418 return status;
421 if (CVAL(lenbuf,0) == 0 && min_recv_size &&
422 (smb_len_large(lenbuf) > /* Could be a UNIX large writeX. */
423 (min_recv_size + STANDARD_WRITE_AND_X_HEADER_SIZE)) &&
424 !srv_is_signing_active(sconn) &&
425 sconn->smb1.echo_handler.trusted_fde == NULL) {
427 return receive_smb_raw_talloc_partial_read(
428 mem_ctx, lenbuf, sconn, sock, buffer, timeout,
429 p_unread, plen);
432 if (!valid_packet_size(len)) {
433 return NT_STATUS_INVALID_PARAMETER;
437 * The +4 here can't wrap, we've checked the length above already.
440 *buffer = talloc_array(mem_ctx, char, len+4);
442 if (*buffer == NULL) {
443 DEBUG(0, ("Could not allocate inbuf of length %d\n",
444 (int)len+4));
445 return NT_STATUS_NO_MEMORY;
448 memcpy(*buffer, lenbuf, sizeof(lenbuf));
450 status = read_packet_remainder(sock, (*buffer)+4, timeout, len);
451 if (!NT_STATUS_IS_OK(status)) {
452 return status;
455 *plen = len + 4;
456 return NT_STATUS_OK;
459 static NTSTATUS receive_smb_talloc(TALLOC_CTX *mem_ctx,
460 struct smbd_server_connection *sconn,
461 int sock,
462 char **buffer, unsigned int timeout,
463 size_t *p_unread, bool *p_encrypted,
464 size_t *p_len,
465 uint32_t *seqnum,
466 bool trusted_channel)
468 size_t len = 0;
469 NTSTATUS status;
471 *p_encrypted = false;
473 status = receive_smb_raw_talloc(mem_ctx, sconn, sock, buffer, timeout,
474 p_unread, &len);
475 if (!NT_STATUS_IS_OK(status)) {
476 DEBUG(NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)?5:1,
477 ("receive_smb_raw_talloc failed for client %s "
478 "read error = %s.\n",
479 tsocket_address_string(sconn->remote_address,
480 talloc_tos()),
481 nt_errstr(status)) );
482 return status;
485 if (is_encrypted_packet(sconn, (uint8_t *)*buffer)) {
486 status = srv_decrypt_buffer(sconn, *buffer);
487 if (!NT_STATUS_IS_OK(status)) {
488 DEBUG(0, ("receive_smb_talloc: SMB decryption failed on "
489 "incoming packet! Error %s\n",
490 nt_errstr(status) ));
491 return status;
493 *p_encrypted = true;
496 /* Check the incoming SMB signature. */
497 if (!srv_check_sign_mac(sconn, *buffer, seqnum, trusted_channel)) {
498 DEBUG(0, ("receive_smb: SMB Signature verification failed on "
499 "incoming packet!\n"));
500 return NT_STATUS_INVALID_NETWORK_RESPONSE;
503 *p_len = len;
504 return NT_STATUS_OK;
508 * Initialize a struct smb_request from an inbuf
511 static bool init_smb_request(struct smb_request *req,
512 struct smbd_server_connection *sconn,
513 const uint8 *inbuf,
514 size_t unread_bytes, bool encrypted,
515 uint32_t seqnum)
517 size_t req_size = smb_len(inbuf) + 4;
518 /* Ensure we have at least smb_size bytes. */
519 if (req_size < smb_size) {
520 DEBUG(0,("init_smb_request: invalid request size %u\n",
521 (unsigned int)req_size ));
522 return false;
524 req->cmd = CVAL(inbuf, smb_com);
525 req->flags2 = SVAL(inbuf, smb_flg2);
526 req->smbpid = SVAL(inbuf, smb_pid);
527 req->mid = (uint64_t)SVAL(inbuf, smb_mid);
528 req->seqnum = seqnum;
529 req->vuid = SVAL(inbuf, smb_uid);
530 req->tid = SVAL(inbuf, smb_tid);
531 req->wct = CVAL(inbuf, smb_wct);
532 req->vwv = (const uint16_t *)(inbuf+smb_vwv);
533 req->buflen = smb_buflen(inbuf);
534 req->buf = (const uint8_t *)smb_buf_const(inbuf);
535 req->unread_bytes = unread_bytes;
536 req->encrypted = encrypted;
537 req->sconn = sconn;
538 req->conn = conn_find(sconn,req->tid);
539 req->chain_fsp = NULL;
540 req->chain_outbuf = NULL;
541 req->done = false;
542 req->smb2req = NULL;
543 smb_init_perfcount_data(&req->pcd);
545 /* Ensure we have at least wct words and 2 bytes of bcc. */
546 if (smb_size + req->wct*2 > req_size) {
547 DEBUG(0,("init_smb_request: invalid wct number %u (size %u)\n",
548 (unsigned int)req->wct,
549 (unsigned int)req_size));
550 return false;
552 /* Ensure bcc is correct. */
553 if (((const uint8_t *)smb_buf_const(inbuf)) + req->buflen > inbuf + req_size) {
554 DEBUG(0,("init_smb_request: invalid bcc number %u "
555 "(wct = %u, size %u)\n",
556 (unsigned int)req->buflen,
557 (unsigned int)req->wct,
558 (unsigned int)req_size));
559 return false;
562 req->outbuf = NULL;
563 return true;
566 static void process_smb(struct smbd_server_connection *conn,
567 uint8_t *inbuf, size_t nread, size_t unread_bytes,
568 uint32_t seqnum, bool encrypted,
569 struct smb_perfcount_data *deferred_pcd);
571 static void smbd_deferred_open_timer(struct event_context *ev,
572 struct timed_event *te,
573 struct timeval _tval,
574 void *private_data)
576 struct pending_message_list *msg = talloc_get_type(private_data,
577 struct pending_message_list);
578 struct smbd_server_connection *sconn = msg->sconn;
579 TALLOC_CTX *mem_ctx = talloc_tos();
580 uint64_t mid = (uint64_t)SVAL(msg->buf.data,smb_mid);
581 uint8_t *inbuf;
583 inbuf = (uint8_t *)talloc_memdup(mem_ctx, msg->buf.data,
584 msg->buf.length);
585 if (inbuf == NULL) {
586 exit_server("smbd_deferred_open_timer: talloc failed\n");
587 return;
590 /* We leave this message on the queue so the open code can
591 know this is a retry. */
592 DEBUG(5,("smbd_deferred_open_timer: trigger mid %llu.\n",
593 (unsigned long long)mid ));
595 /* Mark the message as processed so this is not
596 * re-processed in error. */
597 msg->processed = true;
599 process_smb(sconn, inbuf,
600 msg->buf.length, 0,
601 msg->seqnum, msg->encrypted, &msg->pcd);
603 /* If it's still there and was processed, remove it. */
604 msg = get_deferred_open_message_smb(sconn, mid);
605 if (msg && msg->processed) {
606 remove_deferred_open_message_smb(sconn, mid);
610 /****************************************************************************
611 Function to push a message onto the tail of a linked list of smb messages ready
612 for processing.
613 ****************************************************************************/
615 static bool push_queued_message(struct smb_request *req,
616 struct timeval request_time,
617 struct timeval end_time,
618 char *private_data, size_t private_len)
620 int msg_len = smb_len(req->inbuf) + 4;
621 struct pending_message_list *msg;
623 msg = talloc_zero(NULL, struct pending_message_list);
625 if(msg == NULL) {
626 DEBUG(0,("push_message: malloc fail (1)\n"));
627 return False;
629 msg->sconn = req->sconn;
631 msg->buf = data_blob_talloc(msg, req->inbuf, msg_len);
632 if(msg->buf.data == NULL) {
633 DEBUG(0,("push_message: malloc fail (2)\n"));
634 TALLOC_FREE(msg);
635 return False;
638 msg->request_time = request_time;
639 msg->seqnum = req->seqnum;
640 msg->encrypted = req->encrypted;
641 msg->processed = false;
642 SMB_PERFCOUNT_DEFER_OP(&req->pcd, &msg->pcd);
644 if (private_data) {
645 msg->private_data = data_blob_talloc(msg, private_data,
646 private_len);
647 if (msg->private_data.data == NULL) {
648 DEBUG(0,("push_message: malloc fail (3)\n"));
649 TALLOC_FREE(msg);
650 return False;
654 msg->te = tevent_add_timer(msg->sconn->ev_ctx,
655 msg,
656 end_time,
657 smbd_deferred_open_timer,
658 msg);
659 if (!msg->te) {
660 DEBUG(0,("push_message: event_add_timed failed\n"));
661 TALLOC_FREE(msg);
662 return false;
665 DLIST_ADD_END(req->sconn->deferred_open_queue, msg,
666 struct pending_message_list *);
668 DEBUG(10,("push_message: pushed message length %u on "
669 "deferred_open_queue\n", (unsigned int)msg_len));
671 return True;
674 /****************************************************************************
675 Function to delete a sharing violation open message by mid.
676 ****************************************************************************/
678 void remove_deferred_open_message_smb(struct smbd_server_connection *sconn,
679 uint64_t mid)
681 struct pending_message_list *pml;
683 if (sconn->using_smb2) {
684 remove_deferred_open_message_smb2(sconn, mid);
685 return;
688 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
689 if (mid == (uint64_t)SVAL(pml->buf.data,smb_mid)) {
690 DEBUG(10,("remove_deferred_open_message_smb: "
691 "deleting mid %llu len %u\n",
692 (unsigned long long)mid,
693 (unsigned int)pml->buf.length ));
694 DLIST_REMOVE(sconn->deferred_open_queue, pml);
695 TALLOC_FREE(pml);
696 return;
701 /****************************************************************************
702 Move a sharing violation open retry message to the front of the list and
703 schedule it for immediate processing.
704 ****************************************************************************/
706 void schedule_deferred_open_message_smb(struct smbd_server_connection *sconn,
707 uint64_t mid)
709 struct pending_message_list *pml;
710 int i = 0;
712 if (sconn->using_smb2) {
713 schedule_deferred_open_message_smb2(sconn, mid);
714 return;
717 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
718 uint64_t msg_mid = (uint64_t)SVAL(pml->buf.data,smb_mid);
720 DEBUG(10,("schedule_deferred_open_message_smb: [%d] "
721 "msg_mid = %llu\n",
722 i++,
723 (unsigned long long)msg_mid ));
725 if (mid == msg_mid) {
726 struct timed_event *te;
728 if (pml->processed) {
729 /* A processed message should not be
730 * rescheduled. */
731 DEBUG(0,("schedule_deferred_open_message_smb: LOGIC ERROR "
732 "message mid %llu was already processed\n",
733 (unsigned long long)msg_mid ));
734 continue;
737 DEBUG(10,("schedule_deferred_open_message_smb: "
738 "scheduling mid %llu\n",
739 (unsigned long long)mid ));
741 te = tevent_add_timer(pml->sconn->ev_ctx,
742 pml,
743 timeval_zero(),
744 smbd_deferred_open_timer,
745 pml);
746 if (!te) {
747 DEBUG(10,("schedule_deferred_open_message_smb: "
748 "event_add_timed() failed, "
749 "skipping mid %llu\n",
750 (unsigned long long)msg_mid ));
753 TALLOC_FREE(pml->te);
754 pml->te = te;
755 DLIST_PROMOTE(sconn->deferred_open_queue, pml);
756 return;
760 DEBUG(10,("schedule_deferred_open_message_smb: failed to "
761 "find message mid %llu\n",
762 (unsigned long long)mid ));
765 /****************************************************************************
766 Return true if this mid is on the deferred queue and was not yet processed.
767 ****************************************************************************/
769 bool open_was_deferred(struct smbd_server_connection *sconn, uint64_t mid)
771 struct pending_message_list *pml;
773 if (sconn->using_smb2) {
774 return open_was_deferred_smb2(sconn, mid);
777 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
778 if (((uint64_t)SVAL(pml->buf.data,smb_mid)) == mid && !pml->processed) {
779 return True;
782 return False;
785 /****************************************************************************
786 Return the message queued by this mid.
787 ****************************************************************************/
789 static struct pending_message_list *get_deferred_open_message_smb(
790 struct smbd_server_connection *sconn, uint64_t mid)
792 struct pending_message_list *pml;
794 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
795 if (((uint64_t)SVAL(pml->buf.data,smb_mid)) == mid) {
796 return pml;
799 return NULL;
802 /****************************************************************************
803 Get the state data queued by this mid.
804 ****************************************************************************/
806 bool get_deferred_open_message_state(struct smb_request *smbreq,
807 struct timeval *p_request_time,
808 void **pp_state)
810 struct pending_message_list *pml;
812 if (smbreq->sconn->using_smb2) {
813 return get_deferred_open_message_state_smb2(smbreq->smb2req,
814 p_request_time,
815 pp_state);
818 pml = get_deferred_open_message_smb(smbreq->sconn, smbreq->mid);
819 if (!pml) {
820 return false;
822 if (p_request_time) {
823 *p_request_time = pml->request_time;
825 if (pp_state) {
826 *pp_state = (void *)pml->private_data.data;
828 return true;
831 /****************************************************************************
832 Function to push a deferred open smb message onto a linked list of local smb
833 messages ready for processing.
834 ****************************************************************************/
836 bool push_deferred_open_message_smb(struct smb_request *req,
837 struct timeval request_time,
838 struct timeval timeout,
839 struct file_id id,
840 char *private_data, size_t priv_len)
842 struct timeval end_time;
844 if (req->smb2req) {
845 return push_deferred_open_message_smb2(req->smb2req,
846 request_time,
847 timeout,
849 private_data,
850 priv_len);
853 if (req->unread_bytes) {
854 DEBUG(0,("push_deferred_open_message_smb: logic error ! "
855 "unread_bytes = %u\n",
856 (unsigned int)req->unread_bytes ));
857 smb_panic("push_deferred_open_message_smb: "
858 "logic error unread_bytes != 0" );
861 end_time = timeval_sum(&request_time, &timeout);
863 DEBUG(10,("push_deferred_open_message_smb: pushing message "
864 "len %u mid %llu timeout time [%u.%06u]\n",
865 (unsigned int) smb_len(req->inbuf)+4,
866 (unsigned long long)req->mid,
867 (unsigned int)end_time.tv_sec,
868 (unsigned int)end_time.tv_usec));
870 return push_queued_message(req, request_time, end_time,
871 private_data, priv_len);
874 static void smbd_sig_term_handler(struct tevent_context *ev,
875 struct tevent_signal *se,
876 int signum,
877 int count,
878 void *siginfo,
879 void *private_data)
881 exit_server_cleanly("termination signal");
884 void smbd_setup_sig_term_handler(struct smbd_server_connection *sconn)
886 struct tevent_signal *se;
888 se = tevent_add_signal(sconn->ev_ctx,
889 sconn,
890 SIGTERM, 0,
891 smbd_sig_term_handler,
892 sconn);
893 if (!se) {
894 exit_server("failed to setup SIGTERM handler");
898 static void smbd_sig_hup_handler(struct tevent_context *ev,
899 struct tevent_signal *se,
900 int signum,
901 int count,
902 void *siginfo,
903 void *private_data)
905 struct smbd_server_connection *sconn =
906 talloc_get_type_abort(private_data,
907 struct smbd_server_connection);
909 change_to_root_user();
910 DEBUG(1,("Reloading services after SIGHUP\n"));
911 reload_services(sconn, conn_snum_used, false);
914 void smbd_setup_sig_hup_handler(struct smbd_server_connection *sconn)
916 struct tevent_signal *se;
918 se = tevent_add_signal(sconn->ev_ctx,
919 sconn,
920 SIGHUP, 0,
921 smbd_sig_hup_handler,
922 sconn);
923 if (!se) {
924 exit_server("failed to setup SIGHUP handler");
928 static void smbd_conf_updated(struct messaging_context *msg,
929 void *private_data,
930 uint32_t msg_type,
931 struct server_id server_id,
932 DATA_BLOB *data)
934 struct smbd_server_connection *sconn =
935 talloc_get_type_abort(private_data,
936 struct smbd_server_connection);
938 DEBUG(10,("smbd_conf_updated: Got message saying smb.conf was "
939 "updated. Reloading.\n"));
940 change_to_root_user();
941 reload_services(sconn, conn_snum_used, false);
945 * Only allow 5 outstanding trans requests. We're allocating memory, so
946 * prevent a DoS.
949 NTSTATUS allow_new_trans(struct trans_state *list, uint64_t mid)
951 int count = 0;
952 for (; list != NULL; list = list->next) {
954 if (list->mid == mid) {
955 return NT_STATUS_INVALID_PARAMETER;
958 count += 1;
960 if (count > 5) {
961 return NT_STATUS_INSUFFICIENT_RESOURCES;
964 return NT_STATUS_OK;
968 These flags determine some of the permissions required to do an operation
970 Note that I don't set NEED_WRITE on some write operations because they
971 are used by some brain-dead clients when printing, and I don't want to
972 force write permissions on print services.
974 #define AS_USER (1<<0)
975 #define NEED_WRITE (1<<1) /* Must be paired with AS_USER */
976 #define TIME_INIT (1<<2)
977 #define CAN_IPC (1<<3) /* Must be paired with AS_USER */
978 #define AS_GUEST (1<<5) /* Must *NOT* be paired with AS_USER */
979 #define DO_CHDIR (1<<6)
982 define a list of possible SMB messages and their corresponding
983 functions. Any message that has a NULL function is unimplemented -
984 please feel free to contribute implementations!
986 static const struct smb_message_struct {
987 const char *name;
988 void (*fn)(struct smb_request *req);
989 int flags;
990 } smb_messages[256] = {
992 /* 0x00 */ { "SMBmkdir",reply_mkdir,AS_USER | NEED_WRITE},
993 /* 0x01 */ { "SMBrmdir",reply_rmdir,AS_USER | NEED_WRITE},
994 /* 0x02 */ { "SMBopen",reply_open,AS_USER },
995 /* 0x03 */ { "SMBcreate",reply_mknew,AS_USER},
996 /* 0x04 */ { "SMBclose",reply_close,AS_USER | CAN_IPC },
997 /* 0x05 */ { "SMBflush",reply_flush,AS_USER},
998 /* 0x06 */ { "SMBunlink",reply_unlink,AS_USER | NEED_WRITE },
999 /* 0x07 */ { "SMBmv",reply_mv,AS_USER | NEED_WRITE },
1000 /* 0x08 */ { "SMBgetatr",reply_getatr,AS_USER},
1001 /* 0x09 */ { "SMBsetatr",reply_setatr,AS_USER | NEED_WRITE},
1002 /* 0x0a */ { "SMBread",reply_read,AS_USER},
1003 /* 0x0b */ { "SMBwrite",reply_write,AS_USER | CAN_IPC },
1004 /* 0x0c */ { "SMBlock",reply_lock,AS_USER},
1005 /* 0x0d */ { "SMBunlock",reply_unlock,AS_USER},
1006 /* 0x0e */ { "SMBctemp",reply_ctemp,AS_USER },
1007 /* 0x0f */ { "SMBmknew",reply_mknew,AS_USER},
1008 /* 0x10 */ { "SMBcheckpath",reply_checkpath,AS_USER},
1009 /* 0x11 */ { "SMBexit",reply_exit,DO_CHDIR},
1010 /* 0x12 */ { "SMBlseek",reply_lseek,AS_USER},
1011 /* 0x13 */ { "SMBlockread",reply_lockread,AS_USER},
1012 /* 0x14 */ { "SMBwriteunlock",reply_writeunlock,AS_USER},
1013 /* 0x15 */ { NULL, NULL, 0 },
1014 /* 0x16 */ { NULL, NULL, 0 },
1015 /* 0x17 */ { NULL, NULL, 0 },
1016 /* 0x18 */ { NULL, NULL, 0 },
1017 /* 0x19 */ { NULL, NULL, 0 },
1018 /* 0x1a */ { "SMBreadbraw",reply_readbraw,AS_USER},
1019 /* 0x1b */ { "SMBreadBmpx",reply_readbmpx,AS_USER},
1020 /* 0x1c */ { "SMBreadBs",reply_readbs,AS_USER },
1021 /* 0x1d */ { "SMBwritebraw",reply_writebraw,AS_USER},
1022 /* 0x1e */ { "SMBwriteBmpx",reply_writebmpx,AS_USER},
1023 /* 0x1f */ { "SMBwriteBs",reply_writebs,AS_USER},
1024 /* 0x20 */ { "SMBwritec", NULL,0},
1025 /* 0x21 */ { NULL, NULL, 0 },
1026 /* 0x22 */ { "SMBsetattrE",reply_setattrE,AS_USER | NEED_WRITE },
1027 /* 0x23 */ { "SMBgetattrE",reply_getattrE,AS_USER },
1028 /* 0x24 */ { "SMBlockingX",reply_lockingX,AS_USER },
1029 /* 0x25 */ { "SMBtrans",reply_trans,AS_USER | CAN_IPC },
1030 /* 0x26 */ { "SMBtranss",reply_transs,AS_USER | CAN_IPC},
1031 /* 0x27 */ { "SMBioctl",reply_ioctl,0},
1032 /* 0x28 */ { "SMBioctls", NULL,AS_USER},
1033 /* 0x29 */ { "SMBcopy",reply_copy,AS_USER | NEED_WRITE },
1034 /* 0x2a */ { "SMBmove", NULL,AS_USER | NEED_WRITE },
1035 /* 0x2b */ { "SMBecho",reply_echo,0},
1036 /* 0x2c */ { "SMBwriteclose",reply_writeclose,AS_USER},
1037 /* 0x2d */ { "SMBopenX",reply_open_and_X,AS_USER | CAN_IPC },
1038 /* 0x2e */ { "SMBreadX",reply_read_and_X,AS_USER | CAN_IPC },
1039 /* 0x2f */ { "SMBwriteX",reply_write_and_X,AS_USER | CAN_IPC },
1040 /* 0x30 */ { NULL, NULL, 0 },
1041 /* 0x31 */ { NULL, NULL, 0 },
1042 /* 0x32 */ { "SMBtrans2",reply_trans2, AS_USER | CAN_IPC },
1043 /* 0x33 */ { "SMBtranss2",reply_transs2, AS_USER | CAN_IPC },
1044 /* 0x34 */ { "SMBfindclose",reply_findclose,AS_USER},
1045 /* 0x35 */ { "SMBfindnclose",reply_findnclose,AS_USER},
1046 /* 0x36 */ { NULL, NULL, 0 },
1047 /* 0x37 */ { NULL, NULL, 0 },
1048 /* 0x38 */ { NULL, NULL, 0 },
1049 /* 0x39 */ { NULL, NULL, 0 },
1050 /* 0x3a */ { NULL, NULL, 0 },
1051 /* 0x3b */ { NULL, NULL, 0 },
1052 /* 0x3c */ { NULL, NULL, 0 },
1053 /* 0x3d */ { NULL, NULL, 0 },
1054 /* 0x3e */ { NULL, NULL, 0 },
1055 /* 0x3f */ { NULL, NULL, 0 },
1056 /* 0x40 */ { NULL, NULL, 0 },
1057 /* 0x41 */ { NULL, NULL, 0 },
1058 /* 0x42 */ { NULL, NULL, 0 },
1059 /* 0x43 */ { NULL, NULL, 0 },
1060 /* 0x44 */ { NULL, NULL, 0 },
1061 /* 0x45 */ { NULL, NULL, 0 },
1062 /* 0x46 */ { NULL, NULL, 0 },
1063 /* 0x47 */ { NULL, NULL, 0 },
1064 /* 0x48 */ { NULL, NULL, 0 },
1065 /* 0x49 */ { NULL, NULL, 0 },
1066 /* 0x4a */ { NULL, NULL, 0 },
1067 /* 0x4b */ { NULL, NULL, 0 },
1068 /* 0x4c */ { NULL, NULL, 0 },
1069 /* 0x4d */ { NULL, NULL, 0 },
1070 /* 0x4e */ { NULL, NULL, 0 },
1071 /* 0x4f */ { NULL, NULL, 0 },
1072 /* 0x50 */ { NULL, NULL, 0 },
1073 /* 0x51 */ { NULL, NULL, 0 },
1074 /* 0x52 */ { NULL, NULL, 0 },
1075 /* 0x53 */ { NULL, NULL, 0 },
1076 /* 0x54 */ { NULL, NULL, 0 },
1077 /* 0x55 */ { NULL, NULL, 0 },
1078 /* 0x56 */ { NULL, NULL, 0 },
1079 /* 0x57 */ { NULL, NULL, 0 },
1080 /* 0x58 */ { NULL, NULL, 0 },
1081 /* 0x59 */ { NULL, NULL, 0 },
1082 /* 0x5a */ { NULL, NULL, 0 },
1083 /* 0x5b */ { NULL, NULL, 0 },
1084 /* 0x5c */ { NULL, NULL, 0 },
1085 /* 0x5d */ { NULL, NULL, 0 },
1086 /* 0x5e */ { NULL, NULL, 0 },
1087 /* 0x5f */ { NULL, NULL, 0 },
1088 /* 0x60 */ { NULL, NULL, 0 },
1089 /* 0x61 */ { NULL, NULL, 0 },
1090 /* 0x62 */ { NULL, NULL, 0 },
1091 /* 0x63 */ { NULL, NULL, 0 },
1092 /* 0x64 */ { NULL, NULL, 0 },
1093 /* 0x65 */ { NULL, NULL, 0 },
1094 /* 0x66 */ { NULL, NULL, 0 },
1095 /* 0x67 */ { NULL, NULL, 0 },
1096 /* 0x68 */ { NULL, NULL, 0 },
1097 /* 0x69 */ { NULL, NULL, 0 },
1098 /* 0x6a */ { NULL, NULL, 0 },
1099 /* 0x6b */ { NULL, NULL, 0 },
1100 /* 0x6c */ { NULL, NULL, 0 },
1101 /* 0x6d */ { NULL, NULL, 0 },
1102 /* 0x6e */ { NULL, NULL, 0 },
1103 /* 0x6f */ { NULL, NULL, 0 },
1104 /* 0x70 */ { "SMBtcon",reply_tcon,0},
1105 /* 0x71 */ { "SMBtdis",reply_tdis,DO_CHDIR},
1106 /* 0x72 */ { "SMBnegprot",reply_negprot,0},
1107 /* 0x73 */ { "SMBsesssetupX",reply_sesssetup_and_X,0},
1108 /* 0x74 */ { "SMBulogoffX",reply_ulogoffX, 0}, /* ulogoff doesn't give a valid TID */
1109 /* 0x75 */ { "SMBtconX",reply_tcon_and_X,0},
1110 /* 0x76 */ { NULL, NULL, 0 },
1111 /* 0x77 */ { NULL, NULL, 0 },
1112 /* 0x78 */ { NULL, NULL, 0 },
1113 /* 0x79 */ { NULL, NULL, 0 },
1114 /* 0x7a */ { NULL, NULL, 0 },
1115 /* 0x7b */ { NULL, NULL, 0 },
1116 /* 0x7c */ { NULL, NULL, 0 },
1117 /* 0x7d */ { NULL, NULL, 0 },
1118 /* 0x7e */ { NULL, NULL, 0 },
1119 /* 0x7f */ { NULL, NULL, 0 },
1120 /* 0x80 */ { "SMBdskattr",reply_dskattr,AS_USER},
1121 /* 0x81 */ { "SMBsearch",reply_search,AS_USER},
1122 /* 0x82 */ { "SMBffirst",reply_search,AS_USER},
1123 /* 0x83 */ { "SMBfunique",reply_search,AS_USER},
1124 /* 0x84 */ { "SMBfclose",reply_fclose,AS_USER},
1125 /* 0x85 */ { NULL, NULL, 0 },
1126 /* 0x86 */ { NULL, NULL, 0 },
1127 /* 0x87 */ { NULL, NULL, 0 },
1128 /* 0x88 */ { NULL, NULL, 0 },
1129 /* 0x89 */ { NULL, NULL, 0 },
1130 /* 0x8a */ { NULL, NULL, 0 },
1131 /* 0x8b */ { NULL, NULL, 0 },
1132 /* 0x8c */ { NULL, NULL, 0 },
1133 /* 0x8d */ { NULL, NULL, 0 },
1134 /* 0x8e */ { NULL, NULL, 0 },
1135 /* 0x8f */ { NULL, NULL, 0 },
1136 /* 0x90 */ { NULL, NULL, 0 },
1137 /* 0x91 */ { NULL, NULL, 0 },
1138 /* 0x92 */ { NULL, NULL, 0 },
1139 /* 0x93 */ { NULL, NULL, 0 },
1140 /* 0x94 */ { NULL, NULL, 0 },
1141 /* 0x95 */ { NULL, NULL, 0 },
1142 /* 0x96 */ { NULL, NULL, 0 },
1143 /* 0x97 */ { NULL, NULL, 0 },
1144 /* 0x98 */ { NULL, NULL, 0 },
1145 /* 0x99 */ { NULL, NULL, 0 },
1146 /* 0x9a */ { NULL, NULL, 0 },
1147 /* 0x9b */ { NULL, NULL, 0 },
1148 /* 0x9c */ { NULL, NULL, 0 },
1149 /* 0x9d */ { NULL, NULL, 0 },
1150 /* 0x9e */ { NULL, NULL, 0 },
1151 /* 0x9f */ { NULL, NULL, 0 },
1152 /* 0xa0 */ { "SMBnttrans",reply_nttrans, AS_USER | CAN_IPC },
1153 /* 0xa1 */ { "SMBnttranss",reply_nttranss, AS_USER | CAN_IPC },
1154 /* 0xa2 */ { "SMBntcreateX",reply_ntcreate_and_X, AS_USER | CAN_IPC },
1155 /* 0xa3 */ { NULL, NULL, 0 },
1156 /* 0xa4 */ { "SMBntcancel",reply_ntcancel, 0 },
1157 /* 0xa5 */ { "SMBntrename",reply_ntrename, AS_USER | NEED_WRITE },
1158 /* 0xa6 */ { NULL, NULL, 0 },
1159 /* 0xa7 */ { NULL, NULL, 0 },
1160 /* 0xa8 */ { NULL, NULL, 0 },
1161 /* 0xa9 */ { NULL, NULL, 0 },
1162 /* 0xaa */ { NULL, NULL, 0 },
1163 /* 0xab */ { NULL, NULL, 0 },
1164 /* 0xac */ { NULL, NULL, 0 },
1165 /* 0xad */ { NULL, NULL, 0 },
1166 /* 0xae */ { NULL, NULL, 0 },
1167 /* 0xaf */ { NULL, NULL, 0 },
1168 /* 0xb0 */ { NULL, NULL, 0 },
1169 /* 0xb1 */ { NULL, NULL, 0 },
1170 /* 0xb2 */ { NULL, NULL, 0 },
1171 /* 0xb3 */ { NULL, NULL, 0 },
1172 /* 0xb4 */ { NULL, NULL, 0 },
1173 /* 0xb5 */ { NULL, NULL, 0 },
1174 /* 0xb6 */ { NULL, NULL, 0 },
1175 /* 0xb7 */ { NULL, NULL, 0 },
1176 /* 0xb8 */ { NULL, NULL, 0 },
1177 /* 0xb9 */ { NULL, NULL, 0 },
1178 /* 0xba */ { NULL, NULL, 0 },
1179 /* 0xbb */ { NULL, NULL, 0 },
1180 /* 0xbc */ { NULL, NULL, 0 },
1181 /* 0xbd */ { NULL, NULL, 0 },
1182 /* 0xbe */ { NULL, NULL, 0 },
1183 /* 0xbf */ { NULL, NULL, 0 },
1184 /* 0xc0 */ { "SMBsplopen",reply_printopen,AS_USER},
1185 /* 0xc1 */ { "SMBsplwr",reply_printwrite,AS_USER},
1186 /* 0xc2 */ { "SMBsplclose",reply_printclose,AS_USER},
1187 /* 0xc3 */ { "SMBsplretq",reply_printqueue,AS_USER},
1188 /* 0xc4 */ { NULL, NULL, 0 },
1189 /* 0xc5 */ { NULL, NULL, 0 },
1190 /* 0xc6 */ { NULL, NULL, 0 },
1191 /* 0xc7 */ { NULL, NULL, 0 },
1192 /* 0xc8 */ { NULL, NULL, 0 },
1193 /* 0xc9 */ { NULL, NULL, 0 },
1194 /* 0xca */ { NULL, NULL, 0 },
1195 /* 0xcb */ { NULL, NULL, 0 },
1196 /* 0xcc */ { NULL, NULL, 0 },
1197 /* 0xcd */ { NULL, NULL, 0 },
1198 /* 0xce */ { NULL, NULL, 0 },
1199 /* 0xcf */ { NULL, NULL, 0 },
1200 /* 0xd0 */ { "SMBsends",reply_sends,AS_GUEST},
1201 /* 0xd1 */ { "SMBsendb", NULL,AS_GUEST},
1202 /* 0xd2 */ { "SMBfwdname", NULL,AS_GUEST},
1203 /* 0xd3 */ { "SMBcancelf", NULL,AS_GUEST},
1204 /* 0xd4 */ { "SMBgetmac", NULL,AS_GUEST},
1205 /* 0xd5 */ { "SMBsendstrt",reply_sendstrt,AS_GUEST},
1206 /* 0xd6 */ { "SMBsendend",reply_sendend,AS_GUEST},
1207 /* 0xd7 */ { "SMBsendtxt",reply_sendtxt,AS_GUEST},
1208 /* 0xd8 */ { NULL, NULL, 0 },
1209 /* 0xd9 */ { NULL, NULL, 0 },
1210 /* 0xda */ { NULL, NULL, 0 },
1211 /* 0xdb */ { NULL, NULL, 0 },
1212 /* 0xdc */ { NULL, NULL, 0 },
1213 /* 0xdd */ { NULL, NULL, 0 },
1214 /* 0xde */ { NULL, NULL, 0 },
1215 /* 0xdf */ { NULL, NULL, 0 },
1216 /* 0xe0 */ { NULL, NULL, 0 },
1217 /* 0xe1 */ { NULL, NULL, 0 },
1218 /* 0xe2 */ { NULL, NULL, 0 },
1219 /* 0xe3 */ { NULL, NULL, 0 },
1220 /* 0xe4 */ { NULL, NULL, 0 },
1221 /* 0xe5 */ { NULL, NULL, 0 },
1222 /* 0xe6 */ { NULL, NULL, 0 },
1223 /* 0xe7 */ { NULL, NULL, 0 },
1224 /* 0xe8 */ { NULL, NULL, 0 },
1225 /* 0xe9 */ { NULL, NULL, 0 },
1226 /* 0xea */ { NULL, NULL, 0 },
1227 /* 0xeb */ { NULL, NULL, 0 },
1228 /* 0xec */ { NULL, NULL, 0 },
1229 /* 0xed */ { NULL, NULL, 0 },
1230 /* 0xee */ { NULL, NULL, 0 },
1231 /* 0xef */ { NULL, NULL, 0 },
1232 /* 0xf0 */ { NULL, NULL, 0 },
1233 /* 0xf1 */ { NULL, NULL, 0 },
1234 /* 0xf2 */ { NULL, NULL, 0 },
1235 /* 0xf3 */ { NULL, NULL, 0 },
1236 /* 0xf4 */ { NULL, NULL, 0 },
1237 /* 0xf5 */ { NULL, NULL, 0 },
1238 /* 0xf6 */ { NULL, NULL, 0 },
1239 /* 0xf7 */ { NULL, NULL, 0 },
1240 /* 0xf8 */ { NULL, NULL, 0 },
1241 /* 0xf9 */ { NULL, NULL, 0 },
1242 /* 0xfa */ { NULL, NULL, 0 },
1243 /* 0xfb */ { NULL, NULL, 0 },
1244 /* 0xfc */ { NULL, NULL, 0 },
1245 /* 0xfd */ { NULL, NULL, 0 },
1246 /* 0xfe */ { NULL, NULL, 0 },
1247 /* 0xff */ { NULL, NULL, 0 }
1251 /*******************************************************************
1252 allocate and initialize a reply packet
1253 ********************************************************************/
1255 static bool create_outbuf(TALLOC_CTX *mem_ctx, struct smb_request *req,
1256 const char *inbuf, char **outbuf, uint8_t num_words,
1257 uint32_t num_bytes)
1260 * Protect against integer wrap
1262 if ((num_bytes > 0xffffff)
1263 || ((num_bytes + smb_size + num_words*2) > 0xffffff)) {
1264 char *msg;
1265 if (asprintf(&msg, "num_bytes too large: %u",
1266 (unsigned)num_bytes) == -1) {
1267 msg = discard_const_p(char, "num_bytes too large");
1269 smb_panic(msg);
1272 *outbuf = talloc_array(mem_ctx, char,
1273 smb_size + num_words*2 + num_bytes);
1274 if (*outbuf == NULL) {
1275 return false;
1278 construct_reply_common(req, inbuf, *outbuf);
1279 srv_set_message(*outbuf, num_words, num_bytes, false);
1281 * Zero out the word area, the caller has to take care of the bcc area
1282 * himself
1284 if (num_words != 0) {
1285 memset(*outbuf + smb_vwv0, 0, num_words*2);
1288 return true;
1291 void reply_outbuf(struct smb_request *req, uint8 num_words, uint32 num_bytes)
1293 char *outbuf;
1294 if (!create_outbuf(req, req, (const char *)req->inbuf, &outbuf, num_words,
1295 num_bytes)) {
1296 smb_panic("could not allocate output buffer\n");
1298 req->outbuf = (uint8_t *)outbuf;
1302 /*******************************************************************
1303 Dump a packet to a file.
1304 ********************************************************************/
1306 static void smb_dump(const char *name, int type, const char *data, ssize_t len)
1308 int fd, i;
1309 char *fname = NULL;
1310 if (DEBUGLEVEL < 50) {
1311 return;
1314 if (len < 4) len = smb_len(data)+4;
1315 for (i=1;i<100;i++) {
1316 if (asprintf(&fname, "/tmp/%s.%d.%s", name, i,
1317 type ? "req" : "resp") == -1) {
1318 return;
1320 fd = open(fname, O_WRONLY|O_CREAT|O_EXCL, 0644);
1321 if (fd != -1 || errno != EEXIST) break;
1323 if (fd != -1) {
1324 ssize_t ret = write(fd, data, len);
1325 if (ret != len)
1326 DEBUG(0,("smb_dump: problem: write returned %d\n", (int)ret ));
1327 close(fd);
1328 DEBUG(0,("created %s len %lu\n", fname, (unsigned long)len));
1330 SAFE_FREE(fname);
1333 /****************************************************************************
1334 Prepare everything for calling the actual request function, and potentially
1335 call the request function via the "new" interface.
1337 Return False if the "legacy" function needs to be called, everything is
1338 prepared.
1340 Return True if we're done.
1342 I know this API sucks, but it is the one with the least code change I could
1343 find.
1344 ****************************************************************************/
1346 static connection_struct *switch_message(uint8 type, struct smb_request *req, int size)
1348 int flags;
1349 uint16 session_tag;
1350 connection_struct *conn = NULL;
1351 struct smbd_server_connection *sconn = req->sconn;
1352 char *raddr;
1354 errno = 0;
1356 if (smb_messages[type].fn == NULL) {
1357 DEBUG(0,("Unknown message type %d!\n",type));
1358 smb_dump("Unknown", 1, (const char *)req->inbuf, size);
1359 reply_unknown_new(req, type);
1360 return NULL;
1363 flags = smb_messages[type].flags;
1365 /* In share mode security we must ignore the vuid. */
1366 session_tag = (lp_security() == SEC_SHARE)
1367 ? UID_FIELD_INVALID : req->vuid;
1368 conn = req->conn;
1370 DEBUG(3,("switch message %s (pid %d) conn 0x%lx\n", smb_fn_name(type),
1371 (int)sys_getpid(), (unsigned long)conn));
1373 smb_dump(smb_fn_name(type), 1, (const char *)req->inbuf, size);
1375 /* Ensure this value is replaced in the incoming packet. */
1376 SSVAL(discard_const_p(uint8_t, req->inbuf),smb_uid,session_tag);
1379 * Ensure the correct username is in current_user_info. This is a
1380 * really ugly bugfix for problems with multiple session_setup_and_X's
1381 * being done and allowing %U and %G substitutions to work correctly.
1382 * There is a reason this code is done here, don't move it unless you
1383 * know what you're doing... :-).
1384 * JRA.
1387 if (session_tag != sconn->smb1.sessions.last_session_tag) {
1388 user_struct *vuser = NULL;
1390 sconn->smb1.sessions.last_session_tag = session_tag;
1391 if(session_tag != UID_FIELD_INVALID) {
1392 vuser = get_valid_user_struct(sconn, session_tag);
1393 if (vuser) {
1394 set_current_user_info(
1395 vuser->session_info->unix_info->sanitized_username,
1396 vuser->session_info->unix_info->unix_name,
1397 vuser->session_info->info->domain_name);
1402 /* Does this call need to be run as the connected user? */
1403 if (flags & AS_USER) {
1405 /* Does this call need a valid tree connection? */
1406 if (!conn) {
1408 * Amazingly, the error code depends on the command
1409 * (from Samba4).
1411 if (type == SMBntcreateX) {
1412 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
1413 } else {
1414 reply_nterror(req, NT_STATUS_NETWORK_NAME_DELETED);
1416 return NULL;
1419 if (!change_to_user(conn,session_tag)) {
1420 DEBUG(0, ("Error: Could not change to user. Removing "
1421 "deferred open, mid=%llu.\n",
1422 (unsigned long long)req->mid));
1423 reply_force_doserror(req, ERRSRV, ERRbaduid);
1424 return conn;
1427 /* All NEED_WRITE and CAN_IPC flags must also have AS_USER. */
1429 /* Does it need write permission? */
1430 if ((flags & NEED_WRITE) && !CAN_WRITE(conn)) {
1431 reply_nterror(req, NT_STATUS_MEDIA_WRITE_PROTECTED);
1432 return conn;
1435 /* IPC services are limited */
1436 if (IS_IPC(conn) && !(flags & CAN_IPC)) {
1437 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1438 return conn;
1440 } else {
1441 /* This call needs to be run as root */
1442 change_to_root_user();
1445 /* load service specific parameters */
1446 if (conn) {
1447 if (req->encrypted) {
1448 conn->encrypted_tid = true;
1449 /* encrypted required from now on. */
1450 conn->encrypt_level = Required;
1451 } else if (ENCRYPTION_REQUIRED(conn)) {
1452 if (req->cmd != SMBtrans2 && req->cmd != SMBtranss2) {
1453 exit_server_cleanly("encryption required "
1454 "on connection");
1455 return conn;
1459 if (!set_current_service(conn,SVAL(req->inbuf,smb_flg),
1460 (flags & (AS_USER|DO_CHDIR)
1461 ?True:False))) {
1462 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1463 return conn;
1465 conn->num_smb_operations++;
1468 raddr = tsocket_address_inet_addr_string(sconn->remote_address,
1469 talloc_tos());
1470 if (raddr == NULL) {
1471 reply_nterror(req, NT_STATUS_NO_MEMORY);
1472 return conn;
1475 /* does this protocol need to be run as guest? */
1476 if ((flags & AS_GUEST)
1477 && (!change_to_guest() ||
1478 !allow_access(lp_hostsdeny(-1), lp_hostsallow(-1),
1479 sconn->remote_hostname,
1480 raddr))) {
1481 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1482 return conn;
1485 smb_messages[type].fn(req);
1486 return req->conn;
1489 /****************************************************************************
1490 Construct a reply to the incoming packet.
1491 ****************************************************************************/
1493 static void construct_reply(struct smbd_server_connection *sconn,
1494 char *inbuf, int size, size_t unread_bytes,
1495 uint32_t seqnum, bool encrypted,
1496 struct smb_perfcount_data *deferred_pcd)
1498 connection_struct *conn;
1499 struct smb_request *req;
1501 if (!(req = talloc(talloc_tos(), struct smb_request))) {
1502 smb_panic("could not allocate smb_request");
1505 if (!init_smb_request(req, sconn, (uint8 *)inbuf, unread_bytes,
1506 encrypted, seqnum)) {
1507 exit_server_cleanly("Invalid SMB request");
1510 req->inbuf = (uint8_t *)talloc_move(req, &inbuf);
1512 /* we popped this message off the queue - keep original perf data */
1513 if (deferred_pcd)
1514 req->pcd = *deferred_pcd;
1515 else {
1516 SMB_PERFCOUNT_START(&req->pcd);
1517 SMB_PERFCOUNT_SET_OP(&req->pcd, req->cmd);
1518 SMB_PERFCOUNT_SET_MSGLEN_IN(&req->pcd, size);
1521 conn = switch_message(req->cmd, req, size);
1523 if (req->unread_bytes) {
1524 /* writeX failed. drain socket. */
1525 if (drain_socket(req->sconn->sock, req->unread_bytes) !=
1526 req->unread_bytes) {
1527 smb_panic("failed to drain pending bytes");
1529 req->unread_bytes = 0;
1532 if (req->done) {
1533 TALLOC_FREE(req);
1534 return;
1537 if (req->outbuf == NULL) {
1538 return;
1541 if (CVAL(req->outbuf,0) == 0) {
1542 show_msg((char *)req->outbuf);
1545 if (!srv_send_smb(req->sconn,
1546 (char *)req->outbuf,
1547 true, req->seqnum+1,
1548 IS_CONN_ENCRYPTED(conn)||req->encrypted,
1549 &req->pcd)) {
1550 exit_server_cleanly("construct_reply: srv_send_smb failed.");
1553 TALLOC_FREE(req);
1555 return;
1558 /****************************************************************************
1559 Process an smb from the client
1560 ****************************************************************************/
1561 static void process_smb(struct smbd_server_connection *sconn,
1562 uint8_t *inbuf, size_t nread, size_t unread_bytes,
1563 uint32_t seqnum, bool encrypted,
1564 struct smb_perfcount_data *deferred_pcd)
1566 int msg_type = CVAL(inbuf,0);
1568 DO_PROFILE_INC(smb_count);
1570 DEBUG( 6, ( "got message type 0x%x of len 0x%x\n", msg_type,
1571 smb_len(inbuf) ) );
1572 DEBUG(3, ("Transaction %d of length %d (%u toread)\n",
1573 sconn->trans_num, (int)nread, (unsigned int)unread_bytes));
1575 if (msg_type != NBSSmessage) {
1577 * NetBIOS session request, keepalive, etc.
1579 reply_special(sconn, (char *)inbuf, nread);
1580 goto done;
1583 if (sconn->using_smb2) {
1584 /* At this point we're not really using smb2,
1585 * we make the decision here.. */
1586 if (smbd_is_smb2_header(inbuf, nread)) {
1587 smbd_smb2_first_negprot(sconn, inbuf, nread);
1588 return;
1589 } else if (nread >= smb_size && valid_smb_header(sconn, inbuf)
1590 && CVAL(inbuf, smb_com) != 0x72) {
1591 /* This is a non-negprot SMB1 packet.
1592 Disable SMB2 from now on. */
1593 sconn->using_smb2 = false;
1597 /* Make sure this is an SMB packet. smb_size contains NetBIOS header
1598 * so subtract 4 from it. */
1599 if ((nread < (smb_size - 4)) || !valid_smb_header(sconn, inbuf)) {
1600 DEBUG(2,("Non-SMB packet of length %d. Terminating server\n",
1601 smb_len(inbuf)));
1603 /* special magic for immediate exit */
1604 if ((nread == 9) &&
1605 (IVAL(inbuf, 4) == 0x74697865) &&
1606 lp_parm_bool(-1, "smbd", "suicide mode", false)) {
1607 uint8_t exitcode = CVAL(inbuf, 8);
1608 DEBUG(1, ("Exiting immediately with code %d\n",
1609 (int)exitcode));
1610 exit(exitcode);
1613 exit_server_cleanly("Non-SMB packet");
1616 show_msg((char *)inbuf);
1618 construct_reply(sconn, (char *)inbuf, nread, unread_bytes, seqnum,
1619 encrypted, deferred_pcd);
1620 sconn->trans_num++;
1622 done:
1623 sconn->num_requests++;
1625 /* The timeout_processing function isn't run nearly
1626 often enough to implement 'max log size' without
1627 overrunning the size of the file by many megabytes.
1628 This is especially true if we are running at debug
1629 level 10. Checking every 50 SMBs is a nice
1630 tradeoff of performance vs log file size overrun. */
1632 if ((sconn->num_requests % 50) == 0 &&
1633 need_to_check_log_size()) {
1634 change_to_root_user();
1635 check_log_size();
1639 /****************************************************************************
1640 Return a string containing the function name of a SMB command.
1641 ****************************************************************************/
1643 const char *smb_fn_name(int type)
1645 const char *unknown_name = "SMBunknown";
1647 if (smb_messages[type].name == NULL)
1648 return(unknown_name);
1650 return(smb_messages[type].name);
1653 /****************************************************************************
1654 Helper functions for contruct_reply.
1655 ****************************************************************************/
1657 void add_to_common_flags2(uint32 v)
1659 common_flags2 |= v;
1662 void remove_from_common_flags2(uint32 v)
1664 common_flags2 &= ~v;
1667 static void construct_reply_common(struct smb_request *req, const char *inbuf,
1668 char *outbuf)
1670 uint16_t in_flags2 = SVAL(inbuf,smb_flg2);
1671 uint16_t out_flags2 = common_flags2;
1673 out_flags2 |= in_flags2 & FLAGS2_UNICODE_STRINGS;
1674 out_flags2 |= in_flags2 & FLAGS2_SMB_SECURITY_SIGNATURES;
1675 out_flags2 |= in_flags2 & FLAGS2_SMB_SECURITY_SIGNATURES_REQUIRED;
1677 srv_set_message(outbuf,0,0,false);
1679 SCVAL(outbuf, smb_com, req->cmd);
1680 SIVAL(outbuf,smb_rcls,0);
1681 SCVAL(outbuf,smb_flg, FLAG_REPLY | (CVAL(inbuf,smb_flg) & FLAG_CASELESS_PATHNAMES));
1682 SSVAL(outbuf,smb_flg2, out_flags2);
1683 memset(outbuf+smb_pidhigh,'\0',(smb_tid-smb_pidhigh));
1684 memcpy(outbuf+smb_ss_field, inbuf+smb_ss_field, 8);
1686 SSVAL(outbuf,smb_tid,SVAL(inbuf,smb_tid));
1687 SSVAL(outbuf,smb_pid,SVAL(inbuf,smb_pid));
1688 SSVAL(outbuf,smb_uid,SVAL(inbuf,smb_uid));
1689 SSVAL(outbuf,smb_mid,SVAL(inbuf,smb_mid));
1692 void construct_reply_common_req(struct smb_request *req, char *outbuf)
1694 construct_reply_common(req, (const char *)req->inbuf, outbuf);
1698 * How many bytes have we already accumulated up to the current wct field
1699 * offset?
1702 size_t req_wct_ofs(struct smb_request *req)
1704 size_t buf_size;
1706 if (req->chain_outbuf == NULL) {
1707 return smb_wct - 4;
1709 buf_size = talloc_get_size(req->chain_outbuf);
1710 if ((buf_size % 4) != 0) {
1711 buf_size += (4 - (buf_size % 4));
1713 return buf_size - 4;
1717 * Hack around reply_nterror & friends not being aware of chained requests,
1718 * generating illegal (i.e. wct==0) chain replies.
1721 static void fixup_chain_error_packet(struct smb_request *req)
1723 uint8_t *outbuf = req->outbuf;
1724 req->outbuf = NULL;
1725 reply_outbuf(req, 2, 0);
1726 memcpy(req->outbuf, outbuf, smb_wct);
1727 TALLOC_FREE(outbuf);
1728 SCVAL(req->outbuf, smb_vwv0, 0xff);
1732 * @brief Find the smb_cmd offset of the last command pushed
1733 * @param[in] buf The buffer we're building up
1734 * @retval Where can we put our next andx cmd?
1736 * While chaining requests, the "next" request we're looking at needs to put
1737 * its SMB_Command before the data the previous request already built up added
1738 * to the chain. Find the offset to the place where we have to put our cmd.
1741 static bool find_andx_cmd_ofs(uint8_t *buf, size_t *pofs)
1743 uint8_t cmd;
1744 size_t ofs;
1746 cmd = CVAL(buf, smb_com);
1748 if (!is_andx_req(cmd)) {
1749 return false;
1752 ofs = smb_vwv0;
1754 while (CVAL(buf, ofs) != 0xff) {
1756 if (!is_andx_req(CVAL(buf, ofs))) {
1757 return false;
1761 * ofs is from start of smb header, so add the 4 length
1762 * bytes. The next cmd is right after the wct field.
1764 ofs = SVAL(buf, ofs+2) + 4 + 1;
1766 if (ofs+4 >= talloc_get_size(buf)) {
1767 return false;
1771 *pofs = ofs;
1772 return true;
1776 * @brief Do the smb chaining at a buffer level
1777 * @param[in] poutbuf Pointer to the talloc'ed buffer to be modified
1778 * @param[in] andx_buf Buffer to be appended
1781 static bool smb_splice_chain(uint8_t **poutbuf, const uint8_t *andx_buf)
1783 uint8_t smb_command = CVAL(andx_buf, smb_com);
1784 uint8_t wct = CVAL(andx_buf, smb_wct);
1785 const uint16_t *vwv = (const uint16_t *)(andx_buf + smb_vwv);
1786 uint32_t num_bytes = smb_buflen(andx_buf);
1787 const uint8_t *bytes = (const uint8_t *)smb_buf(andx_buf);
1789 uint8_t *outbuf;
1790 size_t old_size, new_size;
1791 size_t ofs;
1792 size_t chain_padding = 0;
1793 size_t andx_cmd_ofs;
1796 old_size = talloc_get_size(*poutbuf);
1798 if ((old_size % 4) != 0) {
1800 * Align the wct field of subsequent requests to a 4-byte
1801 * boundary
1803 chain_padding = 4 - (old_size % 4);
1807 * After the old request comes the new wct field (1 byte), the vwv's
1808 * and the num_bytes field.
1811 new_size = old_size + chain_padding + 1 + wct * sizeof(uint16_t) + 2;
1812 new_size += num_bytes;
1814 if ((smb_command != SMBwriteX) && (new_size > 0xffff)) {
1815 DEBUG(1, ("smb_splice_chain: %u bytes won't fit\n",
1816 (unsigned)new_size));
1817 return false;
1820 outbuf = talloc_realloc(NULL, *poutbuf, uint8_t, new_size);
1821 if (outbuf == NULL) {
1822 DEBUG(0, ("talloc failed\n"));
1823 return false;
1825 *poutbuf = outbuf;
1827 if (!find_andx_cmd_ofs(outbuf, &andx_cmd_ofs)) {
1828 DEBUG(1, ("invalid command chain\n"));
1829 *poutbuf = talloc_realloc(NULL, *poutbuf, uint8_t, old_size);
1830 return false;
1833 if (chain_padding != 0) {
1834 memset(outbuf + old_size, 0, chain_padding);
1835 old_size += chain_padding;
1838 SCVAL(outbuf, andx_cmd_ofs, smb_command);
1839 SSVAL(outbuf, andx_cmd_ofs + 2, old_size - 4);
1841 ofs = old_size;
1844 * Push the chained request:
1846 * wct field
1849 SCVAL(outbuf, ofs, wct);
1850 ofs += 1;
1853 * vwv array
1856 memcpy(outbuf + ofs, vwv, sizeof(uint16_t) * wct);
1857 ofs += sizeof(uint16_t) * wct;
1860 * bcc (byte count)
1863 SSVAL(outbuf, ofs, num_bytes);
1864 ofs += sizeof(uint16_t);
1867 * The bytes field
1870 memcpy(outbuf + ofs, bytes, num_bytes);
1872 return true;
1875 /****************************************************************************
1876 Construct a chained reply and add it to the already made reply
1877 ****************************************************************************/
1879 void chain_reply(struct smb_request *req)
1881 size_t smblen = smb_len(req->inbuf);
1882 size_t already_used, length_needed;
1883 uint8_t chain_cmd;
1884 uint32_t chain_offset; /* uint32_t to avoid overflow */
1886 uint8_t wct;
1887 const uint16_t *vwv;
1888 uint16_t buflen;
1889 const uint8_t *buf;
1891 if (IVAL(req->outbuf, smb_rcls) != 0) {
1892 fixup_chain_error_packet(req);
1896 * Any of the AndX requests and replies have at least a wct of
1897 * 2. vwv[0] is the next command, vwv[1] is the offset from the
1898 * beginning of the SMB header to the next wct field.
1900 * None of the AndX requests put anything valuable in vwv[0] and [1],
1901 * so we can overwrite it here to form the chain.
1904 if ((req->wct < 2) || (CVAL(req->outbuf, smb_wct) < 2)) {
1905 if (req->chain_outbuf == NULL) {
1906 req->chain_outbuf = talloc_realloc(
1907 req, req->outbuf, uint8_t,
1908 smb_len(req->outbuf) + 4);
1909 if (req->chain_outbuf == NULL) {
1910 smb_panic("talloc failed");
1913 req->outbuf = NULL;
1914 goto error;
1918 * Here we assume that this is the end of the chain. For that we need
1919 * to set "next command" to 0xff and the offset to 0. If we later find
1920 * more commands in the chain, this will be overwritten again.
1923 SCVAL(req->outbuf, smb_vwv0, 0xff);
1924 SCVAL(req->outbuf, smb_vwv0+1, 0);
1925 SSVAL(req->outbuf, smb_vwv1, 0);
1927 if (req->chain_outbuf == NULL) {
1929 * In req->chain_outbuf we collect all the replies. Start the
1930 * chain by copying in the first reply.
1932 * We do the realloc because later on we depend on
1933 * talloc_get_size to determine the length of
1934 * chain_outbuf. The reply_xxx routines might have
1935 * over-allocated (reply_pipe_read_and_X used to be such an
1936 * example).
1938 req->chain_outbuf = talloc_realloc(
1939 req, req->outbuf, uint8_t, smb_len(req->outbuf) + 4);
1940 if (req->chain_outbuf == NULL) {
1941 smb_panic("talloc failed");
1943 req->outbuf = NULL;
1944 } else {
1946 * Update smb headers where subsequent chained commands
1947 * may have updated them.
1949 SSVAL(req->chain_outbuf, smb_tid, SVAL(req->outbuf, smb_tid));
1950 SSVAL(req->chain_outbuf, smb_uid, SVAL(req->outbuf, smb_uid));
1952 if (!smb_splice_chain(&req->chain_outbuf, req->outbuf)) {
1953 goto error;
1955 TALLOC_FREE(req->outbuf);
1959 * We use the old request's vwv field to grab the next chained command
1960 * and offset into the chained fields.
1963 chain_cmd = CVAL(req->vwv+0, 0);
1964 chain_offset = SVAL(req->vwv+1, 0);
1966 if (chain_cmd == 0xff) {
1968 * End of chain, no more requests from the client. So ship the
1969 * replies.
1971 smb_setlen((char *)(req->chain_outbuf),
1972 talloc_get_size(req->chain_outbuf) - 4);
1974 if (!srv_send_smb(req->sconn, (char *)req->chain_outbuf,
1975 true, req->seqnum+1,
1976 IS_CONN_ENCRYPTED(req->conn)
1977 ||req->encrypted,
1978 &req->pcd)) {
1979 exit_server_cleanly("chain_reply: srv_send_smb "
1980 "failed.");
1982 TALLOC_FREE(req->chain_outbuf);
1983 req->done = true;
1984 return;
1987 /* add a new perfcounter for this element of chain */
1988 SMB_PERFCOUNT_ADD(&req->pcd);
1989 SMB_PERFCOUNT_SET_OP(&req->pcd, chain_cmd);
1990 SMB_PERFCOUNT_SET_MSGLEN_IN(&req->pcd, smblen);
1993 * Check if the client tries to fool us. The chain offset
1994 * needs to point beyond the current request in the chain, it
1995 * needs to strictly grow. Otherwise we might be tricked into
1996 * an endless loop always processing the same request over and
1997 * over again. We used to assume that vwv and the byte buffer
1998 * array in a chain are always attached, but OS/2 the
1999 * Write&X/Read&X chain puts the Read&X vwv array right behind
2000 * the Write&X vwv chain. The Write&X bcc array is put behind
2001 * the Read&X vwv array. So now we check whether the chain
2002 * offset points strictly behind the previous vwv
2003 * array. req->buf points right after the vwv array of the
2004 * previous request. See
2005 * https://bugzilla.samba.org/show_bug.cgi?id=8360 for more
2006 * information.
2009 already_used = PTR_DIFF(req->buf, smb_base(req->inbuf));
2010 if (chain_offset <= already_used) {
2011 goto error;
2015 * Next check: Make sure the chain offset does not point beyond the
2016 * overall smb request length.
2019 length_needed = chain_offset+1; /* wct */
2020 if (length_needed > smblen) {
2021 goto error;
2025 * Now comes the pointer magic. Goal here is to set up req->vwv and
2026 * req->buf correctly again to be able to call the subsequent
2027 * switch_message(). The chain offset (the former vwv[1]) points at
2028 * the new wct field.
2031 wct = CVAL(smb_base(req->inbuf), chain_offset);
2034 * Next consistency check: Make the new vwv array fits in the overall
2035 * smb request.
2038 length_needed += (wct+1)*sizeof(uint16_t); /* vwv+buflen */
2039 if (length_needed > smblen) {
2040 goto error;
2042 vwv = (const uint16_t *)(smb_base(req->inbuf) + chain_offset + 1);
2045 * Now grab the new byte buffer....
2048 buflen = SVAL(vwv+wct, 0);
2051 * .. and check that it fits.
2054 length_needed += buflen;
2055 if (length_needed > smblen) {
2056 goto error;
2058 buf = (const uint8_t *)(vwv+wct+1);
2060 req->cmd = chain_cmd;
2061 req->wct = wct;
2062 req->vwv = discard_const_p(uint16_t, vwv);
2063 req->buflen = buflen;
2064 req->buf = buf;
2066 switch_message(chain_cmd, req, smblen);
2068 if (req->outbuf == NULL) {
2070 * This happens if the chained command has suspended itself or
2071 * if it has called srv_send_smb() itself.
2073 return;
2077 * We end up here if the chained command was not itself chained or
2078 * suspended, but for example a close() command. We now need to splice
2079 * the chained commands' outbuf into the already built up chain_outbuf
2080 * and ship the result.
2082 goto done;
2084 error:
2086 * We end up here if there's any error in the chain syntax. Report a
2087 * DOS error, just like Windows does.
2089 reply_force_doserror(req, ERRSRV, ERRerror);
2090 fixup_chain_error_packet(req);
2092 done:
2094 * This scary statement intends to set the
2095 * FLAGS2_32_BIT_ERROR_CODES flg2 field in req->chain_outbuf
2096 * to the value req->outbuf carries
2098 SSVAL(req->chain_outbuf, smb_flg2,
2099 (SVAL(req->chain_outbuf, smb_flg2) & ~FLAGS2_32_BIT_ERROR_CODES)
2100 | (SVAL(req->outbuf, smb_flg2) & FLAGS2_32_BIT_ERROR_CODES));
2103 * Transfer the error codes from the subrequest to the main one
2105 SSVAL(req->chain_outbuf, smb_rcls, SVAL(req->outbuf, smb_rcls));
2106 SSVAL(req->chain_outbuf, smb_err, SVAL(req->outbuf, smb_err));
2108 if (!smb_splice_chain(&req->chain_outbuf, req->outbuf)) {
2109 exit_server_cleanly("chain_reply: smb_splice_chain failed\n");
2111 TALLOC_FREE(req->outbuf);
2113 smb_setlen((char *)(req->chain_outbuf),
2114 talloc_get_size(req->chain_outbuf) - 4);
2116 show_msg((char *)(req->chain_outbuf));
2118 if (!srv_send_smb(req->sconn, (char *)req->chain_outbuf,
2119 true, req->seqnum+1,
2120 IS_CONN_ENCRYPTED(req->conn)||req->encrypted,
2121 &req->pcd)) {
2122 exit_server_cleanly("chain_reply: srv_send_smb failed.");
2124 TALLOC_FREE(req->chain_outbuf);
2125 req->done = true;
2128 /****************************************************************************
2129 Check if services need reloading.
2130 ****************************************************************************/
2132 static void check_reload(struct smbd_server_connection *sconn, time_t t)
2135 if (last_smb_conf_reload_time == 0) {
2136 last_smb_conf_reload_time = t;
2139 if (t >= last_smb_conf_reload_time+SMBD_RELOAD_CHECK) {
2140 reload_services(sconn, conn_snum_used, true);
2141 last_smb_conf_reload_time = t;
2145 static bool fd_is_readable(int fd)
2147 int ret, revents;
2149 ret = poll_one_fd(fd, POLLIN|POLLHUP, 0, &revents);
2151 return ((ret > 0) && ((revents & (POLLIN|POLLHUP|POLLERR)) != 0));
2155 static void smbd_server_connection_write_handler(
2156 struct smbd_server_connection *sconn)
2158 /* TODO: make write nonblocking */
2161 static void smbd_server_connection_read_handler(
2162 struct smbd_server_connection *sconn, int fd)
2164 uint8_t *inbuf = NULL;
2165 size_t inbuf_len = 0;
2166 size_t unread_bytes = 0;
2167 bool encrypted = false;
2168 TALLOC_CTX *mem_ctx = talloc_tos();
2169 NTSTATUS status;
2170 uint32_t seqnum;
2172 bool from_client;
2174 if (lp_async_smb_echo_handler()
2175 && fd_is_readable(sconn->smb1.echo_handler.trusted_fd)) {
2177 * This is the super-ugly hack to prefer the packets
2178 * forwarded by the echo handler over the ones by the
2179 * client directly
2181 fd = sconn->smb1.echo_handler.trusted_fd;
2184 from_client = (sconn->sock == fd);
2186 if (from_client) {
2187 smbd_lock_socket(sconn);
2189 if (!fd_is_readable(fd)) {
2190 DEBUG(10,("the echo listener was faster\n"));
2191 smbd_unlock_socket(sconn);
2192 return;
2196 /* TODO: make this completely nonblocking */
2197 status = receive_smb_talloc(mem_ctx, sconn, fd,
2198 (char **)(void *)&inbuf,
2199 0, /* timeout */
2200 &unread_bytes,
2201 &encrypted,
2202 &inbuf_len, &seqnum,
2203 false /* trusted channel */);
2205 if (from_client) {
2206 smbd_unlock_socket(sconn);
2209 if (NT_STATUS_EQUAL(status, NT_STATUS_RETRY)) {
2210 goto process;
2212 if (NT_STATUS_IS_ERR(status)) {
2213 exit_server_cleanly("failed to receive smb request");
2215 if (!NT_STATUS_IS_OK(status)) {
2216 return;
2219 process:
2220 process_smb(sconn, inbuf, inbuf_len, unread_bytes,
2221 seqnum, encrypted, NULL);
2224 static void smbd_server_connection_handler(struct event_context *ev,
2225 struct fd_event *fde,
2226 uint16_t flags,
2227 void *private_data)
2229 struct smbd_server_connection *conn = talloc_get_type(private_data,
2230 struct smbd_server_connection);
2232 if (flags & EVENT_FD_WRITE) {
2233 smbd_server_connection_write_handler(conn);
2234 return;
2236 if (flags & EVENT_FD_READ) {
2237 smbd_server_connection_read_handler(conn, conn->sock);
2238 return;
2242 static void smbd_server_echo_handler(struct event_context *ev,
2243 struct fd_event *fde,
2244 uint16_t flags,
2245 void *private_data)
2247 struct smbd_server_connection *conn = talloc_get_type(private_data,
2248 struct smbd_server_connection);
2250 if (flags & EVENT_FD_WRITE) {
2251 smbd_server_connection_write_handler(conn);
2252 return;
2254 if (flags & EVENT_FD_READ) {
2255 smbd_server_connection_read_handler(
2256 conn, conn->smb1.echo_handler.trusted_fd);
2257 return;
2261 #ifdef CLUSTER_SUPPORT
2262 /****************************************************************************
2263 received when we should release a specific IP
2264 ****************************************************************************/
2265 static void release_ip(const char *ip, void *priv)
2267 const char *addr = (const char *)priv;
2268 const char *p = addr;
2270 if (strncmp("::ffff:", addr, 7) == 0) {
2271 p = addr + 7;
2274 DEBUG(10, ("Got release IP message for %s, "
2275 "our address is %s\n", ip, p));
2277 if ((strcmp(p, ip) == 0) || ((p != addr) && strcmp(addr, ip) == 0)) {
2278 /* we can't afford to do a clean exit - that involves
2279 database writes, which would potentially mean we
2280 are still running after the failover has finished -
2281 we have to get rid of this process ID straight
2282 away */
2283 DEBUG(0,("Got release IP message for our IP %s - exiting immediately\n",
2284 ip));
2285 /* note we must exit with non-zero status so the unclean handler gets
2286 called in the parent, so that the brl database is tickled */
2287 _exit(1);
2291 static int client_get_tcp_info(int sock, struct sockaddr_storage *server,
2292 struct sockaddr_storage *client)
2294 socklen_t length;
2295 length = sizeof(*server);
2296 if (getsockname(sock, (struct sockaddr *)server, &length) != 0) {
2297 return -1;
2299 length = sizeof(*client);
2300 if (getpeername(sock, (struct sockaddr *)client, &length) != 0) {
2301 return -1;
2303 return 0;
2305 #endif
2308 * Send keepalive packets to our client
2310 static bool keepalive_fn(const struct timeval *now, void *private_data)
2312 struct smbd_server_connection *sconn = talloc_get_type_abort(
2313 private_data, struct smbd_server_connection);
2314 bool ret;
2316 if (sconn->using_smb2) {
2317 /* Don't do keepalives on an SMB2 connection. */
2318 return false;
2321 smbd_lock_socket(sconn);
2322 ret = send_keepalive(sconn->sock);
2323 smbd_unlock_socket(sconn);
2325 if (!ret) {
2326 char addr[INET6_ADDRSTRLEN];
2328 * Try and give an error message saying what
2329 * client failed.
2331 DEBUG(0, ("send_keepalive failed for client %s. "
2332 "Error %s - exiting\n",
2333 get_peer_addr(sconn->sock, addr, sizeof(addr)),
2334 strerror(errno)));
2335 return False;
2337 return True;
2341 * Do the recurring check if we're idle
2343 static bool deadtime_fn(const struct timeval *now, void *private_data)
2345 struct smbd_server_connection *sconn =
2346 (struct smbd_server_connection *)private_data;
2348 if ((conn_num_open(sconn) == 0)
2349 || (conn_idle_all(sconn, now->tv_sec))) {
2350 DEBUG( 2, ( "Closing idle connection\n" ) );
2351 messaging_send(sconn->msg_ctx,
2352 messaging_server_id(sconn->msg_ctx),
2353 MSG_SHUTDOWN, &data_blob_null);
2354 return False;
2357 return True;
2361 * Do the recurring log file and smb.conf reload checks.
2364 static bool housekeeping_fn(const struct timeval *now, void *private_data)
2366 struct smbd_server_connection *sconn = talloc_get_type_abort(
2367 private_data, struct smbd_server_connection);
2369 DEBUG(5, ("housekeeping\n"));
2371 change_to_root_user();
2373 /* update printer queue caches if necessary */
2374 update_monitored_printq_cache(sconn->msg_ctx);
2376 /* check if we need to reload services */
2377 check_reload(sconn, time_mono(NULL));
2379 /* Change machine password if neccessary. */
2380 attempt_machine_password_change();
2383 * Force a log file check.
2385 force_check_log_size();
2386 check_log_size();
2387 return true;
2391 * Read an smb packet in the echo handler child, giving the parent
2392 * smbd one second to react once the socket becomes readable.
2395 struct smbd_echo_read_state {
2396 struct tevent_context *ev;
2397 struct smbd_server_connection *sconn;
2399 char *buf;
2400 size_t buflen;
2401 uint32_t seqnum;
2404 static void smbd_echo_read_readable(struct tevent_req *subreq);
2405 static void smbd_echo_read_waited(struct tevent_req *subreq);
2407 static struct tevent_req *smbd_echo_read_send(
2408 TALLOC_CTX *mem_ctx, struct tevent_context *ev,
2409 struct smbd_server_connection *sconn)
2411 struct tevent_req *req, *subreq;
2412 struct smbd_echo_read_state *state;
2414 req = tevent_req_create(mem_ctx, &state,
2415 struct smbd_echo_read_state);
2416 if (req == NULL) {
2417 return NULL;
2419 state->ev = ev;
2420 state->sconn = sconn;
2422 subreq = wait_for_read_send(state, ev, sconn->sock);
2423 if (tevent_req_nomem(subreq, req)) {
2424 return tevent_req_post(req, ev);
2426 tevent_req_set_callback(subreq, smbd_echo_read_readable, req);
2427 return req;
2430 static void smbd_echo_read_readable(struct tevent_req *subreq)
2432 struct tevent_req *req = tevent_req_callback_data(
2433 subreq, struct tevent_req);
2434 struct smbd_echo_read_state *state = tevent_req_data(
2435 req, struct smbd_echo_read_state);
2436 bool ok;
2437 int err;
2439 ok = wait_for_read_recv(subreq, &err);
2440 TALLOC_FREE(subreq);
2441 if (!ok) {
2442 tevent_req_nterror(req, map_nt_error_from_unix(err));
2443 return;
2447 * Give the parent smbd one second to step in
2450 subreq = tevent_wakeup_send(
2451 state, state->ev, timeval_current_ofs(1, 0));
2452 if (tevent_req_nomem(subreq, req)) {
2453 return;
2455 tevent_req_set_callback(subreq, smbd_echo_read_waited, req);
2458 static void smbd_echo_read_waited(struct tevent_req *subreq)
2460 struct tevent_req *req = tevent_req_callback_data(
2461 subreq, struct tevent_req);
2462 struct smbd_echo_read_state *state = tevent_req_data(
2463 req, struct smbd_echo_read_state);
2464 struct smbd_server_connection *sconn = state->sconn;
2465 bool ok;
2466 NTSTATUS status;
2467 size_t unread = 0;
2468 bool encrypted;
2470 ok = tevent_wakeup_recv(subreq);
2471 TALLOC_FREE(subreq);
2472 if (!ok) {
2473 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
2474 return;
2477 ok = smbd_lock_socket_internal(sconn);
2478 if (!ok) {
2479 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2480 DEBUG(0, ("%s: failed to lock socket\n", __location__));
2481 return;
2484 if (!fd_is_readable(sconn->sock)) {
2485 DEBUG(10,("echo_handler[%d] the parent smbd was faster\n",
2486 (int)sys_getpid()));
2488 ok = smbd_unlock_socket_internal(sconn);
2489 if (!ok) {
2490 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2491 DEBUG(1, ("%s: failed to unlock socket\n",
2492 __location__));
2493 return;
2496 subreq = wait_for_read_send(state, state->ev, sconn->sock);
2497 if (tevent_req_nomem(subreq, req)) {
2498 return;
2500 tevent_req_set_callback(subreq, smbd_echo_read_readable, req);
2501 return;
2504 status = receive_smb_talloc(state, sconn, sconn->sock, &state->buf,
2505 0 /* timeout */,
2506 &unread,
2507 &encrypted,
2508 &state->buflen,
2509 &state->seqnum,
2510 false /* trusted_channel*/);
2512 if (tevent_req_nterror(req, status)) {
2513 tevent_req_nterror(req, status);
2514 DEBUG(1, ("echo_handler[%d]: receive_smb_raw_talloc failed: %s\n",
2515 (int)sys_getpid(), nt_errstr(status)));
2516 return;
2519 ok = smbd_unlock_socket_internal(sconn);
2520 if (!ok) {
2521 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2522 DEBUG(1, ("%s: failed to unlock socket\n", __location__));
2523 return;
2525 tevent_req_done(req);
2528 static NTSTATUS smbd_echo_read_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
2529 char **pbuf, size_t *pbuflen, uint32_t *pseqnum)
2531 struct smbd_echo_read_state *state = tevent_req_data(
2532 req, struct smbd_echo_read_state);
2533 NTSTATUS status;
2535 if (tevent_req_is_nterror(req, &status)) {
2536 return status;
2538 *pbuf = talloc_move(mem_ctx, &state->buf);
2539 *pbuflen = state->buflen;
2540 *pseqnum = state->seqnum;
2541 return NT_STATUS_OK;
2544 struct smbd_echo_state {
2545 struct tevent_context *ev;
2546 struct iovec *pending;
2547 struct smbd_server_connection *sconn;
2548 int parent_pipe;
2550 struct tevent_fd *parent_fde;
2552 struct tevent_req *write_req;
2555 static void smbd_echo_writer_done(struct tevent_req *req);
2557 static void smbd_echo_activate_writer(struct smbd_echo_state *state)
2559 int num_pending;
2561 if (state->write_req != NULL) {
2562 return;
2565 num_pending = talloc_array_length(state->pending);
2566 if (num_pending == 0) {
2567 return;
2570 state->write_req = writev_send(state, state->ev, NULL,
2571 state->parent_pipe, false,
2572 state->pending, num_pending);
2573 if (state->write_req == NULL) {
2574 DEBUG(1, ("writev_send failed\n"));
2575 exit(1);
2578 talloc_steal(state->write_req, state->pending);
2579 state->pending = NULL;
2581 tevent_req_set_callback(state->write_req, smbd_echo_writer_done,
2582 state);
2585 static void smbd_echo_writer_done(struct tevent_req *req)
2587 struct smbd_echo_state *state = tevent_req_callback_data(
2588 req, struct smbd_echo_state);
2589 ssize_t written;
2590 int err;
2592 written = writev_recv(req, &err);
2593 TALLOC_FREE(req);
2594 state->write_req = NULL;
2595 if (written == -1) {
2596 DEBUG(1, ("writev to parent failed: %s\n", strerror(err)));
2597 exit(1);
2599 DEBUG(10,("echo_handler[%d]: forwarded pdu to main\n", (int)sys_getpid()));
2600 smbd_echo_activate_writer(state);
2603 static bool smbd_echo_reply(struct smbd_echo_state *state,
2604 uint8_t *inbuf, size_t inbuf_len,
2605 uint32_t seqnum)
2607 struct smb_request req;
2608 uint16_t num_replies;
2609 char *outbuf;
2610 bool ok;
2612 if ((inbuf_len == 4) && (CVAL(inbuf, 0) == NBSSkeepalive)) {
2613 DEBUG(10, ("Got netbios keepalive\n"));
2615 * Just swallow it
2617 return true;
2620 if (inbuf_len < smb_size) {
2621 DEBUG(10, ("Got short packet: %d bytes\n", (int)inbuf_len));
2622 return false;
2624 if (!valid_smb_header(state->sconn, inbuf)) {
2625 DEBUG(10, ("Got invalid SMB header\n"));
2626 return false;
2629 if (!init_smb_request(&req, state->sconn, inbuf, 0, false,
2630 seqnum)) {
2631 return false;
2633 req.inbuf = inbuf;
2635 DEBUG(10, ("smbecho handler got cmd %d (%s)\n", (int)req.cmd,
2636 smb_messages[req.cmd].name
2637 ? smb_messages[req.cmd].name : "unknown"));
2639 if (req.cmd != SMBecho) {
2640 return false;
2642 if (req.wct < 1) {
2643 return false;
2646 num_replies = SVAL(req.vwv+0, 0);
2647 if (num_replies != 1) {
2648 /* Not a Windows "Hey, you're still there?" request */
2649 return false;
2652 if (!create_outbuf(talloc_tos(), &req, (const char *)req.inbuf, &outbuf,
2653 1, req.buflen)) {
2654 DEBUG(10, ("create_outbuf failed\n"));
2655 return false;
2657 req.outbuf = (uint8_t *)outbuf;
2659 SSVAL(req.outbuf, smb_vwv0, num_replies);
2661 if (req.buflen > 0) {
2662 memcpy(smb_buf(req.outbuf), req.buf, req.buflen);
2665 ok = srv_send_smb(req.sconn,
2666 (char *)outbuf,
2667 true, seqnum+1,
2668 false, &req.pcd);
2669 TALLOC_FREE(outbuf);
2670 if (!ok) {
2671 exit(1);
2674 return true;
2677 static void smbd_echo_exit(struct tevent_context *ev,
2678 struct tevent_fd *fde, uint16_t flags,
2679 void *private_data)
2681 DEBUG(2, ("smbd_echo_exit: lost connection to parent\n"));
2682 exit(0);
2685 static void smbd_echo_got_packet(struct tevent_req *req);
2687 static void smbd_echo_loop(struct smbd_server_connection *sconn,
2688 int parent_pipe)
2690 struct smbd_echo_state *state;
2691 struct tevent_req *read_req;
2693 state = talloc_zero(sconn, struct smbd_echo_state);
2694 if (state == NULL) {
2695 DEBUG(1, ("talloc failed\n"));
2696 return;
2698 state->sconn = sconn;
2699 state->parent_pipe = parent_pipe;
2700 state->ev = s3_tevent_context_init(state);
2701 if (state->ev == NULL) {
2702 DEBUG(1, ("tevent_context_init failed\n"));
2703 TALLOC_FREE(state);
2704 return;
2706 state->parent_fde = tevent_add_fd(state->ev, state, parent_pipe,
2707 TEVENT_FD_READ, smbd_echo_exit,
2708 state);
2709 if (state->parent_fde == NULL) {
2710 DEBUG(1, ("tevent_add_fd failed\n"));
2711 TALLOC_FREE(state);
2712 return;
2715 read_req = smbd_echo_read_send(state, state->ev, sconn);
2716 if (read_req == NULL) {
2717 DEBUG(1, ("smbd_echo_read_send failed\n"));
2718 TALLOC_FREE(state);
2719 return;
2721 tevent_req_set_callback(read_req, smbd_echo_got_packet, state);
2723 while (true) {
2724 if (tevent_loop_once(state->ev) == -1) {
2725 DEBUG(1, ("tevent_loop_once failed: %s\n",
2726 strerror(errno)));
2727 break;
2730 TALLOC_FREE(state);
2733 static void smbd_echo_got_packet(struct tevent_req *req)
2735 struct smbd_echo_state *state = tevent_req_callback_data(
2736 req, struct smbd_echo_state);
2737 NTSTATUS status;
2738 char *buf = NULL;
2739 size_t buflen = 0;
2740 uint32_t seqnum = 0;
2741 bool reply;
2743 status = smbd_echo_read_recv(req, state, &buf, &buflen, &seqnum);
2744 TALLOC_FREE(req);
2745 if (!NT_STATUS_IS_OK(status)) {
2746 DEBUG(1, ("smbd_echo_read_recv returned %s\n",
2747 nt_errstr(status)));
2748 exit(1);
2751 reply = smbd_echo_reply(state, (uint8_t *)buf, buflen, seqnum);
2752 if (!reply) {
2753 size_t num_pending;
2754 struct iovec *tmp;
2755 struct iovec *iov;
2757 num_pending = talloc_array_length(state->pending);
2758 tmp = talloc_realloc(state, state->pending, struct iovec,
2759 num_pending+1);
2760 if (tmp == NULL) {
2761 DEBUG(1, ("talloc_realloc failed\n"));
2762 exit(1);
2764 state->pending = tmp;
2766 if (buflen >= smb_size) {
2768 * place the seqnum in the packet so that the main process
2769 * can reply with signing
2771 SIVAL(buf, smb_ss_field, seqnum);
2772 SIVAL(buf, smb_ss_field+4, NT_STATUS_V(NT_STATUS_OK));
2775 iov = &state->pending[num_pending];
2776 iov->iov_base = buf;
2777 iov->iov_len = buflen;
2779 DEBUG(10,("echo_handler[%d]: forward to main\n",
2780 (int)sys_getpid()));
2781 smbd_echo_activate_writer(state);
2784 req = smbd_echo_read_send(state, state->ev, state->sconn);
2785 if (req == NULL) {
2786 DEBUG(1, ("smbd_echo_read_send failed\n"));
2787 exit(1);
2789 tevent_req_set_callback(req, smbd_echo_got_packet, state);
2794 * Handle SMBecho requests in a forked child process
2796 bool fork_echo_handler(struct smbd_server_connection *sconn)
2798 int listener_pipe[2];
2799 int res;
2800 pid_t child;
2802 res = pipe(listener_pipe);
2803 if (res == -1) {
2804 DEBUG(1, ("pipe() failed: %s\n", strerror(errno)));
2805 return false;
2807 sconn->smb1.echo_handler.socket_lock_fd = create_unlink_tmp(lp_lockdir());
2808 if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
2809 DEBUG(1, ("Could not create lock fd: %s\n", strerror(errno)));
2810 goto fail;
2813 child = sys_fork();
2814 if (child == 0) {
2815 NTSTATUS status;
2817 close(listener_pipe[0]);
2818 set_blocking(listener_pipe[1], false);
2820 status = reinit_after_fork(sconn->msg_ctx,
2821 sconn->ev_ctx,
2822 false);
2823 if (!NT_STATUS_IS_OK(status)) {
2824 DEBUG(1, ("reinit_after_fork failed: %s\n",
2825 nt_errstr(status)));
2826 exit(1);
2828 smbd_echo_loop(sconn, listener_pipe[1]);
2829 exit(0);
2831 close(listener_pipe[1]);
2832 listener_pipe[1] = -1;
2833 sconn->smb1.echo_handler.trusted_fd = listener_pipe[0];
2835 DEBUG(10,("fork_echo_handler: main[%d] echo_child[%d]\n", (int)sys_getpid(), child));
2838 * Without smb signing this is the same as the normal smbd
2839 * listener. This needs to change once signing comes in.
2841 sconn->smb1.echo_handler.trusted_fde = tevent_add_fd(sconn->ev_ctx,
2842 sconn,
2843 sconn->smb1.echo_handler.trusted_fd,
2844 TEVENT_FD_READ,
2845 smbd_server_echo_handler,
2846 sconn);
2847 if (sconn->smb1.echo_handler.trusted_fde == NULL) {
2848 DEBUG(1, ("event_add_fd failed\n"));
2849 goto fail;
2852 return true;
2854 fail:
2855 if (listener_pipe[0] != -1) {
2856 close(listener_pipe[0]);
2858 if (listener_pipe[1] != -1) {
2859 close(listener_pipe[1]);
2861 sconn->smb1.echo_handler.trusted_fd = -1;
2862 if (sconn->smb1.echo_handler.socket_lock_fd != -1) {
2863 close(sconn->smb1.echo_handler.socket_lock_fd);
2865 sconn->smb1.echo_handler.trusted_fd = -1;
2866 sconn->smb1.echo_handler.socket_lock_fd = -1;
2867 return false;
2870 #if CLUSTER_SUPPORT
2872 static NTSTATUS smbd_register_ips(struct smbd_server_connection *sconn,
2873 struct sockaddr_storage *srv,
2874 struct sockaddr_storage *clnt)
2876 struct ctdbd_connection *cconn;
2877 char tmp_addr[INET6_ADDRSTRLEN];
2878 char *addr;
2880 cconn = messaging_ctdbd_connection();
2881 if (cconn == NULL) {
2882 return NT_STATUS_NO_MEMORY;
2885 client_socket_addr(sconn->sock, tmp_addr, sizeof(tmp_addr));
2886 addr = talloc_strdup(cconn, tmp_addr);
2887 if (addr == NULL) {
2888 return NT_STATUS_NO_MEMORY;
2890 return ctdbd_register_ips(cconn, srv, clnt, release_ip, addr);
2893 #endif
2895 static bool uid_in_use(const struct user_struct *user, uid_t uid)
2897 while (user) {
2898 if (user->session_info &&
2899 (user->session_info->unix_token->uid == uid)) {
2900 return true;
2902 user = user->next;
2904 return false;
2907 static bool gid_in_use(const struct user_struct *user, gid_t gid)
2909 while (user) {
2910 if (user->session_info != NULL) {
2911 int i;
2912 struct security_unix_token *utok;
2914 utok = user->session_info->unix_token;
2915 if (utok->gid == gid) {
2916 return true;
2918 for(i=0; i<utok->ngroups; i++) {
2919 if (utok->groups[i] == gid) {
2920 return true;
2924 user = user->next;
2926 return false;
2929 static bool sid_in_use(const struct user_struct *user,
2930 const struct dom_sid *psid)
2932 while (user) {
2933 struct security_token *tok;
2935 if (user->session_info == NULL) {
2936 continue;
2938 tok = user->session_info->security_token;
2939 if (tok == NULL) {
2941 * Not sure session_info->security_token can
2942 * ever be NULL. This check might be not
2943 * necessary.
2945 continue;
2947 if (security_token_has_sid(tok, psid)) {
2948 return true;
2950 user = user->next;
2952 return false;
2955 static bool id_in_use(const struct user_struct *user,
2956 const struct id_cache_ref *id)
2958 switch(id->type) {
2959 case UID:
2960 return uid_in_use(user, id->id.uid);
2961 case GID:
2962 return gid_in_use(user, id->id.gid);
2963 case SID:
2964 return sid_in_use(user, &id->id.sid);
2965 default:
2966 break;
2968 return false;
2971 static void smbd_id_cache_kill(struct messaging_context *msg_ctx,
2972 void *private_data,
2973 uint32_t msg_type,
2974 struct server_id server_id,
2975 DATA_BLOB* data)
2977 const char *msg = (data && data->data)
2978 ? (const char *)data->data : "<NULL>";
2979 struct user_struct *validated_users;
2980 struct id_cache_ref id;
2981 struct smbd_server_connection *sconn =
2982 talloc_get_type_abort(private_data,
2983 struct smbd_server_connection);
2985 validated_users = sconn->smb1.sessions.validated_users;
2987 if (!id_cache_ref_parse(msg, &id)) {
2988 DEBUG(0, ("Invalid ?ID: %s\n", msg));
2989 return;
2992 if (id_in_use(validated_users, &id)) {
2993 exit_server_cleanly(msg);
2995 id_cache_delete_from_cache(&id);
2998 /****************************************************************************
2999 Process commands from the client
3000 ****************************************************************************/
3002 void smbd_process(struct tevent_context *ev_ctx,
3003 struct smbd_server_connection *sconn)
3005 TALLOC_CTX *frame = talloc_stackframe();
3006 struct sockaddr_storage ss;
3007 struct sockaddr *sa = NULL;
3008 socklen_t sa_socklen;
3009 struct tsocket_address *local_address = NULL;
3010 struct tsocket_address *remote_address = NULL;
3011 const char *locaddr = NULL;
3012 const char *remaddr = NULL;
3013 char *rhost;
3014 int ret;
3016 if (lp_maxprotocol() >= PROTOCOL_SMB2_02) {
3018 * We're not making the decision here,
3019 * we're just allowing the client
3020 * to decide between SMB1 and SMB2
3021 * with the first negprot
3022 * packet.
3024 sconn->using_smb2 = true;
3027 /* Ensure child is set to blocking mode */
3028 set_blocking(sconn->sock,True);
3030 set_socket_options(sconn->sock, "SO_KEEPALIVE");
3031 set_socket_options(sconn->sock, lp_socket_options());
3033 sa = (struct sockaddr *)(void *)&ss;
3034 sa_socklen = sizeof(ss);
3035 ret = getpeername(sconn->sock, sa, &sa_socklen);
3036 if (ret != 0) {
3037 int level = (errno == ENOTCONN)?2:0;
3038 DEBUG(level,("getpeername() failed - %s\n", strerror(errno)));
3039 exit_server_cleanly("getpeername() failed.\n");
3041 ret = tsocket_address_bsd_from_sockaddr(sconn,
3042 sa, sa_socklen,
3043 &remote_address);
3044 if (ret != 0) {
3045 DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
3046 __location__, strerror(errno)));
3047 exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
3050 sa = (struct sockaddr *)(void *)&ss;
3051 sa_socklen = sizeof(ss);
3052 ret = getsockname(sconn->sock, sa, &sa_socklen);
3053 if (ret != 0) {
3054 int level = (errno == ENOTCONN)?2:0;
3055 DEBUG(level,("getsockname() failed - %s\n", strerror(errno)));
3056 exit_server_cleanly("getsockname() failed.\n");
3058 ret = tsocket_address_bsd_from_sockaddr(sconn,
3059 sa, sa_socklen,
3060 &local_address);
3061 if (ret != 0) {
3062 DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
3063 __location__, strerror(errno)));
3064 exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
3067 sconn->local_address = local_address;
3068 sconn->remote_address = remote_address;
3070 if (tsocket_address_is_inet(local_address, "ip")) {
3071 locaddr = tsocket_address_inet_addr_string(
3072 sconn->local_address,
3073 talloc_tos());
3074 if (locaddr == NULL) {
3075 DEBUG(0,("%s: tsocket_address_inet_addr_string local failed - %s\n",
3076 __location__, strerror(errno)));
3077 exit_server_cleanly("tsocket_address_inet_addr_string local failed.\n");
3079 } else {
3080 locaddr = "0.0.0.0";
3083 if (tsocket_address_is_inet(remote_address, "ip")) {
3084 remaddr = tsocket_address_inet_addr_string(
3085 sconn->remote_address,
3086 talloc_tos());
3087 if (remaddr == NULL) {
3088 DEBUG(0,("%s: tsocket_address_inet_addr_string remote failed - %s\n",
3089 __location__, strerror(errno)));
3090 exit_server_cleanly("tsocket_address_inet_addr_string remote failed.\n");
3092 } else {
3093 remaddr = "0.0.0.0";
3096 /* this is needed so that we get decent entries
3097 in smbstatus for port 445 connects */
3098 set_remote_machine_name(remaddr, false);
3099 reload_services(sconn, conn_snum_used, true);
3102 * Before the first packet, check the global hosts allow/ hosts deny
3103 * parameters before doing any parsing of packets passed to us by the
3104 * client. This prevents attacks on our parsing code from hosts not in
3105 * the hosts allow list.
3108 ret = get_remote_hostname(remote_address,
3109 &rhost,
3110 talloc_tos());
3111 if (ret < 0) {
3112 DEBUG(0,("%s: get_remote_hostname failed - %s\n",
3113 __location__, strerror(errno)));
3114 exit_server_cleanly("get_remote_hostname failed.\n");
3116 if (strequal(rhost, "UNKNOWN")) {
3117 rhost = talloc_strdup(talloc_tos(), remaddr);
3119 sconn->remote_hostname = talloc_move(sconn, &rhost);
3121 sub_set_socket_ids(remaddr,
3122 sconn->remote_hostname,
3123 locaddr);
3125 if (!allow_access(lp_hostsdeny(-1), lp_hostsallow(-1),
3126 sconn->remote_hostname,
3127 remaddr)) {
3129 * send a negative session response "not listening on calling
3130 * name"
3132 unsigned char buf[5] = {0x83, 0, 0, 1, 0x81};
3133 DEBUG( 1, ("Connection denied from %s to %s\n",
3134 tsocket_address_string(remote_address, talloc_tos()),
3135 tsocket_address_string(local_address, talloc_tos())));
3136 (void)srv_send_smb(sconn,(char *)buf, false,
3137 0, false, NULL);
3138 exit_server_cleanly("connection denied");
3141 DEBUG(10, ("Connection allowed from %s to %s\n",
3142 tsocket_address_string(remote_address, talloc_tos()),
3143 tsocket_address_string(local_address, talloc_tos())));
3145 init_modules();
3147 smb_perfcount_init();
3149 if (!init_account_policy()) {
3150 exit_server("Could not open account policy tdb.\n");
3153 if (*lp_rootdir()) {
3154 if (chroot(lp_rootdir()) != 0) {
3155 DEBUG(0,("Failed to change root to %s\n", lp_rootdir()));
3156 exit_server("Failed to chroot()");
3158 if (chdir("/") == -1) {
3159 DEBUG(0,("Failed to chdir to / on chroot to %s\n", lp_rootdir()));
3160 exit_server("Failed to chroot()");
3162 DEBUG(0,("Changed root to %s\n", lp_rootdir()));
3165 if (!srv_init_signing(sconn)) {
3166 exit_server("Failed to init smb_signing");
3169 /* Setup oplocks */
3170 if (!init_oplocks(sconn))
3171 exit_server("Failed to init oplocks");
3173 /* register our message handlers */
3174 messaging_register(sconn->msg_ctx, sconn,
3175 MSG_SMB_FORCE_TDIS, msg_force_tdis);
3176 messaging_register(sconn->msg_ctx, sconn,
3177 MSG_SMB_CLOSE_FILE, msg_close_file);
3178 messaging_register(sconn->msg_ctx, sconn,
3179 MSG_SMB_FILE_RENAME, msg_file_was_renamed);
3181 id_cache_register_msgs(sconn->msg_ctx);
3182 messaging_deregister(sconn->msg_ctx, ID_CACHE_KILL, NULL);
3183 messaging_register(sconn->msg_ctx, sconn,
3184 ID_CACHE_KILL, smbd_id_cache_kill);
3186 messaging_deregister(sconn->msg_ctx,
3187 MSG_SMB_CONF_UPDATED, sconn->ev_ctx);
3188 messaging_register(sconn->msg_ctx, sconn,
3189 MSG_SMB_CONF_UPDATED, smbd_conf_updated);
3192 * Use the default MSG_DEBUG handler to avoid rebroadcasting
3193 * MSGs to all child processes
3195 messaging_deregister(sconn->msg_ctx,
3196 MSG_DEBUG, NULL);
3197 messaging_register(sconn->msg_ctx, NULL,
3198 MSG_DEBUG, debug_message);
3200 if ((lp_keepalive() != 0)
3201 && !(event_add_idle(ev_ctx, NULL,
3202 timeval_set(lp_keepalive(), 0),
3203 "keepalive", keepalive_fn,
3204 sconn))) {
3205 DEBUG(0, ("Could not add keepalive event\n"));
3206 exit(1);
3209 if (!(event_add_idle(ev_ctx, NULL,
3210 timeval_set(IDLE_CLOSED_TIMEOUT, 0),
3211 "deadtime", deadtime_fn, sconn))) {
3212 DEBUG(0, ("Could not add deadtime event\n"));
3213 exit(1);
3216 if (!(event_add_idle(ev_ctx, NULL,
3217 timeval_set(SMBD_HOUSEKEEPING_INTERVAL, 0),
3218 "housekeeping", housekeeping_fn, sconn))) {
3219 DEBUG(0, ("Could not add housekeeping event\n"));
3220 exit(1);
3223 #ifdef CLUSTER_SUPPORT
3225 if (lp_clustering()) {
3227 * We need to tell ctdb about our client's TCP
3228 * connection, so that for failover ctdbd can send
3229 * tickle acks, triggering a reconnection by the
3230 * client.
3233 struct sockaddr_storage srv, clnt;
3235 if (client_get_tcp_info(sconn->sock, &srv, &clnt) == 0) {
3236 NTSTATUS status;
3237 status = smbd_register_ips(sconn, &srv, &clnt);
3238 if (!NT_STATUS_IS_OK(status)) {
3239 DEBUG(0, ("ctdbd_register_ips failed: %s\n",
3240 nt_errstr(status)));
3242 } else
3244 DEBUG(0,("Unable to get tcp info for "
3245 "CTDB_CONTROL_TCP_CLIENT: %s\n",
3246 strerror(errno)));
3250 #endif
3252 sconn->nbt.got_session = false;
3254 sconn->smb1.negprot.max_recv = MIN(lp_maxxmit(),BUFFER_SIZE);
3256 sconn->smb1.sessions.done_sesssetup = false;
3257 sconn->smb1.sessions.max_send = BUFFER_SIZE;
3258 sconn->smb1.sessions.last_session_tag = UID_FIELD_INVALID;
3259 /* users from session setup */
3260 sconn->smb1.sessions.session_userlist = NULL;
3261 /* workgroup from session setup. */
3262 sconn->smb1.sessions.session_workgroup = NULL;
3263 /* this holds info on user ids that are already validated for this VC */
3264 sconn->smb1.sessions.validated_users = NULL;
3265 sconn->smb1.sessions.next_vuid = VUID_OFFSET;
3266 sconn->smb1.sessions.num_validated_vuids = 0;
3268 conn_init(sconn);
3269 if (!init_dptrs(sconn)) {
3270 exit_server("init_dptrs() failed");
3273 sconn->smb1.fde = event_add_fd(ev_ctx,
3274 sconn,
3275 sconn->sock,
3276 EVENT_FD_READ,
3277 smbd_server_connection_handler,
3278 sconn);
3279 if (!sconn->smb1.fde) {
3280 exit_server("failed to create smbd_server_connection fde");
3283 TALLOC_FREE(frame);
3285 while (True) {
3286 frame = talloc_stackframe_pool(8192);
3288 errno = 0;
3289 if (tevent_loop_once(ev_ctx) == -1) {
3290 if (errno != EINTR) {
3291 DEBUG(3, ("tevent_loop_once failed: %s,"
3292 " exiting\n", strerror(errno) ));
3293 break;
3297 TALLOC_FREE(frame);
3300 exit_server_cleanly(NULL);
3303 bool req_is_in_chain(struct smb_request *req)
3305 if (req->vwv != (const uint16_t *)(req->inbuf+smb_vwv)) {
3307 * We're right now handling a subsequent request, so we must
3308 * be in a chain
3310 return true;
3313 if (!is_andx_req(req->cmd)) {
3314 return false;
3317 if (req->wct < 2) {
3319 * Okay, an illegal request, but definitely not chained :-)
3321 return false;
3324 return (CVAL(req->vwv+0, 0) != 0xFF);