geom_disk.c revision 110477
192108Sphk/*- 292108Sphk * Copyright (c) 2002 Poul-Henning Kamp 392108Sphk * Copyright (c) 2002 Networks Associates Technology, Inc. 492108Sphk * All rights reserved. 592108Sphk * 692108Sphk * This software was developed for the FreeBSD Project by Poul-Henning Kamp 792108Sphk * and NAI Labs, the Security Research Division of Network Associates, Inc. 892108Sphk * under DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"), as part of the 992108Sphk * DARPA CHATS research program. 1092108Sphk * 1192108Sphk * Redistribution and use in source and binary forms, with or without 1292108Sphk * modification, are permitted provided that the following conditions 1392108Sphk * are met: 1492108Sphk * 1. Redistributions of source code must retain the above copyright 1592108Sphk * notice, this list of conditions and the following disclaimer. 1692108Sphk * 2. Redistributions in binary form must reproduce the above copyright 1792108Sphk * notice, this list of conditions and the following disclaimer in the 1892108Sphk * documentation and/or other materials provided with the distribution. 1992108Sphk * 3. The names of the authors may not be used to endorse or promote 2092108Sphk * products derived from this software without specific prior written 2192108Sphk * permission. 2292108Sphk * 2392108Sphk * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 2492108Sphk * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2592108Sphk * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2692108Sphk * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 2792108Sphk * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2892108Sphk * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2992108Sphk * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 3092108Sphk * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 3192108Sphk * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3292108Sphk * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3392108Sphk * SUCH DAMAGE. 3492108Sphk * 3592108Sphk * $FreeBSD: head/sys/geom/geom_disk.c 110477 2003-02-06 22:00:47Z phk $ 3692108Sphk */ 3792108Sphk 38104519Sphk#include "opt_geom.h" 39104519Sphk 4092108Sphk#include <sys/param.h> 4192108Sphk#include <sys/systm.h> 4292108Sphk#include <sys/kernel.h> 4392108Sphk#include <sys/sysctl.h> 4492108Sphk#include <sys/bio.h> 4592108Sphk#include <sys/conf.h> 4692108Sphk#include <sys/disk.h> 47110119Sphk#include <sys/fcntl.h> 4892108Sphk#include <sys/malloc.h> 4992108Sphk#include <sys/sysctl.h> 50103284Sphk#include <sys/stdint.h> 5192108Sphk#include <machine/md_var.h> 5292108Sphk 5392108Sphk 5492108Sphk#include <sys/lock.h> 5592108Sphk#include <sys/mutex.h> 5692108Sphk#include <geom/geom.h> 5792108Sphk 5892108Sphkstatic g_access_t g_disk_access; 5992108Sphk 6093248Sphkstruct g_class g_disk_class = { 6197075Sphk "DISK", 6292108Sphk NULL, 6392108Sphk NULL, 6498066Sphk G_CLASS_INITIALIZER 6592108Sphk}; 6692108Sphk 67104936SphkDECLARE_GEOM_CLASS(g_disk_class, g_disk); 68104936Sphk 69110052Sphkstatic void __inline 70110052Sphkg_disk_lock_giant(struct disk *dp) 71110052Sphk{ 72110119Sphk if (dp->d_flags & DISKFLAG_NOGIANT) 73110119Sphk return; 74110119Sphk mtx_lock(&Giant); 75110052Sphk} 76110052Sphk 77110052Sphkstatic void __inline 78110052Sphkg_disk_unlock_giant(struct disk *dp) 79110052Sphk{ 80110119Sphk if (dp->d_flags & DISKFLAG_NOGIANT) 81110119Sphk return; 82110119Sphk mtx_unlock(&Giant); 83110052Sphk} 84110052Sphk 8592108Sphkstatic int 8692108Sphkg_disk_access(struct g_provider *pp, int r, int w, int e) 8792108Sphk{ 8892108Sphk struct disk *dp; 8992108Sphk dev_t dev; 9092108Sphk int error; 9192108Sphk 9292108Sphk g_trace(G_T_ACCESS, "g_disk_access(%s, %d, %d, %d)", 9392108Sphk pp->name, r, w, e); 9492108Sphk g_topology_assert(); 9592108Sphk r += pp->acr; 9692108Sphk w += pp->acw; 9792108Sphk e += pp->ace; 9892108Sphk dp = pp->geom->softc; 9992108Sphk dev = dp->d_dev; 100110119Sphk error = 0; 10192108Sphk if ((pp->acr + pp->acw + pp->ace) == 0 && (r + w + e) > 0) { 102110419Sphk if (dp->d_open != NULL || dp->d_copen != NULL) { 103110119Sphk g_disk_lock_giant(dp); 104110419Sphk if (dp->d_open != NULL) 105110419Sphk error = dp->d_open(dp); 106110419Sphk else 107110419Sphk error = dp->d_copen(dev, FREAD|FWRITE, 0, NULL); 108110119Sphk if (error != 0) 109110119Sphk printf("Opened disk %s -> %d\n", 110110119Sphk pp->name, error); 111110119Sphk g_disk_unlock_giant(dp); 112110119Sphk } 113105551Sphk pp->mediasize = dp->d_mediasize; 114105551Sphk pp->sectorsize = dp->d_sectorsize; 115110119Sphk dp->d_flags |= DISKFLAG_OPEN; 11692108Sphk } else if ((pp->acr + pp->acw + pp->ace) > 0 && (r + w + e) == 0) { 117110419Sphk if (dp->d_close != NULL || dp->d_cclose != NULL) { 118110119Sphk g_disk_lock_giant(dp); 119110419Sphk if (dp->d_close != NULL) 120110419Sphk error = dp->d_close(dp); 121110419Sphk else 122110419Sphk error = dp->d_cclose(dev, FREAD|FWRITE, 0, NULL); 123110119Sphk if (error != 0) 124110119Sphk printf("Closed disk %s -> %d\n", 125110119Sphk pp->name, error); 126110119Sphk g_disk_unlock_giant(dp); 127110119Sphk } 128110119Sphk dp->d_flags &= ~DISKFLAG_OPEN; 12992108Sphk } 13092108Sphk return (error); 13192108Sphk} 13292108Sphk 13392108Sphkstatic void 13495038Sphkg_disk_kerneldump(struct bio *bp, struct disk *dp) 13595038Sphk{ 13695038Sphk int error; 13795038Sphk struct g_kerneldump *gkd; 13895038Sphk struct dumperinfo di; 139104450Sphk struct g_geom *gp; 14095038Sphk 14195038Sphk gkd = (struct g_kerneldump*)bp->bio_data; 142104450Sphk gp = bp->bio_to->geom; 143104450Sphk g_trace(G_T_TOPOLOGY, "g_disk_kernedump(%s, %jd, %jd)", 144104450Sphk gp->name, (intmax_t)gkd->offset, (intmax_t)gkd->length); 145110119Sphk di.dumper = dp->d_dump; 14695038Sphk di.priv = dp->d_dev; 147103714Sphk di.blocksize = dp->d_sectorsize; 14895038Sphk di.mediaoffset = gkd->offset; 14995038Sphk di.mediasize = gkd->length; 15095038Sphk error = set_dumper(&di); 151104195Sphk g_io_deliver(bp, error); 15295038Sphk} 15395038Sphk 15495038Sphkstatic void 15592108Sphkg_disk_done(struct bio *bp) 15692108Sphk{ 157110475Sphk#ifdef maybe_not 158110052Sphk struct disk *dp; 15992108Sphk 160110230Sphk dp = bp->bio_disk; 16192108Sphk bp->bio_completed = bp->bio_length - bp->bio_resid; 162110119Sphk if (!(dp->d_flags & DISKFLAG_NOGIANT)) { 163110052Sphk DROP_GIANT(); 164110052Sphk g_std_done(bp); 165110052Sphk PICKUP_GIANT(); 166110052Sphk } else { 167110052Sphk g_std_done(bp); 168110052Sphk } 169110475Sphk#else 170110475Sphk bp->bio_completed = bp->bio_length - bp->bio_resid; 171110475Sphk g_std_done(bp); 172110475Sphk#endif 17392108Sphk} 17492108Sphk 17592108Sphkstatic void 17692108Sphkg_disk_start(struct bio *bp) 17792108Sphk{ 17892108Sphk struct bio *bp2; 17992108Sphk dev_t dev; 18092108Sphk struct disk *dp; 18192403Sphk struct g_ioctl *gio; 18292403Sphk int error; 18392108Sphk 18492108Sphk dp = bp->bio_to->geom->softc; 18592108Sphk dev = dp->d_dev; 186104609Sphk error = EJUSTRETURN; 18792108Sphk switch(bp->bio_cmd) { 188104609Sphk case BIO_DELETE: 189110119Sphk if (!(dp->d_flags & DISKFLAG_CANDELETE)) { 190104609Sphk error = 0; 191104609Sphk break; 192104609Sphk } 193104609Sphk /* fall-through */ 19492108Sphk case BIO_READ: 19592108Sphk case BIO_WRITE: 19692108Sphk bp2 = g_clone_bio(bp); 197110477Sphk if (bp2 == NULL) { 198110477Sphk error = ENOMEM; 199110477Sphk break; 200110477Sphk } 20192108Sphk bp2->bio_done = g_disk_done; 20292108Sphk bp2->bio_blkno = bp2->bio_offset >> DEV_BSHIFT; 203104609Sphk bp2->bio_pblkno = bp2->bio_offset / dp->d_sectorsize; 20492108Sphk bp2->bio_bcount = bp2->bio_length; 20592108Sphk bp2->bio_dev = dev; 206110230Sphk bp2->bio_disk = dp; 207110052Sphk g_disk_lock_giant(dp); 208110119Sphk dp->d_strategy(bp2); 209110052Sphk g_disk_unlock_giant(dp); 21092108Sphk break; 21192108Sphk case BIO_GETATTR: 212105551Sphk if (g_handleattr_int(bp, "GEOM::fwsectors", dp->d_fwsectors)) 21392403Sphk break; 214103714Sphk else if (g_handleattr_int(bp, "GEOM::fwheads", dp->d_fwheads)) 21592403Sphk break; 21698066Sphk else if (g_handleattr_off_t(bp, "GEOM::frontstuff", 0)) 21794287Sphk break; 21895038Sphk else if (!strcmp(bp->bio_attribute, "GEOM::kerneldump")) 21995038Sphk g_disk_kerneldump(bp, dp); 220110419Sphk else if ((dp->d_ioctl != NULL || dp->d_cioctl != NULL) && 221110119Sphk !strcmp(bp->bio_attribute, "GEOM::ioctl") && 22292403Sphk bp->bio_length == sizeof *gio) { 223104602Sphk gio = (struct g_ioctl *)bp->bio_data; 224110419Sphk if (dp->d_ioctl != NULL) { 225110419Sphk gio->func = (d_ioctl_t *)(dp->d_ioctl); 226110419Sphk gio->dev = dp; 227110419Sphk } else { 228110419Sphk gio->func = dp->d_cioctl; 229110419Sphk gio->dev = dp->d_dev; 230110419Sphk } 231104602Sphk error = EDIRIOCTL; 23292403Sphk } else 23392403Sphk error = ENOIOCTL; 23492403Sphk break; 235104542Sphk case BIO_SETATTR: 236104542Sphk if (!strcmp(bp->bio_attribute, "GEOM::ioctl") && 237104542Sphk bp->bio_length == sizeof *gio) { 238104602Sphk gio = (struct g_ioctl *)bp->bio_data; 239104602Sphk gio->func = devsw(dp->d_dev)->d_ioctl; 240104602Sphk gio->dev = dp->d_dev; 241104602Sphk error = EDIRIOCTL; 242104542Sphk } else 243104542Sphk error = ENOIOCTL; 244104542Sphk break; 24592108Sphk default: 24692403Sphk error = EOPNOTSUPP; 24792403Sphk break; 24892403Sphk } 249104609Sphk if (error != EJUSTRETURN) 250104195Sphk g_io_deliver(bp, error); 25192403Sphk return; 25292108Sphk} 25392108Sphk 254104936Sphkstatic void 255107953Sphkg_disk_dumpconf(struct sbuf *sb, const char *indent, struct g_geom *gp, struct g_consumer *cp, struct g_provider *pp) 256105537Sphk{ 257105537Sphk struct disk *dp; 258105537Sphk 259105537Sphk dp = gp->softc; 260106101Sphk if (indent == NULL) { 261106101Sphk sbuf_printf(sb, " hd %u", dp->d_fwheads); 262106101Sphk sbuf_printf(sb, " sc %u", dp->d_fwsectors); 263106101Sphk return; 264106101Sphk } 265105539Sphk if (pp != NULL) { 266105537Sphk sbuf_printf(sb, "%s<fwheads>%u</fwheads>\n", 267105537Sphk indent, dp->d_fwheads); 268105537Sphk sbuf_printf(sb, "%s<fwsectors>%u</fwsectors>\n", 269105537Sphk indent, dp->d_fwsectors); 270105537Sphk } 271105537Sphk} 272105537Sphk 273105537Sphkstatic void 274104936Sphkg_disk_create(void *arg) 27592108Sphk{ 27692108Sphk struct g_geom *gp; 27792108Sphk struct g_provider *pp; 27892108Sphk dev_t dev; 27992108Sphk 280104936Sphk g_topology_assert(); 281104936Sphk dev = arg; 282104936Sphk gp = g_new_geomf(&g_disk_class, dev->si_name); 283104936Sphk gp->start = g_disk_start; 284104936Sphk gp->access = g_disk_access; 285104936Sphk gp->softc = dev->si_disk; 286105537Sphk gp->dumpconf = g_disk_dumpconf; 287110419Sphk dev->si_disk->d_geom = gp; 288104936Sphk pp = g_new_providerf(gp, "%s", gp->name); 289105551Sphk pp->mediasize = dev->si_disk->d_mediasize; 290105551Sphk pp->sectorsize = dev->si_disk->d_sectorsize; 291104936Sphk g_error_provider(pp, 0); 292105957Sphk if (bootverbose) 293105957Sphk printf("GEOM: new disk %s\n", gp->name); 294104936Sphk} 295104936Sphk 296104936Sphk 297104936Sphk 298104936Sphkdev_t 299110119Sphkdisk_create(int unit, struct disk *dp, int flags, struct cdevsw *cdevsw, void * unused __unused) 300104936Sphk{ 301104936Sphk dev_t dev; 302104936Sphk 303109623Salfred dev = g_malloc(sizeof *dev, M_ZERO); 30492108Sphk dp->d_dev = dev; 305110119Sphk dp->d_flags = flags; 30693642Sphk dp->d_devsw = cdevsw; 30792108Sphk dev->si_devsw = cdevsw; 308110119Sphk if (cdevsw != NULL) { 309110419Sphk dp->d_copen = cdevsw->d_open; 310110419Sphk dp->d_cclose = cdevsw->d_close; 311110419Sphk dp->d_cioctl = cdevsw->d_ioctl; 312110119Sphk dp->d_strategy = cdevsw->d_strategy; 313110119Sphk dp->d_dump = (dumper_t *)cdevsw->d_dump; 314110119Sphk dp->d_name = cdevsw->d_name; 315110119Sphk } 316110119Sphk KASSERT(dp->d_strategy != NULL, ("disk_create need d_strategy")); 317110119Sphk KASSERT(dp->d_name != NULL, ("disk_create need d_name")); 318110119Sphk KASSERT(*dp->d_name != 0, ("disk_create need d_name")); 319110317Sphk KASSERT(strlen(dp->d_name) < SPECNAMELEN - 4, ("disk name too long")); 32092108Sphk dev->si_disk = dp; 321110118Sphk dev->si_udev = 0x10002; /* XXX: Needed ? */ 322110317Sphk dev->si_name = dev->__si_namebuf; 323110119Sphk sprintf(dev->si_name, "%s%d", dp->d_name, unit); 324104936Sphk g_call_me(g_disk_create, dev); 32592108Sphk return (dev); 32692108Sphk} 32792108Sphk 32892108Sphkvoid 32992108Sphkdisk_destroy(dev_t dev) 33092108Sphk{ 33192108Sphk struct disk *dp; 33292108Sphk struct g_geom *gp; 33392108Sphk 33492108Sphk dp = dev->si_disk; 335110419Sphk gp = dp->d_geom; 33692108Sphk g_free(dev); 33792108Sphk gp->flags |= G_GEOM_WITHER; 338110119Sphk gp->softc = NULL; 33992108Sphk g_orphan_provider(LIST_FIRST(&gp->provider), ENXIO); 34092108Sphk} 34192108Sphk 342104451Sphkstatic void 343104451Sphkg_kern_disks(void *p) 344104451Sphk{ 345104451Sphk struct sbuf *sb; 346104451Sphk struct g_geom *gp; 347104451Sphk char *sp; 348104451Sphk 349104451Sphk sb = p; 350104451Sphk sp = ""; 351104451Sphk g_topology_assert(); 352104451Sphk LIST_FOREACH(gp, &g_disk_class.geom, geom) { 353104451Sphk sbuf_printf(sb, "%s%s", sp, gp->name); 354104451Sphk sp = " "; 355104451Sphk } 356104451Sphk sbuf_finish(sb); 357104451Sphk wakeup(sb); 358104451Sphk} 359104451Sphk 360104451Sphkstatic int 361104451Sphksysctl_disks(SYSCTL_HANDLER_ARGS) 362104451Sphk{ 363104451Sphk int error; 364104451Sphk struct sbuf *sb; 365104451Sphk 366104451Sphk sb = sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND); 367104451Sphk sbuf_clear(sb); 368104451Sphk g_call_me(g_kern_disks, sb); 369104451Sphk do { 370104451Sphk tsleep(sb, PZERO, "kern.disks", hz); 371104451Sphk } while(!sbuf_done(sb)); 372105350Sphk error = SYSCTL_OUT(req, sbuf_data(sb), sbuf_len(sb) + 1); 373104451Sphk sbuf_delete(sb); 374104451Sphk return error; 375104451Sphk} 376104451Sphk 377104451SphkSYSCTL_PROC(_kern, OID_AUTO, disks, CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_NOLOCK, 0, 0, 378104451Sphk sysctl_disks, "A", "names of available disks"); 379104519Sphk 380