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