geom_disk.c revision 125539
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 36116196Sobrien#include <sys/cdefs.h> 37116196Sobrien__FBSDID("$FreeBSD: head/sys/geom/geom_disk.c 125539 2004-02-06 23:10:49Z pjd $"); 38116196Sobrien 39104519Sphk#include "opt_geom.h" 40104519Sphk 4192108Sphk#include <sys/param.h> 4292108Sphk#include <sys/systm.h> 4392108Sphk#include <sys/kernel.h> 4492108Sphk#include <sys/sysctl.h> 4592108Sphk#include <sys/bio.h> 4692108Sphk#include <sys/conf.h> 47110119Sphk#include <sys/fcntl.h> 4892108Sphk#include <sys/malloc.h> 4992108Sphk#include <sys/sysctl.h> 50111979Sphk#include <sys/devicestat.h> 5192108Sphk#include <machine/md_var.h> 5292108Sphk 5392108Sphk#include <sys/lock.h> 5492108Sphk#include <sys/mutex.h> 5592108Sphk#include <geom/geom.h> 56112952Sphk#include <geom/geom_disk.h> 57112476Sphk#include <geom/geom_int.h> 5892108Sphk 59110720Sphkstatic struct mtx g_disk_done_mtx; 60110720Sphk 6192108Sphkstatic g_access_t g_disk_access; 62115473Sphkstatic g_init_t g_disk_init; 63115473Sphkstatic g_fini_t g_disk_fini; 6492108Sphk 6593248Sphkstruct g_class g_disk_class = { 66112552Sphk .name = "DISK", 67115473Sphk .init = g_disk_init, 68115473Sphk .fini = g_disk_fini, 6992108Sphk}; 7092108Sphk 71110720Sphkstatic void 72115473Sphkg_disk_init(struct g_class *mp __unused) 73110720Sphk{ 74115473Sphk 75123271Struckman mtx_init(&g_disk_done_mtx, "g_disk_done", NULL, MTX_DEF); 76110720Sphk} 77104936Sphk 78115473Sphkstatic void 79115473Sphkg_disk_fini(struct g_class *mp __unused) 80115473Sphk{ 81110720Sphk 82115473Sphk mtx_destroy(&g_disk_done_mtx); 83115473Sphk} 84115473Sphk 85115473SphkDECLARE_GEOM_CLASS(g_disk_class, g_disk); 86115473Sphk 87110052Sphkstatic void __inline 88110052Sphkg_disk_lock_giant(struct disk *dp) 89110052Sphk{ 90110119Sphk if (dp->d_flags & DISKFLAG_NOGIANT) 91110119Sphk return; 92110119Sphk mtx_lock(&Giant); 93110052Sphk} 94110052Sphk 95110052Sphkstatic void __inline 96110052Sphkg_disk_unlock_giant(struct disk *dp) 97110052Sphk{ 98110119Sphk if (dp->d_flags & DISKFLAG_NOGIANT) 99110119Sphk return; 100110119Sphk mtx_unlock(&Giant); 101110052Sphk} 102110052Sphk 10392108Sphkstatic int 10492108Sphkg_disk_access(struct g_provider *pp, int r, int w, int e) 10592108Sphk{ 10692108Sphk struct disk *dp; 10792108Sphk int error; 10892108Sphk 10992108Sphk g_trace(G_T_ACCESS, "g_disk_access(%s, %d, %d, %d)", 11092108Sphk pp->name, r, w, e); 11192108Sphk g_topology_assert(); 112125539Spjd dp = pp->geom->softc; 113125539Spjd if (dp == NULL) { 114125539Spjd /* 115125539Spjd * Allow decreasing access count even if disk is not 116125539Spjd * avaliable anymore. 117125539Spjd */ 118125539Spjd if (r <= 0 && w <= 0 && e <= 0) 119125539Spjd return (0); 120125539Spjd else 121125539Spjd return (ENXIO); 122125539Spjd } 12392108Sphk r += pp->acr; 12492108Sphk w += pp->acw; 12592108Sphk e += pp->ace; 126110119Sphk error = 0; 12792108Sphk if ((pp->acr + pp->acw + pp->ace) == 0 && (r + w + e) > 0) { 128111668Sphk if (dp->d_open != NULL) { 129110119Sphk g_disk_lock_giant(dp); 130111668Sphk error = dp->d_open(dp); 131110119Sphk if (error != 0) 132110119Sphk printf("Opened disk %s -> %d\n", 133110119Sphk pp->name, error); 134110119Sphk g_disk_unlock_giant(dp); 135110119Sphk } 136105551Sphk pp->mediasize = dp->d_mediasize; 137105551Sphk pp->sectorsize = dp->d_sectorsize; 138110119Sphk dp->d_flags |= DISKFLAG_OPEN; 139110727Sphk if (dp->d_maxsize == 0) { 140110727Sphk printf("WARNING: Disk drive %s%d has no d_maxsize\n", 141110727Sphk dp->d_name, dp->d_unit); 142110727Sphk dp->d_maxsize = DFLTPHYS; 143110727Sphk } 14492108Sphk } else if ((pp->acr + pp->acw + pp->ace) > 0 && (r + w + e) == 0) { 145111668Sphk if (dp->d_close != NULL) { 146110119Sphk g_disk_lock_giant(dp); 147111668Sphk error = dp->d_close(dp); 148110119Sphk if (error != 0) 149110119Sphk printf("Closed disk %s -> %d\n", 150110119Sphk pp->name, error); 151110119Sphk g_disk_unlock_giant(dp); 152110119Sphk } 153110119Sphk dp->d_flags &= ~DISKFLAG_OPEN; 15492108Sphk } 15592108Sphk return (error); 15692108Sphk} 15792108Sphk 15892108Sphkstatic void 15995038Sphkg_disk_kerneldump(struct bio *bp, struct disk *dp) 16095038Sphk{ 16195038Sphk int error; 16295038Sphk struct g_kerneldump *gkd; 16395038Sphk struct dumperinfo di; 164104450Sphk struct g_geom *gp; 16595038Sphk 16695038Sphk gkd = (struct g_kerneldump*)bp->bio_data; 167104450Sphk gp = bp->bio_to->geom; 168104450Sphk g_trace(G_T_TOPOLOGY, "g_disk_kernedump(%s, %jd, %jd)", 169104450Sphk gp->name, (intmax_t)gkd->offset, (intmax_t)gkd->length); 170120572Sphk if (dp->d_dump == NULL) { 171120572Sphk g_io_deliver(bp, ENODEV); 172120572Sphk return; 173120572Sphk } 174110119Sphk di.dumper = dp->d_dump; 175111220Sphk di.priv = dp; 176103714Sphk di.blocksize = dp->d_sectorsize; 17795038Sphk di.mediaoffset = gkd->offset; 17895038Sphk di.mediasize = gkd->length; 17995038Sphk error = set_dumper(&di); 180104195Sphk g_io_deliver(bp, error); 18195038Sphk} 18295038Sphk 18395038Sphkstatic void 18492108Sphkg_disk_done(struct bio *bp) 18592108Sphk{ 186111979Sphk struct bio *bp2; 187111979Sphk struct disk *dp; 18892108Sphk 189110720Sphk /* See "notes" for why we need a mutex here */ 190110720Sphk /* XXX: will witness accept a mix of Giant/unGiant drivers here ? */ 191110720Sphk mtx_lock(&g_disk_done_mtx); 19292108Sphk bp->bio_completed = bp->bio_length - bp->bio_resid; 193111979Sphk 194111979Sphk bp2 = bp->bio_parent; 195111979Sphk dp = bp2->bio_to->geom->softc; 196111979Sphk if (bp2->bio_error == 0) 197111979Sphk bp2->bio_error = bp->bio_error; 198111979Sphk bp2->bio_completed += bp->bio_completed; 199111979Sphk g_destroy_bio(bp); 200111979Sphk bp2->bio_inbed++; 201111979Sphk if (bp2->bio_children == bp2->bio_inbed) { 202112259Sphk bp2->bio_resid = bp2->bio_bcount - bp2->bio_completed; 203112259Sphk devstat_end_transaction_bio(dp->d_devstat, bp2); 204111979Sphk g_io_deliver(bp2, bp2->bio_error); 205111979Sphk } 206110720Sphk mtx_unlock(&g_disk_done_mtx); 20792108Sphk} 20892108Sphk 209119660Sphkstatic int 210119660Sphkg_disk_ioctl(struct g_provider *pp, u_long cmd, void * data, struct thread *td) 211119660Sphk{ 212119660Sphk struct g_geom *gp; 213119660Sphk struct disk *dp; 214119660Sphk int error; 215119660Sphk 216119660Sphk gp = pp->geom; 217119660Sphk dp = gp->softc; 218119660Sphk 219119660Sphk if (dp->d_ioctl == NULL) 220119660Sphk return (ENOIOCTL); 221119660Sphk g_disk_lock_giant(dp); 222119660Sphk error = dp->d_ioctl(dp, cmd, data, 0, td); 223119660Sphk g_disk_unlock_giant(dp); 224119660Sphk return(error); 225119660Sphk} 226119660Sphk 22792108Sphkstatic void 22892108Sphkg_disk_start(struct bio *bp) 22992108Sphk{ 230110720Sphk struct bio *bp2, *bp3; 23192108Sphk struct disk *dp; 23292403Sphk int error; 233110720Sphk off_t off; 23492108Sphk 23592108Sphk dp = bp->bio_to->geom->softc; 236115214Sphk if (dp == NULL) 237115214Sphk g_io_deliver(bp, ENXIO); 238104609Sphk error = EJUSTRETURN; 23992108Sphk switch(bp->bio_cmd) { 240104609Sphk case BIO_DELETE: 241110119Sphk if (!(dp->d_flags & DISKFLAG_CANDELETE)) { 242104609Sphk error = 0; 243104609Sphk break; 244104609Sphk } 245104609Sphk /* fall-through */ 24692108Sphk case BIO_READ: 24792108Sphk case BIO_WRITE: 248110720Sphk off = 0; 249110720Sphk bp3 = NULL; 25092108Sphk bp2 = g_clone_bio(bp); 251110477Sphk if (bp2 == NULL) { 252110477Sphk error = ENOMEM; 253110477Sphk break; 254110477Sphk } 255112259Sphk devstat_start_transaction_bio(dp->d_devstat, bp); 256110720Sphk do { 257110720Sphk bp2->bio_offset += off; 258110720Sphk bp2->bio_length -= off; 259110766Stegge bp2->bio_data += off; 260110720Sphk if (bp2->bio_length > dp->d_maxsize) { 261110720Sphk /* 262110720Sphk * XXX: If we have a stripesize we should really 263110720Sphk * use it here. 264110720Sphk */ 265110720Sphk bp2->bio_length = dp->d_maxsize; 266110720Sphk off += dp->d_maxsize; 267110720Sphk /* 268110720Sphk * To avoid a race, we need to grab the next bio 269110720Sphk * before we schedule this one. See "notes". 270110720Sphk */ 271110720Sphk bp3 = g_clone_bio(bp); 272110720Sphk if (bp3 == NULL) 273110720Sphk bp->bio_error = ENOMEM; 274110720Sphk } 275110720Sphk bp2->bio_done = g_disk_done; 276110720Sphk bp2->bio_pblkno = bp2->bio_offset / dp->d_sectorsize; 277110720Sphk bp2->bio_bcount = bp2->bio_length; 278110720Sphk bp2->bio_disk = dp; 279110720Sphk g_disk_lock_giant(dp); 280110720Sphk dp->d_strategy(bp2); 281110720Sphk g_disk_unlock_giant(dp); 282110720Sphk bp2 = bp3; 283110720Sphk bp3 = NULL; 284110720Sphk } while (bp2 != NULL); 28592108Sphk break; 28692108Sphk case BIO_GETATTR: 287105551Sphk if (g_handleattr_int(bp, "GEOM::fwsectors", dp->d_fwsectors)) 28892403Sphk break; 289103714Sphk else if (g_handleattr_int(bp, "GEOM::fwheads", dp->d_fwheads)) 29092403Sphk break; 29198066Sphk else if (g_handleattr_off_t(bp, "GEOM::frontstuff", 0)) 29294287Sphk break; 29395038Sphk else if (!strcmp(bp->bio_attribute, "GEOM::kerneldump")) 29495038Sphk g_disk_kerneldump(bp, dp); 295119660Sphk else 29692403Sphk error = ENOIOCTL; 29792403Sphk break; 29892108Sphk default: 29992403Sphk error = EOPNOTSUPP; 30092403Sphk break; 30192403Sphk } 302104609Sphk if (error != EJUSTRETURN) 303104195Sphk g_io_deliver(bp, error); 30492403Sphk return; 30592108Sphk} 30692108Sphk 307104936Sphkstatic void 308107953Sphkg_disk_dumpconf(struct sbuf *sb, const char *indent, struct g_geom *gp, struct g_consumer *cp, struct g_provider *pp) 309105537Sphk{ 310105537Sphk struct disk *dp; 311105537Sphk 312105537Sphk dp = gp->softc; 313120374Sphk if (dp == NULL) 314120374Sphk return; 315106101Sphk if (indent == NULL) { 316106101Sphk sbuf_printf(sb, " hd %u", dp->d_fwheads); 317106101Sphk sbuf_printf(sb, " sc %u", dp->d_fwsectors); 318106101Sphk return; 319106101Sphk } 320105539Sphk if (pp != NULL) { 321105537Sphk sbuf_printf(sb, "%s<fwheads>%u</fwheads>\n", 322105537Sphk indent, dp->d_fwheads); 323105537Sphk sbuf_printf(sb, "%s<fwsectors>%u</fwsectors>\n", 324105537Sphk indent, dp->d_fwsectors); 325105537Sphk } 326105537Sphk} 327105537Sphk 328105537Sphkstatic void 329115309Sphkg_disk_create(void *arg, int flag) 33092108Sphk{ 33192108Sphk struct g_geom *gp; 33292108Sphk struct g_provider *pp; 333110708Sphk struct disk *dp; 33492108Sphk 335115309Sphk if (flag == EV_CANCEL) 336115309Sphk return; 337104936Sphk g_topology_assert(); 338110708Sphk dp = arg; 339110708Sphk gp = g_new_geomf(&g_disk_class, "%s%d", dp->d_name, dp->d_unit); 340104936Sphk gp->start = g_disk_start; 341104936Sphk gp->access = g_disk_access; 342119660Sphk gp->ioctl = g_disk_ioctl; 343110708Sphk gp->softc = dp; 344105537Sphk gp->dumpconf = g_disk_dumpconf; 345104936Sphk pp = g_new_providerf(gp, "%s", gp->name); 346110708Sphk pp->mediasize = dp->d_mediasize; 347110708Sphk pp->sectorsize = dp->d_sectorsize; 348110708Sphk if (dp->d_flags & DISKFLAG_CANDELETE) 349110708Sphk pp->flags |= G_PF_CANDELETE; 350110710Sphk pp->stripeoffset = dp->d_stripeoffset; 351110710Sphk pp->stripesize = dp->d_stripesize; 352105957Sphk if (bootverbose) 353105957Sphk printf("GEOM: new disk %s\n", gp->name); 354112989Sphk dp->d_geom = gp; 355112989Sphk g_error_provider(pp, 0); 356104936Sphk} 357104936Sphk 358119652Sphkstatic void 359119652Sphkg_disk_destroy(void *ptr, int flag) 360119652Sphk{ 361119652Sphk struct g_geom *gp; 362104936Sphk 363119652Sphk g_topology_assert(); 364119652Sphk gp = ptr; 365119652Sphk gp->softc = NULL; 366119652Sphk g_wither_geom(gp, ENXIO); 367119652Sphk} 368104936Sphk 369111668Sphkvoid 370111668Sphkdisk_create(int unit, struct disk *dp, int flags, void *unused __unused, void * unused2 __unused) 371104936Sphk{ 372104936Sphk 373110708Sphk dp->d_unit = unit; 374110119Sphk dp->d_flags = flags; 375110119Sphk KASSERT(dp->d_strategy != NULL, ("disk_create need d_strategy")); 376110119Sphk KASSERT(dp->d_name != NULL, ("disk_create need d_name")); 377110119Sphk KASSERT(*dp->d_name != 0, ("disk_create need d_name")); 378110317Sphk KASSERT(strlen(dp->d_name) < SPECNAMELEN - 4, ("disk name too long")); 379120374Sphk if (dp->d_devstat == NULL) 380120374Sphk dp->d_devstat = devstat_new_entry(dp->d_name, dp->d_unit, 381120374Sphk dp->d_sectorsize, DEVSTAT_ALL_SUPPORTED, 382120374Sphk DEVSTAT_TYPE_DIRECT, DEVSTAT_PRIORITY_MAX); 383112989Sphk dp->d_geom = NULL; 384113937Sphk g_post_event(g_disk_create, dp, M_WAITOK, dp, NULL); 38592108Sphk} 38692108Sphk 38792108Sphkvoid 388111216Sphkdisk_destroy(struct disk *dp) 38992108Sphk{ 39092108Sphk struct g_geom *gp; 39192108Sphk 392112989Sphk g_cancel_event(dp); 393110419Sphk gp = dp->d_geom; 394112989Sphk if (gp == NULL) 395112989Sphk return; 396110119Sphk gp->softc = NULL; 397111979Sphk devstat_remove_entry(dp->d_devstat); 398119652Sphk g_waitfor_event(g_disk_destroy, gp, M_WAITOK, NULL, NULL); 399119652Sphk dp->d_geom = NULL; 40092108Sphk} 40192108Sphk 402104451Sphkstatic void 403112988Sphkg_kern_disks(void *p, int flag __unused) 404104451Sphk{ 405104451Sphk struct sbuf *sb; 406104451Sphk struct g_geom *gp; 407104451Sphk char *sp; 408104451Sphk 409104451Sphk sb = p; 410104451Sphk sp = ""; 411104451Sphk g_topology_assert(); 412104451Sphk LIST_FOREACH(gp, &g_disk_class.geom, geom) { 413104451Sphk sbuf_printf(sb, "%s%s", sp, gp->name); 414104451Sphk sp = " "; 415104451Sphk } 416104451Sphk sbuf_finish(sb); 417104451Sphk} 418104451Sphk 419104451Sphkstatic int 420104451Sphksysctl_disks(SYSCTL_HANDLER_ARGS) 421104451Sphk{ 422104451Sphk int error; 423104451Sphk struct sbuf *sb; 424104451Sphk 425104451Sphk sb = sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND); 426104451Sphk sbuf_clear(sb); 427113940Sphk g_waitfor_event(g_kern_disks, sb, M_WAITOK, NULL); 428113937Sphk error = SYSCTL_OUT(req, sbuf_data(sb), sbuf_len(sb) + 1); 429104451Sphk sbuf_delete(sb); 430104451Sphk return error; 431104451Sphk} 432104451Sphk 433104451SphkSYSCTL_PROC(_kern, OID_AUTO, disks, CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_NOLOCK, 0, 0, 434104451Sphk sysctl_disks, "A", "names of available disks"); 435104519Sphk 436