From a4544c617c302a7fff52ea5f5e19468f4664c49b Mon Sep 17 00:00:00 2001
From: Kyle Lin <minecraft.kyle.train@gmail.com>
Date: Sat, 9 Nov 2024 18:39:43 +0800
Subject: [PATCH] Support escaped NULL character in string and character
 literal

---
 src/lexer.c     | 4 ++++
 tests/driver.sh | 6 ++++++
 2 files changed, 10 insertions(+)

diff --git a/src/lexer.c b/src/lexer.c
index c3c7bf73..75ad1fce 100644
--- a/src/lexer.c
+++ b/src/lexer.c
@@ -304,6 +304,8 @@ token_t lex_token_internal(bool aliasing)
                     token_str[i - 1] = '\t';
                 else if (next_char == '\\')
                     token_str[i - 1] = '\\';
+                else if (next_char == '0')
+                    token_str[i - 1] = '\0';
                 else
                     abort();
             } else {
@@ -334,6 +336,8 @@ token_t lex_token_internal(bool aliasing)
                 token_str[0] = '\t';
             else if (next_char == '\\')
                 token_str[0] = '\\';
+            else if (next_char == '0')
+                token_str[0] = '\0';
             else
                 abort();
         } else {
diff --git a/tests/driver.sh b/tests/driver.sh
index dc26151e..4ed17f49 100755
--- a/tests/driver.sh
+++ b/tests/driver.sh
@@ -630,6 +630,12 @@ int main() {
 }
 EOF
 
+try_ 0 << EOF
+int main() {
+    return '\0';
+}
+EOF
+
 # function-like macro
 try_ 1 << EOF
 #define MAX(a, b) ((a) > (b) ? (a) : (b))