mirror/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2024-11-10 10:41:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use streaming zstd API.

Browse Source
This commit is contained in:
Bartosz Taudul 2021-05-06 22:19:25 +02:00
parent 3c6a06f97e
commit 7f0191d19f
No known key found for this signature in database
GPG Key ID: B7FE2008B7575DF3
1 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
import-chrome/src/import-chrome.cpp
View File

@ -81,19 +81,42 @@ int main( int argc, char** argv )
exit( 1 ); exit( 1 );
} }
const auto sz = ZSTD_getDecompressedSize( zbuf, zsz ); auto zctx = ZSTD_createDStream();
auto buf = new char[sz]; ZSTD_initDStream( zctx );
const auto res = ZSTD_decompress( buf, sz, zbuf, zsz );
munmap( zbuf, zsz ); enum { tmpSize = 64*1024 };
if( ZSTD_isError( res ) ) auto tmp = new char[tmpSize];
ZSTD_inBuffer_s zin = { zbuf, (size_t)zsz };
ZSTD_outBuffer_s zout = { tmp, (size_t)tmpSize };
std::vector<uint8_t> buf;
buf.reserve( 1024*1024 );
while( zin.pos < zin.size )
{ {
delete[] buf; const auto res = ZSTD_decompressStream( zctx, &zout, &zin );
fprintf( stderr, "Couldn't decompress input file (%s)!\n", ZSTD_getErrorName( res ) ); if( ZSTD_isError( res ) )
exit( 1 ); {
ZSTD_freeDStream( zctx );
delete[] tmp;
fprintf( stderr, "Couldn't decompress input file (%s)!\n", ZSTD_getErrorName( res ) );
exit( 1 );
}
if( zout.pos > 0 )
{
const auto bsz = buf.size();
buf.resize( bsz + zout.pos );
memcpy( buf.data() + bsz, tmp, zout.size );
zout.pos = 0;
}
} }
j = json::parse( buf, buf+sz ); ZSTD_freeDStream( zctx );
delete[] buf; delete[] tmp;
munmap( zbuf, zsz );
j = json::parse( buf.begin(), buf.end() );
} }
else else
{ {