Home | History | Annotate | Line # | Download | only in eject
eject.c revision 1.3
      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.3  thorpej #define DISK   0x00000002
     68  1.2       pk #define TAPE   0x00010000
     69  1.2       pk 
     70  1.2       pk #define MOUNTABLE(x) ((x) & 0x0000ffff)
     71  1.2       pk 
     72  1.2       pk static DEVTAB devtab[] = {
     73  1.3  thorpej 	{ "diskette", "/dev/fd0", 'a', DISK },
     74  1.3  thorpej 	{ "diskette0", "/dev/fd0", 'a', DISK },
     75  1.3  thorpej 	{ "diskette1", "/dev/fd1", 'a', DISK },
     76  1.3  thorpej 	{ "floppy", "/dev/fd0", 'a', DISK },
     77  1.3  thorpej 	{ "floppy0", "/dev/fd0", 'a', DISK },
     78  1.3  thorpej 	{ "floppy1", "/dev/fd1", 'a', DISK },
     79  1.3  thorpej 	{ "fd", "/dev/fd0", 'a', DISK },
     80  1.3  thorpej 	{ "fd0", "/dev/fd0", 'a', DISK },
     81  1.3  thorpej 	{ "fd1", "/dev/fd1", 'a', DISK },
     82  1.3  thorpej 	{ "cdrom", "/dev/cd0", 'a', DISK },
     83  1.3  thorpej 	{ "cdrom0", "/dev/cd0", 'a', DISK },
     84  1.3  thorpej 	{ "cdrom1", "/dev/cd1", 'a', DISK },
     85  1.3  thorpej 	{ "cd", "/dev/cd0", 'a', DISK },
     86  1.3  thorpej 	{ "cd0", "/dev/cd0", 'a', DISK },
     87  1.3  thorpej 	{ "cd1", "/dev/cd1", 'a', DISK },
     88  1.3  thorpej 	{ "mcd", "/dev/mcd0", 'a', DISK },
     89  1.3  thorpej 	{ "mcd0", "/dev/mcd0", 'a', DISK },
     90  1.3  thorpej 	{ "mcd1", "/dev/mcd1", 'a', DISK },
     91  1.2       pk 	{ "tape", "/dev/rst0", '\0', TAPE },
     92  1.2       pk 	{ "tape0", "/dev/rst0", '\0', TAPE },
     93  1.2       pk 	{ "tape1", "/dev/rst1", '\0', TAPE },
     94  1.2       pk 	{ "st", "/dev/rst0", '\0', TAPE },
     95  1.2       pk 	{ "st0", "/dev/rst0", '\0', TAPE },
     96  1.2       pk 	{ "st1", "/dev/rst1", '\0', TAPE },
     97  1.2       pk 	{ "dat", "/dev/rst0", '\0', TAPE },
     98  1.2       pk 	{ "dat0", "/dev/rst0", '\0', TAPE },
     99  1.2       pk 	{ "dat1", "/dev/rst1", '\0', TAPE },
    100  1.2       pk 	{ "exabyte", "/dev/rst0", '\0', TAPE },
    101  1.2       pk 	{ "exabyte0", "/dev/rst0", '\0', TAPE },
    102  1.2       pk 	{ "exabyte1", "/dev/rst1", '\0', TAPE },
    103  1.2       pk 	{ NULL, NULL }
    104  1.2       pk };
    105  1.2       pk 
    106  1.2       pk struct types {
    107  1.2       pk 	char	*str;
    108  1.2       pk 	int	type;
    109  1.2       pk } types[] = {
    110  1.3  thorpej 	{ "diskette", DISK },
    111  1.3  thorpej 	{ "floppy", DISK },
    112  1.3  thorpej 	{ "cdrom", DISK },
    113  1.3  thorpej 	{ "disk", DISK },
    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.3  thorpej  * eject a disk (including floppy and cdrom)
    180  1.2       pk  */
    181  1.2       pk static void
    182  1.3  thorpej eject_disk(device)
    183  1.2       pk 	char   *device;
    184  1.2       pk {
    185  1.3  thorpej 	int     fd, arg = 0;
    186  1.2       pk 
    187  1.2       pk 	fd = open(device, O_RDONLY);
    188  1.2       pk 	if (fd < 0) {
    189  1.2       pk 		err(1, "%s: open", device);
    190  1.2       pk 	}
    191  1.3  thorpej 	if (ioctl(fd, DIOCLOCK, (char *)&arg) < 0) {
    192  1.3  thorpej 		err(1, "%s: DIOCLOCK", device);
    193  1.2       pk 	}
    194  1.3  thorpej 	if (ioctl(fd, DIOCEJECT, 0) < 0) {
    195  1.3  thorpej 		err(1, "%s: DIOCEJECT", device);
    196  1.2       pk 	}
    197  1.2       pk 	if (close(fd) != 0)
    198  1.2       pk 		err(1, "%s: close", device);
    199  1.3  thorpej } /* eject_disk */
    200  1.2       pk 
    201  1.2       pk /*
    202  1.2       pk  * eject a tape
    203  1.2       pk  */
    204  1.2       pk static void
    205  1.2       pk eject_tape(device)
    206  1.2       pk 	char   *device;
    207  1.2       pk {
    208  1.2       pk 	int     fd;
    209  1.2       pk 	struct mtop mt_com;
    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, "open %s", device);
    214  1.2       pk 	}
    215  1.2       pk 	mt_com.mt_op = MTOFFL;
    216  1.2       pk 
    217  1.2       pk 	if (ioctl(fd, MTIOCTOP, &mt_com) < 0) {
    218  1.2       pk 		err(1, "%s:  MTOFFL", device);
    219  1.2       pk 	}
    220  1.2       pk 	close(fd);
    221  1.2       pk } /* eject_tape */
    222  1.2       pk 
    223  1.2       pk /*
    224  1.2       pk  * test if partitions of a device are mounted
    225  1.2       pk  * and unmount them
    226  1.2       pk  */
    227  1.2       pk static void
    228  1.2       pk umount_mounted(device)
    229  1.2       pk 	char   *device;
    230  1.2       pk {
    231  1.2       pk 	struct statfs *mntbuf;
    232  1.2       pk 	int     i, n, l;
    233  1.2       pk 
    234  1.2       pk 	n = getmntinfo(&mntbuf, MNT_NOWAIT);
    235  1.2       pk 	if (n == 0) {
    236  1.2       pk 		err(1, "getmntinfo");
    237  1.2       pk 	}
    238  1.2       pk 	l = strlen(device);
    239  1.2       pk 	for (i = 0; i < n; i++) {
    240  1.2       pk 		if (strncmp(device, mntbuf[i].f_mntfromname, l) == 0) {
    241  1.2       pk 			if (verbose)
    242  1.2       pk 				printf("Unmounting: %s\n",
    243  1.2       pk 					mntbuf[i].f_mntonname);
    244  1.2       pk 			if (unmount(mntbuf[i].f_mntonname, 0) < 0) {
    245  1.2       pk 				err(1, "umount %s from %s",
    246  1.2       pk 					mntbuf[i].f_mntfromname,
    247  1.2       pk 					mntbuf[i].f_mntonname);
    248  1.2       pk 			}
    249  1.2       pk 		}
    250  1.2       pk 	}
    251  1.2       pk 
    252  1.2       pk }
    253  1.2       pk 
    254  1.2       pk /*
    255  1.2       pk  * Eject - ejects various removable devices, including cdrom, tapes,
    256  1.3  thorpej  * diskettes, and other removable disks (like ZIP drives)
    257  1.2       pk  */
    258  1.2       pk int
    259  1.2       pk main(argc, argv)
    260  1.2       pk 	int     argc;
    261  1.2       pk 	char   *argv[];
    262  1.2       pk {
    263  1.2       pk 	char    device[MAXPATHLEN];
    264  1.2       pk 	char	*devpath;
    265  1.2       pk 	char	qualifier;
    266  1.2       pk 	int     umount_flag, devtype;
    267  1.2       pk 	int     i, ch;
    268  1.2       pk 
    269  1.2       pk 	/* Default options values */
    270  1.2       pk 	devpath = NULL;
    271  1.2       pk 	devtype = -1;
    272  1.2       pk 	umount_flag = 1;
    273  1.2       pk 
    274  1.2       pk 	while ((ch = getopt(argc, argv, "d:fnt:v")) != EOF) {
    275  1.2       pk 		switch (ch) {
    276  1.2       pk 		case 'd':
    277  1.2       pk 			devpath = optarg;
    278  1.2       pk 			break;
    279  1.2       pk 		case 'f':
    280  1.2       pk 			umount_flag = 0;
    281  1.2       pk 			break;
    282  1.2       pk 		case 'n':
    283  1.2       pk 			for (i = 0; devtab[i].name != NULL; i++) {
    284  1.2       pk 				printf("%9s => %s%c\n",
    285  1.2       pk 					devtab[i].name,
    286  1.2       pk 					devtab[i].device, devtab[i].qualifier);
    287  1.2       pk 			}
    288  1.2       pk 			return 0;
    289  1.2       pk 		case 't':
    290  1.2       pk 			for (i = 0; types[i].str != NULL; i++) {
    291  1.2       pk 				if (strcasecmp(optarg, types[i].str) == 0) {
    292  1.2       pk 					devtype = types[i].type;
    293  1.2       pk 					break;
    294  1.2       pk 				}
    295  1.2       pk 			}
    296  1.2       pk 			if (devtype == -1)
    297  1.2       pk 				errx(1, "%s: unknown device type", optarg);
    298  1.2       pk 			break;
    299  1.2       pk 		case 'v':
    300  1.2       pk 			verbose = 1;
    301  1.2       pk 			break;
    302  1.2       pk 		case '?':
    303  1.2       pk 		default:
    304  1.2       pk 			usage();
    305  1.2       pk 			/*NOTREACHED*/
    306  1.2       pk 		}
    307  1.2       pk 	}
    308  1.2       pk 
    309  1.2       pk 	argc -= optind;
    310  1.2       pk 	argv += optind;
    311  1.2       pk 
    312  1.2       pk 	if (devpath != NULL) {
    313  1.2       pk 		/* device specified with 'd' option */
    314  1.2       pk 		if (devtype == -1)
    315  1.2       pk 			device_by_name(devpath, &devtype, &qualifier);
    316  1.2       pk 		else
    317  1.2       pk 			qualifier = '\0';
    318  1.2       pk 	} else {
    319  1.2       pk 		if (argc <= 0) {
    320  1.2       pk 			errx(1, "No device specified");
    321  1.2       pk 			/* NOTREACHED */
    322  1.2       pk 		}
    323  1.2       pk 		if (strncmp(argv[0], "/dev/", 5) == 0) {
    324  1.2       pk 			/*
    325  1.2       pk 			 * If argument begins with "/dev/", assume
    326  1.2       pk 			 * a device name.
    327  1.2       pk 			 */
    328  1.2       pk 			if (devtype == -1) {
    329  1.2       pk 				devpath = device_by_name(argv[0],
    330  1.2       pk 							 &devtype, &qualifier);
    331  1.2       pk 			} else {
    332  1.2       pk 				/* Device type specified; use literally */
    333  1.2       pk 				devpath = argv[0];
    334  1.2       pk 				qualifier = '\0';
    335  1.2       pk 			}
    336  1.2       pk 		} else {
    337  1.2       pk 			/* assume a nickname */
    338  1.2       pk 			devpath = device_by_nickname(argv[0],
    339  1.2       pk 						     &devtype, &qualifier);
    340  1.2       pk 		}
    341  1.2       pk 	}
    342  1.2       pk 
    343  1.2       pk 	if (devpath == NULL) {
    344  1.2       pk 		errx(1, "%s: unknown device", argv[0]);
    345  1.2       pk 		/*NOTREACHED*/
    346  1.2       pk 	}
    347  1.2       pk 
    348  1.2       pk 	if (umount_flag && MOUNTABLE(devtype)) {
    349  1.2       pk 		umount_mounted(devpath);
    350  1.2       pk 	}
    351  1.2       pk 
    352  1.2       pk 	snprintf(device, sizeof(device), "%s%c", devpath, qualifier);
    353  1.2       pk 	if (verbose)
    354  1.2       pk 		printf("Ejecting device `%s'\n", device);
    355  1.2       pk 	switch (devtype) {
    356  1.3  thorpej 	case DISK:
    357  1.3  thorpej 		eject_disk(device);
    358  1.2       pk 		break;
    359  1.2       pk 	case TAPE:
    360  1.2       pk 		eject_tape(device);
    361  1.2       pk 		break;
    362  1.2       pk 	default:
    363  1.2       pk 		errx(1, "impossible... devtype = %d", devtype);
    364  1.2       pk 	}
    365  1.2       pk 
    366  1.2       pk 	exit(0);
    367  1.2       pk }
    368