chiark / gitweb /
rules: do not preprocess 80-drivers.rules + 75-probe_mtd.rules
[elogind.git] / extras / cdrom_id / cdrom_id.c
index 3f9ca4c228391a4c7eabcaadcc632b5ba6dc3cb1..664a00d2c79ed19f329b5c2296132fceda1d09ce 100644 (file)
@@ -41,7 +41,7 @@
 #include "libudev.h"
 #include "libudev-private.h"
 
-static int debug;
+static bool debug;
 
 static void log_fn(struct udev *udev, int priority,
                   const char *file, int line, const char *fn,
@@ -85,6 +85,8 @@ static unsigned int cd_media_cd_rw = 0;
 static unsigned int cd_media_dvd_rom = 0;
 static unsigned int cd_media_dvd_r = 0;
 static unsigned int cd_media_dvd_rw = 0;
+static unsigned int cd_media_dvd_rw_ro = 0; /* restricted overwrite mode */
+static unsigned int cd_media_dvd_rw_seq = 0; /* sequential mode */
 static unsigned int cd_media_dvd_ram = 0;
 static unsigned int cd_media_dvd_plus_r = 0;
 static unsigned int cd_media_dvd_plus_rw = 0;
@@ -154,13 +156,11 @@ struct scsi_cmd {
        struct sg_io_hdr sg_io;
 };
 
-static void scsi_cmd_init(struct udev *udev, struct scsi_cmd *cmd, unsigned char *buf, size_t bufsize)
+static void scsi_cmd_init(struct udev *udev, struct scsi_cmd *cmd)
 {
        memset(cmd, 0x00, sizeof(struct scsi_cmd));
-       memset(buf, 0x00, bufsize);
        cmd->cgc.quiet = 1;
        cmd->cgc.sense = &cmd->_sense.s;
-       memset(&cmd->sg_io, 0, sizeof(cmd->sg_io));
        cmd->sg_io.interface_id = 'S';
        cmd->sg_io.mx_sb_len = sizeof(cmd->_sense);
        cmd->sg_io.cmdp = cmd->cgc.cmd;
@@ -180,9 +180,13 @@ static int scsi_cmd_run(struct udev *udev, struct scsi_cmd *cmd, int fd, unsigne
 {
        int ret = 0;
 
-       cmd->sg_io.dxferp = buf;
-       cmd->sg_io.dxfer_len = bufsize;
-       cmd->sg_io.dxfer_direction = SG_DXFER_FROM_DEV;
+       if (bufsize > 0) {
+               cmd->sg_io.dxferp = buf;
+               cmd->sg_io.dxfer_len = bufsize;
+               cmd->sg_io.dxfer_direction = SG_DXFER_FROM_DEV;
+       } else {
+               cmd->sg_io.dxfer_direction = SG_DXFER_NONE;
+       }
        if (ioctl(fd, SG_IO, &cmd->sg_io))
                return -1;
 
@@ -198,6 +202,39 @@ static int scsi_cmd_run(struct udev *udev, struct scsi_cmd *cmd, int fd, unsigne
        return ret;
 }
 
+static int media_lock(struct udev *udev, int fd, bool lock)
+{
+       int err;
+
+       /* disable the kernel's lock logic */
+       err = ioctl(fd, CDROM_CLEAR_OPTIONS, CDO_LOCK);
+       if (err < 0)
+               info(udev, "CDROM_CLEAR_OPTIONS, CDO_LOCK failed\n");
+
+       err = ioctl(fd, CDROM_LOCKDOOR, lock ? 1 : 0);
+       if (err < 0)
+               info(udev, "CDROM_LOCKDOOR failed\n");
+
+       return err;
+}
+
+static int media_eject(struct udev *udev, int fd)
+{
+       struct scsi_cmd sc;
+       int err;
+
+       scsi_cmd_init(udev, &sc);
+       scsi_cmd_set(udev, &sc, 0, 0x1b);
+       scsi_cmd_set(udev, &sc, 4, 0x02);
+       scsi_cmd_set(udev, &sc, 5, 0);
+       err = scsi_cmd_run(udev, &sc, fd, NULL, 0);
+       if ((err != 0)) {
+               info_scsi_cmd_err(udev, "START_STOP_UNIT", err);
+               return -1;
+       }
+       return 0;
+}
+
 static int cd_capability_compat(struct udev *udev, int fd)
 {
        int capability;
@@ -235,17 +272,18 @@ static int cd_media_compat(struct udev *udev, int fd)
        return 0;
 }
 
-static int cd_inquiry(struct udev *udev, int fd) {
+static int cd_inquiry(struct udev *udev, int fd)
+{
        struct scsi_cmd sc;
        unsigned char inq[128];
        int err;
 
-       scsi_cmd_init(udev, &sc, inq, sizeof(inq));
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x12);
        scsi_cmd_set(udev, &sc, 4, 36);
        scsi_cmd_set(udev, &sc, 5, 0);
        err = scsi_cmd_run(udev, &sc, fd, inq, 36);
-       if ((err < 0)) {
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "INQUIRY", err);
                return -1;
        }
@@ -259,6 +297,115 @@ static int cd_inquiry(struct udev *udev, int fd) {
        return 0;
 }
 
+static void feature_profile_media(struct udev *udev, int cur_profile)
+{
+       switch (cur_profile) {
+       case 0x03:
+       case 0x04:
+       case 0x05:
+               info(udev, "profile 0x%02x \n", cur_profile);
+               cd_media = 1;
+               cd_media_mo = 1;
+               break;
+       case 0x08:
+               info(udev, "profile 0x%02x media_cd_rom\n", cur_profile);
+               cd_media = 1;
+               cd_media_cd_rom = 1;
+               break;
+       case 0x09:
+               info(udev, "profile 0x%02x media_cd_r\n", cur_profile);
+               cd_media = 1;
+               cd_media_cd_r = 1;
+               break;
+       case 0x0a:
+               info(udev, "profile 0x%02x media_cd_rw\n", cur_profile);
+               cd_media = 1;
+               cd_media_cd_rw = 1;
+               break;
+       case 0x10:
+               info(udev, "profile 0x%02x media_dvd_ro\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_rom = 1;
+               break;
+       case 0x11:
+               info(udev, "profile 0x%02x media_dvd_r\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_r = 1;
+               break;
+       case 0x12:
+               info(udev, "profile 0x%02x media_dvd_ram\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_ram = 1;
+               break;
+       case 0x13:
+               info(udev, "profile 0x%02x media_dvd_rw_ro\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_rw = 1;
+               cd_media_dvd_rw_ro = 1;
+               break;
+       case 0x14:
+               info(udev, "profile 0x%02x media_dvd_rw_seq\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_rw = 1;
+               cd_media_dvd_rw_seq = 1;
+               break;
+       case 0x1B:
+               info(udev, "profile 0x%02x media_dvd_plus_r\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_plus_r = 1;
+               break;
+       case 0x1A:
+               info(udev, "profile 0x%02x media_dvd_plus_rw\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_plus_rw = 1;
+               break;
+       case 0x2A:
+               info(udev, "profile 0x%02x media_dvd_plus_rw_dl\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_plus_rw_dl = 1;
+               break;
+       case 0x2B:
+               info(udev, "profile 0x%02x media_dvd_plus_r_dl\n", cur_profile);
+               cd_media = 1;
+               cd_media_dvd_plus_r_dl = 1;
+               break;
+       case 0x40:
+               info(udev, "profile 0x%02x media_bd\n", cur_profile);
+               cd_media = 1;
+               cd_media_bd = 1;
+               break;
+       case 0x41:
+       case 0x42:
+               info(udev, "profile 0x%02x media_bd_r\n", cur_profile);
+               cd_media = 1;
+               cd_media_bd_r = 1;
+               break;
+       case 0x43:
+               info(udev, "profile 0x%02x media_bd_re\n", cur_profile);
+               cd_media = 1;
+               cd_media_bd_re = 1;
+               break;
+       case 0x50:
+               info(udev, "profile 0x%02x media_hddvd\n", cur_profile);
+               cd_media = 1;
+               cd_media_hddvd = 1;
+               break;
+       case 0x51:
+               info(udev, "profile 0x%02x media_hddvd_r\n", cur_profile);
+               cd_media = 1;
+               cd_media_hddvd_r = 1;
+               break;
+       case 0x52:
+               info(udev, "profile 0x%02x media_hddvd_rw\n", cur_profile);
+               cd_media = 1;
+               cd_media_hddvd_rw = 1;
+               break;
+       default:
+               info(udev, "profile 0x%02x <ignored>\n", cur_profile);
+               break;
+       }
+}
+
 static int feature_profiles(struct udev *udev, const unsigned char *profiles, size_t size)
 {
        unsigned int i;
@@ -274,6 +421,18 @@ static int feature_profiles(struct udev *udev, const unsigned char *profiles, si
                        info(udev, "profile 0x%02x mo\n", profile);
                        cd_mo = 1;
                        break;
+               case 0x08:
+                       info(udev, "profile 0x%02x cd_rom\n", profile);
+                       cd_cd_rom = 1;
+                       break;
+               case 0x09:
+                       info(udev, "profile 0x%02x cd_r\n", profile);
+                       cd_cd_r = 1;
+                       break;
+               case 0x0A:
+                       info(udev, "profile 0x%02x cd_rw\n", profile);
+                       cd_cd_rw = 1;
+                       break;
                case 0x10:
                        info(udev, "profile 0x%02x dvd_rom\n", profile);
                        cd_dvd_rom = 1;
@@ -336,6 +495,47 @@ static int feature_profiles(struct udev *udev, const unsigned char *profiles, si
        return 0;
 }
 
+/* returns 0 if media was detected */
+static int cd_profiles_old_mmc(struct udev *udev, int fd)
+{
+       struct scsi_cmd sc;
+       int err;
+
+       unsigned char header[32];
+
+       scsi_cmd_init(udev, &sc);
+       scsi_cmd_set(udev, &sc, 0, 0x51);
+       scsi_cmd_set(udev, &sc, 8, sizeof(header));
+       scsi_cmd_set(udev, &sc, 9, 0);
+       err = scsi_cmd_run(udev, &sc, fd, header, sizeof(header));
+       if ((err != 0)) {
+               info_scsi_cmd_err(udev, "READ DISC INFORMATION", err);
+               if (cd_media == 1) {
+                       info(udev, "no current profile, but disc is present; assuming CD-ROM\n");
+                       cd_media_cd_rom = 1;
+                       return 0;
+               } else {
+                       info(udev, "no current profile, assuming no media\n");
+                       return -1;
+               }
+       };
+
+       cd_media = 1;
+
+       if (header[2] & 16) {
+               cd_media_cd_rw = 1;
+               info(udev, "profile 0x0a media_cd_rw\n");
+       } else if ((header[2] & 3) < 2 && cd_cd_r) {
+               cd_media_cd_r = 1;
+               info(udev, "profile 0x09 media_cd_r\n");
+       } else {
+               cd_media_cd_rom = 1;
+               info(udev, "profile 0x08 media_cd_rom\n");
+       }
+       return 0;
+}
+
+/* returns 0 if media was detected */
 static int cd_profiles(struct udev *udev, int fd)
 {
        struct scsi_cmd sc;
@@ -344,20 +544,61 @@ static int cd_profiles(struct udev *udev, int fd)
        unsigned int len;
        unsigned int i;
        int err;
+       int ret;
+
+       ret = -1;
+
+       /* First query the current profile */
+       scsi_cmd_init(udev, &sc);
+       scsi_cmd_set(udev, &sc, 0, 0x46);
+       scsi_cmd_set(udev, &sc, 8, 8);
+       scsi_cmd_set(udev, &sc, 9, 0);
+       err = scsi_cmd_run(udev, &sc, fd, features, 8);
+       if ((err != 0)) {
+               info_scsi_cmd_err(udev, "GET CONFIGURATION", err);
+               /* handle pre-MMC2 drives which do not support GET CONFIGURATION */
+               if (SK(err) == 0x5 && ASC(err) == 0x20) {
+                       info(udev, "drive is pre-MMC2 and does not support 46h get configuration command\n");
+                       info(udev, "trying to work around the problem\n");
+                       ret = cd_profiles_old_mmc(udev, fd);
+               }
+               goto out;
+       }
 
-       scsi_cmd_init(udev, &sc, features, sizeof(features));
+       cur_profile = features[6] << 8 | features[7];
+       if (cur_profile > 0) {
+               info(udev, "current profile 0x%02x\n", cur_profile);
+               feature_profile_media (udev, cur_profile);
+               ret = 0; /* we have media */
+       } else {
+               info(udev, "no current profile, assuming no media\n");
+       }
+
+       len = features[0] << 24 | features[1] << 16 | features[2] << 8 | features[3];
+       info(udev, "GET CONFIGURATION: size of features buffer 0x%04x\n", len);
+
+       if (len > sizeof(features)) {
+               info(udev, "can not get features in a single query, truncating\n");
+               len = sizeof(features);
+       } else if (len <= 8) {
+               len = sizeof(features);
+       }
+
+       /* Now get the full feature buffer */
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x46);
-       scsi_cmd_set(udev, &sc, 7, (sizeof(features) >> 8) & 0xff);
-       scsi_cmd_set(udev, &sc, 8, sizeof(features) & 0xff);
+       scsi_cmd_set(udev, &sc, 7, ( len >> 8 ) & 0xff);
+       scsi_cmd_set(udev, &sc, 8, len & 0xff);
        scsi_cmd_set(udev, &sc, 9, 0);
-       err = scsi_cmd_run(udev, &sc, fd, features, sizeof(features));
-       if ((err < 0)) {
+       err = scsi_cmd_run(udev, &sc, fd, features, len);
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "GET CONFIGURATION", err);
                return -1;
        }
 
+       /* parse the length once more, in case the drive decided to have other features suddenly :) */
        len = features[0] << 24 | features[1] << 16 | features[2] << 8 | features[3];
-       info(udev, "GET CONFIGURATION: size of features buffer %i\n", len);
+       info(udev, "GET CONFIGURATION: size of features buffer 0x%04x\n", len);
 
        if (len > sizeof(features)) {
                info(udev, "can not get features in a single query, truncating\n");
@@ -374,87 +615,14 @@ static int cd_profiles(struct udev *udev, int fd)
                case 0x00:
                        info(udev, "GET CONFIGURATION: feature 'profiles', with %i entries\n", features[i+3] / 4);
                        feature_profiles(udev, &features[i]+4, features[i+3]);
-
-                       /* set current profile, if we got a profiles section */
-                       cur_profile = features[6] << 8 | features[7];
-                       info(udev, "current profile 0x%02x\n", cur_profile);
                        break;
                default:
-                       info(udev, "GET CONFIGURATION: feature %i <ignored>, with %i bytes\n", feature, features[i+3]);
+                       info(udev, "GET CONFIGURATION: feature 0x%04x <ignored>, with 0x%02x bytes\n", feature, features[i+3]);
                        break;
                }
        }
-
-       if (cur_profile == 0) {
-               info(udev, "no current profile, assuming no media\n");
-               return -1;
-       }
-
-       cd_media = 1;
-
-       switch (cur_profile) {
-       case 0x03:
-       case 0x04:
-       case 0x05:
-               cd_media_mo = 1;
-               break;
-       case 0x08:
-               cd_media_cd_rom = 1;
-               break;
-       case 0x09:
-               cd_media_cd_r = 1;
-               break;
-       case 0x0a:
-               cd_media_cd_rw = 1;
-               break;
-       case 0x10:
-               cd_media_dvd_rom = 1;
-               break;
-       case 0x11:
-               cd_media_dvd_r = 1;
-               break;
-       case 0x12:
-               cd_media_dvd_ram = 1;
-               break;
-       case 0x13:
-       case 0x14:
-               cd_media_dvd_rw = 1;
-               break;
-       case 0x1B:
-               cd_media_dvd_plus_r = 1;
-               break;
-       case 0x1A:
-               cd_media_dvd_plus_rw = 1;
-               break;
-       case 0x2A:
-               cd_media_dvd_plus_rw_dl = 1;
-               break;
-       case 0x2B:
-               cd_media_dvd_plus_r_dl = 1;
-               break;
-       case 0x40:
-               cd_media_bd = 1;
-               break;
-       case 0x41:
-       case 0x42:
-               cd_media_bd_r = 1;
-               break;
-       case 0x43:
-               cd_media_bd_re = 1;
-               break;
-       case 0x50:
-               cd_media_hddvd = 1;
-               break;
-       case 0x51:
-               cd_media_hddvd_r = 1;
-               break;
-       case 0x52:
-               cd_media_hddvd_rw = 1;
-               break;
-       default:
-               break;
-       }
-       return 0;
+out:
+       return ret;
 }
 
 static int cd_media_info(struct udev *udev, int fd)
@@ -469,23 +637,139 @@ static int cd_media_info(struct udev *udev, int fd)
        };
        int err;
 
-       scsi_cmd_init(udev, &sc, header, sizeof(header));
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x51);
-       scsi_cmd_set(udev, &sc, 8, sizeof(header));
+       scsi_cmd_set(udev, &sc, 8, sizeof(header) & 0xff);
        scsi_cmd_set(udev, &sc, 9, 0);
        err = scsi_cmd_run(udev, &sc, fd, header, sizeof(header));
-       if ((err < 0)) {
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "READ DISC INFORMATION", err);
                return -1;
        };
 
+       cd_media = 1;
        info(udev, "disk type %02x\n", header[8]);
+       info(udev, "hardware reported media status: %s\n", media_status[header[2] & 3]);
 
        /* exclude plain CDROM, some fake cdroms return 0 for "blank" media here */
-       if (!cd_media_cd_rom && (header[2] & 3) < 4)
+       if (!cd_media_cd_rom)
                cd_media_state = media_status[header[2] & 3];
 
-       if ((header[2] & 3) != 2)
+       /* fresh DVD-RW in restricted overwite mode reports itself as
+        * "appendable"; change it to "blank" to make it consistent with what
+        * gets reported after blanking, and what userspace expects  */
+       if (cd_media_dvd_rw_ro && (header[2] & 3) == 1)
+               cd_media_state = media_status[0];
+
+       /* DVD+RW discs (and DVD-RW in restricted mode) once formatted are
+        * always "complete", DVD-RAM are "other" or "complete" if the disc is
+        * write protected; we need to check the contents if it is blank */
+       if ((cd_media_dvd_rw_ro || cd_media_dvd_plus_rw || cd_media_dvd_plus_rw_dl || cd_media_dvd_ram) && (header[2] & 3) > 1) {
+               unsigned char buffer[32 * 2048];
+               unsigned char result, len;
+               int block, offset;
+
+               if (cd_media_dvd_ram) {
+                       /* a write protected dvd-ram may report "complete" status */
+
+                       unsigned char dvdstruct[8];
+                       unsigned char format[12];
+
+                       scsi_cmd_init(udev, &sc);
+                       scsi_cmd_set(udev, &sc, 0, 0xAD);
+                       scsi_cmd_set(udev, &sc, 7, 0xC0);
+                       scsi_cmd_set(udev, &sc, 9, sizeof(dvdstruct));
+                       scsi_cmd_set(udev, &sc, 11, 0);
+                       err = scsi_cmd_run(udev, &sc, fd, dvdstruct, sizeof(dvdstruct));
+                       if ((err != 0)) {
+                               info_scsi_cmd_err(udev, "READ DVD STRUCTURE", err);
+                               return -1;
+                       }
+                       if (dvdstruct[4] & 0x02) {
+                               cd_media_state = media_status[2];
+                               info(udev, "write-protected DVD-RAM media inserted\n");
+                               goto determined;
+                       }
+
+                       /* let's make sure we don't try to read unformatted media */
+                       scsi_cmd_init(udev, &sc);
+                       scsi_cmd_set(udev, &sc, 0, 0x23);
+                       scsi_cmd_set(udev, &sc, 8, sizeof(format));
+                       scsi_cmd_set(udev, &sc, 9, 0);
+                       err = scsi_cmd_run(udev, &sc, fd, format, sizeof(format));
+                       if ((err != 0)) {
+                               info_scsi_cmd_err(udev, "READ DVD FORMAT CAPACITIES", err);
+                               return -1;
+                       }
+
+                       len = format[3];
+                       if (len & 7 || len < 16) {
+                               info(udev, "invalid format capacities length\n");
+                               return -1;
+                       }
+
+                       switch(format[8] & 3) {
+                           case 1:
+                               info(udev, "unformatted DVD-RAM media inserted\n");
+                               /* This means that last format was interrupted
+                                * or failed, blank dvd-ram discs are factory
+                                * formatted. Take no action here as it takes
+                                * quite a while to reformat a dvd-ram and it's
+                                * not automatically started */
+                               goto determined;
+
+                           case 2:
+                               info(udev, "formatted DVD-RAM media inserted\n");
+                               break;
+
+                           case 3:
+                               cd_media = 0; //return no media
+                               info(udev, "format capacities returned no media\n");
+                               return -1;
+                       }
+               }
+
+               /* Take a closer look at formatted media (unformatted DVD+RW
+                * has "blank" status", DVD-RAM was examined earlier) and check
+                * for ISO and UDF PVDs or a fs superblock presence and do it
+                * in one ioctl (we need just sectors 0 and 16) */
+               scsi_cmd_init(udev, &sc);
+               scsi_cmd_set(udev, &sc, 0, 0x28);
+               scsi_cmd_set(udev, &sc, 5, 0);
+               scsi_cmd_set(udev, &sc, 8, 32);
+               scsi_cmd_set(udev, &sc, 9, 0);
+               err = scsi_cmd_run(udev, &sc, fd, buffer, sizeof(buffer));
+               if ((err != 0)) {
+                       cd_media = 0;
+                       info_scsi_cmd_err(udev, "READ FIRST 32 BLOCKS", err);
+                       return -1;
+               }
+
+               /* if any non-zero data is found in sector 16 (iso and udf) or
+                * eventually 0 (fat32 boot sector, ext2 superblock, etc), disc
+                * is assumed non-blank */
+               result = 0;
+
+               for (block = 32768; block >= 0 && !result; block -= 32768) {
+                       offset = block;
+                       while (offset < (block + 2048) && !result) {
+                               result = buffer [offset];
+                               offset++;
+                       }
+               }
+
+               if (!result) {
+                       cd_media_state = media_status[0];
+                       info(udev, "no data in blocks 0 or 16, assuming blank\n");
+               } else {
+                       info(udev, "data in blocks 0 or 16, assuming complete\n");
+               }
+       }
+
+determined:
+       /* "other" is e. g. DVD-RAM, can't append sessions there; DVDs in
+        * restricted overwrite mode can never append, only in sequential mode */
+       if ((header[2] & 3) < 2 && !cd_media_dvd_rw_ro)
                cd_media_session_next = header[10] << 8 | header[5];
        cd_media_session_count = header[9] << 8 | header[4];
        cd_media_track_count = header[11] << 8 | header[6];
@@ -497,46 +781,51 @@ static int cd_media_toc(struct udev *udev, int fd)
 {
        struct scsi_cmd sc;
        unsigned char header[12];
-       unsigned char toc[2048];
-       unsigned int len, i;
+       unsigned char toc[65536];
+       unsigned int len, i, num_tracks;
        unsigned char *p;
        int err;
 
-       scsi_cmd_init(udev, &sc, header, sizeof(header));
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x43);
        scsi_cmd_set(udev, &sc, 6, 1);
-       scsi_cmd_set(udev, &sc, 8, sizeof(header));
+       scsi_cmd_set(udev, &sc, 8, sizeof(header) & 0xff);
        scsi_cmd_set(udev, &sc, 9, 0);
        err = scsi_cmd_run(udev, &sc, fd, header, sizeof(header));
-       if ((err < 0)) {
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "READ TOC", err);
                return -1;
        }
 
        len = (header[0] << 8 | header[1]) + 2;
-       info(udev, "READ TOC: len: %d\n", len);
+       info(udev, "READ TOC: len: %d, start track: %d, end track: %d\n", len, header[2], header[3]);
        if (len > sizeof(toc))
                return -1;
        if (len < 2)
                return -1;
+       /* 2: first track, 3: last track */
+       num_tracks = header[3] - header[2] + 1;
 
        /* empty media has no tracks */
        if (len < 8)
                return 0;
 
-       scsi_cmd_init(udev, &sc, toc, sizeof(toc));
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x43);
        scsi_cmd_set(udev, &sc, 6, header[2]); /* First Track/Session Number */
-       scsi_cmd_set(udev, &sc, 7, len >> 8);
-       scsi_cmd_set(udev, &sc, 8, len);
+       scsi_cmd_set(udev, &sc, 7, (len >> 8) & 0xff);
+       scsi_cmd_set(udev, &sc, 8, len & 0xff);
        scsi_cmd_set(udev, &sc, 9, 0);
        err = scsi_cmd_run(udev, &sc, fd, toc, len);
-       if ((err < 0)) {
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "READ TOC (tracks)", err);
                return -1;
        }
 
-       for (p = toc+4, i = 4; i < len-8; i += 8, p += 8) {
+       /* Take care to not iterate beyond the last valid track as specified in
+        * the TOC, but also avoid going beyond the TOC length, just in case
+        * the last track number is invalidly large */
+       for (p = toc+4, i = 4; i < len-8 && num_tracks > 0; i += 8, p += 8, --num_tracks) {
                unsigned int block;
                unsigned int is_data_track;
 
@@ -552,13 +841,13 @@ static int cd_media_toc(struct udev *udev, int fd)
                        cd_media_track_count_audio++;
        }
 
-       scsi_cmd_init(udev, &sc, header, sizeof(header));
+       scsi_cmd_init(udev, &sc);
        scsi_cmd_set(udev, &sc, 0, 0x43);
        scsi_cmd_set(udev, &sc, 2, 1); /* Session Info */
-       scsi_cmd_set(udev, &sc, 8, 12);
+       scsi_cmd_set(udev, &sc, 8, sizeof(header));
        scsi_cmd_set(udev, &sc, 9, 0);
        err = scsi_cmd_run(udev, &sc, fd, header, sizeof(header));
-       if ((err < 0)) {
+       if ((err != 0)) {
                info_scsi_cmd_err(udev, "READ TOC (multi session)", err);
                return -1;
        }
@@ -572,13 +861,17 @@ int main(int argc, char *argv[])
 {
        struct udev *udev;
        static const struct option options[] = {
-               { "export", no_argument, NULL, 'x' },
+               { "lock-media", no_argument, NULL, 'l' },
+               { "unlock-media", no_argument, NULL, 'u' },
+               { "eject-media", no_argument, NULL, 'e' },
                { "debug", no_argument, NULL, 'd' },
                { "help", no_argument, NULL, 'h' },
                {}
        };
+       bool eject = false;
+       bool lock = false;
+       bool unlock = false;
        const char *node = NULL;
-       int export = 0;
        int fd = -1;
        int cnt;
        int rc = 0;
@@ -593,22 +886,30 @@ int main(int argc, char *argv[])
        while (1) {
                int option;
 
-               option = getopt_long(argc, argv, "dxh", options, NULL);
+               option = getopt_long(argc, argv, "deluh", options, NULL);
                if (option == -1)
                        break;
 
                switch (option) {
+               case 'l':
+                       lock = true;
+                       break;
+               case 'u':
+                       unlock = true;
+                       break;
+               case 'e':
+                       eject = true;
+                       break;
                case 'd':
-                       debug = 1;
+                       debug = true;
                        if (udev_get_log_priority(udev) < LOG_INFO)
                                udev_set_log_priority(udev, LOG_INFO);
                        break;
-               case 'x':
-                       export = 1;
-                       break;
                case 'h':
                        printf("Usage: cdrom_id [options] <device>\n"
-                              "  --export        export key/value pairs\n"
+                              "  --lock-media    lock the media (to enable eject request events)\n"
+                              "  --unlock-media  unlock the media\n"
+                              "  --eject-media   eject the media\n"
                               "  --debug         debug to stderr\n"
                               "  --help          print this help text\n\n");
                        goto exit;
@@ -657,21 +958,39 @@ int main(int argc, char *argv[])
 
        /* check if drive talks MMC */
        if (cd_inquiry(udev, fd) < 0)
-               goto print;
+               goto work;
 
        /* read drive and possibly current profile */
-       if (cd_profiles(udev, fd) < 0)
-               goto print;
+       if (cd_profiles(udev, fd) != 0)
+               goto work;
+
+       /* at this point we are guaranteed to have media in the drive - find out more about it */
 
        /* get session/track info */
-       if (cd_media_toc(udev, fd) < 0)
-               goto print;
+       cd_media_toc(udev, fd);
 
        /* get writable media state */
-       if (cd_media_info(udev, fd) < 0)
-               goto print;
+       cd_media_info(udev, fd);
+
+work:
+       /* lock the media, so we enable eject button events */
+       if (lock && cd_media) {
+               info(udev, "PREVENT_ALLOW_MEDIUM_REMOVAL (lock)\n");
+               media_lock(udev, fd, true);
+       }
+
+       if (unlock && cd_media) {
+               info(udev, "PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)\n");
+               media_lock(udev, fd, false);
+       }
+
+       if (eject) {
+               info(udev, "PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)\n");
+               media_lock(udev, fd, false);
+               info(udev, "START_STOP_UNIT (eject)\n");
+               media_eject(udev, fd);
+       }
 
-print:
        printf("ID_CDROM=1\n");
        if (cd_cd_rom)
                printf("ID_CDROM_CD=1\n");
@@ -763,14 +1082,14 @@ print:
                printf("ID_CDROM_MEDIA_SESSION_NEXT=%d\n", cd_media_session_next);
        if (cd_media_session_count > 0)
                printf("ID_CDROM_MEDIA_SESSION_COUNT=%d\n", cd_media_session_count);
+       if (cd_media_session_count > 1 && cd_media_session_last_offset > 0)
+               printf("ID_CDROM_MEDIA_SESSION_LAST_OFFSET=%llu\n", cd_media_session_last_offset);
        if (cd_media_track_count > 0)
                printf("ID_CDROM_MEDIA_TRACK_COUNT=%d\n", cd_media_track_count);
        if (cd_media_track_count_audio > 0)
                printf("ID_CDROM_MEDIA_TRACK_COUNT_AUDIO=%d\n", cd_media_track_count_audio);
        if (cd_media_track_count_data > 0)
                printf("ID_CDROM_MEDIA_TRACK_COUNT_DATA=%d\n", cd_media_track_count_data);
-       if (cd_media_session_last_offset > 0)
-               printf("ID_CDROM_MEDIA_SESSION_LAST_OFFSET=%llu\n", cd_media_session_last_offset);
 exit:
        if (fd >= 0)
                close(fd);
@@ -778,4 +1097,3 @@ exit:
        udev_log_close();
        return rc;
 }
-