mirror of
https://github.com/curl/curl.git
synced 2026-04-11 12:01:42 +08:00
eb14705280
Add protocol.h and protocol.c containing all about libcurl's known URI schemes and their protocol handlers (so they exist). Moves the scheme definitions from the various sources files into protocol.c. Schemes are known and used, even of the protocol handler is not build or just not implemented at all. Closes #20906
249 lines
7.0 KiB
C
249 lines
7.0 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"
|
|
|
|
#include "curl_rtmp.h"
|
|
#include "urldata.h"
|
|
|
|
#ifdef USE_LIBRTMP
|
|
|
|
#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 Curl_recv rtmp_recv;
|
|
static Curl_send rtmp_send;
|
|
|
|
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);
|
|
}
|
|
|
|
/*
|
|
* RTMP protocol handler.h, based on https://rtmpdump.mplayerhq.hu/
|
|
*/
|
|
|
|
const struct Curl_protocol Curl_protocol_rtmp = {
|
|
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_is_dead */
|
|
ZERO_NULL, /* attach connection */
|
|
ZERO_NULL, /* follow */
|
|
};
|
|
|
|
#endif /* USE_LIBRTMP */
|