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 static void open_file_on_demand(const char *file
, FILE **fp_inout
,
113 fp
= fopen(file
, "w");
115 bail_syserr(errno
, "failed to open %s file `%s'", what
, file
);
116 fprintf(fp
, "## %s\n\n", what
);
121 static void check_write(FILE *fp
, const char *what
)
124 if (ferror(fp
)) bail_syserr(errno
, "error writing %s file", what
);
127 static void carefully_fclose(FILE *fp
, const char *what
)
129 if (fp
&& (ferror(fp
) || fclose(fp
)))
130 bail_syserr(errno
, "error writing %s file", what
);
133 #define DEFVEC(vtype, etype) \
134 typedef struct { etype *v; size_t n, sz; } vtype
135 #define VEC_INIT { 0, 0, 0 }
136 #define VEC_FREE(vv) do { \
137 free((vv)->v); (vv)->v 0; (vv)->n = (vv)->sz = 0; \
139 #define VEC_PUSH(p, vv) do { \
141 if ((vv)->n >= (vv)->sz) { \
142 (vv)->sz = (vv)->sz ? 2*(vv)->sz : 32; \
143 _want = (vv)->sz*sizeof(*(vv)->v); \
144 (vv)->v = realloc((vv)->v, _want); \
145 if (!(vv)->v) bail("out of memory allocating %zu bytes", _want); \
147 (p) = &(vv)->v[(vv)->n++]; \
150 enum { RAW
, IFO
, VOB
, BUP
};
151 typedef uint_least32_t ident
;
153 static inline ident
mkident(unsigned kind
, unsigned title
, unsigned part
)
154 { return (((ident
)kind
<< 0) | ((ident
)title
<< 8) | ((ident
)part
<< 16)); }
155 static inline unsigned id_kind(ident id
) { return ((id
>> 0)&0x0ff); }
156 static inline unsigned id_title(ident id
) { return ((id
>> 8)&0x0ff); }
157 static inline unsigned id_part(ident id
) { return ((id
>> 16)&0x0ff); }
159 #define MAXFNSZ (1 + 8 + 1 + 12 + 1)
161 static void store_filename(char *buf
, ident id
)
163 switch (id_kind(id
)) {
165 sprintf(buf
, "#<raw device>");
168 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.IFO");
169 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.IFO", id_title(id
));
172 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.BUP");
173 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.BUP", id_title(id
));
176 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.VOB");
178 sprintf(buf
, "/VIDEO_TS/VTS_%02u_%u.VOB", id_title(id
), id_part(id
));
185 typedef uint_least32_t secaddr
;
186 #define PRIuSEC PRIuLEAST32
187 #define SECLIMIT 0x00400000
189 #define MAXFILES (1 + 2*99 + 1)
194 DEFVEC(file_v
, struct file
);
195 static file_v filetab
= VEC_INIT
;
197 enum { EV_STOP
, EV_BEGIN
, EV_END
, EV_WRITE
};
199 unsigned char ev
, file
;
202 DEFVEC(event_v
, struct event
);
203 static event_v eventq
= VEC_INIT
;
205 static int compare_event(const void *a
, const void *b
)
207 const struct event
*eva
= a
, *evb
= b
;
209 if (eva
->pos
< evb
->pos
) return (-1);
210 else if (eva
->pos
> evb
->pos
) return (+1);
212 if (eva
->ev
< evb
->ev
) return (-1);
213 else if (eva
->ev
> evb
->ev
) return (+1);
215 if (eva
->file
< evb
->file
) return (-1);
216 else if (eva
->file
> evb
->file
) return (+1);
221 typedef uint_least32_t bits
;
222 static bits live
[(MAXFILES
+ 31)/32];
224 static inline int livep(unsigned i
)
225 { return (live
[i
/32]&((bits
)1 << (i
%32))); }
226 static inline void set_live(unsigned i
)
227 { live
[i
/32] |= (bits
)1 << (i
%32); }
228 static inline void clear_live(unsigned i
)
229 { live
[i
/32] &= ~((bits
)1 << (i
%32)); }
230 static inline int least_live(void)
232 unsigned i
, n
= (filetab
.n
+ 32)/32;
235 for (i
= 0; i
< n
; i
++) { b
= live
[i
]; if (b
) goto found
; }
239 if (!(b
&0x0000ffff)) { b
>>= 16; i
+= 16; }
240 if (!(b
&0x000000ff)) { b
>>= 8; i
+= 8; }
241 if (!(b
&0x0000000f)) { b
>>= 4; i
+= 4; }
242 if (!(b
&0x00000003)) { b
>>= 2; i
+= 2; }
243 if (!(b
&0x00000001)) { b
>>= 1; i
+= 1; }
248 static void put_event(unsigned evtype
, unsigned file
, secaddr pos
)
252 VEC_PUSH(ev
, &eventq
);
253 ev
->ev
= evtype
; ev
->file
= file
; ev
->pos
= pos
;
256 static void put_file(ident id
, secaddr start
, secaddr end
)
261 VEC_PUSH(f
, &filetab
); i
= f
- filetab
.v
;
262 f
->id
= id
; f
->start
= start
; f
->end
= end
;
263 put_event(EV_BEGIN
, i
, start
);
264 put_event(EV_END
, i
, end
);
267 static void put_menu(dvd_reader_t
*dvd
, unsigned title
)
269 ident id
= mkident(VOB
, title
, 0);
273 store_filename(fn
, id
);
274 start
= UDFFindFile(dvd
, fn
, &len
); if (!start
) return;
276 printf(";; %8"PRIuSEC
" .. %-8"PRIuSEC
": %s\n",
277 start
, start
+ SECTORS(len
), fn
);
279 put_file(id
, start
, start
+ SECTORS(len
));
282 static void put_title(dvd_reader_t
*dvd
, unsigned title
)
285 secaddr start
[9], len
[9];
288 for (i
= 0; i
< 9; i
++) {
289 store_filename(fn
, mkident(VOB
, title
, i
+ 1));
290 start
[i
] = UDFFindFile(dvd
, fn
, &len
[i
]); if (!start
[i
]) break;
292 npart
= i
; if (!npart
) return;
295 for (i
= 0; i
< npart
; i
++) {
296 store_filename(fn
, mkident(VOB
, title
, i
+ 1));
297 printf(";; %8"PRIuSEC
" .. %-8"PRIuSEC
": %s\n",
298 start
[i
], start
[i
] + SECTORS(len
[i
]), fn
);
303 for (i
= 0; i
< npart
- 1; i
++) {
305 bail("title %u part %u length = %"PRIuSEC
" not a multiple of %d",
306 title
, i
, len
[i
], SECTORSZ
);
307 if (start
[i
] + len
[i
]/SECTORSZ
!= start
[i
+ 1])
309 ("title %u part %u end = %"PRIuSEC
" /= part %u start = %"PRIuSEC
"",
310 title
, i
, start
[i
] + len
[i
]/SECTORSZ
, i
+ 1, start
[i
+ 1]);
313 put_file(mkident(VOB
, title
, 1),
314 start
[0], start
[npart
- 1] + SECTORS(len
[npart
- 1]));
317 static int progresslen
= 0;
319 static void clear_progress_internal(void)
321 while (progresslen
) { fputs("\b \b", stdout
); progresslen
--; }
324 static void clear_progress(void)
325 { clear_progress_internal(); fflush(stdout
); }
327 static void debug_clear_progress(void)
328 { if (progresslen
) { putchar('\n'); progresslen
= 0; } }
330 static void vappend_progress(const char *fmt
, va_list ap
)
331 { progresslen
+= vprintf(fmt
, ap
); }
332 __attribute__((format(printf
, 1, 2)))
333 static void append_progress(const char *fmt
, ...)
338 vappend_progress(fmt
, ap
);
341 __attribute__((format(printf
, 1, 2)))
342 static void print_progress(const char *fmt
, ...)
347 clear_progress_internal();
348 vappend_progress(fmt
, ap
);
353 # define F_ALLPROGRESS 1u
354 static secaddr last_pos
, limit
, nsectors
, ndone
;
355 static struct timeval last_time
;
356 static double wsum
, wcount
;
357 static struct file
*file
;
359 static void report_progress(secaddr pos
)
364 double percent
, t
, f
, g
, rate
;
368 #define BETA (1 - ALPHA)
370 gettimeofday(&now
, 0);
371 t
= (now
.tv_sec
- last_time
.tv_sec
) +
372 (now
.tv_usec
- last_time
.tv_usec
)/1000000.0;
375 g
= wcount ?
pow(BETA
, t
) : 0.0; f
= (1 - g
)/(1 - BETA
);
376 wsum
= f
*(pos
- last_pos
)/t
+ g
*wsum
;
377 wcount
= f
+ g
*wcount
;
378 ndone
+= pos
- last_pos
;
379 last_time
= now
; last_pos
= pos
;
382 if (!wsum
|| !wcount
)
383 { rate
= 0; strcpy(etastr
, "???"); }
386 eta
= (int)((nsectors
- ndone
)/rate
);
387 sprintf(etastr
, "%d:%02d:%02d", eta
/3600, (eta
/60)%60, eta
%60);
390 rate
*= SECTORSZ
; unit
= "";
391 if (rate
> 128) { rate
/= 1024; unit
= "k"; }
392 if (rate
> 128) { rate
/= 1024; unit
= "M"; }
393 if (rate
> 128) { rate
/= 1024; unit
= "G"; }
395 if (flags
&F_ALLPROGRESS
) percent
= pos
*100.0/limit
;
396 else percent
= ndone
*100.0/nsectors
;
398 ("copied %.1f%% (%"PRIuSEC
" of %"PRIuSEC
"; %.1f %sB/s, ETA %s)",
399 percent
, pos
, limit
, rate
, unit
, etastr
);
400 if (file
&& id_kind(file
->id
) == VOB
) {
401 append_progress(" -- %s %d %.1f%%",
402 id_part(file
->id
) ?
"title" : "menu",
404 (pos
- file
->start
)*100.0/
405 (file
->end
- file
->start
));
412 static dvd_reader_t
*dvd
;
413 static int dvdfd
= -1, outfd
= -1;
414 static dvd_file_t
*vob
;
415 static const char *mapfile
; static FILE *mapfp
;
417 struct badblock
{ secaddr start
, end
; };
418 DEFVEC(badblock_v
, struct badblock
);
419 static badblock_v badblocks
= VEC_INIT
;
421 static int compare_badblock(const void *a
, const void *b
)
423 const struct badblock
*ba
= a
, *bb
= b
;
425 if (ba
->start
< bb
->start
) return (-1);
426 else if (ba
->start
> bb
->start
) return (+1);
428 if (ba
->end
< bb
->end
) return (-1);
429 else if (ba
->end
> bb
->end
) return (+1);
434 static ssize_t
read_sectors(secaddr pos
, void *buf
, secaddr want
)
439 struct badblock
*bad
, *best
;
440 unsigned char *p
= buf
;
443 best
= 0; lo
= 0; hi
= badblocks
.n
;
445 debug_clear_progress();
446 printf(";; searching badblocks for %"PRIuSEC
" .. %"PRIuSEC
"\n",
450 mid
= lo
+ (hi
- lo
)/2; bad
= &badblocks
.v
[mid
];
452 printf(";; try %zu (%"PRIuSEC
" .. %"PRIuSEC
")... ",
453 mid
, bad
->start
, bad
->end
);
455 if (pos
< bad
->start
) { D( printf("high\n"); ) best
= bad
; hi
= mid
; }
456 else if (pos
>= bad
->end
) { D( printf("low\n"); ) lo
= mid
+ 1; }
457 else { D( printf("match!\n"); ) errno
= EIO
; return (-1); }
461 printf(";; next is %"PRIuSEC
" .. %"PRIuSEC
"\n",
462 best
->start
, best
->end
);
464 if (best
&& pos
+ want
> best
->start
)
465 { want
= best
->start
- pos
; fakeerr
= EIO
; }
470 { errno
= 0; n
= DVDReadBlocks(vob
, pos
- file
->start
, want
, p
); }
472 if (lseek(dvdfd
, (off_t
)pos
*SECTORSZ
, SEEK_SET
) < 0)
473 bail_syserr(errno
, "failed to seek to sector %"PRIuSEC
"", pos
);
474 errno
= 0; n
= read(dvdfd
, p
, want
*SECTORSZ
);
475 if (n
>= 0) n
/= SECTORSZ
;
477 memset(p
, 0, want
*SECTORSZ
);
481 if (n
> 0) { done
+= n
; pos
+= n
; p
+= n
*SECTORSZ
; want
-= n
; }
482 else if (!n
|| errno
!= EINTR
) break;
484 if (fakeerr
&& !errno
) errno
= fakeerr
;
485 return (!done
&& errno ?
-1 : done
);
488 static void report_bad_blocks_progress(secaddr lo
, secaddr hi
, int err
)
492 if (lo
== hi
) append_progress(": retrying bad sector");
494 append_progress(": %"PRIuSEC
" bad %s",
495 hi
- lo
, hi
== lo
+ 1 ?
"sector" : "sectors");
496 if (err
&& err
!= EIO
) append_progress(" (%s)", strerror(err
));
500 static void recovered(secaddr bad_lo
, secaddr bad_hi
)
503 moan("skipping %"PRIuSEC
" bad sectors (%"PRIuSEC
" .. %"PRIuSEC
")",
504 bad_hi
- bad_lo
, bad_lo
, bad_hi
);
506 open_file_on_demand(mapfile
, &mapfp
, "bad-sector region map");
507 fprintf(mapfp
, "%"PRIuSEC
" %"PRIuSEC
"\n", bad_lo
, bad_hi
);
508 check_write(mapfp
, "bad-sector region map");
510 if (lseek(outfd
, (off_t
)(bad_hi
- bad_lo
)*SECTORSZ
, SEEK_CUR
) < 0)
511 bail_syserr(errno
, "failed to seek past bad sectors");
517 secaddr sz
, pos
, start
, end
;
520 static void rearrange_sectors(struct recoverybuf
*r
,
521 secaddr dest
, secaddr src
, secaddr len
)
523 assert(dest
+ len
<= r
->sz
);
524 assert(src
+ len
<= r
->sz
);
525 memmove(r
->buf
+ dest
*SECTORSZ
, r
->buf
+ src
*SECTORSZ
, len
*SECTORSZ
);
529 __attribute__((format(printf
, 2, 3)))
530 static void show_recovery_buffer_map(const struct recoverybuf
*r
,
531 const char *what
, ...)
536 debug_clear_progress();
537 printf(";; recovery buffer (");
540 "(%"PRIuSEC
") ..%"PRIuSEC
".. "
541 "[%"PRIuSEC
" ..%"PRIuSEC
".. %"PRIuSEC
"] "
542 "..%"PRIuSEC
".. (%"PRIuSEC
")\n",
544 r
->pos
+ r
->start
, r
->end
- r
->start
, r
->pos
+ r
->end
,
545 r
->sz
- r
->end
, r
->pos
+ r
->sz
);
547 assert(r
->start
<= r
->end
);
548 assert(r
->end
<= r
->sz
);
552 static ssize_t
recovery_read_sectors(struct recoverybuf
*r
,
553 secaddr pos
, secaddr off
, secaddr want
)
557 assert(off
<= r
->sz
); assert(want
<= r
->sz
- off
);
558 n
= read_sectors(pos
, r
->buf
+ off
*SECTORSZ
, want
);
562 static ssize_t
recovery_read(struct recoverybuf
*r
,
563 secaddr pos
, secaddr want
)
565 secaddr diff
, pp
, nn
;
569 debug_clear_progress();
570 show_recovery_buffer_map(r
, "begin(%"PRIuSEC
", %"PRIuSEC
")", pos
, want
);
575 if (r
->start
+ diff
>= r
->sz
) {
576 r
->pos
= pos
; r
->start
= r
->end
= 0;
578 show_recovery_buffer_map(r
, "cleared; shift up by %"PRIuSEC
"", diff
);
581 if (r
->end
+ diff
> r
->sz
) r
->end
= r
->sz
- diff
;
582 rearrange_sectors(r
, r
->start
+ diff
, r
->start
, r
->end
- r
->start
);
583 r
->pos
-= diff
; r
->start
+= diff
; r
->end
+= diff
;
585 show_recovery_buffer_map(r
, "shifted up by %"PRIuSEC
"", diff
);
588 } else if (pos
> r
->pos
+ r
->end
) {
589 r
->pos
= pos
; r
->start
= r
->end
= 0;
591 show_recovery_buffer_map(r
, "cleared; beyond previous region");
593 } else if (pos
+ want
> r
->pos
+ r
->sz
) {
594 diff
= (pos
+ want
) - (r
->pos
+ r
->sz
);
595 if (r
->end
<= diff
) {
596 r
->pos
= pos
; r
->start
= r
->end
= 0;
598 show_recovery_buffer_map(r
, "cleared; shift down by %"PRIuSEC
"", diff
);
601 if (r
->start
< diff
) r
->start
= diff
;
602 rearrange_sectors(r
, r
->start
- diff
, r
->start
, r
->end
- r
->start
);
603 r
->pos
+= diff
; r
->start
-= diff
; r
->end
-= diff
;
605 show_recovery_buffer_map(r
, "shifted down by %"PRIuSEC
"", diff
);
610 if (pos
< r
->pos
+ r
->start
) {
611 pp
= pos
- r
->pos
; nn
= r
->start
- pp
;
613 printf(";; read low (%"PRIuSEC
"@%"PRIuSEC
", %"PRIuSEC
")", pos
, pp
, nn
);
616 n
= recovery_read_sectors(r
, pos
, pp
, nn
);
618 printf(" -> %zd\n", n
);
621 if (n
>= 0 && n
> want
) n
= want
;
626 show_recovery_buffer_map(r
, "joined new region");
630 if (pos
+ want
> r
->pos
+ r
->end
) {
631 pp
= r
->end
; nn
= (pos
+ want
) - (r
->pos
+ r
->end
);
633 printf(";; read high (%"PRIuSEC
"@%"PRIuSEC
", %"PRIuSEC
")",
634 r
->pos
+ pp
, pp
, nn
);
637 n
= recovery_read_sectors(r
, pos
+ pp
, pp
, nn
);
639 printf(" -> %zd\n", n
);
644 show_recovery_buffer_map(r
, "joined new region");
649 n
= r
->pos
+ r
->end
- pos
;
650 if (!n
&& want
) n
= -1;
654 show_recovery_buffer_map(r
, "done; return %zd", n
);
659 static secaddr
run_length_wanted(secaddr pos
, secaddr badlen
,
660 secaddr sz
, secaddr end
)
666 if (want
> end
- pos
) want
= end
- pos
;
667 if (want
> sz
) want
= sz
;
671 static ssize_t
find_good_sector(secaddr
*pos_inout
, secaddr end
,
672 unsigned char *buf
, secaddr sz
)
675 secaddr pos
= *pos_inout
, bad_lo
, bad_hi
, good
, step
, want
;
676 struct recoverybuf r
;
679 r
.buf
= buf
; r
.sz
= sz
; r
.pos
= r
.start
= r
.end
= 0;
680 report_bad_blocks_progress(pos
, pos
, errno
);
682 want
= sz
; if (want
> end
- pos
) want
= end
- pos
;
683 for (i
= 0; i
< 4; i
++) {
684 n
= recovery_read(&r
, pos
, want
);
686 debug_clear_progress();
687 printf(";; [retry] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
692 moan("sector %"PRIuSEC
" read ok after retry", pos
);
697 bad_lo
= pos
; bad_hi
= pos
+ 1;
699 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
702 moan("giving up on this extent");
703 recovered(bad_lo
, end
); *pos_inout
= end
; return (0);
705 step
= 2*(bad_hi
- bad_lo
); if (step
> end
- bad_lo
) step
= end
- bad_lo
;
706 pos
= bad_lo
+ step
- 1;
707 want
= run_length_wanted(pos
, step
, sz
, end
);
708 n
= recovery_read(&r
, pos
, want
);
710 debug_clear_progress();
711 printf(";; [bound] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
714 if (n
== want
) break;
716 bad_hi
= pos
+ n
+ 1;
720 while (good
> bad_hi
) {
721 report_bad_blocks_progress(bad_lo
, bad_hi
, errno
);
722 pos
= bad_hi
+ (good
- bad_hi
)/2;
724 want
= run_length_wanted(pos
, step
, sz
, end
);
725 n
= recovery_read(&r
, pos
, want
);
727 debug_clear_progress();
728 printf(";; [limit] try reading %"PRIuSEC
" .. %"PRIuSEC
" -> %zd\n",
732 if (n
== want
) good
= pos
;
733 else bad_hi
= pos
+ n
+ 1;
735 recovered(bad_lo
, bad_hi
); *pos_inout
= good
;
736 if (good
< r
.pos
+ r
.start
|| r
.pos
+ r
.end
<= good
)
739 n
= r
.pos
+ r
.end
- good
;
740 rearrange_sectors(&r
, 0, good
- r
.pos
, n
);
743 show_recovery_buffer_map(&r
, "returning %zd good sectors at %"PRIuSEC
"",
749 static void emit(secaddr start
, secaddr end
)
751 #define BUFSECTORS 512
754 unsigned char buf
[BUFSECTORS
*SECTORSZ
];
758 static int first_time
= 1;
766 least
= least_live();
769 printf(";; %8"PRIuSEC
" .. %"PRIuSEC
"\n", start
, end
);
771 for (i
= 0; i
< filetab
.n
; i
++) {
772 if (!livep(i
)) continue;
773 if (act
== -1) act
= i
;
774 f
= &filetab
.v
[i
]; store_filename(fn
, f
->id
);
775 printf(";;\t\t%8"PRIuSEC
" .. %-8"PRIuSEC
" %s\n",
776 start
- f
->start
, end
- f
->start
, fn
);
778 if (act
== -1) printf(";;\t\t#<no live source>\n");
779 assert(act
== least
);
783 { file
= 0; vob
= 0; }
785 file
= &filetab
.v
[least
];
786 switch (id_kind(file
->id
)) {
791 if (first_time
) { clear_progress(); first_time
= 0; }
792 vob
= DVDOpenFile(dvd
, id_title(file
->id
),
794 ? DVD_READ_TITLE_VOBS
795 : DVD_READ_MENU_VOBS
);
797 bail("failed to open %s %u",
798 id_part(file
->id
) ?
"title" : "menu",
808 want
= end
- pos
; if (want
> BUFSECTORS
) want
= BUFSECTORS
;
809 n
= read_sectors(pos
, buf
, want
);
811 if (n
<= 0) n
= find_good_sector(&pos
, end
, buf
, BUFSECTORS
);
812 if (n
> 0) { carefully_write(outfd
, buf
, n
*SECTORSZ
); pos
+= n
; }
813 report_progress(pos
); fflush(stdout
);
816 if (vob
) { DVDCloseFile(vob
); vob
= 0; }
822 static void logfn(void *p
, dvd_logger_level_t lev
,
823 const char *fmt
, va_list ap
)
826 case DVD_LOGGER_LEVEL_ERROR
:
827 fprintf("%s (libdvdread error): ", prog
);
829 case DVD_LOGGER_LEVEL_WARN
:
830 fprintf("%s (libdvdread warning): ", prog
);
835 vfprintf(stderr
, fmt
, ap
);
838 static const dvd_logger_cb logger
= { logfn
};
845 #define BUF_INIT { 0, 0, 0 }
846 #define BUF_REWIND(b) do { (b)->n = 0; } while (0)
847 #define BUF_FREE(b) do { \
849 free(_b->p); _b->p = 0; _b->n = _b->sz = 0; \
851 #define BUF_PUTC(b, ch) do { \
852 struct buf *_b = (b); \
853 if (_b->n >= _b->sz) { \
854 _b->sz = _b->sz ? 2*_b->sz : 32; \
855 _b->p = realloc(_b->p, _b->sz); \
856 if (!_b->p) bail("out of memory allocating %zu bytes", _b->sz); \
858 _b->p[_b->n] = (ch); \
861 static int read_line(FILE *fp
, struct buf
*b
)
868 else if (ch
!= '\n') do {
869 BUF_PUTC(b
, ch
); b
->n
++;
871 } while (ch
!= EOF
&& ch
!= '\n');
876 #define PRF_HYPHEN 1u
877 static int parse_range(const char *p
, unsigned f
,
878 secaddr
*start_out
, secaddr
*end_out
)
882 unsigned long start
, end
;
887 start
= strtoul(p
, &q
, 0);
888 if (errno
|| start
>= SECLIMIT
) { rc
= -1; goto end
; }
889 *start_out
= start
; p
= q
;
890 } else if (!(f
&PRF_HYPHEN
))
891 { rc
= -1; goto end
; }
896 if (*p
!= '-') { rc
= -1; goto end
; }
899 if (!ISSPACE(*p
)) { rc
= -1; goto end
; }
900 do p
++; while (ISSPACE(*p
));
904 end
= strtoul(p
, &q
, 0);
905 if (errno
|| end
> SECLIMIT
|| end
< start
) { rc
= -1; goto end
; }
906 *end_out
= end
; p
= q
;
907 } else if (!(f
&PRF_HYPHEN
))
908 { rc
= -1; goto end
; }
910 if (!(f
&PRF_HYPHEN
)) while (ISSPACE(*p
)) p
++;
911 if (*p
) { rc
= -1; goto end
; }
919 int main(int argc
, char *argv
[])
926 secaddr start
, end
, last
;
927 const struct event
*ev
;
928 const char *device
, *outfile
;
929 struct badblock
*bad
;
934 struct buf buf
= BUF_INIT
;
937 const struct file
*file
;
942 #define f_continue 2u
946 p
= strrchr(argv
[0], '/'); prog
= p ? p
+ 1 : argv
[0];
948 opt
= getopt(argc
, argv
, "hFR:X:b:cr:"); if (opt
< 0) break;
950 case 'h': usage(stderr
); exit(0);
951 case 'F': f
|= f_fixup
; break;
953 fp
= fopen(optarg
, "r");
955 bail_syserr(errno
, "failed to open ranges 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
);
967 eventq
.v
[eventq
.n
- 1].pos
= end
;
969 put_event(EV_WRITE
, 0, start
);
970 put_event(EV_STOP
, 0, end
);
976 bail_syserr(errno
, "failed to read ranges file `%s'", optarg
);
979 fp
= fopen(optarg
, "r");
981 bail_syserr(errno
, "failed to open bad-blocks file `%s'", optarg
);
984 BUF_REWIND(&buf
); if (read_line(fp
, &buf
)) break;
986 while (ISSPACE(*p
)) p
++;
987 if (!*p
|| *p
== '#') continue;
988 if (parse_range(p
, 0, &start
, &end
) ||
989 (last
<= SECLIMIT
&& start
< last
))
990 bail("bad range `%s' at `%s' line %zu", buf
.p
, optarg
, i
);
992 { VEC_PUSH(bad
, &badblocks
); bad
->start
= start
; bad
->end
= end
; }
995 bail_syserr(errno
, "failed to read bad-blocks file `%s'", optarg
);
998 if (mapfile
) bail("can't have multiple map files");
1001 case 'c': f
|= f_continue
; break;
1003 start
= 0; end
= -1;
1004 if (parse_range(optarg
, PRF_HYPHEN
, &start
, &end
))
1005 bail("bad range `%s'", optarg
);
1007 put_event(EV_WRITE
, 0, start
);
1008 if (end
<= SECLIMIT
) put_event(EV_STOP
, 0, end
);
1011 default: f
|= f_bogus
; break;
1014 if (argc
- optind
!= 2) f
|= f_bogus
;
1015 if (f
&f_bogus
) { usage(stderr
); exit(2); }
1017 device
= argv
[optind
]; outfile
= argv
[optind
+ 1];
1020 qsort(badblocks
.v
, badblocks
.n
, sizeof(struct badblock
),
1023 printf(";; fake bad blocks:\n");
1024 for (i
= 0; i
< badblocks
.n
; i
++)
1025 printf(";;\t%8"PRIuSEC
" .. %"PRIuSEC
"\n",
1026 badblocks
.v
[i
].start
, badblocks
.v
[i
].end
);
1030 dvdfd
= open(device
, O_RDONLY
);
1032 bail_syserr(errno
, "failed to open device `%s'", device
);
1033 if (fstat(dvdfd
, &st
))
1034 bail_syserr(errno
, "failed to stat device `%s'", device
);
1035 if (S_ISREG(st
.st_mode
)) {
1038 } else if (S_ISBLK(st
.st_mode
)) {
1039 if (ioctl(dvdfd
, BLKSSZGET
, &blksz
))
1040 bail_syserr(errno
, "failed to get block size for `%s'", device
);
1041 if (ioctl(dvdfd
, BLKGETSIZE64
, &volsz
))
1042 bail_syserr(errno
, "failed to get volume size for `%s'", device
);
1044 bail("can't use `%s' as source: expected file or block device", device
);
1046 if (blksz
!= SECTORSZ
)
1047 bail("device `%s' block size %d /= %d", device
, blksz
, SECTORSZ
);
1049 bail("device `%s' volume size %"PRIu64
" not a multiple of %d",
1050 device
, volsz
, SECTORSZ
);
1053 outfd
= open(outfile
, O_WRONLY
| O_CREAT
, 0666);
1055 bail_syserr(errno
, "failed to create output file `%s'", outfile
);
1059 off
= lseek(outfd
, 0, SEEK_END
);
1061 bail_syserr(errno
, "failed to seek to end of output file `%s'",
1063 put_event(EV_WRITE
, 0, off
/SECTORSZ
);
1064 } else if (!eventq
.n
&& !(f
&f_fixup
))
1065 put_event(EV_WRITE
, 0, 0);
1068 dvd
= DVDOpen2(0, &logger
, device
);
1070 dvd
= DVDOpen(device
);
1072 if (!dvd
) bail("failed to open DVD on `%s'", device
);
1074 /* It's fast enough just to check everything. */
1076 for (i
= 1; i
< 100; i
++) {
1080 put_file(mkident(RAW
, 0, 0), 0, volsz
/SECTORSZ
);
1081 assert(filetab
.n
<= MAXFILES
);
1083 for (i
= 0, limit
= 0; i
< filetab
.n
; i
++)
1084 if (filetab
.v
[i
].end
> limit
) limit
= filetab
.v
[i
].end
;
1086 if (end
> limit
) end
= limit
;
1089 printf("\n;; files:\n");
1090 for (i
= 0; i
< filetab
.n
; i
++) {
1091 file
= &filetab
.v
[i
];
1092 store_filename(fn
, file
->id
);
1093 printf(";;\t%8"PRIuSEC
" %s\n", file
->start
, fn
);
1097 qsort(eventq
.v
, eventq
.n
, sizeof(struct event
), compare_event
);
1099 f
&= ~f_write
; start
= 0; n
= 0;
1100 for (i
= 0; i
< eventq
.n
; i
++) {
1105 bail("overlapping ranges: range from %"PRIuSEC
" still open at %"PRIuSEC
"",
1107 n
++; f
|= f_write
; start
= ev
->pos
;
1115 f
&= ~f_write
; start
= 0;
1116 for (i
= 0; i
< eventq
.n
; i
++) {
1119 case EV_WRITE
: start
= ev
->pos
; f
|= f_write
; break;
1120 case EV_STOP
: nsectors
+= ev
->pos
- start
; f
&= ~f_write
; break;
1122 if (ev
->pos
>= limit
) break;
1123 if (f
&f_fixup
) start
= ev
->pos
;
1127 put_event(EV_WRITE
, 0, start
);
1131 nsectors
+= limit
- start
;
1132 put_event(EV_STOP
, 0, limit
);
1134 if (n
== 1 && (f
&f_write
)) flags
|= F_ALLPROGRESS
;
1138 printf("\n;; event sweep:\n");
1140 for (pos
= 0, i
= 0; i
< eventq
.n
; i
++) {
1142 if (ev
->pos
> pos
) {
1143 if (f
&f_write
) emit(pos
, ev
->pos
);
1146 debug_clear_progress();
1154 store_filename(fn
, filetab
.v
[ev
->file
].id
);
1155 debug_clear_progress();
1156 printf(";; %8"PRIuSEC
": begin `%s'\n", pos
, fn
);
1160 gettimeofday(&last_time
, 0); last_pos
= pos
;
1161 if (lseek(outfd
, (off_t
)ev
->pos
*SECTORSZ
, SEEK_SET
) < 0)
1163 "failed to seek to resume position "
1164 "(sector %"PRIuSEC
") in output file `%s'",
1167 debug_clear_progress();
1168 printf(";; %8"PRIuSEC
": begin write\n", pos
);
1175 debug_clear_progress();
1176 printf(";; %8"PRIuSEC
": end write\n", pos
);
1180 clear_live(ev
->file
);
1182 store_filename(fn
, filetab
.v
[ev
->file
].id
);
1183 debug_clear_progress();
1184 printf(";; %8"PRIuSEC
": end `%s'\n", pos
, fn
);
1191 if (progresslen
) putchar('\n');
1193 if (ftruncate(outfd
, (off_t
)limit
*SECTORSZ
) < 0)
1194 bail_syserr(errno
, "failed to set output file `%s' length", outfile
);
1196 if (dvd
) DVDClose(dvd
);
1197 if (dvdfd
>= 0) close(dvdfd
);
1198 if (outfd
>= 0) close(outfd
);
1199 carefully_fclose(mapfp
, "bad-sector region map");