Early check if output is writable in capture utility.

1 files changed, 11 insertions, 0 deletions

diff --git a/capture/src/capture.cpp b/capture/src/capture.cpp
index 2a74bd81..7a6544f8 100644
--- a/capture/src/capture.cpp
+++ b/capture/src/capture.cpp
@@ -1,5 +1,7 @@
 #ifdef _WIN32
 #  include <windows.h>
+#else
+#  include <unistd.h>
 #endif
 
 #include <chrono>
@@ -79,6 +81,15 @@ int main( int argc, char** argv )
         return 4;
     }
 
+    FILE* test = fopen( output, "wb" );
+    if( !test )
+    {
+        printf( "Cannot open output file %s for writing!\n", output );
+        return 5;
+    }
+    fclose( test );
+    unlink( output );
+
     printf( "Connecting to %s:%i...", address, port );
     fflush( stdout );
     tracy::Worker worker( address, port );