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