git.sesse.net Git - ffmpeg/blob - libavformat/avio.h

   1 /*
   2  * unbuffered io for ffmpeg system
   3  * copyright (c) 2001 Fabrice Bellard
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21 #ifndef AVIO_H
  22 #define AVIO_H
  23
  24 /* output byte stream handling */
  25
  26 typedef int64_t offset_t;
  27
  28 /* unbuffered I/O */
  29
  30 struct URLContext {
  31     struct URLProtocol *prot;
  32     int flags;
  33     int is_streamed;  /* true if streamed (no seek possible), default = false */
  34     int max_packet_size;  /* if non zero, the stream is packetized with this max packet size */
  35     void *priv_data;
  36 #if LIBAVFORMAT_VERSION_INT >= (52<<16)
  37     char *filename; /* specified filename */
  38 #else
  39     char filename[1]; /* specified filename */
  40 #endif
  41 };
  42
  43 typedef struct URLContext URLContext;
  44
  45 typedef struct URLPollEntry {
  46     URLContext *handle;
  47     int events;
  48     int revents;
  49 } URLPollEntry;
  50
  51 #define URL_RDONLY 0
  52 #define URL_WRONLY 1
  53 #define URL_RDWR   2
  54
  55 typedef int URLInterruptCB(void);
  56
  57 int url_open(URLContext **h, const char *filename, int flags);
  58 int url_read(URLContext *h, unsigned char *buf, int size);
  59 int url_write(URLContext *h, unsigned char *buf, int size);
  60 offset_t url_seek(URLContext *h, offset_t pos, int whence);
  61 int url_close(URLContext *h);
  62 int url_exist(const char *filename);
  63 offset_t url_filesize(URLContext *h);
  64 int url_get_max_packet_size(URLContext *h);
  65 void url_get_filename(URLContext *h, char *buf, int buf_size);
  66
  67 /* the callback is called in blocking functions to test regulary if
  68    asynchronous interruption is needed. -EINTR is returned in this
  69    case by the interrupted function. 'NULL' means no interrupt
  70    callback is given. */
  71 void url_set_interrupt_cb(URLInterruptCB *interrupt_cb);
  72
  73 /* not implemented */
  74 int url_poll(URLPollEntry *poll_table, int n, int timeout);
  75
  76 /**
  77  * passing this as the "whence" parameter to a seek function causes it to
  78  * return the filesize without seeking anywhere, supporting this is optional
  79  * if its not supprted then the seek function will return <0
  80  */
  81 #define AVSEEK_SIZE 0x10000
  82
  83 typedef struct URLProtocol {
  84     const char *name;
  85     int (*url_open)(URLContext *h, const char *filename, int flags);
  86     int (*url_read)(URLContext *h, unsigned char *buf, int size);
  87     int (*url_write)(URLContext *h, unsigned char *buf, int size);
  88     offset_t (*url_seek)(URLContext *h, offset_t pos, int whence);
  89     int (*url_close)(URLContext *h);
  90     struct URLProtocol *next;
  91 } URLProtocol;
  92
  93 extern URLProtocol *first_protocol;
  94 extern URLInterruptCB *url_interrupt_cb;
  95
  96 int register_protocol(URLProtocol *protocol);
  97
  98 typedef struct {
  99     unsigned char *buffer;
 100     int buffer_size;
 101     unsigned char *buf_ptr, *buf_end;
 102     void *opaque;
 103     int (*read_packet)(void *opaque, uint8_t *buf, int buf_size);
 104     int (*write_packet)(void *opaque, uint8_t *buf, int buf_size);
 105     offset_t (*seek)(void *opaque, offset_t offset, int whence);
 106     offset_t pos; /* position in the file of the current buffer */
 107     int must_flush; /* true if the next seek should flush */
 108     int eof_reached; /* true if eof reached */
 109     int write_flag;  /* true if open for writing */
 110     int is_streamed;
 111     int max_packet_size;
 112     unsigned long checksum;
 113     unsigned char *checksum_ptr;
 114     unsigned long (*update_checksum)(unsigned long checksum, const uint8_t *buf, unsigned int size);
 115     int error;         ///< contains the error code or 0 if no error happened
 116 } ByteIOContext;
 117
 118 int init_put_byte(ByteIOContext *s,
 119                   unsigned char *buffer,
 120                   int buffer_size,
 121                   int write_flag,
 122                   void *opaque,
 123                   int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
 124                   int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
 125                   offset_t (*seek)(void *opaque, offset_t offset, int whence));
 126
 127 void put_byte(ByteIOContext *s, int b);
 128 void put_buffer(ByteIOContext *s, const unsigned char *buf, int size);
 129 void put_le64(ByteIOContext *s, uint64_t val);
 130 void put_be64(ByteIOContext *s, uint64_t val);
 131 void put_le32(ByteIOContext *s, unsigned int val);
 132 void put_be32(ByteIOContext *s, unsigned int val);
 133 void put_le24(ByteIOContext *s, unsigned int val);
 134 void put_be24(ByteIOContext *s, unsigned int val);
 135 void put_le16(ByteIOContext *s, unsigned int val);
 136 void put_be16(ByteIOContext *s, unsigned int val);
 137 void put_tag(ByteIOContext *s, const char *tag);
 138
 139 void put_strz(ByteIOContext *s, const char *buf);
 140
 141 offset_t url_fseek(ByteIOContext *s, offset_t offset, int whence);
 142 void url_fskip(ByteIOContext *s, offset_t offset);
 143 offset_t url_ftell(ByteIOContext *s);
 144 offset_t url_fsize(ByteIOContext *s);
 145 int url_feof(ByteIOContext *s);
 146 int url_ferror(ByteIOContext *s);
 147
 148 #define URL_EOF (-1)
 149 int url_fgetc(ByteIOContext *s);
 150 #ifdef __GNUC__
 151 int url_fprintf(ByteIOContext *s, const char *fmt, ...) __attribute__ ((__format__ (__printf__, 2, 3)));
 152 #else
 153 int url_fprintf(ByteIOContext *s, const char *fmt, ...);
 154 #endif
 155 char *url_fgets(ByteIOContext *s, char *buf, int buf_size);
 156
 157 void put_flush_packet(ByteIOContext *s);
 158
 159 int get_buffer(ByteIOContext *s, unsigned char *buf, int size);
 160 int get_partial_buffer(ByteIOContext *s, unsigned char *buf, int size);
 161 int get_byte(ByteIOContext *s);
 162 unsigned int get_le24(ByteIOContext *s);
 163 unsigned int get_le32(ByteIOContext *s);
 164 uint64_t get_le64(ByteIOContext *s);
 165 unsigned int get_le16(ByteIOContext *s);
 166
 167 char *get_strz(ByteIOContext *s, char *buf, int maxlen);
 168 unsigned int get_be16(ByteIOContext *s);
 169 unsigned int get_be24(ByteIOContext *s);
 170 unsigned int get_be32(ByteIOContext *s);
 171 uint64_t get_be64(ByteIOContext *s);
 172
 173 static inline int url_is_streamed(ByteIOContext *s)
 174 {
 175     return s->is_streamed;
 176 }
 177
 178 int url_fdopen(ByteIOContext *s, URLContext *h);
 179 int url_setbufsize(ByteIOContext *s, int buf_size);
 180 int url_fopen(ByteIOContext *s, const char *filename, int flags);
 181 int url_fclose(ByteIOContext *s);
 182 URLContext *url_fileno(ByteIOContext *s);
 183 int url_fget_max_packet_size(ByteIOContext *s);
 184
 185 int url_open_buf(ByteIOContext *s, uint8_t *buf, int buf_size, int flags);
 186 int url_close_buf(ByteIOContext *s);
 187
 188 int url_open_dyn_buf(ByteIOContext *s);
 189 int url_open_dyn_packet_buf(ByteIOContext *s, int max_packet_size);
 190 int url_close_dyn_buf(ByteIOContext *s, uint8_t **pbuffer);
 191
 192 unsigned long get_checksum(ByteIOContext *s);
 193 void init_checksum(ByteIOContext *s, unsigned long (*update_checksum)(unsigned long c, const uint8_t *p, unsigned int len), unsigned long checksum);
 194
 195 /* file.c */
 196 extern URLProtocol file_protocol;
 197 extern URLProtocol pipe_protocol;
 198
 199 /* udp.c */
 200 extern URLProtocol udp_protocol;
 201 int udp_set_remote_url(URLContext *h, const char *uri);
 202 int udp_get_local_port(URLContext *h);
 203 int udp_get_file_handle(URLContext *h);
 204
 205 /* tcp.c  */
 206 extern URLProtocol tcp_protocol;
 207
 208 /* http.c */
 209 extern URLProtocol http_protocol;
 210
 211 #endif
 212