35
36#include <sys/param.h>
37#include <sys/condvar.h>
38#include <sys/kernel.h>
39#include <sys/lock.h>
40#include <sys/malloc.h>
41#include <sys/mbuf.h>
42#include <sys/poll.h>
43#include <sys/proc.h>
44#include <sys/protosw.h>
45#include <sys/signalvar.h>
46#include <sys/socket.h>
47#include <sys/socketvar.h>
48#include <sys/sx.h>
49#include <sys/sysctl.h>
50#include <sys/systm.h>
51#include <sys/uio.h>
52
53#include <net/if.h>
54#include <net/route.h>
55
56#include <netinet/in.h>
57#include <netinet/tcp.h>
58
59#include <sys/mchain.h>
60
61#include <netsmb/netbios.h>
62
63#include <netsmb/smb.h>
64#include <netsmb/smb_conn.h>
65#include <netsmb/smb_tran.h>
66#include <netsmb/smb_trantcp.h>
67#include <netsmb/smb_subr.h>
68
69#define M_NBDATA M_PCB
70
71static int smb_tcpsndbuf = NB_SNDQ - 1;
72static int smb_tcprcvbuf = NB_RCVQ - 1;
73
74SYSCTL_DECL(_net_smb);
75SYSCTL_INT(_net_smb, OID_AUTO, tcpsndbuf, CTLFLAG_RW, &smb_tcpsndbuf, 0, "");
76SYSCTL_INT(_net_smb, OID_AUTO, tcprcvbuf, CTLFLAG_RW, &smb_tcprcvbuf, 0, "");
77
78#define nb_sosend(so,m,flags,td) (so)->so_proto->pr_usrreqs->pru_sosend( \
79 so, NULL, 0, m, 0, flags, td)
80
81static int nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
82 u_int8_t *rpcodep, struct thread *td);
83static int smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td);
84
85static int
86nb_setsockopt_int(struct socket *so, int level, int name, int val)
87{
88 struct sockopt sopt;
89
90 bzero(&sopt, sizeof(sopt));
91 sopt.sopt_level = level;
92 sopt.sopt_name = name;
93 sopt.sopt_val = &val;
94 sopt.sopt_valsize = sizeof(val);
95 return sosetopt(so, &sopt);
96}
97
98static __inline int
99nb_poll(struct nbpcb *nbp, int events, struct thread *td)
100{
101 return nbp->nbp_tso->so_proto->pr_usrreqs->pru_sopoll(nbp->nbp_tso,
102 events, NULL, td);
103}
104
105static int
106nbssn_rselect(struct nbpcb *nbp, struct timeval *tv, int events,
107 struct thread *td)
108{
109 struct timeval atv, rtv, ttv;
110 int ncoll, timo, error;
111
112 if (tv) {
113 atv = *tv;
114 if (itimerfix(&atv)) {
115 error = EINVAL;
116 goto done_noproclock;
117 }
118 getmicrouptime(&rtv);
119 timevaladd(&atv, &rtv);
120 }
121 timo = 0;
122 mtx_lock(&sellock);
123retry:
124
125 ncoll = nselcoll;
126 mtx_lock_spin(&sched_lock);
127 td->td_flags |= TDF_SELECT;
128 mtx_unlock_spin(&sched_lock);
129 mtx_unlock(&sellock);
130
131 /* XXX: Should be done when the thread is initialized. */
132 TAILQ_INIT(&td->td_selq);
133 error = nb_poll(nbp, events, td);
134 mtx_lock(&sellock);
135 if (error) {
136 error = 0;
137 goto done;
138 }
139 if (tv) {
140 getmicrouptime(&rtv);
141 if (timevalcmp(&rtv, &atv, >=))
142 goto done;
143 ttv = atv;
144 timevalsub(&ttv, &rtv);
145 timo = tvtohz(&ttv);
146 }
147 /*
148 * An event of our interest may occur during locking a process.
149 * In order to avoid missing the event that occurred during locking
150 * the process, test P_SELECT and rescan file descriptors if
151 * necessary.
152 */
153 mtx_lock_spin(&sched_lock);
154 if ((td->td_flags & TDF_SELECT) == 0 || nselcoll != ncoll) {
155 mtx_unlock_spin(&sched_lock);
156 goto retry;
157 }
158 mtx_unlock_spin(&sched_lock);
159
160 if (timo > 0)
161 error = cv_timedwait(&selwait, &sellock, timo);
162 else {
163 cv_wait(&selwait, &sellock);
164 error = 0;
165 }
166
167done:
168 clear_selinfo_list(td);
169
170 mtx_lock_spin(&sched_lock);
171 td->td_flags &= ~TDF_SELECT;
172 mtx_unlock_spin(&sched_lock);
173 mtx_unlock(&sellock);
174
175done_noproclock:
176 if (error == ERESTART)
177 return 0;
178 return error;
179}
180
181static int
182nb_intr(struct nbpcb *nbp, struct proc *p)
183{
184 return 0;
185}
186
187static void
188nb_upcall(struct socket *so, void *arg, int waitflag)
189{
190 struct nbpcb *nbp = arg;
191
192 if (arg == NULL || nbp->nbp_selectid == NULL)
193 return;
194 wakeup(nbp->nbp_selectid);
195}
196
197static int
198nb_sethdr(struct mbuf *m, u_int8_t type, u_int32_t len)
199{
200 u_int32_t *p = mtod(m, u_int32_t *);
201
202 *p = htonl((len & 0x1FFFF) | (type << 24));
203 return 0;
204}
205
206static int
207nb_put_name(struct mbchain *mbp, struct sockaddr_nb *snb)
208{
209 int error;
210 u_char seglen, *cp;
211
212 cp = snb->snb_name;
213 if (*cp == 0)
214 return EINVAL;
215 NBDEBUG("[%s]\n", cp);
216 for (;;) {
217 seglen = (*cp) + 1;
218 error = mb_put_mem(mbp, cp, seglen, MB_MSYSTEM);
219 if (error)
220 return error;
221 if (seglen == 1)
222 break;
223 cp += seglen;
224 }
225 return 0;
226}
227
228static int
229nb_connect_in(struct nbpcb *nbp, struct sockaddr_in *to, struct thread *td)
230{
231 struct socket *so;
232 int error, s;
233
234 error = socreate(AF_INET, &so, SOCK_STREAM, IPPROTO_TCP,
235 td->td_ucred, td);
236 if (error)
237 return error;
238 nbp->nbp_tso = so;
239 so->so_upcallarg = (caddr_t)nbp;
240 so->so_upcall = nb_upcall;
241 so->so_rcv.sb_flags |= SB_UPCALL;
242 so->so_rcv.sb_timeo = (5 * hz);
243 so->so_snd.sb_timeo = (5 * hz);
244 error = soreserve(so, nbp->nbp_sndbuf, nbp->nbp_rcvbuf);
245 if (error)
246 goto bad;
247 nb_setsockopt_int(so, SOL_SOCKET, SO_KEEPALIVE, 1);
248 nb_setsockopt_int(so, IPPROTO_TCP, TCP_NODELAY, 1);
249 so->so_rcv.sb_flags &= ~SB_NOINTR;
250 so->so_snd.sb_flags &= ~SB_NOINTR;
251 error = soconnect(so, (struct sockaddr*)to, td);
252 if (error)
253 goto bad;
254 s = splnet();
255 while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
256 tsleep(&so->so_timeo, PSOCK, "nbcon", 2 * hz);
257 if ((so->so_state & SS_ISCONNECTING) && so->so_error == 0 &&
258 (error = nb_intr(nbp, td->td_proc)) != 0) {
259 so->so_state &= ~SS_ISCONNECTING;
260 splx(s);
261 goto bad;
262 }
263 }
264 if (so->so_error) {
265 error = so->so_error;
266 so->so_error = 0;
267 splx(s);
268 goto bad;
269 }
270 splx(s);
271 return 0;
272bad:
273 smb_nbst_disconnect(nbp->nbp_vc, td);
274 return error;
275}
276
277static int
278nbssn_rq_request(struct nbpcb *nbp, struct thread *td)
279{
280 struct mbchain mb, *mbp = &mb;
281 struct mdchain md, *mdp = &md;
282 struct mbuf *m0;
283 struct timeval tv;
284 struct sockaddr_in sin;
285 u_short port;
286 u_int8_t rpcode;
287 int error, rplen;
288
289 error = mb_init(mbp);
290 if (error)
291 return error;
292 mb_put_uint32le(mbp, 0);
293 nb_put_name(mbp, nbp->nbp_paddr);
294 nb_put_name(mbp, nbp->nbp_laddr);
295 nb_sethdr(mbp->mb_top, NB_SSN_REQUEST, mb_fixhdr(mbp) - 4);
296 error = nb_sosend(nbp->nbp_tso, mbp->mb_top, 0, td);
297 if (!error) {
298 nbp->nbp_state = NBST_RQSENT;
299 }
300 mb_detach(mbp);
301 mb_done(mbp);
302 if (error)
303 return error;
304 TIMESPEC_TO_TIMEVAL(&tv, &nbp->nbp_timo);
305 error = nbssn_rselect(nbp, &tv, POLLIN, td);
306 if (error == EWOULDBLOCK) { /* Timeout */
307 NBDEBUG("initial request timeout\n");
308 return ETIMEDOUT;
309 }
310 if (error) /* restart or interrupt */
311 return error;
312 error = nbssn_recv(nbp, &m0, &rplen, &rpcode, td);
313 if (error) {
314 NBDEBUG("recv() error %d\n", error);
315 return error;
316 }
317 /*
318 * Process NETBIOS reply
319 */
320 if (m0)
321 md_initm(mdp, m0);
322 error = 0;
323 do {
324 if (rpcode == NB_SSN_POSRESP) {
325 nbp->nbp_state = NBST_SESSION;
326 nbp->nbp_flags |= NBF_CONNECTED;
327 break;
328 }
329 if (rpcode != NB_SSN_RTGRESP) {
330 error = ECONNABORTED;
331 break;
332 }
333 if (rplen != 6) {
334 error = ECONNABORTED;
335 break;
336 }
337 md_get_mem(mdp, (caddr_t)&sin.sin_addr, 4, MB_MSYSTEM);
338 md_get_uint16(mdp, &port);
339 sin.sin_port = port;
340 nbp->nbp_state = NBST_RETARGET;
341 smb_nbst_disconnect(nbp->nbp_vc, td);
342 error = nb_connect_in(nbp, &sin, td);
343 if (!error)
344 error = nbssn_rq_request(nbp, td);
345 if (error) {
346 smb_nbst_disconnect(nbp->nbp_vc, td);
347 break;
348 }
349 } while(0);
350 if (m0)
351 md_done(mdp);
352 return error;
353}
354
355static int
356nbssn_recvhdr(struct nbpcb *nbp, int *lenp,
357 u_int8_t *rpcodep, int flags, struct thread *td)
358{
359 struct socket *so = nbp->nbp_tso;
360 struct uio auio;
361 struct iovec aio;
362 u_int32_t len;
363 int error;
364
365 aio.iov_base = (caddr_t)&len;
366 aio.iov_len = sizeof(len);
367 auio.uio_iov = &aio;
368 auio.uio_iovcnt = 1;
369 auio.uio_segflg = UIO_SYSSPACE;
370 auio.uio_rw = UIO_READ;
371 auio.uio_offset = 0;
372 auio.uio_resid = sizeof(len);
373 auio.uio_td = td;
374 error = so->so_proto->pr_usrreqs->pru_soreceive
375 (so, (struct sockaddr **)NULL, &auio,
376 (struct mbuf **)NULL, (struct mbuf **)NULL, &flags);
377 if (error)
378 return error;
379 if (auio.uio_resid > 0) {
380 SMBSDEBUG("short reply\n");
381 return EPIPE;
382 }
383 len = ntohl(len);
384 *rpcodep = (len >> 24) & 0xFF;
385 len &= 0x1ffff;
386 if (len > SMB_MAXPKTLEN) {
387 SMBERROR("packet too long (%d)\n", len);
388 return EFBIG;
389 }
390 *lenp = len;
391 return 0;
392}
393
394static int
395nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
396 u_int8_t *rpcodep, struct thread *td)
397{
398 struct socket *so = nbp->nbp_tso;
399 struct uio auio;
400 struct mbuf *m, *tm, *im;
401 u_int8_t rpcode;
402 int len, resid;
403 int error, rcvflg;
404
405 if (so == NULL)
406 return ENOTCONN;
407
408 if (mpp)
409 *mpp = NULL;
410 m = NULL;
411 for(;;) {
412 /*
413 * Poll for a response header.
414 * If we don't have one waiting, return.
415 */
416 error = nbssn_recvhdr(nbp, &len, &rpcode, MSG_DONTWAIT, td);
| 35
36#include <sys/param.h>
37#include <sys/condvar.h>
38#include <sys/kernel.h>
39#include <sys/lock.h>
40#include <sys/malloc.h>
41#include <sys/mbuf.h>
42#include <sys/poll.h>
43#include <sys/proc.h>
44#include <sys/protosw.h>
45#include <sys/signalvar.h>
46#include <sys/socket.h>
47#include <sys/socketvar.h>
48#include <sys/sx.h>
49#include <sys/sysctl.h>
50#include <sys/systm.h>
51#include <sys/uio.h>
52
53#include <net/if.h>
54#include <net/route.h>
55
56#include <netinet/in.h>
57#include <netinet/tcp.h>
58
59#include <sys/mchain.h>
60
61#include <netsmb/netbios.h>
62
63#include <netsmb/smb.h>
64#include <netsmb/smb_conn.h>
65#include <netsmb/smb_tran.h>
66#include <netsmb/smb_trantcp.h>
67#include <netsmb/smb_subr.h>
68
69#define M_NBDATA M_PCB
70
71static int smb_tcpsndbuf = NB_SNDQ - 1;
72static int smb_tcprcvbuf = NB_RCVQ - 1;
73
74SYSCTL_DECL(_net_smb);
75SYSCTL_INT(_net_smb, OID_AUTO, tcpsndbuf, CTLFLAG_RW, &smb_tcpsndbuf, 0, "");
76SYSCTL_INT(_net_smb, OID_AUTO, tcprcvbuf, CTLFLAG_RW, &smb_tcprcvbuf, 0, "");
77
78#define nb_sosend(so,m,flags,td) (so)->so_proto->pr_usrreqs->pru_sosend( \
79 so, NULL, 0, m, 0, flags, td)
80
81static int nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
82 u_int8_t *rpcodep, struct thread *td);
83static int smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td);
84
85static int
86nb_setsockopt_int(struct socket *so, int level, int name, int val)
87{
88 struct sockopt sopt;
89
90 bzero(&sopt, sizeof(sopt));
91 sopt.sopt_level = level;
92 sopt.sopt_name = name;
93 sopt.sopt_val = &val;
94 sopt.sopt_valsize = sizeof(val);
95 return sosetopt(so, &sopt);
96}
97
98static __inline int
99nb_poll(struct nbpcb *nbp, int events, struct thread *td)
100{
101 return nbp->nbp_tso->so_proto->pr_usrreqs->pru_sopoll(nbp->nbp_tso,
102 events, NULL, td);
103}
104
105static int
106nbssn_rselect(struct nbpcb *nbp, struct timeval *tv, int events,
107 struct thread *td)
108{
109 struct timeval atv, rtv, ttv;
110 int ncoll, timo, error;
111
112 if (tv) {
113 atv = *tv;
114 if (itimerfix(&atv)) {
115 error = EINVAL;
116 goto done_noproclock;
117 }
118 getmicrouptime(&rtv);
119 timevaladd(&atv, &rtv);
120 }
121 timo = 0;
122 mtx_lock(&sellock);
123retry:
124
125 ncoll = nselcoll;
126 mtx_lock_spin(&sched_lock);
127 td->td_flags |= TDF_SELECT;
128 mtx_unlock_spin(&sched_lock);
129 mtx_unlock(&sellock);
130
131 /* XXX: Should be done when the thread is initialized. */
132 TAILQ_INIT(&td->td_selq);
133 error = nb_poll(nbp, events, td);
134 mtx_lock(&sellock);
135 if (error) {
136 error = 0;
137 goto done;
138 }
139 if (tv) {
140 getmicrouptime(&rtv);
141 if (timevalcmp(&rtv, &atv, >=))
142 goto done;
143 ttv = atv;
144 timevalsub(&ttv, &rtv);
145 timo = tvtohz(&ttv);
146 }
147 /*
148 * An event of our interest may occur during locking a process.
149 * In order to avoid missing the event that occurred during locking
150 * the process, test P_SELECT and rescan file descriptors if
151 * necessary.
152 */
153 mtx_lock_spin(&sched_lock);
154 if ((td->td_flags & TDF_SELECT) == 0 || nselcoll != ncoll) {
155 mtx_unlock_spin(&sched_lock);
156 goto retry;
157 }
158 mtx_unlock_spin(&sched_lock);
159
160 if (timo > 0)
161 error = cv_timedwait(&selwait, &sellock, timo);
162 else {
163 cv_wait(&selwait, &sellock);
164 error = 0;
165 }
166
167done:
168 clear_selinfo_list(td);
169
170 mtx_lock_spin(&sched_lock);
171 td->td_flags &= ~TDF_SELECT;
172 mtx_unlock_spin(&sched_lock);
173 mtx_unlock(&sellock);
174
175done_noproclock:
176 if (error == ERESTART)
177 return 0;
178 return error;
179}
180
181static int
182nb_intr(struct nbpcb *nbp, struct proc *p)
183{
184 return 0;
185}
186
187static void
188nb_upcall(struct socket *so, void *arg, int waitflag)
189{
190 struct nbpcb *nbp = arg;
191
192 if (arg == NULL || nbp->nbp_selectid == NULL)
193 return;
194 wakeup(nbp->nbp_selectid);
195}
196
197static int
198nb_sethdr(struct mbuf *m, u_int8_t type, u_int32_t len)
199{
200 u_int32_t *p = mtod(m, u_int32_t *);
201
202 *p = htonl((len & 0x1FFFF) | (type << 24));
203 return 0;
204}
205
206static int
207nb_put_name(struct mbchain *mbp, struct sockaddr_nb *snb)
208{
209 int error;
210 u_char seglen, *cp;
211
212 cp = snb->snb_name;
213 if (*cp == 0)
214 return EINVAL;
215 NBDEBUG("[%s]\n", cp);
216 for (;;) {
217 seglen = (*cp) + 1;
218 error = mb_put_mem(mbp, cp, seglen, MB_MSYSTEM);
219 if (error)
220 return error;
221 if (seglen == 1)
222 break;
223 cp += seglen;
224 }
225 return 0;
226}
227
228static int
229nb_connect_in(struct nbpcb *nbp, struct sockaddr_in *to, struct thread *td)
230{
231 struct socket *so;
232 int error, s;
233
234 error = socreate(AF_INET, &so, SOCK_STREAM, IPPROTO_TCP,
235 td->td_ucred, td);
236 if (error)
237 return error;
238 nbp->nbp_tso = so;
239 so->so_upcallarg = (caddr_t)nbp;
240 so->so_upcall = nb_upcall;
241 so->so_rcv.sb_flags |= SB_UPCALL;
242 so->so_rcv.sb_timeo = (5 * hz);
243 so->so_snd.sb_timeo = (5 * hz);
244 error = soreserve(so, nbp->nbp_sndbuf, nbp->nbp_rcvbuf);
245 if (error)
246 goto bad;
247 nb_setsockopt_int(so, SOL_SOCKET, SO_KEEPALIVE, 1);
248 nb_setsockopt_int(so, IPPROTO_TCP, TCP_NODELAY, 1);
249 so->so_rcv.sb_flags &= ~SB_NOINTR;
250 so->so_snd.sb_flags &= ~SB_NOINTR;
251 error = soconnect(so, (struct sockaddr*)to, td);
252 if (error)
253 goto bad;
254 s = splnet();
255 while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
256 tsleep(&so->so_timeo, PSOCK, "nbcon", 2 * hz);
257 if ((so->so_state & SS_ISCONNECTING) && so->so_error == 0 &&
258 (error = nb_intr(nbp, td->td_proc)) != 0) {
259 so->so_state &= ~SS_ISCONNECTING;
260 splx(s);
261 goto bad;
262 }
263 }
264 if (so->so_error) {
265 error = so->so_error;
266 so->so_error = 0;
267 splx(s);
268 goto bad;
269 }
270 splx(s);
271 return 0;
272bad:
273 smb_nbst_disconnect(nbp->nbp_vc, td);
274 return error;
275}
276
277static int
278nbssn_rq_request(struct nbpcb *nbp, struct thread *td)
279{
280 struct mbchain mb, *mbp = &mb;
281 struct mdchain md, *mdp = &md;
282 struct mbuf *m0;
283 struct timeval tv;
284 struct sockaddr_in sin;
285 u_short port;
286 u_int8_t rpcode;
287 int error, rplen;
288
289 error = mb_init(mbp);
290 if (error)
291 return error;
292 mb_put_uint32le(mbp, 0);
293 nb_put_name(mbp, nbp->nbp_paddr);
294 nb_put_name(mbp, nbp->nbp_laddr);
295 nb_sethdr(mbp->mb_top, NB_SSN_REQUEST, mb_fixhdr(mbp) - 4);
296 error = nb_sosend(nbp->nbp_tso, mbp->mb_top, 0, td);
297 if (!error) {
298 nbp->nbp_state = NBST_RQSENT;
299 }
300 mb_detach(mbp);
301 mb_done(mbp);
302 if (error)
303 return error;
304 TIMESPEC_TO_TIMEVAL(&tv, &nbp->nbp_timo);
305 error = nbssn_rselect(nbp, &tv, POLLIN, td);
306 if (error == EWOULDBLOCK) { /* Timeout */
307 NBDEBUG("initial request timeout\n");
308 return ETIMEDOUT;
309 }
310 if (error) /* restart or interrupt */
311 return error;
312 error = nbssn_recv(nbp, &m0, &rplen, &rpcode, td);
313 if (error) {
314 NBDEBUG("recv() error %d\n", error);
315 return error;
316 }
317 /*
318 * Process NETBIOS reply
319 */
320 if (m0)
321 md_initm(mdp, m0);
322 error = 0;
323 do {
324 if (rpcode == NB_SSN_POSRESP) {
325 nbp->nbp_state = NBST_SESSION;
326 nbp->nbp_flags |= NBF_CONNECTED;
327 break;
328 }
329 if (rpcode != NB_SSN_RTGRESP) {
330 error = ECONNABORTED;
331 break;
332 }
333 if (rplen != 6) {
334 error = ECONNABORTED;
335 break;
336 }
337 md_get_mem(mdp, (caddr_t)&sin.sin_addr, 4, MB_MSYSTEM);
338 md_get_uint16(mdp, &port);
339 sin.sin_port = port;
340 nbp->nbp_state = NBST_RETARGET;
341 smb_nbst_disconnect(nbp->nbp_vc, td);
342 error = nb_connect_in(nbp, &sin, td);
343 if (!error)
344 error = nbssn_rq_request(nbp, td);
345 if (error) {
346 smb_nbst_disconnect(nbp->nbp_vc, td);
347 break;
348 }
349 } while(0);
350 if (m0)
351 md_done(mdp);
352 return error;
353}
354
355static int
356nbssn_recvhdr(struct nbpcb *nbp, int *lenp,
357 u_int8_t *rpcodep, int flags, struct thread *td)
358{
359 struct socket *so = nbp->nbp_tso;
360 struct uio auio;
361 struct iovec aio;
362 u_int32_t len;
363 int error;
364
365 aio.iov_base = (caddr_t)&len;
366 aio.iov_len = sizeof(len);
367 auio.uio_iov = &aio;
368 auio.uio_iovcnt = 1;
369 auio.uio_segflg = UIO_SYSSPACE;
370 auio.uio_rw = UIO_READ;
371 auio.uio_offset = 0;
372 auio.uio_resid = sizeof(len);
373 auio.uio_td = td;
374 error = so->so_proto->pr_usrreqs->pru_soreceive
375 (so, (struct sockaddr **)NULL, &auio,
376 (struct mbuf **)NULL, (struct mbuf **)NULL, &flags);
377 if (error)
378 return error;
379 if (auio.uio_resid > 0) {
380 SMBSDEBUG("short reply\n");
381 return EPIPE;
382 }
383 len = ntohl(len);
384 *rpcodep = (len >> 24) & 0xFF;
385 len &= 0x1ffff;
386 if (len > SMB_MAXPKTLEN) {
387 SMBERROR("packet too long (%d)\n", len);
388 return EFBIG;
389 }
390 *lenp = len;
391 return 0;
392}
393
394static int
395nbssn_recv(struct nbpcb *nbp, struct mbuf **mpp, int *lenp,
396 u_int8_t *rpcodep, struct thread *td)
397{
398 struct socket *so = nbp->nbp_tso;
399 struct uio auio;
400 struct mbuf *m, *tm, *im;
401 u_int8_t rpcode;
402 int len, resid;
403 int error, rcvflg;
404
405 if (so == NULL)
406 return ENOTCONN;
407
408 if (mpp)
409 *mpp = NULL;
410 m = NULL;
411 for(;;) {
412 /*
413 * Poll for a response header.
414 * If we don't have one waiting, return.
415 */
416 error = nbssn_recvhdr(nbp, &len, &rpcode, MSG_DONTWAIT, td);
|
419 nbp->nbp_state = NBST_CLOSED;
420 NBDEBUG("session closed by peer\n");
421 return ECONNRESET;
422 }
423 if (error)
424 return error;
425 if (len == 0 && nbp->nbp_state != NBST_SESSION)
426 break;
427 /* no data, try again */
428 if (rpcode == NB_SSN_KEEPALIVE)
429 continue;
430
431 /*
432 * Loop, blocking, for data following the response header.
433 *
434 * Note that we can't simply block here with MSG_WAITALL for the
435 * entire response size, as it may be larger than the TCP
436 * slow-start window that the sender employs. This will result
437 * in the sender stalling until the delayed ACK is sent, then
438 * resuming slow-start, resulting in very poor performance.
439 *
440 * Instead, we never request more than NB_SORECEIVE_CHUNK
441 * bytes at a time, resulting in an ack being pushed by
442 * the TCP code at the completion of each call.
443 */
444 resid = len;
445 while (resid > 0) {
446 tm = NULL;
447 rcvflg = MSG_WAITALL;
448 bzero(&auio, sizeof(auio));
449 auio.uio_resid = min(resid, NB_SORECEIVE_CHUNK);
450 auio.uio_td = td;
451 resid -= auio.uio_resid;
452 /*
453 * Spin until we have collected everything in
454 * this chunk.
455 */
456 do {
457 rcvflg = MSG_WAITALL;
458 error = so->so_proto->pr_usrreqs->pru_soreceive
459 (so, (struct sockaddr **)NULL,
460 &auio, &tm, (struct mbuf **)NULL, &rcvflg);
461 } while (error == EWOULDBLOCK || error == EINTR ||
462 error == ERESTART);
463 if (error)
464 goto out;
465 /* short return guarantees unhappiness */
466 if (auio.uio_resid > 0) {
467 SMBERROR("packet is shorter than expected\n");
468 error = EPIPE;
469 goto out;
470 }
471 /* append received chunk to previous chunk(s) */
472 if (m == NULL) {
473 m = tm;
474 } else {
475 /*
476 * Just glue the new chain on the end.
477 * Consumer will pullup as required.
478 */
479 for (im = m; im->m_next != NULL; im = im->m_next)
480 ;
481 im->m_next = tm;
482 }
483 }
484 /* got a session/message packet? */
485 if (nbp->nbp_state == NBST_SESSION &&
486 rpcode == NB_SSN_MESSAGE)
487 break;
488 /* drop packet and try for another */
489 NBDEBUG("non-session packet %x\n", rpcode);
490 if (m) {
491 m_freem(m);
492 m = NULL;
493 }
494 }
495
496out:
497 if (error) {
498 if (m)
499 m_freem(m);
500 return error;
501 }
502 if (mpp)
503 *mpp = m;
504 else
505 m_freem(m);
506 *lenp = len;
507 *rpcodep = rpcode;
508 return 0;
509}
510
511/*
512 * SMB transport interface
513 */
514static int
515smb_nbst_create(struct smb_vc *vcp, struct thread *td)
516{
517 struct nbpcb *nbp;
518
519 MALLOC(nbp, struct nbpcb *, sizeof *nbp, M_NBDATA, M_WAITOK);
520 bzero(nbp, sizeof *nbp);
521 nbp->nbp_timo.tv_sec = 15; /* XXX: sysctl ? */
522 nbp->nbp_state = NBST_CLOSED;
523 nbp->nbp_vc = vcp;
524 nbp->nbp_sndbuf = smb_tcpsndbuf;
525 nbp->nbp_rcvbuf = smb_tcprcvbuf;
526 vcp->vc_tdata = nbp;
527 return 0;
528}
529
530static int
531smb_nbst_done(struct smb_vc *vcp, struct thread *td)
532{
533 struct nbpcb *nbp = vcp->vc_tdata;
534
535 if (nbp == NULL)
536 return ENOTCONN;
537 smb_nbst_disconnect(vcp, td);
538 if (nbp->nbp_laddr)
539 free(nbp->nbp_laddr, M_SONAME);
540 if (nbp->nbp_paddr)
541 free(nbp->nbp_paddr, M_SONAME);
542 free(nbp, M_NBDATA);
543 return 0;
544}
545
546static int
547smb_nbst_bind(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
548{
549 struct nbpcb *nbp = vcp->vc_tdata;
550 struct sockaddr_nb *snb;
551 int error, slen;
552
553 NBDEBUG("\n");
554 error = EINVAL;
555 do {
556 if (nbp->nbp_flags & NBF_LOCADDR)
557 break;
558 /*
559 * It is possible to create NETBIOS name in the kernel,
560 * but nothing prevents us to do it in the user space.
561 */
562 if (sap == NULL)
563 break;
564 slen = sap->sa_len;
565 if (slen < NB_MINSALEN)
566 break;
567 snb = (struct sockaddr_nb*)sodupsockaddr(sap, M_WAITOK);
568 if (snb == NULL) {
569 error = ENOMEM;
570 break;
571 }
572 nbp->nbp_laddr = snb;
573 nbp->nbp_flags |= NBF_LOCADDR;
574 error = 0;
575 } while(0);
576 return error;
577}
578
579static int
580smb_nbst_connect(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
581{
582 struct nbpcb *nbp = vcp->vc_tdata;
583 struct sockaddr_in sin;
584 struct sockaddr_nb *snb;
585 struct timespec ts1, ts2;
586 int error, slen;
587
588 NBDEBUG("\n");
589 if (nbp->nbp_tso != NULL)
590 return EISCONN;
591 if (nbp->nbp_laddr == NULL)
592 return EINVAL;
593 slen = sap->sa_len;
594 if (slen < NB_MINSALEN)
595 return EINVAL;
596 if (nbp->nbp_paddr) {
597 free(nbp->nbp_paddr, M_SONAME);
598 nbp->nbp_paddr = NULL;
599 }
600 snb = (struct sockaddr_nb*)sodupsockaddr(sap, M_WAITOK);
601 if (snb == NULL)
602 return ENOMEM;
603 nbp->nbp_paddr = snb;
604 sin = snb->snb_addrin;
605 getnanotime(&ts1);
606 error = nb_connect_in(nbp, &sin, td);
607 if (error)
608 return error;
609 getnanotime(&ts2);
610 timespecsub(&ts2, &ts1);
611 if (ts2.tv_sec == 0 && ts2.tv_sec == 0)
612 ts2.tv_sec = 1;
613 nbp->nbp_timo = ts2;
614 timespecadd(&nbp->nbp_timo, &ts2);
615 timespecadd(&nbp->nbp_timo, &ts2);
616 timespecadd(&nbp->nbp_timo, &ts2); /* * 4 */
617 error = nbssn_rq_request(nbp, td);
618 if (error)
619 smb_nbst_disconnect(vcp, td);
620 return error;
621}
622
623static int
624smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td)
625{
626 struct nbpcb *nbp = vcp->vc_tdata;
627 struct socket *so;
628
629 if (nbp == NULL || nbp->nbp_tso == NULL)
630 return ENOTCONN;
631 if ((so = nbp->nbp_tso) != NULL) {
632 nbp->nbp_flags &= ~NBF_CONNECTED;
633 nbp->nbp_tso = (struct socket *)NULL;
634 soshutdown(so, 2);
635 soclose(so);
636 }
637 if (nbp->nbp_state != NBST_RETARGET) {
638 nbp->nbp_state = NBST_CLOSED;
639 }
640 return 0;
641}
642
643static int
644smb_nbst_send(struct smb_vc *vcp, struct mbuf *m0, struct thread *td)
645{
646 struct nbpcb *nbp = vcp->vc_tdata;
647 int error;
648
649 if (nbp->nbp_state != NBST_SESSION) {
650 error = ENOTCONN;
651 goto abort;
652 }
653 M_PREPEND(m0, 4, M_TRYWAIT);
654 if (m0 == NULL)
655 return ENOBUFS;
656 nb_sethdr(m0, NB_SSN_MESSAGE, m_fixhdr(m0) - 4);
657 error = nb_sosend(nbp->nbp_tso, m0, 0, td);
658 return error;
659abort:
660 if (m0)
661 m_freem(m0);
662 return error;
663}
664
665
666static int
667smb_nbst_recv(struct smb_vc *vcp, struct mbuf **mpp, struct thread *td)
668{
669 struct nbpcb *nbp = vcp->vc_tdata;
670 u_int8_t rpcode;
671 int error, rplen;
672
673 nbp->nbp_flags |= NBF_RECVLOCK;
674 error = nbssn_recv(nbp, mpp, &rplen, &rpcode, td);
675 nbp->nbp_flags &= ~NBF_RECVLOCK;
676 return error;
677}
678
679static void
680smb_nbst_timo(struct smb_vc *vcp)
681{
682 return;
683}
684
685static void
686smb_nbst_intr(struct smb_vc *vcp)
687{
688 struct nbpcb *nbp = vcp->vc_tdata;
689
690 if (nbp == NULL || nbp->nbp_tso == NULL)
691 return;
692 sorwakeup(nbp->nbp_tso);
693 sowwakeup(nbp->nbp_tso);
694}
695
696static int
697smb_nbst_getparam(struct smb_vc *vcp, int param, void *data)
698{
699 struct nbpcb *nbp = vcp->vc_tdata;
700
701 switch (param) {
702 case SMBTP_SNDSZ:
703 *(int*)data = nbp->nbp_sndbuf;
704 break;
705 case SMBTP_RCVSZ:
706 *(int*)data = nbp->nbp_rcvbuf;
707 break;
708 case SMBTP_TIMEOUT:
709 *(struct timespec*)data = nbp->nbp_timo;
710 break;
711 default:
712 return EINVAL;
713 }
714 return 0;
715}
716
717static int
718smb_nbst_setparam(struct smb_vc *vcp, int param, void *data)
719{
720 struct nbpcb *nbp = vcp->vc_tdata;
721
722 switch (param) {
723 case SMBTP_SELECTID:
724 nbp->nbp_selectid = data;
725 break;
726 default:
727 return EINVAL;
728 }
729 return 0;
730}
731
732/*
733 * Check for fatal errors
734 */
735static int
736smb_nbst_fatal(struct smb_vc *vcp, int error)
737{
738 switch (error) {
739 case ENOTCONN:
740 case ENETRESET:
741 case ECONNABORTED:
742 return 1;
743 }
744 return 0;
745}
746
747
748struct smb_tran_desc smb_tran_nbtcp_desc = {
749 SMBT_NBTCP,
750 smb_nbst_create, smb_nbst_done,
751 smb_nbst_bind, smb_nbst_connect, smb_nbst_disconnect,
752 smb_nbst_send, smb_nbst_recv,
753 smb_nbst_timo, smb_nbst_intr,
754 smb_nbst_getparam, smb_nbst_setparam,
755 smb_nbst_fatal
756};
757
| 419 nbp->nbp_state = NBST_CLOSED;
420 NBDEBUG("session closed by peer\n");
421 return ECONNRESET;
422 }
423 if (error)
424 return error;
425 if (len == 0 && nbp->nbp_state != NBST_SESSION)
426 break;
427 /* no data, try again */
428 if (rpcode == NB_SSN_KEEPALIVE)
429 continue;
430
431 /*
432 * Loop, blocking, for data following the response header.
433 *
434 * Note that we can't simply block here with MSG_WAITALL for the
435 * entire response size, as it may be larger than the TCP
436 * slow-start window that the sender employs. This will result
437 * in the sender stalling until the delayed ACK is sent, then
438 * resuming slow-start, resulting in very poor performance.
439 *
440 * Instead, we never request more than NB_SORECEIVE_CHUNK
441 * bytes at a time, resulting in an ack being pushed by
442 * the TCP code at the completion of each call.
443 */
444 resid = len;
445 while (resid > 0) {
446 tm = NULL;
447 rcvflg = MSG_WAITALL;
448 bzero(&auio, sizeof(auio));
449 auio.uio_resid = min(resid, NB_SORECEIVE_CHUNK);
450 auio.uio_td = td;
451 resid -= auio.uio_resid;
452 /*
453 * Spin until we have collected everything in
454 * this chunk.
455 */
456 do {
457 rcvflg = MSG_WAITALL;
458 error = so->so_proto->pr_usrreqs->pru_soreceive
459 (so, (struct sockaddr **)NULL,
460 &auio, &tm, (struct mbuf **)NULL, &rcvflg);
461 } while (error == EWOULDBLOCK || error == EINTR ||
462 error == ERESTART);
463 if (error)
464 goto out;
465 /* short return guarantees unhappiness */
466 if (auio.uio_resid > 0) {
467 SMBERROR("packet is shorter than expected\n");
468 error = EPIPE;
469 goto out;
470 }
471 /* append received chunk to previous chunk(s) */
472 if (m == NULL) {
473 m = tm;
474 } else {
475 /*
476 * Just glue the new chain on the end.
477 * Consumer will pullup as required.
478 */
479 for (im = m; im->m_next != NULL; im = im->m_next)
480 ;
481 im->m_next = tm;
482 }
483 }
484 /* got a session/message packet? */
485 if (nbp->nbp_state == NBST_SESSION &&
486 rpcode == NB_SSN_MESSAGE)
487 break;
488 /* drop packet and try for another */
489 NBDEBUG("non-session packet %x\n", rpcode);
490 if (m) {
491 m_freem(m);
492 m = NULL;
493 }
494 }
495
496out:
497 if (error) {
498 if (m)
499 m_freem(m);
500 return error;
501 }
502 if (mpp)
503 *mpp = m;
504 else
505 m_freem(m);
506 *lenp = len;
507 *rpcodep = rpcode;
508 return 0;
509}
510
511/*
512 * SMB transport interface
513 */
514static int
515smb_nbst_create(struct smb_vc *vcp, struct thread *td)
516{
517 struct nbpcb *nbp;
518
519 MALLOC(nbp, struct nbpcb *, sizeof *nbp, M_NBDATA, M_WAITOK);
520 bzero(nbp, sizeof *nbp);
521 nbp->nbp_timo.tv_sec = 15; /* XXX: sysctl ? */
522 nbp->nbp_state = NBST_CLOSED;
523 nbp->nbp_vc = vcp;
524 nbp->nbp_sndbuf = smb_tcpsndbuf;
525 nbp->nbp_rcvbuf = smb_tcprcvbuf;
526 vcp->vc_tdata = nbp;
527 return 0;
528}
529
530static int
531smb_nbst_done(struct smb_vc *vcp, struct thread *td)
532{
533 struct nbpcb *nbp = vcp->vc_tdata;
534
535 if (nbp == NULL)
536 return ENOTCONN;
537 smb_nbst_disconnect(vcp, td);
538 if (nbp->nbp_laddr)
539 free(nbp->nbp_laddr, M_SONAME);
540 if (nbp->nbp_paddr)
541 free(nbp->nbp_paddr, M_SONAME);
542 free(nbp, M_NBDATA);
543 return 0;
544}
545
546static int
547smb_nbst_bind(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
548{
549 struct nbpcb *nbp = vcp->vc_tdata;
550 struct sockaddr_nb *snb;
551 int error, slen;
552
553 NBDEBUG("\n");
554 error = EINVAL;
555 do {
556 if (nbp->nbp_flags & NBF_LOCADDR)
557 break;
558 /*
559 * It is possible to create NETBIOS name in the kernel,
560 * but nothing prevents us to do it in the user space.
561 */
562 if (sap == NULL)
563 break;
564 slen = sap->sa_len;
565 if (slen < NB_MINSALEN)
566 break;
567 snb = (struct sockaddr_nb*)sodupsockaddr(sap, M_WAITOK);
568 if (snb == NULL) {
569 error = ENOMEM;
570 break;
571 }
572 nbp->nbp_laddr = snb;
573 nbp->nbp_flags |= NBF_LOCADDR;
574 error = 0;
575 } while(0);
576 return error;
577}
578
579static int
580smb_nbst_connect(struct smb_vc *vcp, struct sockaddr *sap, struct thread *td)
581{
582 struct nbpcb *nbp = vcp->vc_tdata;
583 struct sockaddr_in sin;
584 struct sockaddr_nb *snb;
585 struct timespec ts1, ts2;
586 int error, slen;
587
588 NBDEBUG("\n");
589 if (nbp->nbp_tso != NULL)
590 return EISCONN;
591 if (nbp->nbp_laddr == NULL)
592 return EINVAL;
593 slen = sap->sa_len;
594 if (slen < NB_MINSALEN)
595 return EINVAL;
596 if (nbp->nbp_paddr) {
597 free(nbp->nbp_paddr, M_SONAME);
598 nbp->nbp_paddr = NULL;
599 }
600 snb = (struct sockaddr_nb*)sodupsockaddr(sap, M_WAITOK);
601 if (snb == NULL)
602 return ENOMEM;
603 nbp->nbp_paddr = snb;
604 sin = snb->snb_addrin;
605 getnanotime(&ts1);
606 error = nb_connect_in(nbp, &sin, td);
607 if (error)
608 return error;
609 getnanotime(&ts2);
610 timespecsub(&ts2, &ts1);
611 if (ts2.tv_sec == 0 && ts2.tv_sec == 0)
612 ts2.tv_sec = 1;
613 nbp->nbp_timo = ts2;
614 timespecadd(&nbp->nbp_timo, &ts2);
615 timespecadd(&nbp->nbp_timo, &ts2);
616 timespecadd(&nbp->nbp_timo, &ts2); /* * 4 */
617 error = nbssn_rq_request(nbp, td);
618 if (error)
619 smb_nbst_disconnect(vcp, td);
620 return error;
621}
622
623static int
624smb_nbst_disconnect(struct smb_vc *vcp, struct thread *td)
625{
626 struct nbpcb *nbp = vcp->vc_tdata;
627 struct socket *so;
628
629 if (nbp == NULL || nbp->nbp_tso == NULL)
630 return ENOTCONN;
631 if ((so = nbp->nbp_tso) != NULL) {
632 nbp->nbp_flags &= ~NBF_CONNECTED;
633 nbp->nbp_tso = (struct socket *)NULL;
634 soshutdown(so, 2);
635 soclose(so);
636 }
637 if (nbp->nbp_state != NBST_RETARGET) {
638 nbp->nbp_state = NBST_CLOSED;
639 }
640 return 0;
641}
642
643static int
644smb_nbst_send(struct smb_vc *vcp, struct mbuf *m0, struct thread *td)
645{
646 struct nbpcb *nbp = vcp->vc_tdata;
647 int error;
648
649 if (nbp->nbp_state != NBST_SESSION) {
650 error = ENOTCONN;
651 goto abort;
652 }
653 M_PREPEND(m0, 4, M_TRYWAIT);
654 if (m0 == NULL)
655 return ENOBUFS;
656 nb_sethdr(m0, NB_SSN_MESSAGE, m_fixhdr(m0) - 4);
657 error = nb_sosend(nbp->nbp_tso, m0, 0, td);
658 return error;
659abort:
660 if (m0)
661 m_freem(m0);
662 return error;
663}
664
665
666static int
667smb_nbst_recv(struct smb_vc *vcp, struct mbuf **mpp, struct thread *td)
668{
669 struct nbpcb *nbp = vcp->vc_tdata;
670 u_int8_t rpcode;
671 int error, rplen;
672
673 nbp->nbp_flags |= NBF_RECVLOCK;
674 error = nbssn_recv(nbp, mpp, &rplen, &rpcode, td);
675 nbp->nbp_flags &= ~NBF_RECVLOCK;
676 return error;
677}
678
679static void
680smb_nbst_timo(struct smb_vc *vcp)
681{
682 return;
683}
684
685static void
686smb_nbst_intr(struct smb_vc *vcp)
687{
688 struct nbpcb *nbp = vcp->vc_tdata;
689
690 if (nbp == NULL || nbp->nbp_tso == NULL)
691 return;
692 sorwakeup(nbp->nbp_tso);
693 sowwakeup(nbp->nbp_tso);
694}
695
696static int
697smb_nbst_getparam(struct smb_vc *vcp, int param, void *data)
698{
699 struct nbpcb *nbp = vcp->vc_tdata;
700
701 switch (param) {
702 case SMBTP_SNDSZ:
703 *(int*)data = nbp->nbp_sndbuf;
704 break;
705 case SMBTP_RCVSZ:
706 *(int*)data = nbp->nbp_rcvbuf;
707 break;
708 case SMBTP_TIMEOUT:
709 *(struct timespec*)data = nbp->nbp_timo;
710 break;
711 default:
712 return EINVAL;
713 }
714 return 0;
715}
716
717static int
718smb_nbst_setparam(struct smb_vc *vcp, int param, void *data)
719{
720 struct nbpcb *nbp = vcp->vc_tdata;
721
722 switch (param) {
723 case SMBTP_SELECTID:
724 nbp->nbp_selectid = data;
725 break;
726 default:
727 return EINVAL;
728 }
729 return 0;
730}
731
732/*
733 * Check for fatal errors
734 */
735static int
736smb_nbst_fatal(struct smb_vc *vcp, int error)
737{
738 switch (error) {
739 case ENOTCONN:
740 case ENETRESET:
741 case ECONNABORTED:
742 return 1;
743 }
744 return 0;
745}
746
747
748struct smb_tran_desc smb_tran_nbtcp_desc = {
749 SMBT_NBTCP,
750 smb_nbst_create, smb_nbst_done,
751 smb_nbst_bind, smb_nbst_connect, smb_nbst_disconnect,
752 smb_nbst_send, smb_nbst_recv,
753 smb_nbst_timo, smb_nbst_intr,
754 smb_nbst_getparam, smb_nbst_setparam,
755 smb_nbst_fatal
756};
757
|