BeefPerf.cpp 38 KB

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