Add .str directive

2 files changed, 35 insertions, 0 deletions

diff --git a/README.md b/README.md
index f555daf..f17875a 100644
--- a/README.md
+++ b/README.md
@@ -294,6 +294,9 @@ and `>`for the high byte.
 **.dw imm [, imm]**  
 Literal word (16-bit value).
 
+**.str "string"**  
+Include the string as literal bytes. Use `\"` to escape a quote and `\\` to escape de backslash. The string is not zero ended, use `.db` to add a zero if needed.
+
 **.ds count, imm**  
 Generates `count` bytes with value `imm`.
 
diff --git a/tr8as.c b/tr8as.c
index 15143b7..64af41d 100644
--- a/tr8as.c
+++ b/tr8as.c
@@ -695,6 +695,37 @@ static uint8_t parse_equ(As *as, char **c)
     return new_define(as, id, value);
 }
 
+static uint8_t parse_str(As *as, char **c)
+{
+    /* .str "string" */
+
+    *c = skip_whitespace(*c);
+    if (**c != '"')
+        return error_l("Syntax error", &as->loc, "expected \"");
+    (*c)++;
+
+    while (**c && **c != '\"' && **c != '\n')
+    {
+        if (**c == '\\')
+        {
+            (*c)++;
+            if (**c != '\\' && **c != '"')
+                return error_l("Syntax error", &as->loc, "invalid escape sequence");
+        }
+
+        as->out[as->addr++] = **c;
+        if (as->addr > as->size)
+            as->size = as->addr;
+        (*c)++;
+    }
+
+    if (**c != '"')
+        return error_l("Syntax error", &as->loc, "expected \"");
+
+    return 1;
+}
+
+
 static uint8_t parse_db(As *as, char **c)
 {
     char word[MAX_ID + 1];
@@ -1582,6 +1613,7 @@ static InstParse insts[] =
     { ".incbin", parse_incbin },
     { ".org", parse_org },
     { ".equ", parse_equ },
+    { ".str", parse_str },
     { ".db", parse_db },
     { ".dw", parse_dw },
     { ".ds", parse_ds },