NetBSD/usr.bin/eject/eject.c
1997-12-07 19:04:36 +00:00

388 lines
8.9 KiB
C

/* $NetBSD: eject.c,v 1.6 1997/12/07 19:04:36 msaitoh Exp $ */
/*
* Copyright (c) 1995
* Matthieu Herrb. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed for the NetBSD Project
* by Matthieu Herrb.
* 4. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: eject.c,v 1.6 1997/12/07 19:04:36 msaitoh Exp $");
#endif
/*
* Eject command
*
* It knows to eject floppies, CD-ROMS and tapes
* and tries to unmount file systems first
*/
#include <sys/param.h>
#include <sys/ioctl.h>
#include <sys/cdio.h>
#include <sys/mtio.h>
#include <sys/disklabel.h>
#include <sys/ucred.h>
#include <sys/mount.h>
#include <sys/cdefs.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <err.h>
#include <string.h>
typedef struct DEVTAB {
char *name;
char *device;
char qualifier;
u_int type;
} DEVTAB;
/*
* known device nicknames and types
* (used for selecting the proper ioctl to eject them)
*/
#define DISK 0x00000002
#define TAPE 0x00010000
#define MOUNTABLE(x) ((x) & 0x0000ffff)
static DEVTAB devtab[] = {
{ "diskette", "/dev/fd0", 'a', DISK },
{ "diskette0", "/dev/fd0", 'a', DISK },
{ "diskette1", "/dev/fd1", 'a', DISK },
{ "floppy", "/dev/fd0", 'a', DISK },
{ "floppy0", "/dev/fd0", 'a', DISK },
{ "floppy1", "/dev/fd1", 'a', DISK },
{ "fd", "/dev/fd0", 'a', DISK },
{ "fd0", "/dev/fd0", 'a', DISK },
{ "fd1", "/dev/fd1", 'a', DISK },
{ "cdrom", "/dev/cd0", 'a', DISK },
{ "cdrom0", "/dev/cd0", 'a', DISK },
{ "cdrom1", "/dev/cd1", 'a', DISK },
{ "cd", "/dev/cd0", 'a', DISK },
{ "cd0", "/dev/cd0", 'a', DISK },
{ "cd1", "/dev/cd1", 'a', DISK },
{ "mcd", "/dev/mcd0", 'a', DISK },
{ "mcd0", "/dev/mcd0", 'a', DISK },
{ "mcd1", "/dev/mcd1", 'a', DISK },
{ "tape", "/dev/rst0", '\0', TAPE },
{ "tape0", "/dev/rst0", '\0', TAPE },
{ "tape1", "/dev/rst1", '\0', TAPE },
{ "st", "/dev/rst0", '\0', TAPE },
{ "st0", "/dev/rst0", '\0', TAPE },
{ "st1", "/dev/rst1", '\0', TAPE },
{ "dat", "/dev/rst0", '\0', TAPE },
{ "dat0", "/dev/rst0", '\0', TAPE },
{ "dat1", "/dev/rst1", '\0', TAPE },
{ "exabyte", "/dev/rst0", '\0', TAPE },
{ "exabyte0", "/dev/rst0", '\0', TAPE },
{ "exabyte1", "/dev/rst1", '\0', TAPE },
{ NULL, NULL }
};
struct types {
char *str;
int type;
} types[] = {
{ "diskette", DISK },
{ "floppy", DISK },
{ "cdrom", DISK },
{ "disk", DISK },
{ "tape", TAPE },
{ NULL, 0 }
};
int verbose;
static void usage __P((void));
static char *device_by_name __P((char *, int *, char *));
static char *device_by_nickname __P((char *, int *, char *));
static void eject_disk __P((char *));
static void eject_tape __P((char *));
int main __P((int, char **));
static void umount_mounted __P((char *));
/*
* remind the syntax of the command to the user
*/
static void
usage()
{
fprintf(stderr,
"usage: eject [-n][-f][-t devtype][[-d] raw device | nickname ]\n");
exit(1);
/*NOTREACHED*/
}
/*
* given a device nick name, find its associated raw device and type
*/
static char *
device_by_nickname(name, pdevtype, pqualifier)
char *name;
int *pdevtype;
char *pqualifier;
{
int i;
for (i = 0; devtab[i].name != NULL; i++) {
if (strcmp(name, devtab[i].name) == 0) {
*pdevtype = devtab[i].type;
*pqualifier = devtab[i].qualifier;
return devtab[i].device;
}
}
*pdevtype = -1;
return NULL;
}
/*
* Given a raw device name, find its type and partition tag
* from the name.
*/
static char *
device_by_name(device, pdevtype, pqualifier)
char *device;
int *pdevtype;
char *pqualifier;
{
int i;
for (i = 0; devtab[i].name != NULL; i++) {
if (strncmp(devtab[i].device, device,
strlen(devtab[i].device)) == 0) {
*pdevtype = devtab[i].type;
*pqualifier = devtab[i].qualifier;
return devtab[i].device;
}
}
*pdevtype = -1;
return NULL;
}
/*
* eject a disk (including floppy and cdrom)
*/
static void
eject_disk(device)
char *device;
{
int fd, arg = 0;
fd = open(device, O_RDONLY);
if (fd < 0) {
err(1, "%s: open", device);
}
if (ioctl(fd, DIOCLOCK, (char *)&arg) < 0) {
err(1, "%s: DIOCLOCK", device);
}
if (ioctl(fd, DIOCEJECT, 0) < 0) {
err(1, "%s: DIOCEJECT", device);
}
if (close(fd) != 0)
err(1, "%s: close", device);
} /* eject_disk */
/*
* eject a tape
*/
static void
eject_tape(device)
char *device;
{
int fd;
struct mtop mt_com;
fd = open(device, O_RDONLY);
if (fd < 0) {
err(1, "open %s", device);
}
mt_com.mt_op = MTOFFL;
if (ioctl(fd, MTIOCTOP, &mt_com) < 0) {
err(1, "%s: MTOFFL", device);
}
close(fd);
} /* eject_tape */
/*
* test if partitions of a device are mounted
* and unmount them
*/
static void
umount_mounted(device)
char *device;
{
struct statfs *mntbuf;
int i, n, l;
n = getmntinfo(&mntbuf, MNT_NOWAIT);
if (n == 0) {
err(1, "getmntinfo");
}
l = strlen(device);
for (i = 0; i < n; i++) {
if (strncmp(device, mntbuf[i].f_mntfromname, l) == 0) {
if (verbose)
printf("Unmounting: %s\n",
mntbuf[i].f_mntonname);
if (unmount(mntbuf[i].f_mntonname, 0) < 0) {
err(1, "umount %s from %s",
mntbuf[i].f_mntfromname,
mntbuf[i].f_mntonname);
}
}
}
}
/*
* Eject - ejects various removable devices, including cdrom, tapes,
* diskettes, and other removable disks (like ZIP drives)
*/
int
main(argc, argv)
int argc;
char *argv[];
{
char device[MAXPATHLEN];
char *devpath;
char qualifier;
int umount_flag, devtype;
int i, ch;
/* Default options values */
devpath = NULL;
devtype = -1;
umount_flag = 1;
while ((ch = getopt(argc, argv, "d:fnt:v")) != -1) {
switch (ch) {
case 'd':
devpath = optarg;
break;
case 'f':
umount_flag = 0;
break;
case 'n':
for (i = 0; devtab[i].name != NULL; i++) {
if (devtab[i].qualifier != '\0') {
printf("%9s => %s%c\n", devtab[i].name,
devtab[i].device,
devtab[i].qualifier);
} else {
printf("%9s => %s\n", devtab[i].name,
devtab[i].device);
}
}
return 0;
case 't':
for (i = 0; types[i].str != NULL; i++) {
if (strcasecmp(optarg, types[i].str) == 0) {
devtype = types[i].type;
break;
}
}
if (devtype == -1)
errx(1, "%s: unknown device type", optarg);
break;
case 'v':
verbose = 1;
break;
case '?':
default:
usage();
/*NOTREACHED*/
}
}
argc -= optind;
argv += optind;
if (devpath != NULL) {
/* device specified with 'd' option */
if (devtype == -1)
device_by_name(devpath, &devtype, &qualifier);
else
qualifier = '\0';
} else {
if (argc <= 0) {
errx(1, "No device specified");
/* NOTREACHED */
}
if (strncmp(argv[0], "/dev/", 5) == 0) {
/*
* If argument begins with "/dev/", assume
* a device name.
*/
if (devtype == -1) {
devpath = device_by_name(argv[0],
&devtype, &qualifier);
} else {
/* Device type specified; use literally */
devpath = argv[0];
qualifier = '\0';
}
} else {
/* assume a nickname */
devpath = device_by_nickname(argv[0],
&devtype, &qualifier);
}
}
if (devpath == NULL) {
errx(1, "%s: unknown device", argv[0]);
/*NOTREACHED*/
}
if (umount_flag && MOUNTABLE(devtype)) {
umount_mounted(devpath);
}
snprintf(device, sizeof(device), "%s%c", devpath, qualifier);
if (verbose)
printf("Ejecting device `%s'\n", device);
switch (devtype) {
case DISK:
eject_disk(device);
break;
case TAPE:
eject_tape(device);
break;
default:
errx(1, "impossible... devtype = %d", devtype);
}
exit(0);
}