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.

846 lines
24KB

  1. /*
  2. * AVPacket functions for libavcodec
  3. * Copyright (c) 2000, 2001, 2002 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 <string.h>
  22. #include "libavutil/avassert.h"
  23. #include "libavutil/common.h"
  24. #include "libavutil/internal.h"
  25. #include "libavutil/mathematics.h"
  26. #include "libavutil/mem.h"
  27. #include "bytestream.h"
  28. #include "internal.h"
  29. #include "packet.h"
  30. #include "packet_internal.h"
  31. void av_init_packet(AVPacket *pkt)
  32. {
  33. pkt->pts = AV_NOPTS_VALUE;
  34. pkt->dts = AV_NOPTS_VALUE;
  35. pkt->pos = -1;
  36. pkt->duration = 0;
  37. #if FF_API_CONVERGENCE_DURATION
  38. FF_DISABLE_DEPRECATION_WARNINGS
  39. pkt->convergence_duration = 0;
  40. FF_ENABLE_DEPRECATION_WARNINGS
  41. #endif
  42. pkt->flags = 0;
  43. pkt->stream_index = 0;
  44. pkt->buf = NULL;
  45. pkt->side_data = NULL;
  46. pkt->side_data_elems = 0;
  47. }
  48. AVPacket *av_packet_alloc(void)
  49. {
  50. AVPacket *pkt = av_mallocz(sizeof(AVPacket));
  51. if (!pkt)
  52. return pkt;
  53. av_init_packet(pkt);
  54. return pkt;
  55. }
  56. void av_packet_free(AVPacket **pkt)
  57. {
  58. if (!pkt || !*pkt)
  59. return;
  60. av_packet_unref(*pkt);
  61. av_freep(pkt);
  62. }
  63. static int packet_alloc(AVBufferRef **buf, int size)
  64. {
  65. int ret;
  66. if (size < 0 || size >= INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE)
  67. return AVERROR(EINVAL);
  68. ret = av_buffer_realloc(buf, size + AV_INPUT_BUFFER_PADDING_SIZE);
  69. if (ret < 0)
  70. return ret;
  71. memset((*buf)->data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
  72. return 0;
  73. }
  74. int av_new_packet(AVPacket *pkt, int size)
  75. {
  76. AVBufferRef *buf = NULL;
  77. int ret = packet_alloc(&buf, size);
  78. if (ret < 0)
  79. return ret;
  80. av_init_packet(pkt);
  81. pkt->buf = buf;
  82. pkt->data = buf->data;
  83. pkt->size = size;
  84. return 0;
  85. }
  86. void av_shrink_packet(AVPacket *pkt, int size)
  87. {
  88. if (pkt->size <= size)
  89. return;
  90. pkt->size = size;
  91. memset(pkt->data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
  92. }
  93. int av_grow_packet(AVPacket *pkt, int grow_by)
  94. {
  95. int new_size;
  96. av_assert0((unsigned)pkt->size <= INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE);
  97. if ((unsigned)grow_by >
  98. INT_MAX - (pkt->size + AV_INPUT_BUFFER_PADDING_SIZE))
  99. return AVERROR(ENOMEM);
  100. new_size = pkt->size + grow_by + AV_INPUT_BUFFER_PADDING_SIZE;
  101. if (pkt->buf) {
  102. size_t data_offset;
  103. uint8_t *old_data = pkt->data;
  104. if (pkt->data == NULL) {
  105. data_offset = 0;
  106. pkt->data = pkt->buf->data;
  107. } else {
  108. data_offset = pkt->data - pkt->buf->data;
  109. if (data_offset > INT_MAX - new_size)
  110. return AVERROR(ENOMEM);
  111. }
  112. if (new_size + data_offset > pkt->buf->size ||
  113. !av_buffer_is_writable(pkt->buf)) {
  114. int ret = av_buffer_realloc(&pkt->buf, new_size + data_offset);
  115. if (ret < 0) {
  116. pkt->data = old_data;
  117. return ret;
  118. }
  119. pkt->data = pkt->buf->data + data_offset;
  120. }
  121. } else {
  122. pkt->buf = av_buffer_alloc(new_size);
  123. if (!pkt->buf)
  124. return AVERROR(ENOMEM);
  125. if (pkt->size > 0)
  126. memcpy(pkt->buf->data, pkt->data, pkt->size);
  127. pkt->data = pkt->buf->data;
  128. }
  129. pkt->size += grow_by;
  130. memset(pkt->data + pkt->size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
  131. return 0;
  132. }
  133. int av_packet_from_data(AVPacket *pkt, uint8_t *data, int size)
  134. {
  135. if (size >= INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE)
  136. return AVERROR(EINVAL);
  137. pkt->buf = av_buffer_create(data, size + AV_INPUT_BUFFER_PADDING_SIZE,
  138. av_buffer_default_free, NULL, 0);
  139. if (!pkt->buf)
  140. return AVERROR(ENOMEM);
  141. pkt->data = data;
  142. pkt->size = size;
  143. return 0;
  144. }
  145. #if FF_API_AVPACKET_OLD_API
  146. FF_DISABLE_DEPRECATION_WARNINGS
  147. #define ALLOC_MALLOC(data, size) data = av_malloc(size)
  148. #define ALLOC_BUF(data, size) \
  149. do { \
  150. av_buffer_realloc(&pkt->buf, size); \
  151. data = pkt->buf ? pkt->buf->data : NULL; \
  152. } while (0)
  153. #define DUP_DATA(dst, src, size, padding, ALLOC) \
  154. do { \
  155. void *data; \
  156. if (padding) { \
  157. if ((unsigned)(size) > \
  158. (unsigned)(size) + AV_INPUT_BUFFER_PADDING_SIZE) \
  159. goto failed_alloc; \
  160. ALLOC(data, size + AV_INPUT_BUFFER_PADDING_SIZE); \
  161. } else { \
  162. ALLOC(data, size); \
  163. } \
  164. if (!data) \
  165. goto failed_alloc; \
  166. memcpy(data, src, size); \
  167. if (padding) \
  168. memset((uint8_t *)data + size, 0, \
  169. AV_INPUT_BUFFER_PADDING_SIZE); \
  170. dst = data; \
  171. } while (0)
  172. /* Makes duplicates of data, side_data, but does not copy any other fields */
  173. static int copy_packet_data(AVPacket *pkt, const AVPacket *src, int dup)
  174. {
  175. pkt->data = NULL;
  176. pkt->side_data = NULL;
  177. pkt->side_data_elems = 0;
  178. if (pkt->buf) {
  179. AVBufferRef *ref = av_buffer_ref(src->buf);
  180. if (!ref)
  181. return AVERROR(ENOMEM);
  182. pkt->buf = ref;
  183. pkt->data = ref->data;
  184. } else {
  185. DUP_DATA(pkt->data, src->data, pkt->size, 1, ALLOC_BUF);
  186. }
  187. if (src->side_data_elems && dup) {
  188. pkt->side_data = src->side_data;
  189. pkt->side_data_elems = src->side_data_elems;
  190. }
  191. if (src->side_data_elems && !dup) {
  192. return av_copy_packet_side_data(pkt, src);
  193. }
  194. return 0;
  195. failed_alloc:
  196. av_packet_unref(pkt);
  197. return AVERROR(ENOMEM);
  198. }
  199. int av_copy_packet_side_data(AVPacket *pkt, const AVPacket *src)
  200. {
  201. if (src->side_data_elems) {
  202. int i;
  203. DUP_DATA(pkt->side_data, src->side_data,
  204. src->side_data_elems * sizeof(*src->side_data), 0, ALLOC_MALLOC);
  205. if (src != pkt) {
  206. memset(pkt->side_data, 0,
  207. src->side_data_elems * sizeof(*src->side_data));
  208. }
  209. for (i = 0; i < src->side_data_elems; i++) {
  210. DUP_DATA(pkt->side_data[i].data, src->side_data[i].data,
  211. src->side_data[i].size, 1, ALLOC_MALLOC);
  212. pkt->side_data[i].size = src->side_data[i].size;
  213. pkt->side_data[i].type = src->side_data[i].type;
  214. }
  215. }
  216. pkt->side_data_elems = src->side_data_elems;
  217. return 0;
  218. failed_alloc:
  219. av_packet_unref(pkt);
  220. return AVERROR(ENOMEM);
  221. }
  222. int av_dup_packet(AVPacket *pkt)
  223. {
  224. AVPacket tmp_pkt;
  225. if (!pkt->buf && pkt->data) {
  226. tmp_pkt = *pkt;
  227. return copy_packet_data(pkt, &tmp_pkt, 1);
  228. }
  229. return 0;
  230. }
  231. int av_copy_packet(AVPacket *dst, const AVPacket *src)
  232. {
  233. *dst = *src;
  234. return copy_packet_data(dst, src, 0);
  235. }
  236. FF_ENABLE_DEPRECATION_WARNINGS
  237. #endif
  238. void av_packet_free_side_data(AVPacket *pkt)
  239. {
  240. int i;
  241. for (i = 0; i < pkt->side_data_elems; i++)
  242. av_freep(&pkt->side_data[i].data);
  243. av_freep(&pkt->side_data);
  244. pkt->side_data_elems = 0;
  245. }
  246. #if FF_API_AVPACKET_OLD_API
  247. FF_DISABLE_DEPRECATION_WARNINGS
  248. void av_free_packet(AVPacket *pkt)
  249. {
  250. if (pkt) {
  251. if (pkt->buf)
  252. av_buffer_unref(&pkt->buf);
  253. pkt->data = NULL;
  254. pkt->size = 0;
  255. av_packet_free_side_data(pkt);
  256. }
  257. }
  258. FF_ENABLE_DEPRECATION_WARNINGS
  259. #endif
  260. int av_packet_add_side_data(AVPacket *pkt, enum AVPacketSideDataType type,
  261. uint8_t *data, size_t size)
  262. {
  263. AVPacketSideData *tmp;
  264. int i, elems = pkt->side_data_elems;
  265. for (i = 0; i < elems; i++) {
  266. AVPacketSideData *sd = &pkt->side_data[i];
  267. if (sd->type == type) {
  268. av_free(sd->data);
  269. sd->data = data;
  270. sd->size = size;
  271. return 0;
  272. }
  273. }
  274. if ((unsigned)elems + 1 > AV_PKT_DATA_NB)
  275. return AVERROR(ERANGE);
  276. tmp = av_realloc(pkt->side_data, (elems + 1) * sizeof(*tmp));
  277. if (!tmp)
  278. return AVERROR(ENOMEM);
  279. pkt->side_data = tmp;
  280. pkt->side_data[elems].data = data;
  281. pkt->side_data[elems].size = size;
  282. pkt->side_data[elems].type = type;
  283. pkt->side_data_elems++;
  284. return 0;
  285. }
  286. uint8_t *av_packet_new_side_data(AVPacket *pkt, enum AVPacketSideDataType type,
  287. buffer_size_t size)
  288. {
  289. int ret;
  290. uint8_t *data;
  291. #if FF_API_BUFFER_SIZE_T
  292. if ((unsigned)size > INT_MAX - AV_INPUT_BUFFER_PADDING_SIZE)
  293. #else
  294. if (size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE)
  295. #endif
  296. return NULL;
  297. data = av_mallocz(size + AV_INPUT_BUFFER_PADDING_SIZE);
  298. if (!data)
  299. return NULL;
  300. ret = av_packet_add_side_data(pkt, type, data, size);
  301. if (ret < 0) {
  302. av_freep(&data);
  303. return NULL;
  304. }
  305. return data;
  306. }
  307. uint8_t *av_packet_get_side_data(const AVPacket *pkt, enum AVPacketSideDataType type,
  308. buffer_size_t *size)
  309. {
  310. int i;
  311. for (i = 0; i < pkt->side_data_elems; i++) {
  312. if (pkt->side_data[i].type == type) {
  313. if (size)
  314. *size = pkt->side_data[i].size;
  315. return pkt->side_data[i].data;
  316. }
  317. }
  318. if (size)
  319. *size = 0;
  320. return NULL;
  321. }
  322. const char *av_packet_side_data_name(enum AVPacketSideDataType type)
  323. {
  324. switch(type) {
  325. case AV_PKT_DATA_PALETTE: return "Palette";
  326. case AV_PKT_DATA_NEW_EXTRADATA: return "New Extradata";
  327. case AV_PKT_DATA_PARAM_CHANGE: return "Param Change";
  328. case AV_PKT_DATA_H263_MB_INFO: return "H263 MB Info";
  329. case AV_PKT_DATA_REPLAYGAIN: return "Replay Gain";
  330. case AV_PKT_DATA_DISPLAYMATRIX: return "Display Matrix";
  331. case AV_PKT_DATA_STEREO3D: return "Stereo 3D";
  332. case AV_PKT_DATA_AUDIO_SERVICE_TYPE: return "Audio Service Type";
  333. case AV_PKT_DATA_QUALITY_STATS: return "Quality stats";
  334. case AV_PKT_DATA_FALLBACK_TRACK: return "Fallback track";
  335. case AV_PKT_DATA_CPB_PROPERTIES: return "CPB properties";
  336. case AV_PKT_DATA_SKIP_SAMPLES: return "Skip Samples";
  337. case AV_PKT_DATA_JP_DUALMONO: return "JP Dual Mono";
  338. case AV_PKT_DATA_STRINGS_METADATA: return "Strings Metadata";
  339. case AV_PKT_DATA_SUBTITLE_POSITION: return "Subtitle Position";
  340. case AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL: return "Matroska BlockAdditional";
  341. case AV_PKT_DATA_WEBVTT_IDENTIFIER: return "WebVTT ID";
  342. case AV_PKT_DATA_WEBVTT_SETTINGS: return "WebVTT Settings";
  343. case AV_PKT_DATA_METADATA_UPDATE: return "Metadata Update";
  344. case AV_PKT_DATA_MPEGTS_STREAM_ID: return "MPEGTS Stream ID";
  345. case AV_PKT_DATA_MASTERING_DISPLAY_METADATA: return "Mastering display metadata";
  346. case AV_PKT_DATA_CONTENT_LIGHT_LEVEL: return "Content light level metadata";
  347. case AV_PKT_DATA_SPHERICAL: return "Spherical Mapping";
  348. case AV_PKT_DATA_A53_CC: return "A53 Closed Captions";
  349. case AV_PKT_DATA_ENCRYPTION_INIT_INFO: return "Encryption initialization data";
  350. case AV_PKT_DATA_ENCRYPTION_INFO: return "Encryption info";
  351. case AV_PKT_DATA_AFD: return "Active Format Description data";
  352. case AV_PKT_DATA_PRFT: return "Producer Reference Time";
  353. case AV_PKT_DATA_ICC_PROFILE: return "ICC Profile";
  354. case AV_PKT_DATA_DOVI_CONF: return "DOVI configuration record";
  355. case AV_PKT_DATA_S12M_TIMECODE: return "SMPTE ST 12-1:2014 timecode";
  356. }
  357. return NULL;
  358. }
  359. #if FF_API_MERGE_SD_API
  360. #define FF_MERGE_MARKER 0x8c4d9d108e25e9feULL
  361. int av_packet_merge_side_data(AVPacket *pkt){
  362. if(pkt->side_data_elems){
  363. AVBufferRef *buf;
  364. int i;
  365. uint8_t *p;
  366. uint64_t size= pkt->size + 8LL + AV_INPUT_BUFFER_PADDING_SIZE;
  367. AVPacket old= *pkt;
  368. for (i=0; i<old.side_data_elems; i++) {
  369. size += old.side_data[i].size + 5LL;
  370. }
  371. if (size > INT_MAX)
  372. return AVERROR(EINVAL);
  373. buf = av_buffer_alloc(size);
  374. if (!buf)
  375. return AVERROR(ENOMEM);
  376. pkt->buf = buf;
  377. pkt->data = p = buf->data;
  378. pkt->size = size - AV_INPUT_BUFFER_PADDING_SIZE;
  379. bytestream_put_buffer(&p, old.data, old.size);
  380. for (i=old.side_data_elems-1; i>=0; i--) {
  381. bytestream_put_buffer(&p, old.side_data[i].data, old.side_data[i].size);
  382. bytestream_put_be32(&p, old.side_data[i].size);
  383. *p++ = old.side_data[i].type | ((i==old.side_data_elems-1)*128);
  384. }
  385. bytestream_put_be64(&p, FF_MERGE_MARKER);
  386. av_assert0(p-pkt->data == pkt->size);
  387. memset(p, 0, AV_INPUT_BUFFER_PADDING_SIZE);
  388. av_packet_unref(&old);
  389. pkt->side_data_elems = 0;
  390. pkt->side_data = NULL;
  391. return 1;
  392. }
  393. return 0;
  394. }
  395. int av_packet_split_side_data(AVPacket *pkt){
  396. if (!pkt->side_data_elems && pkt->size >12 && AV_RB64(pkt->data + pkt->size - 8) == FF_MERGE_MARKER){
  397. int i;
  398. unsigned int size;
  399. uint8_t *p;
  400. p = pkt->data + pkt->size - 8 - 5;
  401. for (i=1; ; i++){
  402. size = AV_RB32(p);
  403. if (size>INT_MAX - 5 || p - pkt->data < size)
  404. return 0;
  405. if (p[4]&128)
  406. break;
  407. if (p - pkt->data < size + 5)
  408. return 0;
  409. p-= size+5;
  410. }
  411. if (i > AV_PKT_DATA_NB)
  412. return AVERROR(ERANGE);
  413. pkt->side_data = av_malloc_array(i, sizeof(*pkt->side_data));
  414. if (!pkt->side_data)
  415. return AVERROR(ENOMEM);
  416. p= pkt->data + pkt->size - 8 - 5;
  417. for (i=0; ; i++){
  418. size= AV_RB32(p);
  419. av_assert0(size<=INT_MAX - 5 && p - pkt->data >= size);
  420. pkt->side_data[i].data = av_mallocz(size + AV_INPUT_BUFFER_PADDING_SIZE);
  421. pkt->side_data[i].size = size;
  422. pkt->side_data[i].type = p[4]&127;
  423. if (!pkt->side_data[i].data)
  424. return AVERROR(ENOMEM);
  425. memcpy(pkt->side_data[i].data, p-size, size);
  426. pkt->size -= size + 5;
  427. if(p[4]&128)
  428. break;
  429. p-= size+5;
  430. }
  431. pkt->size -= 8;
  432. pkt->side_data_elems = i+1;
  433. return 1;
  434. }
  435. return 0;
  436. }
  437. #endif
  438. uint8_t *av_packet_pack_dictionary(AVDictionary *dict, int *size)
  439. {
  440. AVDictionaryEntry *t = NULL;
  441. uint8_t *data = NULL;
  442. *size = 0;
  443. if (!dict)
  444. return NULL;
  445. while ((t = av_dict_get(dict, "", t, AV_DICT_IGNORE_SUFFIX))) {
  446. const size_t keylen = strlen(t->key);
  447. const size_t valuelen = strlen(t->value);
  448. const size_t new_size = *size + keylen + 1 + valuelen + 1;
  449. uint8_t *const new_data = av_realloc(data, new_size);
  450. if (!new_data)
  451. goto fail;
  452. data = new_data;
  453. if (new_size > INT_MAX)
  454. goto fail;
  455. memcpy(data + *size, t->key, keylen + 1);
  456. memcpy(data + *size + keylen + 1, t->value, valuelen + 1);
  457. *size = new_size;
  458. }
  459. return data;
  460. fail:
  461. av_freep(&data);
  462. *size = 0;
  463. return NULL;
  464. }
  465. int av_packet_unpack_dictionary(const uint8_t *data, int size, AVDictionary **dict)
  466. {
  467. const uint8_t *end;
  468. int ret;
  469. if (!dict || !data || !size)
  470. return 0;
  471. end = data + size;
  472. if (size && end[-1])
  473. return AVERROR_INVALIDDATA;
  474. while (data < end) {
  475. const uint8_t *key = data;
  476. const uint8_t *val = data + strlen(key) + 1;
  477. if (val >= end || !*key)
  478. return AVERROR_INVALIDDATA;
  479. ret = av_dict_set(dict, key, val, 0);
  480. if (ret < 0)
  481. return ret;
  482. data = val + strlen(val) + 1;
  483. }
  484. return 0;
  485. }
  486. int av_packet_shrink_side_data(AVPacket *pkt, enum AVPacketSideDataType type,
  487. buffer_size_t size)
  488. {
  489. int i;
  490. for (i = 0; i < pkt->side_data_elems; i++) {
  491. if (pkt->side_data[i].type == type) {
  492. if (size > pkt->side_data[i].size)
  493. return AVERROR(ENOMEM);
  494. pkt->side_data[i].size = size;
  495. return 0;
  496. }
  497. }
  498. return AVERROR(ENOENT);
  499. }
  500. int av_packet_copy_props(AVPacket *dst, const AVPacket *src)
  501. {
  502. int i;
  503. dst->pts = src->pts;
  504. dst->dts = src->dts;
  505. dst->pos = src->pos;
  506. dst->duration = src->duration;
  507. #if FF_API_CONVERGENCE_DURATION
  508. FF_DISABLE_DEPRECATION_WARNINGS
  509. dst->convergence_duration = src->convergence_duration;
  510. FF_ENABLE_DEPRECATION_WARNINGS
  511. #endif
  512. dst->flags = src->flags;
  513. dst->stream_index = src->stream_index;
  514. dst->side_data = NULL;
  515. dst->side_data_elems = 0;
  516. for (i = 0; i < src->side_data_elems; i++) {
  517. enum AVPacketSideDataType type = src->side_data[i].type;
  518. buffer_size_t size = src->side_data[i].size;
  519. uint8_t *src_data = src->side_data[i].data;
  520. uint8_t *dst_data = av_packet_new_side_data(dst, type, size);
  521. if (!dst_data) {
  522. av_packet_free_side_data(dst);
  523. return AVERROR(ENOMEM);
  524. }
  525. memcpy(dst_data, src_data, size);
  526. }
  527. return 0;
  528. }
  529. void av_packet_unref(AVPacket *pkt)
  530. {
  531. av_packet_free_side_data(pkt);
  532. av_buffer_unref(&pkt->buf);
  533. av_init_packet(pkt);
  534. pkt->data = NULL;
  535. pkt->size = 0;
  536. }
  537. int av_packet_ref(AVPacket *dst, const AVPacket *src)
  538. {
  539. int ret;
  540. dst->buf = NULL;
  541. ret = av_packet_copy_props(dst, src);
  542. if (ret < 0)
  543. goto fail;
  544. if (!src->buf) {
  545. ret = packet_alloc(&dst->buf, src->size);
  546. if (ret < 0)
  547. goto fail;
  548. av_assert1(!src->size || src->data);
  549. if (src->size)
  550. memcpy(dst->buf->data, src->data, src->size);
  551. dst->data = dst->buf->data;
  552. } else {
  553. dst->buf = av_buffer_ref(src->buf);
  554. if (!dst->buf) {
  555. ret = AVERROR(ENOMEM);
  556. goto fail;
  557. }
  558. dst->data = src->data;
  559. }
  560. dst->size = src->size;
  561. return 0;
  562. fail:
  563. av_packet_unref(dst);
  564. return ret;
  565. }
  566. AVPacket *av_packet_clone(const AVPacket *src)
  567. {
  568. AVPacket *ret = av_packet_alloc();
  569. if (!ret)
  570. return ret;
  571. if (av_packet_ref(ret, src))
  572. av_packet_free(&ret);
  573. return ret;
  574. }
  575. void av_packet_move_ref(AVPacket *dst, AVPacket *src)
  576. {
  577. *dst = *src;
  578. av_init_packet(src);
  579. src->data = NULL;
  580. src->size = 0;
  581. }
  582. int av_packet_make_refcounted(AVPacket *pkt)
  583. {
  584. int ret;
  585. if (pkt->buf)
  586. return 0;
  587. ret = packet_alloc(&pkt->buf, pkt->size);
  588. if (ret < 0)
  589. return ret;
  590. av_assert1(!pkt->size || pkt->data);
  591. if (pkt->size)
  592. memcpy(pkt->buf->data, pkt->data, pkt->size);
  593. pkt->data = pkt->buf->data;
  594. return 0;
  595. }
  596. int av_packet_make_writable(AVPacket *pkt)
  597. {
  598. AVBufferRef *buf = NULL;
  599. int ret;
  600. if (pkt->buf && av_buffer_is_writable(pkt->buf))
  601. return 0;
  602. ret = packet_alloc(&buf, pkt->size);
  603. if (ret < 0)
  604. return ret;
  605. av_assert1(!pkt->size || pkt->data);
  606. if (pkt->size)
  607. memcpy(buf->data, pkt->data, pkt->size);
  608. av_buffer_unref(&pkt->buf);
  609. pkt->buf = buf;
  610. pkt->data = buf->data;
  611. return 0;
  612. }
  613. void av_packet_rescale_ts(AVPacket *pkt, AVRational src_tb, AVRational dst_tb)
  614. {
  615. if (pkt->pts != AV_NOPTS_VALUE)
  616. pkt->pts = av_rescale_q(pkt->pts, src_tb, dst_tb);
  617. if (pkt->dts != AV_NOPTS_VALUE)
  618. pkt->dts = av_rescale_q(pkt->dts, src_tb, dst_tb);
  619. if (pkt->duration > 0)
  620. pkt->duration = av_rescale_q(pkt->duration, src_tb, dst_tb);
  621. #if FF_API_CONVERGENCE_DURATION
  622. FF_DISABLE_DEPRECATION_WARNINGS
  623. if (pkt->convergence_duration > 0)
  624. pkt->convergence_duration = av_rescale_q(pkt->convergence_duration, src_tb, dst_tb);
  625. FF_ENABLE_DEPRECATION_WARNINGS
  626. #endif
  627. }
  628. int avpriv_packet_list_put(AVPacketList **packet_buffer,
  629. AVPacketList **plast_pktl,
  630. AVPacket *pkt,
  631. int (*copy)(AVPacket *dst, const AVPacket *src),
  632. int flags)
  633. {
  634. AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
  635. int ret;
  636. if (!pktl)
  637. return AVERROR(ENOMEM);
  638. if (copy) {
  639. ret = copy(&pktl->pkt, pkt);
  640. if (ret < 0) {
  641. av_free(pktl);
  642. return ret;
  643. }
  644. } else {
  645. ret = av_packet_make_refcounted(pkt);
  646. if (ret < 0) {
  647. av_free(pktl);
  648. return ret;
  649. }
  650. av_packet_move_ref(&pktl->pkt, pkt);
  651. }
  652. if (*packet_buffer)
  653. (*plast_pktl)->next = pktl;
  654. else
  655. *packet_buffer = pktl;
  656. /* Add the packet in the buffered packet list. */
  657. *plast_pktl = pktl;
  658. return 0;
  659. }
  660. int avpriv_packet_list_get(AVPacketList **pkt_buffer,
  661. AVPacketList **pkt_buffer_end,
  662. AVPacket *pkt)
  663. {
  664. AVPacketList *pktl;
  665. if (!*pkt_buffer)
  666. return AVERROR(EAGAIN);
  667. pktl = *pkt_buffer;
  668. *pkt = pktl->pkt;
  669. *pkt_buffer = pktl->next;
  670. if (!pktl->next)
  671. *pkt_buffer_end = NULL;
  672. av_freep(&pktl);
  673. return 0;
  674. }
  675. void avpriv_packet_list_free(AVPacketList **pkt_buf, AVPacketList **pkt_buf_end)
  676. {
  677. AVPacketList *tmp = *pkt_buf;
  678. while (tmp) {
  679. AVPacketList *pktl = tmp;
  680. tmp = pktl->next;
  681. av_packet_unref(&pktl->pkt);
  682. av_freep(&pktl);
  683. }
  684. *pkt_buf = NULL;
  685. *pkt_buf_end = NULL;
  686. }
  687. int ff_side_data_set_encoder_stats(AVPacket *pkt, int quality, int64_t *error, int error_count, int pict_type)
  688. {
  689. uint8_t *side_data;
  690. buffer_size_t side_data_size;
  691. int i;
  692. side_data = av_packet_get_side_data(pkt, AV_PKT_DATA_QUALITY_STATS, &side_data_size);
  693. if (!side_data) {
  694. side_data_size = 4+4+8*error_count;
  695. side_data = av_packet_new_side_data(pkt, AV_PKT_DATA_QUALITY_STATS,
  696. side_data_size);
  697. }
  698. if (!side_data || side_data_size < 4+4+8*error_count)
  699. return AVERROR(ENOMEM);
  700. AV_WL32(side_data , quality );
  701. side_data[4] = pict_type;
  702. side_data[5] = error_count;
  703. for (i = 0; i<error_count; i++)
  704. AV_WL64(side_data+8 + 8*i , error[i]);
  705. return 0;
  706. }
  707. int ff_side_data_set_prft(AVPacket *pkt, int64_t timestamp)
  708. {
  709. AVProducerReferenceTime *prft;
  710. uint8_t *side_data;
  711. buffer_size_t side_data_size;
  712. side_data = av_packet_get_side_data(pkt, AV_PKT_DATA_PRFT, &side_data_size);
  713. if (!side_data) {
  714. side_data_size = sizeof(AVProducerReferenceTime);
  715. side_data = av_packet_new_side_data(pkt, AV_PKT_DATA_PRFT, side_data_size);
  716. }
  717. if (!side_data || side_data_size < sizeof(AVProducerReferenceTime))
  718. return AVERROR(ENOMEM);
  719. prft = (AVProducerReferenceTime *)side_data;
  720. prft->wallclock = timestamp;
  721. prft->flags = 0;
  722. return 0;
  723. }