dksubr.c revision 1.93 1 /* $NetBSD: dksubr.c,v 1.93 2016/12/08 12:22:56 mlelstv Exp $ */
2
3 /*-
4 * Copyright (c) 1996, 1997, 1998, 1999, 2002, 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Jason R. Thorpe and Roland C. Dowdeswell.
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 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #include <sys/cdefs.h>
33 __KERNEL_RCSID(0, "$NetBSD: dksubr.c,v 1.93 2016/12/08 12:22:56 mlelstv Exp $");
34
35 #include <sys/param.h>
36 #include <sys/systm.h>
37 #include <sys/stat.h>
38 #include <sys/proc.h>
39 #include <sys/ioctl.h>
40 #include <sys/device.h>
41 #include <sys/disk.h>
42 #include <sys/disklabel.h>
43 #include <sys/buf.h>
44 #include <sys/bufq.h>
45 #include <sys/vnode.h>
46 #include <sys/fcntl.h>
47 #include <sys/namei.h>
48 #include <sys/module.h>
49 #include <sys/syslog.h>
50
51 #include <dev/dkvar.h>
52 #include <miscfs/specfs/specdev.h> /* for v_rdev */
53
54 int dkdebug = 0;
55
56 #ifdef DEBUG
57 #define DKDB_FOLLOW 0x1
58 #define DKDB_INIT 0x2
59 #define DKDB_VNODE 0x4
60 #define DKDB_DUMP 0x8
61
62 #define IFDEBUG(x,y) if (dkdebug & (x)) y
63 #define DPRINTF(x,y) IFDEBUG(x, printf y)
64 #define DPRINTF_FOLLOW(y) DPRINTF(DKDB_FOLLOW, y)
65 #else
66 #define IFDEBUG(x,y)
67 #define DPRINTF(x,y)
68 #define DPRINTF_FOLLOW(y)
69 #endif
70
71 #define DKF_READYFORDUMP (DKF_INITED|DKF_TAKEDUMP)
72
73 static int dk_subr_modcmd(modcmd_t, void *);
74
75 #define DKLABELDEV(dev) \
76 (MAKEDISKDEV(major((dev)), DISKUNIT((dev)), RAW_PART))
77
78 static void dk_makedisklabel(struct dk_softc *);
79 static int dk_translate(struct dk_softc *, struct buf *);
80 static void dk_done1(struct dk_softc *, struct buf *, bool);
81
82 void
83 dk_init(struct dk_softc *dksc, device_t dev, int dtype)
84 {
85
86 memset(dksc, 0x0, sizeof(*dksc));
87 dksc->sc_dtype = dtype;
88 dksc->sc_dev = dev;
89
90 strlcpy(dksc->sc_xname, device_xname(dev), DK_XNAME_SIZE);
91 dksc->sc_dkdev.dk_name = dksc->sc_xname;
92 }
93
94 void
95 dk_attach(struct dk_softc *dksc)
96 {
97 KASSERT(dksc->sc_dev != NULL);
98
99 mutex_init(&dksc->sc_iolock, MUTEX_DEFAULT, IPL_VM);
100 dksc->sc_flags |= DKF_READYFORDUMP;
101 #ifdef DIAGNOSTIC
102 dksc->sc_flags |= DKF_WARNLABEL | DKF_LABELSANITY;
103 #endif
104
105 /* Attach the device into the rnd source list. */
106 rnd_attach_source(&dksc->sc_rnd_source, dksc->sc_xname,
107 RND_TYPE_DISK, RND_FLAG_DEFAULT);
108 }
109
110 void
111 dk_detach(struct dk_softc *dksc)
112 {
113 /* Unhook the entropy source. */
114 rnd_detach_source(&dksc->sc_rnd_source);
115
116 dksc->sc_flags &= ~DKF_READYFORDUMP;
117 mutex_destroy(&dksc->sc_iolock);
118 }
119
120 /* ARGSUSED */
121 int
122 dk_open(struct dk_softc *dksc, dev_t dev,
123 int flags, int fmt, struct lwp *l)
124 {
125 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
126 struct disklabel *lp = dksc->sc_dkdev.dk_label;
127 int part = DISKPART(dev);
128 int pmask = 1 << part;
129 int ret = 0;
130 struct disk *dk = &dksc->sc_dkdev;
131
132 DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%x)\n", __func__,
133 dksc->sc_xname, dksc, dev, flags));
134
135 mutex_enter(&dk->dk_openlock);
136
137 /*
138 * If there are wedges, and this is not RAW_PART, then we
139 * need to fail.
140 */
141 if (dk->dk_nwedges != 0 && part != RAW_PART) {
142 ret = EBUSY;
143 goto done;
144 }
145
146 /*
147 * initialize driver for the first opener
148 */
149 if (dk->dk_openmask == 0 && dkd->d_firstopen != NULL) {
150 ret = (*dkd->d_firstopen)(dksc->sc_dev, dev, flags, fmt);
151 if (ret)
152 goto done;
153 }
154
155 /*
156 * If we're init'ed and there are no other open partitions then
157 * update the in-core disklabel.
158 */
159 if ((dksc->sc_flags & DKF_INITED)) {
160 if ((dksc->sc_flags & DKF_VLABEL) == 0) {
161 dksc->sc_flags |= DKF_VLABEL;
162 dk_getdisklabel(dksc, dev);
163 }
164 }
165
166 /* Fail if we can't find the partition. */
167 if (part != RAW_PART &&
168 ((dksc->sc_flags & DKF_VLABEL) == 0 ||
169 part >= lp->d_npartitions ||
170 lp->d_partitions[part].p_fstype == FS_UNUSED)) {
171 ret = ENXIO;
172 goto done;
173 }
174
175 /* Mark our unit as open. */
176 switch (fmt) {
177 case S_IFCHR:
178 dk->dk_copenmask |= pmask;
179 break;
180 case S_IFBLK:
181 dk->dk_bopenmask |= pmask;
182 break;
183 }
184
185 dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
186
187 done:
188 mutex_exit(&dk->dk_openlock);
189 return ret;
190 }
191
192 /* ARGSUSED */
193 int
194 dk_close(struct dk_softc *dksc, dev_t dev,
195 int flags, int fmt, struct lwp *l)
196 {
197 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
198 int part = DISKPART(dev);
199 int pmask = 1 << part;
200 struct disk *dk = &dksc->sc_dkdev;
201
202 DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%x)\n", __func__,
203 dksc->sc_xname, dksc, dev, flags));
204
205 mutex_enter(&dk->dk_openlock);
206
207 switch (fmt) {
208 case S_IFCHR:
209 dk->dk_copenmask &= ~pmask;
210 break;
211 case S_IFBLK:
212 dk->dk_bopenmask &= ~pmask;
213 break;
214 }
215 dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
216
217 if (dk->dk_openmask == 0) {
218 if (dkd->d_lastclose != NULL)
219 (*dkd->d_lastclose)(dksc->sc_dev);
220 if ((dksc->sc_flags & DKF_KLABEL) == 0)
221 dksc->sc_flags &= ~DKF_VLABEL;
222 }
223
224 mutex_exit(&dk->dk_openlock);
225 return 0;
226 }
227
228 static int
229 dk_translate(struct dk_softc *dksc, struct buf *bp)
230 {
231 int part;
232 int wlabel;
233 daddr_t blkno;
234 struct disklabel *lp;
235 struct disk *dk;
236 uint64_t numsecs;
237 unsigned secsize;
238
239 lp = dksc->sc_dkdev.dk_label;
240 dk = &dksc->sc_dkdev;
241
242 part = DISKPART(bp->b_dev);
243 numsecs = dk->dk_geom.dg_secperunit;
244 secsize = dk->dk_geom.dg_secsize;
245
246 /*
247 * The transfer must be a whole number of blocks and the offset must
248 * not be negative.
249 */
250 if ((bp->b_bcount % secsize) != 0 || bp->b_blkno < 0) {
251 bp->b_error = EINVAL;
252 goto done;
253 }
254
255 /* If there is nothing to do, then we are done */
256 if (bp->b_bcount == 0)
257 goto done;
258
259 wlabel = dksc->sc_flags & (DKF_WLABEL|DKF_LABELLING);
260 if (part == RAW_PART) {
261 uint64_t numblocks = btodb(numsecs * secsize);
262 if (bounds_check_with_mediasize(bp, DEV_BSIZE, numblocks) <= 0)
263 goto done;
264 } else {
265 if (bounds_check_with_label(&dksc->sc_dkdev, bp, wlabel) <= 0)
266 goto done;
267 }
268
269 /*
270 * Convert the block number to absolute and put it in terms
271 * of the device's logical block size.
272 */
273 if (secsize >= DEV_BSIZE)
274 blkno = bp->b_blkno / (secsize / DEV_BSIZE);
275 else
276 blkno = bp->b_blkno * (DEV_BSIZE / secsize);
277
278 if (part != RAW_PART)
279 blkno += lp->d_partitions[DISKPART(bp->b_dev)].p_offset;
280 bp->b_rawblkno = blkno;
281
282 return -1;
283
284 done:
285 bp->b_resid = bp->b_bcount;
286 return bp->b_error;
287 }
288
289 static int
290 dk_strategy1(struct dk_softc *dksc, struct buf *bp)
291 {
292 int error;
293
294 DPRINTF_FOLLOW(("%s(%s, %p, %p)\n", __func__,
295 dksc->sc_xname, dksc, bp));
296
297 if (!(dksc->sc_flags & DKF_INITED)) {
298 DPRINTF_FOLLOW(("%s: not inited\n", __func__));
299 bp->b_error = ENXIO;
300 bp->b_resid = bp->b_bcount;
301 biodone(bp);
302 return 1;
303 }
304
305 error = dk_translate(dksc, bp);
306 if (error >= 0) {
307 biodone(bp);
308 return 1;
309 }
310
311 return 0;
312 }
313
314 void
315 dk_strategy(struct dk_softc *dksc, struct buf *bp)
316 {
317 int error;
318
319 error = dk_strategy1(dksc, bp);
320 if (error)
321 return;
322
323 /*
324 * Queue buffer and start unit
325 */
326 dk_start(dksc, bp);
327 }
328
329 int
330 dk_strategy_defer(struct dk_softc *dksc, struct buf *bp)
331 {
332 int error;
333
334 error = dk_strategy1(dksc, bp);
335 if (error)
336 return error;
337
338 /*
339 * Queue buffer only
340 */
341 mutex_enter(&dksc->sc_iolock);
342 bufq_put(dksc->sc_bufq, bp);
343 mutex_exit(&dksc->sc_iolock);
344
345 return 0;
346 }
347
348 int
349 dk_strategy_pending(struct dk_softc *dksc)
350 {
351 struct buf *bp;
352
353 if (!(dksc->sc_flags & DKF_INITED)) {
354 DPRINTF_FOLLOW(("%s: not inited\n", __func__));
355 return 0;
356 }
357
358 mutex_enter(&dksc->sc_iolock);
359 bp = bufq_peek(dksc->sc_bufq);
360 mutex_exit(&dksc->sc_iolock);
361
362 return bp != NULL;
363 }
364
365 void
366 dk_start(struct dk_softc *dksc, struct buf *bp)
367 {
368 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
369 int error;
370
371 if (!(dksc->sc_flags & DKF_INITED)) {
372 DPRINTF_FOLLOW(("%s: not inited\n", __func__));
373 return;
374 }
375
376 mutex_enter(&dksc->sc_iolock);
377
378 if (bp != NULL)
379 bufq_put(dksc->sc_bufq, bp);
380
381 if (dksc->sc_busy)
382 goto done;
383 dksc->sc_busy = true;
384
385 /*
386 * Peeking at the buffer queue and committing the operation
387 * only after success isn't atomic.
388 *
389 * So when a diskstart fails, the buffer is saved
390 * and tried again before the next buffer is fetched.
391 * dk_drain() handles flushing of a saved buffer.
392 *
393 * This keeps order of I/O operations, unlike bufq_put.
394 */
395
396 bp = dksc->sc_deferred;
397 dksc->sc_deferred = NULL;
398
399 if (bp == NULL)
400 bp = bufq_get(dksc->sc_bufq);
401
402 while (bp != NULL) {
403
404 disk_busy(&dksc->sc_dkdev);
405 mutex_exit(&dksc->sc_iolock);
406 error = dkd->d_diskstart(dksc->sc_dev, bp);
407 mutex_enter(&dksc->sc_iolock);
408 if (error == EAGAIN) {
409 dksc->sc_deferred = bp;
410 disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
411 break;
412 }
413
414 if (error != 0) {
415 bp->b_error = error;
416 bp->b_resid = bp->b_bcount;
417 dk_done1(dksc, bp, false);
418 }
419
420 bp = bufq_get(dksc->sc_bufq);
421 }
422
423 dksc->sc_busy = false;
424 done:
425 mutex_exit(&dksc->sc_iolock);
426 }
427
428 static void
429 dk_done1(struct dk_softc *dksc, struct buf *bp, bool lock)
430 {
431 struct disk *dk = &dksc->sc_dkdev;
432
433 if (bp->b_error != 0) {
434 struct cfdriver *cd = device_cfdriver(dksc->sc_dev);
435
436 diskerr(bp, cd->cd_name, "error", LOG_PRINTF, 0,
437 dk->dk_label);
438 printf("\n");
439 }
440
441 if (lock)
442 mutex_enter(&dksc->sc_iolock);
443 disk_unbusy(dk, bp->b_bcount - bp->b_resid, (bp->b_flags & B_READ));
444 if (lock)
445 mutex_exit(&dksc->sc_iolock);
446
447 rnd_add_uint32(&dksc->sc_rnd_source, bp->b_rawblkno);
448
449 biodone(bp);
450 }
451
452 void
453 dk_done(struct dk_softc *dksc, struct buf *bp)
454 {
455 dk_done1(dksc, bp, true);
456 }
457
458 void
459 dk_drain(struct dk_softc *dksc)
460 {
461 struct buf *bp;
462
463 mutex_enter(&dksc->sc_iolock);
464 bp = dksc->sc_deferred;
465 dksc->sc_deferred = NULL;
466 if (bp != NULL) {
467 bp->b_error = EIO;
468 bp->b_resid = bp->b_bcount;
469 biodone(bp);
470 }
471 bufq_drain(dksc->sc_bufq);
472 mutex_exit(&dksc->sc_iolock);
473 }
474
475 int
476 dk_discard(struct dk_softc *dksc, dev_t dev, off_t pos, off_t len)
477 {
478 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
479 unsigned secsize = dksc->sc_dkdev.dk_geom.dg_secsize;
480 struct buf tmp, *bp = &tmp;
481 int error;
482
483 DPRINTF_FOLLOW(("%s(%s, %p, 0x"PRIx64", %jd, %jd)\n", __func__,
484 dksc->sc_xname, dksc, (intmax_t)pos, (intmax_t)len));
485
486 if (!(dksc->sc_flags & DKF_INITED)) {
487 DPRINTF_FOLLOW(("%s: not inited\n", __func__));
488 return ENXIO;
489 }
490
491 if (secsize == 0 || (pos % secsize) != 0)
492 return EINVAL;
493
494 /* enough data to please the bounds checking code */
495 bp->b_dev = dev;
496 bp->b_blkno = (daddr_t)(pos / secsize);
497 bp->b_bcount = len;
498 bp->b_flags = B_WRITE;
499
500 error = dk_translate(dksc, bp);
501 if (error >= 0)
502 return error;
503
504 error = dkd->d_discard(dksc->sc_dev,
505 (off_t)bp->b_rawblkno * secsize,
506 (off_t)bp->b_bcount);
507
508 return error;
509 }
510
511 int
512 dk_size(struct dk_softc *dksc, dev_t dev)
513 {
514 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
515 struct disklabel *lp;
516 int is_open;
517 int part;
518 int size;
519
520 if ((dksc->sc_flags & DKF_INITED) == 0)
521 return -1;
522
523 part = DISKPART(dev);
524 is_open = dksc->sc_dkdev.dk_openmask & (1 << part);
525
526 if (!is_open && dkd->d_open(dev, 0, S_IFBLK, curlwp))
527 return -1;
528
529 lp = dksc->sc_dkdev.dk_label;
530 if (lp->d_partitions[part].p_fstype != FS_SWAP)
531 size = -1;
532 else
533 size = lp->d_partitions[part].p_size *
534 (lp->d_secsize / DEV_BSIZE);
535
536 if (!is_open && dkd->d_close(dev, 0, S_IFBLK, curlwp))
537 return -1;
538
539 return size;
540 }
541
542 int
543 dk_ioctl(struct dk_softc *dksc, dev_t dev,
544 u_long cmd, void *data, int flag, struct lwp *l)
545 {
546 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
547 struct disklabel *lp;
548 struct disk *dk = &dksc->sc_dkdev;
549 #ifdef __HAVE_OLD_DISKLABEL
550 struct disklabel newlabel;
551 #endif
552 int error;
553
554 DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%lx)\n", __func__,
555 dksc->sc_xname, dksc, dev, cmd));
556
557 /* ensure that the pseudo disk is open for writes for these commands */
558 switch (cmd) {
559 case DIOCSDINFO:
560 case DIOCWDINFO:
561 #ifdef __HAVE_OLD_DISKLABEL
562 case ODIOCSDINFO:
563 case ODIOCWDINFO:
564 #endif
565 case DIOCKLABEL:
566 case DIOCWLABEL:
567 case DIOCAWEDGE:
568 case DIOCDWEDGE:
569 case DIOCSSTRATEGY:
570 if ((flag & FWRITE) == 0)
571 return EBADF;
572 }
573
574 /* ensure that the pseudo-disk is initialized for these */
575 switch (cmd) {
576 case DIOCGDINFO:
577 case DIOCSDINFO:
578 case DIOCWDINFO:
579 case DIOCGPARTINFO:
580 case DIOCKLABEL:
581 case DIOCWLABEL:
582 case DIOCGDEFLABEL:
583 case DIOCAWEDGE:
584 case DIOCDWEDGE:
585 case DIOCLWEDGES:
586 case DIOCMWEDGES:
587 case DIOCCACHESYNC:
588 #ifdef __HAVE_OLD_DISKLABEL
589 case ODIOCGDINFO:
590 case ODIOCSDINFO:
591 case ODIOCWDINFO:
592 case ODIOCGDEFLABEL:
593 #endif
594 if ((dksc->sc_flags & DKF_INITED) == 0)
595 return ENXIO;
596 }
597
598 error = disk_ioctl(dk, dev, cmd, data, flag, l);
599 if (error != EPASSTHROUGH)
600 return error;
601 else
602 error = 0;
603
604 switch (cmd) {
605 case DIOCWDINFO:
606 case DIOCSDINFO:
607 #ifdef __HAVE_OLD_DISKLABEL
608 case ODIOCWDINFO:
609 case ODIOCSDINFO:
610 #endif
611 #ifdef __HAVE_OLD_DISKLABEL
612 if (cmd == ODIOCSDINFO || cmd == ODIOCWDINFO) {
613 memset(&newlabel, 0, sizeof newlabel);
614 memcpy(&newlabel, data, sizeof (struct olddisklabel));
615 lp = &newlabel;
616 } else
617 #endif
618 lp = (struct disklabel *)data;
619
620 mutex_enter(&dk->dk_openlock);
621 dksc->sc_flags |= DKF_LABELLING;
622
623 error = setdisklabel(dksc->sc_dkdev.dk_label,
624 lp, 0, dksc->sc_dkdev.dk_cpulabel);
625 if (error == 0) {
626 if (cmd == DIOCWDINFO
627 #ifdef __HAVE_OLD_DISKLABEL
628 || cmd == ODIOCWDINFO
629 #endif
630 )
631 error = writedisklabel(DKLABELDEV(dev),
632 dkd->d_strategy, dksc->sc_dkdev.dk_label,
633 dksc->sc_dkdev.dk_cpulabel);
634 }
635
636 dksc->sc_flags &= ~DKF_LABELLING;
637 mutex_exit(&dk->dk_openlock);
638 break;
639
640 case DIOCKLABEL:
641 if (*(int *)data != 0)
642 dksc->sc_flags |= DKF_KLABEL;
643 else
644 dksc->sc_flags &= ~DKF_KLABEL;
645 break;
646
647 case DIOCWLABEL:
648 if (*(int *)data != 0)
649 dksc->sc_flags |= DKF_WLABEL;
650 else
651 dksc->sc_flags &= ~DKF_WLABEL;
652 break;
653
654 case DIOCGDEFLABEL:
655 dk_getdefaultlabel(dksc, (struct disklabel *)data);
656 break;
657
658 #ifdef __HAVE_OLD_DISKLABEL
659 case ODIOCGDEFLABEL:
660 dk_getdefaultlabel(dksc, &newlabel);
661 if (newlabel.d_npartitions > OLDMAXPARTITIONS)
662 return ENOTTY;
663 memcpy(data, &newlabel, sizeof (struct olddisklabel));
664 break;
665 #endif
666
667 case DIOCGSTRATEGY:
668 {
669 struct disk_strategy *dks = (void *)data;
670
671 mutex_enter(&dksc->sc_iolock);
672 if (dksc->sc_bufq != NULL)
673 strlcpy(dks->dks_name,
674 bufq_getstrategyname(dksc->sc_bufq),
675 sizeof(dks->dks_name));
676 else
677 error = EINVAL;
678 mutex_exit(&dksc->sc_iolock);
679 dks->dks_paramlen = 0;
680 break;
681 }
682
683 case DIOCSSTRATEGY:
684 {
685 struct disk_strategy *dks = (void *)data;
686 struct bufq_state *new;
687 struct bufq_state *old;
688
689 if (dks->dks_param != NULL) {
690 return EINVAL;
691 }
692 dks->dks_name[sizeof(dks->dks_name) - 1] = 0; /* ensure term */
693 error = bufq_alloc(&new, dks->dks_name,
694 BUFQ_EXACT|BUFQ_SORT_RAWBLOCK);
695 if (error) {
696 return error;
697 }
698 mutex_enter(&dksc->sc_iolock);
699 old = dksc->sc_bufq;
700 if (old)
701 bufq_move(new, old);
702 dksc->sc_bufq = new;
703 mutex_exit(&dksc->sc_iolock);
704 if (old)
705 bufq_free(old);
706 break;
707 }
708
709 default:
710 error = ENOTTY;
711 }
712
713 return error;
714 }
715
716 /*
717 * dk_dump dumps all of physical memory into the partition specified.
718 * This requires substantially more framework than {s,w}ddump, and hence
719 * is probably much more fragile.
720 *
721 */
722
723 #define DKFF_READYFORDUMP(x) (((x) & DKF_READYFORDUMP) == DKF_READYFORDUMP)
724 static volatile int dk_dumping = 0;
725
726 /* ARGSUSED */
727 int
728 dk_dump(struct dk_softc *dksc, dev_t dev,
729 daddr_t blkno, void *vav, size_t size)
730 {
731 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
732 char *va = vav;
733 struct disklabel *lp;
734 struct partition *p;
735 int part, towrt, nsects, sectoff, maxblkcnt, nblk;
736 int maxxfer, rv = 0;
737
738 /*
739 * ensure that we consider this device to be safe for dumping,
740 * and that the device is configured.
741 */
742 if (!DKFF_READYFORDUMP(dksc->sc_flags)) {
743 DPRINTF(DKDB_DUMP, ("%s: bad dump flags 0x%x\n", __func__,
744 dksc->sc_flags));
745 return ENXIO;
746 }
747
748 /* ensure that we are not already dumping */
749 if (dk_dumping)
750 return EFAULT;
751 dk_dumping = 1;
752
753 if (dkd->d_dumpblocks == NULL) {
754 DPRINTF(DKDB_DUMP, ("%s: no dumpblocks\n", __func__));
755 return ENXIO;
756 }
757
758 /* device specific max transfer size */
759 maxxfer = MAXPHYS;
760 if (dkd->d_iosize != NULL)
761 (*dkd->d_iosize)(dksc->sc_dev, &maxxfer);
762
763 /* Convert to disk sectors. Request must be a multiple of size. */
764 part = DISKPART(dev);
765 lp = dksc->sc_dkdev.dk_label;
766 if ((size % lp->d_secsize) != 0) {
767 DPRINTF(DKDB_DUMP, ("%s: odd size %zu\n", __func__, size));
768 return EFAULT;
769 }
770 towrt = size / lp->d_secsize;
771 blkno = dbtob(blkno) / lp->d_secsize; /* blkno in secsize units */
772
773 p = &lp->d_partitions[part];
774 if (p->p_fstype != FS_SWAP) {
775 DPRINTF(DKDB_DUMP, ("%s: bad fstype %d\n", __func__,
776 p->p_fstype));
777 return ENXIO;
778 }
779 nsects = p->p_size;
780 sectoff = p->p_offset;
781
782 /* Check transfer bounds against partition size. */
783 if ((blkno < 0) || ((blkno + towrt) > nsects)) {
784 DPRINTF(DKDB_DUMP, ("%s: out of bounds blkno=%jd, towrt=%d, "
785 "nsects=%d\n", __func__, (intmax_t)blkno, towrt, nsects));
786 return EINVAL;
787 }
788
789 /* Offset block number to start of partition. */
790 blkno += sectoff;
791
792 /* Start dumping and return when done. */
793 maxblkcnt = howmany(maxxfer, lp->d_secsize);
794 while (towrt > 0) {
795 nblk = min(maxblkcnt, towrt);
796
797 if ((rv = (*dkd->d_dumpblocks)(dksc->sc_dev, va, blkno, nblk))
798 != 0) {
799 DPRINTF(DKDB_DUMP, ("%s: dumpblocks %d\n", __func__,
800 rv));
801 return rv;
802 }
803
804 towrt -= nblk;
805 blkno += nblk;
806 va += nblk * lp->d_secsize;
807 }
808
809 dk_dumping = 0;
810
811 return 0;
812 }
813
814 /* ARGSUSED */
815 void
816 dk_getdefaultlabel(struct dk_softc *dksc, struct disklabel *lp)
817 {
818 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
819 struct disk_geom *dg = &dksc->sc_dkdev.dk_geom;
820
821 memset(lp, 0, sizeof(*lp));
822
823 if (dg->dg_secperunit > UINT32_MAX)
824 lp->d_secperunit = UINT32_MAX;
825 else
826 lp->d_secperunit = dg->dg_secperunit;
827 lp->d_secsize = dg->dg_secsize;
828 lp->d_nsectors = dg->dg_nsectors;
829 lp->d_ntracks = dg->dg_ntracks;
830 lp->d_ncylinders = dg->dg_ncylinders;
831 lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
832
833 strlcpy(lp->d_typename, dksc->sc_xname, sizeof(lp->d_typename));
834 lp->d_type = dksc->sc_dtype;
835 strlcpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
836 lp->d_rpm = 3600;
837 lp->d_interleave = 1;
838 lp->d_flags = 0;
839
840 lp->d_partitions[RAW_PART].p_offset = 0;
841 lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
842 lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
843 lp->d_npartitions = RAW_PART + 1;
844
845 lp->d_magic = DISKMAGIC;
846 lp->d_magic2 = DISKMAGIC;
847
848 if (dkd->d_label)
849 dkd->d_label(dksc->sc_dev, lp);
850
851 lp->d_checksum = dkcksum(lp);
852 }
853
854 /* ARGSUSED */
855 void
856 dk_getdisklabel(struct dk_softc *dksc, dev_t dev)
857 {
858 const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
859 struct disklabel *lp = dksc->sc_dkdev.dk_label;
860 struct cpu_disklabel *clp = dksc->sc_dkdev.dk_cpulabel;
861 struct disk_geom *dg = &dksc->sc_dkdev.dk_geom;
862 struct partition *pp;
863 int i;
864 const char *errstring;
865
866 memset(clp, 0x0, sizeof(*clp));
867 dk_getdefaultlabel(dksc, lp);
868 errstring = readdisklabel(DKLABELDEV(dev), dkd->d_strategy,
869 dksc->sc_dkdev.dk_label, dksc->sc_dkdev.dk_cpulabel);
870 if (errstring) {
871 dk_makedisklabel(dksc);
872 if (dksc->sc_flags & DKF_WARNLABEL)
873 printf("%s: %s\n", dksc->sc_xname, errstring);
874 return;
875 }
876
877 if ((dksc->sc_flags & DKF_LABELSANITY) == 0)
878 return;
879
880 /* Sanity check */
881 if (lp->d_secperunit < UINT32_MAX ?
882 lp->d_secperunit != dg->dg_secperunit :
883 lp->d_secperunit > dg->dg_secperunit)
884 printf("WARNING: %s: total sector size in disklabel (%ju) "
885 "!= the size of %s (%ju)\n", dksc->sc_xname,
886 (uintmax_t)lp->d_secperunit, dksc->sc_xname,
887 (uintmax_t)dg->dg_secperunit);
888
889 for (i=0; i < lp->d_npartitions; i++) {
890 pp = &lp->d_partitions[i];
891 if (pp->p_offset + pp->p_size > dg->dg_secperunit)
892 printf("WARNING: %s: end of partition `%c' exceeds "
893 "the size of %s (%ju)\n", dksc->sc_xname,
894 'a' + i, dksc->sc_xname,
895 (uintmax_t)dg->dg_secperunit);
896 }
897 }
898
899 /* ARGSUSED */
900 static void
901 dk_makedisklabel(struct dk_softc *dksc)
902 {
903 struct disklabel *lp = dksc->sc_dkdev.dk_label;
904
905 lp->d_partitions[RAW_PART].p_fstype = FS_BSDFFS;
906 strlcpy(lp->d_packname, "default label", sizeof(lp->d_packname));
907 lp->d_checksum = dkcksum(lp);
908 }
909
910 /* This function is taken from ccd.c:1.76 --rcd */
911
912 /*
913 * XXX this function looks too generic for dksubr.c, shouldn't we
914 * put it somewhere better?
915 */
916
917 /*
918 * Lookup the provided name in the filesystem. If the file exists,
919 * is a valid block device, and isn't being used by anyone else,
920 * set *vpp to the file's vnode.
921 */
922 int
923 dk_lookup(struct pathbuf *pb, struct lwp *l, struct vnode **vpp)
924 {
925 struct nameidata nd;
926 struct vnode *vp;
927 int error;
928
929 if (l == NULL)
930 return ESRCH; /* Is ESRCH the best choice? */
931
932 NDINIT(&nd, LOOKUP, FOLLOW, pb);
933 if ((error = vn_open(&nd, FREAD | FWRITE, 0)) != 0) {
934 DPRINTF((DKDB_FOLLOW|DKDB_INIT),
935 ("%s: vn_open error = %d\n", __func__, error));
936 return error;
937 }
938
939 vp = nd.ni_vp;
940 if (vp->v_type != VBLK) {
941 error = ENOTBLK;
942 goto out;
943 }
944
945 /* Reopen as anonymous vnode to protect against forced unmount. */
946 if ((error = bdevvp(vp->v_rdev, vpp)) != 0)
947 goto out;
948 VOP_UNLOCK(vp);
949 if ((error = vn_close(vp, FREAD | FWRITE, l->l_cred)) != 0) {
950 vrele(*vpp);
951 return error;
952 }
953 if ((error = VOP_OPEN(*vpp, FREAD | FWRITE, l->l_cred)) != 0) {
954 vrele(*vpp);
955 return error;
956 }
957 mutex_enter((*vpp)->v_interlock);
958 (*vpp)->v_writecount++;
959 mutex_exit((*vpp)->v_interlock);
960
961 IFDEBUG(DKDB_VNODE, vprint("dk_lookup: vnode info", *vpp));
962
963 return 0;
964 out:
965 VOP_UNLOCK(vp);
966 (void) vn_close(vp, FREAD | FWRITE, l->l_cred);
967 return error;
968 }
969
970 MODULE(MODULE_CLASS_MISC, dk_subr, NULL);
971
972 static int
973 dk_subr_modcmd(modcmd_t cmd, void *arg)
974 {
975 switch (cmd) {
976 case MODULE_CMD_INIT:
977 case MODULE_CMD_FINI:
978 return 0;
979 case MODULE_CMD_STAT:
980 case MODULE_CMD_AUTOUNLOAD:
981 default:
982 return ENOTTY;
983 }
984 }
985