src/udev/cdrom_id/cdrom_id.c

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