- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- kek
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous
- Anonymous

1. /*
2. * Copyright (C) 2005-2008 Team XBMC
3. * http://www.xbmc.org
4. * Copyright (C) 2008-2009 Andrej Stepanchuk
5. * Copyright (C) 2009-2010 Howard Chu
6. *
7. * This file is part of librtmp.
8. *
9. * librtmp is free software; you can redistribute it and/or modify
10. * it under the terms of the GNU Lesser General Public License as
11. * published by the Free Software Foundation; either version 2.1,
12. * or (at your option) any later version.
13. *
14. * librtmp is distributed in the hope that it will be useful,
15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17. * GNU General Public License for more details.
18. *
19. * You should have received a copy of the GNU Lesser General Public License
20. * along with librtmp see the file COPYING. If not, write to
21. * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
22. * Boston, MA 02110-1301, USA.
23. * http://www.gnu.org/copyleft/lgpl.html
24. */
26. #include <stdint.h>
27. #include <stdlib.h>
28. #include <string.h>
29. #include <assert.h>
30. #include <stdio.h>
32. #include "rtmp_sys.h"
33. #include "log.h"
35. #ifdef CRYPTO
36. #ifdef USE_POLARSSL
37. #include <polarssl/havege.h>
38. #elif defined(USE_GNUTLS)
39. #include <gnutls/gnutls.h>
40. #else /* USE_OPENSSL */
41. #include <openssl/ssl.h>
42. #include <openssl/rc4.h>
43. #endif
44. TLS_CTX RTMP_TLS_ctx;
45. #endif
47. #define RTMP_SIG_SIZE 1536
48. #define RTMP_LARGE_HEADER_SIZE 12
50. static const int packetSize[] = { 12, 8, 4, 1 };
52. int RTMP_ctrlC;
54. const char RTMPProtocolStrings[][7] = {
55. "RTMP",
56. "RTMPT",
57. "RTMPE",
58. "RTMPTE",
59. "RTMPS",
60. "RTMPTS",
61. "",
62. "",
63. "RTMFP"
64. };
66. const char RTMPProtocolStringsLower[][7] = {
67. "rtmp",
68. "rtmpt",
69. "rtmpe",
70. "rtmpte",
71. "rtmps",
72. "rtmpts",
73. "",
74. "",
75. "rtmfp"
76. };
78. static const char *RTMPT_cmds[] = {
79. "open",
80. "send",
81. "idle",
82. "close"
83. };
85. typedef enum {
86. RTMPT_OPEN=0, RTMPT_SEND, RTMPT_IDLE, RTMPT_CLOSE
87. } RTMPTCmd;
89. static int DumpMetaData(AMFObject *obj);
90. static int HandShake(RTMP *r, int FP9HandShake);
91. static int SocksNegotiate(RTMP *r);
93. static int SendConnectPacket(RTMP *r, RTMPPacket *cp);
94. static int SendCheckBW(RTMP *r);
95. static int SendCheckBWResult(RTMP *r, double txn);
96. static int SendDeleteStream(RTMP *r, double dStreamId);
97. static int SendFCSubscribe(RTMP *r, AVal *subscribepath);
98. static int SendPlay(RTMP *r);
99. static int SendBytesReceived(RTMP *r);
100. static int SendUsherToken(RTMP *r, AVal *usherToken);
102. #if 0 /* unused */
103. static int SendBGHasStream(RTMP *r, double dId, AVal *playpath);
104. #endif
106. static int HandleInvoke(RTMP *r, const char *body, unsigned int nBodySize);
107. static int HandleMetadata(RTMP *r, char *body, unsigned int len);
108. static void HandleChangeChunkSize(RTMP *r, const RTMPPacket *packet);
109. static void HandleAudio(RTMP *r, const RTMPPacket *packet);
110. static void HandleVideo(RTMP *r, const RTMPPacket *packet);
111. static void HandleCtrl(RTMP *r, const RTMPPacket *packet);
112. static void HandleServerBW(RTMP *r, const RTMPPacket *packet);
113. static void HandleClientBW(RTMP *r, const RTMPPacket *packet);
115. static int ReadN(RTMP *r, char *buffer, int n);
116. static int WriteN(RTMP *r, const char *buffer, int n);
118. static void DecodeTEA(AVal *key, AVal *text);
120. static int HTTP_Post(RTMP *r, RTMPTCmd cmd, const char *buf, int len);
121. static int HTTP_read(RTMP *r, int fill);
123. #ifndef _WIN32
124. static int clk_tck;
125. #endif
127. #ifdef CRYPTO
128. #include "handshake.h"
129. #endif
131. uint32_t
132. RTMP_GetTime()
133. {
134. #ifdef _DEBUG
135. return 0;
136. #elif defined(_WIN32)
137. return timeGetTime();
138. #else
139. struct tms t;
140. if (!clk_tck) clk_tck = sysconf(_SC_CLK_TCK);
141. return times(&t) * 1000 / clk_tck;
142. #endif
143. }
145. void
146. RTMP_UserInterrupt()
147. {
148. RTMP_ctrlC = TRUE;
149. }
151. void
152. RTMPPacket_Reset(RTMPPacket *p)
153. {
154. p->m_headerType = 0;
155. p->m_packetType = 0;
156. p->m_nChannel = 0;
157. p->m_nTimeStamp = 0;
158. p->m_nInfoField2 = 0;
159. p->m_hasAbsTimestamp = FALSE;
160. p->m_nBodySize = 0;
161. p->m_nBytesRead = 0;
162. }
164. int
165. RTMPPacket_Alloc(RTMPPacket *p, int nSize)
166. {
167. char *ptr = calloc(1, nSize + RTMP_MAX_HEADER_SIZE);
168. if (!ptr)
169. return FALSE;
170. p->m_body = ptr + RTMP_MAX_HEADER_SIZE;
171. p->m_nBytesRead = 0;
172. return TRUE;
173. }
175. void
176. RTMPPacket_Free(RTMPPacket *p)
177. {
178. if (p->m_body)
179. {
180. free(p->m_body - RTMP_MAX_HEADER_SIZE);
181. p->m_body = NULL;
182. }
183. }
185. void
186. RTMPPacket_Dump(RTMPPacket *p)
187. {
188. RTMP_Log(RTMP_LOGDEBUG,
189. "RTMP PACKET: packet type: 0x%02x. channel: 0x%02x. info 1: %d info 2: %d. Body size: %u. body: 0x%02x",
190. p->m_packetType, p->m_nChannel, p->m_nTimeStamp, p->m_nInfoField2,
191. p->m_nBodySize, p->m_body ? (unsigned char)p->m_body[0] : 0);
192. }
194. int
195. RTMP_LibVersion()
196. {
197. return RTMP_LIB_VERSION;
198. }
200. void
201. RTMP_TLS_Init()
202. {
203. #ifdef CRYPTO
204. #ifdef USE_POLARSSL
205. /* Do this regardless of NO_SSL, we use havege for rtmpe too */
206. RTMP_TLS_ctx = calloc(1,sizeof(struct tls_ctx));
207. havege_init(&RTMP_TLS_ctx->hs);
208. #elif defined(USE_GNUTLS) && !defined(NO_SSL)
209. /* Technically we need to initialize libgcrypt ourselves if
210. * we're not going to call gnutls_global_init(). Ignoring this
211. * for now.
212. */
213. gnutls_global_init();
214. RTMP_TLS_ctx = malloc(sizeof(struct tls_ctx));
215. gnutls_certificate_allocate_credentials(&RTMP_TLS_ctx->cred);
216. gnutls_priority_init(&RTMP_TLS_ctx->prios, "NORMAL", NULL);
217. gnutls_certificate_set_x509_trust_file(RTMP_TLS_ctx->cred,
218. "ca.pem", GNUTLS_X509_FMT_PEM);
219. #elif !defined(NO_SSL) /* USE_OPENSSL */
220. /* libcrypto doesn't need anything special */
221. SSL_load_error_strings();
222. SSL_library_init();
223. OpenSSL_add_all_digests();
224. RTMP_TLS_ctx = SSL_CTX_new(SSLv23_method());
225. SSL_CTX_set_options(RTMP_TLS_ctx, SSL_OP_ALL);
226. SSL_CTX_set_default_verify_paths(RTMP_TLS_ctx);
227. #endif
228. #endif
229. }
231. RTMP *
232. RTMP_Alloc()
233. {
234. return calloc(1, sizeof(RTMP));
235. }
237. void
238. RTMP_Free(RTMP *r)
239. {
240. free(r);
241. }
243. void
244. RTMP_Init(RTMP *r)
245. {
246. #ifdef CRYPTO
247. if (!RTMP_TLS_ctx)
248. RTMP_TLS_Init();
249. #endif
251. memset(r, 0, sizeof(RTMP));
252. r->m_sb.sb_socket = -1;
253. r->m_inChunkSize = RTMP_DEFAULT_CHUNKSIZE;
254. r->m_outChunkSize = RTMP_DEFAULT_CHUNKSIZE;
255. r->m_nBufferMS = 30000;
256. r->m_nClientBW = 2500000;
257. r->m_nClientBW2 = 2;
258. r->m_nServerBW = 2500000;
259. r->m_fAudioCodecs = 3191.0;
260. r->m_fVideoCodecs = 252.0;
261. r->Link.timeout = 30;
262. r->Link.swfAge = 30;
263. }
265. void
266. RTMP_EnableWrite(RTMP *r)
267. {
268. r->Link.protocol |= RTMP_FEATURE_WRITE;
269. }
271. double
272. RTMP_GetDuration(RTMP *r)
273. {
274. return r->m_fDuration;
275. }
277. int
278. RTMP_IsConnected(RTMP *r)
279. {
280. return r->m_sb.sb_socket != -1;
281. }
283. int
284. RTMP_Socket(RTMP *r)
285. {
286. return r->m_sb.sb_socket;
287. }
289. int
290. RTMP_IsTimedout(RTMP *r)
291. {
292. return r->m_sb.sb_timedout;
293. }
295. void
296. RTMP_SetBufferMS(RTMP *r, int size)
297. {
298. r->m_nBufferMS = size;
299. }
301. void
302. RTMP_UpdateBufferMS(RTMP *r)
303. {
304. RTMP_SendCtrl(r, 3, r->m_stream_id, r->m_nBufferMS);
305. }
307. #undef OSS
308. #ifdef _WIN32
309. #define OSS "WIN"
310. #elif defined(__sun__)
311. #define OSS "SOL"
312. #elif defined(__APPLE__)
313. #define OSS "MAC"
314. #elif defined(__linux__)
315. #define OSS "LNX"
316. #else
317. #define OSS "GNU"
318. #endif
319. #define DEF_VERSTR OSS " 10,0,32,18"
320. static const char DEFAULT_FLASH_VER[] = DEF_VERSTR;
321. const AVal RTMP_DefaultFlashVer =
322. { (char *)DEFAULT_FLASH_VER, sizeof(DEFAULT_FLASH_VER) - 1 };
324. void
325. RTMP_SetupStream(RTMP *r,
326. int protocol,
327. AVal *host,
328. unsigned int port,
329. AVal *sockshost,
330. AVal *playpath,
331. AVal *tcUrl,
332. AVal *swfUrl,
333. AVal *pageUrl,
334. AVal *app,
335. AVal *auth,
336. AVal *swfSHA256Hash,
337. uint32_t swfSize,
338. AVal *flashVer,
339. AVal *subscribepath,
340. AVal *usherToken,
341. int dStart,
342. int dStop, int bLiveStream, long int timeout)
343. {
344. RTMP_Log(RTMP_LOGDEBUG, "Protocol : %s", RTMPProtocolStrings[protocol&7]);
345. RTMP_Log(RTMP_LOGDEBUG, "Hostname : %.*s", host->av_len, host->av_val);
346. RTMP_Log(RTMP_LOGDEBUG, "Port : %d", port);
347. RTMP_Log(RTMP_LOGDEBUG, "Playpath : %s", playpath->av_val);
349. if (tcUrl && tcUrl->av_val)
350. RTMP_Log(RTMP_LOGDEBUG, "tcUrl : %s", tcUrl->av_val);
351. if (swfUrl && swfUrl->av_val)
352. RTMP_Log(RTMP_LOGDEBUG, "swfUrl : %s", swfUrl->av_val);
353. if (pageUrl && pageUrl->av_val)
354. RTMP_Log(RTMP_LOGDEBUG, "pageUrl : %s", pageUrl->av_val);
355. if (app && app->av_val)
356. RTMP_Log(RTMP_LOGDEBUG, "app : %.*s", app->av_len, app->av_val);
357. if (auth && auth->av_val)
358. RTMP_Log(RTMP_LOGDEBUG, "auth : %s", auth->av_val);
359. if (subscribepath && subscribepath->av_val)
360. RTMP_Log(RTMP_LOGDEBUG, "subscribepath : %s", subscribepath->av_val);
361. if (usherToken && usherToken->av_val)
362. RTMP_Log(RTMP_LOGDEBUG, "NetStream.Authenticate.UsherToken : %s", usherToken->av_val);
363. if (flashVer && flashVer->av_val)
364. RTMP_Log(RTMP_LOGDEBUG, "flashVer : %s", flashVer->av_val);
365. if (dStart > 0)
366. RTMP_Log(RTMP_LOGDEBUG, "StartTime : %d msec", dStart);
367. if (dStop > 0)
368. RTMP_Log(RTMP_LOGDEBUG, "StopTime : %d msec", dStop);
370. RTMP_Log(RTMP_LOGDEBUG, "live : %s", bLiveStream ? "yes" : "no");
371. RTMP_Log(RTMP_LOGDEBUG, "timeout : %ld sec", timeout);
373. #ifdef CRYPTO
374. if (swfSHA256Hash != NULL && swfSize > 0)
375. {
376. memcpy(r->Link.SWFHash, swfSHA256Hash->av_val, sizeof(r->Link.SWFHash));
377. r->Link.SWFSize = swfSize;
378. RTMP_Log(RTMP_LOGDEBUG, "SWFSHA256:");
379. RTMP_LogHex(RTMP_LOGDEBUG, r->Link.SWFHash, sizeof(r->Link.SWFHash));
380. RTMP_Log(RTMP_LOGDEBUG, "SWFSize : %u", r->Link.SWFSize);
381. }
382. else
383. {
384. r->Link.SWFSize = 0;
385. }
386. #endif
388. if (sockshost->av_len)
389. {
390. const char *socksport = strchr(sockshost->av_val, ':');
391. char *hostname = strdup(sockshost->av_val);
393. if (socksport)
394. hostname[socksport - sockshost->av_val] = '\0';
395. r->Link.sockshost.av_val = hostname;
396. r->Link.sockshost.av_len = strlen(hostname);
398. r->Link.socksport = socksport ? atoi(socksport + 1) : 1080;
399. RTMP_Log(RTMP_LOGDEBUG, "Connecting via SOCKS proxy: %s:%d", r->Link.sockshost.av_val,
400. r->Link.socksport);
401. }
402. else
403. {
404. r->Link.sockshost.av_val = NULL;
405. r->Link.sockshost.av_len = 0;
406. r->Link.socksport = 0;
407. }
409. if (tcUrl && tcUrl->av_len)
410. r->Link.tcUrl = *tcUrl;
411. if (swfUrl && swfUrl->av_len)
412. r->Link.swfUrl = *swfUrl;
413. if (pageUrl && pageUrl->av_len)
414. r->Link.pageUrl = *pageUrl;
415. if (app && app->av_len)
416. r->Link.app = *app;
417. if (auth && auth->av_len)
418. {
419. r->Link.auth = *auth;
420. r->Link.lFlags |= RTMP_LF_AUTH;
421. }
422. if (flashVer && flashVer->av_len)
423. r->Link.flashVer = *flashVer;
424. else
425. r->Link.flashVer = RTMP_DefaultFlashVer;
426. if (subscribepath && subscribepath->av_len)
427. r->Link.subscribepath = *subscribepath;
428. if (usherToken && usherToken->av_len)
429. r->Link.usherToken = *usherToken;
430. r->Link.seekTime = dStart;
431. r->Link.stopTime = dStop;
432. if (bLiveStream)
433. r->Link.lFlags |= RTMP_LF_LIVE;
434. r->Link.timeout = timeout;
436. r->Link.protocol = protocol;
437. r->Link.hostname = *host;
438. r->Link.port = port;
439. r->Link.playpath = *playpath;
441. if (r->Link.port == 0)
442. {
443. if (protocol & RTMP_FEATURE_SSL)
444. r->Link.port = 443;
445. else if (protocol & RTMP_FEATURE_HTTP)
446. r->Link.port = 80;
447. else
448. r->Link.port = 1935;
449. }
450. }
452. enum { OPT_STR=0, OPT_INT, OPT_BOOL, OPT_CONN };
453. static const char *optinfo[] = {
454. "string", "integer", "boolean", "AMF" };
456. #define OFF(x) offsetof(struct RTMP,x)
458. static struct urlopt {
459. AVal name;
460. off_t off;
461. int otype;
462. int omisc;
463. char *use;
464. } options[] = {
465. { AVC("socks"), OFF(Link.sockshost), OPT_STR, 0,
466. "Use the specified SOCKS proxy" },
467. { AVC("app"), OFF(Link.app), OPT_STR, 0,
468. "Name of target app on server" },
469. { AVC("tcUrl"), OFF(Link.tcUrl), OPT_STR, 0,
470. "URL to played stream" },
471. { AVC("pageUrl"), OFF(Link.pageUrl), OPT_STR, 0,
472. "URL of played media's web page" },
473. { AVC("swfUrl"), OFF(Link.swfUrl), OPT_STR, 0,
474. "URL to player SWF file" },
475. { AVC("flashver"), OFF(Link.flashVer), OPT_STR, 0,
476. "Flash version string (default " DEF_VERSTR ")" },
477. { AVC("conn"), OFF(Link.extras), OPT_CONN, 0,
478. "Append arbitrary AMF data to Connect message" },
479. { AVC("playpath"), OFF(Link.playpath), OPT_STR, 0,
480. "Path to target media on server" },
481. { AVC("playlist"), OFF(Link.lFlags), OPT_BOOL, RTMP_LF_PLST,
482. "Set playlist before play command" },
483. { AVC("live"), OFF(Link.lFlags), OPT_BOOL, RTMP_LF_LIVE,
484. "Stream is live, no seeking possible" },
485. { AVC("subscribe"), OFF(Link.subscribepath), OPT_STR, 0,
486. "Stream to subscribe to" },
487. { AVC("jtv"), OFF(Link.usherToken), OPT_STR, 0,
488. "Justin.tv authentication token" },
489. { AVC("token"), OFF(Link.token), OPT_STR, 0,
490. "Key for SecureToken response" },
491. { AVC("swfVfy"), OFF(Link.lFlags), OPT_BOOL, RTMP_LF_SWFV,
492. "Perform SWF Verification" },
493. { AVC("swfAge"), OFF(Link.swfAge), OPT_INT, 0,
494. "Number of days to use cached SWF hash" },
495. { AVC("start"), OFF(Link.seekTime), OPT_INT, 0,
496. "Stream start position in milliseconds" },
497. { AVC("stop"), OFF(Link.stopTime), OPT_INT, 0,
498. "Stream stop position in milliseconds" },
499. { AVC("buffer"), OFF(m_nBufferMS), OPT_INT, 0,
500. "Buffer time in milliseconds" },
501. { AVC("timeout"), OFF(Link.timeout), OPT_INT, 0,
502. "Session timeout in seconds" },
503. { {NULL,0}, 0, 0}
504. };
506. static const AVal truth[] = {
507. AVC("1"),
508. AVC("on"),
509. AVC("yes"),
510. AVC("true"),
511. {0,0}
512. };
514. static void RTMP_OptUsage()
515. {
516. int i;
518. RTMP_Log(RTMP_LOGERROR, "Valid RTMP options are:\n");
519. for (i=0; options[i].name.av_len; i++) {
520. RTMP_Log(RTMP_LOGERROR, "%10s %-7s %s\n", options[i].name.av_val,
521. optinfo[options[i].otype], options[i].use);
522. }
523. }
525. static int
526. parseAMF(AMFObject *obj, AVal *av, int *depth)
527. {
528. AMFObjectProperty prop = {{0,0}};
529. int i;
530. char *p, *arg = av->av_val;
532. if (arg[1] == ':')
533. {
534. p = (char *)arg+2;
535. switch(arg[0])
536. {
537. case 'B':
538. prop.p_type = AMF_BOOLEAN;
539. prop.p_vu.p_number = atoi(p);
540. break;
541. case 'S':
542. prop.p_type = AMF_STRING;
543. prop.p_vu.p_aval.av_val = p;
544. prop.p_vu.p_aval.av_len = av->av_len - (p-arg);
545. break;
546. case 'N':
547. prop.p_type = AMF_NUMBER;
548. prop.p_vu.p_number = strtod(p, NULL);
549. break;
550. case 'Z':
551. prop.p_type = AMF_NULL;
552. break;
553. case 'O':
554. i = atoi(p);
555. if (i)
556. {
557. prop.p_type = AMF_OBJECT;
558. }
559. else
560. {
561. (*depth)--;
562. return 0;
563. }
564. break;
565. default:
566. return -1;
567. }
568. }
569. else if (arg[2] == ':' && arg[0] == 'N')
570. {
571. p = strchr(arg+3, ':');
572. if (!p || !*depth)
573. return -1;
574. prop.p_name.av_val = (char *)arg+3;
575. prop.p_name.av_len = p - (arg+3);
577. p++;
578. switch(arg[1])
579. {
580. case 'B':
581. prop.p_type = AMF_BOOLEAN;
582. prop.p_vu.p_number = atoi(p);
583. break;
584. case 'S':
585. prop.p_type = AMF_STRING;
586. prop.p_vu.p_aval.av_val = p;
587. prop.p_vu.p_aval.av_len = av->av_len - (p-arg);
588. break;
589. case 'N':
590. prop.p_type = AMF_NUMBER;
591. prop.p_vu.p_number = strtod(p, NULL);
592. break;
593. case 'O':
594. prop.p_type = AMF_OBJECT;
595. break;
596. default:
597. return -1;
598. }
599. }
600. else
601. return -1;
603. if (*depth)
604. {
605. AMFObject *o2;
606. for (i=0; i<*depth; i++)
607. {
608. o2 = &obj->o_props[obj->o_num-1].p_vu.p_object;
609. obj = o2;
610. }
611. }
612. AMF_AddProp(obj, &prop);
613. if (prop.p_type == AMF_OBJECT)
614. (*depth)++;
615. return 0;
616. }
618. int RTMP_SetOpt(RTMP *r, const AVal *opt, AVal *arg)
619. {
620. int i;
621. void *v;
623. for (i=0; options[i].name.av_len; i++) {
624. if (opt->av_len != options[i].name.av_len) continue;
625. if (strcasecmp(opt->av_val, options[i].name.av_val)) continue;
626. v = (char *)r + options[i].off;
627. switch(options[i].otype) {
628. case OPT_STR: {
629. AVal *aptr = v;
630. *aptr = *arg; }
631. break;
632. case OPT_INT: {
633. long l = strtol(arg->av_val, NULL, 0);
634. *(int *)v = l; }
635. break;
636. case OPT_BOOL: {
637. int j, fl;
638. fl = *(int *)v;
639. for (j=0; truth[j].av_len; j++) {
640. if (arg->av_len != truth[j].av_len) continue;
641. if (strcasecmp(arg->av_val, truth[j].av_val)) continue;
642. fl |= options[i].omisc; break; }
643. *(int *)v = fl;
644. }
645. break;
646. case OPT_CONN:
647. if (parseAMF(&r->Link.extras, arg, &r->Link.edepth))
648. return FALSE;
649. break;
650. }
651. break;
652. }
653. if (!options[i].name.av_len) {
654. RTMP_Log(RTMP_LOGERROR, "Unknown option %s", opt->av_val);
655. RTMP_OptUsage();
656. return FALSE;
657. }
659. return TRUE;
660. }
662. int RTMP_SetupURL(RTMP *r, char *url)
663. {
664. AVal opt, arg;
665. char *p1, *p2, *ptr = strchr(url, ' ');
666. int ret, len;
667. unsigned int port = 0;
669. if (ptr)
670. *ptr = '\0';
672. len = strlen(url);
673. ret = RTMP_ParseURL(url, &r->Link.protocol, &r->Link.hostname,
674. &port, &r->Link.playpath0, &r->Link.app);
675. if (!ret)
676. return ret;
677. r->Link.port = port;
678. r->Link.playpath = r->Link.playpath0;
680. while (ptr) {
681. *ptr++ = '\0';
682. p1 = ptr;
683. p2 = strchr(p1, '=');
684. if (!p2)
685. break;
686. opt.av_val = p1;
687. opt.av_len = p2 - p1;
688. *p2++ = '\0';
689. arg.av_val = p2;
690. ptr = strchr(p2, ' ');
691. if (ptr) {
692. *ptr = '\0';
693. arg.av_len = ptr - p2;
694. /* skip repeated spaces */
695. while(ptr[1] == ' ')
696. *ptr++ = '\0';
697. } else {
698. arg.av_len = strlen(p2);
699. }
701. /* unescape */
702. port = arg.av_len;
703. for (p1=p2; port >0;) {
704. if (*p1 == '\\') {
705. unsigned int c;
706. if (port < 3)
707. return FALSE;
708. sscanf(p1+1, "%02x", &c);
709. *p2++ = c;
710. port -= 3;
711. p1 += 3;
712. } else {
713. *p2++ = *p1++;
714. port--;
715. }
716. }
717. arg.av_len = p2 - arg.av_val;
719. ret = RTMP_SetOpt(r, &opt, &arg);
720. if (!ret)
721. return ret;
722. }
724. if (!r->Link.tcUrl.av_len)
725. {
726. r->Link.tcUrl.av_val = url;
727. if (r->Link.app.av_len)
728. {
729. if (r->Link.app.av_val < url + len)
730. {
731. /* if app is part of original url, just use it */
732. r->Link.tcUrl.av_len = r->Link.app.av_len + (r->Link.app.av_val - url);
733. }
734. else
735. {
736. len = r->Link.hostname.av_len + r->Link.app.av_len +
737. sizeof("rtmpte://:65535/");
738. r->Link.tcUrl.av_val = malloc(len);
739. r->Link.tcUrl.av_len = snprintf(r->Link.tcUrl.av_val, len,
740. "%s://%.*s:%d/%.*s",
741. RTMPProtocolStringsLower[r->Link.protocol],
742. r->Link.hostname.av_len, r->Link.hostname.av_val,
743. r->Link.port,
744. r->Link.app.av_len, r->Link.app.av_val);
745. r->Link.lFlags |= RTMP_LF_FTCU;
746. }
747. }
748. else
749. {
750. r->Link.tcUrl.av_len = strlen(url);
751. }
752. }
754. #ifdef CRYPTO
755. if ((r->Link.lFlags & RTMP_LF_SWFV) && r->Link.swfUrl.av_len)
756. RTMP_HashSWF(r->Link.swfUrl.av_val, &r->Link.SWFSize,
757. (unsigned char *)r->Link.SWFHash, r->Link.swfAge);
758. #endif
760. if (r->Link.port == 0)
761. {
762. if (r->Link.protocol & RTMP_FEATURE_SSL)
763. r->Link.port = 443;
764. else if (r->Link.protocol & RTMP_FEATURE_HTTP)
765. r->Link.port = 80;
766. else
767. r->Link.port = 1935;
768. }
769. return TRUE;
770. }
772. static int
773. add_addr_info(struct sockaddr_in *service, AVal *host, int port)
774. {
775. char *hostname;
776. int ret = TRUE;
777. if (host->av_val[host->av_len])
778. {
779. hostname = malloc(host->av_len+1);
780. memcpy(hostname, host->av_val, host->av_len);
781. hostname[host->av_len] = '\0';
782. }
783. else
784. {
785. hostname = host->av_val;
786. }
788. service->sin_addr.s_addr = inet_addr(hostname);
789. if (service->sin_addr.s_addr == INADDR_NONE)
790. {
791. struct hostent *host = gethostbyname(hostname);
792. if (host == NULL || host->h_addr == NULL)
793. {
794. RTMP_Log(RTMP_LOGERROR, "Problem accessing the DNS. (addr: %s)", hostname);
795. ret = FALSE;
796. goto finish;
797. }
798. service->sin_addr = *(struct in_addr *)host->h_addr;
799. }
801. service->sin_port = htons(port);
802. finish:
803. if (hostname != host->av_val)
804. free(hostname);
805. return ret;
806. }
808. int
809. RTMP_Connect0(RTMP *r, struct sockaddr * service)
810. {
811. int on = 1;
812. r->m_sb.sb_timedout = FALSE;
813. r->m_pausing = 0;
814. r->m_fDuration = 0.0;
816. r->m_sb.sb_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
817. if (r->m_sb.sb_socket != -1)
818. {
819. if (connect(r->m_sb.sb_socket, service, sizeof(struct sockaddr)) < 0)
820. {
821. int err = GetSockError();
822. RTMP_Log(RTMP_LOGERROR, "%s, failed to connect socket. %d (%s)",
823. __FUNCTION__, err, strerror(err));
824. RTMP_Close(r);
825. return FALSE;
826. }
828. if (r->Link.socksport)
829. {
830. RTMP_Log(RTMP_LOGDEBUG, "%s ... SOCKS negotiation", __FUNCTION__);
831. if (!SocksNegotiate(r))
832. {
833. RTMP_Log(RTMP_LOGERROR, "%s, SOCKS negotiation failed.", __FUNCTION__);
834. RTMP_Close(r);
835. return FALSE;
836. }
837. }
838. }
839. else
840. {
841. RTMP_Log(RTMP_LOGERROR, "%s, failed to create socket. Error: %d", __FUNCTION__,
842. GetSockError());
843. return FALSE;
844. }
846. /* set timeout */
847. {
848. SET_RCVTIMEO(tv, r->Link.timeout);
849. if (setsockopt
850. (r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(tv)))
851. {
852. RTMP_Log(RTMP_LOGERROR, "%s, Setting socket timeout to %ds failed!",
853. __FUNCTION__, r->Link.timeout);
854. }
855. }
857. setsockopt(r->m_sb.sb_socket, IPPROTO_TCP, TCP_NODELAY, (char *) &on, sizeof(on));
859. return TRUE;
860. }
862. int
863. RTMP_Connect1(RTMP *r, RTMPPacket *cp)
864. {
865. if (r->Link.protocol & RTMP_FEATURE_SSL)
866. {
867. #if defined(CRYPTO) && !defined(NO_SSL)
868. TLS_client(RTMP_TLS_ctx, r->m_sb.sb_ssl);
869. TLS_setfd(r->m_sb.sb_ssl, r->m_sb.sb_socket);
870. if (TLS_connect(r->m_sb.sb_ssl) < 0)
871. {
872. RTMP_Log(RTMP_LOGERROR, "%s, TLS_Connect failed", __FUNCTION__);
873. RTMP_Close(r);
874. return FALSE;
875. }
876. #else
877. RTMP_Log(RTMP_LOGERROR, "%s, no SSL/TLS support", __FUNCTION__);
878. RTMP_Close(r);
879. return FALSE;
881. #endif
882. }
883. if (r->Link.protocol & RTMP_FEATURE_HTTP)
884. {
885. r->m_msgCounter = 1;
886. r->m_clientID.av_val = NULL;
887. r->m_clientID.av_len = 0;
888. HTTP_Post(r, RTMPT_OPEN, "", 1);
889. if (HTTP_read(r, 1) != 0)
890. {
891. r->m_msgCounter = 0;
892. RTMP_Log(RTMP_LOGDEBUG, "%s, Could not connect for handshake", __FUNCTION__);
893. RTMP_Close(r);
894. return 0;
895. }
896. r->m_msgCounter = 0;
897. }
898. RTMP_Log(RTMP_LOGDEBUG, "%s, ... connected, handshaking", __FUNCTION__);
899. if (!HandShake(r, TRUE))
900. {
901. RTMP_Log(RTMP_LOGERROR, "%s, handshake failed.", __FUNCTION__);
902. RTMP_Close(r);
903. return FALSE;
904. }
905. RTMP_Log(RTMP_LOGDEBUG, "%s, handshaked", __FUNCTION__);
907. if (!SendConnectPacket(r, cp))
908. {
909. RTMP_Log(RTMP_LOGERROR, "%s, RTMP connect failed.", __FUNCTION__);
910. RTMP_Close(r);
911. return FALSE;
912. }
913. return TRUE;
914. }
916. int
917. RTMP_Connect(RTMP *r, RTMPPacket *cp)
918. {
919. struct sockaddr_in service;
920. if (!r->Link.hostname.av_len)
921. return FALSE;
923. memset(&service, 0, sizeof(struct sockaddr_in));
924. service.sin_family = AF_INET;
926. if (r->Link.socksport)
927. {
928. /* Connect via SOCKS */
929. if (!add_addr_info(&service, &r->Link.sockshost, r->Link.socksport))
930. return FALSE;
931. }
932. else
933. {
934. /* Connect directly */
935. if (!add_addr_info(&service, &r->Link.hostname, r->Link.port))
936. return FALSE;
937. }
939. if (!RTMP_Connect0(r, (struct sockaddr *)&service))
940. return FALSE;
942. r->m_bSendCounter = TRUE;
944. return RTMP_Connect1(r, cp);
945. }
947. static int
948. SocksNegotiate(RTMP *r)
949. {
950. unsigned long addr;
951. struct sockaddr_in service;
952. memset(&service, 0, sizeof(struct sockaddr_in));
954. add_addr_info(&service, &r->Link.hostname, r->Link.port);
955. addr = htonl(service.sin_addr.s_addr);
957. {
958. char packet[] = {
959. 4, 1, /* SOCKS 4, connect */
960. (r->Link.port >> 8) & 0xFF,
961. (r->Link.port) & 0xFF,
962. (char)(addr >> 24) & 0xFF, (char)(addr >> 16) & 0xFF,
963. (char)(addr >> 8) & 0xFF, (char)addr & 0xFF,
964. 0
965. }; /* NULL terminate */
967. WriteN(r, packet, sizeof packet);
969. if (ReadN(r, packet, 8) != 8)
970. return FALSE;
972. if (packet[0] == 0 && packet[1] == 90)
973. {
974. return TRUE;
975. }
976. else
977. {
978. RTMP_Log(RTMP_LOGERROR, "%s, SOCKS returned error code %d", __FUNCTION__, packet[1]);
979. return FALSE;
980. }
981. }
982. }
984. int
985. RTMP_ConnectStream(RTMP *r, int seekTime)
986. {
987. RTMPPacket packet = { 0 };
989. /* seekTime was already set by SetupStream / SetupURL.
990. * This is only needed by ReconnectStream.
991. */
992. if (seekTime > 0)
993. r->Link.seekTime = seekTime;
995. r->m_mediaChannel = 0;
997. while (!r->m_bPlaying && RTMP_IsConnected(r) && RTMP_ReadPacket(r, &packet))
998. {
999. if (RTMPPacket_IsReady(&packet))
1000. {
1001. if (!packet.m_nBodySize)
1002. continue;
1003. if ((packet.m_packetType == RTMP_PACKET_TYPE_AUDIO) ||
1004. (packet.m_packetType == RTMP_PACKET_TYPE_VIDEO) ||
1005. (packet.m_packetType == RTMP_PACKET_TYPE_INFO))
1006. {
1007. RTMP_Log(RTMP_LOGWARNING, "Received FLV packet before play()! Ignoring.");
1008. RTMPPacket_Free(&packet);
1009. continue;
1010. }
1012. RTMP_ClientPacket(r, &packet);
1013. RTMPPacket_Free(&packet);
1014. }
1015. }
1017. return r->m_bPlaying;
1018. }
1020. int
1021. RTMP_ReconnectStream(RTMP *r, int seekTime)
1022. {
1023. RTMP_DeleteStream(r);
1025. RTMP_SendCreateStream(r);
1027. return RTMP_ConnectStream(r, seekTime);
1028. }
1030. int
1031. RTMP_ToggleStream(RTMP *r)
1032. {
1033. int res;
1035. if (!r->m_pausing)
1036. {
1037. if (RTMP_IsTimedout(r) && r->m_read.status == RTMP_READ_EOF)
1038. r->m_read.status = 0;
1040. res = RTMP_SendPause(r, TRUE, r->m_pauseStamp);
1041. if (!res)
1042. return res;
1044. r->m_pausing = 1;
1045. sleep(1);
1046. }
1047. res = RTMP_SendPause(r, FALSE, r->m_pauseStamp);
1048. r->m_pausing = 3;
1049. return res;
1050. }
1052. void
1053. RTMP_DeleteStream(RTMP *r)
1054. {
1055. if (r->m_stream_id < 0)
1056. return;
1058. r->m_bPlaying = FALSE;
1060. SendDeleteStream(r, r->m_stream_id);
1061. r->m_stream_id = -1;
1062. }
1064. int
1065. RTMP_GetNextMediaPacket(RTMP *r, RTMPPacket *packet)
1066. {
1067. int bHasMediaPacket = 0;
1069. while (!bHasMediaPacket && RTMP_IsConnected(r)
1070. && RTMP_ReadPacket(r, packet))
1071. {
1072. if (!RTMPPacket_IsReady(packet))
1073. {
1074. continue;
1075. }
1077. bHasMediaPacket = RTMP_ClientPacket(r, packet);
1079. if (!bHasMediaPacket)
1080. {
1081. RTMPPacket_Free(packet);
1082. }
1083. else if (r->m_pausing == 3)
1084. {
1085. if (packet->m_nTimeStamp <= r->m_mediaStamp)
1086. {
1087. bHasMediaPacket = 0;
1088. #ifdef _DEBUG
1089. RTMP_Log(RTMP_LOGDEBUG,
1090. "Skipped type: %02X, size: %d, TS: %d ms, abs TS: %d, pause: %d ms",
1091. packet->m_packetType, packet->m_nBodySize,
1092. packet->m_nTimeStamp, packet->m_hasAbsTimestamp,
1093. r->m_mediaStamp);
1094. #endif
1095. continue;
1096. }
1097. r->m_pausing = 0;
1098. }
1099. }
1101. if (bHasMediaPacket)
1102. r->m_bPlaying = TRUE;
1103. else if (r->m_sb.sb_timedout && !r->m_pausing)
1104. r->m_pauseStamp = r->m_channelTimestamp[r->m_mediaChannel];
1106. return bHasMediaPacket;
1107. }
1109. int
1110. RTMP_ClientPacket(RTMP *r, RTMPPacket *packet)
1111. {
1112. int bHasMediaPacket = 0;
1113. switch (packet->m_packetType)
1114. {
1115. case RTMP_PACKET_TYPE_CHUNK_SIZE:
1116. /* chunk size */
1117. HandleChangeChunkSize(r, packet);
1118. break;
1120. case RTMP_PACKET_TYPE_BYTES_READ_REPORT:
1121. /* bytes read report */
1122. RTMP_Log(RTMP_LOGDEBUG, "%s, received: bytes read report", __FUNCTION__);
1123. break;
1125. case RTMP_PACKET_TYPE_CONTROL:
1126. /* ctrl */
1127. HandleCtrl(r, packet);
1128. break;
1130. case RTMP_PACKET_TYPE_SERVER_BW:
1131. /* server bw */
1132. HandleServerBW(r, packet);
1133. break;
1135. case RTMP_PACKET_TYPE_CLIENT_BW:
1136. /* client bw */
1137. HandleClientBW(r, packet);
1138. break;
1140. case RTMP_PACKET_TYPE_AUDIO:
1141. /* audio data */
1142. /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: audio %lu bytes", __FUNCTION__, packet.m_nBodySize); */
1143. HandleAudio(r, packet);
1144. bHasMediaPacket = 1;
1145. if (!r->m_mediaChannel)
1146. r->m_mediaChannel = packet->m_nChannel;
1147. if (!r->m_pausing)
1148. r->m_mediaStamp = packet->m_nTimeStamp;
1149. break;
1151. case RTMP_PACKET_TYPE_VIDEO:
1152. /* video data */
1153. /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: video %lu bytes", __FUNCTION__, packet.m_nBodySize); */
1154. HandleVideo(r, packet);
1155. bHasMediaPacket = 1;
1156. if (!r->m_mediaChannel)
1157. r->m_mediaChannel = packet->m_nChannel;
1158. if (!r->m_pausing)
1159. r->m_mediaStamp = packet->m_nTimeStamp;
1160. break;
1162. case RTMP_PACKET_TYPE_FLEX_STREAM_SEND:
1163. /* flex stream send */
1164. RTMP_Log(RTMP_LOGDEBUG,
1165. "%s, flex stream send, size %u bytes, not supported, ignoring",
1166. __FUNCTION__, packet->m_nBodySize);
1167. break;
1169. case RTMP_PACKET_TYPE_FLEX_SHARED_OBJECT:
1170. /* flex shared object */
1171. RTMP_Log(RTMP_LOGDEBUG,
1172. "%s, flex shared object, size %u bytes, not supported, ignoring",
1173. __FUNCTION__, packet->m_nBodySize);
1174. break;
1176. case RTMP_PACKET_TYPE_FLEX_MESSAGE:
1177. /* flex message */
1178. {
1179. RTMP_Log(RTMP_LOGDEBUG,
1180. "%s, flex message, size %u bytes, not fully supported",
1181. __FUNCTION__, packet->m_nBodySize);
1182. /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */
1184. /* some DEBUG code */
1185. #if 0
1186. RTMP_LIB_AMFObject obj;
1187. int nRes = obj.Decode(packet.m_body+1, packet.m_nBodySize-1);
1188. if(nRes < 0) {
1189. RTMP_Log(RTMP_LOGERROR, "%s, error decoding AMF3 packet", __FUNCTION__);
1190. /*return; */
1191. }
1193. obj.Dump();
1194. #endif
1196. if (HandleInvoke(r, packet->m_body + 1, packet->m_nBodySize - 1) == 1)
1197. bHasMediaPacket = 2;
1198. break;
1199. }
1200. case RTMP_PACKET_TYPE_INFO:
1201. /* metadata (notify) */
1202. RTMP_Log(RTMP_LOGDEBUG, "%s, received: notify %u bytes", __FUNCTION__,
1203. packet->m_nBodySize);
1204. if (HandleMetadata(r, packet->m_body, packet->m_nBodySize))
1205. bHasMediaPacket = 1;
1206. break;
1208. case RTMP_PACKET_TYPE_SHARED_OBJECT:
1209. RTMP_Log(RTMP_LOGDEBUG, "%s, shared object, not supported, ignoring",
1210. __FUNCTION__);
1211. break;
1213. case RTMP_PACKET_TYPE_INVOKE:
1214. /* invoke */
1215. RTMP_Log(RTMP_LOGDEBUG, "%s, received: invoke %u bytes", __FUNCTION__,
1216. packet->m_nBodySize);
1217. /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */
1219. if (HandleInvoke(r, packet->m_body, packet->m_nBodySize) == 1)
1220. bHasMediaPacket = 2;
1221. break;
1223. case RTMP_PACKET_TYPE_FLASH_VIDEO:
1224. {
1225. /* go through FLV packets and handle metadata packets */
1226. unsigned int pos = 0;
1227. uint32_t nTimeStamp = packet->m_nTimeStamp;
1229. while (pos + 11 < packet->m_nBodySize)
1230. {
1231. uint32_t dataSize = AMF_DecodeInt24(packet->m_body + pos + 1); /* size without header (11) and prevTagSize (4) */
1233. if (pos + 11 + dataSize + 4 > packet->m_nBodySize)
1234. {
1235. RTMP_Log(RTMP_LOGWARNING, "Stream corrupt?!");
1236. break;
1237. }
1238. if (packet->m_body[pos] == 0x12)
1239. {
1240. HandleMetadata(r, packet->m_body + pos + 11, dataSize);
1241. }
1242. else if (packet->m_body[pos] == 8 || packet->m_body[pos] == 9)
1243. {
1244. nTimeStamp = AMF_DecodeInt24(packet->m_body + pos + 4);
1245. nTimeStamp |= (packet->m_body[pos + 7] << 24);
1246. }
1247. pos += (11 + dataSize + 4);
1248. }
1249. if (!r->m_pausing)
1250. r->m_mediaStamp = nTimeStamp;
1252. /* FLV tag(s) */
1253. /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: FLV tag(s) %lu bytes", __FUNCTION__, packet.m_nBodySize); */
1254. bHasMediaPacket = 1;
1255. break;
1256. }
1257. default:
1258. RTMP_Log(RTMP_LOGDEBUG, "%s, unknown packet type received: 0x%02x", __FUNCTION__,
1259. packet->m_packetType);
1260. #ifdef _DEBUG
1261. RTMP_LogHex(RTMP_LOGDEBUG, packet->m_body, packet->m_nBodySize);
1262. #endif
1263. }
1265. return bHasMediaPacket;
1266. }
1268. #ifdef _DEBUG
1269. extern FILE *netstackdump;
1270. extern FILE *netstackdump_read;
1271. #endif
1273. static int
1274. ReadN(RTMP *r, char *buffer, int n)
1275. {
1276. int nOriginalSize = n;
1277. int avail;
1278. char *ptr;
1280. r->m_sb.sb_timedout = FALSE;
1282. #ifdef _DEBUG
1283. memset(buffer, 0, n);
1284. #endif
1286. ptr = buffer;
1287. while (n > 0)
1288. {
1289. int nBytes = 0, nRead;
1290. if (r->Link.protocol & RTMP_FEATURE_HTTP)
1291. {
1292. while (!r->m_resplen)
1293. {
1294. if (r->m_sb.sb_size < 144)
1295. {
1296. if (!r->m_unackd)
1297. HTTP_Post(r, RTMPT_IDLE, "", 1);
1298. if (RTMPSockBuf_Fill(&r->m_sb) < 1)
1299. {
1300. if (!r->m_sb.sb_timedout)
1301. RTMP_Close(r);
1302. return 0;
1303. }
1304. }
1305. if (HTTP_read(r, 0) == -1)
1306. {
1307. RTMP_Log(RTMP_LOGDEBUG, "%s, No valid HTTP response found", __FUNCTION__);
1308. RTMP_Close(r);
1309. return 0;
1310. }
1311. }
1312. if (r->m_resplen && !r->m_sb.sb_size)
1313. RTMPSockBuf_Fill(&r->m_sb);
1314. avail = r->m_sb.sb_size;
1315. if (avail > r->m_resplen)
1316. avail = r->m_resplen;
1317. }
1318. else
1319. {
1320. avail = r->m_sb.sb_size;
1321. if (avail == 0)
1322. {
1323. if (RTMPSockBuf_Fill(&r->m_sb) < 1)
1324. {
1325. if (!r->m_sb.sb_timedout)
1326. RTMP_Close(r);
1327. return 0;
1328. }
1329. avail = r->m_sb.sb_size;
1330. }
1331. }
1332. nRead = ((n < avail) ? n : avail);
1333. if (nRead > 0)
1334. {
1335. memcpy(ptr, r->m_sb.sb_start, nRead);
1336. r->m_sb.sb_start += nRead;
1337. r->m_sb.sb_size -= nRead;
1338. nBytes = nRead;
1339. r->m_nBytesIn += nRead;
1340. if (r->m_bSendCounter
1341. && r->m_nBytesIn > ( r->m_nBytesInSent + r->m_nClientBW / 10))
1342. if (!SendBytesReceived(r))
1343. return FALSE;
1344. }
1345. /*RTMP_Log(RTMP_LOGDEBUG, "%s: %d bytes\n", __FUNCTION__, nBytes); */
1346. #ifdef _DEBUG
1347. fwrite(ptr, 1, nBytes, netstackdump_read);
1348. #endif
1350. if (nBytes == 0)
1351. {
1352. RTMP_Log(RTMP_LOGDEBUG, "%s, RTMP socket closed by peer", __FUNCTION__);
1353. /*goto again; */
1354. RTMP_Close(r);
1355. break;
1356. }
1358. if (r->Link.protocol & RTMP_FEATURE_HTTP)
1359. r->m_resplen -= nBytes;
1361. #ifdef CRYPTO
1362. if (r->Link.rc4keyIn)
1363. {
1364. RC4_encrypt(r->Link.rc4keyIn, nBytes, ptr);
1365. }
1366. #endif
1368. n -= nBytes;
1369. ptr += nBytes;
1370. }
1372. return nOriginalSize - n;
1373. }
1375. static int
1376. WriteN(RTMP *r, const char *buffer, int n)
1377. {
1378. const char *ptr = buffer;
1379. #ifdef CRYPTO
1380. char *encrypted = 0;
1381. char buf[RTMP_BUFFER_CACHE_SIZE];
1383. if (r->Link.rc4keyOut)
1384. {
1385. if (n > sizeof(buf))
1386. encrypted = (char *)malloc(n);
1387. else
1388. encrypted = (char *)buf;
1389. ptr = encrypted;
1390. RC4_encrypt2(r->Link.rc4keyOut, n, buffer, ptr);
1391. }
1392. #endif
1394. while (n > 0)
1395. {
1396. int nBytes;
1398. if (r->Link.protocol & RTMP_FEATURE_HTTP)
1399. nBytes = HTTP_Post(r, RTMPT_SEND, ptr, n);
1400. else
1401. nBytes = RTMPSockBuf_Send(&r->m_sb, ptr, n);
1402. /*RTMP_Log(RTMP_LOGDEBUG, "%s: %d\n", __FUNCTION__, nBytes); */
1404. if (nBytes < 0)
1405. {
1406. int sockerr = GetSockError();
1407. RTMP_Log(RTMP_LOGERROR, "%s, RTMP send error %d (%d bytes)", __FUNCTION__,
1408. sockerr, n);
1410. if (sockerr == EINTR && !RTMP_ctrlC)
1411. continue;
1413. RTMP_Close(r);
1414. n = 1;
1415. break;
1416. }
1418. if (nBytes == 0)
1419. break;
1421. n -= nBytes;
1422. ptr += nBytes;
1423. }
1425. #ifdef CRYPTO
1426. if (encrypted && encrypted != buf)
1427. free(encrypted);
1428. #endif
1430. return n == 0;
1431. }
1433. #define SAVC(x) static const AVal av_##x = AVC(#x)
1435. SAVC(app);
1436. SAVC(connect);
1437. SAVC(flashVer);
1438. SAVC(swfUrl);
1439. SAVC(pageUrl);
1440. SAVC(tcUrl);
1441. SAVC(fpad);
1442. SAVC(capabilities);
1443. SAVC(audioCodecs);
1444. SAVC(videoCodecs);
1445. SAVC(videoFunction);
1446. SAVC(objectEncoding);
1447. SAVC(secureToken);
1448. SAVC(secureTokenResponse);
1449. SAVC(type);
1450. SAVC(nonprivate);
1452. static int
1453. SendConnectPacket(RTMP *r, RTMPPacket *cp)
1454. {
1455. RTMPPacket packet;
1456. char pbuf[4096], *pend = pbuf + sizeof(pbuf);
1457. char *enc;
1459. if (cp)
1460. return RTMP_SendPacket(r, cp, TRUE);
1462. packet.m_nChannel = 0x03; /* control channel (invoke) */
1463. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
1464. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1465. packet.m_nTimeStamp = 0;
1466. packet.m_nInfoField2 = 0;
1467. packet.m_hasAbsTimestamp = 0;
1468. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1470. enc = packet.m_body;
1471. enc = AMF_EncodeString(enc, pend, &av_connect);
1472. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1473. *enc++ = AMF_OBJECT;
1475. enc = AMF_EncodeNamedString(enc, pend, &av_app, &r->Link.app);
1476. if (!enc)
1477. return FALSE;
1478. if (r->Link.protocol & RTMP_FEATURE_WRITE)
1479. {
1480. enc = AMF_EncodeNamedString(enc, pend, &av_type, &av_nonprivate);
1481. if (!enc)
1482. return FALSE;
1483. }
1484. if (r->Link.flashVer.av_len)
1485. {
1486. enc = AMF_EncodeNamedString(enc, pend, &av_flashVer, &r->Link.flashVer);
1487. if (!enc)
1488. return FALSE;
1489. }
1490. if (r->Link.swfUrl.av_len)
1491. {
1492. enc = AMF_EncodeNamedString(enc, pend, &av_swfUrl, &r->Link.swfUrl);
1493. if (!enc)
1494. return FALSE;
1495. }
1496. if (r->Link.tcUrl.av_len)
1497. {
1498. enc = AMF_EncodeNamedString(enc, pend, &av_tcUrl, &r->Link.tcUrl);
1499. if (!enc)
1500. return FALSE;
1501. }
1502. if (!(r->Link.protocol & RTMP_FEATURE_WRITE))
1503. {
1504. enc = AMF_EncodeNamedBoolean(enc, pend, &av_fpad, FALSE);
1505. if (!enc)
1506. return FALSE;
1507. enc = AMF_EncodeNamedNumber(enc, pend, &av_capabilities, 15.0);
1508. if (!enc)
1509. return FALSE;
1510. enc = AMF_EncodeNamedNumber(enc, pend, &av_audioCodecs, r->m_fAudioCodecs);
1511. if (!enc)
1512. return FALSE;
1513. enc = AMF_EncodeNamedNumber(enc, pend, &av_videoCodecs, r->m_fVideoCodecs);
1514. if (!enc)
1515. return FALSE;
1516. enc = AMF_EncodeNamedNumber(enc, pend, &av_videoFunction, 1.0);
1517. if (!enc)
1518. return FALSE;
1519. if (r->Link.pageUrl.av_len)
1520. {
1521. enc = AMF_EncodeNamedString(enc, pend, &av_pageUrl, &r->Link.pageUrl);
1522. if (!enc)
1523. return FALSE;
1524. }
1525. }
1526. if (r->m_fEncoding != 0.0 || r->m_bSendEncoding)
1527. { /* AMF0, AMF3 not fully supported yet */
1528. enc = AMF_EncodeNamedNumber(enc, pend, &av_objectEncoding, r->m_fEncoding);
1529. if (!enc)
1530. return FALSE;
1531. }
1532. if (enc + 3 >= pend)
1533. return FALSE;
1534. *enc++ = 0;
1535. *enc++ = 0; /* end of object - 0x00 0x00 0x09 */
1536. *enc++ = AMF_OBJECT_END;
1538. /* add auth string */
1539. if (r->Link.auth.av_len)
1540. {
1541. enc = AMF_EncodeBoolean(enc, pend, r->Link.lFlags & RTMP_LF_AUTH);
1542. if (!enc)
1543. return FALSE;
1544. enc = AMF_EncodeString(enc, pend, &r->Link.auth);
1545. if (!enc)
1546. return FALSE;
1547. }
1548. if (r->Link.extras.o_num)
1549. {
1550. int i;
1551. for (i = 0; i < r->Link.extras.o_num; i++)
1552. {
1553. enc = AMFProp_Encode(&r->Link.extras.o_props[i], enc, pend);
1554. if (!enc)
1555. return FALSE;
1556. }
1557. }
1558. packet.m_nBodySize = enc - packet.m_body;
1560. return RTMP_SendPacket(r, &packet, TRUE);
1561. }
1563. #if 0 /* unused */
1564. SAVC(bgHasStream);
1566. static int
1567. SendBGHasStream(RTMP *r, double dId, AVal *playpath)
1568. {
1569. RTMPPacket packet;
1570. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1571. char *enc;
1573. packet.m_nChannel = 0x03; /* control channel (invoke) */
1574. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1575. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1576. packet.m_nTimeStamp = 0;
1577. packet.m_nInfoField2 = 0;
1578. packet.m_hasAbsTimestamp = 0;
1579. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1581. enc = packet.m_body;
1582. enc = AMF_EncodeString(enc, pend, &av_bgHasStream);
1583. enc = AMF_EncodeNumber(enc, pend, dId);
1584. *enc++ = AMF_NULL;
1586. enc = AMF_EncodeString(enc, pend, playpath);
1587. if (enc == NULL)
1588. return FALSE;
1590. packet.m_nBodySize = enc - packet.m_body;
1592. return RTMP_SendPacket(r, &packet, TRUE);
1593. }
1594. #endif
1596. SAVC(createStream);
1598. int
1599. RTMP_SendCreateStream(RTMP *r)
1600. {
1601. RTMPPacket packet;
1602. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1603. char *enc;
1605. packet.m_nChannel = 0x03; /* control channel (invoke) */
1606. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1607. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1608. packet.m_nTimeStamp = 0;
1609. packet.m_nInfoField2 = 0;
1610. packet.m_hasAbsTimestamp = 0;
1611. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1613. enc = packet.m_body;
1614. enc = AMF_EncodeString(enc, pend, &av_createStream);
1615. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1616. *enc++ = AMF_NULL; /* NULL */
1618. packet.m_nBodySize = enc - packet.m_body;
1620. return RTMP_SendPacket(r, &packet, TRUE);
1621. }
1623. SAVC(FCSubscribe);
1625. static int
1626. SendFCSubscribe(RTMP *r, AVal *subscribepath)
1627. {
1628. RTMPPacket packet;
1629. char pbuf[512], *pend = pbuf + sizeof(pbuf);
1630. char *enc;
1631. packet.m_nChannel = 0x03; /* control channel (invoke) */
1632. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1633. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1634. packet.m_nTimeStamp = 0;
1635. packet.m_nInfoField2 = 0;
1636. packet.m_hasAbsTimestamp = 0;
1637. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1639. RTMP_Log(RTMP_LOGDEBUG, "FCSubscribe: %s", subscribepath->av_val);
1640. enc = packet.m_body;
1641. enc = AMF_EncodeString(enc, pend, &av_FCSubscribe);
1642. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1643. *enc++ = AMF_NULL;
1644. enc = AMF_EncodeString(enc, pend, subscribepath);
1646. if (!enc)
1647. return FALSE;
1649. packet.m_nBodySize = enc - packet.m_body;
1651. return RTMP_SendPacket(r, &packet, TRUE);
1652. }
1654. /* Justin.tv specific authentication */
1655. static const AVal av_NetStream_Authenticate_UsherToken = AVC("NetStream.Authenticate.UsherToken");
1657. static int
1658. SendUsherToken(RTMP *r, AVal *usherToken)
1659. {
1660. RTMPPacket packet;
1661. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1662. char *enc;
1663. packet.m_nChannel = 0x03; /* control channel (invoke) */
1664. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1665. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1666. packet.m_nTimeStamp = 0;
1667. packet.m_nInfoField2 = 0;
1668. packet.m_hasAbsTimestamp = 0;
1669. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1671. RTMP_Log(RTMP_LOGDEBUG, "UsherToken: %s", usherToken->av_val);
1672. enc = packet.m_body;
1673. enc = AMF_EncodeString(enc, pend, &av_NetStream_Authenticate_UsherToken);
1674. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1675. *enc++ = AMF_NULL;
1676. enc = AMF_EncodeString(enc, pend, usherToken);
1678. if (!enc)
1679. return FALSE;
1681. packet.m_nBodySize = enc - packet.m_body;
1683. return RTMP_SendPacket(r, &packet, FALSE);
1684. }
1685. /******************************************/
1687. SAVC(releaseStream);
1689. static int
1690. SendReleaseStream(RTMP *r)
1691. {
1692. RTMPPacket packet;
1693. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1694. char *enc;
1696. packet.m_nChannel = 0x03; /* control channel (invoke) */
1697. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1698. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1699. packet.m_nTimeStamp = 0;
1700. packet.m_nInfoField2 = 0;
1701. packet.m_hasAbsTimestamp = 0;
1702. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1704. enc = packet.m_body;
1705. enc = AMF_EncodeString(enc, pend, &av_releaseStream);
1706. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1707. *enc++ = AMF_NULL;
1708. enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
1709. if (!enc)
1710. return FALSE;
1712. packet.m_nBodySize = enc - packet.m_body;
1714. return RTMP_SendPacket(r, &packet, FALSE);
1715. }
1717. SAVC(FCPublish);
1719. static int
1720. SendFCPublish(RTMP *r)
1721. {
1722. RTMPPacket packet;
1723. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1724. char *enc;
1726. packet.m_nChannel = 0x03; /* control channel (invoke) */
1727. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1728. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1729. packet.m_nTimeStamp = 0;
1730. packet.m_nInfoField2 = 0;
1731. packet.m_hasAbsTimestamp = 0;
1732. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1734. enc = packet.m_body;
1735. enc = AMF_EncodeString(enc, pend, &av_FCPublish);
1736. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1737. *enc++ = AMF_NULL;
1738. enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
1739. if (!enc)
1740. return FALSE;
1742. packet.m_nBodySize = enc - packet.m_body;
1744. return RTMP_SendPacket(r, &packet, FALSE);
1745. }
1747. SAVC(FCUnpublish);
1749. static int
1750. SendFCUnpublish(RTMP *r)
1751. {
1752. RTMPPacket packet;
1753. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1754. char *enc;
1756. packet.m_nChannel = 0x03; /* control channel (invoke) */
1757. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1758. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1759. packet.m_nTimeStamp = 0;
1760. packet.m_nInfoField2 = 0;
1761. packet.m_hasAbsTimestamp = 0;
1762. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1764. enc = packet.m_body;
1765. enc = AMF_EncodeString(enc, pend, &av_FCUnpublish);
1766. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1767. *enc++ = AMF_NULL;
1768. enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
1769. if (!enc)
1770. return FALSE;
1772. packet.m_nBodySize = enc - packet.m_body;
1774. return RTMP_SendPacket(r, &packet, FALSE);
1775. }
1777. SAVC(publish);
1778. SAVC(live);
1779. SAVC(record);
1781. static int
1782. SendPublish(RTMP *r)
1783. {
1784. RTMPPacket packet;
1785. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
1786. char *enc;
1788. packet.m_nChannel = 0x04; /* source channel (invoke) */
1789. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
1790. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1791. packet.m_nTimeStamp = 0;
1792. packet.m_nInfoField2 = r->m_stream_id;
1793. packet.m_hasAbsTimestamp = 0;
1794. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1796. enc = packet.m_body;
1797. enc = AMF_EncodeString(enc, pend, &av_publish);
1798. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1799. *enc++ = AMF_NULL;
1800. enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
1801. if (!enc)
1802. return FALSE;
1804. /* FIXME: should we choose live based on Link.lFlags & RTMP_LF_LIVE? */
1805. enc = AMF_EncodeString(enc, pend, &av_live);
1806. if (!enc)
1807. return FALSE;
1809. packet.m_nBodySize = enc - packet.m_body;
1811. return RTMP_SendPacket(r, &packet, TRUE);
1812. }
1814. SAVC(deleteStream);
1816. static int
1817. SendDeleteStream(RTMP *r, double dStreamId)
1818. {
1819. RTMPPacket packet;
1820. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1821. char *enc;
1823. packet.m_nChannel = 0x03; /* control channel (invoke) */
1824. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1825. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1826. packet.m_nTimeStamp = 0;
1827. packet.m_nInfoField2 = 0;
1828. packet.m_hasAbsTimestamp = 0;
1829. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1831. enc = packet.m_body;
1832. enc = AMF_EncodeString(enc, pend, &av_deleteStream);
1833. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1834. *enc++ = AMF_NULL;
1835. enc = AMF_EncodeNumber(enc, pend, dStreamId);
1837. packet.m_nBodySize = enc - packet.m_body;
1839. /* no response expected */
1840. return RTMP_SendPacket(r, &packet, FALSE);
1841. }
1843. SAVC(pause);
1845. int
1846. RTMP_SendPause(RTMP *r, int DoPause, int iTime)
1847. {
1848. RTMPPacket packet;
1849. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1850. char *enc;
1852. packet.m_nChannel = 0x08; /* video channel */
1853. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1854. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1855. packet.m_nTimeStamp = 0;
1856. packet.m_nInfoField2 = 0;
1857. packet.m_hasAbsTimestamp = 0;
1858. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1860. enc = packet.m_body;
1861. enc = AMF_EncodeString(enc, pend, &av_pause);
1862. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1863. *enc++ = AMF_NULL;
1864. enc = AMF_EncodeBoolean(enc, pend, DoPause);
1865. enc = AMF_EncodeNumber(enc, pend, (double)iTime);
1867. packet.m_nBodySize = enc - packet.m_body;
1869. RTMP_Log(RTMP_LOGDEBUG, "%s, %d, pauseTime=%d", __FUNCTION__, DoPause, iTime);
1870. return RTMP_SendPacket(r, &packet, TRUE);
1871. }
1873. int RTMP_Pause(RTMP *r, int DoPause)
1874. {
1875. if (DoPause)
1876. r->m_pauseStamp = r->m_channelTimestamp[r->m_mediaChannel];
1877. return RTMP_SendPause(r, DoPause, r->m_pauseStamp);
1878. }
1880. SAVC(seek);
1882. int
1883. RTMP_SendSeek(RTMP *r, int iTime)
1884. {
1885. RTMPPacket packet;
1886. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1887. char *enc;
1889. packet.m_nChannel = 0x08; /* video channel */
1890. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1891. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1892. packet.m_nTimeStamp = 0;
1893. packet.m_nInfoField2 = 0;
1894. packet.m_hasAbsTimestamp = 0;
1895. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1897. enc = packet.m_body;
1898. enc = AMF_EncodeString(enc, pend, &av_seek);
1899. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1900. *enc++ = AMF_NULL;
1901. enc = AMF_EncodeNumber(enc, pend, (double)iTime);
1903. packet.m_nBodySize = enc - packet.m_body;
1905. r->m_read.flags |= RTMP_READ_SEEKING;
1906. r->m_read.nResumeTS = 0;
1908. return RTMP_SendPacket(r, &packet, TRUE);
1909. }
1911. int
1912. RTMP_SendServerBW(RTMP *r)
1913. {
1914. RTMPPacket packet;
1915. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1917. packet.m_nChannel = 0x02; /* control channel (invoke) */
1918. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
1919. packet.m_packetType = RTMP_PACKET_TYPE_SERVER_BW;
1920. packet.m_nTimeStamp = 0;
1921. packet.m_nInfoField2 = 0;
1922. packet.m_hasAbsTimestamp = 0;
1923. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1925. packet.m_nBodySize = 4;
1927. AMF_EncodeInt32(packet.m_body, pend, r->m_nServerBW);
1928. return RTMP_SendPacket(r, &packet, FALSE);
1929. }
1931. int
1932. RTMP_SendClientBW(RTMP *r)
1933. {
1934. RTMPPacket packet;
1935. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1937. packet.m_nChannel = 0x02; /* control channel (invoke) */
1938. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
1939. packet.m_packetType = RTMP_PACKET_TYPE_CLIENT_BW;
1940. packet.m_nTimeStamp = 0;
1941. packet.m_nInfoField2 = 0;
1942. packet.m_hasAbsTimestamp = 0;
1943. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1945. packet.m_nBodySize = 5;
1947. AMF_EncodeInt32(packet.m_body, pend, r->m_nClientBW);
1948. packet.m_body[4] = r->m_nClientBW2;
1949. return RTMP_SendPacket(r, &packet, FALSE);
1950. }
1952. static int
1953. SendBytesReceived(RTMP *r)
1954. {
1955. RTMPPacket packet;
1956. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1958. packet.m_nChannel = 0x02; /* control channel (invoke) */
1959. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
1960. packet.m_packetType = RTMP_PACKET_TYPE_BYTES_READ_REPORT;
1961. packet.m_nTimeStamp = 0;
1962. packet.m_nInfoField2 = 0;
1963. packet.m_hasAbsTimestamp = 0;
1964. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1966. packet.m_nBodySize = 4;
1968. AMF_EncodeInt32(packet.m_body, pend, r->m_nBytesIn); /* hard coded for now */
1969. r->m_nBytesInSent = r->m_nBytesIn;
1971. /*RTMP_Log(RTMP_LOGDEBUG, "Send bytes report. 0x%x (%d bytes)", (unsigned int)m_nBytesIn, m_nBytesIn); */
1972. return RTMP_SendPacket(r, &packet, FALSE);
1973. }
1975. SAVC(_checkbw);
1977. static int
1978. SendCheckBW(RTMP *r)
1979. {
1980. RTMPPacket packet;
1981. char pbuf[256], *pend = pbuf + sizeof(pbuf);
1982. char *enc;
1984. packet.m_nChannel = 0x03; /* control channel (invoke) */
1985. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
1986. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
1987. packet.m_nTimeStamp = 0; /* RTMP_GetTime(); */
1988. packet.m_nInfoField2 = 0;
1989. packet.m_hasAbsTimestamp = 0;
1990. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
1992. enc = packet.m_body;
1993. enc = AMF_EncodeString(enc, pend, &av__checkbw);
1994. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
1995. *enc++ = AMF_NULL;
1997. packet.m_nBodySize = enc - packet.m_body;
1999. /* triggers _onbwcheck and eventually results in _onbwdone */
2000. return RTMP_SendPacket(r, &packet, FALSE);
2001. }
2003. SAVC(_result);
2005. static int
2006. SendCheckBWResult(RTMP *r, double txn)
2007. {
2008. RTMPPacket packet;
2009. char pbuf[256], *pend = pbuf + sizeof(pbuf);
2010. char *enc;
2012. packet.m_nChannel = 0x03; /* control channel (invoke) */
2013. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
2014. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
2015. packet.m_nTimeStamp = 0x16 * r->m_nBWCheckCounter; /* temp inc value. till we figure it out. */
2016. packet.m_nInfoField2 = 0;
2017. packet.m_hasAbsTimestamp = 0;
2018. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2020. enc = packet.m_body;
2021. enc = AMF_EncodeString(enc, pend, &av__result);
2022. enc = AMF_EncodeNumber(enc, pend, txn);
2023. *enc++ = AMF_NULL;
2024. enc = AMF_EncodeNumber(enc, pend, (double)r->m_nBWCheckCounter++);
2026. packet.m_nBodySize = enc - packet.m_body;
2028. return RTMP_SendPacket(r, &packet, FALSE);
2029. }
2031. SAVC(ping);
2032. SAVC(pong);
2034. static int
2035. SendPong(RTMP *r, double txn)
2036. {
2037. RTMPPacket packet;
2038. char pbuf[256], *pend = pbuf + sizeof(pbuf);
2039. char *enc;
2041. packet.m_nChannel = 0x03; /* control channel (invoke) */
2042. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
2043. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
2044. packet.m_nTimeStamp = 0x16 * r->m_nBWCheckCounter; /* temp inc value. till we figure it out. */
2045. packet.m_nInfoField2 = 0;
2046. packet.m_hasAbsTimestamp = 0;
2047. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2049. enc = packet.m_body;
2050. enc = AMF_EncodeString(enc, pend, &av_pong);
2051. enc = AMF_EncodeNumber(enc, pend, txn);
2052. *enc++ = AMF_NULL;
2054. packet.m_nBodySize = enc - packet.m_body;
2056. return RTMP_SendPacket(r, &packet, FALSE);
2057. }
2059. SAVC(play);
2061. static int
2062. SendPlay(RTMP *r)
2063. {
2064. RTMPPacket packet;
2065. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
2066. char *enc;
2068. packet.m_nChannel = 0x08; /* we make 8 our stream channel */
2069. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
2070. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
2071. packet.m_nTimeStamp = 0;
2072. packet.m_nInfoField2 = r->m_stream_id; /*0x01000000; */
2073. packet.m_hasAbsTimestamp = 0;
2074. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2076. enc = packet.m_body;
2077. enc = AMF_EncodeString(enc, pend, &av_play);
2078. enc = AMF_EncodeNumber(enc, pend, ++r->m_numInvokes);
2079. *enc++ = AMF_NULL;
2081. RTMP_Log(RTMP_LOGDEBUG, "%s, seekTime=%d, stopTime=%d, sending play: %s",
2082. __FUNCTION__, r->Link.seekTime, r->Link.stopTime,
2083. r->Link.playpath.av_val);
2084. enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
2085. if (!enc)
2086. return FALSE;
2088. /* Optional parameters start and len.
2089. *
2090. * start: -2, -1, 0, positive number
2091. * -2: looks for a live stream, then a recorded stream,
2092. * if not found any open a live stream
2093. * -1: plays a live stream
2094. * >=0: plays a recorded streams from 'start' milliseconds
2095. */
2096. if (r->Link.lFlags & RTMP_LF_LIVE)
2097. enc = AMF_EncodeNumber(enc, pend, -1000.0);
2098. else
2099. {
2100. if (r->Link.seekTime > 0.0)
2101. enc = AMF_EncodeNumber(enc, pend, r->Link.seekTime); /* resume from here */
2102. else
2103. enc = AMF_EncodeNumber(enc, pend, 0.0); /*-2000.0);*/ /* recorded as default, -2000.0 is not reliable since that freezes the player if the stream is not found */
2104. }
2105. if (!enc)
2106. return FALSE;
2108. /* len: -1, 0, positive number
2109. * -1: plays live or recorded stream to the end (default)
2110. * 0: plays a frame 'start' ms away from the beginning
2111. * >0: plays a live or recoded stream for 'len' milliseconds
2112. */
2113. /*enc += EncodeNumber(enc, -1.0); */ /* len */
2114. if (r->Link.stopTime)
2115. {
2116. enc = AMF_EncodeNumber(enc, pend, r->Link.stopTime - r->Link.seekTime);
2117. if (!enc)
2118. return FALSE;
2119. }
2121. packet.m_nBodySize = enc - packet.m_body;
2123. return RTMP_SendPacket(r, &packet, TRUE);
2124. }
2126. SAVC(set_playlist);
2127. SAVC(0);
2129. static int
2130. SendPlaylist(RTMP *r)
2131. {
2132. RTMPPacket packet;
2133. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
2134. char *enc;
2136. packet.m_nChannel = 0x08; /* we make 8 our stream channel */
2137. packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
2138. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
2139. packet.m_nTimeStamp = 0;
2140. packet.m_nInfoField2 = r->m_stream_id; /*0x01000000; */
2141. packet.m_hasAbsTimestamp = 0;
2142. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2144. enc = packet.m_body;
2145. enc = AMF_EncodeString(enc, pend, &av_set_playlist);
2146. enc = AMF_EncodeNumber(enc, pend, 0);
2147. *enc++ = AMF_NULL;
2148. *enc++ = AMF_ECMA_ARRAY;
2149. *enc++ = 0;
2150. *enc++ = 0;
2151. *enc++ = 0;
2152. *enc++ = AMF_OBJECT;
2153. enc = AMF_EncodeNamedString(enc, pend, &av_0, &r->Link.playpath);
2154. if (!enc)
2155. return FALSE;
2156. if (enc + 3 >= pend)
2157. return FALSE;
2158. *enc++ = 0;
2159. *enc++ = 0;
2160. *enc++ = AMF_OBJECT_END;
2162. packet.m_nBodySize = enc - packet.m_body;
2164. return RTMP_SendPacket(r, &packet, TRUE);
2165. }
2167. static int
2168. SendSecureTokenResponse(RTMP *r, AVal *resp)
2169. {
2170. RTMPPacket packet;
2171. char pbuf[1024], *pend = pbuf + sizeof(pbuf);
2172. char *enc;
2174. packet.m_nChannel = 0x03; /* control channel (invoke) */
2175. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
2176. packet.m_packetType = RTMP_PACKET_TYPE_INVOKE;
2177. packet.m_nTimeStamp = 0;
2178. packet.m_nInfoField2 = 0;
2179. packet.m_hasAbsTimestamp = 0;
2180. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2182. enc = packet.m_body;
2183. enc = AMF_EncodeString(enc, pend, &av_secureTokenResponse);
2184. enc = AMF_EncodeNumber(enc, pend, 0.0);
2185. *enc++ = AMF_NULL;
2186. enc = AMF_EncodeString(enc, pend, resp);
2187. if (!enc)
2188. return FALSE;
2190. packet.m_nBodySize = enc - packet.m_body;
2192. return RTMP_SendPacket(r, &packet, FALSE);
2193. }
2195. /*
2196. from http://jira.red5.org/confluence/display/docs/Ping:
2198. Ping is the most mysterious message in RTMP and till now we haven't fully interpreted it yet. In summary, Ping message is used as a special command that are exchanged between client and server. This page aims to document all known Ping messages. Expect the list to grow.
2200. The type of Ping packet is 0x4 and contains two mandatory parameters and two optional parameters. The first parameter is the type of Ping and in short integer. The second parameter is the target of the ping. As Ping is always sent in Channel 2 (control channel) and the target object in RTMP header is always 0 which means the Connection object, it's necessary to put an extra parameter to indicate the exact target object the Ping is sent to. The second parameter takes this responsibility. The value has the same meaning as the target object field in RTMP header. (The second value could also be used as other purposes, like RTT Ping/Pong. It is used as the timestamp.) The third and fourth parameters are optional and could be looked upon as the parameter of the Ping packet. Below is an unexhausted list of Ping messages.
2202. * type 0: Clear the stream. No third and fourth parameters. The second parameter could be 0. After the connection is established, a Ping 0,0 will be sent from server to client. The message will also be sent to client on the start of Play and in response of a Seek or Pause/Resume request. This Ping tells client to re-calibrate the clock with the timestamp of the next packet server sends.
2203. * type 1: Tell the stream to clear the playing buffer.
2204. * type 3: Buffer time of the client. The third parameter is the buffer time in millisecond.
2205. * type 4: Reset a stream. Used together with type 0 in the case of VOD. Often sent before type 0.
2206. * type 6: Ping the client from server. The second parameter is the current time.
2207. * type 7: Pong reply from client. The second parameter is the time the server sent with his ping request.
2208. * type 26: SWFVerification request
2209. * type 27: SWFVerification response
2210. */
2211. int
2212. RTMP_SendCtrl(RTMP *r, short nType, unsigned int nObject, unsigned int nTime)
2213. {
2214. RTMPPacket packet;
2215. char pbuf[256], *pend = pbuf + sizeof(pbuf);
2216. int nSize;
2217. char *buf;
2219. RTMP_Log(RTMP_LOGDEBUG, "sending ctrl. type: 0x%04x", (unsigned short)nType);
2221. packet.m_nChannel = 0x02; /* control channel (ping) */
2222. packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
2223. packet.m_packetType = RTMP_PACKET_TYPE_CONTROL;
2224. packet.m_nTimeStamp = 0; /* RTMP_GetTime(); */
2225. packet.m_nInfoField2 = 0;
2226. packet.m_hasAbsTimestamp = 0;
2227. packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE;
2229. switch(nType) {
2230. case 0x03: nSize = 10; break; /* buffer time */
2231. case 0x1A: nSize = 3; break; /* SWF verify request */
2232. case 0x1B: nSize = 44; break; /* SWF verify response */
2233. default: nSize = 6; break;
2234. }
2236. packet.m_nBodySize = nSize;
2238. buf = packet.m_body;
2239. buf = AMF_EncodeInt16(buf, pend, nType);
2241. if (nType == 0x1B)
2242. {
2243. #ifdef CRYPTO
2244. memcpy(buf, r->Link.SWFVerificationResponse, 42);
2245. RTMP_Log(RTMP_LOGDEBUG, "Sending SWFVerification response: ");
2246. RTMP_LogHex(RTMP_LOGDEBUG, (uint8_t *)packet.m_body, packet.m_nBodySize);
2247. #endif
2248. }
2249. else if (nType == 0x1A)
2250. {
2251. *buf = nObject & 0xff;
2252. }
2253. else
2254. {
2255. if (nSize > 2)
2256. buf = AMF_EncodeInt32(buf, pend, nObject);
2258. if (nSize > 6)
2259. buf = AMF_EncodeInt32(buf, pend, nTime);
2260. }
2262. return RTMP_SendPacket(r, &packet, FALSE);
2263. }
2265. static void
2266. AV_erase(RTMP_METHOD *vals, int *num, int i, int freeit)
2267. {
2268. if (freeit)
2269. free(vals[i].name.av_val);
2270. (*num)--;
2271. for (; i < *num; i++)
2272. {
2273. vals[i] = vals[i + 1];
2274. }
2275. vals[i].name.av_val = NULL;
2276. vals[i].name.av_len = 0;
2277. vals[i].num = 0;
2278. }
2280. void
2281. RTMP_DropRequest(RTMP *r, int i, int freeit)
2282. {
2283. AV_erase(r->m_methodCalls, &r->m_numCalls, i, freeit);
2284. }
2286. static void
2287. AV_queue(RTMP_METHOD **vals, int *num, AVal *av, int txn)
2288. {
2289. char *tmp;
2290. if (!(*num & 0x0f))
2291. *vals = realloc(*vals, (*num + 16) * sizeof(RTMP_METHOD));
2292. tmp = malloc(av->av_len + 1);
2293. memcpy(tmp, av->av_val, av->av_len);
2294. tmp[av->av_len] = '\0';
2295. (*vals)[*num].num = txn;
2296. (*vals)[*num].name.av_len = av->av_len;
2297. (*vals)[(*num)++].name.av_val = tmp;
2298. }
2300. static void
2301. AV_clear(RTMP_METHOD *vals, int num)
2302. {
2303. int i;
2304. for (i = 0; i < num; i++)
2305. free(vals[i].name.av_val);
2306. free(vals);
2307. }
2309. SAVC(onBWDone);
2310. SAVC(onFCSubscribe);
2311. SAVC(onFCUnsubscribe);
2312. SAVC(_onbwcheck);
2313. SAVC(_onbwdone);
2314. SAVC(_error);
2315. SAVC(close);
2316. SAVC(code);
2317. SAVC(level);
2318. SAVC(onStatus);
2319. SAVC(playlist_ready);
2320. static const AVal av_NetStream_Failed = AVC("NetStream.Failed");
2321. static const AVal av_NetStream_Play_Failed = AVC("NetStream.Play.Failed");
2322. static const AVal av_NetStream_Play_StreamNotFound =
2323. AVC("NetStream.Play.StreamNotFound");
2324. static const AVal av_NetConnection_Connect_InvalidApp =
2325. AVC("NetConnection.Connect.InvalidApp");
2326. static const AVal av_NetStream_Play_Start = AVC("NetStream.Play.Start");
2327. static const AVal av_NetStream_Play_Complete = AVC("NetStream.Play.Complete");
2328. static const AVal av_NetStream_Play_Stop = AVC("NetStream.Play.Stop");
2329. static const AVal av_NetStream_Seek_Notify = AVC("NetStream.Seek.Notify");
2330. static const AVal av_NetStream_Pause_Notify = AVC("NetStream.Pause.Notify");
2331. static const AVal av_NetStream_Play_PublishNotify =
2332. AVC("NetStream.Play.PublishNotify");
2333. static const AVal av_NetStream_Play_UnpublishNotify =
2334. AVC("NetStream.Play.UnpublishNotify");
2335. static const AVal av_NetStream_Publish_Start = AVC("NetStream.Publish.Start");
2337. /* Returns 0 for OK/Failed/error, 1 for 'Stop or Complete' */
2338. static int
2339. HandleInvoke(RTMP *r, const char *body, unsigned int nBodySize)
2340. {
2341. AMFObject obj;
2342. AVal method;
2343. double txn;
2344. int ret = 0, nRes;
2345. if (body[0] != 0x02) /* make sure it is a string method name we start with */
2346. {
2347. RTMP_Log(RTMP_LOGWARNING, "%s, Sanity failed. no string method in invoke packet",
2348. __FUNCTION__);
2349. return 0;
2350. }
2352. nRes = AMF_Decode(&obj, body, nBodySize, FALSE);
2353. if (nRes < 0)
2354. {
2355. RTMP_Log(RTMP_LOGERROR, "%s, error decoding invoke packet", __FUNCTION__);
2356. return 0;
2357. }
2359. AMF_Dump(&obj);
2360. AMFProp_GetString(AMF_GetProp(&obj, NULL, 0), &method);
2361. txn = AMFProp_GetNumber(AMF_GetProp(&obj, NULL, 1));
2362. RTMP_Log(RTMP_LOGDEBUG, "%s, server invoking <%s>", __FUNCTION__, method.av_val);
2364. if (AVMATCH(&method, &av__result))
2365. {
2366. AVal methodInvoked = {0};
2367. int i;
2369. for (i=0; i<r->m_numCalls; i++) {
2370. if (r->m_methodCalls[i].num == (int)txn) {
2371. methodInvoked = r->m_methodCalls[i].name;
2372. AV_erase(r->m_methodCalls, &r->m_numCalls, i, FALSE);
2373. break;
2374. }
2375. }
2376. if (!methodInvoked.av_val) {
2377. RTMP_Log(RTMP_LOGDEBUG, "%s, received result id %f without matching request",
2378. __FUNCTION__, txn);
2379. goto leave;
2380. }
2382. RTMP_Log(RTMP_LOGDEBUG, "%s, received result for method call <%s>", __FUNCTION__,
2383. methodInvoked.av_val);
2385. if (AVMATCH(&methodInvoked, &av_connect))
2386. {
2387. if (r->Link.token.av_len)
2388. {
2389. AMFObjectProperty p;
2390. if (RTMP_FindFirstMatchingProperty(&obj, &av_secureToken, &p))
2391. {
2392. DecodeTEA(&r->Link.token, &p.p_vu.p_aval);
2393. SendSecureTokenResponse(r, &p.p_vu.p_aval);
2394. }
2395. }
2396. if (r->Link.protocol & RTMP_FEATURE_WRITE)
2397. {
2398. SendReleaseStream(r);
2399. SendFCPublish(r);
2400. }
2401. else
2402. {
2403. RTMP_SendServerBW(r);
2404. RTMP_SendCtrl(r, 3, 0, 300);
2405. }
2406. RTMP_SendCreateStream(r);
2408. if (!(r->Link.protocol & RTMP_FEATURE_WRITE))
2409. {
2410. /* Authenticate on Justin.tv legacy servers before sending FCSubscribe */
2411. if (r->Link.usherToken.av_len)
2412. SendUsherToken(r, &r->Link.usherToken);
2413. /* Send the FCSubscribe if live stream or if subscribepath is set */
2414. if (r->Link.subscribepath.av_len)
2415. SendFCSubscribe(r, &r->Link.subscribepath);
2416. else if (r->Link.lFlags & RTMP_LF_LIVE)
2417. SendFCSubscribe(r, &r->Link.playpath);
2418. }
2419. }
2420. else if (AVMATCH(&methodInvoked, &av_createStream))
2421. {
2422. r->m_stream_id = (int)AMFProp_GetNumber(AMF_GetProp(&obj, NULL, 3));
2424. if (r->Link.protocol & RTMP_FEATURE_WRITE)
2425. {
2426. SendPublish(r);
2427. }
2428. else
2429. {
2430. if (r->Link.lFlags & RTMP_LF_PLST)
2431. SendPlaylist(r);
2432. SendPlay(r);
2433. RTMP_SendCtrl(r, 3, r->m_stream_id, r->m_nBufferMS);
2434. }
2435. }
2436. else if (AVMATCH(&methodInvoked, &av_play) ||
2437. AVMATCH(&methodInvoked, &av_publish))
2438. {
2439. r->m_bPlaying = TRUE;
2440. }
2441. free(methodInvoked.av_val);
2442. }
2443. else if (AVMATCH(&method, &av_onBWDone))
2444. {
2445. if (!r->m_nBWCheckCounter)
2446. SendCheckBW(r);
2447. }
2448. else if (AVMATCH(&method, &av_onFCSubscribe))
2449. {
2450. /* SendOnFCSubscribe(); */
2451. }
2452. else if (AVMATCH(&method, &av_onFCUnsubscribe))
2453. {
2454. RTMP_Close(r);
2455. ret = 1;
2456. }
2457. else if (AVMATCH(&method, &av_ping))
2458. {
2459. SendPong(r, txn);
2460. }
2461. else if (AVMATCH(&method, &av__onbwcheck))
2462. {
2463. SendCheckBWResult(r, txn);
2464. }
2465. else if (AVMATCH(&method, &av__onbwdone))
2466. {
2467. int i;
2468. for (i = 0; i < r->m_numCalls; i++)
2469. if (AVMATCH(&r->m_methodCalls[i].name, &av__checkbw))
2470. {
2471. AV_erase(r->m_methodCalls, &r->m_numCalls, i, TRUE);
2472. break;
2473. }
2474. }
2475. else if (AVMATCH(&method, &av__error))
2476. {
2477. RTMP_Log(RTMP_LOGERROR, "rtmp server sent error");
2478. }
2479. else if (AVMATCH(&method, &av_close))
2480. {
2481. RTMP_Log(RTMP_LOGERROR, "rtmp server requested close");
2482. RTMP_Close(r);
2483. }
2484. else if (AVMATCH(&method, &av_onStatus))
2485. {
2486. AMFObject obj2;
2487. AVal code, level;
2488. AMFProp_GetObject(AMF_GetProp(&obj, NULL, 3), &obj2);
2489. AMFProp_GetString(AMF_GetProp(&obj2, &av_code, -1), &code);
2490. AMFProp_GetString(AMF_GetProp(&obj2, &av_level, -1), &level);
2492. RTMP_Log(RTMP_LOGDEBUG, "%s, onStatus: %s", __FUNCTION__, code.av_val);
2493. if (AVMATCH(&code, &av_NetStream_Failed)
2494. || AVMATCH(&code, &av_NetStream_Play_Failed)
2495. || AVMATCH(&code, &av_NetStream_Play_StreamNotFound)
2496. || AVMATCH(&code, &av_NetConnection_Connect_InvalidApp))
2497. {
2498. r->m_stream_id = -1;
2499. RTMP_Close(r);
2500. RTMP_Log(RTMP_LOGERROR, "Closing connection: %s", code.av_val);
2501. }
2503. else if (AVMATCH(&code, &av_NetStream_Play_Start)
2504. || AVMATCH(&code, &av_NetStream_Play_PublishNotify))
2505. {
2506. int i;
2507. r->m_bPlaying = TRUE;
2508. for (i = 0; i < r->m_numCalls; i++)
2509. {
2510. if (AVMATCH(&r->m_methodCalls[i].name, &av_play))
2511. {
2512. AV_erase(r->m_methodCalls, &r->m_numCalls, i, TRUE);
2513. break;
2514. }
2515. }
2516. }
2518. else if (AVMATCH(&code, &av_NetStream_Publish_Start))
2519. {
2520. int i;
2521. r->m_bPlaying = TRUE;
2522. for (i = 0; i < r->m_numCalls; i++)
2523. {
2524. if (AVMATCH(&r->m_methodCalls[i].name, &av_publish))
2525. {
2526. AV_erase(r->m_methodCalls, &r->m_numCalls, i, TRUE);
2527. break;
2528. }
2529. }
2530. }
2532. /* Return 1 if this is a Play.Complete or Play.Stop */
2533. else if (AVMATCH(&code, &av_NetStream_Play_Complete)
2534. || AVMATCH(&code, &av_NetStream_Play_Stop)
2535. || AVMATCH(&code, &av_NetStream_Play_UnpublishNotify))
2536. {
2537. RTMP_Close(r);
2538. ret = 1;
2539. }
2541. else if (AVMATCH(&code, &av_NetStream_Seek_Notify))
2542. {
2543. r->m_read.flags &= ~RTMP_READ_SEEKING;
2544. }
2546. else if (AVMATCH(&code, &av_NetStream_Pause_Notify))
2547. {
2548. if (r->m_pausing == 1 || r->m_pausing == 2)
2549. {
2550. RTMP_SendPause(r, FALSE, r->m_pauseStamp);
2551. r->m_pausing = 3;
2552. }
2553. }
2554. }
2555. else if (AVMATCH(&method, &av_playlist_ready))
2556. {
2557. int i;
2558. for (i = 0; i < r->m_numCalls; i++)
2559. {
2560. if (AVMATCH(&r->m_methodCalls[i].name, &av_set_playlist))
2561. {
2562. AV_erase(r->m_methodCalls, &r->m_numCalls, i, TRUE);
2563. break;
2564. }
2565. }
2566. }
2567. else
2568. {
2570. }
2571. leave:
2572. AMF_Reset(&obj);
2573. return ret;
2574. }
2576. int
2577. RTMP_FindFirstMatchingProperty(AMFObject *obj, const AVal *name,
2578. AMFObjectProperty * p)
2579. {
2580. int n;
2581. /* this is a small object search to locate the "duration" property */
2582. for (n = 0; n < obj->o_num; n++)
2583. {
2584. AMFObjectProperty *prop = AMF_GetProp(obj, NULL, n);
2586. if (AVMATCH(&prop->p_name, name))
2587. {
2588. memcpy(p, prop, sizeof(*prop));
2589. return TRUE;
2590. }
2592. if (prop->p_type == AMF_OBJECT)
2593. {
2594. if (RTMP_FindFirstMatchingProperty(&prop->p_vu.p_object, name, p))
2595. return TRUE;
2596. }
2597. }
2598. return FALSE;
2599. }
2601. /* Like above, but only check if name is a prefix of property */
2602. int
2603. RTMP_FindPrefixProperty(AMFObject *obj, const AVal *name,
2604. AMFObjectProperty * p)
2605. {
2606. int n;
2607. for (n = 0; n < obj->o_num; n++)
2608. {
2609. AMFObjectProperty *prop = AMF_GetProp(obj, NULL, n);
2611. if (prop->p_name.av_len > name->av_len &&
2612. !memcmp(prop->p_name.av_val, name->av_val, name->av_len))
2613. {
2614. memcpy(p, prop, sizeof(*prop));
2615. return TRUE;
2616. }
2618. if (prop->p_type == AMF_OBJECT)
2619. {
2620. if (RTMP_FindPrefixProperty(&prop->p_vu.p_object, name, p))
2621. return TRUE;
2622. }
2623. }
2624. return FALSE;
2625. }
2627. static int
2628. DumpMetaData(AMFObject *obj)
2629. {
2630. AMFObjectProperty *prop;
2631. int n;
2632. for (n = 0; n < obj->o_num; n++)
2633. {
2634. prop = AMF_GetProp(obj, NULL, n);
2635. if (prop->p_type != AMF_OBJECT)
2636. {
2637. char str[256] = "";
2638. switch (prop->p_type)
2639. {
2640. case AMF_NUMBER:
2641. snprintf(str, 255, "%.2f", prop->p_vu.p_number);
2642. break;
2643. case AMF_BOOLEAN:
2644. snprintf(str, 255, "%s",
2645. prop->p_vu.p_number != 0. ? "TRUE" : "FALSE");
2646. break;
2647. case AMF_STRING:
2648. snprintf(str, 255, "%.*s", prop->p_vu.p_aval.av_len,
2649. prop->p_vu.p_aval.av_val);
2650. break;
2651. case AMF_DATE:
2652. snprintf(str, 255, "timestamp:%.2f", prop->p_vu.p_number);
2653. break;
2654. default:
2655. snprintf(str, 255, "INVALID TYPE 0x%02x",
2656. (unsigned char)prop->p_type);
2657. }
2658. if (prop->p_name.av_len)
2659. {
2660. /* chomp */
2661. if (strlen(str) >= 1 && str[strlen(str) - 1] == '\n')
2662. str[strlen(str) - 1] = '\0';
2663. RTMP_Log(RTMP_LOGINFO, " %-22.*s%s", prop->p_name.av_len,
2664. prop->p_name.av_val, str);
2665. }
2666. }
2667. else
2668. {
2669. if (prop->p_name.av_len)
2670. RTMP_Log(RTMP_LOGINFO, "%.*s:", prop->p_name.av_len, prop->p_name.av_val);
2671. DumpMetaData(&prop->p_vu.p_object);
2672. }
2673. }
2674. return FALSE;
2675. }
2677. SAVC(onMetaData);
2678. SAVC(duration);
2679. SAVC(video);
2680. SAVC(audio);
2682. static int
2683. HandleMetadata(RTMP *r, char *body, unsigned int len)
2684. {
2685. /* allright we get some info here, so parse it and print it */
2686. /* also keep duration or filesize to make a nice progress bar */
2688. AMFObject obj;
2689. AVal metastring;
2690. int ret = FALSE;
2692. int nRes = AMF_Decode(&obj, body, len, FALSE);
2693. if (nRes < 0)
2694. {
2695. RTMP_Log(RTMP_LOGERROR, "%s, error decoding meta data packet", __FUNCTION__);
2696. return FALSE;
2697. }
2699. AMF_Dump(&obj);
2700. AMFProp_GetString(AMF_GetProp(&obj, NULL, 0), &metastring);
2702. if (AVMATCH(&metastring, &av_onMetaData))
2703. {
2704. AMFObjectProperty prop;
2705. /* Show metadata */
2706. RTMP_Log(RTMP_LOGINFO, "Metadata:");
2707. DumpMetaData(&obj);
2708. if (RTMP_FindFirstMatchingProperty(&obj, &av_duration, &prop))
2709. {
2710. r->m_fDuration = prop.p_vu.p_number;
2711. /*RTMP_Log(RTMP_LOGDEBUG, "Set duration: %.2f", m_fDuration); */
2712. }
2713. /* Search for audio or video tags */
2714. if (RTMP_FindPrefixProperty(&obj, &av_video, &prop))
2715. r->m_read.dataType |= 1;
2716. if (RTMP_FindPrefixProperty(&obj, &av_audio, &prop))
2717. r->m_read.dataType |= 4;
2718. ret = TRUE;
2719. }
2720. AMF_Reset(&obj);
2721. return ret;
2722. }
2724. static void
2725. HandleChangeChunkSize(RTMP *r, const RTMPPacket *packet)
2726. {
2727. if (packet->m_nBodySize >= 4)
2728. {
2729. r->m_inChunkSize = AMF_DecodeInt32(packet->m_body);
2730. RTMP_Log(RTMP_LOGDEBUG, "%s, received: chunk size change to %d", __FUNCTION__,
2731. r->m_inChunkSize);
2732. }
2733. }
2735. static void
2736. HandleAudio(RTMP *r, const RTMPPacket *packet)
2737. {
2738. }
2740. static void
2741. HandleVideo(RTMP *r, const RTMPPacket *packet)
2742. {
2743. }
2745. static void
2746. HandleCtrl(RTMP *r, const RTMPPacket *packet)
2747. {
2748. short nType = -1;
2749. unsigned int tmp;
2750. if (packet->m_body && packet->m_nBodySize >= 2)
2751. nType = AMF_DecodeInt16(packet->m_body);
2752. RTMP_Log(RTMP_LOGDEBUG, "%s, received ctrl. type: %d, len: %d", __FUNCTION__, nType,
2753. packet->m_nBodySize);
2754. /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */
2756. if (packet->m_nBodySize >= 6)
2757. {
2758. switch (nType)
2759. {
2760. case 0:
2761. tmp = AMF_DecodeInt32(packet->m_body + 2);
2762. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream Begin %d", __FUNCTION__, tmp);
2763. break;
2765. case 1:
2766. tmp = AMF_DecodeInt32(packet->m_body + 2);
2767. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream EOF %d", __FUNCTION__, tmp);
2768. if (r->m_pausing == 1)
2769. r->m_pausing = 2;
2770. break;
2772. case 2:
2773. tmp = AMF_DecodeInt32(packet->m_body + 2);
2774. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream Dry %d", __FUNCTION__, tmp);
2775. break;
2777. case 4:
2778. tmp = AMF_DecodeInt32(packet->m_body + 2);
2779. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream IsRecorded %d", __FUNCTION__, tmp);
2780. break;
2782. case 6: /* server ping. reply with pong. */
2783. tmp = AMF_DecodeInt32(packet->m_body + 2);
2784. RTMP_Log(RTMP_LOGDEBUG, "%s, Ping %d", __FUNCTION__, tmp);
2785. RTMP_SendCtrl(r, 0x07, tmp, 0);
2786. break;
2788. /* FMS 3.5 servers send the following two controls to let the client
2789. * know when the server has sent a complete buffer. I.e., when the
2790. * server has sent an amount of data equal to m_nBufferMS in duration.
2791. * The server meters its output so that data arrives at the client
2792. * in realtime and no faster.
2793. *
2794. * The rtmpdump program tries to set m_nBufferMS as large as
2795. * possible, to force the server to send data as fast as possible.
2796. * In practice, the server appears to cap this at about 1 hour's
2797. * worth of data. After the server has sent a complete buffer, and
2798. * sends this BufferEmpty message, it will wait until the play
2799. * duration of that buffer has passed before sending a new buffer.
2800. * The BufferReady message will be sent when the new buffer starts.
2801. * (There is no BufferReady message for the very first buffer;
2802. * presumably the Stream Begin message is sufficient for that
2803. * purpose.)
2804. *
2805. * If the network speed is much faster than the data bitrate, then
2806. * there may be long delays between the end of one buffer and the
2807. * start of the next.
2808. *
2809. * Since usually the network allows data to be sent at
2810. * faster than realtime, and rtmpdump wants to download the data
2811. * as fast as possible, we use this RTMP_LF_BUFX hack: when we
2812. * get the BufferEmpty message, we send a Pause followed by an
2813. * Unpause. This causes the server to send the next buffer immediately
2814. * instead of waiting for the full duration to elapse. (That's
2815. * also the purpose of the ToggleStream function, which rtmpdump
2816. * calls if we get a read timeout.)
2817. *
2818. * Media player apps don't need this hack since they are just
2819. * going to play the data in realtime anyway. It also doesn't work
2820. * for live streams since they obviously can only be sent in
2821. * realtime. And it's all moot if the network speed is actually
2822. * slower than the media bitrate.
2823. */
2824. case 31:
2825. tmp = AMF_DecodeInt32(packet->m_body + 2);
2826. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream BufferEmpty %d", __FUNCTION__, tmp);
2827. if (!(r->Link.lFlags & RTMP_LF_BUFX))
2828. break;
2829. if (!r->m_pausing)
2830. {
2831. r->m_pauseStamp = r->m_channelTimestamp[r->m_mediaChannel];
2832. RTMP_SendPause(r, TRUE, r->m_pauseStamp);
2833. r->m_pausing = 1;
2834. }
2835. else if (r->m_pausing == 2)
2836. {
2837. RTMP_SendPause(r, FALSE, r->m_pauseStamp);
2838. r->m_pausing = 3;
2839. }
2840. break;
2842. case 32:
2843. tmp = AMF_DecodeInt32(packet->m_body + 2);
2844. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream BufferReady %d", __FUNCTION__, tmp);
2845. break;
2847. default:
2848. tmp = AMF_DecodeInt32(packet->m_body + 2);
2849. RTMP_Log(RTMP_LOGDEBUG, "%s, Stream xx %d", __FUNCTION__, tmp);
2850. break;
2851. }
2853. }
2855. if (nType == 0x1A)
2856. {
2857. RTMP_Log(RTMP_LOGDEBUG, "%s, SWFVerification ping received: ", __FUNCTION__);
2858. if (packet->m_nBodySize > 2 && packet->m_body[2] > 0x01)
2859. {
2860. RTMP_Log(RTMP_LOGERROR,
2861. "%s: SWFVerification Type %d request not supported! Patches welcome...",
2862. __FUNCTION__, packet->m_body[2]);
2863. }
2864. #ifdef CRYPTO
2865. /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */
2867. /* respond with HMAC SHA256 of decompressed SWF, key is the 30byte player key, also the last 30 bytes of the server handshake are applied */
2868. else if (r->Link.SWFSize)
2869. {
2870. RTMP_SendCtrl(r, 0x1B, 0, 0);
2871. }
2872. else
2873. {
2874. RTMP_Log(RTMP_LOGERROR,
2875. "%s: Ignoring SWFVerification request, use --swfVfy!",
2876. __FUNCTION__);
2877. }
2878. #else
2879. RTMP_Log(RTMP_LOGERROR,
2880. "%s: Ignoring SWFVerification request, no CRYPTO support!",
2881. __FUNCTION__);
2882. #endif
2883. }
2884. }
2886. static void
2887. HandleServerBW(RTMP *r, const RTMPPacket *packet)
2888. {
2889. r->m_nServerBW = AMF_DecodeInt32(packet->m_body);
2890. RTMP_Log(RTMP_LOGDEBUG, "%s: server BW = %d", __FUNCTION__, r->m_nServerBW);
2891. }
2893. static void
2894. HandleClientBW(RTMP *r, const RTMPPacket *packet)
2895. {
2896. r->m_nClientBW = AMF_DecodeInt32(packet->m_body);
2897. if (packet->m_nBodySize > 4)
2898. r->m_nClientBW2 = packet->m_body[4];
2899. else
2900. r->m_nClientBW2 = -1;
2901. RTMP_Log(RTMP_LOGDEBUG, "%s: client BW = %d %d", __FUNCTION__, r->m_nClientBW,
2902. r->m_nClientBW2);
2903. }
2905. static int
2906. DecodeInt32LE(const char *data)
2907. {
2908. unsigned char *c = (unsigned char *)data;
2909. unsigned int val;
2911. val = (c[3] << 24) | (c[2] << 16) | (c[1] << 8) | c[0];
2912. return val;
2913. }
2915. static int
2916. EncodeInt32LE(char *output, int nVal)
2917. {
2918. output[0] = nVal;
2919. nVal >>= 8;
2920. output[1] = nVal;
2921. nVal >>= 8;
2922. output[2] = nVal;
2923. nVal >>= 8;
2924. output[3] = nVal;
2925. return 4;
2926. }
2928. int
2929. RTMP_ReadPacket(RTMP *r, RTMPPacket *packet)
2930. {
2931. uint8_t hbuf[RTMP_MAX_HEADER_SIZE] = { 0 };
2932. char *header = (char *)hbuf;
2933. int nSize, hSize, nToRead, nChunk;
2934. int didAlloc = FALSE;
2936. RTMP_Log(RTMP_LOGDEBUG2, "%s: fd=%d", __FUNCTION__, r->m_sb.sb_socket);
2938. if (ReadN(r, (char *)hbuf, 1) == 0)
2939. {
2940. RTMP_Log(RTMP_LOGERROR, "%s, failed to read RTMP packet header", __FUNCTION__);
2941. return FALSE;
2942. }
2944. packet->m_headerType = (hbuf[0] & 0xc0) >> 6;
2945. packet->m_nChannel = (hbuf[0] & 0x3f);
2946. header++;
2947. if (packet->m_nChannel == 0)
2948. {
2949. if (ReadN(r, (char *)&hbuf[1], 1) != 1)
2950. {
2951. RTMP_Log(RTMP_LOGERROR, "%s, failed to read RTMP packet header 2nd byte",
2952. __FUNCTION__);
2953. return FALSE;
2954. }
2955. packet->m_nChannel = hbuf[1];
2956. packet->m_nChannel += 64;
2957. header++;
2958. }
2959. else if (packet->m_nChannel == 1)
2960. {
2961. int tmp;
2962. if (ReadN(r, (char *)&hbuf[1], 2) != 2)
2963. {
2964. RTMP_Log(RTMP_LOGERROR, "%s, failed to read RTMP packet header 3nd byte",
2965. __FUNCTION__);
2966. return FALSE;
2967. }
2968. tmp = (hbuf[2] << 8) + hbuf[1];
2969. packet->m_nChannel = tmp + 64;
2970. RTMP_Log(RTMP_LOGDEBUG, "%s, m_nChannel: %0x", __FUNCTION__, packet->m_nChannel);
2971. header += 2;
2972. }
2974. nSize = packetSize[packet->m_headerType];
2976. if (nSize == RTMP_LARGE_HEADER_SIZE) /* if we get a full header the timestamp is absolute */
2977. packet->m_hasAbsTimestamp = TRUE;
2979. else if (nSize < RTMP_LARGE_HEADER_SIZE)
2980. { /* using values from the last message of this channel */
2981. if (r->m_vecChannelsIn[packet->m_nChannel])
2982. memcpy(packet, r->m_vecChannelsIn[packet->m_nChannel],
2983. sizeof(RTMPPacket));
2984. }
2986. nSize--;
2988. if (nSize > 0 && ReadN(r, header, nSize) != nSize)
2989. {
2990. RTMP_Log(RTMP_LOGERROR, "%s, failed to read RTMP packet header. type: %x",
2991. __FUNCTION__, (unsigned int)hbuf[0]);
2992. return FALSE;
2993. }
2995. hSize = nSize + (header - (char *)hbuf);
2997. if (nSize >= 3)
2998. {
2999. packet->m_nTimeStamp = AMF_DecodeInt24(header);
3001. /*RTMP_Log(RTMP_LOGDEBUG, "%s, reading RTMP packet chunk on channel %x, headersz %i, timestamp %i, abs timestamp %i", __FUNCTION__, packet.m_nChannel, nSize, packet.m_nTimeStamp, packet.m_hasAbsTimestamp); */
3003. if (nSize >= 6)
3004. {
3005. packet->m_nBodySize = AMF_DecodeInt24(header + 3);
3006. packet->m_nBytesRead = 0;
3007. RTMPPacket_Free(packet);
3009. if (nSize > 6)
3010. {
3011. packet->m_packetType = header[6];
3013. if (nSize == 11)
3014. packet->m_nInfoField2 = DecodeInt32LE(header + 7);
3015. }
3016. }
3017. if (packet->m_nTimeStamp == 0xffffff)
3018. {
3019. if (ReadN(r, header + nSize, 4) != 4)
3020. {
3021. RTMP_Log(RTMP_LOGERROR, "%s, failed to read extended timestamp",
3022. __FUNCTION__);
3023. return FALSE;
3024. }
3025. packet->m_nTimeStamp = AMF_DecodeInt32(header + nSize);
3026. hSize += 4;
3027. }
3028. }
3030. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)hbuf, hSize);
3032. if (packet->m_nBodySize > 0 && packet->m_body == NULL)
3033. {
3034. if (!RTMPPacket_Alloc(packet, packet->m_nBodySize))
3035. {
3036. RTMP_Log(RTMP_LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__);
3037. return FALSE;
3038. }
3039. didAlloc = TRUE;
3040. packet->m_headerType = (hbuf[0] & 0xc0) >> 6;
3041. }
3043. nToRead = packet->m_nBodySize - packet->m_nBytesRead;
3044. nChunk = r->m_inChunkSize;
3045. if (nToRead < nChunk)
3046. nChunk = nToRead;
3048. /* Does the caller want the raw chunk? */
3049. if (packet->m_chunk)
3050. {
3051. packet->m_chunk->c_headerSize = hSize;
3052. memcpy(packet->m_chunk->c_header, hbuf, hSize);
3053. packet->m_chunk->c_chunk = packet->m_body + packet->m_nBytesRead;
3054. packet->m_chunk->c_chunkSize = nChunk;
3055. }
3057. if (ReadN(r, packet->m_body + packet->m_nBytesRead, nChunk) != nChunk)
3058. {
3059. RTMP_Log(RTMP_LOGERROR, "%s, failed to read RTMP packet body. len: %u",
3060. __FUNCTION__, packet->m_nBodySize);
3061. return FALSE;
3062. }
3064. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)packet->m_body + packet->m_nBytesRead, nChunk);
3066. packet->m_nBytesRead += nChunk;
3068. /* keep the packet as ref for other packets on this channel */
3069. if (!r->m_vecChannelsIn[packet->m_nChannel])
3070. r->m_vecChannelsIn[packet->m_nChannel] = malloc(sizeof(RTMPPacket));
3071. memcpy(r->m_vecChannelsIn[packet->m_nChannel], packet, sizeof(RTMPPacket));
3073. if (RTMPPacket_IsReady(packet))
3074. {
3075. /* make packet's timestamp absolute */
3076. if (!packet->m_hasAbsTimestamp)
3077. packet->m_nTimeStamp += r->m_channelTimestamp[packet->m_nChannel]; /* timestamps seem to be always relative!! */
3079. r->m_channelTimestamp[packet->m_nChannel] = packet->m_nTimeStamp;
3081. /* reset the data from the stored packet. we keep the header since we may use it later if a new packet for this channel */
3082. /* arrives and requests to re-use some info (small packet header) */
3083. r->m_vecChannelsIn[packet->m_nChannel]->m_body = NULL;
3084. r->m_vecChannelsIn[packet->m_nChannel]->m_nBytesRead = 0;
3085. r->m_vecChannelsIn[packet->m_nChannel]->m_hasAbsTimestamp = FALSE; /* can only be false if we reuse header */
3086. }
3087. else
3088. {
3089. packet->m_body = NULL; /* so it won't be erased on free */
3090. }
3092. return TRUE;
3093. }
3095. #ifndef CRYPTO
3096. static int
3097. HandShake(RTMP *r, int FP9HandShake)
3098. {
3099. int i;
3100. uint32_t uptime, suptime;
3101. int bMatch;
3102. char type;
3103. char clientbuf[RTMP_SIG_SIZE + 1], *clientsig = clientbuf + 1;
3104. char serversig[RTMP_SIG_SIZE];
3106. clientbuf[0] = 0x03; /* not encrypted */
3108. uptime = htonl(RTMP_GetTime());
3109. memcpy(clientsig, &uptime, 4);
3111. memset(&clientsig[4], 0, 4);
3113. #ifdef _DEBUG
3114. for (i = 8; i < RTMP_SIG_SIZE; i++)
3115. clientsig[i] = 0xff;
3116. #else
3117. for (i = 8; i < RTMP_SIG_SIZE; i++)
3118. clientsig[i] = (char)(rand() % 256);
3119. #endif
3121. if (!WriteN(r, clientbuf, RTMP_SIG_SIZE + 1))
3122. return FALSE;
3124. if (ReadN(r, &type, 1) != 1) /* 0x03 or 0x06 */
3125. return FALSE;
3127. RTMP_Log(RTMP_LOGDEBUG, "%s: Type Answer : %02X", __FUNCTION__, type);
3129. if (type != clientbuf[0])
3130. RTMP_Log(RTMP_LOGWARNING, "%s: Type mismatch: client sent %d, server answered %d",
3131. __FUNCTION__, clientbuf[0], type);
3133. if (ReadN(r, serversig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE)
3134. return FALSE;
3136. /* decode server response */
3138. memcpy(&suptime, serversig, 4);
3139. suptime = ntohl(suptime);
3141. RTMP_Log(RTMP_LOGDEBUG, "%s: Server Uptime : %d", __FUNCTION__, suptime);
3142. RTMP_Log(RTMP_LOGDEBUG, "%s: FMS Version : %d.%d.%d.%d", __FUNCTION__,
3143. serversig[4], serversig[5], serversig[6], serversig[7]);
3145. /* 2nd part of handshake */
3146. if (!WriteN(r, serversig, RTMP_SIG_SIZE))
3147. return FALSE;
3149. if (ReadN(r, serversig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE)
3150. return FALSE;
3152. bMatch = (memcmp(serversig, clientsig, RTMP_SIG_SIZE) == 0);
3153. if (!bMatch)
3154. {
3155. RTMP_Log(RTMP_LOGWARNING, "%s, client signature does not match!", __FUNCTION__);
3156. }
3157. return TRUE;
3158. }
3160. static int
3161. SHandShake(RTMP *r)
3162. {
3163. int i;
3164. char serverbuf[RTMP_SIG_SIZE + 1], *serversig = serverbuf + 1;
3165. char clientsig[RTMP_SIG_SIZE];
3166. uint32_t uptime;
3167. int bMatch;
3169. if (ReadN(r, serverbuf, 1) != 1) /* 0x03 or 0x06 */
3170. return FALSE;
3172. RTMP_Log(RTMP_LOGDEBUG, "%s: Type Request : %02X", __FUNCTION__, serverbuf[0]);
3174. if (serverbuf[0] != 3)
3175. {
3176. RTMP_Log(RTMP_LOGERROR, "%s: Type unknown: client sent %02X",
3177. __FUNCTION__, serverbuf[0]);
3178. return FALSE;
3179. }
3181. uptime = htonl(RTMP_GetTime());
3182. memcpy(serversig, &uptime, 4);
3184. memset(&serversig[4], 0, 4);
3185. #ifdef _DEBUG
3186. for (i = 8; i < RTMP_SIG_SIZE; i++)
3187. serversig[i] = 0xff;
3188. #else
3189. for (i = 8; i < RTMP_SIG_SIZE; i++)
3190. serversig[i] = (char)(rand() % 256);
3191. #endif
3193. if (!WriteN(r, serverbuf, RTMP_SIG_SIZE + 1))
3194. return FALSE;
3196. if (ReadN(r, clientsig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE)
3197. return FALSE;
3199. /* decode client response */
3201. memcpy(&uptime, clientsig, 4);
3202. uptime = ntohl(uptime);
3204. RTMP_Log(RTMP_LOGDEBUG, "%s: Client Uptime : %d", __FUNCTION__, uptime);
3205. RTMP_Log(RTMP_LOGDEBUG, "%s: Player Version: %d.%d.%d.%d", __FUNCTION__,
3206. clientsig[4], clientsig[5], clientsig[6], clientsig[7]);
3208. /* 2nd part of handshake */
3209. if (!WriteN(r, clientsig, RTMP_SIG_SIZE))
3210. return FALSE;
3212. if (ReadN(r, clientsig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE)
3213. return FALSE;
3215. bMatch = (memcmp(serversig, clientsig, RTMP_SIG_SIZE) == 0);
3216. if (!bMatch)
3217. {
3218. RTMP_Log(RTMP_LOGWARNING, "%s, client signature does not match!", __FUNCTION__);
3219. }
3220. return TRUE;
3221. }
3222. #endif
3224. int
3225. RTMP_SendChunk(RTMP *r, RTMPChunk *chunk)
3226. {
3227. int wrote;
3228. char hbuf[RTMP_MAX_HEADER_SIZE];
3230. RTMP_Log(RTMP_LOGDEBUG2, "%s: fd=%d, size=%d", __FUNCTION__, r->m_sb.sb_socket,
3231. chunk->c_chunkSize);
3232. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)chunk->c_header, chunk->c_headerSize);
3233. if (chunk->c_chunkSize)
3234. {
3235. char *ptr = chunk->c_chunk - chunk->c_headerSize;
3236. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)chunk->c_chunk, chunk->c_chunkSize);
3237. /* save header bytes we're about to overwrite */
3238. memcpy(hbuf, ptr, chunk->c_headerSize);
3239. memcpy(ptr, chunk->c_header, chunk->c_headerSize);
3240. wrote = WriteN(r, ptr, chunk->c_headerSize + chunk->c_chunkSize);
3241. memcpy(ptr, hbuf, chunk->c_headerSize);
3242. }
3243. else
3244. wrote = WriteN(r, chunk->c_header, chunk->c_headerSize);
3245. return wrote;
3246. }
3248. int
3249. RTMP_SendPacket(RTMP *r, RTMPPacket *packet, int queue)
3250. {
3251. const RTMPPacket *prevPacket = r->m_vecChannelsOut[packet->m_nChannel];
3252. uint32_t last = 0;
3253. int nSize;
3254. int hSize, cSize;
3255. char *header, *hptr, *hend, hbuf[RTMP_MAX_HEADER_SIZE], c;
3256. uint32_t t;
3257. char *buffer, *tbuf = NULL, *toff = NULL;
3258. int nChunkSize;
3259. int tlen;
3261. if (prevPacket && packet->m_headerType != RTMP_PACKET_SIZE_LARGE)
3262. {
3263. /* compress a bit by using the prev packet's attributes */
3264. if (prevPacket->m_nBodySize == packet->m_nBodySize
3265. && prevPacket->m_packetType == packet->m_packetType
3266. && packet->m_headerType == RTMP_PACKET_SIZE_MEDIUM)
3267. packet->m_headerType = RTMP_PACKET_SIZE_SMALL;
3269. if (prevPacket->m_nTimeStamp == packet->m_nTimeStamp
3270. && packet->m_headerType == RTMP_PACKET_SIZE_SMALL)
3271. packet->m_headerType = RTMP_PACKET_SIZE_MINIMUM;
3272. last = prevPacket->m_nTimeStamp;
3273. }
3275. if (packet->m_headerType > 3) /* sanity */
3276. {
3277. RTMP_Log(RTMP_LOGERROR, "sanity failed!! trying to send header of type: 0x%02x.",
3278. (unsigned char)packet->m_headerType);
3279. return FALSE;
3280. }
3282. nSize = packetSize[packet->m_headerType];
3283. hSize = nSize; cSize = 0;
3284. t = packet->m_nTimeStamp - last;
3286. if (packet->m_body)
3287. {
3288. header = packet->m_body - nSize;
3289. hend = packet->m_body;
3290. }
3291. else
3292. {
3293. header = hbuf + 6;
3294. hend = hbuf + sizeof(hbuf);
3295. }
3297. if (packet->m_nChannel > 319)
3298. cSize = 2;
3299. else if (packet->m_nChannel > 63)
3300. cSize = 1;
3301. if (cSize)
3302. {
3303. header -= cSize;
3304. hSize += cSize;
3305. }
3307. if (nSize > 1 && t >= 0xffffff)
3308. {
3309. header -= 4;
3310. hSize += 4;
3311. }
3313. hptr = header;
3314. c = packet->m_headerType << 6;
3315. switch (cSize)
3316. {
3317. case 0:
3318. c |= packet->m_nChannel;
3319. break;
3320. case 1:
3321. break;
3322. case 2:
3323. c |= 1;
3324. break;
3325. }
3326. *hptr++ = c;
3327. if (cSize)
3328. {
3329. int tmp = packet->m_nChannel - 64;
3330. *hptr++ = tmp & 0xff;
3331. if (cSize == 2)
3332. *hptr++ = tmp >> 8;
3333. }
3335. if (nSize > 1)
3336. {
3337. hptr = AMF_EncodeInt24(hptr, hend, t > 0xffffff ? 0xffffff : t);
3338. }
3340. if (nSize > 4)
3341. {
3342. hptr = AMF_EncodeInt24(hptr, hend, packet->m_nBodySize);
3343. *hptr++ = packet->m_packetType;
3344. }
3346. if (nSize > 8)
3347. hptr += EncodeInt32LE(hptr, packet->m_nInfoField2);
3349. if (nSize > 1 && t >= 0xffffff)
3350. hptr = AMF_EncodeInt32(hptr, hend, t);
3352. nSize = packet->m_nBodySize;
3353. buffer = packet->m_body;
3354. nChunkSize = r->m_outChunkSize;
3356. RTMP_Log(RTMP_LOGDEBUG2, "%s: fd=%d, size=%d", __FUNCTION__, r->m_sb.sb_socket,
3357. nSize);
3358. /* send all chunks in one HTTP request */
3359. if (r->Link.protocol & RTMP_FEATURE_HTTP)
3360. {
3361. int chunks = (nSize+nChunkSize-1) / nChunkSize;
3362. if (chunks > 1)
3363. {
3364. tlen = chunks * (cSize + 1) + nSize + hSize;
3365. tbuf = malloc(tlen);
3366. if (!tbuf)
3367. return FALSE;
3368. toff = tbuf;
3369. }
3370. }
3371. while (nSize + hSize)
3372. {
3373. int wrote;
3375. if (nSize < nChunkSize)
3376. nChunkSize = nSize;
3378. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)header, hSize);
3379. RTMP_LogHexString(RTMP_LOGDEBUG2, (uint8_t *)buffer, nChunkSize);
3380. if (tbuf)
3381. {
3382. memcpy(toff, header, nChunkSize + hSize);
3383. toff += nChunkSize + hSize;
3384. }
3385. else
3386. {
3387. wrote = WriteN(r, header, nChunkSize + hSize);
3388. if (!wrote)
3389. return FALSE;
3390. }
3391. nSize -= nChunkSize;
3392. buffer += nChunkSize;
3393. hSize = 0;
3395. if (nSize > 0)
3396. {
3397. header = buffer - 1;
3398. hSize = 1;
3399. if (cSize)
3400. {
3401. header -= cSize;
3402. hSize += cSize;
3403. }
3404. *header = (0xc0 | c);
3405. if (cSize)
3406. {
3407. int tmp = packet->m_nChannel - 64;
3408. header[1] = tmp & 0xff;
3409. if (cSize == 2)
3410. header[2] = tmp >> 8;
3411. }
3412. }
3413. }
3414. if (tbuf)
3415. {
3416. int wrote = WriteN(r, tbuf, toff-tbuf);
3417. free(tbuf);
3418. tbuf = NULL;
3419. if (!wrote)
3420. return FALSE;
3421. }
3423. /* we invoked a remote method */
3424. if (packet->m_packetType == RTMP_PACKET_TYPE_INVOKE)
3425. {
3426. AVal method;
3427. char *ptr;
3428. ptr = packet->m_body + 1;
3429. AMF_DecodeString(ptr, &method);
3430. RTMP_Log(RTMP_LOGDEBUG, "Invoking %s", method.av_val);
3431. /* keep it in call queue till result arrives */
3432. if (queue) {
3433. int txn;
3434. ptr += 3 + method.av_len;
3435. txn = (int)AMF_DecodeNumber(ptr);
3436. AV_queue(&r->m_methodCalls, &r->m_numCalls, &method, txn);
3437. }
3438. }
3440. if (!r->m_vecChannelsOut[packet->m_nChannel])
3441. r->m_vecChannelsOut[packet->m_nChannel] = malloc(sizeof(RTMPPacket));
3442. memcpy(r->m_vecChannelsOut[packet->m_nChannel], packet, sizeof(RTMPPacket));
3443. return TRUE;
3444. }
3446. int
3447. RTMP_Serve(RTMP *r)
3448. {
3449. return SHandShake(r);
3450. }
3452. void
3453. RTMP_Close(RTMP *r)
3454. {
3455. int i;
3457. if (RTMP_IsConnected(r))
3458. {
3459. if (r->m_stream_id > 0)
3460. {
3461. i = r->m_stream_id;
3462. r->m_stream_id = 0;
3463. if ((r->Link.protocol & RTMP_FEATURE_WRITE))
3464. SendFCUnpublish(r);
3465. SendDeleteStream(r, i);
3466. }
3467. if (r->m_clientID.av_val)
3468. {
3469. HTTP_Post(r, RTMPT_CLOSE, "", 1);
3470. free(r->m_clientID.av_val);
3471. r->m_clientID.av_val = NULL;
3472. r->m_clientID.av_len = 0;
3473. }
3474. RTMPSockBuf_Close(&r->m_sb);
3475. }
3477. r->m_stream_id = -1;
3478. r->m_sb.sb_socket = -1;
3479. r->m_nBWCheckCounter = 0;
3480. r->m_nBytesIn = 0;
3481. r->m_nBytesInSent = 0;
3483. if (r->m_read.flags & RTMP_READ_HEADER) {
3484. free(r->m_read.buf);
3485. r->m_read.buf = NULL;
3486. }
3487. r->m_read.dataType = 0;
3488. r->m_read.flags = 0;
3489. r->m_read.status = 0;
3490. r->m_read.nResumeTS = 0;
3491. r->m_read.nIgnoredFrameCounter = 0;
3492. r->m_read.nIgnoredFlvFrameCounter = 0;
3494. r->m_write.m_nBytesRead = 0;
3495. RTMPPacket_Free(&r->m_write);
3497. for (i = 0; i < RTMP_CHANNELS; i++)
3498. {
3499. if (r->m_vecChannelsIn[i])
3500. {
3501. RTMPPacket_Free(r->m_vecChannelsIn[i]);
3502. free(r->m_vecChannelsIn[i]);
3503. r->m_vecChannelsIn[i] = NULL;
3504. }
3505. if (r->m_vecChannelsOut[i])
3506. {
3507. free(r->m_vecChannelsOut[i]);
3508. r->m_vecChannelsOut[i] = NULL;
3509. }
3510. }
3511. AV_clear(r->m_methodCalls, r->m_numCalls);
3512. r->m_methodCalls = NULL;
3513. r->m_numCalls = 0;
3514. r->m_numInvokes = 0;
3516. r->m_bPlaying = FALSE;
3517. r->m_sb.sb_size = 0;
3519. r->m_msgCounter = 0;
3520. r->m_resplen = 0;
3521. r->m_unackd = 0;
3523. free(r->Link.playpath0.av_val);
3524. r->Link.playpath0.av_val = NULL;
3526. if (r->Link.lFlags & RTMP_LF_FTCU)
3527. {
3528. free(r->Link.tcUrl.av_val);
3529. r->Link.tcUrl.av_val = NULL;
3530. r->Link.lFlags ^= RTMP_LF_FTCU;
3531. }
3533. #ifdef CRYPTO
3534. if (r->Link.dh)
3535. {
3536. MDH_free(r->Link.dh);
3537. r->Link.dh = NULL;
3538. }
3539. if (r->Link.rc4keyIn)
3540. {
3541. RC4_free(r->Link.rc4keyIn);
3542. r->Link.rc4keyIn = NULL;
3543. }
3544. if (r->Link.rc4keyOut)
3545. {
3546. RC4_free(r->Link.rc4keyOut);
3547. r->Link.rc4keyOut = NULL;
3548. }
3549. #endif
3550. }
3552. int
3553. RTMPSockBuf_Fill(RTMPSockBuf *sb)
3554. {
3555. int nBytes;
3557. if (!sb->sb_size)
3558. sb->sb_start = sb->sb_buf;
3560. while (1)
3561. {
3562. nBytes = sizeof(sb->sb_buf) - sb->sb_size - (sb->sb_start - sb->sb_buf);
3563. #if defined(CRYPTO) && !defined(NO_SSL)
3564. if (sb->sb_ssl)
3565. {
3566. nBytes = TLS_read(sb->sb_ssl, sb->sb_start + sb->sb_size, nBytes);
3567. }
3568. else
3569. #endif
3570. {
3571. nBytes = recv(sb->sb_socket, sb->sb_start + sb->sb_size, nBytes, 0);
3572. }
3573. if (nBytes != -1)
3574. {
3575. sb->sb_size += nBytes;
3576. }
3577. else
3578. {
3579. int sockerr = GetSockError();
3580. RTMP_Log(RTMP_LOGDEBUG, "%s, recv returned %d. GetSockError(): %d (%s)",
3581. __FUNCTION__, nBytes, sockerr, strerror(sockerr));
3582. if (sockerr == EINTR && !RTMP_ctrlC)
3583. continue;
3585. if (sockerr == EWOULDBLOCK || sockerr == EAGAIN)
3586. {
3587. sb->sb_timedout = TRUE;
3588. nBytes = 0;
3589. }
3590. }
3591. break;
3592. }
3594. return nBytes;
3595. }
3597. int
3598. RTMPSockBuf_Send(RTMPSockBuf *sb, const char *buf, int len)
3599. {
3600. int rc;
3602. #ifdef _DEBUG
3603. fwrite(buf, 1, len, netstackdump);
3604. #endif
3606. #if defined(CRYPTO) && !defined(NO_SSL)
3607. if (sb->sb_ssl)
3608. {
3609. rc = TLS_write(sb->sb_ssl, buf, len);
3610. }
3611. else
3612. #endif
3613. {
3614. rc = send(sb->sb_socket, buf, len, 0);
3615. }
3616. return rc;
3617. }
3619. int
3620. RTMPSockBuf_Close(RTMPSockBuf *sb)
3621. {
3622. #if defined(CRYPTO) && !defined(NO_SSL)
3623. if (sb->sb_ssl)
3624. {
3625. TLS_shutdown(sb->sb_ssl);
3626. TLS_close(sb->sb_ssl);
3627. sb->sb_ssl = NULL;
3628. }
3629. #endif
3630. if (sb->sb_socket != -1)
3631. return closesocket(sb->sb_socket);
3632. return 0;
3633. }
3635. #define HEX2BIN(a) (((a)&0x40)?((a)&0xf)+9:((a)&0xf))
3637. static void
3638. DecodeTEA(AVal *key, AVal *text)
3639. {
3640. uint32_t *v, k[4] = { 0 }, u;
3641. uint32_t z, y, sum = 0, e, DELTA = 0x9e3779b9;
3642. int32_t p, q;
3643. int i, n;
3644. unsigned char *ptr, *out;
3646. /* prep key: pack 1st 16 chars into 4 LittleEndian ints */
3647. ptr = (unsigned char *)key->av_val;
3648. u = 0;
3649. n = 0;
3650. v = k;
3651. p = key->av_len > 16 ? 16 : key->av_len;
3652. for (i = 0; i < p; i++)
3653. {
3654. u |= ptr[i] << (n * 8);
3655. if (n == 3)
3656. {
3657. *v++ = u;
3658. u = 0;
3659. n = 0;
3660. }
3661. else
3662. {
3663. n++;
3664. }
3665. }
3666. /* any trailing chars */
3667. if (u)
3668. *v = u;
3670. /* prep text: hex2bin, multiples of 4 */
3671. n = (text->av_len + 7) / 8;
3672. out = malloc(n * 8);
3673. ptr = (unsigned char *)text->av_val;
3674. v = (uint32_t *) out;
3675. for (i = 0; i < n; i++)
3676. {
3677. u = (HEX2BIN(ptr[0]) << 4) + HEX2BIN(ptr[1]);
3678. u |= ((HEX2BIN(ptr[2]) << 4) + HEX2BIN(ptr[3])) << 8;
3679. u |= ((HEX2BIN(ptr[4]) << 4) + HEX2BIN(ptr[5])) << 16;
3680. u |= ((HEX2BIN(ptr[6]) << 4) + HEX2BIN(ptr[7])) << 24;
3681. *v++ = u;
3682. ptr += 8;
3683. }
3684. v = (uint32_t *) out;
3686. /* http://www.movable-type.co.uk/scripts/tea-block.html */
3687. #define MX (((z>>5)^(y<<2)) + ((y>>3)^(z<<4))) ^ ((sum^y) + (k[(p&3)^e]^z));
3688. z = v[n - 1];
3689. y = v[0];
3690. q = 6 + 52 / n;
3691. sum = q * DELTA;
3692. while (sum != 0)
3693. {
3694. e = sum >> 2 & 3;
3695. for (p = n - 1; p > 0; p--)
3696. z = v[p - 1], y = v[p] -= MX;
3697. z = v[n - 1];
3698. y = v[0] -= MX;
3699. sum -= DELTA;
3700. }
3702. text->av_len /= 2;
3703. memcpy(text->av_val, out, text->av_len);
3704. free(out);
3705. }
3707. static int
3708. HTTP_Post(RTMP *r, RTMPTCmd cmd, const char *buf, int len)
3709. {
3710. char hbuf[512];
3711. int hlen = snprintf(hbuf, sizeof(hbuf), "POST /%s%s/%d HTTP/1.1\r\n"
3712. "Host: %.*s:%d\r\n"
3713. "Accept: */*\r\n"
3714. "User-Agent: Shockwave Flash\n"
3715. "Connection: Keep-Alive\n"
3716. "Cache-Control: no-cache\r\n"
3717. "Content-type: application/x-fcs\r\n"
3718. "Content-length: %d\r\n\r\n", RTMPT_cmds[cmd],
3719. r->m_clientID.av_val ? r->m_clientID.av_val : "",
3720. r->m_msgCounter, r->Link.hostname.av_len, r->Link.hostname.av_val,
3721. r->Link.port, len);
3722. RTMPSockBuf_Send(&r->m_sb, hbuf, hlen);
3723. hlen = RTMPSockBuf_Send(&r->m_sb, buf, len);
3724. r->m_msgCounter++;
3725. r->m_unackd++;
3726. return hlen;
3727. }
3729. static int
3730. HTTP_read(RTMP *r, int fill)
3731. {
3732. char *ptr;
3733. int hlen;
3735. if (fill)
3736. RTMPSockBuf_Fill(&r->m_sb);
3737. if (r->m_sb.sb_size < 144)
3738. return -2;
3739. if (strncmp(r->m_sb.sb_start, "HTTP/1.1 200 ", 13))
3740. return -1;
3741. ptr = r->m_sb.sb_start + sizeof("HTTP/1.1 200");
3742. while ((ptr = strstr(ptr, "Content-"))) {
3743. if (!strncasecmp(ptr+8, "length:", 7)) break;
3744. ptr += 8;
3745. }
3746. if (!ptr)
3747. return -1;
3748. hlen = atoi(ptr+16);
3749. ptr = strstr(ptr+16, "\r\n\r\n");
3750. if (!ptr)
3751. return -1;
3752. ptr += 4;
3753. r->m_sb.sb_size -= ptr - r->m_sb.sb_start;
3754. r->m_sb.sb_start = ptr;
3755. r->m_unackd--;
3757. if (!r->m_clientID.av_val)
3758. {
3759. r->m_clientID.av_len = hlen;
3760. r->m_clientID.av_val = malloc(hlen+1);
3761. if (!r->m_clientID.av_val)
3762. return -1;
3763. r->m_clientID.av_val[0] = '/';
3764. memcpy(r->m_clientID.av_val+1, ptr, hlen-1);
3765. r->m_clientID.av_val[hlen] = 0;
3766. r->m_sb.sb_size = 0;
3767. }
3768. else
3769. {
3770. r->m_polling = *ptr++;
3771. r->m_resplen = hlen - 1;
3772. r->m_sb.sb_start++;
3773. r->m_sb.sb_size--;
3774. }
3775. return 0;
3776. }
3778. #define MAX_IGNORED_FRAMES 50
3780. /* Read from the stream until we get a media packet.
3781. * Returns -3 if Play.Close/Stop, -2 if fatal error, -1 if no more media
3782. * packets, 0 if ignorable error, >0 if there is a media packet
3783. */
3784. static int
3785. Read_1_Packet(RTMP *r, char *buf, unsigned int buflen)
3786. {
3787. uint32_t prevTagSize = 0;
3788. int rtnGetNextMediaPacket = 0, ret = RTMP_READ_EOF;
3789. RTMPPacket packet = { 0 };
3790. int recopy = FALSE;
3791. unsigned int size;
3792. char *ptr, *pend;
3793. uint32_t nTimeStamp = 0;
3794. unsigned int len;
3796. rtnGetNextMediaPacket = RTMP_GetNextMediaPacket(r, &packet);
3797. while (rtnGetNextMediaPacket)
3798. {
3799. char *packetBody = packet.m_body;
3800. unsigned int nPacketLen = packet.m_nBodySize;
3802. /* Return RTMP_READ_COMPLETE if this was completed nicely with
3803. * invoke message Play.Stop or Play.Complete
3804. */
3805. if (rtnGetNextMediaPacket == 2)
3806. {
3807. RTMP_Log(RTMP_LOGDEBUG,
3808. "Got Play.Complete or Play.Stop from server. "
3809. "Assuming stream is complete");
3810. ret = RTMP_READ_COMPLETE;
3811. break;
3812. }
3814. r->m_read.dataType |= (((packet.m_packetType == RTMP_PACKET_TYPE_AUDIO) << 2) |
3815. (packet.m_packetType == RTMP_PACKET_TYPE_VIDEO));
3817. if (packet.m_packetType == RTMP_PACKET_TYPE_VIDEO && nPacketLen <= 5)
3818. {
3819. RTMP_Log(RTMP_LOGDEBUG, "ignoring too small video packet: size: %d",
3820. nPacketLen);
3821. ret = RTMP_READ_IGNORE;
3822. break;
3823. }
3824. if (packet.m_packetType == RTMP_PACKET_TYPE_AUDIO && nPacketLen <= 1)
3825. {
3826. RTMP_Log(RTMP_LOGDEBUG, "ignoring too small audio packet: size: %d",
3827. nPacketLen);
3828. ret = RTMP_READ_IGNORE;
3829. break;
3830. }
3832. if (r->m_read.flags & RTMP_READ_SEEKING)
3833. {
3834. ret = RTMP_READ_IGNORE;
3835. break;
3836. }
3837. #ifdef _DEBUG
3838. RTMP_Log(RTMP_LOGDEBUG, "type: %02X, size: %d, TS: %d ms, abs TS: %d",
3839. packet.m_packetType, nPacketLen, packet.m_nTimeStamp,
3840. packet.m_hasAbsTimestamp);
3841. if (packet.m_packetType == RTMP_PACKET_TYPE_VIDEO)
3842. RTMP_Log(RTMP_LOGDEBUG, "frametype: %02X", (*packetBody & 0xf0));
3843. #endif
3845. if (r->m_read.flags & RTMP_READ_RESUME)
3846. {
3847. /* check the header if we get one */
3848. if (packet.m_nTimeStamp == 0)
3849. {
3850. if (r->m_read.nMetaHeaderSize > 0
3851. && packet.m_packetType == RTMP_PACKET_TYPE_INFO)
3852. {
3853. AMFObject metaObj;
3854. int nRes =
3855. AMF_Decode(&metaObj, packetBody, nPacketLen, FALSE);
3856. if (nRes >= 0)
3857. {
3858. AVal metastring;
3859. AMFProp_GetString(AMF_GetProp(&metaObj, NULL, 0),
3860. &metastring);
3862. if (AVMATCH(&metastring, &av_onMetaData))
3863. {
3864. /* compare */
3865. if ((r->m_read.nMetaHeaderSize != nPacketLen) ||
3866. (memcmp
3867. (r->m_read.metaHeader, packetBody,
3868. r->m_read.nMetaHeaderSize) != 0))
3869. {
3870. ret = RTMP_READ_ERROR;
3871. }
3872. }
3873. AMF_Reset(&metaObj);
3874. if (ret == RTMP_READ_ERROR)
3875. break;
3876. }
3877. }
3879. /* check first keyframe to make sure we got the right position
3880. * in the stream! (the first non ignored frame)
3881. */
3882. if (r->m_read.nInitialFrameSize > 0)
3883. {
3884. /* video or audio data */
3885. if (packet.m_packetType == r->m_read.initialFrameType
3886. && r->m_read.nInitialFrameSize == nPacketLen)
3887. {
3888. /* we don't compare the sizes since the packet can
3889. * contain several FLV packets, just make sure the
3890. * first frame is our keyframe (which we are going
3891. * to rewrite)
3892. */
3893. if (memcmp
3894. (r->m_read.initialFrame, packetBody,
3895. r->m_read.nInitialFrameSize) == 0)
3896. {
3897. RTMP_Log(RTMP_LOGDEBUG, "Checked keyframe successfully!");
3898. r->m_read.flags |= RTMP_READ_GOTKF;
3899. /* ignore it! (what about audio data after it? it is
3900. * handled by ignoring all 0ms frames, see below)
3901. */
3902. ret = RTMP_READ_IGNORE;
3903. break;
3904. }
3905. }
3907. /* hande FLV streams, even though the server resends the
3908. * keyframe as an extra video packet it is also included
3909. * in the first FLV stream chunk and we have to compare
3910. * it and filter it out !!
3911. */
3912. if (packet.m_packetType == RTMP_PACKET_TYPE_FLASH_VIDEO)
3913. {
3914. /* basically we have to find the keyframe with the
3915. * correct TS being nResumeTS
3916. */
3917. unsigned int pos = 0;
3918. uint32_t ts = 0;
3920. while (pos + 11 < nPacketLen)
3921. {
3922. /* size without header (11) and prevTagSize (4) */
3923. uint32_t dataSize =
3924. AMF_DecodeInt24(packetBody + pos + 1);
3925. ts = AMF_DecodeInt24(packetBody + pos + 4);
3926. ts |= (packetBody[pos + 7] << 24);
3928. #ifdef _DEBUG
3929. RTMP_Log(RTMP_LOGDEBUG,
3930. "keyframe search: FLV Packet: type %02X, dataSize: %d, timeStamp: %d ms",
3931. packetBody[pos], dataSize, ts);
3932. #endif
3933. /* ok, is it a keyframe?:
3934. * well doesn't work for audio!
3935. */
3936. if (packetBody[pos /*6928, test 0 */ ] ==
3937. r->m_read.initialFrameType
3938. /* && (packetBody[11]&0xf0) == 0x10 */ )
3939. {
3940. if (ts == r->m_read.nResumeTS)
3941. {
3942. RTMP_Log(RTMP_LOGDEBUG,
3943. "Found keyframe with resume-keyframe timestamp!");
3944. if (r->m_read.nInitialFrameSize != dataSize
3945. || memcmp(r->m_read.initialFrame,
3946. packetBody + pos + 11,
3947. r->m_read.
3948. nInitialFrameSize) != 0)
3949. {
3950. RTMP_Log(RTMP_LOGERROR,
3951. "FLV Stream: Keyframe doesn't match!");
3952. ret = RTMP_READ_ERROR;
3953. break;
3954. }
3955. r->m_read.flags |= RTMP_READ_GOTFLVK;
3957. /* skip this packet?
3958. * check whether skippable:
3959. */
3960. if (pos + 11 + dataSize + 4 > nPacketLen)
3961. {
3962. RTMP_Log(RTMP_LOGWARNING,
3963. "Non skipable packet since it doesn't end with chunk, stream corrupt!");
3964. ret = RTMP_READ_ERROR;
3965. break;
3966. }
3967. packetBody += (pos + 11 + dataSize + 4);
3968. nPacketLen -= (pos + 11 + dataSize + 4);
3970. goto stopKeyframeSearch;
3972. }
3973. else if (r->m_read.nResumeTS < ts)
3974. {
3975. /* the timestamp ts will only increase with
3976. * further packets, wait for seek
3977. */
3978. goto stopKeyframeSearch;
3979. }
3980. }
3981. pos += (11 + dataSize + 4);
3982. }
3983. if (ts < r->m_read.nResumeTS)
3984. {
3985. RTMP_Log(RTMP_LOGERROR,
3986. "First packet does not contain keyframe, all "
3987. "timestamps are smaller than the keyframe "
3988. "timestamp; probably the resume seek failed?");
3989. }
3990. stopKeyframeSearch:
3991. ;
3992. if (!(r->m_read.flags & RTMP_READ_GOTFLVK))
3993. {
3994. RTMP_Log(RTMP_LOGERROR,
3995. "Couldn't find the seeked keyframe in this chunk!");
3996. ret = RTMP_READ_IGNORE;
3997. break;
3998. }
3999. }
4000. }
4001. }
4003. if (packet.m_nTimeStamp > 0
4004. && (r->m_read.flags & (RTMP_READ_GOTKF|RTMP_READ_GOTFLVK)))
4005. {
4006. /* another problem is that the server can actually change from
4007. * 09/08 video/audio packets to an FLV stream or vice versa and
4008. * our keyframe check will prevent us from going along with the
4009. * new stream if we resumed.
4010. *
4011. * in this case set the 'found keyframe' variables to true.
4012. * We assume that if we found one keyframe somewhere and were
4013. * already beyond TS > 0 we have written data to the output
4014. * which means we can accept all forthcoming data including the
4015. * change between 08/09 <-> FLV packets
4016. */
4017. r->m_read.flags |= (RTMP_READ_GOTKF|RTMP_READ_GOTFLVK);
4018. }
4020. /* skip till we find our keyframe
4021. * (seeking might put us somewhere before it)
4022. */
4023. if (!(r->m_read.flags & RTMP_READ_GOTKF) &&
4024. packet.m_packetType != RTMP_PACKET_TYPE_FLASH_VIDEO)
4025. {
4026. RTMP_Log(RTMP_LOGWARNING,
4027. "Stream does not start with requested frame, ignoring data... ");
4028. r->m_read.nIgnoredFrameCounter++;
4029. if (r->m_read.nIgnoredFrameCounter > MAX_IGNORED_FRAMES)
4030. ret = RTMP_READ_ERROR; /* fatal error, couldn't continue stream */
4031. else
4032. ret = RTMP_READ_IGNORE;
4033. break;
4034. }
4035. /* ok, do the same for FLV streams */
4036. if (!(r->m_read.flags & RTMP_READ_GOTFLVK) &&
4037. packet.m_packetType == RTMP_PACKET_TYPE_FLASH_VIDEO)
4038. {
4039. RTMP_Log(RTMP_LOGWARNING,
4040. "Stream does not start with requested FLV frame, ignoring data... ");
4041. r->m_read.nIgnoredFlvFrameCounter++;
4042. if (r->m_read.nIgnoredFlvFrameCounter > MAX_IGNORED_FRAMES)
4043. ret = RTMP_READ_ERROR;
4044. else
4045. ret = RTMP_READ_IGNORE;
4046. break;
4047. }
4049. /* we have to ignore the 0ms frames since these are the first
4050. * keyframes; we've got these so don't mess around with multiple
4051. * copies sent by the server to us! (if the keyframe is found at a
4052. * later position there is only one copy and it will be ignored by
4053. * the preceding if clause)
4054. */
4055. if (!(r->m_read.flags & RTMP_READ_NO_IGNORE) &&
4056. packet.m_packetType != RTMP_PACKET_TYPE_FLASH_VIDEO)
4057. {
4058. /* exclude type RTMP_PACKET_TYPE_FLASH_VIDEO since it can
4059. * contain several FLV packets
4060. */
4061. if (packet.m_nTimeStamp == 0)
4062. {
4063. ret = RTMP_READ_IGNORE;
4064. break;
4065. }
4066. else
4067. {
4068. /* stop ignoring packets */
4069. r->m_read.flags |= RTMP_READ_NO_IGNORE;
4070. }
4071. }
4072. }
4074. /* calculate packet size and allocate slop buffer if necessary */
4075. size = nPacketLen +
4076. ((packet.m_packetType == RTMP_PACKET_TYPE_AUDIO
4077. || packet.m_packetType == RTMP_PACKET_TYPE_VIDEO
4078. || packet.m_packetType == RTMP_PACKET_TYPE_INFO) ? 11 : 0) +
4079. (packet.m_packetType != RTMP_PACKET_TYPE_FLASH_VIDEO ? 4 : 0);
4081. if (size + 4 > buflen)
4082. {
4083. /* the extra 4 is for the case of an FLV stream without a last
4084. * prevTagSize (we need extra 4 bytes to append it) */
4085. r->m_read.buf = malloc(size + 4);
4086. if (r->m_read.buf == 0)
4087. {
4088. RTMP_Log(RTMP_LOGERROR, "Couldn't allocate memory!");
4089. ret = RTMP_READ_ERROR; /* fatal error */
4090. break;
4091. }
4092. recopy = TRUE;
4093. ptr = r->m_read.buf;
4094. }
4095. else
4096. {
4097. ptr = buf;
4098. }
4099. pend = ptr + size + 4;
4101. /* use to return timestamp of last processed packet */
4103. /* audio (0x08), video (0x09) or metadata (0x12) packets :
4104. * construct 11 byte header then add rtmp packet's data */
4105. if (packet.m_packetType == RTMP_PACKET_TYPE_AUDIO
4106. || packet.m_packetType == RTMP_PACKET_TYPE_VIDEO
4107. || packet.m_packetType == RTMP_PACKET_TYPE_INFO)
4108. {
4109. nTimeStamp = r->m_read.nResumeTS + packet.m_nTimeStamp;
4110. prevTagSize = 11 + nPacketLen;
4112. *ptr = packet.m_packetType;
4113. ptr++;
4114. ptr = AMF_EncodeInt24(ptr, pend, nPacketLen);
4116. #if 0
4117. if(packet.m_packetType == RTMP_PACKET_TYPE_VIDEO) {
4119. /* H264 fix: */
4120. if((packetBody[0] & 0x0f) == 7) { /* CodecId = H264 */
4121. uint8_t packetType = *(packetBody+1);
4123. uint32_t ts = AMF_DecodeInt24(packetBody+2); /* composition time */
4124. int32_t cts = (ts+0xff800000)^0xff800000;
4125. RTMP_Log(RTMP_LOGDEBUG, "cts : %d\n", cts);
4127. nTimeStamp -= cts;
4128. /* get rid of the composition time */
4129. CRTMP::EncodeInt24(packetBody+2, 0);
4130. }
4131. RTMP_Log(RTMP_LOGDEBUG, "VIDEO: nTimeStamp: 0x%08X (%d)\n", nTimeStamp, nTimeStamp);
4132. }
4133. #endif
4135. ptr = AMF_EncodeInt24(ptr, pend, nTimeStamp);
4136. *ptr = (char)((nTimeStamp & 0xFF000000) >> 24);
4137. ptr++;
4139. /* stream id */
4140. ptr = AMF_EncodeInt24(ptr, pend, 0);
4141. }
4143. memcpy(ptr, packetBody, nPacketLen);
4144. len = nPacketLen;
4146. /* correct tagSize and obtain timestamp if we have an FLV stream */
4147. if (packet.m_packetType == RTMP_PACKET_TYPE_FLASH_VIDEO)
4148. {
4149. unsigned int pos = 0;
4150. int delta;
4152. /* grab first timestamp and see if it needs fixing */
4153. nTimeStamp = AMF_DecodeInt24(packetBody + 4);
4154. nTimeStamp |= (packetBody[7] << 24);
4155. delta = packet.m_nTimeStamp - nTimeStamp + r->m_read.nResumeTS;
4157. while (pos + 11 < nPacketLen)
4158. {
4159. /* size without header (11) and without prevTagSize (4) */
4160. uint32_t dataSize = AMF_DecodeInt24(packetBody + pos + 1);
4161. nTimeStamp = AMF_DecodeInt24(packetBody + pos + 4);
4162. nTimeStamp |= (packetBody[pos + 7] << 24);
4164. if (delta)
4165. {
4166. nTimeStamp += delta;
4167. AMF_EncodeInt24(ptr+pos+4, pend, nTimeStamp);
4168. ptr[pos+7] = nTimeStamp>>24;
4169. }
4171. /* set data type */
4172. r->m_read.dataType |= (((*(packetBody + pos) == 0x08) << 2) |
4173. (*(packetBody + pos) == 0x09));
4175. if (pos + 11 + dataSize + 4 > nPacketLen)
4176. {
4177. if (pos + 11 + dataSize > nPacketLen)
4178. {
4179. RTMP_Log(RTMP_LOGERROR,
4180. "Wrong data size (%u), stream corrupted, aborting!",
4181. dataSize);
4182. ret = RTMP_READ_ERROR;
4183. break;
4184. }
4185. RTMP_Log(RTMP_LOGWARNING, "No tagSize found, appending!");
4187. /* we have to append a last tagSize! */
4188. prevTagSize = dataSize + 11;
4189. AMF_EncodeInt32(ptr + pos + 11 + dataSize, pend,
4190. prevTagSize);
4191. size += 4;
4192. len += 4;
4193. }
4194. else
4195. {
4196. prevTagSize =
4197. AMF_DecodeInt32(packetBody + pos + 11 + dataSize);
4199. #ifdef _DEBUG
4200. RTMP_Log(RTMP_LOGDEBUG,
4201. "FLV Packet: type %02X, dataSize: %lu, tagSize: %lu, timeStamp: %lu ms",
4202. (unsigned char)packetBody[pos], dataSize, prevTagSize,
4203. nTimeStamp);
4204. #endif
4206. if (prevTagSize != (dataSize + 11))
4207. {
4208. #ifdef _DEBUG
4209. RTMP_Log(RTMP_LOGWARNING,
4210. "Tag and data size are not consitent, writing tag size according to dataSize+11: %d",
4211. dataSize + 11);
4212. #endif
4214. prevTagSize = dataSize + 11;
4215. AMF_EncodeInt32(ptr + pos + 11 + dataSize, pend,
4216. prevTagSize);
4217. }
4218. }
4220. pos += prevTagSize + 4; /*(11+dataSize+4); */
4221. }
4222. }
4223. ptr += len;
4225. if (packet.m_packetType != RTMP_PACKET_TYPE_FLASH_VIDEO)
4226. {
4227. /* FLV tag packets contain their own prevTagSize */
4228. AMF_EncodeInt32(ptr, pend, prevTagSize);
4229. }
4231. /* In non-live this nTimeStamp can contain an absolute TS.
4232. * Update ext timestamp with this absolute offset in non-live mode
4233. * otherwise report the relative one
4234. */
4235. /* RTMP_Log(RTMP_LOGDEBUG, "type: %02X, size: %d, pktTS: %dms, TS: %dms, bLiveStream: %d", packet.m_packetType, nPacketLen, packet.m_nTimeStamp, nTimeStamp, r->Link.lFlags & RTMP_LF_LIVE); */
4236. r->m_read.timestamp = (r->Link.lFlags & RTMP_LF_LIVE) ? packet.m_nTimeStamp : nTimeStamp;
4238. ret = size;
4239. break;
4240. }
4242. if (rtnGetNextMediaPacket)
4243. RTMPPacket_Free(&packet);
4245. if (recopy)
4246. {
4247. len = ret > buflen ? buflen : ret;
4248. memcpy(buf, r->m_read.buf, len);
4249. r->m_read.bufpos = r->m_read.buf + len;
4250. r->m_read.buflen = ret - len;
4251. }
4252. return ret;
4253. }
4255. static const char flvHeader[] = { 'F', 'L', 'V', 0x01,
4256. 0x00, /* 0x04 == audio, 0x01 == video */
4257. 0x00, 0x00, 0x00, 0x09,
4258. 0x00, 0x00, 0x00, 0x00
4259. };
4261. #define HEADERBUF (128*1024)
4262. int
4263. RTMP_Read(RTMP *r, char *buf, int size)
4264. {
4265. int nRead = 0, total = 0;
4267. /* can't continue */
4268. fail:
4269. switch (r->m_read.status) {
4270. case RTMP_READ_EOF:
4271. case RTMP_READ_COMPLETE:
4272. return 0;
4273. case RTMP_READ_ERROR: /* corrupted stream, resume failed */
4274. SetSockError(EINVAL);
4275. return -1;
4276. default:
4277. break;
4278. }
4280. /* first time thru */
4281. if (!(r->m_read.flags & RTMP_READ_HEADER))
4282. {
4283. if (!(r->m_read.flags & RTMP_READ_RESUME))
4284. {
4285. char *mybuf = malloc(HEADERBUF), *end = mybuf + HEADERBUF;
4286. int cnt = 0;
4287. r->m_read.buf = mybuf;
4288. r->m_read.buflen = HEADERBUF;
4290. memcpy(mybuf, flvHeader, sizeof(flvHeader));
4291. r->m_read.buf += sizeof(flvHeader);
4292. r->m_read.buflen -= sizeof(flvHeader);
4294. while (r->m_read.timestamp == 0)
4295. {
4296. nRead = Read_1_Packet(r, r->m_read.buf, r->m_read.buflen);
4297. if (nRead < 0)
4298. {
4299. free(mybuf);
4300. r->m_read.buf = NULL;
4301. r->m_read.buflen = 0;
4302. r->m_read.status = nRead;
4303. goto fail;
4304. }
4305. /* buffer overflow, fix buffer and give up */
4306. if (r->m_read.buf < mybuf || r->m_read.buf > end) {
4307. mybuf = realloc(mybuf, cnt + nRead);
4308. memcpy(mybuf+cnt, r->m_read.buf, nRead);
4309. r->m_read.buf = mybuf+cnt+nRead;
4310. break;
4311. }
4312. cnt += nRead;
4313. r->m_read.buf += nRead;
4314. r->m_read.buflen -= nRead;
4315. if (r->m_read.dataType == 5)
4316. break;
4317. }
4318. mybuf[4] = r->m_read.dataType;
4319. r->m_read.buflen = r->m_read.buf - mybuf;
4320. r->m_read.buf = mybuf;
4321. r->m_read.bufpos = mybuf;
4322. }
4323. r->m_read.flags |= RTMP_READ_HEADER;
4324. }
4326. if ((r->m_read.flags & RTMP_READ_SEEKING) && r->m_read.buf)
4327. {
4328. /* drop whatever's here */
4329. free(r->m_read.buf);
4330. r->m_read.buf = NULL;
4331. r->m_read.bufpos = NULL;
4332. r->m_read.buflen = 0;
4333. }
4335. /* If there's leftover data buffered, use it up */
4336. if (r->m_read.buf)
4337. {
4338. nRead = r->m_read.buflen;
4339. if (nRead > size)
4340. nRead = size;
4341. memcpy(buf, r->m_read.bufpos, nRead);
4342. r->m_read.buflen -= nRead;
4343. if (!r->m_read.buflen)
4344. {
4345. free(r->m_read.buf);
4346. r->m_read.buf = NULL;
4347. r->m_read.bufpos = NULL;
4348. }
4349. else
4350. {
4351. r->m_read.bufpos += nRead;
4352. }
4353. buf += nRead;
4354. total += nRead;
4355. size -= nRead;
4356. }
4358. while (size > 0 && (nRead = Read_1_Packet(r, buf, size)) >= 0)
4359. {
4360. if (!nRead) continue;
4361. buf += nRead;
4362. total += nRead;
4363. size -= nRead;
4364. break;
4365. }
4366. if (nRead < 0)
4367. r->m_read.status = nRead;
4369. if (size < 0)
4370. total += size;
4371. return total;
4372. }
4374. static const AVal av_setDataFrame = AVC("@setDataFrame");
4376. int
4377. RTMP_Write(RTMP *r, const char *buf, int size)
4378. {
4379. RTMPPacket *pkt = &r->m_write;
4380. char *pend, *enc;
4381. int s2 = size, ret, num;
4383. pkt->m_nChannel = 0x04; /* source channel */
4384. pkt->m_nInfoField2 = r->m_stream_id;
4386. while (s2)
4387. {
4388. if (!pkt->m_nBytesRead)
4389. {
4390. if (size < 11) {
4391. /* FLV pkt too small */
4392. return 0;
4393. }
4395. if (buf[0] == 'F' && buf[1] == 'L' && buf[2] == 'V')
4396. {
4397. buf += 13;
4398. s2 -= 13;
4399. }
4401. pkt->m_packetType = *buf++;
4402. pkt->m_nBodySize = AMF_DecodeInt24(buf);
4403. buf += 3;
4404. pkt->m_nTimeStamp = AMF_DecodeInt24(buf);
4405. buf += 3;
4406. pkt->m_nTimeStamp |= *buf++ << 24;
4407. buf += 3;
4408. s2 -= 11;
4410. if (((pkt->m_packetType == RTMP_PACKET_TYPE_AUDIO
4411. || pkt->m_packetType == RTMP_PACKET_TYPE_VIDEO) &&
4412. !pkt->m_nTimeStamp) || pkt->m_packetType == RTMP_PACKET_TYPE_INFO)
4413. {
4414. pkt->m_headerType = RTMP_PACKET_SIZE_LARGE;
4415. if (pkt->m_packetType == RTMP_PACKET_TYPE_INFO)
4416. pkt->m_nBodySize += 16;
4417. }
4418. else
4419. {
4420. pkt->m_headerType = RTMP_PACKET_SIZE_MEDIUM;
4421. }
4423. if (!RTMPPacket_Alloc(pkt, pkt->m_nBodySize))
4424. {
4425. RTMP_Log(RTMP_LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__);
4426. return FALSE;
4427. }
4428. enc = pkt->m_body;
4429. pend = enc + pkt->m_nBodySize;
4430. if (pkt->m_packetType == RTMP_PACKET_TYPE_INFO)
4431. {
4432. enc = AMF_EncodeString(enc, pend, &av_setDataFrame);
4433. pkt->m_nBytesRead = enc - pkt->m_body;
4434. }
4435. }
4436. else
4437. {
4438. enc = pkt->m_body + pkt->m_nBytesRead;
4439. }
4440. num = pkt->m_nBodySize - pkt->m_nBytesRead;
4441. if (num > s2)
4442. num = s2;
4443. memcpy(enc, buf, num);
4444. pkt->m_nBytesRead += num;
4445. s2 -= num;
4446. buf += num;
4447. if (pkt->m_nBytesRead == pkt->m_nBodySize)
4448. {
4449. ret = RTMP_SendPacket(r, pkt, FALSE);
4450. RTMPPacket_Free(pkt);
4451. pkt->m_nBytesRead = 0;
4452. if (!ret)
4453. return -1;
4454. buf += 4;
4455. s2 -= 4;
4456. if (s2 < 0)
4457. break;
4458. }
4459. }
4460. return size+s2;
4461. }