fix #337 - line splicing in comment not handled properly (#431)

Author: clock999

simplecpp.cpp

@@ -758,17 +758,34 @@ void simplecpp::TokenList::readfile(Stream &stream, const std::string &filename,
 
         // comment
         else if (ch == '/' && stream.peekChar() == '/') {
-            while (stream.good() && ch != '\r' && ch != '\n') {
+            while (stream.good() && ch != '\n') {
                 currentToken += ch;
                 ch = stream.readChar();
+                if(ch == '\\') {
+                    TokenString tmp;
+                    char tmp_ch = ch;
+                    while((stream.good()) && (tmp_ch == '\\' || tmp_ch == ' ' || tmp_ch == '\t')) {
+                        tmp += tmp_ch;
+                        tmp_ch = stream.readChar();
+                    }
+                    if(!stream.good()) {
+                        break;
+                    }
+
+                    if(tmp_ch != '\n') {
+                        currentToken += tmp;
+                    } else {
+                        TokenString check_portability = currentToken + tmp;
+                        const std::string::size_type pos = check_portability.find_last_not_of(" \t");
+                        if (pos < check_portability.size() - 1U && check_portability[pos] == '\\')
+                            portabilityBackslash(outputList, files, location);
+                        ++multiline;
+                        tmp_ch = stream.readChar();
+                    }
+                    ch = tmp_ch;
+                }
             }
-            const std::string::size_type pos = currentToken.find_last_not_of(" \t");
-            if (pos < currentToken.size() - 1U && currentToken[pos] == '\\')
-                portabilityBackslash(outputList, files, location);
-            if (currentToken[currentToken.size() - 1U] == '\\') {
-                ++multiline;
-                currentToken.erase(currentToken.size() - 1U);
-            } else {
+            if (ch == '\n') {
                 stream.ungetChar();
             }
         }

test.cpp

@@ -434,7 +434,30 @@ static void comment_multiline()
     const char code[] = "#define ABC {// \\\n"
                         "}\n"
                         "void f() ABC\n";
-    ASSERT_EQUALS("\n\nvoid f ( ) { }", preprocess(code));
+    ASSERT_EQUALS("\n\nvoid f ( ) {", preprocess(code));
+
+    const char code1[] = "#define ABC {// \\\r\n"
+                        "}\n"
+                        "void f() ABC\n";
+    ASSERT_EQUALS("\n\nvoid f ( ) {", preprocess(code1));
+
+    const char code2[] = "#define A 1// \\\r"
+                        "\r"
+                        "2\r"
+                        "A\r";
+    ASSERT_EQUALS("\n\n2\n1", preprocess(code2));
+
+    const char code3[] = "void f() {// \\ \n}\n";
+    ASSERT_EQUALS("void f ( ) {", preprocess(code3));
+
+    const char code4[] = "void f() {// \\\\\\\t\t\n}\n";
+    ASSERT_EQUALS("void f ( ) {", preprocess(code4));
+
+    const char code5[] = "void f() {// \\\\\\a\n}\n";
+    ASSERT_EQUALS("void f ( ) {\n}", preprocess(code5));
+
+    const char code6[] = "void f() {// \\\n\n\n}\n";
+    ASSERT_EQUALS("void f ( ) {\n\n\n}", preprocess(code6));
 }