xref: /illumos-gate/usr/src/uts/common/fs/smbsrv/smb_session.c (revision 92a0208178405fef708b0283ffcaa02fbc3468ff)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 #include <sys/atomic.h>
27 #include <sys/strsubr.h>
28 #include <sys/synch.h>
29 #include <sys/types.h>
30 #include <sys/socketvar.h>
31 #include <sys/sdt.h>
32 #include <smbsrv/netbios.h>
33 #include <smbsrv/smb_incl.h>
34 #include <smbsrv/smb_i18n.h>
35 
36 static volatile uint64_t smb_kids;
37 
38 uint32_t smb_keep_alive = SSN_KEEP_ALIVE_TIMEOUT;
39 
40 static int smb_session_message(smb_session_t *);
41 static int smb_session_xprt_puthdr(smb_session_t *, smb_xprt_t *,
42     uint8_t *, size_t);
43 
44 static void smb_request_init_command_mbuf(smb_request_t *sr);
45 
46 
47 void
48 smb_session_timers(smb_session_list_t *se)
49 {
50 	smb_session_t	*session;
51 
52 	rw_enter(&se->se_lock, RW_READER);
53 	session = list_head(&se->se_act.lst);
54 	while (session) {
55 		/*
56 		 * Walk through the table and decrement each keep_alive
57 		 * timer that has not timed out yet. (keepalive > 0)
58 		 */
59 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
60 		if (session->keep_alive &&
61 		    (session->keep_alive != (uint32_t)-1))
62 			session->keep_alive--;
63 		session = list_next(&se->se_act.lst, session);
64 	}
65 	rw_exit(&se->se_lock);
66 }
67 
68 void
69 smb_session_correct_keep_alive_values(
70     smb_session_list_t	*se,
71     uint32_t		new_keep_alive)
72 {
73 	smb_session_t		*sn;
74 
75 	if (new_keep_alive == smb_keep_alive)
76 		return;
77 	/*
78 	 * keep alive == 0 means do not drop connection if it's idle
79 	 */
80 	smb_keep_alive = (new_keep_alive) ? new_keep_alive : -1;
81 
82 	/*
83 	 * Walk through the table and set each session to the new keep_alive
84 	 * value if they have not already timed out.  Block clock interrupts.
85 	 */
86 	rw_enter(&se->se_lock, RW_READER);
87 	sn = list_head(&se->se_rdy.lst);
88 	while (sn) {
89 		ASSERT(sn->s_magic == SMB_SESSION_MAGIC);
90 		sn->keep_alive = new_keep_alive;
91 		sn = list_next(&se->se_rdy.lst, sn);
92 	}
93 	sn = list_head(&se->se_act.lst);
94 	while (sn) {
95 		ASSERT(sn->s_magic == SMB_SESSION_MAGIC);
96 		if (sn->keep_alive)
97 			sn->keep_alive = new_keep_alive;
98 		sn = list_next(&se->se_act.lst, sn);
99 	}
100 	rw_exit(&se->se_lock);
101 }
102 
103 /*
104  * smb_reconnection_check
105  *
106  * This function is called when a client indicates its current connection
107  * should be the only one it has with the server, as indicated by VC=0 in
108  * a SessionSetupX request. We go through the session list and destroy any
109  * stale connections for that client.
110  *
111  * Clients don't associate IP addresses and servers. So a client may make
112  * independent connections (i.e. with VC=0) to a server with multiple
113  * IP addresses. So, when checking for a reconnection, we need to include
114  * the local IP address, to which the client is connecting, when checking
115  * for stale sessions.
116  *
117  * Also check the server's NetBIOS name to support simultaneous access by
118  * multiple clients behind a NAT server.  This will only work for SMB over
119  * NetBIOS on TCP port 139, it will not work SMB over TCP port 445 because
120  * there is no NetBIOS name.  See also Knowledge Base article Q301673.
121  */
122 void
123 smb_session_reconnection_check(smb_session_list_t *se, smb_session_t *session)
124 {
125 	smb_session_t	*sn;
126 
127 	rw_enter(&se->se_lock, RW_READER);
128 	sn = list_head(&se->se_act.lst);
129 	while (sn) {
130 		ASSERT(sn->s_magic == SMB_SESSION_MAGIC);
131 		if ((sn != session) &&
132 		    (sn->ipaddr == session->ipaddr) &&
133 		    (sn->local_ipaddr == session->local_ipaddr) &&
134 		    (strcasecmp(sn->workstation, session->workstation) == 0) &&
135 		    (sn->opentime <= session->opentime) &&
136 		    (sn->s_kid < session->s_kid)) {
137 			tsignal(sn->s_thread, SIGINT);
138 		}
139 		sn = list_next(&se->se_act.lst, sn);
140 	}
141 	rw_exit(&se->se_lock);
142 }
143 
144 /*
145  * Send a session message - supports SMB-over-NBT and SMB-over-TCP.
146  *
147  * The mbuf chain is copied into a contiguous buffer so that the whole
148  * message is submitted to smb_sosend as a single request.  This should
149  * help Ethereal/Wireshark delineate the packets correctly even though
150  * TCP_NODELAY has been set on the socket.
151  *
152  * If an mbuf chain is provided, it will be freed and set to NULL here.
153  */
154 int
155 smb_session_send(smb_session_t *session, uint8_t type, mbuf_chain_t *mbc)
156 {
157 	smb_txreq_t	*txr;
158 	smb_xprt_t	hdr;
159 	int		rc;
160 
161 	switch (session->s_state) {
162 	case SMB_SESSION_STATE_DISCONNECTED:
163 	case SMB_SESSION_STATE_TERMINATED:
164 		if ((mbc != NULL) && (mbc->chain != NULL)) {
165 			m_freem(mbc->chain);
166 			mbc->chain = NULL;
167 			mbc->flags = 0;
168 		}
169 		return (ENOTCONN);
170 	default:
171 		break;
172 	}
173 
174 	txr = smb_net_txr_alloc();
175 
176 	if ((mbc != NULL) && (mbc->chain != NULL)) {
177 		rc = mbc_moveout(mbc, (caddr_t)&txr->tr_buf[NETBIOS_HDR_SZ],
178 		    sizeof (txr->tr_buf) - NETBIOS_HDR_SZ, &txr->tr_len);
179 		if (rc != 0) {
180 			smb_net_txr_free(txr);
181 			return (rc);
182 		}
183 	}
184 
185 	hdr.xh_type = type;
186 	hdr.xh_length = (uint32_t)txr->tr_len;
187 
188 	rc = smb_session_xprt_puthdr(session, &hdr, txr->tr_buf,
189 	    NETBIOS_HDR_SZ);
190 
191 	if (rc != 0) {
192 		smb_net_txr_free(txr);
193 		return (rc);
194 	}
195 	txr->tr_len += NETBIOS_HDR_SZ;
196 	return (smb_net_txr_send(session->sock, &session->s_txlst, txr));
197 }
198 
199 /*
200  * Read, process and respond to a NetBIOS session request.
201  *
202  * A NetBIOS session must be established for SMB-over-NetBIOS.  Validate
203  * the calling and called name format and save the client NetBIOS name,
204  * which is used when a NetBIOS session is established to check for and
205  * cleanup leftover state from a previous session.
206  *
207  * Session requests are not valid for SMB-over-TCP, which is unfortunate
208  * because without the client name leftover state cannot be cleaned up
209  * if the client is behind a NAT server.
210  */
211 static int
212 smb_session_request(struct smb_session *session)
213 {
214 	int			rc;
215 	char			*calling_name;
216 	char			*called_name;
217 	char 			client_name[NETBIOS_NAME_SZ];
218 	struct mbuf_chain 	mbc;
219 	char 			*names = NULL;
220 	mts_wchar_t		*wbuf = NULL;
221 	smb_xprt_t		hdr;
222 	char *p;
223 	unsigned int cpid = oem_get_smb_cpid();
224 	int rc1, rc2;
225 
226 	session->keep_alive = smb_keep_alive;
227 
228 	if ((rc = smb_session_xprt_gethdr(session, &hdr)) != 0)
229 		return (rc);
230 
231 	DTRACE_PROBE2(receive__session__req__xprthdr, struct session *, session,
232 	    smb_xprt_t *, &hdr);
233 
234 	if ((hdr.xh_type != SESSION_REQUEST) ||
235 	    (hdr.xh_length != NETBIOS_SESSION_REQUEST_DATA_LENGTH)) {
236 		DTRACE_PROBE1(receive__session__req__failed,
237 		    struct session *, session);
238 		return (EINVAL);
239 	}
240 
241 	names = kmem_alloc(hdr.xh_length, KM_SLEEP);
242 
243 	if ((rc = smb_sorecv(session->sock, names, hdr.xh_length)) != 0) {
244 		kmem_free(names, hdr.xh_length);
245 		DTRACE_PROBE1(receive__session__req__failed,
246 		    struct session *, session);
247 		return (rc);
248 	}
249 
250 	DTRACE_PROBE3(receive__session__req__data, struct session *, session,
251 	    char *, names, uint32_t, hdr.xh_length);
252 
253 	called_name = &names[0];
254 	calling_name = &names[NETBIOS_ENCODED_NAME_SZ + 2];
255 
256 	rc1 = netbios_name_isvalid(called_name, 0);
257 	rc2 = netbios_name_isvalid(calling_name, client_name);
258 
259 	if (rc1 == 0 || rc2 == 0) {
260 
261 		DTRACE_PROBE3(receive__invalid__session__req,
262 		    struct session *, session, char *, names,
263 		    uint32_t, hdr.xh_length);
264 
265 		kmem_free(names, hdr.xh_length);
266 		MBC_INIT(&mbc, MAX_DATAGRAM_LENGTH);
267 		(void) smb_mbc_encodef(&mbc, "b",
268 		    DATAGRAM_INVALID_SOURCE_NAME_FORMAT);
269 		(void) smb_session_send(session, NEGATIVE_SESSION_RESPONSE,
270 		    &mbc);
271 		return (EINVAL);
272 	}
273 
274 	DTRACE_PROBE3(receive__session__req__calling__decoded,
275 	    struct session *, session,
276 	    char *, calling_name, char *, client_name);
277 
278 	/*
279 	 * The client NetBIOS name is in oem codepage format.
280 	 * We need to convert it to unicode and store it in
281 	 * multi-byte format.  We also need to strip off any
282 	 * spaces added as part of the NetBIOS name encoding.
283 	 */
284 	wbuf = kmem_alloc((SMB_PI_MAX_HOST * sizeof (mts_wchar_t)), KM_SLEEP);
285 	(void) oemstounicodes(wbuf, client_name, SMB_PI_MAX_HOST, cpid);
286 	(void) mts_wcstombs(session->workstation, wbuf, SMB_PI_MAX_HOST);
287 	kmem_free(wbuf, (SMB_PI_MAX_HOST * sizeof (mts_wchar_t)));
288 
289 	if ((p = strchr(session->workstation, ' ')) != 0)
290 		*p = '\0';
291 
292 	kmem_free(names, hdr.xh_length);
293 	return (smb_session_send(session, POSITIVE_SESSION_RESPONSE, NULL));
294 }
295 
296 /*
297  * Read 4-byte header from the session socket and build an in-memory
298  * session transport header.  See smb_xprt_t definition for header
299  * format information.
300  *
301  * Direct hosted NetBIOS-less SMB (SMB-over-TCP) uses port 445.  The
302  * first byte of the four-byte header must be 0 and the next three
303  * bytes contain the length of the remaining data.
304  */
305 int
306 smb_session_xprt_gethdr(smb_session_t *session, smb_xprt_t *ret_hdr)
307 {
308 	int		rc;
309 	unsigned char	buf[NETBIOS_HDR_SZ];
310 
311 	if ((rc = smb_sorecv(session->sock, buf, NETBIOS_HDR_SZ)) != 0)
312 		return (rc);
313 
314 	switch (session->s_local_port) {
315 	case SSN_SRVC_TCP_PORT:
316 		ret_hdr->xh_type = buf[0];
317 		ret_hdr->xh_length = (((uint32_t)buf[1] & 1) << 16) |
318 		    ((uint32_t)buf[2] << 8) |
319 		    ((uint32_t)buf[3]);
320 		break;
321 
322 	case SMB_SRVC_TCP_PORT:
323 		ret_hdr->xh_type = buf[0];
324 
325 		if (ret_hdr->xh_type != 0) {
326 			cmn_err(CE_WARN, "0x%08x: invalid type (%u)",
327 			    session->ipaddr, ret_hdr->xh_type);
328 			return (EPROTO);
329 		}
330 
331 		ret_hdr->xh_length = ((uint32_t)buf[1] << 16) |
332 		    ((uint32_t)buf[2] << 8) |
333 		    ((uint32_t)buf[3]);
334 		break;
335 
336 	default:
337 		cmn_err(CE_WARN, "0x%08x: invalid port %u",
338 		    session->ipaddr, session->s_local_port);
339 		return (EPROTO);
340 	}
341 
342 	return (0);
343 }
344 
345 /*
346  * Encode a transport session packet header into a 4-byte buffer.
347  * See smb_xprt_t definition for header format information.
348  */
349 static int
350 smb_session_xprt_puthdr(smb_session_t *session, smb_xprt_t *hdr,
351     uint8_t *buf, size_t buflen)
352 {
353 	if (session == NULL || hdr == NULL ||
354 	    buf == NULL || buflen < NETBIOS_HDR_SZ) {
355 		return (-1);
356 	}
357 
358 	switch (session->s_local_port) {
359 	case SSN_SRVC_TCP_PORT:
360 		buf[0] = hdr->xh_type;
361 		buf[1] = ((hdr->xh_length >> 16) & 1);
362 		buf[2] = (hdr->xh_length >> 8) & 0xff;
363 		buf[3] = hdr->xh_length & 0xff;
364 		break;
365 
366 	case SMB_SRVC_TCP_PORT:
367 		buf[0] = hdr->xh_type;
368 		buf[1] = (hdr->xh_length >> 16) & 0xff;
369 		buf[2] = (hdr->xh_length >> 8) & 0xff;
370 		buf[3] = hdr->xh_length & 0xff;
371 		break;
372 
373 	default:
374 		cmn_err(CE_WARN, "0x%08x: invalid port (%u)",
375 		    session->ipaddr, session->s_local_port);
376 		return (-1);
377 	}
378 
379 	return (0);
380 }
381 
382 static void
383 smb_request_init_command_mbuf(smb_request_t *sr)
384 {
385 	MGET(sr->command.chain, 0, MT_DATA);
386 
387 	/*
388 	 * Setup mbuf, mimic MCLGET but use the complete packet buffer.
389 	 */
390 	sr->command.chain->m_ext.ext_buf = sr->sr_request_buf;
391 	sr->command.chain->m_data = sr->command.chain->m_ext.ext_buf;
392 	sr->command.chain->m_len = sr->sr_req_length;
393 	sr->command.chain->m_flags |= M_EXT;
394 	sr->command.chain->m_ext.ext_size = sr->sr_req_length;
395 	sr->command.chain->m_ext.ext_ref = &mclrefnoop;
396 
397 	/*
398 	 * Initialize the rest of the mbuf_chain fields
399 	 */
400 	sr->command.flags = 0;
401 	sr->command.shadow_of = 0;
402 	sr->command.max_bytes = sr->sr_req_length;
403 	sr->command.chain_offset = 0;
404 }
405 
406 /*
407  * smb_request_cancel
408  *
409  * Handle a cancel for a request properly depending on the current request
410  * state.
411  */
412 void
413 smb_request_cancel(smb_request_t *sr)
414 {
415 	mutex_enter(&sr->sr_mutex);
416 	switch (sr->sr_state) {
417 
418 	case SMB_REQ_STATE_SUBMITTED:
419 	case SMB_REQ_STATE_ACTIVE:
420 	case SMB_REQ_STATE_CLEANED_UP:
421 		sr->sr_state = SMB_REQ_STATE_CANCELED;
422 		break;
423 
424 	case SMB_REQ_STATE_WAITING_LOCK:
425 		/*
426 		 * This request is waiting on a lock.  Wakeup everything
427 		 * waiting on the lock so that the relevant thread regains
428 		 * control and notices that is has been canceled.  The
429 		 * other lock request threads waiting on this lock will go
430 		 * back to sleep when they discover they are still blocked.
431 		 */
432 		sr->sr_state = SMB_REQ_STATE_CANCELED;
433 
434 		ASSERT(sr->sr_awaiting != NULL);
435 		mutex_enter(&sr->sr_awaiting->l_mutex);
436 		cv_broadcast(&sr->sr_awaiting->l_cv);
437 		mutex_exit(&sr->sr_awaiting->l_mutex);
438 
439 		break;
440 
441 	case SMB_REQ_STATE_WAITING_EVENT:
442 	case SMB_REQ_STATE_EVENT_OCCURRED:
443 		/*
444 		 * Cancellations for these states are handled by the
445 		 * notify-change code
446 		 */
447 		break;
448 
449 	case SMB_REQ_STATE_COMPLETED:
450 	case SMB_REQ_STATE_CANCELED:
451 		/*
452 		 * No action required for these states since the request
453 		 * is completing.
454 		 */
455 		break;
456 	/*
457 	 * Cases included:
458 	 *	SMB_REQ_STATE_FREE:
459 	 *	SMB_REQ_STATE_INITIALIZING:
460 	 */
461 	default:
462 		ASSERT(0);
463 		break;
464 	}
465 	mutex_exit(&sr->sr_mutex);
466 }
467 
468 /*
469  * This is the entry point for processing SMB messages over NetBIOS or
470  * SMB-over-TCP.
471  *
472  * NetBIOS connections require a session request to establish a session
473  * on which to send session messages.
474  *
475  * Session requests are not valid on SMB-over-TCP.  We don't need to do
476  * anything here as session requests will be treated as an error when
477  * handling session messages.
478  */
479 int
480 smb_session_daemon(smb_session_list_t *se)
481 {
482 	int		rc = 0;
483 	smb_session_t	*session;
484 
485 	session = smb_session_list_activate_head(se);
486 	if (session == NULL)
487 		return (EINVAL);
488 
489 	if (session->s_local_port == SSN_SRVC_TCP_PORT) {
490 		rc = smb_session_request(session);
491 		if (rc) {
492 			smb_rwx_rwenter(&session->s_lock, RW_WRITER);
493 			session->s_state = SMB_SESSION_STATE_DISCONNECTED;
494 			smb_rwx_rwexit(&session->s_lock);
495 			smb_session_list_terminate(se, session);
496 			return (rc);
497 		}
498 	}
499 
500 	smb_rwx_rwenter(&session->s_lock, RW_WRITER);
501 	session->s_state = SMB_SESSION_STATE_ESTABLISHED;
502 	smb_rwx_rwexit(&session->s_lock);
503 
504 	rc = smb_session_message(session);
505 
506 	smb_rwx_rwenter(&session->s_lock, RW_WRITER);
507 	session->s_state = SMB_SESSION_STATE_DISCONNECTED;
508 	smb_rwx_rwexit(&session->s_lock);
509 
510 	smb_soshutdown(session->sock);
511 
512 	DTRACE_PROBE2(session__drop, struct session *, session, int, rc);
513 
514 	smb_session_cancel(session);
515 
516 	/*
517 	 * At this point everything related to the session should have been
518 	 * cleaned up and we expect that nothing will attempt to use the
519 	 * socket.
520 	 */
521 	smb_session_list_terminate(se, session);
522 
523 	return (rc);
524 }
525 
526 /*
527  * Read and process SMB requests.
528  *
529  * Returns:
530  *	0	Success
531  *	1	Unable to read transport header
532  *	2	Invalid transport header type
533  *	3	Invalid SMB length (too small)
534  *	4	Unable to read SMB header
535  *	5	Invalid SMB header (bad magic number)
536  *	6	Unable to read SMB data
537  *	2x	Write raw failed
538  */
539 static int
540 smb_session_message(smb_session_t *session)
541 {
542 	smb_request_t	*sr = NULL;
543 	smb_xprt_t	hdr;
544 	uint8_t		*req_buf;
545 	uint32_t	resid;
546 	int		rc;
547 
548 	for (;;) {
549 
550 		rc = smb_session_xprt_gethdr(session, &hdr);
551 		if (rc)
552 			return (rc);
553 
554 		DTRACE_PROBE2(session__receive__xprthdr, session_t *, session,
555 		    smb_xprt_t *, &hdr);
556 
557 		if (hdr.xh_type != SESSION_MESSAGE) {
558 			/*
559 			 * Anything other than SESSION_MESSAGE or
560 			 * SESSION_KEEP_ALIVE is an error.  A SESSION_REQUEST
561 			 * may indicate a new session request but we need to
562 			 * close this session and we can treat it as an error
563 			 * here.
564 			 */
565 			if (hdr.xh_type == SESSION_KEEP_ALIVE) {
566 				session->keep_alive = smb_keep_alive;
567 				continue;
568 			}
569 			return (EPROTO);
570 		}
571 
572 		if (hdr.xh_length < SMB_HEADER_LEN)
573 			return (EPROTO);
574 
575 		session->keep_alive = smb_keep_alive;
576 
577 		/*
578 		 * Allocate a request context, read the SMB header and validate
579 		 * it. The sr includes a buffer large enough to hold the SMB
580 		 * request payload.  If the header looks valid, read any
581 		 * remaining data.
582 		 */
583 		sr = smb_request_alloc(session, hdr.xh_length);
584 
585 		req_buf = (uint8_t *)sr->sr_request_buf;
586 		resid = hdr.xh_length;
587 
588 		rc = smb_sorecv(session->sock, req_buf, SMB_HEADER_LEN);
589 		if (rc) {
590 			smb_request_free(sr);
591 			return (rc);
592 		}
593 
594 		if (SMB_PROTOCOL_MAGIC_INVALID(sr)) {
595 			smb_request_free(sr);
596 			return (EPROTO);
597 		}
598 
599 		if (resid > SMB_HEADER_LEN) {
600 			req_buf += SMB_HEADER_LEN;
601 			resid -= SMB_HEADER_LEN;
602 
603 			rc = smb_sorecv(session->sock, req_buf, resid);
604 			if (rc) {
605 				smb_request_free(sr);
606 				return (rc);
607 			}
608 		}
609 
610 		/*
611 		 * Initialize command MBC to represent the received data.
612 		 */
613 		smb_request_init_command_mbuf(sr);
614 
615 		DTRACE_PROBE1(session__receive__smb, smb_request_t *, sr);
616 
617 		/*
618 		 * If this is a raw write, hand off the request.  The handler
619 		 * will retrieve the remaining raw data and process the request.
620 		 */
621 		if (SMB_IS_WRITERAW(sr)) {
622 			rc = smb_handle_write_raw(session, sr);
623 			/* XXX smb_request_free(sr); ??? */
624 			return (rc);
625 		}
626 
627 		sr->sr_state = SMB_REQ_STATE_SUBMITTED;
628 		(void) taskq_dispatch(session->s_server->sv_thread_pool,
629 		    smb_session_worker, sr, TQ_SLEEP);
630 	}
631 }
632 
633 /*
634  * Port will be SSN_SRVC_TCP_PORT or SMB_SRVC_TCP_PORT.
635  */
636 smb_session_t *
637 smb_session_create(struct sonode *new_so, uint16_t port, smb_server_t *sv)
638 {
639 	uint32_t		ipaddr;
640 	uint32_t		local_ipaddr;
641 	struct sockaddr_in	sin;
642 	smb_session_t		*session;
643 
644 	session = kmem_cache_alloc(sv->si_cache_session, KM_SLEEP);
645 	bzero(session, sizeof (smb_session_t));
646 
647 	if (smb_idpool_constructor(&session->s_uid_pool)) {
648 		kmem_cache_free(sv->si_cache_session, session);
649 		return (NULL);
650 	}
651 
652 	session->s_kid = SMB_NEW_KID();
653 	session->s_state = SMB_SESSION_STATE_INITIALIZED;
654 	session->native_os = NATIVE_OS_UNKNOWN;
655 	session->opentime = lbolt64;
656 	session->keep_alive = smb_keep_alive;
657 	session->activity_timestamp = lbolt64;
658 
659 	smb_slist_constructor(&session->s_req_list, sizeof (smb_request_t),
660 	    offsetof(smb_request_t, sr_session_lnd));
661 
662 	smb_llist_constructor(&session->s_user_list, sizeof (smb_user_t),
663 	    offsetof(smb_user_t, u_lnd));
664 
665 	smb_llist_constructor(&session->s_xa_list, sizeof (smb_xa_t),
666 	    offsetof(smb_xa_t, xa_lnd));
667 
668 	smb_net_txl_constructor(&session->s_txlst);
669 
670 	smb_rwx_init(&session->s_lock);
671 
672 	if (new_so) {
673 		bcopy(new_so->so_faddr_sa, &sin, new_so->so_faddr_len);
674 		ipaddr = sin.sin_addr.s_addr;
675 		bcopy(new_so->so_laddr_sa, &sin, new_so->so_faddr_len);
676 		local_ipaddr = sin.sin_addr.s_addr;
677 		session->s_local_port = port;
678 		session->ipaddr = ipaddr;
679 		session->local_ipaddr = local_ipaddr;
680 		session->sock = new_so;
681 	}
682 
683 	session->s_server = sv;
684 	smb_server_get_cfg(sv, &session->s_cfg);
685 	session->s_cache_request = sv->si_cache_request;
686 	session->s_cache = sv->si_cache_session;
687 	session->s_magic = SMB_SESSION_MAGIC;
688 	return (session);
689 }
690 
691 void
692 smb_session_delete(smb_session_t *session)
693 {
694 	ASSERT(session->s_magic == SMB_SESSION_MAGIC);
695 
696 	session->s_magic = (uint32_t)~SMB_SESSION_MAGIC;
697 
698 	smb_rwx_destroy(&session->s_lock);
699 	smb_net_txl_destructor(&session->s_txlst);
700 	smb_slist_destructor(&session->s_req_list);
701 	smb_llist_destructor(&session->s_user_list);
702 	smb_llist_destructor(&session->s_xa_list);
703 
704 	ASSERT(session->s_tree_cnt == 0);
705 	ASSERT(session->s_file_cnt == 0);
706 	ASSERT(session->s_dir_cnt == 0);
707 
708 	smb_idpool_destructor(&session->s_uid_pool);
709 	kmem_cache_free(session->s_cache, session);
710 }
711 
712 void
713 smb_session_cancel(smb_session_t *session)
714 {
715 	smb_xa_t	*xa, *nextxa;
716 
717 	/* All the request currently being treated must be canceled. */
718 	smb_session_cancel_requests(session, NULL, NULL);
719 
720 	/*
721 	 * We wait for the completion of all the requests associated with
722 	 * this session.
723 	 */
724 	smb_slist_wait_for_empty(&session->s_req_list);
725 
726 	/*
727 	 * At this point the reference count of the users, trees, files,
728 	 * directories should be zero. It should be possible to destroy them
729 	 * without any problem.
730 	 */
731 	xa = smb_llist_head(&session->s_xa_list);
732 	while (xa) {
733 		nextxa = smb_llist_next(&session->s_xa_list, xa);
734 		smb_xa_close(xa);
735 		xa = nextxa;
736 	}
737 	smb_user_logoff_all(session);
738 }
739 
740 /*
741  * Cancel requests.  If a non-null tree is specified, only requests specific
742  * to that tree will be cancelled.  If a non-null sr is specified, that sr
743  * will be not be cancelled - this would typically be the caller's sr.
744  */
745 void
746 smb_session_cancel_requests(
747     smb_session_t	*session,
748     smb_tree_t		*tree,
749     smb_request_t	*exclude_sr)
750 {
751 	smb_request_t	*sr;
752 
753 	smb_process_session_notify_change_queue(session, tree);
754 
755 	smb_slist_enter(&session->s_req_list);
756 	sr = smb_slist_head(&session->s_req_list);
757 
758 	while (sr) {
759 		ASSERT(sr->sr_magic == SMB_REQ_MAGIC);
760 		if ((sr != exclude_sr) &&
761 		    (tree == NULL || sr->tid_tree == tree))
762 			smb_request_cancel(sr);
763 
764 		sr = smb_slist_next(&session->s_req_list, sr);
765 	}
766 
767 	smb_slist_exit(&session->s_req_list);
768 }
769 
770 void
771 smb_session_worker(
772     void	*arg)
773 {
774 	smb_request_t	*sr;
775 
776 	sr = (smb_request_t *)arg;
777 
778 	ASSERT(sr->sr_magic == SMB_REQ_MAGIC);
779 
780 
781 	mutex_enter(&sr->sr_mutex);
782 	switch (sr->sr_state) {
783 	case SMB_REQ_STATE_SUBMITTED:
784 		mutex_exit(&sr->sr_mutex);
785 		smb_dispatch_request(sr);
786 		mutex_enter(&sr->sr_mutex);
787 		if (!sr->sr_keep) {
788 			sr->sr_state = SMB_REQ_STATE_COMPLETED;
789 			mutex_exit(&sr->sr_mutex);
790 			smb_request_free(sr);
791 			break;
792 		}
793 		mutex_exit(&sr->sr_mutex);
794 		break;
795 
796 	default:
797 		ASSERT(sr->sr_state == SMB_REQ_STATE_CANCELED);
798 		sr->sr_state = SMB_REQ_STATE_COMPLETED;
799 		mutex_exit(&sr->sr_mutex);
800 		smb_request_free(sr);
801 		break;
802 	}
803 }
804 
805 /*
806  * smb_session_disconnect_share
807  *
808  * Disconnects the specified share. This function should be called after the
809  * share passed in has been made unavailable by the "share manager".
810  */
811 void
812 smb_session_disconnect_share(smb_session_list_t *se, char *sharename)
813 {
814 	smb_session_t	*session;
815 
816 	rw_enter(&se->se_lock, RW_READER);
817 	session = list_head(&se->se_act.lst);
818 	while (session) {
819 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
820 		smb_rwx_rwenter(&session->s_lock, RW_READER);
821 		switch (session->s_state) {
822 		case SMB_SESSION_STATE_NEGOTIATED:
823 		case SMB_SESSION_STATE_OPLOCK_BREAKING:
824 		case SMB_SESSION_STATE_WRITE_RAW_ACTIVE: {
825 			smb_user_t	*user;
826 			smb_user_t	*next;
827 
828 			user = smb_user_lookup_by_state(session, NULL);
829 			while (user) {
830 				smb_user_disconnect_share(user, sharename);
831 				next = smb_user_lookup_by_state(session, user);
832 				smb_user_release(user);
833 				user = next;
834 			}
835 			break;
836 
837 		}
838 		default:
839 			break;
840 		}
841 		smb_rwx_rwexit(&session->s_lock);
842 		session = list_next(&se->se_act.lst, session);
843 	}
844 	rw_exit(&se->se_lock);
845 }
846 
847 void
848 smb_session_list_constructor(smb_session_list_t *se)
849 {
850 	bzero(se, sizeof (*se));
851 	rw_init(&se->se_lock, NULL, RW_DEFAULT, NULL);
852 	list_create(&se->se_rdy.lst, sizeof (smb_session_t),
853 	    offsetof(smb_session_t, s_lnd));
854 	list_create(&se->se_act.lst, sizeof (smb_session_t),
855 	    offsetof(smb_session_t, s_lnd));
856 }
857 
858 void
859 smb_session_list_destructor(smb_session_list_t *se)
860 {
861 	list_destroy(&se->se_rdy.lst);
862 	list_destroy(&se->se_act.lst);
863 	rw_destroy(&se->se_lock);
864 }
865 
866 void
867 smb_session_list_append(smb_session_list_t *se, smb_session_t *session)
868 {
869 	ASSERT(session->s_magic == SMB_SESSION_MAGIC);
870 	ASSERT(session->s_state == SMB_SESSION_STATE_INITIALIZED);
871 
872 	rw_enter(&se->se_lock, RW_WRITER);
873 	list_insert_tail(&se->se_rdy.lst, session);
874 	se->se_rdy.count++;
875 	se->se_wrop++;
876 	rw_exit(&se->se_lock);
877 }
878 
879 void
880 smb_session_list_delete_tail(smb_session_list_t *se)
881 {
882 	smb_session_t	*session;
883 
884 	rw_enter(&se->se_lock, RW_WRITER);
885 	session = list_tail(&se->se_rdy.lst);
886 	if (session) {
887 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
888 		ASSERT(session->s_state == SMB_SESSION_STATE_INITIALIZED);
889 		list_remove(&se->se_rdy.lst, session);
890 		ASSERT(se->se_rdy.count);
891 		se->se_rdy.count--;
892 		rw_exit(&se->se_lock);
893 		smb_session_delete(session);
894 		return;
895 	}
896 	rw_exit(&se->se_lock);
897 }
898 
899 smb_session_t *
900 smb_session_list_activate_head(smb_session_list_t *se)
901 {
902 	smb_session_t	*session;
903 
904 	rw_enter(&se->se_lock, RW_WRITER);
905 	session = list_head(&se->se_rdy.lst);
906 	if (session) {
907 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
908 		smb_rwx_rwenter(&session->s_lock, RW_WRITER);
909 		ASSERT(session->s_state == SMB_SESSION_STATE_INITIALIZED);
910 		session->s_thread = curthread;
911 		session->s_ktdid = session->s_thread->t_did;
912 		smb_rwx_rwexit(&session->s_lock);
913 		list_remove(&se->se_rdy.lst, session);
914 		se->se_rdy.count--;
915 		list_insert_tail(&se->se_act.lst, session);
916 		se->se_act.count++;
917 		se->se_wrop++;
918 	}
919 	rw_exit(&se->se_lock);
920 	return (session);
921 }
922 
923 void
924 smb_session_list_terminate(smb_session_list_t *se, smb_session_t *session)
925 {
926 	ASSERT(session->s_magic == SMB_SESSION_MAGIC);
927 
928 	rw_enter(&se->se_lock, RW_WRITER);
929 
930 	smb_rwx_rwenter(&session->s_lock, RW_WRITER);
931 	ASSERT(session->s_state == SMB_SESSION_STATE_DISCONNECTED);
932 	session->s_state = SMB_SESSION_STATE_TERMINATED;
933 	smb_sodestroy(session->sock);
934 	session->sock = NULL;
935 	smb_rwx_rwexit(&session->s_lock);
936 
937 	list_remove(&se->se_act.lst, session);
938 	se->se_act.count--;
939 	se->se_wrop++;
940 
941 	ASSERT(session->s_thread == curthread);
942 
943 	rw_exit(&se->se_lock);
944 
945 	smb_session_delete(session);
946 }
947 
948 /*
949  * smb_session_list_signal
950  *
951  * This function signals all the session threads. The intent is to terminate
952  * them. The sessions still in the SMB_SESSION_STATE_INITIALIZED are delete
953  * immediately.
954  *
955  * This function must only be called by the threads listening and accepting
956  * connections. They must pass in their respective session list.
957  */
958 void
959 smb_session_list_signal(smb_session_list_t *se)
960 {
961 	smb_session_t	*session;
962 
963 	rw_enter(&se->se_lock, RW_WRITER);
964 	while (session = list_head(&se->se_rdy.lst)) {
965 
966 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
967 
968 		smb_rwx_rwenter(&session->s_lock, RW_WRITER);
969 		ASSERT(session->s_state == SMB_SESSION_STATE_INITIALIZED);
970 		session->s_state = SMB_SESSION_STATE_TERMINATED;
971 		smb_sodestroy(session->sock);
972 		session->sock = NULL;
973 		smb_rwx_rwexit(&session->s_lock);
974 
975 		list_remove(&se->se_rdy.lst, session);
976 		se->se_rdy.count--;
977 		se->se_wrop++;
978 
979 		rw_exit(&se->se_lock);
980 		smb_session_delete(session);
981 		rw_enter(&se->se_lock, RW_WRITER);
982 	}
983 	rw_downgrade(&se->se_lock);
984 
985 	session = list_head(&se->se_act.lst);
986 	while (session) {
987 
988 		ASSERT(session->s_magic == SMB_SESSION_MAGIC);
989 		tsignal(session->s_thread, SIGINT);
990 		session = list_next(&se->se_act.lst, session);
991 	}
992 	rw_exit(&se->se_lock);
993 }
994 
995 /*
996  * smb_request_alloc
997  *
998  * Allocate an smb_request_t structure from the kmem_cache.  Partially
999  * initialize the found/new request.
1000  *
1001  * Returns pointer to a request
1002  */
1003 smb_request_t *
1004 smb_request_alloc(smb_session_t *session, int req_length)
1005 {
1006 	smb_request_t	*sr;
1007 
1008 	ASSERT(session->s_magic == SMB_SESSION_MAGIC);
1009 
1010 	sr = kmem_cache_alloc(session->s_cache_request, KM_SLEEP);
1011 
1012 	/*
1013 	 * Future:  Use constructor to pre-initialize some fields.  For now
1014 	 * there are so many fields that it is easiest just to zero the
1015 	 * whole thing and start over.
1016 	 */
1017 	bzero(sr, sizeof (smb_request_t));
1018 
1019 	mutex_init(&sr->sr_mutex, NULL, MUTEX_DEFAULT, NULL);
1020 	sr->session = session;
1021 	sr->sr_server = session->s_server;
1022 	sr->sr_gmtoff = session->s_server->si_gmtoff;
1023 	sr->sr_cache = session->s_server->si_cache_request;
1024 	sr->sr_cfg = &session->s_cfg;
1025 	sr->request_storage.forw = &sr->request_storage;
1026 	sr->request_storage.back = &sr->request_storage;
1027 	sr->command.max_bytes = req_length;
1028 	sr->reply.max_bytes = smb_maxbufsize;
1029 	sr->sr_req_length = req_length;
1030 	if (req_length)
1031 		sr->sr_request_buf = kmem_alloc(req_length, KM_SLEEP);
1032 	sr->sr_magic = SMB_REQ_MAGIC;
1033 	sr->sr_state = SMB_REQ_STATE_INITIALIZING;
1034 	smb_slist_insert_tail(&session->s_req_list, sr);
1035 	return (sr);
1036 }
1037 
1038 /*
1039  * smb_request_free
1040  *
1041  * release the memories which have been allocated for a smb request.
1042  */
1043 void
1044 smb_request_free(smb_request_t *sr)
1045 {
1046 	ASSERT(sr->sr_magic == SMB_REQ_MAGIC);
1047 	ASSERT(sr->session);
1048 	ASSERT(sr->fid_ofile == NULL);
1049 	ASSERT(sr->sid_odir == NULL);
1050 	ASSERT(sr->r_xa == NULL);
1051 
1052 	if (sr->tid_tree)
1053 		smb_tree_release(sr->tid_tree);
1054 
1055 	if (sr->uid_user)
1056 		smb_user_release(sr->uid_user);
1057 
1058 	smb_slist_remove(&sr->session->s_req_list, sr);
1059 
1060 	sr->session = NULL;
1061 
1062 	/* Release any temp storage */
1063 	smbsr_free_malloc_list(&sr->request_storage);
1064 
1065 	if (sr->sr_request_buf)
1066 		kmem_free(sr->sr_request_buf, sr->sr_req_length);
1067 	if (sr->command.chain)
1068 		m_freem(sr->command.chain);
1069 	if (sr->reply.chain)
1070 		m_freem(sr->reply.chain);
1071 	if (sr->raw_data.chain)
1072 		m_freem(sr->raw_data.chain);
1073 
1074 	sr->sr_magic = 0;
1075 	mutex_destroy(&sr->sr_mutex);
1076 	kmem_cache_free(sr->sr_cache, sr);
1077 }
1078