assembler.c

   1 /*
   2  * Copyright (C) 2012
   3  *      Dale Weiler
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a copy of
   6  * this software and associated documentation files (the "Software"), to deal in
   7  * the Software without restriction, including without limitation the rights to
   8  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
   9  * of the Software, and to permit persons to whom the Software is furnished to do
  10  * so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in all
  13  * copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  */
  23 #include "gmqcc.h"
  24 /*
  25  * This is the assembler, gmqas, this is being implemented because I'm
  26  * not exactly sure how codegen would work for the C compiler as of yet
  27  * and also I plan to allow inline assembly for the compiler.
  28  */
  29 static const struct {
  30         const char  *m; /* menomic     */
  31         const size_t o; /* operands    */
  32         const size_t l; /* menomic len */
  33 } const asm_instr[] = {
  34         [INSTR_DONE]       = { "DONE"      , 1, 4 },
  35         [INSTR_MUL_F]      = { "MUL_F"     , 3, 5 },
  36         [INSTR_MUL_V]      = { "MUL_V"     , 3, 5 },
  37         [INSTR_MUL_FV]     = { "MUL_FV"    , 3, 6 },
  38         [INSTR_MUL_VF]     = { "MUL_VF"    , 3, 6 },
  39         [INSTR_DIV_F]      = { "DIV"       , 0, 3 },
  40         [INSTR_ADD_F]      = { "ADD_F"     , 3, 5 },
  41         [INSTR_ADD_V]      = { "ADD_V"     , 3, 5 },
  42         [INSTR_SUB_F]      = { "SUB_F"     , 3, 5 },
  43         [INSTR_SUB_V]      = { "DUB_V"     , 3, 5 },
  44         [INSTR_EQ_F]       = { "EQ_F"      , 0, 4 },
  45         [INSTR_EQ_V]       = { "EQ_V"      , 0, 4 },
  46         [INSTR_EQ_S]       = { "EQ_S"      , 0, 4 },
  47         [INSTR_EQ_E]       = { "EQ_E"      , 0, 4 },
  48         [INSTR_EQ_FNC]     = { "ES_FNC"    , 0, 6 },
  49         [INSTR_NE_F]       = { "NE_F"      , 0, 4 },
  50         [INSTR_NE_V]       = { "NE_V"      , 0, 4 },
  51         [INSTR_NE_S]       = { "NE_S"      , 0, 4 },
  52         [INSTR_NE_E]       = { "NE_E"      , 0, 4 },
  53         [INSTR_NE_FNC]     = { "NE_FNC"    , 0, 6 },
  54         [INSTR_LE]         = { "LE"        , 0, 2 },
  55         [INSTR_GE]         = { "GE"        , 0, 2 },
  56         [INSTR_LT]         = { "LT"        , 0, 2 },
  57         [INSTR_GT]         = { "GT"        , 0, 2 },
  58         [INSTR_LOAD_F]     = { "FIELD_F"   , 0, 7 },
  59         [INSTR_LOAD_V]     = { "FIELD_V"   , 0, 7 },
  60         [INSTR_LOAD_S]     = { "FIELD_S"   , 0, 7 },
  61         [INSTR_LOAD_ENT]   = { "FIELD_ENT" , 0, 9 },
  62         [INSTR_LOAD_FLD]   = { "FIELD_FLD" , 0, 9 },
  63         [INSTR_LOAD_FNC]   = { "FIELD_FNC" , 0, 9 },
  64         [INSTR_ADDRESS]    = { "ADDRESS"   , 0, 7 },
  65         [INSTR_STORE_F]    = { "STORE_F"   , 0, 7 },
  66         [INSTR_STORE_V]    = { "STORE_V"   , 0, 7 },
  67         [INSTR_STORE_S]    = { "STORE_S"   , 0, 7 },
  68         [INSTR_STORE_ENT]  = { "STORE_ENT" , 0, 9 },
  69         [INSTR_STORE_FLD]  = { "STORE_FLD" , 0, 9 },
  70         [INSTR_STORE_FNC]  = { "STORE_FNC" , 0, 9 },
  71         [INSTR_STOREP_F]   = { "STOREP_F"  , 0, 8 },
  72         [INSTR_STOREP_V]   = { "STOREP_V"  , 0, 8 },
  73         [INSTR_STOREP_S]   = { "STOREP_S"  , 0, 8 },
  74         [INSTR_STOREP_ENT] = { "STOREP_ENT", 0, 10},
  75         [INSTR_STOREP_FLD] = { "STOREP_FLD", 0, 10},
  76         [INSTR_STOREP_FNC] = { "STOREP_FNC", 0, 10},
  77         [INSTR_RETURN]     = { "RETURN"    , 0, 6 },
  78         [INSTR_NOT_F]      = { "NOT_F"     , 0, 5 },
  79         [INSTR_NOT_V]      = { "NOT_V"     , 0, 5 },
  80         [INSTR_NOT_S]      = { "NOT_S"     , 0, 5 },
  81         [INSTR_NOT_ENT]    = { "NOT_ENT"   , 0, 7 },
  82         [INSTR_NOT_FNC]    = { "NOT_FNC"   , 0, 7 },
  83         [INSTR_IF]         = { "IF"        , 0, 2 },
  84         [INSTR_IFNOT]      = { "IFNOT"     , 0, 5 },
  85         [INSTR_CALL0]      = { "CALL0"     , 0, 5 },
  86         [INSTR_CALL1]      = { "CALL1"     , 0, 5 },
  87         [INSTR_CALL2]      = { "CALL2"     , 0, 5 },
  88         [INSTR_CALL3]      = { "CALL3"     , 0, 5 },
  89         [INSTR_CALL4]      = { "CALL4"     , 0, 5 },
  90         [INSTR_CALL5]      = { "CALL5"     , 0, 5 },
  91         [INSTR_CALL6]      = { "CALL6"     , 0, 5 },
  92         [INSTR_CALL7]      = { "CALL7"     , 0, 5 },
  93         [INSTR_CALL8]      = { "CALL8"     , 0, 5 },
  94         [INSTR_STATE]      = { "STATE"     , 0, 5 },
  95         [INSTR_GOTO]       = { "GOTO"      , 0, 4 },
  96         [INSTR_AND]        = { "AND"       , 0, 3 },
  97         [INSTR_OR]         = { "OR"        , 0, 2 },
  98         [INSTR_BITAND]     = { "BITAND"    , 0, 6 },
  99         [INSTR_BITOR]      = { "BITOR"     , 0, 5 }
 100 };
 101
 102 /*
 103  * Some assembler keywords not part of the opcodes above: these are
 104  * for creating functions, or constants.
 105  */
 106 const char *const asm_keys[] = {
 107         "FLOAT"    , /* define float  */
 108         "VECTOR"   , /* define vector */
 109         "ENTITY"   , /* define ent    */
 110         "FIELD"    , /* define field  */
 111         "STRING"   , /* define string */
 112         "FUNCTION"
 113 };
 114
 115 static char *const asm_getline(size_t *byte, FILE *fp) {
 116         char   *line = NULL;
 117         ssize_t read = util_getline(&line, byte, fp);
 118         *byte = read;
 119         if (read == -1) {
 120                 mem_d (line);
 121                 return NULL;
 122         }
 123         return line;
 124 }
 125
 126 #define asm_rmnewline(L,S) *((L)+*(S)-1) = '\0'
 127 #define asm_skipwhite(L)             \
 128     while((*(L)==' '||*(L)=='\t')) { \
 129         (L)++;                       \
 130     }
 131
 132 void asm_init(const char *file, FILE **fp) {
 133         *fp = fopen(file, "r");
 134         code_init();
 135 }
 136
 137 void asm_close(FILE *fp) {
 138         fclose(fp);
 139         code_write();
 140 }
 141
 142 /*
 143  * Following parse states:
 144  *      ASM_FUNCTION -- in a function accepting input statements
 145  *      ....
 146  */
 147 typedef enum {
 148         ASM_NULL,
 149         ASM_FUNCTION
 150 } asm_state;
 151
 152 typedef struct {
 153         char *name;   /* name of constant    */
 154         int   offset; /* location in globals */
 155 } globals;
 156 VECTOR_MAKE(globals, assembly_constants);
 157
 158 void asm_parse(FILE *fp) {
 159         char     *data  = NULL;
 160         char     *skip  = NULL;
 161         long      line  = 1; /* current line */
 162         size_t    size  = 0; /* size of line */
 163         asm_state state = ASM_NULL;
 164
 165         while ((data = asm_getline(&size, fp)) != NULL) {
 166                 skip = data;
 167                 asm_skipwhite(skip);
 168                 asm_rmnewline(skip, &size);
 169
 170                 #define DECLTYPE(X, CODE)                                         \
 171                     if (!strncmp(X, skip, strlen(X))) {                           \
 172                         if (skip[strlen(X)] != ':') {                             \
 173                             printf("%li: Missing `:` after decltype\n",line);     \
 174                             exit (1);                                             \
 175                         }                                                         \
 176                         skip += strlen(X)+1;                                      \
 177                         asm_skipwhite(skip);                                      \
 178                         if(!isalpha(*skip)) {                                     \
 179                             printf("%li: Invalid identififer: %s\n", line, skip); \
 180                             exit (1);                                             \
 181                         } else {                                                  \
 182                             size_t offset_code      = code_statements_elements+1; \
 183                             size_t offset_chars     = code_strings_elements   +1; \
 184                             size_t offset_globals   = code_globals_elements   +1; \
 185                             size_t offset_functions = code_functions_elements +1; \
 186                             size_t offset_fields    = code_fields_elements    +1; \
 187                             size_t offset_defs      = code_defs_elements      +1; \
 188                             CODE                                                  \
 189                             /* silent unused warnings */                          \
 190                             (void)offset_code;                                    \
 191                             (void)offset_chars;                                   \
 192                             (void)offset_globals;                                 \
 193                             (void)offset_functions;                               \
 194                             (void)offset_fields;                                  \
 195                             (void)offset_defs;                                    \
 196                             assembly_constants_add((globals){                     \
 197                                 .name   = util_strdup(skip),                      \
 198                                 .offset = offset_globals                          \
 199                             });                                                   \
 200                         }                                                         \
 201                         goto end;                                                 \
 202                     }
 203
 204                 /* FLOAT    */
 205                 DECLTYPE(asm_keys[0], {
 206                         code_defs_add((prog_section_def){
 207                                 .type   = TYPE_FLOAT,
 208                                 .offset = offset_globals, /* global table */
 209                                 .name   = offset_chars    /* string table TODO */
 210                         });
 211                         float f = 0; /*TODO*/
 212                         code_globals_add(*(int*)&f);
 213
 214                 });
 215                 DECLTYPE(asm_keys[1], {
 216                         code_defs_add((prog_section_def){
 217                                 .type   = TYPE_FLOAT,
 218                                 .offset = offset_globals, /* global table */
 219                                 .name   = offset_chars    /* string table TODO */
 220                         });
 221                         float f1 = 0;
 222                         float f2 = 0;
 223                         float f3 = 0;
 224                         code_globals_add(*(int*)&f1);
 225                         code_globals_add(*(int*)&f2);
 226                         code_globals_add(*(int*)&f3);
 227                 });
 228                 /* ENTITY   */ DECLTYPE(asm_keys[2], {});
 229                 /* FIELD    */ DECLTYPE(asm_keys[3], {});
 230                 /* STRING   */
 231                 DECLTYPE(asm_keys[4], {
 232                         code_defs_add((prog_section_def){
 233                                 .type   = TYPE_STRING,
 234                                 .offset = offset_globals, /* offset to offset in string table (for data)*/
 235                                 .name   = offset_chars    /* location of name in string table (for name)*/
 236                         });
 237                         code_strings_add('h');
 238                 });
 239                 /* FUNCTION */
 240                 DECLTYPE(asm_keys[5], {
 241                         /* TODO: parse */
 242                         if (state != ASM_NULL) {
 243                                 printf("%li: Error unfinished function block, expected DONE or RETURN\n", line);
 244                                 goto end;
 245                         }
 246
 247                         state = ASM_FUNCTION;
 248                         code_defs_add((prog_section_def){
 249                                 .type   = TYPE_VOID,
 250                                 .offset = offset_globals,
 251                                 .name   = offset_chars
 252                         });
 253                         code_globals_add(offset_functions);
 254                         code_functions_add((prog_section_function){
 255                                 .entry      =  offset_code,
 256                                 .firstlocal =  0,
 257                                 .locals     =  0,
 258                                 .profile    =  0,
 259                                 .name       =  offset_chars,
 260                                 .file       =  0,
 261                                 .nargs      =  0,
 262                                 .argsize    = {0}
 263                         });
 264                 });
 265
 266                 /* if we make it this far then we have statements */
 267                 {
 268                         size_t i = 0;    /* counter   */
 269                         size_t o = 0;    /* operands  */
 270                         char  *t = NULL; /* token     */
 271
 272                         /*
 273                          * Most ops a single statement can have is three.
 274                          * lets allocate some space for all of those here.
 275                          */
 276                         char op[3][32768] = {{0},{0},{0}};
 277                         for (; i < sizeof(asm_instr)/sizeof(*asm_instr); i++) {
 278                                 if (!strncmp(skip, asm_instr[i].m, asm_instr[i].l)) {
 279                                         if (state != ASM_FUNCTION) {
 280                                                 printf("%li: Statement not inside function block\n", line);
 281                                                 goto end;
 282                                         }
 283
 284                                         /* update parser state */
 285                                         if (i == INSTR_DONE || i == INSTR_RETURN) {
 286                                                 goto end;
 287                                                 state = ASM_NULL;
 288                                         }
 289
 290                                         /* parse the statement */
 291                                         o     = asm_instr[i].o; /* operands         */
 292                                         skip += asm_instr[i].l; /* skip instruction */
 293                                         t     = strtok(skip, " ,");
 294                                         i     = 0;
 295                                         while (t != NULL && i < 3) {
 296                                                 strcpy(op[i], t);
 297                                                 t = strtok(NULL, " ,");
 298                                                 i ++;
 299                                         }
 300
 301                                         util_debug("ASM", "Operand 1: %s\n", util_strrnl(op[0]));
 302                                         util_debug("ASM", "Operand 2: %s\n", util_strrnl(op[1]));
 303                                         util_debug("ASM", "Operand 3: %s\n", util_strrnl(op[2]));
 304
 305                                         /* check */
 306                                         if (i != o) {
 307                                                 printf("not enough operands, expected: %li, got %li\n", o, i);
 308                                         }
 309
 310                                         /* TODO: hashtable value LOAD .... etc */
 311                                         code_statements_add((prog_section_statement){
 312                                                 i, {0}, {0}, {0}
 313                                         });
 314                                         goto end;
 315                                 }
 316                         }
 317                 }
 318
 319                 /* if we made it this far something is wrong */
 320                 if (*skip != '\0')
 321                         printf("%li: Invalid statement, expression, or decleration\n", line);
 322
 323                 end:
 324                 //free(data);
 325                 mem_d(data);
 326                 line ++;
 327         }
 328 }