src/udev/cdrom_id/cdrom_id.c

   1 /* SPDX-License-Identifier: GPL-2.0+ */
   2 /*
   3  * cdrom_id - optical drive and media information prober
   4  */
   5
   6 #include <errno.h>
   7 #include <fcntl.h>
   8 #include <getopt.h>
   9 #include <limits.h>
  10 #include <linux/cdrom.h>
  11 #include <scsi/sg.h>
  12 #include <stddef.h>
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <sys/ioctl.h>
  17 #include <sys/stat.h>
  18 #include <sys/time.h>
  19 #include <sys/types.h>
  20 #include <time.h>
  21 #include <unistd.h>
  22
  23 #include "log.h"
  24 #include "memory-util.h"
  25 #include "random-util.h"
  26 #include "udev-util.h"
  27
  28 /* device info */
  29 static unsigned cd_cd_rom;
  30 static unsigned cd_cd_r;
  31 static unsigned cd_cd_rw;
  32 static unsigned cd_dvd_rom;
  33 static unsigned cd_dvd_r;
  34 static unsigned cd_dvd_rw;
  35 static unsigned cd_dvd_ram;
  36 static unsigned cd_dvd_plus_r;
  37 static unsigned cd_dvd_plus_rw;
  38 static unsigned cd_dvd_plus_r_dl;
  39 static unsigned cd_dvd_plus_rw_dl;
  40 static unsigned cd_bd;
  41 static unsigned cd_bd_r;
  42 static unsigned cd_bd_re;
  43 static unsigned cd_hddvd;
  44 static unsigned cd_hddvd_r;
  45 static unsigned cd_hddvd_rw;
  46 static unsigned cd_mo;
  47 static unsigned cd_mrw;
  48 static unsigned cd_mrw_w;
  49
  50 /* media info */
  51 static unsigned cd_media;
  52 static unsigned cd_media_cd_rom;
  53 static unsigned cd_media_cd_r;
  54 static unsigned cd_media_cd_rw;
  55 static unsigned cd_media_dvd_rom;
  56 static unsigned cd_media_dvd_r;
  57 static unsigned cd_media_dvd_rw;
  58 static unsigned cd_media_dvd_rw_ro; /* restricted overwrite mode */
  59 static unsigned cd_media_dvd_rw_seq; /* sequential mode */
  60 static unsigned cd_media_dvd_ram;
  61 static unsigned cd_media_dvd_plus_r;
  62 static unsigned cd_media_dvd_plus_rw;
  63 static unsigned cd_media_dvd_plus_r_dl;
  64 static unsigned cd_media_dvd_plus_rw_dl;
  65 static unsigned cd_media_bd;
  66 static unsigned cd_media_bd_r;
  67 static unsigned cd_media_bd_re;
  68 static unsigned cd_media_hddvd;
  69 static unsigned cd_media_hddvd_r;
  70 static unsigned cd_media_hddvd_rw;
  71 static unsigned cd_media_mo;
  72 static unsigned cd_media_mrw;
  73 static unsigned cd_media_mrw_w;
  74
  75 static const char *cd_media_state = NULL;
  76 static unsigned cd_media_session_next;
  77 static unsigned cd_media_session_count;
  78 static unsigned cd_media_track_count;
  79 static unsigned cd_media_track_count_data;
  80 static unsigned cd_media_track_count_audio;
  81 static unsigned long long int cd_media_session_last_offset;
  82
  83 #define ERRCODE(s)        ((((s)[2] & 0x0F) << 16) | ((s)[12] << 8) | ((s)[13]))
  84 #define SK(errcode)        (((errcode) >> 16) & 0xF)
  85 #define ASC(errcode)        (((errcode) >> 8) & 0xFF)
  86 #define ASCQ(errcode)        ((errcode) & 0xFF)
  87
  88 static bool is_mounted(const char *device) {
  89         struct stat statbuf;
  90         FILE *fp;
  91         int maj, min;
  92         bool mounted = false;
  93
  94         if (stat(device, &statbuf) < 0)
  95                 return false;
  96
  97         fp = fopen("/proc/self/mountinfo", "re");
  98         if (!fp)
  99                 return false;
 100         while (fscanf(fp, "%*s %*s %i:%i %*[^\n]", &maj, &min) == 2) {
 101                 if (makedev(maj, min) == statbuf.st_rdev) {
 102                         mounted = true;
 103                         break;
 104                 }
 105         }
 106         fclose(fp);
 107         return mounted;
 108 }
 109
 110 static void info_scsi_cmd_err(const char *cmd, int err) {
 111         if (err == -1)
 112                 log_debug("%s failed", cmd);
 113         else
 114                 log_debug("%s failed with SK=%Xh/ASC=%02Xh/ACQ=%02Xh", cmd, SK(err), ASC(err), ASCQ(err));
 115 }
 116
 117 struct scsi_cmd {
 118         struct cdrom_generic_command cgc;
 119         union {
 120                 struct request_sense s;
 121                 unsigned char u[18];
 122         } _sense;
 123         struct sg_io_hdr sg_io;
 124 };
 125
 126 static void scsi_cmd_init(struct scsi_cmd *cmd) {
 127         memzero(cmd, sizeof(struct scsi_cmd));
 128         cmd->cgc.quiet = 1;
 129         cmd->cgc.sense = &cmd->_sense.s;
 130         cmd->sg_io.interface_id = 'S';
 131         cmd->sg_io.mx_sb_len = sizeof(cmd->_sense);
 132         cmd->sg_io.cmdp = cmd->cgc.cmd;
 133         cmd->sg_io.sbp = cmd->_sense.u;
 134         cmd->sg_io.flags = SG_FLAG_LUN_INHIBIT | SG_FLAG_DIRECT_IO;
 135 }
 136
 137 static void scsi_cmd_set(struct scsi_cmd *cmd, size_t i, unsigned char arg) {
 138         cmd->sg_io.cmd_len = i + 1;
 139         cmd->cgc.cmd[i] = arg;
 140 }
 141
 142 #define CHECK_CONDITION 0x01
 143
 144 static int scsi_cmd_run(struct scsi_cmd *cmd, int fd, unsigned char *buf, size_t bufsize) {
 145         int ret = 0;
 146
 147         if (bufsize > 0) {
 148                 cmd->sg_io.dxferp = buf;
 149                 cmd->sg_io.dxfer_len = bufsize;
 150                 cmd->sg_io.dxfer_direction = SG_DXFER_FROM_DEV;
 151         } else
 152                 cmd->sg_io.dxfer_direction = SG_DXFER_NONE;
 153
 154         if (ioctl(fd, SG_IO, &cmd->sg_io))
 155                 return -1;
 156
 157         if ((cmd->sg_io.info & SG_INFO_OK_MASK) != SG_INFO_OK) {
 158                 errno = EIO;
 159                 ret = -1;
 160                 if (cmd->sg_io.masked_status & CHECK_CONDITION) {
 161                         ret = ERRCODE(cmd->_sense.u);
 162                         if (ret == 0)
 163                                 ret = -1;
 164                 }
 165         }
 166         return ret;
 167 }
 168
 169 static int media_lock(int fd, bool lock) {
 170         int err;
 171
 172         /* disable the kernel's lock logic */
 173         err = ioctl(fd, CDROM_CLEAR_OPTIONS, CDO_LOCK);
 174         if (err < 0)
 175                 log_debug("CDROM_CLEAR_OPTIONS, CDO_LOCK failed");
 176
 177         err = ioctl(fd, CDROM_LOCKDOOR, lock ? 1 : 0);
 178         if (err < 0)
 179                 log_debug("CDROM_LOCKDOOR failed");
 180
 181         return err;
 182 }
 183
 184 static int media_eject(int fd) {
 185         struct scsi_cmd sc;
 186         int err;
 187
 188         scsi_cmd_init(&sc);
 189         scsi_cmd_set(&sc, 0, 0x1b);
 190         scsi_cmd_set(&sc, 4, 0x02);
 191         scsi_cmd_set(&sc, 5, 0);
 192         err = scsi_cmd_run(&sc, fd, NULL, 0);
 193         if (err != 0) {
 194                 info_scsi_cmd_err("START_STOP_UNIT", err);
 195                 return -1;
 196         }
 197         return 0;
 198 }
 199
 200 static int cd_capability_compat(int fd) {
 201         int capability;
 202
 203         capability = ioctl(fd, CDROM_GET_CAPABILITY, NULL);
 204         if (capability < 0)
 205                 return log_debug_errno(errno, "CDROM_GET_CAPABILITY failed");
 206
 207         if (capability & CDC_CD_R)
 208                 cd_cd_r = 1;
 209         if (capability & CDC_CD_RW)
 210                 cd_cd_rw = 1;
 211         if (capability & CDC_DVD)
 212                 cd_dvd_rom = 1;
 213         if (capability & CDC_DVD_R)
 214                 cd_dvd_r = 1;
 215         if (capability & CDC_DVD_RAM)
 216                 cd_dvd_ram = 1;
 217         if (capability & CDC_MRW)
 218                 cd_mrw = 1;
 219         if (capability & CDC_MRW_W)
 220                 cd_mrw_w = 1;
 221         return 0;
 222 }
 223
 224 static int cd_media_compat(int fd) {
 225         if (ioctl(fd, CDROM_DRIVE_STATUS, CDSL_CURRENT) != CDS_DISC_OK)
 226                 return log_debug_errno(errno, "CDROM_DRIVE_STATUS != CDS_DISC_OK");
 227
 228         cd_media = 1;
 229         return 0;
 230 }
 231
 232 static int cd_inquiry(int fd) {
 233         struct scsi_cmd sc;
 234         unsigned char inq[128];
 235         int err;
 236
 237         scsi_cmd_init(&sc);
 238         scsi_cmd_set(&sc, 0, 0x12);
 239         scsi_cmd_set(&sc, 4, 36);
 240         scsi_cmd_set(&sc, 5, 0);
 241         err = scsi_cmd_run(&sc, fd, inq, 36);
 242         if (err != 0) {
 243                 info_scsi_cmd_err("INQUIRY", err);
 244                 return -1;
 245         }
 246
 247         if ((inq[0] & 0x1F) != 5)
 248                 return log_debug_errno(SYNTHETIC_ERRNO(EINVAL), "not an MMC unit");
 249
 250         log_debug("INQUIRY: [%.8s][%.16s][%.4s]", inq + 8, inq + 16, inq + 32);
 251         return 0;
 252 }
 253
 254 static void feature_profile_media(int cur_profile) {
 255         switch (cur_profile) {
 256         case 0x03:
 257         case 0x04:
 258         case 0x05:
 259                 log_debug("profile 0x%02x ", cur_profile);
 260                 cd_media = 1;
 261                 cd_media_mo = 1;
 262                 break;
 263         case 0x08:
 264                 log_debug("profile 0x%02x media_cd_rom", cur_profile);
 265                 cd_media = 1;
 266                 cd_media_cd_rom = 1;
 267                 break;
 268         case 0x09:
 269                 log_debug("profile 0x%02x media_cd_r", cur_profile);
 270                 cd_media = 1;
 271                 cd_media_cd_r = 1;
 272                 break;
 273         case 0x0a:
 274                 log_debug("profile 0x%02x media_cd_rw", cur_profile);
 275                 cd_media = 1;
 276                 cd_media_cd_rw = 1;
 277                 break;
 278         case 0x10:
 279                 log_debug("profile 0x%02x media_dvd_ro", cur_profile);
 280                 cd_media = 1;
 281                 cd_media_dvd_rom = 1;
 282                 break;
 283         case 0x11:
 284                 log_debug("profile 0x%02x media_dvd_r", cur_profile);
 285                 cd_media = 1;
 286                 cd_media_dvd_r = 1;
 287                 break;
 288         case 0x12:
 289                 log_debug("profile 0x%02x media_dvd_ram", cur_profile);
 290                 cd_media = 1;
 291                 cd_media_dvd_ram = 1;
 292                 break;
 293         case 0x13:
 294                 log_debug("profile 0x%02x media_dvd_rw_ro", cur_profile);
 295                 cd_media = 1;
 296                 cd_media_dvd_rw = 1;
 297                 cd_media_dvd_rw_ro = 1;
 298                 break;
 299         case 0x14:
 300                 log_debug("profile 0x%02x media_dvd_rw_seq", cur_profile);
 301                 cd_media = 1;
 302                 cd_media_dvd_rw = 1;
 303                 cd_media_dvd_rw_seq = 1;
 304                 break;
 305         case 0x1B:
 306                 log_debug("profile 0x%02x media_dvd_plus_r", cur_profile);
 307                 cd_media = 1;
 308                 cd_media_dvd_plus_r = 1;
 309                 break;
 310         case 0x1A:
 311                 log_debug("profile 0x%02x media_dvd_plus_rw", cur_profile);
 312                 cd_media = 1;
 313                 cd_media_dvd_plus_rw = 1;
 314                 break;
 315         case 0x2A:
 316                 log_debug("profile 0x%02x media_dvd_plus_rw_dl", cur_profile);
 317                 cd_media = 1;
 318                 cd_media_dvd_plus_rw_dl = 1;
 319                 break;
 320         case 0x2B:
 321                 log_debug("profile 0x%02x media_dvd_plus_r_dl", cur_profile);
 322                 cd_media = 1;
 323                 cd_media_dvd_plus_r_dl = 1;
 324                 break;
 325         case 0x40:
 326                 log_debug("profile 0x%02x media_bd", cur_profile);
 327                 cd_media = 1;
 328                 cd_media_bd = 1;
 329                 break;
 330         case 0x41:
 331         case 0x42:
 332                 log_debug("profile 0x%02x media_bd_r", cur_profile);
 333                 cd_media = 1;
 334                 cd_media_bd_r = 1;
 335                 break;
 336         case 0x43:
 337                 log_debug("profile 0x%02x media_bd_re", cur_profile);
 338                 cd_media = 1;
 339                 cd_media_bd_re = 1;
 340                 break;
 341         case 0x50:
 342                 log_debug("profile 0x%02x media_hddvd", cur_profile);
 343                 cd_media = 1;
 344                 cd_media_hddvd = 1;
 345                 break;
 346         case 0x51:
 347                 log_debug("profile 0x%02x media_hddvd_r", cur_profile);
 348                 cd_media = 1;
 349                 cd_media_hddvd_r = 1;
 350                 break;
 351         case 0x52:
 352                 log_debug("profile 0x%02x media_hddvd_rw", cur_profile);
 353                 cd_media = 1;
 354                 cd_media_hddvd_rw = 1;
 355                 break;
 356         default:
 357                 log_debug("profile 0x%02x <ignored>", cur_profile);
 358                 break;
 359         }
 360 }
 361
 362 static int feature_profiles(const unsigned char *profiles, size_t size) {
 363         unsigned i;
 364
 365         for (i = 0; i+4 <= size; i += 4) {
 366                 int profile;
 367
 368                 profile = profiles[i] << 8 | profiles[i+1];
 369                 switch (profile) {
 370                 case 0x03:
 371                 case 0x04:
 372                 case 0x05:
 373                         log_debug("profile 0x%02x mo", profile);
 374                         cd_mo = 1;
 375                         break;
 376                 case 0x08:
 377                         log_debug("profile 0x%02x cd_rom", profile);
 378                         cd_cd_rom = 1;
 379                         break;
 380                 case 0x09:
 381                         log_debug("profile 0x%02x cd_r", profile);
 382                         cd_cd_r = 1;
 383                         break;
 384                 case 0x0A:
 385                         log_debug("profile 0x%02x cd_rw", profile);
 386                         cd_cd_rw = 1;
 387                         break;
 388                 case 0x10:
 389                         log_debug("profile 0x%02x dvd_rom", profile);
 390                         cd_dvd_rom = 1;
 391                         break;
 392                 case 0x12:
 393                         log_debug("profile 0x%02x dvd_ram", profile);
 394                         cd_dvd_ram = 1;
 395                         break;
 396                 case 0x13:
 397                 case 0x14:
 398                         log_debug("profile 0x%02x dvd_rw", profile);
 399                         cd_dvd_rw = 1;
 400                         break;
 401                 case 0x1B:
 402                         log_debug("profile 0x%02x dvd_plus_r", profile);
 403                         cd_dvd_plus_r = 1;
 404                         break;
 405                 case 0x1A:
 406                         log_debug("profile 0x%02x dvd_plus_rw", profile);
 407                         cd_dvd_plus_rw = 1;
 408                         break;
 409                 case 0x2A:
 410                         log_debug("profile 0x%02x dvd_plus_rw_dl", profile);
 411                         cd_dvd_plus_rw_dl = 1;
 412                         break;
 413                 case 0x2B:
 414                         log_debug("profile 0x%02x dvd_plus_r_dl", profile);
 415                         cd_dvd_plus_r_dl = 1;
 416                         break;
 417                 case 0x40:
 418                         cd_bd = 1;
 419                         log_debug("profile 0x%02x bd", profile);
 420                         break;
 421                 case 0x41:
 422                 case 0x42:
 423                         cd_bd_r = 1;
 424                         log_debug("profile 0x%02x bd_r", profile);
 425                         break;
 426                 case 0x43:
 427                         cd_bd_re = 1;
 428                         log_debug("profile 0x%02x bd_re", profile);
 429                         break;
 430                 case 0x50:
 431                         cd_hddvd = 1;
 432                         log_debug("profile 0x%02x hddvd", profile);
 433                         break;
 434                 case 0x51:
 435                         cd_hddvd_r = 1;
 436                         log_debug("profile 0x%02x hddvd_r", profile);
 437                         break;
 438                 case 0x52:
 439                         cd_hddvd_rw = 1;
 440                         log_debug("profile 0x%02x hddvd_rw", profile);
 441                         break;
 442                 default:
 443                         log_debug("profile 0x%02x <ignored>", profile);
 444                         break;
 445                 }
 446         }
 447         return 0;
 448 }
 449
 450 /* returns 0 if media was detected */
 451 static int cd_profiles_old_mmc(int fd) {
 452         struct scsi_cmd sc;
 453         int err;
 454
 455         unsigned char header[32];
 456
 457         scsi_cmd_init(&sc);
 458         scsi_cmd_set(&sc, 0, 0x51);
 459         scsi_cmd_set(&sc, 8, sizeof(header));
 460         scsi_cmd_set(&sc, 9, 0);
 461         err = scsi_cmd_run(&sc, fd, header, sizeof(header));
 462         if (err != 0) {
 463                 info_scsi_cmd_err("READ DISC INFORMATION", err);
 464                 if (cd_media == 1) {
 465                         log_debug("no current profile, but disc is present; assuming CD-ROM");
 466                         cd_media_cd_rom = 1;
 467                         cd_media_track_count = 1;
 468                         cd_media_track_count_data = 1;
 469                         return 0;
 470                 } else
 471                         return log_debug_errno(SYNTHETIC_ERRNO(ENOMEDIUM),
 472                                                "no current profile, assuming no media");
 473         };
 474
 475         cd_media = 1;
 476
 477         if (header[2] & 16) {
 478                 cd_media_cd_rw = 1;
 479                 log_debug("profile 0x0a media_cd_rw");
 480         } else if ((header[2] & 3) < 2 && cd_cd_r) {
 481                 cd_media_cd_r = 1;
 482                 log_debug("profile 0x09 media_cd_r");
 483         } else {
 484                 cd_media_cd_rom = 1;
 485                 log_debug("profile 0x08 media_cd_rom");
 486         }
 487         return 0;
 488 }
 489
 490 /* returns 0 if media was detected */
 491 static int cd_profiles(int fd) {
 492         struct scsi_cmd sc;
 493         unsigned char features[65530];
 494         unsigned cur_profile = 0;
 495         unsigned len;
 496         unsigned i;
 497         int err;
 498         int ret;
 499
 500         ret = -1;
 501
 502         /* First query the current profile */
 503         scsi_cmd_init(&sc);
 504         scsi_cmd_set(&sc, 0, 0x46);
 505         scsi_cmd_set(&sc, 8, 8);
 506         scsi_cmd_set(&sc, 9, 0);
 507         err = scsi_cmd_run(&sc, fd, features, 8);
 508         if (err != 0) {
 509                 info_scsi_cmd_err("GET CONFIGURATION", err);
 510                 /* handle pre-MMC2 drives which do not support GET CONFIGURATION */
 511                 if (SK(err) == 0x5 && IN_SET(ASC(err), 0x20, 0x24)) {
 512                         log_debug("drive is pre-MMC2 and does not support 46h get configuration command");
 513                         log_debug("trying to work around the problem");
 514                         ret = cd_profiles_old_mmc(fd);
 515                 }
 516                 goto out;
 517         }
 518
 519         cur_profile = features[6] << 8 | features[7];
 520         if (cur_profile > 0) {
 521                 log_debug("current profile 0x%02x", cur_profile);
 522                 feature_profile_media(cur_profile);
 523                 ret = 0; /* we have media */
 524         } else
 525                 log_debug("no current profile, assuming no media");
 526
 527         len = features[0] << 24 | features[1] << 16 | features[2] << 8 | features[3];
 528         log_debug("GET CONFIGURATION: size of features buffer 0x%04x", len);
 529
 530         if (len > sizeof(features)) {
 531                 log_debug("cannot get features in a single query, truncating");
 532                 len = sizeof(features);
 533         } else if (len <= 8)
 534                 len = sizeof(features);
 535
 536         /* Now get the full feature buffer */
 537         scsi_cmd_init(&sc);
 538         scsi_cmd_set(&sc, 0, 0x46);
 539         scsi_cmd_set(&sc, 7, ( len >> 8 ) & 0xff);
 540         scsi_cmd_set(&sc, 8, len & 0xff);
 541         scsi_cmd_set(&sc, 9, 0);
 542         err = scsi_cmd_run(&sc, fd, features, len);
 543         if (err != 0) {
 544                 info_scsi_cmd_err("GET CONFIGURATION", err);
 545                 return -1;
 546         }
 547
 548         /* parse the length once more, in case the drive decided to have other features suddenly :) */
 549         len = features[0] << 24 | features[1] << 16 | features[2] << 8 | features[3];
 550         log_debug("GET CONFIGURATION: size of features buffer 0x%04x", len);
 551
 552         if (len > sizeof(features)) {
 553                 log_debug("cannot get features in a single query, truncating");
 554                 len = sizeof(features);
 555         }
 556
 557         /* device features */
 558         for (i = 8; i+4 < len; i += (4 + features[i+3])) {
 559                 unsigned feature;
 560
 561                 feature = features[i] << 8 | features[i+1];
 562
 563                 switch (feature) {
 564                 case 0x00:
 565                         log_debug("GET CONFIGURATION: feature 'profiles', with %i entries", features[i+3] / 4);
 566                         feature_profiles(&features[i]+4, MIN(features[i+3], len - i - 4));
 567                         break;
 568                 default:
 569                         log_debug("GET CONFIGURATION: feature 0x%04x <ignored>, with 0x%02x bytes", feature, features[i+3]);
 570                         break;
 571                 }
 572         }
 573 out:
 574         return ret;
 575 }
 576
 577 static int cd_media_info(int fd) {
 578         struct scsi_cmd sc;
 579         unsigned char header[32];
 580         static const char *const media_status[] = {
 581                 "blank",
 582                 "appendable",
 583                 "complete",
 584                 "other"
 585         };
 586         int err;
 587
 588         scsi_cmd_init(&sc);
 589         scsi_cmd_set(&sc, 0, 0x51);
 590         scsi_cmd_set(&sc, 8, sizeof(header) & 0xff);
 591         scsi_cmd_set(&sc, 9, 0);
 592         err = scsi_cmd_run(&sc, fd, header, sizeof(header));
 593         if (err != 0) {
 594                 info_scsi_cmd_err("READ DISC INFORMATION", err);
 595                 return -1;
 596         };
 597
 598         cd_media = 1;
 599         log_debug("disk type %02x", header[8]);
 600         log_debug("hardware reported media status: %s", media_status[header[2] & 3]);
 601
 602         /* exclude plain CDROM, some fake cdroms return 0 for "blank" media here */
 603         if (!cd_media_cd_rom)
 604                 cd_media_state = media_status[header[2] & 3];
 605
 606         /* fresh DVD-RW in restricted overwrite mode reports itself as
 607          * "appendable"; change it to "blank" to make it consistent with what
 608          * gets reported after blanking, and what userspace expects  */
 609         if (cd_media_dvd_rw_ro && (header[2] & 3) == 1)
 610                 cd_media_state = media_status[0];
 611
 612         /* DVD+RW discs (and DVD-RW in restricted mode) once formatted are
 613          * always "complete", DVD-RAM are "other" or "complete" if the disc is
 614          * write protected; we need to check the contents if it is blank */
 615         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) {
 616                 unsigned char buffer[32 * 2048];
 617                 unsigned char len;
 618                 int offset;
 619
 620                 if (cd_media_dvd_ram) {
 621                         /* a write protected dvd-ram may report "complete" status */
 622
 623                         unsigned char dvdstruct[8];
 624                         unsigned char format[12];
 625
 626                         scsi_cmd_init(&sc);
 627                         scsi_cmd_set(&sc, 0, 0xAD);
 628                         scsi_cmd_set(&sc, 7, 0xC0);
 629                         scsi_cmd_set(&sc, 9, sizeof(dvdstruct));
 630                         scsi_cmd_set(&sc, 11, 0);
 631                         err = scsi_cmd_run(&sc, fd, dvdstruct, sizeof(dvdstruct));
 632                         if (err != 0) {
 633                                 info_scsi_cmd_err("READ DVD STRUCTURE", err);
 634                                 return -1;
 635                         }
 636                         if (dvdstruct[4] & 0x02) {
 637                                 cd_media_state = media_status[2];
 638                                 log_debug("write-protected DVD-RAM media inserted");
 639                                 goto determined;
 640                         }
 641
 642                         /* let's make sure we don't try to read unformatted media */
 643                         scsi_cmd_init(&sc);
 644                         scsi_cmd_set(&sc, 0, 0x23);
 645                         scsi_cmd_set(&sc, 8, sizeof(format));
 646                         scsi_cmd_set(&sc, 9, 0);
 647                         err = scsi_cmd_run(&sc, fd, format, sizeof(format));
 648                         if (err != 0) {
 649                                 info_scsi_cmd_err("READ DVD FORMAT CAPACITIES", err);
 650                                 return -1;
 651                         }
 652
 653                         len = format[3];
 654                         if (len & 7 || len < 16)
 655                                 return log_debug_errno(SYNTHETIC_ERRNO(EINVAL),
 656                                                        "invalid format capacities length");
 657
 658                         switch(format[8] & 3) {
 659                             case 1:
 660                                 log_debug("unformatted DVD-RAM media inserted");
 661                                 /* This means that last format was interrupted
 662                                  * or failed, blank dvd-ram discs are factory
 663                                  * formatted. Take no action here as it takes
 664                                  * quite a while to reformat a dvd-ram and it's
 665                                  * not automatically started */
 666                                 goto determined;
 667
 668                             case 2:
 669                                 log_debug("formatted DVD-RAM media inserted");
 670                                 break;
 671
 672                             case 3:
 673                                 cd_media = 0; //return no media
 674                                 return log_debug_errno(SYNTHETIC_ERRNO(ENOMEDIUM),
 675                                                        "format capacities returned no media");
 676                         }
 677                 }
 678
 679                 /* Take a closer look at formatted media (unformatted DVD+RW
 680                  * has "blank" status", DVD-RAM was examined earlier) and check
 681                  * for ISO and UDF PVDs or a fs superblock presence and do it
 682                  * in one ioctl (we need just sectors 0 and 16) */
 683                 scsi_cmd_init(&sc);
 684                 scsi_cmd_set(&sc, 0, 0x28);
 685                 scsi_cmd_set(&sc, 5, 0);
 686                 scsi_cmd_set(&sc, 8, 32);
 687                 scsi_cmd_set(&sc, 9, 0);
 688                 err = scsi_cmd_run(&sc, fd, buffer, sizeof(buffer));
 689                 if (err != 0) {
 690                         cd_media = 0;
 691                         info_scsi_cmd_err("READ FIRST 32 BLOCKS", err);
 692                         return -1;
 693                 }
 694
 695                 /* if any non-zero data is found in sector 16 (iso and udf) or
 696                  * eventually 0 (fat32 boot sector, ext2 superblock, etc), disc
 697                  * is assumed non-blank */
 698
 699                 for (offset = 32768; offset < (32768 + 2048); offset++) {
 700                         if (buffer [offset]) {
 701                                 log_debug("data in block 16, assuming complete");
 702                                 goto determined;
 703                         }
 704                 }
 705
 706                 for (offset = 0; offset < 2048; offset++) {
 707                         if (buffer [offset]) {
 708                                 log_debug("data in block 0, assuming complete");
 709                                 goto determined;
 710                         }
 711                 }
 712
 713                 cd_media_state = media_status[0];
 714                 log_debug("no data in blocks 0 or 16, assuming blank");
 715         }
 716
 717 determined:
 718         /* "other" is e. g. DVD-RAM, can't append sessions there; DVDs in
 719          * restricted overwrite mode can never append, only in sequential mode */
 720         if ((header[2] & 3) < 2 && !cd_media_dvd_rw_ro)
 721                 cd_media_session_next = header[10] << 8 | header[5];
 722         cd_media_session_count = header[9] << 8 | header[4];
 723         cd_media_track_count = header[11] << 8 | header[6];
 724
 725         return 0;
 726 }
 727
 728 static int cd_media_toc(int fd) {
 729         struct scsi_cmd sc;
 730         unsigned char header[12];
 731         unsigned char toc[65536];
 732         unsigned len, i, num_tracks;
 733         unsigned char *p;
 734         int err;
 735
 736         scsi_cmd_init(&sc);
 737         scsi_cmd_set(&sc, 0, 0x43);
 738         scsi_cmd_set(&sc, 6, 1);
 739         scsi_cmd_set(&sc, 8, sizeof(header) & 0xff);
 740         scsi_cmd_set(&sc, 9, 0);
 741         err = scsi_cmd_run(&sc, fd, header, sizeof(header));
 742         if (err != 0) {
 743                 info_scsi_cmd_err("READ TOC", err);
 744                 return -1;
 745         }
 746
 747         len = (header[0] << 8 | header[1]) + 2;
 748         log_debug("READ TOC: len: %d, start track: %d, end track: %d", len, header[2], header[3]);
 749         if (len > sizeof(toc))
 750                 return -1;
 751         if (len < 2)
 752                 return -1;
 753         /* 2: first track, 3: last track */
 754         num_tracks = header[3] - header[2] + 1;
 755
 756         /* empty media has no tracks */
 757         if (len < 8)
 758                 return 0;
 759
 760         scsi_cmd_init(&sc);
 761         scsi_cmd_set(&sc, 0, 0x43);
 762         scsi_cmd_set(&sc, 6, header[2]); /* First Track/Session Number */
 763         scsi_cmd_set(&sc, 7, (len >> 8) & 0xff);
 764         scsi_cmd_set(&sc, 8, len & 0xff);
 765         scsi_cmd_set(&sc, 9, 0);
 766         err = scsi_cmd_run(&sc, fd, toc, len);
 767         if (err != 0) {
 768                 info_scsi_cmd_err("READ TOC (tracks)", err);
 769                 return -1;
 770         }
 771
 772         /* Take care to not iterate beyond the last valid track as specified in
 773          * the TOC, but also avoid going beyond the TOC length, just in case
 774          * the last track number is invalidly large */
 775         for (p = toc+4, i = 4; i < len-8 && num_tracks > 0; i += 8, p += 8, --num_tracks) {
 776                 unsigned block;
 777                 unsigned is_data_track;
 778
 779                 is_data_track = (p[1] & 0x04) != 0;
 780
 781                 block = p[4] << 24 | p[5] << 16 | p[6] << 8 | p[7];
 782                 log_debug("track=%u info=0x%x(%s) start_block=%u",
 783                      p[2], p[1] & 0x0f, is_data_track ? "data":"audio", block);
 784
 785                 if (is_data_track)
 786                         cd_media_track_count_data++;
 787                 else
 788                         cd_media_track_count_audio++;
 789         }
 790
 791         scsi_cmd_init(&sc);
 792         scsi_cmd_set(&sc, 0, 0x43);
 793         scsi_cmd_set(&sc, 2, 1); /* Session Info */
 794         scsi_cmd_set(&sc, 8, sizeof(header));
 795         scsi_cmd_set(&sc, 9, 0);
 796         err = scsi_cmd_run(&sc, fd, header, sizeof(header));
 797         if (err != 0) {
 798                 info_scsi_cmd_err("READ TOC (multi session)", err);
 799                 return -1;
 800         }
 801         len = header[4+4] << 24 | header[4+5] << 16 | header[4+6] << 8 | header[4+7];
 802         log_debug("last track %u starts at block %u", header[4+2], len);
 803         cd_media_session_last_offset = (unsigned long long int)len * 2048;
 804         return 0;
 805 }
 806
 807 int main(int argc, char *argv[]) {
 808         static const struct option options[] = {
 809                 { "lock-media", no_argument, NULL, 'l' },
 810                 { "unlock-media", no_argument, NULL, 'u' },
 811                 { "eject-media", no_argument, NULL, 'e' },
 812                 { "debug", no_argument, NULL, 'd' },
 813                 { "help", no_argument, NULL, 'h' },
 814                 {}
 815         };
 816         bool eject = false;
 817         bool lock = false;
 818         bool unlock = false;
 819         const char *node = NULL;
 820         int fd = -1;
 821         int cnt;
 822         int rc = 0;
 823
 824         log_set_target(LOG_TARGET_AUTO);
 825         udev_parse_config();
 826         log_parse_environment();
 827         log_open();
 828
 829         for (;;) {
 830                 int option;
 831
 832                 option = getopt_long(argc, argv, "deluh", options, NULL);
 833                 if (option == -1)
 834                         break;
 835
 836                 switch (option) {
 837                 case 'l':
 838                         lock = true;
 839                         break;
 840                 case 'u':
 841                         unlock = true;
 842                         break;
 843                 case 'e':
 844                         eject = true;
 845                         break;
 846                 case 'd':
 847                         log_set_target(LOG_TARGET_CONSOLE);
 848                         log_set_max_level(LOG_DEBUG);
 849                         log_open();
 850                         break;
 851                 case 'h':
 852                         printf("Usage: cdrom_id [options] <device>\n"
 853                                "  -l,--lock-media    lock the media (to enable eject request events)\n"
 854                                "  -u,--unlock-media  unlock the media\n"
 855                                "  -e,--eject-media   eject the media\n"
 856                                "  -d,--debug         debug to stderr\n"
 857                                "  -h,--help          print this help text\n\n");
 858                         goto exit;
 859                 default:
 860                         rc = 1;
 861                         goto exit;
 862                 }
 863         }
 864
 865         node = argv[optind];
 866         if (!node) {
 867                 log_error("no device");
 868                 rc = 1;
 869                 goto exit;
 870         }
 871
 872         initialize_srand();
 873         for (cnt = 20; cnt > 0; cnt--) {
 874                 struct timespec duration;
 875
 876                 fd = open(node, O_RDONLY|O_NONBLOCK|O_CLOEXEC|(is_mounted(node) ? 0 : O_EXCL));
 877                 if (fd >= 0 || errno != EBUSY)
 878                         break;
 879                 duration.tv_sec = 0;
 880                 duration.tv_nsec = (100 * 1000 * 1000) + (rand() % 100 * 1000 * 1000);
 881                 nanosleep(&duration, NULL);
 882         }
 883         if (fd < 0) {
 884                 log_debug("unable to open '%s'", node);
 885                 rc = 1;
 886                 goto exit;
 887         }
 888         log_debug("probing: '%s'", node);
 889
 890         /* same data as original cdrom_id */
 891         if (cd_capability_compat(fd) < 0) {
 892                 rc = 1;
 893                 goto exit;
 894         }
 895
 896         /* check for media - don't bail if there's no media as we still need to
 897          * to read profiles */
 898         cd_media_compat(fd);
 899
 900         /* check if drive talks MMC */
 901         if (cd_inquiry(fd) < 0)
 902                 goto work;
 903
 904         /* read drive and possibly current profile */
 905         if (cd_profiles(fd) != 0)
 906                 goto work;
 907
 908         /* at this point we are guaranteed to have media in the drive - find out more about it */
 909
 910         /* get session/track info */
 911         cd_media_toc(fd);
 912
 913         /* get writable media state */
 914         cd_media_info(fd);
 915
 916 work:
 917         /* lock the media, so we enable eject button events */
 918         if (lock && cd_media) {
 919                 log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (lock)");
 920                 media_lock(fd, true);
 921         }
 922
 923         if (unlock && cd_media) {
 924                 log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)");
 925                 media_lock(fd, false);
 926         }
 927
 928         if (eject) {
 929                 log_debug("PREVENT_ALLOW_MEDIUM_REMOVAL (unlock)");
 930                 media_lock(fd, false);
 931                 log_debug("START_STOP_UNIT (eject)");
 932                 media_eject(fd);
 933         }
 934
 935         printf("ID_CDROM=1\n");
 936         if (cd_cd_rom)
 937                 printf("ID_CDROM_CD=1\n");
 938         if (cd_cd_r)
 939                 printf("ID_CDROM_CD_R=1\n");
 940         if (cd_cd_rw)
 941                 printf("ID_CDROM_CD_RW=1\n");
 942         if (cd_dvd_rom)
 943                 printf("ID_CDROM_DVD=1\n");
 944         if (cd_dvd_r)
 945                 printf("ID_CDROM_DVD_R=1\n");
 946         if (cd_dvd_rw)
 947                 printf("ID_CDROM_DVD_RW=1\n");
 948         if (cd_dvd_ram)
 949                 printf("ID_CDROM_DVD_RAM=1\n");
 950         if (cd_dvd_plus_r)
 951                 printf("ID_CDROM_DVD_PLUS_R=1\n");
 952         if (cd_dvd_plus_rw)
 953                 printf("ID_CDROM_DVD_PLUS_RW=1\n");
 954         if (cd_dvd_plus_r_dl)
 955                 printf("ID_CDROM_DVD_PLUS_R_DL=1\n");
 956         if (cd_dvd_plus_rw_dl)
 957                 printf("ID_CDROM_DVD_PLUS_RW_DL=1\n");
 958         if (cd_bd)
 959                 printf("ID_CDROM_BD=1\n");
 960         if (cd_bd_r)
 961                 printf("ID_CDROM_BD_R=1\n");
 962         if (cd_bd_re)
 963                 printf("ID_CDROM_BD_RE=1\n");
 964         if (cd_hddvd)
 965                 printf("ID_CDROM_HDDVD=1\n");
 966         if (cd_hddvd_r)
 967                 printf("ID_CDROM_HDDVD_R=1\n");
 968         if (cd_hddvd_rw)
 969                 printf("ID_CDROM_HDDVD_RW=1\n");
 970         if (cd_mo)
 971                 printf("ID_CDROM_MO=1\n");
 972         if (cd_mrw)
 973                 printf("ID_CDROM_MRW=1\n");
 974         if (cd_mrw_w)
 975                 printf("ID_CDROM_MRW_W=1\n");
 976
 977         if (cd_media)
 978                 printf("ID_CDROM_MEDIA=1\n");
 979         if (cd_media_mo)
 980                 printf("ID_CDROM_MEDIA_MO=1\n");
 981         if (cd_media_mrw)
 982                 printf("ID_CDROM_MEDIA_MRW=1\n");
 983         if (cd_media_mrw_w)
 984                 printf("ID_CDROM_MEDIA_MRW_W=1\n");
 985         if (cd_media_cd_rom)
 986                 printf("ID_CDROM_MEDIA_CD=1\n");
 987         if (cd_media_cd_r)
 988                 printf("ID_CDROM_MEDIA_CD_R=1\n");
 989         if (cd_media_cd_rw)
 990                 printf("ID_CDROM_MEDIA_CD_RW=1\n");
 991         if (cd_media_dvd_rom)
 992                 printf("ID_CDROM_MEDIA_DVD=1\n");
 993         if (cd_media_dvd_r)
 994                 printf("ID_CDROM_MEDIA_DVD_R=1\n");
 995         if (cd_media_dvd_ram)
 996                 printf("ID_CDROM_MEDIA_DVD_RAM=1\n");
 997         if (cd_media_dvd_rw)
 998                 printf("ID_CDROM_MEDIA_DVD_RW=1\n");
 999         if (cd_media_dvd_plus_r)
1000                 printf("ID_CDROM_MEDIA_DVD_PLUS_R=1\n");
1001         if (cd_media_dvd_plus_rw)
1002                 printf("ID_CDROM_MEDIA_DVD_PLUS_RW=1\n");
1003         if (cd_media_dvd_plus_rw_dl)
1004                 printf("ID_CDROM_MEDIA_DVD_PLUS_RW_DL=1\n");
1005         if (cd_media_dvd_plus_r_dl)
1006                 printf("ID_CDROM_MEDIA_DVD_PLUS_R_DL=1\n");
1007         if (cd_media_bd)
1008                 printf("ID_CDROM_MEDIA_BD=1\n");
1009         if (cd_media_bd_r)
1010                 printf("ID_CDROM_MEDIA_BD_R=1\n");
1011         if (cd_media_bd_re)
1012                 printf("ID_CDROM_MEDIA_BD_RE=1\n");
1013         if (cd_media_hddvd)
1014                 printf("ID_CDROM_MEDIA_HDDVD=1\n");
1015         if (cd_media_hddvd_r)
1016                 printf("ID_CDROM_MEDIA_HDDVD_R=1\n");
1017         if (cd_media_hddvd_rw)
1018                 printf("ID_CDROM_MEDIA_HDDVD_RW=1\n");
1019
1020         if (cd_media_state)
1021                 printf("ID_CDROM_MEDIA_STATE=%s\n", cd_media_state);
1022         if (cd_media_session_next > 0)
1023                 printf("ID_CDROM_MEDIA_SESSION_NEXT=%u\n", cd_media_session_next);
1024         if (cd_media_session_count > 0)
1025                 printf("ID_CDROM_MEDIA_SESSION_COUNT=%u\n", cd_media_session_count);
1026         if (cd_media_session_count > 1 && cd_media_session_last_offset > 0)
1027                 printf("ID_CDROM_MEDIA_SESSION_LAST_OFFSET=%llu\n", cd_media_session_last_offset);
1028         if (cd_media_track_count > 0)
1029                 printf("ID_CDROM_MEDIA_TRACK_COUNT=%u\n", cd_media_track_count);
1030         if (cd_media_track_count_audio > 0)
1031                 printf("ID_CDROM_MEDIA_TRACK_COUNT_AUDIO=%u\n", cd_media_track_count_audio);
1032         if (cd_media_track_count_data > 0)
1033                 printf("ID_CDROM_MEDIA_TRACK_COUNT_DATA=%u\n", cd_media_track_count_data);
1034 exit:
1035         if (fd >= 0)
1036                 close(fd);
1037         log_close();
1038         return rc;
1039 }