dvdrip: Leave a note in the working directory about which device we're using.

[dvdrip] / lib.c

#include "lib.h"

const char *prog = "<unset>";

void set_prog(const char *p)
  { const char *q = strrchr(p, '/'); prog = q ? q + 1 : p; }

void vmoan(const char *fmt, va_list ap)
  { fprintf(stderr, "%s: ", prog); vfprintf(stderr, fmt, ap); }

__attribute__((format(printf, 1, 2)))
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)))
void bail(const char *fmt, ...)
{
  va_list ap;

  va_start(ap, fmt); vmoan(fmt, ap); va_end(ap);
  fputc('\n', stderr);
  exit(2);
}

__attribute__((noreturn, format(printf, 2, 3)))
void bail_syserr(int err, const char *fmt, ...)
{
  va_list ap;

  va_start(ap, fmt); vmoan(fmt, ap); va_end(ap);
  if (err) fprintf(stderr, ": %s", strerror(errno));
  fputc('\n', stderr);
  exit(2);
}

void sit(double t)
{
  struct timeval tv;
  double whole = floor(t);

  if (t) {
    tv.tv_sec = whole; tv.tv_usec = floor((t - whole)*1.0e6) + 1;
    if (select(0, 0, 0, 0, &tv) < 0) bail_syserr(errno, "failed to sleep");
  }
}

void carefully_write(int fd, const void *buf, size_t sz)
{
  const unsigned char *p = buf;
  ssize_t n;

  if (fd < 0) return;
  while (sz) {
    n = write(fd, p, sz);
    if (n < 0) {
      if (errno == EINTR) continue;
      bail_syserr(errno, "failed to write to output file");
    }
    if (!n) bail("unexpected short write to output file");
    p += n; sz -= n;
  }
}

void open_file_on_demand(const char *file, FILE **fp_inout, const char *what)
{
  FILE *fp;

  if (!*fp_inout) {
    fp = fopen(file, "w");
    if (!fp) bail_syserr(errno, "failed to open %s file `%s'", what, file);
    fprintf(fp, "## %s\n\n", what);
    *fp_inout = fp;
  }
}

void check_write(FILE *fp, const char *what)
{
  fflush(fp);
  if (ferror(fp)) bail_syserr(errno, "error writing %s file", what);
}

void carefully_fclose(FILE *fp, const char *what)
{
  if (fp && (ferror(fp) || fclose(fp)))
    bail_syserr(errno, "error writing %s file", what);
}

void store_filename(char *buf, ident id)
{
  switch (id_kind(id)) {
    case RAW:
      sprintf(buf, "#<raw device>");
      break;
    case IFO:
      if (!id_title(id)) sprintf(buf, "/VIDEO_TS/VIDEO_TS.IFO");
      else sprintf(buf, "/VIDEO_TS/VTS_%02u_0.IFO", id_title(id));
      break;
    case BUP:
      if (!id_title(id)) sprintf(buf, "/VIDEO_TS/VIDEO_TS.BUP");
      else sprintf(buf, "/VIDEO_TS/VTS_%02u_0.BUP", id_title(id));
      break;
    case VOB:
      if (!id_title(id)) sprintf(buf, "/VIDEO_TS/VIDEO_TS.VOB");
      else
	sprintf(buf, "/VIDEO_TS/VTS_%02u_%u.VOB", id_title(id), id_part(id));
      break;
    default:
      abort();
  }
}

struct progress_state progress = PROGRESS_STATE_INIT;
static struct banner_progress_item banner_progress;

static void render_banner_progress(struct progress_item *item,
			    struct progress_render_state *render)
{
  struct banner_progress_item *bi = (struct banner_progress_item *)item;

  progress_putleft(render, " %s", bi->msg);
  progress_shownotice(render, 4, 7);
}

void show_banner(const char *msg)
{
  banner_progress._base.render = render_banner_progress;
  progress_additem(&progress, &banner_progress._base);
  banner_progress.msg = msg;
  progress_update(&progress);
}

void hide_banner(void)
{
  if (!progress_removeitem(&progress, &banner_progress._base))
    progress_update(&progress);
}

#ifdef notdef
static void logfn(void *p, dvd_logger_level_t lev,
		  const char *fmt, va_list ap)
{
  switch (lev) {
    case DVD_LOGGER_LEVEL_ERROR:
      fprintf("%s (libdvdread error): ", prog);
      break;
    case DVD_LOGGER_LEVEL_WARN:
      fprintf("%s (libdvdread warning): ", prog);
      break;
    default:
      return;
  }
  vfprintf(stderr, fmt, ap);
  fputc('\n', stderr);
}
static const dvd_logger_cb logger = { logfn };
#endif

void open_dvd(const char *device, int *fd_out, dvd_reader_t **dvd_out)
{
  int fd;
  dvd_reader_t *dvd;
  int bannerp = 0;

  for (;;) {
    fd = open(device, O_RDONLY);
    if (fd >= 0 || errno != ENOMEDIUM) break;
    if (!bannerp) {
      show_banner("Waiting for disc to be inserted...");
      bannerp = 1;
    }
    sit(0.2);
  }
  if (bannerp) hide_banner();
  if (fd < 0) bail_syserr(errno, "failed to open device `%s'", device);
  if (dvd_out) {
#ifdef notdef
    dvd = DVDOpen2(0, &logger, device);
#else
    dvd = DVDOpen(device);
#endif
    if (!dvd) bail("failed to open DVD on `%s'", device);
    *dvd_out = dvd;
  }
  if (fd_out) *fd_out = fd;
  else close(fd);
}