mirror of
https://github.com/ossrs/srs.git
synced 2025-02-15 04:42:04 +00:00
118 lines
3.7 KiB
C++
118 lines
3.7 KiB
C++
/**
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2013-2020 Winlin
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
* this software and associated documentation files (the "Software"), to deal in
|
|
* the Software without restriction, including without limitation the rights to
|
|
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
* the Software, and to permit persons to whom the Software is furnished to do so,
|
|
* subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#ifndef SRS_APP_MPEGTS_UDP_HPP
|
|
#define SRS_APP_MPEGTS_UDP_HPP
|
|
|
|
#include <srs_core.hpp>
|
|
|
|
struct sockaddr;
|
|
#include <string>
|
|
#include <map>
|
|
|
|
class SrsBuffer;
|
|
class SrsTsContext;
|
|
class SrsConfDirective;
|
|
class SrsSimpleStream;
|
|
class SrsRtmpClient;
|
|
class SrsStSocket;
|
|
class SrsRequest;
|
|
class SrsRawH264Stream;
|
|
class SrsSharedPtrMessage;
|
|
class SrsRawAacStream;
|
|
struct SrsRawAacStreamCodec;
|
|
class SrsPithyPrint;
|
|
class SrsSimpleRtmpClient;
|
|
|
|
#include <srs_app_st.hpp>
|
|
#include <srs_kernel_ts.hpp>
|
|
#include <srs_app_listener.hpp>
|
|
|
|
// The queue for mpegts over udp to send packets.
|
|
// For the aac in mpegts contains many flv packets in a pes packet,
|
|
// we must recalc the timestamp.
|
|
class SrsMpegtsQueue
|
|
{
|
|
private:
|
|
// The key: dts, value: msg.
|
|
std::map<int64_t, SrsSharedPtrMessage*> msgs;
|
|
int nb_audios;
|
|
int nb_videos;
|
|
public:
|
|
SrsMpegtsQueue();
|
|
virtual ~SrsMpegtsQueue();
|
|
public:
|
|
virtual srs_error_t push(SrsSharedPtrMessage* msg);
|
|
virtual SrsSharedPtrMessage* dequeue();
|
|
};
|
|
|
|
// The mpegts over udp stream caster.
|
|
class SrsMpegtsOverUdp : virtual public ISrsTsHandler, virtual public ISrsUdpHandler
|
|
{
|
|
private:
|
|
SrsTsContext* context;
|
|
SrsSimpleStream* buffer;
|
|
std::string output;
|
|
private:
|
|
SrsSimpleRtmpClient* sdk;
|
|
private:
|
|
SrsRawH264Stream* avc;
|
|
std::string h264_sps;
|
|
bool h264_sps_changed;
|
|
std::string h264_pps;
|
|
bool h264_pps_changed;
|
|
bool h264_sps_pps_sent;
|
|
private:
|
|
SrsRawAacStream* aac;
|
|
std::string aac_specific_config;
|
|
private:
|
|
SrsMpegtsQueue* queue;
|
|
SrsPithyPrint* pprint;
|
|
public:
|
|
SrsMpegtsOverUdp(SrsConfDirective* c);
|
|
virtual ~SrsMpegtsOverUdp();
|
|
// Interface ISrsUdpHandler
|
|
public:
|
|
virtual srs_error_t on_udp_packet(const sockaddr* from, const int fromlen, char* buf, int nb_buf);
|
|
private:
|
|
virtual srs_error_t on_udp_bytes(std::string host, int port, char* buf, int nb_buf);
|
|
// Interface ISrsTsHandler
|
|
public:
|
|
virtual srs_error_t on_ts_message(SrsTsMessage* msg);
|
|
private:
|
|
virtual srs_error_t on_ts_video(SrsTsMessage* msg, SrsBuffer* avs);
|
|
virtual srs_error_t write_h264_sps_pps(uint32_t dts, uint32_t pts);
|
|
virtual srs_error_t write_h264_ipb_frame(char* frame, int frame_size, uint32_t dts, uint32_t pts);
|
|
virtual srs_error_t on_ts_audio(SrsTsMessage* msg, SrsBuffer* avs);
|
|
virtual srs_error_t write_audio_raw_frame(char* frame, int frame_size, SrsRawAacStreamCodec* codec, uint32_t dts);
|
|
private:
|
|
virtual srs_error_t rtmp_write_packet(char type, uint32_t timestamp, char* data, int size);
|
|
private:
|
|
// Connect to RTMP server.
|
|
virtual srs_error_t connect();
|
|
// Close the connection to RTMP server.
|
|
virtual void close();
|
|
};
|
|
|
|
#endif
|
|
|