add check for .enc extension to support encryption

microsoft · Oct 3, 2023 · 78c41fa · 78c41fa
1 parent ced3b4f
commit 78c41fa
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/include/utils.h b/include/utils.h
@@ -57,7 +57,7 @@ typedef int FileHandle;
 #define PBSTR "||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||"
 #define PBWIDTH 60
 
-inline bool file_exists(const std::string &name, bool dirCheck = false)
+inline bool file_exists_impl(const std::string &name, bool dirCheck = false)
 {
     int val;
 #ifndef _WINDOWS
@@ -94,6 +94,29 @@ inline bool file_exists(const std::string &name, bool dirCheck = false)
     }
 }
 
+inline bool file_exists(const std::string &name, bool dirCheck = false)
+{
+#ifdef EXEC_ENV_OLS
+    bool exists = file_exists_impl(name, dirCheck);
+    if (exists)
+    {
+        return true;
+    }
+    if (!dirCheck)
+    {
+        // try with .enc extension
+        std::string enc_name = name + ENCRYPTED_EXTENSION;
+        return file_exists_impl(enc_name, dirCheck);
+    }
+    else
+    {
+        return exists;
+    }
+#else
+    return file_exists_impl(name, dirCheck);
+#endif
+}
+
 inline void open_file_to_write(std::ofstream &writer, const std::string &filename)
 {
     writer.exceptions(std::ofstream::failbit | std::ofstream::badbit);