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.

704 lines
22KB

  1. /*
  2. * RTMP input format
  3. * Copyright (c) 2009 Konstantin Shishkov
  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 "libavcodec/bytestream.h"
  22. #include "libavutil/avstring.h"
  23. #include "libavutil/intfloat.h"
  24. #include "avformat.h"
  25. #include "rtmppkt.h"
  26. #include "flv.h"
  27. #include "url.h"
  28. void ff_amf_write_bool(uint8_t **dst, int val)
  29. {
  30. bytestream_put_byte(dst, AMF_DATA_TYPE_BOOL);
  31. bytestream_put_byte(dst, val);
  32. }
  33. void ff_amf_write_number(uint8_t **dst, double val)
  34. {
  35. bytestream_put_byte(dst, AMF_DATA_TYPE_NUMBER);
  36. bytestream_put_be64(dst, av_double2int(val));
  37. }
  38. void ff_amf_write_string(uint8_t **dst, const char *str)
  39. {
  40. bytestream_put_byte(dst, AMF_DATA_TYPE_STRING);
  41. bytestream_put_be16(dst, strlen(str));
  42. bytestream_put_buffer(dst, str, strlen(str));
  43. }
  44. void ff_amf_write_string2(uint8_t **dst, const char *str1, const char *str2)
  45. {
  46. int len1 = 0, len2 = 0;
  47. if (str1)
  48. len1 = strlen(str1);
  49. if (str2)
  50. len2 = strlen(str2);
  51. bytestream_put_byte(dst, AMF_DATA_TYPE_STRING);
  52. bytestream_put_be16(dst, len1 + len2);
  53. bytestream_put_buffer(dst, str1, len1);
  54. bytestream_put_buffer(dst, str2, len2);
  55. }
  56. void ff_amf_write_null(uint8_t **dst)
  57. {
  58. bytestream_put_byte(dst, AMF_DATA_TYPE_NULL);
  59. }
  60. void ff_amf_write_object_start(uint8_t **dst)
  61. {
  62. bytestream_put_byte(dst, AMF_DATA_TYPE_OBJECT);
  63. }
  64. void ff_amf_write_field_name(uint8_t **dst, const char *str)
  65. {
  66. bytestream_put_be16(dst, strlen(str));
  67. bytestream_put_buffer(dst, str, strlen(str));
  68. }
  69. void ff_amf_write_object_end(uint8_t **dst)
  70. {
  71. /* first two bytes are field name length = 0,
  72. * AMF object should end with it and end marker
  73. */
  74. bytestream_put_be24(dst, AMF_DATA_TYPE_OBJECT_END);
  75. }
  76. int ff_amf_read_bool(GetByteContext *bc, int *val)
  77. {
  78. if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_BOOL)
  79. return AVERROR_INVALIDDATA;
  80. *val = bytestream2_get_byte(bc);
  81. return 0;
  82. }
  83. int ff_amf_read_number(GetByteContext *bc, double *val)
  84. {
  85. uint64_t read;
  86. if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_NUMBER)
  87. return AVERROR_INVALIDDATA;
  88. read = bytestream2_get_be64(bc);
  89. *val = av_int2double(read);
  90. return 0;
  91. }
  92. int ff_amf_get_string(GetByteContext *bc, uint8_t *str,
  93. int strsize, int *length)
  94. {
  95. int stringlen = 0;
  96. int readsize;
  97. stringlen = bytestream2_get_be16(bc);
  98. if (stringlen + 1 > strsize)
  99. return AVERROR(EINVAL);
  100. readsize = bytestream2_get_buffer(bc, str, stringlen);
  101. if (readsize != stringlen) {
  102. av_log(NULL, AV_LOG_WARNING,
  103. "Unable to read as many bytes as AMF string signaled\n");
  104. }
  105. str[readsize] = '\0';
  106. *length = FFMIN(stringlen, readsize);
  107. return 0;
  108. }
  109. int ff_amf_read_string(GetByteContext *bc, uint8_t *str,
  110. int strsize, int *length)
  111. {
  112. if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_STRING)
  113. return AVERROR_INVALIDDATA;
  114. return ff_amf_get_string(bc, str, strsize, length);
  115. }
  116. int ff_amf_read_null(GetByteContext *bc)
  117. {
  118. if (bytestream2_get_byte(bc) != AMF_DATA_TYPE_NULL)
  119. return AVERROR_INVALIDDATA;
  120. return 0;
  121. }
  122. int ff_rtmp_check_alloc_array(RTMPPacket **prev_pkt, int *nb_prev_pkt,
  123. int channel)
  124. {
  125. int nb_alloc;
  126. RTMPPacket *ptr;
  127. if (channel < *nb_prev_pkt)
  128. return 0;
  129. nb_alloc = channel + 16;
  130. // This can't use the av_reallocp family of functions, since we
  131. // would need to free each element in the array before the array
  132. // itself is freed.
  133. ptr = av_realloc_array(*prev_pkt, nb_alloc, sizeof(**prev_pkt));
  134. if (!ptr)
  135. return AVERROR(ENOMEM);
  136. memset(ptr + *nb_prev_pkt, 0, (nb_alloc - *nb_prev_pkt) * sizeof(*ptr));
  137. *prev_pkt = ptr;
  138. *nb_prev_pkt = nb_alloc;
  139. return 0;
  140. }
  141. int ff_rtmp_packet_read(URLContext *h, RTMPPacket *p,
  142. int chunk_size, RTMPPacket **prev_pkt, int *nb_prev_pkt)
  143. {
  144. uint8_t hdr;
  145. if (ffurl_read(h, &hdr, 1) != 1)
  146. return AVERROR(EIO);
  147. return ff_rtmp_packet_read_internal(h, p, chunk_size, prev_pkt,
  148. nb_prev_pkt, hdr);
  149. }
  150. static int rtmp_packet_read_one_chunk(URLContext *h, RTMPPacket *p,
  151. int chunk_size, RTMPPacket **prev_pkt_ptr,
  152. int *nb_prev_pkt, uint8_t hdr)
  153. {
  154. uint8_t buf[16];
  155. int channel_id, timestamp, size;
  156. uint32_t ts_field; // non-extended timestamp or delta field
  157. uint32_t extra = 0;
  158. enum RTMPPacketType type;
  159. int written = 0;
  160. int ret, toread;
  161. RTMPPacket *prev_pkt;
  162. written++;
  163. channel_id = hdr & 0x3F;
  164. if (channel_id < 2) { //special case for channel number >= 64
  165. buf[1] = 0;
  166. if (ffurl_read_complete(h, buf, channel_id + 1) != channel_id + 1)
  167. return AVERROR(EIO);
  168. written += channel_id + 1;
  169. channel_id = AV_RL16(buf) + 64;
  170. }
  171. if ((ret = ff_rtmp_check_alloc_array(prev_pkt_ptr, nb_prev_pkt,
  172. channel_id)) < 0)
  173. return ret;
  174. prev_pkt = *prev_pkt_ptr;
  175. size = prev_pkt[channel_id].size;
  176. type = prev_pkt[channel_id].type;
  177. extra = prev_pkt[channel_id].extra;
  178. hdr >>= 6; // header size indicator
  179. if (hdr == RTMP_PS_ONEBYTE) {
  180. ts_field = prev_pkt[channel_id].ts_field;
  181. } else {
  182. if (ffurl_read_complete(h, buf, 3) != 3)
  183. return AVERROR(EIO);
  184. written += 3;
  185. ts_field = AV_RB24(buf);
  186. if (hdr != RTMP_PS_FOURBYTES) {
  187. if (ffurl_read_complete(h, buf, 3) != 3)
  188. return AVERROR(EIO);
  189. written += 3;
  190. size = AV_RB24(buf);
  191. if (ffurl_read_complete(h, buf, 1) != 1)
  192. return AVERROR(EIO);
  193. written++;
  194. type = buf[0];
  195. if (hdr == RTMP_PS_TWELVEBYTES) {
  196. if (ffurl_read_complete(h, buf, 4) != 4)
  197. return AVERROR(EIO);
  198. written += 4;
  199. extra = AV_RL32(buf);
  200. }
  201. }
  202. }
  203. if (ts_field == 0xFFFFFF) {
  204. if (ffurl_read_complete(h, buf, 4) != 4)
  205. return AVERROR(EIO);
  206. timestamp = AV_RB32(buf);
  207. } else {
  208. timestamp = ts_field;
  209. }
  210. if (hdr != RTMP_PS_TWELVEBYTES)
  211. timestamp += prev_pkt[channel_id].timestamp;
  212. if (prev_pkt[channel_id].read && size != prev_pkt[channel_id].size) {
  213. av_log(NULL, AV_LOG_ERROR, "RTMP packet size mismatch %d != %d\n",
  214. size,
  215. prev_pkt[channel_id].size);
  216. ff_rtmp_packet_destroy(&prev_pkt[channel_id]);
  217. prev_pkt[channel_id].read = 0;
  218. }
  219. if (!prev_pkt[channel_id].read) {
  220. if ((ret = ff_rtmp_packet_create(p, channel_id, type, timestamp,
  221. size)) < 0)
  222. return ret;
  223. p->read = written;
  224. p->offset = 0;
  225. prev_pkt[channel_id].ts_field = ts_field;
  226. prev_pkt[channel_id].timestamp = timestamp;
  227. } else {
  228. // previous packet in this channel hasn't completed reading
  229. RTMPPacket *prev = &prev_pkt[channel_id];
  230. p->data = prev->data;
  231. p->size = prev->size;
  232. p->channel_id = prev->channel_id;
  233. p->type = prev->type;
  234. p->ts_field = prev->ts_field;
  235. p->extra = prev->extra;
  236. p->offset = prev->offset;
  237. p->read = prev->read + written;
  238. p->timestamp = prev->timestamp;
  239. prev->data = NULL;
  240. }
  241. p->extra = extra;
  242. // save history
  243. prev_pkt[channel_id].channel_id = channel_id;
  244. prev_pkt[channel_id].type = type;
  245. prev_pkt[channel_id].size = size;
  246. prev_pkt[channel_id].extra = extra;
  247. size = size - p->offset;
  248. toread = FFMIN(size, chunk_size);
  249. if (ffurl_read_complete(h, p->data + p->offset, toread) != toread) {
  250. ff_rtmp_packet_destroy(p);
  251. return AVERROR(EIO);
  252. }
  253. size -= toread;
  254. p->read += toread;
  255. p->offset += toread;
  256. if (size > 0) {
  257. RTMPPacket *prev = &prev_pkt[channel_id];
  258. prev->data = p->data;
  259. prev->read = p->read;
  260. prev->offset = p->offset;
  261. p->data = NULL;
  262. return AVERROR(EAGAIN);
  263. }
  264. prev_pkt[channel_id].read = 0; // read complete; reset if needed
  265. return p->read;
  266. }
  267. int ff_rtmp_packet_read_internal(URLContext *h, RTMPPacket *p, int chunk_size,
  268. RTMPPacket **prev_pkt, int *nb_prev_pkt,
  269. uint8_t hdr)
  270. {
  271. while (1) {
  272. int ret = rtmp_packet_read_one_chunk(h, p, chunk_size, prev_pkt,
  273. nb_prev_pkt, hdr);
  274. if (ret > 0 || ret != AVERROR(EAGAIN))
  275. return ret;
  276. if (ffurl_read(h, &hdr, 1) != 1)
  277. return AVERROR(EIO);
  278. }
  279. }
  280. int ff_rtmp_packet_write(URLContext *h, RTMPPacket *pkt,
  281. int chunk_size, RTMPPacket **prev_pkt_ptr,
  282. int *nb_prev_pkt)
  283. {
  284. uint8_t pkt_hdr[16], *p = pkt_hdr;
  285. int mode = RTMP_PS_TWELVEBYTES;
  286. int off = 0;
  287. int written = 0;
  288. int ret;
  289. RTMPPacket *prev_pkt;
  290. int use_delta; // flag if using timestamp delta, not RTMP_PS_TWELVEBYTES
  291. uint32_t timestamp; // full 32-bit timestamp or delta value
  292. if ((ret = ff_rtmp_check_alloc_array(prev_pkt_ptr, nb_prev_pkt,
  293. pkt->channel_id)) < 0)
  294. return ret;
  295. prev_pkt = *prev_pkt_ptr;
  296. //if channel_id = 0, this is first presentation of prev_pkt, send full hdr.
  297. use_delta = prev_pkt[pkt->channel_id].channel_id &&
  298. pkt->extra == prev_pkt[pkt->channel_id].extra &&
  299. pkt->timestamp >= prev_pkt[pkt->channel_id].timestamp;
  300. timestamp = pkt->timestamp;
  301. if (use_delta) {
  302. timestamp -= prev_pkt[pkt->channel_id].timestamp;
  303. }
  304. if (timestamp >= 0xFFFFFF) {
  305. pkt->ts_field = 0xFFFFFF;
  306. } else {
  307. pkt->ts_field = timestamp;
  308. }
  309. if (use_delta) {
  310. if (pkt->type == prev_pkt[pkt->channel_id].type &&
  311. pkt->size == prev_pkt[pkt->channel_id].size) {
  312. mode = RTMP_PS_FOURBYTES;
  313. if (pkt->ts_field == prev_pkt[pkt->channel_id].ts_field)
  314. mode = RTMP_PS_ONEBYTE;
  315. } else {
  316. mode = RTMP_PS_EIGHTBYTES;
  317. }
  318. }
  319. if (pkt->channel_id < 64) {
  320. bytestream_put_byte(&p, pkt->channel_id | (mode << 6));
  321. } else if (pkt->channel_id < 64 + 256) {
  322. bytestream_put_byte(&p, 0 | (mode << 6));
  323. bytestream_put_byte(&p, pkt->channel_id - 64);
  324. } else {
  325. bytestream_put_byte(&p, 1 | (mode << 6));
  326. bytestream_put_le16(&p, pkt->channel_id - 64);
  327. }
  328. if (mode != RTMP_PS_ONEBYTE) {
  329. bytestream_put_be24(&p, pkt->ts_field);
  330. if (mode != RTMP_PS_FOURBYTES) {
  331. bytestream_put_be24(&p, pkt->size);
  332. bytestream_put_byte(&p, pkt->type);
  333. if (mode == RTMP_PS_TWELVEBYTES)
  334. bytestream_put_le32(&p, pkt->extra);
  335. }
  336. }
  337. if (pkt->ts_field == 0xFFFFFF)
  338. bytestream_put_be32(&p, timestamp);
  339. // save history
  340. prev_pkt[pkt->channel_id].channel_id = pkt->channel_id;
  341. prev_pkt[pkt->channel_id].type = pkt->type;
  342. prev_pkt[pkt->channel_id].size = pkt->size;
  343. prev_pkt[pkt->channel_id].timestamp = pkt->timestamp;
  344. prev_pkt[pkt->channel_id].ts_field = pkt->ts_field;
  345. prev_pkt[pkt->channel_id].extra = pkt->extra;
  346. if ((ret = ffurl_write(h, pkt_hdr, p - pkt_hdr)) < 0)
  347. return ret;
  348. written = p - pkt_hdr + pkt->size;
  349. while (off < pkt->size) {
  350. int towrite = FFMIN(chunk_size, pkt->size - off);
  351. if ((ret = ffurl_write(h, pkt->data + off, towrite)) < 0)
  352. return ret;
  353. off += towrite;
  354. if (off < pkt->size) {
  355. uint8_t marker = 0xC0 | pkt->channel_id;
  356. if ((ret = ffurl_write(h, &marker, 1)) < 0)
  357. return ret;
  358. written++;
  359. if (pkt->ts_field == 0xFFFFFF) {
  360. uint8_t ts_header[4];
  361. AV_WB32(ts_header, timestamp);
  362. if ((ret = ffurl_write(h, ts_header, 4)) < 0)
  363. return ret;
  364. written += 4;
  365. }
  366. }
  367. }
  368. return written;
  369. }
  370. int ff_rtmp_packet_create(RTMPPacket *pkt, int channel_id, RTMPPacketType type,
  371. int timestamp, int size)
  372. {
  373. if (size) {
  374. pkt->data = av_realloc(NULL, size);
  375. if (!pkt->data)
  376. return AVERROR(ENOMEM);
  377. }
  378. pkt->size = size;
  379. pkt->channel_id = channel_id;
  380. pkt->type = type;
  381. pkt->timestamp = timestamp;
  382. pkt->extra = 0;
  383. pkt->ts_field = 0;
  384. return 0;
  385. }
  386. void ff_rtmp_packet_destroy(RTMPPacket *pkt)
  387. {
  388. if (!pkt)
  389. return;
  390. av_freep(&pkt->data);
  391. pkt->size = 0;
  392. }
  393. static int amf_tag_skip(GetByteContext *gb)
  394. {
  395. AMFDataType type;
  396. unsigned nb = -1;
  397. int parse_key = 1;
  398. if (bytestream2_get_bytes_left(gb) < 1)
  399. return -1;
  400. type = bytestream2_get_byte(gb);
  401. switch (type) {
  402. case AMF_DATA_TYPE_NUMBER:
  403. bytestream2_get_be64(gb);
  404. return 0;
  405. case AMF_DATA_TYPE_BOOL:
  406. bytestream2_get_byte(gb);
  407. return 0;
  408. case AMF_DATA_TYPE_STRING:
  409. bytestream2_skip(gb, bytestream2_get_be16(gb));
  410. return 0;
  411. case AMF_DATA_TYPE_LONG_STRING:
  412. bytestream2_skip(gb, bytestream2_get_be32(gb));
  413. return 0;
  414. case AMF_DATA_TYPE_NULL:
  415. return 0;
  416. case AMF_DATA_TYPE_DATE:
  417. bytestream2_skip(gb, 10);
  418. return 0;
  419. case AMF_DATA_TYPE_ARRAY:
  420. parse_key = 0;
  421. case AMF_DATA_TYPE_MIXEDARRAY:
  422. nb = bytestream2_get_be32(gb);
  423. case AMF_DATA_TYPE_OBJECT:
  424. while (nb-- > 0 || type != AMF_DATA_TYPE_ARRAY) {
  425. int t;
  426. if (parse_key) {
  427. int size = bytestream2_get_be16(gb);
  428. if (!size) {
  429. bytestream2_get_byte(gb);
  430. break;
  431. }
  432. if (size < 0 || size >= bytestream2_get_bytes_left(gb))
  433. return -1;
  434. bytestream2_skip(gb, size);
  435. }
  436. t = amf_tag_skip(gb);
  437. if (t < 0 || bytestream2_get_bytes_left(gb) <= 0)
  438. return -1;
  439. }
  440. return 0;
  441. case AMF_DATA_TYPE_OBJECT_END: return 0;
  442. default: return -1;
  443. }
  444. }
  445. int ff_amf_tag_size(const uint8_t *data, const uint8_t *data_end)
  446. {
  447. GetByteContext gb;
  448. int ret;
  449. if (data >= data_end)
  450. return -1;
  451. bytestream2_init(&gb, data, data_end - data);
  452. ret = amf_tag_skip(&gb);
  453. if (ret < 0 || bytestream2_get_bytes_left(&gb) <= 0)
  454. return -1;
  455. av_assert0(bytestream2_tell(&gb) >= 0 && bytestream2_tell(&gb) <= data_end - data);
  456. return bytestream2_tell(&gb);
  457. }
  458. int ff_amf_get_field_value(const uint8_t *data, const uint8_t *data_end,
  459. const uint8_t *name, uint8_t *dst, int dst_size)
  460. {
  461. int namelen = strlen(name);
  462. int len;
  463. while (*data != AMF_DATA_TYPE_OBJECT && data < data_end) {
  464. len = ff_amf_tag_size(data, data_end);
  465. if (len < 0)
  466. len = data_end - data;
  467. data += len;
  468. }
  469. if (data_end - data < 3)
  470. return -1;
  471. data++;
  472. for (;;) {
  473. int size = bytestream_get_be16(&data);
  474. if (!size)
  475. break;
  476. if (size < 0 || size >= data_end - data)
  477. return -1;
  478. data += size;
  479. if (size == namelen && !memcmp(data-size, name, namelen)) {
  480. switch (*data++) {
  481. case AMF_DATA_TYPE_NUMBER:
  482. snprintf(dst, dst_size, "%g", av_int2double(AV_RB64(data)));
  483. break;
  484. case AMF_DATA_TYPE_BOOL:
  485. snprintf(dst, dst_size, "%s", *data ? "true" : "false");
  486. break;
  487. case AMF_DATA_TYPE_STRING:
  488. len = bytestream_get_be16(&data);
  489. av_strlcpy(dst, data, FFMIN(len+1, dst_size));
  490. break;
  491. default:
  492. return -1;
  493. }
  494. return 0;
  495. }
  496. len = ff_amf_tag_size(data, data_end);
  497. if (len < 0 || len >= data_end - data)
  498. return -1;
  499. data += len;
  500. }
  501. return -1;
  502. }
  503. static const char* rtmp_packet_type(int type)
  504. {
  505. switch (type) {
  506. case RTMP_PT_CHUNK_SIZE: return "chunk size";
  507. case RTMP_PT_BYTES_READ: return "bytes read";
  508. case RTMP_PT_PING: return "ping";
  509. case RTMP_PT_SERVER_BW: return "server bandwidth";
  510. case RTMP_PT_CLIENT_BW: return "client bandwidth";
  511. case RTMP_PT_AUDIO: return "audio packet";
  512. case RTMP_PT_VIDEO: return "video packet";
  513. case RTMP_PT_FLEX_STREAM: return "Flex shared stream";
  514. case RTMP_PT_FLEX_OBJECT: return "Flex shared object";
  515. case RTMP_PT_FLEX_MESSAGE: return "Flex shared message";
  516. case RTMP_PT_NOTIFY: return "notification";
  517. case RTMP_PT_SHARED_OBJ: return "shared object";
  518. case RTMP_PT_INVOKE: return "invoke";
  519. case RTMP_PT_METADATA: return "metadata";
  520. default: return "unknown";
  521. }
  522. }
  523. static void amf_tag_contents(void *ctx, const uint8_t *data,
  524. const uint8_t *data_end)
  525. {
  526. unsigned int size, nb = -1;
  527. char buf[1024];
  528. AMFDataType type;
  529. int parse_key = 1;
  530. if (data >= data_end)
  531. return;
  532. switch ((type = *data++)) {
  533. case AMF_DATA_TYPE_NUMBER:
  534. av_log(ctx, AV_LOG_DEBUG, " number %g\n", av_int2double(AV_RB64(data)));
  535. return;
  536. case AMF_DATA_TYPE_BOOL:
  537. av_log(ctx, AV_LOG_DEBUG, " bool %d\n", *data);
  538. return;
  539. case AMF_DATA_TYPE_STRING:
  540. case AMF_DATA_TYPE_LONG_STRING:
  541. if (type == AMF_DATA_TYPE_STRING) {
  542. size = bytestream_get_be16(&data);
  543. } else {
  544. size = bytestream_get_be32(&data);
  545. }
  546. size = FFMIN(size, sizeof(buf) - 1);
  547. memcpy(buf, data, size);
  548. buf[size] = 0;
  549. av_log(ctx, AV_LOG_DEBUG, " string '%s'\n", buf);
  550. return;
  551. case AMF_DATA_TYPE_NULL:
  552. av_log(ctx, AV_LOG_DEBUG, " NULL\n");
  553. return;
  554. case AMF_DATA_TYPE_ARRAY:
  555. parse_key = 0;
  556. case AMF_DATA_TYPE_MIXEDARRAY:
  557. nb = bytestream_get_be32(&data);
  558. case AMF_DATA_TYPE_OBJECT:
  559. av_log(ctx, AV_LOG_DEBUG, " {\n");
  560. while (nb-- > 0 || type != AMF_DATA_TYPE_ARRAY) {
  561. int t;
  562. if (parse_key) {
  563. size = bytestream_get_be16(&data);
  564. size = FFMIN(size, sizeof(buf) - 1);
  565. if (!size) {
  566. av_log(ctx, AV_LOG_DEBUG, " }\n");
  567. data++;
  568. break;
  569. }
  570. memcpy(buf, data, size);
  571. buf[size] = 0;
  572. if (size >= data_end - data)
  573. return;
  574. data += size;
  575. av_log(ctx, AV_LOG_DEBUG, " %s: ", buf);
  576. }
  577. amf_tag_contents(ctx, data, data_end);
  578. t = ff_amf_tag_size(data, data_end);
  579. if (t < 0 || t >= data_end - data)
  580. return;
  581. data += t;
  582. }
  583. return;
  584. case AMF_DATA_TYPE_OBJECT_END:
  585. av_log(ctx, AV_LOG_DEBUG, " }\n");
  586. return;
  587. default:
  588. return;
  589. }
  590. }
  591. void ff_rtmp_packet_dump(void *ctx, RTMPPacket *p)
  592. {
  593. av_log(ctx, AV_LOG_DEBUG, "RTMP packet type '%s'(%d) for channel %d, timestamp %d, extra field %d size %d\n",
  594. rtmp_packet_type(p->type), p->type, p->channel_id, p->timestamp, p->extra, p->size);
  595. if (p->type == RTMP_PT_INVOKE || p->type == RTMP_PT_NOTIFY) {
  596. uint8_t *src = p->data, *src_end = p->data + p->size;
  597. while (src < src_end) {
  598. int sz;
  599. amf_tag_contents(ctx, src, src_end);
  600. sz = ff_amf_tag_size(src, src_end);
  601. if (sz < 0)
  602. break;
  603. src += sz;
  604. }
  605. } else if (p->type == RTMP_PT_SERVER_BW){
  606. av_log(ctx, AV_LOG_DEBUG, "Server BW = %d\n", AV_RB32(p->data));
  607. } else if (p->type == RTMP_PT_CLIENT_BW){
  608. av_log(ctx, AV_LOG_DEBUG, "Client BW = %d\n", AV_RB32(p->data));
  609. } else if (p->type != RTMP_PT_AUDIO && p->type != RTMP_PT_VIDEO && p->type != RTMP_PT_METADATA) {
  610. int i;
  611. for (i = 0; i < p->size; i++)
  612. av_log(ctx, AV_LOG_DEBUG, " %02X", p->data[i]);
  613. av_log(ctx, AV_LOG_DEBUG, "\n");
  614. }
  615. }
  616. int ff_amf_match_string(const uint8_t *data, int size, const char *str)
  617. {
  618. int len = strlen(str);
  619. int amf_len, type;
  620. if (size < 1)
  621. return 0;
  622. type = *data++;
  623. if (type != AMF_DATA_TYPE_LONG_STRING &&
  624. type != AMF_DATA_TYPE_STRING)
  625. return 0;
  626. if (type == AMF_DATA_TYPE_LONG_STRING) {
  627. if ((size -= 4 + 1) < 0)
  628. return 0;
  629. amf_len = bytestream_get_be32(&data);
  630. } else {
  631. if ((size -= 2 + 1) < 0)
  632. return 0;
  633. amf_len = bytestream_get_be16(&data);
  634. }
  635. if (amf_len > size)
  636. return 0;
  637. if (amf_len != len)
  638. return 0;
  639. return !memcmp(data, str, len);
  640. }