mirror of
https://github.com/curl/curl.git
synced 2026-04-22 09:32:11 +03:00
884b5ea921
To make it available for all files. Drop includes from individual sources. This header was already included from most sources and not specific to any internal subsystem. Also to ensure that two system symbol redefines on Windows (`read()` and `write()`) get applied to all sources. Move them to `curl_setup.h`. Closes #20056
387 lines
14 KiB
C
387 lines
14 KiB
C
/***************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
|
|
* Copyright (C) Howard Chu, <hyc@highlandsun.com>
|
|
*
|
|
* This software is licensed as described in the file COPYING, which
|
|
* you should have received as part of this distribution. The terms
|
|
* are also available at https://curl.se/docs/copyright.html.
|
|
*
|
|
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
|
|
* copies of the Software, and permit persons to whom the Software is
|
|
* furnished to do so, under the terms of the COPYING file.
|
|
*
|
|
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
|
* KIND, either express or implied.
|
|
*
|
|
* SPDX-License-Identifier: curl
|
|
*
|
|
***************************************************************************/
|
|
|
|
#include "curl_setup.h"
|
|
|
|
#ifdef USE_LIBRTMP
|
|
|
|
#include "curl_rtmp.h"
|
|
#include "urldata.h"
|
|
#include "url.h"
|
|
#include "curlx/nonblock.h"
|
|
#include "progress.h" /* for Curl_pgrsSetUploadSize */
|
|
#include "transfer.h"
|
|
#include "bufref.h"
|
|
|
|
#include <librtmp/rtmp.h>
|
|
|
|
#if defined(USE_WINSOCK) || defined(LWIP_SO_SNDRCVTIMEO_NONSTANDARD)
|
|
#define SET_RCVTIMEO(tv, s) int tv = s * 1000
|
|
#else
|
|
#define SET_RCVTIMEO(tv, s) struct timeval tv = { s, 0 }
|
|
#endif
|
|
|
|
#define DEF_BUFTIME (2 * 60 * 60 * 1000) /* 2 hours */
|
|
|
|
/* meta key for storing RTMP* at connection */
|
|
#define CURL_META_RTMP_CONN "meta:proto:rtmp:conn"
|
|
|
|
|
|
static CURLcode rtmp_setup_connection(struct Curl_easy *data,
|
|
struct connectdata *conn);
|
|
static CURLcode rtmp_do(struct Curl_easy *data, bool *done);
|
|
static CURLcode rtmp_done(struct Curl_easy *data, CURLcode, bool premature);
|
|
static CURLcode rtmp_connect(struct Curl_easy *data, bool *done);
|
|
static CURLcode rtmp_disconnect(struct Curl_easy *data,
|
|
struct connectdata *conn, bool dead);
|
|
|
|
static Curl_recv rtmp_recv;
|
|
static Curl_send rtmp_send;
|
|
|
|
/*
|
|
* RTMP protocol handler.h, based on https://rtmpdump.mplayerhq.hu/
|
|
*/
|
|
|
|
const struct Curl_handler Curl_handler_rtmp = {
|
|
"rtmp", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMP, /* defport */
|
|
CURLPROTO_RTMP, /* protocol */
|
|
CURLPROTO_RTMP, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
const struct Curl_handler Curl_handler_rtmpt = {
|
|
"rtmpt", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMPT, /* defport */
|
|
CURLPROTO_RTMPT, /* protocol */
|
|
CURLPROTO_RTMPT, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
const struct Curl_handler Curl_handler_rtmpe = {
|
|
"rtmpe", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMP, /* defport */
|
|
CURLPROTO_RTMPE, /* protocol */
|
|
CURLPROTO_RTMPE, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
const struct Curl_handler Curl_handler_rtmpte = {
|
|
"rtmpte", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMPT, /* defport */
|
|
CURLPROTO_RTMPTE, /* protocol */
|
|
CURLPROTO_RTMPTE, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
const struct Curl_handler Curl_handler_rtmps = {
|
|
"rtmps", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMPS, /* defport */
|
|
CURLPROTO_RTMPS, /* protocol */
|
|
CURLPROTO_RTMP, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
const struct Curl_handler Curl_handler_rtmpts = {
|
|
"rtmpts", /* scheme */
|
|
rtmp_setup_connection, /* setup_connection */
|
|
rtmp_do, /* do_it */
|
|
rtmp_done, /* done */
|
|
ZERO_NULL, /* do_more */
|
|
rtmp_connect, /* connect_it */
|
|
ZERO_NULL, /* connecting */
|
|
ZERO_NULL, /* doing */
|
|
ZERO_NULL, /* proto_pollset */
|
|
ZERO_NULL, /* doing_pollset */
|
|
ZERO_NULL, /* domore_pollset */
|
|
ZERO_NULL, /* perform_pollset */
|
|
rtmp_disconnect, /* disconnect */
|
|
ZERO_NULL, /* write_resp */
|
|
ZERO_NULL, /* write_resp_hd */
|
|
ZERO_NULL, /* connection_check */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
PORT_RTMPS, /* defport */
|
|
CURLPROTO_RTMPTS, /* protocol */
|
|
CURLPROTO_RTMPT, /* family */
|
|
PROTOPT_NONE /* flags */
|
|
};
|
|
|
|
static void rtmp_conn_dtor(void *key, size_t klen, void *entry)
|
|
{
|
|
RTMP *r = entry;
|
|
(void)key;
|
|
(void)klen;
|
|
RTMP_Close(r);
|
|
RTMP_Free(r);
|
|
}
|
|
|
|
static CURLcode rtmp_setup_connection(struct Curl_easy *data,
|
|
struct connectdata *conn)
|
|
{
|
|
RTMP *r = RTMP_Alloc();
|
|
if(!r ||
|
|
Curl_conn_meta_set(conn, CURL_META_RTMP_CONN, r, rtmp_conn_dtor))
|
|
return CURLE_OUT_OF_MEMORY;
|
|
|
|
RTMP_Init(r);
|
|
RTMP_SetBufferMS(r, DEF_BUFTIME);
|
|
if(!RTMP_SetupURL(r, CURL_UNCONST(Curl_bufref_ptr(&data->state.url))))
|
|
/* rtmp_conn_dtor() performs the cleanup */
|
|
return CURLE_URL_MALFORMAT;
|
|
return CURLE_OK;
|
|
}
|
|
|
|
static CURLcode rtmp_connect(struct Curl_easy *data, bool *done)
|
|
{
|
|
struct connectdata *conn = data->conn;
|
|
RTMP *r = Curl_conn_meta_get(conn, CURL_META_RTMP_CONN);
|
|
SET_RCVTIMEO(tv, 10);
|
|
|
|
if(!r)
|
|
return CURLE_FAILED_INIT;
|
|
|
|
if(conn->sock[FIRSTSOCKET] > INT_MAX) {
|
|
/* The socket value is invalid for rtmp. */
|
|
return CURLE_FAILED_INIT;
|
|
}
|
|
|
|
r->m_sb.sb_socket = (int)conn->sock[FIRSTSOCKET];
|
|
|
|
/* We have to know if it is a write before we send the
|
|
* connect request packet
|
|
*/
|
|
if(data->state.upload)
|
|
r->Link.protocol |= RTMP_FEATURE_WRITE;
|
|
|
|
/* For plain streams, use the buffer toggle trick to keep data flowing */
|
|
if(!(r->Link.lFlags & RTMP_LF_LIVE) &&
|
|
!(r->Link.protocol & RTMP_FEATURE_HTTP))
|
|
r->Link.lFlags |= RTMP_LF_BUFX;
|
|
|
|
(void)curlx_nonblock(r->m_sb.sb_socket, FALSE);
|
|
setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO,
|
|
(char *)&tv, sizeof(tv));
|
|
|
|
if(!RTMP_Connect1(r, NULL))
|
|
return CURLE_FAILED_INIT;
|
|
|
|
/* Clients must send a periodic BytesReceived report to the server */
|
|
r->m_bSendCounter = TRUE;
|
|
|
|
*done = TRUE;
|
|
conn->recv[FIRSTSOCKET] = rtmp_recv;
|
|
conn->send[FIRSTSOCKET] = rtmp_send;
|
|
return CURLE_OK;
|
|
}
|
|
|
|
static CURLcode rtmp_do(struct Curl_easy *data, bool *done)
|
|
{
|
|
struct connectdata *conn = data->conn;
|
|
RTMP *r = Curl_conn_meta_get(conn, CURL_META_RTMP_CONN);
|
|
|
|
if(!r || !RTMP_ConnectStream(r, 0))
|
|
return CURLE_FAILED_INIT;
|
|
|
|
if(data->state.upload) {
|
|
Curl_pgrsSetUploadSize(data, data->state.infilesize);
|
|
Curl_xfer_setup_send(data, FIRSTSOCKET);
|
|
}
|
|
else
|
|
Curl_xfer_setup_recv(data, FIRSTSOCKET, -1);
|
|
*done = TRUE;
|
|
return CURLE_OK;
|
|
}
|
|
|
|
static CURLcode rtmp_done(struct Curl_easy *data, CURLcode status,
|
|
bool premature)
|
|
{
|
|
(void)data;
|
|
(void)status;
|
|
(void)premature;
|
|
|
|
return CURLE_OK;
|
|
}
|
|
|
|
static CURLcode rtmp_disconnect(struct Curl_easy *data,
|
|
struct connectdata *conn,
|
|
bool dead_connection)
|
|
{
|
|
RTMP *r = Curl_conn_meta_get(conn, CURL_META_RTMP_CONN);
|
|
(void)data;
|
|
(void)dead_connection;
|
|
if(r)
|
|
Curl_conn_meta_remove(conn, CURL_META_RTMP_CONN);
|
|
return CURLE_OK;
|
|
}
|
|
|
|
static CURLcode rtmp_recv(struct Curl_easy *data, int sockindex, char *buf,
|
|
size_t len, size_t *pnread)
|
|
{
|
|
struct connectdata *conn = data->conn;
|
|
RTMP *r = Curl_conn_meta_get(conn, CURL_META_RTMP_CONN);
|
|
CURLcode result = CURLE_OK;
|
|
ssize_t rv;
|
|
|
|
(void)sockindex;
|
|
*pnread = 0;
|
|
if(!r)
|
|
return CURLE_FAILED_INIT;
|
|
|
|
rv = RTMP_Read(r, buf, curlx_uztosi(len));
|
|
if(!curlx_sztouz(rv, pnread)) {
|
|
if(r->m_read.status == RTMP_READ_COMPLETE ||
|
|
r->m_read.status == RTMP_READ_EOF) {
|
|
data->req.size = data->req.bytecount;
|
|
}
|
|
else
|
|
result = CURLE_RECV_ERROR;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
static CURLcode rtmp_send(struct Curl_easy *data, int sockindex,
|
|
const uint8_t *buf, size_t len, bool eos,
|
|
size_t *pnwritten)
|
|
{
|
|
struct connectdata *conn = data->conn;
|
|
RTMP *r = Curl_conn_meta_get(conn, CURL_META_RTMP_CONN);
|
|
ssize_t rv;
|
|
|
|
(void)sockindex;
|
|
(void)eos;
|
|
*pnwritten = 0;
|
|
if(!r)
|
|
return CURLE_FAILED_INIT;
|
|
|
|
rv = RTMP_Write(r, (const char *)buf, curlx_uztosi(len));
|
|
if(!curlx_sztouz(rv, pnwritten))
|
|
return CURLE_SEND_ERROR;
|
|
|
|
return CURLE_OK;
|
|
}
|
|
|
|
void Curl_rtmp_version(char *version, size_t len)
|
|
{
|
|
char suff[2];
|
|
if(RTMP_LIB_VERSION & 0xff) {
|
|
suff[0] = (RTMP_LIB_VERSION & 0xff) + 'a' - 1;
|
|
suff[1] = '\0';
|
|
}
|
|
else
|
|
suff[0] = '\0';
|
|
|
|
curl_msnprintf(version, len, "librtmp/%d.%d%s",
|
|
RTMP_LIB_VERSION >> 16, (RTMP_LIB_VERSION >> 8) & 0xff,
|
|
suff);
|
|
}
|
|
|
|
#endif /* USE_LIBRTMP */
|