BeefPerf.cpp 37 KB

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