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

Add support for importing zstd-compressed chrome traces.

Browse Source
This commit is contained in:
Bartosz Taudul 2021-05-06 18:10:08 +02:00
parent a6c6943a6c
commit fa1a717e3c
No known key found for this signature in database
GPG Key ID: B7FE2008B7575DF3
1 changed files with 64 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,12 +6,24 @@
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unordered_map>
#include <sys/stat.h>
#ifdef _MSC_VER
# define stat64 _stat64
#endif
#if defined __CYGWIN__ || defined __APPLE__
# define stat64 stat
#endif
#include "json.hpp"
#include "../../server/TracyFileWrite.hpp"
#include "../../server/TracyMmap.hpp"
#include "../../server/TracyWorker.hpp"
#include "../../zstd/zstd.h"
using json = nlohmann::json;
@ -41,15 +53,59 @@ int main( int argc, char** argv )
printf( "Loading...\r" );
fflush( stdout );
std::ifstream is( input );
if( !is.is_open() )
{
fprintf( stderr, "Cannot open input file!\n" );
exit( 1 );
}
json j;
is >> j;
is.close();
const auto fnsz = strlen( input );
if( fnsz > 4 && memcmp( input+fnsz-4, ".zst", 4 ) == 0 )
{
FILE* f = fopen( input, "rb" );
if( !f )
{
fprintf( stderr, "Cannot open input file!\n" );
exit( 1 );
}
struct stat64 sb;
if( stat64( input, &sb ) != 0 )
{
fprintf( stderr, "Cannot open input file!\n" );
fclose( f );
exit( 1 );
}
const auto zsz = sb.st_size;
auto zbuf = (char*)mmap( nullptr, zsz, PROT_READ, MAP_SHARED, fileno( f ), 0 );
fclose( f );
if( !zbuf )
{
fprintf( stderr, "Cannot mmap input file!\n" );
exit( 1 );
}
const auto sz = ZSTD_getDecompressedSize( zbuf, zsz );
auto buf = new char[sz];
const auto res = ZSTD_decompress( buf, sz, zbuf, zsz );
munmap( zbuf, zsz );
if( ZSTD_isError( res ) )
{
delete[] buf;
fprintf( stderr, "Couldn't decompress input file (%s)!\n", ZSTD_getErrorName( res ) );
exit( 1 );
}
j = json::parse( buf, buf+sz );
delete[] buf;
}
else
{
std::ifstream is( input );
if( !is.is_open() )
{
fprintf( stderr, "Cannot open input file!\n" );
exit( 1 );
}
is >> j;
is.close();
}
printf( "\33[2KParsing...\r" );
fflush( stdout );