stream/stream_ffmpeg.c

   1 #include "config.h"
   2
   3 #include "libavformat/avformat.h"
   4 #include "libavformat/avio.h"
   5 #include "mp_msg.h"
   6 #include "stream.h"
   7 #include "m_option.h"
   8 #include "m_struct.h"
   9
  10 static int fill_buffer(stream_t *s, char *buffer, int max_len)
  11 {
  12     int r = url_read(s->priv, buffer, max_len);
  13     return (r <= 0) ? -1 : r;
  14 }
  15
  16 static int write_buffer(stream_t *s, char *buffer, int len)
  17 {
  18     int r = url_write(s->priv, buffer, len);
  19     return (r <= 0) ? -1 : r;
  20 }
  21
  22 static int seek(stream_t *s, off_t newpos)
  23 {
  24     s->pos = newpos;
  25     if (url_seek(s->priv, s->pos, SEEK_SET) < 0) {
  26         s->eof = 1;
  27         return 0;
  28     }
  29     return 1;
  30 }
  31
  32 static int control(stream_t *s, int cmd, void *arg)
  33 {
  34     int64_t size;
  35     switch(cmd) {
  36     case STREAM_CTRL_GET_SIZE:
  37         size = url_filesize(s->priv);
  38         if(size >= 0) {
  39             *(off_t *)arg = size;
  40             return 1;
  41         }
  42     }
  43     return STREAM_UNSUPPORTED;
  44 }
  45
  46 static void close_f(stream_t *stream)
  47 {
  48     url_close(stream->priv);
  49 }
  50
  51 static const char prefix[] = "ffmpeg://";
  52
  53 static int open_f(stream_t *stream, int mode, void *opts, int *file_format)
  54 {
  55     int flags = 0;
  56     const char *filename;
  57     URLContext *ctx = NULL;
  58     int res = STREAM_ERROR;
  59     int64_t size;
  60
  61     av_register_all();
  62     if (mode == STREAM_READ)
  63         flags = URL_RDONLY;
  64     else if (mode == STREAM_WRITE)
  65         flags = URL_WRONLY;
  66     else {
  67         mp_msg(MSGT_OPEN, MSGL_ERR, "[ffmpeg] Unknown open mode %d\n", mode);
  68         res = STREAM_UNSUPPORTED;
  69         goto out;
  70     }
  71
  72     if (stream->url)
  73         filename = stream->url;
  74     else {
  75         mp_msg(MSGT_OPEN, MSGL_ERR, "[ffmpeg] No URL\n");
  76         goto out;
  77     }
  78     if (!strncmp(filename, prefix, strlen(prefix)))
  79         filename += strlen(prefix);
  80     mp_msg(MSGT_OPEN, MSGL_V, "[ffmpeg] Opening %s\n", filename);
  81
  82     if (url_open(&ctx, filename, flags) < 0)
  83         goto out;
  84
  85     stream->priv = ctx;
  86     size = url_filesize(ctx);
  87     if (size >= 0)
  88         stream->end_pos = size;
  89     stream->type = STREAMTYPE_FILE;
  90     stream->seek = seek;
  91     if (ctx->is_streamed) {
  92         stream->type = STREAMTYPE_STREAM;
  93         stream->seek = NULL;
  94     }
  95     stream->fill_buffer = fill_buffer;
  96     stream->write_buffer = write_buffer;
  97     stream->control = control;
  98     stream->close = close_f;
  99     res = STREAM_OK;
 100
 101 out:
 102     return res;
 103 }
 104
 105 const stream_info_t stream_info_ffmpeg = {
 106   "FFmpeg",
 107   "ffmpeg",
 108   "",
 109   "",
 110   open_f,
 111   { "ffmpeg", "rtmp", NULL },
 112   NULL,
 113   1 // Urls are an option string
 114 };