avio.h 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460
  1. /*
  2. * copyright (c) 2001 Fabrice Bellard
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #ifndef AVFORMAT_AVIO_H
  21. #define AVFORMAT_AVIO_H
  22. /**
  23. * @file
  24. * @ingroup lavf_io
  25. * Buffered I/O operations
  26. */
  27. #include <stdint.h>
  28. #include "libavutil/common.h"
  29. #include "libavutil/dict.h"
  30. #include "libavutil/log.h"
  31. #include "libavformat/version.h"
  32. #define AVIO_SEEKABLE_NORMAL 0x0001 /**< Seeking works like for a local file */
  33. /**
  34. * Callback for checking whether to abort blocking functions.
  35. * AVERROR_EXIT is returned in this case by the interrupted
  36. * function. During blocking operations, callback is called with
  37. * opaque as parameter. If the callback returns 1, the
  38. * blocking operation will be aborted.
  39. *
  40. * No members can be added to this struct without a major bump, if
  41. * new elements have been added after this struct in AVFormatContext
  42. * or AVIOContext.
  43. */
  44. typedef struct AVIOInterruptCB {
  45. int (*callback)(void*);
  46. void *opaque;
  47. } AVIOInterruptCB;
  48. /**
  49. * Bytestream IO Context.
  50. * New fields can be added to the end with minor version bumps.
  51. * Removal, reordering and changes to existing fields require a major
  52. * version bump.
  53. * sizeof(AVIOContext) must not be used outside libav*.
  54. *
  55. * @note None of the function pointers in AVIOContext should be called
  56. * directly, they should only be set by the client application
  57. * when implementing custom I/O. Normally these are set to the
  58. * function pointers specified in avio_alloc_context()
  59. */
  60. typedef struct AVIOContext {
  61. /**
  62. * A class for private options.
  63. *
  64. * If this AVIOContext is created by avio_open2(), av_class is set and
  65. * passes the options down to protocols.
  66. *
  67. * If this AVIOContext is manually allocated, then av_class may be set by
  68. * the caller.
  69. *
  70. * warning -- this field can be NULL, be sure to not pass this AVIOContext
  71. * to any av_opt_* functions in that case.
  72. */
  73. const AVClass *av_class;
  74. unsigned char *buffer; /**< Start of the buffer. */
  75. int buffer_size; /**< Maximum buffer size */
  76. unsigned char *buf_ptr; /**< Current position in the buffer */
  77. unsigned char *buf_end; /**< End of the data, may be less than
  78. buffer+buffer_size if the read function returned
  79. less data than requested, e.g. for streams where
  80. no more data has been received yet. */
  81. void *opaque; /**< A private pointer, passed to the read/write/seek/...
  82. functions. */
  83. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size);
  84. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size);
  85. int64_t (*seek)(void *opaque, int64_t offset, int whence);
  86. int64_t pos; /**< position in the file of the current buffer */
  87. int must_flush; /**< true if the next seek should flush */
  88. int eof_reached; /**< true if eof reached */
  89. int write_flag; /**< true if open for writing */
  90. int max_packet_size;
  91. unsigned long checksum;
  92. unsigned char *checksum_ptr;
  93. unsigned long (*update_checksum)(unsigned long checksum, const uint8_t *buf, unsigned int size);
  94. int error; /**< contains the error code or 0 if no error happened */
  95. /**
  96. * Pause or resume playback for network streaming protocols - e.g. MMS.
  97. */
  98. int (*read_pause)(void *opaque, int pause);
  99. /**
  100. * Seek to a given timestamp in stream with the specified stream_index.
  101. * Needed for some network streaming protocols which don't support seeking
  102. * to byte position.
  103. */
  104. int64_t (*read_seek)(void *opaque, int stream_index,
  105. int64_t timestamp, int flags);
  106. /**
  107. * A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
  108. */
  109. int seekable;
  110. /**
  111. * max filesize, used to limit allocations
  112. * This field is internal to libavformat and access from outside is not allowed.
  113. */
  114. int64_t maxsize;
  115. /**
  116. * avio_read and avio_write should if possible be satisfied directly
  117. * instead of going through a buffer, and avio_seek will always
  118. * call the underlying seek function directly.
  119. */
  120. int direct;
  121. /**
  122. * Bytes read statistic
  123. * This field is internal to libavformat and access from outside is not allowed.
  124. */
  125. int64_t bytes_read;
  126. /**
  127. * seek statistic
  128. * This field is internal to libavformat and access from outside is not allowed.
  129. */
  130. int seek_count;
  131. } AVIOContext;
  132. /* unbuffered I/O */
  133. /**
  134. * Return AVIO_FLAG_* access flags corresponding to the access permissions
  135. * of the resource in url, or a negative value corresponding to an
  136. * AVERROR code in case of failure. The returned access flags are
  137. * masked by the value in flags.
  138. *
  139. * @note This function is intrinsically unsafe, in the sense that the
  140. * checked resource may change its existence or permission status from
  141. * one call to another. Thus you should not trust the returned value,
  142. * unless you are sure that no other processes are accessing the
  143. * checked resource.
  144. */
  145. int avio_check(const char *url, int flags);
  146. /**
  147. * Allocate and initialize an AVIOContext for buffered I/O. It must be later
  148. * freed with av_free().
  149. *
  150. * @param buffer Memory block for input/output operations via AVIOContext.
  151. * The buffer must be allocated with av_malloc() and friends.
  152. * @param buffer_size The buffer size is very important for performance.
  153. * For protocols with fixed blocksize it should be set to this blocksize.
  154. * For others a typical size is a cache page, e.g. 4kb.
  155. * @param write_flag Set to 1 if the buffer should be writable, 0 otherwise.
  156. * @param opaque An opaque pointer to user-specific data.
  157. * @param read_packet A function for refilling the buffer, may be NULL.
  158. * @param write_packet A function for writing the buffer contents, may be NULL.
  159. * The function may not change the input buffers content.
  160. * @param seek A function for seeking to specified byte position, may be NULL.
  161. *
  162. * @return Allocated AVIOContext or NULL on failure.
  163. */
  164. AVIOContext *avio_alloc_context(
  165. unsigned char *buffer,
  166. int buffer_size,
  167. int write_flag,
  168. void *opaque,
  169. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
  170. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
  171. int64_t (*seek)(void *opaque, int64_t offset, int whence));
  172. void avio_w8(AVIOContext *s, int b);
  173. void avio_write(AVIOContext *s, const unsigned char *buf, int size);
  174. void avio_wl64(AVIOContext *s, uint64_t val);
  175. void avio_wb64(AVIOContext *s, uint64_t val);
  176. void avio_wl32(AVIOContext *s, unsigned int val);
  177. void avio_wb32(AVIOContext *s, unsigned int val);
  178. void avio_wl24(AVIOContext *s, unsigned int val);
  179. void avio_wb24(AVIOContext *s, unsigned int val);
  180. void avio_wl16(AVIOContext *s, unsigned int val);
  181. void avio_wb16(AVIOContext *s, unsigned int val);
  182. /**
  183. * Write a NULL-terminated string.
  184. * @return number of bytes written.
  185. */
  186. int avio_put_str(AVIOContext *s, const char *str);
  187. /**
  188. * Convert an UTF-8 string to UTF-16LE and write it.
  189. * @return number of bytes written.
  190. */
  191. int avio_put_str16le(AVIOContext *s, const char *str);
  192. /**
  193. * Passing this as the "whence" parameter to a seek function causes it to
  194. * return the filesize without seeking anywhere. Supporting this is optional.
  195. * If it is not supported then the seek function will return <0.
  196. */
  197. #define AVSEEK_SIZE 0x10000
  198. /**
  199. * Oring this flag as into the "whence" parameter to a seek function causes it to
  200. * seek by any means (like reopening and linear reading) or other normally unreasonble
  201. * means that can be extreemly slow.
  202. * This may be ignored by the seek code.
  203. */
  204. #define AVSEEK_FORCE 0x20000
  205. /**
  206. * fseek() equivalent for AVIOContext.
  207. * @return new position or AVERROR.
  208. */
  209. int64_t avio_seek(AVIOContext *s, int64_t offset, int whence);
  210. /**
  211. * Skip given number of bytes forward
  212. * @return new position or AVERROR.
  213. */
  214. int64_t avio_skip(AVIOContext *s, int64_t offset);
  215. /**
  216. * ftell() equivalent for AVIOContext.
  217. * @return position or AVERROR.
  218. */
  219. static av_always_inline int64_t avio_tell(AVIOContext *s)
  220. {
  221. return avio_seek(s, 0, SEEK_CUR);
  222. }
  223. /**
  224. * Get the filesize.
  225. * @return filesize or AVERROR
  226. */
  227. int64_t avio_size(AVIOContext *s);
  228. /**
  229. * feof() equivalent for AVIOContext.
  230. * @return non zero if and only if end of file
  231. */
  232. int url_feof(AVIOContext *s);
  233. /** @warning currently size is limited */
  234. int avio_printf(AVIOContext *s, const char *fmt, ...) av_printf_format(2, 3);
  235. /**
  236. * Force flushing of buffered data to the output s.
  237. *
  238. * Force the buffered data to be immediately written to the output,
  239. * without to wait to fill the internal buffer.
  240. */
  241. void avio_flush(AVIOContext *s);
  242. /**
  243. * Read size bytes from AVIOContext into buf.
  244. * @return number of bytes read or AVERROR
  245. */
  246. int avio_read(AVIOContext *s, unsigned char *buf, int size);
  247. /**
  248. * @name Functions for reading from AVIOContext
  249. * @{
  250. *
  251. * @note return 0 if EOF, so you cannot use it if EOF handling is
  252. * necessary
  253. */
  254. int avio_r8 (AVIOContext *s);
  255. unsigned int avio_rl16(AVIOContext *s);
  256. unsigned int avio_rl24(AVIOContext *s);
  257. unsigned int avio_rl32(AVIOContext *s);
  258. uint64_t avio_rl64(AVIOContext *s);
  259. unsigned int avio_rb16(AVIOContext *s);
  260. unsigned int avio_rb24(AVIOContext *s);
  261. unsigned int avio_rb32(AVIOContext *s);
  262. uint64_t avio_rb64(AVIOContext *s);
  263. /**
  264. * @}
  265. */
  266. /**
  267. * Read a string from pb into buf. The reading will terminate when either
  268. * a NULL character was encountered, maxlen bytes have been read, or nothing
  269. * more can be read from pb. The result is guaranteed to be NULL-terminated, it
  270. * will be truncated if buf is too small.
  271. * Note that the string is not interpreted or validated in any way, it
  272. * might get truncated in the middle of a sequence for multi-byte encodings.
  273. *
  274. * @return number of bytes read (is always <= maxlen).
  275. * If reading ends on EOF or error, the return value will be one more than
  276. * bytes actually read.
  277. */
  278. int avio_get_str(AVIOContext *pb, int maxlen, char *buf, int buflen);
  279. /**
  280. * Read a UTF-16 string from pb and convert it to UTF-8.
  281. * The reading will terminate when either a null or invalid character was
  282. * encountered or maxlen bytes have been read.
  283. * @return number of bytes read (is always <= maxlen)
  284. */
  285. int avio_get_str16le(AVIOContext *pb, int maxlen, char *buf, int buflen);
  286. int avio_get_str16be(AVIOContext *pb, int maxlen, char *buf, int buflen);
  287. /**
  288. * @name URL open modes
  289. * The flags argument to avio_open must be one of the following
  290. * constants, optionally ORed with other flags.
  291. * @{
  292. */
  293. #define AVIO_FLAG_READ 1 /**< read-only */
  294. #define AVIO_FLAG_WRITE 2 /**< write-only */
  295. #define AVIO_FLAG_READ_WRITE (AVIO_FLAG_READ|AVIO_FLAG_WRITE) /**< read-write pseudo flag */
  296. /**
  297. * @}
  298. */
  299. /**
  300. * Use non-blocking mode.
  301. * If this flag is set, operations on the context will return
  302. * AVERROR(EAGAIN) if they can not be performed immediately.
  303. * If this flag is not set, operations on the context will never return
  304. * AVERROR(EAGAIN).
  305. * Note that this flag does not affect the opening/connecting of the
  306. * context. Connecting a protocol will always block if necessary (e.g. on
  307. * network protocols) but never hang (e.g. on busy devices).
  308. * Warning: non-blocking protocols is work-in-progress; this flag may be
  309. * silently ignored.
  310. */
  311. #define AVIO_FLAG_NONBLOCK 8
  312. /**
  313. * Use direct mode.
  314. * avio_read and avio_write should if possible be satisfied directly
  315. * instead of going through a buffer, and avio_seek will always
  316. * call the underlying seek function directly.
  317. */
  318. #define AVIO_FLAG_DIRECT 0x8000
  319. /**
  320. * Create and initialize a AVIOContext for accessing the
  321. * resource indicated by url.
  322. * @note When the resource indicated by url has been opened in
  323. * read+write mode, the AVIOContext can be used only for writing.
  324. *
  325. * @param s Used to return the pointer to the created AVIOContext.
  326. * In case of failure the pointed to value is set to NULL.
  327. * @param flags flags which control how the resource indicated by url
  328. * is to be opened
  329. * @return 0 in case of success, a negative value corresponding to an
  330. * AVERROR code in case of failure
  331. */
  332. int avio_open(AVIOContext **s, const char *url, int flags);
  333. /**
  334. * Create and initialize a AVIOContext for accessing the
  335. * resource indicated by url.
  336. * @note When the resource indicated by url has been opened in
  337. * read+write mode, the AVIOContext can be used only for writing.
  338. *
  339. * @param s Used to return the pointer to the created AVIOContext.
  340. * In case of failure the pointed to value is set to NULL.
  341. * @param flags flags which control how the resource indicated by url
  342. * is to be opened
  343. * @param int_cb an interrupt callback to be used at the protocols level
  344. * @param options A dictionary filled with protocol-private options. On return
  345. * this parameter will be destroyed and replaced with a dict containing options
  346. * that were not found. May be NULL.
  347. * @return 0 in case of success, a negative value corresponding to an
  348. * AVERROR code in case of failure
  349. */
  350. int avio_open2(AVIOContext **s, const char *url, int flags,
  351. const AVIOInterruptCB *int_cb, AVDictionary **options);
  352. /**
  353. * Close the resource accessed by the AVIOContext s and free it.
  354. * This function can only be used if s was opened by avio_open().
  355. *
  356. * The internal buffer is automatically flushed before closing the
  357. * resource.
  358. *
  359. * @return 0 on success, an AVERROR < 0 on error.
  360. */
  361. int avio_close(AVIOContext *s);
  362. /**
  363. * Open a write only memory stream.
  364. *
  365. * @param s new IO context
  366. * @return zero if no error.
  367. */
  368. int avio_open_dyn_buf(AVIOContext **s);
  369. /**
  370. * Return the written size and a pointer to the buffer. The buffer
  371. * must be freed with av_free().
  372. * Padding of FF_INPUT_BUFFER_PADDING_SIZE is added to the buffer.
  373. *
  374. * @param s IO context
  375. * @param pbuffer pointer to a byte buffer
  376. * @return the length of the byte buffer
  377. */
  378. int avio_close_dyn_buf(AVIOContext *s, uint8_t **pbuffer);
  379. /**
  380. * Iterate through names of available protocols.
  381. *
  382. * @param opaque A private pointer representing current protocol.
  383. * It must be a pointer to NULL on first iteration and will
  384. * be updated by successive calls to avio_enum_protocols.
  385. * @param output If set to 1, iterate over output protocols,
  386. * otherwise over input protocols.
  387. *
  388. * @return A static string containing the name of current protocol or NULL
  389. */
  390. const char *avio_enum_protocols(void **opaque, int output);
  391. /**
  392. * Pause and resume playing - only meaningful if using a network streaming
  393. * protocol (e.g. MMS).
  394. * @param pause 1 for pause, 0 for resume
  395. */
  396. int avio_pause(AVIOContext *h, int pause);
  397. /**
  398. * Seek to a given timestamp relative to some component stream.
  399. * Only meaningful if using a network streaming protocol (e.g. MMS.).
  400. * @param stream_index The stream index that the timestamp is relative to.
  401. * If stream_index is (-1) the timestamp should be in AV_TIME_BASE
  402. * units from the beginning of the presentation.
  403. * If a stream_index >= 0 is used and the protocol does not support
  404. * seeking based on component streams, the call will fail.
  405. * @param timestamp timestamp in AVStream.time_base units
  406. * or if there is no stream specified then in AV_TIME_BASE units.
  407. * @param flags Optional combination of AVSEEK_FLAG_BACKWARD, AVSEEK_FLAG_BYTE
  408. * and AVSEEK_FLAG_ANY. The protocol may silently ignore
  409. * AVSEEK_FLAG_BACKWARD and AVSEEK_FLAG_ANY, but AVSEEK_FLAG_BYTE will
  410. * fail if used and not supported.
  411. * @return >= 0 on success
  412. * @see AVInputFormat::read_seek
  413. */
  414. int64_t avio_seek_time(AVIOContext *h, int stream_index,
  415. int64_t timestamp, int flags);
  416. #endif /* AVFORMAT_AVIO_H */