You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

965 lines
23KB

  1. /*
  2. * Buffered I/O for ffmpeg system
  3. * Copyright (c) 2000,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. #include "libavutil/crc.h"
  22. #include "libavutil/intreadwrite.h"
  23. #include "avformat.h"
  24. #include "avio.h"
  25. #include "internal.h"
  26. #include <stdarg.h>
  27. #define IO_BUFFER_SIZE 32768
  28. /**
  29. * Do seeks within this distance ahead of the current buffer by skipping
  30. * data instead of calling the protocol seek function, for seekable
  31. * protocols.
  32. */
  33. #define SHORT_SEEK_THRESHOLD 4096
  34. static void fill_buffer(ByteIOContext *s);
  35. #if !FF_API_URL_RESETBUF
  36. static int url_resetbuf(ByteIOContext *s, int flags);
  37. #endif
  38. int init_put_byte(ByteIOContext *s,
  39. unsigned char *buffer,
  40. int buffer_size,
  41. int write_flag,
  42. void *opaque,
  43. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
  44. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
  45. int64_t (*seek)(void *opaque, int64_t offset, int whence))
  46. {
  47. s->buffer = buffer;
  48. s->buffer_size = buffer_size;
  49. s->buf_ptr = buffer;
  50. s->opaque = opaque;
  51. url_resetbuf(s, write_flag ? URL_WRONLY : URL_RDONLY);
  52. s->write_packet = write_packet;
  53. s->read_packet = read_packet;
  54. s->seek = seek;
  55. s->pos = 0;
  56. s->must_flush = 0;
  57. s->eof_reached = 0;
  58. s->error = 0;
  59. s->is_streamed = 0;
  60. s->max_packet_size = 0;
  61. s->update_checksum= NULL;
  62. if(!read_packet && !write_flag){
  63. s->pos = buffer_size;
  64. s->buf_end = s->buffer + buffer_size;
  65. }
  66. s->read_pause = NULL;
  67. s->read_seek = NULL;
  68. return 0;
  69. }
  70. ByteIOContext *av_alloc_put_byte(
  71. unsigned char *buffer,
  72. int buffer_size,
  73. int write_flag,
  74. void *opaque,
  75. int (*read_packet)(void *opaque, uint8_t *buf, int buf_size),
  76. int (*write_packet)(void *opaque, uint8_t *buf, int buf_size),
  77. int64_t (*seek)(void *opaque, int64_t offset, int whence))
  78. {
  79. ByteIOContext *s = av_mallocz(sizeof(ByteIOContext));
  80. init_put_byte(s, buffer, buffer_size, write_flag, opaque,
  81. read_packet, write_packet, seek);
  82. return s;
  83. }
  84. static void flush_buffer(ByteIOContext *s)
  85. {
  86. if (s->buf_ptr > s->buffer) {
  87. if (s->write_packet && !s->error){
  88. int ret= s->write_packet(s->opaque, s->buffer, s->buf_ptr - s->buffer);
  89. if(ret < 0){
  90. s->error = ret;
  91. }
  92. }
  93. if(s->update_checksum){
  94. s->checksum= s->update_checksum(s->checksum, s->checksum_ptr, s->buf_ptr - s->checksum_ptr);
  95. s->checksum_ptr= s->buffer;
  96. }
  97. s->pos += s->buf_ptr - s->buffer;
  98. }
  99. s->buf_ptr = s->buffer;
  100. }
  101. void put_byte(ByteIOContext *s, int b)
  102. {
  103. *(s->buf_ptr)++ = b;
  104. if (s->buf_ptr >= s->buf_end)
  105. flush_buffer(s);
  106. }
  107. void put_nbyte(ByteIOContext *s, int b, int count)
  108. {
  109. while (count > 0) {
  110. int len = FFMIN(s->buf_end - s->buf_ptr, count);
  111. memset(s->buf_ptr, b, len);
  112. s->buf_ptr += len;
  113. if (s->buf_ptr >= s->buf_end)
  114. flush_buffer(s);
  115. count -= len;
  116. }
  117. }
  118. void put_buffer(ByteIOContext *s, const unsigned char *buf, int size)
  119. {
  120. while (size > 0) {
  121. int len = FFMIN(s->buf_end - s->buf_ptr, size);
  122. memcpy(s->buf_ptr, buf, len);
  123. s->buf_ptr += len;
  124. if (s->buf_ptr >= s->buf_end)
  125. flush_buffer(s);
  126. buf += len;
  127. size -= len;
  128. }
  129. }
  130. void put_flush_packet(ByteIOContext *s)
  131. {
  132. flush_buffer(s);
  133. s->must_flush = 0;
  134. }
  135. int64_t url_fseek(ByteIOContext *s, int64_t offset, int whence)
  136. {
  137. int64_t offset1;
  138. int64_t pos;
  139. int force = whence & AVSEEK_FORCE;
  140. whence &= ~AVSEEK_FORCE;
  141. if(!s)
  142. return AVERROR(EINVAL);
  143. pos = s->pos - (s->write_flag ? 0 : (s->buf_end - s->buffer));
  144. if (whence != SEEK_CUR && whence != SEEK_SET)
  145. return AVERROR(EINVAL);
  146. if (whence == SEEK_CUR) {
  147. offset1 = pos + (s->buf_ptr - s->buffer);
  148. if (offset == 0)
  149. return offset1;
  150. offset += offset1;
  151. }
  152. offset1 = offset - pos;
  153. if (!s->must_flush &&
  154. offset1 >= 0 && offset1 <= (s->buf_end - s->buffer)) {
  155. /* can do the seek inside the buffer */
  156. s->buf_ptr = s->buffer + offset1;
  157. } else if ((s->is_streamed ||
  158. offset1 <= s->buf_end + SHORT_SEEK_THRESHOLD - s->buffer) &&
  159. !s->write_flag && offset1 >= 0 &&
  160. (whence != SEEK_END || force)) {
  161. while(s->pos < offset && !s->eof_reached)
  162. fill_buffer(s);
  163. if (s->eof_reached)
  164. return AVERROR_EOF;
  165. s->buf_ptr = s->buf_end + offset - s->pos;
  166. } else {
  167. int64_t res;
  168. #if CONFIG_MUXERS || CONFIG_NETWORK
  169. if (s->write_flag) {
  170. flush_buffer(s);
  171. s->must_flush = 1;
  172. }
  173. #endif /* CONFIG_MUXERS || CONFIG_NETWORK */
  174. if (!s->seek)
  175. return AVERROR(EPIPE);
  176. if ((res = s->seek(s->opaque, offset, SEEK_SET)) < 0)
  177. return res;
  178. if (!s->write_flag)
  179. s->buf_end = s->buffer;
  180. s->buf_ptr = s->buffer;
  181. s->pos = offset;
  182. }
  183. s->eof_reached = 0;
  184. return offset;
  185. }
  186. int url_fskip(ByteIOContext *s, int64_t offset)
  187. {
  188. int64_t ret = url_fseek(s, offset, SEEK_CUR);
  189. return ret < 0 ? ret : 0;
  190. }
  191. int64_t url_ftell(ByteIOContext *s)
  192. {
  193. return url_fseek(s, 0, SEEK_CUR);
  194. }
  195. int64_t url_fsize(ByteIOContext *s)
  196. {
  197. int64_t size;
  198. if(!s)
  199. return AVERROR(EINVAL);
  200. if (!s->seek)
  201. return AVERROR(ENOSYS);
  202. size = s->seek(s->opaque, 0, AVSEEK_SIZE);
  203. if(size<0){
  204. if ((size = s->seek(s->opaque, -1, SEEK_END)) < 0)
  205. return size;
  206. size++;
  207. s->seek(s->opaque, s->pos, SEEK_SET);
  208. }
  209. return size;
  210. }
  211. int url_feof(ByteIOContext *s)
  212. {
  213. if(!s)
  214. return 0;
  215. return s->eof_reached;
  216. }
  217. int url_ferror(ByteIOContext *s)
  218. {
  219. if(!s)
  220. return 0;
  221. return s->error;
  222. }
  223. void put_le32(ByteIOContext *s, unsigned int val)
  224. {
  225. put_byte(s, val);
  226. put_byte(s, val >> 8);
  227. put_byte(s, val >> 16);
  228. put_byte(s, val >> 24);
  229. }
  230. void put_be32(ByteIOContext *s, unsigned int val)
  231. {
  232. put_byte(s, val >> 24);
  233. put_byte(s, val >> 16);
  234. put_byte(s, val >> 8);
  235. put_byte(s, val);
  236. }
  237. void put_strz(ByteIOContext *s, const char *str)
  238. {
  239. if (str)
  240. put_buffer(s, (const unsigned char *) str, strlen(str) + 1);
  241. else
  242. put_byte(s, 0);
  243. }
  244. int ff_get_v_length(uint64_t val){
  245. int i=1;
  246. while(val>>=7)
  247. i++;
  248. return i;
  249. }
  250. void ff_put_v(ByteIOContext *bc, uint64_t val){
  251. int i= ff_get_v_length(val);
  252. while(--i>0)
  253. put_byte(bc, 128 | (val>>(7*i)));
  254. put_byte(bc, val&127);
  255. }
  256. void put_le64(ByteIOContext *s, uint64_t val)
  257. {
  258. put_le32(s, (uint32_t)(val & 0xffffffff));
  259. put_le32(s, (uint32_t)(val >> 32));
  260. }
  261. void put_be64(ByteIOContext *s, uint64_t val)
  262. {
  263. put_be32(s, (uint32_t)(val >> 32));
  264. put_be32(s, (uint32_t)(val & 0xffffffff));
  265. }
  266. void put_le16(ByteIOContext *s, unsigned int val)
  267. {
  268. put_byte(s, val);
  269. put_byte(s, val >> 8);
  270. }
  271. void put_be16(ByteIOContext *s, unsigned int val)
  272. {
  273. put_byte(s, val >> 8);
  274. put_byte(s, val);
  275. }
  276. void put_le24(ByteIOContext *s, unsigned int val)
  277. {
  278. put_le16(s, val & 0xffff);
  279. put_byte(s, val >> 16);
  280. }
  281. void put_be24(ByteIOContext *s, unsigned int val)
  282. {
  283. put_be16(s, val >> 8);
  284. put_byte(s, val);
  285. }
  286. void put_tag(ByteIOContext *s, const char *tag)
  287. {
  288. while (*tag) {
  289. put_byte(s, *tag++);
  290. }
  291. }
  292. /* Input stream */
  293. static void fill_buffer(ByteIOContext *s)
  294. {
  295. uint8_t *dst= !s->max_packet_size && s->buf_end - s->buffer < s->buffer_size ? s->buf_ptr : s->buffer;
  296. int len= s->buffer_size - (dst - s->buffer);
  297. int max_buffer_size = s->max_packet_size ? s->max_packet_size : IO_BUFFER_SIZE;
  298. /* no need to do anything if EOF already reached */
  299. if (s->eof_reached)
  300. return;
  301. if(s->update_checksum && dst == s->buffer){
  302. if(s->buf_end > s->checksum_ptr)
  303. s->checksum= s->update_checksum(s->checksum, s->checksum_ptr, s->buf_end - s->checksum_ptr);
  304. s->checksum_ptr= s->buffer;
  305. }
  306. /* make buffer smaller in case it ended up large after probing */
  307. if (s->buffer_size > max_buffer_size) {
  308. url_setbufsize(s, max_buffer_size);
  309. s->checksum_ptr = dst = s->buffer;
  310. len = s->buffer_size;
  311. }
  312. if(s->read_packet)
  313. len = s->read_packet(s->opaque, dst, len);
  314. else
  315. len = 0;
  316. if (len <= 0) {
  317. /* do not modify buffer if EOF reached so that a seek back can
  318. be done without rereading data */
  319. s->eof_reached = 1;
  320. if(len<0)
  321. s->error= len;
  322. } else {
  323. s->pos += len;
  324. s->buf_ptr = dst;
  325. s->buf_end = dst + len;
  326. }
  327. }
  328. unsigned long ff_crc04C11DB7_update(unsigned long checksum, const uint8_t *buf,
  329. unsigned int len)
  330. {
  331. return av_crc(av_crc_get_table(AV_CRC_32_IEEE), checksum, buf, len);
  332. }
  333. unsigned long get_checksum(ByteIOContext *s)
  334. {
  335. s->checksum= s->update_checksum(s->checksum, s->checksum_ptr, s->buf_ptr - s->checksum_ptr);
  336. s->update_checksum= NULL;
  337. return s->checksum;
  338. }
  339. void init_checksum(ByteIOContext *s,
  340. unsigned long (*update_checksum)(unsigned long c, const uint8_t *p, unsigned int len),
  341. unsigned long checksum)
  342. {
  343. s->update_checksum= update_checksum;
  344. if(s->update_checksum){
  345. s->checksum= checksum;
  346. s->checksum_ptr= s->buf_ptr;
  347. }
  348. }
  349. /* XXX: put an inline version */
  350. int get_byte(ByteIOContext *s)
  351. {
  352. if (s->buf_ptr >= s->buf_end)
  353. fill_buffer(s);
  354. if (s->buf_ptr < s->buf_end)
  355. return *s->buf_ptr++;
  356. return 0;
  357. }
  358. int url_fgetc(ByteIOContext *s)
  359. {
  360. if (s->buf_ptr >= s->buf_end)
  361. fill_buffer(s);
  362. if (s->buf_ptr < s->buf_end)
  363. return *s->buf_ptr++;
  364. return URL_EOF;
  365. }
  366. int get_buffer(ByteIOContext *s, unsigned char *buf, int size)
  367. {
  368. int len, size1;
  369. size1 = size;
  370. while (size > 0) {
  371. len = s->buf_end - s->buf_ptr;
  372. if (len > size)
  373. len = size;
  374. if (len == 0) {
  375. if(size > s->buffer_size && !s->update_checksum){
  376. if(s->read_packet)
  377. len = s->read_packet(s->opaque, buf, size);
  378. if (len <= 0) {
  379. /* do not modify buffer if EOF reached so that a seek back can
  380. be done without rereading data */
  381. s->eof_reached = 1;
  382. if(len<0)
  383. s->error= len;
  384. break;
  385. } else {
  386. s->pos += len;
  387. size -= len;
  388. buf += len;
  389. s->buf_ptr = s->buffer;
  390. s->buf_end = s->buffer/* + len*/;
  391. }
  392. }else{
  393. fill_buffer(s);
  394. len = s->buf_end - s->buf_ptr;
  395. if (len == 0)
  396. break;
  397. }
  398. } else {
  399. memcpy(buf, s->buf_ptr, len);
  400. buf += len;
  401. s->buf_ptr += len;
  402. size -= len;
  403. }
  404. }
  405. if (size1 == size) {
  406. if (url_ferror(s)) return url_ferror(s);
  407. if (url_feof(s)) return AVERROR_EOF;
  408. }
  409. return size1 - size;
  410. }
  411. int get_partial_buffer(ByteIOContext *s, unsigned char *buf, int size)
  412. {
  413. int len;
  414. if(size<0)
  415. return -1;
  416. len = s->buf_end - s->buf_ptr;
  417. if (len == 0) {
  418. fill_buffer(s);
  419. len = s->buf_end - s->buf_ptr;
  420. }
  421. if (len > size)
  422. len = size;
  423. memcpy(buf, s->buf_ptr, len);
  424. s->buf_ptr += len;
  425. if (!len) {
  426. if (url_ferror(s)) return url_ferror(s);
  427. if (url_feof(s)) return AVERROR_EOF;
  428. }
  429. return len;
  430. }
  431. unsigned int get_le16(ByteIOContext *s)
  432. {
  433. unsigned int val;
  434. val = get_byte(s);
  435. val |= get_byte(s) << 8;
  436. return val;
  437. }
  438. unsigned int get_le24(ByteIOContext *s)
  439. {
  440. unsigned int val;
  441. val = get_le16(s);
  442. val |= get_byte(s) << 16;
  443. return val;
  444. }
  445. unsigned int get_le32(ByteIOContext *s)
  446. {
  447. unsigned int val;
  448. val = get_le16(s);
  449. val |= get_le16(s) << 16;
  450. return val;
  451. }
  452. uint64_t get_le64(ByteIOContext *s)
  453. {
  454. uint64_t val;
  455. val = (uint64_t)get_le32(s);
  456. val |= (uint64_t)get_le32(s) << 32;
  457. return val;
  458. }
  459. unsigned int get_be16(ByteIOContext *s)
  460. {
  461. unsigned int val;
  462. val = get_byte(s) << 8;
  463. val |= get_byte(s);
  464. return val;
  465. }
  466. unsigned int get_be24(ByteIOContext *s)
  467. {
  468. unsigned int val;
  469. val = get_be16(s) << 8;
  470. val |= get_byte(s);
  471. return val;
  472. }
  473. unsigned int get_be32(ByteIOContext *s)
  474. {
  475. unsigned int val;
  476. val = get_be16(s) << 16;
  477. val |= get_be16(s);
  478. return val;
  479. }
  480. char *get_strz(ByteIOContext *s, char *buf, int maxlen)
  481. {
  482. int i = 0;
  483. char c;
  484. while ((c = get_byte(s))) {
  485. if (i < maxlen-1)
  486. buf[i++] = c;
  487. }
  488. buf[i] = 0; /* Ensure null terminated, but may be truncated */
  489. return buf;
  490. }
  491. int ff_get_line(ByteIOContext *s, char *buf, int maxlen)
  492. {
  493. int i = 0;
  494. char c;
  495. do {
  496. c = get_byte(s);
  497. if (c && i < maxlen-1)
  498. buf[i++] = c;
  499. } while (c != '\n' && c);
  500. buf[i] = 0;
  501. return i;
  502. }
  503. uint64_t get_be64(ByteIOContext *s)
  504. {
  505. uint64_t val;
  506. val = (uint64_t)get_be32(s) << 32;
  507. val |= (uint64_t)get_be32(s);
  508. return val;
  509. }
  510. uint64_t ff_get_v(ByteIOContext *bc){
  511. uint64_t val = 0;
  512. int tmp;
  513. do{
  514. tmp = get_byte(bc);
  515. val= (val<<7) + (tmp&127);
  516. }while(tmp&128);
  517. return val;
  518. }
  519. int url_fdopen(ByteIOContext **s, URLContext *h)
  520. {
  521. uint8_t *buffer;
  522. int buffer_size, max_packet_size;
  523. max_packet_size = url_get_max_packet_size(h);
  524. if (max_packet_size) {
  525. buffer_size = max_packet_size; /* no need to bufferize more than one packet */
  526. } else {
  527. buffer_size = IO_BUFFER_SIZE;
  528. }
  529. buffer = av_malloc(buffer_size);
  530. if (!buffer)
  531. return AVERROR(ENOMEM);
  532. *s = av_mallocz(sizeof(ByteIOContext));
  533. if(!*s) {
  534. av_free(buffer);
  535. return AVERROR(ENOMEM);
  536. }
  537. if (init_put_byte(*s, buffer, buffer_size,
  538. (h->flags & URL_WRONLY || h->flags & URL_RDWR), h,
  539. url_read, url_write, url_seek) < 0) {
  540. av_free(buffer);
  541. av_freep(s);
  542. return AVERROR(EIO);
  543. }
  544. (*s)->is_streamed = h->is_streamed;
  545. (*s)->max_packet_size = max_packet_size;
  546. if(h->prot) {
  547. (*s)->read_pause = (int (*)(void *, int))h->prot->url_read_pause;
  548. (*s)->read_seek = (int64_t (*)(void *, int, int64_t, int))h->prot->url_read_seek;
  549. }
  550. return 0;
  551. }
  552. int url_setbufsize(ByteIOContext *s, int buf_size)
  553. {
  554. uint8_t *buffer;
  555. buffer = av_malloc(buf_size);
  556. if (!buffer)
  557. return AVERROR(ENOMEM);
  558. av_free(s->buffer);
  559. s->buffer = buffer;
  560. s->buffer_size = buf_size;
  561. s->buf_ptr = buffer;
  562. url_resetbuf(s, s->write_flag ? URL_WRONLY : URL_RDONLY);
  563. return 0;
  564. }
  565. #if FF_API_URL_RESETBUF
  566. int url_resetbuf(ByteIOContext *s, int flags)
  567. #else
  568. static int url_resetbuf(ByteIOContext *s, int flags)
  569. #endif
  570. {
  571. #if FF_API_URL_RESETBUF
  572. if (flags & URL_RDWR)
  573. return AVERROR(EINVAL);
  574. #else
  575. assert(flags == URL_WRONLY || flags == URL_RDONLY);
  576. #endif
  577. if (flags & URL_WRONLY) {
  578. s->buf_end = s->buffer + s->buffer_size;
  579. s->write_flag = 1;
  580. } else {
  581. s->buf_end = s->buffer;
  582. s->write_flag = 0;
  583. }
  584. return 0;
  585. }
  586. int ff_rewind_with_probe_data(ByteIOContext *s, unsigned char *buf, int buf_size)
  587. {
  588. int64_t buffer_start;
  589. int buffer_size;
  590. int overlap, new_size, alloc_size;
  591. if (s->write_flag)
  592. return AVERROR(EINVAL);
  593. buffer_size = s->buf_end - s->buffer;
  594. /* the buffers must touch or overlap */
  595. if ((buffer_start = s->pos - buffer_size) > buf_size)
  596. return AVERROR(EINVAL);
  597. overlap = buf_size - buffer_start;
  598. new_size = buf_size + buffer_size - overlap;
  599. alloc_size = FFMAX(s->buffer_size, new_size);
  600. if (alloc_size > buf_size)
  601. if (!(buf = av_realloc(buf, alloc_size)))
  602. return AVERROR(ENOMEM);
  603. if (new_size > buf_size) {
  604. memcpy(buf + buf_size, s->buffer + overlap, buffer_size - overlap);
  605. buf_size = new_size;
  606. }
  607. av_free(s->buffer);
  608. s->buf_ptr = s->buffer = buf;
  609. s->buffer_size = alloc_size;
  610. s->pos = buf_size;
  611. s->buf_end = s->buf_ptr + buf_size;
  612. s->eof_reached = 0;
  613. s->must_flush = 0;
  614. return 0;
  615. }
  616. int url_fopen(ByteIOContext **s, const char *filename, int flags)
  617. {
  618. URLContext *h;
  619. int err;
  620. err = url_open(&h, filename, flags);
  621. if (err < 0)
  622. return err;
  623. err = url_fdopen(s, h);
  624. if (err < 0) {
  625. url_close(h);
  626. return err;
  627. }
  628. return 0;
  629. }
  630. int url_fclose(ByteIOContext *s)
  631. {
  632. URLContext *h = s->opaque;
  633. av_free(s->buffer);
  634. av_free(s);
  635. return url_close(h);
  636. }
  637. URLContext *url_fileno(ByteIOContext *s)
  638. {
  639. return s->opaque;
  640. }
  641. #if CONFIG_MUXERS
  642. int url_fprintf(ByteIOContext *s, const char *fmt, ...)
  643. {
  644. va_list ap;
  645. char buf[4096];
  646. int ret;
  647. va_start(ap, fmt);
  648. ret = vsnprintf(buf, sizeof(buf), fmt, ap);
  649. va_end(ap);
  650. put_buffer(s, buf, strlen(buf));
  651. return ret;
  652. }
  653. #endif //CONFIG_MUXERS
  654. char *url_fgets(ByteIOContext *s, char *buf, int buf_size)
  655. {
  656. int c;
  657. char *q;
  658. c = url_fgetc(s);
  659. if (c == EOF)
  660. return NULL;
  661. q = buf;
  662. for(;;) {
  663. if (c == EOF || c == '\n')
  664. break;
  665. if ((q - buf) < buf_size - 1)
  666. *q++ = c;
  667. c = url_fgetc(s);
  668. }
  669. if (buf_size > 0)
  670. *q = '\0';
  671. return buf;
  672. }
  673. int url_fget_max_packet_size(ByteIOContext *s)
  674. {
  675. return s->max_packet_size;
  676. }
  677. int av_url_read_fpause(ByteIOContext *s, int pause)
  678. {
  679. if (!s->read_pause)
  680. return AVERROR(ENOSYS);
  681. return s->read_pause(s->opaque, pause);
  682. }
  683. int64_t av_url_read_fseek(ByteIOContext *s, int stream_index,
  684. int64_t timestamp, int flags)
  685. {
  686. URLContext *h = s->opaque;
  687. int64_t ret;
  688. if (!s->read_seek)
  689. return AVERROR(ENOSYS);
  690. ret = s->read_seek(h, stream_index, timestamp, flags);
  691. if(ret >= 0) {
  692. int64_t pos;
  693. s->buf_ptr = s->buf_end; // Flush buffer
  694. pos = s->seek(h, 0, SEEK_CUR);
  695. if (pos >= 0)
  696. s->pos = pos;
  697. else if (pos != AVERROR(ENOSYS))
  698. ret = pos;
  699. }
  700. return ret;
  701. }
  702. /* url_open_dyn_buf and url_close_dyn_buf are used in rtp.c to send a response
  703. * back to the server even if CONFIG_MUXERS is false. */
  704. #if CONFIG_MUXERS || CONFIG_NETWORK
  705. /* buffer handling */
  706. int url_open_buf(ByteIOContext **s, uint8_t *buf, int buf_size, int flags)
  707. {
  708. int ret;
  709. *s = av_mallocz(sizeof(ByteIOContext));
  710. if(!*s)
  711. return AVERROR(ENOMEM);
  712. ret = init_put_byte(*s, buf, buf_size,
  713. (flags & URL_WRONLY || flags & URL_RDWR),
  714. NULL, NULL, NULL, NULL);
  715. if(ret != 0)
  716. av_freep(s);
  717. return ret;
  718. }
  719. int url_close_buf(ByteIOContext *s)
  720. {
  721. put_flush_packet(s);
  722. return s->buf_ptr - s->buffer;
  723. }
  724. /* output in a dynamic buffer */
  725. typedef struct DynBuffer {
  726. int pos, size, allocated_size;
  727. uint8_t *buffer;
  728. int io_buffer_size;
  729. uint8_t io_buffer[1];
  730. } DynBuffer;
  731. static int dyn_buf_write(void *opaque, uint8_t *buf, int buf_size)
  732. {
  733. DynBuffer *d = opaque;
  734. unsigned new_size, new_allocated_size;
  735. /* reallocate buffer if needed */
  736. new_size = d->pos + buf_size;
  737. new_allocated_size = d->allocated_size;
  738. if(new_size < d->pos || new_size > INT_MAX/2)
  739. return -1;
  740. while (new_size > new_allocated_size) {
  741. if (!new_allocated_size)
  742. new_allocated_size = new_size;
  743. else
  744. new_allocated_size += new_allocated_size / 2 + 1;
  745. }
  746. if (new_allocated_size > d->allocated_size) {
  747. d->buffer = av_realloc(d->buffer, new_allocated_size);
  748. if(d->buffer == NULL)
  749. return AVERROR(ENOMEM);
  750. d->allocated_size = new_allocated_size;
  751. }
  752. memcpy(d->buffer + d->pos, buf, buf_size);
  753. d->pos = new_size;
  754. if (d->pos > d->size)
  755. d->size = d->pos;
  756. return buf_size;
  757. }
  758. static int dyn_packet_buf_write(void *opaque, uint8_t *buf, int buf_size)
  759. {
  760. unsigned char buf1[4];
  761. int ret;
  762. /* packetized write: output the header */
  763. AV_WB32(buf1, buf_size);
  764. ret= dyn_buf_write(opaque, buf1, 4);
  765. if(ret < 0)
  766. return ret;
  767. /* then the data */
  768. return dyn_buf_write(opaque, buf, buf_size);
  769. }
  770. static int64_t dyn_buf_seek(void *opaque, int64_t offset, int whence)
  771. {
  772. DynBuffer *d = opaque;
  773. if (whence == SEEK_CUR)
  774. offset += d->pos;
  775. else if (whence == SEEK_END)
  776. offset += d->size;
  777. if (offset < 0 || offset > 0x7fffffffLL)
  778. return -1;
  779. d->pos = offset;
  780. return 0;
  781. }
  782. static int url_open_dyn_buf_internal(ByteIOContext **s, int max_packet_size)
  783. {
  784. DynBuffer *d;
  785. int ret;
  786. unsigned io_buffer_size = max_packet_size ? max_packet_size : 1024;
  787. if(sizeof(DynBuffer) + io_buffer_size < io_buffer_size)
  788. return -1;
  789. d = av_mallocz(sizeof(DynBuffer) + io_buffer_size);
  790. if (!d)
  791. return AVERROR(ENOMEM);
  792. *s = av_mallocz(sizeof(ByteIOContext));
  793. if(!*s) {
  794. av_free(d);
  795. return AVERROR(ENOMEM);
  796. }
  797. d->io_buffer_size = io_buffer_size;
  798. ret = init_put_byte(*s, d->io_buffer, io_buffer_size,
  799. 1, d, NULL,
  800. max_packet_size ? dyn_packet_buf_write : dyn_buf_write,
  801. max_packet_size ? NULL : dyn_buf_seek);
  802. if (ret == 0) {
  803. (*s)->max_packet_size = max_packet_size;
  804. } else {
  805. av_free(d);
  806. av_freep(s);
  807. }
  808. return ret;
  809. }
  810. int url_open_dyn_buf(ByteIOContext **s)
  811. {
  812. return url_open_dyn_buf_internal(s, 0);
  813. }
  814. int url_open_dyn_packet_buf(ByteIOContext **s, int max_packet_size)
  815. {
  816. if (max_packet_size <= 0)
  817. return -1;
  818. return url_open_dyn_buf_internal(s, max_packet_size);
  819. }
  820. int url_close_dyn_buf(ByteIOContext *s, uint8_t **pbuffer)
  821. {
  822. DynBuffer *d = s->opaque;
  823. int size;
  824. static const char padbuf[FF_INPUT_BUFFER_PADDING_SIZE] = {0};
  825. int padding = 0;
  826. /* don't attempt to pad fixed-size packet buffers */
  827. if (!s->max_packet_size) {
  828. put_buffer(s, padbuf, sizeof(padbuf));
  829. padding = FF_INPUT_BUFFER_PADDING_SIZE;
  830. }
  831. put_flush_packet(s);
  832. *pbuffer = d->buffer;
  833. size = d->size;
  834. av_free(d);
  835. av_free(s);
  836. return size - padding;
  837. }
  838. #endif /* CONFIG_MUXERS || CONFIG_NETWORK */