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