19bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels/*************************************************************************** 29bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * _ _ ____ _ 39bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * Project ___| | | | _ \| | 49bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * / __| | | | |_) | | 59bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * | (__| |_| | _ <| |___ 69bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * \___|\___/|_| \_\_____| 79bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * 8e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET * Copyright (C) 2012 - 2015, Daniel Stenberg, <daniel@haxx.se>, et al. 99bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * Copyright (C) 2010, Howard Chu, <hyc@highlandsun.com> 109bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * 119bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * This software is licensed as described in the file COPYING, which 129bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * you should have received as part of this distribution. The terms 138f1a214b8a21b66f33454790dfba97ae2f818289Alex Deymo * are also available at https://curl.haxx.se/docs/copyright.html. 149bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * 159bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * You may opt to use, copy, modify, merge, publish, distribute and/or sell 169bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * copies of the Software, and permit persons to whom the Software is 179bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * furnished to do so, under the terms of the COPYING file. 189bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * 199bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY 209bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * KIND, either express or implied. 219bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * 229bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ***************************************************************************/ 239bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 24e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET#include "curl_setup.h" 259bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 269bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#ifdef USE_LIBRTMP 279bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 289bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "urldata.h" 299bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "nonblock.h" /* for curlx_nonblock */ 309bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "progress.h" /* for Curl_pgrsSetUploadSize */ 319bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "transfer.h" 32e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET#include "warnless.h" 339bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include <curl/curl.h> 349bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include <librtmp/rtmp.h> 359bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "curl_memory.h" 369bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels/* The last #include file should be: */ 379bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#include "memdebug.h" 389bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 399bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#ifdef _WIN32 409bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#define setsockopt(a,b,c,d,e) (setsockopt)(a,b,c,(const char *)d,(int)e) 419bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#define SET_RCVTIMEO(tv,s) int tv = s*1000 429bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#else 439bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#define SET_RCVTIMEO(tv,s) struct timeval tv = {s,0} 449bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#endif 459bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 469bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#define DEF_BUFTIME (2*60*60*1000) /* 2 hours */ 479bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 48e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNETstatic CURLcode rtmp_setup_connection(struct connectdata *conn); 499bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_do(struct connectdata *conn, bool *done); 509bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_done(struct connectdata *conn, CURLcode, bool premature); 519bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_connect(struct connectdata *conn, bool *done); 52e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNETstatic CURLcode rtmp_disconnect(struct connectdata *conn, bool dead); 539bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 549bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic Curl_recv rtmp_recv; 559bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic Curl_send rtmp_send; 569bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 579bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels/* 588f1a214b8a21b66f33454790dfba97ae2f818289Alex Deymo * RTMP protocol handler.h, based on https://rtmpdump.mplayerhq.hu 599bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels */ 609bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 619bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmp = { 629bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMP", /* scheme */ 63e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 649bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 659bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 669bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 679bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 689bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 699bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 709bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 719bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 72e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 739bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 749bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 75e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 769bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMP, /* defport */ 77e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMP, /* protocol */ 78e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 799bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 809bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 819bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmpt = { 829bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMPT", /* scheme */ 83e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 849bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 859bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 869bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 879bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 889bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 899bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 909bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 919bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 92e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 939bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 949bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 95e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 969bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMPT, /* defport */ 97e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMPT, /* protocol */ 98e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 999bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 1009bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 1019bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmpe = { 1029bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMPE", /* scheme */ 103e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 1049bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 1059bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 1069bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 1079bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 1089bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 1099bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 1109bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 1119bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 112e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 1139bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 1149bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 115e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 1169bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMP, /* defport */ 117e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMPE, /* protocol */ 118e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 1199bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 1209bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 1219bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmpte = { 1229bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMPTE", /* scheme */ 123e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 1249bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 1259bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 1269bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 1279bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 1289bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 1299bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 1309bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 1319bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 132e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 1339bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 1349bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 135e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 1369bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMPT, /* defport */ 137e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMPTE, /* protocol */ 138e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 1399bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 1409bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 1419bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmps = { 1429bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMPS", /* scheme */ 143e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 1449bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 1459bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 1469bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 1479bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 1489bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 1499bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 1509bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 1519bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 152e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 1539bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 1549bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 155e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 1569bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMPS, /* defport */ 157e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMPS, /* protocol */ 158e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 1599bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 160e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET 1619bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsconst struct Curl_handler Curl_handler_rtmpts = { 1629bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels "RTMPTS", /* scheme */ 163e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET rtmp_setup_connection, /* setup_connection */ 1649bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_do, /* do_it */ 1659bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_done, /* done */ 1669bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* do_more */ 1679bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_connect, /* connect_it */ 1689bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* connecting */ 1699bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing */ 1709bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* proto_getsock */ 1719bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* doing_getsock */ 172e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* domore_getsock */ 1739bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ZERO_NULL, /* perform_getsock */ 1749bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels rtmp_disconnect, /* disconnect */ 175e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET ZERO_NULL, /* readwrite */ 1769bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels PORT_RTMPS, /* defport */ 177e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET CURLPROTO_RTMPTS, /* protocol */ 178e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET PROTOPT_NONE /* flags*/ 1799bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels}; 1809bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 181e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNETstatic CURLcode rtmp_setup_connection(struct connectdata *conn) 1829bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 1839bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = RTMP_Alloc(); 184e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(!r) 1859bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OUT_OF_MEMORY; 1869bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 1879bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP_Init(r); 1889bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP_SetBufferMS(r, DEF_BUFTIME); 189e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(!RTMP_SetupURL(r, conn->data->change.url)) { 1909bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP_Free(r); 1919bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_URL_MALFORMAT; 1929bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels } 1939bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels conn->proto.generic = r; 1949bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OK; 1959bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 1969bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 1979bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_connect(struct connectdata *conn, bool *done) 1989bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 1999bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = conn->proto.generic; 200e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET SET_RCVTIMEO(tv, 10); 2019bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2029bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels r->m_sb.sb_socket = conn->sock[FIRSTSOCKET]; 2039bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2049bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels /* We have to know if it's a write before we send the 2059bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels * connect request packet 2069bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels */ 207e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(conn->data->set.upload) 2089bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels r->Link.protocol |= RTMP_FEATURE_WRITE; 2099bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2109bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels /* For plain streams, use the buffer toggle trick to keep data flowing */ 211e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(!(r->Link.lFlags & RTMP_LF_LIVE) && 212e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET !(r->Link.protocol & RTMP_FEATURE_HTTP)) 2139bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels r->Link.lFlags |= RTMP_LF_BUFX; 2149bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 215e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET (void)curlx_nonblock(r->m_sb.sb_socket, FALSE); 216e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, 217e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET (char *)&tv, sizeof(tv)); 2189bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 219e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(!RTMP_Connect1(r, NULL)) 2209bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_FAILED_INIT; 2219bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2229bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels /* Clients must send a periodic BytesReceived report to the server */ 2239bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels r->m_bSendCounter = true; 2249bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2259bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels *done = TRUE; 2269bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels conn->recv[FIRSTSOCKET] = rtmp_recv; 2279bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels conn->send[FIRSTSOCKET] = rtmp_send; 2289bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OK; 2299bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 2309bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2319bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_do(struct connectdata *conn, bool *done) 2329bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 2339bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = conn->proto.generic; 2349bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 235e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(!RTMP_ConnectStream(r, 0)) 2369bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_FAILED_INIT; 2379bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 238e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(conn->data->set.upload) { 239e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET Curl_pgrsSetUploadSize(conn->data, conn->data->state.infilesize); 2409bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels Curl_setup_transfer(conn, -1, -1, FALSE, NULL, FIRSTSOCKET, NULL); 241e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET } 242e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET else 2439bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels Curl_setup_transfer(conn, FIRSTSOCKET, -1, FALSE, NULL, -1, NULL); 2449bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels *done = TRUE; 2459bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OK; 2469bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 2479bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2489bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic CURLcode rtmp_done(struct connectdata *conn, CURLcode status, 2499bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels bool premature) 2509bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 2519bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels (void)conn; /* unused */ 2529bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels (void)status; /* unused */ 2539bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels (void)premature; /* unused */ 2549bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2559bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OK; 2569bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 2579bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 258e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNETstatic CURLcode rtmp_disconnect(struct connectdata *conn, 259e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET bool dead_connection) 2609bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 2619bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = conn->proto.generic; 262e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET (void)dead_connection; 263e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(r) { 2649bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels conn->proto.generic = NULL; 2659bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP_Close(r); 2669bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP_Free(r); 2679bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels } 2689bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return CURLE_OK; 2699bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 2709bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2719bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic ssize_t rtmp_recv(struct connectdata *conn, int sockindex, char *buf, 2729bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels size_t len, CURLcode *err) 2739bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 2749bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = conn->proto.generic; 2759bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ssize_t nread; 2769bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2779bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels (void)sockindex; /* unused */ 2789bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 279e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET nread = RTMP_Read(r, buf, curlx_uztosi(len)); 280e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(nread < 0) { 281e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(r->m_read.status == RTMP_READ_COMPLETE || 2829bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels r->m_read.status == RTMP_READ_EOF) { 2839bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels conn->data->req.size = conn->data->req.bytecount; 2849bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels nread = 0; 285e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET } 286e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET else 2879bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels *err = CURLE_RECV_ERROR; 2889bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels } 2899bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return nread; 2909bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 2919bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2929bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckelsstatic ssize_t rtmp_send(struct connectdata *conn, int sockindex, 2939bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels const void *buf, size_t len, CURLcode *err) 2949bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels{ 2959bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels RTMP *r = conn->proto.generic; 2969bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels ssize_t num; 2979bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 2989bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels (void)sockindex; /* unused */ 2999bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels 300e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET num = RTMP_Write(r, (char *)buf, curlx_uztosi(len)); 301e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET if(num < 0) 3029bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels *err = CURLE_SEND_ERROR; 303e6cd738ed3716c02557fb3a47515244e949ade39Bertrand SIMONNET 3049bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels return num; 3059bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels} 3069bd90e6e25f1e55f50201c87a1b5837de7e5b64aLucas Eckels#endif /* USE_LIBRTMP */ 307