Merge remote-tracking branch 'origin/master' into dev

This commit is contained in:
xiongguangjie 2024-01-11 16:28:06 +08:00
commit 08f9e83476
19 changed files with 174 additions and 118 deletions

@ -1 +1 @@
Subproject commit 6a8bdfc80db3930d2ba8067d06db467dad8baddf Subproject commit e5535a7164f55eb9062213f40ddc68c0294e6f57

View File

@ -370,7 +370,7 @@ if(ENABLE_JEMALLOC_STATIC)
set(ENABLE_JEMALLOC_STAT OFF) set(ENABLE_JEMALLOC_STAT OFF)
endif () endif ()
include(Jemalloc) include(Jemalloc)
include_directories(SYSTEM ${DEP_ROOT_DIR}/${JEMALLOC_NAME}/include/jemalloc) include_directories(SYSTEM ${DEP_ROOT_DIR}/${JEMALLOC_NAME}/include)
link_directories(${DEP_ROOT_DIR}/${JEMALLOC_NAME}/lib) link_directories(${DEP_ROOT_DIR}/${JEMALLOC_NAME}/lib)
# #
# Used to affect subsequent lookup process # Used to affect subsequent lookup process

View File

@ -135,6 +135,8 @@ segDur=2
#m3u8索引中,hls保留切片个数(实际保留切片个数大2~3个) #m3u8索引中,hls保留切片个数(实际保留切片个数大2~3个)
#如果设置为0则不删除切片而是保存为点播 #如果设置为0则不删除切片而是保存为点播
segNum=3 segNum=3
#HLS切片延迟个数大于0将生成hls_delay.m3u8文件0则不生成
segDelay=0
#HLS切片从m3u8文件中移除后继续保留在磁盘上的个数 #HLS切片从m3u8文件中移除后继续保留在磁盘上的个数
segRetain=5 segRetain=5
#是否广播 hls切片(ts/fmp4)完成通知(on_record_ts) #是否广播 hls切片(ts/fmp4)完成通知(on_record_ts)

View File

@ -307,6 +307,7 @@ namespace Hls {
const string kSegmentDuration = HLS_FIELD "segDur"; const string kSegmentDuration = HLS_FIELD "segDur";
const string kSegmentNum = HLS_FIELD "segNum"; const string kSegmentNum = HLS_FIELD "segNum";
const string kSegmentKeep = HLS_FIELD "segKeep"; const string kSegmentKeep = HLS_FIELD "segKeep";
const string kSegmentDelay = HLS_FIELD "segDelay";
const string kSegmentRetain = HLS_FIELD "segRetain"; const string kSegmentRetain = HLS_FIELD "segRetain";
const string kFileBufSize = HLS_FIELD "fileBufSize"; const string kFileBufSize = HLS_FIELD "fileBufSize";
const string kBroadcastRecordTs = HLS_FIELD "broadcastRecordTs"; const string kBroadcastRecordTs = HLS_FIELD "broadcastRecordTs";
@ -317,6 +318,7 @@ static onceToken token([]() {
mINI::Instance()[kSegmentDuration] = 2; mINI::Instance()[kSegmentDuration] = 2;
mINI::Instance()[kSegmentNum] = 3; mINI::Instance()[kSegmentNum] = 3;
mINI::Instance()[kSegmentKeep] = false; mINI::Instance()[kSegmentKeep] = false;
mINI::Instance()[kSegmentDelay] = 0;
mINI::Instance()[kSegmentRetain] = 5; mINI::Instance()[kSegmentRetain] = 5;
mINI::Instance()[kFileBufSize] = 64 * 1024; mINI::Instance()[kFileBufSize] = 64 * 1024;
mINI::Instance()[kBroadcastRecordTs] = false; mINI::Instance()[kBroadcastRecordTs] = false;

View File

@ -349,6 +349,8 @@ extern const std::string kSegmentDuration;
extern const std::string kSegmentNum; extern const std::string kSegmentNum;
// 如果设置为0则不保留切片设置为1则一直保留切片 // 如果设置为0则不保留切片设置为1则一直保留切片
extern const std::string kSegmentKeep; extern const std::string kSegmentKeep;
// HLS切片延迟个数大于0将生成hls_delay.m3u8文件0则不生成
extern const std::string kSegmentDelay;
// HLS切片从m3u8文件中移除后继续保留在磁盘上的个数 // HLS切片从m3u8文件中移除后继续保留在磁盘上的个数
extern const std::string kSegmentRetain; extern const std::string kSegmentRetain;
// HLS文件写缓存大小 // HLS文件写缓存大小

View File

@ -69,12 +69,16 @@ CodecId getCodecByMovId(int object_id) {
if (object_id == MOV_OBJECT_NONE) { if (object_id == MOV_OBJECT_NONE) {
return CodecInvalid; return CodecInvalid;
} }
switch (object_id) {
#define XX(name, type, value, str, mpeg_id, mp4_id) case mp4_id : return name; #define XX(name, type, value, str, mpeg_id, mp4_id) { mp4_id, name },
CODEC_MAP(XX) static map<int, CodecId> s_map = { CODEC_MAP(XX) };
#undef XX #undef XX
default : WarnL << "Unsupported mov: " << object_id; return CodecInvalid; auto it = s_map.find(object_id);
if (it == s_map.end()) {
WarnL << "Unsupported mov: " << object_id;
return CodecInvalid;
} }
return it->second;
} }
#endif #endif
@ -89,17 +93,20 @@ int getMpegIdByCodec(CodecId codec) {
} }
CodecId getCodecByMpegId(int mpeg_id) { CodecId getCodecByMpegId(int mpeg_id) {
if (mpeg_id == PSI_STREAM_RESERVED) { if (mpeg_id == PSI_STREAM_RESERVED || mpeg_id == 0xBD) {
// 海康的 PS 流中会有0xBD 的包
return CodecInvalid; return CodecInvalid;
} }
switch (mpeg_id) {
#define XX(name, type, value, str, mpeg_id, mp4_id) case mpeg_id : return name; #define XX(name, type, value, str, mpeg_id, mp4_id) { mpeg_id, name },
CODEC_MAP(XX) static map<int, CodecId> s_map = { CODEC_MAP(XX) };
#undef XX #undef XX
// 海康的 PS 流中会有0xBD 的包 auto it = s_map.find(mpeg_id);
case 0xBD: return CodecInvalid; if (it == s_map.end()) {
default : WarnL << "Unsupported mpeg: " << mpeg_id; return CodecInvalid; WarnL << "Unsupported mpeg: " << mpeg_id;
return CodecInvalid;
} }
return it->second;
} }
#endif #endif

View File

@ -129,8 +129,8 @@ public:
_fps = fps; _fps = fps;
} }
int getVideoHeight() const override { return _width; } int getVideoWidth() const override { return _width; }
int getVideoWidth() const override { return _height; } int getVideoHeight() const override { return _height; }
float getVideoFps() const override { return _fps; } float getVideoFps() const override { return _fps; }
bool ready() const override { return true; } bool ready() const override { return true; }

View File

@ -24,15 +24,40 @@ HlsMaker::HlsMaker(bool is_fmp4, float seg_duration, uint32_t seg_number, bool s
_seg_keep = seg_keep; _seg_keep = seg_keep;
} }
void HlsMaker::makeIndexFile(bool eof) { void HlsMaker::makeIndexFile(bool include_delay, bool eof) {
GET_CONFIG(uint32_t, segDelay, Hls::kSegmentDelay);
GET_CONFIG(uint32_t, segRetain, Hls::kSegmentRetain);
std::deque<std::tuple<int, std::string>> temp(_seg_dur_list);
if (!include_delay) {
while (temp.size() > _seg_number) {
temp.pop_front();
}
}
int maxSegmentDuration = 0; int maxSegmentDuration = 0;
for (auto &tp : _seg_dur_list) { for (auto &tp : temp) {
int dur = std::get<0>(tp); int dur = std::get<0>(tp);
if (dur > maxSegmentDuration) { if (dur > maxSegmentDuration) {
maxSegmentDuration = dur; maxSegmentDuration = dur;
} }
} }
auto index_seq = _seg_number ? (_file_index > _seg_number ? _file_index - _seg_number : 0LL) : 0LL; uint64_t index_seq;
if (_seg_number) {
if (include_delay) {
if (_file_index > _seg_number + segDelay) {
index_seq = _file_index - _seg_number - segDelay;
} else {
index_seq = 0LL;
}
} else {
if (_file_index > _seg_number) {
index_seq = _file_index - _seg_number;
} else {
index_seq = 0LL;
}
}
} else {
index_seq = 0LL;
}
string index_str; string index_str;
index_str.reserve(2048); index_str.reserve(2048);
@ -50,7 +75,7 @@ void HlsMaker::makeIndexFile(bool eof) {
} }
stringstream ss; stringstream ss;
for (auto &tp : _seg_dur_list) { for (auto &tp : temp) {
ss << "#EXTINF:" << std::setprecision(3) << std::get<0>(tp) / 1000.0 << ",\n" << std::get<1>(tp) << "\n"; ss << "#EXTINF:" << std::setprecision(3) << std::get<0>(tp) / 1000.0 << ",\n" << std::get<1>(tp) << "\n";
} }
index_str += ss.str(); index_str += ss.str();
@ -58,7 +83,7 @@ void HlsMaker::makeIndexFile(bool eof) {
if (eof) { if (eof) {
index_str += "#EXT-X-ENDLIST\n"; index_str += "#EXT-X-ENDLIST\n";
} }
onWriteHls(index_str); onWriteHls(index_str, include_delay);
} }
void HlsMaker::inputInitSegment(const char *data, size_t len) { void HlsMaker::inputInitSegment(const char *data, size_t len) {
@ -91,12 +116,13 @@ void HlsMaker::inputData(const char *data, size_t len, uint64_t timestamp, bool
} }
void HlsMaker::delOldSegment() { void HlsMaker::delOldSegment() {
GET_CONFIG(uint32_t, segDelay, Hls::kSegmentDelay);
if (_seg_number == 0) { if (_seg_number == 0) {
//如果设置为保留0个切片则认为是保存为点播 //如果设置为保留0个切片则认为是保存为点播
return; return;
} }
//在hls m3u8索引文件中,我们保存的切片个数跟_seg_number相关设置一致 //在hls m3u8索引文件中,我们保存的切片个数跟_seg_number相关设置一致
if (_file_index > _seg_number) { if (_file_index > _seg_number + segDelay) {
_seg_dur_list.pop_front(); _seg_dur_list.pop_front();
} }
//如果设置为一直保存,就不删除 //如果设置为一直保存,就不删除
@ -105,8 +131,8 @@ void HlsMaker::delOldSegment() {
} }
GET_CONFIG(uint32_t, segRetain, Hls::kSegmentRetain); GET_CONFIG(uint32_t, segRetain, Hls::kSegmentRetain);
//但是实际保存的切片个数比m3u8所述多若干个,这样做的目的是防止播放器在切片删除前能下载完毕 //但是实际保存的切片个数比m3u8所述多若干个,这样做的目的是防止播放器在切片删除前能下载完毕
if (_file_index > _seg_number + segRetain) { if (_file_index > _seg_number + segDelay + segRetain) {
onDelSegment(_file_index - _seg_number - segRetain - 1); onDelSegment(_file_index - _seg_number - segDelay - segRetain - 1);
} }
} }
@ -125,6 +151,7 @@ void HlsMaker::addNewSegment(uint64_t stamp) {
} }
void HlsMaker::flushLastSegment(bool eof){ void HlsMaker::flushLastSegment(bool eof){
GET_CONFIG(uint32_t, segDelay, Hls::kSegmentDelay);
if (_last_file_name.empty()) { if (_last_file_name.empty()) {
//不存在上个切片 //不存在上个切片
return; return;
@ -139,7 +166,11 @@ void HlsMaker::flushLastSegment(bool eof){
//先flush ts切片否则可能存在ts文件未写入完毕就被访问的情况 //先flush ts切片否则可能存在ts文件未写入完毕就被访问的情况
onFlushLastSegment(seg_dur); onFlushLastSegment(seg_dur);
//然后写m3u8文件 //然后写m3u8文件
makeIndexFile(eof); makeIndexFile(false, eof);
//写入切片延迟的m3u8文件
if (segDelay) {
makeIndexFile(true, eof);
}
} }
bool HlsMaker::isLive() const { bool HlsMaker::isLive() const {

View File

@ -96,7 +96,7 @@ protected:
/** /**
* m3u8文件回调 * m3u8文件回调
*/ */
virtual void onWriteHls(const std::string &data) = 0; virtual void onWriteHls(const std::string &data, bool include_delay) = 0;
/** /**
* ts , * ts ,
@ -115,7 +115,7 @@ private:
* m3u8文件 * m3u8文件
* @param eof true代表点播 * @param eof true代表点播
*/ */
void makeIndexFile(bool eof = false); void makeIndexFile(bool include_delay, bool eof = false);
/** /**
* ts切片 * ts切片

View File

@ -21,11 +21,20 @@ using namespace toolkit;
namespace mediakit { namespace mediakit {
std::string getDelayPath(const std::string& originalPath) {
std::size_t pos = originalPath.find(".m3u8");
if (pos != std::string::npos) {
return originalPath.substr(0, pos) + "_delay.m3u8";
}
return originalPath;
}
HlsMakerImp::HlsMakerImp(bool is_fmp4, const string &m3u8_file, const string &params, uint32_t bufSize, float seg_duration, HlsMakerImp::HlsMakerImp(bool is_fmp4, const string &m3u8_file, const string &params, uint32_t bufSize, float seg_duration,
uint32_t seg_number, bool seg_keep) : HlsMaker(is_fmp4, seg_duration, seg_number, seg_keep) { uint32_t seg_number, bool seg_keep) : HlsMaker(is_fmp4, seg_duration, seg_number, seg_keep) {
_poller = EventPollerPool::Instance().getPoller(); _poller = EventPollerPool::Instance().getPoller();
_path_prefix = m3u8_file.substr(0, m3u8_file.rfind('/')); _path_prefix = m3u8_file.substr(0, m3u8_file.rfind('/'));
_path_hls = m3u8_file; _path_hls = m3u8_file;
_path_hls_delay = getDelayPath(m3u8_file);
_params = params; _params = params;
_buf_size = bufSize; _buf_size = bufSize;
_file_buf.reset(new char[bufSize], [](char *ptr) { delete[] ptr; }); _file_buf.reset(new char[bufSize], [](char *ptr) { delete[] ptr; });
@ -62,6 +71,7 @@ void HlsMakerImp::clearCache(bool immediately, bool eof) {
{ {
std::list<std::string> lst; std::list<std::string> lst;
lst.emplace_back(_path_hls); lst.emplace_back(_path_hls);
lst.emplace_back(_path_hls_delay);
if (!_path_init.empty()) { if (!_path_init.empty()) {
lst.emplace_back(_path_init); lst.emplace_back(_path_init);
} }
@ -146,16 +156,17 @@ void HlsMakerImp::onWriteSegment(const char *data, size_t len) {
} }
} }
void HlsMakerImp::onWriteHls(const std::string &data) { void HlsMakerImp::onWriteHls(const std::string &data, bool include_delay) {
auto hls = makeFile(_path_hls); auto path = include_delay ? _path_hls_delay : _path_hls;
auto hls = makeFile(path);
if (hls) { if (hls) {
fwrite(data.data(), data.size(), 1, hls.get()); fwrite(data.data(), data.size(), 1, hls.get());
hls.reset(); hls.reset();
if (_media_src) { if (_media_src && !include_delay) {
_media_src->setIndexFile(data); _media_src->setIndexFile(data);
} }
} else { } else {
WarnL << "Create hls file failed," << _path_hls << " " << get_uv_errmsg(); WarnL << "Create hls file failed," << path << " " << get_uv_errmsg();
} }
} }

View File

@ -49,7 +49,7 @@ protected:
void onDelSegment(uint64_t index) override; void onDelSegment(uint64_t index) override;
void onWriteInitSegment(const char *data, size_t len) override; void onWriteInitSegment(const char *data, size_t len) override;
void onWriteSegment(const char *data, size_t len) override; void onWriteSegment(const char *data, size_t len) override;
void onWriteHls(const std::string &data) override; void onWriteHls(const std::string &data, bool include_delay) override;
void onFlushLastSegment(uint64_t duration_ms) override; void onFlushLastSegment(uint64_t duration_ms) override;
private: private:
@ -60,6 +60,7 @@ private:
int _buf_size; int _buf_size;
std::string _params; std::string _params;
std::string _path_hls; std::string _path_hls;
std::string _path_hls_delay;
std::string _path_init; std::string _path_init;
std::string _path_prefix; std::string _path_prefix;
RecordInfo _info; RecordInfo _info;

View File

@ -210,7 +210,7 @@ string FCI_NACK::dumpString() const {
} }
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma pack(push, 1)
class RunLengthChunk { class RunLengthChunk {
public: public:
static size_t constexpr kSize = 2; static size_t constexpr kSize = 2;
@ -241,6 +241,7 @@ public:
// 打印本对象 // 打印本对象
string dumpString() const; string dumpString() const;
}; };
#pragma pack(pop)
RunLengthChunk::RunLengthChunk(SymbolStatus status, uint16_t run_length) { RunLengthChunk::RunLengthChunk(SymbolStatus status, uint16_t run_length) {
type = 0; type = 0;
@ -261,7 +262,7 @@ string RunLengthChunk::dumpString() const {
} }
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#pragma pack(push, 1)
class StatusVecChunk { class StatusVecChunk {
public: public:
static size_t constexpr kSize = 2; static size_t constexpr kSize = 2;
@ -292,6 +293,7 @@ public:
// 打印本对象 // 打印本对象
string dumpString() const; string dumpString() const;
}; };
#pragma pack(pop)
StatusVecChunk::StatusVecChunk(bool symbol_bit, const vector<SymbolStatus> &status) { StatusVecChunk::StatusVecChunk(bool symbol_bit, const vector<SymbolStatus> &status) {
CHECK(status.size() << symbol_bit <= 14); CHECK(status.size() << symbol_bit <= 14);

View File

@ -14,6 +14,7 @@
#include "Rtcp.h" #include "Rtcp.h"
namespace mediakit { namespace mediakit {
#pragma pack(push, 1)
/////////////////////////////////////////// PSFB //////////////////////////////////////////////////// /////////////////////////////////////////// PSFB ////////////////////////////////////////////////////
@ -375,6 +376,6 @@ private:
// feedback packet count,反馈包号,本包是第几个transport-cc包每次加1 | // feedback packet count,反馈包号,本包是第几个transport-cc包每次加1 |
uint8_t fb_pkt_count; uint8_t fb_pkt_count;
}; };
#pragma pack(pop)
} // namespace mediakit } // namespace mediakit
#endif // ZLMEDIAKIT_RTCPFCI_H #endif // ZLMEDIAKIT_RTCPFCI_H

View File

@ -180,7 +180,7 @@ int main(int argc, char *argv[]) {
auto pusher = std::make_shared<MediaPusher>(src); auto pusher = std::make_shared<MediaPusher>(src);
pusher->setOnCreateSocket([](const EventPoller::Ptr &poller) { pusher->setOnCreateSocket([](const EventPoller::Ptr &poller) {
//socket关闭互斥锁提高性能 //socket关闭互斥锁提高性能
return std::make_shared<Socket>(poller, false); return Socket::createSocket(poller, false);
}); });
//设置推流失败监听 //设置推流失败监听
pusher->setOnPublished([&mtx, &pusher_map, index](const SockException &ex) { pusher->setOnPublished([&mtx, &pusher_map, index](const SockException &ex) {

View File

@ -123,7 +123,7 @@ int main(int argc, char *argv[]) {
auto tag = player.get(); auto tag = player.get();
player->setOnCreateSocket([](const EventPoller::Ptr &poller) { player->setOnCreateSocket([](const EventPoller::Ptr &poller) {
//socket关闭互斥锁提高性能 //socket关闭互斥锁提高性能
return std::make_shared<Socket>(poller, false); return Socket::createSocket(poller, false);
}); });
//设置播放失败监听 //设置播放失败监听
player->setOnPlayResult([&mtx, &player_map, tag](const SockException &ex) { player->setOnPlayResult([&mtx, &player_map, tag](const SockException &ex) {

View File

@ -166,7 +166,7 @@ int main(int argc, char *argv[]) {
auto tag = pusher.get(); auto tag = pusher.get();
pusher->setOnCreateSocket([](const EventPoller::Ptr &poller) { pusher->setOnCreateSocket([](const EventPoller::Ptr &poller) {
//socket关闭互斥锁提高性能 //socket关闭互斥锁提高性能
return std::make_shared<Socket>(poller, false); return Socket::createSocket(poller, false);
}); });
//设置推流失败监听 //设置推流失败监听
pusher->setOnPublished([&mtx, &pusher_map, tag](const SockException &ex) { pusher->setOnPublished([&mtx, &pusher_map, tag](const SockException &ex) {

View File

@ -10,11 +10,8 @@
#include <map> #include <map>
#include <iostream> #include <iostream>
#include "Util/MD5.h"
#include "Util/File.h"
#include "Util/logger.h"
#include "Util/SSLBox.h"
#include "Util/util.h" #include "Util/util.h"
#include "Util/logger.h"
#include "Network/TcpServer.h" #include "Network/TcpServer.h"
#include "Common/config.h" #include "Common/config.h"
#include "Rtsp/RtspSession.h" #include "Rtsp/RtspSession.h"
@ -29,102 +26,102 @@ using namespace mediakit;
static semaphore sem; static semaphore sem;
#if defined(ENABLE_RTPPROXY) #if defined(ENABLE_RTPPROXY)
static bool loadFile(const char *path, const EventPoller::Ptr &poller){ static bool loadFile(const char *path, const EventPoller::Ptr &poller) {
FILE *fp = fopen(path, "rb"); std::shared_ptr<FILE> fp(fopen(path, "rb"), [](FILE *fp) {
sem.post();
if (fp) {
fclose(fp);
}
});
if (!fp) { if (!fp) {
WarnL << "open file failed:" << path; WarnL << "open file failed:" << path;
return false; return false;
} }
uint64_t timeStamp_last = 0;
uint16_t len;
char rtp[0xFFFF];
struct sockaddr_storage addr; struct sockaddr_storage addr;
memset(&addr, 0, sizeof(addr)); memset(&addr, 0, sizeof(addr));
addr.ss_family = AF_INET; addr.ss_family = AF_INET;
auto sock = Socket::createSocket(poller); auto sock = Socket::createSocket(poller);
size_t total_size = 0; auto process = RtpSelector::Instance().getProcess("test", true);
RtpProcess::Ptr process;
uint32_t ssrc = 0;
while (true) {
if (2 != fread(&len, 1, 2, fp)) {
WarnL;
break;
}
len = ntohs(len);
if (len < 12 || len > sizeof(rtp)) {
WarnL << len;
break;
}
if (len != fread(rtp, 1, len, fp)) { uint64_t stamp_last = 0;
WarnL; auto total_size = std::make_shared<size_t>(0);
break; auto do_read = [fp, total_size, sock, addr, process, stamp_last]() mutable -> int {
} uint16_t len;
total_size += len; char rtp[0xFFFF];
uint64_t timeStamp = 0; while (true) {
if (2 != fread(&len, 1, 2, fp.get())) {
if (!process) { WarnL << "Read rtp size failed";
if (!RtpSelector::getSSRC(rtp, len, ssrc)) { // 重新播放
WarnL << "get ssrc from rtp failed:" << len; fseek(fp.get(), 0, SEEK_SET);
return false; return 1;
} }
process = RtpSelector::Instance().getProcess(printSSRC(ssrc), true); len = ntohs(len);
} if (len < 12 || len > sizeof(rtp)) {
if (process) { WarnL << "Invalid rtp size: " << len;
return 0;
}
if (len != fread(rtp, 1, len, fp.get())) {
WarnL << "Read rtp data failed";
return 0;
}
(*total_size) += len;
uint64_t stamp = 0;
try { try {
process->inputRtp(true, sock, rtp, len, (struct sockaddr *)&addr, &timeStamp); process->inputRtp(true, sock, rtp, len, (struct sockaddr *)&addr, &stamp);
} catch (...) { } catch (std::exception &ex) {
RtpSelector::Instance().delProcess(printSSRC(ssrc), process.get()); WarnL << "Input rtp failed: " << ex.what();
throw; return 0;
}
auto diff = stamp - stamp_last;
if (diff < 0 || diff > 500) {
diff = 1;
}
if (diff) {
stamp_last = stamp;
return diff;
} }
} }
};
auto diff = timeStamp - timeStamp_last; poller->doDelayTask(1, [do_read, total_size, process]() mutable {
if (diff > 0 && diff < 500) { auto ret = do_read();
usleep(diff * 1000); if (!ret) {
} else { WarnL << *total_size / 1024 << "KB";
usleep(1 * 1000); RtpSelector::Instance().delProcess("test", process.get());
} }
timeStamp_last = timeStamp; return ret;
} });
WarnL << total_size / 1024 << "KB";
fclose(fp);
return true; return true;
} }
#endif//#if defined(ENABLE_RTPPROXY) #endif // #if defined(ENABLE_RTPPROXY)
int main(int argc,char *argv[]) { int main(int argc, char *argv[]) {
//设置日志 // 设置日志
Logger::Instance().add(std::make_shared<ConsoleChannel>("ConsoleChannel")); Logger::Instance().add(std::make_shared<ConsoleChannel>("ConsoleChannel"));
#if defined(ENABLE_RTPPROXY) #if defined(ENABLE_RTPPROXY)
//启动异步日志线程 // 启动异步日志线程
Logger::Instance().setWriter(std::make_shared<AsyncLogWriter>()); Logger::Instance().setWriter(std::make_shared<AsyncLogWriter>());
loadIniConfig((exeDir() + "config.ini").data()); loadIniConfig((exeDir() + "config.ini").data());
TcpServer::Ptr rtspSrv(new TcpServer()); TcpServer::Ptr rtspSrv(new TcpServer());
TcpServer::Ptr rtmpSrv(new TcpServer()); TcpServer::Ptr rtmpSrv(new TcpServer());
TcpServer::Ptr httpSrv(new TcpServer()); TcpServer::Ptr httpSrv(new TcpServer());
rtspSrv->start<RtspSession>(554);//默认554 rtspSrv->start<RtspSession>(554); // 默认554
rtmpSrv->start<RtmpSession>(1935);//默认1935 rtmpSrv->start<RtmpSession>(1935); // 默认1935
httpSrv->start<HttpSession>(80);//默认80 httpSrv->start<HttpSession>(80); // 默认80
//此处选择是否导出调试文件 // 此处选择是否导出调试文件
// mINI::Instance()[RtpProxy::kDumpDir] = "/Users/xzl/Desktop/"; // mINI::Instance()[RtpProxy::kDumpDir] = "/Users/xzl/Desktop/";
if (argc == 2){ if (argc == 2) {
auto poller = EventPollerPool::Instance().getPoller(); loadFile(argv[1], EventPollerPool::Instance().getPoller());
poller->async_first([poller,argv](){
loadFile(argv[1],poller);
sem.post();
});
sem.wait(); sem.wait();
sleep(1); } else {
ErrorL << "parameter error.";
} }
else
ErrorL << "parameter error.";
#else #else
ErrorL << "please ENABLE_RTPPROXY and then test"; ErrorL << "please ENABLE_RTPPROXY and then test";
#endif//#if defined(ENABLE_RTPPROXY) #endif // #if defined(ENABLE_RTPPROXY)
return 0; return 0;
} }

View File

@ -677,9 +677,9 @@ namespace RTC
if (notification->sn_header.sn_length > 0) if (notification->sn_header.sn_length > 0)
{ {
static const size_t BufferSize{ 1024 }; static const size_t BufferSize{ 1024 };
static char buffer[BufferSize]; thread_local static char buffer[BufferSize];
uint32_t len = notification->sn_header.sn_length; uint32_t len = notification->sn_assoc_change.sac_length - sizeof(struct sctp_assoc_change);
for (uint32_t i{ 0 }; i < len; ++i) for (uint32_t i{ 0 }; i < len; ++i)
{ {
@ -745,9 +745,9 @@ namespace RTC
if (notification->sn_header.sn_length > 0) if (notification->sn_header.sn_length > 0)
{ {
static const size_t BufferSize{ 1024 }; static const size_t BufferSize{ 1024 };
static char buffer[BufferSize]; thread_local static char buffer[BufferSize];
uint32_t len = notification->sn_header.sn_length; uint32_t len = notification->sn_assoc_change.sac_length - sizeof(struct sctp_assoc_change);
for (uint32_t i{ 0 }; i < len; ++i) for (uint32_t i{ 0 }; i < len; ++i)
{ {
@ -786,7 +786,7 @@ namespace RTC
case SCTP_REMOTE_ERROR: case SCTP_REMOTE_ERROR:
{ {
static const size_t BufferSize{ 1024 }; static const size_t BufferSize{ 1024 };
static char buffer[BufferSize]; thread_local static char buffer[BufferSize];
uint32_t len = notification->sn_remote_error.sre_length - sizeof(struct sctp_remote_error); uint32_t len = notification->sn_remote_error.sre_length - sizeof(struct sctp_remote_error);
@ -822,7 +822,7 @@ namespace RTC
case SCTP_SEND_FAILED_EVENT: case SCTP_SEND_FAILED_EVENT:
{ {
static const size_t BufferSize{ 1024 }; static const size_t BufferSize{ 1024 };
static char buffer[BufferSize]; thread_local static char buffer[BufferSize];
uint32_t len = uint32_t len =
notification->sn_send_failed_event.ssfe_length - sizeof(struct sctp_send_failed_event); notification->sn_send_failed_event.ssfe_length - sizeof(struct sctp_send_failed_event);
@ -1004,4 +1004,4 @@ namespace RTC
} // namespace RTC } // namespace RTC
#endif //ENABLE_SCTP #endif //ENABLE_SCTP

View File

@ -59,8 +59,8 @@ static onceToken token([]() {
mINI::Instance()[kTimeOutSec] = 15; mINI::Instance()[kTimeOutSec] = 15;
mINI::Instance()[kExternIP] = ""; mINI::Instance()[kExternIP] = "";
mINI::Instance()[kRembBitRate] = 0; mINI::Instance()[kRembBitRate] = 0;
mINI::Instance()[kPort] = 0; mINI::Instance()[kPort] = 8000;
mINI::Instance()[kTcpPort] = 0; mINI::Instance()[kTcpPort] = 8000;
mINI::Instance()[kStartBitrate] = 0; mINI::Instance()[kStartBitrate] = 0;
mINI::Instance()[kMaxBitrate] = 0; mINI::Instance()[kMaxBitrate] = 0;
@ -1380,4 +1380,4 @@ static onceToken s_rtc_auto_register([]() {
}); });
}); });
}// namespace mediakit }// namespace mediakit