12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790 |
- #ifndef NOMINMAX
- #define NOMINMAX
- #endif
- #include "BeefPerf.h"
- #include "platform/PlatformHelper.h"
- #ifndef BP_DISABLED
- #ifndef BF_PLATFORM_WINDOWS
- #include <netinet/in.h>
- #include <netinet/ip.h>
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <netdb.h>
- typedef fd_set FD_SET;
- #define closesocket close
- #endif
- #ifdef BF_PLATFORM_MACOS
- #include <sys/socket.h>
- #include <mach/error.h>
- #include <mach/mach.h>
- #endif
- #ifdef BF_PLATFORM_POSIX
- #include <cerrno>
- #endif
- #pragma comment(lib,"wsock32.lib")
- #pragma warning(disable:4996)
- USING_NS_BF;
- // All DLLs must use the same ABI version
- #define BP_ABI_VERSION 2
- BpManager* BpManager::sBpManager = NULL;
- static bool gOwnsBpManager = false;
- static BpThreadInfo sFakeThreadInfo;
- struct BpManagerOwner
- {
- BpManager* mOwnedManager;
- bool mDidShutdown;
- BpManagerOwner()
- {
- mOwnedManager = NULL;
- mDidShutdown = false;
- }
- ~BpManagerOwner()
- {
- mDidShutdown = true;
- delete mOwnedManager;
- }
- };
- static BpManagerOwner gBpManagerOwner;
- BF_TLS_DECLSPEC BpThreadInfo* Beefy::BpManager::sBpThreadInfo;
- inline void EncodeSLEB128(uint8*& buf, int value)
- {
- bool hasMore;
- do
- {
- uint8 curByte = (uint8)(value & 0x7f);
- value >>= 7;
- hasMore = !((((value == 0) && ((curByte & 0x40) == 0)) ||
- ((value == -1) && ((curByte & 0x40) != 0))));
- if (hasMore)
- curByte |= 0x80;
- *(buf++) = curByte;
- }
- while (hasMore);
- }
- inline void EncodeSLEB128(uint8*& buf, int64_t value)
- {
- bool hasMore;
- do
- {
- uint8 curByte = (uint8)(value & 0x7f);
- value >>= 7;
- hasMore = !((((value == 0) && ((curByte & 0x40) == 0)) ||
- ((value == -1) && ((curByte & 0x40) != 0))));
- if (hasMore)
- curByte |= 0x80;
- *(buf++) = curByte;
- }
- while (hasMore);
- }
- inline void EncodeULEB32(uint8*& buf, uint64 value)
- {
- do
- {
- uint8 byteVal = value & 0x1f;
- value >>= 5;
- if (value != 0)
- byteVal |= 0x20; // Mark this byte to show that more bytes will follow
- *(buf++) = byteVal;
- }
- while (value != 0);
- }
- //////////////////////////////////////////////////////////////////////////
- Buffer::Buffer()
- {
- mPtr = NULL;
- mBufSize = 0;
- mDataSize = 0;
- }
- Buffer::~Buffer()
- {
- delete mPtr;
- }
- uint8* Buffer::Alloc(int size)
- {
- if (mDataSize + size > mBufSize)
- {
- // Grow factor 50%
- int wantSize = max(mDataSize + size, mBufSize + mBufSize / 2);
- auto bpManager = BpManager::Get();
- //uint8* newPtr = new uint8[wantSize];
- uint8* newPtr = (uint8*)bpManager->AllocBytes(wantSize);
- memcpy(newPtr, mPtr, mDataSize);
- //delete mPtr;
- bpManager->FreeBytes(mPtr);
- mPtr = newPtr;
- mBufSize = wantSize;
- }
- uint8* ptr = mPtr + mDataSize;
- mDataSize += size;
- return ptr;
- }
- void Buffer::Free()
- {
- //delete mPtr;
- BpManager::Get()->FreeBytes(mPtr);
- mPtr = NULL;
- mDataSize = 0;
- mBufSize = 0;
- }
- void Buffer::Clear()
- {
- mDataSize = 0;
- }
- //////////////////////////////////////////////////////////////////////////
- CircularBuffer::View::View()
- {
- mPtr = NULL;
- mCircularBuffer = NULL;
- mTempBuf = NULL;
- mTempBufSize = 0;
- mSrcIdx = 0;
- mSrcSize = 0;
- }
- CircularBuffer::View::~View()
- {
- delete mTempBuf;
- }
- void CircularBuffer::View::Commit(int size)
- {
- if (mPtr == mTempBuf)
- {
- if (size == -1)
- size = mSrcSize;
- else
- BF_ASSERT(size <= mSrcSize);
- mCircularBuffer->Write(mTempBuf, mSrcIdx, size);
- }
- else if (size != -1)
- BF_ASSERT(size <= mSrcSize);
- }
- CircularBuffer::CircularBuffer()
- {
- mTail = 0;
- mBufSize = 0;
- mDataSize = 0;
- mBuffer = NULL;
- }
- CircularBuffer::~CircularBuffer()
- {
- delete mBuffer;
- }
- void CircularBuffer::Clear()
- {
- mTail = 0;
- mDataSize = 0;
- }
- void CircularBuffer::Resize(int newSize)
- {
- uint8* newBuffer = new uint8[newSize];
- Read(newBuffer, 0, mDataSize);
- delete mBuffer;
- mBuffer = newBuffer;
- mBufSize = newSize;
- mTail = 0;
- }
- void CircularBuffer::GrowReserve(int addSize)
- {
- if (mDataSize + addSize <= mBufSize)
- return;
- Resize(max(mDataSize + addSize, mDataSize + mDataSize/2));
- }
- void CircularBuffer::Grow(int addSize)
- {
- GrowReserve(addSize);
- mDataSize += addSize;
- }
- void CircularBuffer::GrowFront(int addSize)
- {
- if (mDataSize + addSize > mBufSize)
- {
- Resize(mDataSize + addSize);
- }
- mDataSize += addSize;
- mTail = (mTail + mBufSize - addSize) % mBufSize;
- }
- int CircularBuffer::GetSize()
- {
- return mDataSize;
- }
- void CircularBuffer::MapView(int idx, int len, CircularBuffer::View& view)
- {
- view.mCircularBuffer = this;
- view.mSrcIdx = idx;
- view.mSrcSize = len;
- if (mTail + idx + len <= mBufSize)
- {
- view.mPtr = mBuffer + mTail + idx;
- }
- else
- {
- if (view.mTempBufSize < len)
- {
- delete view.mTempBuf;
- view.mTempBuf = new uint8[len];
- view.mTempBufSize = len;
- }
- view.mPtr = view.mTempBuf;
- Read(view.mTempBuf, idx, len);
- }
- }
- void CircularBuffer::Read(void* ptr, int idx, int len)
- {
- BF_ASSERT(len <= mBufSize);
- if (len == 0)
- return;
- int absIdx = (mTail + idx) % mBufSize;
- if (absIdx + len > mBufSize)
- {
- int lowSize = mBufSize - absIdx;
- memcpy(ptr, mBuffer + absIdx, lowSize);
- memcpy((uint8*)ptr + lowSize, mBuffer, len - lowSize);
- }
- else
- {
- memcpy(ptr, mBuffer + absIdx, len);
- }
- }
- void CircularBuffer::Write(void* ptr, int idx, int len)
- {
- BF_ASSERT(len <= mBufSize);
- if (len == 0)
- return;
- int absIdx = (mTail + idx) % mBufSize;
- if (absIdx + len > mBufSize)
- {
- int lowSize = mBufSize - absIdx;
- memcpy(mBuffer + absIdx, ptr, lowSize);
- memcpy(mBuffer, (uint8*)ptr + lowSize, len - lowSize);
- }
- else
- {
- memcpy(mBuffer + absIdx, ptr, len);
- }
- }
- void CircularBuffer::RemoveFront(int len)
- {
- mTail = (mTail + len) % mBufSize;
- mDataSize -= len;
- }
- //////////////////////////////////////////////////////////////////////////
- BpCmdTarget::BpCmdTarget()
- {
- mCurDynStrIdx = 0;
- mCurDepth = 0;
- mThreadName = NULL;
- }
- void BpCmdTarget::Disable()
- {
- AutoCrit autoCrit(mCritSect);
- mOutBuffer.Free();
- }
- const char* BpCmdTarget::DynamicString(const char* str)
- {
- int usedIdx = mCurDynStrIdx;
- mDynStrs[usedIdx] = str;
- mCurDynStrIdx = (mCurDynStrIdx + 1) % BF_ARRAY_COUNT(mDynStrs);
- return (const char*)(intptr)usedIdx;
- }
- const char* BpCmdTarget::ToStrPtr(const char* str)
- {
- if ((intptr)str < BF_ARRAY_COUNT(mDynStrs))
- return mDynStrs[(intptr)str];
- return str;
- }
- #define BPCMD_PREPARE if ((gBpManagerOwner.mDidShutdown) || (!BpManager::Get()->mCollectData)) return; AutoCrit autoCrit(mCritSect)
- #define GET_FROM(ptr, T) *((T*)(ptr += sizeof(T)) - 1)
- //#define BPCMD_RESERVE(addSize) mOutBuffer.resize(mOutBuffer.size() + (addSize)); uint8* data = &mOutBuffer[mOutBuffer.size() - (addSize)];
- #define BPCMD_RESERVE(addSize) uint8* data = mOutBuffer.Alloc(addSize)
- #define BPCMD_RESERVE_UNDECL(addSize) data = mOutBuffer.Alloc(addSize)
- #define BPCMD_MEMBER(T) *((T*)(data += sizeof(T)) - 1)
- #define BPCMD_MEMCPY(ptr, size) memcpy(data, ptr, size); data += size
- #define BPCMD_END() BF_ASSERT(data == mOutBuffer.mPtr + mOutBuffer.mDataSize)
- static int64 GetTimestamp()
- {
- #ifdef BF_PLATFORM_WINDOWS
- return __rdtsc() / 100;
- #else
- return BfpSystem_GetCPUTick();
- #endif
- }
- #define MAX_DEPTH 8192
- void BpCmdTarget::Enter(const char* name)
- {
- BPCMD_PREPARE;
- // Failure here could be from unbalanced enter/leave calls
- BF_ASSERT((uint32)mCurDepth <= MAX_DEPTH);
- if ((intptr)name < BF_ARRAY_COUNT(mDynStrs))
- {
- const char* dynStr = mDynStrs[(intptr)name];
- int len = (int)strlen(dynStr);
- BPCMD_RESERVE(1 + 8 + len + 1);
- BPCMD_MEMBER(uint8) = BpCmd_EnterDyn;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMCPY(dynStr, len + 1);
- BPCMD_END();
- }
- else
- {
- BPCMD_RESERVE(1 + 8 + sizeof(const char*));
- BPCMD_MEMBER(uint8) = BpCmd_Enter;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMBER(const char*) = name;
- BPCMD_END();
- }
- mCurDepth++;
- }
- void BpCmdTarget::Enter(const char* name, va_list args)
- {
- BPCMD_PREPARE;
- // Failure here could be from unbalanced enter/leave calls
- BF_ASSERT((uint32)mCurDepth <= MAX_DEPTH);
- int len = 0;
- int paramSize = 0;
- //va_list origArgs = args;
- va_list origArgs;
- va_copy(origArgs, args);
- //va_start(args, name);
- const char* cPtr = ToStrPtr(name);
- while (true)
- {
- char c = *(cPtr++);
- if (c == 0)
- break;
- len++;
- if (c == '%')
- {
- len++;
- char nextC = *(cPtr++);
- if (nextC != '%')
- {
- if (nextC == 'f')
- {
- va_arg(args, double);
- paramSize += 4; // float
- }
- else if (nextC == 'd')
- {
- intptr val = va_arg(args, intptr);
- paramSize += 4; // int32
- }
- else if (nextC == 's')
- {
- const char* str = ToStrPtr(va_arg(args, char*));
- paramSize += (int)strlen(str) + 1;
- }
- else
- {
- BF_FATAL("Invalid format flag");
- }
- }
- }
- }
- //va_end(args);
- uint8* data;
- if ((intptr)name < BF_ARRAY_COUNT(mDynStrs))
- {
- const char* dynStr = mDynStrs[(intptr)name];
- int len = (int)strlen(dynStr);
- BPCMD_RESERVE_UNDECL(1 + 8 + len + 1 + paramSize);
- BPCMD_MEMBER(uint8) = BpCmd_EnterDyn;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMCPY(dynStr, len + 1);
- }
- else
- {
- BPCMD_RESERVE_UNDECL(1 + 8 + sizeof(const char*) + paramSize);
- BPCMD_MEMBER(uint8) = BpCmd_Enter;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMBER(const char*) = name;
- }
- /*BPCMD_RESERVE(1 + 8 + len + 1 + paramSize);
- BPCMD_MEMBER(uint8) = BpCmd_Enter;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMCPY(name, len + 1);*/
- args = origArgs;
- //va_start(args, name);
- cPtr = ToStrPtr(name);
- while (true)
- {
- char c = *(cPtr++);
- if (c == 0)
- break;
- if (c == '%')
- {
- char nextC = *(cPtr++);
- if (nextC != '%')
- {
- if (nextC == 'f')
- {
- BPCMD_MEMBER(float) = (float)va_arg(args, double);
- }
- else if (nextC == 'd')
- {
- BPCMD_MEMBER(int32) = (int32)va_arg(args, intptr);
- }
- else if (nextC == 's')
- {
- const char* str = ToStrPtr(va_arg(args, char*));
- BPCMD_MEMCPY(str, (int)strlen(str) + 1);
- }
- else
- {
- BF_FATAL("Invalid format flag");
- }
- }
- }
- }
- va_end(args);
- BPCMD_END();
- mCurDepth++;
- }
- void BpCmdTarget::EnterF(const char* name, ...)
- {
- va_list args;
- va_start(args, name);
- Enter(name, args);
- va_end(args);
- }
- void BpCmdTarget::Leave()
- {
- BPCMD_PREPARE;
- if (mCurDepth <= 0)
- {
- // This is either due to improperly balanced Enter/Leaves or from a Reconnnect
- BF_ASSERT(BpManager::Get()->mInitCount > 1);
- return;
- }
- BPCMD_RESERVE(1 + 8);
- BPCMD_MEMBER(uint8) = BpCmd_Leave;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_END();
- mCurDepth--;
- }
- void BpCmdTarget::Event(const char* name, const char* details)
- {
- name = ToStrPtr(name);
- details = ToStrPtr(details);
- BPCMD_PREPARE;
- int nameLen = (int)strlen(name);
- int detailsLen = (int)strlen(details);
- BPCMD_RESERVE(1 + 8 + nameLen+1 + detailsLen+1);
- BPCMD_MEMBER(uint8) = BpCmd_Event;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMCPY(name, nameLen + 1);
- BPCMD_MEMCPY(details, detailsLen + 1);
- BPCMD_END();
- }
- void BpRootCmdTarget::Init()
- {
- BPCMD_PREPARE;
- BPCMD_RESERVE(1);
- BPCMD_MEMBER(uint8) = BpCmd_Init;
- BPCMD_END();
- }
- void BpRootCmdTarget::Tick()
- {
- BPCMD_PREPARE;
- BPCMD_RESERVE(1 + 8);
- BPCMD_MEMBER(uint8) = BpCmd_Tick;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_END();
- }
- void BpRootCmdTarget::KeepAlive()
- {
- BPCMD_PREPARE;
- BPCMD_RESERVE(1 + 8);
- BPCMD_MEMBER(uint8) = BpCmd_KeepAlive;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_END();
- }
- void BpRootCmdTarget::AddThread(int threadId, BfpThreadId nativeThreadId)
- {
- BPCMD_PREPARE;
- BPCMD_RESERVE(1 + 8 + 4 + 4);
- BPCMD_MEMBER(uint8) = BpCmd_ThreadAdd;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_MEMBER(int32) = threadId;
- BPCMD_MEMBER(int32) = (int32)nativeThreadId;
- BPCMD_END();
- }
- BpThreadInfo::BpThreadInfo()
- {
- mNativeThreadId = -1;
- mThreadId = 0;
- mReadyToSend = false;
- mHasTerminated = false;
- }
- BpThreadInfo::~BpThreadInfo()
- {
- if (mThreadName != NULL)
- BpManager::Get()->FreeBytes(mThreadName);
- }
- void BpThreadInfo::SetThreadName(const char* name)
- {
- int len = (int)strlen(name);
- if (mThreadName == NULL)
- {
- mThreadName = (char*)BpManager::Get()->AllocBytes(len + 1);
- memcpy(mThreadName, name, len + 1);
- }
- BPCMD_PREPARE;
-
- BPCMD_RESERVE(1 + len + 1);
- BPCMD_MEMBER(uint8) = BpCmd_ThreadName;
- BPCMD_MEMCPY(name, len + 1);
- BPCMD_END();
- }
- void BpThreadInfo::RemoveThread()
- {
- BPCMD_PREPARE;
- BPCMD_RESERVE(1 + 8);
- BPCMD_MEMBER(uint8) = BpCmd_ThreadRemove;
- BPCMD_MEMBER(int64) = GetTimestamp();
- BPCMD_END();
- }
- //////////////////////////////////////////////////////////////////////////
- static void NTAPI FlsFreeFunc(void* ptr)
- {
- BpThreadInfo* threadInfo = (BpThreadInfo*)ptr;
- auto bpManager = BpManager::Get();
- AutoCrit autoCrit(bpManager->mCritSect);
- threadInfo->mHasTerminated = true;
- threadInfo->RemoveThread();
- if (!bpManager->mThreadRunning)
- {
- bpManager->mThreadInfos.Remove(threadInfo);
- delete threadInfo;
- // It's possible that other thread-specific destructors will occur after this call, which may use BeefPerf,
- // and in that case we will re-add this thread info
- BpManager::sBpThreadInfo = NULL;
- }
- }
- BpManager::BpManager() : mShutdownEvent(true), mTLSDtor(&FlsFreeFunc)
- {
- #ifdef BF_PLATFORM_WINDOWS
- mMutex = NULL;
- mSharedMemoryFile = NULL;
- #endif
- mSocket = INVALID_SOCKET;
- mConnectState = BpConnectState_NotConnected;
- mThread = NULL;
- mCurTick = 0;
- mCurThreadId = 0;
- mOutBlockSizeLeft = 0;
- mPauseCount = 0;
- mInitCount = 0;
-
- mCollectData = false;
- mThreadRunning = false;
- mInitTimeStamp = GetTimestamp();
- mInitTickCount = BFTickCount();
- }
- BpManager::~BpManager()
- {
- Shutdown();
-
- #ifdef BF_PLATFORM_WINDOWS
- if (mMutex != NULL)
- ::CloseHandle(mMutex);
- if (mSharedMemoryFile != NULL)
- ::CloseHandle(mSharedMemoryFile);
- #endif
- }
- bool BpManager::Connect()
- {
- struct sockaddr_in server;
- struct hostent * hp;
- server.sin_family = PF_INET;
- hp = gethostbyname(mServerName.c_str());
- if (hp == NULL)
- return false;
- memcpy(&server.sin_addr, hp->h_addr_list[0], sizeof(server.sin_addr));
- server.sin_port = htons(4208);
- #ifdef BF_PLATFORM_WINDOWS
- bool isLocalhost = server.sin_addr.S_un.S_addr == 0x0100007f;
- #else
- bool isLocalhost = server.sin_addr.s_addr == 0x0100007f;
- #endif
- int result = ::connect(mSocket, (sockaddr*)&server, sizeof(server));
- if (result != 0)
- {
- #ifdef BF_PLATFORM_WINDOWS
- int err = WSAGetLastError();
- if (err != WSAEWOULDBLOCK)
- return false;
- #else
- if (errno != EINPROGRESS)
- return false;
- #endif
- }
-
- int totalWaitedMS = 0;
- // Wait for connection - normally we wait for either the connection to occur or an error to occur,
- // but if we shutdown the app then we need to ensure we waited "long enough", but we don't want
- // to keep very short programs from exiting in a timely manner so we have a short localhost timeout
- // since Windows will delay for 2s before failing
- while (true)
- {
- int selectTimeoutMS = 20;
- timeval timeout;
- timeout.tv_sec = 0;
- timeout.tv_usec = selectTimeoutMS * 1000;
- FD_SET socketWriteSet;
- FD_ZERO(&socketWriteSet);
- FD_SET(mSocket, &socketWriteSet);
- FD_SET socketErrorSet;
- FD_ZERO(&socketErrorSet);
- FD_SET(mSocket, &socketErrorSet);
- int result = select((int)mSocket + 1, NULL, &socketWriteSet, &socketErrorSet, &timeout);
- if (result == -1)
- return false;
- if (FD_ISSET(mSocket, &socketWriteSet))
- break;
- if (FD_ISSET(mSocket, &socketErrorSet))
- return false;
- totalWaitedMS += selectTimeoutMS;
- if (mShutdownEvent.WaitFor(0))
- {
- // We are shutting down - have we waited enough?
- int minWaitMS = isLocalhost ? 50 : 20*1000;
- if (totalWaitedMS >= minWaitMS)
- return false;
- }
- }
-
- return true;
- }
- uint8* BpManager::StartCmd(uint8 cmd, CircularBuffer::View& view, int maxLen)
- {
- mOutBuffer.GrowReserve(maxLen);
- mOutBuffer.MapView(mOutBuffer.GetSize(), maxLen, view);
- uint8* dataOut = view.mPtr;
- GET_FROM(dataOut, uint8) = cmd;
- return dataOut;
- }
- void BpManager::EndCmd(CircularBuffer::View& view, uint8* ptr)
- {
- int actualSize = (int)(ptr - view.mPtr);
- view.Commit(actualSize);
- mOutBuffer.Grow(actualSize);
- }
- void BpManager::TrySendData()
- {
- CircularBuffer::View outView;
- while (true)
- {
- int sizeLeft = mOutBuffer.GetSize();
- if (sizeLeft == 0)
- return;
- int trySend = std::min(sizeLeft, 8192);
- mOutBuffer.MapView(0, trySend, outView);
- int result = send(mSocket, (const char*)outView.mPtr, trySend, 0);
-
- if (result < 0)
- {
- #ifdef BF_PLATFORM_WINDOWS
- int err = WSAGetLastError();
- switch (err)
- {
- case WSAECONNABORTED:
- case WSAECONNRESET:
- mConnectState = BpConnectState_NotConnected;
- }
- #else
- switch (errno)
- {
- case ECONNRESET:
- mConnectState = BpConnectState_NotConnected;
- }
- #endif
- return;
- }
- mOutBuffer.RemoveFront(result);
- }
- }
- void BpManager::LostConnection()
- {
- mCollectData = false;
- AutoCrit autoCrit(mCritSect);
- mRootCmdTarget.Disable();
- for (auto threadInfo : mThreadInfos)
- threadInfo->Disable();
- #ifdef BF_PLATFORM_WINDOWS
- closesocket(mSocket);
- #else
- close(mSocket);
- #endif
- mSocket = INVALID_SOCKET;
- }
- BpThreadInfo* BpManager::SlowGetCurThreadInfo()
- {
- BfpThreadId curThreadId = BfpThread_GetCurrentId();
- // Try to find an existing one
- {
- AutoCrit autoCrit(mCritSect);
- for (auto threadInfo : mThreadInfos)
- if (threadInfo->mNativeThreadId == curThreadId)
- return threadInfo;
- }
- auto threadInfo = new BpThreadInfo();
- threadInfo->mThreadId = mCurThreadId++;
- threadInfo->mNativeThreadId = curThreadId;
- sBpThreadInfo = threadInfo;
-
- AutoCrit autoCrit(mCritSect);
- mThreadInfos.push_back(threadInfo);
- if (mThreadRunning)
- mRootCmdTarget.AddThread(threadInfo->mThreadId, threadInfo->mNativeThreadId);
- mTLSDtor.Add((void*)threadInfo);
- return threadInfo;
- }
- void* BpManager::AllocBytes(int size)
- {
- return new uint8[size];
- }
- void BpManager::FreeBytes(void* ptr)
- {
- delete (uint8*)ptr;
- }
- void BpManager::FinishWorkThread()
- {
- AutoCrit autoCrit(mCritSect);
- for (int i = 0; i < (int)mThreadInfos.size(); i++)
- {
- auto threadInfo = mThreadInfos[i];
- if (threadInfo->mHasTerminated)
- {
- delete threadInfo;
- mThreadInfos.erase(mThreadInfos.begin() + i);
- i--;
- }
- }
- mThreadRunning = false;
- }
- void BpManager::ThreadProc()
- {
- if (!Connect())
- {
- mConnectState = BpConnectState_Failed;
- LostConnection();
- FinishWorkThread();
- return;
- }
- Buffer threadBuffer;
- String tempStr;
- CircularBuffer::View outView;
- timeval timeout;
- timeout.tv_sec = 0;
- timeout.tv_usec = 20 * 1000; // 20ms
- uint32 gLastMsgTick = BFTickCount();
- DWORD lastTimeTick = 0;
- while (mConnectState != BpConnectState_NotConnected)
- {
- bool wantsExit = mShutdownEvent.WaitFor(0);
- FD_SET socketReadSet;
- FD_ZERO(&socketReadSet);
- FD_SET(mSocket, &socketReadSet);
- FD_SET socketWriteSet;
- FD_ZERO(&socketWriteSet);
- if (mOutBuffer.GetSize() > 0)
- FD_SET(mSocket, &socketWriteSet);
- FD_SET socketErrorSet;
- FD_ZERO(&socketErrorSet);
- FD_SET(mSocket, &socketErrorSet);
- int selResult = select((int)mSocket + 1, &socketReadSet, &socketWriteSet, &socketErrorSet, &timeout);
- if (FD_ISSET(mSocket, &socketWriteSet))
- {
- TrySendData();
- //continue;
- }
- if (FD_ISSET(mSocket, &socketReadSet))
- {
- // Just eat the data
- uint8 data[4096];
- int len = recv(mSocket, (char*)data, 4096, 0);
- int b = 0;
- }
- if (FD_ISSET(mSocket, &socketErrorSet))
- {
- #ifdef BF_PLATFORM_WINDOWS
- int err = WSAGetLastError();
- #endif
- mConnectState = BpConnectState_NotConnected;
- LostConnection();
- FinishWorkThread();
- return;
- }
- // Alloc space for size
- mOutBuffer.Grow(4);
- int startBufferSize = mOutBuffer.GetSize();
- int threadIdx = -1;
- int threadId = -1;
- //
- {
- // We need to send the BpCmd_ThreadAdd before sending any data from the thread,
- AutoCrit autoCrit(mCritSect);
- for (int threadIdx = 0; threadIdx < (int)mThreadInfos.size(); threadIdx++)
- {
- auto threadInfo = mThreadInfos[threadIdx];
- if ((threadInfo->mHasTerminated) && (threadInfo->mOutBuffer.mDataSize == 0))
- {
- mThreadInfos.erase(mThreadInfos.begin() + threadIdx);
- delete threadInfo;
- threadIdx--;
- }
- else
- {
- threadInfo->mReadyToSend = true;
- }
- }
- }
- while (true)
- {
- BpCmdTarget* cmdTarget = NULL;
- if (threadIdx == -1)
- {
- cmdTarget = &mRootCmdTarget;
- threadIdx++;
- }
- else
- {
- AutoCrit autoCrit(mCritSect);
- if (threadIdx >= (int)mThreadInfos.size())
- break;
- auto threadInfo = mThreadInfos[threadIdx++];
- threadId = threadInfo->mThreadId;
- cmdTarget = threadInfo;
- if (!threadInfo->mReadyToSend)
- continue;
- }
- //
- {
- AutoCrit autoCrit(cmdTarget->mCritSect);
- BF_ASSERT(threadBuffer.mDataSize == 0);
- memcpy(threadBuffer.Alloc(cmdTarget->mOutBuffer.mDataSize), cmdTarget->mOutBuffer.mPtr, cmdTarget->mOutBuffer.mDataSize);
- cmdTarget->mOutBuffer.Clear();
- }
- if (threadBuffer.mDataSize == 0)
- continue;
- uint8* dataOut = StartCmd(BpCmd_SetThread, outView, 5);
- EncodeSLEB128(dataOut, threadId);
- EndCmd(outView, dataOut);
- uint8* dataIn = threadBuffer.mPtr;
- uint8* dataInEnd = dataIn + threadBuffer.mDataSize;
- static int sIdx = 0;
- while (dataIn < dataInEnd)
- {
- BpCmd cmd = (BpCmd)*(dataIn++);
- int nameMsgLen;
- if (cmd == BpCmd_Init)
- {
- AutoCrit autoCrit(mRootCmdTarget.mCritSect);
- String env;
- env += "SessionID\t";
- env += mSessionID;
- env += "\n";
- if (!mSessionName.IsEmpty())
- {
- env += "SessionName\t";
- env += mSessionName;
- env += "\n";
- }
- if (!mClientName.IsEmpty())
- {
- env += "ClientName\t";
- env += mClientName;
- env += "\n";
- }
-
- int32 nameLen = (int32)env.length();
- uint8* dataOut = StartCmd(BpCmd_Init, outView, 1 + 4 + nameLen + 1);
- EncodeSLEB128(dataOut, BP_CLIENT_VERSION);
- memcpy(dataOut, env.c_str(), nameLen + 1);
- dataOut += nameLen + 1;
- EndCmd(outView, dataOut);
- dataOut = StartCmd(BpCmd_ClockInfo, outView, 1 + 8 + 8 + 8 + 1);
- EncodeSLEB128(dataOut, mCurTick);
- EncodeSLEB128(dataOut, GetTimestamp());
- EncodeSLEB128(dataOut, (int)BFTickCount());
- EncodeSLEB128(dataOut, BfpSystem_GetCPUTickFreq());
- EndCmd(outView, dataOut);
- }
- else if ((cmd == BpCmd_Enter) || (cmd == BpCmd_EnterDyn))
- {
- int64 tick = GET_FROM(dataIn, int64);
- const char* name;
- int strIdx;
- int paramsSize = 0;
- if (cmd == BpCmd_EnterDyn)
- {
- name = (const char*)dataIn;
- int nameLen = (int)strlen(name);
- dataIn += nameLen + 1;
- paramsSize = -1;
- strIdx = -nameLen;
- nameMsgLen = 4+1 + nameLen;
- }
- else
- {
- name = GET_FROM(dataIn, const char*);
- BpZoneName* zoneName;
- if (mZoneNameMap.TryAdd(name, NULL, &zoneName))
- {
- int nameLen = (int)strlen(name);
- strIdx = (int)mZoneNameMap.size() - 1;
- zoneName->mIdx = strIdx;
- zoneName->mSize = 0;
- bool isDyn = false;
- const char* cPtr = name;
- while (true)
- {
- char c = *(cPtr++);
- if (c == 0)
- break;
- if (c == '%')
- {
- char nextC = *(cPtr++);
- if (nextC != '%')
- {
- if (nextC == 'f')
- {
- zoneName->mSize += 4;
- }
- else if (nextC == 'd')
- {
- zoneName->mSize += 4;
- }
- else if (nextC == 's')
- {
- isDyn = true;
- }
- }
- }
- }
- if (isDyn)
- zoneName->mSize = -1;
-
- uint8* dataOut = StartCmd(BpCmd_StrEntry, outView, 1 + nameLen + 1);
- memcpy(dataOut, name, nameLen + 1);
- dataOut += nameLen + 1;
- EndCmd(outView, dataOut);
- paramsSize = zoneName->mSize;
- }
- else
- {
- strIdx = zoneName->mIdx;
- paramsSize = zoneName->mSize;
- }
- nameMsgLen = 4+1;
- }
- bool isDynSize = false;
- int addParamSize = 0;
- if (paramsSize == -1)
- {
- isDynSize = true;
- addParamSize = 4 + 1; // For size param
- // Calc size
- auto checkDataIn = dataIn;
- const char* cPtr = name;
- while (true)
- {
- char c = *(cPtr++);
- if (c == 0)
- break;
- if (c == '%')
- {
- char nextC = *(cPtr++);
- if (nextC != '%')
- {
- if (nextC == 'f')
- {
- checkDataIn += 4;
- }
- else if (nextC == 'd')
- {
- checkDataIn += 4;
- }
- else if (nextC == 's')
- {
- int len = (int)strlen((const char*)checkDataIn);
- checkDataIn += len + 1;
- }
- }
- }
- }
- paramsSize = (int)(checkDataIn - dataIn);
- }
- uint8* dataOut = StartCmd(BpCmd_Enter, outView, 1 + 8+1 + nameMsgLen + paramsSize + addParamSize);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- EncodeSLEB128(dataOut, tickDelta);
- if (strIdx < 0)
- {
- EncodeSLEB128(dataOut, strIdx);
- memcpy(dataOut, name, -strIdx);
- dataOut += -strIdx;
- }
- else
- EncodeSLEB128(dataOut, strIdx);
- if (isDynSize)
- EncodeSLEB128(dataOut, paramsSize);
- if (paramsSize != 0)
- {
- memcpy(dataOut, dataIn, paramsSize);
- dataOut += paramsSize;
- dataIn += paramsSize;
- }
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_Leave)
- {
- uint8* dataOut = StartCmd(BpCmd_Leave, outView, 1 + 8+1);
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- EncodeSLEB128(dataOut, tickDelta);
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_ThreadName)
- {
- const char* name = (const char*)dataIn;
- int nameLen = (int)strlen(name);
- uint8* dataOut = StartCmd(BpCmd_ThreadName, outView, 1 + nameLen + 1);
- memcpy(dataOut, dataIn, nameLen + 1);
- dataIn += nameLen + 1;
- dataOut += nameLen + 1;
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_Tick)
- {
- uint8* dataOut = StartCmd(BpCmd_Tick, outView, 1 + 8+1);
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- EncodeSLEB128(dataOut, tickDelta);
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_KeepAlive)
- {
- uint8* dataOut = StartCmd(BpCmd_KeepAlive, outView, 1 + 8+1);
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- EncodeSLEB128(dataOut, tickDelta);
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_ThreadAdd)
- {
- uint8* dataOut = StartCmd(BpCmd_ThreadAdd, outView, 1 + 8+1 + 4+1 + 4+1);
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- int32 threadId = GET_FROM(dataIn, int32);
- int32 nativeThreadId = GET_FROM(dataIn, int32);
- EncodeSLEB128(dataOut, tickDelta);
- EncodeSLEB128(dataOut, threadId);
- EncodeSLEB128(dataOut, nativeThreadId);
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_ThreadRemove)
- {
- uint8* dataOut = StartCmd(BpCmd_ThreadRemove, outView, 1 + 8+1);
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- EncodeSLEB128(dataOut, tickDelta);
- EndCmd(outView, dataOut);
- }
- else if (cmd == BpCmd_Event)
- {
- int64 tick = GET_FROM(dataIn, int64);
- int64 tickDelta = tick - mCurTick;
- mCurTick = tick;
- const char* name = (const char*)dataIn;
- int nameLen = (int)strlen(name);
- const char* details = (const char*)(dataIn + nameLen + 1);
- int detailsLen = (int)strlen(details);
- uint8* dataOut = StartCmd(BpCmd_Event, outView, 1 + 8+1 + nameLen + 1 + detailsLen + 1);
- EncodeSLEB128(dataOut, tickDelta);
- memcpy(dataOut, dataIn, nameLen + 1);
- dataIn += nameLen + 1;
- dataOut += nameLen + 1;
- memcpy(dataOut, dataIn, detailsLen + 1);
- dataIn += detailsLen + 1;
- dataOut += detailsLen + 1;
- EndCmd(outView, dataOut);
- }
- else
- BF_FATAL("Not handled");
-
- // If we get a large backlog then try to break it up into smaller chunks and send out periodically
- int bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
- BF_ASSERT(bufSizeAdded >= 0);
- if (bufSizeAdded >= 64*1024)
- {
- // Set chunk size
- CircularBuffer::View view;
- mOutBuffer.MapView(startBufferSize - 4, 4, view);
- uint8* data = view.mPtr;
- GET_FROM(data, int32) = bufSizeAdded;
- view.Commit();
- // We can attempt some more sending now...
- TrySendData();
- // Start next header
- mOutBuffer.Grow(4);
- startBufferSize = mOutBuffer.GetSize();
- }
- }
- threadBuffer.Clear();
- }
- int bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
- DWORD curTick = BFTickCount();
-
- // Encode clock info
- if ((lastTimeTick == 0) || ((bufSizeAdded > 0) && (curTick - lastTimeTick >= 1000)))
- {
- uint8* dataOut = StartCmd(BpCmd_ClockInfo, outView, 1 + 8 + 8 + 8 + 1);
- EncodeSLEB128(dataOut, mCurTick);
- EncodeSLEB128(dataOut, GetTimestamp());
- EncodeSLEB128(dataOut, (int)BFTickCount());
- EncodeSLEB128(dataOut, BfpSystem_GetCPUTickFreq());
- EndCmd(outView, dataOut);
- lastTimeTick = curTick;
- }
- bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
- //BF_ASSERT((bufSizeAdded <= 64*1024) && (bufSizeAdded >= 0));
- if (bufSizeAdded != 0)
- {
- // Set chunk size
- CircularBuffer::View view;
- mOutBuffer.MapView(startBufferSize - 4, 4, view);
- uint8* data = view.mPtr;
- GET_FROM(data, int32) = bufSizeAdded;
- view.Commit();
- }
- else
- mOutBuffer.Grow(-4); // No data added, pop chunk size off
- if (mOutBuffer.GetSize() == 0)
- {
- if (wantsExit)
- {
- break;
- }
- uint32 tickNow = BFTickCount();
- if ((tickNow - gLastMsgTick >= 1000) /*&& (!wantsExit)*/)
- {
- mRootCmdTarget.KeepAlive();
- }
- }
- else
- {
- gLastMsgTick = BFTickCount();
- }
- }
- mOutBuffer.Clear();
- mConnectState = BpConnectState_NotConnected;
- closesocket(mSocket);
- mSocket = INVALID_SOCKET;
- FinishWorkThread();
- }
- void BFP_CALLTYPE BpManager::ThreadProcThunk(void* ptr)
- {
- ((BpManager*)ptr)->ThreadProc();
- }
- void BpManager::Clear()
- {
- int threadIdx = 0;
- while (true)
- {
- BpThreadInfo* threadInfo = NULL;
- //
- {
- AutoCrit autoCrit(mCritSect);
- if (threadIdx >= (int)mThreadInfos.size())
- break;
- threadInfo = mThreadInfos[threadIdx++];
- }
- AutoCrit autoCrit(threadInfo->mCritSect);
- threadInfo->mOutBuffer.Clear();
- }
- }
- void BpManager::SetClientName(const StringImpl& clientName)
- {
- AutoCrit autoCrit(mRootCmdTarget.mCritSect);
- mClientName = clientName;
- }
- BpResult BpManager::Init(const char* serverName, const char* sessionName)
- {
- if (serverName == NULL)
- {
- FinishWorkThread();
- return BpResult_Ok;
- }
- if (mSocket != INVALID_SOCKET)
- return BpResult_AlreadyInitialized;
- BfpGUID guid;
- BfpSystem_CreateGUID(&guid);
- mSessionID = StrFormat("%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X",
- guid.mData1, guid.mData2, guid.mData3,
- guid.mData4[0], guid.mData4[1], guid.mData4[2], guid.mData4[3],
- guid.mData4[4], guid.mData4[5], guid.mData4[6], guid.mData4[7]);
- if (mClientName.IsEmpty())
- {
- BfpSystemResult result;
- BFP_GETSTR_HELPER(mClientName, result, BfpSystem_GetComputerName(__STR, __STRLEN, &result));
- }
- #ifdef BF_PLATFORM_WINDOWS
- WSADATA wsa;
- int result = WSAStartup(MAKEWORD(2, 0), &wsa);
- if (result != 0)
- {
- return BpResult_InternalError;
- }
- #endif
- mSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
- if (mSocket == INVALID_SOCKET)
- return BpResult_InternalError;
- u_long iMode = 1;
- #ifdef BF_PLATFORM_WINDOWS
- result = ioctlsocket(mSocket, FIONBIO, &iMode);
- #else
- int result = ioctl(mSocket, FIONBIO, &iMode);
- #endif
- AutoCrit autoCrit(mCritSect);
- AutoCrit autoCrit2(mRootCmdTarget.mCritSect);
-
- mInitCount++;
- mThreadRunning = true;
- mCollectData = true;
- mServerName = serverName;
- mSessionName = sessionName;
- mConnectState = BpConnectState_Connecting;
- mCurTick = 0;
- mZoneNameMap.Clear();
- bool isReinit = mInitCount > 1;
- if (isReinit)
- {
- // We may have old data in the command targets, so clear all that out
- mRootCmdTarget.Disable();
- for (auto threadInfo : mThreadInfos)
- {
- AutoCrit autoCrit(mCritSect);
- threadInfo->Disable();
- if (threadInfo->mThreadName != NULL)
- threadInfo->SetThreadName(threadInfo->mThreadName);
- threadInfo->mCurDepth = 0;
- }
- }
- for (auto threadInfo : mThreadInfos)
- {
- mRootCmdTarget.AddThread(threadInfo->mThreadId, threadInfo->mNativeThreadId);
- }
- mRootCmdTarget.Init();
-
- mThread = BfpThread_Create(ThreadProcThunk, (void*)this, 64 * 1024, BfpThreadCreateFlag_StackSizeReserve);
- return BpResult_Ok;
- }
- void BpManager::RetryConnect()
- {
- {
- AutoCrit autoCrit(mCritSect);
- if ((mConnectState == BpConnectState_Connecting) ||
- (mConnectState == BpConnectState_Connected))
- return;
- }
- Shutdown();
- Init(mServerName.c_str(), mSessionName.c_str());
- }
- void BpManager::Pause()
- {
- BfpSystem_InterlockedExchangeAdd32((uint32*)&mPauseCount, 1);
- }
- void BpManager::Unpause()
- {
- BfpSystem_InterlockedExchangeAdd32((uint32*)&mPauseCount, -1);
- }
- void BpManager::Shutdown()
- {
- BfpThread* workerThread = NULL;
- {
- AutoCrit autoCrit(mCritSect);
- mCollectData = false;
- workerThread = mThread;
- if (workerThread != NULL)
- mShutdownEvent.Set(true);
- }
-
- if (workerThread != NULL)
- {
- BfpThread_WaitFor(mThread, -1);
- mShutdownEvent.Reset();
- }
- }
- bool BpManager::IsDisconnected()
- {
- return mConnectState == BpConnectState_NotConnected;
- }
- BpContext* Beefy::BpManager::CreateContext(const char* name)
- {
- return nullptr;
- }
- void Beefy::BpManager::CloseContext()
- {
- }
- void BpManager::Tick()
- {
- mRootCmdTarget.Tick();
- }
- BpThreadInfo* BpManager::GetCurThreadInfo()
- {
- BpThreadInfo* threadInfo = sBpThreadInfo;
- if (threadInfo == NULL)
- {
- if (gBpManagerOwner.mDidShutdown)
- return &sFakeThreadInfo;
- threadInfo = Get()->SlowGetCurThreadInfo();
- sBpThreadInfo = threadInfo;
- }
-
- return threadInfo;
- }
- struct BpSharedMemory
- {
- public:
- BpManager* mBpManager;
- int mABIVersion;
- };
- BpManager* BpManager::Get()
- {
- if (sBpManager != NULL)
- return sBpManager;
- #ifdef BF_PLATFORM_WINDOWS
- char mutexName[128];
- sprintf(mutexName, "BeefPerf_mutex_%d", GetCurrentProcessId());
- char memName[128];
- sprintf(memName, "BeefPerf_mem_%d", GetCurrentProcessId());
- auto mutex = ::CreateMutexA(NULL, TRUE, mutexName);
- if (mutex != NULL)
- {
- HANDLE fileMapping = ::OpenFileMappingA(FILE_MAP_ALL_ACCESS, FALSE, memName);
- if (fileMapping != NULL)
- {
- BpSharedMemory* sharedMem = (BpSharedMemory*)MapViewOfFile(fileMapping, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(BpSharedMemory));
- if (sharedMem != NULL)
- {
- if (sharedMem->mABIVersion == BP_ABI_VERSION)
- {
- sBpManager = sharedMem->mBpManager;
- }
- else
- OutputDebugStringA("*** BeefPerf ABI mismatch! ***\r\n");
- ::UnmapViewOfFile(sharedMem);
- }
- else
- BF_FATAL("BpManager::Get MapViewOfFile error");
- ::CloseHandle(fileMapping);
- }
- else
- {
- fileMapping = ::CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, sizeof(BpSharedMemory), memName);
- if (fileMapping != NULL)
- {
- BpSharedMemory* sharedMem = (BpSharedMemory*)MapViewOfFile(fileMapping, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(BpSharedMemory));
- if (sharedMem != NULL)
- {
- sBpManager = new BpManager();
- sBpManager->mMutex = mutex;
- sBpManager->mSharedMemoryFile = fileMapping;
- sharedMem->mBpManager = sBpManager;
- sharedMem->mABIVersion = BP_ABI_VERSION;
- ::UnmapViewOfFile(sharedMem);
- ::ReleaseMutex(mutex);
- gBpManagerOwner.mOwnedManager = sBpManager;
- }
- else
- {
- BF_FATAL("BpManager::Get MapViewOfFile error");
- ::CloseHandle(fileMapping);
- ::CloseHandle(mutex);
- }
- }
- else
- BF_FATAL("BpManager::Get CreateFileMapping error");
- }
- }
- else
- {
- BF_FATAL("BpManager::Get CreateMutex error");
- }
- #endif //BF_PLATFORM_WINDOWS
-
- if (sBpManager == NULL)
- {
- sBpManager = new BpManager();
- gBpManagerOwner.mOwnedManager = sBpManager;
- }
- return sBpManager;
- }
- //////////////////////////////////////////////////////////////////////////
- BP_EXPORT void BP_CALLTYPE BpShutdown()
- {
- BpManager::Get()->Shutdown();
- }
- BP_EXPORT void BP_CALLTYPE BpSetClientName(const char* clientName)
- {
- BpManager::Get()->SetClientName(clientName);
- }
- BP_EXPORT void BP_CALLTYPE BpInit(const char* serverName, const char* sessionName)
- {
- BpManager::Get()->Init(serverName, sessionName);
- }
- BP_EXPORT BpConnectState BP_CALLTYPE BpGetConnectState()
- {
- return BpManager::Get()->mConnectState;
- }
- BP_EXPORT void BP_CALLTYPE BpRetryConnect()
- {
- return BpManager::Get()->RetryConnect();
- }
- BP_EXPORT void BP_CALLTYPE BpPause()
- {
- BpManager::Get()->Pause();
- }
- BP_EXPORT void BP_CALLTYPE BpUnpause()
- {
- BpManager::Get()->Unpause();
- }
- BP_EXPORT void BP_CALLTYPE BpSetThreadName(const char* threadName)
- {
- BpManager::GetCurThreadInfo()->SetThreadName(threadName);
- }
- BP_EXPORT void BP_CALLTYPE BpEnter(const char* zoneName)
- {
- BpManager::GetCurThreadInfo()->Enter(zoneName);
- }
- BP_EXPORT void BP_CALLTYPE BpEnterF(const char* zoneName, ...)
- {
- va_list args;
- va_start(args, zoneName);
- BpManager::GetCurThreadInfo()->Enter(zoneName, args);
- }
- BP_EXPORT void BP_CALLTYPE BpLeave()
- {
- BpManager::GetCurThreadInfo()->Leave();
- }
- BP_EXPORT void BP_CALLTYPE BpFrameTick()
- {
- BpManager::Get()->Tick();
- }
- BP_EXPORT void BP_CALLTYPE BpEvent(const char* name, const char* details)
- {
- BpManager::GetCurThreadInfo()->Event(name, details);
- }
- BP_EXPORT const char* BP_CALLTYPE BpDynStr(const char* str)
- {
- return BpManager::GetCurThreadInfo()->DynamicString(str);
- }
- #else
- BP_EXPORT void BP_CALLTYPE BpShutdown()
- {
-
- }
- BP_EXPORT BpConnectState BP_CALLTYPE BpGetConnectState()
- {
- return BpConnectState_NotConnected;
- }
- BP_EXPORT void BP_CALLTYPE BpRetryConnect()
- {
- }
- BP_EXPORT void BP_CALLTYPE BpPause()
- {
- }
- BP_EXPORT void BP_CALLTYPE BpUnpause()
- {
- }
- BP_EXPORT void BP_CALLTYPE BpSetClientName(const char* clientName)
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpInit(const char* serverName, const char* sessionName)
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpSetThreadName(const char* threadName)
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpEnter(const char* zoneName)
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpEnterF(const char* zoneName, ...)
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpLeave()
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpFrameTick()
- {
-
- }
- BP_EXPORT void BP_CALLTYPE BpEvent(const char* name, const char* details)
- {
-
- }
- BP_EXPORT const char* BP_CALLTYPE BpDynStr(const char* str)
- {
- return str;
- }
- #endif
|