?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s3:smbd: move sconn->sock to xconn->transport.sock
[Samba.git] / source3 / smbd / process.c
blob41547bbe15957110a4b347c5118a516975bf0c4c
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
6
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.
11
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.
16
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/>.
19 */
20
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"
42 #include "serverid.h"
43 #include "system/threads.h"
44
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 tevent_timer *te;
51 struct smb_perfcount_data pcd;
52 uint32_t seqnum;
53 bool encrypted;
54 bool processed;
55 DATA_BLOB buf;
56 struct deferred_open_record *open_rec;
57 };
58
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);
63 static bool smb_splice_chain(uint8_t **poutbuf, const uint8_t *andx_buf);
64
65 void smbd_echo_init(struct smbd_server_connection *sconn)
66 {
67 sconn->smb1.echo_handler.trusted_fd = -1;
68 sconn->smb1.echo_handler.socket_lock_fd = -1;
69 #ifdef HAVE_ROBUST_MUTEXES
70 sconn->smb1.echo_handler.socket_mutex = NULL;
71 #endif
72 }
73
74 static bool smbd_echo_active(struct smbd_server_connection *sconn)
75 {
76 if (sconn->smb1.echo_handler.socket_lock_fd != -1) {
77 return true;
78 }
79
80 #ifdef HAVE_ROBUST_MUTEXES
81 if (sconn->smb1.echo_handler.socket_mutex != NULL) {
82 return true;
83 }
84 #endif
85
86 return false;
87 }
88
89 static bool smbd_lock_socket_internal(struct smbd_server_connection *sconn)
90 {
91 if (!smbd_echo_active(sconn)) {
92 return true;
93 }
94
95 sconn->smb1.echo_handler.ref_count++;
96
97 if (sconn->smb1.echo_handler.ref_count > 1) {
98 return true;
99 }
100
101 DEBUG(10,("pid[%d] wait for socket lock\n", (int)getpid()));
102
103 #ifdef HAVE_ROBUST_MUTEXES
104 if (sconn->smb1.echo_handler.socket_mutex != NULL) {
105 int ret = EINTR;
106
107 while (ret == EINTR) {
108 ret = pthread_mutex_lock(
109 sconn->smb1.echo_handler.socket_mutex);
110 if (ret == 0) {
111 break;
112 }
113 }
114 if (ret != 0) {
115 DEBUG(1, ("pthread_mutex_lock failed: %s\n",
116 strerror(ret)));
117 return false;
118 }
119 }
120 #endif
121
122 if (sconn->smb1.echo_handler.socket_lock_fd != -1) {
123 bool ok;
124
125 do {
126 ok = fcntl_lock(
127 sconn->smb1.echo_handler.socket_lock_fd,
128 F_SETLKW, 0, 0, F_WRLCK);
129 } while (!ok && (errno == EINTR));
130
131 if (!ok) {
132 DEBUG(1, ("fcntl_lock failed: %s\n", strerror(errno)));
133 return false;
134 }
135 }
136
137 DEBUG(10,("pid[%d] got socket lock\n", (int)getpid()));
138
139 return true;
140 }
141
142 void smbd_lock_socket(struct smbd_server_connection *sconn)
143 {
144 if (!smbd_lock_socket_internal(sconn)) {
145 exit_server_cleanly("failed to lock socket");
146 }
147 }
148
149 static bool smbd_unlock_socket_internal(struct smbd_server_connection *sconn)
150 {
151 if (!smbd_echo_active(sconn)) {
152 return true;
153 }
154
155 sconn->smb1.echo_handler.ref_count--;
156
157 if (sconn->smb1.echo_handler.ref_count > 0) {
158 return true;
159 }
160
161 #ifdef HAVE_ROBUST_MUTEXES
162 if (sconn->smb1.echo_handler.socket_mutex != NULL) {
163 int ret = EINTR;
164
165 while (ret == EINTR) {
166 ret = pthread_mutex_unlock(
167 sconn->smb1.echo_handler.socket_mutex);
168 if (ret == 0) {
169 break;
170 }
171 }
172 if (ret != 0) {
173 DEBUG(1, ("pthread_mutex_unlock failed: %s\n",
174 strerror(ret)));
175 return false;
176 }
177 }
178 #endif
179
180 if (sconn->smb1.echo_handler.socket_lock_fd != -1) {
181 bool ok;
182
183 do {
184 ok = fcntl_lock(
185 sconn->smb1.echo_handler.socket_lock_fd,
186 F_SETLKW, 0, 0, F_UNLCK);
187 } while (!ok && (errno == EINTR));
188
189 if (!ok) {
190 DEBUG(1, ("fcntl_lock failed: %s\n", strerror(errno)));
191 return false;
192 }
193 }
194
195 DEBUG(10,("pid[%d] unlocked socket\n", (int)getpid()));
196
197 return true;
198 }
199
200 void smbd_unlock_socket(struct smbd_server_connection *sconn)
201 {
202 if (!smbd_unlock_socket_internal(sconn)) {
203 exit_server_cleanly("failed to unlock socket");
204 }
205 }
206
207 /* Accessor function for smb_read_error for smbd functions. */
208
209 /****************************************************************************
210 Send an smb to a fd.
211 ****************************************************************************/
212
213 bool srv_send_smb(struct smbd_server_connection *sconn, char *buffer,
214 bool do_signing, uint32_t seqnum,
215 bool do_encrypt,
216 struct smb_perfcount_data *pcd)
217 {
218 struct smbXsrv_connection *xconn = sconn->conn;
219 size_t len = 0;
220 ssize_t ret;
221 char *buf_out = buffer;
222
223 if (!NT_STATUS_IS_OK(sconn->status)) {
224 /*
225 * we're not supposed to do any io
226 */
227 return true;
228 }
229
230 smbd_lock_socket(sconn);
231
232 if (do_signing) {
233 /* Sign the outgoing packet if required. */
234 srv_calculate_sign_mac(sconn, buf_out, seqnum);
235 }
236
237 if (do_encrypt) {
238 NTSTATUS status = srv_encrypt_buffer(sconn, buffer, &buf_out);
239 if (!NT_STATUS_IS_OK(status)) {
240 DEBUG(0, ("send_smb: SMB encryption failed "
241 "on outgoing packet! Error %s\n",
242 nt_errstr(status) ));
243 ret = -1;
244 goto out;
245 }
246 }
247
248 len = smb_len_large(buf_out) + 4;
249
250 ret = write_data(xconn->transport.sock, buf_out, len);
251 if (ret <= 0) {
252 int saved_errno = errno;
253 /*
254 * Try and give an error message saying what
255 * client failed.
256 */
257 DEBUG(1,("pid[%d] Error writing %d bytes to client %s. %d. (%s)\n",
258 (int)getpid(), (int)len,
259 smbXsrv_connection_dbg(xconn),
260 (int)ret, strerror(saved_errno)));
261 errno = saved_errno;
262
263 srv_free_enc_buffer(sconn, buf_out);
264 goto out;
265 }
266
267 SMB_PERFCOUNT_SET_MSGLEN_OUT(pcd, len);
268 srv_free_enc_buffer(sconn, buf_out);
269 out:
270 SMB_PERFCOUNT_END(pcd);
271
272 smbd_unlock_socket(sconn);
273 return (ret > 0);
274 }
275
276 /*******************************************************************
277 Setup the word count and byte count for a smb message.
278 ********************************************************************/
279
280 int srv_set_message(char *buf,
281 int num_words,
282 int num_bytes,
283 bool zero)
284 {
285 if (zero && (num_words || num_bytes)) {
286 memset(buf + smb_size,'\0',num_words*2 + num_bytes);
287 }
288 SCVAL(buf,smb_wct,num_words);
289 SSVAL(buf,smb_vwv + num_words*SIZEOFWORD,num_bytes);
290 smb_setlen(buf,(smb_size + num_words*2 + num_bytes - 4));
291 return (smb_size + num_words*2 + num_bytes);
292 }
293
294 static bool valid_smb_header(struct smbd_server_connection *sconn,
295 const uint8_t *inbuf)
296 {
297 if (is_encrypted_packet(sconn, inbuf)) {
298 return true;
299 }
300 /*
301 * This used to be (strncmp(smb_base(inbuf),"\377SMB",4) == 0)
302 * but it just looks weird to call strncmp for this one.
303 */
304 return (IVAL(smb_base(inbuf), 0) == 0x424D53FF);
305 }
306
307 /* Socket functions for smbd packet processing. */
308
309 static bool valid_packet_size(size_t len)
310 {
311 /*
312 * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
313 * of header. Don't print the error if this fits.... JRA.
314 */
315
316 if (len > (LARGE_WRITEX_BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
317 DEBUG(0,("Invalid packet length! (%lu bytes).\n",
318 (unsigned long)len));
319 return false;
320 }
321 return true;
322 }
323
324 static NTSTATUS read_packet_remainder(int fd, char *buffer,
325 unsigned int timeout, ssize_t len)
326 {
327 NTSTATUS status;
328
329 if (len <= 0) {
330 return NT_STATUS_OK;
331 }
332
333 status = read_fd_with_timeout(fd, buffer, len, len, timeout, NULL);
334 if (!NT_STATUS_IS_OK(status)) {
335 char addr[INET6_ADDRSTRLEN];
336 DEBUG(0, ("read_fd_with_timeout failed for client %s read "
337 "error = %s.\n",
338 get_peer_addr(fd, addr, sizeof(addr)),
339 nt_errstr(status)));
340 }
341 return status;
342 }
343
344 /****************************************************************************
345 Attempt a zerocopy writeX read. We know here that len > smb_size-4
346 ****************************************************************************/
347
348 /*
349 * Unfortunately, earlier versions of smbclient/libsmbclient
350 * don't send this "standard" writeX header. I've fixed this
351 * for 3.2 but we'll use the old method with earlier versions.
352 * Windows and CIFSFS at least use this standard size. Not
353 * sure about MacOSX.
354 */
355
356 #define STANDARD_WRITE_AND_X_HEADER_SIZE (smb_size - 4 + /* basic header */ \
357 (2*14) + /* word count (including bcc) */ \
358 1 /* pad byte */)
359
360 static NTSTATUS receive_smb_raw_talloc_partial_read(TALLOC_CTX *mem_ctx,
361 const char lenbuf[4],
362 struct smbd_server_connection *sconn,
363 int sock,
364 char **buffer,
365 unsigned int timeout,
366 size_t *p_unread,
367 size_t *len_ret)
368 {
369 /* Size of a WRITEX call (+4 byte len). */
370 char writeX_header[4 + STANDARD_WRITE_AND_X_HEADER_SIZE];
371 ssize_t len = smb_len_large(lenbuf); /* Could be a UNIX large writeX. */
372 ssize_t toread;
373 NTSTATUS status;
374
375 memcpy(writeX_header, lenbuf, 4);
376
377 status = read_fd_with_timeout(
378 sock, writeX_header + 4,
379 STANDARD_WRITE_AND_X_HEADER_SIZE,
380 STANDARD_WRITE_AND_X_HEADER_SIZE,
381 timeout, NULL);
382
383 if (!NT_STATUS_IS_OK(status)) {
384 DEBUG(0, ("read_fd_with_timeout failed for client %s read "
385 "error = %s.\n",
386 tsocket_address_string(sconn->remote_address,
387 talloc_tos()),
388 nt_errstr(status)));
389 return status;
390 }
391
392 /*
393 * Ok - now try and see if this is a possible
394 * valid writeX call.
395 */
396
397 if (is_valid_writeX_buffer(sconn, (uint8_t *)writeX_header)) {
398 /*
399 * If the data offset is beyond what
400 * we've read, drain the extra bytes.
401 */
402 uint16_t doff = SVAL(writeX_header,smb_vwv11);
403 ssize_t newlen;
404
405 if (doff > STANDARD_WRITE_AND_X_HEADER_SIZE) {
406 size_t drain = doff - STANDARD_WRITE_AND_X_HEADER_SIZE;
407 if (drain_socket(sock, drain) != drain) {
408 smb_panic("receive_smb_raw_talloc_partial_read:"
409 " failed to drain pending bytes");
410 }
411 } else {
412 doff = STANDARD_WRITE_AND_X_HEADER_SIZE;
413 }
414
415 /* Spoof down the length and null out the bcc. */
416 set_message_bcc(writeX_header, 0);
417 newlen = smb_len(writeX_header);
418
419 /* Copy the header we've written. */
420
421 *buffer = (char *)talloc_memdup(mem_ctx,
422 writeX_header,
423 sizeof(writeX_header));
424
425 if (*buffer == NULL) {
426 DEBUG(0, ("Could not allocate inbuf of length %d\n",
427 (int)sizeof(writeX_header)));
428 return NT_STATUS_NO_MEMORY;
429 }
430
431 /* Work out the remaining bytes. */
432 *p_unread = len - STANDARD_WRITE_AND_X_HEADER_SIZE;
433 *len_ret = newlen + 4;
434 return NT_STATUS_OK;
435 }
436
437 if (!valid_packet_size(len)) {
438 return NT_STATUS_INVALID_PARAMETER;
439 }
440
441 /*
442 * Not a valid writeX call. Just do the standard
443 * talloc and return.
444 */
445
446 *buffer = talloc_array(mem_ctx, char, len+4);
447
448 if (*buffer == NULL) {
449 DEBUG(0, ("Could not allocate inbuf of length %d\n",
450 (int)len+4));
451 return NT_STATUS_NO_MEMORY;
452 }
453
454 /* Copy in what we already read. */
455 memcpy(*buffer,
456 writeX_header,
457 4 + STANDARD_WRITE_AND_X_HEADER_SIZE);
458 toread = len - STANDARD_WRITE_AND_X_HEADER_SIZE;
459
460 if(toread > 0) {
461 status = read_packet_remainder(
462 sock,
463 (*buffer) + 4 + STANDARD_WRITE_AND_X_HEADER_SIZE,
464 timeout, toread);
465
466 if (!NT_STATUS_IS_OK(status)) {
467 DEBUG(10, ("receive_smb_raw_talloc_partial_read: %s\n",
468 nt_errstr(status)));
469 return status;
470 }
471 }
472
473 *len_ret = len + 4;
474 return NT_STATUS_OK;
475 }
476
477 static NTSTATUS receive_smb_raw_talloc(TALLOC_CTX *mem_ctx,
478 struct smbd_server_connection *sconn,
479 int sock,
480 char **buffer, unsigned int timeout,
481 size_t *p_unread, size_t *plen)
482 {
483 char lenbuf[4];
484 size_t len;
485 int min_recv_size = lp_min_receive_file_size();
486 NTSTATUS status;
487
488 *p_unread = 0;
489
490 status = read_smb_length_return_keepalive(sock, lenbuf, timeout,
491 &len);
492 if (!NT_STATUS_IS_OK(status)) {
493 return status;
494 }
495
496 if (CVAL(lenbuf,0) == 0 && min_recv_size &&
497 (smb_len_large(lenbuf) > /* Could be a UNIX large writeX. */
498 (min_recv_size + STANDARD_WRITE_AND_X_HEADER_SIZE)) &&
499 !srv_is_signing_active(sconn) &&
500 sconn->smb1.echo_handler.trusted_fde == NULL) {
501
502 return receive_smb_raw_talloc_partial_read(
503 mem_ctx, lenbuf, sconn, sock, buffer, timeout,
504 p_unread, plen);
505 }
506
507 if (!valid_packet_size(len)) {
508 return NT_STATUS_INVALID_PARAMETER;
509 }
510
511 /*
512 * The +4 here can't wrap, we've checked the length above already.
513 */
514
515 *buffer = talloc_array(mem_ctx, char, len+4);
516
517 if (*buffer == NULL) {
518 DEBUG(0, ("Could not allocate inbuf of length %d\n",
519 (int)len+4));
520 return NT_STATUS_NO_MEMORY;
521 }
522
523 memcpy(*buffer, lenbuf, sizeof(lenbuf));
524
525 status = read_packet_remainder(sock, (*buffer)+4, timeout, len);
526 if (!NT_STATUS_IS_OK(status)) {
527 return status;
528 }
529
530 *plen = len + 4;
531 return NT_STATUS_OK;
532 }
533
534 static NTSTATUS receive_smb_talloc(TALLOC_CTX *mem_ctx,
535 struct smbd_server_connection *sconn,
536 int sock,
537 char **buffer, unsigned int timeout,
538 size_t *p_unread, bool *p_encrypted,
539 size_t *p_len,
540 uint32_t *seqnum,
541 bool trusted_channel)
542 {
543 size_t len = 0;
544 NTSTATUS status;
545
546 *p_encrypted = false;
547
548 status = receive_smb_raw_talloc(mem_ctx, sconn, sock, buffer, timeout,
549 p_unread, &len);
550 if (!NT_STATUS_IS_OK(status)) {
551 DEBUG(NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)?5:1,
552 ("receive_smb_raw_talloc failed for client %s "
553 "read error = %s.\n",
554 tsocket_address_string(sconn->remote_address,
555 talloc_tos()),
556 nt_errstr(status)) );
557 return status;
558 }
559
560 if (is_encrypted_packet(sconn, (uint8_t *)*buffer)) {
561 status = srv_decrypt_buffer(sconn, *buffer);
562 if (!NT_STATUS_IS_OK(status)) {
563 DEBUG(0, ("receive_smb_talloc: SMB decryption failed on "
564 "incoming packet! Error %s\n",
565 nt_errstr(status) ));
566 return status;
567 }
568 *p_encrypted = true;
569 }
570
571 /* Check the incoming SMB signature. */
572 if (!srv_check_sign_mac(sconn, *buffer, seqnum, trusted_channel)) {
573 DEBUG(0, ("receive_smb: SMB Signature verification failed on "
574 "incoming packet!\n"));
575 return NT_STATUS_INVALID_NETWORK_RESPONSE;
576 }
577
578 *p_len = len;
579 return NT_STATUS_OK;
580 }
581
582 /*
583 * Initialize a struct smb_request from an inbuf
584 */
585
586 static bool init_smb_request(struct smb_request *req,
587 struct smbd_server_connection *sconn,
588 const uint8 *inbuf,
589 size_t unread_bytes, bool encrypted,
590 uint32_t seqnum)
591 {
592 struct smbXsrv_tcon *tcon;
593 NTSTATUS status;
594 NTTIME now;
595 size_t req_size = smb_len(inbuf) + 4;
596
597 /* Ensure we have at least smb_size bytes. */
598 if (req_size < smb_size) {
599 DEBUG(0,("init_smb_request: invalid request size %u\n",
600 (unsigned int)req_size ));
601 return false;
602 }
603
604 req->request_time = timeval_current();
605 now = timeval_to_nttime(&req->request_time);
606
607 req->cmd = CVAL(inbuf, smb_com);
608 req->flags2 = SVAL(inbuf, smb_flg2);
609 req->smbpid = SVAL(inbuf, smb_pid);
610 req->mid = (uint64_t)SVAL(inbuf, smb_mid);
611 req->seqnum = seqnum;
612 req->vuid = SVAL(inbuf, smb_uid);
613 req->tid = SVAL(inbuf, smb_tid);
614 req->wct = CVAL(inbuf, smb_wct);
615 req->vwv = (const uint16_t *)(inbuf+smb_vwv);
616 req->buflen = smb_buflen(inbuf);
617 req->buf = (const uint8_t *)smb_buf_const(inbuf);
618 req->unread_bytes = unread_bytes;
619 req->encrypted = encrypted;
620 req->sconn = sconn;
621 status = smb1srv_tcon_lookup(sconn->conn, req->tid, now, &tcon);
622 if (NT_STATUS_IS_OK(status)) {
623 req->conn = tcon->compat;
624 } else {
625 req->conn = NULL;
626 }
627 req->chain_fsp = NULL;
628 req->smb2req = NULL;
629 req->priv_paths = NULL;
630 req->chain = NULL;
631 smb_init_perfcount_data(&req->pcd);
632
633 /* Ensure we have at least wct words and 2 bytes of bcc. */
634 if (smb_size + req->wct*2 > req_size) {
635 DEBUG(0,("init_smb_request: invalid wct number %u (size %u)\n",
636 (unsigned int)req->wct,
637 (unsigned int)req_size));
638 return false;
639 }
640 /* Ensure bcc is correct. */
641 if (((const uint8_t *)smb_buf_const(inbuf)) + req->buflen > inbuf + req_size) {
642 DEBUG(0,("init_smb_request: invalid bcc number %u "
643 "(wct = %u, size %u)\n",
644 (unsigned int)req->buflen,
645 (unsigned int)req->wct,
646 (unsigned int)req_size));
647 return false;
648 }
649
650 req->outbuf = NULL;
651 return true;
652 }
653
654 static void process_smb(struct smbd_server_connection *conn,
655 uint8_t *inbuf, size_t nread, size_t unread_bytes,
656 uint32_t seqnum, bool encrypted,
657 struct smb_perfcount_data *deferred_pcd);
658
659 static void smbd_deferred_open_timer(struct tevent_context *ev,
660 struct tevent_timer *te,
661 struct timeval _tval,
662 void *private_data)
663 {
664 struct pending_message_list *msg = talloc_get_type(private_data,
665 struct pending_message_list);
666 struct smbd_server_connection *sconn = msg->sconn;
667 TALLOC_CTX *mem_ctx = talloc_tos();
668 uint64_t mid = (uint64_t)SVAL(msg->buf.data,smb_mid);
669 uint8_t *inbuf;
670
671 inbuf = (uint8_t *)talloc_memdup(mem_ctx, msg->buf.data,
672 msg->buf.length);
673 if (inbuf == NULL) {
674 exit_server("smbd_deferred_open_timer: talloc failed\n");
675 return;
676 }
677
678 /* We leave this message on the queue so the open code can
679 know this is a retry. */
680 DEBUG(5,("smbd_deferred_open_timer: trigger mid %llu.\n",
681 (unsigned long long)mid ));
682
683 /* Mark the message as processed so this is not
684 * re-processed in error. */
685 msg->processed = true;
686
687 process_smb(sconn, inbuf,
688 msg->buf.length, 0,
689 msg->seqnum, msg->encrypted, &msg->pcd);
690
691 /* If it's still there and was processed, remove it. */
692 msg = get_deferred_open_message_smb(sconn, mid);
693 if (msg && msg->processed) {
694 remove_deferred_open_message_smb(sconn, mid);
695 }
696 }
697
698 /****************************************************************************
699 Function to push a message onto the tail of a linked list of smb messages ready
700 for processing.
701 ****************************************************************************/
702
703 static bool push_queued_message(struct smb_request *req,
704 struct timeval request_time,
705 struct timeval end_time,
706 struct deferred_open_record *open_rec)
707 {
708 int msg_len = smb_len(req->inbuf) + 4;
709 struct pending_message_list *msg;
710
711 msg = talloc_zero(NULL, struct pending_message_list);
712
713 if(msg == NULL) {
714 DEBUG(0,("push_message: malloc fail (1)\n"));
715 return False;
716 }
717 msg->sconn = req->sconn;
718
719 msg->buf = data_blob_talloc(msg, req->inbuf, msg_len);
720 if(msg->buf.data == NULL) {
721 DEBUG(0,("push_message: malloc fail (2)\n"));
722 TALLOC_FREE(msg);
723 return False;
724 }
725
726 msg->request_time = request_time;
727 msg->seqnum = req->seqnum;
728 msg->encrypted = req->encrypted;
729 msg->processed = false;
730 SMB_PERFCOUNT_DEFER_OP(&req->pcd, &msg->pcd);
731
732 if (open_rec) {
733 msg->open_rec = talloc_move(msg, &open_rec);
734 }
735
736 #if 0
737 msg->te = tevent_add_timer(msg->sconn->ev_ctx,
738 msg,
739 end_time,
740 smbd_deferred_open_timer,
741 msg);
742 if (!msg->te) {
743 DEBUG(0,("push_message: event_add_timed failed\n"));
744 TALLOC_FREE(msg);
745 return false;
746 }
747 #endif
748
749 DLIST_ADD_END(req->sconn->deferred_open_queue, msg,
750 struct pending_message_list *);
751
752 DEBUG(10,("push_message: pushed message length %u on "
753 "deferred_open_queue\n", (unsigned int)msg_len));
754
755 return True;
756 }
757
758 /****************************************************************************
759 Function to delete a sharing violation open message by mid.
760 ****************************************************************************/
761
762 void remove_deferred_open_message_smb(struct smbd_server_connection *sconn,
763 uint64_t mid)
764 {
765 struct pending_message_list *pml;
766
767 if (sconn->using_smb2) {
768 remove_deferred_open_message_smb2(sconn, mid);
769 return;
770 }
771
772 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
773 if (mid == (uint64_t)SVAL(pml->buf.data,smb_mid)) {
774 DEBUG(10,("remove_deferred_open_message_smb: "
775 "deleting mid %llu len %u\n",
776 (unsigned long long)mid,
777 (unsigned int)pml->buf.length ));
778 DLIST_REMOVE(sconn->deferred_open_queue, pml);
779 TALLOC_FREE(pml);
780 return;
781 }
782 }
783 }
784
785 /****************************************************************************
786 Move a sharing violation open retry message to the front of the list and
787 schedule it for immediate processing.
788 ****************************************************************************/
789
790 bool schedule_deferred_open_message_smb(struct smbd_server_connection *sconn,
791 uint64_t mid)
792 {
793 struct pending_message_list *pml;
794 int i = 0;
795
796 if (sconn->using_smb2) {
797 return schedule_deferred_open_message_smb2(sconn, mid);
798 }
799
800 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
801 uint64_t msg_mid = (uint64_t)SVAL(pml->buf.data,smb_mid);
802
803 DEBUG(10,("schedule_deferred_open_message_smb: [%d] "
804 "msg_mid = %llu\n",
805 i++,
806 (unsigned long long)msg_mid ));
807
808 if (mid == msg_mid) {
809 struct tevent_timer *te;
810
811 if (pml->processed) {
812 /* A processed message should not be
813 * rescheduled. */
814 DEBUG(0,("schedule_deferred_open_message_smb: LOGIC ERROR "
815 "message mid %llu was already processed\n",
816 (unsigned long long)msg_mid ));
817 continue;
818 }
819
820 DEBUG(10,("schedule_deferred_open_message_smb: "
821 "scheduling mid %llu\n",
822 (unsigned long long)mid ));
823
824 te = tevent_add_timer(pml->sconn->ev_ctx,
825 pml,
826 timeval_zero(),
827 smbd_deferred_open_timer,
828 pml);
829 if (!te) {
830 DEBUG(10,("schedule_deferred_open_message_smb: "
831 "event_add_timed() failed, "
832 "skipping mid %llu\n",
833 (unsigned long long)msg_mid ));
834 }
835
836 TALLOC_FREE(pml->te);
837 pml->te = te;
838 DLIST_PROMOTE(sconn->deferred_open_queue, pml);
839 return true;
840 }
841 }
842
843 DEBUG(10,("schedule_deferred_open_message_smb: failed to "
844 "find message mid %llu\n",
845 (unsigned long long)mid ));
846
847 return false;
848 }
849
850 /****************************************************************************
851 Return true if this mid is on the deferred queue and was not yet processed.
852 ****************************************************************************/
853
854 bool open_was_deferred(struct smbd_server_connection *sconn, uint64_t mid)
855 {
856 struct pending_message_list *pml;
857
858 if (sconn->using_smb2) {
859 return open_was_deferred_smb2(sconn, mid);
860 }
861
862 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
863 if (((uint64_t)SVAL(pml->buf.data,smb_mid)) == mid && !pml->processed) {
864 return True;
865 }
866 }
867 return False;
868 }
869
870 /****************************************************************************
871 Return the message queued by this mid.
872 ****************************************************************************/
873
874 static struct pending_message_list *get_deferred_open_message_smb(
875 struct smbd_server_connection *sconn, uint64_t mid)
876 {
877 struct pending_message_list *pml;
878
879 for (pml = sconn->deferred_open_queue; pml; pml = pml->next) {
880 if (((uint64_t)SVAL(pml->buf.data,smb_mid)) == mid) {
881 return pml;
882 }
883 }
884 return NULL;
885 }
886
887 /****************************************************************************
888 Get the state data queued by this mid.
889 ****************************************************************************/
890
891 bool get_deferred_open_message_state(struct smb_request *smbreq,
892 struct timeval *p_request_time,
893 struct deferred_open_record **open_rec)
894 {
895 struct pending_message_list *pml;
896
897 if (smbreq->sconn->using_smb2) {
898 return get_deferred_open_message_state_smb2(smbreq->smb2req,
899 p_request_time,
900 open_rec);
901 }
902
903 pml = get_deferred_open_message_smb(smbreq->sconn, smbreq->mid);
904 if (!pml) {
905 return false;
906 }
907 if (p_request_time) {
908 *p_request_time = pml->request_time;
909 }
910 if (open_rec != NULL) {
911 *open_rec = pml->open_rec;
912 }
913 return true;
914 }
915
916 /****************************************************************************
917 Function to push a deferred open smb message onto a linked list of local smb
918 messages ready for processing.
919 ****************************************************************************/
920
921 bool push_deferred_open_message_smb(struct smb_request *req,
922 struct timeval request_time,
923 struct timeval timeout,
924 struct file_id id,
925 struct deferred_open_record *open_rec)
926 {
927 struct timeval end_time;
928
929 if (req->smb2req) {
930 return push_deferred_open_message_smb2(req->smb2req,
931 request_time,
932 timeout,
933 id,
934 open_rec);
935 }
936
937 if (req->unread_bytes) {
938 DEBUG(0,("push_deferred_open_message_smb: logic error ! "
939 "unread_bytes = %u\n",
940 (unsigned int)req->unread_bytes ));
941 smb_panic("push_deferred_open_message_smb: "
942 "logic error unread_bytes != 0" );
943 }
944
945 end_time = timeval_sum(&request_time, &timeout);
946
947 DEBUG(10,("push_deferred_open_message_smb: pushing message "
948 "len %u mid %llu timeout time [%u.%06u]\n",
949 (unsigned int) smb_len(req->inbuf)+4,
950 (unsigned long long)req->mid,
951 (unsigned int)end_time.tv_sec,
952 (unsigned int)end_time.tv_usec));
953
954 return push_queued_message(req, request_time, end_time, open_rec);
955 }
956
957 static void smbd_sig_term_handler(struct tevent_context *ev,
958 struct tevent_signal *se,
959 int signum,
960 int count,
961 void *siginfo,
962 void *private_data)
963 {
964 exit_server_cleanly("termination signal");
965 }
966
967 void smbd_setup_sig_term_handler(struct smbd_server_connection *sconn)
968 {
969 struct tevent_signal *se;
970
971 se = tevent_add_signal(sconn->ev_ctx,
972 sconn,
973 SIGTERM, 0,
974 smbd_sig_term_handler,
975 sconn);
976 if (!se) {
977 exit_server("failed to setup SIGTERM handler");
978 }
979 }
980
981 static void smbd_sig_hup_handler(struct tevent_context *ev,
982 struct tevent_signal *se,
983 int signum,
984 int count,
985 void *siginfo,
986 void *private_data)
987 {
988 struct smbd_server_connection *sconn =
989 talloc_get_type_abort(private_data,
990 struct smbd_server_connection);
991
992 change_to_root_user();
993 DEBUG(1,("Reloading services after SIGHUP\n"));
994 reload_services(sconn, conn_snum_used, false);
995 }
996
997 void smbd_setup_sig_hup_handler(struct smbd_server_connection *sconn)
998 {
999 struct tevent_signal *se;
1000
1001 se = tevent_add_signal(sconn->ev_ctx,
1002 sconn,
1003 SIGHUP, 0,
1004 smbd_sig_hup_handler,
1005 sconn);
1006 if (!se) {
1007 exit_server("failed to setup SIGHUP handler");
1008 }
1009 }
1010
1011 static void smbd_conf_updated(struct messaging_context *msg,
1012 void *private_data,
1013 uint32_t msg_type,
1014 struct server_id server_id,
1015 DATA_BLOB *data)
1016 {
1017 struct smbd_server_connection *sconn =
1018 talloc_get_type_abort(private_data,
1019 struct smbd_server_connection);
1020
1021 DEBUG(10,("smbd_conf_updated: Got message saying smb.conf was "
1022 "updated. Reloading.\n"));
1023 change_to_root_user();
1024 reload_services(sconn, conn_snum_used, false);
1025 }
1026
1027 /*
1028 * Only allow 5 outstanding trans requests. We're allocating memory, so
1029 * prevent a DoS.
1030 */
1031
1032 NTSTATUS allow_new_trans(struct trans_state *list, uint64_t mid)
1033 {
1034 int count = 0;
1035 for (; list != NULL; list = list->next) {
1036
1037 if (list->mid == mid) {
1038 return NT_STATUS_INVALID_PARAMETER;
1039 }
1040
1041 count += 1;
1042 }
1043 if (count > 5) {
1044 return NT_STATUS_INSUFFICIENT_RESOURCES;
1045 }
1046
1047 return NT_STATUS_OK;
1048 }
1049
1050 /*
1051 These flags determine some of the permissions required to do an operation
1052
1053 Note that I don't set NEED_WRITE on some write operations because they
1054 are used by some brain-dead clients when printing, and I don't want to
1055 force write permissions on print services.
1056 */
1057 #define AS_USER (1<<0)
1058 #define NEED_WRITE (1<<1) /* Must be paired with AS_USER */
1059 #define TIME_INIT (1<<2)
1060 #define CAN_IPC (1<<3) /* Must be paired with AS_USER */
1061 #define AS_GUEST (1<<5) /* Must *NOT* be paired with AS_USER */
1062 #define DO_CHDIR (1<<6)
1063
1064 /*
1065 define a list of possible SMB messages and their corresponding
1066 functions. Any message that has a NULL function is unimplemented -
1067 please feel free to contribute implementations!
1068 */
1069 static const struct smb_message_struct {
1070 const char *name;
1071 void (*fn)(struct smb_request *req);
1072 int flags;
1073 } smb_messages[256] = {
1074
1075 /* 0x00 */ { "SMBmkdir",reply_mkdir,AS_USER | NEED_WRITE},
1076 /* 0x01 */ { "SMBrmdir",reply_rmdir,AS_USER | NEED_WRITE},
1077 /* 0x02 */ { "SMBopen",reply_open,AS_USER },
1078 /* 0x03 */ { "SMBcreate",reply_mknew,AS_USER},
1079 /* 0x04 */ { "SMBclose",reply_close,AS_USER | CAN_IPC },
1080 /* 0x05 */ { "SMBflush",reply_flush,AS_USER},
1081 /* 0x06 */ { "SMBunlink",reply_unlink,AS_USER | NEED_WRITE },
1082 /* 0x07 */ { "SMBmv",reply_mv,AS_USER | NEED_WRITE },
1083 /* 0x08 */ { "SMBgetatr",reply_getatr,AS_USER},
1084 /* 0x09 */ { "SMBsetatr",reply_setatr,AS_USER | NEED_WRITE},
1085 /* 0x0a */ { "SMBread",reply_read,AS_USER},
1086 /* 0x0b */ { "SMBwrite",reply_write,AS_USER | CAN_IPC },
1087 /* 0x0c */ { "SMBlock",reply_lock,AS_USER},
1088 /* 0x0d */ { "SMBunlock",reply_unlock,AS_USER},
1089 /* 0x0e */ { "SMBctemp",reply_ctemp,AS_USER },
1090 /* 0x0f */ { "SMBmknew",reply_mknew,AS_USER},
1091 /* 0x10 */ { "SMBcheckpath",reply_checkpath,AS_USER},
1092 /* 0x11 */ { "SMBexit",reply_exit,DO_CHDIR},
1093 /* 0x12 */ { "SMBlseek",reply_lseek,AS_USER},
1094 /* 0x13 */ { "SMBlockread",reply_lockread,AS_USER},
1095 /* 0x14 */ { "SMBwriteunlock",reply_writeunlock,AS_USER},
1096 /* 0x15 */ { NULL, NULL, 0 },
1097 /* 0x16 */ { NULL, NULL, 0 },
1098 /* 0x17 */ { NULL, NULL, 0 },
1099 /* 0x18 */ { NULL, NULL, 0 },
1100 /* 0x19 */ { NULL, NULL, 0 },
1101 /* 0x1a */ { "SMBreadbraw",reply_readbraw,AS_USER},
1102 /* 0x1b */ { "SMBreadBmpx",reply_readbmpx,AS_USER},
1103 /* 0x1c */ { "SMBreadBs",reply_readbs,AS_USER },
1104 /* 0x1d */ { "SMBwritebraw",reply_writebraw,AS_USER},
1105 /* 0x1e */ { "SMBwriteBmpx",reply_writebmpx,AS_USER},
1106 /* 0x1f */ { "SMBwriteBs",reply_writebs,AS_USER},
1107 /* 0x20 */ { "SMBwritec", NULL,0},
1108 /* 0x21 */ { NULL, NULL, 0 },
1109 /* 0x22 */ { "SMBsetattrE",reply_setattrE,AS_USER | NEED_WRITE },
1110 /* 0x23 */ { "SMBgetattrE",reply_getattrE,AS_USER },
1111 /* 0x24 */ { "SMBlockingX",reply_lockingX,AS_USER },
1112 /* 0x25 */ { "SMBtrans",reply_trans,AS_USER | CAN_IPC },
1113 /* 0x26 */ { "SMBtranss",reply_transs,AS_USER | CAN_IPC},
1114 /* 0x27 */ { "SMBioctl",reply_ioctl,0},
1115 /* 0x28 */ { "SMBioctls", NULL,AS_USER},
1116 /* 0x29 */ { "SMBcopy",reply_copy,AS_USER | NEED_WRITE },
1117 /* 0x2a */ { "SMBmove", NULL,AS_USER | NEED_WRITE },
1118 /* 0x2b */ { "SMBecho",reply_echo,0},
1119 /* 0x2c */ { "SMBwriteclose",reply_writeclose,AS_USER},
1120 /* 0x2d */ { "SMBopenX",reply_open_and_X,AS_USER | CAN_IPC },
1121 /* 0x2e */ { "SMBreadX",reply_read_and_X,AS_USER | CAN_IPC },
1122 /* 0x2f */ { "SMBwriteX",reply_write_and_X,AS_USER | CAN_IPC },
1123 /* 0x30 */ { NULL, NULL, 0 },
1124 /* 0x31 */ { NULL, NULL, 0 },
1125 /* 0x32 */ { "SMBtrans2",reply_trans2, AS_USER | CAN_IPC },
1126 /* 0x33 */ { "SMBtranss2",reply_transs2, AS_USER | CAN_IPC },
1127 /* 0x34 */ { "SMBfindclose",reply_findclose,AS_USER},
1128 /* 0x35 */ { "SMBfindnclose",reply_findnclose,AS_USER},
1129 /* 0x36 */ { NULL, NULL, 0 },
1130 /* 0x37 */ { NULL, NULL, 0 },
1131 /* 0x38 */ { NULL, NULL, 0 },
1132 /* 0x39 */ { NULL, NULL, 0 },
1133 /* 0x3a */ { NULL, NULL, 0 },
1134 /* 0x3b */ { NULL, NULL, 0 },
1135 /* 0x3c */ { NULL, NULL, 0 },
1136 /* 0x3d */ { NULL, NULL, 0 },
1137 /* 0x3e */ { NULL, NULL, 0 },
1138 /* 0x3f */ { NULL, NULL, 0 },
1139 /* 0x40 */ { NULL, NULL, 0 },
1140 /* 0x41 */ { NULL, NULL, 0 },
1141 /* 0x42 */ { NULL, NULL, 0 },
1142 /* 0x43 */ { NULL, NULL, 0 },
1143 /* 0x44 */ { NULL, NULL, 0 },
1144 /* 0x45 */ { NULL, NULL, 0 },
1145 /* 0x46 */ { NULL, NULL, 0 },
1146 /* 0x47 */ { NULL, NULL, 0 },
1147 /* 0x48 */ { NULL, NULL, 0 },
1148 /* 0x49 */ { NULL, NULL, 0 },
1149 /* 0x4a */ { NULL, NULL, 0 },
1150 /* 0x4b */ { NULL, NULL, 0 },
1151 /* 0x4c */ { NULL, NULL, 0 },
1152 /* 0x4d */ { NULL, NULL, 0 },
1153 /* 0x4e */ { NULL, NULL, 0 },
1154 /* 0x4f */ { NULL, NULL, 0 },
1155 /* 0x50 */ { NULL, NULL, 0 },
1156 /* 0x51 */ { NULL, NULL, 0 },
1157 /* 0x52 */ { NULL, NULL, 0 },
1158 /* 0x53 */ { NULL, NULL, 0 },
1159 /* 0x54 */ { NULL, NULL, 0 },
1160 /* 0x55 */ { NULL, NULL, 0 },
1161 /* 0x56 */ { NULL, NULL, 0 },
1162 /* 0x57 */ { NULL, NULL, 0 },
1163 /* 0x58 */ { NULL, NULL, 0 },
1164 /* 0x59 */ { NULL, NULL, 0 },
1165 /* 0x5a */ { NULL, NULL, 0 },
1166 /* 0x5b */ { NULL, NULL, 0 },
1167 /* 0x5c */ { NULL, NULL, 0 },
1168 /* 0x5d */ { NULL, NULL, 0 },
1169 /* 0x5e */ { NULL, NULL, 0 },
1170 /* 0x5f */ { NULL, NULL, 0 },
1171 /* 0x60 */ { NULL, NULL, 0 },
1172 /* 0x61 */ { NULL, NULL, 0 },
1173 /* 0x62 */ { NULL, NULL, 0 },
1174 /* 0x63 */ { NULL, NULL, 0 },
1175 /* 0x64 */ { NULL, NULL, 0 },
1176 /* 0x65 */ { NULL, NULL, 0 },
1177 /* 0x66 */ { NULL, NULL, 0 },
1178 /* 0x67 */ { NULL, NULL, 0 },
1179 /* 0x68 */ { NULL, NULL, 0 },
1180 /* 0x69 */ { NULL, NULL, 0 },
1181 /* 0x6a */ { NULL, NULL, 0 },
1182 /* 0x6b */ { NULL, NULL, 0 },
1183 /* 0x6c */ { NULL, NULL, 0 },
1184 /* 0x6d */ { NULL, NULL, 0 },
1185 /* 0x6e */ { NULL, NULL, 0 },
1186 /* 0x6f */ { NULL, NULL, 0 },
1187 /* 0x70 */ { "SMBtcon",reply_tcon,0},
1188 /* 0x71 */ { "SMBtdis",reply_tdis,DO_CHDIR},
1189 /* 0x72 */ { "SMBnegprot",reply_negprot,0},
1190 /* 0x73 */ { "SMBsesssetupX",reply_sesssetup_and_X,0},
1191 /* 0x74 */ { "SMBulogoffX",reply_ulogoffX, 0}, /* ulogoff doesn't give a valid TID */
1192 /* 0x75 */ { "SMBtconX",reply_tcon_and_X,0},
1193 /* 0x76 */ { NULL, NULL, 0 },
1194 /* 0x77 */ { NULL, NULL, 0 },
1195 /* 0x78 */ { NULL, NULL, 0 },
1196 /* 0x79 */ { NULL, NULL, 0 },
1197 /* 0x7a */ { NULL, NULL, 0 },
1198 /* 0x7b */ { NULL, NULL, 0 },
1199 /* 0x7c */ { NULL, NULL, 0 },
1200 /* 0x7d */ { NULL, NULL, 0 },
1201 /* 0x7e */ { NULL, NULL, 0 },
1202 /* 0x7f */ { NULL, NULL, 0 },
1203 /* 0x80 */ { "SMBdskattr",reply_dskattr,AS_USER},
1204 /* 0x81 */ { "SMBsearch",reply_search,AS_USER},
1205 /* 0x82 */ { "SMBffirst",reply_search,AS_USER},
1206 /* 0x83 */ { "SMBfunique",reply_search,AS_USER},
1207 /* 0x84 */ { "SMBfclose",reply_fclose,AS_USER},
1208 /* 0x85 */ { NULL, NULL, 0 },
1209 /* 0x86 */ { NULL, NULL, 0 },
1210 /* 0x87 */ { NULL, NULL, 0 },
1211 /* 0x88 */ { NULL, NULL, 0 },
1212 /* 0x89 */ { NULL, NULL, 0 },
1213 /* 0x8a */ { NULL, NULL, 0 },
1214 /* 0x8b */ { NULL, NULL, 0 },
1215 /* 0x8c */ { NULL, NULL, 0 },
1216 /* 0x8d */ { NULL, NULL, 0 },
1217 /* 0x8e */ { NULL, NULL, 0 },
1218 /* 0x8f */ { NULL, NULL, 0 },
1219 /* 0x90 */ { NULL, NULL, 0 },
1220 /* 0x91 */ { NULL, NULL, 0 },
1221 /* 0x92 */ { NULL, NULL, 0 },
1222 /* 0x93 */ { NULL, NULL, 0 },
1223 /* 0x94 */ { NULL, NULL, 0 },
1224 /* 0x95 */ { NULL, NULL, 0 },
1225 /* 0x96 */ { NULL, NULL, 0 },
1226 /* 0x97 */ { NULL, NULL, 0 },
1227 /* 0x98 */ { NULL, NULL, 0 },
1228 /* 0x99 */ { NULL, NULL, 0 },
1229 /* 0x9a */ { NULL, NULL, 0 },
1230 /* 0x9b */ { NULL, NULL, 0 },
1231 /* 0x9c */ { NULL, NULL, 0 },
1232 /* 0x9d */ { NULL, NULL, 0 },
1233 /* 0x9e */ { NULL, NULL, 0 },
1234 /* 0x9f */ { NULL, NULL, 0 },
1235 /* 0xa0 */ { "SMBnttrans",reply_nttrans, AS_USER | CAN_IPC },
1236 /* 0xa1 */ { "SMBnttranss",reply_nttranss, AS_USER | CAN_IPC },
1237 /* 0xa2 */ { "SMBntcreateX",reply_ntcreate_and_X, AS_USER | CAN_IPC },
1238 /* 0xa3 */ { NULL, NULL, 0 },
1239 /* 0xa4 */ { "SMBntcancel",reply_ntcancel, 0 },
1240 /* 0xa5 */ { "SMBntrename",reply_ntrename, AS_USER | NEED_WRITE },
1241 /* 0xa6 */ { NULL, NULL, 0 },
1242 /* 0xa7 */ { NULL, NULL, 0 },
1243 /* 0xa8 */ { NULL, NULL, 0 },
1244 /* 0xa9 */ { NULL, NULL, 0 },
1245 /* 0xaa */ { NULL, NULL, 0 },
1246 /* 0xab */ { NULL, NULL, 0 },
1247 /* 0xac */ { NULL, NULL, 0 },
1248 /* 0xad */ { NULL, NULL, 0 },
1249 /* 0xae */ { NULL, NULL, 0 },
1250 /* 0xaf */ { NULL, NULL, 0 },
1251 /* 0xb0 */ { NULL, NULL, 0 },
1252 /* 0xb1 */ { NULL, NULL, 0 },
1253 /* 0xb2 */ { NULL, NULL, 0 },
1254 /* 0xb3 */ { NULL, NULL, 0 },
1255 /* 0xb4 */ { NULL, NULL, 0 },
1256 /* 0xb5 */ { NULL, NULL, 0 },
1257 /* 0xb6 */ { NULL, NULL, 0 },
1258 /* 0xb7 */ { NULL, NULL, 0 },
1259 /* 0xb8 */ { NULL, NULL, 0 },
1260 /* 0xb9 */ { NULL, NULL, 0 },
1261 /* 0xba */ { NULL, NULL, 0 },
1262 /* 0xbb */ { NULL, NULL, 0 },
1263 /* 0xbc */ { NULL, NULL, 0 },
1264 /* 0xbd */ { NULL, NULL, 0 },
1265 /* 0xbe */ { NULL, NULL, 0 },
1266 /* 0xbf */ { NULL, NULL, 0 },
1267 /* 0xc0 */ { "SMBsplopen",reply_printopen,AS_USER},
1268 /* 0xc1 */ { "SMBsplwr",reply_printwrite,AS_USER},
1269 /* 0xc2 */ { "SMBsplclose",reply_printclose,AS_USER},
1270 /* 0xc3 */ { "SMBsplretq",reply_printqueue,AS_USER},
1271 /* 0xc4 */ { NULL, NULL, 0 },
1272 /* 0xc5 */ { NULL, NULL, 0 },
1273 /* 0xc6 */ { NULL, NULL, 0 },
1274 /* 0xc7 */ { NULL, NULL, 0 },
1275 /* 0xc8 */ { NULL, NULL, 0 },
1276 /* 0xc9 */ { NULL, NULL, 0 },
1277 /* 0xca */ { NULL, NULL, 0 },
1278 /* 0xcb */ { NULL, NULL, 0 },
1279 /* 0xcc */ { NULL, NULL, 0 },
1280 /* 0xcd */ { NULL, NULL, 0 },
1281 /* 0xce */ { NULL, NULL, 0 },
1282 /* 0xcf */ { NULL, NULL, 0 },
1283 /* 0xd0 */ { "SMBsends",reply_sends,AS_GUEST},
1284 /* 0xd1 */ { "SMBsendb", NULL,AS_GUEST},
1285 /* 0xd2 */ { "SMBfwdname", NULL,AS_GUEST},
1286 /* 0xd3 */ { "SMBcancelf", NULL,AS_GUEST},
1287 /* 0xd4 */ { "SMBgetmac", NULL,AS_GUEST},
1288 /* 0xd5 */ { "SMBsendstrt",reply_sendstrt,AS_GUEST},
1289 /* 0xd6 */ { "SMBsendend",reply_sendend,AS_GUEST},
1290 /* 0xd7 */ { "SMBsendtxt",reply_sendtxt,AS_GUEST},
1291 /* 0xd8 */ { NULL, NULL, 0 },
1292 /* 0xd9 */ { NULL, NULL, 0 },
1293 /* 0xda */ { NULL, NULL, 0 },
1294 /* 0xdb */ { NULL, NULL, 0 },
1295 /* 0xdc */ { NULL, NULL, 0 },
1296 /* 0xdd */ { NULL, NULL, 0 },
1297 /* 0xde */ { NULL, NULL, 0 },
1298 /* 0xdf */ { NULL, NULL, 0 },
1299 /* 0xe0 */ { NULL, NULL, 0 },
1300 /* 0xe1 */ { NULL, NULL, 0 },
1301 /* 0xe2 */ { NULL, NULL, 0 },
1302 /* 0xe3 */ { NULL, NULL, 0 },
1303 /* 0xe4 */ { NULL, NULL, 0 },
1304 /* 0xe5 */ { NULL, NULL, 0 },
1305 /* 0xe6 */ { NULL, NULL, 0 },
1306 /* 0xe7 */ { NULL, NULL, 0 },
1307 /* 0xe8 */ { NULL, NULL, 0 },
1308 /* 0xe9 */ { NULL, NULL, 0 },
1309 /* 0xea */ { NULL, NULL, 0 },
1310 /* 0xeb */ { NULL, NULL, 0 },
1311 /* 0xec */ { NULL, NULL, 0 },
1312 /* 0xed */ { NULL, NULL, 0 },
1313 /* 0xee */ { NULL, NULL, 0 },
1314 /* 0xef */ { NULL, NULL, 0 },
1315 /* 0xf0 */ { NULL, NULL, 0 },
1316 /* 0xf1 */ { NULL, NULL, 0 },
1317 /* 0xf2 */ { NULL, NULL, 0 },
1318 /* 0xf3 */ { NULL, NULL, 0 },
1319 /* 0xf4 */ { NULL, NULL, 0 },
1320 /* 0xf5 */ { NULL, NULL, 0 },
1321 /* 0xf6 */ { NULL, NULL, 0 },
1322 /* 0xf7 */ { NULL, NULL, 0 },
1323 /* 0xf8 */ { NULL, NULL, 0 },
1324 /* 0xf9 */ { NULL, NULL, 0 },
1325 /* 0xfa */ { NULL, NULL, 0 },
1326 /* 0xfb */ { NULL, NULL, 0 },
1327 /* 0xfc */ { NULL, NULL, 0 },
1328 /* 0xfd */ { NULL, NULL, 0 },
1329 /* 0xfe */ { NULL, NULL, 0 },
1330 /* 0xff */ { NULL, NULL, 0 }
1331
1332 };
1333
1334 /*******************************************************************
1335 allocate and initialize a reply packet
1336 ********************************************************************/
1337
1338 static bool create_outbuf(TALLOC_CTX *mem_ctx, struct smb_request *req,
1339 const char *inbuf, char **outbuf, uint8_t num_words,
1340 uint32_t num_bytes)
1341 {
1342 size_t smb_len = MIN_SMB_SIZE + VWV(num_words) + num_bytes;
1343
1344 /*
1345 * Protect against integer wrap.
1346 * The SMB layer reply can be up to 0xFFFFFF bytes.
1347 */
1348 if ((num_bytes > 0xffffff) || (smb_len > 0xffffff)) {
1349 char *msg;
1350 if (asprintf(&msg, "num_bytes too large: %u",
1351 (unsigned)num_bytes) == -1) {
1352 msg = discard_const_p(char, "num_bytes too large");
1353 }
1354 smb_panic(msg);
1355 }
1356
1357 /*
1358 * Here we include the NBT header for now.
1359 */
1360 *outbuf = talloc_array(mem_ctx, char,
1361 NBT_HDR_SIZE + smb_len);
1362 if (*outbuf == NULL) {
1363 return false;
1364 }
1365
1366 construct_reply_common(req, inbuf, *outbuf);
1367 srv_set_message(*outbuf, num_words, num_bytes, false);
1368 /*
1369 * Zero out the word area, the caller has to take care of the bcc area
1370 * himself
1371 */
1372 if (num_words != 0) {
1373 memset(*outbuf + (NBT_HDR_SIZE + HDR_VWV), 0, VWV(num_words));
1374 }
1375
1376 return true;
1377 }
1378
1379 void reply_outbuf(struct smb_request *req, uint8 num_words, uint32 num_bytes)
1380 {
1381 char *outbuf;
1382 if (!create_outbuf(req, req, (const char *)req->inbuf, &outbuf, num_words,
1383 num_bytes)) {
1384 smb_panic("could not allocate output buffer\n");
1385 }
1386 req->outbuf = (uint8_t *)outbuf;
1387 }
1388
1389
1390 /*******************************************************************
1391 Dump a packet to a file.
1392 ********************************************************************/
1393
1394 static void smb_dump(const char *name, int type, const char *data)
1395 {
1396 size_t len;
1397 int fd, i;
1398 char *fname = NULL;
1399 if (DEBUGLEVEL < 50) {
1400 return;
1401 }
1402
1403 len = smb_len_tcp(data)+4;
1404 for (i=1;i<100;i++) {
1405 fname = talloc_asprintf(talloc_tos(),
1406 "/tmp/%s.%d.%s",
1407 name,
1408 i,
1409 type ? "req" : "resp");
1410 if (fname == NULL) {
1411 return;
1412 }
1413 fd = open(fname, O_WRONLY|O_CREAT|O_EXCL, 0644);
1414 if (fd != -1 || errno != EEXIST) break;
1415 TALLOC_FREE(fname);
1416 }
1417 if (fd != -1) {
1418 ssize_t ret = write(fd, data, len);
1419 if (ret != len)
1420 DEBUG(0,("smb_dump: problem: write returned %d\n", (int)ret ));
1421 close(fd);
1422 DEBUG(0,("created %s len %lu\n", fname, (unsigned long)len));
1423 }
1424 TALLOC_FREE(fname);
1425 }
1426
1427 /****************************************************************************
1428 Prepare everything for calling the actual request function, and potentially
1429 call the request function via the "new" interface.
1430
1431 Return False if the "legacy" function needs to be called, everything is
1432 prepared.
1433
1434 Return True if we're done.
1435
1436 I know this API sucks, but it is the one with the least code change I could
1437 find.
1438 ****************************************************************************/
1439
1440 static connection_struct *switch_message(uint8 type, struct smb_request *req)
1441 {
1442 int flags;
1443 uint64_t session_tag;
1444 connection_struct *conn = NULL;
1445 struct smbd_server_connection *sconn = req->sconn;
1446 NTTIME now = timeval_to_nttime(&req->request_time);
1447 struct smbXsrv_session *session = NULL;
1448 NTSTATUS status;
1449
1450 errno = 0;
1451
1452 if (smb_messages[type].fn == NULL) {
1453 DEBUG(0,("Unknown message type %d!\n",type));
1454 smb_dump("Unknown", 1, (const char *)req->inbuf);
1455 reply_unknown_new(req, type);
1456 return NULL;
1457 }
1458
1459 flags = smb_messages[type].flags;
1460
1461 /* In share mode security we must ignore the vuid. */
1462 session_tag = req->vuid;
1463 conn = req->conn;
1464
1465 DEBUG(3,("switch message %s (pid %d) conn 0x%lx\n", smb_fn_name(type),
1466 (int)getpid(), (unsigned long)conn));
1467
1468 smb_dump(smb_fn_name(type), 1, (const char *)req->inbuf);
1469
1470 /* Ensure this value is replaced in the incoming packet. */
1471 SSVAL(discard_const_p(uint8_t, req->inbuf),smb_uid,session_tag);
1472
1473 /*
1474 * Ensure the correct username is in current_user_info. This is a
1475 * really ugly bugfix for problems with multiple session_setup_and_X's
1476 * being done and allowing %U and %G substitutions to work correctly.
1477 * There is a reason this code is done here, don't move it unless you
1478 * know what you're doing... :-).
1479 * JRA.
1480 */
1481
1482 /*
1483 * lookup an existing session
1484 *
1485 * Note: for now we only check for NT_STATUS_NETWORK_SESSION_EXPIRED
1486 * here, the main check is still in change_to_user()
1487 */
1488 status = smb1srv_session_lookup(sconn->conn,
1489 session_tag,
1490 now,
1491 &session);
1492 if (NT_STATUS_EQUAL(status, NT_STATUS_NETWORK_SESSION_EXPIRED)) {
1493 switch (type) {
1494 case SMBsesssetupX:
1495 status = NT_STATUS_OK;
1496 break;
1497 default:
1498 DEBUG(1,("Error: session %llu is expired, mid=%llu.\n",
1499 (unsigned long long)session_tag,
1500 (unsigned long long)req->mid));
1501 reply_nterror(req, NT_STATUS_NETWORK_SESSION_EXPIRED);
1502 return conn;
1503 }
1504 }
1505
1506 if (session_tag != sconn->conn->last_session_id) {
1507 struct user_struct *vuser = NULL;
1508
1509 sconn->conn->last_session_id = session_tag;
1510 if (session) {
1511 vuser = session->compat;
1512 }
1513 if (vuser) {
1514 set_current_user_info(
1515 vuser->session_info->unix_info->sanitized_username,
1516 vuser->session_info->unix_info->unix_name,
1517 vuser->session_info->info->domain_name);
1518 }
1519 }
1520
1521 /* Does this call need to be run as the connected user? */
1522 if (flags & AS_USER) {
1523
1524 /* Does this call need a valid tree connection? */
1525 if (!conn) {
1526 /*
1527 * Amazingly, the error code depends on the command
1528 * (from Samba4).
1529 */
1530 if (type == SMBntcreateX) {
1531 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
1532 } else {
1533 reply_nterror(req, NT_STATUS_NETWORK_NAME_DELETED);
1534 }
1535 return NULL;
1536 }
1537
1538 if (!change_to_user(conn,session_tag)) {
1539 DEBUG(0, ("Error: Could not change to user. Removing "
1540 "deferred open, mid=%llu.\n",
1541 (unsigned long long)req->mid));
1542 reply_force_doserror(req, ERRSRV, ERRbaduid);
1543 return conn;
1544 }
1545
1546 /* All NEED_WRITE and CAN_IPC flags must also have AS_USER. */
1547
1548 /* Does it need write permission? */
1549 if ((flags & NEED_WRITE) && !CAN_WRITE(conn)) {
1550 reply_nterror(req, NT_STATUS_MEDIA_WRITE_PROTECTED);
1551 return conn;
1552 }
1553
1554 /* IPC services are limited */
1555 if (IS_IPC(conn) && !(flags & CAN_IPC)) {
1556 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1557 return conn;
1558 }
1559 } else {
1560 /* This call needs to be run as root */
1561 change_to_root_user();
1562 }
1563
1564 /* load service specific parameters */
1565 if (conn) {
1566 if (req->encrypted) {
1567 conn->encrypted_tid = true;
1568 /* encrypted required from now on. */
1569 conn->encrypt_level = SMB_SIGNING_REQUIRED;
1570 } else if (ENCRYPTION_REQUIRED(conn)) {
1571 if (req->cmd != SMBtrans2 && req->cmd != SMBtranss2) {
1572 DEBUG(1,("service[%s] requires encryption"
1573 "%s ACCESS_DENIED. mid=%llu\n",
1574 lp_servicename(talloc_tos(), SNUM(conn)),
1575 smb_fn_name(type),
1576 (unsigned long long)req->mid));
1577 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1578 return conn;
1579 }
1580 }
1581
1582 if (!set_current_service(conn,SVAL(req->inbuf,smb_flg),
1583 (flags & (AS_USER|DO_CHDIR)
1584 ?True:False))) {
1585 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1586 return conn;
1587 }
1588 conn->num_smb_operations++;
1589 }
1590
1591 /*
1592 * Does this protocol need to be run as guest? (Only archane
1593 * messenger service requests have this...)
1594 */
1595 if (flags & AS_GUEST) {
1596 char *raddr;
1597 bool ok;
1598
1599 if (!change_to_guest()) {
1600 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1601 return conn;
1602 }
1603
1604 raddr = tsocket_address_inet_addr_string(sconn->remote_address,
1605 talloc_tos());
1606 if (raddr == NULL) {
1607 reply_nterror(req, NT_STATUS_NO_MEMORY);
1608 return conn;
1609 }
1610
1611 /*
1612 * Haven't we checked this in smbd_process already???
1613 */
1614
1615 ok = allow_access(lp_hosts_deny(-1), lp_hosts_allow(-1),
1616 sconn->remote_hostname, raddr);
1617 TALLOC_FREE(raddr);
1618
1619 if (!ok) {
1620 reply_nterror(req, NT_STATUS_ACCESS_DENIED);
1621 return conn;
1622 }
1623 }
1624
1625 smb_messages[type].fn(req);
1626 return req->conn;
1627 }
1628
1629 /****************************************************************************
1630 Construct a reply to the incoming packet.
1631 ****************************************************************************/
1632
1633 static void construct_reply(struct smbd_server_connection *sconn,
1634 char *inbuf, int size, size_t unread_bytes,
1635 uint32_t seqnum, bool encrypted,
1636 struct smb_perfcount_data *deferred_pcd)
1637 {
1638 connection_struct *conn;
1639 struct smb_request *req;
1640
1641 if (!(req = talloc(talloc_tos(), struct smb_request))) {
1642 smb_panic("could not allocate smb_request");
1643 }
1644
1645 if (!init_smb_request(req, sconn, (uint8 *)inbuf, unread_bytes,
1646 encrypted, seqnum)) {
1647 exit_server_cleanly("Invalid SMB request");
1648 }
1649
1650 req->inbuf = (uint8_t *)talloc_move(req, &inbuf);
1651
1652 /* we popped this message off the queue - keep original perf data */
1653 if (deferred_pcd)
1654 req->pcd = *deferred_pcd;
1655 else {
1656 SMB_PERFCOUNT_START(&req->pcd);
1657 SMB_PERFCOUNT_SET_OP(&req->pcd, req->cmd);
1658 SMB_PERFCOUNT_SET_MSGLEN_IN(&req->pcd, size);
1659 }
1660
1661 conn = switch_message(req->cmd, req);
1662
1663 if (req->outbuf == NULL) {
1664 return;
1665 }
1666
1667 if (CVAL(req->outbuf,0) == 0) {
1668 show_msg((char *)req->outbuf);
1669 }
1670
1671 if (!srv_send_smb(req->sconn,
1672 (char *)req->outbuf,
1673 true, req->seqnum+1,
1674 IS_CONN_ENCRYPTED(conn)||req->encrypted,
1675 &req->pcd)) {
1676 exit_server_cleanly("construct_reply: srv_send_smb failed.");
1677 }
1678
1679 TALLOC_FREE(req);
1680
1681 return;
1682 }
1683
1684 static void construct_reply_chain(struct smbd_server_connection *sconn,
1685 char *inbuf, int size, uint32_t seqnum,
1686 bool encrypted,
1687 struct smb_perfcount_data *deferred_pcd)
1688 {
1689 struct smb_request **reqs = NULL;
1690 struct smb_request *req;
1691 unsigned num_reqs;
1692 bool ok;
1693
1694 ok = smb1_parse_chain(talloc_tos(), (uint8_t *)inbuf, sconn, encrypted,
1695 seqnum, &reqs, &num_reqs);
1696 if (!ok) {
1697 char errbuf[smb_size];
1698 error_packet(errbuf, 0, 0, NT_STATUS_INVALID_PARAMETER,
1699 __LINE__, __FILE__);
1700 if (!srv_send_smb(sconn, errbuf, true, seqnum, encrypted,
1701 NULL)) {
1702 exit_server_cleanly("construct_reply_chain: "
1703 "srv_send_smb failed.");
1704 }
1705 return;
1706 }
1707
1708 req = reqs[0];
1709 req->inbuf = (uint8_t *)talloc_move(reqs, &inbuf);
1710
1711 req->conn = switch_message(req->cmd, req);
1712
1713 if (req->outbuf == NULL) {
1714 /*
1715 * Request has suspended itself, will come
1716 * back here.
1717 */
1718 return;
1719 }
1720 smb_request_done(req);
1721 }
1722
1723 /*
1724 * To be called from an async SMB handler that is potentially chained
1725 * when it is finished for shipping.
1726 */
1727
1728 void smb_request_done(struct smb_request *req)
1729 {
1730 struct smb_request **reqs = NULL;
1731 struct smb_request *first_req;
1732 size_t i, num_reqs, next_index;
1733 NTSTATUS status;
1734
1735 if (req->chain == NULL) {
1736 first_req = req;
1737 goto shipit;
1738 }
1739
1740 reqs = req->chain;
1741 num_reqs = talloc_array_length(reqs);
1742
1743 for (i=0; i<num_reqs; i++) {
1744 if (reqs[i] == req) {
1745 break;
1746 }
1747 }
1748 if (i == num_reqs) {
1749 /*
1750 * Invalid chain, should not happen
1751 */
1752 status = NT_STATUS_INTERNAL_ERROR;
1753 goto error;
1754 }
1755 next_index = i+1;
1756
1757 while ((next_index < num_reqs) && (IVAL(req->outbuf, smb_rcls) == 0)) {
1758 struct smb_request *next = reqs[next_index];
1759 struct smbXsrv_tcon *tcon;
1760 NTTIME now = timeval_to_nttime(&req->request_time);
1761
1762 next->vuid = SVAL(req->outbuf, smb_uid);
1763 next->tid = SVAL(req->outbuf, smb_tid);
1764 status = smb1srv_tcon_lookup(req->sconn->conn, req->tid,
1765 now, &tcon);
1766 if (NT_STATUS_IS_OK(status)) {
1767 req->conn = tcon->compat;
1768 } else {
1769 req->conn = NULL;
1770 }
1771 next->chain_fsp = req->chain_fsp;
1772 next->inbuf = req->inbuf;
1773
1774 req = next;
1775 req->conn = switch_message(req->cmd, req);
1776
1777 if (req->outbuf == NULL) {
1778 /*
1779 * Request has suspended itself, will come
1780 * back here.
1781 */
1782 return;
1783 }
1784 next_index += 1;
1785 }
1786
1787 first_req = reqs[0];
1788
1789 for (i=1; i<next_index; i++) {
1790 bool ok;
1791
1792 ok = smb_splice_chain(&first_req->outbuf, reqs[i]->outbuf);
1793 if (!ok) {
1794 status = NT_STATUS_INTERNAL_ERROR;
1795 goto error;
1796 }
1797 }
1798
1799 SSVAL(first_req->outbuf, smb_uid, SVAL(req->outbuf, smb_uid));
1800 SSVAL(first_req->outbuf, smb_tid, SVAL(req->outbuf, smb_tid));
1801
1802 /*
1803 * This scary statement intends to set the
1804 * FLAGS2_32_BIT_ERROR_CODES flg2 field in first_req->outbuf
1805 * to the value last_req->outbuf carries
1806 */
1807 SSVAL(first_req->outbuf, smb_flg2,
1808 (SVAL(first_req->outbuf, smb_flg2) & ~FLAGS2_32_BIT_ERROR_CODES)
1809 |(SVAL(req->outbuf, smb_flg2) & FLAGS2_32_BIT_ERROR_CODES));
1810
1811 /*
1812 * Transfer the error codes from the subrequest to the main one
1813 */
1814 SSVAL(first_req->outbuf, smb_rcls, SVAL(req->outbuf, smb_rcls));
1815 SSVAL(first_req->outbuf, smb_err, SVAL(req->outbuf, smb_err));
1816
1817 _smb_setlen_large(
1818 first_req->outbuf, talloc_get_size(first_req->outbuf) - 4);
1819
1820 shipit:
1821 if (!srv_send_smb(first_req->sconn,
1822 (char *)first_req->outbuf,
1823 true, first_req->seqnum+1,
1824 IS_CONN_ENCRYPTED(req->conn)||first_req->encrypted,
1825 &first_req->pcd)) {
1826 exit_server_cleanly("construct_reply_chain: srv_send_smb "
1827 "failed.");
1828 }
1829 TALLOC_FREE(req); /* non-chained case */
1830 TALLOC_FREE(reqs); /* chained case */
1831 return;
1832
1833 error:
1834 {
1835 char errbuf[smb_size];
1836 error_packet(errbuf, 0, 0, status, __LINE__, __FILE__);
1837 if (!srv_send_smb(req->sconn, errbuf, true,
1838 req->seqnum+1, req->encrypted,
1839 NULL)) {
1840 exit_server_cleanly("construct_reply_chain: "
1841 "srv_send_smb failed.");
1842 }
1843 }
1844 TALLOC_FREE(req); /* non-chained case */
1845 TALLOC_FREE(reqs); /* chained case */
1846 }
1847
1848 /****************************************************************************
1849 Process an smb from the client
1850 ****************************************************************************/
1851 static void process_smb(struct smbd_server_connection *sconn,
1852 uint8_t *inbuf, size_t nread, size_t unread_bytes,
1853 uint32_t seqnum, bool encrypted,
1854 struct smb_perfcount_data *deferred_pcd)
1855 {
1856 int msg_type = CVAL(inbuf,0);
1857
1858 DO_PROFILE_INC(smb_count);
1859
1860 DEBUG( 6, ( "got message type 0x%x of len 0x%x\n", msg_type,
1861 smb_len(inbuf) ) );
1862 DEBUG(3, ("Transaction %d of length %d (%u toread)\n",
1863 sconn->trans_num, (int)nread, (unsigned int)unread_bytes));
1864
1865 if (msg_type != NBSSmessage) {
1866 /*
1867 * NetBIOS session request, keepalive, etc.
1868 */
1869 reply_special(sconn, (char *)inbuf, nread);
1870 goto done;
1871 }
1872
1873 if (sconn->using_smb2) {
1874 /* At this point we're not really using smb2,
1875 * we make the decision here.. */
1876 if (smbd_is_smb2_header(inbuf, nread)) {
1877 smbd_smb2_first_negprot(sconn, inbuf, nread);
1878 return;
1879 } else if (nread >= smb_size && valid_smb_header(sconn, inbuf)
1880 && CVAL(inbuf, smb_com) != 0x72) {
1881 /* This is a non-negprot SMB1 packet.
1882 Disable SMB2 from now on. */
1883 sconn->using_smb2 = false;
1884 }
1885 }
1886
1887 /* Make sure this is an SMB packet. smb_size contains NetBIOS header
1888 * so subtract 4 from it. */
1889 if ((nread < (smb_size - 4)) || !valid_smb_header(sconn, inbuf)) {
1890 DEBUG(2,("Non-SMB packet of length %d. Terminating server\n",
1891 smb_len(inbuf)));
1892
1893 /* special magic for immediate exit */
1894 if ((nread == 9) &&
1895 (IVAL(inbuf, 4) == 0x74697865) &&
1896 lp_parm_bool(-1, "smbd", "suicide mode", false)) {
1897 uint8_t exitcode = CVAL(inbuf, 8);
1898 DEBUG(1, ("Exiting immediately with code %d\n",
1899 (int)exitcode));
1900 exit(exitcode);
1901 }
1902
1903 exit_server_cleanly("Non-SMB packet");
1904 }
1905
1906 show_msg((char *)inbuf);
1907
1908 if ((unread_bytes == 0) && smb1_is_chain(inbuf)) {
1909 construct_reply_chain(sconn, (char *)inbuf, nread,
1910 seqnum, encrypted, deferred_pcd);
1911 } else {
1912 construct_reply(sconn, (char *)inbuf, nread, unread_bytes,
1913 seqnum, encrypted, deferred_pcd);
1914 }
1915
1916 sconn->trans_num++;
1917
1918 done:
1919 sconn->num_requests++;
1920
1921 /* The timeout_processing function isn't run nearly
1922 often enough to implement 'max log size' without
1923 overrunning the size of the file by many megabytes.
1924 This is especially true if we are running at debug
1925 level 10. Checking every 50 SMBs is a nice
1926 tradeoff of performance vs log file size overrun. */
1927
1928 if ((sconn->num_requests % 50) == 0 &&
1929 need_to_check_log_size()) {
1930 change_to_root_user();
1931 check_log_size();
1932 }
1933 }
1934
1935 /****************************************************************************
1936 Return a string containing the function name of a SMB command.
1937 ****************************************************************************/
1938
1939 const char *smb_fn_name(int type)
1940 {
1941 const char *unknown_name = "SMBunknown";
1942
1943 if (smb_messages[type].name == NULL)
1944 return(unknown_name);
1945
1946 return(smb_messages[type].name);
1947 }
1948
1949 /****************************************************************************
1950 Helper functions for contruct_reply.
1951 ****************************************************************************/
1952
1953 void add_to_common_flags2(uint32 v)
1954 {
1955 common_flags2 |= v;
1956 }
1957
1958 void remove_from_common_flags2(uint32 v)
1959 {
1960 common_flags2 &= ~v;
1961 }
1962
1963 static void construct_reply_common(struct smb_request *req, const char *inbuf,
1964 char *outbuf)
1965 {
1966 uint16_t in_flags2 = SVAL(inbuf,smb_flg2);
1967 uint16_t out_flags2 = common_flags2;
1968
1969 out_flags2 |= in_flags2 & FLAGS2_UNICODE_STRINGS;
1970 out_flags2 |= in_flags2 & FLAGS2_SMB_SECURITY_SIGNATURES;
1971 out_flags2 |= in_flags2 & FLAGS2_SMB_SECURITY_SIGNATURES_REQUIRED;
1972
1973 srv_set_message(outbuf,0,0,false);
1974
1975 SCVAL(outbuf, smb_com, req->cmd);
1976 SIVAL(outbuf,smb_rcls,0);
1977 SCVAL(outbuf,smb_flg, FLAG_REPLY | (CVAL(inbuf,smb_flg) & FLAG_CASELESS_PATHNAMES));
1978 SSVAL(outbuf,smb_flg2, out_flags2);
1979 memset(outbuf+smb_pidhigh,'\0',(smb_tid-smb_pidhigh));
1980 memcpy(outbuf+smb_ss_field, inbuf+smb_ss_field, 8);
1981
1982 SSVAL(outbuf,smb_tid,SVAL(inbuf,smb_tid));
1983 SSVAL(outbuf,smb_pid,SVAL(inbuf,smb_pid));
1984 SSVAL(outbuf,smb_uid,SVAL(inbuf,smb_uid));
1985 SSVAL(outbuf,smb_mid,SVAL(inbuf,smb_mid));
1986 }
1987
1988 void construct_reply_common_req(struct smb_request *req, char *outbuf)
1989 {
1990 construct_reply_common(req, (const char *)req->inbuf, outbuf);
1991 }
1992
1993 /**
1994 * @brief Find the smb_cmd offset of the last command pushed
1995 * @param[in] buf The buffer we're building up
1996 * @retval Where can we put our next andx cmd?
1997 *
1998 * While chaining requests, the "next" request we're looking at needs to put
1999 * its SMB_Command before the data the previous request already built up added
2000 * to the chain. Find the offset to the place where we have to put our cmd.
2001 */
2002
2003 static bool find_andx_cmd_ofs(uint8_t *buf, size_t *pofs)
2004 {
2005 uint8_t cmd;
2006 size_t ofs;
2007
2008 cmd = CVAL(buf, smb_com);
2009
2010 if (!is_andx_req(cmd)) {
2011 return false;
2012 }
2013
2014 ofs = smb_vwv0;
2015
2016 while (CVAL(buf, ofs) != 0xff) {
2017
2018 if (!is_andx_req(CVAL(buf, ofs))) {
2019 return false;
2020 }
2021
2022 /*
2023 * ofs is from start of smb header, so add the 4 length
2024 * bytes. The next cmd is right after the wct field.
2025 */
2026 ofs = SVAL(buf, ofs+2) + 4 + 1;
2027
2028 if (ofs+4 >= talloc_get_size(buf)) {
2029 return false;
2030 }
2031 }
2032
2033 *pofs = ofs;
2034 return true;
2035 }
2036
2037 /**
2038 * @brief Do the smb chaining at a buffer level
2039 * @param[in] poutbuf Pointer to the talloc'ed buffer to be modified
2040 * @param[in] andx_buf Buffer to be appended
2041 */
2042
2043 static bool smb_splice_chain(uint8_t **poutbuf, const uint8_t *andx_buf)
2044 {
2045 uint8_t smb_command = CVAL(andx_buf, smb_com);
2046 uint8_t wct = CVAL(andx_buf, smb_wct);
2047 const uint16_t *vwv = (const uint16_t *)(andx_buf + smb_vwv);
2048 uint32_t num_bytes = smb_buflen(andx_buf);
2049 const uint8_t *bytes = (const uint8_t *)smb_buf_const(andx_buf);
2050
2051 uint8_t *outbuf;
2052 size_t old_size, new_size;
2053 size_t ofs;
2054 size_t chain_padding = 0;
2055 size_t andx_cmd_ofs;
2056
2057
2058 old_size = talloc_get_size(*poutbuf);
2059
2060 if ((old_size % 4) != 0) {
2061 /*
2062 * Align the wct field of subsequent requests to a 4-byte
2063 * boundary
2064 */
2065 chain_padding = 4 - (old_size % 4);
2066 }
2067
2068 /*
2069 * After the old request comes the new wct field (1 byte), the vwv's
2070 * and the num_bytes field.
2071 */
2072
2073 new_size = old_size + chain_padding + 1 + wct * sizeof(uint16_t) + 2;
2074 new_size += num_bytes;
2075
2076 if ((smb_command != SMBwriteX) && (new_size > 0xffff)) {
2077 DEBUG(1, ("smb_splice_chain: %u bytes won't fit\n",
2078 (unsigned)new_size));
2079 return false;
2080 }
2081
2082 outbuf = talloc_realloc(NULL, *poutbuf, uint8_t, new_size);
2083 if (outbuf == NULL) {
2084 DEBUG(0, ("talloc failed\n"));
2085 return false;
2086 }
2087 *poutbuf = outbuf;
2088
2089 if (!find_andx_cmd_ofs(outbuf, &andx_cmd_ofs)) {
2090 DEBUG(1, ("invalid command chain\n"));
2091 *poutbuf = talloc_realloc(NULL, *poutbuf, uint8_t, old_size);
2092 return false;
2093 }
2094
2095 if (chain_padding != 0) {
2096 memset(outbuf + old_size, 0, chain_padding);
2097 old_size += chain_padding;
2098 }
2099
2100 SCVAL(outbuf, andx_cmd_ofs, smb_command);
2101 SSVAL(outbuf, andx_cmd_ofs + 2, old_size - 4);
2102
2103 ofs = old_size;
2104
2105 /*
2106 * Push the chained request:
2107 *
2108 * wct field
2109 */
2110
2111 SCVAL(outbuf, ofs, wct);
2112 ofs += 1;
2113
2114 /*
2115 * vwv array
2116 */
2117
2118 memcpy(outbuf + ofs, vwv, sizeof(uint16_t) * wct);
2119
2120 /*
2121 * HACK ALERT
2122 *
2123 * Read&X has an offset into its data buffer at
2124 * vwv[6]. reply_read_andx has no idea anymore that it's
2125 * running from within a chain, so we have to fix up the
2126 * offset here.
2127 *
2128 * Although it looks disgusting at this place, I want to keep
2129 * it here. The alternative would be to push knowledge about
2130 * the andx chain down into read&x again.
2131 */
2132
2133 if (smb_command == SMBreadX) {
2134 uint8_t *bytes_addr;
2135
2136 if (wct < 7) {
2137 /*
2138 * Invalid read&x response
2139 */
2140 return false;
2141 }
2142
2143 bytes_addr = outbuf + ofs /* vwv start */
2144 + sizeof(uint16_t) * wct /* vwv array */
2145 + sizeof(uint16_t); /* bcc */
2146
2147 SSVAL(outbuf + ofs, 6 * sizeof(uint16_t),
2148 bytes_addr - outbuf - 4);
2149 }
2150
2151 ofs += sizeof(uint16_t) * wct;
2152
2153 /*
2154 * bcc (byte count)
2155 */
2156
2157 SSVAL(outbuf, ofs, num_bytes);
2158 ofs += sizeof(uint16_t);
2159
2160 /*
2161 * The bytes field
2162 */
2163
2164 memcpy(outbuf + ofs, bytes, num_bytes);
2165
2166 return true;
2167 }
2168
2169 bool smb1_is_chain(const uint8_t *buf)
2170 {
2171 uint8_t cmd, wct, andx_cmd;
2172
2173 cmd = CVAL(buf, smb_com);
2174 if (!is_andx_req(cmd)) {
2175 return false;
2176 }
2177 wct = CVAL(buf, smb_wct);
2178 if (wct < 2) {
2179 return false;
2180 }
2181 andx_cmd = CVAL(buf, smb_vwv);
2182 return (andx_cmd != 0xFF);
2183 }
2184
2185 bool smb1_walk_chain(const uint8_t *buf,
2186 bool (*fn)(uint8_t cmd,
2187 uint8_t wct, const uint16_t *vwv,
2188 uint16_t num_bytes, const uint8_t *bytes,
2189 void *private_data),
2190 void *private_data)
2191 {
2192 size_t smblen = smb_len(buf);
2193 const char *smb_buf = smb_base(buf);
2194 uint8_t cmd, chain_cmd;
2195 uint8_t wct;
2196 const uint16_t *vwv;
2197 uint16_t num_bytes;
2198 const uint8_t *bytes;
2199
2200 cmd = CVAL(buf, smb_com);
2201 wct = CVAL(buf, smb_wct);
2202 vwv = (const uint16_t *)(buf + smb_vwv);
2203 num_bytes = smb_buflen(buf);
2204 bytes = (const uint8_t *)smb_buf_const(buf);
2205
2206 if (!fn(cmd, wct, vwv, num_bytes, bytes, private_data)) {
2207 return false;
2208 }
2209
2210 if (!is_andx_req(cmd)) {
2211 return true;
2212 }
2213 if (wct < 2) {
2214 return false;
2215 }
2216
2217 chain_cmd = CVAL(vwv, 0);
2218
2219 while (chain_cmd != 0xff) {
2220 uint32_t chain_offset; /* uint32_t to avoid overflow */
2221 size_t length_needed;
2222 ptrdiff_t vwv_offset;
2223
2224 chain_offset = SVAL(vwv+1, 0);
2225
2226 /*
2227 * Check if the client tries to fool us. The chain
2228 * offset needs to point beyond the current request in
2229 * the chain, it needs to strictly grow. Otherwise we
2230 * might be tricked into an endless loop always
2231 * processing the same request over and over again. We
2232 * used to assume that vwv and the byte buffer array
2233 * in a chain are always attached, but OS/2 the
2234 * Write&X/Read&X chain puts the Read&X vwv array
2235 * right behind the Write&X vwv chain. The Write&X bcc
2236 * array is put behind the Read&X vwv array. So now we
2237 * check whether the chain offset points strictly
2238 * behind the previous vwv array. req->buf points
2239 * right after the vwv array of the previous
2240 * request. See
2241 * https://bugzilla.samba.org/show_bug.cgi?id=8360 for
2242 * more information.
2243 */
2244
2245 vwv_offset = ((const char *)vwv - smb_buf);
2246 if (chain_offset <= vwv_offset) {
2247 return false;
2248 }
2249
2250 /*
2251 * Next check: Make sure the chain offset does not
2252 * point beyond the overall smb request length.
2253 */
2254
2255 length_needed = chain_offset+1; /* wct */
2256 if (length_needed > smblen) {
2257 return false;
2258 }
2259
2260 /*
2261 * Now comes the pointer magic. Goal here is to set up
2262 * vwv and buf correctly again. The chain offset (the
2263 * former vwv[1]) points at the new wct field.
2264 */
2265
2266 wct = CVAL(smb_buf, chain_offset);
2267
2268 if (is_andx_req(chain_cmd) && (wct < 2)) {
2269 return false;
2270 }
2271
2272 /*
2273 * Next consistency check: Make the new vwv array fits
2274 * in the overall smb request.
2275 */
2276
2277 length_needed += (wct+1)*sizeof(uint16_t); /* vwv+buflen */
2278 if (length_needed > smblen) {
2279 return false;
2280 }
2281 vwv = (const uint16_t *)(smb_buf + chain_offset + 1);
2282
2283 /*
2284 * Now grab the new byte buffer....
2285 */
2286
2287 num_bytes = SVAL(vwv+wct, 0);
2288
2289 /*
2290 * .. and check that it fits.
2291 */
2292
2293 length_needed += num_bytes;
2294 if (length_needed > smblen) {
2295 return false;
2296 }
2297 bytes = (const uint8_t *)(vwv+wct+1);
2298
2299 if (!fn(chain_cmd, wct, vwv, num_bytes, bytes, private_data)) {
2300 return false;
2301 }
2302
2303 if (!is_andx_req(chain_cmd)) {
2304 return true;
2305 }
2306 chain_cmd = CVAL(vwv, 0);
2307 }
2308 return true;
2309 }
2310
2311 static bool smb1_chain_length_cb(uint8_t cmd,
2312 uint8_t wct, const uint16_t *vwv,
2313 uint16_t num_bytes, const uint8_t *bytes,
2314 void *private_data)
2315 {
2316 unsigned *count = (unsigned *)private_data;
2317 *count += 1;
2318 return true;
2319 }
2320
2321 unsigned smb1_chain_length(const uint8_t *buf)
2322 {
2323 unsigned count = 0;
2324
2325 if (!smb1_walk_chain(buf, smb1_chain_length_cb, &count)) {
2326 return 0;
2327 }
2328 return count;
2329 }
2330
2331 struct smb1_parse_chain_state {
2332 TALLOC_CTX *mem_ctx;
2333 const uint8_t *buf;
2334 struct smbd_server_connection *sconn;
2335 bool encrypted;
2336 uint32_t seqnum;
2337
2338 struct smb_request **reqs;
2339 unsigned num_reqs;
2340 };
2341
2342 static bool smb1_parse_chain_cb(uint8_t cmd,
2343 uint8_t wct, const uint16_t *vwv,
2344 uint16_t num_bytes, const uint8_t *bytes,
2345 void *private_data)
2346 {
2347 struct smb1_parse_chain_state *state =
2348 (struct smb1_parse_chain_state *)private_data;
2349 struct smb_request **reqs;
2350 struct smb_request *req;
2351 bool ok;
2352
2353 reqs = talloc_realloc(state->mem_ctx, state->reqs,
2354 struct smb_request *, state->num_reqs+1);
2355 if (reqs == NULL) {
2356 return false;
2357 }
2358 state->reqs = reqs;
2359
2360 req = talloc(reqs, struct smb_request);
2361 if (req == NULL) {
2362 return false;
2363 }
2364
2365 ok = init_smb_request(req, state->sconn, state->buf, 0,
2366 state->encrypted, state->seqnum);
2367 if (!ok) {
2368 return false;
2369 }
2370 req->cmd = cmd;
2371 req->wct = wct;
2372 req->vwv = vwv;
2373 req->buflen = num_bytes;
2374 req->buf = bytes;
2375
2376 reqs[state->num_reqs] = req;
2377 state->num_reqs += 1;
2378 return true;
2379 }
2380
2381 bool smb1_parse_chain(TALLOC_CTX *mem_ctx, const uint8_t *buf,
2382 struct smbd_server_connection *sconn,
2383 bool encrypted, uint32_t seqnum,
2384 struct smb_request ***reqs, unsigned *num_reqs)
2385 {
2386 struct smb1_parse_chain_state state;
2387 unsigned i;
2388
2389 state.mem_ctx = mem_ctx;
2390 state.buf = buf;
2391 state.sconn = sconn;
2392 state.encrypted = encrypted;
2393 state.seqnum = seqnum;
2394 state.reqs = NULL;
2395 state.num_reqs = 0;
2396
2397 if (!smb1_walk_chain(buf, smb1_parse_chain_cb, &state)) {
2398 TALLOC_FREE(state.reqs);
2399 return false;
2400 }
2401 for (i=0; i<state.num_reqs; i++) {
2402 state.reqs[i]->chain = state.reqs;
2403 }
2404 *reqs = state.reqs;
2405 *num_reqs = state.num_reqs;
2406 return true;
2407 }
2408
2409 /****************************************************************************
2410 Check if services need reloading.
2411 ****************************************************************************/
2412
2413 static void check_reload(struct smbd_server_connection *sconn, time_t t)
2414 {
2415
2416 if (last_smb_conf_reload_time == 0) {
2417 last_smb_conf_reload_time = t;
2418 }
2419
2420 if (t >= last_smb_conf_reload_time+SMBD_RELOAD_CHECK) {
2421 reload_services(sconn, conn_snum_used, true);
2422 last_smb_conf_reload_time = t;
2423 }
2424 }
2425
2426 static bool fd_is_readable(int fd)
2427 {
2428 int ret, revents;
2429
2430 ret = poll_one_fd(fd, POLLIN|POLLHUP, 0, &revents);
2431
2432 return ((ret > 0) && ((revents & (POLLIN|POLLHUP|POLLERR)) != 0));
2433
2434 }
2435
2436 static void smbd_server_connection_write_handler(
2437 struct smbd_server_connection *sconn)
2438 {
2439 /* TODO: make write nonblocking */
2440 }
2441
2442 static void smbd_server_connection_read_handler(
2443 struct smbd_server_connection *sconn, int fd)
2444 {
2445 struct smbXsrv_connection *xconn = sconn->conn;
2446 uint8_t *inbuf = NULL;
2447 size_t inbuf_len = 0;
2448 size_t unread_bytes = 0;
2449 bool encrypted = false;
2450 TALLOC_CTX *mem_ctx = talloc_tos();
2451 NTSTATUS status;
2452 uint32_t seqnum;
2453
2454 bool async_echo = lp_async_smb_echo_handler();
2455 bool from_client = false;
2456
2457 if (async_echo) {
2458 if (fd_is_readable(sconn->smb1.echo_handler.trusted_fd)) {
2459 /*
2460 * This is the super-ugly hack to prefer the packets
2461 * forwarded by the echo handler over the ones by the
2462 * client directly
2463 */
2464 fd = sconn->smb1.echo_handler.trusted_fd;
2465 }
2466 }
2467
2468 from_client = (xconn->transport.sock == fd);
2469
2470 if (async_echo && from_client) {
2471 smbd_lock_socket(sconn);
2472
2473 if (!fd_is_readable(fd)) {
2474 DEBUG(10,("the echo listener was faster\n"));
2475 smbd_unlock_socket(sconn);
2476 return;
2477 }
2478 }
2479
2480 /* TODO: make this completely nonblocking */
2481 status = receive_smb_talloc(mem_ctx, sconn, fd,
2482 (char **)(void *)&inbuf,
2483 0, /* timeout */
2484 &unread_bytes,
2485 &encrypted,
2486 &inbuf_len, &seqnum,
2487 !from_client /* trusted channel */);
2488
2489 if (async_echo && from_client) {
2490 smbd_unlock_socket(sconn);
2491 }
2492
2493 if (NT_STATUS_EQUAL(status, NT_STATUS_RETRY)) {
2494 goto process;
2495 }
2496 if (NT_STATUS_IS_ERR(status)) {
2497 exit_server_cleanly("failed to receive smb request");
2498 }
2499 if (!NT_STATUS_IS_OK(status)) {
2500 return;
2501 }
2502
2503 process:
2504 process_smb(sconn, inbuf, inbuf_len, unread_bytes,
2505 seqnum, encrypted, NULL);
2506 }
2507
2508 static void smbd_server_connection_handler(struct tevent_context *ev,
2509 struct tevent_fd *fde,
2510 uint16_t flags,
2511 void *private_data)
2512 {
2513 struct smbd_server_connection *conn = talloc_get_type(private_data,
2514 struct smbd_server_connection);
2515 struct smbXsrv_connection *xconn = conn->conn;
2516
2517 if (!NT_STATUS_IS_OK(conn->status)) {
2518 /*
2519 * we're not supposed to do any io
2520 */
2521 TEVENT_FD_NOT_READABLE(conn->smb1.fde);
2522 TEVENT_FD_NOT_WRITEABLE(conn->smb1.fde);
2523 return;
2524 }
2525
2526 if (flags & TEVENT_FD_WRITE) {
2527 smbd_server_connection_write_handler(conn);
2528 return;
2529 }
2530 if (flags & TEVENT_FD_READ) {
2531 smbd_server_connection_read_handler(conn, xconn->transport.sock);
2532 return;
2533 }
2534 }
2535
2536 static void smbd_server_echo_handler(struct tevent_context *ev,
2537 struct tevent_fd *fde,
2538 uint16_t flags,
2539 void *private_data)
2540 {
2541 struct smbd_server_connection *conn = talloc_get_type(private_data,
2542 struct smbd_server_connection);
2543
2544 if (!NT_STATUS_IS_OK(conn->status)) {
2545 /*
2546 * we're not supposed to do any io
2547 */
2548 TEVENT_FD_NOT_READABLE(conn->smb1.echo_handler.trusted_fde);
2549 TEVENT_FD_NOT_WRITEABLE(conn->smb1.echo_handler.trusted_fde);
2550 return;
2551 }
2552
2553 if (flags & TEVENT_FD_WRITE) {
2554 smbd_server_connection_write_handler(conn);
2555 return;
2556 }
2557 if (flags & TEVENT_FD_READ) {
2558 smbd_server_connection_read_handler(
2559 conn, conn->smb1.echo_handler.trusted_fd);
2560 return;
2561 }
2562 }
2563
2564 struct smbd_release_ip_state {
2565 struct smbd_server_connection *sconn;
2566 struct tevent_immediate *im;
2567 char addr[INET6_ADDRSTRLEN];
2568 };
2569
2570 static void smbd_release_ip_immediate(struct tevent_context *ctx,
2571 struct tevent_immediate *im,
2572 void *private_data)
2573 {
2574 struct smbd_release_ip_state *state =
2575 talloc_get_type_abort(private_data,
2576 struct smbd_release_ip_state);
2577
2578 if (!NT_STATUS_EQUAL(state->sconn->status, NT_STATUS_ADDRESS_CLOSED)) {
2579 /*
2580 * smbd_server_connection_terminate() already triggered ?
2581 */
2582 return;
2583 }
2584
2585 smbd_server_connection_terminate(state->sconn, "CTDB_SRVID_RELEASE_IP");
2586 }
2587
2588 /****************************************************************************
2589 received when we should release a specific IP
2590 ****************************************************************************/
2591 static bool release_ip(const char *ip, void *priv)
2592 {
2593 struct smbd_release_ip_state *state =
2594 talloc_get_type_abort(priv,
2595 struct smbd_release_ip_state);
2596 const char *addr = state->addr;
2597 const char *p = addr;
2598
2599 if (!NT_STATUS_IS_OK(state->sconn->status)) {
2600 /* avoid recursion */
2601 return false;
2602 }
2603
2604 if (strncmp("::ffff:", addr, 7) == 0) {
2605 p = addr + 7;
2606 }
2607
2608 DEBUG(10, ("Got release IP message for %s, "
2609 "our address is %s\n", ip, p));
2610
2611 if ((strcmp(p, ip) == 0) || ((p != addr) && strcmp(addr, ip) == 0)) {
2612 DEBUG(0,("Got release IP message for our IP %s - exiting immediately\n",
2613 ip));
2614 /*
2615 * With SMB2 we should do a clean disconnect,
2616 * the previous_session_id in the session setup
2617 * will cleanup the old session, tcons and opens.
2618 *
2619 * A clean disconnect is needed in order to support
2620 * durable handles.
2621 *
2622 * Note: typically this is never triggered
2623 * as we got a TCP RST (triggered by ctdb event scripts)
2624 * before we get CTDB_SRVID_RELEASE_IP.
2625 *
2626 * We used to call _exit(1) here, but as this was mostly never
2627 * triggered and has implication on our process model,
2628 * we can just use smbd_server_connection_terminate()
2629 * (also for SMB1).
2630 *
2631 * We don't call smbd_server_connection_terminate() directly
2632 * as we might be called from within ctdbd_migrate(),
2633 * we need to defer our action to the next event loop
2634 */
2635 tevent_schedule_immediate(state->im, state->sconn->ev_ctx,
2636 smbd_release_ip_immediate, state);
2637
2638 /*
2639 * Make sure we don't get any io on the connection.
2640 */
2641 state->sconn->status = NT_STATUS_ADDRESS_CLOSED;
2642 return true;
2643 }
2644
2645 return false;
2646 }
2647
2648 static NTSTATUS smbd_register_ips(struct smbd_server_connection *sconn,
2649 struct sockaddr_storage *srv,
2650 struct sockaddr_storage *clnt)
2651 {
2652 struct smbd_release_ip_state *state;
2653 struct ctdbd_connection *cconn;
2654
2655 cconn = messaging_ctdbd_connection();
2656 if (cconn == NULL) {
2657 return NT_STATUS_NO_MEMORY;
2658 }
2659
2660 state = talloc_zero(sconn, struct smbd_release_ip_state);
2661 if (state == NULL) {
2662 return NT_STATUS_NO_MEMORY;
2663 }
2664 state->sconn = sconn;
2665 state->im = tevent_create_immediate(state);
2666 if (state->im == NULL) {
2667 return NT_STATUS_NO_MEMORY;
2668 }
2669 if (print_sockaddr(state->addr, sizeof(state->addr), srv) == NULL) {
2670 return NT_STATUS_NO_MEMORY;
2671 }
2672
2673 return ctdbd_register_ips(cconn, srv, clnt, release_ip, state);
2674 }
2675
2676 static void msg_kill_client_ip(struct messaging_context *msg_ctx,
2677 void *private_data, uint32_t msg_type,
2678 struct server_id server_id, DATA_BLOB *data)
2679 {
2680 struct smbd_server_connection *sconn = talloc_get_type_abort(
2681 private_data, struct smbd_server_connection);
2682 const char *ip = (char *) data->data;
2683 char *client_ip;
2684
2685 DEBUG(10, ("Got kill request for client IP %s\n", ip));
2686
2687 client_ip = tsocket_address_inet_addr_string(sconn->remote_address,
2688 talloc_tos());
2689 if (client_ip == NULL) {
2690 return;
2691 }
2692
2693 if (strequal(ip, client_ip)) {
2694 DEBUG(1, ("Got kill client message for %s - "
2695 "exiting immediately\n", ip));
2696 exit_server_cleanly("Forced disconnect for client");
2697 }
2698
2699 TALLOC_FREE(client_ip);
2700 }
2701
2702 /*
2703 * Send keepalive packets to our client
2704 */
2705 static bool keepalive_fn(const struct timeval *now, void *private_data)
2706 {
2707 struct smbd_server_connection *sconn = talloc_get_type_abort(
2708 private_data, struct smbd_server_connection);
2709 struct smbXsrv_connection *xconn = sconn->conn;
2710 bool ret;
2711
2712 if (sconn->using_smb2) {
2713 /* Don't do keepalives on an SMB2 connection. */
2714 return false;
2715 }
2716
2717 smbd_lock_socket(sconn);
2718 ret = send_keepalive(xconn->transport.sock);
2719 smbd_unlock_socket(sconn);
2720
2721 if (!ret) {
2722 int saved_errno = errno;
2723 /*
2724 * Try and give an error message saying what
2725 * client failed.
2726 */
2727 DEBUG(0, ("send_keepalive failed for client %s. "
2728 "Error %s - exiting\n",
2729 smbXsrv_connection_dbg(xconn),
2730 strerror(saved_errno)));
2731 errno = saved_errno;
2732 return False;
2733 }
2734 return True;
2735 }
2736
2737 /*
2738 * Do the recurring check if we're idle
2739 */
2740 static bool deadtime_fn(const struct timeval *now, void *private_data)
2741 {
2742 struct smbd_server_connection *sconn =
2743 (struct smbd_server_connection *)private_data;
2744
2745 if ((conn_num_open(sconn) == 0)
2746 || (conn_idle_all(sconn, now->tv_sec))) {
2747 DEBUG( 2, ( "Closing idle connection\n" ) );
2748 messaging_send(sconn->msg_ctx,
2749 messaging_server_id(sconn->msg_ctx),
2750 MSG_SHUTDOWN, &data_blob_null);
2751 return False;
2752 }
2753
2754 return True;
2755 }
2756
2757 /*
2758 * Do the recurring log file and smb.conf reload checks.
2759 */
2760
2761 static bool housekeeping_fn(const struct timeval *now, void *private_data)
2762 {
2763 struct smbd_server_connection *sconn = talloc_get_type_abort(
2764 private_data, struct smbd_server_connection);
2765
2766 DEBUG(5, ("housekeeping\n"));
2767
2768 change_to_root_user();
2769
2770 /* update printer queue caches if necessary */
2771 update_monitored_printq_cache(sconn->msg_ctx);
2772
2773 /* check if we need to reload services */
2774 check_reload(sconn, time_mono(NULL));
2775
2776 /*
2777 * Force a log file check.
2778 */
2779 force_check_log_size();
2780 check_log_size();
2781 return true;
2782 }
2783
2784 /*
2785 * Read an smb packet in the echo handler child, giving the parent
2786 * smbd one second to react once the socket becomes readable.
2787 */
2788
2789 struct smbd_echo_read_state {
2790 struct tevent_context *ev;
2791 struct smbd_server_connection *sconn;
2792
2793 char *buf;
2794 size_t buflen;
2795 uint32_t seqnum;
2796 };
2797
2798 static void smbd_echo_read_readable(struct tevent_req *subreq);
2799 static void smbd_echo_read_waited(struct tevent_req *subreq);
2800
2801 static struct tevent_req *smbd_echo_read_send(
2802 TALLOC_CTX *mem_ctx, struct tevent_context *ev,
2803 struct smbd_server_connection *sconn)
2804 {
2805 struct tevent_req *req, *subreq;
2806 struct smbd_echo_read_state *state;
2807 struct smbXsrv_connection *xconn = sconn->conn;
2808
2809 req = tevent_req_create(mem_ctx, &state,
2810 struct smbd_echo_read_state);
2811 if (req == NULL) {
2812 return NULL;
2813 }
2814 state->ev = ev;
2815 state->sconn = sconn;
2816
2817 subreq = wait_for_read_send(state, ev, xconn->transport.sock);
2818 if (tevent_req_nomem(subreq, req)) {
2819 return tevent_req_post(req, ev);
2820 }
2821 tevent_req_set_callback(subreq, smbd_echo_read_readable, req);
2822 return req;
2823 }
2824
2825 static void smbd_echo_read_readable(struct tevent_req *subreq)
2826 {
2827 struct tevent_req *req = tevent_req_callback_data(
2828 subreq, struct tevent_req);
2829 struct smbd_echo_read_state *state = tevent_req_data(
2830 req, struct smbd_echo_read_state);
2831 bool ok;
2832 int err;
2833
2834 ok = wait_for_read_recv(subreq, &err);
2835 TALLOC_FREE(subreq);
2836 if (!ok) {
2837 tevent_req_nterror(req, map_nt_error_from_unix(err));
2838 return;
2839 }
2840
2841 /*
2842 * Give the parent smbd one second to step in
2843 */
2844
2845 subreq = tevent_wakeup_send(
2846 state, state->ev, timeval_current_ofs(1, 0));
2847 if (tevent_req_nomem(subreq, req)) {
2848 return;
2849 }
2850 tevent_req_set_callback(subreq, smbd_echo_read_waited, req);
2851 }
2852
2853 static void smbd_echo_read_waited(struct tevent_req *subreq)
2854 {
2855 struct tevent_req *req = tevent_req_callback_data(
2856 subreq, struct tevent_req);
2857 struct smbd_echo_read_state *state = tevent_req_data(
2858 req, struct smbd_echo_read_state);
2859 struct smbd_server_connection *sconn = state->sconn;
2860 struct smbXsrv_connection *xconn = sconn->conn;
2861 bool ok;
2862 NTSTATUS status;
2863 size_t unread = 0;
2864 bool encrypted;
2865
2866 ok = tevent_wakeup_recv(subreq);
2867 TALLOC_FREE(subreq);
2868 if (!ok) {
2869 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
2870 return;
2871 }
2872
2873 ok = smbd_lock_socket_internal(sconn);
2874 if (!ok) {
2875 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2876 DEBUG(0, ("%s: failed to lock socket\n", __location__));
2877 return;
2878 }
2879
2880 if (!fd_is_readable(xconn->transport.sock)) {
2881 DEBUG(10,("echo_handler[%d] the parent smbd was faster\n",
2882 (int)getpid()));
2883
2884 ok = smbd_unlock_socket_internal(sconn);
2885 if (!ok) {
2886 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2887 DEBUG(1, ("%s: failed to unlock socket\n",
2888 __location__));
2889 return;
2890 }
2891
2892 subreq = wait_for_read_send(state, state->ev,
2893 xconn->transport.sock);
2894 if (tevent_req_nomem(subreq, req)) {
2895 return;
2896 }
2897 tevent_req_set_callback(subreq, smbd_echo_read_readable, req);
2898 return;
2899 }
2900
2901 status = receive_smb_talloc(state, sconn,
2902 xconn->transport.sock,
2903 &state->buf,
2904 0 /* timeout */,
2905 &unread,
2906 &encrypted,
2907 &state->buflen,
2908 &state->seqnum,
2909 false /* trusted_channel*/);
2910
2911 if (tevent_req_nterror(req, status)) {
2912 tevent_req_nterror(req, status);
2913 DEBUG(1, ("echo_handler[%d]: receive_smb_raw_talloc failed: %s\n",
2914 (int)getpid(), nt_errstr(status)));
2915 return;
2916 }
2917
2918 ok = smbd_unlock_socket_internal(sconn);
2919 if (!ok) {
2920 tevent_req_nterror(req, map_nt_error_from_unix(errno));
2921 DEBUG(1, ("%s: failed to unlock socket\n", __location__));
2922 return;
2923 }
2924 tevent_req_done(req);
2925 }
2926
2927 static NTSTATUS smbd_echo_read_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
2928 char **pbuf, size_t *pbuflen, uint32_t *pseqnum)
2929 {
2930 struct smbd_echo_read_state *state = tevent_req_data(
2931 req, struct smbd_echo_read_state);
2932 NTSTATUS status;
2933
2934 if (tevent_req_is_nterror(req, &status)) {
2935 return status;
2936 }
2937 *pbuf = talloc_move(mem_ctx, &state->buf);
2938 *pbuflen = state->buflen;
2939 *pseqnum = state->seqnum;
2940 return NT_STATUS_OK;
2941 }
2942
2943 struct smbd_echo_state {
2944 struct tevent_context *ev;
2945 struct iovec *pending;
2946 struct smbd_server_connection *sconn;
2947 int parent_pipe;
2948
2949 struct tevent_fd *parent_fde;
2950
2951 struct tevent_req *write_req;
2952 };
2953
2954 static void smbd_echo_writer_done(struct tevent_req *req);
2955
2956 static void smbd_echo_activate_writer(struct smbd_echo_state *state)
2957 {
2958 int num_pending;
2959
2960 if (state->write_req != NULL) {
2961 return;
2962 }
2963
2964 num_pending = talloc_array_length(state->pending);
2965 if (num_pending == 0) {
2966 return;
2967 }
2968
2969 state->write_req = writev_send(state, state->ev, NULL,
2970 state->parent_pipe, false,
2971 state->pending, num_pending);
2972 if (state->write_req == NULL) {
2973 DEBUG(1, ("writev_send failed\n"));
2974 exit(1);
2975 }
2976
2977 talloc_steal(state->write_req, state->pending);
2978 state->pending = NULL;
2979
2980 tevent_req_set_callback(state->write_req, smbd_echo_writer_done,
2981 state);
2982 }
2983
2984 static void smbd_echo_writer_done(struct tevent_req *req)
2985 {
2986 struct smbd_echo_state *state = tevent_req_callback_data(
2987 req, struct smbd_echo_state);
2988 ssize_t written;
2989 int err;
2990
2991 written = writev_recv(req, &err);
2992 TALLOC_FREE(req);
2993 state->write_req = NULL;
2994 if (written == -1) {
2995 DEBUG(1, ("writev to parent failed: %s\n", strerror(err)));
2996 exit(1);
2997 }
2998 DEBUG(10,("echo_handler[%d]: forwarded pdu to main\n", (int)getpid()));
2999 smbd_echo_activate_writer(state);
3000 }
3001
3002 static bool smbd_echo_reply(struct smbd_echo_state *state,
3003 uint8_t *inbuf, size_t inbuf_len,
3004 uint32_t seqnum)
3005 {
3006 struct smb_request req;
3007 uint16_t num_replies;
3008 char *outbuf;
3009 bool ok;
3010
3011 if ((inbuf_len == 4) && (CVAL(inbuf, 0) == NBSSkeepalive)) {
3012 DEBUG(10, ("Got netbios keepalive\n"));
3013 /*
3014 * Just swallow it
3015 */
3016 return true;
3017 }
3018
3019 if (inbuf_len < smb_size) {
3020 DEBUG(10, ("Got short packet: %d bytes\n", (int)inbuf_len));
3021 return false;
3022 }
3023 if (!valid_smb_header(state->sconn, inbuf)) {
3024 DEBUG(10, ("Got invalid SMB header\n"));
3025 return false;
3026 }
3027
3028 if (!init_smb_request(&req, state->sconn, inbuf, 0, false,
3029 seqnum)) {
3030 return false;
3031 }
3032 req.inbuf = inbuf;
3033
3034 DEBUG(10, ("smbecho handler got cmd %d (%s)\n", (int)req.cmd,
3035 smb_messages[req.cmd].name
3036 ? smb_messages[req.cmd].name : "unknown"));
3037
3038 if (req.cmd != SMBecho) {
3039 return false;
3040 }
3041 if (req.wct < 1) {
3042 return false;
3043 }
3044
3045 num_replies = SVAL(req.vwv+0, 0);
3046 if (num_replies != 1) {
3047 /* Not a Windows "Hey, you're still there?" request */
3048 return false;
3049 }
3050
3051 if (!create_outbuf(talloc_tos(), &req, (const char *)req.inbuf, &outbuf,
3052 1, req.buflen)) {
3053 DEBUG(10, ("create_outbuf failed\n"));
3054 return false;
3055 }
3056 req.outbuf = (uint8_t *)outbuf;
3057
3058 SSVAL(req.outbuf, smb_vwv0, num_replies);
3059
3060 if (req.buflen > 0) {
3061 memcpy(smb_buf(req.outbuf), req.buf, req.buflen);
3062 }
3063
3064 ok = srv_send_smb(req.sconn,
3065 (char *)outbuf,
3066 true, seqnum+1,
3067 false, &req.pcd);
3068 TALLOC_FREE(outbuf);
3069 if (!ok) {
3070 exit(1);
3071 }
3072
3073 return true;
3074 }
3075
3076 static void smbd_echo_exit(struct tevent_context *ev,
3077 struct tevent_fd *fde, uint16_t flags,
3078 void *private_data)
3079 {
3080 DEBUG(2, ("smbd_echo_exit: lost connection to parent\n"));
3081 exit(0);
3082 }
3083
3084 static void smbd_echo_got_packet(struct tevent_req *req);
3085
3086 static void smbd_echo_loop(struct smbd_server_connection *sconn,
3087 int parent_pipe)
3088 {
3089 struct smbd_echo_state *state;
3090 struct tevent_req *read_req;
3091
3092 state = talloc_zero(sconn, struct smbd_echo_state);
3093 if (state == NULL) {
3094 DEBUG(1, ("talloc failed\n"));
3095 return;
3096 }
3097 state->sconn = sconn;
3098 state->parent_pipe = parent_pipe;
3099 state->ev = s3_tevent_context_init(state);
3100 if (state->ev == NULL) {
3101 DEBUG(1, ("tevent_context_init failed\n"));
3102 TALLOC_FREE(state);
3103 return;
3104 }
3105 state->parent_fde = tevent_add_fd(state->ev, state, parent_pipe,
3106 TEVENT_FD_READ, smbd_echo_exit,
3107 state);
3108 if (state->parent_fde == NULL) {
3109 DEBUG(1, ("tevent_add_fd failed\n"));
3110 TALLOC_FREE(state);
3111 return;
3112 }
3113
3114 read_req = smbd_echo_read_send(state, state->ev, sconn);
3115 if (read_req == NULL) {
3116 DEBUG(1, ("smbd_echo_read_send failed\n"));
3117 TALLOC_FREE(state);
3118 return;
3119 }
3120 tevent_req_set_callback(read_req, smbd_echo_got_packet, state);
3121
3122 while (true) {
3123 if (tevent_loop_once(state->ev) == -1) {
3124 DEBUG(1, ("tevent_loop_once failed: %s\n",
3125 strerror(errno)));
3126 break;
3127 }
3128 }
3129 TALLOC_FREE(state);
3130 }
3131
3132 static void smbd_echo_got_packet(struct tevent_req *req)
3133 {
3134 struct smbd_echo_state *state = tevent_req_callback_data(
3135 req, struct smbd_echo_state);
3136 NTSTATUS status;
3137 char *buf = NULL;
3138 size_t buflen = 0;
3139 uint32_t seqnum = 0;
3140 bool reply;
3141
3142 status = smbd_echo_read_recv(req, state, &buf, &buflen, &seqnum);
3143 TALLOC_FREE(req);
3144 if (!NT_STATUS_IS_OK(status)) {
3145 DEBUG(1, ("smbd_echo_read_recv returned %s\n",
3146 nt_errstr(status)));
3147 exit(1);
3148 }
3149
3150 reply = smbd_echo_reply(state, (uint8_t *)buf, buflen, seqnum);
3151 if (!reply) {
3152 size_t num_pending;
3153 struct iovec *tmp;
3154 struct iovec *iov;
3155
3156 num_pending = talloc_array_length(state->pending);
3157 tmp = talloc_realloc(state, state->pending, struct iovec,
3158 num_pending+1);
3159 if (tmp == NULL) {
3160 DEBUG(1, ("talloc_realloc failed\n"));
3161 exit(1);
3162 }
3163 state->pending = tmp;
3164
3165 if (buflen >= smb_size) {
3166 /*
3167 * place the seqnum in the packet so that the main process
3168 * can reply with signing
3169 */
3170 SIVAL(buf, smb_ss_field, seqnum);
3171 SIVAL(buf, smb_ss_field+4, NT_STATUS_V(NT_STATUS_OK));
3172 }
3173
3174 iov = &state->pending[num_pending];
3175 iov->iov_base = talloc_move(state->pending, &buf);
3176 iov->iov_len = buflen;
3177
3178 DEBUG(10,("echo_handler[%d]: forward to main\n",
3179 (int)getpid()));
3180 smbd_echo_activate_writer(state);
3181 }
3182
3183 req = smbd_echo_read_send(state, state->ev, state->sconn);
3184 if (req == NULL) {
3185 DEBUG(1, ("smbd_echo_read_send failed\n"));
3186 exit(1);
3187 }
3188 tevent_req_set_callback(req, smbd_echo_got_packet, state);
3189 }
3190
3191
3192 /*
3193 * Handle SMBecho requests in a forked child process
3194 */
3195 bool fork_echo_handler(struct smbd_server_connection *sconn)
3196 {
3197 int listener_pipe[2];
3198 int res;
3199 pid_t child;
3200 bool use_mutex = false;
3201
3202 res = pipe(listener_pipe);
3203 if (res == -1) {
3204 DEBUG(1, ("pipe() failed: %s\n", strerror(errno)));
3205 return false;
3206 }
3207
3208 #ifdef HAVE_ROBUST_MUTEXES
3209 use_mutex = tdb_runtime_check_for_robust_mutexes();
3210
3211 if (use_mutex) {
3212 pthread_mutexattr_t a;
3213
3214 sconn->smb1.echo_handler.socket_mutex =
3215 anonymous_shared_allocate(sizeof(pthread_mutex_t));
3216 if (sconn->smb1.echo_handler.socket_mutex == NULL) {
3217 DEBUG(1, ("Could not create mutex shared memory: %s\n",
3218 strerror(errno)));
3219 goto fail;
3220 }
3221
3222 res = pthread_mutexattr_init(&a);
3223 if (res != 0) {
3224 DEBUG(1, ("pthread_mutexattr_init failed: %s\n",
3225 strerror(res)));
3226 goto fail;
3227 }
3228 res = pthread_mutexattr_settype(&a, PTHREAD_MUTEX_ERRORCHECK);
3229 if (res != 0) {
3230 DEBUG(1, ("pthread_mutexattr_settype failed: %s\n",
3231 strerror(res)));
3232 pthread_mutexattr_destroy(&a);
3233 goto fail;
3234 }
3235 res = pthread_mutexattr_setpshared(&a, PTHREAD_PROCESS_SHARED);
3236 if (res != 0) {
3237 DEBUG(1, ("pthread_mutexattr_setpshared failed: %s\n",
3238 strerror(res)));
3239 pthread_mutexattr_destroy(&a);
3240 goto fail;
3241 }
3242 res = pthread_mutexattr_setrobust(&a, PTHREAD_MUTEX_ROBUST);
3243 if (res != 0) {
3244 DEBUG(1, ("pthread_mutexattr_setrobust failed: "
3245 "%s\n", strerror(res)));
3246 pthread_mutexattr_destroy(&a);
3247 goto fail;
3248 }
3249 res = pthread_mutex_init(sconn->smb1.echo_handler.socket_mutex,
3250 &a);
3251 pthread_mutexattr_destroy(&a);
3252 if (res != 0) {
3253 DEBUG(1, ("pthread_mutex_init failed: %s\n",
3254 strerror(res)));
3255 goto fail;
3256 }
3257 }
3258 #endif
3259
3260 if (!use_mutex) {
3261 sconn->smb1.echo_handler.socket_lock_fd =
3262 create_unlink_tmp(lp_lock_directory());
3263 if (sconn->smb1.echo_handler.socket_lock_fd == -1) {
3264 DEBUG(1, ("Could not create lock fd: %s\n",
3265 strerror(errno)));
3266 goto fail;
3267 }
3268 }
3269
3270 child = fork();
3271 if (child == 0) {
3272 NTSTATUS status;
3273
3274 close(listener_pipe[0]);
3275 set_blocking(listener_pipe[1], false);
3276
3277 status = reinit_after_fork(sconn->msg_ctx,
3278 sconn->ev_ctx,
3279 true);
3280 if (!NT_STATUS_IS_OK(status)) {
3281 DEBUG(1, ("reinit_after_fork failed: %s\n",
3282 nt_errstr(status)));
3283 exit(1);
3284 }
3285 smbd_echo_loop(sconn, listener_pipe[1]);
3286 exit(0);
3287 }
3288 close(listener_pipe[1]);
3289 listener_pipe[1] = -1;
3290 sconn->smb1.echo_handler.trusted_fd = listener_pipe[0];
3291
3292 DEBUG(10,("fork_echo_handler: main[%d] echo_child[%d]\n", (int)getpid(), (int)child));
3293
3294 /*
3295 * Without smb signing this is the same as the normal smbd
3296 * listener. This needs to change once signing comes in.
3297 */
3298 sconn->smb1.echo_handler.trusted_fde = tevent_add_fd(sconn->ev_ctx,
3299 sconn,
3300 sconn->smb1.echo_handler.trusted_fd,
3301 TEVENT_FD_READ,
3302 smbd_server_echo_handler,
3303 sconn);
3304 if (sconn->smb1.echo_handler.trusted_fde == NULL) {
3305 DEBUG(1, ("event_add_fd failed\n"));
3306 goto fail;
3307 }
3308
3309 return true;
3310
3311 fail:
3312 if (listener_pipe[0] != -1) {
3313 close(listener_pipe[0]);
3314 }
3315 if (listener_pipe[1] != -1) {
3316 close(listener_pipe[1]);
3317 }
3318 if (sconn->smb1.echo_handler.socket_lock_fd != -1) {
3319 close(sconn->smb1.echo_handler.socket_lock_fd);
3320 }
3321
3322 #ifdef HAVE_ROBUST_MUTEXES
3323 if (sconn->smb1.echo_handler.socket_mutex != NULL) {
3324 pthread_mutex_destroy(sconn->smb1.echo_handler.socket_mutex);
3325 anonymous_shared_free(sconn->smb1.echo_handler.socket_mutex);
3326 }
3327 #endif
3328 smbd_echo_init(sconn);
3329
3330 return false;
3331 }
3332
3333 static bool uid_in_use(const struct user_struct *user, uid_t uid)
3334 {
3335 while (user) {
3336 if (user->session_info &&
3337 (user->session_info->unix_token->uid == uid)) {
3338 return true;
3339 }
3340 user = user->next;
3341 }
3342 return false;
3343 }
3344
3345 static bool gid_in_use(const struct user_struct *user, gid_t gid)
3346 {
3347 while (user) {
3348 if (user->session_info != NULL) {
3349 int i;
3350 struct security_unix_token *utok;
3351
3352 utok = user->session_info->unix_token;
3353 if (utok->gid == gid) {
3354 return true;
3355 }
3356 for(i=0; i<utok->ngroups; i++) {
3357 if (utok->groups[i] == gid) {
3358 return true;
3359 }
3360 }
3361 }
3362 user = user->next;
3363 }
3364 return false;
3365 }
3366
3367 static bool sid_in_use(const struct user_struct *user,
3368 const struct dom_sid *psid)
3369 {
3370 while (user) {
3371 struct security_token *tok;
3372
3373 if (user->session_info == NULL) {
3374 continue;
3375 }
3376 tok = user->session_info->security_token;
3377 if (tok == NULL) {
3378 /*
3379 * Not sure session_info->security_token can
3380 * ever be NULL. This check might be not
3381 * necessary.
3382 */
3383 continue;
3384 }
3385 if (security_token_has_sid(tok, psid)) {
3386 return true;
3387 }
3388 user = user->next;
3389 }
3390 return false;
3391 }
3392
3393 static bool id_in_use(const struct user_struct *user,
3394 const struct id_cache_ref *id)
3395 {
3396 switch(id->type) {
3397 case UID:
3398 return uid_in_use(user, id->id.uid);
3399 case GID:
3400 return gid_in_use(user, id->id.gid);
3401 case SID:
3402 return sid_in_use(user, &id->id.sid);
3403 default:
3404 break;
3405 }
3406 return false;
3407 }
3408
3409 static void smbd_id_cache_kill(struct messaging_context *msg_ctx,
3410 void *private_data,
3411 uint32_t msg_type,
3412 struct server_id server_id,
3413 DATA_BLOB* data)
3414 {
3415 const char *msg = (data && data->data)
3416 ? (const char *)data->data : "<NULL>";
3417 struct id_cache_ref id;
3418 struct smbd_server_connection *sconn =
3419 talloc_get_type_abort(private_data,
3420 struct smbd_server_connection);
3421
3422 if (!id_cache_ref_parse(msg, &id)) {
3423 DEBUG(0, ("Invalid ?ID: %s\n", msg));
3424 return;
3425 }
3426
3427 if (id_in_use(sconn->users, &id)) {
3428 exit_server_cleanly(msg);
3429 }
3430 id_cache_delete_from_cache(&id);
3431 }
3432
3433 NTSTATUS smbXsrv_connection_init_tables(struct smbXsrv_connection *conn,
3434 enum protocol_types protocol)
3435 {
3436 NTSTATUS status;
3437
3438 set_Protocol(protocol);
3439 conn->protocol = protocol;
3440
3441 if (protocol >= PROTOCOL_SMB2_02) {
3442 status = smb2srv_session_table_init(conn);
3443 if (!NT_STATUS_IS_OK(status)) {
3444 return status;
3445 }
3446
3447 status = smb2srv_open_table_init(conn);
3448 if (!NT_STATUS_IS_OK(status)) {
3449 return status;
3450 }
3451 } else {
3452 status = smb1srv_session_table_init(conn);
3453 if (!NT_STATUS_IS_OK(status)) {
3454 return status;
3455 }
3456
3457 status = smb1srv_tcon_table_init(conn);
3458 if (!NT_STATUS_IS_OK(status)) {
3459 return status;
3460 }
3461
3462 status = smb1srv_open_table_init(conn);
3463 if (!NT_STATUS_IS_OK(status)) {
3464 return status;
3465 }
3466 }
3467
3468 return NT_STATUS_OK;
3469 }
3470
3471 static void smbd_tevent_trace_callback(enum tevent_trace_point point,
3472 void *private_data)
3473 {
3474 struct smbXsrv_connection *conn =
3475 talloc_get_type_abort(private_data,
3476 struct smbXsrv_connection);
3477
3478 switch (point) {
3479 case TEVENT_TRACE_BEFORE_WAIT:
3480 /*
3481 * This just removes compiler warning
3482 * without profile support
3483 */
3484 conn->smbd_idle_profstamp = 0;
3485 START_PROFILE_STAMP(smbd_idle, conn->smbd_idle_profstamp);
3486 break;
3487 case TEVENT_TRACE_AFTER_WAIT:
3488 END_PROFILE_STAMP(smbd_idle, conn->smbd_idle_profstamp);
3489 break;
3490 #ifdef TEVENT_HAS_LOOP_ONCE_TRACE_POINTS
3491 case TEVENT_TRACE_BEFORE_LOOP_ONCE:
3492 case TEVENT_TRACE_AFTER_LOOP_ONCE:
3493 break;
3494 #endif
3495 }
3496 }
3497
3498 /**
3499 * Create a debug string for the connection
3500 *
3501 * This is allocated to talloc_tos() or a string constant
3502 * in certain corner cases. The returned string should
3503 * hence not be free'd directly but only via the talloc stack.
3504 */
3505 const char *smbXsrv_connection_dbg(const struct smbXsrv_connection *xconn)
3506 {
3507 const char *ret;
3508
3509 /*
3510 * TODO: this can be improved later
3511 * maybe including the client guid or more
3512 */
3513 ret = tsocket_address_string(xconn->remote_address, talloc_tos());
3514 if (ret == NULL) {
3515 return "<tsocket_address_string() failed>";
3516 }
3517
3518 return ret;
3519 }
3520
3521 /****************************************************************************
3522 Process commands from the client
3523 ****************************************************************************/
3524
3525 void smbd_process(struct tevent_context *ev_ctx,
3526 struct messaging_context *msg_ctx,
3527 int sock_fd,
3528 bool interactive)
3529 {
3530 TALLOC_CTX *frame = talloc_stackframe();
3531 struct smbXsrv_connection *conn;
3532 struct smbd_server_connection *sconn;
3533 struct sockaddr_storage ss_srv;
3534 void *sp_srv = (void *)&ss_srv;
3535 struct sockaddr *sa_srv = (struct sockaddr *)sp_srv;
3536 struct sockaddr_storage ss_clnt;
3537 void *sp_clnt = (void *)&ss_clnt;
3538 struct sockaddr *sa_clnt = (struct sockaddr *)sp_clnt;
3539 socklen_t sa_socklen;
3540 struct tsocket_address *local_address = NULL;
3541 struct tsocket_address *remote_address = NULL;
3542 const char *locaddr = NULL;
3543 const char *remaddr = NULL;
3544 char *rhost;
3545 int ret;
3546 int tmp;
3547
3548 conn = talloc_zero(ev_ctx, struct smbXsrv_connection);
3549 if (conn == NULL) {
3550 DEBUG(0,("talloc_zero(struct smbXsrv_connection)\n"));
3551 exit_server_cleanly("talloc_zero(struct smbXsrv_connection).\n");
3552 }
3553
3554 conn->ev_ctx = ev_ctx;
3555 conn->msg_ctx = msg_ctx;
3556 conn->transport.sock = sock_fd;
3557
3558 sconn = talloc_zero(conn, struct smbd_server_connection);
3559 if (!sconn) {
3560 exit_server("failed to create smbd_server_connection");
3561 }
3562
3563 conn->sconn = sconn;
3564 sconn->conn = conn;
3565
3566 /*
3567 * TODO: remove this...:-)
3568 */
3569 global_smbXsrv_connection = conn;
3570
3571 sconn->ev_ctx = ev_ctx;
3572 sconn->msg_ctx = msg_ctx;
3573 smbd_echo_init(sconn);
3574
3575 if (!interactive) {
3576 smbd_setup_sig_term_handler(sconn);
3577 smbd_setup_sig_hup_handler(sconn);
3578
3579 if (!serverid_register(messaging_server_id(msg_ctx),
3580 FLAG_MSG_GENERAL|FLAG_MSG_SMBD
3581 |FLAG_MSG_DBWRAP
3582 |FLAG_MSG_PRINT_GENERAL)) {
3583 exit_server_cleanly("Could not register myself in "
3584 "serverid.tdb");
3585 }
3586 }
3587
3588 if (lp_server_max_protocol() >= PROTOCOL_SMB2_02) {
3589 /*
3590 * We're not making the decision here,
3591 * we're just allowing the client
3592 * to decide between SMB1 and SMB2
3593 * with the first negprot
3594 * packet.
3595 */
3596 sconn->using_smb2 = true;
3597 }
3598
3599 /* Ensure child is set to blocking mode */
3600 set_blocking(sock_fd,True);
3601
3602 set_socket_options(sock_fd, "SO_KEEPALIVE");
3603 set_socket_options(sock_fd, lp_socket_options());
3604
3605 sa_socklen = sizeof(ss_clnt);
3606 ret = getpeername(sock_fd, sa_clnt, &sa_socklen);
3607 if (ret != 0) {
3608 int level = (errno == ENOTCONN)?2:0;
3609 DEBUG(level,("getpeername() failed - %s\n", strerror(errno)));
3610 exit_server_cleanly("getpeername() failed.\n");
3611 }
3612 ret = tsocket_address_bsd_from_sockaddr(sconn,
3613 sa_clnt, sa_socklen,
3614 &remote_address);
3615 if (ret != 0) {
3616 DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
3617 __location__, strerror(errno)));
3618 exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
3619 }
3620
3621 sa_socklen = sizeof(ss_srv);
3622 ret = getsockname(sock_fd, sa_srv, &sa_socklen);
3623 if (ret != 0) {
3624 int level = (errno == ENOTCONN)?2:0;
3625 DEBUG(level,("getsockname() failed - %s\n", strerror(errno)));
3626 exit_server_cleanly("getsockname() failed.\n");
3627 }
3628 ret = tsocket_address_bsd_from_sockaddr(sconn,
3629 sa_srv, sa_socklen,
3630 &local_address);
3631 if (ret != 0) {
3632 DEBUG(0,("%s: tsocket_address_bsd_from_sockaddr remote failed - %s\n",
3633 __location__, strerror(errno)));
3634 exit_server_cleanly("tsocket_address_bsd_from_sockaddr remote failed.\n");
3635 }
3636
3637 sconn->local_address = local_address;
3638 sconn->remote_address = remote_address;
3639
3640 if (tsocket_address_is_inet(local_address, "ip")) {
3641 locaddr = tsocket_address_inet_addr_string(
3642 sconn->local_address,
3643 talloc_tos());
3644 if (locaddr == NULL) {
3645 DEBUG(0,("%s: tsocket_address_inet_addr_string local failed - %s\n",
3646 __location__, strerror(errno)));
3647 exit_server_cleanly("tsocket_address_inet_addr_string local failed.\n");
3648 }
3649 } else {
3650 locaddr = "0.0.0.0";
3651 }
3652
3653 if (tsocket_address_is_inet(remote_address, "ip")) {
3654 remaddr = tsocket_address_inet_addr_string(
3655 sconn->remote_address,
3656 talloc_tos());
3657 if (remaddr == NULL) {
3658 DEBUG(0,("%s: tsocket_address_inet_addr_string remote failed - %s\n",
3659 __location__, strerror(errno)));
3660 exit_server_cleanly("tsocket_address_inet_addr_string remote failed.\n");
3661 }
3662 } else {
3663 remaddr = "0.0.0.0";
3664 }
3665
3666 /* this is needed so that we get decent entries
3667 in smbstatus for port 445 connects */
3668 set_remote_machine_name(remaddr, false);
3669 reload_services(sconn, conn_snum_used, true);
3670
3671 /*
3672 * Before the first packet, check the global hosts allow/ hosts deny
3673 * parameters before doing any parsing of packets passed to us by the
3674 * client. This prevents attacks on our parsing code from hosts not in
3675 * the hosts allow list.
3676 */
3677
3678 ret = get_remote_hostname(remote_address,
3679 &rhost,
3680 talloc_tos());
3681 if (ret < 0) {
3682 DEBUG(0,("%s: get_remote_hostname failed - %s\n",
3683 __location__, strerror(errno)));
3684 exit_server_cleanly("get_remote_hostname failed.\n");
3685 }
3686 if (strequal(rhost, "UNKNOWN")) {
3687 rhost = talloc_strdup(talloc_tos(), remaddr);
3688 }
3689 sconn->remote_hostname = talloc_move(sconn, &rhost);
3690
3691 sub_set_socket_ids(remaddr,
3692 sconn->remote_hostname,
3693 locaddr);
3694
3695 if (!allow_access(lp_hosts_deny(-1), lp_hosts_allow(-1),
3696 sconn->remote_hostname,
3697 remaddr)) {
3698 /*
3699 * send a negative session response "not listening on calling
3700 * name"
3701 */
3702 unsigned char buf[5] = {0x83, 0, 0, 1, 0x81};
3703 DEBUG( 1, ("Connection denied from %s to %s\n",
3704 tsocket_address_string(remote_address, talloc_tos()),
3705 tsocket_address_string(local_address, talloc_tos())));
3706 (void)srv_send_smb(sconn,(char *)buf, false,
3707 0, false, NULL);
3708 exit_server_cleanly("connection denied");
3709 }
3710
3711 DEBUG(10, ("Connection allowed from %s to %s\n",
3712 tsocket_address_string(remote_address, talloc_tos()),
3713 tsocket_address_string(local_address, talloc_tos())));
3714
3715 if (lp_preload_modules()) {
3716 smb_load_modules(lp_preload_modules());
3717 }
3718
3719 smb_perfcount_init();
3720
3721 if (!init_account_policy()) {
3722 exit_server("Could not open account policy tdb.\n");
3723 }
3724
3725 if (*lp_root_directory(talloc_tos())) {
3726 if (chroot(lp_root_directory(talloc_tos())) != 0) {
3727 DEBUG(0,("Failed to change root to %s\n",
3728 lp_root_directory(talloc_tos())));
3729 exit_server("Failed to chroot()");
3730 }
3731 if (chdir("/") == -1) {
3732 DEBUG(0,("Failed to chdir to / on chroot to %s\n", lp_root_directory(talloc_tos())));
3733 exit_server("Failed to chroot()");
3734 }
3735 DEBUG(0,("Changed root to %s\n", lp_root_directory(talloc_tos())));
3736 }
3737
3738 if (!srv_init_signing(sconn)) {
3739 exit_server("Failed to init smb_signing");
3740 }
3741
3742 if (!file_init(sconn)) {
3743 exit_server("file_init() failed");
3744 }
3745
3746 /* Setup oplocks */
3747 if (!init_oplocks(sconn))
3748 exit_server("Failed to init oplocks");
3749
3750 /* register our message handlers */
3751 messaging_register(sconn->msg_ctx, sconn,
3752 MSG_SMB_FORCE_TDIS, msg_force_tdis);
3753 messaging_register(sconn->msg_ctx, sconn,
3754 MSG_SMB_CLOSE_FILE, msg_close_file);
3755 messaging_register(sconn->msg_ctx, sconn,
3756 MSG_SMB_FILE_RENAME, msg_file_was_renamed);
3757
3758 id_cache_register_msgs(sconn->msg_ctx);
3759 messaging_deregister(sconn->msg_ctx, ID_CACHE_KILL, NULL);
3760 messaging_register(sconn->msg_ctx, sconn,
3761 ID_CACHE_KILL, smbd_id_cache_kill);
3762
3763 messaging_deregister(sconn->msg_ctx,
3764 MSG_SMB_CONF_UPDATED, sconn->ev_ctx);
3765 messaging_register(sconn->msg_ctx, sconn,
3766 MSG_SMB_CONF_UPDATED, smbd_conf_updated);
3767
3768 messaging_deregister(sconn->msg_ctx, MSG_SMB_KILL_CLIENT_IP,
3769 NULL);
3770 messaging_register(sconn->msg_ctx, sconn,
3771 MSG_SMB_KILL_CLIENT_IP,
3772 msg_kill_client_ip);
3773
3774 messaging_deregister(sconn->msg_ctx, MSG_SMB_TELL_NUM_CHILDREN, NULL);
3775
3776 /*
3777 * Use the default MSG_DEBUG handler to avoid rebroadcasting
3778 * MSGs to all child processes
3779 */
3780 messaging_deregister(sconn->msg_ctx,
3781 MSG_DEBUG, NULL);
3782 messaging_register(sconn->msg_ctx, NULL,
3783 MSG_DEBUG, debug_message);
3784
3785 if ((lp_keepalive() != 0)
3786 && !(event_add_idle(ev_ctx, NULL,
3787 timeval_set(lp_keepalive(), 0),
3788 "keepalive", keepalive_fn,
3789 sconn))) {
3790 DEBUG(0, ("Could not add keepalive event\n"));
3791 exit(1);
3792 }
3793
3794 if (!(event_add_idle(ev_ctx, NULL,
3795 timeval_set(IDLE_CLOSED_TIMEOUT, 0),
3796 "deadtime", deadtime_fn, sconn))) {
3797 DEBUG(0, ("Could not add deadtime event\n"));
3798 exit(1);
3799 }
3800
3801 if (!(event_add_idle(ev_ctx, NULL,
3802 timeval_set(SMBD_HOUSEKEEPING_INTERVAL, 0),
3803 "housekeeping", housekeeping_fn, sconn))) {
3804 DEBUG(0, ("Could not add housekeeping event\n"));
3805 exit(1);
3806 }
3807
3808 if (lp_clustering()) {
3809 /*
3810 * We need to tell ctdb about our client's TCP
3811 * connection, so that for failover ctdbd can send
3812 * tickle acks, triggering a reconnection by the
3813 * client.
3814 */
3815 NTSTATUS status;
3816
3817 status = smbd_register_ips(sconn, &ss_srv, &ss_clnt);
3818 if (!NT_STATUS_IS_OK(status)) {
3819 DEBUG(0, ("ctdbd_register_ips failed: %s\n",
3820 nt_errstr(status)));
3821 }
3822 }
3823
3824 tmp = lp_max_xmit();
3825 tmp = MAX(tmp, SMB_BUFFER_SIZE_MIN);
3826 tmp = MIN(tmp, SMB_BUFFER_SIZE_MAX);
3827
3828 sconn->smb1.negprot.max_recv = tmp;
3829
3830 sconn->smb1.sessions.done_sesssetup = false;
3831 sconn->smb1.sessions.max_send = SMB_BUFFER_SIZE_MAX;
3832
3833 if (!init_dptrs(sconn)) {
3834 exit_server("init_dptrs() failed");
3835 }
3836
3837 sconn->smb1.fde = tevent_add_fd(ev_ctx,
3838 sconn,
3839 sock_fd,
3840 TEVENT_FD_READ,
3841 smbd_server_connection_handler,
3842 sconn);
3843 if (!sconn->smb1.fde) {
3844 exit_server("failed to create smbd_server_connection fde");
3845 }
3846
3847 sconn->conn->local_address = sconn->local_address;
3848 sconn->conn->remote_address = sconn->remote_address;
3849 sconn->conn->remote_hostname = sconn->remote_hostname;
3850 sconn->conn->protocol = PROTOCOL_NONE;
3851
3852 TALLOC_FREE(frame);
3853
3854 tevent_set_trace_callback(ev_ctx, smbd_tevent_trace_callback, conn);
3855
3856 while (True) {
3857 frame = talloc_stackframe_pool(8192);
3858
3859 errno = 0;
3860 if (tevent_loop_once(ev_ctx) == -1) {
3861 if (errno != EINTR) {
3862 DEBUG(3, ("tevent_loop_once failed: %s,"
3863 " exiting\n", strerror(errno) ));
3864 break;
3865 }
3866 }
3867
3868 TALLOC_FREE(frame);
3869 }
3870
3871 exit_server_cleanly(NULL);
3872 }
3873
3874 bool req_is_in_chain(const struct smb_request *req)
3875 {
3876 if (req->vwv != (const uint16_t *)(req->inbuf+smb_vwv)) {
3877 /*
3878 * We're right now handling a subsequent request, so we must
3879 * be in a chain
3880 */
3881 return true;
3882 }
3883
3884 if (!is_andx_req(req->cmd)) {
3885 return false;
3886 }
3887
3888 if (req->wct < 2) {
3889 /*
3890 * Okay, an illegal request, but definitely not chained :-)
3891 */
3892 return false;
3893 }
3894
3895 return (CVAL(req->vwv+0, 0) != 0xFF);
3896 }
Samba GIT mirror