added support for built-in defines __DATE__ and __TIME__ (#251)

Author: firewave

simplecpp.cpp

@@ -25,6 +25,7 @@
 #include <algorithm>
 #include <climits>
 #include <cstring>
+#include <ctime>
 #include <exception>
 #include <fstream>
 #include <iostream>
@@ -2867,6 +2868,28 @@ static bool preprocessToken(simplecpp::TokenList &output, const simplecpp::Token
     return true;
 }
 
+static void getLocaltime(struct tm &ltime) {
+    time_t t;
+    time(&t);
+#ifndef _WIN32
+    localtime_r(&t, &ltime);
+#else
+    localtime_s(&ltime, &t);
+#endif
+}
+
+static std::string getDateDefine(struct tm *timep) {
+    char buf[] = "??? ?? ????";
+    strftime(buf, sizeof(buf), "%b %d %Y", timep);
+    return std::string("\"").append(buf).append("\"");
+}
+
+static std::string getTimeDefine(struct tm *timep) {
+    char buf[] = "??:??:??";
+    strftime(buf, sizeof(buf), "%T", timep);
+    return std::string("\"").append(buf).append("\"");
+}
+
 void simplecpp::preprocess(simplecpp::TokenList &output, const simplecpp::TokenList &rawtokens, std::vector<std::string> &files, std::map<std::string, simplecpp::TokenList *> &filedata, const simplecpp::DUI &dui, simplecpp::OutputList *outputList, std::list<simplecpp::MacroUsage> *macroUsage, std::list<simplecpp::IfCond> *ifCond)
 {
     std::map<std::string, std::size_t> sizeOfType(rawtokens.sizeOfType);
@@ -2909,6 +2932,10 @@ void simplecpp::preprocess(simplecpp::TokenList &output, const simplecpp::TokenL
     macros.insert(std::make_pair("__FILE__", Macro("__FILE__", "__FILE__", files)));
     macros.insert(std::make_pair("__LINE__", Macro("__LINE__", "__LINE__", files)));
     macros.insert(std::make_pair("__COUNTER__", Macro("__COUNTER__", "__COUNTER__", files)));
+    struct tm ltime = {};
+    getLocaltime(ltime);
+    macros.insert(std::make_pair("__DATE__", Macro("__DATE__", getDateDefine(&ltime), files)));
+    macros.insert(std::make_pair("__TIME__", Macro("__TIME__", getTimeDefine(&ltime), files)));
 
     if (dui.std == "c++11")
         macros.insert(std::make_pair("__cplusplus", Macro("__cplusplus", "201103L", files)));

test.cpp

@@ -2161,6 +2161,47 @@ static void preprocessSizeOf()
     ASSERT_EQUALS("file0,1,syntax_error,failed to evaluate #if condition, invalid sizeof expression\n", toString(outputList));
 }
 
+static void timeDefine()
+{
+    const char code[] = "__TIME__";
+    const std::string t = preprocess(code);
+    // "19:09:53"
+    ASSERT_EQUALS(10, t.size());
+    // TODO: split string and check proper ranges instead
+    ASSERT_EQUALS('"', t[0]);
+    ASSERT_EQUALS(true, isdigit(t[1]) != 0);
+    ASSERT_EQUALS(true, isdigit(t[2]) != 0);
+    ASSERT_EQUALS(':', t[3]);
+    ASSERT_EQUALS(true, isdigit(t[4]) != 0);
+    ASSERT_EQUALS(true, isdigit(t[5]) != 0);
+    ASSERT_EQUALS(':', t[6]);
+    ASSERT_EQUALS(true, isdigit(t[7]) != 0);
+    ASSERT_EQUALS(true, isdigit(t[8]) != 0);
+    ASSERT_EQUALS('"', t[9]);
+}
+
+static void dateDefine()
+{
+    const char code[] = "__DATE__";
+    const std::string dt = preprocess(code);
+    // "\"Mar 11 2022\""
+    ASSERT_EQUALS(13, dt.size());
+    // TODO: split string and check proper ranges instead
+    ASSERT_EQUALS('"', dt[0]);
+    ASSERT_EQUALS(true, dt[1] >= 'A' && dt[1] <= 'Z'); // uppercase letter
+    ASSERT_EQUALS(true, dt[2] >= 'a' && dt[2] <= 'z'); // lowercase letter
+    ASSERT_EQUALS(true, dt[3] >= 'a' && dt[3] <= 'z'); // lowercase letter
+    ASSERT_EQUALS(' ', dt[4]);
+    ASSERT_EQUALS(true, isdigit(dt[5]) != 0);
+    ASSERT_EQUALS(true, isdigit(dt[6]) != 0);
+    ASSERT_EQUALS(' ', dt[7]);
+    ASSERT_EQUALS(true, isdigit(dt[8]) != 0);
+    ASSERT_EQUALS(true, isdigit(dt[9]) != 0);
+    ASSERT_EQUALS(true, isdigit(dt[10]) != 0);
+    ASSERT_EQUALS(true, isdigit(dt[11]) != 0);
+    ASSERT_EQUALS('"', dt[12]);
+}
+
 int main(int argc, char **argv)
 {
     TEST_CASE(backslash);
@@ -2342,5 +2383,8 @@ int main(int argc, char **argv)
 
     TEST_CASE(preprocessSizeOf);
 
+    TEST_CASE(timeDefine);
+    TEST_CASE(dateDefine);
+
     return numberOfFailedAssertions > 0 ? EXIT_FAILURE : EXIT_SUCCESS;
 }