eject.c revision 1.2 1 1.2 pk /*
2 1.2 pk * Copyright (c) 1995
3 1.2 pk * Matthieu Herrb. All rights reserved.
4 1.2 pk *
5 1.2 pk * Redistribution and use in source and binary forms, with or without
6 1.2 pk * modification, are permitted provided that the following conditions
7 1.2 pk * are met:
8 1.2 pk * 1. Redistributions of source code must retain the above copyright
9 1.2 pk * notice, this list of conditions and the following disclaimer.
10 1.2 pk * 2. Redistributions in binary form must reproduce the above copyright
11 1.2 pk * notice, this list of conditions and the following disclaimer in the
12 1.2 pk * documentation and/or other materials provided with the distribution.
13 1.2 pk * 3. All advertising materials mentioning features or use of this software
14 1.2 pk * must display the following acknowledgement:
15 1.2 pk * This product includes software developed for the NetBSD Project
16 1.2 pk * by Matthieu Herrb.
17 1.2 pk * 4. The name of the author may not be used to endorse or promote products
18 1.2 pk * derived from this software without specific prior written permission.
19 1.2 pk *
20 1.2 pk * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21 1.2 pk * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22 1.2 pk * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23 1.2 pk * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
24 1.2 pk * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
25 1.2 pk * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26 1.2 pk * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
27 1.2 pk * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 1.2 pk * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 1.2 pk * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 1.2 pk * SUCH DAMAGE.
31 1.2 pk */
32 1.2 pk
33 1.2 pk /*
34 1.2 pk * Eject command
35 1.2 pk *
36 1.2 pk * It knows to eject floppies, CD-ROMS and tapes
37 1.2 pk * and tries to unmount file systems first
38 1.2 pk */
39 1.2 pk
40 1.2 pk #include <sys/param.h>
41 1.2 pk #include <sys/ioctl.h>
42 1.2 pk #include <sys/cdio.h>
43 1.2 pk #include <sys/mtio.h>
44 1.2 pk #include <sys/disklabel.h>
45 1.2 pk #include <sys/ucred.h>
46 1.2 pk #include <sys/mount.h>
47 1.2 pk #include <sys/cdefs.h>
48 1.2 pk
49 1.2 pk #include <stdio.h>
50 1.2 pk #include <stdlib.h>
51 1.2 pk #include <unistd.h>
52 1.2 pk #include <fcntl.h>
53 1.2 pk #include <err.h>
54 1.2 pk #include <string.h>
55 1.2 pk
56 1.2 pk typedef struct DEVTAB {
57 1.2 pk char *name;
58 1.2 pk char *device;
59 1.2 pk char qualifier;
60 1.2 pk u_int type;
61 1.2 pk } DEVTAB;
62 1.2 pk
63 1.2 pk /*
64 1.2 pk * known device nicknames and types
65 1.2 pk * (used for selecting the proper ioctl to eject them)
66 1.2 pk */
67 1.2 pk #define FLOPPY 0x00000001
68 1.2 pk #define CDROM 0x00000002
69 1.2 pk #define TAPE 0x00010000
70 1.2 pk
71 1.2 pk #define MOUNTABLE(x) ((x) & 0x0000ffff)
72 1.2 pk
73 1.2 pk static DEVTAB devtab[] = {
74 1.2 pk { "diskette", "/dev/fd0", 'a', FLOPPY },
75 1.2 pk { "diskette0", "/dev/fd0", 'a', FLOPPY },
76 1.2 pk { "diskette1", "/dev/fd1", 'a', FLOPPY },
77 1.2 pk { "floppy", "/dev/fd0", 'a', FLOPPY },
78 1.2 pk { "floppy0", "/dev/fd0", 'a', FLOPPY },
79 1.2 pk { "floppy1", "/dev/fd1", 'a', FLOPPY },
80 1.2 pk { "fd", "/dev/fd0", 'a', FLOPPY },
81 1.2 pk { "fd0", "/dev/fd0", 'a', FLOPPY },
82 1.2 pk { "fd1", "/dev/fd1", 'a', FLOPPY },
83 1.2 pk { "cdrom", "/dev/cd0", 'a', CDROM },
84 1.2 pk { "cdrom0", "/dev/cd0", 'a', CDROM },
85 1.2 pk { "cdrom1", "/dev/cd1", 'a', CDROM },
86 1.2 pk { "cd", "/dev/cd0", 'a', CDROM },
87 1.2 pk { "cd0", "/dev/cd0", 'a', CDROM },
88 1.2 pk { "cd1", "/dev/cd1", 'a', CDROM },
89 1.2 pk { "mcd", "/dev/mcd0", 'a', CDROM },
90 1.2 pk { "mcd0", "/dev/mcd0", 'a', CDROM },
91 1.2 pk { "mcd1", "/dev/mcd1", 'a', CDROM },
92 1.2 pk { "tape", "/dev/rst0", '\0', TAPE },
93 1.2 pk { "tape0", "/dev/rst0", '\0', TAPE },
94 1.2 pk { "tape1", "/dev/rst1", '\0', TAPE },
95 1.2 pk { "st", "/dev/rst0", '\0', TAPE },
96 1.2 pk { "st0", "/dev/rst0", '\0', TAPE },
97 1.2 pk { "st1", "/dev/rst1", '\0', TAPE },
98 1.2 pk { "dat", "/dev/rst0", '\0', TAPE },
99 1.2 pk { "dat0", "/dev/rst0", '\0', TAPE },
100 1.2 pk { "dat1", "/dev/rst1", '\0', TAPE },
101 1.2 pk { "exabyte", "/dev/rst0", '\0', TAPE },
102 1.2 pk { "exabyte0", "/dev/rst0", '\0', TAPE },
103 1.2 pk { "exabyte1", "/dev/rst1", '\0', TAPE },
104 1.2 pk { NULL, NULL }
105 1.2 pk };
106 1.2 pk
107 1.2 pk struct types {
108 1.2 pk char *str;
109 1.2 pk int type;
110 1.2 pk } types[] = {
111 1.2 pk { "diskette", FLOPPY },
112 1.2 pk { "floppy", FLOPPY },
113 1.2 pk { "cdrom", CDROM },
114 1.2 pk { "tape", TAPE },
115 1.2 pk { NULL, 0 }
116 1.2 pk };
117 1.2 pk
118 1.2 pk int verbose;
119 1.2 pk
120 1.2 pk /*
121 1.2 pk * remind the syntax of the command to the user
122 1.2 pk */
123 1.2 pk static void
124 1.2 pk usage()
125 1.2 pk {
126 1.2 pk errx(1, "usage: eject [-n][-f][-t devtype][[-d] raw device | nickname ]");
127 1.2 pk /*NOTREACHED*/
128 1.2 pk
129 1.2 pk }
130 1.2 pk
131 1.2 pk
132 1.2 pk /*
133 1.2 pk * given a device nick name, find its associated raw device and type
134 1.2 pk */
135 1.2 pk static char *
136 1.2 pk device_by_nickname(name, pdevtype, pqualifier)
137 1.2 pk char *name;
138 1.2 pk int *pdevtype;
139 1.2 pk char *pqualifier;
140 1.2 pk {
141 1.2 pk int i;
142 1.2 pk
143 1.2 pk for (i = 0; devtab[i].name != NULL; i++) {
144 1.2 pk if (strcmp(name, devtab[i].name) == 0) {
145 1.2 pk *pdevtype = devtab[i].type;
146 1.2 pk *pqualifier = devtab[i].qualifier;
147 1.2 pk return devtab[i].device;
148 1.2 pk }
149 1.2 pk }
150 1.2 pk *pdevtype = -1;
151 1.2 pk return NULL;
152 1.2 pk }
153 1.2 pk
154 1.2 pk /*
155 1.2 pk * Given a raw device name, find its type and partition tag
156 1.2 pk * from the name.
157 1.2 pk */
158 1.2 pk static char *
159 1.2 pk device_by_name(device, pdevtype, pqualifier)
160 1.2 pk char *device;
161 1.2 pk int *pdevtype;
162 1.2 pk char *pqualifier;
163 1.2 pk {
164 1.2 pk int i;
165 1.2 pk
166 1.2 pk for (i = 0; devtab[i].name != NULL; i++) {
167 1.2 pk if (strncmp(devtab[i].device, device,
168 1.2 pk strlen(devtab[i].device)) == 0) {
169 1.2 pk *pdevtype = devtab[i].type;
170 1.2 pk *pqualifier = devtab[i].qualifier;
171 1.2 pk return devtab[i].device;
172 1.2 pk }
173 1.2 pk }
174 1.2 pk *pdevtype = -1;
175 1.2 pk return NULL;
176 1.2 pk }
177 1.2 pk
178 1.2 pk /*
179 1.2 pk * eject a floppy
180 1.2 pk */
181 1.2 pk static void
182 1.2 pk eject_floppy(device)
183 1.2 pk char *device;
184 1.2 pk {
185 1.2 pk #ifdef __i386
186 1.2 pk printf("You may now press the eject button on the floppy drive...\n");
187 1.2 pk return;
188 1.2 pk #else
189 1.2 pk int fd;
190 1.2 pk
191 1.2 pk fd = open(device, O_RDONLY);
192 1.2 pk if (fd < 0) {
193 1.2 pk err(1, "open %s", device);
194 1.2 pk }
195 1.2 pk if (ioctl(fd, DIOCEJECT, 0) < 0) {
196 1.2 pk err(1, device);
197 1.2 pk }
198 1.2 pk close(fd);
199 1.2 pk #endif
200 1.2 pk } /* eject_floppy */
201 1.2 pk
202 1.2 pk /*
203 1.2 pk * eject a cd
204 1.2 pk */
205 1.2 pk static void
206 1.2 pk eject_cdrom(device)
207 1.2 pk char *device;
208 1.2 pk {
209 1.2 pk int fd;
210 1.2 pk
211 1.2 pk fd = open(device, O_RDONLY);
212 1.2 pk if (fd < 0) {
213 1.2 pk err(1, "%s: open", device);
214 1.2 pk }
215 1.2 pk if (ioctl(fd, CDIOCALLOW) < 0) {
216 1.2 pk err(1, "%s: CDIOCALLOW", device);
217 1.2 pk }
218 1.2 pk if (ioctl(fd, CDIOCEJECT, 0) < 0) {
219 1.2 pk err(1, "%s: CDIOCEJECT", device);
220 1.2 pk }
221 1.2 pk if (close(fd) != 0)
222 1.2 pk err(1, "%s: close", device);
223 1.2 pk } /* eject_cdrom */
224 1.2 pk
225 1.2 pk /*
226 1.2 pk * eject a tape
227 1.2 pk */
228 1.2 pk static void
229 1.2 pk eject_tape(device)
230 1.2 pk char *device;
231 1.2 pk {
232 1.2 pk int fd;
233 1.2 pk struct mtop mt_com;
234 1.2 pk
235 1.2 pk fd = open(device, O_RDONLY);
236 1.2 pk if (fd < 0) {
237 1.2 pk err(1, "open %s", device);
238 1.2 pk }
239 1.2 pk mt_com.mt_op = MTOFFL;
240 1.2 pk
241 1.2 pk if (ioctl(fd, MTIOCTOP, &mt_com) < 0) {
242 1.2 pk err(1, "%s: MTOFFL", device);
243 1.2 pk }
244 1.2 pk close(fd);
245 1.2 pk } /* eject_tape */
246 1.2 pk
247 1.2 pk /*
248 1.2 pk * test if partitions of a device are mounted
249 1.2 pk * and unmount them
250 1.2 pk */
251 1.2 pk static void
252 1.2 pk umount_mounted(device)
253 1.2 pk char *device;
254 1.2 pk {
255 1.2 pk struct statfs *mntbuf;
256 1.2 pk int i, n, l;
257 1.2 pk
258 1.2 pk n = getmntinfo(&mntbuf, MNT_NOWAIT);
259 1.2 pk if (n == 0) {
260 1.2 pk err(1, "getmntinfo");
261 1.2 pk }
262 1.2 pk l = strlen(device);
263 1.2 pk for (i = 0; i < n; i++) {
264 1.2 pk if (strncmp(device, mntbuf[i].f_mntfromname, l) == 0) {
265 1.2 pk if (verbose)
266 1.2 pk printf("Unmounting: %s\n",
267 1.2 pk mntbuf[i].f_mntonname);
268 1.2 pk if (unmount(mntbuf[i].f_mntonname, 0) < 0) {
269 1.2 pk err(1, "umount %s from %s",
270 1.2 pk mntbuf[i].f_mntfromname,
271 1.2 pk mntbuf[i].f_mntonname);
272 1.2 pk }
273 1.2 pk }
274 1.2 pk }
275 1.2 pk
276 1.2 pk }
277 1.2 pk
278 1.2 pk /*
279 1.2 pk * Eject - ejects various removable devices, including cdrom, tapes,
280 1.2 pk * diskettes
281 1.2 pk */
282 1.2 pk int
283 1.2 pk main(argc, argv)
284 1.2 pk int argc;
285 1.2 pk char *argv[];
286 1.2 pk {
287 1.2 pk char device[MAXPATHLEN];
288 1.2 pk char *devpath;
289 1.2 pk char qualifier;
290 1.2 pk int umount_flag, devtype;
291 1.2 pk int i, ch;
292 1.2 pk
293 1.2 pk /* Default options values */
294 1.2 pk devpath = NULL;
295 1.2 pk devtype = -1;
296 1.2 pk umount_flag = 1;
297 1.2 pk
298 1.2 pk while ((ch = getopt(argc, argv, "d:fnt:v")) != EOF) {
299 1.2 pk switch (ch) {
300 1.2 pk case 'd':
301 1.2 pk devpath = optarg;
302 1.2 pk break;
303 1.2 pk case 'f':
304 1.2 pk umount_flag = 0;
305 1.2 pk break;
306 1.2 pk case 'n':
307 1.2 pk for (i = 0; devtab[i].name != NULL; i++) {
308 1.2 pk printf("%9s => %s%c\n",
309 1.2 pk devtab[i].name,
310 1.2 pk devtab[i].device, devtab[i].qualifier);
311 1.2 pk }
312 1.2 pk return 0;
313 1.2 pk case 't':
314 1.2 pk for (i = 0; types[i].str != NULL; i++) {
315 1.2 pk if (strcasecmp(optarg, types[i].str) == 0) {
316 1.2 pk devtype = types[i].type;
317 1.2 pk break;
318 1.2 pk }
319 1.2 pk }
320 1.2 pk if (devtype == -1)
321 1.2 pk errx(1, "%s: unknown device type", optarg);
322 1.2 pk break;
323 1.2 pk case 'v':
324 1.2 pk verbose = 1;
325 1.2 pk break;
326 1.2 pk case '?':
327 1.2 pk default:
328 1.2 pk usage();
329 1.2 pk /*NOTREACHED*/
330 1.2 pk }
331 1.2 pk }
332 1.2 pk
333 1.2 pk argc -= optind;
334 1.2 pk argv += optind;
335 1.2 pk
336 1.2 pk if (devpath != NULL) {
337 1.2 pk /* device specified with 'd' option */
338 1.2 pk if (devtype == -1)
339 1.2 pk device_by_name(devpath, &devtype, &qualifier);
340 1.2 pk else
341 1.2 pk qualifier = '\0';
342 1.2 pk } else {
343 1.2 pk if (argc <= 0) {
344 1.2 pk errx(1, "No device specified");
345 1.2 pk /* NOTREACHED */
346 1.2 pk }
347 1.2 pk if (strncmp(argv[0], "/dev/", 5) == 0) {
348 1.2 pk /*
349 1.2 pk * If argument begins with "/dev/", assume
350 1.2 pk * a device name.
351 1.2 pk */
352 1.2 pk if (devtype == -1) {
353 1.2 pk devpath = device_by_name(argv[0],
354 1.2 pk &devtype, &qualifier);
355 1.2 pk } else {
356 1.2 pk /* Device type specified; use literally */
357 1.2 pk devpath = argv[0];
358 1.2 pk qualifier = '\0';
359 1.2 pk }
360 1.2 pk } else {
361 1.2 pk /* assume a nickname */
362 1.2 pk devpath = device_by_nickname(argv[0],
363 1.2 pk &devtype, &qualifier);
364 1.2 pk }
365 1.2 pk }
366 1.2 pk
367 1.2 pk if (devpath == NULL) {
368 1.2 pk errx(1, "%s: unknown device", argv[0]);
369 1.2 pk /*NOTREACHED*/
370 1.2 pk }
371 1.2 pk
372 1.2 pk if (umount_flag && MOUNTABLE(devtype)) {
373 1.2 pk umount_mounted(devpath);
374 1.2 pk }
375 1.2 pk
376 1.2 pk snprintf(device, sizeof(device), "%s%c", devpath, qualifier);
377 1.2 pk if (verbose)
378 1.2 pk printf("Ejecting device `%s'\n", device);
379 1.2 pk switch (devtype) {
380 1.2 pk case FLOPPY:
381 1.2 pk eject_floppy(device);
382 1.2 pk break;
383 1.2 pk case CDROM:
384 1.2 pk eject_cdrom(device);
385 1.2 pk break;
386 1.2 pk case TAPE:
387 1.2 pk eject_tape(device);
388 1.2 pk break;
389 1.2 pk default:
390 1.2 pk errx(1, "impossible... devtype = %d", devtype);
391 1.2 pk }
392 1.2 pk
393 1.2 pk exit(0);
394 1.2 pk }
395