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.

1356 lines
45KB

  1. /*
  2. * RTMP network protocol
  3. * Copyright (c) 2009 Kostya Shishkov
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * RTMP protocol
  24. */
  25. #include "libavcodec/bytestream.h"
  26. #include "libavutil/avstring.h"
  27. #include "libavutil/intfloat.h"
  28. #include "libavutil/lfg.h"
  29. #include "libavutil/opt.h"
  30. #include "libavutil/sha.h"
  31. #include "avformat.h"
  32. #include "internal.h"
  33. #include "network.h"
  34. #include "flv.h"
  35. #include "rtmp.h"
  36. #include "rtmppkt.h"
  37. #include "url.h"
  38. //#define DEBUG
  39. #define APP_MAX_LENGTH 128
  40. #define PLAYPATH_MAX_LENGTH 256
  41. #define TCURL_MAX_LENGTH 512
  42. #define FLASHVER_MAX_LENGTH 64
  43. /** RTMP protocol handler state */
  44. typedef enum {
  45. STATE_START, ///< client has not done anything yet
  46. STATE_HANDSHAKED, ///< client has performed handshake
  47. STATE_RELEASING, ///< client releasing stream before publish it (for output)
  48. STATE_FCPUBLISH, ///< client FCPublishing stream (for output)
  49. STATE_CONNECTING, ///< client connected to server successfully
  50. STATE_READY, ///< client has sent all needed commands and waits for server reply
  51. STATE_PLAYING, ///< client has started receiving multimedia data from server
  52. STATE_PUBLISHING, ///< client has started sending multimedia data to server (for output)
  53. STATE_STOPPED, ///< the broadcast has been stopped
  54. } ClientState;
  55. /** protocol handler context */
  56. typedef struct RTMPContext {
  57. const AVClass *class;
  58. URLContext* stream; ///< TCP stream used in interactions with RTMP server
  59. RTMPPacket prev_pkt[2][RTMP_CHANNELS]; ///< packet history used when reading and sending packets
  60. int chunk_size; ///< size of the chunks RTMP packets are divided into
  61. int is_input; ///< input/output flag
  62. char *playpath; ///< stream identifier to play (with possible "mp4:" prefix)
  63. int live; ///< 0: recorded, -1: live, -2: both
  64. char *app; ///< name of application
  65. char *conn; ///< append arbitrary AMF data to the Connect message
  66. ClientState state; ///< current state
  67. int main_channel_id; ///< an additional channel ID which is used for some invocations
  68. uint8_t* flv_data; ///< buffer with data for demuxer
  69. int flv_size; ///< current buffer size
  70. int flv_off; ///< number of bytes read from current buffer
  71. RTMPPacket out_pkt; ///< rtmp packet, created from flv a/v or metadata (for output)
  72. uint32_t client_report_size; ///< number of bytes after which client should report to server
  73. uint32_t bytes_read; ///< number of bytes read from server
  74. uint32_t last_bytes_read; ///< number of bytes read last reported to server
  75. int skip_bytes; ///< number of bytes to skip from the input FLV stream in the next write call
  76. uint8_t flv_header[11]; ///< partial incoming flv packet header
  77. int flv_header_bytes; ///< number of initialized bytes in flv_header
  78. int nb_invokes; ///< keeps track of invoke messages
  79. int create_stream_invoke; ///< invoke id for the create stream command
  80. char* tcurl; ///< url of the target stream
  81. char* flashver; ///< version of the flash plugin
  82. char* swfurl; ///< url of the swf player
  83. } RTMPContext;
  84. #define PLAYER_KEY_OPEN_PART_LEN 30 ///< length of partial key used for first client digest signing
  85. /** Client key used for digest signing */
  86. static const uint8_t rtmp_player_key[] = {
  87. 'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
  88. 'F', 'l', 'a', 's', 'h', ' ', 'P', 'l', 'a', 'y', 'e', 'r', ' ', '0', '0', '1',
  89. 0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
  90. 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
  91. 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
  92. };
  93. #define SERVER_KEY_OPEN_PART_LEN 36 ///< length of partial key used for first server digest signing
  94. /** Key used for RTMP server digest signing */
  95. static const uint8_t rtmp_server_key[] = {
  96. 'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
  97. 'F', 'l', 'a', 's', 'h', ' ', 'M', 'e', 'd', 'i', 'a', ' ',
  98. 'S', 'e', 'r', 'v', 'e', 'r', ' ', '0', '0', '1',
  99. 0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02,
  100. 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8,
  101. 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE
  102. };
  103. static int rtmp_write_amf_data(URLContext *s, char *param, uint8_t **p)
  104. {
  105. char *field, *value, *saveptr;
  106. char type;
  107. /* The type must be B for Boolean, N for number, S for string, O for
  108. * object, or Z for null. For Booleans the data must be either 0 or 1 for
  109. * FALSE or TRUE, respectively. Likewise for Objects the data must be
  110. * 0 or 1 to end or begin an object, respectively. Data items in subobjects
  111. * may be named, by prefixing the type with 'N' and specifying the name
  112. * before the value (ie. NB:myFlag:1). This option may be used multiple times
  113. * to construct arbitrary AMF sequences. */
  114. if (param[0] && param[1] == ':') {
  115. type = param[0];
  116. value = param + 2;
  117. } else if (param[0] == 'N' && param[1] && param[2] == ':') {
  118. type = param[1];
  119. field = strtok_r(param + 3, ":", &saveptr);
  120. value = strtok_r(NULL, ":", &saveptr);
  121. if (!field || !value)
  122. goto fail;
  123. ff_amf_write_field_name(p, field);
  124. } else {
  125. goto fail;
  126. }
  127. switch (type) {
  128. case 'B':
  129. ff_amf_write_bool(p, value[0] != '0');
  130. break;
  131. case 'S':
  132. ff_amf_write_string(p, value);
  133. break;
  134. case 'N':
  135. ff_amf_write_number(p, strtod(value, NULL));
  136. break;
  137. case 'Z':
  138. ff_amf_write_null(p);
  139. break;
  140. case 'O':
  141. if (value[0] != '0')
  142. ff_amf_write_object_start(p);
  143. else
  144. ff_amf_write_object_end(p);
  145. break;
  146. default:
  147. goto fail;
  148. break;
  149. }
  150. return 0;
  151. fail:
  152. av_log(s, AV_LOG_ERROR, "Invalid AMF parameter: %s\n", param);
  153. return AVERROR(EINVAL);
  154. }
  155. /**
  156. * Generate 'connect' call and send it to the server.
  157. */
  158. static int gen_connect(URLContext *s, RTMPContext *rt)
  159. {
  160. RTMPPacket pkt;
  161. uint8_t *p;
  162. int ret;
  163. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  164. 0, 4096)) < 0)
  165. return ret;
  166. p = pkt.data;
  167. ff_amf_write_string(&p, "connect");
  168. ff_amf_write_number(&p, ++rt->nb_invokes);
  169. ff_amf_write_object_start(&p);
  170. ff_amf_write_field_name(&p, "app");
  171. ff_amf_write_string(&p, rt->app);
  172. if (!rt->is_input) {
  173. ff_amf_write_field_name(&p, "type");
  174. ff_amf_write_string(&p, "nonprivate");
  175. }
  176. ff_amf_write_field_name(&p, "flashVer");
  177. ff_amf_write_string(&p, rt->flashver);
  178. if (rt->swfurl) {
  179. ff_amf_write_field_name(&p, "swfUrl");
  180. ff_amf_write_string(&p, rt->swfurl);
  181. }
  182. ff_amf_write_field_name(&p, "tcUrl");
  183. ff_amf_write_string(&p, rt->tcurl);
  184. if (rt->is_input) {
  185. ff_amf_write_field_name(&p, "fpad");
  186. ff_amf_write_bool(&p, 0);
  187. ff_amf_write_field_name(&p, "capabilities");
  188. ff_amf_write_number(&p, 15.0);
  189. /* Tell the server we support all the audio codecs except
  190. * SUPPORT_SND_INTEL (0x0008) and SUPPORT_SND_UNUSED (0x0010)
  191. * which are unused in the RTMP protocol implementation. */
  192. ff_amf_write_field_name(&p, "audioCodecs");
  193. ff_amf_write_number(&p, 4071.0);
  194. ff_amf_write_field_name(&p, "videoCodecs");
  195. ff_amf_write_number(&p, 252.0);
  196. ff_amf_write_field_name(&p, "videoFunction");
  197. ff_amf_write_number(&p, 1.0);
  198. }
  199. ff_amf_write_object_end(&p);
  200. if (rt->conn) {
  201. char *param, *saveptr;
  202. // Write arbitrary AMF data to the Connect message.
  203. param = strtok_r(rt->conn, " ", &saveptr);
  204. while (param != NULL) {
  205. if ((ret = rtmp_write_amf_data(s, param, &p)) < 0) {
  206. // Invalid AMF parameter.
  207. ff_rtmp_packet_destroy(&pkt);
  208. return ret;
  209. }
  210. param = strtok_r(NULL, " ", &saveptr);
  211. }
  212. }
  213. pkt.data_size = p - pkt.data;
  214. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  215. rt->prev_pkt[1]);
  216. ff_rtmp_packet_destroy(&pkt);
  217. return ret;
  218. }
  219. /**
  220. * Generate 'releaseStream' call and send it to the server. It should make
  221. * the server release some channel for media streams.
  222. */
  223. static int gen_release_stream(URLContext *s, RTMPContext *rt)
  224. {
  225. RTMPPacket pkt;
  226. uint8_t *p;
  227. int ret;
  228. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  229. 0, 29 + strlen(rt->playpath))) < 0)
  230. return ret;
  231. av_log(s, AV_LOG_DEBUG, "Releasing stream...\n");
  232. p = pkt.data;
  233. ff_amf_write_string(&p, "releaseStream");
  234. ff_amf_write_number(&p, ++rt->nb_invokes);
  235. ff_amf_write_null(&p);
  236. ff_amf_write_string(&p, rt->playpath);
  237. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  238. rt->prev_pkt[1]);
  239. ff_rtmp_packet_destroy(&pkt);
  240. return ret;
  241. }
  242. /**
  243. * Generate 'FCPublish' call and send it to the server. It should make
  244. * the server preapare for receiving media streams.
  245. */
  246. static int gen_fcpublish_stream(URLContext *s, RTMPContext *rt)
  247. {
  248. RTMPPacket pkt;
  249. uint8_t *p;
  250. int ret;
  251. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  252. 0, 25 + strlen(rt->playpath))) < 0)
  253. return ret;
  254. av_log(s, AV_LOG_DEBUG, "FCPublish stream...\n");
  255. p = pkt.data;
  256. ff_amf_write_string(&p, "FCPublish");
  257. ff_amf_write_number(&p, ++rt->nb_invokes);
  258. ff_amf_write_null(&p);
  259. ff_amf_write_string(&p, rt->playpath);
  260. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  261. rt->prev_pkt[1]);
  262. ff_rtmp_packet_destroy(&pkt);
  263. return ret;
  264. }
  265. /**
  266. * Generate 'FCUnpublish' call and send it to the server. It should make
  267. * the server destroy stream.
  268. */
  269. static int gen_fcunpublish_stream(URLContext *s, RTMPContext *rt)
  270. {
  271. RTMPPacket pkt;
  272. uint8_t *p;
  273. int ret;
  274. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  275. 0, 27 + strlen(rt->playpath))) < 0)
  276. return ret;
  277. av_log(s, AV_LOG_DEBUG, "UnPublishing stream...\n");
  278. p = pkt.data;
  279. ff_amf_write_string(&p, "FCUnpublish");
  280. ff_amf_write_number(&p, ++rt->nb_invokes);
  281. ff_amf_write_null(&p);
  282. ff_amf_write_string(&p, rt->playpath);
  283. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  284. rt->prev_pkt[1]);
  285. ff_rtmp_packet_destroy(&pkt);
  286. return ret;
  287. }
  288. /**
  289. * Generate 'createStream' call and send it to the server. It should make
  290. * the server allocate some channel for media streams.
  291. */
  292. static int gen_create_stream(URLContext *s, RTMPContext *rt)
  293. {
  294. RTMPPacket pkt;
  295. uint8_t *p;
  296. int ret;
  297. av_log(s, AV_LOG_DEBUG, "Creating stream...\n");
  298. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  299. 0, 25)) < 0)
  300. return ret;
  301. p = pkt.data;
  302. ff_amf_write_string(&p, "createStream");
  303. ff_amf_write_number(&p, ++rt->nb_invokes);
  304. ff_amf_write_null(&p);
  305. rt->create_stream_invoke = rt->nb_invokes;
  306. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  307. rt->prev_pkt[1]);
  308. ff_rtmp_packet_destroy(&pkt);
  309. return ret;
  310. }
  311. /**
  312. * Generate 'deleteStream' call and send it to the server. It should make
  313. * the server remove some channel for media streams.
  314. */
  315. static int gen_delete_stream(URLContext *s, RTMPContext *rt)
  316. {
  317. RTMPPacket pkt;
  318. uint8_t *p;
  319. int ret;
  320. av_log(s, AV_LOG_DEBUG, "Deleting stream...\n");
  321. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  322. 0, 34)) < 0)
  323. return ret;
  324. p = pkt.data;
  325. ff_amf_write_string(&p, "deleteStream");
  326. ff_amf_write_number(&p, ++rt->nb_invokes);
  327. ff_amf_write_null(&p);
  328. ff_amf_write_number(&p, rt->main_channel_id);
  329. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  330. rt->prev_pkt[1]);
  331. ff_rtmp_packet_destroy(&pkt);
  332. return ret;
  333. }
  334. /**
  335. * Generate 'play' call and send it to the server, then ping the server
  336. * to start actual playing.
  337. */
  338. static int gen_play(URLContext *s, RTMPContext *rt)
  339. {
  340. RTMPPacket pkt;
  341. uint8_t *p;
  342. int ret;
  343. av_log(s, AV_LOG_DEBUG, "Sending play command for '%s'\n", rt->playpath);
  344. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_VIDEO_CHANNEL, RTMP_PT_INVOKE,
  345. 0, 29 + strlen(rt->playpath))) < 0)
  346. return ret;
  347. pkt.extra = rt->main_channel_id;
  348. p = pkt.data;
  349. ff_amf_write_string(&p, "play");
  350. ff_amf_write_number(&p, ++rt->nb_invokes);
  351. ff_amf_write_null(&p);
  352. ff_amf_write_string(&p, rt->playpath);
  353. ff_amf_write_number(&p, rt->live);
  354. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  355. rt->prev_pkt[1]);
  356. ff_rtmp_packet_destroy(&pkt);
  357. if (ret < 0)
  358. return ret;
  359. // set client buffer time disguised in ping packet
  360. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
  361. 1, 10)) < 0)
  362. return ret;
  363. p = pkt.data;
  364. bytestream_put_be16(&p, 3);
  365. bytestream_put_be32(&p, 1);
  366. bytestream_put_be32(&p, 256); //TODO: what is a good value here?
  367. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  368. rt->prev_pkt[1]);
  369. ff_rtmp_packet_destroy(&pkt);
  370. return ret;
  371. }
  372. /**
  373. * Generate 'publish' call and send it to the server.
  374. */
  375. static int gen_publish(URLContext *s, RTMPContext *rt)
  376. {
  377. RTMPPacket pkt;
  378. uint8_t *p;
  379. int ret;
  380. av_log(s, AV_LOG_DEBUG, "Sending publish command for '%s'\n", rt->playpath);
  381. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SOURCE_CHANNEL, RTMP_PT_INVOKE,
  382. 0, 30 + strlen(rt->playpath))) < 0)
  383. return ret;
  384. pkt.extra = rt->main_channel_id;
  385. p = pkt.data;
  386. ff_amf_write_string(&p, "publish");
  387. ff_amf_write_number(&p, ++rt->nb_invokes);
  388. ff_amf_write_null(&p);
  389. ff_amf_write_string(&p, rt->playpath);
  390. ff_amf_write_string(&p, "live");
  391. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  392. rt->prev_pkt[1]);
  393. ff_rtmp_packet_destroy(&pkt);
  394. return ret;
  395. }
  396. /**
  397. * Generate ping reply and send it to the server.
  398. */
  399. static int gen_pong(URLContext *s, RTMPContext *rt, RTMPPacket *ppkt)
  400. {
  401. RTMPPacket pkt;
  402. uint8_t *p;
  403. int ret;
  404. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_PING,
  405. ppkt->timestamp + 1, 6)) < 0)
  406. return ret;
  407. p = pkt.data;
  408. bytestream_put_be16(&p, 7);
  409. bytestream_put_be32(&p, AV_RB32(ppkt->data+2));
  410. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  411. rt->prev_pkt[1]);
  412. ff_rtmp_packet_destroy(&pkt);
  413. return ret;
  414. }
  415. /**
  416. * Generate server bandwidth message and send it to the server.
  417. */
  418. static int gen_server_bw(URLContext *s, RTMPContext *rt)
  419. {
  420. RTMPPacket pkt;
  421. uint8_t *p;
  422. int ret;
  423. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_SERVER_BW,
  424. 0, 4)) < 0)
  425. return ret;
  426. p = pkt.data;
  427. bytestream_put_be32(&p, 2500000);
  428. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  429. rt->prev_pkt[1]);
  430. ff_rtmp_packet_destroy(&pkt);
  431. return ret;
  432. }
  433. /**
  434. * Generate check bandwidth message and send it to the server.
  435. */
  436. static int gen_check_bw(URLContext *s, RTMPContext *rt)
  437. {
  438. RTMPPacket pkt;
  439. uint8_t *p;
  440. int ret;
  441. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_SYSTEM_CHANNEL, RTMP_PT_INVOKE,
  442. 0, 21)) < 0)
  443. return ret;
  444. p = pkt.data;
  445. ff_amf_write_string(&p, "_checkbw");
  446. ff_amf_write_number(&p, ++rt->nb_invokes);
  447. ff_amf_write_null(&p);
  448. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  449. rt->prev_pkt[1]);
  450. ff_rtmp_packet_destroy(&pkt);
  451. return ret;
  452. }
  453. /**
  454. * Generate report on bytes read so far and send it to the server.
  455. */
  456. static int gen_bytes_read(URLContext *s, RTMPContext *rt, uint32_t ts)
  457. {
  458. RTMPPacket pkt;
  459. uint8_t *p;
  460. int ret;
  461. if ((ret = ff_rtmp_packet_create(&pkt, RTMP_NETWORK_CHANNEL, RTMP_PT_BYTES_READ,
  462. ts, 4)) < 0)
  463. return ret;
  464. p = pkt.data;
  465. bytestream_put_be32(&p, rt->bytes_read);
  466. ret = ff_rtmp_packet_write(rt->stream, &pkt, rt->chunk_size,
  467. rt->prev_pkt[1]);
  468. ff_rtmp_packet_destroy(&pkt);
  469. return ret;
  470. }
  471. //TODO: Move HMAC code somewhere. Eventually.
  472. #define HMAC_IPAD_VAL 0x36
  473. #define HMAC_OPAD_VAL 0x5C
  474. /**
  475. * Calculate HMAC-SHA2 digest for RTMP handshake packets.
  476. *
  477. * @param src input buffer
  478. * @param len input buffer length (should be 1536)
  479. * @param gap offset in buffer where 32 bytes should not be taken into account
  480. * when calculating digest (since it will be used to store that digest)
  481. * @param key digest key
  482. * @param keylen digest key length
  483. * @param dst buffer where calculated digest will be stored (32 bytes)
  484. */
  485. static int rtmp_calc_digest(const uint8_t *src, int len, int gap,
  486. const uint8_t *key, int keylen, uint8_t *dst)
  487. {
  488. struct AVSHA *sha;
  489. uint8_t hmac_buf[64+32] = {0};
  490. int i;
  491. sha = av_mallocz(av_sha_size);
  492. if (!sha)
  493. return AVERROR(ENOMEM);
  494. if (keylen < 64) {
  495. memcpy(hmac_buf, key, keylen);
  496. } else {
  497. av_sha_init(sha, 256);
  498. av_sha_update(sha,key, keylen);
  499. av_sha_final(sha, hmac_buf);
  500. }
  501. for (i = 0; i < 64; i++)
  502. hmac_buf[i] ^= HMAC_IPAD_VAL;
  503. av_sha_init(sha, 256);
  504. av_sha_update(sha, hmac_buf, 64);
  505. if (gap <= 0) {
  506. av_sha_update(sha, src, len);
  507. } else { //skip 32 bytes used for storing digest
  508. av_sha_update(sha, src, gap);
  509. av_sha_update(sha, src + gap + 32, len - gap - 32);
  510. }
  511. av_sha_final(sha, hmac_buf + 64);
  512. for (i = 0; i < 64; i++)
  513. hmac_buf[i] ^= HMAC_IPAD_VAL ^ HMAC_OPAD_VAL; //reuse XORed key for opad
  514. av_sha_init(sha, 256);
  515. av_sha_update(sha, hmac_buf, 64+32);
  516. av_sha_final(sha, dst);
  517. av_free(sha);
  518. return 0;
  519. }
  520. /**
  521. * Put HMAC-SHA2 digest of packet data (except for the bytes where this digest
  522. * will be stored) into that packet.
  523. *
  524. * @param buf handshake data (1536 bytes)
  525. * @return offset to the digest inside input data
  526. */
  527. static int rtmp_handshake_imprint_with_digest(uint8_t *buf)
  528. {
  529. int i, digest_pos = 0;
  530. int ret;
  531. for (i = 8; i < 12; i++)
  532. digest_pos += buf[i];
  533. digest_pos = (digest_pos % 728) + 12;
  534. ret = rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
  535. rtmp_player_key, PLAYER_KEY_OPEN_PART_LEN,
  536. buf + digest_pos);
  537. if (ret < 0)
  538. return ret;
  539. return digest_pos;
  540. }
  541. /**
  542. * Verify that the received server response has the expected digest value.
  543. *
  544. * @param buf handshake data received from the server (1536 bytes)
  545. * @param off position to search digest offset from
  546. * @return 0 if digest is valid, digest position otherwise
  547. */
  548. static int rtmp_validate_digest(uint8_t *buf, int off)
  549. {
  550. int i, digest_pos = 0;
  551. uint8_t digest[32];
  552. int ret;
  553. for (i = 0; i < 4; i++)
  554. digest_pos += buf[i + off];
  555. digest_pos = (digest_pos % 728) + off + 4;
  556. ret = rtmp_calc_digest(buf, RTMP_HANDSHAKE_PACKET_SIZE, digest_pos,
  557. rtmp_server_key, SERVER_KEY_OPEN_PART_LEN,
  558. digest);
  559. if (ret < 0)
  560. return ret;
  561. if (!memcmp(digest, buf + digest_pos, 32))
  562. return digest_pos;
  563. return 0;
  564. }
  565. /**
  566. * Perform handshake with the server by means of exchanging pseudorandom data
  567. * signed with HMAC-SHA2 digest.
  568. *
  569. * @return 0 if handshake succeeds, negative value otherwise
  570. */
  571. static int rtmp_handshake(URLContext *s, RTMPContext *rt)
  572. {
  573. AVLFG rnd;
  574. uint8_t tosend [RTMP_HANDSHAKE_PACKET_SIZE+1] = {
  575. 3, // unencrypted data
  576. 0, 0, 0, 0, // client uptime
  577. RTMP_CLIENT_VER1,
  578. RTMP_CLIENT_VER2,
  579. RTMP_CLIENT_VER3,
  580. RTMP_CLIENT_VER4,
  581. };
  582. uint8_t clientdata[RTMP_HANDSHAKE_PACKET_SIZE];
  583. uint8_t serverdata[RTMP_HANDSHAKE_PACKET_SIZE+1];
  584. int i;
  585. int server_pos, client_pos;
  586. uint8_t digest[32];
  587. int ret;
  588. av_log(s, AV_LOG_DEBUG, "Handshaking...\n");
  589. av_lfg_init(&rnd, 0xDEADC0DE);
  590. // generate handshake packet - 1536 bytes of pseudorandom data
  591. for (i = 9; i <= RTMP_HANDSHAKE_PACKET_SIZE; i++)
  592. tosend[i] = av_lfg_get(&rnd) >> 24;
  593. client_pos = rtmp_handshake_imprint_with_digest(tosend + 1);
  594. if (client_pos < 0)
  595. return client_pos;
  596. if ((ret = ffurl_write(rt->stream, tosend,
  597. RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
  598. av_log(s, AV_LOG_ERROR, "Cannot write RTMP handshake request\n");
  599. return ret;
  600. }
  601. if ((ret = ffurl_read_complete(rt->stream, serverdata,
  602. RTMP_HANDSHAKE_PACKET_SIZE + 1)) < 0) {
  603. av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
  604. return ret;
  605. }
  606. if ((ret = ffurl_read_complete(rt->stream, clientdata,
  607. RTMP_HANDSHAKE_PACKET_SIZE)) < 0) {
  608. av_log(s, AV_LOG_ERROR, "Cannot read RTMP handshake response\n");
  609. return ret;
  610. }
  611. av_log(s, AV_LOG_DEBUG, "Server version %d.%d.%d.%d\n",
  612. serverdata[5], serverdata[6], serverdata[7], serverdata[8]);
  613. if (rt->is_input && serverdata[5] >= 3) {
  614. server_pos = rtmp_validate_digest(serverdata + 1, 772);
  615. if (server_pos < 0)
  616. return server_pos;
  617. if (!server_pos) {
  618. server_pos = rtmp_validate_digest(serverdata + 1, 8);
  619. if (server_pos < 0)
  620. return server_pos;
  621. if (!server_pos) {
  622. av_log(s, AV_LOG_ERROR, "Server response validating failed\n");
  623. return AVERROR(EIO);
  624. }
  625. }
  626. ret = rtmp_calc_digest(tosend + 1 + client_pos, 32, 0, rtmp_server_key,
  627. sizeof(rtmp_server_key), digest);
  628. if (ret < 0)
  629. return ret;
  630. ret = rtmp_calc_digest(clientdata, RTMP_HANDSHAKE_PACKET_SIZE - 32, 0,
  631. digest, 32, digest);
  632. if (ret < 0)
  633. return ret;
  634. if (memcmp(digest, clientdata + RTMP_HANDSHAKE_PACKET_SIZE - 32, 32)) {
  635. av_log(s, AV_LOG_ERROR, "Signature mismatch\n");
  636. return AVERROR(EIO);
  637. }
  638. for (i = 0; i < RTMP_HANDSHAKE_PACKET_SIZE; i++)
  639. tosend[i] = av_lfg_get(&rnd) >> 24;
  640. ret = rtmp_calc_digest(serverdata + 1 + server_pos, 32, 0,
  641. rtmp_player_key, sizeof(rtmp_player_key),
  642. digest);
  643. if (ret < 0)
  644. return ret;
  645. ret = rtmp_calc_digest(tosend, RTMP_HANDSHAKE_PACKET_SIZE - 32, 0,
  646. digest, 32,
  647. tosend + RTMP_HANDSHAKE_PACKET_SIZE - 32);
  648. if (ret < 0)
  649. return ret;
  650. // write reply back to the server
  651. if ((ret = ffurl_write(rt->stream, tosend,
  652. RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
  653. return ret;
  654. } else {
  655. if ((ret = ffurl_write(rt->stream, serverdata + 1,
  656. RTMP_HANDSHAKE_PACKET_SIZE)) < 0)
  657. return ret;
  658. }
  659. return 0;
  660. }
  661. /**
  662. * Parse received packet and possibly perform some action depending on
  663. * the packet contents.
  664. * @return 0 for no errors, negative values for serious errors which prevent
  665. * further communications, positive values for uncritical errors
  666. */
  667. static int rtmp_parse_result(URLContext *s, RTMPContext *rt, RTMPPacket *pkt)
  668. {
  669. int i, t;
  670. const uint8_t *data_end = pkt->data + pkt->data_size;
  671. int ret;
  672. #ifdef DEBUG
  673. ff_rtmp_packet_dump(s, pkt);
  674. #endif
  675. switch (pkt->type) {
  676. case RTMP_PT_CHUNK_SIZE:
  677. if (pkt->data_size != 4) {
  678. av_log(s, AV_LOG_ERROR,
  679. "Chunk size change packet is not 4 bytes long (%d)\n", pkt->data_size);
  680. return -1;
  681. }
  682. if (!rt->is_input)
  683. if ((ret = ff_rtmp_packet_write(rt->stream, pkt, rt->chunk_size,
  684. rt->prev_pkt[1])) < 0)
  685. return ret;
  686. rt->chunk_size = AV_RB32(pkt->data);
  687. if (rt->chunk_size <= 0) {
  688. av_log(s, AV_LOG_ERROR, "Incorrect chunk size %d\n", rt->chunk_size);
  689. return -1;
  690. }
  691. av_log(s, AV_LOG_DEBUG, "New chunk size = %d\n", rt->chunk_size);
  692. break;
  693. case RTMP_PT_PING:
  694. t = AV_RB16(pkt->data);
  695. if (t == 6)
  696. if ((ret = gen_pong(s, rt, pkt)) < 0)
  697. return ret;
  698. break;
  699. case RTMP_PT_CLIENT_BW:
  700. if (pkt->data_size < 4) {
  701. av_log(s, AV_LOG_ERROR,
  702. "Client bandwidth report packet is less than 4 bytes long (%d)\n",
  703. pkt->data_size);
  704. return -1;
  705. }
  706. av_log(s, AV_LOG_DEBUG, "Client bandwidth = %d\n", AV_RB32(pkt->data));
  707. rt->client_report_size = AV_RB32(pkt->data) >> 1;
  708. break;
  709. case RTMP_PT_INVOKE:
  710. //TODO: check for the messages sent for wrong state?
  711. if (!memcmp(pkt->data, "\002\000\006_error", 9)) {
  712. uint8_t tmpstr[256];
  713. if (!ff_amf_get_field_value(pkt->data + 9, data_end,
  714. "description", tmpstr, sizeof(tmpstr)))
  715. av_log(s, AV_LOG_ERROR, "Server error: %s\n",tmpstr);
  716. return -1;
  717. } else if (!memcmp(pkt->data, "\002\000\007_result", 10)) {
  718. switch (rt->state) {
  719. case STATE_HANDSHAKED:
  720. if (!rt->is_input) {
  721. if ((ret = gen_release_stream(s, rt)) < 0)
  722. return ret;
  723. if ((ret = gen_fcpublish_stream(s, rt)) < 0)
  724. return ret;
  725. rt->state = STATE_RELEASING;
  726. } else {
  727. if ((ret = gen_server_bw(s, rt)) < 0)
  728. return ret;
  729. rt->state = STATE_CONNECTING;
  730. }
  731. if ((ret = gen_create_stream(s, rt)) < 0)
  732. return ret;
  733. break;
  734. case STATE_FCPUBLISH:
  735. rt->state = STATE_CONNECTING;
  736. break;
  737. case STATE_RELEASING:
  738. rt->state = STATE_FCPUBLISH;
  739. /* hack for Wowza Media Server, it does not send result for
  740. * releaseStream and FCPublish calls */
  741. if (!pkt->data[10]) {
  742. int pkt_id = av_int2double(AV_RB64(pkt->data + 11));
  743. if (pkt_id == rt->create_stream_invoke)
  744. rt->state = STATE_CONNECTING;
  745. }
  746. if (rt->state != STATE_CONNECTING)
  747. break;
  748. case STATE_CONNECTING:
  749. //extract a number from the result
  750. if (pkt->data[10] || pkt->data[19] != 5 || pkt->data[20]) {
  751. av_log(s, AV_LOG_WARNING, "Unexpected reply on connect()\n");
  752. } else {
  753. rt->main_channel_id = av_int2double(AV_RB64(pkt->data + 21));
  754. }
  755. if (rt->is_input) {
  756. if ((ret = gen_play(s, rt)) < 0)
  757. return ret;
  758. } else {
  759. if ((ret = gen_publish(s, rt)) < 0)
  760. return ret;
  761. }
  762. rt->state = STATE_READY;
  763. break;
  764. }
  765. } else if (!memcmp(pkt->data, "\002\000\010onStatus", 11)) {
  766. const uint8_t* ptr = pkt->data + 11;
  767. uint8_t tmpstr[256];
  768. for (i = 0; i < 2; i++) {
  769. t = ff_amf_tag_size(ptr, data_end);
  770. if (t < 0)
  771. return 1;
  772. ptr += t;
  773. }
  774. t = ff_amf_get_field_value(ptr, data_end,
  775. "level", tmpstr, sizeof(tmpstr));
  776. if (!t && !strcmp(tmpstr, "error")) {
  777. if (!ff_amf_get_field_value(ptr, data_end,
  778. "description", tmpstr, sizeof(tmpstr)))
  779. av_log(s, AV_LOG_ERROR, "Server error: %s\n",tmpstr);
  780. return -1;
  781. }
  782. t = ff_amf_get_field_value(ptr, data_end,
  783. "code", tmpstr, sizeof(tmpstr));
  784. if (!t && !strcmp(tmpstr, "NetStream.Play.Start")) rt->state = STATE_PLAYING;
  785. if (!t && !strcmp(tmpstr, "NetStream.Play.Stop")) rt->state = STATE_STOPPED;
  786. if (!t && !strcmp(tmpstr, "NetStream.Play.UnpublishNotify")) rt->state = STATE_STOPPED;
  787. if (!t && !strcmp(tmpstr, "NetStream.Publish.Start")) rt->state = STATE_PUBLISHING;
  788. } else if (!memcmp(pkt->data, "\002\000\010onBWDone", 11)) {
  789. if ((ret = gen_check_bw(s, rt)) < 0)
  790. return ret;
  791. }
  792. break;
  793. }
  794. return 0;
  795. }
  796. /**
  797. * Interact with the server by receiving and sending RTMP packets until
  798. * there is some significant data (media data or expected status notification).
  799. *
  800. * @param s reading context
  801. * @param for_header non-zero value tells function to work until it
  802. * gets notification from the server that playing has been started,
  803. * otherwise function will work until some media data is received (or
  804. * an error happens)
  805. * @return 0 for successful operation, negative value in case of error
  806. */
  807. static int get_packet(URLContext *s, int for_header)
  808. {
  809. RTMPContext *rt = s->priv_data;
  810. int ret;
  811. uint8_t *p;
  812. const uint8_t *next;
  813. uint32_t data_size;
  814. uint32_t ts, cts, pts=0;
  815. if (rt->state == STATE_STOPPED)
  816. return AVERROR_EOF;
  817. for (;;) {
  818. RTMPPacket rpkt = { 0 };
  819. if ((ret = ff_rtmp_packet_read(rt->stream, &rpkt,
  820. rt->chunk_size, rt->prev_pkt[0])) <= 0) {
  821. if (ret == 0) {
  822. return AVERROR(EAGAIN);
  823. } else {
  824. return AVERROR(EIO);
  825. }
  826. }
  827. rt->bytes_read += ret;
  828. if (rt->bytes_read > rt->last_bytes_read + rt->client_report_size) {
  829. av_log(s, AV_LOG_DEBUG, "Sending bytes read report\n");
  830. if ((ret = gen_bytes_read(s, rt, rpkt.timestamp + 1)) < 0)
  831. return ret;
  832. rt->last_bytes_read = rt->bytes_read;
  833. }
  834. ret = rtmp_parse_result(s, rt, &rpkt);
  835. if (ret < 0) {//serious error in current packet
  836. ff_rtmp_packet_destroy(&rpkt);
  837. return ret;
  838. }
  839. if (rt->state == STATE_STOPPED) {
  840. ff_rtmp_packet_destroy(&rpkt);
  841. return AVERROR_EOF;
  842. }
  843. if (for_header && (rt->state == STATE_PLAYING || rt->state == STATE_PUBLISHING)) {
  844. ff_rtmp_packet_destroy(&rpkt);
  845. return 0;
  846. }
  847. if (!rpkt.data_size || !rt->is_input) {
  848. ff_rtmp_packet_destroy(&rpkt);
  849. continue;
  850. }
  851. if (rpkt.type == RTMP_PT_VIDEO || rpkt.type == RTMP_PT_AUDIO ||
  852. (rpkt.type == RTMP_PT_NOTIFY && !memcmp("\002\000\012onMetaData", rpkt.data, 13))) {
  853. ts = rpkt.timestamp;
  854. // generate packet header and put data into buffer for FLV demuxer
  855. rt->flv_off = 0;
  856. rt->flv_size = rpkt.data_size + 15;
  857. rt->flv_data = p = av_realloc(rt->flv_data, rt->flv_size);
  858. bytestream_put_byte(&p, rpkt.type);
  859. bytestream_put_be24(&p, rpkt.data_size);
  860. bytestream_put_be24(&p, ts);
  861. bytestream_put_byte(&p, ts >> 24);
  862. bytestream_put_be24(&p, 0);
  863. bytestream_put_buffer(&p, rpkt.data, rpkt.data_size);
  864. bytestream_put_be32(&p, 0);
  865. ff_rtmp_packet_destroy(&rpkt);
  866. return 0;
  867. } else if (rpkt.type == RTMP_PT_METADATA) {
  868. // we got raw FLV data, make it available for FLV demuxer
  869. rt->flv_off = 0;
  870. rt->flv_size = rpkt.data_size;
  871. rt->flv_data = av_realloc(rt->flv_data, rt->flv_size);
  872. /* rewrite timestamps */
  873. next = rpkt.data;
  874. ts = rpkt.timestamp;
  875. while (next - rpkt.data < rpkt.data_size - 11) {
  876. next++;
  877. data_size = bytestream_get_be24(&next);
  878. p=next;
  879. cts = bytestream_get_be24(&next);
  880. cts |= bytestream_get_byte(&next) << 24;
  881. if (pts==0)
  882. pts=cts;
  883. ts += cts - pts;
  884. pts = cts;
  885. bytestream_put_be24(&p, ts);
  886. bytestream_put_byte(&p, ts >> 24);
  887. next += data_size + 3 + 4;
  888. }
  889. memcpy(rt->flv_data, rpkt.data, rpkt.data_size);
  890. ff_rtmp_packet_destroy(&rpkt);
  891. return 0;
  892. }
  893. ff_rtmp_packet_destroy(&rpkt);
  894. }
  895. }
  896. static int rtmp_close(URLContext *h)
  897. {
  898. RTMPContext *rt = h->priv_data;
  899. int ret = 0;
  900. if (!rt->is_input) {
  901. rt->flv_data = NULL;
  902. if (rt->out_pkt.data_size)
  903. ff_rtmp_packet_destroy(&rt->out_pkt);
  904. if (rt->state > STATE_FCPUBLISH)
  905. ret = gen_fcunpublish_stream(h, rt);
  906. }
  907. if (rt->state > STATE_HANDSHAKED)
  908. ret = gen_delete_stream(h, rt);
  909. av_freep(&rt->flv_data);
  910. ffurl_close(rt->stream);
  911. return ret;
  912. }
  913. /**
  914. * Open RTMP connection and verify that the stream can be played.
  915. *
  916. * URL syntax: rtmp://server[:port][/app][/playpath]
  917. * where 'app' is first one or two directories in the path
  918. * (e.g. /ondemand/, /flash/live/, etc.)
  919. * and 'playpath' is a file name (the rest of the path,
  920. * may be prefixed with "mp4:")
  921. */
  922. static int rtmp_open(URLContext *s, const char *uri, int flags)
  923. {
  924. RTMPContext *rt = s->priv_data;
  925. char proto[8], hostname[256], path[1024], *fname;
  926. char *old_app;
  927. uint8_t buf[2048];
  928. int port;
  929. int ret;
  930. rt->is_input = !(flags & AVIO_FLAG_WRITE);
  931. av_url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), &port,
  932. path, sizeof(path), s->filename);
  933. if (port < 0)
  934. port = RTMP_DEFAULT_PORT;
  935. ff_url_join(buf, sizeof(buf), "tcp", NULL, hostname, port, NULL);
  936. if ((ret = ffurl_open(&rt->stream, buf, AVIO_FLAG_READ_WRITE,
  937. &s->interrupt_callback, NULL)) < 0) {
  938. av_log(s , AV_LOG_ERROR, "Cannot open connection %s\n", buf);
  939. goto fail;
  940. }
  941. rt->state = STATE_START;
  942. if ((ret = rtmp_handshake(s, rt)) < 0)
  943. goto fail;
  944. rt->chunk_size = 128;
  945. rt->state = STATE_HANDSHAKED;
  946. // Keep the application name when it has been defined by the user.
  947. old_app = rt->app;
  948. rt->app = av_malloc(APP_MAX_LENGTH);
  949. if (!rt->app) {
  950. ret = AVERROR(ENOMEM);
  951. goto fail;
  952. }
  953. //extract "app" part from path
  954. if (!strncmp(path, "/ondemand/", 10)) {
  955. fname = path + 10;
  956. memcpy(rt->app, "ondemand", 9);
  957. } else {
  958. char *next = *path ? path + 1 : path;
  959. char *p = strchr(next, '/');
  960. if (!p) {
  961. fname = next;
  962. rt->app[0] = '\0';
  963. } else {
  964. // make sure we do not mismatch a playpath for an application instance
  965. char *c = strchr(p + 1, ':');
  966. fname = strchr(p + 1, '/');
  967. if (!fname || (c && c < fname)) {
  968. fname = p + 1;
  969. av_strlcpy(rt->app, path + 1, p - path);
  970. } else {
  971. fname++;
  972. av_strlcpy(rt->app, path + 1, fname - path - 1);
  973. }
  974. }
  975. }
  976. if (old_app) {
  977. // The name of application has been defined by the user, override it.
  978. av_free(rt->app);
  979. rt->app = old_app;
  980. }
  981. if (!rt->playpath) {
  982. rt->playpath = av_malloc(PLAYPATH_MAX_LENGTH);
  983. if (!rt->playpath) {
  984. ret = AVERROR(ENOMEM);
  985. goto fail;
  986. }
  987. if (!strchr(fname, ':') &&
  988. (!strcmp(fname + strlen(fname) - 4, ".f4v") ||
  989. !strcmp(fname + strlen(fname) - 4, ".mp4"))) {
  990. memcpy(rt->playpath, "mp4:", 5);
  991. } else {
  992. rt->playpath[0] = 0;
  993. }
  994. strncat(rt->playpath, fname, PLAYPATH_MAX_LENGTH - 5);
  995. }
  996. if (!rt->tcurl) {
  997. rt->tcurl = av_malloc(TCURL_MAX_LENGTH);
  998. if (!rt->tcurl) {
  999. ret = AVERROR(ENOMEM);
  1000. goto fail;
  1001. }
  1002. ff_url_join(rt->tcurl, TCURL_MAX_LENGTH, proto, NULL, hostname,
  1003. port, "/%s", rt->app);
  1004. }
  1005. if (!rt->flashver) {
  1006. rt->flashver = av_malloc(FLASHVER_MAX_LENGTH);
  1007. if (!rt->flashver) {
  1008. ret = AVERROR(ENOMEM);
  1009. goto fail;
  1010. }
  1011. if (rt->is_input) {
  1012. snprintf(rt->flashver, FLASHVER_MAX_LENGTH, "%s %d,%d,%d,%d",
  1013. RTMP_CLIENT_PLATFORM, RTMP_CLIENT_VER1, RTMP_CLIENT_VER2,
  1014. RTMP_CLIENT_VER3, RTMP_CLIENT_VER4);
  1015. } else {
  1016. snprintf(rt->flashver, FLASHVER_MAX_LENGTH,
  1017. "FMLE/3.0 (compatible; %s)", LIBAVFORMAT_IDENT);
  1018. }
  1019. }
  1020. rt->client_report_size = 1048576;
  1021. rt->bytes_read = 0;
  1022. rt->last_bytes_read = 0;
  1023. av_log(s, AV_LOG_DEBUG, "Proto = %s, path = %s, app = %s, fname = %s\n",
  1024. proto, path, rt->app, rt->playpath);
  1025. if ((ret = gen_connect(s, rt)) < 0)
  1026. goto fail;
  1027. do {
  1028. ret = get_packet(s, 1);
  1029. } while (ret == EAGAIN);
  1030. if (ret < 0)
  1031. goto fail;
  1032. if (rt->is_input) {
  1033. // generate FLV header for demuxer
  1034. rt->flv_size = 13;
  1035. rt->flv_data = av_realloc(rt->flv_data, rt->flv_size);
  1036. rt->flv_off = 0;
  1037. memcpy(rt->flv_data, "FLV\1\5\0\0\0\011\0\0\0\0", rt->flv_size);
  1038. } else {
  1039. rt->flv_size = 0;
  1040. rt->flv_data = NULL;
  1041. rt->flv_off = 0;
  1042. rt->skip_bytes = 13;
  1043. }
  1044. s->max_packet_size = rt->stream->max_packet_size;
  1045. s->is_streamed = 1;
  1046. return 0;
  1047. fail:
  1048. rtmp_close(s);
  1049. return ret;
  1050. }
  1051. static int rtmp_read(URLContext *s, uint8_t *buf, int size)
  1052. {
  1053. RTMPContext *rt = s->priv_data;
  1054. int orig_size = size;
  1055. int ret;
  1056. while (size > 0) {
  1057. int data_left = rt->flv_size - rt->flv_off;
  1058. if (data_left >= size) {
  1059. memcpy(buf, rt->flv_data + rt->flv_off, size);
  1060. rt->flv_off += size;
  1061. return orig_size;
  1062. }
  1063. if (data_left > 0) {
  1064. memcpy(buf, rt->flv_data + rt->flv_off, data_left);
  1065. buf += data_left;
  1066. size -= data_left;
  1067. rt->flv_off = rt->flv_size;
  1068. return data_left;
  1069. }
  1070. if ((ret = get_packet(s, 0)) < 0)
  1071. return ret;
  1072. }
  1073. return orig_size;
  1074. }
  1075. static int rtmp_write(URLContext *s, const uint8_t *buf, int size)
  1076. {
  1077. RTMPContext *rt = s->priv_data;
  1078. int size_temp = size;
  1079. int pktsize, pkttype;
  1080. uint32_t ts;
  1081. const uint8_t *buf_temp = buf;
  1082. int ret;
  1083. do {
  1084. if (rt->skip_bytes) {
  1085. int skip = FFMIN(rt->skip_bytes, size_temp);
  1086. buf_temp += skip;
  1087. size_temp -= skip;
  1088. rt->skip_bytes -= skip;
  1089. continue;
  1090. }
  1091. if (rt->flv_header_bytes < 11) {
  1092. const uint8_t *header = rt->flv_header;
  1093. int copy = FFMIN(11 - rt->flv_header_bytes, size_temp);
  1094. bytestream_get_buffer(&buf_temp, rt->flv_header + rt->flv_header_bytes, copy);
  1095. rt->flv_header_bytes += copy;
  1096. size_temp -= copy;
  1097. if (rt->flv_header_bytes < 11)
  1098. break;
  1099. pkttype = bytestream_get_byte(&header);
  1100. pktsize = bytestream_get_be24(&header);
  1101. ts = bytestream_get_be24(&header);
  1102. ts |= bytestream_get_byte(&header) << 24;
  1103. bytestream_get_be24(&header);
  1104. rt->flv_size = pktsize;
  1105. //force 12bytes header
  1106. if (((pkttype == RTMP_PT_VIDEO || pkttype == RTMP_PT_AUDIO) && ts == 0) ||
  1107. pkttype == RTMP_PT_NOTIFY) {
  1108. if (pkttype == RTMP_PT_NOTIFY)
  1109. pktsize += 16;
  1110. rt->prev_pkt[1][RTMP_SOURCE_CHANNEL].channel_id = 0;
  1111. }
  1112. //this can be a big packet, it's better to send it right here
  1113. if ((ret = ff_rtmp_packet_create(&rt->out_pkt, RTMP_SOURCE_CHANNEL,
  1114. pkttype, ts, pktsize)) < 0)
  1115. return ret;
  1116. rt->out_pkt.extra = rt->main_channel_id;
  1117. rt->flv_data = rt->out_pkt.data;
  1118. if (pkttype == RTMP_PT_NOTIFY)
  1119. ff_amf_write_string(&rt->flv_data, "@setDataFrame");
  1120. }
  1121. if (rt->flv_size - rt->flv_off > size_temp) {
  1122. bytestream_get_buffer(&buf_temp, rt->flv_data + rt->flv_off, size_temp);
  1123. rt->flv_off += size_temp;
  1124. size_temp = 0;
  1125. } else {
  1126. bytestream_get_buffer(&buf_temp, rt->flv_data + rt->flv_off, rt->flv_size - rt->flv_off);
  1127. size_temp -= rt->flv_size - rt->flv_off;
  1128. rt->flv_off += rt->flv_size - rt->flv_off;
  1129. }
  1130. if (rt->flv_off == rt->flv_size) {
  1131. rt->skip_bytes = 4;
  1132. if ((ret = ff_rtmp_packet_write(rt->stream, &rt->out_pkt,
  1133. rt->chunk_size, rt->prev_pkt[1])) < 0)
  1134. return ret;
  1135. ff_rtmp_packet_destroy(&rt->out_pkt);
  1136. rt->flv_size = 0;
  1137. rt->flv_off = 0;
  1138. rt->flv_header_bytes = 0;
  1139. }
  1140. } while (buf_temp - buf < size);
  1141. return size;
  1142. }
  1143. #define OFFSET(x) offsetof(RTMPContext, x)
  1144. #define DEC AV_OPT_FLAG_DECODING_PARAM
  1145. #define ENC AV_OPT_FLAG_ENCODING_PARAM
  1146. static const AVOption rtmp_options[] = {
  1147. {"rtmp_app", "Name of application to connect to on the RTMP server", OFFSET(app), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1148. {"rtmp_conn", "Append arbitrary AMF data to the Connect message", OFFSET(conn), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1149. {"rtmp_flashver", "Version of the Flash plugin used to run the SWF player.", OFFSET(flashver), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1150. {"rtmp_live", "Specify that the media is a live stream.", OFFSET(live), AV_OPT_TYPE_INT, {-2}, INT_MIN, INT_MAX, DEC, "rtmp_live"},
  1151. {"any", "both", 0, AV_OPT_TYPE_CONST, {-2}, 0, 0, DEC, "rtmp_live"},
  1152. {"live", "live stream", 0, AV_OPT_TYPE_CONST, {-1}, 0, 0, DEC, "rtmp_live"},
  1153. {"recorded", "recorded stream", 0, AV_OPT_TYPE_CONST, {0}, 0, 0, DEC, "rtmp_live"},
  1154. {"rtmp_playpath", "Stream identifier to play or to publish", OFFSET(playpath), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1155. {"rtmp_swfurl", "URL of the SWF player. By default no value will be sent", OFFSET(swfurl), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1156. {"rtmp_tcurl", "URL of the target stream. Defaults to rtmp://host[:port]/app.", OFFSET(tcurl), AV_OPT_TYPE_STRING, {.str = NULL }, 0, 0, DEC|ENC},
  1157. { NULL },
  1158. };
  1159. static const AVClass rtmp_class = {
  1160. .class_name = "rtmp",
  1161. .item_name = av_default_item_name,
  1162. .option = rtmp_options,
  1163. .version = LIBAVUTIL_VERSION_INT,
  1164. };
  1165. URLProtocol ff_rtmp_protocol = {
  1166. .name = "rtmp",
  1167. .url_open = rtmp_open,
  1168. .url_read = rtmp_read,
  1169. .url_write = rtmp_write,
  1170. .url_close = rtmp_close,
  1171. .priv_data_size = sizeof(RTMPContext),
  1172. .flags = URL_PROTOCOL_FLAG_NETWORK,
  1173. .priv_data_class= &rtmp_class,
  1174. };