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
;
413 best
= 0; lo
= 0; hi
= badblocks
.n
;
415 debug_clear_progress();
416 printf(";; searching badblocks for %"PRIuSEC
" .. %"PRIuSEC
"\n",
420 mid
= lo
+ (hi
- lo
)/2; bad
= &badblocks
.v
[mid
];
422 printf(";; try %zu (%"PRIuSEC
" .. %"PRIuSEC
")... ",
423 mid
, bad
->start
, bad
->end
);
425 if (pos
< bad
->start
) { D( printf("high\n"); ) best
= bad
; hi
= mid
; }
426 else if (pos
>= bad
->end
) { D( printf("low\n"); ) lo
= mid
+ 1; }
427 else { D( printf("match!\n"); ) errno
= EIO
; return (-1); }
431 printf(";; next is %"PRIuSEC
" .. %"PRIuSEC
"\n",
432 best
->start
, best
->end
);
434 if (best
&& pos
+ want
> best
->start
)
435 { want
= best
->start
- pos
; fakeerr
= EIO
; }
440 { errno
= 0; n
= DVDReadBlocks(vob
, pos
- file
->start
, want
, p
); }
442 if (lseek(dvdfd
, (off_t
)pos
*SECTORSZ
, SEEK_SET
) < 0)
443 bail_syserr(errno
, "failed to seek to sector %"PRIuSEC
"", pos
);
444 errno
= 0; n
= read(dvdfd
, p
, want
*SECTORSZ
);
445 if (n
>= 0) n
/= SECTORSZ
;
447 memset(p
, 0, want
*SECTORSZ
);
451 if (n
> 0) { done
+= n
; pos
+= n
; p
+= n
*SECTORSZ
; want
-= n
; }
452 else if (!n
|| errno
!= EINTR
) break;
454 if (fakeerr
&& !errno
) errno
= fakeerr
;
455 return (!done
&& errno ?
-1 : done
);
458 static void report_bad_blocks_progress(secaddr lo
, secaddr hi
, int err
)
462 if (lo
== hi
) append_progress(": retrying bad sector");
464 append_progress(": %"PRIuSEC
" bad %s",
465 hi
- lo
, hi
== lo
+ 1 ?
"sector" : "sectors");
466 if (err
&& err
!= EIO
) append_progress(" (%s)", strerror(err
));
470 static void recovered(secaddr bad_lo
, secaddr bad_hi
)
473 moan("skipping %"PRIuSEC
" bad sectors (%"PRIuSEC
" .. %"PRIuSEC
")",
474 bad_hi
- bad_lo
, bad_lo
, bad_hi
);
477 mapfp
= fopen(mapfile
, "w");
479 bail_syserr(errno
, "failed to open bad-sector map file `%s'",
481 fprintf(mapfp
, "## bad sector map\n\n");
483 fprintf(mapfp
, "%"PRIuSEC
" %"PRIuSEC
"\n", bad_lo
, bad_hi
);
486 bail_syserr(errno
, "error writing bad-sector map file");
488 if (lseek(outfd
, (off_t
)(bad_hi
- bad_lo
)*SECTORSZ
, SEEK_CUR
) < 0)
489 bail_syserr(errno
, "failed to seek past bad sectors");
495 secaddr sz
, pos
, start
, end
;
498 static void rearrange_sectors(struct recoverybuf
*r
,
499 secaddr dest
, secaddr src
, secaddr len
)
501 assert(dest
+ len
<= r
->sz
);
502 assert(src
+ len
<= r
->sz
);
503 memmove(r
->buf
+ dest
*SECTORSZ
, r
->buf
+ src
*SECTORSZ
, len
*SECTORSZ
);
507 __attribute__((format(printf
, 2, 3)))
508 static void show_recovery_buffer_map(const struct recoverybuf
*r
,
509 const char *what
, ...)
515 printf(";; recovery buffer (");
518 "(%"PRIuSEC
") ..%"PRIuSEC
".. "
519 "[%"PRIuSEC
" ..%"PRIuSEC
".. %"PRIuSEC
"] "
520 "..%"PRIuSEC
".. (%"PRIuSEC
")\n",
522 r
->pos
+ r
->start
, r
->end
- r
->start
, r
->pos
+ r
->end
,
523 r
->sz
- r
->end
, r
->pos
+ r
->sz
);
525 assert(r
->start
<= r
->end
);
526 assert(r
->end
<= r
->sz
);
530 static ssize_t
recovery_read_sectors(struct recoverybuf
*r
,
531 secaddr pos
, secaddr off
, secaddr want
)
535 assert(off
<= r
->sz
); assert(want
<= r
->sz
- off
);
536 n
= read_sectors(pos
, r
->buf
+ off
*SECTORSZ
, want
);
540 static ssize_t
recovery_read(struct recoverybuf
*r
,
541 secaddr pos
, secaddr want
)
543 secaddr diff
, pp
, nn
;
548 show_recovery_buffer_map(r
, "begin(%"PRIuSEC
", %"PRIuSEC
")", pos
, want
);
553 if (r
->start
+ diff
>= r
->sz
) {
554 r
->pos
= pos
; r
->start
= r
->end
= 0;
556 show_recovery_buffer_map(r
, "cleared; shift up by %"PRIuSEC
"", diff
);
559 if (r
->end
+ diff
> r
->sz
) r
->end
= r
->sz
- diff
;
560 rearrange_sectors(r
, r
->start
+ diff
, r
->start
, r
->end
- r
->start
);
561 r
->pos
-= diff
; r
->start
+= diff
; r
->end
+= diff
;
563 show_recovery_buffer_map(r
, "shifted up by %"PRIuSEC
"", diff
);
566 } else if (pos
> r
->pos
+ r
->end
) {
567 r
->pos
= pos
; r
->start
= r
->end
= 0;
569 show_recovery_buffer_map(r
, "cleared; beyond previous region");
571 } else if (pos
+ want
> r
->pos
+ r
->sz
) {
572 diff
= (pos
+ want
) - (r
->pos
+ r
->sz
);
573 if (r
->end
<= diff
) {
574 r
->pos
= pos
; r
->start
= r
->end
= 0;
576 show_recovery_buffer_map(r
, "cleared; shift down by %"PRIuSEC
"", diff
);
579 if (r
->start
< diff
) r
->start
= diff
;
580 rearrange_sectors(r
, r
->start
- diff
, r
->start
, r
->end
- r
->start
);
581 r
->pos
+= diff
; r
->start
-= diff
; r
->end
-= diff
;
583 show_recovery_buffer_map(r
, "shifted down by %"PRIuSEC
"", diff
);
588 if (pos
< r
->pos
+ r
->start
) {
589 pp
= pos
- r
->pos
; nn
= r
->start
- pp
;
591 printf(";; read low (%"PRIuSEC
"@%"PRIuSEC
", %"PRIuSEC
")", pos
, pp
, nn
);
594 n
= recovery_read_sectors(r
, pos
, pp
, nn
);
596 printf(" -> %zd\n", n
);
599 if (n
>= 0 && n
> want
) n
= want
;
604 show_recovery_buffer_map(r
, "joined new region");
608 if (pos
+ want
> r
->pos
+ r
->end
) {
609 pp
= r
->end
; nn
= (pos
+ want
) - (r
->pos
+ r
->end
);
611 printf(";; read high (%"PRIuSEC
"@%"PRIuSEC
", %"PRIuSEC
")",
612 r
->pos
+ pp
, pp
, nn
);
615 n
= recovery_read_sectors(r
, pos
+ pp
, pp
, nn
);
617 printf(" -> %zd\n", n
);
622 show_recovery_buffer_map(r
, "joined new region");
627 n
= r
->pos
+ r
->end
- pos
;
628 if (!n
&& want
) n
= -1;
632 show_recovery_buffer_map(r
, "done; return %zd", n
);
637 static secaddr
run_length_wanted(secaddr pos
, secaddr badlen
,
638 secaddr sz
, secaddr end
)
644 if (want
> end
- pos
) want
= end
- pos
;
645 if (want
> sz
) want
= sz
;
649 static ssize_t
find_good_sector(secaddr
*pos_inout
, secaddr end
,
650 unsigned char *buf
, secaddr sz
)
653 secaddr pos
= *pos_inout
, bad_lo
, bad_hi
, good
, step
, want
;
654 struct recoverybuf r
;
657 r
.buf
= buf
; r
.sz
= sz
; r
.pos
= r
.start
= r
.end
= 0;
658 report_bad_blocks_progress(pos
, pos
, errno
);
660 want
= sz
; if (want
> end
- pos
) want
= end
- pos
;
661 for (i
= 0; i
< 4; i
++) {
662 n
= recovery_read(&r
, pos
, want
);
665 printf(";; [retry] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
670 moan("sector %"PRIuSEC
" read ok after retry", pos
);
675 bad_lo
= pos
; bad_hi
= pos
+ 1;
677 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
680 moan("giving up on this extent");
681 recovered(bad_lo
, end
); *pos_inout
= end
; return (0);
683 step
= 2*(bad_hi
- bad_lo
); if (step
> end
- bad_lo
) step
= end
- bad_lo
;
684 pos
= bad_lo
+ step
- 1;
685 want
= run_length_wanted(pos
, step
, sz
, end
);
686 n
= recovery_read(&r
, pos
, want
);
689 printf(";; [bound] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
692 if (n
== want
) break;
694 bad_hi
= pos
+ n
+ 1;
698 while (good
> bad_hi
) {
699 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
700 pos
= bad_hi
+ (good
- bad_hi
)/2;
702 want
= run_length_wanted(pos
, step
, sz
, end
);
703 n
= recovery_read(&r
, pos
, want
);
706 printf(";; [limit] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
710 if (n
== want
) good
= pos
;
711 else bad_hi
= pos
+ n
+ 1;
713 recovered(bad_lo
, bad_hi
); *pos_inout
= good
;
714 if (good
< r
.pos
+ r
.start
|| r
.pos
+ r
.end
<= good
)
717 n
= r
.pos
+ r
.end
- good
;
718 rearrange_sectors(&r
, 0, good
- r
.pos
, n
);
723 static void emit(secaddr start
, secaddr end
)
725 #define BUFSECTORS 512
728 unsigned char buf
[BUFSECTORS
*SECTORSZ
];
732 static int first_time
= 1;
740 least
= least_live();
743 printf(";; %8"PRIuSEC
" .. %"PRIuSEC
"\n", start
, end
);
745 for (i
= 0; i
< filetab
.n
; i
++) {
746 if (!livep(i
)) continue;
747 if (act
== -1) act
= i
;
748 f
= &filetab
.v
[i
]; store_filename(fn
, f
->id
);
749 printf(";;\t\t%8"PRIuSEC
" .. %-8"PRIuSEC
" %s\n",
750 start
- f
->start
, end
- f
->start
, fn
);
752 if (act
== -1) printf(";;\t\t#<no live source>\n");
753 assert(act
== least
);
757 { file
= 0; vob
= 0; }
759 file
= &filetab
.v
[least
];
760 switch (id_kind(file
->id
)) {
765 if (first_time
) { clear_progress(); first_time
= 0; }
766 vob
= DVDOpenFile(dvd
, id_title(file
->id
),
768 ? DVD_READ_TITLE_VOBS
769 : DVD_READ_MENU_VOBS
);
771 bail("failed to open %s %u",
772 id_part(file
->id
) ?
"title" : "menu",
782 want
= end
- pos
; if (want
> BUFSECTORS
) want
= BUFSECTORS
;
783 n
= read_sectors(pos
, buf
, want
);
785 if (n
<= 0) n
= find_good_sector(&pos
, end
, buf
, BUFSECTORS
);
786 if (n
> 0) { carefully_write(outfd
, buf
, n
*SECTORSZ
); pos
+= n
; }
787 report_progress(pos
); fflush(stdout
);
790 if (vob
) { DVDCloseFile(vob
); vob
= 0; }
796 static void logfn(void *p
, dvd_logger_level_t lev
,
797 const char *fmt
, va_list ap
)
800 case DVD_LOGGER_LEVEL_ERROR
:
801 fprintf("%s (libdvdread error): ", prog
);
803 case DVD_LOGGER_LEVEL_WARN
:
804 fprintf("%s (libdvdread warning): ", prog
);
809 vfprintf(stderr
, fmt
, ap
);
812 static const dvd_logger_cb logger
= { logfn
};
819 #define BUF_INIT { 0, 0, 0 }
820 #define BUF_REWIND(b) do { (b)->n = 0; } while (0)
821 #define BUF_FREE(b) do { \
823 free(_b->p); _b->p = 0; _b->n = _b->sz = 0; \
825 #define BUF_PUTC(b, ch) do { \
826 struct buf *_b = (b); \
827 if (_b->n >= _b->sz) { \
828 _b->sz = _b->sz ? 2*_b->sz : 32; \
829 _b->p = realloc(_b->p, _b->sz); \
830 if (!_b->p) bail("out of memory allocating %zu bytes", _b->sz); \
832 _b->p[_b->n] = (ch); \
835 static int read_line(FILE *fp
, struct buf
*b
)
842 else if (ch
!= '\n') do {
843 BUF_PUTC(b
, ch
); b
->n
++;
845 } while (ch
!= EOF
&& ch
!= '\n');
850 #define PRF_HYPHEN 1u
851 static int parse_range(const char *p
, unsigned f
,
852 secaddr
*start_out
, secaddr
*end_out
)
856 unsigned long start
, end
;
861 start
= strtoul(p
, &q
, 0);
862 if (errno
|| start
>= SECLIMIT
) { rc
= -1; goto end
; }
863 *start_out
= start
; p
= q
;
864 } else if (!(f
&PRF_HYPHEN
))
865 { rc
= -1; goto end
; }
870 if (*p
!= '-') { rc
= -1; goto end
; }
873 if (!ISSPACE(*p
)) { rc
= -1; goto end
; }
874 do p
++; while (ISSPACE(*p
));
878 end
= strtoul(p
, &q
, 0);
879 if (errno
|| end
> SECLIMIT
|| end
< start
) { rc
= -1; goto end
; }
880 *end_out
= end
; p
= q
;
881 } else if (!(f
&PRF_HYPHEN
))
882 { rc
= -1; goto end
; }
884 if (!(f
&PRF_HYPHEN
)) while (ISSPACE(*p
)) p
++;
885 if (*p
) { rc
= -1; goto end
; }
893 int main(int argc
, char *argv
[])
900 secaddr start
, end
, last
;
901 const struct event
*ev
;
902 const char *device
, *outfile
;
903 struct badblock
*bad
;
908 struct buf buf
= BUF_INIT
;
911 const struct file
*file
;
916 #define f_continue 2u
920 p
= strrchr(argv
[0], '/'); prog
= p ? p
+ 1 : argv
[0];
922 opt
= getopt(argc
, argv
, "hFR:X:b:cr:"); if (opt
< 0) break;
924 case 'h': usage(stderr
); exit(0);
925 case 'F': f
|= f_fixup
; break;
927 fp
= fopen(optarg
, "r");
929 bail_syserr(errno
, "failed to open ranges file `%s'", optarg
);
932 BUF_REWIND(&buf
); if (read_line(fp
, &buf
)) break;
934 while (ISSPACE(*p
)) p
++;
935 if (!*p
|| *p
== '#') continue;
936 if (parse_range(p
, 0, &start
, &end
) ||
937 (last
<= SECLIMIT
&& start
< last
))
938 bail("bad range `%s' at `%s' line %zu", buf
.p
, optarg
, i
);
941 eventq
.v
[eventq
.n
- 1].pos
= end
;
943 put_event(EV_WRITE
, 0, start
);
944 put_event(EV_STOP
, 0, end
);
950 bail_syserr(errno
, "failed to read ranges file `%s'", optarg
);
953 fp
= fopen(optarg
, "r");
955 bail_syserr(errno
, "failed to open bad-blocks file `%s'", optarg
);
958 BUF_REWIND(&buf
); if (read_line(fp
, &buf
)) break;
960 while (ISSPACE(*p
)) p
++;
961 if (!*p
|| *p
== '#') continue;
962 if (parse_range(p
, 0, &start
, &end
) ||
963 (last
<= SECLIMIT
&& start
< last
))
964 bail("bad range `%s' at `%s' line %zu", buf
.p
, optarg
, i
);
966 { VEC_PUSH(bad
, &badblocks
); bad
->start
= start
; bad
->end
= end
; }
969 bail_syserr(errno
, "failed to read bad-blocks file `%s'", optarg
);
972 if (mapfile
) bail("can't have multiple map files");
975 case 'c': f
|= f_continue
; break;
978 if (parse_range(optarg
, PRF_HYPHEN
, &start
, &end
))
979 bail("bad range `%s'", optarg
);
981 put_event(EV_WRITE
, 0, start
);
982 if (end
<= SECLIMIT
) put_event(EV_STOP
, 0, end
);
985 default: f
|= f_bogus
; break;
988 if (argc
- optind
!= 2) f
|= f_bogus
;
989 if (f
&f_bogus
) { usage(stderr
); exit(2); }
991 device
= argv
[optind
]; outfile
= argv
[optind
+ 1];
994 qsort(badblocks
.v
, badblocks
.n
, sizeof(struct badblock
),
997 printf(";; fake bad blocks:\n");
998 for (i
= 0; i
< badblocks
.n
; i
++)
999 printf(";;\t%8"PRIuSEC
" .. %"PRIuSEC
"\n",
1000 badblocks
.v
[i
].start
, badblocks
.v
[i
].end
);
1004 dvdfd
= open(device
, O_RDONLY
);
1006 bail_syserr(errno
, "failed to open device `%s'", device
);
1007 if (fstat(dvdfd
, &st
))
1008 bail_syserr(errno
, "failed to stat device `%s'", device
);
1009 if (S_ISREG(st
.st_mode
)) {
1012 } else if (S_ISBLK(st
.st_mode
)) {
1013 if (ioctl(dvdfd
, BLKSSZGET
, &blksz
))
1014 bail_syserr(errno
, "failed to get block size for `%s'", device
);
1015 if (ioctl(dvdfd
, BLKGETSIZE64
, &volsz
))
1016 bail_syserr(errno
, "failed to get volume size for `%s'", device
);
1018 bail("can't use `%s' as source: expected file or block device", device
);
1020 if (blksz
!= SECTORSZ
)
1021 bail("device `%s' block size %d /= %d", device
, blksz
, SECTORSZ
);
1023 bail("device `%s' volume size %"PRIu64
" not a multiple of %d",
1024 device
, volsz
, SECTORSZ
);
1027 outfd
= open(outfile
, O_WRONLY
| O_CREAT
, 0666);
1029 bail_syserr(errno
, "failed to create output file `%s'", outfile
);
1033 off
= lseek(outfd
, 0, SEEK_END
);
1035 bail_syserr(errno
, "failed to seek to end of output file `%s'",
1037 put_event(EV_WRITE
, 0, off
/SECTORSZ
);
1038 } else if (!eventq
.n
&& !(f
&f_fixup
))
1039 put_event(EV_WRITE
, 0, 0);
1042 dvd
= DVDOpen2(0, &logger
, device
);
1044 dvd
= DVDOpen(device
);
1046 if (!dvd
) bail("failed to open DVD on `%s'", device
);
1048 /* It's fast enough just to check everything. */
1050 for (i
= 1; i
< 100; i
++) {
1054 put_file(mkident(RAW
, 0, 0), 0, volsz
/SECTORSZ
);
1055 assert(filetab
.n
<= MAXFILES
);
1057 for (i
= 0, limit
= 0; i
< filetab
.n
; i
++)
1058 if (filetab
.v
[i
].end
> limit
) limit
= filetab
.v
[i
].end
;
1060 if (end
> limit
) end
= limit
;
1063 printf("\n;; files:\n");
1064 for (i
= 0; i
< filetab
.n
; i
++) {
1065 file
= &filetab
.v
[i
];
1066 store_filename(fn
, file
->id
);
1067 printf(";;\t%8"PRIuSEC
" %s\n", file
->start
, fn
);
1071 qsort(eventq
.v
, eventq
.n
, sizeof(struct event
), compare_event
);
1073 f
&= ~f_write
; start
= 0; n
= 0;
1074 for (i
= 0; i
< eventq
.n
; i
++) {
1079 bail("overlapping ranges: range from %"PRIuSEC
" still open at %"PRIuSEC
"",
1081 n
++; f
|= f_write
; start
= ev
->pos
;
1089 f
&= ~f_write
; start
= 0;
1090 for (i
= 0; i
< eventq
.n
; i
++) {
1093 case EV_WRITE
: start
= ev
->pos
; f
|= f_write
; break;
1094 case EV_STOP
: nsectors
+= ev
->pos
- start
; f
&= ~f_write
; break;
1096 if (ev
->pos
>= limit
) break;
1097 if (f
&f_fixup
) start
= ev
->pos
;
1101 put_event(EV_WRITE
, 0, start
);
1105 nsectors
+= limit
- start
;
1106 put_event(EV_STOP
, 0, limit
);
1108 if (n
== 1 && (f
&f_write
)) flags
|= F_ALLPROGRESS
;
1112 printf("\n;; event sweep:\n");
1114 for (pos
= 0, i
= 0; i
< eventq
.n
; i
++) {
1116 if (ev
->pos
> pos
) {
1117 if (f
&f_write
) emit(pos
, ev
->pos
);
1128 store_filename(fn
, filetab
.v
[ev
->file
].id
);
1130 printf(";; %8"PRIuSEC
": begin `%s'\n", pos
, fn
);
1134 gettimeofday(&last_time
, 0); last_pos
= pos
;
1135 if (lseek(outfd
, (off_t
)ev
->pos
*SECTORSZ
, SEEK_SET
) < 0)
1137 "failed to seek to resume position "
1138 "(sector %"PRIuSEC
") in output file `%s'",
1142 printf(";; %8"PRIuSEC
": begin write\n", pos
);
1150 printf(";; %8"PRIuSEC
": end write\n", pos
);
1154 clear_live(ev
->file
);
1156 store_filename(fn
, filetab
.v
[ev
->file
].id
);
1158 printf(";; %8"PRIuSEC
": end `%s'\n", pos
, fn
);
1165 if (progresslen
) putchar('\n');
1167 if (ftruncate(outfd
, (off_t
)limit
*SECTORSZ
) < 0)
1168 bail_syserr(errno
, "failed to set output file `%s' length", outfile
);
1170 if (dvd
) DVDClose(dvd
);
1171 if (dvdfd
>= 0) close(dvdfd
);
1172 if (outfd
>= 0) close(outfd
);
1174 if (ferror(mapfp
) || fclose(mapfp
))
1175 bail_syserr(errno
, "error writing bad-sector map file");