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