avio_internal.h 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. /*
  2. * This file is part of FFmpeg.
  3. *
  4. * FFmpeg is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * FFmpeg is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with FFmpeg; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. #ifndef AVFORMAT_AVIO_INTERNAL_H
  19. #define AVFORMAT_AVIO_INTERNAL_H
  20. #include "avio.h"
  21. #include "url.h"
  22. #include "libavutil/log.h"
  23. extern const AVClass ff_avio_class;
  24. int ffio_init_context(AVIOContext *s,
  25. unsigned char *buffer,
  26. int buffer_size,
  27. int write_flag,
  28. void *opaque,
  29. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
  30. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
  31. int64_t (*seek)(void *opaque, int64_t offset, int whence));
  32. /**
  33. * Read size bytes from AVIOContext, returning a pointer.
  34. * Note that the data pointed at by the returned pointer is only
  35. * valid until the next call that references the same IO context.
  36. * @param s IO context
  37. * @param buf pointer to buffer into which to assemble the requested
  38. * data if it is not available in contiguous addresses in the
  39. * underlying buffer
  40. * @param size number of bytes requested
  41. * @param data address at which to store pointer: this will be a
  42. * a direct pointer into the underlying buffer if the requested
  43. * number of bytes are available at contiguous addresses, otherwise
  44. * will be a copy of buf
  45. * @return number of bytes read or AVERROR
  46. */
  47. int ffio_read_indirect(AVIOContext *s, unsigned char *buf, int size, const unsigned char **data);
  48. void ffio_fill(AVIOContext *s, int b, int count);
  49. static av_always_inline void ffio_wfourcc(AVIOContext *pb, const uint8_t *s)
  50. {
  51. avio_wl32(pb, MKTAG(s[0], s[1], s[2], s[3]));
  52. }
  53. /**
  54. * Rewind the AVIOContext using the specified buffer containing the first buf_size bytes of the file.
  55. * Used after probing to avoid seeking.
  56. * Joins buf and s->buffer, taking any overlap into consideration.
  57. * @note s->buffer must overlap with buf or they can't be joined and the function fails
  58. *
  59. * @param s The read-only AVIOContext to rewind
  60. * @param buf The probe buffer containing the first buf_size bytes of the file
  61. * @param buf_size The size of buf
  62. * @return >= 0 in case of success, a negative value corresponding to an
  63. * AVERROR code in case of failure
  64. */
  65. int ffio_rewind_with_probe_data(AVIOContext *s, unsigned char **buf, int buf_size);
  66. uint64_t ffio_read_varlen(AVIOContext *bc);
  67. /**
  68. * Read size bytes from AVIOContext into buf.
  69. * Check that exactly size bytes have been read.
  70. * @return number of bytes read or AVERROR
  71. */
  72. int ffio_read_size(AVIOContext *s, unsigned char *buf, int size);
  73. /** @warning must be called before any I/O */
  74. int ffio_set_buf_size(AVIOContext *s, int buf_size);
  75. /**
  76. * Reallocate a given buffer for AVIOContext.
  77. *
  78. * @param s the AVIOContext to realloc.
  79. * @param buf_size required new buffer size.
  80. * @return 0 on success, a negative AVERROR on failure.
  81. */
  82. int ffio_realloc_buf(AVIOContext *s, int buf_size);
  83. /**
  84. * Ensures that the requested seekback buffer size will be available
  85. *
  86. * Will ensure that when reading sequentially up to buf_size, seeking
  87. * within the current pos and pos+buf_size is possible.
  88. * Once the stream position moves outside this window this guarantee is lost.
  89. */
  90. int ffio_ensure_seekback(AVIOContext *s, int64_t buf_size);
  91. int ffio_limit(AVIOContext *s, int size);
  92. void ffio_init_checksum(AVIOContext *s,
  93. unsigned long (*update_checksum)(unsigned long c, const uint8_t *p, unsigned int len),
  94. unsigned long checksum);
  95. unsigned long ffio_get_checksum(AVIOContext *s);
  96. unsigned long ff_crc04C11DB7_update(unsigned long checksum, const uint8_t *buf,
  97. unsigned int len);
  98. unsigned long ff_crcEDB88320_update(unsigned long checksum, const uint8_t *buf,
  99. unsigned int len);
  100. unsigned long ff_crcA001_update(unsigned long checksum, const uint8_t *buf,
  101. unsigned int len);
  102. /**
  103. * Open a write only packetized memory stream with a maximum packet
  104. * size of 'max_packet_size'. The stream is stored in a memory buffer
  105. * with a big-endian 4 byte header giving the packet size in bytes.
  106. *
  107. * @param s new IO context
  108. * @param max_packet_size maximum packet size (must be > 0)
  109. * @return zero if no error.
  110. */
  111. int ffio_open_dyn_packet_buf(AVIOContext **s, int max_packet_size);
  112. /**
  113. * Create and initialize a AVIOContext for accessing the
  114. * resource referenced by the URLContext h.
  115. * @note When the URLContext h has been opened in read+write mode, the
  116. * AVIOContext can be used only for writing.
  117. *
  118. * @param s Used to return the pointer to the created AVIOContext.
  119. * In case of failure the pointed to value is set to NULL.
  120. * @return >= 0 in case of success, a negative value corresponding to an
  121. * AVERROR code in case of failure
  122. */
  123. int ffio_fdopen(AVIOContext **s, URLContext *h);
  124. /**
  125. * Return the URLContext associated with the AVIOContext
  126. *
  127. * @param s IO context
  128. * @return pointer to URLContext or NULL.
  129. */
  130. URLContext *ffio_geturlcontext(AVIOContext *s);
  131. /**
  132. * Open a write-only fake memory stream. The written data is not stored
  133. * anywhere - this is only used for measuring the amount of data
  134. * written.
  135. *
  136. * @param s new IO context
  137. * @return zero if no error.
  138. */
  139. int ffio_open_null_buf(AVIOContext **s);
  140. int ffio_open_whitelist(AVIOContext **s, const char *url, int flags,
  141. const AVIOInterruptCB *int_cb, AVDictionary **options,
  142. const char *whitelist, const char *blacklist);
  143. /**
  144. * Close a null buffer.
  145. *
  146. * @param s an IO context opened by ffio_open_null_buf
  147. * @return the number of bytes written to the null buffer
  148. */
  149. int ffio_close_null_buf(AVIOContext *s);
  150. /**
  151. * Reset a dynamic buffer.
  152. *
  153. * Resets everything, but keeps the allocated buffer for later use.
  154. */
  155. void ffio_reset_dyn_buf(AVIOContext *s);
  156. /**
  157. * Free a dynamic buffer.
  158. *
  159. * @param s a pointer to an IO context opened by avio_open_dyn_buf()
  160. */
  161. void ffio_free_dyn_buf(AVIOContext **s);
  162. #endif /* AVFORMAT_AVIO_INTERNAL_H */