chiark / gitweb /
dvd-sector-copy.c: Eliminate `struct source'; just make the things global.
[dvdrip] / dvd-sector-copy.c
index 9c1c7d8ffaabdaf70bda567c60523a1ecd1fc285..6fec2deaaa863377d617ed3e97683f3a20be6341 100644 (file)
@@ -16,6 +16,7 @@
 #include <unistd.h>
 #include <fcntl.h>
 #include <sys/ioctl.h>
+#include <sys/stat.h>
 #include <sys/time.h>
 
 #include <getopt.h>
 #include <dvdread/ifo_read.h>
 #include <dvdread/ifo_types.h>
 
-#define SECTORSZ 2048
-#define SECTORS(n) (((n) + (SECTORSZ - 1))/SECTORSZ)
-
 #define CTYPE_HACK(fn, ch) fn((unsigned char)(ch))
 #define ISDIGIT(ch) CTYPE_HACK(isdigit, ch)
 #define ISSPACE(ch) CTYPE_HACK(isspace, ch)
 
 #define N(v) (sizeof(v)/sizeof((v)[0]))
 
-enum { RAW, IFO, VOB, BUP };
-typedef uint_least32_t ident;
-
-static inline ident mkident(unsigned kind, unsigned title, unsigned part)
-  { return (((ident)kind << 0) | ((ident)title << 8) | ((ident)part << 16)); }
-static inline unsigned id_kind(ident id) { return ((id >> 0)&0x0ff); }
-static inline unsigned id_title(ident id) { return ((id >> 8)&0x0ff); }
-static inline unsigned id_part(ident id) { return ((id >> 16)&0x0ff); }
+#define SECTORSZ 2048
+#define SECTORS(n) (((n) + (SECTORSZ - 1))/SECTORSZ)
 
 static const char *prog = "<unset>";
 static int status = 0;
@@ -58,6 +50,16 @@ static void usage(FILE *fp)
 
 static void vmoan(const char *fmt, va_list ap)
   { fprintf(stderr, "%s: ", prog); vfprintf(stderr, fmt, ap); }
+
+__attribute__((format(printf, 1, 2)))
+static void moan(const char *fmt, ...)
+{
+  va_list ap;
+
+  va_start(ap, fmt); vmoan(fmt, ap); va_end(ap);
+  fputc('\n', stderr);
+}
+
 __attribute__((noreturn, format(printf, 1, 2)))
 static void bail(const char *fmt, ...)
 {
@@ -67,6 +69,7 @@ static void bail(const char *fmt, ...)
   fputc('\n', stderr);
   exit(2);
 }
+
 __attribute__((noreturn, format(printf, 2, 3)))
 static void bail_syserr(int err, const char *fmt, ...)
 {
@@ -78,6 +81,32 @@ static void bail_syserr(int err, const char *fmt, ...)
   exit(2);
 }
 
+#define DEFVEC(vtype, etype)                                           \
+       typedef struct { etype *v; size_t n, sz; } vtype
+#define VEC_INIT { 0, 0, 0 }
+#define VEC_FREE(vv) do {                                              \
+  free((vv)->v); (vv)->v 0; (vv)->n = (vv)->sz = 0;                    \
+} while (0)
+#define VEC_PUSH(p, vv) do {                                           \
+  size_t _want;                                                                \
+  if ((vv)->n >= (vv)->sz) {                                           \
+    (vv)->sz = (vv)->sz ? 2*(vv)->sz : 32;                             \
+    _want = (vv)->sz*sizeof(*(vv)->v);                                 \
+    (vv)->v = realloc((vv)->v, _want);                                 \
+    if (!(vv)->v) bail("out of memory allocating %zu bytes", _want);   \
+  }                                                                    \
+  (p) = &(vv)->v[(vv)->n++];                                           \
+} while (0)
+
+enum { RAW, IFO, VOB, BUP };
+typedef uint_least32_t ident;
+
+static inline ident mkident(unsigned kind, unsigned title, unsigned part)
+  { return (((ident)kind << 0) | ((ident)title << 8) | ((ident)part << 16)); }
+static inline unsigned id_kind(ident id) { return ((id >> 0)&0x0ff); }
+static inline unsigned id_title(ident id) { return ((id >> 8)&0x0ff); }
+static inline unsigned id_part(ident id) { return ((id >> 16)&0x0ff); }
+
 #define MAXFNSZ (1 + 8 + 1 + 12 + 1)
 
 static void store_filename(char *buf, ident id)
@@ -104,39 +133,41 @@ static void store_filename(char *buf, ident id)
   }
 }
 
-#define DEFVEC(vtype, etype)                                           \
-       typedef struct { etype *v; size_t n, sz; } vtype
-#define VEC_INIT { 0, 0, 0 }
-#define VEC_FREE(vv) do {                                              \
-  free((vv)->v); (vv)->v 0; (vv)->n = (vv)->sz = 0;                    \
-} while (0)
-#define VEC_PUSH(p, vv) do {                                           \
-  size_t _want;                                                                \
-  if ((vv)->n >= (vv)->sz) {                                           \
-    (vv)->sz = (vv)->sz ? 2*(vv)->sz : 32;                             \
-    _want = (vv)->sz*sizeof(*(vv)->v);                                 \
-    (vv)->v = realloc((vv)->v, _want);                                 \
-    if (!(vv)->v) bail("out of memory allocating %zu bytes", _want);   \
-  }                                                                    \
-  (p) = &(vv)->v[(vv)->n++];                                           \
-} while (0)
+typedef uint_least32_t secaddr;
+#define PRIuSEC PRIuLEAST32
 
 #define MAXFILES (1 + 2*99 + 1)
 struct file {
   ident id;
-  uint32_t start, end;
+  secaddr start, end;
 };
 DEFVEC(file_v, struct file);
 static file_v filetab = VEC_INIT;
 
-enum { EV_WRITE, EV_BEGIN, EV_END, EV_STOP };
+enum { EV_STOP, EV_BEGIN, EV_END, EV_WRITE };
 struct event {
   unsigned char ev, file;
-  uint32_t pos;
+  secaddr pos;
 };
 DEFVEC(event_v, struct event);
 static event_v eventq = VEC_INIT;
 
+static int compare_event(const void *a, const void *b)
+{
+  const struct event *eva = a, *evb = b;
+
+  if (eva->pos < evb->pos) return (-1);
+  else if (eva->pos > evb->pos) return (+1);
+
+  if (eva->ev < evb->ev) return (-1);
+  else if (eva->ev > evb->ev) return (+1);
+
+  if (eva->file < evb->file) return (-1);
+  else if (eva->file > evb->file) return (+1);
+
+  return (0);
+}
+
 typedef uint_least32_t bits;
 static bits live[(MAXFILES + 31)/32];
 
@@ -164,7 +195,7 @@ found:
   return (i);
 }
 
-static void put_event(unsigned evtype, unsigned file, uint32_t pos)
+static void put_event(unsigned evtype, unsigned file, secaddr pos)
 {
   struct event *ev;
 
@@ -172,7 +203,7 @@ static void put_event(unsigned evtype, unsigned file, uint32_t pos)
   ev->ev = evtype; ev->file = file; ev->pos = pos;
 }
 
-static void put_file(ident id, uint32_t start, uint32_t end)
+static void put_file(ident id, secaddr start, secaddr end)
 {
   struct file *f;
   size_t i;
@@ -187,12 +218,12 @@ static void put_menu(dvd_reader_t *dvd, unsigned title)
 {
   ident id = mkident(VOB, title, 0);
   char fn[MAXFNSZ];
-  uint32_t start, len;
+  secaddr start, len;
 
   store_filename(fn, id);
   start = UDFFindFile(dvd, fn, &len); if (!start) return;
 #ifdef DEBUG
-      printf(";; %8"PRIu32" .. %-8"PRIu32": %s\n",
+      printf(";; %8"PRIuSEC" .. %-8"PRIuSEC": %s\n",
             start, start + SECTORS(len), fn);
 #endif
   put_file(id, start, start + SECTORS(len));
@@ -201,7 +232,7 @@ static void put_menu(dvd_reader_t *dvd, unsigned title)
 static void put_title(dvd_reader_t *dvd, unsigned title)
 {
   char fn[MAXFNSZ];
-  uint32_t start[9], len[9];
+  secaddr start[9], len[9];
   unsigned i, npart;
 
   for (i = 0; i < 9; i++) {
@@ -213,7 +244,7 @@ static void put_title(dvd_reader_t *dvd, unsigned title)
 #ifdef DEBUG
   for (i = 0; i < npart; i++) {
     store_filename(fn, mkident(VOB, title, i + 1));
-    printf(";; %8"PRIu32" .. %-8"PRIu32": %s\n",
+    printf(";; %8"PRIuSEC" .. %-8"PRIuSEC": %s\n",
           start[i], start[i] + SECTORS(len[i]), fn);
   }
 #endif
@@ -221,33 +252,18 @@ static void put_title(dvd_reader_t *dvd, unsigned title)
   if (npart > 1)
     for (i = 0; i < npart - 1; i++) {
       if (len[i]%SECTORSZ)
-       bail("title %u part %u length = %"PRIu32" not a multiple of %d",
+       bail("title %u part %u length = %"PRIuSEC" not a multiple of %d",
             title, i, len[i], SECTORSZ);
       if (start[i] + len[i]/SECTORSZ != start[i + 1])
-       bail("title %u part %u end = %"PRIu32" /= part %u start = %"PRIu32"",
-            title, i, start[i] + len[i]/SECTORSZ, i + 1, start[i + 1]);
+       bail
+         ("title %u part %u end = %"PRIuSEC" /= part %u start = %"PRIuSEC"",
+          title, i, start[i] + len[i]/SECTORSZ, i + 1, start[i + 1]);
     }
 
   put_file(mkident(VOB, title, 1),
           start[0], start[npart - 1] + SECTORS(len[npart - 1]));
 }
 
-static int compare_event(const void *a, const void *b)
-{
-  const struct event *eva = a, *evb = b;
-
-  if (eva->pos < evb->pos) return (-1);
-  else if (eva->pos > evb->pos) return (+1);
-
-  if (eva->ev < evb->ev) return (-1);
-  else if (eva->ev > evb->ev) return (+1);
-
-  if (eva->file < evb->file) return (-1);
-  else if (eva->file > evb->file) return (+1);
-
-  return (0);
-}
-
 static int progresslen = 0;
 
 static void clear_progress_internal(void)
@@ -276,47 +292,46 @@ static void print_progress(const char *fmt, ...)
   va_end(ap);
 }
 
-struct source {
-  dvd_reader_t *dvd;
-  int dvdfd;
-  struct file *file;
-  dvd_file_t *vob;
+static dvd_reader_t *dvd;
+static int dvdfd = -1;
+static struct file *file;
+static dvd_file_t *vob;
 
-  uint32_t last_pos, limit, nsectors, ndone;
-  struct timeval last_time;
-  double wsum, wcount;
-  const char *mapfile; FILE *mapfp;
-};
-#define SOURCE_INIT { 0, -1, 0, 0,  0, 0, 0, 0, { 0, 0 }, 0.0, 0.0, 0, 0 }
+unsigned flags;
+#  define F_ALLPROGRESS 1u
+static secaddr last_pos, limit, nsectors, ndone;
+static struct timeval last_time;
+static double wsum, wcount;
+static const char *mapfile; static FILE *mapfp;
 
-static void report_progress(struct source *src, uint32_t pos)
+static void report_progress(secaddr pos)
 {
   char etastr[32];
   struct timeval now;
   int eta;
-  double t, f, g, rate;
+  double percent, t, f, g, rate;
   char *unit;
 
 #define ALPHA 0.02
 #define BETA (1 - ALPHA)
 
   gettimeofday(&now, 0);
-  t = (now.tv_sec - src->last_time.tv_sec) +
-    (now.tv_usec - src->last_time.tv_usec)/1000000.0;
+  t = (now.tv_sec - last_time.tv_sec) +
+    (now.tv_usec - last_time.tv_usec)/1000000.0;
 
   if (t) {
-    g = src->wcount ? pow(BETA, t) : 0.0; f = (1 - g)/(1 - BETA);
-    src->wsum = f*(pos - src->last_pos)/t + g*src->wsum;
-    src->wcount = f + g*src->wcount;
-    src->ndone += pos - src->last_pos;
-    src->last_time = now; src->last_pos = pos;
+    g = wcount ? pow(BETA, t) : 0.0; f = (1 - g)/(1 - BETA);
+    wsum = f*(pos - last_pos)/t + g*wsum;
+    wcount = f + g*wcount;
+    ndone += pos - last_pos;
+    last_time = now; last_pos = pos;
   }
 
-  if (!src->wsum || !src->wcount)
+  if (!wsum || !wcount)
     { rate = 0; strcpy(etastr, "???"); }
   else {
-    rate = src->wsum/src->wcount;
-    eta = (int)((src->nsectors - src->ndone)/rate);
+    rate = wsum/wcount;
+    eta = (int)((nsectors - ndone)/rate);
     sprintf(etastr, "%d:%02d:%02d", eta/3600, (eta/60)%60, eta%60);
   }
 
@@ -325,46 +340,46 @@ static void report_progress(struct source *src, uint32_t pos)
   if (rate > 128) { rate /= 1024; unit = "M"; }
   if (rate > 128) { rate /= 1024; unit = "G"; }
 
-  print_progress("copied %.1f%% (%"PRIu32" of %"PRIu32"; %.1f %sB/s, ETA %s)",
-                src->ndone*100.0/src->nsectors, pos, src->limit,
-                rate, unit, etastr);
-  if (src->file && id_kind(src->file->id) == VOB) {
+  if (flags&F_ALLPROGRESS) percent = pos*100.0/limit;
+  else percent = ndone*100.0/nsectors;
+  print_progress
+    ("copied %.1f%% (%"PRIuSEC" of %"PRIuSEC"; %.1f %sB/s, ETA %s)",
+     percent, pos, limit,  rate, unit, etastr);
+  if (file && id_kind(file->id) == VOB) {
     append_progress(" -- %s %d %3.1f%%",
-                   id_part(src->file->id) ? "title" : "menu",
-                   id_title(src->file->id),
-                   (pos - src->file->start)*100.0/
-                     (src->file->end - src->file->start));
+                   id_part(file->id) ? "title" : "menu",
+                   id_title(file->id),
+                   (pos - file->start)*100.0/
+                     (file->end - file->start));
   }
 
 #undef ALPHA
 #undef BETA
 }
 
-static void report_bad_blocks_progress(struct source *src,
-                                      uint32_t lo, uint32_t hi, int err)
+static void report_bad_blocks_progress(secaddr lo, secaddr hi, int err)
 {
-  report_progress(src, hi);
+  report_progress(hi);
 
   if (lo == hi) append_progress(": retrying bad sector");
   else
-    append_progress(": %"PRIu32" bad %s",
+    append_progress(": %"PRIuSEC" bad %s",
                    hi - lo, hi == lo + 1 ? "sector" : "sectors");
   if (err != EIO) append_progress(" (%s)", strerror(err));
   fflush(stdout);
 }
 
-static ssize_t read_sectors(struct source *src, uint32_t pos,
-                           void *buf, uint32_t want)
+static ssize_t read_sectors(secaddr pos, void *buf, secaddr want)
 {
   ssize_t n;
 
 again:
-  if (src->vob)
-    n = DVDReadBlocks(src->vob, pos - src->file->start, want, buf);
-  else if (src->file) {
-    if (lseek(src->dvdfd, (off_t)pos*SECTORSZ, SEEK_SET) < 0)
-      bail_syserr(errno, "failed to seek to sector %"PRIu32"", pos);
-    n = read(src->dvdfd, buf, want*SECTORSZ);
+  if (vob)
+    n = DVDReadBlocks(vob, pos - file->start, want, buf);
+  else if (file) {
+    if (lseek(dvdfd, (off_t)pos*SECTORSZ, SEEK_SET) < 0)
+      bail_syserr(errno, "failed to seek to sector %"PRIuSEC"", pos);
+    n = read(dvdfd, buf, want*SECTORSZ);
     if (n >= 0) n /= SECTORSZ;
   } else {
     memset(buf, 0, want*SECTORSZ);
@@ -392,14 +407,14 @@ static void carefully_write(int fd, const void *buf, size_t sz)
   }
 }
 
-static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
+static void emit(int outfd, secaddr start, secaddr end)
 {
 #define BUFSECTORS 512
 
   int least, i;
   unsigned char buf[BUFSECTORS*SECTORSZ];
-  uint32_t pos;
-  uint32_t bad_lo, bad_hi, good, step;
+  secaddr pos;
+  secaddr bad_lo, bad_hi, good, step;
   size_t want;
   ssize_t n;
   static int first_time = 1;
@@ -412,13 +427,13 @@ static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
   least = least_live();
 
 #ifdef DEBUG
-  printf(";; %8"PRIu32" .. %"PRIu32"\n", start, end);
+  printf(";; %8"PRIuSEC" .. %"PRIuSEC"\n", start, end);
 
   for (i = 0; i < filetab.n; i++) {
     if (!livep(i)) continue;
     if (act == -1) act = i;
     f = &filetab.v[i]; store_filename(fn, f->id);
-    printf(";;\t\t%8"PRIu32" .. %-8"PRIu32" %s\n",
+    printf(";;\t\t%8"PRIuSEC" .. %-8"PRIuSEC" %s\n",
           start - f->start, end - f->start, fn);
   }
   if (act == -1) printf(";;\t\t#<no live source>\n");
@@ -426,23 +441,23 @@ static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
 #endif
 
   if (least == -1)
-    { src->file = 0; src->vob = 0; }
+    { file = 0; vob = 0; }
   else {
-    src->file = &filetab.v[least];
-    switch (id_kind(src->file->id)) {
+    file = &filetab.v[least];
+    switch (id_kind(file->id)) {
       case RAW:
-       src->vob = 0;
+       vob = 0;
        break;
       case VOB:
        if (first_time) { clear_progress(); first_time = 0; }
-       src->vob = DVDOpenFile(src->dvd, id_title(src->file->id),
-                              id_part(src->file->id)
-                                ? DVD_READ_TITLE_VOBS
-                                : DVD_READ_MENU_VOBS);
-       if (!src->vob)
+       vob = DVDOpenFile(dvd, id_title(file->id),
+                         id_part(file->id)
+                           ? DVD_READ_TITLE_VOBS
+                           : DVD_READ_MENU_VOBS);
+       if (!vob)
          bail("failed to open %s %u",
-              id_part(src->file->id) ? "title" : "menu",
-              id_title(src->file->id));
+              id_part(file->id) ? "title" : "menu",
+              id_title(file->id));
        break;
       default:
        abort();
@@ -452,16 +467,15 @@ static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
   pos = start;
   while (pos < end) {
     want = end - pos; if (want > BUFSECTORS) want = BUFSECTORS;
-    n = read_sectors(src, pos, buf, want);
+    n = read_sectors(pos, buf, want);
 
     if (n <= 0) {
-      report_bad_blocks_progress(src, pos, pos, errno);
+      report_bad_blocks_progress(pos, pos, errno);
       for (i = 0; i < 4; i++) {
-       n = read_sectors(src, pos, buf, 1);
+       n = read_sectors(pos, buf, 1);
        if (n > 0) {
          clear_progress();
-         fprintf(stderr, "%s: sector %"PRIu32" read ok after retry\n",
-                 prog, pos);
+         moan("sector %"PRIuSEC" read ok after retry", pos);
          bad_lo = bad_hi = pos;
          goto recovered;
        }
@@ -469,45 +483,44 @@ static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
 
       bad_lo = pos; step = 1; bad_hi = pos + 1;
       for (;;) {
-       report_bad_blocks_progress(src, bad_lo, bad_hi, errno);
+       report_bad_blocks_progress(bad_lo, bad_hi, errno);
        if (bad_hi >= end) {
          clear_progress();
-         fprintf(stderr, "%s: giving up on this extent\n", prog);
+         moan("giving up on this extent");
          n = 0; goto recovered;
        }
        step *= 2;
        if (step > end - bad_lo) step = end - bad_lo;
        pos = bad_lo + step - 1;
-       n = read_sectors(src, pos, buf, 1);
+       n = read_sectors(pos, buf, 1);
        if (n > 0) break;
        bad_hi = pos + 1;
       }
 
       good = pos;
       while (good > bad_hi) {
-       report_bad_blocks_progress(src, bad_lo, bad_hi, errno);
+       report_bad_blocks_progress(bad_lo, bad_hi, errno);
        pos = bad_hi + (good - bad_hi)/2;
-       n = read_sectors(src, pos, buf, 1);
+       n = read_sectors(pos, buf, 1);
        if (n > 0) good = pos;
        else bad_hi = pos + 1;
       }
     recovered:
       if (bad_hi > bad_lo) {
        clear_progress();
-       fprintf(stderr, "%s: skipping %"PRIu32" bad sectors "
-               "(%"PRIu32" .. %"PRIu32")\n",
-               prog, bad_hi - bad_lo, bad_lo, bad_hi);
-       if (src->mapfile) {
-         if (!src->mapfp) {
-           src->mapfp = fopen(src->mapfile, "w");
-           if (!src->mapfp)
+       moan("skipping %"PRIuSEC" bad sectors (%"PRIuSEC" .. %"PRIuSEC")",
+            bad_hi - bad_lo, bad_lo, bad_hi);
+       if (mapfile) {
+         if (!mapfp) {
+           mapfp = fopen(mapfile, "w");
+           if (!mapfp)
              bail_syserr(errno, "failed to open bad-sector map file `%s'",
                          optarg);
-           fprintf(src->mapfp, "## bad sector map\n\n");
+           fprintf(mapfp, "## bad sector map\n\n");
          }
-         fprintf(src->mapfp, "%"PRIu32" %"PRIu32"\n", bad_lo, bad_hi);
-         fflush(src->mapfp);
-         if (ferror(src->mapfp))
+         fprintf(mapfp, "%"PRIuSEC" %"PRIuSEC"\n", bad_lo, bad_hi);
+         fflush(mapfp);
+         if (ferror(mapfp))
            bail_syserr(errno, "error writing bad-sector map file");
        }
        if (outfd >= 0 &&
@@ -519,10 +532,10 @@ static void emit(struct source *src, int outfd, uint32_t start, uint32_t end)
     }
 
     if (n > 0) { carefully_write(outfd, buf, n*SECTORSZ); pos += n; }
-    report_progress(src, pos); fflush(stdout);
+    report_progress(pos); fflush(stdout);
   }
 
-  if (src->vob) { DVDCloseFile(src->vob); src->vob = 0; }
+  if (vob) { DVDCloseFile(vob); vob = 0; }
 
 #undef BUFSECTORS
 }
@@ -587,16 +600,17 @@ int main(int argc, char *argv[])
   unsigned f = 0;
   char *p;
   uint64_t volsz;
-  uint32_t pos;
+  secaddr pos;
   off_t off;
-  struct source src = SOURCE_INIT;
   unsigned long start, end;
   const struct event *ev;
   const char *device = "/dev/dvd", *outfile = 0;
   int opt, err, outfd = -1, blksz;
+  unsigned n;
   size_t i;
   FILE *fp;
   struct buf buf = BUF_INIT;
+  struct stat st;
 #ifdef DEBUG
   const struct file *file;
   char fn[MAXFNSZ];
@@ -645,8 +659,8 @@ int main(int argc, char *argv[])
       bad_range_file:
        bail("bad range `%s' at `%s' line %zu", buf.p, optarg, i);
       case 'b':
-       if (src.mapfile) bail("can't have multiple map files");
-       src.mapfile = optarg;
+       if (mapfile) bail("can't have multiple map files");
+       mapfile = optarg;
        break;
       case 'c': f |= f_continue; break;
       case 'o': outfile = optarg; break;
@@ -684,12 +698,21 @@ int main(int argc, char *argv[])
   if (optind < argc) f |= f_bogus;
   if (f&f_bogus) { usage(stderr); exit(2); }
 
-  src.dvdfd = open(device, O_RDONLY);
-  if (src.dvdfd < 0) bail_syserr(errno, "failed to open device `%s'", device);
-  if (ioctl(src.dvdfd, BLKSSZGET, &blksz))
-    bail_syserr(errno, "failed to get block size for `%s'", device);
-  if (ioctl(src.dvdfd, BLKGETSIZE64, &volsz))
-    bail_syserr(errno, "failed to get volume size for `%s'", device);
+  dvdfd = open(device, O_RDONLY);
+  if (dvdfd < 0)
+    bail_syserr(errno, "failed to open device `%s'", device);
+  if (fstat(dvdfd, &st))
+    bail_syserr(errno, "failed to stat device `%s'", device);
+  if (S_ISREG(st.st_mode)) {
+    blksz = SECTORSZ;
+    volsz = st.st_size;
+  } else if (S_ISBLK(st.st_mode)) {
+    if (ioctl(dvdfd, BLKSSZGET, &blksz))
+      bail_syserr(errno, "failed to get block size for `%s'", device);
+    if (ioctl(dvdfd, BLKGETSIZE64, &volsz))
+      bail_syserr(errno, "failed to get volume size for `%s'", device);
+  } else
+    bail("can't use `%s' as source: expected file or block device", device);
 
   if (blksz != SECTORSZ)
     bail("device `%s' block size %d /= %d", device, blksz, SECTORSZ);
@@ -714,46 +737,46 @@ int main(int argc, char *argv[])
     put_event(EV_WRITE, 0, 0);
 
 #ifdef notdef
-  src.dvd = DVDOpen2(0, &logger, device);
+  dvd = DVDOpen2(0, &logger, device);
 #else
-  src.dvd = DVDOpen(device);
+  dvd = DVDOpen(device);
 #endif
-  if (!src.dvd) bail("failed to open DVD on `%s'", device);
+  if (!dvd) bail("failed to open DVD on `%s'", device);
 
   /* It's fast enough just to check everything. */
-  put_menu(src.dvd, 0);
+  put_menu(dvd, 0);
   for (i = 1; i < 100; i++) {
-    put_menu(src.dvd, i);
-    put_title(src.dvd, i);
+    put_menu(dvd, i);
+    put_title(dvd, i);
   }
   put_file(mkident(RAW, 0, 0), 0, volsz/SECTORSZ);
   assert(filetab.n <= MAXFILES);
 
-  for (i = 0, src.limit = 0; i < filetab.n; i++)
-    if (filetab.v[i].end > src.limit) src.limit = filetab.v[i].end;
+  for (i = 0, limit = 0; i < filetab.n; i++)
+    if (filetab.v[i].end > limit) limit = filetab.v[i].end;
 
-  if (end > src.limit) end = src.limit;
+  if (end > limit) end = limit;
 
 #ifdef DEBUG
   printf("\n;; files:\n");
   for (i = 0; i < filetab.n; i++) {
     file = &filetab.v[i];
     store_filename(fn, file->id);
-    printf(";;\t%8"PRIu32" %s\n", file->start, fn);
+    printf(";;\t%8"PRIuSEC" %s\n", file->start, fn);
   }
 #endif
 
   qsort(eventq.v, eventq.n, sizeof(struct event), compare_event);
 
-  f &= ~f_write; start = 0;
+  f &= ~f_write; start = 0; n = 0;
   for (i = 0; i < eventq.n; i++) {
     ev = &eventq.v[i];
     switch (ev->ev) {
       case EV_WRITE:
        if (f&f_write)
-         bail("overlapping ranges: range from %lu still open at %"PRIu32"",
+         bail("overlapping ranges: range from %lu still open at %"PRIuSEC"",
               start, ev->pos);
-       f |= f_write; start = ev->pos;
+       n++; f |= f_write; start = ev->pos;
        break;
       case EV_STOP:
        f &= ~f_write;
@@ -766,20 +789,21 @@ int main(int argc, char *argv[])
     ev = &eventq.v[i];
     switch (ev->ev) {
       case EV_WRITE: start = ev->pos; f |= f_write; break;
-      case EV_STOP: src.nsectors += ev->pos - start; f &= ~f_write; break;
+      case EV_STOP: nsectors += ev->pos - start; f &= ~f_write; break;
     }
-    if (ev->pos >= src.limit) break;
+    if (ev->pos >= limit) break;
     if (f&f_fixup) start = ev->pos;
   }
   eventq.n = i;
   if (f&f_fixup) {
     put_event(EV_WRITE, 0, start);
-    f |= f_write;
+    n++; f |= f_write;
   }
   if (f&f_write) {
-    src.nsectors += src.limit - start;
-    put_event(EV_STOP, 0, src.limit);
+    nsectors += limit - start;
+    put_event(EV_STOP, 0, limit);
   }
+  if (n == 1 && (f&f_write)) flags |= F_ALLPROGRESS;
   f &= ~f_write;
 
 #ifdef DEBUG
@@ -788,7 +812,7 @@ int main(int argc, char *argv[])
   for (pos = 0, i = 0; i < eventq.n; i++) {
     ev = &eventq.v[i];
     if (ev->pos > pos) {
-      if (f&f_write) emit(&src, outfd, pos, ev->pos);
+      if (f&f_write) emit(outfd, pos, ev->pos);
       pos = ev->pos;
 #ifdef DEBUG
       clear_progress();
@@ -801,20 +825,20 @@ int main(int argc, char *argv[])
 #ifdef DEBUG
        store_filename(fn, filetab.v[ev->file].id);
        clear_progress();
-       printf(";; %8"PRIu32": begin `%s'\n", pos, fn);
+       printf(";; %8"PRIuSEC": begin `%s'\n", pos, fn);
 #endif
        break;
       case EV_WRITE:
-       gettimeofday(&src.last_time, 0); src.last_pos = pos;
+       gettimeofday(&last_time, 0); last_pos = pos;
        if (outfd >= 0 &&
            lseek(outfd, (off_t)ev->pos*SECTORSZ, SEEK_SET) < 0)
          bail_syserr(errno,
                      "failed to seek to resume position "
-                     "(sector %"PRIu32") in output file `%s'",
+                     "(sector %"PRIuSEC") in output file `%s'",
                      ev->pos, outfile);
 #ifdef DEBUG
        clear_progress();
-       printf(";; %8"PRIu32": begin write\n", pos);
+       printf(";; %8"PRIuSEC": begin write\n", pos);
 #endif
        f |= f_write;
        break;
@@ -822,7 +846,7 @@ int main(int argc, char *argv[])
        f &= ~f_write;
 #ifdef DEBUG
        clear_progress();
-       printf(";; %8"PRIu32": end write\n", pos);
+       printf(";; %8"PRIuSEC": end write\n", pos);
 #endif
        break;
       case EV_END:
@@ -830,7 +854,7 @@ int main(int argc, char *argv[])
 #ifdef DEBUG
        store_filename(fn, filetab.v[ev->file].id);
        clear_progress();
-       printf(";; %8"PRIu32": end `%s'\n", pos, fn);
+       printf(";; %8"PRIuSEC": end `%s'\n", pos, fn);
 #endif
        break;
       default: abort();
@@ -839,14 +863,14 @@ int main(int argc, char *argv[])
 
   if (progresslen) putchar('\n');
 
-  if (outfd >= 0 && ftruncate(outfd, (off_t)src.limit*SECTORSZ) < 0)
+  if (outfd >= 0 && ftruncate(outfd, (off_t)limit*SECTORSZ) < 0)
     bail_syserr(errno, "failed to set output file `%s' length", outfile);
 
-  if (src.dvd) DVDClose(src.dvd);
-  if (src.dvdfd >= 0) close(src.dvdfd);
+  if (dvd) DVDClose(dvd);
+  if (dvdfd >= 0) close(dvdfd);
   if (outfd >= 0) close(outfd);
-  if (src.mapfp) {
-    if (ferror(src.mapfp) || fclose(src.mapfp))
+  if (mapfp) {
+    if (ferror(mapfp) || fclose(mapfp))
       bail_syserr(errno, "error writing bad-sector map file");
   }