sys/geom/geom_dev.c

   1 /*-
   2  * Copyright (c) 2002 Poul-Henning Kamp
   3  * Copyright (c) 2002 Networks Associates Technology, Inc.
   4  * All rights reserved.
   5  *
   6  * This software was developed for the FreeBSD Project by Poul-Henning Kamp
   7  * and NAI Labs, the Security Research Division of Network Associates, Inc.
   8  * under DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"), as part of the
   9  * DARPA CHATS research program.
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  * 3. The names of the authors may not be used to endorse or promote
  20  *    products derived from this software without specific prior written
  21  *    permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35
  36 #include <sys/cdefs.h>
  37 __FBSDID("$FreeBSD$");
  38
  39 #include <sys/param.h>
  40 #include <sys/systm.h>
  41 #include <sys/malloc.h>
  42 #include <sys/kernel.h>
  43 #include <sys/conf.h>
  44 #include <sys/ctype.h>
  45 #include <sys/bio.h>
  46 #include <sys/bus.h>
  47 #include <sys/lock.h>
  48 #include <sys/mutex.h>
  49 #include <sys/proc.h>
  50 #include <sys/errno.h>
  51 #include <sys/time.h>
  52 #include <sys/disk.h>
  53 #include <sys/fcntl.h>
  54 #include <sys/limits.h>
  55 #include <sys/sysctl.h>
  56 #include <geom/geom.h>
  57 #include <geom/geom_int.h>
  58 #include <machine/stdarg.h>
  59
  60 struct g_dev_softc {
  61         struct mtx       sc_mtx;
  62         struct cdev     *sc_dev;
  63         struct cdev     *sc_alias;
  64         int              sc_open;
  65         u_int            sc_active;
  66 #define SC_A_DESTROY    (1 << 31)
  67 #define SC_A_OPEN       (1 << 30)
  68 #define SC_A_ACTIVE     (SC_A_OPEN - 1)
  69 };
  70
  71 static d_open_t         g_dev_open;
  72 static d_close_t        g_dev_close;
  73 static d_strategy_t     g_dev_strategy;
  74 static d_ioctl_t        g_dev_ioctl;
  75
  76 static struct cdevsw g_dev_cdevsw = {
  77         .d_version =    D_VERSION,
  78         .d_open =       g_dev_open,
  79         .d_close =      g_dev_close,
  80         .d_read =       physread,
  81         .d_write =      physwrite,
  82         .d_ioctl =      g_dev_ioctl,
  83         .d_strategy =   g_dev_strategy,
  84         .d_name =       "g_dev",
  85         .d_flags =      D_DISK | D_TRACKCLOSE,
  86 };
  87
  88 static g_init_t g_dev_init;
  89 static g_fini_t g_dev_fini;
  90 static g_taste_t g_dev_taste;
  91 static g_orphan_t g_dev_orphan;
  92 static g_attrchanged_t g_dev_attrchanged;
  93
  94 static struct g_class g_dev_class       = {
  95         .name = "DEV",
  96         .version = G_VERSION,
  97         .init = g_dev_init,
  98         .fini = g_dev_fini,
  99         .taste = g_dev_taste,
 100         .orphan = g_dev_orphan,
 101         .attrchanged = g_dev_attrchanged
 102 };
 103
 104 /*
 105  * We target 262144 (8 x 32768) sectors by default as this significantly
 106  * increases the throughput on commonly used SSD's with a marginal
 107  * increase in non-interruptible request latency.
 108  */
 109 static uint64_t g_dev_del_max_sectors = 262144;
 110 SYSCTL_DECL(_kern_geom);
 111 SYSCTL_NODE(_kern_geom, OID_AUTO, dev, CTLFLAG_RW, 0, "GEOM_DEV stuff");
 112 SYSCTL_QUAD(_kern_geom_dev, OID_AUTO, delete_max_sectors, CTLFLAG_RW,
 113     &g_dev_del_max_sectors, 0, "Maximum number of sectors in a single "
 114     "delete request sent to the provider. Larger requests are chunked "
 115     "so they can be interrupted. (0 = disable chunking)");
 116
 117 static char *dumpdev = NULL;
 118 static void
 119 g_dev_init(struct g_class *mp)
 120 {
 121
 122         dumpdev = kern_getenv("dumpdev");
 123 }
 124
 125 static void
 126 g_dev_fini(struct g_class *mp)
 127 {
 128
 129         freeenv(dumpdev);
 130         dumpdev = NULL;
 131 }
 132
 133 static int
 134 g_dev_setdumpdev(struct cdev *dev, struct thread *td)
 135 {
 136         struct g_kerneldump kd;
 137         struct g_consumer *cp;
 138         int error, len;
 139
 140         if (dev == NULL)
 141                 return (set_dumper(NULL, NULL, td));
 142
 143         cp = dev->si_drv2;
 144         len = sizeof(kd);
 145         kd.offset = 0;
 146         kd.length = OFF_MAX;
 147         error = g_io_getattr("GEOM::kerneldump", cp, &len, &kd);
 148         if (error == 0) {
 149                 error = set_dumper(&kd.di, devtoname(dev), td);
 150                 if (error == 0)
 151                         dev->si_flags |= SI_DUMPDEV;
 152         }
 153         return (error);
 154 }
 155
 156 static int
 157 init_dumpdev(struct cdev *dev)
 158 {
 159         struct g_consumer *cp;
 160         const char *devprefix = "/dev/", *devname;
 161         int error;
 162         size_t len;
 163
 164         if (dumpdev == NULL)
 165                 return (0);
 166
 167         len = strlen(devprefix);
 168         devname = devtoname(dev);
 169         if (strcmp(devname, dumpdev) != 0 &&
 170            (strncmp(dumpdev, devprefix, len) != 0 ||
 171             strcmp(devname, dumpdev + len) != 0))
 172                 return (0);
 173
 174         cp = (struct g_consumer *)dev->si_drv2;
 175         error = g_access(cp, 1, 0, 0);
 176         if (error != 0)
 177                 return (error);
 178
 179         error = g_dev_setdumpdev(dev, curthread);
 180         if (error == 0) {
 181                 freeenv(dumpdev);
 182                 dumpdev = NULL;
 183         }
 184
 185         (void)g_access(cp, -1, 0, 0);
 186
 187         return (error);
 188 }
 189
 190 static void
 191 g_dev_destroy(void *arg, int flags __unused)
 192 {
 193         struct g_consumer *cp;
 194         struct g_geom *gp;
 195         struct g_dev_softc *sc;
 196         char buf[SPECNAMELEN + 6];
 197
 198         g_topology_assert();
 199         cp = arg;
 200         gp = cp->geom;
 201         sc = cp->private;
 202         g_trace(G_T_TOPOLOGY, "g_dev_destroy(%p(%s))", cp, gp->name);
 203         snprintf(buf, sizeof(buf), "cdev=%s", gp->name);
 204         devctl_notify_f("GEOM", "DEV", "DESTROY", buf, M_WAITOK);
 205         if (cp->acr > 0 || cp->acw > 0 || cp->ace > 0)
 206                 g_access(cp, -cp->acr, -cp->acw, -cp->ace);
 207         g_detach(cp);
 208         g_destroy_consumer(cp);
 209         g_destroy_geom(gp);
 210         mtx_destroy(&sc->sc_mtx);
 211         g_free(sc);
 212 }
 213
 214 void
 215 g_dev_print(void)
 216 {
 217         struct g_geom *gp;
 218         char const *p = "";
 219
 220         LIST_FOREACH(gp, &g_dev_class.geom, geom) {
 221                 printf("%s%s", p, gp->name);
 222                 p = " ";
 223         }
 224         printf("\n");
 225 }
 226
 227 static void
 228 g_dev_set_physpath(struct g_consumer *cp)
 229 {
 230         struct g_dev_softc *sc;
 231         char *physpath;
 232         int error, physpath_len;
 233
 234         if (g_access(cp, 1, 0, 0) != 0)
 235                 return;
 236
 237         sc = cp->private;
 238         physpath_len = MAXPATHLEN;
 239         physpath = g_malloc(physpath_len, M_WAITOK|M_ZERO);
 240         error = g_io_getattr("GEOM::physpath", cp, &physpath_len, physpath);
 241         g_access(cp, -1, 0, 0);
 242         if (error == 0 && strlen(physpath) != 0) {
 243                 struct cdev *dev, *old_alias_dev;
 244                 struct cdev **alias_devp;
 245
 246                 dev = sc->sc_dev;
 247                 old_alias_dev = sc->sc_alias;
 248                 alias_devp = (struct cdev **)&sc->sc_alias;
 249                 make_dev_physpath_alias(MAKEDEV_WAITOK, alias_devp, dev,
 250                     old_alias_dev, physpath);
 251         } else if (sc->sc_alias) {
 252                 destroy_dev((struct cdev *)sc->sc_alias);
 253                 sc->sc_alias = NULL;
 254         }
 255         g_free(physpath);
 256 }
 257
 258 static void
 259 g_dev_set_media(struct g_consumer *cp)
 260 {
 261         struct g_dev_softc *sc;
 262         struct cdev *dev;
 263         char buf[SPECNAMELEN + 6];
 264
 265         sc = cp->private;
 266         dev = sc->sc_dev;
 267         snprintf(buf, sizeof(buf), "cdev=%s", dev->si_name);
 268         devctl_notify_f("DEVFS", "CDEV", "MEDIACHANGE", buf, M_WAITOK);
 269         devctl_notify_f("GEOM", "DEV", "MEDIACHANGE", buf, M_WAITOK);
 270         dev = sc->sc_alias;
 271         if (dev != NULL) {
 272                 snprintf(buf, sizeof(buf), "cdev=%s", dev->si_name);
 273                 devctl_notify_f("DEVFS", "CDEV", "MEDIACHANGE", buf, M_WAITOK);
 274                 devctl_notify_f("GEOM", "DEV", "MEDIACHANGE", buf, M_WAITOK);
 275         }
 276 }
 277
 278 static void
 279 g_dev_attrchanged(struct g_consumer *cp, const char *attr)
 280 {
 281
 282         if (strcmp(attr, "GEOM::media") == 0) {
 283                 g_dev_set_media(cp);
 284                 return;
 285         }
 286
 287         if (strcmp(attr, "GEOM::physpath") == 0) {
 288                 g_dev_set_physpath(cp);
 289                 return;
 290         }
 291 }
 292
 293 struct g_provider *
 294 g_dev_getprovider(struct cdev *dev)
 295 {
 296         struct g_consumer *cp;
 297
 298         g_topology_assert();
 299         if (dev == NULL)
 300                 return (NULL);
 301         if (dev->si_devsw != &g_dev_cdevsw)
 302                 return (NULL);
 303         cp = dev->si_drv2;
 304         return (cp->provider);
 305 }
 306
 307 static struct g_geom *
 308 g_dev_taste(struct g_class *mp, struct g_provider *pp, int insist __unused)
 309 {
 310         struct g_geom *gp;
 311         struct g_consumer *cp;
 312         struct g_dev_softc *sc;
 313         int error;
 314         struct cdev *dev;
 315         char buf[SPECNAMELEN + 6];
 316
 317         g_trace(G_T_TOPOLOGY, "dev_taste(%s,%s)", mp->name, pp->name);
 318         g_topology_assert();
 319         gp = g_new_geomf(mp, "%s", pp->name);
 320         sc = g_malloc(sizeof(*sc), M_WAITOK | M_ZERO);
 321         mtx_init(&sc->sc_mtx, "g_dev", NULL, MTX_DEF);
 322         cp = g_new_consumer(gp);
 323         cp->private = sc;
 324         cp->flags |= G_CF_DIRECT_SEND | G_CF_DIRECT_RECEIVE;
 325         error = g_attach(cp, pp);
 326         KASSERT(error == 0,
 327             ("g_dev_taste(%s) failed to g_attach, err=%d", pp->name, error));
 328         error = make_dev_p(MAKEDEV_CHECKNAME | MAKEDEV_WAITOK, &dev,
 329             &g_dev_cdevsw, NULL, UID_ROOT, GID_OPERATOR, 0640, "%s", gp->name);
 330         if (error != 0) {
 331                 printf("%s: make_dev_p() failed (gp->name=%s, error=%d)\n",
 332                     __func__, gp->name, error);
 333                 g_detach(cp);
 334                 g_destroy_consumer(cp);
 335                 g_destroy_geom(gp);
 336                 mtx_destroy(&sc->sc_mtx);
 337                 g_free(sc);
 338                 return (NULL);
 339         }
 340         dev->si_flags |= SI_UNMAPPED;
 341         sc->sc_dev = dev;
 342
 343         dev->si_iosize_max = MAXPHYS;
 344         dev->si_drv2 = cp;
 345         error = init_dumpdev(dev);
 346         if (error != 0)
 347                 printf("%s: init_dumpdev() failed (gp->name=%s, error=%d)\n",
 348                     __func__, gp->name, error);
 349
 350         g_dev_attrchanged(cp, "GEOM::physpath");
 351         snprintf(buf, sizeof(buf), "cdev=%s", gp->name);
 352         devctl_notify_f("GEOM", "DEV", "CREATE", buf, M_WAITOK);
 353
 354         return (gp);
 355 }
 356
 357 static int
 358 g_dev_open(struct cdev *dev, int flags, int fmt, struct thread *td)
 359 {
 360         struct g_consumer *cp;
 361         struct g_dev_softc *sc;
 362         int error, r, w, e;
 363
 364         cp = dev->si_drv2;
 365         if (cp == NULL)
 366                 return (ENXIO);         /* g_dev_taste() not done yet */
 367         g_trace(G_T_ACCESS, "g_dev_open(%s, %d, %d, %p)",
 368             cp->geom->name, flags, fmt, td);
 369
 370         r = flags & FREAD ? 1 : 0;
 371         w = flags & FWRITE ? 1 : 0;
 372 #ifdef notyet
 373         e = flags & O_EXCL ? 1 : 0;
 374 #else
 375         e = 0;
 376 #endif
 377
 378         /*
 379          * This happens on attempt to open a device node with O_EXEC.
 380          */
 381         if (r + w + e == 0)
 382                 return (EINVAL);
 383
 384         if (w) {
 385                 /*
 386                  * When running in very secure mode, do not allow
 387                  * opens for writing of any disks.
 388                  */
 389                 error = securelevel_ge(td->td_ucred, 2);
 390                 if (error)
 391                         return (error);
 392         }
 393         g_topology_lock();
 394         error = g_access(cp, r, w, e);
 395         g_topology_unlock();
 396         if (error == 0) {
 397                 sc = cp->private;
 398                 mtx_lock(&sc->sc_mtx);
 399                 if (sc->sc_open == 0 && (sc->sc_active & SC_A_ACTIVE) != 0)
 400                         wakeup(&sc->sc_active);
 401                 sc->sc_open += r + w + e;
 402                 if (sc->sc_open == 0)
 403                         atomic_clear_int(&sc->sc_active, SC_A_OPEN);
 404                 else
 405                         atomic_set_int(&sc->sc_active, SC_A_OPEN);
 406                 mtx_unlock(&sc->sc_mtx);
 407         }
 408         return (error);
 409 }
 410
 411 static int
 412 g_dev_close(struct cdev *dev, int flags, int fmt, struct thread *td)
 413 {
 414         struct g_consumer *cp;
 415         struct g_dev_softc *sc;
 416         int error, r, w, e;
 417
 418         cp = dev->si_drv2;
 419         if (cp == NULL)
 420                 return (ENXIO);
 421         g_trace(G_T_ACCESS, "g_dev_close(%s, %d, %d, %p)",
 422             cp->geom->name, flags, fmt, td);
 423
 424         r = flags & FREAD ? -1 : 0;
 425         w = flags & FWRITE ? -1 : 0;
 426 #ifdef notyet
 427         e = flags & O_EXCL ? -1 : 0;
 428 #else
 429         e = 0;
 430 #endif
 431
 432         /*
 433          * The vgonel(9) - caused by eg. forced unmount of devfs - calls
 434          * VOP_CLOSE(9) on devfs vnode without any FREAD or FWRITE flags,
 435          * which would result in zero deltas, which in turn would cause
 436          * panic in g_access(9).
 437          *
 438          * Note that we cannot zero the counters (ie. do "r = cp->acr"
 439          * etc) instead, because the consumer might be opened in another
 440          * devfs instance.
 441          */
 442         if (r + w + e == 0)
 443                 return (EINVAL);
 444
 445         sc = cp->private;
 446         mtx_lock(&sc->sc_mtx);
 447         sc->sc_open += r + w + e;
 448         if (sc->sc_open == 0)
 449                 atomic_clear_int(&sc->sc_active, SC_A_OPEN);
 450         else
 451                 atomic_set_int(&sc->sc_active, SC_A_OPEN);
 452         while (sc->sc_open == 0 && (sc->sc_active & SC_A_ACTIVE) != 0)
 453                 msleep(&sc->sc_active, &sc->sc_mtx, 0, "g_dev_close", hz / 10);
 454         mtx_unlock(&sc->sc_mtx);
 455         g_topology_lock();
 456         error = g_access(cp, r, w, e);
 457         g_topology_unlock();
 458         return (error);
 459 }
 460
 461 /*
 462  * XXX: Until we have unmessed the ioctl situation, there is a race against
 463  * XXX: a concurrent orphanization.  We cannot close it by holding topology
 464  * XXX: since that would prevent us from doing our job, and stalling events
 465  * XXX: will break (actually: stall) the BSD disklabel hacks.
 466  */
 467 static int
 468 g_dev_ioctl(struct cdev *dev, u_long cmd, caddr_t data, int fflag, struct thread *td)
 469 {
 470         struct g_consumer *cp;
 471         struct g_provider *pp;
 472         off_t offset, length, chunk, odd;
 473         int i, error;
 474
 475         cp = dev->si_drv2;
 476         pp = cp->provider;
 477
 478         error = 0;
 479         KASSERT(cp->acr || cp->acw,
 480             ("Consumer with zero access count in g_dev_ioctl"));
 481
 482         i = IOCPARM_LEN(cmd);
 483         switch (cmd) {
 484         case DIOCGSECTORSIZE:
 485                 *(u_int *)data = cp->provider->sectorsize;
 486                 if (*(u_int *)data == 0)
 487                         error = ENOENT;
 488                 break;
 489         case DIOCGMEDIASIZE:
 490                 *(off_t *)data = cp->provider->mediasize;
 491                 if (*(off_t *)data == 0)
 492                         error = ENOENT;
 493                 break;
 494         case DIOCGFWSECTORS:
 495                 error = g_io_getattr("GEOM::fwsectors", cp, &i, data);
 496                 if (error == 0 && *(u_int *)data == 0)
 497                         error = ENOENT;
 498                 break;
 499         case DIOCGFWHEADS:
 500                 error = g_io_getattr("GEOM::fwheads", cp, &i, data);
 501                 if (error == 0 && *(u_int *)data == 0)
 502                         error = ENOENT;
 503                 break;
 504         case DIOCGFRONTSTUFF:
 505                 error = g_io_getattr("GEOM::frontstuff", cp, &i, data);
 506                 break;
 507         case DIOCSKERNELDUMP:
 508                 if (*(u_int *)data == 0)
 509                         error = g_dev_setdumpdev(NULL, td);
 510                 else
 511                         error = g_dev_setdumpdev(dev, td);
 512                 break;
 513         case DIOCGFLUSH:
 514                 error = g_io_flush(cp);
 515                 break;
 516         case DIOCGDELETE:
 517                 offset = ((off_t *)data)[0];
 518                 length = ((off_t *)data)[1];
 519                 if ((offset % cp->provider->sectorsize) != 0 ||
 520                     (length % cp->provider->sectorsize) != 0 || length <= 0) {
 521                         printf("%s: offset=%jd length=%jd\n", __func__, offset,
 522                             length);
 523                         error = EINVAL;
 524                         break;
 525                 }
 526                 while (length > 0) {
 527                         chunk = length;
 528                         if (g_dev_del_max_sectors != 0 && chunk >
 529                             g_dev_del_max_sectors * cp->provider->sectorsize) {
 530                                 chunk = g_dev_del_max_sectors *
 531                                     cp->provider->sectorsize;
 532                                 if (cp->provider->stripesize > 0) {
 533                                         odd = (offset + chunk +
 534                                             cp->provider->stripeoffset) %
 535                                             cp->provider->stripesize;
 536                                         if (chunk > odd)
 537                                                 chunk -= odd;
 538                                 }
 539                         }
 540                         error = g_delete_data(cp, offset, chunk);
 541                         length -= chunk;
 542                         offset += chunk;
 543                         if (error)
 544                                 break;
 545                         /*
 546                          * Since the request size can be large, the service
 547                          * time can be is likewise.  We make this ioctl
 548                          * interruptible by checking for signals for each bio.
 549                          */
 550                         if (SIGPENDING(td))
 551                                 break;
 552                 }
 553                 break;
 554         case DIOCGIDENT:
 555                 error = g_io_getattr("GEOM::ident", cp, &i, data);
 556                 break;
 557         case DIOCGPROVIDERNAME:
 558                 if (pp == NULL)
 559                         return (ENOENT);
 560                 strlcpy(data, pp->name, i);
 561                 break;
 562         case DIOCGSTRIPESIZE:
 563                 *(off_t *)data = cp->provider->stripesize;
 564                 break;
 565         case DIOCGSTRIPEOFFSET:
 566                 *(off_t *)data = cp->provider->stripeoffset;
 567                 break;
 568         case DIOCGPHYSPATH:
 569                 error = g_io_getattr("GEOM::physpath", cp, &i, data);
 570                 if (error == 0 && *(char *)data == '\0')
 571                         error = ENOENT;
 572                 break;
 573         case DIOCGATTR: {
 574                 struct diocgattr_arg *arg = (struct diocgattr_arg *)data;
 575
 576                 if (arg->len > sizeof(arg->value)) {
 577                         error = EINVAL;
 578                         break;
 579                 }
 580                 error = g_io_getattr(arg->name, cp, &arg->len, &arg->value);
 581                 break;
 582         }
 583         case DIOCZONECMD: {
 584                 struct disk_zone_args *zone_args =(struct disk_zone_args *)data;
 585                 struct disk_zone_rep_entry *new_entries, *old_entries;
 586                 struct disk_zone_report *rep;
 587                 size_t alloc_size;
 588
 589                 old_entries = NULL;
 590                 new_entries = NULL;
 591                 rep = NULL;
 592                 alloc_size = 0;
 593
 594                 if (zone_args->zone_cmd == DISK_ZONE_REPORT_ZONES) {
 595                         rep = &zone_args->zone_params.report;
 596 #define MAXENTRIES      (MAXPHYS / sizeof(struct disk_zone_rep_entry))
 597                         if (rep->entries_allocated > MAXENTRIES)
 598                                 rep->entries_allocated = MAXENTRIES;
 599                         alloc_size = rep->entries_allocated *
 600                             sizeof(struct disk_zone_rep_entry);
 601                         if (alloc_size != 0)
 602                                 new_entries = g_malloc(alloc_size,
 603                                     M_WAITOK| M_ZERO);
 604                         old_entries = rep->entries;
 605                         rep->entries = new_entries;
 606                 }
 607                 error = g_io_zonecmd(zone_args, cp);
 608                 if (zone_args->zone_cmd == DISK_ZONE_REPORT_ZONES &&
 609                     alloc_size != 0 && error == 0)
 610                         error = copyout(new_entries, old_entries, alloc_size);
 611                 if (old_entries != NULL && rep != NULL)
 612                         rep->entries = old_entries;
 613                 if (new_entries != NULL)
 614                         g_free(new_entries);
 615                 break;
 616         }
 617         default:
 618                 if (cp->provider->geom->ioctl != NULL) {
 619                         error = cp->provider->geom->ioctl(cp->provider, cmd, data, fflag, td);
 620                 } else {
 621                         error = ENOIOCTL;
 622                 }
 623         }
 624
 625         return (error);
 626 }
 627
 628 static void
 629 g_dev_done(struct bio *bp2)
 630 {
 631         struct g_consumer *cp;
 632         struct g_dev_softc *sc;
 633         struct bio *bp;
 634         int active;
 635
 636         cp = bp2->bio_from;
 637         sc = cp->private;
 638         bp = bp2->bio_parent;
 639         bp->bio_error = bp2->bio_error;
 640         bp->bio_completed = bp2->bio_completed;
 641         bp->bio_resid = bp->bio_length - bp2->bio_completed;
 642         if (bp2->bio_cmd == BIO_ZONE)
 643                 bcopy(&bp2->bio_zone, &bp->bio_zone, sizeof(bp->bio_zone));
 644
 645         if (bp2->bio_error != 0) {
 646                 g_trace(G_T_BIO, "g_dev_done(%p) had error %d",
 647                     bp2, bp2->bio_error);
 648                 bp->bio_flags |= BIO_ERROR;
 649         } else {
 650                 g_trace(G_T_BIO, "g_dev_done(%p/%p) resid %ld completed %jd",
 651                     bp2, bp, bp2->bio_resid, (intmax_t)bp2->bio_completed);
 652         }
 653         g_destroy_bio(bp2);
 654         active = atomic_fetchadd_int(&sc->sc_active, -1) - 1;
 655         if ((active & SC_A_ACTIVE) == 0) {
 656                 if ((active & SC_A_OPEN) == 0)
 657                         wakeup(&sc->sc_active);
 658                 if (active & SC_A_DESTROY)
 659                         g_post_event(g_dev_destroy, cp, M_NOWAIT, NULL);
 660         }
 661         biodone(bp);
 662 }
 663
 664 static void
 665 g_dev_strategy(struct bio *bp)
 666 {
 667         struct g_consumer *cp;
 668         struct bio *bp2;
 669         struct cdev *dev;
 670         struct g_dev_softc *sc;
 671
 672         KASSERT(bp->bio_cmd == BIO_READ ||
 673                 bp->bio_cmd == BIO_WRITE ||
 674                 bp->bio_cmd == BIO_DELETE ||
 675                 bp->bio_cmd == BIO_FLUSH ||
 676                 bp->bio_cmd == BIO_ZONE,
 677                 ("Wrong bio_cmd bio=%p cmd=%d", bp, bp->bio_cmd));
 678         dev = bp->bio_dev;
 679         cp = dev->si_drv2;
 680         sc = cp->private;
 681         KASSERT(cp->acr || cp->acw,
 682             ("Consumer with zero access count in g_dev_strategy"));
 683 #ifdef INVARIANTS
 684         if ((bp->bio_offset % cp->provider->sectorsize) != 0 ||
 685             (bp->bio_bcount % cp->provider->sectorsize) != 0) {
 686                 bp->bio_resid = bp->bio_bcount;
 687                 biofinish(bp, NULL, EINVAL);
 688                 return;
 689         }
 690 #endif
 691         KASSERT(sc->sc_open > 0, ("Closed device in g_dev_strategy"));
 692         atomic_add_int(&sc->sc_active, 1);
 693
 694         for (;;) {
 695                 /*
 696                  * XXX: This is not an ideal solution, but I believe it to
 697                  * XXX: deadlock safely, all things considered.
 698                  */
 699                 bp2 = g_clone_bio(bp);
 700                 if (bp2 != NULL)
 701                         break;
 702                 pause("gdstrat", hz / 10);
 703         }
 704         KASSERT(bp2 != NULL, ("XXX: ENOMEM in a bad place"));
 705         bp2->bio_done = g_dev_done;
 706         g_trace(G_T_BIO,
 707             "g_dev_strategy(%p/%p) offset %jd length %jd data %p cmd %d",
 708             bp, bp2, (intmax_t)bp->bio_offset, (intmax_t)bp2->bio_length,
 709             bp2->bio_data, bp2->bio_cmd);
 710         g_io_request(bp2, cp);
 711         KASSERT(cp->acr || cp->acw,
 712             ("g_dev_strategy raced with g_dev_close and lost"));
 713
 714 }
 715
 716 /*
 717  * g_dev_callback()
 718  *
 719  * Called by devfs when asynchronous device destruction is completed.
 720  * - Mark that we have no attached device any more.
 721  * - If there are no outstanding requests, schedule geom destruction.
 722  *   Otherwise destruction will be scheduled later by g_dev_done().
 723  */
 724
 725 static void
 726 g_dev_callback(void *arg)
 727 {
 728         struct g_consumer *cp;
 729         struct g_dev_softc *sc;
 730         int active;
 731
 732         cp = arg;
 733         sc = cp->private;
 734         g_trace(G_T_TOPOLOGY, "g_dev_callback(%p(%s))", cp, cp->geom->name);
 735
 736         sc->sc_dev = NULL;
 737         sc->sc_alias = NULL;
 738         active = atomic_fetchadd_int(&sc->sc_active, SC_A_DESTROY);
 739         if ((active & SC_A_ACTIVE) == 0)
 740                 g_post_event(g_dev_destroy, cp, M_WAITOK, NULL);
 741 }
 742
 743 /*
 744  * g_dev_orphan()
 745  *
 746  * Called from below when the provider orphaned us.
 747  * - Clear any dump settings.
 748  * - Request asynchronous device destruction to prevent any more requests
 749  *   from coming in.  The provider is already marked with an error, so
 750  *   anything which comes in the interim will be returned immediately.
 751  */
 752
 753 static void
 754 g_dev_orphan(struct g_consumer *cp)
 755 {
 756         struct cdev *dev;
 757         struct g_dev_softc *sc;
 758
 759         g_topology_assert();
 760         sc = cp->private;
 761         dev = sc->sc_dev;
 762         g_trace(G_T_TOPOLOGY, "g_dev_orphan(%p(%s))", cp, cp->geom->name);
 763
 764         /* Reset any dump-area set on this device */
 765         if (dev->si_flags & SI_DUMPDEV)
 766                 (void)set_dumper(NULL, NULL, curthread);
 767
 768         /* Destroy the struct cdev *so we get no more requests */
 769         delist_dev(dev);
 770         destroy_dev_sched_cb(dev, g_dev_callback, cp);
 771 }
 772
 773 DECLARE_GEOM_CLASS(g_dev_class, g_dev);