mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-25 20:27:34 +08:00
139 lines
3.9 KiB
C++
139 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT license that can be found in the
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef ZLMEDIAKIT_RTPEXT_H
|
|
#define ZLMEDIAKIT_RTPEXT_H
|
|
|
|
#include <stdint.h>
|
|
#include <map>
|
|
#include <string>
|
|
#include "Common/macros.h"
|
|
#include "Rtsp/Rtsp.h"
|
|
|
|
using namespace std;
|
|
using namespace mediakit;
|
|
|
|
enum class RtpExtType : uint8_t {
|
|
padding = 0,
|
|
ssrc_audio_level = 1,
|
|
abs_send_time = 2,
|
|
transport_cc = 3,
|
|
sdes_mid = 4,
|
|
sdes_rtp_stream_id = 5,
|
|
sdes_repaired_rtp_stream_id = 6,
|
|
video_timing = 7,
|
|
color_space = 8,
|
|
//for firefox
|
|
csrc_audio_level = 9,
|
|
//svc ?
|
|
framemarking = 10,
|
|
video_content_type = 11,
|
|
playout_delay = 12,
|
|
video_orientation = 13,
|
|
toffset = 14,
|
|
reserved = 15,
|
|
// e2e ?
|
|
encrypt = reserved
|
|
};
|
|
|
|
class RtcMedia;
|
|
|
|
//使用次对象的方法前需保证RtpHeader内存未释放
|
|
class RtpExt {
|
|
public:
|
|
template<typename Type>
|
|
friend void appendExt(map<uint8_t, RtpExt> &ret, uint8_t *ptr, const uint8_t *end);
|
|
|
|
~RtpExt() = default;
|
|
|
|
static map<uint8_t/*id*/, RtpExt/*data*/> getExtValue(const RtpHeader *header);
|
|
static RtpExtType getExtType(const string &url);
|
|
static const string& getExtUrl(RtpExtType type);
|
|
static const char *getExtName(RtpExtType type);
|
|
|
|
void setType(RtpExtType type);
|
|
RtpExtType getType() const;
|
|
string dumpString() const;
|
|
|
|
uint8_t getAudioLevel(bool *vad) const;
|
|
uint32_t getAbsSendTime() const;
|
|
uint16_t getTransportCCSeq() const;
|
|
string getSdesMid() const;
|
|
string getRtpStreamId() const;
|
|
string getRepairedRtpStreamId() const;
|
|
|
|
void getVideoTiming(uint8_t &flags,
|
|
uint16_t &encode_start,
|
|
uint16_t &encode_finish,
|
|
uint16_t &packetization_complete,
|
|
uint16_t &last_pkt_left_pacer,
|
|
uint16_t &reserved_net0,
|
|
uint16_t &reserved_net1) const;
|
|
|
|
uint8_t getVideoContentType() const;
|
|
|
|
void getVideoOrientation(bool &camera_bit,
|
|
bool &flip_bit,
|
|
bool &first_rotation,
|
|
bool &second_rotation) const;
|
|
|
|
void getPlayoutDelay(uint16_t &min_delay, uint16_t &max_delay) const;
|
|
|
|
uint32_t getTransmissionOffset() const;
|
|
|
|
uint8_t getFramemarkingTID() const;
|
|
|
|
void setExtId(uint8_t ext_id);
|
|
void clearExt();
|
|
|
|
private:
|
|
RtpExt(void *ptr, bool one_byte_ext, const char *str, size_t size);
|
|
const char *data() const;
|
|
size_t size() const;
|
|
const char& operator[](size_t pos) const;
|
|
operator std::string() const;
|
|
|
|
private:
|
|
void *_ext = nullptr;
|
|
const char *_data;
|
|
size_t _size;
|
|
bool _one_byte_ext = true;
|
|
RtpExtType _type = RtpExtType::padding;
|
|
};
|
|
|
|
class RtcMedia;
|
|
class RtpExtContext {
|
|
public:
|
|
using Ptr = std::shared_ptr<RtpExtContext>;
|
|
using OnGetRtp = function<void(uint8_t pt, uint32_t ssrc, const string &rid)>;
|
|
|
|
RtpExtContext(const RtcMedia &media);
|
|
~RtpExtContext() = default;
|
|
|
|
void setOnGetRtp(OnGetRtp cb);
|
|
string getRid(uint32_t ssrc) const;
|
|
void setRid(uint32_t ssrc, const string &rid);
|
|
void changeRtpExtId(const RtpHeader *header, bool is_recv, string *rid_ptr = nullptr);
|
|
|
|
private:
|
|
void onGetRtp(uint8_t pt, uint32_t ssrc, const string &rid);
|
|
|
|
private:
|
|
OnGetRtp _cb;
|
|
//发送rtp时需要修改rtp ext id
|
|
map<RtpExtType, uint8_t> _rtp_ext_type_to_id;
|
|
//接收rtp时需要修改rtp ext id
|
|
unordered_map<uint8_t, RtpExtType> _rtp_ext_id_to_type;
|
|
//ssrc --> rid
|
|
unordered_map<uint32_t/*simulcast ssrc*/, string/*rid*/> _ssrc_to_rid;
|
|
};
|
|
|
|
#endif //ZLMEDIAKIT_RTPEXT_H
|