1 /***************************************************************************
3 * Project ___| | | | _ \| |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
8 * Copyright (C) 2012 - 2014, Daniel Stenberg, <daniel@haxx.se>, et al.
9 * Copyright (C) 2010, Howard Chu, <hyc@highlandsun.com>
11 * This software is licensed as described in the file COPYING, which
12 * you should have received as part of this distribution. The terms
13 * are also available at http://curl.haxx.se/docs/copyright.html.
15 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
16 * copies of the Software, and permit persons to whom the Software is
17 * furnished to do so, under the terms of the COPYING file.
19 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
20 * KIND, either express or implied.
22 ***************************************************************************/
24 #include "curl_setup.h"
29 #include "nonblock.h" /* for curlx_nonblock */
30 #include "progress.h" /* for Curl_pgrsSetUploadSize */
33 #include <curl/curl.h>
34 #include <librtmp/rtmp.h>
36 #define _MPRINTF_REPLACE /* use our functions only */
37 #include <curl/mprintf.h>
39 #include "curl_memory.h"
40 /* The last #include file should be: */
44 #define setsockopt(a,b,c,d,e) (setsockopt)(a,b,c,(const char *)d,(int)e)
45 #define SET_RCVTIMEO(tv,s) int tv = s*1000
47 #define SET_RCVTIMEO(tv,s) struct timeval tv = {s,0}
50 #define DEF_BUFTIME (2*60*60*1000) /* 2 hours */
52 static CURLcode rtmp_setup_connection(struct connectdata *conn);
53 static CURLcode rtmp_do(struct connectdata *conn, bool *done);
54 static CURLcode rtmp_done(struct connectdata *conn, CURLcode, bool premature);
55 static CURLcode rtmp_connect(struct connectdata *conn, bool *done);
56 static CURLcode rtmp_disconnect(struct connectdata *conn, bool dead);
58 static Curl_recv rtmp_recv;
59 static Curl_send rtmp_send;
62 * RTMP protocol handler.h, based on http://rtmpdump.mplayerhq.hu
65 const struct Curl_handler Curl_handler_rtmp = {
67 rtmp_setup_connection, /* setup_connection */
70 ZERO_NULL, /* do_more */
71 rtmp_connect, /* connect_it */
72 ZERO_NULL, /* connecting */
73 ZERO_NULL, /* doing */
74 ZERO_NULL, /* proto_getsock */
75 ZERO_NULL, /* doing_getsock */
76 ZERO_NULL, /* domore_getsock */
77 ZERO_NULL, /* perform_getsock */
78 rtmp_disconnect, /* disconnect */
79 ZERO_NULL, /* readwrite */
80 PORT_RTMP, /* defport */
81 CURLPROTO_RTMP, /* protocol */
82 PROTOPT_NONE /* flags*/
85 const struct Curl_handler Curl_handler_rtmpt = {
87 rtmp_setup_connection, /* setup_connection */
90 ZERO_NULL, /* do_more */
91 rtmp_connect, /* connect_it */
92 ZERO_NULL, /* connecting */
93 ZERO_NULL, /* doing */
94 ZERO_NULL, /* proto_getsock */
95 ZERO_NULL, /* doing_getsock */
96 ZERO_NULL, /* domore_getsock */
97 ZERO_NULL, /* perform_getsock */
98 rtmp_disconnect, /* disconnect */
99 ZERO_NULL, /* readwrite */
100 PORT_RTMPT, /* defport */
101 CURLPROTO_RTMPT, /* protocol */
102 PROTOPT_NONE /* flags*/
105 const struct Curl_handler Curl_handler_rtmpe = {
106 "RTMPE", /* scheme */
107 rtmp_setup_connection, /* setup_connection */
109 rtmp_done, /* done */
110 ZERO_NULL, /* do_more */
111 rtmp_connect, /* connect_it */
112 ZERO_NULL, /* connecting */
113 ZERO_NULL, /* doing */
114 ZERO_NULL, /* proto_getsock */
115 ZERO_NULL, /* doing_getsock */
116 ZERO_NULL, /* domore_getsock */
117 ZERO_NULL, /* perform_getsock */
118 rtmp_disconnect, /* disconnect */
119 ZERO_NULL, /* readwrite */
120 PORT_RTMP, /* defport */
121 CURLPROTO_RTMPE, /* protocol */
122 PROTOPT_NONE /* flags*/
125 const struct Curl_handler Curl_handler_rtmpte = {
126 "RTMPTE", /* scheme */
127 rtmp_setup_connection, /* setup_connection */
129 rtmp_done, /* done */
130 ZERO_NULL, /* do_more */
131 rtmp_connect, /* connect_it */
132 ZERO_NULL, /* connecting */
133 ZERO_NULL, /* doing */
134 ZERO_NULL, /* proto_getsock */
135 ZERO_NULL, /* doing_getsock */
136 ZERO_NULL, /* domore_getsock */
137 ZERO_NULL, /* perform_getsock */
138 rtmp_disconnect, /* disconnect */
139 ZERO_NULL, /* readwrite */
140 PORT_RTMPT, /* defport */
141 CURLPROTO_RTMPTE, /* protocol */
142 PROTOPT_NONE /* flags*/
145 const struct Curl_handler Curl_handler_rtmps = {
146 "RTMPS", /* scheme */
147 rtmp_setup_connection, /* setup_connection */
149 rtmp_done, /* done */
150 ZERO_NULL, /* do_more */
151 rtmp_connect, /* connect_it */
152 ZERO_NULL, /* connecting */
153 ZERO_NULL, /* doing */
154 ZERO_NULL, /* proto_getsock */
155 ZERO_NULL, /* doing_getsock */
156 ZERO_NULL, /* domore_getsock */
157 ZERO_NULL, /* perform_getsock */
158 rtmp_disconnect, /* disconnect */
159 ZERO_NULL, /* readwrite */
160 PORT_RTMPS, /* defport */
161 CURLPROTO_RTMPS, /* protocol */
162 PROTOPT_NONE /* flags*/
165 const struct Curl_handler Curl_handler_rtmpts = {
166 "RTMPTS", /* scheme */
167 rtmp_setup_connection, /* setup_connection */
169 rtmp_done, /* done */
170 ZERO_NULL, /* do_more */
171 rtmp_connect, /* connect_it */
172 ZERO_NULL, /* connecting */
173 ZERO_NULL, /* doing */
174 ZERO_NULL, /* proto_getsock */
175 ZERO_NULL, /* doing_getsock */
176 ZERO_NULL, /* domore_getsock */
177 ZERO_NULL, /* perform_getsock */
178 rtmp_disconnect, /* disconnect */
179 ZERO_NULL, /* readwrite */
180 PORT_RTMPS, /* defport */
181 CURLPROTO_RTMPTS, /* protocol */
182 PROTOPT_NONE /* flags*/
185 static CURLcode rtmp_setup_connection(struct connectdata *conn)
187 RTMP *r = RTMP_Alloc();
189 return CURLE_OUT_OF_MEMORY;
192 RTMP_SetBufferMS(r, DEF_BUFTIME);
193 if(!RTMP_SetupURL(r, conn->data->change.url)) {
195 return CURLE_URL_MALFORMAT;
197 conn->proto.generic = r;
201 static CURLcode rtmp_connect(struct connectdata *conn, bool *done)
203 RTMP *r = conn->proto.generic;
206 r->m_sb.sb_socket = conn->sock[FIRSTSOCKET];
208 /* We have to know if it's a write before we send the
209 * connect request packet
211 if(conn->data->set.upload)
212 r->Link.protocol |= RTMP_FEATURE_WRITE;
214 /* For plain streams, use the buffer toggle trick to keep data flowing */
215 if(!(r->Link.lFlags & RTMP_LF_LIVE) &&
216 !(r->Link.protocol & RTMP_FEATURE_HTTP))
217 r->Link.lFlags |= RTMP_LF_BUFX;
219 (void)curlx_nonblock(r->m_sb.sb_socket, FALSE);
220 setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO,
221 (char *)&tv, sizeof(tv));
223 if(!RTMP_Connect1(r, NULL))
224 return CURLE_FAILED_INIT;
226 /* Clients must send a periodic BytesReceived report to the server */
227 r->m_bSendCounter = true;
230 conn->recv[FIRSTSOCKET] = rtmp_recv;
231 conn->send[FIRSTSOCKET] = rtmp_send;
235 static CURLcode rtmp_do(struct connectdata *conn, bool *done)
237 RTMP *r = conn->proto.generic;
239 if(!RTMP_ConnectStream(r, 0))
240 return CURLE_FAILED_INIT;
242 if(conn->data->set.upload) {
243 Curl_pgrsSetUploadSize(conn->data, conn->data->state.infilesize);
244 Curl_setup_transfer(conn, -1, -1, FALSE, NULL, FIRSTSOCKET, NULL);
247 Curl_setup_transfer(conn, FIRSTSOCKET, -1, FALSE, NULL, -1, NULL);
252 static CURLcode rtmp_done(struct connectdata *conn, CURLcode status,
255 (void)conn; /* unused */
256 (void)status; /* unused */
257 (void)premature; /* unused */
262 static CURLcode rtmp_disconnect(struct connectdata *conn,
263 bool dead_connection)
265 RTMP *r = conn->proto.generic;
266 (void)dead_connection;
268 conn->proto.generic = NULL;
275 static ssize_t rtmp_recv(struct connectdata *conn, int sockindex, char *buf,
276 size_t len, CURLcode *err)
278 RTMP *r = conn->proto.generic;
281 (void)sockindex; /* unused */
283 nread = RTMP_Read(r, buf, curlx_uztosi(len));
285 if(r->m_read.status == RTMP_READ_COMPLETE ||
286 r->m_read.status == RTMP_READ_EOF) {
287 conn->data->req.size = conn->data->req.bytecount;
291 *err = CURLE_RECV_ERROR;
296 static ssize_t rtmp_send(struct connectdata *conn, int sockindex,
297 const void *buf, size_t len, CURLcode *err)
299 RTMP *r = conn->proto.generic;
302 (void)sockindex; /* unused */
304 num = RTMP_Write(r, (char *)buf, curlx_uztosi(len));
306 *err = CURLE_SEND_ERROR;
310 #endif /* USE_LIBRTMP */