Skip to content

Commit

Permalink
Organized source codes related to RTMP Provider
Browse files Browse the repository at this point in the history
  • Loading branch information
@dimiden
dimiden committed Nov 14, 2024
1 parent d2b5c26 commit 342e901
Show file tree
Hide file tree
Showing 7 changed files with 170 additions and 244 deletions.
5 changes: 4 additions & 1 deletion src/projects/base/ovlibrary/type.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,13 @@
namespace ov
{
template <typename T>
std::underlying_type_t<T> ToUnderlyingType(T enum_value)
inline std::underlying_type_t<T> ToUnderlyingType(T enum_value)
{
return static_cast<std::underlying_type_t<T>>(enum_value);
}

template <typename T>
using UnderylingType = std::underlying_type_t<T>;
} // namespace ov

#define OV_DEFINE_SETTER(type, setter, member, extra_qualifier) \
Expand Down
2 changes: 1 addition & 1 deletion src/projects/providers/rtmp/chunk/rtmp_datastructure.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ struct RtmpChunkHeader
int64_t timestamp = 0LL;
uint32_t timestamp_delta = 0U;

RtmpMessageTypeID type_id = RtmpMessageTypeID::UNKNOWN;
RtmpMessageTypeID type_id = RtmpMessageTypeID::Unknown;
uint32_t stream_id = 0U;
} completed;

Expand Down
300 changes: 104 additions & 196 deletions src/projects/providers/rtmp/chunk/rtmp_define.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,223 +2,154 @@
//
// RtmpProvider
//
// Created by Jaejong Bong
// Created by Hyunjun Jang
// Copyright (c) 2018 AirenSoft. All rights reserved.
//
//==============================================================================
#pragma once

#include <arpa/inet.h>
#include <base/ovlibrary/ovlibrary.h>
#include <string.h>

#include <map>
#include <memory>
#include <vector>

#ifdef WIN32

# include <winsock2.h>

#else
# include <arpa/inet.h>
#endif

#pragma pack(1)
#define RTMP_AVC_NAL_HEADER_SIZE (4) // 00 00 00 01 or 00 00 01
#define RTMP_ADTS_HEADER_SIZE (7)
#define RTMP_MAX_PACKET_SIZE (20 * 1024 * 1024) // 20M

//Avc Nal Header
const char g_rtmp_avc_nal_header[RTMP_AVC_NAL_HEADER_SIZE] = {0, 0, 0, 1};

//Rtmp - 44100(4) 22050(7) 11025(10) 사용
const int g_rtmp_sample_rate_table[] = {96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050, 16000, 12000, 11025,
8000, 7350, 0, 0, 0};
#define RTMP_SAMPLERATE_TABLE_SIZE (16)
#define RTMP_TIME_SCALE 1000

#define RTMP_SESSION_KEY_MAX_SIZE (1024)
#define RTMP_TIME_SCALE (1000)
#define RTMP_HANDSHAKE_VERSION 0x03
#define RTMP_HANDSHAKE_PACKET_SIZE 1536

// HANDSHAKE
#define RTMP_HANDSHAKE_VERSION (0x03)
#define RTMP_HANDSHAKE_PACKET_SIZE (1536)

// CHUNK
#define RTMP_CHUNK_BASIC_HEADER_SIZE_MAX (3)
#define RTMP_CHUNK_BASIC_HEADER_SIZE_MAX 3
#define RTMP_CHUNK_MSG_HEADER_SIZE_MAX (11 + 4) // header + extended timestamp
#define RTMP_PACKET_HEADER_SIZE_MAX (RTMP_CHUNK_BASIC_HEADER_SIZE_MAX + RTMP_CHUNK_MSG_HEADER_SIZE_MAX)

#define RTMP_CHUNK_TYPE_MASK (0xc0)
#define RTMP_CHUNK_STREAM_ID_MASK (0x3f) // 00111111b
#define RTMP_EXTEND_TIMESTAMP 0x00ffffff
#define RTMP_EXTEND_TIMESTAMP_SIZE 4

#define RTMP_EXTEND_TIMESTAMP (0x00ffffff)
#define RTMP_EXTEND_TIMESTAMP_SIZE (4) //sizeof(uint)
enum class RtmpChunkStreamId : uint32_t
{
Urgent = 2,
Control = 3,
Media = 8,

// CHUNK STREAM ID
#define RTMP_CHUNK_STREAM_ID_URGENT (2)
#define RTMP_CHUNK_STREAM_ID_CONTROL (3)
#define RTMP_CHUNK_STREAM_ID_MEDIA (8)
Mask = 0b00111111,
};

// MESSAGE ID
enum class RtmpMessageTypeID : uint8_t
// Overload operator to allow usage like `10 & RtmpChunkStreamId::Mask`
inline ov::UnderylingType<RtmpChunkStreamId> operator&(ov::UnderylingType<RtmpChunkStreamId> number, RtmpChunkStreamId id)
{
UNKNOWN = 0,
return number & ov::ToUnderlyingType(id);
}

SET_CHUNK_SIZE = 1,
ABORT = 2,
ACKNOWLEDGEMENT = 3,
USER_CONTROL = 4,
WINDOW_ACKNOWLEDGEMENT_SIZE = 5,
SET_PEER_BANDWIDTH = 6,
AUDIO = 8,
VIDEO = 9,
AMF3_DATA = 15,
AMF3_SHARED_OBJECT = 16,
AMF3_COMMAND = 17,
AMF0_DATA = 18,
AMF0_SHARED_OBJECT = 19,
AMF0_COMMAND = 20,
AGGREGATE = 22,
enum class RtmpMessageTypeID : uint8_t
{
Unknown = 0,

SetChunkSize = 1,
Abort = 2,
Acknowledgement = 3,
UserControl = 4,
WindowAcknowledgementSize = 5,
SetPeerBandwidth = 6,
Audio = 8,
Video = 9,
Amf3Data = 15,
Amf3SharedObject = 16,
Amf3Command = 17,
Amf0Data = 18,
Amf0SharedObject = 19,
Amf0Command = 20,
Aggregate = 22,
};

#define _DECLARE_ENUM_TO_STRING(type_id) \
case RtmpMessageTypeID::type_id: \
return #type_id

inline constexpr const char *StringFromRtmpMessageTypeID(RtmpMessageTypeID type)
{
switch (type)
{
case RtmpMessageTypeID::UNKNOWN:
return "Unknown";
case RtmpMessageTypeID::SET_CHUNK_SIZE:
return "SetChunkSize";
case RtmpMessageTypeID::ABORT:
return "Abort";
case RtmpMessageTypeID::ACKNOWLEDGEMENT:
return "Acknowledgement";
case RtmpMessageTypeID::USER_CONTROL:
return "UserControl";
case RtmpMessageTypeID::WINDOW_ACKNOWLEDGEMENT_SIZE:
return "WindowAcknowledgementSize";
case RtmpMessageTypeID::SET_PEER_BANDWIDTH:
return "SetPeerBandwidth";
case RtmpMessageTypeID::AUDIO:
return "Audio";
case RtmpMessageTypeID::VIDEO:
return "Video";
case RtmpMessageTypeID::AMF3_DATA:
return "AMF3Data";
case RtmpMessageTypeID::AMF3_SHARED_OBJECT:
return "AMF3SharedObject";
case RtmpMessageTypeID::AMF3_COMMAND:
return "AMF3Command";
case RtmpMessageTypeID::AMF0_DATA:
return "AMF0Data";
case RtmpMessageTypeID::AMF0_SHARED_OBJECT:
return "AMF0SharedObject";
case RtmpMessageTypeID::AMF0_COMMAND:
return "AMF0Command";
case RtmpMessageTypeID::AGGREGATE:
return "Aggregate";
_DECLARE_ENUM_TO_STRING(Unknown);
_DECLARE_ENUM_TO_STRING(SetChunkSize);
_DECLARE_ENUM_TO_STRING(Abort);
_DECLARE_ENUM_TO_STRING(Acknowledgement);
_DECLARE_ENUM_TO_STRING(UserControl);
_DECLARE_ENUM_TO_STRING(WindowAcknowledgementSize);
_DECLARE_ENUM_TO_STRING(SetPeerBandwidth);
_DECLARE_ENUM_TO_STRING(Audio);
_DECLARE_ENUM_TO_STRING(Video);
_DECLARE_ENUM_TO_STRING(Amf3Data);
_DECLARE_ENUM_TO_STRING(Amf3SharedObject);
_DECLARE_ENUM_TO_STRING(Amf3Command);
_DECLARE_ENUM_TO_STRING(Amf0Data);
_DECLARE_ENUM_TO_STRING(Amf0SharedObject);
_DECLARE_ENUM_TO_STRING(Amf0Command);
_DECLARE_ENUM_TO_STRING(Aggregate);
}

return "Invalid";
}

// USER CONTROL MESSAGE ID
#define RTMP_UCMID_STREAMBEGIN (0)
#define RTMP_UCMID_STREAMEOF (1)
#define RTMP_UCMID_STREAMDRY (2)
#define RTMP_UCMID_SETBUFFERLENGTH (3)
#define RTMP_UCMID_STREAMISRECORDED (4)
#define RTMP_UCMID_PINGREQUEST (6)
#define RTMP_UCMID_PINGRESPONSE (7)
#define RTMP_UCMID_BUFFEREMPTY (31)
#define RTMP_UCMID_BUFFERREADY (32)

// COMMAND TRANSACTION ID
#define RTMP_CMD_TRID_NOREPONSE (0.0)
#define RTMP_CMD_TRID_CONNECT (1.0)
#define RTMP_CMD_TRID_CREATESTREAM (2.0)
#define RTMP_CMD_TRID_RELEASESTREAM (2.0)
#define RTMP_CMD_TRID_DELETESTREAM RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_CLOSESTREAM RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_PLAY RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_PLAY2 RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_SEEK RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_PAUSE RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_RECEIVEAUDIO RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_RECEIVEVIDEO RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_PUBLISH RTMP_CMD_TRID_NOREPONSE
#define RTMP_CMD_TRID_FCPUBLISH (3.0)
#define RTMP_CMD_TRID_FCSUBSCRIBE (4.0)
#define RTMP_CMD_TRID_FCUNSUBSCRIBE (5.0)
#define RTMP_CMD_TRID_FCUNPUBLISH (5.0)
enum class UserControlMessageId : uint16_t
{
StreamBegin = 0,
StreamEof = 1,
StreamDry = 2,
SetBufferLength = 3,
StreamIsRecorded = 4,
PingRequest = 6,
PingResponse = 7,
BufferEmpty = 31,
BufferReady = 32,
};

// COMMAND NAME
#define RTMP_CMD_NAME_CONNECT ("connect")
#define RTMP_CMD_NAME_CREATESTREAM ("createStream")
#define RTMP_CMD_NAME_RELEASESTREAM ("releaseStream")
#define RTMP_CMD_NAME_DELETESTREAM ("deleteStream")
#define RTMP_CMD_NAME_CLOSESTREAM ("closeStream")
#define RTMP_CMD_NAME_PLAY ("play")
#define RTMP_CMD_NAME_ONSTATUS ("onStatus")
#define RTMP_CMD_NAME_PUBLISH ("publish")
#define RTMP_CMD_NAME_FCPUBLISH ("FCPublish")
#define RTMP_CMD_NAME_FCUNPUBLISH ("FCUnpublish")
#define RTMP_CMD_NAME_ONFCPUBLISH ("onFCPublish")
#define RTMP_CMD_NAME_ONFCUNPUBLISH ("onFCUnpublish")
#define RTMP_CMD_NAME_CLOSE ("close")
#define RTMP_CMD_NAME_SETCHALLENGE ("setChallenge")
#define RTMP_CMD_DATA_SETDATAFRAME ("@setDataFrame")
#define RTMP_CMD_NAME_ONFI ("onFI")
#define RTMP_CMD_NAME_ONCLIENTLOGIN ("onClientLogin")
#define RTMP_CMD_DATA_ONMETADATA ("onMetaData")
#define RTMP_ACK_NAME_RESULT ("_result")
#define RTMP_ACK_NAME_ERROR ("_error")
#define RTMP_ACK_NAME_BWDONE ("onBWDone")
#define RTMP_PING ("ping")

//RtmpStream/RtmpPublish
#define RTMP_H264_I_FRAME_TYPE (0x17)
#define RTMP_H264_P_FRAME_TYPE (0x27)
#define RTMP_DEFAULT_ACKNOWNLEDGEMENT_SIZE (2500000)
#define RTMP_DEFAULT_PEER_BANDWIDTH (2500000)
#define RTMP_DEFAULT_CHUNK_SIZE (128)

#define RTMP_SEQUENCE_INFO_TYPE (0x00)
#define RTMP_FRAME_DATA_TYPE (0x01)
#define RTMP_VIDEO_DATA_MIN_SIZE (5) // control(1) + sequence(1) + offsettime(3)

#define RTMP_AUDIO_CONTROL_HEADER_INDEX (0)
#define RTMP_AAC_AUDIO_SEQUENCE_HEADER_INDEX (1)
#define RTMP_AAC_AUDIO_DATA_MIN_SIZE (2) // control(1) + sequence(1)
#define RTMP_AAC_AUDIO_FRAME_INDEX (2)
#define RTMP_SPEEX_AUDIO_FRAME_INDEX (1)
#define RTMP_MP3_AUDIO_FRAME_INDEX (1)
#define RTMP_SPS_PPS_MIN_DATA_SIZE (14)

#define RTMP_VIDEO_CONTROL_HEADER_INDEX (0)
#define RTMP_VIDEO_SEQUENCE_HEADER_INDEX (1)
#define RTMP_VIDEO_COMPOSITION_OFFSET_INDEX (2)
#define RTMP_VIDEO_FRAME_SIZE_INDEX (5)
#define RTMP_VIDEO_FRAME_INDEX (9)
#define RTMP_VIDEO_FRAME_SIZE_INFO_SIZE (4)

#define RTMP_UNKNOWN_DEVICE_TYPE_STRING ("unknown_device_type")
#define RTMP_DEFAULT_CLIENT_VERSION ("rtmp client 1.0") //"afcCli 11,0,100,1 (compatible; FMSc/1.0)"
#define RTMP_CMD_NAME_CONNECT "connect"
#define RTMP_CMD_NAME_CREATESTREAM "createStream"
#define RTMP_CMD_NAME_RELEASESTREAM "releaseStream"
#define RTMP_CMD_NAME_DELETESTREAM "deleteStream"
#define RTMP_CMD_NAME_CLOSESTREAM "closeStream"
#define RTMP_CMD_NAME_PLAY "play"
#define RTMP_CMD_NAME_ONSTATUS "onStatus"
#define RTMP_CMD_NAME_PUBLISH "publish"
#define RTMP_CMD_NAME_FCPUBLISH "FCPublish"
#define RTMP_CMD_NAME_FCUNPUBLISH "FCUnpublish"
#define RTMP_CMD_NAME_ONFCPUBLISH "onFCPublish"
#define RTMP_CMD_NAME_ONFCUNPUBLISH "onFCUnpublish"
#define RTMP_CMD_NAME_CLOSE "close"
#define RTMP_CMD_NAME_SETCHALLENGE "setChallenge"
#define RTMP_CMD_DATA_SETDATAFRAME "@setDataFrame"
#define RTMP_CMD_NAME_ONFI "onFI"
#define RTMP_CMD_NAME_ONCLIENTLOGIN "onClientLogin"
#define RTMP_CMD_DATA_ONMETADATA "onMetaData"
#define RTMP_ACK_NAME_RESULT "_result"
#define RTMP_ACK_NAME_ERROR "_error"
#define RTMP_ACK_NAME_BWDONE "onBWDone"
#define RTMP_PING "ping"

#define RTMP_DEFAULT_ACKNOWNLEDGEMENT_SIZE 2500000
#define RTMP_DEFAULT_PEER_BANDWIDTH 2500000
#define RTMP_DEFAULT_CHUNK_SIZE 128

#define RTMP_VIDEO_DATA_MIN_SIZE (1 + 1 + 3) // control(1) + sequence(1) + offsettime(3)
#define RTMP_AAC_AUDIO_DATA_MIN_SIZE (1 + 1) // control(1) + sequence(1)

#define RTMP_UNKNOWN_DEVICE_TYPE_STRING "unknown_device_type"
#define RTMP_DEFAULT_CLIENT_VERSION "rtmp client 1.0" //"afcCli 11,0,100,1 (compatible; FMSc/1.0)"

enum class RtmpEncoderType : int32_t
{
Custom = 0, // 일반 Rtmp 클라이언트(기타 구분 불가능 Client 포함)
Custom = 0, // General Rtmp client (Client that cannot be distinguished from others)
Xsplit, // XSplit
OBS, // OBS
Lavf, // Libavformat (lavf)
};

enum class RtmpFrameType : int32_t
{
VideoIFrame = 'I', // VIDEO I Frame
VideoPFrame = 'P', // VIDEO P Frame
AudioFrame = 'A', // AUDIO Frame
};

enum class RtmpCodecType : int32_t
{
Unknown,
Expand All @@ -228,27 +159,6 @@ enum class RtmpCodecType : int32_t
SPEEX, // SPEEX(11)
};

struct RtmpFrameInfo
{
public:
RtmpFrameInfo(uint32_t timestamp_, int composition_time_offset_, RtmpFrameType frame_type_, int frame_size, uint8_t *frame)
{
timestamp = timestamp_;
composition_time_offset = composition_time_offset_;
frame_type = frame_type_;
frame_data = std::make_shared<std::vector<uint8_t>>(frame, frame + frame_size);
}

public:
uint32_t timestamp;
int composition_time_offset;
RtmpFrameType frame_type;
std::shared_ptr<std::vector<uint8_t>> frame_data;
};

//===============================================================================================
// Rtmp Handshake State
//===============================================================================================
enum class RtmpHandshakeState
{
Uninitialized = 0,
Expand Down Expand Up @@ -283,5 +193,3 @@ struct RtmpMediaInfo
uint32_t timestamp_scale = RTMP_TIME_SCALE;
RtmpEncoderType encoder_type = RtmpEncoderType::Custom;
};

#pragma pack()
Loading

0 comments on commit 342e901

Please sign in to comment.