Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-08-24 03:58:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use proper function to read data from socket.

Browse Source
This commit is contained in:
Bartosz Taudul
2017-09-13 02:08:35 +02:00
parent 3dd744019a
commit f0d76af15c
2 changed files with 10 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
server/TracyView.cpp
View File

@@ -35,6 +35,11 @@ View::~View()
m_thread.join();
}
bool View::ShouldExit()
{
return s_instance->m_shutdown.load( std::memory_order_relaxed );
}
void View::Worker()
{
Socket sock;
@@ -48,26 +53,10 @@ void View::Worker()
if( m_shutdown.load( std::memory_order_relaxed ) ) return;
if( !sock.Connect( m_addr.c_str(), "8086" ) ) continue;
auto left = sizeof( m_timeBegin );
auto ptr = (char*)&m_timeBegin;
do
{
if( m_shutdown.load( std::memory_order_relaxed ) ) return;
auto sz = sock.Recv( ptr, left, &tv );
if( sz == 0 ) goto close;
if( sz > 0 )
{
left -= sz;
ptr += sz;
}
}
while( left > 0 );
uint8_t lz4;
while( sock.Recv( &lz4, 1, nullptr ) == -1 )
{
if( m_shutdown.load( std::memory_order_relaxed ) ) return;
}
if( !sock.Read( &m_timeBegin, sizeof( m_timeBegin ), &tv, ShouldExit ) ) goto close;
if( !sock.Read( &lz4, sizeof( lz4 ), &tv, ShouldExit ) ) goto close;
for(;;)
{