sys/kern/subr_disk.c

   1 /*
   2  * ----------------------------------------------------------------------------
   3  * "THE BEER-WARE LICENSE" (Revision 42):
   4  * <phk@FreeBSD.ORG> wrote this file.  As long as you retain this notice you
   5  * can do whatever you want with this stuff. If we meet some day, and you think
   6  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
   7  * ----------------------------------------------------------------------------
   8  *
   9  * Copyright (c) 2004 Matthew Dillon.
  10  * Copyright (c) 1982, 1986, 1988, 1993
  11  *      The Regents of the University of California.  All rights reserved.
  12  * (c) UNIX System Laboratories, Inc.
  13  * All or some portions of this file are derived from material licensed
  14  * to the University of California by American Telephone and Telegraph
  15  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
  16  * the permission of UNIX System Laboratories, Inc.
  17  *
  18  * Redistribution and use in source and binary forms, with or without
  19  * modification, are permitted provided that the following conditions
  20  * are met:
  21  * 1. Redistributions of source code must retain the above copyright
  22  *    notice, this list of conditions and the following disclaimer.
  23  * 2. Redistributions in binary form must reproduce the above copyright
  24  *    notice, this list of conditions and the following disclaimer in the
  25  *    documentation and/or other materials provided with the distribution.
  26  * 3. All advertising materials mentioning features or use of this software
  27  *    must display the following acknowledgement:
  28  *      This product includes software developed by the University of
  29  *      California, Berkeley and its contributors.
  30  * 4. Neither the name of the University nor the names of its contributors
  31  *    may be used to endorse or promote products derived from this software
  32  *    without specific prior written permission.
  33  *
  34  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  35  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  36  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  37  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  38  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  39  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  40  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  41  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  42  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  43  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  44  * SUCH DAMAGE.
  45  *
  46  *      @(#)ufs_disksubr.c      8.5 (Berkeley) 1/21/94
  47  * $FreeBSD: src/sys/kern/subr_disk.c,v 1.20.2.6 2001/10/05 07:14:57 peter Exp $
  48  * $FreeBSD: src/sys/ufs/ufs/ufs_disksubr.c,v 1.44.2.3 2001/03/05 05:42:19 obrien Exp $
  49  * $DragonFly: src/sys/kern/subr_disk.c,v 1.10 2004/05/19 22:52:58 dillon Exp $
  50  */
  51
  52 #include <sys/param.h>
  53 #include <sys/systm.h>
  54 #include <sys/kernel.h>
  55 #include <sys/proc.h>
  56 #include <sys/sysctl.h>
  57 #include <sys/buf.h>
  58 #include <sys/conf.h>
  59 #include <sys/disklabel.h>
  60 #include <sys/diskslice.h>
  61 #include <sys/disk.h>
  62 #include <sys/malloc.h>
  63 #include <sys/sysctl.h>
  64 #include <machine/md_var.h>
  65 #include <sys/ctype.h>
  66 #include <sys/syslog.h>
  67 #include <sys/device.h>
  68 #include <sys/msgport.h>
  69 #include <sys/msgport2.h>
  70 #include <sys/buf2.h>
  71
  72 static MALLOC_DEFINE(M_DISK, "disk", "disk data");
  73
  74 static d_strategy_t diskstrategy;
  75 static d_open_t diskopen;
  76 static d_close_t diskclose;
  77 static d_ioctl_t diskioctl;
  78 static d_psize_t diskpsize;
  79 static d_clone_t diskclone;
  80 static int disk_putport(lwkt_port_t port, lwkt_msg_t msg);
  81
  82 static LIST_HEAD(, disk) disklist = LIST_HEAD_INITIALIZER(&disklist);
  83
  84 /*
  85  * Create a slice and unit managed disk.
  86  *
  87  * Our port layer will be responsible for assigning pblkno and handling
  88  * high level partition operations, then forwarding the requests to the
  89  * raw device.
  90  *
  91  * The raw device (based on rawsw) is returned to the caller, NOT the
  92  * slice and unit managed cdev.  The caller typically sets various
  93  * driver parameters and IO limits on the returned rawdev which we must
  94  * inherit when our managed device is opened.
  95  */
  96 dev_t
  97 disk_create(int unit, struct disk *dp, int flags, struct cdevsw *rawsw)
  98 {
  99         dev_t rawdev;
 100         struct cdevsw *devsw;
 101
 102         /*
 103          * Create the raw backing device
 104          */
 105         compile_devsw(rawsw);
 106         rawdev = make_dev(rawsw,
 107                             dkmakeminor(unit, WHOLE_DISK_SLICE, RAW_PART),
 108                             UID_ROOT, GID_OPERATOR, 0640,
 109                             "%s%d", rawsw->d_name, unit);
 110
 111         /*
 112          * Initialize our intercept port
 113          */
 114         bzero(dp, sizeof(*dp));
 115         lwkt_initport(&dp->d_port, NULL);
 116         dp->d_port.mp_putport = disk_putport;
 117         dp->d_rawsw = rawsw;
 118
 119         /*
 120          * We install a custom cdevsw rather then the passed cdevsw,
 121          * and save our disk structure in d_data so we can get at it easily
 122          * without any complex cloning code.
 123          */
 124         devsw = cdevsw_add_override(rawdev, dkunitmask(), dkmakeunit(unit));
 125         devsw->d_port = &dp->d_port;
 126         devsw->d_data = dp;
 127         devsw->d_clone = diskclone;
 128         dp->d_devsw = devsw;
 129         dp->d_rawdev = rawdev;
 130         dp->d_cdev = make_dev(devsw,
 131                             dkmakeminor(unit, WHOLE_DISK_SLICE, RAW_PART),
 132                             UID_ROOT, GID_OPERATOR, 0640,
 133                             "%s%d", devsw->d_name, unit);
 134
 135         dp->d_dsflags = flags;
 136         LIST_INSERT_HEAD(&disklist, dp, d_list);
 137         return (dp->d_rawdev);
 138 }
 139
 140 /*
 141  * This routine is called when an adapter detaches.  The higher level
 142  * managed disk device is destroyed while the lower level raw device is
 143  * released.
 144  */
 145 void
 146 disk_destroy(struct disk *disk)
 147 {
 148         if (disk->d_devsw) {
 149             cdevsw_remove(disk->d_devsw, dkunitmask(), dkunit(disk->d_cdev));
 150             LIST_REMOVE(disk, d_list);
 151         }
 152         if (disk->d_rawsw)
 153             destroy_all_dev(disk->d_rawsw, dkunitmask(), dkunit(disk->d_rawdev));
 154         bzero(disk, sizeof(*disk));
 155 }
 156
 157 int
 158 disk_dumpcheck(dev_t dev, u_int *count, u_int *blkno, u_int *secsize)
 159 {
 160         struct disk *dp;
 161         struct disklabel *dl;
 162         u_int boff;
 163
 164         dp = dev->si_disk;
 165         if (!dp)
 166                 return (ENXIO);
 167         if (!dp->d_slice)
 168                 return (ENXIO);
 169         dl = dsgetlabel(dev, dp->d_slice);
 170         if (!dl)
 171                 return (ENXIO);
 172         *count = Maxmem * (PAGE_SIZE / dl->d_secsize);
 173         if (dumplo <= LABELSECTOR ||
 174             (dumplo + *count > dl->d_partitions[dkpart(dev)].p_size))
 175                 return (EINVAL);
 176         boff = dl->d_partitions[dkpart(dev)].p_offset +
 177             dp->d_slice->dss_slices[dkslice(dev)].ds_offset;
 178         *blkno = boff + dumplo;
 179         *secsize = dl->d_secsize;
 180         return (0);
 181
 182 }
 183
 184 void
 185 disk_invalidate (struct disk *disk)
 186 {
 187         if (disk->d_slice)
 188                 dsgone(&disk->d_slice);
 189 }
 190
 191 struct disk *
 192 disk_enumerate(struct disk *disk)
 193 {
 194         if (!disk)
 195                 return (LIST_FIRST(&disklist));
 196         else
 197                 return (LIST_NEXT(disk, d_list));
 198 }
 199
 200 static int
 201 sysctl_disks(SYSCTL_HANDLER_ARGS)
 202 {
 203         struct disk *disk;
 204         int error, first;
 205
 206         disk = NULL;
 207         first = 1;
 208
 209         while ((disk = disk_enumerate(disk))) {
 210                 if (!first) {
 211                         error = SYSCTL_OUT(req, " ", 1);
 212                         if (error)
 213                                 return error;
 214                 } else {
 215                         first = 0;
 216                 }
 217                 error = SYSCTL_OUT(req, disk->d_rawdev->si_name, strlen(disk->d_rawdev->si_name));
 218                 if (error)
 219                         return error;
 220         }
 221         error = SYSCTL_OUT(req, "", 1);
 222         return error;
 223 }
 224
 225 SYSCTL_PROC(_kern, OID_AUTO, disks, CTLTYPE_STRING | CTLFLAG_RD, 0, NULL,
 226     sysctl_disks, "A", "names of available disks");
 227
 228 /*
 229  * The port intercept functions
 230  */
 231 static
 232 int
 233 disk_putport(lwkt_port_t port, lwkt_msg_t lmsg)
 234 {
 235         struct disk *disk = (struct disk *)port;
 236         cdevallmsg_t msg = (cdevallmsg_t)lmsg;
 237         int error;
 238
 239         switch(msg->am_lmsg.ms_cmd.cm_op) {
 240         case CDEV_CMD_OPEN:
 241                 error = diskopen(
 242                             msg->am_open.msg.dev,
 243                             msg->am_open.oflags,
 244                             msg->am_open.devtype,
 245                             msg->am_open.td);
 246                 break;
 247         case CDEV_CMD_CLOSE:
 248                 error = diskclose(
 249                             msg->am_close.msg.dev,
 250                             msg->am_close.fflag,
 251                             msg->am_close.devtype,
 252                             msg->am_close.td);
 253                 break;
 254         case CDEV_CMD_IOCTL:
 255                 error = diskioctl(
 256                             msg->am_ioctl.msg.dev,
 257                             msg->am_ioctl.cmd,
 258                             msg->am_ioctl.data,
 259                             msg->am_ioctl.fflag,
 260                             msg->am_ioctl.td);
 261                 break;
 262         case CDEV_CMD_STRATEGY:
 263                 diskstrategy(msg->am_strategy.bp);
 264                 error = 0;
 265                 break;
 266         case CDEV_CMD_PSIZE:
 267                 msg->am_psize.result = diskpsize(msg->am_psize.msg.dev);
 268                 error = 0;      /* XXX */
 269                 break;
 270         case CDEV_CMD_READ:
 271                 error = physio(msg->am_read.msg.dev,
 272                                 msg->am_read.uio, msg->am_read.ioflag);
 273                 break;
 274         case CDEV_CMD_WRITE:
 275                 error = physio(msg->am_write.msg.dev,
 276                                 msg->am_write.uio, msg->am_write.ioflag);
 277                 break;
 278         case CDEV_CMD_POLL:
 279         case CDEV_CMD_KQFILTER:
 280                 error = ENODEV;
 281         case CDEV_CMD_MMAP:
 282                 error = -1;
 283                 break;
 284         case CDEV_CMD_DUMP:
 285                 error = disk_dumpcheck(msg->am_dump.msg.dev,
 286                                 &msg->am_dump.count,
 287                                 &msg->am_dump.blkno,
 288                                 &msg->am_dump.secsize);
 289                 if (error == 0) {
 290                         msg->am_dump.msg.dev = disk->d_rawdev;
 291                         error = lwkt_forwardmsg(disk->d_rawdev->si_port,
 292                                                 &msg->am_dump.msg.msg);
 293                         printf("error2 %d\n", error);
 294                 }
 295                 break;
 296         default:
 297                 error = ENOTSUP;
 298                 break;
 299         }
 300         return(error);
 301 }
 302
 303 /*
 304  * When new device entries are instantiated, make sure they inherit our
 305  * si_disk structure and block and iosize limits from the raw device.
 306  *
 307  * This routine is always called synchronously in the context of the
 308  * client.
 309  *
 310  * XXX The various io and block size constraints are not always initialized
 311  * properly by devices.
 312  */
 313 static
 314 int
 315 diskclone(dev_t dev)
 316 {
 317         struct disk *dp;
 318
 319         dp = dev->si_devsw->d_data;
 320         KKASSERT(dp != NULL);
 321         dev->si_disk = dp;
 322         dev->si_iosize_max = dp->d_rawdev->si_iosize_max;
 323         dev->si_bsize_phys = dp->d_rawdev->si_bsize_phys;
 324         dev->si_bsize_best = dp->d_rawdev->si_bsize_best;
 325         return(0);
 326 }
 327
 328 /*
 329  * Open a disk device or partition.
 330  */
 331 static int
 332 diskopen(dev_t dev, int oflags, int devtype, struct thread *td)
 333 {
 334         struct disk *dp;
 335         int error;
 336
 337         /*
 338          * dp can't be NULL here XXX.
 339          */
 340         error = 0;
 341         dp = dev->si_disk;
 342         if (dp == NULL)
 343                 return (ENXIO);
 344
 345         /*
 346          * Deal with open races
 347          */
 348         while (dp->d_flags & DISKFLAG_LOCK) {
 349                 dp->d_flags |= DISKFLAG_WANTED;
 350                 error = tsleep(dp, PCATCH, "diskopen", hz);
 351                 if (error)
 352                         return (error);
 353         }
 354         dp->d_flags |= DISKFLAG_LOCK;
 355
 356         /*
 357          * Open the underlying raw device.
 358          */
 359         if (!dsisopen(dp->d_slice)) {
 360 #if 0
 361                 if (!pdev->si_iosize_max)
 362                         pdev->si_iosize_max = dev->si_iosize_max;
 363 #endif
 364                 error = dev_dopen(dp->d_rawdev, oflags, devtype, td);
 365         }
 366
 367         /*
 368          * Inherit properties from the underlying device now that it is
 369          * open.
 370          */
 371         diskclone(dev);
 372
 373         if (error)
 374                 goto out;
 375
 376         error = dsopen(dev, devtype, dp->d_dsflags, &dp->d_slice, &dp->d_label);
 377
 378         if (!dsisopen(dp->d_slice))
 379                 dev_dclose(dp->d_rawdev, oflags, devtype, td);
 380 out:
 381         dp->d_flags &= ~DISKFLAG_LOCK;
 382         if (dp->d_flags & DISKFLAG_WANTED) {
 383                 dp->d_flags &= ~DISKFLAG_WANTED;
 384                 wakeup(dp);
 385         }
 386
 387         return(error);
 388 }
 389
 390 /*
 391  * Close a disk device or partition
 392  */
 393 static int
 394 diskclose(dev_t dev, int fflag, int devtype, struct thread *td)
 395 {
 396         struct disk *dp;
 397         int error;
 398
 399         error = 0;
 400         dp = dev->si_disk;
 401
 402         dsclose(dev, devtype, dp->d_slice);
 403         if (!dsisopen(dp->d_slice))
 404                 error = dev_dclose(dp->d_rawdev, fflag, devtype, td);
 405         return (error);
 406 }
 407
 408 /*
 409  * Execute strategy routine
 410  */
 411 static void
 412 diskstrategy(struct buf *bp)
 413 {
 414         struct disk *dp;
 415
 416         dp = bp->b_dev->si_disk;
 417
 418         if (dp == NULL) {
 419                 bp->b_error = ENXIO;
 420                 bp->b_flags |= B_ERROR;
 421                 biodone(bp);
 422                 return;
 423         }
 424         KKASSERT(bp->b_dev->si_disk == dp);
 425
 426         if (dscheck(bp, dp->d_slice) <= 0) {
 427                 biodone(bp);
 428                 return;
 429         }
 430         bp->b_dev = dp->d_rawdev;
 431         dev_dstrategy(dp->d_rawdev, bp);
 432 }
 433
 434 /*
 435  * First execute the ioctl on the disk device, and if it isn't supported
 436  * try running it on the backing device.
 437  */
 438 static int
 439 diskioctl(dev_t dev, u_long cmd, caddr_t data, int fflag, struct thread *td)
 440 {
 441         struct disk *dp;
 442         int error;
 443
 444         dp = dev->si_disk;
 445         if (dp == NULL)
 446                 return (ENXIO);
 447
 448         error = dsioctl(dev, cmd, data, fflag, &dp->d_slice);
 449         if (error == ENOIOCTL)
 450                 error = dev_dioctl(dp->d_rawdev, cmd, data, fflag, td);
 451         return (error);
 452 }
 453
 454 /*
 455  *
 456  */
 457 static int
 458 diskpsize(dev_t dev)
 459 {
 460         struct disk *dp;
 461
 462         dp = dev->si_disk;
 463         if (dp == NULL)
 464                 return (-1);
 465         return(dssize(dev, &dp->d_slice));
 466 #if 0
 467         if (dp != dev->si_disk) {
 468                 dev->si_drv1 = pdev->si_drv1;
 469                 dev->si_drv2 = pdev->si_drv2;
 470                 /* XXX: don't set bp->b_dev->si_disk (?) */
 471         }
 472 #endif
 473 }
 474
 475 SYSCTL_DECL(_debug_sizeof);
 476
 477 SYSCTL_INT(_debug_sizeof, OID_AUTO, disklabel, CTLFLAG_RD,
 478     0, sizeof(struct disklabel), "sizeof(struct disklabel)");
 479
 480 SYSCTL_INT(_debug_sizeof, OID_AUTO, diskslices, CTLFLAG_RD,
 481     0, sizeof(struct diskslices), "sizeof(struct diskslices)");
 482
 483 SYSCTL_INT(_debug_sizeof, OID_AUTO, disk, CTLFLAG_RD,
 484     0, sizeof(struct disk), "sizeof(struct disk)");
 485
 486
 487 /*
 488  * Seek sort for disks.
 489  *
 490  * The buf_queue keep two queues, sorted in ascending block order.  The first
 491  * queue holds those requests which are positioned after the current block
 492  * (in the first request); the second, which starts at queue->switch_point,
 493  * holds requests which came in after their block number was passed.  Thus
 494  * we implement a one way scan, retracting after reaching the end of the drive
 495  * to the first request on the second queue, at which time it becomes the
 496  * first queue.
 497  *
 498  * A one-way scan is natural because of the way UNIX read-ahead blocks are
 499  * allocated.
 500  */
 501 void
 502 bufqdisksort(bufq, bp)
 503         struct buf_queue_head *bufq;
 504         struct buf *bp;
 505 {
 506         struct buf *bq;
 507         struct buf *bn;
 508         struct buf *be;
 509
 510         be = TAILQ_LAST(&bufq->queue, buf_queue);
 511         /*
 512          * If the queue is empty or we are an
 513          * ordered transaction, then it's easy.
 514          */
 515         if ((bq = bufq_first(bufq)) == NULL
 516          || (bp->b_flags & B_ORDERED) != 0) {
 517                 bufq_insert_tail(bufq, bp);
 518                 return;
 519         } else if (bufq->insert_point != NULL) {
 520
 521                 /*
 522                  * A certain portion of the list is
 523                  * "locked" to preserve ordering, so
 524                  * we can only insert after the insert
 525                  * point.
 526                  */
 527                 bq = bufq->insert_point;
 528         } else {
 529
 530                 /*
 531                  * If we lie before the last removed (currently active)
 532                  * request, and are not inserting ourselves into the
 533                  * "locked" portion of the list, then we must add ourselves
 534                  * to the second request list.
 535                  */
 536                 if (bp->b_pblkno < bufq->last_pblkno) {
 537
 538                         bq = bufq->switch_point;
 539                         /*
 540                          * If we are starting a new secondary list,
 541                          * then it's easy.
 542                          */
 543                         if (bq == NULL) {
 544                                 bufq->switch_point = bp;
 545                                 bufq_insert_tail(bufq, bp);
 546                                 return;
 547                         }
 548                         /*
 549                          * If we lie ahead of the current switch point,
 550                          * insert us before the switch point and move
 551                          * the switch point.
 552                          */
 553                         if (bp->b_pblkno < bq->b_pblkno) {
 554                                 bufq->switch_point = bp;
 555                                 TAILQ_INSERT_BEFORE(bq, bp, b_act);
 556                                 return;
 557                         }
 558                 } else {
 559                         if (bufq->switch_point != NULL)
 560                                 be = TAILQ_PREV(bufq->switch_point,
 561                                                 buf_queue, b_act);
 562                         /*
 563                          * If we lie between last_pblkno and bq,
 564                          * insert before bq.
 565                          */
 566                         if (bp->b_pblkno < bq->b_pblkno) {
 567                                 TAILQ_INSERT_BEFORE(bq, bp, b_act);
 568                                 return;
 569                         }
 570                 }
 571         }
 572
 573         /*
 574          * Request is at/after our current position in the list.
 575          * Optimize for sequential I/O by seeing if we go at the tail.
 576          */
 577         if (bp->b_pblkno > be->b_pblkno) {
 578                 TAILQ_INSERT_AFTER(&bufq->queue, be, bp, b_act);
 579                 return;
 580         }
 581
 582         /* Otherwise, insertion sort */
 583         while ((bn = TAILQ_NEXT(bq, b_act)) != NULL) {
 584
 585                 /*
 586                  * We want to go after the current request if it is the end
 587                  * of the first request list, or if the next request is a
 588                  * larger cylinder than our request.
 589                  */
 590                 if (bn == bufq->switch_point
 591                  || bp->b_pblkno < bn->b_pblkno)
 592                         break;
 593                 bq = bn;
 594         }
 595         TAILQ_INSERT_AFTER(&bufq->queue, bq, bp, b_act);
 596 }
 597
 598
 599 /*
 600  * Attempt to read a disk label from a device using the indicated strategy
 601  * routine.  The label must be partly set up before this: secpercyl, secsize
 602  * and anything required in the strategy routine (e.g., dummy bounds for the
 603  * partition containing the label) must be filled in before calling us.
 604  * Returns NULL on success and an error string on failure.
 605  */
 606 char *
 607 readdisklabel(dev_t dev, struct disklabel *lp)
 608 {
 609         struct buf *bp;
 610         struct disklabel *dlp;
 611         char *msg = NULL;
 612
 613         bp = geteblk((int)lp->d_secsize);
 614         bp->b_dev = dev;
 615         bp->b_blkno = LABELSECTOR * ((int)lp->d_secsize/DEV_BSIZE);
 616         bp->b_bcount = lp->d_secsize;
 617         bp->b_flags &= ~B_INVAL;
 618         bp->b_flags |= B_READ;
 619         BUF_STRATEGY(bp, 1);
 620         if (biowait(bp))
 621                 msg = "I/O error";
 622         else for (dlp = (struct disklabel *)bp->b_data;
 623             dlp <= (struct disklabel *)((char *)bp->b_data +
 624             lp->d_secsize - sizeof(*dlp));
 625             dlp = (struct disklabel *)((char *)dlp + sizeof(long))) {
 626                 if (dlp->d_magic != DISKMAGIC || dlp->d_magic2 != DISKMAGIC) {
 627                         if (msg == NULL)
 628                                 msg = "no disk label";
 629                 } else if (dlp->d_npartitions > MAXPARTITIONS ||
 630                            dkcksum(dlp) != 0)
 631                         msg = "disk label corrupted";
 632                 else {
 633                         *lp = *dlp;
 634                         msg = NULL;
 635                         break;
 636                 }
 637         }
 638         bp->b_flags |= B_INVAL | B_AGE;
 639         brelse(bp);
 640         return (msg);
 641 }
 642
 643 /*
 644  * Check new disk label for sensibility before setting it.
 645  */
 646 int
 647 setdisklabel(olp, nlp, openmask)
 648         struct disklabel *olp, *nlp;
 649         u_long openmask;
 650 {
 651         int i;
 652         struct partition *opp, *npp;
 653
 654         /*
 655          * Check it is actually a disklabel we are looking at.
 656          */
 657         if (nlp->d_magic != DISKMAGIC || nlp->d_magic2 != DISKMAGIC ||
 658             dkcksum(nlp) != 0)
 659                 return (EINVAL);
 660         /*
 661          * For each partition that we think is open,
 662          */
 663         while ((i = ffs((long)openmask)) != 0) {
 664                 i--;
 665                 /*
 666                  * Check it is not changing....
 667                  */
 668                 openmask &= ~(1 << i);
 669                 if (nlp->d_npartitions <= i)
 670                         return (EBUSY);
 671                 opp = &olp->d_partitions[i];
 672                 npp = &nlp->d_partitions[i];
 673                 if (npp->p_offset != opp->p_offset || npp->p_size < opp->p_size)
 674                         return (EBUSY);
 675                 /*
 676                  * Copy internally-set partition information
 677                  * if new label doesn't include it.             XXX
 678                  * (If we are using it then we had better stay the same type)
 679                  * This is possibly dubious, as someone else noted (XXX)
 680                  */
 681                 if (npp->p_fstype == FS_UNUSED && opp->p_fstype != FS_UNUSED) {
 682                         npp->p_fstype = opp->p_fstype;
 683                         npp->p_fsize = opp->p_fsize;
 684                         npp->p_frag = opp->p_frag;
 685                         npp->p_cpg = opp->p_cpg;
 686                 }
 687         }
 688         nlp->d_checksum = 0;
 689         nlp->d_checksum = dkcksum(nlp);
 690         *olp = *nlp;
 691         return (0);
 692 }
 693
 694 /*
 695  * Write disk label back to device after modification.
 696  */
 697 int
 698 writedisklabel(dev_t dev, struct disklabel *lp)
 699 {
 700         struct buf *bp;
 701         struct disklabel *dlp;
 702         int error = 0;
 703
 704         if (lp->d_partitions[RAW_PART].p_offset != 0)
 705                 return (EXDEV);                 /* not quite right */
 706         bp = geteblk((int)lp->d_secsize);
 707         bp->b_dev = dkmodpart(dev, RAW_PART);
 708         bp->b_blkno = LABELSECTOR * ((int)lp->d_secsize/DEV_BSIZE);
 709         bp->b_bcount = lp->d_secsize;
 710 #if 1
 711         /*
 712          * We read the label first to see if it's there,
 713          * in which case we will put ours at the same offset into the block..
 714          * (I think this is stupid [Julian])
 715          * Note that you can't write a label out over a corrupted label!
 716          * (also stupid.. how do you write the first one? by raw writes?)
 717          */
 718         bp->b_flags &= ~B_INVAL;
 719         bp->b_flags |= B_READ;
 720         BUF_STRATEGY(bp, 1);
 721         error = biowait(bp);
 722         if (error)
 723                 goto done;
 724         for (dlp = (struct disklabel *)bp->b_data;
 725             dlp <= (struct disklabel *)
 726               ((char *)bp->b_data + lp->d_secsize - sizeof(*dlp));
 727             dlp = (struct disklabel *)((char *)dlp + sizeof(long))) {
 728                 if (dlp->d_magic == DISKMAGIC && dlp->d_magic2 == DISKMAGIC &&
 729                     dkcksum(dlp) == 0) {
 730                         *dlp = *lp;
 731                         bp->b_flags &= ~(B_DONE | B_READ);
 732                         bp->b_flags |= B_WRITE;
 733                         bp->b_dev = dkmodpart(dev, RAW_PART);
 734 #ifdef __alpha__
 735                         alpha_fix_srm_checksum(bp);
 736 #endif
 737                         BUF_STRATEGY(bp, 1);
 738                         error = biowait(bp);
 739                         goto done;
 740                 }
 741         }
 742         error = ESRCH;
 743 done:
 744 #else
 745         bzero(bp->b_data, lp->d_secsize);
 746         dlp = (struct disklabel *)bp->b_data;
 747         *dlp = *lp;
 748         bp->b_flags &= ~B_INVAL;
 749         bp->b_flags |= B_WRITE;
 750         BUF_STRATEGY(bp, 1);
 751         error = biowait(bp);
 752 #endif
 753         bp->b_flags |= B_INVAL | B_AGE;
 754         brelse(bp);
 755         return (error);
 756 }
 757
 758 /*
 759  * Disk error is the preface to plaintive error messages
 760  * about failing disk transfers.  It prints messages of the form
 761
 762 hp0g: hard error reading fsbn 12345 of 12344-12347 (hp0 bn %d cn %d tn %d sn %d)
 763
 764  * if the offset of the error in the transfer and a disk label
 765  * are both available.  blkdone should be -1 if the position of the error
 766  * is unknown; the disklabel pointer may be null from drivers that have not
 767  * been converted to use them.  The message is printed with printf
 768  * if pri is LOG_PRINTF, otherwise it uses log at the specified priority.
 769  * The message should be completed (with at least a newline) with printf
 770  * or addlog, respectively.  There is no trailing space.
 771  */
 772 void
 773 diskerr(bp, what, pri, blkdone, lp)
 774         struct buf *bp;
 775         char *what;
 776         int pri, blkdone;
 777         struct disklabel *lp;
 778 {
 779         int unit = dkunit(bp->b_dev);
 780         int slice = dkslice(bp->b_dev);
 781         int part = dkpart(bp->b_dev);
 782         char partname[2];
 783         char *sname;
 784         daddr_t sn;
 785
 786         sname = dsname(bp->b_dev, unit, slice, part, partname);
 787         printf("%s%s: %s %sing fsbn ", sname, partname, what,
 788               bp->b_flags & B_READ ? "read" : "writ");
 789         sn = bp->b_blkno;
 790         if (bp->b_bcount <= DEV_BSIZE)
 791                 printf("%ld", (long)sn);
 792         else {
 793                 if (blkdone >= 0) {
 794                         sn += blkdone;
 795                         printf("%ld of ", (long)sn);
 796                 }
 797                 printf("%ld-%ld", (long)bp->b_blkno,
 798                     (long)(bp->b_blkno + (bp->b_bcount - 1) / DEV_BSIZE));
 799         }
 800         if (lp && (blkdone >= 0 || bp->b_bcount <= lp->d_secsize)) {
 801 #ifdef tahoe
 802                 sn *= DEV_BSIZE / lp->d_secsize;                /* XXX */
 803 #endif
 804                 sn += lp->d_partitions[part].p_offset;
 805                 /*
 806                  * XXX should add slice offset and not print the slice,
 807                  * but we don't know the slice pointer.
 808                  * XXX should print bp->b_pblkno so that this will work
 809                  * independent of slices, labels and bad sector remapping,
 810                  * but some drivers don't set bp->b_pblkno.
 811                  */
 812                 printf(" (%s bn %ld; cn %ld", sname, (long)sn,
 813                     (long)(sn / lp->d_secpercyl));
 814                 sn %= (long)lp->d_secpercyl;
 815                 printf(" tn %ld sn %ld)", (long)(sn / lp->d_nsectors),
 816                     (long)(sn % lp->d_nsectors));
 817         }
 818 }