12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046 |
- /***************************************************************************
- * _ _ ____ _
- * 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)
- #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);
- 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 are 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, const 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 */
- ZERO_NULL, /* write_resp_hd */
- 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);
- }
- if(data->set.rtspreq == RTSPREQ_RECEIVE &&
- data->req.eos_written) {
- failf(data, "Server prematurely closed the RTSP connection.");
- return CURLE_RECV_ERROR;
- }
- }
- 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;
- 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;
- /* Initialize a dynamic send buffer */
- Curl_dyn_init(&req_buffer, DYN_RTSP_REQ_HEADER);
- 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 do not 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_setup1(data, CURL_XFER_RECV, -1, TRUE);
- goto out;
- }
- p_session_id = data->set.str[STRING_RTSP_SESSION_ID];
- if(!p_session_id &&
- (rtspreq & ~(Curl_RtspReq)(RTSPREQ_OPTIONS |
- RTSPREQ_DESCRIBE |
- RTSPREQ_SETUP))) {
- failf(data, "Refusing to issue an RTSP request [%s] without a session ID.",
- p_request);
- result = CURLE_BAD_FUNCTION_ARGUMENT;
- goto out;
- }
- /* 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.");
- result = CURLE_BAD_FUNCTION_ARGUMENT;
- goto out;
- }
- 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) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- 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)
- goto out;
- #ifndef CURL_DISABLE_PROXY
- p_proxyuserpwd = data->state.aptr.proxyuserpwd;
- #endif
- 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.");
- result = CURLE_RTSP_CSEQ_ERROR;
- goto out;
- }
- if(Curl_checkheaders(data, STRCONST("Session"))) {
- failf(data, "Session ID cannot be set as a custom header.");
- result = CURLE_BAD_FUNCTION_ARGUMENT;
- goto out;
- }
- 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)
- goto out;
- /*
- * 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)
- goto out;
- }
- /*
- * 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)
- goto out;
- if((rtspreq == RTSPREQ_SETUP) || (rtspreq == RTSPREQ_DESCRIBE)) {
- result = Curl_add_timecondition(data, &req_buffer);
- if(result)
- goto out;
- }
- result = Curl_add_custom_headers(data, FALSE, &req_buffer);
- if(result)
- goto out;
- if(rtspreq == RTSPREQ_ANNOUNCE ||
- rtspreq == RTSPREQ_SET_PARAMETER ||
- rtspreq == RTSPREQ_GET_PARAMETER) {
- curl_off_t req_clen; /* request content length */
- if(data->state.upload) {
- req_clen = data->state.infilesize;
- data->state.httpreq = HTTPREQ_PUT;
- result = Curl_creader_set_fread(data, req_clen);
- if(result)
- goto out;
- }
- else {
- if(data->set.postfields) {
- size_t plen = strlen(data->set.postfields);
- req_clen = (curl_off_t)plen;
- result = Curl_creader_set_buf(data, data->set.postfields, plen);
- }
- else if(data->state.infilesize >= 0) {
- req_clen = data->state.infilesize;
- result = Curl_creader_set_fread(data, req_clen);
- }
- else {
- req_clen = 0;
- result = Curl_creader_set_null(data);
- }
- if(result)
- goto out;
- }
- if(req_clen > 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: %" FMT_OFF_T"\r\n",
- req_clen);
- if(result)
- goto out;
- }
- 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)
- goto out;
- }
- }
- 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)
- goto out;
- }
- }
- }
- else if(rtspreq == RTSPREQ_GET_PARAMETER) {
- /* Check for an empty GET_PARAMETER (heartbeat) request */
- data->state.httpreq = HTTPREQ_HEAD;
- data->req.no_body = TRUE;
- }
- }
- else {
- result = Curl_creader_set_null(data);
- if(result)
- goto out;
- }
- /* Finish the request buffer */
- result = Curl_dyn_addn(&req_buffer, STRCONST("\r\n"));
- if(result)
- goto out;
- Curl_xfer_setup1(data, CURL_XFER_SENDRECV, -1, TRUE);
- /* issue the request */
- result = Curl_req_send(data, &req_buffer);
- if(result) {
- failf(data, "Failed sending RTSP request");
- goto out;
- }
- /* 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;
- }
- out:
- Curl_dyn_free(&req_buffer);
- 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)
- {
- struct rtsp_conn *rtspc = &(data->conn->proto.rtspc);
- CURLcode result = CURLE_OK;
- size_t consumed = 0;
- if(!data->req.header)
- rtspc->in_header = 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 ongoing, 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);
- 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)
- data->req.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=%" FMT_OFF_T ")",
- blen, rtspc->in_header, data->req.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, const char *header)
- {
- if(checkprefix("CSeq:", header)) {
- long CSeq = 0;
- char *endp;
- const 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)) {
- const char *start, *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 is not 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, const 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' */
- const char *start, *end;
- start = transport;
- while(start && *start) {
- while(*start && ISBLANK(*start) )
- start++;
- end = strchr(start, ';');
- if(checkprefix("interleaved=", start)) {
- long chan1, chan2, chan;
- char *endp;
- const 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 */
|