diskprobe.c revision 1.1.2.2 1 1.1.2.2 skrll /* $NetBSD: diskprobe.c,v 1.1.2.2 2009/03/03 18:29:50 skrll Exp $ */
2 1.1.2.2 skrll /* $OpenBSD: diskprobe.c,v 1.3 2006/10/13 00:00:55 krw Exp $ */
3 1.1.2.2 skrll
4 1.1.2.2 skrll /*
5 1.1.2.2 skrll * Copyright (c) 1997 Tobias Weingartner
6 1.1.2.2 skrll * All rights reserved.
7 1.1.2.2 skrll *
8 1.1.2.2 skrll * Redistribution and use in source and binary forms, with or without
9 1.1.2.2 skrll * modification, are permitted provided that the following conditions
10 1.1.2.2 skrll * are met:
11 1.1.2.2 skrll * 1. Redistributions of source code must retain the above copyright
12 1.1.2.2 skrll * notice, this list of conditions and the following disclaimer.
13 1.1.2.2 skrll * 2. Redistributions in binary form must reproduce the above copyright
14 1.1.2.2 skrll * notice, this list of conditions and the following disclaimer in the
15 1.1.2.2 skrll * documentation and/or other materials provided with the distribution.
16 1.1.2.2 skrll *
17 1.1.2.2 skrll * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 1.1.2.2 skrll * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 1.1.2.2 skrll * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 1.1.2.2 skrll * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 1.1.2.2 skrll * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 1.1.2.2 skrll * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 1.1.2.2 skrll * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 1.1.2.2 skrll * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 1.1.2.2 skrll * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 1.1.2.2 skrll * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 1.1.2.2 skrll * SUCH DAMAGE.
28 1.1.2.2 skrll *
29 1.1.2.2 skrll */
30 1.1.2.2 skrll
31 1.1.2.2 skrll /* We want the disk type names from disklabel.h */
32 1.1.2.2 skrll #undef DKTYPENAMES
33 1.1.2.2 skrll
34 1.1.2.2 skrll #include <sys/param.h>
35 1.1.2.2 skrll #include <sys/bootblock.h>
36 1.1.2.2 skrll #include <sys/disklabel.h>
37 1.1.2.2 skrll #include <sys/queue.h>
38 1.1.2.2 skrll #include <sys/reboot.h>
39 1.1.2.2 skrll
40 1.1.2.2 skrll #include "boot.h"
41 1.1.2.2 skrll #include "disk.h"
42 1.1.2.2 skrll #include "unixdev.h"
43 1.1.2.2 skrll #include "compat_linux.h"
44 1.1.2.2 skrll
45 1.1.2.2 skrll /* Disk spin-up wait timeout. */
46 1.1.2.2 skrll static u_int timeout = 10;
47 1.1.2.2 skrll
48 1.1.2.2 skrll /* Local Prototypes */
49 1.1.2.2 skrll static void hardprobe(char *buf, size_t bufsiz);
50 1.1.2.2 skrll
51 1.1.2.2 skrll /* List of disk devices we found/probed */
52 1.1.2.2 skrll struct disklist_lh disklist;
53 1.1.2.2 skrll
54 1.1.2.2 skrll static char disk_devname[MAXDEVNAME];
55 1.1.2.2 skrll
56 1.1.2.2 skrll /*
57 1.1.2.2 skrll * Probe for all hard disks.
58 1.1.2.2 skrll */
59 1.1.2.2 skrll static void
60 1.1.2.2 skrll hardprobe(char *buf, size_t bufsiz)
61 1.1.2.2 skrll {
62 1.1.2.2 skrll /* XXX probe disks in this order */
63 1.1.2.2 skrll static const int order[] = { 0x80, 0x82, 0x00 };
64 1.1.2.2 skrll char devname[MAXDEVNAME];
65 1.1.2.2 skrll struct diskinfo *dip;
66 1.1.2.2 skrll u_int disk = 0;
67 1.1.2.2 skrll u_int hd_disk = 0;
68 1.1.2.2 skrll u_int mmcd_disk = 0;
69 1.1.2.2 skrll uint unit = 0;
70 1.1.2.2 skrll int first = 1;
71 1.1.2.2 skrll int i;
72 1.1.2.2 skrll
73 1.1.2.2 skrll buf[0] = '\0';
74 1.1.2.2 skrll
75 1.1.2.2 skrll /* Hard disks */
76 1.1.2.2 skrll for (i = 0; i < __arraycount(order); i++) {
77 1.1.2.2 skrll dip = alloc(sizeof(struct diskinfo));
78 1.1.2.2 skrll memset(dip, 0, sizeof(*dip));
79 1.1.2.2 skrll
80 1.1.2.2 skrll if (bios_getdiskinfo(order[i], &dip->bios_info) != NULL) {
81 1.1.2.2 skrll dealloc(dip, 0);
82 1.1.2.2 skrll continue;
83 1.1.2.2 skrll }
84 1.1.2.2 skrll
85 1.1.2.2 skrll bios_devname(order[i], devname, sizeof(devname));
86 1.1.2.2 skrll if (order[i] & 0x80) {
87 1.1.2.2 skrll unit = hd_disk;
88 1.1.2.2 skrll snprintf(dip->devname, sizeof(dip->devname), "%s%d",
89 1.1.2.2 skrll devname, hd_disk++);
90 1.1.2.2 skrll } else {
91 1.1.2.2 skrll unit = mmcd_disk;
92 1.1.2.2 skrll snprintf(dip->devname, sizeof(dip->devname), "%s%d",
93 1.1.2.2 skrll devname, mmcd_disk++);
94 1.1.2.2 skrll }
95 1.1.2.2 skrll strlcat(buf, dip->devname, bufsiz);
96 1.1.2.2 skrll disk++;
97 1.1.2.2 skrll
98 1.1.2.2 skrll /* Try to find the label, to figure out device type. */
99 1.1.2.2 skrll if (bios_getdisklabel(&dip->bios_info, &dip->disklabel)
100 1.1.2.2 skrll == NULL) {
101 1.1.2.2 skrll strlcat(buf, "*", bufsiz);
102 1.1.2.2 skrll if (first) {
103 1.1.2.2 skrll first = 0;
104 1.1.2.2 skrll strlcpy(disk_devname, devname,
105 1.1.2.2 skrll sizeof(disk_devname));
106 1.1.2.2 skrll default_devname = disk_devname;
107 1.1.2.2 skrll default_unit = unit;
108 1.1.2.2 skrll }
109 1.1.2.2 skrll } else {
110 1.1.2.2 skrll /* Best guess */
111 1.1.2.2 skrll switch (dip->disklabel.d_type) {
112 1.1.2.2 skrll case DTYPE_SCSI:
113 1.1.2.2 skrll case DTYPE_ESDI:
114 1.1.2.2 skrll case DTYPE_ST506:
115 1.1.2.2 skrll dip->bios_info.flags |= BDI_GOODLABEL;
116 1.1.2.2 skrll break;
117 1.1.2.2 skrll
118 1.1.2.2 skrll default:
119 1.1.2.2 skrll dip->bios_info.flags |= BDI_BADLABEL;
120 1.1.2.2 skrll }
121 1.1.2.2 skrll }
122 1.1.2.2 skrll
123 1.1.2.2 skrll /* Add to queue of disks. */
124 1.1.2.2 skrll TAILQ_INSERT_TAIL(&disklist, dip, list);
125 1.1.2.2 skrll
126 1.1.2.2 skrll strlcat(buf, " ", bufsiz);
127 1.1.2.2 skrll }
128 1.1.2.2 skrll if (disk == 0)
129 1.1.2.2 skrll strlcat(buf, "none...", bufsiz);
130 1.1.2.2 skrll }
131 1.1.2.2 skrll
132 1.1.2.2 skrll /* Probe for all BIOS supported disks */
133 1.1.2.2 skrll void
134 1.1.2.2 skrll diskprobe(char *buf, size_t bufsiz)
135 1.1.2.2 skrll {
136 1.1.2.2 skrll
137 1.1.2.2 skrll /* Init stuff */
138 1.1.2.2 skrll TAILQ_INIT(&disklist);
139 1.1.2.2 skrll
140 1.1.2.2 skrll /* Do probes */
141 1.1.2.2 skrll hardprobe(buf, bufsiz);
142 1.1.2.2 skrll }
143 1.1.2.2 skrll
144 1.1.2.2 skrll /*
145 1.1.2.2 skrll * Find info on the disk given by major + unit number.
146 1.1.2.2 skrll */
147 1.1.2.2 skrll struct diskinfo *
148 1.1.2.2 skrll dkdevice(const char *devname, uint unit)
149 1.1.2.2 skrll {
150 1.1.2.2 skrll char name[MAXDEVNAME];
151 1.1.2.2 skrll struct diskinfo *dip;
152 1.1.2.2 skrll
153 1.1.2.2 skrll snprintf(name, sizeof(name), "%s%d", devname, unit);
154 1.1.2.2 skrll for (dip = TAILQ_FIRST(&disklist); dip != NULL;
155 1.1.2.2 skrll dip = TAILQ_NEXT(dip, list)) {
156 1.1.2.2 skrll if (strcmp(name, dip->devname) == 0) {
157 1.1.2.2 skrll return dip;
158 1.1.2.2 skrll }
159 1.1.2.2 skrll }
160 1.1.2.2 skrll return NULL;
161 1.1.2.2 skrll }
162 1.1.2.2 skrll
163 1.1.2.2 skrll int
164 1.1.2.2 skrll bios_devname(int biosdev, char *devname, int size)
165 1.1.2.2 skrll {
166 1.1.2.2 skrll
167 1.1.2.2 skrll if ((biosdev & 0x80) != 0) {
168 1.1.2.2 skrll strlcpy(devname, devname_hd, size);
169 1.1.2.2 skrll } else {
170 1.1.2.2 skrll strlcpy(devname, devname_mmcd, size);
171 1.1.2.2 skrll }
172 1.1.2.2 skrll return 0;
173 1.1.2.2 skrll }
174 1.1.2.2 skrll
175 1.1.2.2 skrll /*
176 1.1.2.2 skrll * Find the Linux device path that corresponds to the given "BIOS" disk,
177 1.1.2.2 skrll * where 0x80 corresponds to /dev/hda, 0x81 to /dev/hdb, and so on.
178 1.1.2.2 skrll */
179 1.1.2.2 skrll void
180 1.1.2.2 skrll bios_devpath(int dev, int part, char *p)
181 1.1.2.2 skrll {
182 1.1.2.2 skrll char devname[MAXDEVNAME];
183 1.1.2.2 skrll const char *q;
184 1.1.2.2 skrll
185 1.1.2.2 skrll *p++ = '/';
186 1.1.2.2 skrll *p++ = 'd';
187 1.1.2.2 skrll *p++ = 'e';
188 1.1.2.2 skrll *p++ = 'v';
189 1.1.2.2 skrll *p++ = '/';
190 1.1.2.2 skrll
191 1.1.2.2 skrll bios_devname(dev, devname, sizeof(devname));
192 1.1.2.2 skrll q = devname;
193 1.1.2.2 skrll while (*q != '\0')
194 1.1.2.2 skrll *p++ = *q++;
195 1.1.2.2 skrll
196 1.1.2.2 skrll *p++ = 'a' + (dev & 0x7f);
197 1.1.2.2 skrll if (part >= 0)
198 1.1.2.2 skrll *p++ = '1' + part;
199 1.1.2.2 skrll *p = '\0';
200 1.1.2.2 skrll }
201 1.1.2.2 skrll
202 1.1.2.2 skrll /*
203 1.1.2.2 skrll * Fill out a bios_diskinfo_t for this device.
204 1.1.2.2 skrll */
205 1.1.2.2 skrll char *
206 1.1.2.2 skrll bios_getdiskinfo(int dev, bios_diskinfo_t *bdi)
207 1.1.2.2 skrll {
208 1.1.2.2 skrll static char path[PATH_MAX];
209 1.1.2.2 skrll struct linux_stat sb;
210 1.1.2.2 skrll
211 1.1.2.2 skrll memset(bdi, 0, sizeof *bdi);
212 1.1.2.2 skrll bdi->bios_number = -1;
213 1.1.2.2 skrll
214 1.1.2.2 skrll bios_devpath(dev, -1, path);
215 1.1.2.2 skrll
216 1.1.2.2 skrll if (ustat(path, &sb) != 0)
217 1.1.2.2 skrll return "no device node";
218 1.1.2.2 skrll
219 1.1.2.2 skrll bdi->bios_number = dev;
220 1.1.2.2 skrll
221 1.1.2.2 skrll if (bios_getdospart(bdi) < 0)
222 1.1.2.2 skrll return "no NetBSD partition";
223 1.1.2.2 skrll
224 1.1.2.2 skrll return NULL;
225 1.1.2.2 skrll }
226 1.1.2.2 skrll
227 1.1.2.2 skrll int
228 1.1.2.2 skrll bios_getdospart(bios_diskinfo_t *bdi)
229 1.1.2.2 skrll {
230 1.1.2.2 skrll char path[PATH_MAX];
231 1.1.2.2 skrll char buf[DEV_BSIZE];
232 1.1.2.2 skrll struct mbr_partition *mp;
233 1.1.2.2 skrll int fd;
234 1.1.2.2 skrll u_int part;
235 1.1.2.2 skrll size_t rsize;
236 1.1.2.2 skrll
237 1.1.2.2 skrll bios_devpath(bdi->bios_number, -1, path);
238 1.1.2.2 skrll
239 1.1.2.2 skrll /*
240 1.1.2.2 skrll * Give disk devices some time to become ready when the first open
241 1.1.2.2 skrll * fails. Even when open succeeds the disk is sometimes not ready.
242 1.1.2.2 skrll */
243 1.1.2.2 skrll if ((fd = uopen(path, LINUX_O_RDONLY)) == -1 && errno == ENXIO) {
244 1.1.2.2 skrll while (fd == -1 && timeout > 0) {
245 1.1.2.2 skrll timeout--;
246 1.1.2.2 skrll sleep(1);
247 1.1.2.2 skrll fd = uopen(path, LINUX_O_RDONLY);
248 1.1.2.2 skrll }
249 1.1.2.2 skrll if (fd != -1)
250 1.1.2.2 skrll sleep(2);
251 1.1.2.2 skrll }
252 1.1.2.2 skrll if (fd == -1)
253 1.1.2.2 skrll return -1;
254 1.1.2.2 skrll
255 1.1.2.2 skrll /* Read the disk's MBR. */
256 1.1.2.2 skrll if (unixstrategy((void *)fd, F_READ, MBR_BBSECTOR, DEV_BSIZE, buf,
257 1.1.2.2 skrll &rsize) != 0 || rsize != DEV_BSIZE) {
258 1.1.2.2 skrll uclose(fd);
259 1.1.2.2 skrll errno = EIO;
260 1.1.2.2 skrll return -1;
261 1.1.2.2 skrll }
262 1.1.2.2 skrll
263 1.1.2.2 skrll /* Find NetBSD primary partition in the disk's MBR. */
264 1.1.2.2 skrll mp = (struct mbr_partition *)&buf[MBR_PART_OFFSET];
265 1.1.2.2 skrll for (part = 0; part < MBR_PART_COUNT; part++) {
266 1.1.2.2 skrll if (mp[part].mbrp_type == MBR_PTYPE_NETBSD)
267 1.1.2.2 skrll break;
268 1.1.2.2 skrll }
269 1.1.2.2 skrll if (part == MBR_PART_COUNT) {
270 1.1.2.2 skrll uclose(fd);
271 1.1.2.2 skrll errno = ERDLAB;
272 1.1.2.2 skrll return -1;
273 1.1.2.2 skrll }
274 1.1.2.2 skrll uclose(fd);
275 1.1.2.2 skrll
276 1.1.2.2 skrll return part;
277 1.1.2.2 skrll }
278 1.1.2.2 skrll
279 1.1.2.2 skrll char *
280 1.1.2.2 skrll bios_getdisklabel(bios_diskinfo_t *bdi, struct disklabel *label)
281 1.1.2.2 skrll {
282 1.1.2.2 skrll char path[PATH_MAX];
283 1.1.2.2 skrll char buf[DEV_BSIZE];
284 1.1.2.2 skrll int part;
285 1.1.2.2 skrll int fd;
286 1.1.2.2 skrll size_t rsize;
287 1.1.2.2 skrll
288 1.1.2.2 skrll part = bios_getdospart(bdi);
289 1.1.2.2 skrll if (part < 0)
290 1.1.2.2 skrll return "no NetBSD partition";
291 1.1.2.2 skrll
292 1.1.2.2 skrll bios_devpath(bdi->bios_number, part, path);
293 1.1.2.2 skrll
294 1.1.2.2 skrll /* Test if the NetBSD partition has a valid disklabel. */
295 1.1.2.2 skrll if ((fd = uopen(path, LINUX_O_RDONLY)) != -1) {
296 1.1.2.2 skrll char *msg = "failed to read disklabel";
297 1.1.2.2 skrll
298 1.1.2.2 skrll if (unixstrategy((void *)fd, F_READ, LABELSECTOR,
299 1.1.2.2 skrll DEV_BSIZE, buf, &rsize) == 0 && rsize == DEV_BSIZE)
300 1.1.2.2 skrll msg = getdisklabel(buf, label);
301 1.1.2.2 skrll uclose(fd);
302 1.1.2.2 skrll /* Don't wait for other disks if this label is ok. */
303 1.1.2.2 skrll if (msg == NULL)
304 1.1.2.2 skrll timeout = 0;
305 1.1.2.2 skrll return msg;
306 1.1.2.2 skrll }
307 1.1.2.2 skrll
308 1.1.2.2 skrll return "failed to open partition";
309 1.1.2.2 skrll }
310