| /* SPDX-License-Identifier: GPL-2.0-or-later */ |
| /* |
| * cdrom_id - optical drive and media information prober |
| */ |
| |
| #include <fcntl.h> |
| #include <getopt.h> |
| #include <linux/cdrom.h> |
| #include <scsi/sg.h> |
| #include <sys/ioctl.h> |
| #include <unistd.h> |
| |
| #include "fd-util.h" |
| #include "main-func.h" |
| #include "memory-util.h" |
| #include "random-util.h" |
| #include "sort-util.h" |
| #include "string-table.h" |
| #include "string-util.h" |
| #include "udev-util.h" |
| #include "unaligned.h" |
| |
| static bool arg_eject = false; |
| static bool arg_lock = false; |
| static bool arg_unlock = false; |
| static const char *arg_node = NULL; |
| |
| typedef enum Feature { |
| FEATURE_RW_NONREMOVABLE = 0x01, |
| FEATURE_RW_REMOVABLE = 0x02, |
| |
| FEATURE_MO_SE = 0x03, /* sector erase */ |
| FEATURE_MO_WO = 0x04, /* write once */ |
| FEATURE_MO_AS = 0x05, /* advance storage */ |
| |
| FEATURE_CD_ROM = 0x08, |
| FEATURE_CD_R = 0x09, |
| FEATURE_CD_RW = 0x0a, |
| |
| FEATURE_DVD_ROM = 0x10, |
| FEATURE_DVD_R = 0x11, |
| FEATURE_DVD_RAM = 0x12, |
| FEATURE_DVD_RW_RO = 0x13, /* restricted overwrite mode */ |
| FEATURE_DVD_RW_SEQ = 0x14, /* sequential mode */ |
| FEATURE_DVD_R_DL_SEQ = 0x15, /* sequential recording */ |
| FEATURE_DVD_R_DL_JR = 0x16, /* jump recording */ |
| FEATURE_DVD_RW_DL = 0x17, |
| FEATURE_DVD_R_DDR = 0x18, /* download disc recording - dvd for css managed recording */ |
| FEATURE_DVD_PLUS_RW = 0x1a, |
| FEATURE_DVD_PLUS_R = 0x1b, |
| |
| FEATURE_DDCD_ROM = 0x20, |
| FEATURE_DDCD_R = 0x21, |
| FEATURE_DDCD_RW = 0x22, |
| |
| FEATURE_DVD_PLUS_RW_DL = 0x2a, |
| FEATURE_DVD_PLUS_R_DL = 0x2b, |
| |
| FEATURE_BD = 0x40, |
| FEATURE_BD_R_SRM = 0x41, /* sequential recording mode */ |
| FEATURE_BD_R_RRM = 0x42, /* random recording mode */ |
| FEATURE_BD_RE = 0x43, |
| |
| FEATURE_HDDVD = 0x50, |
| FEATURE_HDDVD_R = 0x51, |
| FEATURE_HDDVD_RAM = 0x52, |
| FEATURE_HDDVD_RW = 0x53, |
| FEATURE_HDDVD_R_DL = 0x58, |
| FEATURE_HDDVD_RW_DL = 0x5a, |
| |
| FEATURE_MRW, |
| FEATURE_MRW_W, |
| |
| _FEATURE_MAX, |
| _FEATURE_INVALID = -EINVAL, |
| } Feature; |
| |
| typedef enum MediaState { |
| MEDIA_STATE_BLANK = 0, |
| MEDIA_STATE_APPENDABLE = 1, |
| MEDIA_STATE_COMPLETE = 2, |
| MEDIA_STATE_OTHER = 3, |
| _MEDIA_STATE_MAX, |
| _MEDIA_STATE_INVALID = -EINVAL, |
| } MediaState; |
| |
| typedef struct Context { |
| int fd; |
| |
| Feature *drive_features; |
| size_t n_drive_feature; |
| |
| Feature media_feature; |
| bool has_media; |
| |
| MediaState media_state; |
| unsigned media_session_next; |
| unsigned media_session_count; |
| unsigned media_track_count; |
| unsigned media_track_count_data; |
| unsigned media_track_count_audio; |
| uint64_t media_session_last_offset; |
| } Context; |
| |
| #define CONTEXT_EMPTY { \ |
| .fd = -EBADF, \ |
| .media_feature = _FEATURE_INVALID, \ |
| .media_state = _MEDIA_STATE_INVALID, \ |
| } |
| |
| static void context_clear(Context *c) { |
| if (!c) |
| return; |
| |
| safe_close(c->fd); |
| free(c->drive_features); |
| } |
| |
| static bool drive_has_feature(const Context *c, Feature f) { |
| assert(c); |
| |
| for (size_t i = 0; i < c->n_drive_feature; i++) |
| if (c->drive_features[i] == f) |
| return true; |
| |
| return false; |
| } |
| |
| static int set_drive_feature(Context *c, Feature f) { |
| assert(c); |
| |
| if (drive_has_feature(c, f)) |
| return 0; |
| |
| if (!GREEDY_REALLOC(c->drive_features, c->n_drive_feature + 1)) |
| return -ENOMEM; |
| |
| c->drive_features[c->n_drive_feature++] = f; |
| return 1; |
| } |
| |
| #define ERRCODE(s) ((((s)[2] & 0x0F) << 16) | ((s)[12] << 8) | ((s)[13])) |
| #define SK(errcode) (((errcode) >> 16) & 0xFU) |
| #define ASC(errcode) (((errcode) >> 8) & 0xFFU) |
| #define ASCQ(errcode) ((errcode) & 0xFFU) |
| #define CHECK_CONDITION 0x01 |
| |
| static int log_scsi_debug_errno(int error, const char *msg) { |
| assert(error != 0); |
| |
| /* error < 0 means errno-style error, error > 0 means SCSI error */ |
| |
| if (error < 0) |
| return log_debug_errno(error, "Failed to %s: %m", msg); |
| |
| return log_debug_errno(SYNTHETIC_ERRNO(EIO), |
| "Failed to %s with SK=%X/ASC=%02X/ACQ=%02X", |
| msg, SK(error), ASC(error), ASCQ(error)); |
| } |
| |
| struct scsi_cmd { |
| struct cdrom_generic_command cgc; |
| union { |
| struct request_sense s; |
| unsigned char u[18]; |
| } _sense; |
| struct sg_io_hdr sg_io; |
| }; |
| |
| static void scsi_cmd_init(struct scsi_cmd *cmd) { |
| memzero(cmd, sizeof(struct scsi_cmd)); |
| cmd->cgc.quiet = 1; |
| cmd->cgc.sense = &cmd->_sense.s; |
| cmd->sg_io.interface_id = 'S'; |
| cmd->sg_io.mx_sb_len = sizeof(cmd->_sense); |
| cmd->sg_io.cmdp = cmd->cgc.cmd; |
| cmd->sg_io.sbp = cmd->_sense.u; |
| cmd->sg_io.flags = SG_FLAG_LUN_INHIBIT | SG_FLAG_DIRECT_IO; |
| } |
| |
| static void scsi_cmd_set(struct scsi_cmd *cmd, size_t i, unsigned char arg) { |
| cmd->sg_io.cmd_len = i + 1; |
| cmd->cgc.cmd[i] = arg; |
| } |
| |
| static int scsi_cmd_run(struct scsi_cmd *cmd, int fd, unsigned char *buf, size_t bufsize) { |
| int r; |
| |
| assert(cmd); |
| assert(fd >= 0); |
| assert(buf || bufsize == 0); |
| |
| /* Return 0 on success. On failure, return negative errno or positive error code. */ |
| |
| 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) < 0) |
| return -errno; |
| |
| if ((cmd->sg_io.info & SG_INFO_OK_MASK) != SG_INFO_OK) { |
| if (cmd->sg_io.masked_status & CHECK_CONDITION) { |
| r = ERRCODE(cmd->_sense.u); |
| if (r != 0) |
| return r; |
| } |
| return -EIO; |
| } |
| |
| return 0; |
| } |
| |
| static int scsi_cmd_run_and_log(struct scsi_cmd *cmd, int fd, unsigned char *buf, size_t bufsize, const char *msg) { |
| int r; |
| |
| assert(msg); |
| |
| r = scsi_cmd_run(cmd, fd, buf, bufsize); |
| if (r != 0) |
| return log_scsi_debug_errno(r, msg); |
| |
| return 0; |
| } |
| |
| static int media_lock(int fd, bool lock) { |
| /* disable the kernel's lock logic */ |
| if (ioctl(fd, CDROM_CLEAR_OPTIONS, CDO_LOCK) < 0) |
| log_debug_errno(errno, "Failed to issue ioctl(CDROM_CLEAR_OPTIONS, CDO_LOCK), ignoring: %m"); |
| |
| if (ioctl(fd, CDROM_LOCKDOOR, lock ? 1 : 0) < 0) |
| return log_debug_errno(errno, "Failed to issue ioctl(CDROM_LOCKDOOR): %m"); |
| |
| return 0; |
| } |
| |
| static int media_eject(int fd) { |
| struct scsi_cmd sc; |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_START_STOP_UNIT); |
| scsi_cmd_set(&sc, 4, 0x02); |
| scsi_cmd_set(&sc, 5, 0); |
| |
| return scsi_cmd_run_and_log(&sc, fd, NULL, 0, "start/stop unit"); |
| } |
| |
| static int cd_capability_compat(Context *c) { |
| int capability, r; |
| |
| assert(c); |
| |
| capability = ioctl(c->fd, CDROM_GET_CAPABILITY, NULL); |
| if (capability < 0) |
| return log_debug_errno(errno, "CDROM_GET_CAPABILITY failed"); |
| |
| if (capability & CDC_CD_R) { |
| r = set_drive_feature(c, FEATURE_CD_R); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_CD_RW) { |
| r = set_drive_feature(c, FEATURE_CD_RW); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_DVD) { |
| r = set_drive_feature(c, FEATURE_DVD_ROM); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_DVD_R) { |
| r = set_drive_feature(c, FEATURE_DVD_R); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_DVD_RAM) { |
| r = set_drive_feature(c, FEATURE_DVD_RAM); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_MRW) { |
| r = set_drive_feature(c, FEATURE_MRW); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| if (capability & CDC_MRW_W) { |
| r = set_drive_feature(c, FEATURE_MRW_W); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| |
| return 0; |
| } |
| |
| static int cd_media_compat(Context *c) { |
| int r; |
| |
| assert(c); |
| |
| r = ioctl(c->fd, CDROM_DRIVE_STATUS, CDSL_CURRENT); |
| if (r < 0) |
| return log_debug_errno(errno, "ioctl(CDROM_DRIVE_STATUS) failed: %m"); |
| if (r != CDS_DISC_OK) |
| return log_debug_errno(SYNTHETIC_ERRNO(ENOMEDIUM), |
| "ioctl(CDROM_DRIVE_STATUS) → %d (%s), ignoring.", |
| r, |
| r == CDS_NO_INFO ? "no info" : |
| r == CDS_NO_DISC ? "no disc" : |
| r == CDS_TRAY_OPEN ? "tray open" : |
| r == CDS_DRIVE_NOT_READY ? "drive not ready" : |
| "unknown status"); |
| |
| c->has_media = true; |
| return 0; |
| } |
| |
| static int cd_inquiry(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char inq[36]; |
| int r; |
| |
| assert(c); |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_INQUIRY); |
| scsi_cmd_set(&sc, 4, sizeof(inq)); |
| scsi_cmd_set(&sc, 5, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, inq, sizeof(inq), "inquire"); |
| if (r < 0) |
| return r; |
| |
| if ((inq[0] & 0x1F) != 5) |
| return log_debug_errno(SYNTHETIC_ERRNO(EINVAL), "Not an MMC unit."); |
| |
| log_debug("INQUIRY: [%.8s][%.16s][%.4s]", inq + 8, inq + 16, inq + 32); |
| return 0; |
| } |
| |
| static int feature_profiles(Context *c, const unsigned char *profiles, size_t size) { |
| int r; |
| |
| assert(c); |
| |
| for (size_t i = 0; i + 4 <= size; i += 4) { |
| r = set_drive_feature(c, (Feature) unaligned_read_be16(&profiles[i])); |
| if (r < 0) |
| return log_oom_debug(); |
| } |
| |
| return 1; |
| } |
| |
| static int cd_profiles_old_mmc(Context *c) { |
| disc_information discinfo; |
| struct scsi_cmd sc; |
| size_t len; |
| int r; |
| |
| assert(c); |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_DISC_INFO); |
| scsi_cmd_set(&sc, 8, sizeof(discinfo.disc_information_length)); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, (unsigned char *)&discinfo.disc_information_length, sizeof(discinfo.disc_information_length), "read disc information"); |
| if (r >= 0) { |
| /* Not all drives have the same disc_info length, so requeue |
| * packet with the length the drive tells us it can supply */ |
| len = be16toh(discinfo.disc_information_length) + sizeof(discinfo.disc_information_length); |
| if (len > sizeof(discinfo)) |
| len = sizeof(discinfo); |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_DISC_INFO); |
| scsi_cmd_set(&sc, 8, len); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, (unsigned char *)&discinfo, len, "read disc information"); |
| } |
| if (r < 0) { |
| if (c->has_media) { |
| log_debug("No current profile, but disc is present; assuming CD-ROM."); |
| c->media_feature = FEATURE_CD_ROM; |
| c->media_track_count = 1; |
| c->media_track_count_data = 1; |
| return 1; |
| } else |
| return log_debug_errno(SYNTHETIC_ERRNO(ENOMEDIUM), |
| "no current profile, assuming no media."); |
| }; |
| |
| c->has_media = true; |
| |
| if (discinfo.erasable) |
| c->media_feature = FEATURE_CD_RW; |
| else if (discinfo.disc_status < 2 && drive_has_feature(c, FEATURE_CD_R)) |
| c->media_feature = FEATURE_CD_R; |
| else |
| c->media_feature = FEATURE_CD_ROM; |
| |
| return 0; |
| } |
| |
| static int cd_profiles(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char features[65530]; |
| unsigned cur_profile; |
| size_t len; |
| int r; |
| |
| assert(c); |
| |
| /* First query the current profile */ |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_GET_CONFIGURATION); |
| scsi_cmd_set(&sc, 8, 8); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run(&sc, c->fd, features, 8); |
| if (r != 0) { |
| /* handle pre-MMC2 drives which do not support GET CONFIGURATION */ |
| if (r > 0 && SK(r) == 0x5 && IN_SET(ASC(r), 0x20, 0x24)) { |
| log_debug("Drive is pre-MMC2 and does not support 46h get configuration command; " |
| "trying to work around the problem."); |
| return cd_profiles_old_mmc(c); |
| } |
| |
| return log_scsi_debug_errno(r, "get configuration"); |
| } |
| |
| cur_profile = unaligned_read_be16(&features[6]); |
| if (cur_profile > 0) { |
| log_debug("current profile 0x%02x", cur_profile); |
| c->media_feature = (Feature) cur_profile; |
| c->has_media = true; |
| } else { |
| log_debug("no current profile, assuming no media"); |
| c->has_media = false; |
| } |
| |
| len = unaligned_read_be32(features); |
| log_debug("GET CONFIGURATION: size of features buffer %zu", len); |
| |
| if (len > sizeof(features)) { |
| log_debug("Cannot get features in a single query, truncating."); |
| len = sizeof(features); |
| } else if (len <= 8) |
| len = sizeof(features); |
| |
| /* Now get the full feature buffer */ |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_GET_CONFIGURATION); |
| scsi_cmd_set(&sc, 7, (len >> 8) & 0xff); |
| scsi_cmd_set(&sc, 8, len & 0xff); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, features, len, "get configuration"); |
| if (r < 0) |
| return r; |
| |
| /* parse the length once more, in case the drive decided to have other features suddenly :) */ |
| len = unaligned_read_be32(features); |
| log_debug("GET CONFIGURATION: size of features buffer %zu", len); |
| |
| if (len > sizeof(features)) { |
| log_debug("Cannot get features in a single query, truncating."); |
| len = sizeof(features); |
| } |
| |
| /* device features */ |
| for (size_t i = 8; i + 4 < len; i += 4 + features[i + 3]) { |
| unsigned feature; |
| |
| feature = unaligned_read_be16(&features[i]); |
| |
| switch (feature) { |
| case 0x00: |
| log_debug("GET CONFIGURATION: feature 'profiles', with %u entries", features[i + 3] / 4); |
| feature_profiles(c, features + i + 4, MIN(features[i + 3], len - i - 4)); |
| break; |
| default: |
| log_debug("GET CONFIGURATION: feature 0x%04x <ignored>, with 0x%02x bytes", feature, features[i + 3]); |
| break; |
| } |
| } |
| |
| return c->has_media; |
| } |
| |
| static const char * const media_state_table[_MEDIA_STATE_MAX] = { |
| [MEDIA_STATE_BLANK] = "blank", |
| [MEDIA_STATE_APPENDABLE] = "appendable", |
| [MEDIA_STATE_COMPLETE] = "complete", |
| [MEDIA_STATE_OTHER] = "other", |
| }; |
| |
| DEFINE_PRIVATE_STRING_TABLE_LOOKUP_TO_STRING(media_state, MediaState); |
| |
| static int dvd_ram_media_update_state(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char dvdstruct[8]; |
| unsigned char format[12]; |
| unsigned char len; |
| int r; |
| |
| assert(c); |
| |
| /* Return 1 if media state is determined. */ |
| |
| if (c->media_feature != FEATURE_DVD_RAM) |
| return 0; |
| |
| /* a write protected dvd-ram may report "complete" status */ |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_DVD_STRUCTURE); |
| scsi_cmd_set(&sc, 7, 0xC0); |
| scsi_cmd_set(&sc, 9, sizeof(dvdstruct)); |
| scsi_cmd_set(&sc, 11, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, dvdstruct, sizeof(dvdstruct), "read DVD structure"); |
| if (r < 0) |
| return r; |
| |
| if (dvdstruct[4] & 0x02) { |
| c->media_state = MEDIA_STATE_COMPLETE; |
| log_debug("Write-protected DVD-RAM media inserted"); |
| return 1; |
| } |
| |
| /* let's make sure we don't try to read unformatted media */ |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_FORMAT_CAPACITIES); |
| scsi_cmd_set(&sc, 8, sizeof(format)); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, format, sizeof(format), "read DVD format capacities"); |
| if (r < 0) |
| return r; |
| |
| len = format[3]; |
| if (len & 7 || len < 16) |
| return log_debug_errno(SYNTHETIC_ERRNO(EINVAL), |
| "Invalid format capacities length."); |
| |
| switch (format[8] & 3) { |
| case 1: |
| /* 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. */ |
| log_debug("Unformatted DVD-RAM media inserted."); |
| return 1; |
| |
| case 2: |
| log_debug("Formatted DVD-RAM media inserted."); |
| return 0; |
| |
| case 3: |
| c->has_media = false; |
| return log_debug_errno(SYNTHETIC_ERRNO(ENOMEDIUM), |
| "Format capacities returned no media."); |
| } |
| |
| return 0; |
| } |
| |
| static int dvd_media_update_state(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char buffer[32 * 2048]; |
| int r; |
| |
| r = dvd_ram_media_update_state(c); |
| if (r != 0) |
| return r; |
| |
| /* 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(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_10); |
| scsi_cmd_set(&sc, 5, 0); |
| scsi_cmd_set(&sc, 8, sizeof(buffer)/2048); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, buffer, sizeof(buffer), "read first 32 blocks"); |
| if (r < 0) { |
| c->has_media = false; |
| return r; |
| } |
| |
| /* 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 */ |
| |
| for (size_t offset = 32768; offset < 32768 + 2048; offset++) |
| if (buffer[offset] != 0) { |
| log_debug("Data in block 16, assuming complete."); |
| return 0; |
| } |
| |
| for (size_t offset = 0; offset < 2048; offset++) |
| if (buffer[offset] != 0) { |
| log_debug("Data in block 0, assuming complete."); |
| return 0; |
| } |
| |
| log_debug("No data in blocks 0 or 16, assuming blank."); |
| c->media_state = MEDIA_STATE_BLANK; |
| return 0; |
| } |
| |
| static int cd_media_info(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char header[32]; |
| MediaState state; |
| int r; |
| |
| assert(c); |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_DISC_INFO); |
| scsi_cmd_set(&sc, 8, sizeof(header)); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, header, sizeof(header), "read disc information"); |
| if (r < 0) |
| return r; |
| |
| c->has_media = true; |
| log_debug("disk type %02x", header[8]); |
| |
| state = (MediaState) (header[2] & 0x03); |
| log_debug("hardware reported media status: %s", strna(media_state_to_string(state))); |
| |
| /* exclude plain CDROM, some fake cdroms return 0 for "blank" media here */ |
| if (c->media_feature != FEATURE_CD_ROM) |
| c->media_state = state; |
| |
| /* fresh DVD-RW in restricted overwrite mode reports itself as |
| * "appendable"; change it to "blank" to make it consistent with what |
| * gets reported after blanking, and what userspace expects. */ |
| if (c->media_feature == FEATURE_DVD_RW_RO && state == MEDIA_STATE_APPENDABLE) |
| c->media_state = MEDIA_STATE_BLANK; |
| |
| /* 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 (IN_SET(c->media_feature, FEATURE_DVD_RW_RO, FEATURE_DVD_PLUS_RW, FEATURE_DVD_PLUS_RW_DL, FEATURE_DVD_RAM) && |
| IN_SET(state, MEDIA_STATE_COMPLETE, MEDIA_STATE_OTHER)) { |
| r = dvd_media_update_state(c); |
| if (r < 0) |
| return r; |
| } |
| |
| /* "other" is e. g. DVD-RAM, can't append sessions there; DVDs in |
| * restricted overwrite mode can never append, only in sequential mode */ |
| if (c->media_feature != FEATURE_DVD_RW_RO && IN_SET(state, MEDIA_STATE_BLANK, MEDIA_STATE_APPENDABLE)) |
| c->media_session_next = header[10] << 8 | header[5]; |
| c->media_session_count = header[9] << 8 | header[4]; |
| c->media_track_count = header[11] << 8 | header[6]; |
| |
| return 0; |
| } |
| |
| static int cd_media_toc(Context *c) { |
| struct scsi_cmd sc; |
| unsigned char header[12]; |
| unsigned char toc[65536]; |
| unsigned num_tracks; |
| size_t len; |
| int r; |
| |
| assert(c); |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_TOC_PMA_ATIP); |
| scsi_cmd_set(&sc, 6, 1); |
| scsi_cmd_set(&sc, 8, sizeof(header)); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, header, sizeof(header), "read TOC"); |
| if (r < 0) |
| return r; |
| |
| len = unaligned_read_be16(header) + 2; |
| log_debug("READ TOC: len: %zu, start track: %u, end track: %u", len, header[2], header[3]); |
| |
| if (len > sizeof(toc)) |
| return -1; |
| /* empty media has no tracks */ |
| if (len < 8) |
| return 0; |
| |
| /* 2: first track, 3: last track */ |
| num_tracks = header[3] - header[2] + 1; |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_TOC_PMA_ATIP); |
| scsi_cmd_set(&sc, 6, header[2]); /* First Track/Session Number */ |
| scsi_cmd_set(&sc, 7, (len >> 8) & 0xff); |
| scsi_cmd_set(&sc, 8, len & 0xff); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, toc, len, "read TOC (tracks)"); |
| if (r < 0) |
| return r; |
| |
| /* 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 (size_t i = 4; i + 8 <= len && num_tracks > 0; i += 8, --num_tracks) { |
| bool is_data_track; |
| uint32_t block; |
| |
| is_data_track = (toc[i + 1] & 0x04) != 0; |
| block = unaligned_read_be32(&toc[i + 4]); |
| |
| log_debug("track=%u info=0x%x(%s) start_block=%"PRIu32, |
| toc[i + 2], toc[i + 1] & 0x0FU, is_data_track ? "data":"audio", block); |
| |
| if (is_data_track) |
| c->media_track_count_data++; |
| else |
| c->media_track_count_audio++; |
| } |
| |
| scsi_cmd_init(&sc); |
| scsi_cmd_set(&sc, 0, GPCMD_READ_TOC_PMA_ATIP); |
| scsi_cmd_set(&sc, 2, 1); /* Session Info */ |
| scsi_cmd_set(&sc, 8, sizeof(header)); |
| scsi_cmd_set(&sc, 9, 0); |
| r = scsi_cmd_run_and_log(&sc, c->fd, header, sizeof(header), "read TOC (multi session)"); |
| if (r < 0) |
| return r; |
| |
| len = unaligned_read_be32(&header[8]); |
| log_debug("last track %u starts at block %zu", header[4+2], len); |
| c->media_session_last_offset = (uint64_t) len * 2048; |
| |
| return 0; |
| } |
| |
| static int open_drive(Context *c) { |
| int fd; |
| |
| assert(c); |
| assert(c->fd < 0); |
| |
| for (int cnt = 0;; cnt++) { |
| fd = open(arg_node, O_RDONLY|O_NONBLOCK|O_CLOEXEC|O_NOCTTY); |
| if (fd >= 0) |
| break; |
| if (++cnt >= 20 || errno != EBUSY) |
| return log_debug_errno(errno, "Unable to open '%s': %m", arg_node); |
| |
| (void) usleep(100 * USEC_PER_MSEC + random_u64_range(100 * USEC_PER_MSEC)); |
| } |
| |
| log_debug("probing: '%s'", arg_node); |
| c->fd = fd; |
| return 0; |
| } |
| |
| typedef struct FeatureToString { |
| Feature feature; |
| const char *str; |
| } FeatureToString; |
| |
| static const FeatureToString feature_to_string[] = { |
| { .feature = FEATURE_RW_NONREMOVABLE, .str = "RW_NONREMOVABLE", }, |
| { .feature = FEATURE_RW_REMOVABLE, .str = "RW_REMOVABLE", }, |
| |
| { .feature = FEATURE_MO_SE, .str = "MO_SE", }, |
| { .feature = FEATURE_MO_WO, .str = "MO_WO", }, |
| { .feature = FEATURE_MO_AS, .str = "MO_AS", }, |
| |
| { .feature = FEATURE_CD_ROM, .str = "CD", }, |
| { .feature = FEATURE_CD_R, .str = "CD_R", }, |
| { .feature = FEATURE_CD_RW, .str = "CD_RW", }, |
| |
| { .feature = FEATURE_DVD_ROM, .str = "DVD", }, |
| { .feature = FEATURE_DVD_R, .str = "DVD_R", }, |
| { .feature = FEATURE_DVD_RAM, .str = "DVD_RAM", }, |
| { .feature = FEATURE_DVD_RW_RO, .str = "DVD_RW_RO", }, |
| { .feature = FEATURE_DVD_RW_SEQ, .str = "DVD_RW_SEQ", }, |
| { .feature = FEATURE_DVD_R_DL_SEQ, .str = "DVD_R_DL_SEQ", }, |
| { .feature = FEATURE_DVD_R_DL_JR, .str = "DVD_R_DL_JR", }, |
| { .feature = FEATURE_DVD_RW_DL, .str = "DVD_RW_DL", }, |
| { .feature = FEATURE_DVD_R_DDR, .str = "DVD_R_DDR", }, |
| { .feature = FEATURE_DVD_PLUS_RW, .str = "DVD_PLUS_RW", }, |
| { .feature = FEATURE_DVD_PLUS_R, .str = "DVD_PLUS_R", }, |
| |
| { .feature = FEATURE_DDCD_ROM, .str = "DDCD", }, |
| { .feature = FEATURE_DDCD_R, .str = "DDCD_R", }, |
| { .feature = FEATURE_DDCD_RW, .str = "DDCD_RW", }, |
| |
| { .feature = FEATURE_DVD_PLUS_RW_DL, .str = "DVD_PLUS_RW_DL", }, |
| { .feature = FEATURE_DVD_PLUS_R_DL, .str = "DVD_PLUS_R_DL", }, |
| |
| { .feature = FEATURE_BD, .str = "BD", }, |
| { .feature = FEATURE_BD_R_SRM, .str = "BD_R_SRM", }, |
| { .feature = FEATURE_BD_R_RRM, .str = "BD_R_RRM", }, |
| { .feature = FEATURE_BD_RE, .str = "BD_RE", }, |
| |
| { .feature = FEATURE_HDDVD, .str = "HDDVD", }, |
| { .feature = FEATURE_HDDVD_R, .str = "HDDVD_R", }, |
| { .feature = FEATURE_HDDVD_RAM, .str = "HDDVD_RAM", }, |
| { .feature = FEATURE_HDDVD_RW, .str = "HDDVD_RW", }, |
| { .feature = FEATURE_HDDVD_R_DL, .str = "HDDVD_R_DL", }, |
| { .feature = FEATURE_HDDVD_RW_DL, .str = "HDDVD_RW_DL", }, |
| |
| { .feature = FEATURE_MRW, .str = "MRW", }, |
| { .feature = FEATURE_MRW_W, .str = "MRW_W", }, |
| }; |
| |
| static int feature_to_string_compare_func(const FeatureToString *a, const FeatureToString *b) { |
| assert(a); |
| assert(b); |
| |
| return CMP(a->feature, b->feature); |
| } |
| |
| static void print_feature(Feature feature, const char *prefix) { |
| FeatureToString *found, in = { |
| .feature = feature, |
| }; |
| |
| assert(prefix); |
| |
| found = typesafe_bsearch(&in, feature_to_string, ELEMENTSOF(feature_to_string), feature_to_string_compare_func); |
| if (!found) |
| return (void) log_debug("Unknown feature 0x%02x, ignoring.", (unsigned) feature); |
| |
| printf("%s_%s=1\n", prefix, found->str); |
| } |
| |
| static void print_properties(const Context *c) { |
| const char *state; |
| |
| assert(c); |
| |
| printf("ID_CDROM=1\n"); |
| for (size_t i = 0; i < c->n_drive_feature; i++) |
| print_feature(c->drive_features[i], "ID_CDROM"); |
| |
| if (drive_has_feature(c, FEATURE_MO_SE) || |
| drive_has_feature(c, FEATURE_MO_WO) || |
| drive_has_feature(c, FEATURE_MO_AS)) |
| printf("ID_CDROM_MO=1\n"); |
| |
| if (drive_has_feature(c, FEATURE_DVD_RW_RO) || |
| drive_has_feature(c, FEATURE_DVD_RW_SEQ)) |
| printf("ID_CDROM_DVD_RW=1\n"); |
| |
| if (drive_has_feature(c, FEATURE_DVD_R_DL_SEQ) || |
| drive_has_feature(c, FEATURE_DVD_R_DL_JR)) |
| printf("ID_CDROM_DVD_R_DL=1\n"); |
| |
| if (drive_has_feature(c, FEATURE_DVD_R_DDR)) |
| printf("ID_CDROM_DVD_R=1\n"); |
| |
| if (drive_has_feature(c, FEATURE_BD_R_SRM) || |
| drive_has_feature(c, FEATURE_BD_R_RRM)) |
| printf("ID_CDROM_BD_R=1\n"); |
| |
| if (c->has_media) { |
| printf("ID_CDROM_MEDIA=1\n"); |
| print_feature(c->media_feature, "ID_CDROM_MEDIA"); |
| |
| if (IN_SET(c->media_feature, FEATURE_MO_SE, FEATURE_MO_WO, FEATURE_MO_AS)) |
| printf("ID_CDROM_MEDIA_MO=1\n"); |
| |
| if (IN_SET(c->media_feature, FEATURE_DVD_RW_RO, FEATURE_DVD_RW_SEQ)) |
| printf("ID_CDROM_MEDIA_DVD_RW=1\n"); |
| |
| if (IN_SET(c->media_feature, FEATURE_DVD_R_DL_SEQ, FEATURE_DVD_R_DL_JR)) |
| printf("ID_CDROM_MEDIA_DVD_R_DL=1\n"); |
| |
| if (c->media_feature == FEATURE_DVD_R_DDR) |
| printf("ID_CDROM_MEDIA_DVD_R=1\n"); |
| |
| if (IN_SET(c->media_feature, FEATURE_BD_R_SRM, FEATURE_BD_R_RRM)) |
| printf("ID_CDROM_MEDIA_BD_R=1\n"); |
| } |
| |
| state = media_state_to_string(c->media_state); |
| if (state) |
| printf("ID_CDROM_MEDIA_STATE=%s\n", state); |
| if (c->media_session_next > 0) |
| printf("ID_CDROM_MEDIA_SESSION_NEXT=%u\n", c->media_session_next); |
| if (c->media_session_count > 0) |
| printf("ID_CDROM_MEDIA_SESSION_COUNT=%u\n", c->media_session_count); |
| if (c->media_session_count > 1 && c->media_session_last_offset > 0) |
| printf("ID_CDROM_MEDIA_SESSION_LAST_OFFSET=%" PRIu64 "\n", c->media_session_last_offset); |
| if (c->media_track_count > 0) |
| printf("ID_CDROM_MEDIA_TRACK_COUNT=%u\n", c->media_track_count); |
| if (c->media_track_count_audio > 0) |
| printf("ID_CDROM_MEDIA_TRACK_COUNT_AUDIO=%u\n", c->media_track_count_audio); |
| if (c->media_track_count_data > 0) |
| printf("ID_CDROM_MEDIA_TRACK_COUNT_DATA=%u\n", c->media_track_count_data); |
| } |
| |
| static int help(void) { |
| printf("Usage: %s [options] <device>\n" |
| " -l --lock-media lock the media (to enable eject request events)\n" |
| " -u --unlock-media unlock the media\n" |
| " -e --eject-media eject the media\n" |
| " -d --debug print debug messages to stderr\n" |
| " -h --help print this help text\n" |
| "\n", |
| program_invocation_short_name); |
| |
| return 0; |
| } |
| |
| static int parse_argv(int argc, char *argv[]) { |
| static const struct option options[] = { |
| { "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' }, |
| {} |
| }; |
| int c; |
| |
| while ((c = getopt_long(argc, argv, "deluh", options, NULL)) >= 0) |
| switch (c) { |
| case 'l': |
| arg_lock = true; |
| break; |
| case 'u': |
| arg_unlock = true; |
| break; |
| case 'e': |
| arg_eject = true; |
| break; |
| case 'd': |
| log_set_target(LOG_TARGET_CONSOLE); |
| log_set_max_level(LOG_DEBUG); |
| log_open(); |
| break; |
| case 'h': |
| return help(); |
| default: |
| assert_not_reached(); |
| } |
| |
| arg_node = argv[optind]; |
| if (!arg_node) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "No device specified."); |
| |
| return 1; |
| } |
| |
| static int run(int argc, char *argv[]) { |
| _cleanup_(context_clear) Context c = CONTEXT_EMPTY; |
| int r; |
| |
| log_set_target(LOG_TARGET_AUTO); |
| udev_parse_config(); |
| log_parse_environment(); |
| log_open(); |
| |
| r = parse_argv(argc, argv); |
| if (r <= 0) |
| return r; |
| |
| r = open_drive(&c); |
| if (r < 0) |
| return r; |
| |
| /* same data as original cdrom_id */ |
| r = cd_capability_compat(&c); |
| if (r < 0) |
| return r; |
| |
| /* check for media - don't bail if there's no media as we still need to |
| * to read profiles */ |
| (void) cd_media_compat(&c); |
| |
| /* check if drive talks MMC */ |
| if (cd_inquiry(&c) < 0) |
| goto work; |
| |
| r = cd_profiles(&c); /* read drive and possibly current profile */ |
| if (r > 0) { |
| /* at this point we are guaranteed to have media in the drive - find out more about it */ |
| |
| /* get session/track info */ |
| (void) cd_media_toc(&c); |
| |
| /* get writable media state */ |
| (void) cd_media_info(&c); |
| } |
| |
| work: |
| /* lock the media, so we enable eject button events */ |
| if (arg_lock && c.has_media) { |
| log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (lock)"); |
| (void) media_lock(c.fd, true); |
| } |
| |
| if (arg_unlock && c.has_media) { |
| log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)"); |
| (void) media_lock(c.fd, false); |
| } |
| |
| if (arg_eject) { |
| log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)"); |
| (void) media_lock(c.fd, false); |
| log_debug("START_STOP_UNIT (eject)"); |
| (void) media_eject(c.fd); |
| } |
| |
| print_properties(&c); |
| |
| return 0; |
| } |
| |
| DEFINE_MAIN_FUNCTION(run); |