busybox/miscutils/eject.c

/* vi: set sw=4 ts=4: */
/*
 * eject implementation for busybox
 *
 * Copyright (C) 2004  Peter Willis <psyphreak@phreaker.net>
 * Copyright (C) 2005  Tito Ragusa <farmatito@tiscali.it>
 *
 * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
 */

/*
 * This is a simple hack of eject based on something Erik posted in #uclibc.
 * Most of the dirty work blatantly ripped off from cat.c =)
 */

#include "libbb.h"

/* various defines swiped from linux/cdrom.h */
#define CDROMCLOSETRAY            0x5319  /* pendant of CDROMEJECT  */
#define CDROMEJECT                0x5309  /* Ejects the cdrom media */
#define CDROM_DRIVE_STATUS        0x5326  /* Get tray position, etc. */
/* drive status possibilities returned by CDROM_DRIVE_STATUS ioctl */
#define CDS_TRAY_OPEN        2

#define dev_fd 3

/* Code taken from the original eject (http://eject.sourceforge.net/),
 * refactored it a bit for busybox (ne-bb@nicoerfurth.de) */

#include <scsi/sg.h>
#include <scsi/scsi.h>

static void eject_scsi(const char *dev)
{
	static const char sg_commands[3][6] = {
		{ ALLOW_MEDIUM_REMOVAL, 0, 0, 0, 0, 0 },
		{ START_STOP, 0, 0, 0, 1, 0 },
		{ START_STOP, 0, 0, 0, 2, 0 }
	};

	unsigned i;
	unsigned char sense_buffer[32];
	unsigned char inqBuff[2];
	sg_io_hdr_t io_hdr;

	if ((ioctl(dev_fd, SG_GET_VERSION_NUM, &i) < 0) || (i < 30000))
		bb_error_msg_and_die("not a sg device or old sg driver");

	memset(&io_hdr, 0, sizeof(sg_io_hdr_t));
	io_hdr.interface_id = 'S';
	io_hdr.cmd_len = 6;
	io_hdr.mx_sb_len = sizeof(sense_buffer);
	io_hdr.dxfer_direction = SG_DXFER_NONE;
	/* io_hdr.dxfer_len = 0; */
	io_hdr.dxferp = inqBuff;
	io_hdr.sbp = sense_buffer;
	io_hdr.timeout = 2000;

	for (i = 0; i < 3; i++) {
		io_hdr.cmdp = (void *)sg_commands[i];
		ioctl_or_perror_and_die(dev_fd, SG_IO, (void *)&io_hdr, "%s", dev);
	}

	/* force kernel to reread partition table when new disc is inserted */
	ioctl(dev_fd, BLKRRPART);
}

#define FLAG_CLOSE  1
#define FLAG_SMART  2
#define FLAG_SCSI   4

static void eject_cdrom(unsigned flags, const char *dev)
{
	int cmd = CDROMEJECT;

	if (flags & FLAG_CLOSE
	 || (flags & FLAG_SMART && ioctl(dev_fd, CDROM_DRIVE_STATUS) == CDS_TRAY_OPEN)
	) {
		cmd = CDROMCLOSETRAY;
	}

	ioctl_or_perror_and_die(dev_fd, cmd, NULL, "%s", dev);
}

int eject_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int eject_main(int argc UNUSED_PARAM, char **argv)
{
	unsigned flags;
	const char *device;

	opt_complementary = "?1:t--T:T--t";
	flags = getopt32(argv, "tT" USE_FEATURE_EJECT_SCSI("s"));
	device = argv[optind] ? argv[optind] : "/dev/cdrom";

	/* We used to do "umount <device>" here, but it was buggy
	   if something was mounted OVER cdrom and
	   if cdrom is mounted many times.

	   This works equally well (or better):
	   #!/bin/sh
	   umount /dev/cdrom
	   eject /dev/cdrom
	*/

	xmove_fd(xopen(device, O_RDONLY|O_NONBLOCK), dev_fd);

	if (ENABLE_FEATURE_EJECT_SCSI && (flags & FLAG_SCSI))
		eject_scsi(device);
	else
		eject_cdrom(flags, device);

	if (ENABLE_FEATURE_CLEAN_UP)
		close(dev_fd);

	return EXIT_SUCCESS;
}
1	/* vi: set sw=4 ts=4: */
2	/*
3	* eject implementation for busybox
4	*
5	* Copyright (C) 2004 Peter Willis <psyphreak@phreaker.net>
6	* Copyright (C) 2005 Tito Ragusa <farmatito@tiscali.it>
7	*
8	* Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
9	*/
10
11	/*
12	* This is a simple hack of eject based on something Erik posted in #uclibc.
13	* Most of the dirty work blatantly ripped off from cat.c =)
14	*/
15
16	#include "libbb.h"
17
18	/* various defines swiped from linux/cdrom.h */
19	#define CDROMCLOSETRAY 0x5319 /* pendant of CDROMEJECT */
20	#define CDROMEJECT 0x5309 /* Ejects the cdrom media */
21	#define CDROM_DRIVE_STATUS 0x5326 /* Get tray position, etc. */
22	/* drive status possibilities returned by CDROM_DRIVE_STATUS ioctl */
23	#define CDS_TRAY_OPEN 2
24
25	#define dev_fd 3
26
27	/* Code taken from the original eject (http://eject.sourceforge.net/),
28	* refactored it a bit for busybox (ne-bb@nicoerfurth.de) */
29
30	#include <scsi/sg.h>
31	#include <scsi/scsi.h>
32
33	static void eject_scsi(const char *dev)
34	{
35	static const char sg_commands[3][6] = {
36	{ ALLOW_MEDIUM_REMOVAL, 0, 0, 0, 0, 0 },
37	{ START_STOP, 0, 0, 0, 1, 0 },
38	{ START_STOP, 0, 0, 0, 2, 0 }
39	};
40
41	unsigned i;
42	unsigned char sense_buffer[32];
43	unsigned char inqBuff[2];
44	sg_io_hdr_t io_hdr;
45
46	if ((ioctl(dev_fd, SG_GET_VERSION_NUM, &i) < 0) \|\| (i < 30000))
47	bb_error_msg_and_die("not a sg device or old sg driver");
48
49	memset(&io_hdr, 0, sizeof(sg_io_hdr_t));
50	io_hdr.interface_id = 'S';
51	io_hdr.cmd_len = 6;
52	io_hdr.mx_sb_len = sizeof(sense_buffer);
53	io_hdr.dxfer_direction = SG_DXFER_NONE;
54	/* io_hdr.dxfer_len = 0; */
55	io_hdr.dxferp = inqBuff;
56	io_hdr.sbp = sense_buffer;
57	io_hdr.timeout = 2000;
58
59	for (i = 0; i < 3; i++) {
60	io_hdr.cmdp = (void *)sg_commands[i];
61	ioctl_or_perror_and_die(dev_fd, SG_IO, (void *)&io_hdr, "%s", dev);
62	}
63
64	/* force kernel to reread partition table when new disc is inserted */
65	ioctl(dev_fd, BLKRRPART);
66	}
67
68	#define FLAG_CLOSE 1
69	#define FLAG_SMART 2
70	#define FLAG_SCSI 4
71
72	static void eject_cdrom(unsigned flags, const char *dev)
73	{
74	int cmd = CDROMEJECT;
75
76	if (flags & FLAG_CLOSE
77	\|\| (flags & FLAG_SMART && ioctl(dev_fd, CDROM_DRIVE_STATUS) == CDS_TRAY_OPEN)
78	) {
79	cmd = CDROMCLOSETRAY;
80	}
81
82	ioctl_or_perror_and_die(dev_fd, cmd, NULL, "%s", dev);
83	}
84
85	int eject_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
86	int eject_main(int argc UNUSED_PARAM, char **argv)
87	{
88	unsigned flags;
89	const char *device;
90
91	opt_complementary = "?1:t--T:T--t";
92	flags = getopt32(argv, "tT" USE_FEATURE_EJECT_SCSI("s"));
93	device = argv[optind] ? argv[optind] : "/dev/cdrom";
94
95	/* We used to do "umount <device>" here, but it was buggy
96	if something was mounted OVER cdrom and
97	if cdrom is mounted many times.
98
99	This works equally well (or better):
100	#!/bin/sh
101	umount /dev/cdrom
102	eject /dev/cdrom
103	*/
104
105	xmove_fd(xopen(device, O_RDONLY\|O_NONBLOCK), dev_fd);
106
107	if (ENABLE_FEATURE_EJECT_SCSI && (flags & FLAG_SCSI))
108	eject_scsi(device);
109	else
110	eject_cdrom(flags, device);
111
112	if (ENABLE_FEATURE_CLEAN_UP)
113	close(dev_fd);
114
115	return EXIT_SUCCESS;
116	}