UltraGrid/src/rtsp/BasicRTSPOnlySubsession.cpp

/*
 * FILE:    rtsp/BasicRTSPOnlySubsession.cpp
 * AUTHORS: David Cassany    <david.cassany@i2cat.net>
 *          Gerard Castillo  <gerard.castillo@i2cat.net>
 *
 * Copyright (c) 2005-2010 Fundació i2CAT, Internet I Innovació Digital a Catalunya
 * Copyright (c) 2014-2023 CESNET, z. s. p. o.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, is permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *
 *      This product includes software developed by the Fundació i2CAT,
 *      Internet I Innovació Digital a Catalunya. This product also includes
 *      software developed by CESNET z.s.p.o.
 *
 * 4. Neither the name of the University nor of the Institute may be used
 *    to endorse or promote products derived from this software without
 *    specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
 * EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include "rtsp/BasicRTSPOnlySubsession.hh"
#include <cassert>
#include <BasicUsageEnvironment.hh>
#include <RTSPServer.hh>
#include <GroupsockHelper.hh>

#include "debug.h"                // for MSG
#include "messaging.h"
#include "utils/macros.h"
#include "utils/net.h"
#include "utils/sdp.h"
#include "video_codec.h"          // for get_codec_name

#define MOD_NAME "[RTSP] "

BasicRTSPOnlySubsession*
BasicRTSPOnlySubsession::createNew(UsageEnvironment& env,
		Boolean reuseFirstSource, rtps_types_t avType,
		struct rtsp_server_parameters params) {
	return new BasicRTSPOnlySubsession(env, reuseFirstSource, avType, params);
}

BasicRTSPOnlySubsession::BasicRTSPOnlySubsession(UsageEnvironment& env,
		Boolean reuseFirstSource, rtps_types_t avType,
		struct rtsp_server_parameters params) :
		ServerMediaSubsession(env), fSDPLines(NULL), fReuseFirstSource(
				reuseFirstSource), fLastStreamToken(NULL),
		rtsp_params(params)
{
	Vdestination = NULL;
	Adestination = NULL;
	gethostname(fCNAME, sizeof fCNAME);
	this->avType = avType;
	fCNAME[sizeof fCNAME - 1] = '\0';
}

BasicRTSPOnlySubsession::~BasicRTSPOnlySubsession() {
	delete[] fSDPLines;
	delete Adestination;
	delete Vdestination;
}

char const* BasicRTSPOnlySubsession::sdpLines(int addressFamily) {
	if (fSDPLines == NULL) {
                setSDPLines(addressFamily);
	}
	return fSDPLines;
}

void BasicRTSPOnlySubsession::setSDPLines(int addressFamily) {
	//TODO: should be more dynamic
        const char *ip_ver_list_addr =
            addressFamily == AF_INET ? "4 0.0.0.0" : "6 ::";

	//VStream
	if (avType == video || avType == av) {
		unsigned estBitrate = 5000;
		char const* mediaType = "video";
                char rtpmapLine[STR_LEN];
                const int rtpPayloadType = get_video_rtp_pt_rtpmap(
                    rtsp_params.video_codec, rtpmapLine);
                if (rtpPayloadType < 0) {
                        MSG(ERROR, "Unsupported video codec %s!\n",
                            get_codec_name(rtsp_params.video_codec));
                }
		//char const* auxSDPLine = "";

		char const* const sdpFmt = "m=%s %u RTP/AVP %d\r\n"
				"c=IN IP%s\r\n"
				"b=AS:%u\r\n"
				"a=rtcp:%d\r\n"
				"%s"
				"a=control:%s\r\n";
		unsigned sdpFmtSize = strlen(sdpFmt) + strlen(mediaType) + 5 /* max short len */
				+ 3 /* max char len */
				+ strlen(ip_ver_list_addr) + 20 /* max int len */
				+ strlen(rtpmapLine) + strlen(trackId());
		char* sdpLines = new char[sdpFmtSize];

		snprintf(sdpLines, sdpFmtSize, sdpFmt, mediaType, // m= <media>
				rtsp_params.rtp_port,//fPortNumForSDP, // m= <port>
				rtpPayloadType, // m= <fmt list>
				ip_ver_list_addr, // c= address
				estBitrate, // b=AS:<bandwidth>
				rtsp_params.rtp_port + 1,
				rtpmapLine, // a=rtpmap:... (if present)
				trackId()); // a=control:<track-id>

		fSDPLines = sdpLines;
	}
	//AStream
	if (avType == audio || avType == av) {
		unsigned estBitrate = 384;
		char const* mediaType = "audio";

                char rtpmapLine[STR_LEN];
		//char const* auxSDPLine = "";
                const uint8_t rtpPayloadType = get_audio_rtp_pt_rtpmap(
                    rtsp_params.audio_codec, rtsp_params.audio_sample_rate,
                    rtsp_params.audio_channels, rtpmapLine);

		char const* const sdpFmt = "m=%s %u RTP/AVP %u\r\n"
				"c=IN IP%s\r\n"
				"b=AS:%u\r\n"
				"a=rtcp:%d\r\n"
				"%s"
				"a=control:%s\r\n";
		unsigned sdpFmtSize = strlen(sdpFmt) + strlen(mediaType) + 5 /* max short len */
				+ 3 /* max char len */
				+ strlen(ip_ver_list_addr) + 20 /* max int len */
				+ strlen(rtpmapLine) + strlen(trackId());
		char* sdpLines = new char[sdpFmtSize];

		snprintf(sdpLines, sdpFmtSize, sdpFmt,
				mediaType, // m= <media>
				rtsp_params.rtp_port_audio,//fPortNumForSDP, // m= <port>
				rtpPayloadType, // m= <fmt list>
				ip_ver_list_addr, // c= address
				estBitrate, // b=AS:<bandwidth>
				rtsp_params.rtp_port_audio + 1,
				rtpmapLine, // a=rtpmap:... (if present)
				trackId()); // a=control:<track-id>

		fSDPLines = sdpLines;
	}
	MSG(VERBOSE, "SDP:\n%s\n", fSDPLines);
}

void BasicRTSPOnlySubsession::getStreamParameters(unsigned /* clientSessionId */,
		struct sockaddr_storage const &clientAddress, Port const& clientRTPPort,
		Port const& clientRTCPPort, int /* tcpSocketNum */,
		unsigned char /* rtpChannelId */, unsigned char /* rtcpChannelId */,
                TLSState * /* tlsState */,
		struct sockaddr_storage& /*destinationAddress*/, uint8_t& /*destinationTTL*/,
		Boolean& /* isMulticast */, Port& serverRTPPort, Port& serverRTCPPort,
		void*& /* streamToken */) {
	if (avType == video || avType == av) {
		Port rtp(rtsp_params.rtp_port);
		serverRTPPort = rtp;
		Port rtcp(rtsp_params.rtp_port + 1);
		serverRTCPPort = rtcp;

		delete Vdestination;
		Vdestination = new Destinations(clientAddress, clientRTPPort,
				clientRTCPPort);
	}
	if (avType == audio || avType == av) {
		Port rtp(rtsp_params.rtp_port_audio);
		serverRTPPort = rtp;
		Port rtcp(rtsp_params.rtp_port_audio + 1);
		serverRTCPPort = rtcp;

		delete Adestination;
		Adestination = new Destinations(clientAddress, clientRTPPort,
				clientRTCPPort);
	}
}

void BasicRTSPOnlySubsession::startStream(unsigned /* clientSessionId */,
		void* /* streamToken */, TaskFunc* /* rtcpRRHandler */,
		void* /* rtcpRRHandlerClientData */, unsigned short& /* rtpSeqNum */,
		unsigned& /* rtpTimestamp */,
		ServerRequestAlternativeByteHandler* /* serverRequestAlternativeByteHandler */,
		void* /* serverRequestAlternativeByteHandlerClientData */) {
	struct response *resp = NULL;

	if (Vdestination != NULL) {
		if (avType == video || avType == av) {
			char pathV[1024];

			memset(pathV, 0, sizeof(pathV));
			enum module_class path_sender[] = { MODULE_CLASS_SENDER,
					MODULE_CLASS_NONE };
			append_message_path(pathV, sizeof(pathV), path_sender);

			//CHANGE DST PORT
			struct msg_sender *msgV1 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
			msgV1->tx_port = ntohs(Vdestination->rtpPort.num());
			msgV1->type = SENDER_MSG_CHANGE_PORT;
			resp = send_message(rtsp_params.parent, pathV, (struct message *) msgV1);
			free_response(resp);

			//CHANGE DST ADDRESS
			struct msg_sender *msgV2 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
                        char host[IN6_MAX_ASCII_LEN + 1];
                        const int ret =
                            getnameinfo((struct sockaddr *) &Vdestination->addr,
                                        sizeof Vdestination->addr, host,
                                        sizeof host, nullptr, 0, NI_NUMERICHOST);
                        assert(ret == 0);
			strncpy(msgV2->receiver, host,
					sizeof(msgV2->receiver) - 1);
			msgV2->type = SENDER_MSG_CHANGE_RECEIVER;

			resp = send_message(rtsp_params.parent, pathV, (struct message *) msgV2);
			free_response(resp);
		}
	}

	if (Adestination != NULL) {
		if (avType == audio || avType == av) {
			char pathA[1024];

			memset(pathA, 0, sizeof(pathA));
			enum module_class path_sender[] = { MODULE_CLASS_AUDIO,
					MODULE_CLASS_SENDER, MODULE_CLASS_NONE };
			append_message_path(pathA, sizeof(pathA), path_sender);

			//CHANGE DST PORT
			struct msg_sender *msgA1 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
			msgA1->tx_port = ntohs(Adestination->rtpPort.num());
			msgA1->type = SENDER_MSG_CHANGE_PORT;
			resp = send_message(rtsp_params.parent, pathA, (struct message *) msgA1);
			free_response(resp);
			resp = NULL;

			//CHANGE DST ADDRESS
			struct msg_sender *msgA2 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
                        char host[IN6_MAX_ASCII_LEN + 1];
                        const int ret =
                            getnameinfo((struct sockaddr *) &Adestination->addr,
                                        sizeof Adestination->addr, host,
                                        sizeof host, nullptr, 0, NI_NUMERICHOST);
                        assert(ret == 0);
                        strncpy(msgA2->receiver, host,
					sizeof(msgA2->receiver) - 1);
			msgA2->type = SENDER_MSG_CHANGE_RECEIVER;

			resp = send_message(rtsp_params.parent, pathA, (struct message *) msgA2);
			free_response(resp);
			resp = NULL;
		}
	}
}

void BasicRTSPOnlySubsession::deleteStream(unsigned /* clientSessionId */,
		void*& /* streamToken */) {
	if (Vdestination != NULL) {
		if (avType == video || avType == av) {
			char pathV[1024];
			delete Vdestination;
			Vdestination = NULL;
			memset(pathV, 0, sizeof(pathV));
			enum module_class path_sender[] = { MODULE_CLASS_SENDER,
					MODULE_CLASS_NONE };
			append_message_path(pathV, sizeof(pathV), path_sender);

			//CHANGE DST PORT
			struct msg_sender *msgV1 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
			msgV1->tx_port = rtsp_params.rtp_port;
			msgV1->type = SENDER_MSG_CHANGE_PORT;
			struct response *resp;
			resp = send_message(rtsp_params.parent, pathV, (struct message *) msgV1);
			free_response(resp);

			//CHANGE DST ADDRESS
			struct msg_sender *msgV2 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
			strncpy(msgV2->receiver, "127.0.0.1", sizeof(msgV2->receiver) - 1);
			msgV2->type = SENDER_MSG_CHANGE_RECEIVER;
			resp = send_message(rtsp_params.parent, pathV, (struct message *) msgV2);
			free_response(resp);
		}
	}

	if (Adestination != NULL) {
		if (avType == audio || avType == av) {
			char pathA[1024];
			delete Adestination;
			Adestination = NULL;
			memset(pathA, 0, sizeof(pathA));
			enum module_class path_sender[] = { MODULE_CLASS_AUDIO,
					MODULE_CLASS_SENDER, MODULE_CLASS_NONE };
			append_message_path(pathA, sizeof(pathA), path_sender);

			//CHANGE DST PORT
			struct msg_sender *msgA1 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));

			//TODO: GET AUDIO PORT SET (NOT A COMMON CASE WHEN RTSP IS ENABLED: DEFAULT -> vport + 2)
			msgA1->tx_port = rtsp_params.rtp_port_audio;
			msgA1->type = SENDER_MSG_CHANGE_PORT;
			struct response *resp;
                        resp = send_message(rtsp_params.parent, pathA, (struct message *) msgA1);
			free_response(resp);

			//CHANGE DST ADDRESS
			struct msg_sender *msgA2 = (struct msg_sender *) new_message(
					sizeof(struct msg_sender));
			strncpy(msgA2->receiver, "127.0.0.1", sizeof(msgA2->receiver) - 1);
			msgA2->type = SENDER_MSG_CHANGE_RECEIVER;
			resp = send_message(rtsp_params.parent, pathA, (struct message *) msgA2);
			free_response(resp);
		}
	}
}
/* vi: set noexpandtab: */