xref: /freebsd/sys/netsmb/smb_trantcp.c (revision a854ed98931b2e365eddd24cd2a1bb53a3f1828f)
1681a5bbeSBoris Popov /*
2681a5bbeSBoris Popov  * Copyright (c) 2000-2001 Boris Popov
3681a5bbeSBoris Popov  * All rights reserved.
4681a5bbeSBoris Popov  *
5681a5bbeSBoris Popov  * Redistribution and use in source and binary forms, with or without
6681a5bbeSBoris Popov  * modification, are permitted provided that the following conditions
7681a5bbeSBoris Popov  * are met:
8681a5bbeSBoris Popov  * 1. Redistributions of source code must retain the above copyright
9681a5bbeSBoris Popov  *    notice, this list of conditions and the following disclaimer.
10681a5bbeSBoris Popov  * 2. Redistributions in binary form must reproduce the above copyright
11681a5bbeSBoris Popov  *    notice, this list of conditions and the following disclaimer in the
12681a5bbeSBoris Popov  *    documentation and/or other materials provided with the distribution.
13681a5bbeSBoris Popov  * 3. All advertising materials mentioning features or use of this software
14681a5bbeSBoris Popov  *    must display the following acknowledgement:
15681a5bbeSBoris Popov  *    This product includes software developed by Boris Popov.
16681a5bbeSBoris Popov  * 4. Neither the name of the author nor the names of any co-contributors
17681a5bbeSBoris Popov  *    may be used to endorse or promote products derived from this software
18681a5bbeSBoris Popov  *    without specific prior written permission.
19681a5bbeSBoris Popov  *
20681a5bbeSBoris Popov  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21681a5bbeSBoris Popov  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22681a5bbeSBoris Popov  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23681a5bbeSBoris Popov  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24681a5bbeSBoris Popov  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25681a5bbeSBoris Popov  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26681a5bbeSBoris Popov  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27681a5bbeSBoris Popov  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28681a5bbeSBoris Popov  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29681a5bbeSBoris Popov  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30681a5bbeSBoris Popov  * SUCH DAMAGE.
31681a5bbeSBoris Popov  *
32681a5bbeSBoris Popov  * $FreeBSD$
33681a5bbeSBoris Popov  */
34681a5bbeSBoris Popov #include <sys/param.h>
35681a5bbeSBoris Popov #include <sys/systm.h>
36681a5bbeSBoris Popov #include <sys/kernel.h>
37681a5bbeSBoris Popov #include <sys/malloc.h>
38681a5bbeSBoris Popov #include <sys/mbuf.h>
39681a5bbeSBoris Popov #include <sys/proc.h>
40681a5bbeSBoris Popov #include <sys/protosw.h>
41681a5bbeSBoris Popov #include <sys/socket.h>
42681a5bbeSBoris Popov #include <sys/socketvar.h>
43681a5bbeSBoris Popov #include <sys/poll.h>
44681a5bbeSBoris Popov #include <sys/uio.h>
45681a5bbeSBoris Popov #include <sys/sysctl.h>
46265fc98fSSeigo Tanimura #include <sys/condvar.h>
47681a5bbeSBoris Popov 
48681a5bbeSBoris Popov #include <net/if.h>
49681a5bbeSBoris Popov #include <net/route.h>
50681a5bbeSBoris Popov 
51681a5bbeSBoris Popov #include <netinet/in.h>
52681a5bbeSBoris Popov #include <netinet/tcp.h>
53681a5bbeSBoris Popov 
54681a5bbeSBoris Popov #include <sys/mchain.h>
55681a5bbeSBoris Popov 
56681a5bbeSBoris Popov #include <netsmb/netbios.h>
57681a5bbeSBoris Popov 
58681a5bbeSBoris Popov #include <netsmb/smb.h>
59681a5bbeSBoris Popov #include <netsmb/smb_conn.h>
60681a5bbeSBoris Popov #include <netsmb/smb_tran.h>
61681a5bbeSBoris Popov #include <netsmb/smb_trantcp.h>
62681a5bbeSBoris Popov #include <netsmb/smb_subr.h>
63681a5bbeSBoris Popov 
64681a5bbeSBoris Popov #define M_NBDATA	M_PCB
65681a5bbeSBoris Popov 
66681a5bbeSBoris Popov static int smb_tcpsndbuf = 10 * 1024;
67681a5bbeSBoris Popov static int smb_tcprcvbuf = 10 * 1024;
68681a5bbeSBoris Popov 
69681a5bbeSBoris Popov SYSCTL_DECL(_net_smb);
70681a5bbeSBoris Popov SYSCTL_INT(_net_smb, OID_AUTO, tcpsndbuf, CTLFLAG_RW, &smb_tcpsndbuf, 0, "");
71681a5bbeSBoris Popov SYSCTL_INT(_net_smb, OID_AUTO, tcprcvbuf, CTLFLAG_RW, &smb_tcprcvbuf, 0, "");
72681a5bbeSBoris Popov 
73fce6fbfaSBoris Popov #define nb_sosend(so,m,flags,td) (so)->so_proto->pr_usrreqs->pru_sosend( \
74fce6fbfaSBoris Popov 				    so, NULL, 0, m, 0, flags, td)
75681a5bbeSBoris Popov 
76681a5bbeSBoris Popov static int  nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
77fce6fbfaSBoris Popov 	u_int8_t *rpcodep, struct thread *td);
78fce6fbfaSBoris Popov static int  smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td);
79681a5bbeSBoris Popov 
80681a5bbeSBoris Popov static int
81681a5bbeSBoris Popov nb_setsockopt_int(struct socket *so, int level, int name, int val)
82681a5bbeSBoris Popov {
83681a5bbeSBoris Popov 	struct sockopt sopt;
84681a5bbeSBoris Popov 
85681a5bbeSBoris Popov 	bzero(&sopt, sizeof(sopt));
86681a5bbeSBoris Popov 	sopt.sopt_level = level;
87681a5bbeSBoris Popov 	sopt.sopt_name = name;
88681a5bbeSBoris Popov 	sopt.sopt_val = &val;
89681a5bbeSBoris Popov 	sopt.sopt_valsize = sizeof(val);
90681a5bbeSBoris Popov 	return sosetopt(so, &sopt);
91681a5bbeSBoris Popov }
92681a5bbeSBoris Popov 
93681a5bbeSBoris Popov static __inline int
94fce6fbfaSBoris Popov nb_poll(struct nbpcb *nbp, int events, struct thread *td)
95681a5bbeSBoris Popov {
96681a5bbeSBoris Popov 	return nbp->nbp_tso->so_proto->pr_usrreqs->pru_sopoll(nbp->nbp_tso,
97fce6fbfaSBoris Popov 	    events, NULL, td);
98681a5bbeSBoris Popov }
99681a5bbeSBoris Popov 
100681a5bbeSBoris Popov static int
101fce6fbfaSBoris Popov nbssn_rselect(struct nbpcb *nbp, struct timeval *tv, int events,
102fce6fbfaSBoris Popov 	struct thread *td)
103681a5bbeSBoris Popov {
104681a5bbeSBoris Popov 	struct timeval atv, rtv, ttv;
105fce6fbfaSBoris Popov 	struct proc *p;
106265fc98fSSeigo Tanimura 	int timo, error;
107681a5bbeSBoris Popov 
108681a5bbeSBoris Popov 	if (tv) {
109681a5bbeSBoris Popov 		atv = *tv;
110681a5bbeSBoris Popov 		if (itimerfix(&atv)) {
111681a5bbeSBoris Popov 			error = EINVAL;
112265fc98fSSeigo Tanimura 			goto done_noproclock;
113681a5bbeSBoris Popov 		}
114681a5bbeSBoris Popov 		getmicrouptime(&rtv);
115681a5bbeSBoris Popov 		timevaladd(&atv, &rtv);
116681a5bbeSBoris Popov 	}
117681a5bbeSBoris Popov 	timo = 0;
118fce6fbfaSBoris Popov 	p = td->td_proc;
119265fc98fSSeigo Tanimura 	PROC_LOCK(p);
120fce6fbfaSBoris Popov 	mtx_lock_spin(&sched_lock);
121fce6fbfaSBoris Popov 	td->td_flags |= TDF_SELECT;
122fce6fbfaSBoris Popov 	mtx_unlock_spin(&sched_lock);
1231b369704SSeigo Tanimura 	PROC_UNLOCK(p);
124fce6fbfaSBoris Popov 	error = nb_poll(nbp, events, td);
1251b369704SSeigo Tanimura 	PROC_LOCK(p);
126681a5bbeSBoris Popov 	if (error) {
127681a5bbeSBoris Popov 		error = 0;
128681a5bbeSBoris Popov 		goto done;
129681a5bbeSBoris Popov 	}
130681a5bbeSBoris Popov 	if (tv) {
131681a5bbeSBoris Popov 		getmicrouptime(&rtv);
1321b369704SSeigo Tanimura 		if (timevalcmp(&rtv, &atv, >=)) {
1331b369704SSeigo Tanimura 			/*
1341b369704SSeigo Tanimura 			 * An event of our interest may occur during locking a process.
135fc75194cSBoris Popov 			 * In order to avoid missing the event that occurred during locking
1361b369704SSeigo Tanimura 			 * the process, test P_SELECT and rescan file descriptors if
1371b369704SSeigo Tanimura 			 * necessary.
1381b369704SSeigo Tanimura 			 */
139fce6fbfaSBoris Popov 			mtx_lock_spin(&sched_lock);
140fce6fbfaSBoris Popov 			if ((td->td_flags & TDF_SELECT) == 0) {
141fce6fbfaSBoris Popov 				td->td_flags |= TDF_SELECT;
142fce6fbfaSBoris Popov 				mtx_unlock_spin(&sched_lock);
1431b369704SSeigo Tanimura 				PROC_UNLOCK(p);
144fce6fbfaSBoris Popov 				error = nb_poll(nbp, events, td);
1451b369704SSeigo Tanimura 				PROC_LOCK(p);
146fce6fbfaSBoris Popov 			} else
147fce6fbfaSBoris Popov 				mtx_unlock_spin(&sched_lock);
148681a5bbeSBoris Popov 			goto done;
1491b369704SSeigo Tanimura 		}
150681a5bbeSBoris Popov 		ttv = atv;
151681a5bbeSBoris Popov 		timevalsub(&ttv, &rtv);
152681a5bbeSBoris Popov 		timo = tvtohz(&ttv);
153681a5bbeSBoris Popov 	}
154fce6fbfaSBoris Popov 	mtx_lock_spin(&sched_lock);
155fce6fbfaSBoris Popov 	td->td_flags &= ~TDF_SELECT;
156fce6fbfaSBoris Popov 	mtx_unlock_spin(&sched_lock);
157265fc98fSSeigo Tanimura 	if (timo > 0)
158265fc98fSSeigo Tanimura 		error = cv_timedwait(&selwait, &p->p_mtx, timo);
159265fc98fSSeigo Tanimura 	else {
160265fc98fSSeigo Tanimura 		cv_wait(&selwait, &p->p_mtx);
161265fc98fSSeigo Tanimura 		error = 0;
162681a5bbeSBoris Popov 	}
163265fc98fSSeigo Tanimura 
164681a5bbeSBoris Popov done:
165fce6fbfaSBoris Popov 	mtx_lock_spin(&sched_lock);
166fce6fbfaSBoris Popov 	td->td_flags &= ~TDF_SELECT;
167fce6fbfaSBoris Popov 	mtx_unlock_spin(&sched_lock);
168265fc98fSSeigo Tanimura 	PROC_UNLOCK(p);
169fce6fbfaSBoris Popov 
170265fc98fSSeigo Tanimura done_noproclock:
171681a5bbeSBoris Popov 	if (error == ERESTART)
172681a5bbeSBoris Popov 		return 0;
173681a5bbeSBoris Popov 	return error;
174681a5bbeSBoris Popov }
175681a5bbeSBoris Popov 
176681a5bbeSBoris Popov static int
177681a5bbeSBoris Popov nb_intr(struct nbpcb *nbp, struct proc *p)
178681a5bbeSBoris Popov {
179681a5bbeSBoris Popov 	return 0;
180681a5bbeSBoris Popov }
181681a5bbeSBoris Popov 
182681a5bbeSBoris Popov static void
183681a5bbeSBoris Popov nb_upcall(struct socket *so, void *arg, int waitflag)
184681a5bbeSBoris Popov {
185681a5bbeSBoris Popov 	struct nbpcb *nbp = arg;
186681a5bbeSBoris Popov 
187681a5bbeSBoris Popov 	if (arg == NULL || nbp->nbp_selectid == NULL)
188681a5bbeSBoris Popov 		return;
189681a5bbeSBoris Popov 	wakeup(nbp->nbp_selectid);
190681a5bbeSBoris Popov }
191681a5bbeSBoris Popov 
192681a5bbeSBoris Popov static int
193681a5bbeSBoris Popov nb_sethdr(struct mbuf *m, u_int8_t type, u_int32_t len)
194681a5bbeSBoris Popov {
195681a5bbeSBoris Popov 	u_int32_t *p = mtod(m, u_int32_t *);
196681a5bbeSBoris Popov 
197681a5bbeSBoris Popov 	*p = htonl((len & 0x1FFFF) | (type << 24));
198681a5bbeSBoris Popov 	return 0;
199681a5bbeSBoris Popov }
200681a5bbeSBoris Popov 
201681a5bbeSBoris Popov static int
202681a5bbeSBoris Popov nb_put_name(struct mbchain *mbp, struct sockaddr_nb *snb)
203681a5bbeSBoris Popov {
204681a5bbeSBoris Popov 	int error;
205681a5bbeSBoris Popov 	u_char seglen, *cp;
206681a5bbeSBoris Popov 
207681a5bbeSBoris Popov 	cp = snb->snb_name;
208681a5bbeSBoris Popov 	if (*cp == 0)
209681a5bbeSBoris Popov 		return EINVAL;
210681a5bbeSBoris Popov 	NBDEBUG("[%s]\n", cp);
211681a5bbeSBoris Popov 	for (;;) {
212681a5bbeSBoris Popov 		seglen = (*cp) + 1;
213681a5bbeSBoris Popov 		error = mb_put_mem(mbp, cp, seglen, MB_MSYSTEM);
214681a5bbeSBoris Popov 		if (error)
215681a5bbeSBoris Popov 			return error;
216681a5bbeSBoris Popov 		if (seglen == 1)
217681a5bbeSBoris Popov 			break;
218681a5bbeSBoris Popov 		cp += seglen;
219681a5bbeSBoris Popov 	}
220681a5bbeSBoris Popov 	return 0;
221681a5bbeSBoris Popov }
222681a5bbeSBoris Popov 
223681a5bbeSBoris Popov static int
224fce6fbfaSBoris Popov nb_connect_in(struct nbpcb *nbp, struct sockaddr_in *to, struct thread *td)
225681a5bbeSBoris Popov {
226681a5bbeSBoris Popov 	struct socket *so;
227681a5bbeSBoris Popov 	int error, s;
228681a5bbeSBoris Popov 
2299c4d63daSRobert Watson 	error = socreate(AF_INET, &so, SOCK_STREAM, IPPROTO_TCP,
230a854ed98SJohn Baldwin 	    td->td_ucred, td);
231681a5bbeSBoris Popov 	if (error)
232681a5bbeSBoris Popov 		return error;
233681a5bbeSBoris Popov 	nbp->nbp_tso = so;
234681a5bbeSBoris Popov 	so->so_upcallarg = (caddr_t)nbp;
235681a5bbeSBoris Popov 	so->so_upcall = nb_upcall;
236681a5bbeSBoris Popov 	so->so_rcv.sb_flags |= SB_UPCALL;
237681a5bbeSBoris Popov 	so->so_rcv.sb_timeo = (5 * hz);
238681a5bbeSBoris Popov 	so->so_snd.sb_timeo = (5 * hz);
239681a5bbeSBoris Popov 	error = soreserve(so, nbp->nbp_sndbuf, nbp->nbp_rcvbuf);
240681a5bbeSBoris Popov 	if (error)
241681a5bbeSBoris Popov 		goto bad;
242681a5bbeSBoris Popov 	nb_setsockopt_int(so, SOL_SOCKET, SO_KEEPALIVE, 1);
243681a5bbeSBoris Popov 	nb_setsockopt_int(so, IPPROTO_TCP, TCP_NODELAY, 1);
244681a5bbeSBoris Popov 	so->so_rcv.sb_flags &= ~SB_NOINTR;
245681a5bbeSBoris Popov 	so->so_snd.sb_flags &= ~SB_NOINTR;
246fce6fbfaSBoris Popov 	error = soconnect(so, (struct sockaddr*)to, td);
247681a5bbeSBoris Popov 	if (error)
248681a5bbeSBoris Popov 		goto bad;
249681a5bbeSBoris Popov 	s = splnet();
250681a5bbeSBoris Popov 	while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
251681a5bbeSBoris Popov 		tsleep(&so->so_timeo, PSOCK, "nbcon", 2 * hz);
252681a5bbeSBoris Popov 		if ((so->so_state & SS_ISCONNECTING) && so->so_error == 0 &&
253fce6fbfaSBoris Popov 			(error = nb_intr(nbp, td->td_proc)) != 0) {
254681a5bbeSBoris Popov 			so->so_state &= ~SS_ISCONNECTING;
255681a5bbeSBoris Popov 			splx(s);
256681a5bbeSBoris Popov 			goto bad;
257681a5bbeSBoris Popov 		}
258681a5bbeSBoris Popov 	}
259681a5bbeSBoris Popov 	if (so->so_error) {
260681a5bbeSBoris Popov 		error = so->so_error;
261681a5bbeSBoris Popov 		so->so_error = 0;
262681a5bbeSBoris Popov 		splx(s);
263681a5bbeSBoris Popov 		goto bad;
264681a5bbeSBoris Popov 	}
265681a5bbeSBoris Popov 	splx(s);
266681a5bbeSBoris Popov 	return 0;
267681a5bbeSBoris Popov bad:
268fce6fbfaSBoris Popov 	smb_nbst_disconnect(nbp->nbp_vc, td);
269681a5bbeSBoris Popov 	return error;
270681a5bbeSBoris Popov }
271681a5bbeSBoris Popov 
272681a5bbeSBoris Popov static int
273fce6fbfaSBoris Popov nbssn_rq_request(struct nbpcb *nbp, struct thread *td)
274681a5bbeSBoris Popov {
275681a5bbeSBoris Popov 	struct mbchain mb, *mbp = &mb;
276681a5bbeSBoris Popov 	struct mdchain md, *mdp = &md;
277681a5bbeSBoris Popov 	struct mbuf *m0;
278681a5bbeSBoris Popov 	struct timeval tv;
279681a5bbeSBoris Popov 	struct sockaddr_in sin;
280681a5bbeSBoris Popov 	u_short port;
281681a5bbeSBoris Popov 	u_int8_t rpcode;
282681a5bbeSBoris Popov 	int error, rplen;
283681a5bbeSBoris Popov 
284681a5bbeSBoris Popov 	error = mb_init(mbp);
285681a5bbeSBoris Popov 	if (error)
286681a5bbeSBoris Popov 		return error;
287681a5bbeSBoris Popov 	mb_put_uint32le(mbp, 0);
288681a5bbeSBoris Popov 	nb_put_name(mbp, nbp->nbp_paddr);
289681a5bbeSBoris Popov 	nb_put_name(mbp, nbp->nbp_laddr);
290681a5bbeSBoris Popov 	nb_sethdr(mbp->mb_top, NB_SSN_REQUEST, mb_fixhdr(mbp) - 4);
291fce6fbfaSBoris Popov 	error = nb_sosend(nbp->nbp_tso, mbp->mb_top, 0, td);
292681a5bbeSBoris Popov 	if (!error) {
293681a5bbeSBoris Popov 		nbp->nbp_state = NBST_RQSENT;
294681a5bbeSBoris Popov 	}
295681a5bbeSBoris Popov 	mb_detach(mbp);
296681a5bbeSBoris Popov 	mb_done(mbp);
297681a5bbeSBoris Popov 	if (error)
298681a5bbeSBoris Popov 		return error;
299681a5bbeSBoris Popov 	TIMESPEC_TO_TIMEVAL(&tv, &nbp->nbp_timo);
300fce6fbfaSBoris Popov 	error = nbssn_rselect(nbp, &tv, POLLIN, td);
301681a5bbeSBoris Popov 	if (error == EWOULDBLOCK) {	/* Timeout */
302681a5bbeSBoris Popov 		NBDEBUG("initial request timeout\n");
303681a5bbeSBoris Popov 		return ETIMEDOUT;
304681a5bbeSBoris Popov 	}
305681a5bbeSBoris Popov 	if (error)			/* restart or interrupt */
306681a5bbeSBoris Popov 		return error;
307fce6fbfaSBoris Popov 	error = nbssn_recv(nbp, &m0, &rplen, &rpcode, td);
308681a5bbeSBoris Popov 	if (error) {
309681a5bbeSBoris Popov 		NBDEBUG("recv() error %d\n", error);
310681a5bbeSBoris Popov 		return error;
311681a5bbeSBoris Popov 	}
312681a5bbeSBoris Popov 	/*
313681a5bbeSBoris Popov 	 * Process NETBIOS reply
314681a5bbeSBoris Popov 	 */
315681a5bbeSBoris Popov 	if (m0)
316681a5bbeSBoris Popov 		md_initm(mdp, m0);
317681a5bbeSBoris Popov 	error = 0;
318681a5bbeSBoris Popov 	do {
319681a5bbeSBoris Popov 		if (rpcode == NB_SSN_POSRESP) {
320681a5bbeSBoris Popov 			nbp->nbp_state = NBST_SESSION;
321681a5bbeSBoris Popov 			nbp->nbp_flags |= NBF_CONNECTED;
322681a5bbeSBoris Popov 			break;
323681a5bbeSBoris Popov 		}
324681a5bbeSBoris Popov 		if (rpcode != NB_SSN_RTGRESP) {
325681a5bbeSBoris Popov 			error = ECONNABORTED;
326681a5bbeSBoris Popov 			break;
327681a5bbeSBoris Popov 		}
328681a5bbeSBoris Popov 		if (rplen != 6) {
329681a5bbeSBoris Popov 			error = ECONNABORTED;
330681a5bbeSBoris Popov 			break;
331681a5bbeSBoris Popov 		}
332681a5bbeSBoris Popov 		md_get_mem(mdp, (caddr_t)&sin.sin_addr, 4, MB_MSYSTEM);
333681a5bbeSBoris Popov 		md_get_uint16(mdp, &port);
334681a5bbeSBoris Popov 		sin.sin_port = port;
335681a5bbeSBoris Popov 		nbp->nbp_state = NBST_RETARGET;
336fce6fbfaSBoris Popov 		smb_nbst_disconnect(nbp->nbp_vc, td);
337fce6fbfaSBoris Popov 		error = nb_connect_in(nbp, &sin, td);
338681a5bbeSBoris Popov 		if (!error)
339fce6fbfaSBoris Popov 			error = nbssn_rq_request(nbp, td);
340681a5bbeSBoris Popov 		if (error) {
341fce6fbfaSBoris Popov 			smb_nbst_disconnect(nbp->nbp_vc, td);
342681a5bbeSBoris Popov 			break;
343681a5bbeSBoris Popov 		}
344681a5bbeSBoris Popov 	} while(0);
345681a5bbeSBoris Popov 	if (m0)
346681a5bbeSBoris Popov 		md_done(mdp);
347681a5bbeSBoris Popov 	return error;
348681a5bbeSBoris Popov }
349681a5bbeSBoris Popov 
350681a5bbeSBoris Popov static int
351681a5bbeSBoris Popov nbssn_recvhdr(struct nbpcb *nbp, int *lenp,
352fce6fbfaSBoris Popov 	u_int8_t *rpcodep, int flags, struct thread *td)
353681a5bbeSBoris Popov {
354681a5bbeSBoris Popov 	struct socket *so = nbp->nbp_tso;
355681a5bbeSBoris Popov 	struct uio auio;
356681a5bbeSBoris Popov 	struct iovec aio;
357681a5bbeSBoris Popov 	u_int32_t len;
358681a5bbeSBoris Popov 	int error;
359681a5bbeSBoris Popov 
360681a5bbeSBoris Popov 	aio.iov_base = (caddr_t)&len;
361681a5bbeSBoris Popov 	aio.iov_len = sizeof(len);
362681a5bbeSBoris Popov 	auio.uio_iov = &aio;
363681a5bbeSBoris Popov 	auio.uio_iovcnt = 1;
364681a5bbeSBoris Popov 	auio.uio_segflg = UIO_SYSSPACE;
365681a5bbeSBoris Popov 	auio.uio_rw = UIO_READ;
366681a5bbeSBoris Popov 	auio.uio_offset = 0;
367681a5bbeSBoris Popov 	auio.uio_resid = sizeof(len);
368fce6fbfaSBoris Popov 	auio.uio_td = td;
369681a5bbeSBoris Popov 	error = so->so_proto->pr_usrreqs->pru_soreceive
370681a5bbeSBoris Popov 	    (so, (struct sockaddr **)NULL, &auio,
371681a5bbeSBoris Popov 	    (struct mbuf **)NULL, (struct mbuf **)NULL, &flags);
372681a5bbeSBoris Popov 	if (error)
373681a5bbeSBoris Popov 		return error;
374681a5bbeSBoris Popov 	if (auio.uio_resid > 0) {
375681a5bbeSBoris Popov 		SMBSDEBUG("short reply\n");
376681a5bbeSBoris Popov 		return EPIPE;
377681a5bbeSBoris Popov 	}
378681a5bbeSBoris Popov 	len = ntohl(len);
379681a5bbeSBoris Popov 	*rpcodep = (len >> 24) & 0xFF;
380681a5bbeSBoris Popov 	len &= 0x1ffff;
381681a5bbeSBoris Popov 	if (len > SMB_MAXPKTLEN) {
382681a5bbeSBoris Popov 		SMBERROR("packet too long (%d)\n", len);
383681a5bbeSBoris Popov 		return EFBIG;
384681a5bbeSBoris Popov 	}
385681a5bbeSBoris Popov 	*lenp = len;
386681a5bbeSBoris Popov 	return 0;
387681a5bbeSBoris Popov }
388681a5bbeSBoris Popov 
389681a5bbeSBoris Popov static int
390681a5bbeSBoris Popov nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
391fce6fbfaSBoris Popov 	u_int8_t *rpcodep, struct thread *td)
392681a5bbeSBoris Popov {
393681a5bbeSBoris Popov 	struct socket *so = nbp->nbp_tso;
394681a5bbeSBoris Popov 	struct uio auio;
395681a5bbeSBoris Popov 	struct mbuf *m;
396681a5bbeSBoris Popov 	u_int8_t rpcode;
397681a5bbeSBoris Popov 	int len;
398681a5bbeSBoris Popov 	int error, rcvflg;
399681a5bbeSBoris Popov 
400681a5bbeSBoris Popov 	if (so == NULL)
401681a5bbeSBoris Popov 		return ENOTCONN;
402681a5bbeSBoris Popov 
403681a5bbeSBoris Popov 	if (mpp)
404681a5bbeSBoris Popov 		*mpp = NULL;
405681a5bbeSBoris Popov 	for(;;) {
406681a5bbeSBoris Popov 		m = NULL;
407fce6fbfaSBoris Popov 		error = nbssn_recvhdr(nbp, &len, &rpcode, MSG_DONTWAIT, td);
408681a5bbeSBoris Popov 		if (so->so_state &
409681a5bbeSBoris Popov 		    (SS_ISDISCONNECTING | SS_ISDISCONNECTED | SS_CANTRCVMORE)) {
410681a5bbeSBoris Popov 			nbp->nbp_state = NBST_CLOSED;
411681a5bbeSBoris Popov 			NBDEBUG("session closed by peer\n");
412681a5bbeSBoris Popov 			return ECONNRESET;
413681a5bbeSBoris Popov 		}
414681a5bbeSBoris Popov 		if (error)
415681a5bbeSBoris Popov 			return error;
416681a5bbeSBoris Popov 		if (len == 0 && nbp->nbp_state != NBST_SESSION)
417681a5bbeSBoris Popov 			break;
418681a5bbeSBoris Popov 		if (rpcode == NB_SSN_KEEPALIVE)
419681a5bbeSBoris Popov 			continue;
420681a5bbeSBoris Popov 		bzero(&auio, sizeof(auio));
421681a5bbeSBoris Popov 		auio.uio_resid = len;
422fce6fbfaSBoris Popov 		auio.uio_td = td;
423681a5bbeSBoris Popov 		do {
424681a5bbeSBoris Popov 			rcvflg = MSG_WAITALL;
425681a5bbeSBoris Popov 			error = so->so_proto->pr_usrreqs->pru_soreceive
426681a5bbeSBoris Popov 			    (so, (struct sockaddr **)NULL,
427681a5bbeSBoris Popov 			    &auio, &m, (struct mbuf **)NULL, &rcvflg);
428681a5bbeSBoris Popov 		} while (error == EWOULDBLOCK || error == EINTR ||
429681a5bbeSBoris Popov 				 error == ERESTART);
430681a5bbeSBoris Popov 		if (error)
431681a5bbeSBoris Popov 			break;
432681a5bbeSBoris Popov 		if (auio.uio_resid > 0) {
433681a5bbeSBoris Popov 			SMBERROR("packet is shorter than expected\n");
434681a5bbeSBoris Popov 			error = EPIPE;
435681a5bbeSBoris Popov 			break;
436681a5bbeSBoris Popov 		}
437681a5bbeSBoris Popov 		if (nbp->nbp_state == NBST_SESSION &&
438681a5bbeSBoris Popov 		    rpcode == NB_SSN_MESSAGE)
439681a5bbeSBoris Popov 			break;
440681a5bbeSBoris Popov 		NBDEBUG("non-session packet %x\n", rpcode);
441681a5bbeSBoris Popov 		if (m)
442681a5bbeSBoris Popov 			m_freem(m);
443681a5bbeSBoris Popov 	}
444681a5bbeSBoris Popov 	if (error) {
445681a5bbeSBoris Popov 		if (m)
446681a5bbeSBoris Popov 			m_freem(m);
447681a5bbeSBoris Popov 		return error;
448681a5bbeSBoris Popov 	}
449681a5bbeSBoris Popov 	if (mpp)
450681a5bbeSBoris Popov 		*mpp = m;
451681a5bbeSBoris Popov 	else
452681a5bbeSBoris Popov 		m_freem(m);
453681a5bbeSBoris Popov 	*lenp = len;
454681a5bbeSBoris Popov 	*rpcodep = rpcode;
455681a5bbeSBoris Popov 	return 0;
456681a5bbeSBoris Popov }
457681a5bbeSBoris Popov 
458681a5bbeSBoris Popov /*
459681a5bbeSBoris Popov  * SMB transport interface
460681a5bbeSBoris Popov  */
461681a5bbeSBoris Popov static int
462fce6fbfaSBoris Popov smb_nbst_create(struct smb_vc *vcp, struct thread *td)
463681a5bbeSBoris Popov {
464681a5bbeSBoris Popov 	struct nbpcb *nbp;
465681a5bbeSBoris Popov 
466681a5bbeSBoris Popov 	MALLOC(nbp, struct nbpcb *, sizeof *nbp, M_NBDATA, M_WAITOK);
467681a5bbeSBoris Popov 	bzero(nbp, sizeof *nbp);
468681a5bbeSBoris Popov 	nbp->nbp_timo.tv_sec = 15;	/* XXX: sysctl ? */
469681a5bbeSBoris Popov 	nbp->nbp_state = NBST_CLOSED;
470681a5bbeSBoris Popov 	nbp->nbp_vc = vcp;
471681a5bbeSBoris Popov 	nbp->nbp_sndbuf = smb_tcpsndbuf;
472681a5bbeSBoris Popov 	nbp->nbp_rcvbuf = smb_tcprcvbuf;
473681a5bbeSBoris Popov 	vcp->vc_tdata = nbp;
474681a5bbeSBoris Popov 	return 0;
475681a5bbeSBoris Popov }
476681a5bbeSBoris Popov 
477681a5bbeSBoris Popov static int
478fce6fbfaSBoris Popov smb_nbst_done(struct smb_vc *vcp, struct thread *td)
479681a5bbeSBoris Popov {
480681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
481681a5bbeSBoris Popov 
482681a5bbeSBoris Popov 	if (nbp == NULL)
483681a5bbeSBoris Popov 		return ENOTCONN;
484fce6fbfaSBoris Popov 	smb_nbst_disconnect(vcp, td);
485681a5bbeSBoris Popov 	if (nbp->nbp_laddr)
486681a5bbeSBoris Popov 		free(nbp->nbp_laddr, M_SONAME);
487681a5bbeSBoris Popov 	if (nbp->nbp_paddr)
488681a5bbeSBoris Popov 		free(nbp->nbp_paddr, M_SONAME);
489681a5bbeSBoris Popov 	free(nbp, M_NBDATA);
490681a5bbeSBoris Popov 	return 0;
491681a5bbeSBoris Popov }
492681a5bbeSBoris Popov 
493681a5bbeSBoris Popov static int
494fce6fbfaSBoris Popov smb_nbst_bind(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
495681a5bbeSBoris Popov {
496681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
497681a5bbeSBoris Popov 	struct sockaddr_nb *snb;
498681a5bbeSBoris Popov 	int error, slen;
499681a5bbeSBoris Popov 
500681a5bbeSBoris Popov 	NBDEBUG("\n");
501681a5bbeSBoris Popov 	error = EINVAL;
502681a5bbeSBoris Popov 	do {
503681a5bbeSBoris Popov 		if (nbp->nbp_flags & NBF_LOCADDR)
504681a5bbeSBoris Popov 			break;
505681a5bbeSBoris Popov 		/*
506681a5bbeSBoris Popov 		 * It is possible to create NETBIOS name in the kernel,
507681a5bbeSBoris Popov 		 * but nothing prevents us to do it in the user space.
508681a5bbeSBoris Popov 		 */
509681a5bbeSBoris Popov 		if (sap == NULL)
510681a5bbeSBoris Popov 			break;
511681a5bbeSBoris Popov 		slen = sap->sa_len;
512681a5bbeSBoris Popov 		if (slen < NB_MINSALEN)
513681a5bbeSBoris Popov 			break;
514681a5bbeSBoris Popov 		snb = (struct sockaddr_nb*)dup_sockaddr(sap, 1);
515681a5bbeSBoris Popov 		if (snb == NULL) {
516681a5bbeSBoris Popov 			error = ENOMEM;
517681a5bbeSBoris Popov 			break;
518681a5bbeSBoris Popov 		}
519681a5bbeSBoris Popov 		nbp->nbp_laddr = snb;
520681a5bbeSBoris Popov 		nbp->nbp_flags |= NBF_LOCADDR;
521681a5bbeSBoris Popov 		error = 0;
522681a5bbeSBoris Popov 	} while(0);
523681a5bbeSBoris Popov 	return error;
524681a5bbeSBoris Popov }
525681a5bbeSBoris Popov 
526681a5bbeSBoris Popov static int
527fce6fbfaSBoris Popov smb_nbst_connect(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
528681a5bbeSBoris Popov {
529681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
530681a5bbeSBoris Popov 	struct sockaddr_in sin;
531681a5bbeSBoris Popov 	struct sockaddr_nb *snb;
532681a5bbeSBoris Popov 	struct timespec ts1, ts2;
533681a5bbeSBoris Popov 	int error, slen;
534681a5bbeSBoris Popov 
535681a5bbeSBoris Popov 	NBDEBUG("\n");
536681a5bbeSBoris Popov 	if (nbp->nbp_tso != NULL)
537681a5bbeSBoris Popov 		return EISCONN;
538681a5bbeSBoris Popov 	if (nbp->nbp_laddr == NULL)
539681a5bbeSBoris Popov 		return EINVAL;
540681a5bbeSBoris Popov 	slen = sap->sa_len;
541681a5bbeSBoris Popov 	if (slen < NB_MINSALEN)
542681a5bbeSBoris Popov 		return EINVAL;
543681a5bbeSBoris Popov 	if (nbp->nbp_paddr) {
544681a5bbeSBoris Popov 		free(nbp->nbp_paddr, M_SONAME);
545681a5bbeSBoris Popov 		nbp->nbp_paddr = NULL;
546681a5bbeSBoris Popov 	}
547681a5bbeSBoris Popov 	snb = (struct sockaddr_nb*)dup_sockaddr(sap, 1);
548681a5bbeSBoris Popov 	if (snb == NULL)
549681a5bbeSBoris Popov 		return ENOMEM;
550681a5bbeSBoris Popov 	nbp->nbp_paddr = snb;
551681a5bbeSBoris Popov 	sin = snb->snb_addrin;
552681a5bbeSBoris Popov 	getnanotime(&ts1);
553fce6fbfaSBoris Popov 	error = nb_connect_in(nbp, &sin, td);
554681a5bbeSBoris Popov 	if (error)
555681a5bbeSBoris Popov 		return error;
556681a5bbeSBoris Popov 	getnanotime(&ts2);
557681a5bbeSBoris Popov 	timespecsub(&ts2, &ts1);
558681a5bbeSBoris Popov 	if (ts2.tv_sec == 0 && ts2.tv_sec == 0)
559681a5bbeSBoris Popov 		ts2.tv_sec = 1;
560681a5bbeSBoris Popov 	nbp->nbp_timo = ts2;
561681a5bbeSBoris Popov 	timespecadd(&nbp->nbp_timo, &ts2);
562681a5bbeSBoris Popov 	timespecadd(&nbp->nbp_timo, &ts2);
563681a5bbeSBoris Popov 	timespecadd(&nbp->nbp_timo, &ts2);	/*  * 4 */
564fce6fbfaSBoris Popov 	error = nbssn_rq_request(nbp, td);
565681a5bbeSBoris Popov 	if (error)
566fce6fbfaSBoris Popov 		smb_nbst_disconnect(vcp, td);
567681a5bbeSBoris Popov 	return error;
568681a5bbeSBoris Popov }
569681a5bbeSBoris Popov 
570681a5bbeSBoris Popov static int
571fce6fbfaSBoris Popov smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td)
572681a5bbeSBoris Popov {
573681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
574681a5bbeSBoris Popov 	struct socket *so;
575681a5bbeSBoris Popov 
576681a5bbeSBoris Popov 	if (nbp == NULL || nbp->nbp_tso == NULL)
577681a5bbeSBoris Popov 		return ENOTCONN;
578681a5bbeSBoris Popov 	if ((so = nbp->nbp_tso) != NULL) {
579681a5bbeSBoris Popov 		nbp->nbp_flags &= ~NBF_CONNECTED;
580681a5bbeSBoris Popov 		nbp->nbp_tso = (struct socket *)NULL;
581681a5bbeSBoris Popov 		soshutdown(so, 2);
582681a5bbeSBoris Popov 		soclose(so);
583681a5bbeSBoris Popov 	}
584681a5bbeSBoris Popov 	if (nbp->nbp_state != NBST_RETARGET) {
585681a5bbeSBoris Popov 		nbp->nbp_state = NBST_CLOSED;
586681a5bbeSBoris Popov 	}
587681a5bbeSBoris Popov 	return 0;
588681a5bbeSBoris Popov }
589681a5bbeSBoris Popov 
590681a5bbeSBoris Popov static int
591fce6fbfaSBoris Popov smb_nbst_send(struct smb_vc *vcp, struct mbuf *m0, struct thread *td)
592681a5bbeSBoris Popov {
593681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
594681a5bbeSBoris Popov 	int error;
595681a5bbeSBoris Popov 
596681a5bbeSBoris Popov 	if (nbp->nbp_state != NBST_SESSION) {
597681a5bbeSBoris Popov 		error = ENOTCONN;
598681a5bbeSBoris Popov 		goto abort;
599681a5bbeSBoris Popov 	}
600681a5bbeSBoris Popov 	M_PREPEND(m0, 4, M_WAITOK);
601681a5bbeSBoris Popov 	if (m0 == NULL)
602681a5bbeSBoris Popov 		return ENOBUFS;
603681a5bbeSBoris Popov 	nb_sethdr(m0, NB_SSN_MESSAGE, m_fixhdr(m0) - 4);
604fce6fbfaSBoris Popov 	error = nb_sosend(nbp->nbp_tso, m0, 0, td);
605681a5bbeSBoris Popov 	return error;
606681a5bbeSBoris Popov abort:
607681a5bbeSBoris Popov 	if (m0)
608681a5bbeSBoris Popov 		m_freem(m0);
609681a5bbeSBoris Popov 	return error;
610681a5bbeSBoris Popov }
611681a5bbeSBoris Popov 
612681a5bbeSBoris Popov 
613681a5bbeSBoris Popov static int
614fce6fbfaSBoris Popov smb_nbst_recv(struct smb_vc *vcp, struct mbuf **mpp, struct thread *td)
615681a5bbeSBoris Popov {
616681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
617681a5bbeSBoris Popov 	u_int8_t rpcode;
618681a5bbeSBoris Popov 	int error, rplen;
619681a5bbeSBoris Popov 
620681a5bbeSBoris Popov 	nbp->nbp_flags |= NBF_RECVLOCK;
621fce6fbfaSBoris Popov 	error = nbssn_recv(nbp, mpp, &rplen, &rpcode, td);
622681a5bbeSBoris Popov 	nbp->nbp_flags &= ~NBF_RECVLOCK;
623681a5bbeSBoris Popov 	return error;
624681a5bbeSBoris Popov }
625681a5bbeSBoris Popov 
626681a5bbeSBoris Popov static void
627681a5bbeSBoris Popov smb_nbst_timo(struct smb_vc *vcp)
628681a5bbeSBoris Popov {
629681a5bbeSBoris Popov 	return;
630681a5bbeSBoris Popov }
631681a5bbeSBoris Popov 
632681a5bbeSBoris Popov static void
633681a5bbeSBoris Popov smb_nbst_intr(struct smb_vc *vcp)
634681a5bbeSBoris Popov {
635681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
636681a5bbeSBoris Popov 
637681a5bbeSBoris Popov 	if (nbp == NULL || nbp->nbp_tso == NULL)
638681a5bbeSBoris Popov 		return;
639681a5bbeSBoris Popov 	sorwakeup(nbp->nbp_tso);
640681a5bbeSBoris Popov 	sowwakeup(nbp->nbp_tso);
641681a5bbeSBoris Popov }
642681a5bbeSBoris Popov 
643681a5bbeSBoris Popov static int
644681a5bbeSBoris Popov smb_nbst_getparam(struct smb_vc *vcp, int param, void *data)
645681a5bbeSBoris Popov {
646681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
647681a5bbeSBoris Popov 
648681a5bbeSBoris Popov 	switch (param) {
649681a5bbeSBoris Popov 	    case SMBTP_SNDSZ:
650681a5bbeSBoris Popov 		*(int*)data = nbp->nbp_sndbuf;
651681a5bbeSBoris Popov 		break;
652681a5bbeSBoris Popov 	    case SMBTP_RCVSZ:
653681a5bbeSBoris Popov 		*(int*)data = nbp->nbp_rcvbuf;
654681a5bbeSBoris Popov 		break;
655681a5bbeSBoris Popov 	    case SMBTP_TIMEOUT:
656681a5bbeSBoris Popov 		*(struct timespec*)data = nbp->nbp_timo;
657681a5bbeSBoris Popov 		break;
658681a5bbeSBoris Popov 	    default:
659681a5bbeSBoris Popov 		return EINVAL;
660681a5bbeSBoris Popov 	}
661681a5bbeSBoris Popov 	return 0;
662681a5bbeSBoris Popov }
663681a5bbeSBoris Popov 
664681a5bbeSBoris Popov static int
665681a5bbeSBoris Popov smb_nbst_setparam(struct smb_vc *vcp, int param, void *data)
666681a5bbeSBoris Popov {
667681a5bbeSBoris Popov 	struct nbpcb *nbp = vcp->vc_tdata;
668681a5bbeSBoris Popov 
669681a5bbeSBoris Popov 	switch (param) {
670681a5bbeSBoris Popov 	    case SMBTP_SELECTID:
671681a5bbeSBoris Popov 		nbp->nbp_selectid = data;
672681a5bbeSBoris Popov 		break;
673681a5bbeSBoris Popov 	    default:
674681a5bbeSBoris Popov 		return EINVAL;
675681a5bbeSBoris Popov 	}
676681a5bbeSBoris Popov 	return 0;
677681a5bbeSBoris Popov }
678681a5bbeSBoris Popov 
679681a5bbeSBoris Popov /*
680681a5bbeSBoris Popov  * Check for fatal errors
681681a5bbeSBoris Popov  */
682681a5bbeSBoris Popov static int
683681a5bbeSBoris Popov smb_nbst_fatal(struct smb_vc *vcp, int error)
684681a5bbeSBoris Popov {
685681a5bbeSBoris Popov 	switch (error) {
686681a5bbeSBoris Popov 	    case ENOTCONN:
687681a5bbeSBoris Popov 	    case ENETRESET:
688681a5bbeSBoris Popov 	    case ECONNABORTED:
689681a5bbeSBoris Popov 		return 1;
690681a5bbeSBoris Popov 	}
691681a5bbeSBoris Popov 	return 0;
692681a5bbeSBoris Popov }
693681a5bbeSBoris Popov 
694681a5bbeSBoris Popov 
695681a5bbeSBoris Popov struct smb_tran_desc smb_tran_nbtcp_desc = {
696681a5bbeSBoris Popov 	SMBT_NBTCP,
697681a5bbeSBoris Popov 	smb_nbst_create, smb_nbst_done,
698681a5bbeSBoris Popov 	smb_nbst_bind, smb_nbst_connect, smb_nbst_disconnect,
699681a5bbeSBoris Popov 	smb_nbst_send, smb_nbst_recv,
700681a5bbeSBoris Popov 	smb_nbst_timo, smb_nbst_intr,
701681a5bbeSBoris Popov 	smb_nbst_getparam, smb_nbst_setparam,
702681a5bbeSBoris Popov 	smb_nbst_fatal
703681a5bbeSBoris Popov };
704681a5bbeSBoris Popov 
705