d663c0b00b34ef146556d9e5a57d4a87dd812fbb
2 #define _FILE_OFFSET_BITS 64
18 #include <sys/ioctl.h>
26 #include <dvdread/dvd_reader.h>
27 #include <dvdread/dvd_udf.h>
28 #include <dvdread/ifo_read.h>
29 #include <dvdread/ifo_types.h>
31 #define CTYPE_HACK(fn, ch) fn((unsigned char)(ch))
32 #define ISDIGIT(ch) CTYPE_HACK(isdigit, ch)
33 #define ISSPACE(ch) CTYPE_HACK(isspace, ch)
41 #define N(v) (sizeof(v)/sizeof((v)[0]))
44 #define SECTORS(n) (((n) + (SECTORSZ - 1))/SECTORSZ)
46 static const char *prog
= "<unset>";
47 static int status
= 0;
49 static void usage(FILE *fp
)
52 "usage: %s [-c] [-R MAP] [-b OUTMAP] [-r [START]-[END]]\n"
57 static void vmoan(const char *fmt
, va_list ap
)
58 { fprintf(stderr
, "%s: ", prog
); vfprintf(stderr
, fmt
, ap
); }
60 __attribute__((format(printf
, 1, 2)))
61 static void moan(const char *fmt
, ...)
65 va_start(ap
, fmt
); vmoan(fmt
, ap
); va_end(ap
);
69 __attribute__((noreturn
, format(printf
, 1, 2)))
70 static void bail(const char *fmt
, ...)
74 va_start(ap
, fmt
); vmoan(fmt
, ap
); va_end(ap
);
79 __attribute__((noreturn
, format(printf
, 2, 3)))
80 static void bail_syserr(int err
, const char *fmt
, ...)
84 va_start(ap
, fmt
); vmoan(fmt
, ap
); va_end(ap
);
85 if (err
) fprintf(stderr
, ": %s", strerror(errno
));
90 static void carefully_write(int fd
, const void *buf
, size_t sz
)
92 const unsigned char *p
= buf
;
99 if (errno
== EINTR
) continue;
100 bail_syserr(errno
, "failed to write to output file");
102 if (!n
) bail("unexpected short write to output file");
107 #define DEFVEC(vtype, etype) \
108 typedef struct { etype *v; size_t n, sz; } vtype
109 #define VEC_INIT { 0, 0, 0 }
110 #define VEC_FREE(vv) do { \
111 free((vv)->v); (vv)->v 0; (vv)->n = (vv)->sz = 0; \
113 #define VEC_PUSH(p, vv) do { \
115 if ((vv)->n >= (vv)->sz) { \
116 (vv)->sz = (vv)->sz ? 2*(vv)->sz : 32; \
117 _want = (vv)->sz*sizeof(*(vv)->v); \
118 (vv)->v = realloc((vv)->v, _want); \
119 if (!(vv)->v) bail("out of memory allocating %zu bytes", _want); \
121 (p) = &(vv)->v[(vv)->n++]; \
124 enum { RAW
, IFO
, VOB
, BUP
};
125 typedef uint_least32_t ident
;
127 static inline ident
mkident(unsigned kind
, unsigned title
, unsigned part
)
128 { return (((ident
)kind
<< 0) | ((ident
)title
<< 8) | ((ident
)part
<< 16)); }
129 static inline unsigned id_kind(ident id
) { return ((id
>> 0)&0x0ff); }
130 static inline unsigned id_title(ident id
) { return ((id
>> 8)&0x0ff); }
131 static inline unsigned id_part(ident id
) { return ((id
>> 16)&0x0ff); }
133 #define MAXFNSZ (1 + 8 + 1 + 12 + 1)
135 static void store_filename(char *buf
, ident id
)
137 switch (id_kind(id
)) {
139 sprintf(buf
, "#<raw device>");
142 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.IFO");
143 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.IFO", id_title(id
));
146 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.BUP");
147 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.BUP", id_title(id
));
150 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.VOB");
152 sprintf(buf
, "/VIDEO_TS/VTS_%02u_%u.VOB", id_title(id
), id_part(id
));
159 typedef uint_least32_t secaddr
;
160 #define PRIuSEC PRIuLEAST32
161 #define SECLIMIT 0x00400000
163 #define MAXFILES (1 + 2*99 + 1)
168 DEFVEC(file_v
, struct file
);
169 static file_v filetab
= VEC_INIT
;
171 enum { EV_STOP
, EV_BEGIN
, EV_END
, EV_WRITE
};
173 unsigned char ev
, file
;
176 DEFVEC(event_v
, struct event
);
177 static event_v eventq
= VEC_INIT
;
179 static int compare_event(const void *a
, const void *b
)
181 const struct event
*eva
= a
, *evb
= b
;
183 if (eva
->pos
< evb
->pos
) return (-1);
184 else if (eva
->pos
> evb
->pos
) return (+1);
186 if (eva
->ev
< evb
->ev
) return (-1);
187 else if (eva
->ev
> evb
->ev
) return (+1);
189 if (eva
->file
< evb
->file
) return (-1);
190 else if (eva
->file
> evb
->file
) return (+1);
195 typedef uint_least32_t bits
;
196 static bits live
[(MAXFILES
+ 31)/32];
198 static inline int livep(unsigned i
)
199 { return (live
[i
/32]&((bits
)1 << (i
%32))); }
200 static inline void set_live(unsigned i
)
201 { live
[i
/32] |= (bits
)1 << (i
%32); }
202 static inline void clear_live(unsigned i
)
203 { live
[i
/32] &= ~((bits
)1 << (i
%32)); }
204 static inline int least_live(void)
206 unsigned i
, n
= (filetab
.n
+ 32)/32;
209 for (i
= 0; i
< n
; i
++) { b
= live
[i
]; if (b
) goto found
; }
213 if (!(b
&0x0000ffff)) { b
>>= 16; i
+= 16; }
214 if (!(b
&0x000000ff)) { b
>>= 8; i
+= 8; }
215 if (!(b
&0x0000000f)) { b
>>= 4; i
+= 4; }
216 if (!(b
&0x00000003)) { b
>>= 2; i
+= 2; }
217 if (!(b
&0x00000001)) { b
>>= 1; i
+= 1; }
222 static void put_event(unsigned evtype
, unsigned file
, secaddr pos
)
226 VEC_PUSH(ev
, &eventq
);
227 ev
->ev
= evtype
; ev
->file
= file
; ev
->pos
= pos
;
230 static void put_file(ident id
, secaddr start
, secaddr end
)
235 VEC_PUSH(f
, &filetab
); i
= f
- filetab
.v
;
236 f
->id
= id
; f
->start
= start
; f
->end
= end
;
237 put_event(EV_BEGIN
, i
, start
);
238 put_event(EV_END
, i
, end
);
241 static void put_menu(dvd_reader_t
*dvd
, unsigned title
)
243 ident id
= mkident(VOB
, title
, 0);
247 store_filename(fn
, id
);
248 start
= UDFFindFile(dvd
, fn
, &len
); if (!start
) return;
250 printf(";; %8"PRIuSEC
" .. %-8"PRIuSEC
": %s\n",
251 start
, start
+ SECTORS(len
), fn
);
253 put_file(id
, start
, start
+ SECTORS(len
));
256 static void put_title(dvd_reader_t
*dvd
, unsigned title
)
259 secaddr start
[9], len
[9];
262 for (i
= 0; i
< 9; i
++) {
263 store_filename(fn
, mkident(VOB
, title
, i
+ 1));
264 start
[i
] = UDFFindFile(dvd
, fn
, &len
[i
]); if (!start
[i
]) break;
266 npart
= i
; if (!npart
) return;
269 for (i
= 0; i
< npart
; i
++) {
270 store_filename(fn
, mkident(VOB
, title
, i
+ 1));
271 printf(";; %8"PRIuSEC
" .. %-8"PRIuSEC
": %s\n",
272 start
[i
], start
[i
] + SECTORS(len
[i
]), fn
);
277 for (i
= 0; i
< npart
- 1; i
++) {
279 bail("title %u part %u length = %"PRIuSEC
" not a multiple of %d",
280 title
, i
, len
[i
], SECTORSZ
);
281 if (start
[i
] + len
[i
]/SECTORSZ
!= start
[i
+ 1])
283 ("title %u part %u end = %"PRIuSEC
" /= part %u start = %"PRIuSEC
"",
284 title
, i
, start
[i
] + len
[i
]/SECTORSZ
, i
+ 1, start
[i
+ 1]);
287 put_file(mkident(VOB
, title
, 1),
288 start
[0], start
[npart
- 1] + SECTORS(len
[npart
- 1]));
291 static int progresslen
= 0;
293 static void clear_progress_internal(void)
295 while (progresslen
) { fputs("\b \b", stdout
); progresslen
--; }
298 static void clear_progress(void)
299 { clear_progress_internal(); fflush(stdout
); }
300 static void vappend_progress(const char *fmt
, va_list ap
)
301 { progresslen
+= vprintf(fmt
, ap
); }
302 __attribute__((format(printf
, 1, 2)))
303 static void append_progress(const char *fmt
, ...)
308 vappend_progress(fmt
, ap
);
311 __attribute__((format(printf
, 1, 2)))
312 static void print_progress(const char *fmt
, ...)
317 clear_progress_internal();
318 vappend_progress(fmt
, ap
);
323 # define F_ALLPROGRESS 1u
324 static secaddr last_pos
, limit
, nsectors
, ndone
;
325 static struct timeval last_time
;
326 static double wsum
, wcount
;
327 static struct file
*file
;
329 static void report_progress(secaddr pos
)
334 double percent
, t
, f
, g
, rate
;
338 #define BETA (1 - ALPHA)
340 gettimeofday(&now
, 0);
341 t
= (now
.tv_sec
- last_time
.tv_sec
) +
342 (now
.tv_usec
- last_time
.tv_usec
)/1000000.0;
345 g
= wcount ?
pow(BETA
, t
) : 0.0; f
= (1 - g
)/(1 - BETA
);
346 wsum
= f
*(pos
- last_pos
)/t
+ g
*wsum
;
347 wcount
= f
+ g
*wcount
;
348 ndone
+= pos
- last_pos
;
349 last_time
= now
; last_pos
= pos
;
352 if (!wsum
|| !wcount
)
353 { rate
= 0; strcpy(etastr
, "???"); }
356 eta
= (int)((nsectors
- ndone
)/rate
);
357 sprintf(etastr
, "%d:%02d:%02d", eta
/3600, (eta
/60)%60, eta
%60);
360 rate
*= SECTORSZ
; unit
= "";
361 if (rate
> 128) { rate
/= 1024; unit
= "k"; }
362 if (rate
> 128) { rate
/= 1024; unit
= "M"; }
363 if (rate
> 128) { rate
/= 1024; unit
= "G"; }
365 if (flags
&F_ALLPROGRESS
) percent
= pos
*100.0/limit
;
366 else percent
= ndone
*100.0/nsectors
;
368 ("copied %.1f%% (%"PRIuSEC
" of %"PRIuSEC
"; %.1f %sB/s, ETA %s)",
369 percent
, pos
, limit
, rate
, unit
, etastr
);
370 if (file
&& id_kind(file
->id
) == VOB
) {
371 append_progress(" -- %s %d %3.1f%%",
372 id_part(file
->id
) ?
"title" : "menu",
374 (pos
- file
->start
)*100.0/
375 (file
->end
- file
->start
));
382 static dvd_reader_t
*dvd
;
383 static int dvdfd
= -1, outfd
= -1;
384 static dvd_file_t
*vob
;
385 static const char *mapfile
; static FILE *mapfp
;
387 struct badblock
{ secaddr start
, end
; };
388 DEFVEC(badblock_v
, struct badblock
);
389 static badblock_v badblocks
= VEC_INIT
;
391 static int compare_badblock(const void *a
, const void *b
)
393 const struct badblock
*ba
= a
, *bb
= b
;
395 if (ba
->start
< bb
->start
) return (-1);
396 else if (ba
->start
> bb
->start
) return (+1);
398 if (ba
->end
< bb
->end
) return (-1);
399 else if (ba
->end
> bb
->end
) return (+1);
404 static ssize_t
read_sectors(secaddr pos
, void *buf
, secaddr want
)
409 struct badblock
*bad
, *best
;
410 unsigned char *p
= buf
;
412 best
= 0; lo
= 0; hi
= badblocks
.n
;
415 printf(";; searching badblocks for %"PRIuSEC
" .. %"PRIuSEC
"\n",
419 mid
= lo
+ (hi
- lo
)/2; bad
= &badblocks
.v
[mid
];
421 printf(";; try %zu (%"PRIuSEC
" .. %"PRIuSEC
")... ",
422 mid
, bad
->start
, bad
->end
);
424 if (pos
< bad
->start
) { D( printf("high\n"); ) best
= bad
; hi
= mid
; }
425 else if (pos
>= bad
->end
) { D( printf("low\n"); ) lo
= mid
+ 1; }
426 else { D( printf("match!\n"); ) errno
= EIO
; return (-1); }
430 printf(";; next is %"PRIuSEC
" .. %"PRIuSEC
"\n",
431 best
->start
, best
->end
);
433 if (best
&& pos
+ want
> best
->start
)
434 { want
= best
->start
- pos
; fakeerr
= EIO
; }
439 { errno
= 0; n
= DVDReadBlocks(vob
, pos
- file
->start
, want
, p
); }
441 if (lseek(dvdfd
, (off_t
)pos
*SECTORSZ
, SEEK_SET
) < 0)
442 bail_syserr(errno
, "failed to seek to sector %"PRIuSEC
"", pos
);
443 errno
= 0; n
= read(dvdfd
, p
, want
*SECTORSZ
);
444 if (n
>= 0) n
/= SECTORSZ
;
446 memset(p
, 0, want
*SECTORSZ
);
450 if (n
> 0) { done
+= n
; pos
+= n
; p
+= n
*SECTORSZ
; want
-= n
; }
451 else if (!n
|| errno
!= EINTR
) break;
453 if (fakeerr
&& !errno
) errno
= fakeerr
;
454 return (!done
&& errno ?
-1 : done
);
457 static void report_bad_blocks_progress(secaddr lo
, secaddr hi
, int err
)
461 if (lo
== hi
) append_progress(": retrying bad sector");
463 append_progress(": %"PRIuSEC
" bad %s",
464 hi
- lo
, hi
== lo
+ 1 ?
"sector" : "sectors");
465 if (err
&& err
!= EIO
) append_progress(" (%s)", strerror(err
));
469 static void recovered(secaddr bad_lo
, secaddr bad_hi
)
472 moan("skipping %"PRIuSEC
" bad sectors (%"PRIuSEC
" .. %"PRIuSEC
")",
473 bad_hi
- bad_lo
, bad_lo
, bad_hi
);
476 mapfp
= fopen(mapfile
, "w");
478 bail_syserr(errno
, "failed to open bad-sector map file `%s'",
480 fprintf(mapfp
, "## bad sector map\n\n");
482 fprintf(mapfp
, "%"PRIuSEC
" %"PRIuSEC
"\n", bad_lo
, bad_hi
);
485 bail_syserr(errno
, "error writing bad-sector map file");
487 if (lseek(outfd
, (off_t
)(bad_hi
- bad_lo
)*SECTORSZ
, SEEK_CUR
) < 0)
488 bail_syserr(errno
, "failed to seek past bad sectors");
492 static ssize_t
find_good_sector(secaddr
*pos_inout
, secaddr end
,
493 unsigned char *buf
, secaddr sz
)
496 secaddr pos
= *pos_inout
, bad_lo
, bad_hi
, good
, step
;
499 report_bad_blocks_progress(pos
, pos
, errno
);
500 for (i
= 0; i
< 4; i
++) {
501 n
= read_sectors(pos
, buf
, 1);
504 moan("sector %"PRIuSEC
" read ok after retry", pos
);
505 bad_lo
= bad_hi
= pos
;
510 bad_lo
= pos
; step
= 1; bad_hi
= pos
+ 1;
512 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
515 moan("giving up on this extent");
516 recovered(bad_lo
, end
); *pos_inout
= end
; return (0);
519 if (step
> end
- bad_lo
) step
= end
- bad_lo
;
520 pos
= bad_lo
+ step
- 1;
521 n
= read_sectors(pos
, buf
, 1);
527 while (good
> bad_hi
) {
528 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
529 pos
= bad_hi
+ (good
- bad_hi
)/2;
530 n
= read_sectors(pos
, buf
, 1);
531 if (n
> 0) good
= pos
;
532 else bad_hi
= pos
+ 1;
534 recovered(bad_lo
, bad_hi
); *pos_inout
= good
;
538 static void emit(secaddr start
, secaddr end
)
540 #define BUFSECTORS 512
543 unsigned char buf
[BUFSECTORS
*SECTORSZ
];
547 static int first_time
= 1;
555 least
= least_live();
558 printf(";; %8"PRIuSEC
" .. %"PRIuSEC
"\n", start
, end
);
560 for (i
= 0; i
< filetab
.n
; i
++) {
561 if (!livep(i
)) continue;
562 if (act
== -1) act
= i
;
563 f
= &filetab
.v
[i
]; store_filename(fn
, f
->id
);
564 printf(";;\t\t%8"PRIuSEC
" .. %-8"PRIuSEC
" %s\n",
565 start
- f
->start
, end
- f
->start
, fn
);
567 if (act
== -1) printf(";;\t\t#<no live source>\n");
568 assert(act
== least
);
572 { file
= 0; vob
= 0; }
574 file
= &filetab
.v
[least
];
575 switch (id_kind(file
->id
)) {
580 if (first_time
) { clear_progress(); first_time
= 0; }
581 vob
= DVDOpenFile(dvd
, id_title(file
->id
),
583 ? DVD_READ_TITLE_VOBS
584 : DVD_READ_MENU_VOBS
);
586 bail("failed to open %s %u",
587 id_part(file
->id
) ?
"title" : "menu",
597 want
= end
- pos
; if (want
> BUFSECTORS
) want
= BUFSECTORS
;
598 n
= read_sectors(pos
, buf
, want
);
600 if (n
<= 0) n
= find_good_sector(&pos
, end
, buf
, BUFSECTORS
);
601 if (n
> 0) { carefully_write(outfd
, buf
, n
*SECTORSZ
); pos
+= n
; }
602 report_progress(pos
); fflush(stdout
);
605 if (vob
) { DVDCloseFile(vob
); vob
= 0; }
611 static void logfn(void *p
, dvd_logger_level_t lev
,
612 const char *fmt
, va_list ap
)
615 case DVD_LOGGER_LEVEL_ERROR
:
616 fprintf("%s (libdvdread error): ", prog
);
618 case DVD_LOGGER_LEVEL_WARN
:
619 fprintf("%s (libdvdread warning): ", prog
);
624 vfprintf(stderr
, fmt
, ap
);
627 static const dvd_logger_cb logger
= { logfn
};
634 #define BUF_INIT { 0, 0, 0 }
635 #define BUF_REWIND(b) do { (b)->n = 0; } while (0)
636 #define BUF_FREE(b) do { \
638 free(_b->p); _b->p = 0; _b->n = _b->sz = 0; \
640 #define BUF_PUTC(b, ch) do { \
641 struct buf *_b = (b); \
642 if (_b->n >= _b->sz) { \
643 _b->sz = _b->sz ? 2*_b->sz : 32; \
644 _b->p = realloc(_b->p, _b->sz); \
645 if (!_b->p) bail("out of memory allocating %zu bytes", _b->sz); \
647 _b->p[_b->n] = (ch); \
650 static int read_line(FILE *fp
, struct buf
*b
)
657 else if (ch
!= '\n') do {
658 BUF_PUTC(b
, ch
); b
->n
++;
660 } while (ch
!= EOF
&& ch
!= '\n');
665 #define PRF_HYPHEN 1u
666 static int parse_range(const char *p
, unsigned f
,
667 secaddr
*start_out
, secaddr
*end_out
)
671 unsigned long start
, end
;
676 start
= strtoul(p
, &q
, 0);
677 if (errno
|| start
>= SECLIMIT
) { rc
= -1; goto end
; }
678 *start_out
= start
; p
= q
;
679 } else if (!(f
&PRF_HYPHEN
))
680 { rc
= -1; goto end
; }
685 if (*p
!= '-') { rc
= -1; goto end
; }
688 if (!ISSPACE(*p
)) { rc
= -1; goto end
; }
689 do p
++; while (ISSPACE(*p
));
693 end
= strtoul(p
, &q
, 0);
694 if (errno
|| end
> SECLIMIT
|| end
< start
) { rc
= -1; goto end
; }
695 *end_out
= end
; p
= q
;
696 } else if (!(f
&PRF_HYPHEN
))
697 { rc
= -1; goto end
; }
699 if (!(f
&PRF_HYPHEN
)) while (ISSPACE(*p
)) p
++;
700 if (*p
) { rc
= -1; goto end
; }
708 int main(int argc
, char *argv
[])
715 secaddr start
, end
, last
;
716 const struct event
*ev
;
717 const char *device
, *outfile
;
718 struct badblock
*bad
;
723 struct buf buf
= BUF_INIT
;
726 const struct file
*file
;
731 #define f_continue 2u
735 p
= strrchr(argv
[0], '/'); prog
= p ? p
+ 1 : argv
[0];
737 opt
= getopt(argc
, argv
, "hFR:X:b:cr:"); if (opt
< 0) break;
739 case 'h': usage(stderr
); exit(0);
740 case 'F': f
|= f_fixup
; break;
742 fp
= fopen(optarg
, "r");
744 bail_syserr(errno
, "failed to open ranges file `%s'", optarg
);
747 BUF_REWIND(&buf
); if (read_line(fp
, &buf
)) break;
749 while (ISSPACE(*p
)) p
++;
750 if (!*p
|| *p
== '#') continue;
751 if (parse_range(p
, 0, &start
, &end
) ||
752 (last
<= SECLIMIT
&& start
< last
))
753 bail("bad range `%s' at `%s' line %zu", buf
.p
, optarg
, i
);
756 eventq
.v
[eventq
.n
- 1].pos
= end
;
758 put_event(EV_WRITE
, 0, start
);
759 put_event(EV_STOP
, 0, end
);
765 bail_syserr(errno
, "failed to read ranges file `%s'", optarg
);
768 fp
= fopen(optarg
, "r");
770 bail_syserr(errno
, "failed to open bad-blocks file `%s'", optarg
);
773 BUF_REWIND(&buf
); if (read_line(fp
, &buf
)) break;
775 while (ISSPACE(*p
)) p
++;
776 if (!*p
|| *p
== '#') continue;
777 if (parse_range(p
, 0, &start
, &end
) ||
778 (last
<= SECLIMIT
&& start
< last
))
779 bail("bad range `%s' at `%s' line %zu", buf
.p
, optarg
, i
);
781 { VEC_PUSH(bad
, &badblocks
); bad
->start
= start
; bad
->end
= end
; }
784 bail_syserr(errno
, "failed to read bad-blocks file `%s'", optarg
);
787 if (mapfile
) bail("can't have multiple map files");
790 case 'c': f
|= f_continue
; break;
793 if (parse_range(optarg
, PRF_HYPHEN
, &start
, &end
))
794 bail("bad range `%s'", optarg
);
796 put_event(EV_WRITE
, 0, start
);
797 if (end
<= SECLIMIT
) put_event(EV_STOP
, 0, end
);
800 default: f
|= f_bogus
; break;
803 if (argc
- optind
!= 2) f
|= f_bogus
;
804 if (f
&f_bogus
) { usage(stderr
); exit(2); }
806 device
= argv
[optind
]; outfile
= argv
[optind
+ 1];
809 qsort(badblocks
.v
, badblocks
.n
, sizeof(struct badblock
),
812 printf(";; fake bad blocks:\n");
813 for (i
= 0; i
< badblocks
.n
; i
++)
814 printf(";;\t%8"PRIuSEC
" .. %"PRIuSEC
"\n",
815 badblocks
.v
[i
].start
, badblocks
.v
[i
].end
);
819 dvdfd
= open(device
, O_RDONLY
);
821 bail_syserr(errno
, "failed to open device `%s'", device
);
822 if (fstat(dvdfd
, &st
))
823 bail_syserr(errno
, "failed to stat device `%s'", device
);
824 if (S_ISREG(st
.st_mode
)) {
827 } else if (S_ISBLK(st
.st_mode
)) {
828 if (ioctl(dvdfd
, BLKSSZGET
, &blksz
))
829 bail_syserr(errno
, "failed to get block size for `%s'", device
);
830 if (ioctl(dvdfd
, BLKGETSIZE64
, &volsz
))
831 bail_syserr(errno
, "failed to get volume size for `%s'", device
);
833 bail("can't use `%s' as source: expected file or block device", device
);
835 if (blksz
!= SECTORSZ
)
836 bail("device `%s' block size %d /= %d", device
, blksz
, SECTORSZ
);
838 bail("device `%s' volume size %"PRIu64
" not a multiple of %d",
839 device
, volsz
, SECTORSZ
);
842 outfd
= open(outfile
, O_WRONLY
| O_CREAT
, 0666);
844 bail_syserr(errno
, "failed to create output file `%s'", outfile
);
848 off
= lseek(outfd
, 0, SEEK_END
);
850 bail_syserr(errno
, "failed to seek to end of output file `%s'",
852 put_event(EV_WRITE
, 0, off
/SECTORSZ
);
853 } else if (!eventq
.n
&& !(f
&f_fixup
))
854 put_event(EV_WRITE
, 0, 0);
857 dvd
= DVDOpen2(0, &logger
, device
);
859 dvd
= DVDOpen(device
);
861 if (!dvd
) bail("failed to open DVD on `%s'", device
);
863 /* It's fast enough just to check everything. */
865 for (i
= 1; i
< 100; i
++) {
869 put_file(mkident(RAW
, 0, 0), 0, volsz
/SECTORSZ
);
870 assert(filetab
.n
<= MAXFILES
);
872 for (i
= 0, limit
= 0; i
< filetab
.n
; i
++)
873 if (filetab
.v
[i
].end
> limit
) limit
= filetab
.v
[i
].end
;
875 if (end
> limit
) end
= limit
;
878 printf("\n;; files:\n");
879 for (i
= 0; i
< filetab
.n
; i
++) {
880 file
= &filetab
.v
[i
];
881 store_filename(fn
, file
->id
);
882 printf(";;\t%8"PRIuSEC
" %s\n", file
->start
, fn
);
886 qsort(eventq
.v
, eventq
.n
, sizeof(struct event
), compare_event
);
888 f
&= ~f_write
; start
= 0; n
= 0;
889 for (i
= 0; i
< eventq
.n
; i
++) {
894 bail("overlapping ranges: range from %"PRIuSEC
" still open at %"PRIuSEC
"",
896 n
++; f
|= f_write
; start
= ev
->pos
;
904 f
&= ~f_write
; start
= 0;
905 for (i
= 0; i
< eventq
.n
; i
++) {
908 case EV_WRITE
: start
= ev
->pos
; f
|= f_write
; break;
909 case EV_STOP
: nsectors
+= ev
->pos
- start
; f
&= ~f_write
; break;
911 if (ev
->pos
>= limit
) break;
912 if (f
&f_fixup
) start
= ev
->pos
;
916 put_event(EV_WRITE
, 0, start
);
920 nsectors
+= limit
- start
;
921 put_event(EV_STOP
, 0, limit
);
923 if (n
== 1 && (f
&f_write
)) flags
|= F_ALLPROGRESS
;
927 printf("\n;; event sweep:\n");
929 for (pos
= 0, i
= 0; i
< eventq
.n
; i
++) {
932 if (f
&f_write
) emit(pos
, ev
->pos
);
943 store_filename(fn
, filetab
.v
[ev
->file
].id
);
945 printf(";; %8"PRIuSEC
": begin `%s'\n", pos
, fn
);
949 gettimeofday(&last_time
, 0); last_pos
= pos
;
950 if (lseek(outfd
, (off_t
)ev
->pos
*SECTORSZ
, SEEK_SET
) < 0)
952 "failed to seek to resume position "
953 "(sector %"PRIuSEC
") in output file `%s'",
957 printf(";; %8"PRIuSEC
": begin write\n", pos
);
965 printf(";; %8"PRIuSEC
": end write\n", pos
);
969 clear_live(ev
->file
);
971 store_filename(fn
, filetab
.v
[ev
->file
].id
);
973 printf(";; %8"PRIuSEC
": end `%s'\n", pos
, fn
);
980 if (progresslen
) putchar('\n');
982 if (ftruncate(outfd
, (off_t
)limit
*SECTORSZ
) < 0)
983 bail_syserr(errno
, "failed to set output file `%s' length", outfile
);
985 if (dvd
) DVDClose(dvd
);
986 if (dvdfd
>= 0) close(dvdfd
);
987 if (outfd
>= 0) close(outfd
);
989 if (ferror(mapfp
) || fclose(mapfp
))
990 bail_syserr(errno
, "error writing bad-sector map file");