mirror/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2024-09-20 05:42:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prevent overwriting existing files by the capture utility.

Browse Source
This commit is contained in:
Bartosz Taudul 2020-09-20 20:04:43 +02:00
parent 1e34b22a82
commit 7da17680ae
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
capture/src/capture.cpp
View File

@ -9,6 +9,7 @@
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include "../../common/TracyProtocol.hpp"
#include "../../server/TracyFileWrite.hpp"
@ -27,7 +28,7 @@ void SigInt( int )
void Usage()
{
printf( "Usage: capture -o output.tracy [-a address] [-p port]\n" );
printf( "Usage: capture -o output.tracy [-a address] [-p port] [-f]\n" );
exit( 1 );
}
@ -41,12 +42,13 @@ int main( int argc, char** argv )
}
#endif
bool overwrite = false;
const char* address = "localhost";
const char* output = nullptr;
int port = 8086;
int c;
while( ( c = getopt( argc, argv, "a:o:p:" ) ) != -1 )
while( ( c = getopt( argc, argv, "a:o:p:f" ) ) != -1 )
{
switch( c )
{
@ -59,6 +61,9 @@ int main( int argc, char** argv )
case 'p':
port = atoi( optarg );
break;
case 'f':
overwrite = true;
break;
default:
Usage();
break;
@ -67,6 +72,13 @@ int main( int argc, char** argv )
if( !address || !output ) Usage();
struct stat st;
if( stat( output, &st ) == 0 && !overwrite )
{
printf( "Output file %s already exists! Use -f to force overwrite.\n", output );
return 4;
}
printf( "Connecting to %s:%i...", address, port );
fflush( stdout );
tracy::Worker worker( address, port );