| 1 | /********** |
| 2 | This library is free software; you can redistribute it and/or modify it under |
| 3 | the terms of the GNU Lesser General Public License as published by the |
| 4 | Free Software Foundation; either version 3 of the License, or (at your |
| 5 | option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.) |
| 6 | |
| 7 | This library is distributed in the hope that it will be useful, but WITHOUT |
| 8 | ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
| 9 | FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for |
| 10 | more details. |
| 11 | |
| 12 | You should have received a copy of the GNU Lesser General Public License |
| 13 | along with this library; if not, write to the Free Software Foundation, Inc., |
| 14 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 15 | **********/ |
| 16 | // "liveMedia" |
| 17 | // Copyright (c) 1996-2020 Live Networks, Inc. All rights reserved. |
| 18 | // A 'ServerMediaSubsession' object that creates new, unicast, "RTPSink"s |
| 19 | // on demand. |
| 20 | // C++ header |
| 21 | |
| 22 | #ifndef _ON_DEMAND_SERVER_MEDIA_SUBSESSION_HH |
| 23 | #define _ON_DEMAND_SERVER_MEDIA_SUBSESSION_HH |
| 24 | |
| 25 | #ifndef _SERVER_MEDIA_SESSION_HH |
| 26 | #include "ServerMediaSession.hh" |
| 27 | #endif |
| 28 | #ifndef _RTP_SINK_HH |
| 29 | #include "RTPSink.hh" |
| 30 | #endif |
| 31 | #ifndef _BASIC_UDP_SINK_HH |
| 32 | #include "BasicUDPSink.hh" |
| 33 | #endif |
| 34 | #ifndef _RTCP_HH |
| 35 | #include "RTCP.hh" |
| 36 | #endif |
| 37 | |
| 38 | class OnDemandServerMediaSubsession: public ServerMediaSubsession { |
| 39 | protected: // we're a virtual base class |
| 40 | OnDemandServerMediaSubsession(UsageEnvironment& env, Boolean reuseFirstSource, |
| 41 | portNumBits initialPortNum = 6970, |
| 42 | Boolean multiplexRTCPWithRTP = False); |
| 43 | virtual ~OnDemandServerMediaSubsession(); |
| 44 | |
| 45 | protected: // redefined virtual functions |
| 46 | virtual char const* sdpLines(); |
| 47 | virtual void getStreamParameters(unsigned clientSessionId, |
| 48 | netAddressBits clientAddress, |
| 49 | Port const& clientRTPPort, |
| 50 | Port const& clientRTCPPort, |
| 51 | int tcpSocketNum, |
| 52 | unsigned char rtpChannelId, |
| 53 | unsigned char rtcpChannelId, |
| 54 | netAddressBits& destinationAddress, |
| 55 | u_int8_t& destinationTTL, |
| 56 | Boolean& isMulticast, |
| 57 | Port& serverRTPPort, |
| 58 | Port& serverRTCPPort, |
| 59 | void*& streamToken); |
| 60 | virtual void startStream(unsigned clientSessionId, void* streamToken, |
| 61 | TaskFunc* rtcpRRHandler, |
| 62 | void* rtcpRRHandlerClientData, |
| 63 | unsigned short& rtpSeqNum, |
| 64 | unsigned& rtpTimestamp, |
| 65 | ServerRequestAlternativeByteHandler* serverRequestAlternativeByteHandler, |
| 66 | void* serverRequestAlternativeByteHandlerClientData); |
| 67 | virtual void pauseStream(unsigned clientSessionId, void* streamToken); |
| 68 | virtual void seekStream(unsigned clientSessionId, void* streamToken, double& seekNPT, double streamDuration, u_int64_t& numBytes); |
| 69 | virtual void seekStream(unsigned clientSessionId, void* streamToken, char*& absStart, char*& absEnd); |
| 70 | virtual void nullSeekStream(unsigned clientSessionId, void* streamToken, |
| 71 | double streamEndTime, u_int64_t& numBytes); |
| 72 | virtual void setStreamScale(unsigned clientSessionId, void* streamToken, float scale); |
| 73 | virtual float getCurrentNPT(void* streamToken); |
| 74 | virtual FramedSource* getStreamSource(void* streamToken); |
| 75 | virtual void getRTPSinkandRTCP(void* streamToken, |
| 76 | RTPSink const*& rtpSink, RTCPInstance const*& rtcp); |
| 77 | virtual void deleteStream(unsigned clientSessionId, void*& streamToken); |
| 78 | |
| 79 | protected: // new virtual functions, possibly redefined by subclasses |
| 80 | virtual char const* getAuxSDPLine(RTPSink* rtpSink, |
| 81 | FramedSource* inputSource); |
| 82 | virtual void seekStreamSource(FramedSource* inputSource, double& seekNPT, double streamDuration, u_int64_t& numBytes); |
| 83 | // This routine is used to seek by relative (i.e., NPT) time. |
| 84 | // "streamDuration", if >0.0, specifies how much data to stream, past "seekNPT". (If <=0.0, all remaining data is streamed.) |
| 85 | // "numBytes" returns the size (in bytes) of the data to be streamed, or 0 if unknown or unlimited. |
| 86 | virtual void seekStreamSource(FramedSource* inputSource, char*& absStart, char*& absEnd); |
| 87 | // This routine is used to seek by 'absolute' time. |
| 88 | // "absStart" should be a string of the form "YYYYMMDDTHHMMSSZ" or "YYYYMMDDTHHMMSS.<frac>Z". |
| 89 | // "absEnd" should be either NULL (for no end time), or a string of the same form as "absStart". |
| 90 | // These strings may be modified in-place, or can be reassigned to a newly-allocated value (after delete[]ing the original). |
| 91 | virtual void setStreamSourceScale(FramedSource* inputSource, float scale); |
| 92 | virtual void setStreamSourceDuration(FramedSource* inputSource, double streamDuration, u_int64_t& numBytes); |
| 93 | virtual void closeStreamSource(FramedSource* inputSource); |
| 94 | |
| 95 | protected: // new virtual functions, defined by all subclasses |
| 96 | virtual FramedSource* createNewStreamSource(unsigned clientSessionId, |
| 97 | unsigned& estBitrate) = 0; |
| 98 | // "estBitrate" is the stream's estimated bitrate, in kbps |
| 99 | virtual RTPSink* createNewRTPSink(Groupsock* rtpGroupsock, |
| 100 | unsigned char rtpPayloadTypeIfDynamic, |
| 101 | FramedSource* inputSource) = 0; |
| 102 | |
| 103 | protected: // new virtual functions, may be redefined by a subclass: |
| 104 | virtual Groupsock* createGroupsock(struct in_addr const& addr, Port port); |
| 105 | virtual RTCPInstance* createRTCP(Groupsock* RTCPgs, unsigned totSessionBW, /* in kbps */ |
| 106 | unsigned char const* cname, RTPSink* sink); |
| 107 | |
| 108 | public: |
| 109 | void multiplexRTCPWithRTP() { fMultiplexRTCPWithRTP = True; } |
| 110 | // An alternative to passing the "multiplexRTCPWithRTP" parameter as True in the constructor |
| 111 | |
| 112 | void setRTCPAppPacketHandler(RTCPAppHandlerFunc* handler, void* clientData); |
| 113 | // Sets a handler to be called if a RTCP "APP" packet arrives from any future client. |
| 114 | // (Any current clients are not affected; any "APP" packets from them will continue to be |
| 115 | // handled by whatever handler existed when the client sent its first RTSP "PLAY" command.) |
| 116 | // (Call with (NULL, NULL) to remove an existing handler - for future clients only) |
| 117 | |
| 118 | void sendRTCPAppPacket(u_int8_t subtype, char const* name, |
| 119 | u_int8_t* appDependentData, unsigned appDependentDataSize); |
| 120 | // Sends a custom RTCP "APP" packet to the most recent client (if "reuseFirstSource" was False), |
| 121 | // or to all current clients (if "reuseFirstSource" was True). |
| 122 | // The parameters correspond to their |
| 123 | // respective fields as described in the RTP/RTCP definition (RFC 3550). |
| 124 | // Note that only the low-order 5 bits of "subtype" are used, and only the first 4 bytes |
| 125 | // of "name" are used. (If "name" has fewer than 4 bytes, or is NULL, |
| 126 | // then the remaining bytes are '\0'.) |
| 127 | |
| 128 | protected: |
| 129 | void setSDPLinesFromRTPSink(RTPSink* rtpSink, FramedSource* inputSource, |
| 130 | unsigned estBitrate); |
| 131 | // used to implement "sdpLines()" |
| 132 | |
| 133 | protected: |
| 134 | char* fSDPLines; |
| 135 | HashTable* fDestinationsHashTable; // indexed by client session id |
| 136 | |
| 137 | private: |
| 138 | Boolean fReuseFirstSource; |
| 139 | portNumBits fInitialPortNum; |
| 140 | Boolean fMultiplexRTCPWithRTP; |
| 141 | void* fLastStreamToken; |
| 142 | char fCNAME[100]; // for RTCP |
| 143 | RTCPAppHandlerFunc* fAppHandlerTask; |
| 144 | void* fAppHandlerClientData; |
| 145 | friend class StreamState; |
| 146 | }; |
| 147 | |
| 148 | |
| 149 | // A class that represents the state of an ongoing stream. This is used only internally, in the implementation of |
| 150 | // "OnDemandServerMediaSubsession", but we expose the definition here, in case subclasses of "OnDemandServerMediaSubsession" |
| 151 | // want to access it. |
| 152 | |
| 153 | class Destinations { |
| 154 | public: |
| 155 | Destinations(struct in_addr const& destAddr, |
| 156 | Port const& rtpDestPort, |
| 157 | Port const& rtcpDestPort) |
| 158 | : isTCP(False), addr(destAddr), rtpPort(rtpDestPort), rtcpPort(rtcpDestPort) { |
| 159 | } |
| 160 | Destinations(int tcpSockNum, unsigned char rtpChanId, unsigned char rtcpChanId) |
| 161 | : isTCP(True), rtpPort(0) /*dummy*/, rtcpPort(0) /*dummy*/, |
| 162 | tcpSocketNum(tcpSockNum), rtpChannelId(rtpChanId), rtcpChannelId(rtcpChanId) { |
| 163 | } |
| 164 | |
| 165 | public: |
| 166 | Boolean isTCP; |
| 167 | struct in_addr addr; |
| 168 | Port rtpPort; |
| 169 | Port rtcpPort; |
| 170 | int tcpSocketNum; |
| 171 | unsigned char rtpChannelId, rtcpChannelId; |
| 172 | }; |
| 173 | |
| 174 | class StreamState { |
| 175 | public: |
| 176 | StreamState(OnDemandServerMediaSubsession& master, |
| 177 | Port const& serverRTPPort, Port const& serverRTCPPort, |
| 178 | RTPSink* rtpSink, BasicUDPSink* udpSink, |
| 179 | unsigned totalBW, FramedSource* mediaSource, |
| 180 | Groupsock* rtpGS, Groupsock* rtcpGS); |
| 181 | virtual ~StreamState(); |
| 182 | |
| 183 | void startPlaying(Destinations* destinations, unsigned clientSessionId, |
| 184 | TaskFunc* rtcpRRHandler, void* rtcpRRHandlerClientData, |
| 185 | ServerRequestAlternativeByteHandler* serverRequestAlternativeByteHandler, |
| 186 | void* serverRequestAlternativeByteHandlerClientData); |
| 187 | void pause(); |
| 188 | void sendRTCPAppPacket(u_int8_t subtype, char const* name, |
| 189 | u_int8_t* appDependentData, unsigned appDependentDataSize); |
| 190 | void endPlaying(Destinations* destinations, unsigned clientSessionId); |
| 191 | void reclaim(); |
| 192 | |
| 193 | unsigned& referenceCount() { return fReferenceCount; } |
| 194 | |
| 195 | Port const& serverRTPPort() const { return fServerRTPPort; } |
| 196 | Port const& serverRTCPPort() const { return fServerRTCPPort; } |
| 197 | |
| 198 | RTPSink* rtpSink() const { return fRTPSink; } |
| 199 | RTCPInstance* rtcpInstance() const { return fRTCPInstance; } |
| 200 | |
| 201 | float streamDuration() const { return fStreamDuration; } |
| 202 | |
| 203 | FramedSource* mediaSource() const { return fMediaSource; } |
| 204 | float& startNPT() { return fStartNPT; } |
| 205 | |
| 206 | private: |
| 207 | OnDemandServerMediaSubsession& fMaster; |
| 208 | Boolean fAreCurrentlyPlaying; |
| 209 | unsigned fReferenceCount; |
| 210 | |
| 211 | Port fServerRTPPort, fServerRTCPPort; |
| 212 | |
| 213 | RTPSink* fRTPSink; |
| 214 | BasicUDPSink* fUDPSink; |
| 215 | |
| 216 | float fStreamDuration; |
| 217 | unsigned fTotalBW; |
| 218 | RTCPInstance* fRTCPInstance; |
| 219 | |
| 220 | FramedSource* fMediaSource; |
| 221 | float fStartNPT; // initial 'normal play time'; reset after each seek |
| 222 | |
| 223 | Groupsock* fRTPgs; |
| 224 | Groupsock* fRTCPgs; |
| 225 | }; |
| 226 | |
| 227 | #endif |
| 228 | |