NetBSD/sys/netsmb/smb_dev.c
thorpej 80cc38a1af Fix a partial construction problem that can cause race conditions
between creation of a file descriptor and close(2) when using kernel
assisted threads.  What we do is stick descriptors in the table, but
mark them as "larval".  This causes essentially everything to treat
it as a non-existent descriptor, except for fdalloc(), which sees a
filled slot so that it won't (incorrectly) allocate it again.  When
a descriptor is fully constructed, the code that has constructed it
marks it as "mature" (which actually clears the "larval" flag), and
things continue to work as normal.

While here, gather all the code that gets a descriptor from the table
into a fd_getfile() function, and call it, rather than having the
same (sometimes incorrect) code copied all over the place.
2001-06-14 20:32:41 +00:00

535 lines
12 KiB
C

/* $NetBSD: smb_dev.c,v 1.3 2001/06/14 20:32:48 thorpej Exp $ */
/*
* Copyright (c) 2000, Boris Popov
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by Boris Popov.
* 4. Neither the name of the author nor the names of any co-contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <netsmb.h>
#include <sys/param.h>
#include <sys/ioccom.h>
#include <sys/malloc.h>
#include <sys/uio.h>
#include <sys/kernel.h>
#include <sys/systm.h>
#include <sys/conf.h>
#include <sys/mbuf.h>
#include <sys/proc.h>
#include <sys/fcntl.h>
#include <sys/file.h>
#include <sys/filedesc.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/poll.h>
#include <sys/socketvar.h>
#include <uvm/uvm_extern.h>
#include <sys/sysctl.h>
#include <sys/vnode.h>
#include <sys/conf.h>
#include <net/if.h>
#include <sys/tree.h>
#include <netsmb/smb.h>
#include <netsmb/smb_conn.h>
#include <netsmb/smb_subr.h>
#include <netsmb/smb_dev.h>
#ifndef FB_RELENG3
#define SMB_GETDEV(dev) ((struct smb_dev*)(dev)->si_drv1)
#define SMB_CHECKMINOR(dev) do { \
sdp = SMB_GETDEV(dev); \
if (sdp == NULL) return ENXIO; \
} while(0)
#else
int numnetsmb = 0;
#define SMB_GETDEV(dev) (nsmbdevs + minor(dev))
#define SMB_CHECKMINOR(dev) do { \
if (minor(dev) >= numnetsmb) \
return ENXIO; \
sdp = SMB_GETDEV(dev); \
} while(0)
#include <miscfs/specfs/specdev.h>
static struct smb_dev *nsmbdevs;
#endif /* !FB_RELENG3 */
#ifndef NetBSD
static d_open_t nsmb_dev_open;
static d_close_t nsmb_dev_close;
static d_read_t nsmb_dev_read;
static d_write_t nsmb_dev_write;
static d_ioctl_t nsmb_dev_ioctl;
static d_poll_t nsmb_dev_poll;
#else
cdev_decl(nsmb_dev_);
#endif
#ifdef MODULE_DEPEND
MODULE_DEPEND(netsmb, libiconv, 1, 1, 1);
#endif
#ifdef MODULE_VERSION
MODULE_VERSION(netsmb, 1);
#endif
#ifndef NetBSD
static int smb_version = NSMB_VERSION;
#endif
#ifndef NetBSD
#ifdef SYSCTL_DECL
SYSCTL_DECL(_net_smb);
#endif
SYSCTL_INT(_net_smb, OID_AUTO, version, CTLFLAG_RD, &smb_version, 0, "");
MALLOC_DEFINE(M_NSMBDEV, "NETSMBDEV", "NET/SMB device");
#endif
/*
int smb_dev_queue(struct smb_dev *ndp, struct smb_rq *rqp, int prio);
*/
#ifndef NetBSD
static struct cdevsw nsmb_cdevsw = {
/* open */ nsmb_dev_open,
/* close */ nsmb_dev_close,
/* read */ nsmb_dev_read,
/* write */ nsmb_dev_write,
/* ioctl */ nsmb_dev_ioctl,
#ifdef FB_RELENG3
/* stop */ nostop,
/* reset */ noreset,
/* devtotty */ nodevtotty,
#endif
/* poll */ nsmb_dev_poll,
/* mmap */ nommap,
/* strategy */ nostrategy,
/* name */ NSMB_NAME,
#ifndef FB_RELENG3
/* maj */ NSMB_MAJOR,
/* dump */ nodump,
/* psize */ nopsize,
/* flags */ 0,
#else
/* spare */ NULL,
#endif
/* bmaj */ -1
};
#endif
void netsmbattach __P((int));
void
netsmbattach(num)
int num;
{
if (num <= 0) {
#ifdef DIAGNOSTIC
panic("ccdattach: count <= 0");
#endif
return;
}
nsmbdevs = (struct smb_dev *)malloc(num * sizeof(struct smb_dev),
M_DEVBUF, M_NOWAIT);
if (nsmbdevs == NULL) {
printf("WARNING: no memory for netsmb devices\n");
return;
}
numnetsmb = num;
if (smb_sm_init()) {
#ifdef DIAGNOSTIC
panic("netsmbattach: smb_sm_init failed");
#endif
return;
}
}
int
nsmb_dev_open(dev_t dev, int oflags, int devtype, struct proc *p)
{
struct smb_dev *sdp;
#ifndef FB_RELENG3
struct ucred *cred = p->p_ucred;
#endif
int s;
sdp = SMB_GETDEV(dev);
if (sdp && (sdp->sd_flags & NSMBFL_OPEN))
return EBUSY;
#ifndef FB_RELENG3
if (sdp == NULL) {
MALLOC(sdp, struct smb_dev*, sizeof(*sdp), M_NSMBDEV, M_WAITOK);
if (sdp == NULL)
return ENOMEM;
dev->si_drv1 = (void*)sdp;
}
/*
* XXX: this is just crazy - make a device for an already passed device...
* someone should take care of it.
*/
make_dev(&nsmb_cdevsw, minor(dev), cred->cr_uid, cred->cr_gid, 0700,
"net/smb%d", lminor(dev));
#endif
bzero(sdp, sizeof(*sdp));
/*
STAILQ_INIT(&sdp->sd_rqlist);
STAILQ_INIT(&sdp->sd_rplist);
bzero(&sdp->sd_pollinfo, sizeof(struct selinfo));
*/
s = splnet();
sdp->sd_level = -1;
sdp->sd_flags |= NSMBFL_OPEN;
splx(s);
return 0;
}
int
nsmb_dev_close(dev_t dev, int flag, int fmt, struct proc *p)
{
struct smb_dev *sdp;
struct smb_conn *scp;
struct smb_vc *vcp;
struct smb_share *ssp;
struct smb_cred scred;
int s;
SMB_CHECKMINOR(dev);
s = splnet();
if ((sdp->sd_flags & NSMBFL_OPEN) == 0) {
splx(s);
return EBADF;
}
smb_makescred(&scred, p, NULL);
ssp = sdp->sd_share;
if (ssp != NULL) {
smb_share_rele(ssp, &scred);
}
vcp = sdp->sd_vc;
if (vcp != NULL) {
smb_vc_rele(vcp, &scred);
}
scp = sdp->sd_conn;
if (scp != NULL)
smb_conn_rele(scp, &scred);
/*
smb_flushq(&sdp->sd_rqlist);
smb_flushq(&sdp->sd_rplist);
*/
#if __FreeBSD_version > 400001
dev->si_drv1 = NULL;
free(sdp, M_NSMBDEV);
destroy_dev(dev);
#else
sdp->sd_flags &= ~NSMBFL_OPEN;
#endif
splx(s);
return 0;
}
int
#ifndef NetBSD
#if __FreeBSD_version >= 300003
nsmb_dev_ioctl(dev_t dev, u_long cmd, caddr_t data, int flag, struct proc *p)
#else
nsmb_dev_ioctl(dev_t dev, int cmd, caddr_t data, int flag, struct proc *p)
#endif
#else
nsmb_dev_ioctl(dev_t dev, u_long cmd, caddr_t data, int flag, struct proc *p)
#endif
{
struct smb_dev *sdp;
struct smb_conn *scp;
struct smb_vc *vcp;
struct smb_share *ssp;
struct smb_cred scred;
int error = 0;
SMB_CHECKMINOR(dev);
if ((sdp->sd_flags & NSMBFL_OPEN) == 0)
return EBADF;
smb_makescred(&scred, p, NULL);
switch (cmd) {
case SMBIOC_OPENCONN:
if (sdp->sd_conn)
return EISCONN;
error = smb_usr_openconn((struct smbioc_oconn*)data,
&scred, &scp);
if (error)
break;
sdp->sd_conn = scp;
smb_conn_unlock(scp, p);
sdp->sd_level = SMBL_CONN;
break;
case SMBIOC_OPENSESSION:
if (sdp->sd_vc)
return EISCONN;
if (sdp->sd_conn == NULL)
return ENOTCONN;
error = smb_usr_opensession((struct smbioc_ossn*)data,
sdp->sd_conn, &scred, &vcp);
if (error)
break;
sdp->sd_vc = vcp;
smb_vc_unlock(vcp, 0, &scred);
sdp->sd_level = SMBL_VC;
break;
case SMBIOC_OPENSHARE:
if (sdp->sd_share)
return EISCONN;
if (sdp->sd_vc == NULL)
return ENOTCONN;
error = smb_usr_openshare(sdp->sd_vc,
(struct smbioc_oshare*)data, &scred, &ssp);
if (error)
break;
sdp->sd_share = ssp;
smb_share_unlock(ssp, 0, &scred);
sdp->sd_level = SMBL_SHARE;
break;
case SMBIOC_REQUEST:
if (sdp->sd_share == NULL)
return ENOTCONN;
error = smb_usr_simplerequest(sdp->sd_share,
(struct smbioc_rq*)data, &scred);
break;
case SMBIOC_T2RQ:
if (sdp->sd_share == NULL)
return ENOTCONN;
error = smb_usr_t2request(sdp->sd_share,
(struct smbioc_t2rq*)data, &scred);
break;
case SMBIOC_SETFLAGS: {
struct smbioc_flags *fl = (struct smbioc_flags*)data;
int on;
if (fl->ioc_level == SMBL_VC) {
if (fl->ioc_mask & SMBV_PERMANENT) {
on = fl->ioc_flags & SMBV_PERMANENT;
if ((vcp = sdp->sd_vc) == NULL)
return ENOTCONN;
error = smb_vc_get(vcp, &scred);
if (error)
break;
if (on && (vcp->vc_flags & SMBV_PERMANENT) == 0) {
vcp->vc_flags |= SMBV_PERMANENT;
tref(VCTOTN(vcp));
} else if (!on && (vcp->vc_flags & SMBV_PERMANENT)) {
vcp->vc_flags &= ~SMBV_PERMANENT;
trele(VCTOTN(vcp));
}
smb_vc_put(vcp, &scred);
} else
error = EINVAL;
} else if (fl->ioc_level == SMBL_SHARE) {
if (fl->ioc_mask & SMBS_PERMANENT) {
on = fl->ioc_flags & SMBS_PERMANENT;
if ((ssp = sdp->sd_share) == NULL)
return ENOTCONN;
error = smb_share_get(ssp, &scred);
if (error)
break;
if (on && (ssp->ss_flags & SMBS_PERMANENT) == 0) {
ssp->ss_flags |= SMBS_PERMANENT;
tref(SSTOTN(ssp));
} else if (!on && (ssp->ss_flags & SMBS_PERMANENT)) {
ssp->ss_flags &= ~SMBS_PERMANENT;
trele(SSTOTN(ssp));
}
smb_share_put(ssp, p);
} else
error = EINVAL;
break;
} else
error = EINVAL;
break;
}
case SMBIOC_LOOKUP:
if (sdp->sd_conn || sdp->sd_vc || sdp->sd_share)
return EISCONN;
scp = NULL;
vcp = NULL;
ssp = NULL;
error = smb_usr_lookup((struct smbioc_lookup*)data, &scred, &scp, &vcp, &ssp);
if (error)
break;
if (scp) {
sdp->sd_conn = scp;
smb_conn_unlock(scp, p);
sdp->sd_level = SMBL_CONN;
}
if (vcp) {
sdp->sd_vc = vcp;
smb_vc_unlock(vcp, 0, &scred);
sdp->sd_level = SMBL_VC;
}
if (ssp) {
sdp->sd_share = ssp;
smb_share_unlock(ssp, 0, &scred);
sdp->sd_level = SMBL_SHARE;
}
break;
case SMBIOC_READ: case SMBIOC_WRITE: {
struct smbioc_rw *rwrq = (struct smbioc_rw*)data;
struct uio auio;
struct iovec iov;
if ((ssp = sdp->sd_share) == NULL)
return ENOTCONN;
iov.iov_base = rwrq->ioc_base;
iov.iov_len = rwrq->ioc_cnt;
auio.uio_iov = &iov;
auio.uio_iovcnt = 1;
auio.uio_offset = rwrq->ioc_offset;
auio.uio_resid = rwrq->ioc_cnt;
auio.uio_segflg = UIO_USERSPACE;
auio.uio_rw = (cmd == SMBIOC_READ) ? UIO_READ : UIO_WRITE;
auio.uio_procp = p;
if (cmd == SMBIOC_READ)
error = smb_read(ssp, rwrq->ioc_fh, &auio, &scred);
else
error = smb_write(ssp, rwrq->ioc_fh, &auio, &scred);
rwrq->ioc_cnt -= auio.uio_resid;
break;
}
default:
error = ENODEV;
}
return error;
}
int
nsmb_dev_read(dev_t dev, struct uio *uio, int flag)
{
return EACCES;
}
int
nsmb_dev_write(dev_t dev, struct uio *uio, int flag)
{
return EACCES;
}
int
nsmb_dev_poll(dev_t dev, int events, struct proc *p)
{
return ENODEV;
}
#ifndef NetBSD
static int
nsmb_dev_load(module_t mod, int cmd, void *arg)
{
int error = 0;
switch (cmd) {
case MOD_LOAD:
error = smb_sm_init();
if (error)
break;
#if __FreeBSD_version > 400001
cdevsw_add(&nsmb_cdevsw);
#endif
printf("netsmb_dev: loaded\n");
break;
case MOD_UNLOAD:
error = smb_sm_done();
error = 0;
#if __FreeBSD_version > 400001
cdevsw_remove(&nsmb_cdevsw);
#endif
printf("netsmb_dev: unloaded\n");
break;
default:
error = EINVAL;
break;
}
return error;
}
#if __FreeBSD_version > 400000
DEV_MODULE (dev_netsmb, nsmb_dev_load, 0);
#else
CDEV_MODULE(dev_netsmb, NSMB_MAJOR, nsmb_cdevsw, nsmb_dev_load, 0);
#endif
#endif
/*
* Convert a file descriptor to appropriate smb_share pointer
*/
static struct file*
nsmb_getfp(struct filedesc* fdp, int fd, int flag)
{
struct file* fp;
if ((fp = fd_getfile(fdp, fd)) == NULL)
return (NULL);
if ((fp->f_flag & flag) == 0)
return (NULL);
return (fp);
}
int
smb_dev2share(int fd, int mode, struct smb_cred *scred,
struct smb_share **sspp)
{
struct file *fp;
struct vnode *vp;
struct smb_dev *sdp;
struct smb_share *ssp;
dev_t dev;
int error;
if ((fp = nsmb_getfp(scred->scr_p->p_fd, fd, FREAD | FWRITE)) == NULL)
return EBADF;
vp = (struct vnode*)fp->f_data;
if (vp == NULL)
return EBADF;
dev = vp->v_rdev;
if (dev == NODEV)
return EBADF;
SMB_CHECKMINOR(dev);
ssp = sdp->sd_share;
if (ssp == NULL)
return ENOTCONN;
error = smb_share_get(ssp, scred);
if (error)
return error;
*sspp = ssp;
return 0;
}