mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2026-06-22 08:52:20 +08:00
for rtp server can config pt
This commit is contained in:
xiongguangjie
@@ -8,31 +8,31 @@
|
||||
* may be found in the AUTHORS file in the root of the source tree.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <assert.h>
|
||||
#include "Common/config.h"
|
||||
#include "Util/util.h"
|
||||
#include "Util/NoticeCenter.h"
|
||||
#include "Util/logger.h"
|
||||
#include "Util/onceToken.h"
|
||||
#include "Util/NoticeCenter.h"
|
||||
#include "Util/util.h"
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
|
||||
using namespace std;
|
||||
using namespace toolkit;
|
||||
|
||||
namespace mediakit {
|
||||
|
||||
bool loadIniConfig(const char *ini_path){
|
||||
bool loadIniConfig(const char *ini_path) {
|
||||
string ini;
|
||||
if(ini_path && ini_path[0] != '\0'){
|
||||
if (ini_path && ini_path[0] != '\0') {
|
||||
ini = ini_path;
|
||||
}else{
|
||||
} else {
|
||||
ini = exePath() + ".ini";
|
||||
}
|
||||
try{
|
||||
try {
|
||||
mINI::Instance().parseFile(ini);
|
||||
NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastReloadConfig);
|
||||
return true;
|
||||
}catch (std::exception &) {
|
||||
} catch (std::exception &) {
|
||||
InfoL << "dump ini file to:" << ini;
|
||||
mINI::Instance().dumpFile(ini);
|
||||
return false;
|
||||
@@ -55,36 +55,36 @@ const string kBroadcastShellLogin = "kBroadcastShellLogin";
|
||||
const string kBroadcastNotFoundStream = "kBroadcastNotFoundStream";
|
||||
const string kBroadcastStreamNoneReader = "kBroadcastStreamNoneReader";
|
||||
const string kBroadcastHttpBeforeAccess = "kBroadcastHttpBeforeAccess";
|
||||
} //namespace Broadcast
|
||||
} // namespace Broadcast
|
||||
|
||||
//通用配置项目
|
||||
namespace General{
|
||||
// 通用配置项目
|
||||
namespace General {
|
||||
#define GENERAL_FIELD "general."
|
||||
const string kMediaServerId = GENERAL_FIELD"mediaServerId";
|
||||
const string kFlowThreshold = GENERAL_FIELD"flowThreshold";
|
||||
const string kStreamNoneReaderDelayMS = GENERAL_FIELD"streamNoneReaderDelayMS";
|
||||
const string kMaxStreamWaitTimeMS = GENERAL_FIELD"maxStreamWaitMS";
|
||||
const string kEnableVhost = GENERAL_FIELD"enableVhost";
|
||||
const string kAddMuteAudio = GENERAL_FIELD"addMuteAudio";
|
||||
const string kResetWhenRePlay = GENERAL_FIELD"resetWhenRePlay";
|
||||
const string kPublishToHls = GENERAL_FIELD"publishToHls";
|
||||
const string kPublishToMP4 = GENERAL_FIELD"publishToMP4";
|
||||
const string kMergeWriteMS = GENERAL_FIELD"mergeWriteMS";
|
||||
const string kModifyStamp = GENERAL_FIELD"modifyStamp";
|
||||
const string kHlsDemand = GENERAL_FIELD"hls_demand";
|
||||
const string kRtspDemand = GENERAL_FIELD"rtsp_demand";
|
||||
const string kRtmpDemand = GENERAL_FIELD"rtmp_demand";
|
||||
const string kTSDemand = GENERAL_FIELD"ts_demand";
|
||||
const string kFMP4Demand = GENERAL_FIELD"fmp4_demand";
|
||||
const string kEnableAudio = GENERAL_FIELD"enable_audio";
|
||||
const string kCheckNvidiaDev = GENERAL_FIELD"check_nvidia_dev";
|
||||
const string kEnableFFmpegLog = GENERAL_FIELD"enable_ffmpeg_log";
|
||||
const string kWaitTrackReadyMS = GENERAL_FIELD"wait_track_ready_ms";
|
||||
const string kWaitAddTrackMS = GENERAL_FIELD"wait_add_track_ms";
|
||||
const string kUnreadyFrameCache = GENERAL_FIELD"unready_frame_cache";
|
||||
const string kContinuePushMS = GENERAL_FIELD"continue_push_ms";
|
||||
const string kMediaServerId = GENERAL_FIELD "mediaServerId";
|
||||
const string kFlowThreshold = GENERAL_FIELD "flowThreshold";
|
||||
const string kStreamNoneReaderDelayMS = GENERAL_FIELD "streamNoneReaderDelayMS";
|
||||
const string kMaxStreamWaitTimeMS = GENERAL_FIELD "maxStreamWaitMS";
|
||||
const string kEnableVhost = GENERAL_FIELD "enableVhost";
|
||||
const string kAddMuteAudio = GENERAL_FIELD "addMuteAudio";
|
||||
const string kResetWhenRePlay = GENERAL_FIELD "resetWhenRePlay";
|
||||
const string kPublishToHls = GENERAL_FIELD "publishToHls";
|
||||
const string kPublishToMP4 = GENERAL_FIELD "publishToMP4";
|
||||
const string kMergeWriteMS = GENERAL_FIELD "mergeWriteMS";
|
||||
const string kModifyStamp = GENERAL_FIELD "modifyStamp";
|
||||
const string kHlsDemand = GENERAL_FIELD "hls_demand";
|
||||
const string kRtspDemand = GENERAL_FIELD "rtsp_demand";
|
||||
const string kRtmpDemand = GENERAL_FIELD "rtmp_demand";
|
||||
const string kTSDemand = GENERAL_FIELD "ts_demand";
|
||||
const string kFMP4Demand = GENERAL_FIELD "fmp4_demand";
|
||||
const string kEnableAudio = GENERAL_FIELD "enable_audio";
|
||||
const string kCheckNvidiaDev = GENERAL_FIELD "check_nvidia_dev";
|
||||
const string kEnableFFmpegLog = GENERAL_FIELD "enable_ffmpeg_log";
|
||||
const string kWaitTrackReadyMS = GENERAL_FIELD "wait_track_ready_ms";
|
||||
const string kWaitAddTrackMS = GENERAL_FIELD "wait_add_track_ms";
|
||||
const string kUnreadyFrameCache = GENERAL_FIELD "unready_frame_cache";
|
||||
const string kContinuePushMS = GENERAL_FIELD "continue_push_ms";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kFlowThreshold] = 1024;
|
||||
mINI::Instance()[kStreamNoneReaderDelayMS] = 20 * 1000;
|
||||
mINI::Instance()[kMaxStreamWaitTimeMS] = 15 * 1000;
|
||||
@@ -110,22 +110,22 @@ static onceToken token([](){
|
||||
mINI::Instance()[kContinuePushMS] = 15 * 1000;
|
||||
});
|
||||
|
||||
}//namespace General
|
||||
} // namespace General
|
||||
|
||||
////////////HTTP配置///////////
|
||||
namespace Http {
|
||||
#define HTTP_FIELD "http."
|
||||
const string kSendBufSize = HTTP_FIELD"sendBufSize";
|
||||
const string kMaxReqSize = HTTP_FIELD"maxReqSize";
|
||||
const string kKeepAliveSecond = HTTP_FIELD"keepAliveSecond";
|
||||
const string kCharSet = HTTP_FIELD"charSet";
|
||||
const string kRootPath = HTTP_FIELD"rootPath";
|
||||
const string kSendBufSize = HTTP_FIELD "sendBufSize";
|
||||
const string kMaxReqSize = HTTP_FIELD "maxReqSize";
|
||||
const string kKeepAliveSecond = HTTP_FIELD "keepAliveSecond";
|
||||
const string kCharSet = HTTP_FIELD "charSet";
|
||||
const string kRootPath = HTTP_FIELD "rootPath";
|
||||
const string kVirtualPath = HTTP_FIELD "virtualPath";
|
||||
const string kNotFound = HTTP_FIELD"notFound";
|
||||
const string kDirMenu = HTTP_FIELD"dirMenu";
|
||||
const string kForbidCacheSuffix = HTTP_FIELD"forbidCacheSuffix";
|
||||
const string kNotFound = HTTP_FIELD "notFound";
|
||||
const string kDirMenu = HTTP_FIELD "dirMenu";
|
||||
const string kForbidCacheSuffix = HTTP_FIELD "forbidCacheSuffix";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kSendBufSize] = 64 * 1024;
|
||||
mINI::Instance()[kMaxReqSize] = 4 * 10240;
|
||||
mINI::Instance()[kKeepAliveSecond] = 15;
|
||||
@@ -135,7 +135,7 @@ static onceToken token([](){
|
||||
#if defined(_WIN32)
|
||||
mINI::Instance()[kCharSet] = "gb2312";
|
||||
#else
|
||||
mINI::Instance()[kCharSet] ="utf-8";
|
||||
mINI::Instance()[kCharSet] = "utf-8";
|
||||
#endif
|
||||
|
||||
mINI::Instance()[kRootPath] = "./www";
|
||||
@@ -149,121 +149,118 @@ static onceToken token([](){
|
||||
"</body>"
|
||||
"</html>"
|
||||
<< endl;
|
||||
mINI::Instance()[kForbidCacheSuffix] = "";
|
||||
mINI::Instance()[kForbidCacheSuffix] = "";
|
||||
});
|
||||
|
||||
}//namespace Http
|
||||
} // namespace Http
|
||||
|
||||
////////////SHELL配置///////////
|
||||
namespace Shell {
|
||||
#define SHELL_FIELD "shell."
|
||||
const string kMaxReqSize = SHELL_FIELD"maxReqSize";
|
||||
const string kMaxReqSize = SHELL_FIELD "maxReqSize";
|
||||
|
||||
static onceToken token([](){
|
||||
mINI::Instance()[kMaxReqSize] = 1024;
|
||||
});
|
||||
} //namespace Shell
|
||||
static onceToken token([]() { mINI::Instance()[kMaxReqSize] = 1024; });
|
||||
} // namespace Shell
|
||||
|
||||
////////////RTSP服务器配置///////////
|
||||
namespace Rtsp {
|
||||
#define RTSP_FIELD "rtsp."
|
||||
const string kAuthBasic = RTSP_FIELD"authBasic";
|
||||
const string kHandshakeSecond = RTSP_FIELD"handshakeSecond";
|
||||
const string kKeepAliveSecond = RTSP_FIELD"keepAliveSecond";
|
||||
const string kDirectProxy = RTSP_FIELD"directProxy";
|
||||
const string kAuthBasic = RTSP_FIELD "authBasic";
|
||||
const string kHandshakeSecond = RTSP_FIELD "handshakeSecond";
|
||||
const string kKeepAliveSecond = RTSP_FIELD "keepAliveSecond";
|
||||
const string kDirectProxy = RTSP_FIELD "directProxy";
|
||||
|
||||
static onceToken token([](){
|
||||
//默认Md5方式认证
|
||||
static onceToken token([]() {
|
||||
// 默认Md5方式认证
|
||||
mINI::Instance()[kAuthBasic] = 0;
|
||||
mINI::Instance()[kHandshakeSecond] = 15;
|
||||
mINI::Instance()[kKeepAliveSecond] = 15;
|
||||
mINI::Instance()[kDirectProxy] = 1;
|
||||
});
|
||||
} //namespace Rtsp
|
||||
} // namespace Rtsp
|
||||
|
||||
////////////RTMP服务器配置///////////
|
||||
namespace Rtmp {
|
||||
#define RTMP_FIELD "rtmp."
|
||||
const string kModifyStamp = RTMP_FIELD"modifyStamp";
|
||||
const string kHandshakeSecond = RTMP_FIELD"handshakeSecond";
|
||||
const string kKeepAliveSecond = RTMP_FIELD"keepAliveSecond";
|
||||
const string kModifyStamp = RTMP_FIELD "modifyStamp";
|
||||
const string kHandshakeSecond = RTMP_FIELD "handshakeSecond";
|
||||
const string kKeepAliveSecond = RTMP_FIELD "keepAliveSecond";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kModifyStamp] = false;
|
||||
mINI::Instance()[kHandshakeSecond] = 15;
|
||||
mINI::Instance()[kKeepAliveSecond] = 15;
|
||||
});
|
||||
} //namespace RTMP
|
||||
|
||||
} // namespace Rtmp
|
||||
|
||||
////////////RTP配置///////////
|
||||
namespace Rtp {
|
||||
#define RTP_FIELD "rtp."
|
||||
//RTP打包最大MTU,公网情况下更小
|
||||
const string kVideoMtuSize = RTP_FIELD"videoMtuSize";
|
||||
const string kAudioMtuSize = RTP_FIELD"audioMtuSize";
|
||||
//rtp包最大长度限制，单位是KB
|
||||
const string kRtpMaxSize = RTP_FIELD"rtpMaxSize";
|
||||
// RTP打包最大MTU,公网情况下更小
|
||||
const string kVideoMtuSize = RTP_FIELD "videoMtuSize";
|
||||
const string kAudioMtuSize = RTP_FIELD "audioMtuSize";
|
||||
// rtp包最大长度限制，单位是KB
|
||||
const string kRtpMaxSize = RTP_FIELD "rtpMaxSize";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kVideoMtuSize] = 1400;
|
||||
mINI::Instance()[kAudioMtuSize] = 600;
|
||||
mINI::Instance()[kRtpMaxSize] = 10;
|
||||
});
|
||||
} //namespace Rtsp
|
||||
} // namespace Rtp
|
||||
|
||||
////////////组播配置///////////
|
||||
namespace MultiCast {
|
||||
#define MULTI_FIELD "multicast."
|
||||
//组播分配起始地址
|
||||
const string kAddrMin = MULTI_FIELD"addrMin";
|
||||
//组播分配截止地址
|
||||
const string kAddrMax = MULTI_FIELD"addrMax";
|
||||
//组播TTL
|
||||
const string kUdpTTL = MULTI_FIELD"udpTTL";
|
||||
// 组播分配起始地址
|
||||
const string kAddrMin = MULTI_FIELD "addrMin";
|
||||
// 组播分配截止地址
|
||||
const string kAddrMax = MULTI_FIELD "addrMax";
|
||||
// 组播TTL
|
||||
const string kUdpTTL = MULTI_FIELD "udpTTL";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kAddrMin] = "239.0.0.0";
|
||||
mINI::Instance()[kAddrMax] = "239.255.255.255";
|
||||
mINI::Instance()[kUdpTTL] = 64;
|
||||
});
|
||||
} //namespace MultiCast
|
||||
} // namespace MultiCast
|
||||
|
||||
////////////录像配置///////////
|
||||
namespace Record {
|
||||
#define RECORD_FIELD "record."
|
||||
const string kAppName = RECORD_FIELD"appName";
|
||||
const string kSampleMS = RECORD_FIELD"sampleMS";
|
||||
const string kFileSecond = RECORD_FIELD"fileSecond";
|
||||
const string kFilePath = RECORD_FIELD"filePath";
|
||||
const string kFileBufSize = RECORD_FIELD"fileBufSize";
|
||||
const string kFastStart = RECORD_FIELD"fastStart";
|
||||
const string kFileRepeat = RECORD_FIELD"fileRepeat";
|
||||
const string kAppName = RECORD_FIELD "appName";
|
||||
const string kSampleMS = RECORD_FIELD "sampleMS";
|
||||
const string kFileSecond = RECORD_FIELD "fileSecond";
|
||||
const string kFilePath = RECORD_FIELD "filePath";
|
||||
const string kFileBufSize = RECORD_FIELD "fileBufSize";
|
||||
const string kFastStart = RECORD_FIELD "fastStart";
|
||||
const string kFileRepeat = RECORD_FIELD "fileRepeat";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kAppName] = "record";
|
||||
mINI::Instance()[kSampleMS] = 500;
|
||||
mINI::Instance()[kFileSecond] = 60*60;
|
||||
mINI::Instance()[kFileSecond] = 60 * 60;
|
||||
mINI::Instance()[kFilePath] = "./www";
|
||||
mINI::Instance()[kFileBufSize] = 64 * 1024;
|
||||
mINI::Instance()[kFastStart] = false;
|
||||
mINI::Instance()[kFileRepeat] = false;
|
||||
});
|
||||
} //namespace Record
|
||||
} // namespace Record
|
||||
|
||||
////////////HLS相关配置///////////
|
||||
namespace Hls {
|
||||
#define HLS_FIELD "hls."
|
||||
const string kSegmentDuration = HLS_FIELD"segDur";
|
||||
const string kSegmentNum = HLS_FIELD"segNum";
|
||||
const string kSegmentKeep = HLS_FIELD"segKeep";
|
||||
const string kSegmentRetain = HLS_FIELD"segRetain";
|
||||
const string kFileBufSize = HLS_FIELD"fileBufSize";
|
||||
const string kFilePath = HLS_FIELD"filePath";
|
||||
const string kBroadcastRecordTs = HLS_FIELD"broadcastRecordTs";
|
||||
const string kDeleteDelaySec = HLS_FIELD"deleteDelaySec";
|
||||
const string kSegmentDuration = HLS_FIELD "segDur";
|
||||
const string kSegmentNum = HLS_FIELD "segNum";
|
||||
const string kSegmentKeep = HLS_FIELD "segKeep";
|
||||
const string kSegmentRetain = HLS_FIELD "segRetain";
|
||||
const string kFileBufSize = HLS_FIELD "fileBufSize";
|
||||
const string kFilePath = HLS_FIELD "filePath";
|
||||
const string kBroadcastRecordTs = HLS_FIELD "broadcastRecordTs";
|
||||
const string kDeleteDelaySec = HLS_FIELD "deleteDelaySec";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kSegmentDuration] = 2;
|
||||
mINI::Instance()[kSegmentNum] = 3;
|
||||
mINI::Instance()[kSegmentKeep] = false;
|
||||
@@ -273,28 +270,40 @@ static onceToken token([](){
|
||||
mINI::Instance()[kBroadcastRecordTs] = false;
|
||||
mINI::Instance()[kDeleteDelaySec] = 0;
|
||||
});
|
||||
} //namespace Hls
|
||||
|
||||
} // namespace Hls
|
||||
|
||||
////////////Rtp代理相关配置///////////
|
||||
namespace RtpProxy {
|
||||
#define RTP_PROXY_FIELD "rtp_proxy."
|
||||
const string kDumpDir = RTP_PROXY_FIELD"dumpDir";
|
||||
const string kTimeoutSec = RTP_PROXY_FIELD"timeoutSec";
|
||||
const string kDumpDir = RTP_PROXY_FIELD "dumpDir";
|
||||
const string kTimeoutSec = RTP_PROXY_FIELD "timeoutSec";
|
||||
const string kPortRange = RTP_PROXY_FIELD "port_range";
|
||||
const string KH264PT = RTP_PROXY_FIELD "h264_pt";
|
||||
const string KH265PT = RTP_PROXY_FIELD "h265_pt";
|
||||
const string KPSPT = RTP_PROXY_FIELD "ps_pt";
|
||||
const string KTSPT = RTP_PROXY_FIELD "ts_pt";
|
||||
const string KOpusPT = RTP_PROXY_FIELD "opus_pt";
|
||||
const string KG711UPT = RTP_PROXY_FIELD "g711u_pt";
|
||||
const string KG711APT = RTP_PROXY_FIELD "g711a_pt";
|
||||
|
||||
static onceToken token([](){
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kDumpDir] = "";
|
||||
mINI::Instance()[kTimeoutSec] = 15;
|
||||
mINI::Instance()[kPortRange] = "30000-35000";
|
||||
mINI::Instance()[KH264PT] = 98;
|
||||
mINI::Instance()[KH265PT] = 99;
|
||||
mINI::Instance()[KPSPT] = 96;
|
||||
mINI::Instance()[KTSPT] = 33;
|
||||
mINI::Instance()[KOpusPT] = 100;
|
||||
mINI::Instance()[KG711UPT] = 0;
|
||||
mINI::Instance()[KG711APT] = 8;
|
||||
});
|
||||
} //namespace RtpProxy
|
||||
|
||||
} // namespace RtpProxy
|
||||
|
||||
namespace Client {
|
||||
const string kNetAdapter = "net_adapter";
|
||||
const string kRtpType = "rtp_type";
|
||||
const string kRtspUser = "rtsp_user" ;
|
||||
const string kRtspUser = "rtsp_user";
|
||||
const string kRtspPwd = "rtsp_pwd";
|
||||
const string kRtspPwdIsMD5 = "rtsp_pwd_md5";
|
||||
const string kTimeoutMS = "protocol_timeout_ms";
|
||||
@@ -302,20 +311,20 @@ const string kMediaTimeoutMS = "media_timeout_ms";
|
||||
const string kBeatIntervalMS = "beat_interval_ms";
|
||||
const string kBenchmarkMode = "benchmark_mode";
|
||||
const string kWaitTrackReady = "wait_track_ready";
|
||||
}
|
||||
} // namespace Client
|
||||
|
||||
} // namespace mediakit
|
||||
} // namespace mediakit
|
||||
|
||||
#ifdef ENABLE_MEM_DEBUG
|
||||
|
||||
extern "C" {
|
||||
extern void *__real_malloc(size_t);
|
||||
extern void __real_free(void *);
|
||||
extern void *__real_realloc(void *ptr, size_t c);
|
||||
void *__wrap_malloc(size_t c);
|
||||
void __wrap_free(void *ptr);
|
||||
void *__wrap_calloc(size_t __nmemb, size_t __size);
|
||||
void *__wrap_realloc(void *ptr, size_t c);
|
||||
extern void *__real_malloc(size_t);
|
||||
extern void __real_free(void *);
|
||||
extern void *__real_realloc(void *ptr, size_t c);
|
||||
void *__wrap_malloc(size_t c);
|
||||
void __wrap_free(void *ptr);
|
||||
void *__wrap_calloc(size_t __nmemb, size_t __size);
|
||||
void *__wrap_realloc(void *ptr, size_t c);
|
||||
}
|
||||
|
||||
#define BLOCK_TYPES 16
|
||||
@@ -344,8 +353,8 @@ std::vector<size_t> getBlockTypeSize() {
|
||||
class MemThreadInfo {
|
||||
public:
|
||||
using Ptr = std::shared_ptr<MemThreadInfo>;
|
||||
atomic<uint64_t> mem_usage{0};
|
||||
atomic<uint64_t> mem_block{0};
|
||||
atomic<uint64_t> mem_usage { 0 };
|
||||
atomic<uint64_t> mem_block { 0 };
|
||||
atomic<uint64_t> mem_block_map[BLOCK_TYPES];
|
||||
|
||||
static MemThreadInfo *Instance(bool is_thread_local) {
|
||||
@@ -358,25 +367,21 @@ public:
|
||||
}
|
||||
|
||||
~MemThreadInfo() {
|
||||
//printf("%s %d\r\n", __FUNCTION__, (int) _is_thread_local);
|
||||
// printf("%s %d\r\n", __FUNCTION__, (int) _is_thread_local);
|
||||
}
|
||||
|
||||
MemThreadInfo(bool is_thread_local) {
|
||||
_is_thread_local = is_thread_local;
|
||||
if (_is_thread_local) {
|
||||
//确保所有线程退出后才能释放全局内存统计器
|
||||
// 确保所有线程退出后才能释放全局内存统计器
|
||||
total_mem = Instance(false);
|
||||
}
|
||||
//printf("%s %d\r\n", __FUNCTION__, (int) _is_thread_local);
|
||||
// printf("%s %d\r\n", __FUNCTION__, (int) _is_thread_local);
|
||||
}
|
||||
|
||||
void *operator new(size_t sz) {
|
||||
return __real_malloc(sz);
|
||||
}
|
||||
void *operator new(size_t sz) { return __real_malloc(sz); }
|
||||
|
||||
void operator delete(void *ptr) {
|
||||
__real_free(ptr);
|
||||
}
|
||||
void operator delete(void *ptr) { __real_free(ptr); }
|
||||
|
||||
void addBlock(size_t c) {
|
||||
if (total_mem) {
|
||||
@@ -410,19 +415,15 @@ public:
|
||||
ptr->addBlock(1);
|
||||
}
|
||||
|
||||
~MemThreadInfoLocal() {
|
||||
ptr->delBlock(1);
|
||||
}
|
||||
~MemThreadInfoLocal() { ptr->delBlock(1); }
|
||||
|
||||
MemThreadInfo *get() const {
|
||||
return ptr;
|
||||
}
|
||||
MemThreadInfo *get() const { return ptr; }
|
||||
|
||||
private:
|
||||
MemThreadInfo *ptr;
|
||||
};
|
||||
|
||||
//该变量主要确保线程退出后才能释放MemThreadInfo变量
|
||||
// 该变量主要确保线程退出后才能释放MemThreadInfo变量
|
||||
static thread_local MemThreadInfoLocal s_thread_mem_info;
|
||||
|
||||
uint64_t getTotalMemUsage() {
|
||||
@@ -456,7 +457,7 @@ public:
|
||||
static constexpr uint32_t kMagic = 0xFEFDFCFB;
|
||||
uint32_t magic;
|
||||
uint32_t size;
|
||||
MemThreadInfo* alloc_info;
|
||||
MemThreadInfo *alloc_info;
|
||||
char ptr;
|
||||
};
|
||||
|
||||
@@ -465,10 +466,10 @@ public:
|
||||
#if (defined(__linux__) && !defined(ANDROID)) || defined(__MACH__)
|
||||
#define MAX_STACK_FRAMES 128
|
||||
#define MEM_WARING
|
||||
#include <execinfo.h>
|
||||
#include <limits.h>
|
||||
#include <sys/resource.h>
|
||||
#include <sys/wait.h>
|
||||
#include <execinfo.h>
|
||||
|
||||
static void print_mem_waring(size_t c) {
|
||||
void *array[MAX_STACK_FRAMES];
|
||||
@@ -503,7 +504,7 @@ static void un_init_cookie(MemCookie *cookie) {
|
||||
|
||||
void *__wrap_malloc(size_t c) {
|
||||
c += MEM_OFFSET;
|
||||
auto cookie = (MemCookie *) __real_malloc(c);
|
||||
auto cookie = (MemCookie *)__real_malloc(c);
|
||||
if (cookie) {
|
||||
init_cookie(cookie, c);
|
||||
return &cookie->ptr;
|
||||
@@ -515,7 +516,7 @@ void __wrap_free(void *ptr) {
|
||||
if (!ptr) {
|
||||
return;
|
||||
}
|
||||
auto cookie = (MemCookie *) ((char *) ptr - MEM_OFFSET);
|
||||
auto cookie = (MemCookie *)((char *)ptr - MEM_OFFSET);
|
||||
if (cookie->magic != MemCookie::kMagic) {
|
||||
__real_free(ptr);
|
||||
return;
|
||||
@@ -538,14 +539,14 @@ void *__wrap_realloc(void *ptr, size_t c) {
|
||||
return malloc(c);
|
||||
}
|
||||
|
||||
auto cookie = (MemCookie *) ((char *) ptr - MEM_OFFSET);
|
||||
auto cookie = (MemCookie *)((char *)ptr - MEM_OFFSET);
|
||||
if (cookie->magic != MemCookie::kMagic) {
|
||||
return __real_realloc(ptr, c);
|
||||
}
|
||||
|
||||
un_init_cookie(cookie);
|
||||
c += MEM_OFFSET;
|
||||
cookie = (MemCookie *) __real_realloc(cookie, c);
|
||||
cookie = (MemCookie *)__real_realloc(cookie, c);
|
||||
if (cookie) {
|
||||
init_cookie(cookie, c);
|
||||
return &cookie->ptr;
|
||||
|
||||
Reference in New Issue
Block a user