fixed -D option

tcc.c

@@ -42,7 +42,7 @@
 /* preprocessor debug */
 //#define PP_DEBUG
 
-/* amount of virtual memory associate to a section (currently, we do
+/* amount of virtual memory associated to a section (currently, we do
    not realloc them) */
 #define SECTION_VSIZE       (1024 * 1024)
 
@@ -1306,19 +1306,12 @@ void tok_print(int *str)
 }
 #endif
 
-void preprocess(void)
+/* parse after #define */
+void parse_define(void)
 {
-    int size, i, c, v, t, *str, len;
-    char buf[1024], *q, *p;
-    char buf1[1024];
-    BufferedFile *f;
-    Sym **ps, *first, *s;
+    Sym *s, *first, **ps;
+    int v, t, *str, len;
 
-    cinp();
-    next_nomacro();
- redo:
-    if (tok == TOK_DEFINE) {
-        next_nomacro();
     v = tok;
     /* XXX: should check if same macro (ANSI) */
     first = NULL;
@@ -1357,6 +1350,22 @@ void preprocess(void)
 #endif
     s = sym_push1(&define_stack, v, t, (int)str);
     s->next = first;
+}
+
+void preprocess(void)
+{
+    int size, i, c;
+    char buf[1024], *q, *p;
+    char buf1[1024];
+    BufferedFile *f;
+    Sym *s;
+
+    cinp();
+    next_nomacro();
+ redo:
+    if (tok == TOK_DEFINE) {
+        next_nomacro();
+        parse_define();
     } else if (tok == TOK_UNDEF) {
         next_nomacro();
         s = sym_find1(&define_stack, tok);
@@ -5675,46 +5684,40 @@ int tcc_compile_file(const char *filename1)
    tcc parser, but would need a custom 'FILE *' */
 void define_symbol(const char *sym)
 {
-    TokenSym *ts;
-    int *str, len;
-    CValue cval;
-    const char *p;
-    char buf[256];
+    char *p;
+    BufferedFile bf1, *bf = &bf1;
 
-    p = strchr(sym, '=');
+    pstrcpy(bf->buffer, IO_BUF_SIZE, sym);
+    p = strchr(bf->buffer, '=');
     if (!p) {
-        pstrcpy(buf, sizeof(buf), sym);
-        p = "1";
+        /* default value */
+        pstrcat(bf->buffer, IO_BUF_SIZE, " 1");
     } else {
-        len = p - sym;
-        if (len > sizeof(buf) - 1)
-            len = sizeof(buf) - 1;
-        memcpy(buf, sym, len);
-        buf[len] = '\0';
-        p++;
+        *p = ' ';
     }
     
-    ts = tok_alloc(buf, 0);
-    str = NULL;
-    len = 0;
-    if (isnum(*p)) {
-        /* integer case */
-        cval.i = atoi(p);
-        tok_add2(&str, &len, TOK_CINT, &cval);
-    } else {
-        /* string case */
-        cval.ts = tok_alloc(p, 0);
-        tok_add2(&str, &len, TOK_STR, &cval);
-    }
-    tok_add(&str, &len, 0);
-    sym_push1(&define_stack, ts->tok, MACRO_OBJ, (int)str);
+    /* init file structure */
+    bf->fd = -1;
+    bf->buf_ptr = bf->buffer;
+    bf->buf_end = bf->buffer + strlen(bf->buffer);
+    bf->filename[0] = '\0';
+    bf->line_num = 1;
+    file = bf;
+    
+    include_stack_ptr = include_stack;
+
+    /* parse with define parser */
+    inp();
+    ch = '\n'; /* needed to parse correctly first preprocessor command */
+    next_nomacro();
+    parse_define();
+    file = NULL;
 }
 
 void undef_symbol(const char *sym)
 {
     TokenSym *ts;
     Sym *s;
-    printf("undef %s\n", sym);
     ts = tok_alloc(sym, 0);
     s = sym_find1(&define_stack, tok);
     /* undefine symbol by putting an invalid name */