main.cpp: error out when file/path provided by -I or -include=, or the input do not exist

Author: firewave

main.cpp

@@ -9,6 +9,7 @@
 #include <cstring>
 #include <fstream>
 #include <iostream>
+#include <sys/stat.h>
 #include <string>
 #include <vector>
 
@@ -21,6 +22,7 @@ int main(int argc, char **argv)
 
     // Settings..
     simplecpp::DUI dui;
+    dui.removeComments = true;
     bool quiet = false;
     bool error_only = false;
     for (int i = 1; i < argc; i++) {
@@ -95,6 +97,32 @@ int main(int argc, char **argv)
         return 1;
     }
 
+    // TODO: move this logic into simplecpp?
+    {
+        bool inc_missing = false;
+        for (const std::string& inc : dui.includes) {
+            std::fstream f(inc, std::ios::in|std::ios::out); // check if this is an existing file
+            if (!f.is_open()) {
+                inc_missing = true;
+                std::cout << "error: could not open include '" << inc << "'" << std::endl;
+            }
+        }
+        if (inc_missing)
+            return 1;
+    }
+    {
+        bool inc_missing = false;
+        for (const std::string& inc : dui.includePaths) {
+            struct stat file_stat;
+            if ((stat(inc.c_str(), &file_stat) == -1) || ((file_stat.st_mode & S_IFMT) != S_IFDIR)) {
+                inc_missing = true;
+                std::cout << "error: could not find include path '" << inc << "'" << std::endl;
+            }
+        }
+        if (inc_missing)
+            return 1;
+    }
+
     if (!filename) {
         std::cout << "Syntax:" << std::endl;
         std::cout << "simplecpp [options] filename" << std::endl;
@@ -110,29 +138,31 @@ int main(int argc, char **argv)
         return 0;
     }
 
-    dui.removeComments = true;
+    // TODO: move this logic into simplecpp
+    std::ifstream f(filename, std::ios::in|std::ios::out);
+    if (!f.is_open()) {
+        std::cout << "error: could not open file '" << filename << "'" << std::endl;
+        return 1;
+    }
 
     // Perform preprocessing
     simplecpp::OutputList outputList;
     std::vector<std::string> files;
-    simplecpp::TokenList *rawtokens;
-    if (use_istream) {
-        std::ifstream f(filename);
-        if (!f.is_open()) {
-            std::cout << "error: could not open file '" << filename << "'" << std::endl;
-            return 1;
+    simplecpp::TokenList outputTokens(files);
+    {
+        simplecpp::TokenList *rawtokens;
+        if (use_istream) {
+            rawtokens = new simplecpp::TokenList(f, files,filename,&outputList);
+        } else {
+            f.close();
+            rawtokens = new simplecpp::TokenList(filename,files,&outputList);
         }
-        rawtokens = new simplecpp::TokenList(f, files,filename,&outputList);
-    } else {
-        rawtokens = new simplecpp::TokenList(filename,files,&outputList);
+        rawtokens->removeComments();
+        simplecpp::FileDataCache filedata;
+        simplecpp::preprocess(outputTokens, *rawtokens, files, filedata, dui, &outputList);
+        simplecpp::cleanup(filedata);
+        delete rawtokens;
     }
-    rawtokens->removeComments();
-    simplecpp::TokenList outputTokens(files);
-    simplecpp::FileDataCache filedata;
-    simplecpp::preprocess(outputTokens, *rawtokens, files, filedata, dui, &outputList);
-    simplecpp::cleanup(filedata);
-    delete rawtokens;
-    rawtokens = nullptr;
 
     // Output
     if (!quiet) {