sys/dev/disk/vn/vn.c

   1 /*
   2  * Copyright (c) 1988 University of Utah.
   3  * Copyright (c) 1990, 1993
   4  *      The Regents of the University of California.  All rights reserved.
   5  *
   6  * This code is derived from software contributed to Berkeley by
   7  * the Systems Programming Group of the University of Utah Computer
   8  * Science Department.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. All advertising materials mentioning features or use of this software
  19  *    must display the following acknowledgement:
  20  *      This product includes software developed by the University of
  21  *      California, Berkeley and its contributors.
  22  * 4. Neither the name of the University nor the names of its contributors
  23  *    may be used to endorse or promote products derived from this software
  24  *    without specific prior written permission.
  25  *
  26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  36  * SUCH DAMAGE.
  37  *
  38  * from: Utah Hdr: vn.c 1.13 94/04/02
  39  *
  40  *      from: @(#)vn.c  8.6 (Berkeley) 4/1/94
  41  * $FreeBSD: src/sys/dev/vn/vn.c,v 1.105.2.4 2001/11/18 07:11:00 dillon Exp $
  42  * $DragonFly: src/sys/dev/disk/vn/vn.c,v 1.38 2008/07/01 02:02:53 dillon Exp $
  43  */
  44
  45 /*
  46  * Vnode disk driver.
  47  *
  48  * Block/character interface to a vnode.  Allows one to treat a file
  49  * as a disk (e.g. build a filesystem in it, mount it, etc.).
  50  *
  51  * NOTE 1: There is a security issue involved with this driver.
  52  * Once mounted all access to the contents of the "mapped" file via
  53  * the special file is controlled by the permissions on the special
  54  * file, the protection of the mapped file is ignored (effectively,
  55  * by using root credentials in all transactions).
  56  *
  57  * NOTE 2: Doesn't interact with leases, should it?
  58  */
  59
  60 #include <sys/param.h>
  61 #include <sys/systm.h>
  62 #include <sys/kernel.h>
  63 #include <sys/proc.h>
  64 #include <sys/priv.h>
  65 #include <sys/nlookup.h>
  66 #include <sys/buf.h>
  67 #include <sys/malloc.h>
  68 #include <sys/mount.h>
  69 #include <sys/vnode.h>
  70 #include <sys/fcntl.h>
  71 #include <sys/conf.h>
  72 #include <sys/diskslice.h>
  73 #include <sys/disk.h>
  74 #include <sys/stat.h>
  75 #include <sys/module.h>
  76 #include <sys/vnioctl.h>
  77
  78 #include <vm/vm.h>
  79 #include <vm/vm_object.h>
  80 #include <vm/vm_page.h>
  81 #include <vm/vm_pager.h>
  82 #include <vm/vm_pageout.h>
  83 #include <vm/swap_pager.h>
  84 #include <vm/vm_extern.h>
  85 #include <vm/vm_zone.h>
  86 #include <sys/devfs.h>
  87
  88 static  d_ioctl_t       vnioctl;
  89 static  d_open_t        vnopen;
  90 static  d_close_t       vnclose;
  91 static  d_psize_t       vnsize;
  92 static  d_strategy_t    vnstrategy;
  93 static  d_clone_t       vnclone;
  94
  95 DEVFS_DECLARE_CLONE_BITMAP(vn);
  96 #define VN_PREALLOCATED_UNITS   4
  97
  98 #define CDEV_MAJOR 43
  99
 100 #define VN_BSIZE_BEST   8192
 101
 102 /*
 103  * dev_ops
 104  *      D_DISK          we want to look like a disk
 105  *      D_CANFREE       We support BUF_CMD_FREEBLKS
 106  */
 107
 108 static struct dev_ops vn_ops = {
 109         { "vn", CDEV_MAJOR, D_DISK | D_CANFREE },
 110         .d_open =       vnopen,
 111         .d_close =      vnclose,
 112         .d_read =       physread,
 113         .d_write =      physwrite,
 114         .d_ioctl =      vnioctl,
 115         .d_strategy =   vnstrategy,
 116         .d_psize =      vnsize
 117 };
 118
 119 struct vn_softc {
 120         int             sc_unit;
 121         int             sc_flags;       /* flags                        */
 122         u_int64_t       sc_size;        /* size of vn, sc_secsize scale */
 123         int             sc_secsize;     /* sector size                  */
 124         struct disk     sc_disk;
 125         struct vnode    *sc_vp;         /* vnode if not NULL            */
 126         vm_object_t     sc_object;      /* backing object if not NULL   */
 127         struct ucred    *sc_cred;       /* credentials                  */
 128         int              sc_maxactive;  /* max # of active requests     */
 129         struct buf       sc_tab;        /* transfer queue               */
 130         u_long           sc_options;    /* options                      */
 131         cdev_t           sc_devlist;    /* devices that refer to this unit */
 132         SLIST_ENTRY(vn_softc) sc_list;
 133 };
 134
 135 static SLIST_HEAD(, vn_softc) vn_list;
 136
 137 /* sc_flags */
 138 #define VNF_INITED      0x01
 139 #define VNF_READONLY    0x02
 140
 141 static u_long   vn_options;
 142
 143 #define IFOPT(vn,opt) if (((vn)->sc_options|vn_options) & (opt))
 144 #define TESTOPT(vn,opt) (((vn)->sc_options|vn_options) & (opt))
 145
 146 static int      vnsetcred (struct vn_softc *vn, struct ucred *cred);
 147 static void     vnclear (struct vn_softc *vn);
 148 static int      vnget (cdev_t dev, struct vn_softc *vn , struct vn_user *vnu);
 149 static int      vn_modevent (module_t, int, void *);
 150 static int      vniocattach_file (struct vn_softc *, struct vn_ioctl *, cdev_t dev, int flag, struct ucred *cred);
 151 static int      vniocattach_swap (struct vn_softc *, struct vn_ioctl *, cdev_t dev, int flag, struct ucred *cred);
 152 static cdev_t   vn_create(int unit, struct devfs_bitmap *bitmap);
 153
 154 static int
 155 vnclone(struct dev_clone_args *ap)
 156 {
 157         int unit;
 158
 159         unit = devfs_clone_bitmap_get(&DEVFS_CLONE_BITMAP(vn), 0);
 160         ap->a_dev = vn_create(unit, &DEVFS_CLONE_BITMAP(vn));
 161
 162         return 0;
 163 }
 164
 165 static  int
 166 vnclose(struct dev_close_args *ap)
 167 {
 168         return (0);
 169 }
 170
 171 static struct vn_softc *
 172 vncreatevn(void)
 173 {
 174         struct vn_softc *vn;
 175
 176         vn = kmalloc(sizeof *vn, M_DEVBUF, M_WAITOK | M_ZERO);
 177         return vn;
 178 }
 179
 180 static void
 181 vninitvn(struct vn_softc *vn, cdev_t dev)
 182 {
 183         int unit;
 184
 185         KKASSERT(vn != NULL);
 186         KKASSERT(dev != NULL);
 187         unit = dkunit(dev);
 188
 189         vn->sc_unit = unit;
 190         dev->si_drv1 = vn;
 191         vn->sc_devlist = dev;
 192         if (vn->sc_devlist->si_drv1 == NULL) {
 193                 reference_dev(vn->sc_devlist);
 194                 vn->sc_devlist->si_drv1 = vn;
 195                 vn->sc_devlist->si_drv2 = NULL;
 196         }
 197         if (vn->sc_devlist != dev) {
 198                 dev->si_drv1 = vn;
 199                 dev->si_drv2 = vn->sc_devlist;
 200                 vn->sc_devlist = dev;
 201                 reference_dev(dev);
 202         }
 203         SLIST_INSERT_HEAD(&vn_list, vn, sc_list);
 204 }
 205
 206 /*
 207  * Called only when si_drv1 is NULL.  Locate the associated vn node and
 208  * attach the device to it.
 209  */
 210 static struct vn_softc *
 211 vnfindvn(cdev_t dev)
 212 {
 213         int unit;
 214         struct vn_softc *vn;
 215
 216         unit = dkunit(dev);
 217         SLIST_FOREACH(vn, &vn_list, sc_list) {
 218                 if (vn->sc_unit == unit) {
 219                         dev->si_drv1 = vn;
 220                         dev->si_drv2 = vn->sc_devlist;
 221                         vn->sc_devlist = dev;
 222                         reference_dev(dev);
 223                         break;
 224                 }
 225         }
 226
 227         KKASSERT(vn != NULL);
 228
 229         return vn;
 230 }
 231
 232 static  int
 233 vnopen(struct dev_open_args *ap)
 234 {
 235         cdev_t dev = ap->a_head.a_dev;
 236         struct vn_softc *vn;
 237
 238         /*
 239          * Locate preexisting device
 240          */
 241
 242         if ((vn = dev->si_drv1) == NULL)
 243                 vn = vnfindvn(dev);
 244
 245         /*
 246          * Update si_bsize fields for device.  This data will be overriden by
 247          * the slice/parition code for vn accesses through partitions, and
 248          * used directly if you open the 'whole disk' device.
 249          *
 250          * si_bsize_best must be reinitialized in case VN has been
 251          * reconfigured, plus make it at least VN_BSIZE_BEST for efficiency.
 252          */
 253         dev->si_bsize_phys = vn->sc_secsize;
 254         dev->si_bsize_best = vn->sc_secsize;
 255         if (dev->si_bsize_best < VN_BSIZE_BEST)
 256                 dev->si_bsize_best = VN_BSIZE_BEST;
 257
 258         if ((ap->a_oflags & FWRITE) && (vn->sc_flags & VNF_READONLY))
 259                 return (EACCES);
 260
 261         IFOPT(vn, VN_FOLLOW)
 262                 kprintf("vnopen(%s, 0x%x, 0x%x)\n",
 263                     devtoname(dev), ap->a_oflags, ap->a_devtype);
 264
 265         return(0);
 266 }
 267
 268 /*
 269  *      vnstrategy:
 270  *
 271  *      Run strategy routine for VN device.  We use VOP_READ/VOP_WRITE calls
 272  *      for vnode-backed vn's, and the new vm_pager_strategy() call for
 273  *      vm_object-backed vn's.
 274  */
 275 static int
 276 vnstrategy(struct dev_strategy_args *ap)
 277 {
 278         cdev_t dev = ap->a_head.a_dev;
 279         struct bio *bio = ap->a_bio;
 280         struct buf *bp;
 281         struct bio *nbio;
 282         int unit;
 283         struct vn_softc *vn;
 284         int error;
 285
 286         unit = dkunit(dev);
 287         if ((vn = dev->si_drv1) == NULL)
 288                 vn = vnfindvn(dev);
 289
 290         bp = bio->bio_buf;
 291
 292         IFOPT(vn, VN_DEBUG)
 293                 kprintf("vnstrategy(%p): unit %d\n", bp, unit);
 294
 295         if ((vn->sc_flags & VNF_INITED) == 0) {
 296                 bp->b_error = ENXIO;
 297                 bp->b_flags |= B_ERROR;
 298                 biodone(bio);
 299                 return(0);
 300         }
 301
 302         bp->b_resid = bp->b_bcount;
 303
 304         /*
 305          * The vnode device is using disk/slice label support.
 306          *
 307          * The dscheck() function is called for validating the
 308          * slices that exist ON the vnode device itself, and
 309          * translate the "slice-relative" block number, again.
 310          * dscheck() will call biodone() and return NULL if
 311          * we are at EOF or beyond the device size.
 312          */
 313
 314         nbio = bio;
 315
 316         /*
 317          * Use the translated nbio from this point on
 318          */
 319         if (vn->sc_vp && bp->b_cmd == BUF_CMD_FREEBLKS) {
 320                 /*
 321                  * Freeblks is not handled for vnode-backed elements yet.
 322                  */
 323                 bp->b_resid = 0;
 324                 /* operation complete */
 325         } else if (vn->sc_vp) {
 326                 /*
 327                  * VNODE I/O
 328                  *
 329                  * If an error occurs, we set B_ERROR but we do not set
 330                  * B_INVAL because (for a write anyway), the buffer is
 331                  * still valid.
 332                  */
 333                 struct uio auio;
 334                 struct iovec aiov;
 335
 336                 bzero(&auio, sizeof(auio));
 337
 338                 aiov.iov_base = bp->b_data;
 339                 aiov.iov_len = bp->b_bcount;
 340                 auio.uio_iov = &aiov;
 341                 auio.uio_iovcnt = 1;
 342                 auio.uio_offset = nbio->bio_offset;
 343                 auio.uio_segflg = UIO_SYSSPACE;
 344                 if (bp->b_cmd == BUF_CMD_READ)
 345                         auio.uio_rw = UIO_READ;
 346                 else
 347                         auio.uio_rw = UIO_WRITE;
 348                 auio.uio_resid = bp->b_bcount;
 349                 auio.uio_td = curthread;
 350                 vn_lock(vn->sc_vp, LK_EXCLUSIVE | LK_RETRY);
 351                 if (bp->b_cmd == BUF_CMD_READ)
 352                         error = VOP_READ(vn->sc_vp, &auio, IO_DIRECT | IO_RECURSE, vn->sc_cred);
 353                 else
 354                         error = VOP_WRITE(vn->sc_vp, &auio, IO_DIRECT | IO_RECURSE, vn->sc_cred);
 355                 vn_unlock(vn->sc_vp);
 356                 bp->b_resid = auio.uio_resid;
 357                 if (error) {
 358                         bp->b_error = error;
 359                         bp->b_flags |= B_ERROR;
 360                 }
 361                 /* operation complete */
 362         } else if (vn->sc_object) {
 363                 /*
 364                  * OBJT_SWAP I/O (handles read, write, freebuf)
 365                  *
 366                  * We have nothing to do if freeing  blocks on a reserved
 367                  * swap area, othrewise execute the op.
 368                  */
 369                 if (bp->b_cmd == BUF_CMD_FREEBLKS && TESTOPT(vn, VN_RESERVE)) {
 370                         bp->b_resid = 0;
 371                         /* operation complete */
 372                 } else {
 373                         vm_pager_strategy(vn->sc_object, nbio);
 374                         return(0);
 375                         /* NOT REACHED */
 376                 }
 377         } else {
 378                 bp->b_resid = bp->b_bcount;
 379                 bp->b_flags |= B_ERROR | B_INVAL;
 380                 bp->b_error = EINVAL;
 381                 /* operation complete */
 382         }
 383         biodone(nbio);
 384         return(0);
 385 }
 386
 387 /* ARGSUSED */
 388 static  int
 389 vnioctl(struct dev_ioctl_args *ap)
 390 {
 391         cdev_t dev = ap->a_head.a_dev;
 392         struct vn_softc *vn;
 393         struct vn_ioctl *vio;
 394         int error;
 395         u_long *f;
 396
 397         vn = dev->si_drv1;
 398         IFOPT(vn,VN_FOLLOW) {
 399                 kprintf("vnioctl(%s, 0x%lx, %p, 0x%x): unit %d\n",
 400                     devtoname(dev), ap->a_cmd, ap->a_data, ap->a_fflag,
 401                     dkunit(dev));
 402         }
 403
 404         switch (ap->a_cmd) {
 405         case VNIOCATTACH:
 406         case VNIOCDETACH:
 407         case VNIOCGSET:
 408         case VNIOCGCLEAR:
 409         case VNIOCGET:
 410         case VNIOCUSET:
 411         case VNIOCUCLEAR:
 412                 goto vn_specific;
 413         }
 414
 415 #if 0
 416         if (dkslice(dev) != WHOLE_DISK_SLICE ||
 417                 dkpart(dev) != WHOLE_SLICE_PART)
 418                 return (ENOTTY);
 419 #endif
 420
 421     vn_specific:
 422
 423         error = priv_check_cred(ap->a_cred, PRIV_ROOT, 0);
 424         if (error)
 425                 return (error);
 426
 427         vio = (struct vn_ioctl *)ap->a_data;
 428         f = (u_long*)ap->a_data;
 429
 430         switch (ap->a_cmd) {
 431         case VNIOCATTACH:
 432                 if (vn->sc_flags & VNF_INITED)
 433                         return(EBUSY);
 434
 435                 if (vio->vn_file == NULL)
 436                         error = vniocattach_swap(vn, vio, dev, ap->a_fflag, ap->a_cred);
 437                 else
 438                         error = vniocattach_file(vn, vio, dev, ap->a_fflag, ap->a_cred);
 439                 break;
 440
 441         case VNIOCDETACH:
 442                 if ((vn->sc_flags & VNF_INITED) == 0)
 443                         return(ENXIO);
 444                 /*
 445                  * XXX handle i/o in progress.  Return EBUSY, or wait, or
 446                  * flush the i/o.
 447                  * XXX handle multiple opens of the device.  Return EBUSY,
 448                  * or revoke the fd's.
 449                  * How are these problems handled for removable and failing
 450                  * hardware devices? (Hint: They are not)
 451                  */
 452                 if (count_dev(vn->sc_devlist) > 1)
 453                         return (EBUSY);
 454
 455                 vnclear(vn);
 456                 IFOPT(vn, VN_FOLLOW)
 457                         kprintf("vnioctl: CLRed\n");
 458
 459                 if (dkunit(dev) >= VN_PREALLOCATED_UNITS) {
 460                         devfs_clone_bitmap_put(&DEVFS_CLONE_BITMAP(vn), dkunit(dev));
 461                         disk_destroy(&vn->sc_disk);
 462                 }
 463
 464                 break;
 465
 466         case VNIOCGET:
 467                 error = vnget(dev, vn, (struct vn_user *) ap->a_data);
 468                 break;
 469
 470         case VNIOCGSET:
 471                 vn_options |= *f;
 472                 *f = vn_options;
 473                 break;
 474
 475         case VNIOCGCLEAR:
 476                 vn_options &= ~(*f);
 477                 *f = vn_options;
 478                 break;
 479
 480         case VNIOCUSET:
 481                 vn->sc_options |= *f;
 482                 *f = vn->sc_options;
 483                 break;
 484
 485         case VNIOCUCLEAR:
 486                 vn->sc_options &= ~(*f);
 487                 *f = vn->sc_options;
 488                 break;
 489
 490         default:
 491                 error = ENOTTY;
 492                 break;
 493         }
 494         return(error);
 495 }
 496
 497 /*
 498  *      vniocattach_file:
 499  *
 500  *      Attach a file to a VN partition.  Return the size in the vn_size
 501  *      field.
 502  */
 503
 504 static int
 505 vniocattach_file(struct vn_softc *vn, struct vn_ioctl *vio, cdev_t dev,
 506                  int flag, struct ucred *cred)
 507 {
 508         struct vattr vattr;
 509         struct nlookupdata nd;
 510         int error, flags;
 511         struct vnode *vp;
 512         struct disk_info info;
 513
 514         flags = FREAD|FWRITE;
 515         error = nlookup_init(&nd, vio->vn_file,
 516                                 UIO_USERSPACE, NLC_FOLLOW|NLC_LOCKVP);
 517         if (error)
 518                 return (error);
 519         if ((error = vn_open(&nd, NULL, flags, 0)) != 0) {
 520                 if (error != EACCES && error != EPERM && error != EROFS)
 521                         goto done;
 522                 flags &= ~FWRITE;
 523                 nlookup_done(&nd);
 524                 error = nlookup_init(&nd, vio->vn_file, UIO_USERSPACE, NLC_FOLLOW|NLC_LOCKVP);
 525                 if (error)
 526                         return (error);
 527                 if ((error = vn_open(&nd, NULL, flags, 0)) != 0)
 528                         goto done;
 529         }
 530         vp = nd.nl_open_vp;
 531         if (vp->v_type != VREG ||
 532             (error = VOP_GETATTR(vp, &vattr))) {
 533                 if (error == 0)
 534                         error = EINVAL;
 535                 goto done;
 536         }
 537         vn_unlock(vp);
 538         vn->sc_secsize = DEV_BSIZE;
 539         vn->sc_vp = vp;
 540         nd.nl_open_vp = NULL;
 541
 542         /*
 543          * If the size is specified, override the file attributes.  Note that
 544          * the vn_size argument is in PAGE_SIZE sized blocks.
 545          */
 546         if (vio->vn_size)
 547                 vn->sc_size = vio->vn_size * PAGE_SIZE / vn->sc_secsize;
 548         else
 549                 vn->sc_size = vattr.va_size / vn->sc_secsize;
 550         error = vnsetcred(vn, cred);
 551         if (error) {
 552                 vn->sc_vp = NULL;
 553                 vn_close(vp, flags);
 554                 goto done;
 555         }
 556         vn->sc_flags |= VNF_INITED;
 557         if (flags == FREAD)
 558                 vn->sc_flags |= VNF_READONLY;
 559
 560         /*
 561          * Set the disk info so that probing is triggered
 562          */
 563         bzero(&info, sizeof(struct disk_info));
 564         info.d_media_blksize = vn->sc_secsize;
 565         info.d_media_blocks = vn->sc_size;
 566         /*
 567          * reserve mbr sector for backwards compatibility
 568          * when no slices exist.
 569          */
 570         info.d_dsflags = DSO_COMPATMBR;
 571         info.d_secpertrack = 32;
 572         info.d_nheads = 64 / (vn->sc_secsize / DEV_BSIZE);
 573         info.d_secpercyl = info.d_secpertrack * info.d_nheads;
 574         info.d_ncylinders = vn->sc_size / info.d_secpercyl;
 575         disk_setdiskinfo_sync(&vn->sc_disk, &info);
 576
 577         error = dev_dopen(dev, flag, S_IFCHR, cred);
 578         if (error)
 579                 vnclear(vn);
 580
 581         IFOPT(vn, VN_FOLLOW)
 582                 kprintf("vnioctl: SET vp %p size %llx blks\n",
 583                        vn->sc_vp, (long long)vn->sc_size);
 584 done:
 585         nlookup_done(&nd);
 586         return(error);
 587 }
 588
 589 /*
 590  *      vniocattach_swap:
 591  *
 592  *      Attach swap backing store to a VN partition of the size specified
 593  *      in vn_size.
 594  */
 595
 596 static int
 597 vniocattach_swap(struct vn_softc *vn, struct vn_ioctl *vio, cdev_t dev,
 598                  int flag, struct ucred *cred)
 599 {
 600         int error;
 601         struct disk_info info;
 602
 603         /*
 604          * Range check.  Disallow negative sizes or any size less then the
 605          * size of a page.  Then round to a page.
 606          */
 607
 608         if (vio->vn_size <= 0)
 609                 return(EDOM);
 610
 611         /*
 612          * Allocate an OBJT_SWAP object.
 613          *
 614          * sc_secsize is PAGE_SIZE'd
 615          *
 616          * vio->vn_size is in PAGE_SIZE'd chunks.
 617          * sc_size must be in PAGE_SIZE'd chunks.
 618          * Note the truncation.
 619          */
 620
 621         vn->sc_secsize = PAGE_SIZE;
 622         vn->sc_size = vio->vn_size;
 623         vn->sc_object = vm_pager_allocate(OBJT_SWAP, NULL,
 624                                           vn->sc_secsize * (off_t)vio->vn_size,
 625                                           VM_PROT_DEFAULT, 0);
 626         IFOPT(vn, VN_RESERVE) {
 627                 if (swap_pager_reserve(vn->sc_object, 0, vn->sc_size) < 0) {
 628                         vm_pager_deallocate(vn->sc_object);
 629                         vn->sc_object = NULL;
 630                         return(EDOM);
 631                 }
 632         }
 633         vn->sc_flags |= VNF_INITED;
 634
 635         error = vnsetcred(vn, cred);
 636         if (error == 0) {
 637                 /*
 638                  * Set the disk info so that probing is triggered
 639                  */
 640                 bzero(&info, sizeof(struct disk_info));
 641                 info.d_media_blksize = vn->sc_secsize;
 642                 info.d_media_blocks = vn->sc_size;
 643                 /*
 644                  * reserve mbr sector for backwards compatibility
 645                  * when no slices exist.
 646                  */
 647                 info.d_dsflags = DSO_COMPATMBR;
 648                 info.d_secpertrack = 32;
 649                 info.d_nheads = 64 / (vn->sc_secsize / DEV_BSIZE);
 650                 info.d_secpercyl = info.d_secpertrack * info.d_nheads;
 651                 info.d_ncylinders = vn->sc_size / info.d_secpercyl;
 652                 disk_setdiskinfo_sync(&vn->sc_disk, &info);
 653
 654                 error = dev_dopen(dev, flag, S_IFCHR, cred);
 655         }
 656         if (error == 0) {
 657                 IFOPT(vn, VN_FOLLOW) {
 658                         kprintf("vnioctl: SET vp %p size %llx\n",
 659                                vn->sc_vp, (long long)vn->sc_size);
 660                 }
 661         }
 662         if (error)
 663                 vnclear(vn);
 664         return(error);
 665 }
 666
 667 /*
 668  * Duplicate the current processes' credentials.  Since we are called only
 669  * as the result of a SET ioctl and only root can do that, any future access
 670  * to this "disk" is essentially as root.  Note that credentials may change
 671  * if some other uid can write directly to the mapped file (NFS).
 672  */
 673 int
 674 vnsetcred(struct vn_softc *vn, struct ucred *cred)
 675 {
 676         char *tmpbuf;
 677         int error = 0;
 678
 679         /*
 680          * Set credits in our softc
 681          */
 682
 683         if (vn->sc_cred)
 684                 crfree(vn->sc_cred);
 685         vn->sc_cred = crdup(cred);
 686
 687         /*
 688          * Horrible kludge to establish credentials for NFS  XXX.
 689          */
 690
 691         if (vn->sc_vp) {
 692                 struct uio auio;
 693                 struct iovec aiov;
 694
 695                 tmpbuf = kmalloc(vn->sc_secsize, M_TEMP, M_WAITOK);
 696                 bzero(&auio, sizeof(auio));
 697
 698                 aiov.iov_base = tmpbuf;
 699                 aiov.iov_len = vn->sc_secsize;
 700                 auio.uio_iov = &aiov;
 701                 auio.uio_iovcnt = 1;
 702                 auio.uio_offset = 0;
 703                 auio.uio_rw = UIO_READ;
 704                 auio.uio_segflg = UIO_SYSSPACE;
 705                 auio.uio_resid = aiov.iov_len;
 706                 vn_lock(vn->sc_vp, LK_EXCLUSIVE | LK_RETRY);
 707                 error = VOP_READ(vn->sc_vp, &auio, 0, vn->sc_cred);
 708                 vn_unlock(vn->sc_vp);
 709                 kfree(tmpbuf, M_TEMP);
 710         }
 711         return (error);
 712 }
 713
 714 void
 715 vnclear(struct vn_softc *vn)
 716 {
 717         IFOPT(vn, VN_FOLLOW)
 718                 kprintf("vnclear(%p): vp=%p\n", vn, vn->sc_vp);
 719         vn->sc_flags &= ~VNF_INITED;
 720         if (vn->sc_vp != NULL) {
 721                 vn_close(vn->sc_vp,
 722                     (vn->sc_flags & VNF_READONLY) ?  FREAD : (FREAD|FWRITE));
 723                 vn->sc_vp = NULL;
 724         }
 725         vn->sc_flags &= ~VNF_READONLY;
 726         if (vn->sc_cred) {
 727                 crfree(vn->sc_cred);
 728                 vn->sc_cred = NULL;
 729         }
 730         if (vn->sc_object != NULL) {
 731                 vm_pager_deallocate(vn->sc_object);
 732                 vn->sc_object = NULL;
 733         }
 734
 735         disk_unprobe(&vn->sc_disk);
 736
 737         vn->sc_size = 0;
 738 }
 739
 740 /*
 741  *      vnget:
 742  *
 743  *      populate a struct vn_user for the VNIOCGET ioctl.
 744  *      interface conventions defined in sys/sys/vnioctl.h.
 745  */
 746
 747 static int
 748 vnget(cdev_t dev, struct vn_softc *vn, struct vn_user *vnu)
 749 {
 750         int error, found = 0;
 751         char *freepath, *fullpath;
 752         struct vattr vattr;
 753
 754         if (vnu->vnu_unit == -1) {
 755                 vnu->vnu_unit = dkunit(dev);
 756         }
 757         else if (vnu->vnu_unit < 0)
 758                 return (EINVAL);
 759
 760         SLIST_FOREACH(vn, &vn_list, sc_list) {
 761
 762                 if(vn->sc_unit != vnu->vnu_unit)
 763                         continue;
 764
 765                 found = 1;
 766
 767                 if (vn->sc_flags & VNF_INITED && vn->sc_vp != NULL) {
 768
 769                         /* note: u_cred checked in vnioctl above */
 770                         error = VOP_GETATTR(vn->sc_vp, &vattr);
 771                         if (error) {
 772                                 kprintf("vnget: VOP_GETATTR for %p failed\n",
 773                                         vn->sc_vp);
 774                                 return (error);
 775                         }
 776
 777                         error = vn_fullpath(curproc, vn->sc_vp,
 778                                                 &fullpath, &freepath);
 779
 780                         if (error) {
 781                                 kprintf("vnget: unable to resolve vp %p\n",
 782                                         vn->sc_vp);
 783                                 return(error);
 784                         }
 785
 786                         strlcpy(vnu->vnu_file, fullpath,
 787                                 sizeof(vnu->vnu_file));
 788                         kfree(freepath, M_TEMP);
 789                         vnu->vnu_dev = vattr.va_fsid;
 790                         vnu->vnu_ino = vattr.va_fileid;
 791
 792                 }
 793                 else if (vn->sc_flags & VNF_INITED && vn->sc_object != NULL){
 794
 795                         strlcpy(vnu->vnu_file, _VN_USER_SWAP,
 796                                 sizeof(vnu->vnu_file));
 797                         vnu->vnu_size = vn->sc_size;
 798                         vnu->vnu_secsize = vn->sc_secsize;
 799
 800                 } else {
 801
 802                         bzero(vnu->vnu_file, sizeof(vnu->vnu_file));
 803                         vnu->vnu_dev = 0;
 804                         vnu->vnu_ino = 0;
 805
 806                 }
 807                 break;
 808         }
 809
 810         if (!found)
 811                 return(ENXIO);
 812
 813         return(0);
 814 }
 815
 816 static int
 817 vnsize(struct dev_psize_args *ap)
 818 {
 819         cdev_t dev = ap->a_head.a_dev;
 820         struct vn_softc *vn;
 821
 822         vn = dev->si_drv1;
 823         if (!vn)
 824                 return(ENXIO);
 825         if ((vn->sc_flags & VNF_INITED) == 0)
 826                 return(ENXIO);
 827         ap->a_result = (int64_t)vn->sc_size;
 828         return(0);
 829 }
 830
 831 static cdev_t
 832 vn_create(int unit, struct devfs_bitmap *bitmap)
 833 {
 834         struct vn_softc *vn;
 835         struct disk_info info;
 836         cdev_t dev;
 837
 838         vn = vncreatevn();
 839         dev = disk_create(unit, &vn->sc_disk, &vn_ops);
 840         vninitvn(vn, dev);
 841
 842         bzero(&info, sizeof(struct disk_info));
 843         info.d_media_blksize = 512;
 844         info.d_media_blocks = 0;
 845         info.d_dsflags = DSO_MBRQUIET;
 846         info.d_secpertrack = 32;
 847         info.d_nheads = 64;
 848         info.d_secpercyl = info.d_secpertrack * info.d_nheads;
 849         info.d_ncylinders = 0;
 850         disk_setdiskinfo_sync(&vn->sc_disk, &info);
 851
 852         if (bitmap != NULL)
 853                 devfs_clone_bitmap_set(bitmap, unit);
 854
 855         return dev;
 856 }
 857
 858 static int
 859 vn_modevent(module_t mod, int type, void *data)
 860 {
 861         struct vn_softc *vn;
 862         struct disk_info info;
 863         static cdev_t dev = NULL;
 864         int i;
 865
 866         switch (type) {
 867         case MOD_LOAD:
 868                 dev = make_autoclone_dev(&vn_ops, &DEVFS_CLONE_BITMAP(vn), vnclone, UID_ROOT,
 869                     GID_OPERATOR, 0640, "vn");
 870
 871                 for (i = 0; i < VN_PREALLOCATED_UNITS; i++) {
 872                         vn_create(i, &DEVFS_CLONE_BITMAP(vn));
 873                 }
 874                 break;
 875         case MOD_UNLOAD:
 876                 destroy_autoclone_dev(dev, &DEVFS_CLONE_BITMAP(vn));
 877                 /* fall through */
 878         case MOD_SHUTDOWN:
 879                 while ((vn = SLIST_FIRST(&vn_list)) != NULL) {
 880                         SLIST_REMOVE_HEAD(&vn_list, sc_list);
 881                         if (vn->sc_flags & VNF_INITED)
 882                                 vnclear(vn);
 883                         /* Cleanup all cdev_t's that refer to this unit */
 884                         while ((dev = vn->sc_devlist) != NULL) {
 885                                 vn->sc_devlist = dev->si_drv2;
 886                                 dev->si_drv1 = dev->si_drv2 = NULL;
 887                                 destroy_dev(dev);
 888                         }
 889                         kfree(vn, M_DEVBUF);
 890                 }
 891                 dev_ops_remove_all(&vn_ops);
 892                 break;
 893         default:
 894                 break;
 895         }
 896         return 0;
 897 }
 898
 899 DEV_MODULE(vn, vn_modevent, 0);