BeefPerf.cpp 38 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790
  1. #ifndef NOMINMAX
  2. #define NOMINMAX
  3. #endif
  4. #include "BeefPerf.h"
  5. #include "platform/PlatformHelper.h"
  6. #ifndef BP_DISABLED
  7. #ifndef BF_PLATFORM_WINDOWS
  8. #include <netinet/in.h>
  9. #include <netinet/ip.h>
  10. #include <sys/ioctl.h>
  11. #include <sys/types.h>
  12. #include <netdb.h>
  13. typedef fd_set FD_SET;
  14. #define closesocket close
  15. #endif
  16. #ifdef BF_PLATFORM_MACOS
  17. #include <sys/socket.h>
  18. #include <mach/error.h>
  19. #include <mach/mach.h>
  20. #endif
  21. #ifdef BF_PLATFORM_POSIX
  22. #include <cerrno>
  23. #endif
  24. #pragma comment(lib,"wsock32.lib")
  25. #pragma warning(disable:4996)
  26. USING_NS_BF;
  27. // All DLLs must use the same ABI version
  28. #define BP_ABI_VERSION 2
  29. BpManager* BpManager::sBpManager = NULL;
  30. static bool gOwnsBpManager = false;
  31. static BpThreadInfo sFakeThreadInfo;
  32. struct BpManagerOwner
  33. {
  34. BpManager* mOwnedManager;
  35. bool mDidShutdown;
  36. BpManagerOwner()
  37. {
  38. mOwnedManager = NULL;
  39. mDidShutdown = false;
  40. }
  41. ~BpManagerOwner()
  42. {
  43. mDidShutdown = true;
  44. delete mOwnedManager;
  45. }
  46. };
  47. static BpManagerOwner gBpManagerOwner;
  48. BF_TLS_DECLSPEC BpThreadInfo* Beefy::BpManager::sBpThreadInfo;
  49. inline void EncodeSLEB128(uint8*& buf, int value)
  50. {
  51. bool hasMore;
  52. do
  53. {
  54. uint8 curByte = (uint8)(value & 0x7f);
  55. value >>= 7;
  56. hasMore = !((((value == 0) && ((curByte & 0x40) == 0)) ||
  57. ((value == -1) && ((curByte & 0x40) != 0))));
  58. if (hasMore)
  59. curByte |= 0x80;
  60. *(buf++) = curByte;
  61. }
  62. while (hasMore);
  63. }
  64. inline void EncodeSLEB128(uint8*& buf, int64_t value)
  65. {
  66. bool hasMore;
  67. do
  68. {
  69. uint8 curByte = (uint8)(value & 0x7f);
  70. value >>= 7;
  71. hasMore = !((((value == 0) && ((curByte & 0x40) == 0)) ||
  72. ((value == -1) && ((curByte & 0x40) != 0))));
  73. if (hasMore)
  74. curByte |= 0x80;
  75. *(buf++) = curByte;
  76. }
  77. while (hasMore);
  78. }
  79. inline void EncodeULEB32(uint8*& buf, uint64 value)
  80. {
  81. do
  82. {
  83. uint8 byteVal = value & 0x1f;
  84. value >>= 5;
  85. if (value != 0)
  86. byteVal |= 0x20; // Mark this byte to show that more bytes will follow
  87. *(buf++) = byteVal;
  88. }
  89. while (value != 0);
  90. }
  91. //////////////////////////////////////////////////////////////////////////
  92. Buffer::Buffer()
  93. {
  94. mPtr = NULL;
  95. mBufSize = 0;
  96. mDataSize = 0;
  97. }
  98. Buffer::~Buffer()
  99. {
  100. delete mPtr;
  101. }
  102. uint8* Buffer::Alloc(int size)
  103. {
  104. if (mDataSize + size > mBufSize)
  105. {
  106. // Grow factor 50%
  107. int wantSize = max(mDataSize + size, mBufSize + mBufSize / 2);
  108. auto bpManager = BpManager::Get();
  109. //uint8* newPtr = new uint8[wantSize];
  110. uint8* newPtr = (uint8*)bpManager->AllocBytes(wantSize);
  111. memcpy(newPtr, mPtr, mDataSize);
  112. //delete mPtr;
  113. bpManager->FreeBytes(mPtr);
  114. mPtr = newPtr;
  115. mBufSize = wantSize;
  116. }
  117. uint8* ptr = mPtr + mDataSize;
  118. mDataSize += size;
  119. return ptr;
  120. }
  121. void Buffer::Free()
  122. {
  123. //delete mPtr;
  124. BpManager::Get()->FreeBytes(mPtr);
  125. mPtr = NULL;
  126. mDataSize = 0;
  127. mBufSize = 0;
  128. }
  129. void Buffer::Clear()
  130. {
  131. mDataSize = 0;
  132. }
  133. //////////////////////////////////////////////////////////////////////////
  134. CircularBuffer::View::View()
  135. {
  136. mPtr = NULL;
  137. mCircularBuffer = NULL;
  138. mTempBuf = NULL;
  139. mTempBufSize = 0;
  140. mSrcIdx = 0;
  141. mSrcSize = 0;
  142. }
  143. CircularBuffer::View::~View()
  144. {
  145. delete mTempBuf;
  146. }
  147. void CircularBuffer::View::Commit(int size)
  148. {
  149. if (mPtr == mTempBuf)
  150. {
  151. if (size == -1)
  152. size = mSrcSize;
  153. else
  154. BF_ASSERT(size <= mSrcSize);
  155. mCircularBuffer->Write(mTempBuf, mSrcIdx, size);
  156. }
  157. else if (size != -1)
  158. BF_ASSERT(size <= mSrcSize);
  159. }
  160. CircularBuffer::CircularBuffer()
  161. {
  162. mTail = 0;
  163. mBufSize = 0;
  164. mDataSize = 0;
  165. mBuffer = NULL;
  166. }
  167. CircularBuffer::~CircularBuffer()
  168. {
  169. delete mBuffer;
  170. }
  171. void CircularBuffer::Clear()
  172. {
  173. mTail = 0;
  174. mDataSize = 0;
  175. }
  176. void CircularBuffer::Resize(int newSize)
  177. {
  178. uint8* newBuffer = new uint8[newSize];
  179. Read(newBuffer, 0, mDataSize);
  180. delete mBuffer;
  181. mBuffer = newBuffer;
  182. mBufSize = newSize;
  183. mTail = 0;
  184. }
  185. void CircularBuffer::GrowReserve(int addSize)
  186. {
  187. if (mDataSize + addSize <= mBufSize)
  188. return;
  189. Resize(max(mDataSize + addSize, mDataSize + mDataSize/2));
  190. }
  191. void CircularBuffer::Grow(int addSize)
  192. {
  193. GrowReserve(addSize);
  194. mDataSize += addSize;
  195. }
  196. void CircularBuffer::GrowFront(int addSize)
  197. {
  198. if (mDataSize + addSize > mBufSize)
  199. {
  200. Resize(mDataSize + addSize);
  201. }
  202. mDataSize += addSize;
  203. mTail = (mTail + mBufSize - addSize) % mBufSize;
  204. }
  205. int CircularBuffer::GetSize()
  206. {
  207. return mDataSize;
  208. }
  209. void CircularBuffer::MapView(int idx, int len, CircularBuffer::View& view)
  210. {
  211. view.mCircularBuffer = this;
  212. view.mSrcIdx = idx;
  213. view.mSrcSize = len;
  214. if (mTail + idx + len <= mBufSize)
  215. {
  216. view.mPtr = mBuffer + mTail + idx;
  217. }
  218. else
  219. {
  220. if (view.mTempBufSize < len)
  221. {
  222. delete view.mTempBuf;
  223. view.mTempBuf = new uint8[len];
  224. view.mTempBufSize = len;
  225. }
  226. view.mPtr = view.mTempBuf;
  227. Read(view.mTempBuf, idx, len);
  228. }
  229. }
  230. void CircularBuffer::Read(void* ptr, int idx, int len)
  231. {
  232. BF_ASSERT(len <= mBufSize);
  233. if (len == 0)
  234. return;
  235. int absIdx = (mTail + idx) % mBufSize;
  236. if (absIdx + len > mBufSize)
  237. {
  238. int lowSize = mBufSize - absIdx;
  239. memcpy(ptr, mBuffer + absIdx, lowSize);
  240. memcpy((uint8*)ptr + lowSize, mBuffer, len - lowSize);
  241. }
  242. else
  243. {
  244. memcpy(ptr, mBuffer + absIdx, len);
  245. }
  246. }
  247. void CircularBuffer::Write(void* ptr, int idx, int len)
  248. {
  249. BF_ASSERT(len <= mBufSize);
  250. if (len == 0)
  251. return;
  252. int absIdx = (mTail + idx) % mBufSize;
  253. if (absIdx + len > mBufSize)
  254. {
  255. int lowSize = mBufSize - absIdx;
  256. memcpy(mBuffer + absIdx, ptr, lowSize);
  257. memcpy(mBuffer, (uint8*)ptr + lowSize, len - lowSize);
  258. }
  259. else
  260. {
  261. memcpy(mBuffer + absIdx, ptr, len);
  262. }
  263. }
  264. void CircularBuffer::RemoveFront(int len)
  265. {
  266. mTail = (mTail + len) % mBufSize;
  267. mDataSize -= len;
  268. }
  269. //////////////////////////////////////////////////////////////////////////
  270. BpCmdTarget::BpCmdTarget()
  271. {
  272. mCurDynStrIdx = 0;
  273. mCurDepth = 0;
  274. mThreadName = NULL;
  275. }
  276. void BpCmdTarget::Disable()
  277. {
  278. AutoCrit autoCrit(mCritSect);
  279. mOutBuffer.Free();
  280. }
  281. const char* BpCmdTarget::DynamicString(const char* str)
  282. {
  283. int usedIdx = mCurDynStrIdx;
  284. mDynStrs[usedIdx] = str;
  285. mCurDynStrIdx = (mCurDynStrIdx + 1) % BF_ARRAY_COUNT(mDynStrs);
  286. return (const char*)(intptr)usedIdx;
  287. }
  288. const char* BpCmdTarget::ToStrPtr(const char* str)
  289. {
  290. if ((intptr)str < BF_ARRAY_COUNT(mDynStrs))
  291. return mDynStrs[(intptr)str];
  292. return str;
  293. }
  294. #define BPCMD_PREPARE if ((gBpManagerOwner.mDidShutdown) || (!BpManager::Get()->mCollectData)) return; AutoCrit autoCrit(mCritSect)
  295. #define GET_FROM(ptr, T) *((T*)(ptr += sizeof(T)) - 1)
  296. //#define BPCMD_RESERVE(addSize) mOutBuffer.resize(mOutBuffer.size() + (addSize)); uint8* data = &mOutBuffer[mOutBuffer.size() - (addSize)];
  297. #define BPCMD_RESERVE(addSize) uint8* data = mOutBuffer.Alloc(addSize)
  298. #define BPCMD_RESERVE_UNDECL(addSize) data = mOutBuffer.Alloc(addSize)
  299. #define BPCMD_MEMBER(T) *((T*)(data += sizeof(T)) - 1)
  300. #define BPCMD_MEMCPY(ptr, size) memcpy(data, ptr, size); data += size
  301. #define BPCMD_END() BF_ASSERT(data == mOutBuffer.mPtr + mOutBuffer.mDataSize)
  302. static int64 GetTimestamp()
  303. {
  304. #ifdef BF_PLATFORM_WINDOWS
  305. return __rdtsc() / 100;
  306. #else
  307. return BfpSystem_GetCPUTick();
  308. #endif
  309. }
  310. #define MAX_DEPTH 8192
  311. void BpCmdTarget::Enter(const char* name)
  312. {
  313. BPCMD_PREPARE;
  314. // Failure here could be from unbalanced enter/leave calls
  315. BF_ASSERT((uint32)mCurDepth <= MAX_DEPTH);
  316. if ((intptr)name < BF_ARRAY_COUNT(mDynStrs))
  317. {
  318. const char* dynStr = mDynStrs[(intptr)name];
  319. int len = (int)strlen(dynStr);
  320. BPCMD_RESERVE(1 + 8 + len + 1);
  321. BPCMD_MEMBER(uint8) = BpCmd_EnterDyn;
  322. BPCMD_MEMBER(int64) = GetTimestamp();
  323. BPCMD_MEMCPY(dynStr, len + 1);
  324. BPCMD_END();
  325. }
  326. else
  327. {
  328. BPCMD_RESERVE(1 + 8 + sizeof(const char*));
  329. BPCMD_MEMBER(uint8) = BpCmd_Enter;
  330. BPCMD_MEMBER(int64) = GetTimestamp();
  331. BPCMD_MEMBER(const char*) = name;
  332. BPCMD_END();
  333. }
  334. mCurDepth++;
  335. }
  336. void BpCmdTarget::Enter(const char* name, va_list args)
  337. {
  338. BPCMD_PREPARE;
  339. // Failure here could be from unbalanced enter/leave calls
  340. BF_ASSERT((uint32)mCurDepth <= MAX_DEPTH);
  341. int len = 0;
  342. int paramSize = 0;
  343. //va_list origArgs = args;
  344. va_list origArgs;
  345. va_copy(origArgs, args);
  346. //va_start(args, name);
  347. const char* cPtr = ToStrPtr(name);
  348. while (true)
  349. {
  350. char c = *(cPtr++);
  351. if (c == 0)
  352. break;
  353. len++;
  354. if (c == '%')
  355. {
  356. len++;
  357. char nextC = *(cPtr++);
  358. if (nextC != '%')
  359. {
  360. if (nextC == 'f')
  361. {
  362. va_arg(args, double);
  363. paramSize += 4; // float
  364. }
  365. else if (nextC == 'd')
  366. {
  367. intptr val = va_arg(args, intptr);
  368. paramSize += 4; // int32
  369. }
  370. else if (nextC == 's')
  371. {
  372. const char* str = ToStrPtr(va_arg(args, char*));
  373. paramSize += (int)strlen(str) + 1;
  374. }
  375. else
  376. {
  377. BF_FATAL("Invalid format flag");
  378. }
  379. }
  380. }
  381. }
  382. //va_end(args);
  383. uint8* data;
  384. if ((intptr)name < BF_ARRAY_COUNT(mDynStrs))
  385. {
  386. const char* dynStr = mDynStrs[(intptr)name];
  387. int len = (int)strlen(dynStr);
  388. BPCMD_RESERVE_UNDECL(1 + 8 + len + 1 + paramSize);
  389. BPCMD_MEMBER(uint8) = BpCmd_EnterDyn;
  390. BPCMD_MEMBER(int64) = GetTimestamp();
  391. BPCMD_MEMCPY(dynStr, len + 1);
  392. }
  393. else
  394. {
  395. BPCMD_RESERVE_UNDECL(1 + 8 + sizeof(const char*) + paramSize);
  396. BPCMD_MEMBER(uint8) = BpCmd_Enter;
  397. BPCMD_MEMBER(int64) = GetTimestamp();
  398. BPCMD_MEMBER(const char*) = name;
  399. }
  400. /*BPCMD_RESERVE(1 + 8 + len + 1 + paramSize);
  401. BPCMD_MEMBER(uint8) = BpCmd_Enter;
  402. BPCMD_MEMBER(int64) = GetTimestamp();
  403. BPCMD_MEMCPY(name, len + 1);*/
  404. args = origArgs;
  405. //va_start(args, name);
  406. cPtr = ToStrPtr(name);
  407. while (true)
  408. {
  409. char c = *(cPtr++);
  410. if (c == 0)
  411. break;
  412. if (c == '%')
  413. {
  414. char nextC = *(cPtr++);
  415. if (nextC != '%')
  416. {
  417. if (nextC == 'f')
  418. {
  419. BPCMD_MEMBER(float) = (float)va_arg(args, double);
  420. }
  421. else if (nextC == 'd')
  422. {
  423. BPCMD_MEMBER(int32) = (int32)va_arg(args, intptr);
  424. }
  425. else if (nextC == 's')
  426. {
  427. const char* str = ToStrPtr(va_arg(args, char*));
  428. BPCMD_MEMCPY(str, (int)strlen(str) + 1);
  429. }
  430. else
  431. {
  432. BF_FATAL("Invalid format flag");
  433. }
  434. }
  435. }
  436. }
  437. va_end(args);
  438. BPCMD_END();
  439. mCurDepth++;
  440. }
  441. void BpCmdTarget::EnterF(const char* name, ...)
  442. {
  443. va_list args;
  444. va_start(args, name);
  445. Enter(name, args);
  446. va_end(args);
  447. }
  448. void BpCmdTarget::Leave()
  449. {
  450. BPCMD_PREPARE;
  451. if (mCurDepth <= 0)
  452. {
  453. // This is either due to improperly balanced Enter/Leaves or from a Reconnnect
  454. BF_ASSERT(BpManager::Get()->mInitCount > 1);
  455. return;
  456. }
  457. BPCMD_RESERVE(1 + 8);
  458. BPCMD_MEMBER(uint8) = BpCmd_Leave;
  459. BPCMD_MEMBER(int64) = GetTimestamp();
  460. BPCMD_END();
  461. mCurDepth--;
  462. }
  463. void BpCmdTarget::Event(const char* name, const char* details)
  464. {
  465. name = ToStrPtr(name);
  466. details = ToStrPtr(details);
  467. BPCMD_PREPARE;
  468. int nameLen = (int)strlen(name);
  469. int detailsLen = (int)strlen(details);
  470. BPCMD_RESERVE(1 + 8 + nameLen+1 + detailsLen+1);
  471. BPCMD_MEMBER(uint8) = BpCmd_Event;
  472. BPCMD_MEMBER(int64) = GetTimestamp();
  473. BPCMD_MEMCPY(name, nameLen + 1);
  474. BPCMD_MEMCPY(details, detailsLen + 1);
  475. BPCMD_END();
  476. }
  477. void BpRootCmdTarget::Init()
  478. {
  479. BPCMD_PREPARE;
  480. BPCMD_RESERVE(1);
  481. BPCMD_MEMBER(uint8) = BpCmd_Init;
  482. BPCMD_END();
  483. }
  484. void BpRootCmdTarget::Tick()
  485. {
  486. BPCMD_PREPARE;
  487. BPCMD_RESERVE(1 + 8);
  488. BPCMD_MEMBER(uint8) = BpCmd_Tick;
  489. BPCMD_MEMBER(int64) = GetTimestamp();
  490. BPCMD_END();
  491. }
  492. void BpRootCmdTarget::KeepAlive()
  493. {
  494. BPCMD_PREPARE;
  495. BPCMD_RESERVE(1 + 8);
  496. BPCMD_MEMBER(uint8) = BpCmd_KeepAlive;
  497. BPCMD_MEMBER(int64) = GetTimestamp();
  498. BPCMD_END();
  499. }
  500. void BpRootCmdTarget::AddThread(int threadId, BfpThreadId nativeThreadId)
  501. {
  502. BPCMD_PREPARE;
  503. BPCMD_RESERVE(1 + 8 + 4 + 4);
  504. BPCMD_MEMBER(uint8) = BpCmd_ThreadAdd;
  505. BPCMD_MEMBER(int64) = GetTimestamp();
  506. BPCMD_MEMBER(int32) = threadId;
  507. BPCMD_MEMBER(int32) = (int32)nativeThreadId;
  508. BPCMD_END();
  509. }
  510. BpThreadInfo::BpThreadInfo()
  511. {
  512. mNativeThreadId = -1;
  513. mThreadId = 0;
  514. mReadyToSend = false;
  515. mHasTerminated = false;
  516. }
  517. BpThreadInfo::~BpThreadInfo()
  518. {
  519. if (mThreadName != NULL)
  520. BpManager::Get()->FreeBytes(mThreadName);
  521. }
  522. void BpThreadInfo::SetThreadName(const char* name)
  523. {
  524. int len = (int)strlen(name);
  525. if (mThreadName == NULL)
  526. {
  527. mThreadName = (char*)BpManager::Get()->AllocBytes(len + 1);
  528. memcpy(mThreadName, name, len + 1);
  529. }
  530. BPCMD_PREPARE;
  531. BPCMD_RESERVE(1 + len + 1);
  532. BPCMD_MEMBER(uint8) = BpCmd_ThreadName;
  533. BPCMD_MEMCPY(name, len + 1);
  534. BPCMD_END();
  535. }
  536. void BpThreadInfo::RemoveThread()
  537. {
  538. BPCMD_PREPARE;
  539. BPCMD_RESERVE(1 + 8);
  540. BPCMD_MEMBER(uint8) = BpCmd_ThreadRemove;
  541. BPCMD_MEMBER(int64) = GetTimestamp();
  542. BPCMD_END();
  543. }
  544. //////////////////////////////////////////////////////////////////////////
  545. static void NTAPI FlsFreeFunc(void* ptr)
  546. {
  547. BpThreadInfo* threadInfo = (BpThreadInfo*)ptr;
  548. auto bpManager = BpManager::Get();
  549. AutoCrit autoCrit(bpManager->mCritSect);
  550. threadInfo->mHasTerminated = true;
  551. threadInfo->RemoveThread();
  552. if (!bpManager->mThreadRunning)
  553. {
  554. bpManager->mThreadInfos.Remove(threadInfo);
  555. delete threadInfo;
  556. // It's possible that other thread-specific destructors will occur after this call, which may use BeefPerf,
  557. // and in that case we will re-add this thread info
  558. BpManager::sBpThreadInfo = NULL;
  559. }
  560. }
  561. BpManager::BpManager() : mShutdownEvent(true), mTLSDtor(&FlsFreeFunc)
  562. {
  563. #ifdef BF_PLATFORM_WINDOWS
  564. mMutex = NULL;
  565. mSharedMemoryFile = NULL;
  566. #endif
  567. mSocket = INVALID_SOCKET;
  568. mConnectState = BpConnectState_NotConnected;
  569. mThread = NULL;
  570. mCurTick = 0;
  571. mCurThreadId = 0;
  572. mOutBlockSizeLeft = 0;
  573. mPauseCount = 0;
  574. mInitCount = 0;
  575. mCollectData = false;
  576. mThreadRunning = false;
  577. mInitTimeStamp = GetTimestamp();
  578. mInitTickCount = BFTickCount();
  579. }
  580. BpManager::~BpManager()
  581. {
  582. Shutdown();
  583. #ifdef BF_PLATFORM_WINDOWS
  584. if (mMutex != NULL)
  585. ::CloseHandle(mMutex);
  586. if (mSharedMemoryFile != NULL)
  587. ::CloseHandle(mSharedMemoryFile);
  588. #endif
  589. }
  590. bool BpManager::Connect()
  591. {
  592. struct sockaddr_in server;
  593. struct hostent * hp;
  594. server.sin_family = PF_INET;
  595. hp = gethostbyname(mServerName.c_str());
  596. if (hp == NULL)
  597. return false;
  598. memcpy(&server.sin_addr, hp->h_addr_list[0], sizeof(server.sin_addr));
  599. server.sin_port = htons(4208);
  600. #ifdef BF_PLATFORM_WINDOWS
  601. bool isLocalhost = server.sin_addr.S_un.S_addr == 0x0100007f;
  602. #else
  603. bool isLocalhost = server.sin_addr.s_addr == 0x0100007f;
  604. #endif
  605. int result = ::connect(mSocket, (sockaddr*)&server, sizeof(server));
  606. if (result != 0)
  607. {
  608. #ifdef BF_PLATFORM_WINDOWS
  609. int err = WSAGetLastError();
  610. if (err != WSAEWOULDBLOCK)
  611. return false;
  612. #else
  613. if (errno != EINPROGRESS)
  614. return false;
  615. #endif
  616. }
  617. int totalWaitedMS = 0;
  618. // Wait for connection - normally we wait for either the connection to occur or an error to occur,
  619. // but if we shutdown the app then we need to ensure we waited "long enough", but we don't want
  620. // to keep very short programs from exiting in a timely manner so we have a short localhost timeout
  621. // since Windows will delay for 2s before failing
  622. while (true)
  623. {
  624. int selectTimeoutMS = 20;
  625. timeval timeout;
  626. timeout.tv_sec = 0;
  627. timeout.tv_usec = selectTimeoutMS * 1000;
  628. FD_SET socketWriteSet;
  629. FD_ZERO(&socketWriteSet);
  630. FD_SET(mSocket, &socketWriteSet);
  631. FD_SET socketErrorSet;
  632. FD_ZERO(&socketErrorSet);
  633. FD_SET(mSocket, &socketErrorSet);
  634. int result = select((int)mSocket + 1, NULL, &socketWriteSet, &socketErrorSet, &timeout);
  635. if (result == -1)
  636. return false;
  637. if (FD_ISSET(mSocket, &socketWriteSet))
  638. break;
  639. if (FD_ISSET(mSocket, &socketErrorSet))
  640. return false;
  641. totalWaitedMS += selectTimeoutMS;
  642. if (mShutdownEvent.WaitFor(0))
  643. {
  644. // We are shutting down - have we waited enough?
  645. int minWaitMS = isLocalhost ? 50 : 20*1000;
  646. if (totalWaitedMS >= minWaitMS)
  647. return false;
  648. }
  649. }
  650. return true;
  651. }
  652. uint8* BpManager::StartCmd(uint8 cmd, CircularBuffer::View& view, int maxLen)
  653. {
  654. mOutBuffer.GrowReserve(maxLen);
  655. mOutBuffer.MapView(mOutBuffer.GetSize(), maxLen, view);
  656. uint8* dataOut = view.mPtr;
  657. GET_FROM(dataOut, uint8) = cmd;
  658. return dataOut;
  659. }
  660. void BpManager::EndCmd(CircularBuffer::View& view, uint8* ptr)
  661. {
  662. int actualSize = (int)(ptr - view.mPtr);
  663. view.Commit(actualSize);
  664. mOutBuffer.Grow(actualSize);
  665. }
  666. void BpManager::TrySendData()
  667. {
  668. CircularBuffer::View outView;
  669. while (true)
  670. {
  671. int sizeLeft = mOutBuffer.GetSize();
  672. if (sizeLeft == 0)
  673. return;
  674. int trySend = std::min(sizeLeft, 8192);
  675. mOutBuffer.MapView(0, trySend, outView);
  676. int result = send(mSocket, (const char*)outView.mPtr, trySend, 0);
  677. if (result < 0)
  678. {
  679. #ifdef BF_PLATFORM_WINDOWS
  680. int err = WSAGetLastError();
  681. switch (err)
  682. {
  683. case WSAECONNABORTED:
  684. case WSAECONNRESET:
  685. mConnectState = BpConnectState_NotConnected;
  686. }
  687. #else
  688. switch (errno)
  689. {
  690. case ECONNRESET:
  691. mConnectState = BpConnectState_NotConnected;
  692. }
  693. #endif
  694. return;
  695. }
  696. mOutBuffer.RemoveFront(result);
  697. }
  698. }
  699. void BpManager::LostConnection()
  700. {
  701. mCollectData = false;
  702. AutoCrit autoCrit(mCritSect);
  703. mRootCmdTarget.Disable();
  704. for (auto threadInfo : mThreadInfos)
  705. threadInfo->Disable();
  706. #ifdef BF_PLATFORM_WINDOWS
  707. closesocket(mSocket);
  708. #else
  709. close(mSocket);
  710. #endif
  711. mSocket = INVALID_SOCKET;
  712. }
  713. BpThreadInfo* BpManager::SlowGetCurThreadInfo()
  714. {
  715. BfpThreadId curThreadId = BfpThread_GetCurrentId();
  716. // Try to find an existing one
  717. {
  718. AutoCrit autoCrit(mCritSect);
  719. for (auto threadInfo : mThreadInfos)
  720. if (threadInfo->mNativeThreadId == curThreadId)
  721. return threadInfo;
  722. }
  723. auto threadInfo = new BpThreadInfo();
  724. threadInfo->mThreadId = mCurThreadId++;
  725. threadInfo->mNativeThreadId = curThreadId;
  726. sBpThreadInfo = threadInfo;
  727. AutoCrit autoCrit(mCritSect);
  728. mThreadInfos.push_back(threadInfo);
  729. if (mThreadRunning)
  730. mRootCmdTarget.AddThread(threadInfo->mThreadId, threadInfo->mNativeThreadId);
  731. mTLSDtor.Add((void*)threadInfo);
  732. return threadInfo;
  733. }
  734. void* BpManager::AllocBytes(int size)
  735. {
  736. return new uint8[size];
  737. }
  738. void BpManager::FreeBytes(void* ptr)
  739. {
  740. delete (uint8*)ptr;
  741. }
  742. void BpManager::FinishWorkThread()
  743. {
  744. AutoCrit autoCrit(mCritSect);
  745. for (int i = 0; i < (int)mThreadInfos.size(); i++)
  746. {
  747. auto threadInfo = mThreadInfos[i];
  748. if (threadInfo->mHasTerminated)
  749. {
  750. delete threadInfo;
  751. mThreadInfos.erase(mThreadInfos.begin() + i);
  752. i--;
  753. }
  754. }
  755. mThreadRunning = false;
  756. }
  757. void BpManager::ThreadProc()
  758. {
  759. if (!Connect())
  760. {
  761. mConnectState = BpConnectState_Failed;
  762. LostConnection();
  763. FinishWorkThread();
  764. return;
  765. }
  766. Buffer threadBuffer;
  767. String tempStr;
  768. CircularBuffer::View outView;
  769. timeval timeout;
  770. timeout.tv_sec = 0;
  771. timeout.tv_usec = 20 * 1000; // 20ms
  772. uint32 gLastMsgTick = BFTickCount();
  773. DWORD lastTimeTick = 0;
  774. while (mConnectState != BpConnectState_NotConnected)
  775. {
  776. bool wantsExit = mShutdownEvent.WaitFor(0);
  777. FD_SET socketReadSet;
  778. FD_ZERO(&socketReadSet);
  779. FD_SET(mSocket, &socketReadSet);
  780. FD_SET socketWriteSet;
  781. FD_ZERO(&socketWriteSet);
  782. if (mOutBuffer.GetSize() > 0)
  783. FD_SET(mSocket, &socketWriteSet);
  784. FD_SET socketErrorSet;
  785. FD_ZERO(&socketErrorSet);
  786. FD_SET(mSocket, &socketErrorSet);
  787. int selResult = select((int)mSocket + 1, &socketReadSet, &socketWriteSet, &socketErrorSet, &timeout);
  788. if (FD_ISSET(mSocket, &socketWriteSet))
  789. {
  790. TrySendData();
  791. //continue;
  792. }
  793. if (FD_ISSET(mSocket, &socketReadSet))
  794. {
  795. // Just eat the data
  796. uint8 data[4096];
  797. int len = recv(mSocket, (char*)data, 4096, 0);
  798. int b = 0;
  799. }
  800. if (FD_ISSET(mSocket, &socketErrorSet))
  801. {
  802. #ifdef BF_PLATFORM_WINDOWS
  803. int err = WSAGetLastError();
  804. #endif
  805. mConnectState = BpConnectState_NotConnected;
  806. LostConnection();
  807. FinishWorkThread();
  808. return;
  809. }
  810. // Alloc space for size
  811. mOutBuffer.Grow(4);
  812. int startBufferSize = mOutBuffer.GetSize();
  813. int threadIdx = -1;
  814. int threadId = -1;
  815. //
  816. {
  817. // We need to send the BpCmd_ThreadAdd before sending any data from the thread,
  818. AutoCrit autoCrit(mCritSect);
  819. for (int threadIdx = 0; threadIdx < (int)mThreadInfos.size(); threadIdx++)
  820. {
  821. auto threadInfo = mThreadInfos[threadIdx];
  822. if ((threadInfo->mHasTerminated) && (threadInfo->mOutBuffer.mDataSize == 0))
  823. {
  824. mThreadInfos.erase(mThreadInfos.begin() + threadIdx);
  825. delete threadInfo;
  826. threadIdx--;
  827. }
  828. else
  829. {
  830. threadInfo->mReadyToSend = true;
  831. }
  832. }
  833. }
  834. while (true)
  835. {
  836. BpCmdTarget* cmdTarget = NULL;
  837. if (threadIdx == -1)
  838. {
  839. cmdTarget = &mRootCmdTarget;
  840. threadIdx++;
  841. }
  842. else
  843. {
  844. AutoCrit autoCrit(mCritSect);
  845. if (threadIdx >= (int)mThreadInfos.size())
  846. break;
  847. auto threadInfo = mThreadInfos[threadIdx++];
  848. threadId = threadInfo->mThreadId;
  849. cmdTarget = threadInfo;
  850. if (!threadInfo->mReadyToSend)
  851. continue;
  852. }
  853. //
  854. {
  855. AutoCrit autoCrit(cmdTarget->mCritSect);
  856. BF_ASSERT(threadBuffer.mDataSize == 0);
  857. memcpy(threadBuffer.Alloc(cmdTarget->mOutBuffer.mDataSize), cmdTarget->mOutBuffer.mPtr, cmdTarget->mOutBuffer.mDataSize);
  858. cmdTarget->mOutBuffer.Clear();
  859. }
  860. if (threadBuffer.mDataSize == 0)
  861. continue;
  862. uint8* dataOut = StartCmd(BpCmd_SetThread, outView, 5);
  863. EncodeSLEB128(dataOut, threadId);
  864. EndCmd(outView, dataOut);
  865. uint8* dataIn = threadBuffer.mPtr;
  866. uint8* dataInEnd = dataIn + threadBuffer.mDataSize;
  867. static int sIdx = 0;
  868. while (dataIn < dataInEnd)
  869. {
  870. BpCmd cmd = (BpCmd)*(dataIn++);
  871. int nameMsgLen;
  872. if (cmd == BpCmd_Init)
  873. {
  874. AutoCrit autoCrit(mRootCmdTarget.mCritSect);
  875. String env;
  876. env += "SessionID\t";
  877. env += mSessionID;
  878. env += "\n";
  879. if (!mSessionName.IsEmpty())
  880. {
  881. env += "SessionName\t";
  882. env += mSessionName;
  883. env += "\n";
  884. }
  885. if (!mClientName.IsEmpty())
  886. {
  887. env += "ClientName\t";
  888. env += mClientName;
  889. env += "\n";
  890. }
  891. int32 nameLen = (int32)env.length();
  892. uint8* dataOut = StartCmd(BpCmd_Init, outView, 1 + 4 + nameLen + 1);
  893. EncodeSLEB128(dataOut, BP_CLIENT_VERSION);
  894. memcpy(dataOut, env.c_str(), nameLen + 1);
  895. dataOut += nameLen + 1;
  896. EndCmd(outView, dataOut);
  897. dataOut = StartCmd(BpCmd_ClockInfo, outView, 1 + 8 + 8 + 8 + 1);
  898. EncodeSLEB128(dataOut, mCurTick);
  899. EncodeSLEB128(dataOut, GetTimestamp());
  900. EncodeSLEB128(dataOut, (int)BFTickCount());
  901. EncodeSLEB128(dataOut, BfpSystem_GetCPUTickFreq());
  902. EndCmd(outView, dataOut);
  903. }
  904. else if ((cmd == BpCmd_Enter) || (cmd == BpCmd_EnterDyn))
  905. {
  906. int64 tick = GET_FROM(dataIn, int64);
  907. const char* name;
  908. int strIdx;
  909. int paramsSize = 0;
  910. if (cmd == BpCmd_EnterDyn)
  911. {
  912. name = (const char*)dataIn;
  913. int nameLen = (int)strlen(name);
  914. dataIn += nameLen + 1;
  915. paramsSize = -1;
  916. strIdx = -nameLen;
  917. nameMsgLen = 4+1 + nameLen;
  918. }
  919. else
  920. {
  921. name = GET_FROM(dataIn, const char*);
  922. BpZoneName* zoneName;
  923. if (mZoneNameMap.TryAdd(name, NULL, &zoneName))
  924. {
  925. int nameLen = (int)strlen(name);
  926. strIdx = (int)mZoneNameMap.size() - 1;
  927. zoneName->mIdx = strIdx;
  928. zoneName->mSize = 0;
  929. bool isDyn = false;
  930. const char* cPtr = name;
  931. while (true)
  932. {
  933. char c = *(cPtr++);
  934. if (c == 0)
  935. break;
  936. if (c == '%')
  937. {
  938. char nextC = *(cPtr++);
  939. if (nextC != '%')
  940. {
  941. if (nextC == 'f')
  942. {
  943. zoneName->mSize += 4;
  944. }
  945. else if (nextC == 'd')
  946. {
  947. zoneName->mSize += 4;
  948. }
  949. else if (nextC == 's')
  950. {
  951. isDyn = true;
  952. }
  953. }
  954. }
  955. }
  956. if (isDyn)
  957. zoneName->mSize = -1;
  958. uint8* dataOut = StartCmd(BpCmd_StrEntry, outView, 1 + nameLen + 1);
  959. memcpy(dataOut, name, nameLen + 1);
  960. dataOut += nameLen + 1;
  961. EndCmd(outView, dataOut);
  962. paramsSize = zoneName->mSize;
  963. }
  964. else
  965. {
  966. strIdx = zoneName->mIdx;
  967. paramsSize = zoneName->mSize;
  968. }
  969. nameMsgLen = 4+1;
  970. }
  971. bool isDynSize = false;
  972. int addParamSize = 0;
  973. if (paramsSize == -1)
  974. {
  975. isDynSize = true;
  976. addParamSize = 4 + 1; // For size param
  977. // Calc size
  978. auto checkDataIn = dataIn;
  979. const char* cPtr = name;
  980. while (true)
  981. {
  982. char c = *(cPtr++);
  983. if (c == 0)
  984. break;
  985. if (c == '%')
  986. {
  987. char nextC = *(cPtr++);
  988. if (nextC != '%')
  989. {
  990. if (nextC == 'f')
  991. {
  992. checkDataIn += 4;
  993. }
  994. else if (nextC == 'd')
  995. {
  996. checkDataIn += 4;
  997. }
  998. else if (nextC == 's')
  999. {
  1000. int len = (int)strlen((const char*)checkDataIn);
  1001. checkDataIn += len + 1;
  1002. }
  1003. }
  1004. }
  1005. }
  1006. paramsSize = (int)(checkDataIn - dataIn);
  1007. }
  1008. uint8* dataOut = StartCmd(BpCmd_Enter, outView, 1 + 8+1 + nameMsgLen + paramsSize + addParamSize);
  1009. int64 tickDelta = tick - mCurTick;
  1010. mCurTick = tick;
  1011. EncodeSLEB128(dataOut, tickDelta);
  1012. if (strIdx < 0)
  1013. {
  1014. EncodeSLEB128(dataOut, strIdx);
  1015. memcpy(dataOut, name, -strIdx);
  1016. dataOut += -strIdx;
  1017. }
  1018. else
  1019. EncodeSLEB128(dataOut, strIdx);
  1020. if (isDynSize)
  1021. EncodeSLEB128(dataOut, paramsSize);
  1022. if (paramsSize != 0)
  1023. {
  1024. memcpy(dataOut, dataIn, paramsSize);
  1025. dataOut += paramsSize;
  1026. dataIn += paramsSize;
  1027. }
  1028. EndCmd(outView, dataOut);
  1029. }
  1030. else if (cmd == BpCmd_Leave)
  1031. {
  1032. uint8* dataOut = StartCmd(BpCmd_Leave, outView, 1 + 8+1);
  1033. int64 tick = GET_FROM(dataIn, int64);
  1034. int64 tickDelta = tick - mCurTick;
  1035. mCurTick = tick;
  1036. EncodeSLEB128(dataOut, tickDelta);
  1037. EndCmd(outView, dataOut);
  1038. }
  1039. else if (cmd == BpCmd_ThreadName)
  1040. {
  1041. const char* name = (const char*)dataIn;
  1042. int nameLen = (int)strlen(name);
  1043. uint8* dataOut = StartCmd(BpCmd_ThreadName, outView, 1 + nameLen + 1);
  1044. memcpy(dataOut, dataIn, nameLen + 1);
  1045. dataIn += nameLen + 1;
  1046. dataOut += nameLen + 1;
  1047. EndCmd(outView, dataOut);
  1048. }
  1049. else if (cmd == BpCmd_Tick)
  1050. {
  1051. uint8* dataOut = StartCmd(BpCmd_Tick, outView, 1 + 8+1);
  1052. int64 tick = GET_FROM(dataIn, int64);
  1053. int64 tickDelta = tick - mCurTick;
  1054. mCurTick = tick;
  1055. EncodeSLEB128(dataOut, tickDelta);
  1056. EndCmd(outView, dataOut);
  1057. }
  1058. else if (cmd == BpCmd_KeepAlive)
  1059. {
  1060. uint8* dataOut = StartCmd(BpCmd_KeepAlive, outView, 1 + 8+1);
  1061. int64 tick = GET_FROM(dataIn, int64);
  1062. int64 tickDelta = tick - mCurTick;
  1063. mCurTick = tick;
  1064. EncodeSLEB128(dataOut, tickDelta);
  1065. EndCmd(outView, dataOut);
  1066. }
  1067. else if (cmd == BpCmd_ThreadAdd)
  1068. {
  1069. uint8* dataOut = StartCmd(BpCmd_ThreadAdd, outView, 1 + 8+1 + 4+1 + 4+1);
  1070. int64 tick = GET_FROM(dataIn, int64);
  1071. int64 tickDelta = tick - mCurTick;
  1072. mCurTick = tick;
  1073. int32 threadId = GET_FROM(dataIn, int32);
  1074. int32 nativeThreadId = GET_FROM(dataIn, int32);
  1075. EncodeSLEB128(dataOut, tickDelta);
  1076. EncodeSLEB128(dataOut, threadId);
  1077. EncodeSLEB128(dataOut, nativeThreadId);
  1078. EndCmd(outView, dataOut);
  1079. }
  1080. else if (cmd == BpCmd_ThreadRemove)
  1081. {
  1082. uint8* dataOut = StartCmd(BpCmd_ThreadRemove, outView, 1 + 8+1);
  1083. int64 tick = GET_FROM(dataIn, int64);
  1084. int64 tickDelta = tick - mCurTick;
  1085. mCurTick = tick;
  1086. EncodeSLEB128(dataOut, tickDelta);
  1087. EndCmd(outView, dataOut);
  1088. }
  1089. else if (cmd == BpCmd_Event)
  1090. {
  1091. int64 tick = GET_FROM(dataIn, int64);
  1092. int64 tickDelta = tick - mCurTick;
  1093. mCurTick = tick;
  1094. const char* name = (const char*)dataIn;
  1095. int nameLen = (int)strlen(name);
  1096. const char* details = (const char*)(dataIn + nameLen + 1);
  1097. int detailsLen = (int)strlen(details);
  1098. uint8* dataOut = StartCmd(BpCmd_Event, outView, 1 + 8+1 + nameLen + 1 + detailsLen + 1);
  1099. EncodeSLEB128(dataOut, tickDelta);
  1100. memcpy(dataOut, dataIn, nameLen + 1);
  1101. dataIn += nameLen + 1;
  1102. dataOut += nameLen + 1;
  1103. memcpy(dataOut, dataIn, detailsLen + 1);
  1104. dataIn += detailsLen + 1;
  1105. dataOut += detailsLen + 1;
  1106. EndCmd(outView, dataOut);
  1107. }
  1108. else
  1109. BF_FATAL("Not handled");
  1110. // If we get a large backlog then try to break it up into smaller chunks and send out periodically
  1111. int bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
  1112. BF_ASSERT(bufSizeAdded >= 0);
  1113. if (bufSizeAdded >= 64*1024)
  1114. {
  1115. // Set chunk size
  1116. CircularBuffer::View view;
  1117. mOutBuffer.MapView(startBufferSize - 4, 4, view);
  1118. uint8* data = view.mPtr;
  1119. GET_FROM(data, int32) = bufSizeAdded;
  1120. view.Commit();
  1121. // We can attempt some more sending now...
  1122. TrySendData();
  1123. // Start next header
  1124. mOutBuffer.Grow(4);
  1125. startBufferSize = mOutBuffer.GetSize();
  1126. }
  1127. }
  1128. threadBuffer.Clear();
  1129. }
  1130. int bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
  1131. DWORD curTick = BFTickCount();
  1132. // Encode clock info
  1133. if ((lastTimeTick == 0) || ((bufSizeAdded > 0) && (curTick - lastTimeTick >= 1000)))
  1134. {
  1135. uint8* dataOut = StartCmd(BpCmd_ClockInfo, outView, 1 + 8 + 8 + 8 + 1);
  1136. EncodeSLEB128(dataOut, mCurTick);
  1137. EncodeSLEB128(dataOut, GetTimestamp());
  1138. EncodeSLEB128(dataOut, (int)BFTickCount());
  1139. EncodeSLEB128(dataOut, BfpSystem_GetCPUTickFreq());
  1140. EndCmd(outView, dataOut);
  1141. lastTimeTick = curTick;
  1142. }
  1143. bufSizeAdded = mOutBuffer.GetSize() - startBufferSize;
  1144. //BF_ASSERT((bufSizeAdded <= 64*1024) && (bufSizeAdded >= 0));
  1145. if (bufSizeAdded != 0)
  1146. {
  1147. // Set chunk size
  1148. CircularBuffer::View view;
  1149. mOutBuffer.MapView(startBufferSize - 4, 4, view);
  1150. uint8* data = view.mPtr;
  1151. GET_FROM(data, int32) = bufSizeAdded;
  1152. view.Commit();
  1153. }
  1154. else
  1155. mOutBuffer.Grow(-4); // No data added, pop chunk size off
  1156. if (mOutBuffer.GetSize() == 0)
  1157. {
  1158. if (wantsExit)
  1159. {
  1160. break;
  1161. }
  1162. uint32 tickNow = BFTickCount();
  1163. if ((tickNow - gLastMsgTick >= 1000) /*&& (!wantsExit)*/)
  1164. {
  1165. mRootCmdTarget.KeepAlive();
  1166. }
  1167. }
  1168. else
  1169. {
  1170. gLastMsgTick = BFTickCount();
  1171. }
  1172. }
  1173. mOutBuffer.Clear();
  1174. mConnectState = BpConnectState_NotConnected;
  1175. closesocket(mSocket);
  1176. mSocket = INVALID_SOCKET;
  1177. FinishWorkThread();
  1178. }
  1179. void BFP_CALLTYPE BpManager::ThreadProcThunk(void* ptr)
  1180. {
  1181. ((BpManager*)ptr)->ThreadProc();
  1182. }
  1183. void BpManager::Clear()
  1184. {
  1185. int threadIdx = 0;
  1186. while (true)
  1187. {
  1188. BpThreadInfo* threadInfo = NULL;
  1189. //
  1190. {
  1191. AutoCrit autoCrit(mCritSect);
  1192. if (threadIdx >= (int)mThreadInfos.size())
  1193. break;
  1194. threadInfo = mThreadInfos[threadIdx++];
  1195. }
  1196. AutoCrit autoCrit(threadInfo->mCritSect);
  1197. threadInfo->mOutBuffer.Clear();
  1198. }
  1199. }
  1200. void BpManager::SetClientName(const StringImpl& clientName)
  1201. {
  1202. AutoCrit autoCrit(mRootCmdTarget.mCritSect);
  1203. mClientName = clientName;
  1204. }
  1205. BpResult BpManager::Init(const char* serverName, const char* sessionName)
  1206. {
  1207. if (serverName == NULL)
  1208. {
  1209. FinishWorkThread();
  1210. return BpResult_Ok;
  1211. }
  1212. if (mSocket != INVALID_SOCKET)
  1213. return BpResult_AlreadyInitialized;
  1214. BfpGUID guid;
  1215. BfpSystem_CreateGUID(&guid);
  1216. mSessionID = StrFormat("%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X",
  1217. guid.mData1, guid.mData2, guid.mData3,
  1218. guid.mData4[0], guid.mData4[1], guid.mData4[2], guid.mData4[3],
  1219. guid.mData4[4], guid.mData4[5], guid.mData4[6], guid.mData4[7]);
  1220. if (mClientName.IsEmpty())
  1221. {
  1222. BfpSystemResult result;
  1223. BFP_GETSTR_HELPER(mClientName, result, BfpSystem_GetComputerName(__STR, __STRLEN, &result));
  1224. }
  1225. #ifdef BF_PLATFORM_WINDOWS
  1226. WSADATA wsa;
  1227. int result = WSAStartup(MAKEWORD(2, 0), &wsa);
  1228. if (result != 0)
  1229. {
  1230. return BpResult_InternalError;
  1231. }
  1232. #endif
  1233. mSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  1234. if (mSocket == INVALID_SOCKET)
  1235. return BpResult_InternalError;
  1236. u_long iMode = 1;
  1237. #ifdef BF_PLATFORM_WINDOWS
  1238. result = ioctlsocket(mSocket, FIONBIO, &iMode);
  1239. #else
  1240. int result = ioctl(mSocket, FIONBIO, &iMode);
  1241. #endif
  1242. AutoCrit autoCrit(mCritSect);
  1243. AutoCrit autoCrit2(mRootCmdTarget.mCritSect);
  1244. mInitCount++;
  1245. mThreadRunning = true;
  1246. mCollectData = true;
  1247. mServerName = serverName;
  1248. mSessionName = sessionName;
  1249. mConnectState = BpConnectState_Connecting;
  1250. mCurTick = 0;
  1251. mZoneNameMap.Clear();
  1252. bool isReinit = mInitCount > 1;
  1253. if (isReinit)
  1254. {
  1255. // We may have old data in the command targets, so clear all that out
  1256. mRootCmdTarget.Disable();
  1257. for (auto threadInfo : mThreadInfos)
  1258. {
  1259. AutoCrit autoCrit(mCritSect);
  1260. threadInfo->Disable();
  1261. if (threadInfo->mThreadName != NULL)
  1262. threadInfo->SetThreadName(threadInfo->mThreadName);
  1263. threadInfo->mCurDepth = 0;
  1264. }
  1265. }
  1266. for (auto threadInfo : mThreadInfos)
  1267. {
  1268. mRootCmdTarget.AddThread(threadInfo->mThreadId, threadInfo->mNativeThreadId);
  1269. }
  1270. mRootCmdTarget.Init();
  1271. mThread = BfpThread_Create(ThreadProcThunk, (void*)this, 64 * 1024, BfpThreadCreateFlag_StackSizeReserve);
  1272. return BpResult_Ok;
  1273. }
  1274. void BpManager::RetryConnect()
  1275. {
  1276. {
  1277. AutoCrit autoCrit(mCritSect);
  1278. if ((mConnectState == BpConnectState_Connecting) ||
  1279. (mConnectState == BpConnectState_Connected))
  1280. return;
  1281. }
  1282. Shutdown();
  1283. Init(mServerName.c_str(), mSessionName.c_str());
  1284. }
  1285. void BpManager::Pause()
  1286. {
  1287. BfpSystem_InterlockedExchangeAdd32((uint32*)&mPauseCount, 1);
  1288. }
  1289. void BpManager::Unpause()
  1290. {
  1291. BfpSystem_InterlockedExchangeAdd32((uint32*)&mPauseCount, -1);
  1292. }
  1293. void BpManager::Shutdown()
  1294. {
  1295. BfpThread* workerThread = NULL;
  1296. {
  1297. AutoCrit autoCrit(mCritSect);
  1298. mCollectData = false;
  1299. workerThread = mThread;
  1300. if (workerThread != NULL)
  1301. mShutdownEvent.Set(true);
  1302. }
  1303. if (workerThread != NULL)
  1304. {
  1305. BfpThread_WaitFor(mThread, -1);
  1306. mShutdownEvent.Reset();
  1307. }
  1308. }
  1309. bool BpManager::IsDisconnected()
  1310. {
  1311. return mConnectState == BpConnectState_NotConnected;
  1312. }
  1313. BpContext* Beefy::BpManager::CreateContext(const char* name)
  1314. {
  1315. return nullptr;
  1316. }
  1317. void Beefy::BpManager::CloseContext()
  1318. {
  1319. }
  1320. void BpManager::Tick()
  1321. {
  1322. mRootCmdTarget.Tick();
  1323. }
  1324. BpThreadInfo* BpManager::GetCurThreadInfo()
  1325. {
  1326. BpThreadInfo* threadInfo = sBpThreadInfo;
  1327. if (threadInfo == NULL)
  1328. {
  1329. if (gBpManagerOwner.mDidShutdown)
  1330. return &sFakeThreadInfo;
  1331. threadInfo = Get()->SlowGetCurThreadInfo();
  1332. sBpThreadInfo = threadInfo;
  1333. }
  1334. return threadInfo;
  1335. }
  1336. struct BpSharedMemory
  1337. {
  1338. public:
  1339. BpManager* mBpManager;
  1340. int mABIVersion;
  1341. };
  1342. BpManager* BpManager::Get()
  1343. {
  1344. if (sBpManager != NULL)
  1345. return sBpManager;
  1346. #ifdef BF_PLATFORM_WINDOWS
  1347. char mutexName[128];
  1348. sprintf(mutexName, "BeefPerf_mutex_%d", GetCurrentProcessId());
  1349. char memName[128];
  1350. sprintf(memName, "BeefPerf_mem_%d", GetCurrentProcessId());
  1351. auto mutex = ::CreateMutexA(NULL, TRUE, mutexName);
  1352. if (mutex != NULL)
  1353. {
  1354. HANDLE fileMapping = ::OpenFileMappingA(FILE_MAP_ALL_ACCESS, FALSE, memName);
  1355. if (fileMapping != NULL)
  1356. {
  1357. BpSharedMemory* sharedMem = (BpSharedMemory*)MapViewOfFile(fileMapping, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(BpSharedMemory));
  1358. if (sharedMem != NULL)
  1359. {
  1360. if (sharedMem->mABIVersion == BP_ABI_VERSION)
  1361. {
  1362. sBpManager = sharedMem->mBpManager;
  1363. }
  1364. else
  1365. OutputDebugStringA("*** BeefPerf ABI mismatch! ***\r\n");
  1366. ::UnmapViewOfFile(sharedMem);
  1367. }
  1368. else
  1369. BF_FATAL("BpManager::Get MapViewOfFile error");
  1370. ::CloseHandle(fileMapping);
  1371. }
  1372. else
  1373. {
  1374. fileMapping = ::CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, sizeof(BpSharedMemory), memName);
  1375. if (fileMapping != NULL)
  1376. {
  1377. BpSharedMemory* sharedMem = (BpSharedMemory*)MapViewOfFile(fileMapping, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(BpSharedMemory));
  1378. if (sharedMem != NULL)
  1379. {
  1380. sBpManager = new BpManager();
  1381. sBpManager->mMutex = mutex;
  1382. sBpManager->mSharedMemoryFile = fileMapping;
  1383. sharedMem->mBpManager = sBpManager;
  1384. sharedMem->mABIVersion = BP_ABI_VERSION;
  1385. ::UnmapViewOfFile(sharedMem);
  1386. ::ReleaseMutex(mutex);
  1387. gBpManagerOwner.mOwnedManager = sBpManager;
  1388. }
  1389. else
  1390. {
  1391. BF_FATAL("BpManager::Get MapViewOfFile error");
  1392. ::CloseHandle(fileMapping);
  1393. ::CloseHandle(mutex);
  1394. }
  1395. }
  1396. else
  1397. BF_FATAL("BpManager::Get CreateFileMapping error");
  1398. }
  1399. }
  1400. else
  1401. {
  1402. BF_FATAL("BpManager::Get CreateMutex error");
  1403. }
  1404. #endif //BF_PLATFORM_WINDOWS
  1405. if (sBpManager == NULL)
  1406. {
  1407. sBpManager = new BpManager();
  1408. gBpManagerOwner.mOwnedManager = sBpManager;
  1409. }
  1410. return sBpManager;
  1411. }
  1412. //////////////////////////////////////////////////////////////////////////
  1413. BP_EXPORT void BP_CALLTYPE BpShutdown()
  1414. {
  1415. BpManager::Get()->Shutdown();
  1416. }
  1417. BP_EXPORT void BP_CALLTYPE BpSetClientName(const char* clientName)
  1418. {
  1419. BpManager::Get()->SetClientName(clientName);
  1420. }
  1421. BP_EXPORT void BP_CALLTYPE BpInit(const char* serverName, const char* sessionName)
  1422. {
  1423. BpManager::Get()->Init(serverName, sessionName);
  1424. }
  1425. BP_EXPORT BpConnectState BP_CALLTYPE BpGetConnectState()
  1426. {
  1427. return BpManager::Get()->mConnectState;
  1428. }
  1429. BP_EXPORT void BP_CALLTYPE BpRetryConnect()
  1430. {
  1431. return BpManager::Get()->RetryConnect();
  1432. }
  1433. BP_EXPORT void BP_CALLTYPE BpPause()
  1434. {
  1435. BpManager::Get()->Pause();
  1436. }
  1437. BP_EXPORT void BP_CALLTYPE BpUnpause()
  1438. {
  1439. BpManager::Get()->Unpause();
  1440. }
  1441. BP_EXPORT void BP_CALLTYPE BpSetThreadName(const char* threadName)
  1442. {
  1443. BpManager::GetCurThreadInfo()->SetThreadName(threadName);
  1444. }
  1445. BP_EXPORT void BP_CALLTYPE BpEnter(const char* zoneName)
  1446. {
  1447. BpManager::GetCurThreadInfo()->Enter(zoneName);
  1448. }
  1449. BP_EXPORT void BP_CALLTYPE BpEnterF(const char* zoneName, ...)
  1450. {
  1451. va_list args;
  1452. va_start(args, zoneName);
  1453. BpManager::GetCurThreadInfo()->Enter(zoneName, args);
  1454. }
  1455. BP_EXPORT void BP_CALLTYPE BpLeave()
  1456. {
  1457. BpManager::GetCurThreadInfo()->Leave();
  1458. }
  1459. BP_EXPORT void BP_CALLTYPE BpFrameTick()
  1460. {
  1461. BpManager::Get()->Tick();
  1462. }
  1463. BP_EXPORT void BP_CALLTYPE BpEvent(const char* name, const char* details)
  1464. {
  1465. BpManager::GetCurThreadInfo()->Event(name, details);
  1466. }
  1467. BP_EXPORT const char* BP_CALLTYPE BpDynStr(const char* str)
  1468. {
  1469. return BpManager::GetCurThreadInfo()->DynamicString(str);
  1470. }
  1471. #else
  1472. BP_EXPORT void BP_CALLTYPE BpShutdown()
  1473. {
  1474. }
  1475. BP_EXPORT BpConnectState BP_CALLTYPE BpGetConnectState()
  1476. {
  1477. return BpConnectState_NotConnected;
  1478. }
  1479. BP_EXPORT void BP_CALLTYPE BpRetryConnect()
  1480. {
  1481. }
  1482. BP_EXPORT void BP_CALLTYPE BpPause()
  1483. {
  1484. }
  1485. BP_EXPORT void BP_CALLTYPE BpUnpause()
  1486. {
  1487. }
  1488. BP_EXPORT void BP_CALLTYPE BpSetClientName(const char* clientName)
  1489. {
  1490. }
  1491. BP_EXPORT void BP_CALLTYPE BpInit(const char* serverName, const char* sessionName)
  1492. {
  1493. }
  1494. BP_EXPORT void BP_CALLTYPE BpSetThreadName(const char* threadName)
  1495. {
  1496. }
  1497. BP_EXPORT void BP_CALLTYPE BpEnter(const char* zoneName)
  1498. {
  1499. }
  1500. BP_EXPORT void BP_CALLTYPE BpEnterF(const char* zoneName, ...)
  1501. {
  1502. }
  1503. BP_EXPORT void BP_CALLTYPE BpLeave()
  1504. {
  1505. }
  1506. BP_EXPORT void BP_CALLTYPE BpFrameTick()
  1507. {
  1508. }
  1509. BP_EXPORT void BP_CALLTYPE BpEvent(const char* name, const char* details)
  1510. {
  1511. }
  1512. BP_EXPORT const char* BP_CALLTYPE BpDynStr(const char* str)
  1513. {
  1514. return str;
  1515. }
  1516. #endif