1 | /***************************************************************************
|
---|
2 | * _ _ ____ _
|
---|
3 | * Project ___| | | | _ \| |
|
---|
4 | * / __| | | | |_) | |
|
---|
5 | * | (__| |_| | _ <| |___
|
---|
6 | * \___|\___/|_| \_\_____|
|
---|
7 | *
|
---|
8 | * Copyright (C) 2012 - 2015, Daniel Stenberg, <daniel@haxx.se>, et al.
|
---|
9 | * Copyright (C) 2010, Howard Chu, <hyc@highlandsun.com>
|
---|
10 | *
|
---|
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 https://curl.haxx.se/docs/copyright.html.
|
---|
14 | *
|
---|
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.
|
---|
18 | *
|
---|
19 | * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
---|
20 | * KIND, either express or implied.
|
---|
21 | *
|
---|
22 | ***************************************************************************/
|
---|
23 |
|
---|
24 | #include "curl_setup.h"
|
---|
25 |
|
---|
26 | #ifdef USE_LIBRTMP
|
---|
27 |
|
---|
28 | #include "urldata.h"
|
---|
29 | #include "nonblock.h" /* for curlx_nonblock */
|
---|
30 | #include "progress.h" /* for Curl_pgrsSetUploadSize */
|
---|
31 | #include "transfer.h"
|
---|
32 | #include "warnless.h"
|
---|
33 | #include <curl/curl.h>
|
---|
34 | #include <librtmp/rtmp.h>
|
---|
35 | #include "curl_memory.h"
|
---|
36 | /* The last #include file should be: */
|
---|
37 | #include "memdebug.h"
|
---|
38 |
|
---|
39 | #ifdef _WIN32
|
---|
40 | #define setsockopt(a,b,c,d,e) (setsockopt)(a,b,c,(const char *)d,(int)e)
|
---|
41 | #define SET_RCVTIMEO(tv,s) int tv = s*1000
|
---|
42 | #else
|
---|
43 | #define SET_RCVTIMEO(tv,s) struct timeval tv = {s,0}
|
---|
44 | #endif
|
---|
45 |
|
---|
46 | #define DEF_BUFTIME (2*60*60*1000) /* 2 hours */
|
---|
47 |
|
---|
48 | static CURLcode rtmp_setup_connection(struct connectdata *conn);
|
---|
49 | static CURLcode rtmp_do(struct connectdata *conn, bool *done);
|
---|
50 | static CURLcode rtmp_done(struct connectdata *conn, CURLcode, bool premature);
|
---|
51 | static CURLcode rtmp_connect(struct connectdata *conn, bool *done);
|
---|
52 | static CURLcode rtmp_disconnect(struct connectdata *conn, bool dead);
|
---|
53 |
|
---|
54 | static Curl_recv rtmp_recv;
|
---|
55 | static Curl_send rtmp_send;
|
---|
56 |
|
---|
57 | /*
|
---|
58 | * RTMP protocol handler.h, based on https://rtmpdump.mplayerhq.hu
|
---|
59 | */
|
---|
60 |
|
---|
61 | const struct Curl_handler Curl_handler_rtmp = {
|
---|
62 | "RTMP", /* scheme */
|
---|
63 | rtmp_setup_connection, /* setup_connection */
|
---|
64 | rtmp_do, /* do_it */
|
---|
65 | rtmp_done, /* done */
|
---|
66 | ZERO_NULL, /* do_more */
|
---|
67 | rtmp_connect, /* connect_it */
|
---|
68 | ZERO_NULL, /* connecting */
|
---|
69 | ZERO_NULL, /* doing */
|
---|
70 | ZERO_NULL, /* proto_getsock */
|
---|
71 | ZERO_NULL, /* doing_getsock */
|
---|
72 | ZERO_NULL, /* domore_getsock */
|
---|
73 | ZERO_NULL, /* perform_getsock */
|
---|
74 | rtmp_disconnect, /* disconnect */
|
---|
75 | ZERO_NULL, /* readwrite */
|
---|
76 | PORT_RTMP, /* defport */
|
---|
77 | CURLPROTO_RTMP, /* protocol */
|
---|
78 | PROTOPT_NONE /* flags*/
|
---|
79 | };
|
---|
80 |
|
---|
81 | const struct Curl_handler Curl_handler_rtmpt = {
|
---|
82 | "RTMPT", /* scheme */
|
---|
83 | rtmp_setup_connection, /* setup_connection */
|
---|
84 | rtmp_do, /* do_it */
|
---|
85 | rtmp_done, /* done */
|
---|
86 | ZERO_NULL, /* do_more */
|
---|
87 | rtmp_connect, /* connect_it */
|
---|
88 | ZERO_NULL, /* connecting */
|
---|
89 | ZERO_NULL, /* doing */
|
---|
90 | ZERO_NULL, /* proto_getsock */
|
---|
91 | ZERO_NULL, /* doing_getsock */
|
---|
92 | ZERO_NULL, /* domore_getsock */
|
---|
93 | ZERO_NULL, /* perform_getsock */
|
---|
94 | rtmp_disconnect, /* disconnect */
|
---|
95 | ZERO_NULL, /* readwrite */
|
---|
96 | PORT_RTMPT, /* defport */
|
---|
97 | CURLPROTO_RTMPT, /* protocol */
|
---|
98 | PROTOPT_NONE /* flags*/
|
---|
99 | };
|
---|
100 |
|
---|
101 | const struct Curl_handler Curl_handler_rtmpe = {
|
---|
102 | "RTMPE", /* scheme */
|
---|
103 | rtmp_setup_connection, /* setup_connection */
|
---|
104 | rtmp_do, /* do_it */
|
---|
105 | rtmp_done, /* done */
|
---|
106 | ZERO_NULL, /* do_more */
|
---|
107 | rtmp_connect, /* connect_it */
|
---|
108 | ZERO_NULL, /* connecting */
|
---|
109 | ZERO_NULL, /* doing */
|
---|
110 | ZERO_NULL, /* proto_getsock */
|
---|
111 | ZERO_NULL, /* doing_getsock */
|
---|
112 | ZERO_NULL, /* domore_getsock */
|
---|
113 | ZERO_NULL, /* perform_getsock */
|
---|
114 | rtmp_disconnect, /* disconnect */
|
---|
115 | ZERO_NULL, /* readwrite */
|
---|
116 | PORT_RTMP, /* defport */
|
---|
117 | CURLPROTO_RTMPE, /* protocol */
|
---|
118 | PROTOPT_NONE /* flags*/
|
---|
119 | };
|
---|
120 |
|
---|
121 | const struct Curl_handler Curl_handler_rtmpte = {
|
---|
122 | "RTMPTE", /* scheme */
|
---|
123 | rtmp_setup_connection, /* setup_connection */
|
---|
124 | rtmp_do, /* do_it */
|
---|
125 | rtmp_done, /* done */
|
---|
126 | ZERO_NULL, /* do_more */
|
---|
127 | rtmp_connect, /* connect_it */
|
---|
128 | ZERO_NULL, /* connecting */
|
---|
129 | ZERO_NULL, /* doing */
|
---|
130 | ZERO_NULL, /* proto_getsock */
|
---|
131 | ZERO_NULL, /* doing_getsock */
|
---|
132 | ZERO_NULL, /* domore_getsock */
|
---|
133 | ZERO_NULL, /* perform_getsock */
|
---|
134 | rtmp_disconnect, /* disconnect */
|
---|
135 | ZERO_NULL, /* readwrite */
|
---|
136 | PORT_RTMPT, /* defport */
|
---|
137 | CURLPROTO_RTMPTE, /* protocol */
|
---|
138 | PROTOPT_NONE /* flags*/
|
---|
139 | };
|
---|
140 |
|
---|
141 | const struct Curl_handler Curl_handler_rtmps = {
|
---|
142 | "RTMPS", /* scheme */
|
---|
143 | rtmp_setup_connection, /* setup_connection */
|
---|
144 | rtmp_do, /* do_it */
|
---|
145 | rtmp_done, /* done */
|
---|
146 | ZERO_NULL, /* do_more */
|
---|
147 | rtmp_connect, /* connect_it */
|
---|
148 | ZERO_NULL, /* connecting */
|
---|
149 | ZERO_NULL, /* doing */
|
---|
150 | ZERO_NULL, /* proto_getsock */
|
---|
151 | ZERO_NULL, /* doing_getsock */
|
---|
152 | ZERO_NULL, /* domore_getsock */
|
---|
153 | ZERO_NULL, /* perform_getsock */
|
---|
154 | rtmp_disconnect, /* disconnect */
|
---|
155 | ZERO_NULL, /* readwrite */
|
---|
156 | PORT_RTMPS, /* defport */
|
---|
157 | CURLPROTO_RTMPS, /* protocol */
|
---|
158 | PROTOPT_NONE /* flags*/
|
---|
159 | };
|
---|
160 |
|
---|
161 | const struct Curl_handler Curl_handler_rtmpts = {
|
---|
162 | "RTMPTS", /* scheme */
|
---|
163 | rtmp_setup_connection, /* setup_connection */
|
---|
164 | rtmp_do, /* do_it */
|
---|
165 | rtmp_done, /* done */
|
---|
166 | ZERO_NULL, /* do_more */
|
---|
167 | rtmp_connect, /* connect_it */
|
---|
168 | ZERO_NULL, /* connecting */
|
---|
169 | ZERO_NULL, /* doing */
|
---|
170 | ZERO_NULL, /* proto_getsock */
|
---|
171 | ZERO_NULL, /* doing_getsock */
|
---|
172 | ZERO_NULL, /* domore_getsock */
|
---|
173 | ZERO_NULL, /* perform_getsock */
|
---|
174 | rtmp_disconnect, /* disconnect */
|
---|
175 | ZERO_NULL, /* readwrite */
|
---|
176 | PORT_RTMPS, /* defport */
|
---|
177 | CURLPROTO_RTMPTS, /* protocol */
|
---|
178 | PROTOPT_NONE /* flags*/
|
---|
179 | };
|
---|
180 |
|
---|
181 | static CURLcode rtmp_setup_connection(struct connectdata *conn)
|
---|
182 | {
|
---|
183 | RTMP *r = RTMP_Alloc();
|
---|
184 | if(!r)
|
---|
185 | return CURLE_OUT_OF_MEMORY;
|
---|
186 |
|
---|
187 | RTMP_Init(r);
|
---|
188 | RTMP_SetBufferMS(r, DEF_BUFTIME);
|
---|
189 | if(!RTMP_SetupURL(r, conn->data->change.url)) {
|
---|
190 | RTMP_Free(r);
|
---|
191 | return CURLE_URL_MALFORMAT;
|
---|
192 | }
|
---|
193 | conn->proto.generic = r;
|
---|
194 | return CURLE_OK;
|
---|
195 | }
|
---|
196 |
|
---|
197 | static CURLcode rtmp_connect(struct connectdata *conn, bool *done)
|
---|
198 | {
|
---|
199 | RTMP *r = conn->proto.generic;
|
---|
200 | SET_RCVTIMEO(tv, 10);
|
---|
201 |
|
---|
202 | r->m_sb.sb_socket = conn->sock[FIRSTSOCKET];
|
---|
203 |
|
---|
204 | /* We have to know if it's a write before we send the
|
---|
205 | * connect request packet
|
---|
206 | */
|
---|
207 | if(conn->data->set.upload)
|
---|
208 | r->Link.protocol |= RTMP_FEATURE_WRITE;
|
---|
209 |
|
---|
210 | /* For plain streams, use the buffer toggle trick to keep data flowing */
|
---|
211 | if(!(r->Link.lFlags & RTMP_LF_LIVE) &&
|
---|
212 | !(r->Link.protocol & RTMP_FEATURE_HTTP))
|
---|
213 | r->Link.lFlags |= RTMP_LF_BUFX;
|
---|
214 |
|
---|
215 | (void)curlx_nonblock(r->m_sb.sb_socket, FALSE);
|
---|
216 | setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO,
|
---|
217 | (char *)&tv, sizeof(tv));
|
---|
218 |
|
---|
219 | if(!RTMP_Connect1(r, NULL))
|
---|
220 | return CURLE_FAILED_INIT;
|
---|
221 |
|
---|
222 | /* Clients must send a periodic BytesReceived report to the server */
|
---|
223 | r->m_bSendCounter = true;
|
---|
224 |
|
---|
225 | *done = TRUE;
|
---|
226 | conn->recv[FIRSTSOCKET] = rtmp_recv;
|
---|
227 | conn->send[FIRSTSOCKET] = rtmp_send;
|
---|
228 | return CURLE_OK;
|
---|
229 | }
|
---|
230 |
|
---|
231 | static CURLcode rtmp_do(struct connectdata *conn, bool *done)
|
---|
232 | {
|
---|
233 | RTMP *r = conn->proto.generic;
|
---|
234 |
|
---|
235 | if(!RTMP_ConnectStream(r, 0))
|
---|
236 | return CURLE_FAILED_INIT;
|
---|
237 |
|
---|
238 | if(conn->data->set.upload) {
|
---|
239 | Curl_pgrsSetUploadSize(conn->data, conn->data->state.infilesize);
|
---|
240 | Curl_setup_transfer(conn, -1, -1, FALSE, NULL, FIRSTSOCKET, NULL);
|
---|
241 | }
|
---|
242 | else
|
---|
243 | Curl_setup_transfer(conn, FIRSTSOCKET, -1, FALSE, NULL, -1, NULL);
|
---|
244 | *done = TRUE;
|
---|
245 | return CURLE_OK;
|
---|
246 | }
|
---|
247 |
|
---|
248 | static CURLcode rtmp_done(struct connectdata *conn, CURLcode status,
|
---|
249 | bool premature)
|
---|
250 | {
|
---|
251 | (void)conn; /* unused */
|
---|
252 | (void)status; /* unused */
|
---|
253 | (void)premature; /* unused */
|
---|
254 |
|
---|
255 | return CURLE_OK;
|
---|
256 | }
|
---|
257 |
|
---|
258 | static CURLcode rtmp_disconnect(struct connectdata *conn,
|
---|
259 | bool dead_connection)
|
---|
260 | {
|
---|
261 | RTMP *r = conn->proto.generic;
|
---|
262 | (void)dead_connection;
|
---|
263 | if(r) {
|
---|
264 | conn->proto.generic = NULL;
|
---|
265 | RTMP_Close(r);
|
---|
266 | RTMP_Free(r);
|
---|
267 | }
|
---|
268 | return CURLE_OK;
|
---|
269 | }
|
---|
270 |
|
---|
271 | static ssize_t rtmp_recv(struct connectdata *conn, int sockindex, char *buf,
|
---|
272 | size_t len, CURLcode *err)
|
---|
273 | {
|
---|
274 | RTMP *r = conn->proto.generic;
|
---|
275 | ssize_t nread;
|
---|
276 |
|
---|
277 | (void)sockindex; /* unused */
|
---|
278 |
|
---|
279 | nread = RTMP_Read(r, buf, curlx_uztosi(len));
|
---|
280 | if(nread < 0) {
|
---|
281 | if(r->m_read.status == RTMP_READ_COMPLETE ||
|
---|
282 | r->m_read.status == RTMP_READ_EOF) {
|
---|
283 | conn->data->req.size = conn->data->req.bytecount;
|
---|
284 | nread = 0;
|
---|
285 | }
|
---|
286 | else
|
---|
287 | *err = CURLE_RECV_ERROR;
|
---|
288 | }
|
---|
289 | return nread;
|
---|
290 | }
|
---|
291 |
|
---|
292 | static ssize_t rtmp_send(struct connectdata *conn, int sockindex,
|
---|
293 | const void *buf, size_t len, CURLcode *err)
|
---|
294 | {
|
---|
295 | RTMP *r = conn->proto.generic;
|
---|
296 | ssize_t num;
|
---|
297 |
|
---|
298 | (void)sockindex; /* unused */
|
---|
299 |
|
---|
300 | num = RTMP_Write(r, (char *)buf, curlx_uztosi(len));
|
---|
301 | if(num < 0)
|
---|
302 | *err = CURLE_SEND_ERROR;
|
---|
303 |
|
---|
304 | return num;
|
---|
305 | }
|
---|
306 | #endif /* USE_LIBRTMP */
|
---|