1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
- *
- * 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"
- #if !defined(CURL_DISABLE_RTSP) && !defined(USE_HYPER)
- #include "urldata.h"
- #include <curl/curl.h>
- #include "transfer.h"
- #include "sendf.h"
- #include "multiif.h"
- #include "http.h"
- #include "url.h"
- #include "progress.h"
- #include "rtsp.h"
- #include "strcase.h"
- #include "select.h"
- #include "connect.h"
- #include "cfilters.h"
- #include "strdup.h"
- /* The last 3 #include files should be in this order */
- #include "curl_printf.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- #define RTP_PKT_LENGTH(p) ((((unsigned int)((unsigned char)((p)[2]))) << 8) | \
- ((unsigned int)((unsigned char)((p)[3]))))
- /* protocol-specific functions set up to be called by the main engine */
- static CURLcode rtsp_do(struct Curl_easy *data, bool *done);
- static CURLcode rtsp_done(struct Curl_easy *data, CURLcode, bool premature);
- static CURLcode rtsp_connect(struct Curl_easy *data, bool *done);
- static CURLcode rtsp_disconnect(struct Curl_easy *data,
- struct connectdata *conn, bool dead);
- static int rtsp_getsock_do(struct Curl_easy *data,
- struct connectdata *conn, curl_socket_t *socks);
- /*
- * Parse and write out an RTSP response.
- * @param data the transfer
- * @param conn the connection
- * @param buf data read from connection
- * @param blen amount of data in buf
- * @param is_eos TRUE iff this is the last write
- * @param readmore out, TRUE iff complete buf was consumed and more data
- * is needed
- */
- static CURLcode rtsp_rtp_write_resp(struct Curl_easy *data,
- const char *buf,
- size_t blen,
- bool is_eos,
- bool *done);
- static CURLcode rtsp_setup_connection(struct Curl_easy *data,
- struct connectdata *conn);
- static unsigned int rtsp_conncheck(struct Curl_easy *data,
- struct connectdata *check,
- unsigned int checks_to_perform);
- /* this returns the socket to wait for in the DO and DOING state for the multi
- interface and then we're always _sending_ a request and thus we wait for
- the single socket to become writable only */
- static int rtsp_getsock_do(struct Curl_easy *data, struct connectdata *conn,
- curl_socket_t *socks)
- {
- /* write mode */
- (void)data;
- socks[0] = conn->sock[FIRSTSOCKET];
- return GETSOCK_WRITESOCK(0);
- }
- static
- CURLcode rtp_client_write(struct Curl_easy *data, const char *ptr, size_t len);
- static
- CURLcode rtsp_parse_transport(struct Curl_easy *data, char *transport);
- /*
- * RTSP handler interface.
- */
- const struct Curl_handler Curl_handler_rtsp = {
- "RTSP", /* scheme */
- rtsp_setup_connection, /* setup_connection */
- rtsp_do, /* do_it */
- rtsp_done, /* done */
- ZERO_NULL, /* do_more */
- rtsp_connect, /* connect_it */
- ZERO_NULL, /* connecting */
- ZERO_NULL, /* doing */
- ZERO_NULL, /* proto_getsock */
- rtsp_getsock_do, /* doing_getsock */
- ZERO_NULL, /* domore_getsock */
- ZERO_NULL, /* perform_getsock */
- rtsp_disconnect, /* disconnect */
- rtsp_rtp_write_resp, /* write_resp */
- rtsp_conncheck, /* connection_check */
- ZERO_NULL, /* attach connection */
- PORT_RTSP, /* defport */
- CURLPROTO_RTSP, /* protocol */
- CURLPROTO_RTSP, /* family */
- PROTOPT_NONE /* flags */
- };
- #define MAX_RTP_BUFFERSIZE 1000000 /* arbitrary */
- static CURLcode rtsp_setup_connection(struct Curl_easy *data,
- struct connectdata *conn)
- {
- struct RTSP *rtsp;
- (void)conn;
- data->req.p.rtsp = rtsp = calloc(1, sizeof(struct RTSP));
- if(!rtsp)
- return CURLE_OUT_OF_MEMORY;
- Curl_dyn_init(&conn->proto.rtspc.buf, MAX_RTP_BUFFERSIZE);
- return CURLE_OK;
- }
- /*
- * Function to check on various aspects of a connection.
- */
- static unsigned int rtsp_conncheck(struct Curl_easy *data,
- struct connectdata *conn,
- unsigned int checks_to_perform)
- {
- unsigned int ret_val = CONNRESULT_NONE;
- (void)data;
- if(checks_to_perform & CONNCHECK_ISDEAD) {
- bool input_pending;
- if(!Curl_conn_is_alive(data, conn, &input_pending))
- ret_val |= CONNRESULT_DEAD;
- }
- return ret_val;
- }
- static CURLcode rtsp_connect(struct Curl_easy *data, bool *done)
- {
- CURLcode httpStatus;
- httpStatus = Curl_http_connect(data, done);
- /* Initialize the CSeq if not already done */
- if(data->state.rtsp_next_client_CSeq == 0)
- data->state.rtsp_next_client_CSeq = 1;
- if(data->state.rtsp_next_server_CSeq == 0)
- data->state.rtsp_next_server_CSeq = 1;
- data->conn->proto.rtspc.rtp_channel = -1;
- return httpStatus;
- }
- static CURLcode rtsp_disconnect(struct Curl_easy *data,
- struct connectdata *conn, bool dead)
- {
- (void) dead;
- (void) data;
- Curl_dyn_free(&conn->proto.rtspc.buf);
- return CURLE_OK;
- }
- static CURLcode rtsp_done(struct Curl_easy *data,
- CURLcode status, bool premature)
- {
- struct RTSP *rtsp = data->req.p.rtsp;
- CURLcode httpStatus;
- /* Bypass HTTP empty-reply checks on receive */
- if(data->set.rtspreq == RTSPREQ_RECEIVE)
- premature = TRUE;
- httpStatus = Curl_http_done(data, status, premature);
- if(rtsp && !status && !httpStatus) {
- /* Check the sequence numbers */
- long CSeq_sent = rtsp->CSeq_sent;
- long CSeq_recv = rtsp->CSeq_recv;
- if((data->set.rtspreq != RTSPREQ_RECEIVE) && (CSeq_sent != CSeq_recv)) {
- failf(data,
- "The CSeq of this request %ld did not match the response %ld",
- CSeq_sent, CSeq_recv);
- return CURLE_RTSP_CSEQ_ERROR;
- }
- if(data->set.rtspreq == RTSPREQ_RECEIVE &&
- (data->conn->proto.rtspc.rtp_channel == -1)) {
- infof(data, "Got an RTP Receive with a CSeq of %ld", CSeq_recv);
- }
- }
- return httpStatus;
- }
- static CURLcode rtsp_do(struct Curl_easy *data, bool *done)
- {
- struct connectdata *conn = data->conn;
- CURLcode result = CURLE_OK;
- Curl_RtspReq rtspreq = data->set.rtspreq;
- struct RTSP *rtsp = data->req.p.rtsp;
- struct dynbuf req_buffer;
- curl_off_t postsize = 0; /* for ANNOUNCE and SET_PARAMETER */
- curl_off_t putsize = 0; /* for ANNOUNCE and SET_PARAMETER */
- const char *p_request = NULL;
- const char *p_session_id = NULL;
- const char *p_accept = NULL;
- const char *p_accept_encoding = NULL;
- const char *p_range = NULL;
- const char *p_referrer = NULL;
- const char *p_stream_uri = NULL;
- const char *p_transport = NULL;
- const char *p_uagent = NULL;
- const char *p_proxyuserpwd = NULL;
- const char *p_userpwd = NULL;
- *done = TRUE;
- rtsp->CSeq_sent = data->state.rtsp_next_client_CSeq;
- rtsp->CSeq_recv = 0;
- /* Setup the first_* fields to allow auth details get sent
- to this origin */
- if(!data->state.first_host) {
- data->state.first_host = strdup(conn->host.name);
- if(!data->state.first_host)
- return CURLE_OUT_OF_MEMORY;
- data->state.first_remote_port = conn->remote_port;
- data->state.first_remote_protocol = conn->handler->protocol;
- }
- /* Setup the 'p_request' pointer to the proper p_request string
- * Since all RTSP requests are included here, there is no need to
- * support custom requests like HTTP.
- **/
- data->req.no_body = TRUE; /* most requests don't contain a body */
- switch(rtspreq) {
- default:
- failf(data, "Got invalid RTSP request");
- return CURLE_BAD_FUNCTION_ARGUMENT;
- case RTSPREQ_OPTIONS:
- p_request = "OPTIONS";
- break;
- case RTSPREQ_DESCRIBE:
- p_request = "DESCRIBE";
- data->req.no_body = FALSE;
- break;
- case RTSPREQ_ANNOUNCE:
- p_request = "ANNOUNCE";
- break;
- case RTSPREQ_SETUP:
- p_request = "SETUP";
- break;
- case RTSPREQ_PLAY:
- p_request = "PLAY";
- break;
- case RTSPREQ_PAUSE:
- p_request = "PAUSE";
- break;
- case RTSPREQ_TEARDOWN:
- p_request = "TEARDOWN";
- break;
- case RTSPREQ_GET_PARAMETER:
- /* GET_PARAMETER's no_body status is determined later */
- p_request = "GET_PARAMETER";
- data->req.no_body = FALSE;
- break;
- case RTSPREQ_SET_PARAMETER:
- p_request = "SET_PARAMETER";
- break;
- case RTSPREQ_RECORD:
- p_request = "RECORD";
- break;
- case RTSPREQ_RECEIVE:
- p_request = "";
- /* Treat interleaved RTP as body */
- data->req.no_body = FALSE;
- break;
- case RTSPREQ_LAST:
- failf(data, "Got invalid RTSP request: RTSPREQ_LAST");
- return CURLE_BAD_FUNCTION_ARGUMENT;
- }
- if(rtspreq == RTSPREQ_RECEIVE) {
- Curl_xfer_setup(data, FIRSTSOCKET, -1, TRUE, -1);
- return result;
- }
- p_session_id = data->set.str[STRING_RTSP_SESSION_ID];
- if(!p_session_id &&
- (rtspreq & ~(RTSPREQ_OPTIONS | RTSPREQ_DESCRIBE | RTSPREQ_SETUP))) {
- failf(data, "Refusing to issue an RTSP request [%s] without a session ID.",
- p_request);
- return CURLE_BAD_FUNCTION_ARGUMENT;
- }
- /* Stream URI. Default to server '*' if not specified */
- if(data->set.str[STRING_RTSP_STREAM_URI]) {
- p_stream_uri = data->set.str[STRING_RTSP_STREAM_URI];
- }
- else {
- p_stream_uri = "*";
- }
- /* Transport Header for SETUP requests */
- p_transport = Curl_checkheaders(data, STRCONST("Transport"));
- if(rtspreq == RTSPREQ_SETUP && !p_transport) {
- /* New Transport: setting? */
- if(data->set.str[STRING_RTSP_TRANSPORT]) {
- Curl_safefree(data->state.aptr.rtsp_transport);
- data->state.aptr.rtsp_transport =
- aprintf("Transport: %s\r\n",
- data->set.str[STRING_RTSP_TRANSPORT]);
- if(!data->state.aptr.rtsp_transport)
- return CURLE_OUT_OF_MEMORY;
- }
- else {
- failf(data,
- "Refusing to issue an RTSP SETUP without a Transport: header.");
- return CURLE_BAD_FUNCTION_ARGUMENT;
- }
- p_transport = data->state.aptr.rtsp_transport;
- }
- /* Accept Headers for DESCRIBE requests */
- if(rtspreq == RTSPREQ_DESCRIBE) {
- /* Accept Header */
- p_accept = Curl_checkheaders(data, STRCONST("Accept"))?
- NULL:"Accept: application/sdp\r\n";
- /* Accept-Encoding header */
- if(!Curl_checkheaders(data, STRCONST("Accept-Encoding")) &&
- data->set.str[STRING_ENCODING]) {
- Curl_safefree(data->state.aptr.accept_encoding);
- data->state.aptr.accept_encoding =
- aprintf("Accept-Encoding: %s\r\n", data->set.str[STRING_ENCODING]);
- if(!data->state.aptr.accept_encoding)
- return CURLE_OUT_OF_MEMORY;
- p_accept_encoding = data->state.aptr.accept_encoding;
- }
- }
- /* The User-Agent string might have been allocated in url.c already, because
- it might have been used in the proxy connect, but if we have got a header
- with the user-agent string specified, we erase the previously made string
- here. */
- if(Curl_checkheaders(data, STRCONST("User-Agent")) &&
- data->state.aptr.uagent) {
- Curl_safefree(data->state.aptr.uagent);
- }
- else if(!Curl_checkheaders(data, STRCONST("User-Agent")) &&
- data->set.str[STRING_USERAGENT]) {
- p_uagent = data->state.aptr.uagent;
- }
- /* setup the authentication headers */
- result = Curl_http_output_auth(data, conn, p_request, HTTPREQ_GET,
- p_stream_uri, FALSE);
- if(result)
- return result;
- p_proxyuserpwd = data->state.aptr.proxyuserpwd;
- p_userpwd = data->state.aptr.userpwd;
- /* Referrer */
- Curl_safefree(data->state.aptr.ref);
- if(data->state.referer && !Curl_checkheaders(data, STRCONST("Referer")))
- data->state.aptr.ref = aprintf("Referer: %s\r\n", data->state.referer);
- p_referrer = data->state.aptr.ref;
- /*
- * Range Header
- * Only applies to PLAY, PAUSE, RECORD
- *
- * Go ahead and use the Range stuff supplied for HTTP
- */
- if(data->state.use_range &&
- (rtspreq & (RTSPREQ_PLAY | RTSPREQ_PAUSE | RTSPREQ_RECORD))) {
- /* Check to see if there is a range set in the custom headers */
- if(!Curl_checkheaders(data, STRCONST("Range")) && data->state.range) {
- Curl_safefree(data->state.aptr.rangeline);
- data->state.aptr.rangeline = aprintf("Range: %s\r\n", data->state.range);
- p_range = data->state.aptr.rangeline;
- }
- }
- /*
- * Sanity check the custom headers
- */
- if(Curl_checkheaders(data, STRCONST("CSeq"))) {
- failf(data, "CSeq cannot be set as a custom header.");
- return CURLE_RTSP_CSEQ_ERROR;
- }
- if(Curl_checkheaders(data, STRCONST("Session"))) {
- failf(data, "Session ID cannot be set as a custom header.");
- return CURLE_BAD_FUNCTION_ARGUMENT;
- }
- /* Initialize a dynamic send buffer */
- Curl_dyn_init(&req_buffer, DYN_RTSP_REQ_HEADER);
- result =
- Curl_dyn_addf(&req_buffer,
- "%s %s RTSP/1.0\r\n" /* Request Stream-URI RTSP/1.0 */
- "CSeq: %ld\r\n", /* CSeq */
- p_request, p_stream_uri, rtsp->CSeq_sent);
- if(result)
- return result;
- /*
- * Rather than do a normal alloc line, keep the session_id unformatted
- * to make comparison easier
- */
- if(p_session_id) {
- result = Curl_dyn_addf(&req_buffer, "Session: %s\r\n", p_session_id);
- if(result)
- return result;
- }
- /*
- * Shared HTTP-like options
- */
- result = Curl_dyn_addf(&req_buffer,
- "%s" /* transport */
- "%s" /* accept */
- "%s" /* accept-encoding */
- "%s" /* range */
- "%s" /* referrer */
- "%s" /* user-agent */
- "%s" /* proxyuserpwd */
- "%s" /* userpwd */
- ,
- p_transport ? p_transport : "",
- p_accept ? p_accept : "",
- p_accept_encoding ? p_accept_encoding : "",
- p_range ? p_range : "",
- p_referrer ? p_referrer : "",
- p_uagent ? p_uagent : "",
- p_proxyuserpwd ? p_proxyuserpwd : "",
- p_userpwd ? p_userpwd : "");
- /*
- * Free userpwd now --- cannot reuse this for Negotiate and possibly NTLM
- * with basic and digest, it will be freed anyway by the next request
- */
- Curl_safefree(data->state.aptr.userpwd);
- if(result)
- return result;
- if((rtspreq == RTSPREQ_SETUP) || (rtspreq == RTSPREQ_DESCRIBE)) {
- result = Curl_add_timecondition(data, &req_buffer);
- if(result)
- return result;
- }
- result = Curl_add_custom_headers(data, FALSE, &req_buffer);
- if(result)
- return result;
- if(rtspreq == RTSPREQ_ANNOUNCE ||
- rtspreq == RTSPREQ_SET_PARAMETER ||
- rtspreq == RTSPREQ_GET_PARAMETER) {
- if(data->state.upload) {
- putsize = data->state.infilesize;
- data->state.httpreq = HTTPREQ_PUT;
- }
- else {
- postsize = (data->state.infilesize != -1)?
- data->state.infilesize:
- (data->set.postfields? (curl_off_t)strlen(data->set.postfields):0);
- data->state.httpreq = HTTPREQ_POST;
- }
- if(putsize > 0 || postsize > 0) {
- /* As stated in the http comments, it is probably not wise to
- * actually set a custom Content-Length in the headers */
- if(!Curl_checkheaders(data, STRCONST("Content-Length"))) {
- result =
- Curl_dyn_addf(&req_buffer,
- "Content-Length: %" CURL_FORMAT_CURL_OFF_T"\r\n",
- (data->state.upload ? putsize : postsize));
- if(result)
- return result;
- }
- if(rtspreq == RTSPREQ_SET_PARAMETER ||
- rtspreq == RTSPREQ_GET_PARAMETER) {
- if(!Curl_checkheaders(data, STRCONST("Content-Type"))) {
- result = Curl_dyn_addn(&req_buffer,
- STRCONST("Content-Type: "
- "text/parameters\r\n"));
- if(result)
- return result;
- }
- }
- if(rtspreq == RTSPREQ_ANNOUNCE) {
- if(!Curl_checkheaders(data, STRCONST("Content-Type"))) {
- result = Curl_dyn_addn(&req_buffer,
- STRCONST("Content-Type: "
- "application/sdp\r\n"));
- if(result)
- return result;
- }
- }
- data->state.expect100header = FALSE; /* RTSP posts are simple/small */
- }
- else if(rtspreq == RTSPREQ_GET_PARAMETER) {
- /* Check for an empty GET_PARAMETER (heartbeat) request */
- data->state.httpreq = HTTPREQ_HEAD;
- data->req.no_body = TRUE;
- }
- }
- /* RTSP never allows chunked transfer */
- data->req.forbidchunk = TRUE;
- /* Finish the request buffer */
- result = Curl_dyn_addn(&req_buffer, STRCONST("\r\n"));
- if(result)
- return result;
- if(postsize > 0) {
- result = Curl_dyn_addn(&req_buffer, data->set.postfields,
- (size_t)postsize);
- if(result)
- return result;
- }
- /* issue the request */
- result = Curl_req_send_hds(data, Curl_dyn_ptr(&req_buffer),
- Curl_dyn_len(&req_buffer));
- Curl_dyn_free(&req_buffer);
- if(result) {
- failf(data, "Failed sending RTSP request");
- return result;
- }
- Curl_xfer_setup(data, FIRSTSOCKET, -1, TRUE, putsize?FIRSTSOCKET:-1);
- /* Increment the CSeq on success */
- data->state.rtsp_next_client_CSeq++;
- if(data->req.writebytecount) {
- /* if a request-body has been sent off, we make sure this progress is
- noted properly */
- Curl_pgrsSetUploadCounter(data, data->req.writebytecount);
- if(Curl_pgrsUpdate(data))
- result = CURLE_ABORTED_BY_CALLBACK;
- }
- return result;
- }
- /**
- * write any BODY bytes missing to the client, ignore the rest.
- */
- static CURLcode rtp_write_body_junk(struct Curl_easy *data,
- const char *buf,
- size_t blen)
- {
- struct rtsp_conn *rtspc = &(data->conn->proto.rtspc);
- curl_off_t body_remain;
- bool in_body;
- in_body = (data->req.headerline && !rtspc->in_header) &&
- (data->req.size >= 0) &&
- (data->req.bytecount < data->req.size);
- body_remain = in_body? (data->req.size - data->req.bytecount) : 0;
- DEBUGASSERT(body_remain >= 0);
- if(body_remain) {
- if((curl_off_t)blen > body_remain)
- blen = (size_t)body_remain;
- return Curl_client_write(data, CLIENTWRITE_BODY, (char *)buf, blen);
- }
- return CURLE_OK;
- }
- static CURLcode rtsp_filter_rtp(struct Curl_easy *data,
- const char *buf,
- size_t blen,
- size_t *pconsumed)
- {
- struct rtsp_conn *rtspc = &(data->conn->proto.rtspc);
- CURLcode result = CURLE_OK;
- size_t skip_len = 0;
- *pconsumed = 0;
- while(blen) {
- bool in_body = (data->req.headerline && !rtspc->in_header) &&
- (data->req.size >= 0) &&
- (data->req.bytecount < data->req.size);
- switch(rtspc->state) {
- case RTP_PARSE_SKIP: {
- DEBUGASSERT(Curl_dyn_len(&rtspc->buf) == 0);
- while(blen && buf[0] != '$') {
- if(!in_body && buf[0] == 'R' &&
- data->set.rtspreq != RTSPREQ_RECEIVE) {
- if(strncmp(buf, "RTSP/", (blen < 5) ? blen : 5) == 0) {
- /* This could be the next response, no consume and return */
- if(*pconsumed) {
- DEBUGF(infof(data, "RTP rtsp_filter_rtp[SKIP] RTSP/ prefix, "
- "skipping %zd bytes of junk", *pconsumed));
- }
- rtspc->state = RTP_PARSE_SKIP;
- rtspc->in_header = TRUE;
- goto out;
- }
- }
- /* junk/BODY, consume without buffering */
- *pconsumed += 1;
- ++buf;
- --blen;
- ++skip_len;
- }
- if(blen && buf[0] == '$') {
- /* possible start of an RTP message, buffer */
- if(skip_len) {
- /* end of junk/BODY bytes, flush */
- result = rtp_write_body_junk(data,
- (char *)(buf - skip_len), skip_len);
- skip_len = 0;
- if(result)
- goto out;
- }
- if(Curl_dyn_addn(&rtspc->buf, buf, 1)) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- *pconsumed += 1;
- ++buf;
- --blen;
- rtspc->state = RTP_PARSE_CHANNEL;
- }
- break;
- }
- case RTP_PARSE_CHANNEL: {
- int idx = ((unsigned char)buf[0]) / 8;
- int off = ((unsigned char)buf[0]) % 8;
- DEBUGASSERT(Curl_dyn_len(&rtspc->buf) == 1);
- if(!(data->state.rtp_channel_mask[idx] & (1 << off))) {
- /* invalid channel number, junk or BODY data */
- rtspc->state = RTP_PARSE_SKIP;
- DEBUGASSERT(skip_len == 0);
- /* we do not consume this byte, it is BODY data */
- DEBUGF(infof(data, "RTSP: invalid RTP channel %d, skipping", idx));
- if(*pconsumed == 0) {
- /* We did not consume the initial '$' in our buffer, but had
- * it from an earlier call. We cannot un-consume it and have
- * to write it directly as BODY data */
- result = rtp_write_body_junk(data, Curl_dyn_ptr(&rtspc->buf), 1);
- if(result)
- goto out;
- }
- else {
- /* count the '$' as skip and continue */
- skip_len = 1;
- }
- Curl_dyn_free(&rtspc->buf);
- break;
- }
- /* a valid channel, so we expect this to be a real RTP message */
- rtspc->rtp_channel = (unsigned char)buf[0];
- if(Curl_dyn_addn(&rtspc->buf, buf, 1)) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- *pconsumed += 1;
- ++buf;
- --blen;
- rtspc->state = RTP_PARSE_LEN;
- break;
- }
- case RTP_PARSE_LEN: {
- size_t rtp_len = Curl_dyn_len(&rtspc->buf);
- const char *rtp_buf;
- DEBUGASSERT(rtp_len >= 2 && rtp_len < 4);
- if(Curl_dyn_addn(&rtspc->buf, buf, 1)) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- *pconsumed += 1;
- ++buf;
- --blen;
- if(rtp_len == 2)
- break;
- rtp_buf = Curl_dyn_ptr(&rtspc->buf);
- rtspc->rtp_len = RTP_PKT_LENGTH(rtp_buf) + 4;
- rtspc->state = RTP_PARSE_DATA;
- break;
- }
- case RTP_PARSE_DATA: {
- size_t rtp_len = Curl_dyn_len(&rtspc->buf);
- size_t needed;
- DEBUGASSERT(rtp_len < rtspc->rtp_len);
- needed = rtspc->rtp_len - rtp_len;
- if(needed <= blen) {
- if(Curl_dyn_addn(&rtspc->buf, buf, needed)) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- *pconsumed += needed;
- buf += needed;
- blen -= needed;
- /* complete RTP message in buffer */
- DEBUGF(infof(data, "RTP write channel %d rtp_len %zu",
- rtspc->rtp_channel, rtspc->rtp_len));
- result = rtp_client_write(data, Curl_dyn_ptr(&rtspc->buf),
- rtspc->rtp_len);
- Curl_dyn_free(&rtspc->buf);
- rtspc->state = RTP_PARSE_SKIP;
- if(result)
- goto out;
- }
- else {
- if(Curl_dyn_addn(&rtspc->buf, buf, blen)) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- *pconsumed += blen;
- buf += blen;
- blen = 0;
- }
- break;
- }
- default:
- DEBUGASSERT(0);
- return CURLE_RECV_ERROR;
- }
- }
- out:
- if(!result && skip_len)
- result = rtp_write_body_junk(data, (char *)(buf - skip_len), skip_len);
- return result;
- }
- static CURLcode rtsp_rtp_write_resp(struct Curl_easy *data,
- const char *buf,
- size_t blen,
- bool is_eos,
- bool *done)
- {
- struct rtsp_conn *rtspc = &(data->conn->proto.rtspc);
- CURLcode result = CURLE_OK;
- size_t consumed = 0;
- if(!data->req.header)
- rtspc->in_header = FALSE;
- *done = FALSE;
- if(!blen) {
- goto out;
- }
- DEBUGF(infof(data, "rtsp_rtp_write_resp(len=%zu, in_header=%d, eos=%d)",
- blen, rtspc->in_header, is_eos));
- /* If header parsing is not onging, extract RTP messages */
- if(!rtspc->in_header) {
- result = rtsp_filter_rtp(data, buf, blen, &consumed);
- if(result)
- goto out;
- buf += consumed;
- blen -= consumed;
- /* either we consumed all or are at the start of header parsing */
- if(blen && !data->req.header)
- DEBUGF(infof(data, "RTSP: %zu bytes, possibly excess in response body",
- blen));
- }
- /* we want to parse headers, do so */
- if(data->req.header && blen) {
- rtspc->in_header = TRUE;
- result = Curl_http_write_resp_hds(data, buf, blen, &consumed, done);
- if(result)
- goto out;
- buf += consumed;
- blen -= consumed;
- if(!data->req.header)
- rtspc->in_header = FALSE;
- if(!rtspc->in_header) {
- /* If header parsing is done, extract interleaved RTP messages */
- if(data->req.size <= -1) {
- /* Respect section 4.4 of rfc2326: If the Content-Length header is
- absent, a length 0 must be assumed. */
- data->req.size = 0;
- data->req.download_done = TRUE;
- }
- result = rtsp_filter_rtp(data, buf, blen, &consumed);
- if(result)
- goto out;
- blen -= consumed;
- }
- }
- if(rtspc->state != RTP_PARSE_SKIP)
- *done = FALSE;
- /* we SHOULD have consumed all bytes, unless the response is borked.
- * In which case we write out the left over bytes, letting the client
- * writer deal with it (it will report EXCESS and fail the transfer). */
- DEBUGF(infof(data, "rtsp_rtp_write_resp(len=%zu, in_header=%d, done=%d "
- " rtspc->state=%d, req.size=%" CURL_FORMAT_CURL_OFF_T ")",
- blen, rtspc->in_header, *done, rtspc->state, data->req.size));
- if(!result && (is_eos || blen)) {
- result = Curl_client_write(data, CLIENTWRITE_BODY|
- (is_eos? CLIENTWRITE_EOS:0),
- (char *)buf, blen);
- }
- out:
- if((data->set.rtspreq == RTSPREQ_RECEIVE) &&
- (rtspc->state == RTP_PARSE_SKIP)) {
- /* In special mode RECEIVE, we just process one chunk of network
- * data, so we stop the transfer here, if we have no incomplete
- * RTP message pending. */
- data->req.download_done = TRUE;
- }
- return result;
- }
- static
- CURLcode rtp_client_write(struct Curl_easy *data, const char *ptr, size_t len)
- {
- size_t wrote;
- curl_write_callback writeit;
- void *user_ptr;
- if(len == 0) {
- failf(data, "Cannot write a 0 size RTP packet.");
- return CURLE_WRITE_ERROR;
- }
- /* If the user has configured CURLOPT_INTERLEAVEFUNCTION then use that
- function and any configured CURLOPT_INTERLEAVEDATA to write out the RTP
- data. Otherwise, use the CURLOPT_WRITEFUNCTION with the CURLOPT_WRITEDATA
- pointer to write out the RTP data. */
- if(data->set.fwrite_rtp) {
- writeit = data->set.fwrite_rtp;
- user_ptr = data->set.rtp_out;
- }
- else {
- writeit = data->set.fwrite_func;
- user_ptr = data->set.out;
- }
- Curl_set_in_callback(data, true);
- wrote = writeit((char *)ptr, 1, len, user_ptr);
- Curl_set_in_callback(data, false);
- if(CURL_WRITEFUNC_PAUSE == wrote) {
- failf(data, "Cannot pause RTP");
- return CURLE_WRITE_ERROR;
- }
- if(wrote != len) {
- failf(data, "Failed writing RTP data");
- return CURLE_WRITE_ERROR;
- }
- return CURLE_OK;
- }
- CURLcode Curl_rtsp_parseheader(struct Curl_easy *data, char *header)
- {
- if(checkprefix("CSeq:", header)) {
- long CSeq = 0;
- char *endp;
- char *p = &header[5];
- while(ISBLANK(*p))
- p++;
- CSeq = strtol(p, &endp, 10);
- if(p != endp) {
- struct RTSP *rtsp = data->req.p.rtsp;
- rtsp->CSeq_recv = CSeq; /* mark the request */
- data->state.rtsp_CSeq_recv = CSeq; /* update the handle */
- }
- else {
- failf(data, "Unable to read the CSeq header: [%s]", header);
- return CURLE_RTSP_CSEQ_ERROR;
- }
- }
- else if(checkprefix("Session:", header)) {
- char *start;
- char *end;
- size_t idlen;
- /* Find the first non-space letter */
- start = header + 8;
- while(*start && ISBLANK(*start))
- start++;
- if(!*start) {
- failf(data, "Got a blank Session ID");
- return CURLE_RTSP_SESSION_ERROR;
- }
- /* Find the end of Session ID
- *
- * Allow any non whitespace content, up to the field separator or end of
- * line. RFC 2326 isn't 100% clear on the session ID and for example
- * gstreamer does url-encoded session ID's not covered by the standard.
- */
- end = start;
- while(*end && *end != ';' && !ISSPACE(*end))
- end++;
- idlen = end - start;
- if(data->set.str[STRING_RTSP_SESSION_ID]) {
- /* If the Session ID is set, then compare */
- if(strlen(data->set.str[STRING_RTSP_SESSION_ID]) != idlen ||
- strncmp(start, data->set.str[STRING_RTSP_SESSION_ID], idlen)) {
- failf(data, "Got RTSP Session ID Line [%s], but wanted ID [%s]",
- start, data->set.str[STRING_RTSP_SESSION_ID]);
- return CURLE_RTSP_SESSION_ERROR;
- }
- }
- else {
- /* If the Session ID is not set, and we find it in a response, then set
- * it.
- */
- /* Copy the id substring into a new buffer */
- data->set.str[STRING_RTSP_SESSION_ID] = Curl_memdup0(start, idlen);
- if(!data->set.str[STRING_RTSP_SESSION_ID])
- return CURLE_OUT_OF_MEMORY;
- }
- }
- else if(checkprefix("Transport:", header)) {
- CURLcode result;
- result = rtsp_parse_transport(data, header + 10);
- if(result)
- return result;
- }
- return CURLE_OK;
- }
- static
- CURLcode rtsp_parse_transport(struct Curl_easy *data, char *transport)
- {
- /* If we receive multiple Transport response-headers, the linterleaved
- channels of each response header is recorded and used together for
- subsequent data validity checks.*/
- /* e.g.: ' RTP/AVP/TCP;unicast;interleaved=5-6' */
- char *start;
- char *end;
- start = transport;
- while(start && *start) {
- while(*start && ISBLANK(*start) )
- start++;
- end = strchr(start, ';');
- if(checkprefix("interleaved=", start)) {
- long chan1, chan2, chan;
- char *endp;
- char *p = start + 12;
- chan1 = strtol(p, &endp, 10);
- if(p != endp && chan1 >= 0 && chan1 <= 255) {
- unsigned char *rtp_channel_mask = data->state.rtp_channel_mask;
- chan2 = chan1;
- if(*endp == '-') {
- p = endp + 1;
- chan2 = strtol(p, &endp, 10);
- if(p == endp || chan2 < 0 || chan2 > 255) {
- infof(data, "Unable to read the interleaved parameter from "
- "Transport header: [%s]", transport);
- chan2 = chan1;
- }
- }
- for(chan = chan1; chan <= chan2; chan++) {
- long idx = chan / 8;
- long off = chan % 8;
- rtp_channel_mask[idx] |= (unsigned char)(1 << off);
- }
- }
- else {
- infof(data, "Unable to read the interleaved parameter from "
- "Transport header: [%s]", transport);
- }
- break;
- }
- /* skip to next parameter */
- start = (!end) ? end : (end + 1);
- }
- return CURLE_OK;
- }
- #endif /* CURL_DISABLE_RTSP or using Hyper */
|