* Copyright (c) 2003-2011, Haiku Inc.
* Distributed under the terms of the MIT license.
*
* Authors:
* François Revol <mmu_man@users.sf.net>
* Philippe Houdoin
*
* Description: ejects physical media from a drive.
* This version also loads a media and can query for the status.
*/
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <device/scsi.h>
#include <DiskDevice.h>
#include <DiskDeviceRoster.h>
#include <DiskDeviceVisitor.h>
#include <Drivers.h>
#include <fs_info.h>
#include <ObjectList.h>
#include <Path.h>
#include <String.h>
class RemovableDevice {
public:
RemovableDevice(BDiskDevice* device) {
fName = device->Name();
device->GetPath(&fPath);
};
inline const char* Name() { return fName.String(); }
inline const char* Path() { return fPath.Path(); }
private:
BString fName;
BPath fPath;
};
class RemovableDeviceVisitor : public BDiskDeviceVisitor {
public:
virtual bool Visit(BDiskDevice* device) {
if (device->IsRemovableMedia())
fRemovableDevices.AddItem(new RemovableDevice(device));
return false;
}
virtual bool Visit(BPartition* partition, int32 level) { return false; }
inline BObjectList<RemovableDevice>& RemovableDevices() { return fRemovableDevices; }
private:
BObjectList<RemovableDevice> fRemovableDevices;
};
static int usage(const char *prog)
{
printf("usage: eject [-q|-l|-s|-b|-u] /dev/disk/.../raw\n");
printf(" eject the device, or:\n");
printf(" -l: load it (close the tray)\n");
printf(" -q: query for media status\n");
printf(" -s: swap tray position (close/eject)\n");
printf(" -b: block (lock) tray position (prevent close/eject)\n");
printf(" -u: unblock (unlock) tray position (allow close/eject)\n");
return 0;
}
static int do_eject(char operation, char *device);
int main(int argc, char **argv)
{
char *device = NULL;
char operation = 'e';
int i;
int ret;
for (i = 1; i < argc; i++) {
if (strncmp(argv[i], "--h", 3) == 0) {
return usage("eject");
} else if (strncmp(argv[i], "-", 1) == 0) {
if (strlen(argv[i]) > 1)
operation = argv[i][1];
else {
usage("eject");
return 1;
}
} else {
device = argv[i];
ret = do_eject(operation, device);
if (ret != 0)
return ret;
}
}
if (device == NULL) {
BDiskDeviceRoster diskDeviceRoster;
RemovableDeviceVisitor visitor;
diskDeviceRoster.VisitEachDevice(&visitor);
int32 count = visitor.RemovableDevices().CountItems();
if (count < 1) {
printf("No removable device found!\n");
return 1;
}
if (count > 1) {
printf("Multiple removable devices available:\n");
for (i = 0; i < count; i++) {
RemovableDevice* item = visitor.RemovableDevices().ItemAt(i);
printf(" %s\t\"%s\"\n", item->Path(), item->Name());
}
return 1;
}
device = (char*)visitor.RemovableDevices().FirstItem()->Path();
return do_eject(operation, device);
}
return 0;
}
static int do_eject(char operation, char *device)
{
bool bval;
int fd;
status_t devstatus;
fs_info info;
if (fs_stat_dev(dev_for_path(device), &info) >= B_OK) {
if (strcmp(info.fsh_name, "devfs"))
device = info.device_name;
}
fd = open(device, O_RDONLY);
if (fd < 0) {
perror(device);
return 1;
}
switch (operation) {
case 'e':
if (ioctl(fd, B_EJECT_DEVICE, NULL, 0) < 0) {
perror(device);
close(fd);
return 1;
}
break;
case 'l':
if (ioctl(fd, B_LOAD_MEDIA, NULL, 0) < 0) {
perror(device);
close(fd);
return 1;
}
break;
case 'b':
bval = true;
if (ioctl(fd, B_SCSI_PREVENT_ALLOW, &bval, sizeof(bval)) < 0) {
perror(device);
close(fd);
return 1;
}
break;
case 'u':
bval = false;
if (ioctl(fd, B_SCSI_PREVENT_ALLOW, &bval, sizeof(bval)) < 0) {
perror(device);
close(fd);
return 1;
}
break;
case 'q':
if (ioctl(fd, B_GET_MEDIA_STATUS, &devstatus, sizeof(devstatus))
< 0) {
perror(device);
close(fd);
return 1;
}
switch (devstatus) {
case B_NO_ERROR:
puts("Media present");
break;
default:
puts(strerror(devstatus));
}
break;
case 's':
if (ioctl(fd, B_GET_MEDIA_STATUS, &devstatus, sizeof(devstatus))
< 0) {
perror(device);
close(fd);
return 1;
}
switch (devstatus) {
case B_NO_ERROR:
case B_DEV_NO_MEDIA:
if (ioctl(fd, B_EJECT_DEVICE, NULL, 0) < 0) {
perror(device);
close(fd);
return 1;
}
break;
case B_DEV_DOOR_OPEN:
if (ioctl(fd, B_LOAD_MEDIA, NULL, 0) < 0) {
perror(device);
close(fd);
return 1;
}
break;
default:
perror(device);
}
break;
case 'h':
default:
close(fd);
return usage("eject");
}
close(fd);
return 0;
}
