]> git.ipfire.org Git - thirdparty/util-linux.git/blob - sys-utils/eject.c
Merge branch 'master' of https://github.com/pali/util-linux
[thirdparty/util-linux.git] / sys-utils / eject.c
1 /*
2 * Copyright (C) 1994-2005 Jeff Tranter (tranter@pobox.com)
3 * Copyright (C) 2012 Karel Zak <kzak@redhat.com>
4 * Copyright (C) Michal Luscon <mluscon@redhat.com>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21 #include <unistd.h>
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <fcntl.h>
26 #include <limits.h>
27 #include <err.h>
28 #include <stdarg.h>
29
30 #include <getopt.h>
31 #include <errno.h>
32 #include <regex.h>
33 #include <sys/types.h>
34 #include <sys/stat.h>
35 #include <sys/ioctl.h>
36 #include <sys/wait.h>
37 #include <sys/mtio.h>
38 #include <linux/cdrom.h>
39 #include <linux/fd.h>
40 #include <sys/mount.h>
41 #include <scsi/scsi.h>
42 #include <scsi/sg.h>
43 #include <scsi/scsi_ioctl.h>
44 #include <sys/time.h>
45
46 #include <libmount.h>
47
48 #include "c.h"
49 #include "closestream.h"
50 #include "nls.h"
51 #include "strutils.h"
52 #include "xalloc.h"
53 #include "pathnames.h"
54 #include "sysfs.h"
55 #include "monotonic.h"
56
57 /*
58 * sg_io_hdr_t driver_status -- see kernel include/scsi/scsi.h
59 */
60 #ifndef DRIVER_SENSE
61 # define DRIVER_SENSE 0x08
62 #endif
63
64
65 #define EJECT_DEFAULT_DEVICE "/dev/cdrom"
66
67
68 /* Used by the toggle_tray() function. If ejecting the tray takes this
69 * time or less, the tray was probably already ejected, so we close it
70 * again.
71 */
72 #define TRAY_WAS_ALREADY_OPEN_USECS 200000 /* about 0.2 seconds */
73
74 struct eject_control {
75 struct libmnt_table *mtab;
76 char *device; /* device or mount point to be ejected */
77 int fd; /* file descriptor for device */
78 unsigned int /* command flags and arguments */
79 a_option:1,
80 c_option:1,
81 d_option:1,
82 F_option:1,
83 f_option:1,
84 i_option:1,
85 M_option:1,
86 m_option:1,
87 n_option:1,
88 p_option:1,
89 q_option:1,
90 r_option:1,
91 s_option:1,
92 T_option:1,
93 t_option:1,
94 v_option:1,
95 X_option:1,
96 x_option:1,
97 a_arg:1,
98 i_arg:1;
99 long int c_arg; /* changer slot number */
100 long int x_arg; /* cd speed */
101 };
102
103 static void vinfo(const char *fmt, va_list va)
104 {
105 fprintf(stdout, "%s: ", program_invocation_short_name);
106 vprintf(fmt, va);
107 fputc('\n', stdout);
108 }
109
110 static inline void verbose(const struct eject_control *ctl, const char *fmt, ...)
111 {
112 va_list va;
113
114 if (!ctl->v_option)
115 return;
116
117 va_start(va, fmt);
118 vinfo(fmt, va);
119 va_end(va);
120 }
121
122 static inline void info(const char *fmt, ...)
123 {
124 va_list va;
125 va_start(va, fmt);
126 vinfo(fmt, va);
127 va_end(va);
128 }
129
130 static void __attribute__((__noreturn__)) usage(void)
131 {
132 FILE *out = stdout;
133 fputs(USAGE_HEADER, out);
134 fprintf(out,
135 _(" %s [options] [<device>|<mountpoint>]\n"), program_invocation_short_name);
136
137 fputs(USAGE_SEPARATOR, out);
138 fputs(_("Eject removable media.\n"), out);
139
140 fputs(USAGE_OPTIONS, out);
141 fputs(_(" -a, --auto <on|off> turn auto-eject feature on or off\n"
142 " -c, --changerslot <slot> switch discs on a CD-ROM changer\n"
143 " -d, --default display default device\n"
144 " -f, --floppy eject floppy\n"
145 " -F, --force don't care about device type\n"
146 " -i, --manualeject <on|off> toggle manual eject protection on/off\n"
147 " -m, --no-unmount do not unmount device even if it is mounted\n"
148 " -M, --no-partitions-unmount do not unmount another partitions\n"
149 " -n, --noop don't eject, just show device found\n"
150 " -p, --proc use /proc/mounts instead of /etc/mtab\n"
151 " -q, --tape eject tape\n"
152 " -r, --cdrom eject CD-ROM\n"
153 " -s, --scsi eject SCSI device\n"
154 " -t, --trayclose close tray\n"
155 " -T, --traytoggle toggle tray\n"
156 " -v, --verbose enable verbose output\n"
157 " -x, --cdspeed <speed> set CD-ROM max speed\n"
158 " -X, --listspeed list CD-ROM available speeds\n"),
159 out);
160
161 fputs(USAGE_SEPARATOR, out);
162 printf(USAGE_HELP_OPTIONS(29));
163
164 fputs(_("\nBy default tries -r, -s, -f, and -q in order until success.\n"), out);
165 printf(USAGE_MAN_TAIL("eject(1)"));
166
167 exit(EXIT_SUCCESS);
168 }
169
170
171 /* Handle command line options. */
172 static void parse_args(struct eject_control *ctl, int argc, char **argv)
173 {
174 static const struct option long_opts[] =
175 {
176 {"auto", required_argument, NULL, 'a'},
177 {"cdrom", no_argument, NULL, 'r'},
178 {"cdspeed", required_argument, NULL, 'x'},
179 {"changerslot", required_argument, NULL, 'c'},
180 {"default", no_argument, NULL, 'd'},
181 {"floppy", no_argument, NULL, 'f'},
182 {"force", no_argument, NULL, 'F'},
183 {"help", no_argument, NULL, 'h'},
184 {"listspeed", no_argument, NULL, 'X'},
185 {"manualeject", required_argument, NULL, 'i'},
186 {"noop", no_argument, NULL, 'n'},
187 {"no-unmount", no_argument, NULL, 'm'},
188 {"no-partitions-unmount", no_argument, NULL, 'M' },
189 {"proc", no_argument, NULL, 'p'},
190 {"scsi", no_argument, NULL, 's'},
191 {"tape", no_argument, NULL, 'q'},
192 {"trayclose", no_argument, NULL, 't'},
193 {"traytoggle", no_argument, NULL, 'T'},
194 {"verbose", no_argument, NULL, 'v'},
195 {"version", no_argument, NULL, 'V'},
196 {NULL, 0, NULL, 0}
197 };
198 int c;
199
200 while ((c = getopt_long(argc, argv,
201 "a:c:i:x:dfFhnqrstTXvVpmM", long_opts, NULL)) != -1) {
202 switch (c) {
203 case 'a':
204 ctl->a_option = 1;
205 ctl->a_arg = parse_switch(optarg, _("argument error"),
206 "on", "off", "1", "0", NULL);
207 break;
208 case 'c':
209 ctl->c_option = 1;
210 ctl->c_arg = strtoul_or_err(optarg, _("invalid argument to --changerslot/-c option"));
211 break;
212 case 'x':
213 ctl->x_option = 1;
214 ctl->x_arg = strtoul_or_err(optarg, _("invalid argument to --cdspeed/-x option"));
215 break;
216 case 'd':
217 ctl->d_option = 1;
218 break;
219 case 'f':
220 ctl->f_option = 1;
221 break;
222 case 'F':
223 ctl->F_option = 1;
224 break;
225 case 'h':
226 usage();
227 break;
228 case 'i':
229 ctl->i_option = 1;
230 ctl->i_arg = parse_switch(optarg, _("argument error"),
231 "on", "off", "1", "0", NULL);
232 break;
233 case 'm':
234 ctl->m_option = 1;
235 break;
236 case 'M':
237 ctl->M_option = 1;
238 break;
239 case 'n':
240 ctl->n_option = 1;
241 break;
242 case 'p':
243 ctl->p_option = 1;
244 break;
245 case 'q':
246 ctl->q_option = 1;
247 break;
248 case 'r':
249 ctl->r_option = 1;
250 break;
251 case 's':
252 ctl->s_option = 1;
253 break;
254 case 't':
255 ctl->t_option = 1;
256 break;
257 case 'T':
258 ctl->T_option = 1;
259 break;
260 case 'X':
261 ctl->X_option = 1;
262 break;
263 case 'v':
264 ctl->v_option = 1;
265 break;
266 case 'V':
267 printf(UTIL_LINUX_VERSION);
268 exit(EXIT_SUCCESS);
269 break;
270 default:
271 errtryhelp(EXIT_FAILURE);
272 break;
273 }
274 }
275
276 /* check for a single additional argument */
277 if ((argc - optind) > 1)
278 errx(EXIT_FAILURE, _("too many arguments"));
279
280 if ((argc - optind) == 1)
281 ctl->device = xstrdup(argv[optind]);
282 }
283
284 /*
285 * Given name, such as foo, see if any of the following exist:
286 *
287 * foo (if foo starts with '.' or '/')
288 * /dev/foo
289 *
290 * If found, return the full path. If not found, return 0.
291 * Returns pointer to dynamically allocated string.
292 */
293 static char *find_device(const char *name)
294 {
295 if (!name)
296 return NULL;
297
298 if ((*name == '.' || *name == '/') && access(name, F_OK) == 0)
299 return xstrdup(name);
300 else {
301 char buf[PATH_MAX];
302
303 snprintf(buf, sizeof(buf), "/dev/%s", name);
304 if (access(buf, F_OK) == 0)
305 return xstrdup(buf);
306 }
307
308 return NULL;
309 }
310
311 /* Set or clear auto-eject mode. */
312 static void auto_eject(const struct eject_control *ctl)
313 {
314 int status = -1;
315
316 #if defined(CDROM_SET_OPTIONS) && defined(CDROM_CLEAR_OPTIONS)
317 if (ctl->a_arg)
318 status = ioctl(ctl->fd, CDROM_SET_OPTIONS, CDO_AUTO_EJECT);
319 else
320 status = ioctl(ctl->fd, CDROM_CLEAR_OPTIONS, CDO_AUTO_EJECT);
321 #else
322 errno = ENOSYS;
323 #endif
324 if (status < 0)
325 err(EXIT_FAILURE,_("CD-ROM auto-eject command failed"));
326 }
327
328 /*
329 * Stops CDROM from opening on manual eject button press.
330 * This can be useful when you carry your laptop
331 * in your bag while it's on and no CD inserted in it's drive.
332 * Implemented as found in Documentation/ioctl/cdrom.txt
333 */
334 static void manual_eject(const struct eject_control *ctl)
335 {
336 if (ioctl(ctl->fd, CDROM_LOCKDOOR, ctl->i_arg) < 0) {
337 switch (errno) {
338 case EDRIVE_CANT_DO_THIS:
339 errx(EXIT_FAILURE, _("CD-ROM door lock is not supported"));
340 case EBUSY:
341 errx(EXIT_FAILURE, _("other users have the drive open and not CAP_SYS_ADMIN"));
342 default:
343 err(EXIT_FAILURE, _("CD-ROM lock door command failed"));
344 }
345 }
346
347 if (ctl->i_arg)
348 info(_("CD-Drive may NOT be ejected with device button"));
349 else
350 info(_("CD-Drive may be ejected with device button"));
351 }
352
353 /*
354 * Changer select. CDROM_SELECT_DISC is preferred, older kernels used
355 * CDROMLOADFROMSLOT.
356 */
357 static void changer_select(const struct eject_control *ctl)
358 {
359 #ifdef CDROM_SELECT_DISC
360 if (ioctl(ctl->fd, CDROM_SELECT_DISC, ctl->c_arg) < 0)
361 err(EXIT_FAILURE, _("CD-ROM select disc command failed"));
362
363 #elif defined CDROMLOADFROMSLOT
364 if (ioctl(ctl->fd, CDROMLOADFROMSLOT, ctl->c_arg) != 0)
365 err(EXIT_FAILURE, _("CD-ROM load from slot command failed"));
366 #else
367 warnx(_("IDE/ATAPI CD-ROM changer not supported by this kernel\n") );
368 #endif
369 }
370
371 /*
372 * Close tray. Not supported by older kernels.
373 */
374 static void close_tray(int fd)
375 {
376 int status;
377
378 #if defined(CDROMCLOSETRAY) || defined(CDIOCCLOSE)
379 #if defined(CDROMCLOSETRAY)
380 status = ioctl(fd, CDROMCLOSETRAY);
381 #elif defined(CDIOCCLOSE)
382 status = ioctl(fd, CDIOCCLOSE);
383 #endif
384 if (status != 0)
385 err(EXIT_FAILURE, _("CD-ROM tray close command failed"));
386 #else
387 warnx(_("CD-ROM tray close command not supported by this kernel\n"));
388 #endif
389 }
390
391 /*
392 * Eject using CDROMEJECT ioctl.
393 */
394 static int eject_cdrom(int fd)
395 {
396 #if defined(CDROMEJECT)
397 int ret = ioctl(fd, CDROM_LOCKDOOR, 0);
398 if (ret < 0)
399 return 0;
400 return ioctl(fd, CDROMEJECT) >= 0;
401 #elif defined(CDIOCEJECT)
402 return ioctl(fd, CDIOCEJECT) >= 0;
403 #else
404 warnx(_("CD-ROM eject unsupported"));
405 errno = ENOSYS;
406 return 0;
407 #endif
408 }
409
410 /*
411 * Toggle tray.
412 *
413 * Written by Benjamin Schwenk <benjaminschwenk@yahoo.de> and
414 * Sybren Stuvel <sybren@thirdtower.com>
415 *
416 * Not supported by older kernels because it might use
417 * CloseTray().
418 *
419 */
420 static void toggle_tray(int fd)
421 {
422 #ifdef CDROM_DRIVE_STATUS
423 /* First ask the CDROM for info, otherwise fall back to manual. */
424 switch (ioctl(fd, CDROM_DRIVE_STATUS)) {
425 case CDS_TRAY_OPEN:
426 close_tray(fd);
427 return;
428
429 case CDS_NO_DISC:
430 case CDS_DISC_OK:
431 if (!eject_cdrom(fd))
432 err(EXIT_FAILURE, _("CD-ROM eject command failed"));
433 return;
434 case CDS_NO_INFO:
435 warnx(_("no CD-ROM information available"));
436 return;
437 case CDS_DRIVE_NOT_READY:
438 warnx(_("CD-ROM drive is not ready"));
439 return;
440 default:
441 abort();
442 }
443 #else
444 struct timeval time_start, time_stop;
445 int time_elapsed;
446
447 /* Try to open the CDROM tray and measure the time therefor
448 * needed. In my experience the function needs less than 0.05
449 * seconds if the tray was already open, and at least 1.5 seconds
450 * if it was closed. */
451 gettime_monotonic(&time_start);
452
453 /* Send the CDROMEJECT command to the device. */
454 if (!eject_cdrom(fd))
455 err(EXIT_FAILURE, _("CD-ROM eject command failed"));
456
457 /* Get the second timestamp, to measure the time needed to open
458 * the tray. */
459 gettime_monotonic(&time_stop);
460
461 time_elapsed = (time_stop.tv_sec * 1000000 + time_stop.tv_usec) -
462 (time_start.tv_sec * 1000000 + time_start.tv_usec);
463
464 /* If the tray "opened" too fast, we can be nearly sure, that it
465 * was already open. In this case, close it now. Else the tray was
466 * closed before. This would mean that we are done. */
467 if (time_elapsed < TRAY_WAS_ALREADY_OPEN_USECS)
468 close_tray(fd);
469 #endif
470 }
471
472 /*
473 * Select Speed of CD-ROM drive.
474 * Thanks to Roland Krivanek (krivanek@fmph.uniba.sk)
475 * http://dmpc.dbp.fmph.uniba.sk/~krivanek/cdrom_speed/
476 */
477 static void select_speed(const struct eject_control *ctl)
478 {
479 #ifdef CDROM_SELECT_SPEED
480 if (ioctl(ctl->fd, CDROM_SELECT_SPEED, ctl->x_arg) != 0)
481 err(EXIT_FAILURE, _("CD-ROM select speed command failed"));
482 #else
483 warnx(_("CD-ROM select speed command not supported by this kernel"));
484 #endif
485 }
486
487 /*
488 * Read Speed of CD-ROM drive. From Linux 2.6.13, the current speed
489 * is correctly reported
490 */
491 static int read_speed(const char *devname)
492 {
493 int drive_number = -1;
494 char *name;
495 FILE *f;
496
497 f = fopen(_PATH_PROC_CDROMINFO, "r");
498 if (!f)
499 err(EXIT_FAILURE, _("cannot open %s"), _PATH_PROC_CDROMINFO);
500
501 name = strrchr(devname, '/') + 1;
502
503 while (name && !feof(f)) {
504 char line[512];
505 char *str;
506
507 if (!fgets(line, sizeof(line), f))
508 break;
509
510 /* find drive number in line "drive name" */
511 if (drive_number == -1) {
512 if (strncmp(line, "drive name:", 11) == 0) {
513 str = strtok(&line[11], "\t ");
514 drive_number = 0;
515 while (str && strncmp(name, str, strlen(name)) != 0) {
516 drive_number++;
517 str = strtok(NULL, "\t ");
518 if (!str)
519 errx(EXIT_FAILURE,
520 _("%s: failed to finding CD-ROM name"),
521 _PATH_PROC_CDROMINFO);
522 }
523 }
524 /* find line "drive speed" and read the correct speed */
525 } else {
526 if (strncmp(line, "drive speed:", 12) == 0) {
527 int i;
528
529 str = strtok(&line[12], "\t ");
530 for (i = 1; i < drive_number; i++)
531 str = strtok(NULL, "\t ");
532
533 if (!str)
534 errx(EXIT_FAILURE,
535 _("%s: failed to read speed"),
536 _PATH_PROC_CDROMINFO);
537 fclose(f);
538 return atoi(str);
539 }
540 }
541 }
542
543 errx(EXIT_FAILURE, _("failed to read speed"));
544 }
545
546 /*
547 * List Speed of CD-ROM drive.
548 */
549 static void list_speeds(struct eject_control *ctl)
550 {
551 #ifdef CDROM_SELECT_SPEED
552 int max_speed, curr_speed = 0;
553
554 select_speed(ctl);
555 max_speed = read_speed(ctl->device);
556
557 while (curr_speed < max_speed) {
558 ctl->x_arg = curr_speed + 1;
559 select_speed(ctl);
560 curr_speed = read_speed(ctl->device);
561 if (ctl->x_arg < curr_speed)
562 printf("%d ", curr_speed);
563 else
564 curr_speed = ctl->x_arg + 1;
565 }
566
567 printf("\n");
568 #else
569 warnx(_("CD-ROM select speed command not supported by this kernel"));
570 #endif
571 }
572
573 /*
574 * Eject using SCSI SG_IO commands. Return 1 if successful, 0 otherwise.
575 */
576 static int eject_scsi(const struct eject_control *ctl)
577 {
578 int status, k;
579 sg_io_hdr_t io_hdr;
580 unsigned char allowRmBlk[6] = {ALLOW_MEDIUM_REMOVAL, 0, 0, 0, 0, 0};
581 unsigned char startStop1Blk[6] = {START_STOP, 0, 0, 0, 1, 0};
582 unsigned char startStop2Blk[6] = {START_STOP, 0, 0, 0, 2, 0};
583 unsigned char inqBuff[2];
584 unsigned char sense_buffer[32];
585
586 if ((ioctl(ctl->fd, SG_GET_VERSION_NUM, &k) < 0) || (k < 30000)) {
587 verbose(ctl, _("not an sg device, or old sg driver"));
588 return 0;
589 }
590
591 memset(&io_hdr, 0, sizeof(sg_io_hdr_t));
592 io_hdr.interface_id = 'S';
593 io_hdr.cmd_len = 6;
594 io_hdr.mx_sb_len = sizeof(sense_buffer);
595 io_hdr.dxfer_direction = SG_DXFER_NONE;
596 io_hdr.dxfer_len = 0;
597 io_hdr.dxferp = inqBuff;
598 io_hdr.sbp = sense_buffer;
599 io_hdr.timeout = 10000;
600
601 io_hdr.cmdp = allowRmBlk;
602 status = ioctl(ctl->fd, SG_IO, (void *)&io_hdr);
603 if (status < 0 || io_hdr.host_status || io_hdr.driver_status)
604 return 0;
605
606 io_hdr.cmdp = startStop1Blk;
607 status = ioctl(ctl->fd, SG_IO, (void *)&io_hdr);
608 if (status < 0 || io_hdr.host_status)
609 return 0;
610
611 /* Ignore errors when there is not medium -- in this case driver sense
612 * buffer sets MEDIUM NOT PRESENT (3a) bit. For more details see:
613 * http://www.tldp.org/HOWTO/archived/SCSI-Programming-HOWTO/SCSI-Programming-HOWTO-22.html#sec-sensecodes
614 * -- kzak Jun 2013
615 */
616 if (io_hdr.driver_status != 0 &&
617 !(io_hdr.driver_status == DRIVER_SENSE && io_hdr.sbp &&
618 io_hdr.sbp[12] == 0x3a))
619 return 0;
620
621 io_hdr.cmdp = startStop2Blk;
622 status = ioctl(ctl->fd, SG_IO, (void *)&io_hdr);
623 if (status < 0 || io_hdr.host_status || io_hdr.driver_status)
624 return 0;
625
626 /* force kernel to reread partition table when new disc inserted */
627 ioctl(ctl->fd, BLKRRPART);
628 return 1;
629 }
630
631 /*
632 * Eject using FDEJECT ioctl. Return 1 if successful, 0 otherwise.
633 */
634 static int eject_floppy(int fd)
635 {
636 return ioctl(fd, FDEJECT) >= 0;
637 }
638
639
640 /*
641 * Rewind and eject using tape ioctl. Return 1 if successful, 0 otherwise.
642 */
643 static int eject_tape(int fd)
644 {
645 struct mtop op = { .mt_op = MTOFFL, .mt_count = 0 };
646
647 return ioctl(fd, MTIOCTOP, &op) >= 0;
648 }
649
650
651 /* umount a device. */
652 static void umount_one(const struct eject_control *ctl, const char *name)
653 {
654 int status;
655
656 if (!name)
657 return;
658
659 verbose(ctl, _("%s: unmounting"), name);
660
661 switch (fork()) {
662 case 0: /* child */
663 if (setgid(getgid()) < 0)
664 err(EXIT_FAILURE, _("cannot set group id"));
665
666 if (setuid(getuid()) < 0)
667 err(EXIT_FAILURE, _("cannot set user id"));
668
669 if (ctl->p_option)
670 execl("/bin/umount", "/bin/umount", name, "-n", NULL);
671 else
672 execl("/bin/umount", "/bin/umount", name, NULL);
673
674 errx(EXIT_FAILURE, _("unable to exec /bin/umount of `%s'"), name);
675
676 case -1:
677 warn( _("unable to fork"));
678 break;
679
680 default: /* parent */
681 wait(&status);
682 if (WIFEXITED(status) == 0)
683 errx(EXIT_FAILURE,
684 _("unmount of `%s' did not exit normally"), name);
685
686 if (WEXITSTATUS(status) != 0)
687 errx(EXIT_FAILURE, _("unmount of `%s' failed\n"), name);
688 break;
689 }
690 }
691
692 /* Open a device file. */
693 static void open_device(struct eject_control *ctl)
694 {
695 ctl->fd = open(ctl->device, O_RDWR | O_NONBLOCK);
696 if (ctl->fd < 0)
697 ctl->fd = open(ctl->device, O_RDONLY | O_NONBLOCK);
698 if (ctl->fd == -1)
699 err(EXIT_FAILURE, _("cannot open %s"), ctl->device);
700 }
701
702 /*
703 * See if device has been mounted by looking in mount table. If so, set
704 * device name and mount point name, and return 1, otherwise return 0.
705 */
706 static int device_get_mountpoint(struct eject_control *ctl, char **devname, char **mnt)
707 {
708 struct libmnt_fs *fs;
709 int rc;
710
711 *mnt = NULL;
712
713 if (!ctl->mtab) {
714 struct libmnt_cache *cache;
715
716 ctl->mtab = mnt_new_table();
717 if (!ctl->mtab)
718 err(EXIT_FAILURE, _("failed to initialize libmount table"));
719
720 cache = mnt_new_cache();
721 mnt_table_set_cache(ctl->mtab, cache);
722 mnt_unref_cache(cache);
723
724 if (ctl->p_option)
725 rc = mnt_table_parse_file(ctl->mtab, _PATH_PROC_MOUNTINFO);
726 else
727 rc = mnt_table_parse_mtab(ctl->mtab, NULL);
728 if (rc)
729 err(EXIT_FAILURE, _("failed to parse mount table"));
730 }
731
732 fs = mnt_table_find_source(ctl->mtab, *devname, MNT_ITER_BACKWARD);
733 if (!fs) {
734 /* maybe 'devname' is mountpoint rather than a real device */
735 fs = mnt_table_find_target(ctl->mtab, *devname, MNT_ITER_BACKWARD);
736 if (fs) {
737 free(*devname);
738 *devname = xstrdup(mnt_fs_get_source(fs));
739 }
740 }
741
742 if (fs)
743 *mnt = xstrdup(mnt_fs_get_target(fs));
744 return *mnt ? 0 : -1;
745 }
746
747 static char *get_disk_devname(const char *device)
748 {
749 struct stat st;
750 dev_t diskno = 0;
751 char diskname[128];
752
753 if (stat(device, &st) != 0)
754 return NULL;
755
756 /* get whole-disk devno */
757 if (sysfs_devno_to_wholedisk(st.st_rdev, diskname,
758 sizeof(diskname), &diskno) != 0)
759 return NULL;
760
761 return st.st_rdev == diskno ? NULL : find_device(diskname);
762 }
763
764 static int umount_partitions(struct eject_control *ctl)
765 {
766 struct sysfs_cxt cxt = UL_SYSFSCXT_EMPTY;
767 dev_t devno;
768 DIR *dir = NULL;
769 struct dirent *d;
770 int count = 0;
771
772 devno = sysfs_devname_to_devno(ctl->device, NULL);
773 if (sysfs_init(&cxt, devno, NULL) != 0)
774 return 0;
775
776 /* open /sys/block/<wholedisk> */
777 if (!(dir = sysfs_opendir(&cxt, NULL)))
778 goto done;
779
780 /* scan for partition subdirs */
781 while ((d = readdir(dir))) {
782 if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
783 continue;
784
785 if (sysfs_is_partition_dirent(dir, d, ctl->device)) {
786 char *mnt = NULL;
787 char *dev = find_device(d->d_name);
788
789 if (dev && device_get_mountpoint(ctl, &dev, &mnt) == 0) {
790 verbose(ctl, _("%s: mounted on %s"), dev, mnt);
791 if (!ctl->M_option)
792 umount_one(ctl, mnt);
793 count++;
794 }
795 free(dev);
796 free(mnt);
797 }
798 }
799
800 done:
801 if (dir)
802 closedir(dir);
803 sysfs_deinit(&cxt);
804
805 return count;
806 }
807
808 static int is_hotpluggable(const struct eject_control *ctl)
809 {
810 struct sysfs_cxt cxt = UL_SYSFSCXT_EMPTY;
811 dev_t devno;
812 int rc = 0;
813
814 devno = sysfs_devname_to_devno(ctl->device, NULL);
815 if (sysfs_init(&cxt, devno, NULL) != 0)
816 return 0;
817
818 rc = sysfs_is_hotpluggable(&cxt);
819
820 sysfs_deinit(&cxt);
821 return rc;
822 }
823
824
825 /* handle -x option */
826 static void set_device_speed(struct eject_control *ctl)
827 {
828 if (!ctl->x_option)
829 return;
830
831 if (ctl->x_arg == 0)
832 verbose(ctl, _("setting CD-ROM speed to auto"));
833 else
834 verbose(ctl, _("setting CD-ROM speed to %ldX"), ctl->x_arg);
835
836 open_device(ctl);
837 select_speed(ctl);
838 exit(EXIT_SUCCESS);
839 }
840
841
842 /* main program */
843 int main(int argc, char **argv)
844 {
845 char *disk = NULL;
846 char *mountpoint = NULL;
847 int worked = 0; /* set to 1 when successfully ejected */
848 struct eject_control ctl = { NULL };
849
850 setlocale(LC_ALL,"");
851 bindtextdomain(PACKAGE, LOCALEDIR);
852 textdomain(PACKAGE);
853 atexit(close_stdout);
854
855 /* parse the command line arguments */
856 parse_args(&ctl, argc, argv);
857
858 /* handle -d option */
859 if (ctl.d_option) {
860 info(_("default device: `%s'"), EJECT_DEFAULT_DEVICE);
861 return EXIT_SUCCESS;
862 }
863
864 if (!ctl.device) {
865 ctl.device = mnt_resolve_path(EJECT_DEFAULT_DEVICE, NULL);
866 verbose(&ctl, _("using default device `%s'"), ctl.device);
867 } else {
868 char *p;
869
870 if (ctl.device[strlen(ctl.device) - 1] == '/')
871 ctl.device[strlen(ctl.device) - 1] = '\0';
872
873 /* figure out full device or mount point name */
874 p = find_device(ctl.device);
875 if (p)
876 free(ctl.device);
877 else
878 p = ctl.device;
879
880 ctl.device = mnt_resolve_spec(p, NULL);
881 free(p);
882 }
883
884 if (!ctl.device)
885 errx(EXIT_FAILURE, _("%s: unable to find device"), ctl.device);
886
887 verbose(&ctl, _("device name is `%s'"), ctl.device);
888
889 device_get_mountpoint(&ctl, &ctl.device, &mountpoint);
890 if (mountpoint)
891 verbose(&ctl, _("%s: mounted on %s"), ctl.device, mountpoint);
892 else
893 verbose(&ctl, _("%s: not mounted"), ctl.device);
894
895 disk = get_disk_devname(ctl.device);
896 if (disk) {
897 verbose(&ctl, _("%s: disc device: %s (disk device will be used for eject)"), ctl.device, disk);
898 free(ctl.device);
899 ctl.device = disk;
900 disk = NULL;
901 } else {
902 struct stat st;
903
904 if (stat(ctl.device, &st) != 0 || !S_ISBLK(st.st_mode))
905 errx(EXIT_FAILURE, _("%s: not found mountpoint or device "
906 "with the given name"), ctl.device);
907
908 verbose(&ctl, _("%s: is whole-disk device"), ctl.device);
909 }
910
911 if (ctl.F_option == 0 && is_hotpluggable(&ctl) == 0)
912 errx(EXIT_FAILURE, _("%s: is not hot-pluggable device"), ctl.device);
913
914 /* handle -n option */
915 if (ctl.n_option) {
916 info(_("device is `%s'"), ctl.device);
917 verbose(&ctl, _("exiting due to -n/--noop option"));
918 return EXIT_SUCCESS;
919 }
920
921 /* handle -i option */
922 if (ctl.i_option) {
923 open_device(&ctl);
924 manual_eject(&ctl);
925 return EXIT_SUCCESS;
926 }
927
928 /* handle -a option */
929 if (ctl.a_option) {
930 if (ctl.a_arg)
931 verbose(&ctl, _("%s: enabling auto-eject mode"), ctl.device);
932 else
933 verbose(&ctl, _("%s: disabling auto-eject mode"), ctl.device);
934 open_device(&ctl);
935 auto_eject(&ctl);
936 return EXIT_SUCCESS;
937 }
938
939 /* handle -t option */
940 if (ctl.t_option) {
941 verbose(&ctl, _("%s: closing tray"), ctl.device);
942 open_device(&ctl);
943 close_tray(ctl.fd);
944 set_device_speed(&ctl);
945 return EXIT_SUCCESS;
946 }
947
948 /* handle -T option */
949 if (ctl.T_option) {
950 verbose(&ctl, _("%s: toggling tray"), ctl.device);
951 open_device(&ctl);
952 toggle_tray(ctl.fd);
953 set_device_speed(&ctl);
954 return EXIT_SUCCESS;
955 }
956
957 /* handle -X option */
958 if (ctl.X_option) {
959 verbose(&ctl, _("%s: listing CD-ROM speed"), ctl.device);
960 open_device(&ctl);
961 list_speeds(&ctl);
962 return EXIT_SUCCESS;
963 }
964
965 /* handle -x option only */
966 if (!ctl.c_option)
967 set_device_speed(&ctl);
968
969
970 /*
971 * Unmount all partitions if -m is not specified; or umount given
972 * mountpoint if -M is specified, otherwise print error of another
973 * partition is mounted.
974 */
975 if (!ctl.m_option) {
976 int ct = umount_partitions(&ctl);
977
978 if (ct == 0 && mountpoint)
979 umount_one(&ctl, mountpoint); /* probably whole-device */
980
981 if (ctl.M_option) {
982 if (ct == 1 && mountpoint)
983 umount_one(&ctl, mountpoint);
984 else if (ct)
985 errx(EXIT_FAILURE, _("error: %s: device in use"), ctl.device);
986 }
987 }
988
989 /* handle -c option */
990 if (ctl.c_option) {
991 verbose(&ctl, _("%s: selecting CD-ROM disc #%ld"), ctl.device, ctl.c_arg);
992 open_device(&ctl);
993 changer_select(&ctl);
994 set_device_speed(&ctl);
995 return EXIT_SUCCESS;
996 }
997
998 /* if user did not specify type of eject, try all four methods */
999 if (ctl.r_option + ctl.s_option + ctl.f_option + ctl.q_option == 0)
1000 ctl.r_option = ctl.s_option = ctl.f_option = ctl.q_option = 1;
1001
1002 /* open device */
1003 open_device(&ctl);
1004
1005 /* try various methods of ejecting until it works */
1006 if (ctl.r_option) {
1007 verbose(&ctl, _("%s: trying to eject using CD-ROM eject command"), ctl.device);
1008 worked = eject_cdrom(ctl.fd);
1009 verbose(&ctl, worked ? _("CD-ROM eject command succeeded") :
1010 _("CD-ROM eject command failed"));
1011 }
1012
1013 if (ctl.s_option && !worked) {
1014 verbose(&ctl, _("%s: trying to eject using SCSI commands"), ctl.device);
1015 worked = eject_scsi(&ctl);
1016 verbose(&ctl, worked ? _("SCSI eject succeeded") :
1017 _("SCSI eject failed"));
1018 }
1019
1020 if (ctl.f_option && !worked) {
1021 verbose(&ctl, _("%s: trying to eject using floppy eject command"), ctl.device);
1022 worked = eject_floppy(ctl.fd);
1023 verbose(&ctl, worked ? _("floppy eject command succeeded") :
1024 _("floppy eject command failed"));
1025 }
1026
1027 if (ctl.q_option && !worked) {
1028 verbose(&ctl, _("%s: trying to eject using tape offline command"), ctl.device);
1029 worked = eject_tape(ctl.fd);
1030 verbose(&ctl, worked ? _("tape offline command succeeded") :
1031 _("tape offline command failed"));
1032 }
1033
1034 if (!worked)
1035 errx(EXIT_FAILURE, _("unable to eject"));
1036
1037 /* cleanup */
1038 close(ctl.fd);
1039 free(ctl.device);
1040 free(mountpoint);
1041
1042 mnt_unref_table(ctl.mtab);
1043
1044 return EXIT_SUCCESS;
1045 }