3 * Common functions for the DVDrip C utilities.
5 * (c) 2022 Mark Wooding
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of the DVD ripping toolset.
12 * DVDrip is free software: you can redistribute it and/or modify it
13 * under the terms of the GNU General Public License as published by the
14 * Free Software Foundation; either version 3 of the License, or (at your
15 * option) any later version.
17 * DVDrip is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
22 * You should have received a copy of the GNU General Public License
23 * along with DVDrip. If not, see <https://www.gnu.org/licenses/>.
26 /*----- Header files ------------------------------------------------------*/
30 /*----- Diagnostics -------------------------------------------------------*/
32 const char *prog
= "<unset>";
34 void set_prog(const char *p
)
35 { const char *q
= strrchr(p
, '/'); prog
= q ? q
+ 1 : p
; }
37 void vmoan(const char *fmt
, va_list ap
)
38 { vmoan_syserr(0, fmt
, ap
); }
40 void vmoan_syserr(int err
, const char *fmt
, va_list ap
)
42 fprintf(stderr
, "%s: ", prog
);
43 vfprintf(stderr
, fmt
, ap
);
44 if (err
) fprintf(stderr
, ": %s", strerror(errno
));
48 void moan(const char *fmt
, ...)
49 { va_list ap
; va_start(ap
, fmt
); vmoan(fmt
, ap
); va_end(ap
); }
51 void moan_syserr(int err
, const char *fmt
, ...)
52 { va_list ap
; va_start(ap
, fmt
); vmoan_syserr(err
, fmt
, ap
); va_end(ap
); }
54 void bail(const char *fmt
, ...)
55 { va_list ap
; va_start(ap
, fmt
); vmoan(fmt
, ap
); va_end(ap
); exit(2); }
57 void bail_syserr(int err
, const char *fmt
, ...)
61 va_start(ap
, fmt
); vmoan_syserr(err
, fmt
, ap
); va_end(ap
);
65 /*----- Parsing utilities -------------------------------------------------*/
67 double parse_float(const char **p_inout
, unsigned f
,
68 double min
, double max
, const char *what
)
75 err
= errno
; errno
= 0;
78 if (errno
|| x
< min
|| x
> max
|| (!(f
&PNF_JUNK
) && *q
))
79 bail("bad %s `%s'", what
, p
);
80 *p_inout
= q
; errno
= err
;
84 long parse_int(const char **p_inout
, unsigned f
,
85 long min
, long max
, const char *what
)
92 err
= errno
; errno
= 0;
94 x
= strtoul(p
, &q
, 0);
95 if (errno
|| x
< min
|| x
> max
|| (!(f
&PNF_JUNK
) && *q
))
96 bail("bad %s `%s'", what
, p
);
97 *p_inout
= q
; errno
= err
;
101 /*----- System utilities --------------------------------------------------*/
106 double whole
= floor(t
);
109 tv
.tv_sec
= whole
; tv
.tv_usec
= floor((t
- whole
)*1.0e6
) + 1;
110 if (select(0, 0, 0, 0, &tv
) < 0) bail_syserr(errno
, "failed to sleep");
114 void carefully_write(int fd
, const void *buf
, size_t sz
)
116 const unsigned char *p
= buf
;
121 n
= write(fd
, p
, sz
);
123 if (errno
== EINTR
) continue;
124 bail_syserr(errno
, "failed to write to output file");
126 if (!n
) bail("unexpected short write to output file");
131 void open_file_on_demand(const char *file
, FILE **fp_inout
, const char *what
)
136 fp
= fopen(file
, "w");
137 if (!fp
) bail_syserr(errno
, "failed to open %s file `%s'", what
, file
);
138 fprintf(fp
, "## %s\n\n", what
);
143 void check_write(FILE *fp
, const char *what
)
146 if (ferror(fp
)) bail_syserr(errno
, "error writing %s file", what
);
149 void carefully_fclose(FILE *fp
, const char *what
)
151 if (fp
&& (ferror(fp
) || fclose(fp
)))
152 bail_syserr(errno
, "error writing %s file", what
);
155 off_t
device_size(int fd
, const char *file
, int *blksz_out
)
161 bail_syserr(errno
, "failed to obtain status for `%s'", file
);
162 if (S_ISREG(st
.st_mode
))
164 else if (S_ISBLK(st
.st_mode
)) {
165 if (ioctl(fd
, BLKGETSIZE64
, &volsz
))
166 bail_syserr(errno
, "failed to get volume size for `%s'", file
);
167 if (ioctl(fd
, BLKSSZGET
, blksz_out
))
168 bail_syserr(errno
, "failed to get block size for `%s'", file
);
170 bail("can't read size for `%s': expected file or block device", file
);
171 return ((off_t
)volsz
);
174 /*----- Progress utilities ------------------------------------------------*/
176 struct progress_state progress
= PROGRESS_STATE_INIT
;
177 static struct banner_progress_item banner_progress
;
179 static void render_banner_progress(struct progress_item
*item
,
180 struct progress_render_state
*render
)
182 struct banner_progress_item
*bi
= (struct banner_progress_item
*)item
;
184 progress_putleft(render
, " %s", bi
->msg
);
185 progress_shownotice(render
, 4, 7);
188 void show_banner(const char *msg
)
190 banner_progress
._base
.render
= render_banner_progress
;
191 progress_additem(&progress
, &banner_progress
._base
);
192 banner_progress
.msg
= msg
;
193 progress_update(&progress
);
196 void hide_banner(void)
198 if (!progress_removeitem(&progress
, &banner_progress
._base
))
199 progress_update(&progress
);
202 /*----- DVD utilities -----------------------------------------------------*/
205 static void logfn(void *p
, dvd_logger_level_t lev
,
206 const char *fmt
, va_list ap
)
209 case DVD_LOGGER_LEVEL_ERROR
:
210 fprintf("%s (libdvdread error): ", prog
);
212 case DVD_LOGGER_LEVEL_WARN
:
213 fprintf("%s (libdvdread warning): ", prog
);
218 vfprintf(stderr
, fmt
, ap
);
221 static const dvd_logger_cb logger
= { logfn
};
224 void open_dvd(const char *device
, int mode
,
225 int *fd_out
, dvd_reader_t
**dvd_out
)
232 fd
= open(device
, mode
);
233 if (fd
>= 0 || errno
!= ENOMEDIUM
) break;
235 show_banner("Waiting for disc to be inserted...");
240 if (bannerp
) hide_banner();
241 if (fd
< 0) bail_syserr(errno
, "failed to open device `%s'", device
);
244 dvd
= DVDOpen2(0, &logger
, device
);
246 dvd
= DVDOpen(device
);
248 if (!dvd
) bail("failed to open DVD on `%s'", device
);
251 if (fd_out
) *fd_out
= fd
;
255 void store_filename(char *buf
, ident id
)
257 switch (id_kind(id
)) {
259 sprintf(buf
, "#<raw device>");
262 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.IFO");
263 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.IFO", id_title(id
));
266 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.BUP");
267 else sprintf(buf
, "/VIDEO_TS/VTS_%02u_0.BUP", id_title(id
));
270 if (!id_title(id
)) sprintf(buf
, "/VIDEO_TS/VIDEO_TS.VOB");
272 sprintf(buf
, "/VIDEO_TS/VTS_%02u_%u.VOB", id_title(id
), id_part(id
));
279 static char *copy_string(char *p
, const char *q
)
281 while (*q
) *p
++ = *q
++;
285 static char *copy_hex(char *p
, const unsigned char *q
, size_t sz
)
288 sprintf(p
, "%02x", *q
);
294 int dvd_id(char *p
, dvd_reader_t
*dvd
, unsigned f
, const char *file
)
297 unsigned char volsetid
[16], discid
[16];
301 rc
= DVDUDFVolumeInfo(dvd
,
302 volid
, sizeof(volid
),
303 volsetid
, sizeof(volsetid
));
305 p
= copy_string(p
, volid
);
307 for (n
= sizeof(volsetid
); n
&& !volsetid
[n
- 1]; n
--);
308 p
= copy_hex(p
, volsetid
, n
);
309 } else if (f
&DIF_MUSTVOLINF
) {
310 if (file
) moan("failed to read volume info for `%s'", file
);
311 else moan("failed to read volume info");
314 p
= copy_string(p
, "<error reading volume info>");
317 rc
= DVDDiscID(dvd
, discid
);
319 p
= copy_hex(p
, discid
, sizeof(discid
));
320 else if (f
&DIF_MUSTIFOHASH
) {
321 if (file
) moan("failed to determine disc id of `%s'", file
);
322 else moan("failed to determine disc id");
325 p
= copy_string(p
, "<error reading disc-id>");
330 /*----- That's all, folks -------------------------------------------------*/