#4448 - Fail early if engine file path is unwritable before ONNX parsing

Signed-off-by: Paul Pacheco <[email protected]>

Author: PierreMarieCurie

samples/common/sampleOptions.cpp

@@ -1321,6 +1321,10 @@ void BuildOptions::parse(Arguments& arguments)
 
     if (getAndDelOption(arguments, "--saveEngine", engine))
     {
+        if (!canWriteFile(engine))
+        {
+            throw std::invalid_argument(std::string("Cannot write engine file to path: ") + engine);
+        }
         save = true;
     }
     if (load && save)

samples/common/sampleUtils.cpp

@@ -116,6 +116,36 @@ void loadFromFile(std::string const& fileName, char* dst, size_t size)
     }
 }
 
+// Check if the file at the given path can be written to.
+bool canWriteFile(const std::string& path)
+{
+    // Verify that the target directory exists
+    namespace fs = std::filesystem;
+    fs::path p(path);
+    fs::path dir = p.has_parent_path() ? p.parent_path() : fs::current_path();
+    if (!fs::exists(dir) || !fs::is_directory(dir))
+    {
+        return false;
+    }
+
+    // Try creating and writing to a temporary file in the directory
+    const fs::path tempFilePath = dir / ".writetest.tmp";
+    std::ofstream test(tempFilePath.string(), std::ios::out | std::ios::trunc);
+    if (!test.is_open())
+    {
+        return false;
+    }
+    test << "test";
+    const bool ok = test.good();
+    test.close();
+
+    // Clean up the temporary file without throwing on failure
+    std::error_code ec;
+    fs::remove(tempFilePath, ec);
+
+    return ok;
+}
+
 std::vector<std::string> splitToStringVec(std::string const& s, char separator, int64_t maxSplit)
 {
     std::vector<std::string> splitted;

samples/common/sampleUtils.h

@@ -19,6 +19,7 @@
 #define TRT_SAMPLE_UTILS_H
 
 #include <fstream>
+#include <filesystem>
 #include <iostream>
 #include <memory>
 #include <numeric>
@@ -74,6 +75,8 @@ void dumpBuffer(void const* buffer, std::string const& separator, std::ostream&
 
 void loadFromFile(std::string const& fileName, char* dst, size_t size);
 
+bool canWriteFile(const std::string& path);
+
 std::vector<std::string> splitToStringVec(std::string const& option, char separator, int64_t maxSplit = -1);
 
 bool broadcastIOFormats(std::vector<IOFormat> const& formats, size_t nbBindings, bool isInput = true);