80cc38a1af
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.
535 lines
12 KiB
C
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;
|
|
}
|
|
|