3 files changed, 157 insertions, 62 deletions

diff --git a/src/symboltable.c b/src/symboltable.c
index cdbd442..5c63d3f 100644
--- a/src/symboltable.c
+++ b/src/symboltable.c
@@ -2,28 +2,30 @@
 
 extern int lineno; /* from lexical analyser */
 
-SymbolTable glo_symbol_table = {{{{0},0}},MAXSYMBOLS,0};
-SymbolTable loc_symbol_table = {{{{0},0}},MAXSYMBOLS,0};
+SymbolTable glo_symbol_table = {{{{0}, 0}}, MAXSYMBOLS, 0};
+SymbolTable loc_symbol_table = {{{{0}, 0}}, MAXSYMBOLS, 0};
 
 void glo_addVar(const char name[], int type) {
   int count;
   for (count = 0; count < glo_symbol_table.size; count++) {
     if (!strcmp(glo_symbol_table.entries[count].name, name)) {
-      fprintf(stderr,"semantic error, redefinition of variable %s near line %d\n", name,
-             lineno);
+      fprintf(stderr,
+              "semantic error, redefinition of variable %s near line %d\n",
+              name, lineno);
       return;
     }
   }
   if (++glo_symbol_table.size > glo_symbol_table.maxsize) {
-    fprintf(stderr,"too many variables near line %d\n", lineno);
+    fprintf(stderr, "too many variables near line %d\n", lineno);
     exit(1);
   }
   strcpy(glo_symbol_table.entries[glo_symbol_table.size - 1].name, name);
   glo_symbol_table.entries[glo_symbol_table.size - 1].type = type;
-  glo_symbol_table.entries[glo_symbol_table.size - 1].addr = (glo_symbol_table.size - 1)*8;
+  glo_symbol_table.entries[glo_symbol_table.size - 1].addr =
+      (glo_symbol_table.size - 1) * 8;
 }
 
-//Verifies that the variable exists and returns the type
+// Verifies that the variable exists and returns the type
 int glo_lookup(const char name[]) {
   int count;
 
@@ -32,43 +34,58 @@ int glo_lookup(const char name[]) {
       return glo_symbol_table.entries[count].type;
     }
   }
-  fprintf(stderr,"No definition of the variable %s near line %d\n", name,
-           lineno);
+  fprintf(stderr, "No definition of the variable %s near line %d\n", name,
+          lineno);
   return -1;
 }
+int glo_get_addr(const char name[]) {
+  int count;
 
-
-void glo_display_table(){
-    int count;
-    for (count=0;count<glo_symbol_table.size;count++) {
-        if(glo_symbol_table.entries[count].type == INT)
-            printf("entier: %s, pos: %d     \n", glo_symbol_table.entries[count].name, glo_symbol_table.entries[count].addr);
-        else
-            printf("caractere: %s, pos: %d       \n", glo_symbol_table.entries[count].name, glo_symbol_table.entries[count].addr);
+  for (count = 0; count < glo_symbol_table.size; count++) {
+    if (!strcmp(glo_symbol_table.entries[count].name, name)) {
+      return glo_symbol_table.entries[count].addr;
     }
-    printf("\n");
+  }
+  fprintf(stderr, "No definition of the variable %s near line %d\n", name,
+          lineno);
+  return -1;
 }
 
+void glo_display_table() {
+  int count;
+  for (count = 0; count < glo_symbol_table.size; count++) {
+    if (glo_symbol_table.entries[count].type == INT)
+      printf(";entier: %s, pos: %d     \n",
+             glo_symbol_table.entries[count].name,
+             glo_symbol_table.entries[count].addr);
+    else
+      printf(";caractere: %s, pos: %d       \n",
+             glo_symbol_table.entries[count].name,
+             glo_symbol_table.entries[count].addr);
+  }
+  printf("\n");
+}
 
 void loc_addVar(const char name[], int type) {
   int count;
   for (count = 0; count < loc_symbol_table.size; count++) {
     if (!strcmp(loc_symbol_table.entries[count].name, name)) {
-      fprintf(stderr,"semantic error, redefinition of variable %s near line %d\n", name,
-             lineno);
+      fprintf(stderr,
+              "semantic error, redefinition of variable %s near line %d\n",
+              name, lineno);
       return;
     }
   }
-  if (++glo_symbol_table.size > glo_symbol_table.maxsize) {
-    fprintf(stderr,"too many variables near line %d\n", lineno);
+  if (++loc_symbol_table.size > loc_symbol_table.maxsize) {
+    fprintf(stderr, "too many variables near line %d\n", lineno);
     exit(1);
   }
-  strcpy(glo_symbol_table.entries[glo_symbol_table.size - 1].name, name);
-  glo_symbol_table.entries[glo_symbol_table.size - 1].type = type;
-  glo_symbol_table.entries[glo_symbol_table.size - 1].addr = (glo_symbol_table.size - 1)*8;
+  strcpy(loc_symbol_table.entries[loc_symbol_table.size - 1].name, name);
+  loc_symbol_table.entries[loc_symbol_table.size - 1].type = type;
+  loc_symbol_table.entries[loc_symbol_table.size - 1].addr =
+      (loc_symbol_table.size - 1) * 8;
 }
 
-
 int loc_lookup(const char name[]) {
   int count;
 
@@ -77,17 +94,52 @@ int loc_lookup(const char name[]) {
       return loc_symbol_table.entries[count].type;
     }
   }
-  fprintf(stderr,"No definition of the variable %s near line %d\n", name,
-           lineno);
+  fprintf(stderr, "No definition of the variable %s near line %d\n", name,
+          lineno);
   return -1;
 }
-void loc_display_table(){
-    int count;
-    for (count=0;count<loc_symbol_table.size;count++) {
-        if(loc_symbol_table.entries[count].type == INT)
-            printf("entier: %s, pos: %d     \n", loc_symbol_table.entries[count].name, loc_symbol_table.entries[count].addr);
-        else
-            printf("caractere: %s, pos: %d       \n", loc_symbol_table.entries[count].name, loc_symbol_table.entries[count].addr);
+
+int loc_get_addr(const char name[]) {
+  int count;
+
+  for (count = 0; count < loc_symbol_table.size; count++) {
+    if (!strcmp(loc_symbol_table.entries[count].name, name)) {
+      return loc_symbol_table.entries[count].addr;
     }
-    printf("\n");
+  }
+  fprintf(stderr, "No definition of the variable %s near line %d\n", name,
+          lineno);
+  return -1;
+}
+void loc_display_table() {
+  int count;
+  for (count = 0; count < loc_symbol_table.size; count++) {
+    if (loc_symbol_table.entries[count].type == INT)
+      printf(";entier: %s, pos: %d     \n",
+             loc_symbol_table.entries[count].name,
+             loc_symbol_table.entries[count].addr);
+    else
+      printf(";caractere: %s, pos: %d       \n",
+             loc_symbol_table.entries[count].name,
+             loc_symbol_table.entries[count].addr);
+  }
+  printf("\n");
+}
+
+void loc_clean_table() {
+  int i;
+  for (i = 0; i < loc_symbol_table.size; i++) {
+    printf("pop eax\n");
+  }
+  loc_symbol_table.size = 0;
+}
+
+void check_expected_type(int type_to_check, int type_expected) {
+  if (type_to_check != type_expected)
+    fprintf(stderr, "Expected type : %s -> Got type : %s (near line %d)\n",
+            type_expected == INT ? "INT" : type_expected == CHAR ? "CHAR"
+                                                                 : "UNDEFINED",
+            type_to_check == INT ? "INT" : type_to_check == CHAR ? "CHAR"
+                                                                 : "UNDEFINED",
+            lineno);
 }
\ No newline at end of file
diff --git a/src/symboltable.h b/src/symboltable.h
index bd7a373..ac6db36 100644
--- a/src/symboltable.h
+++ b/src/symboltable.h
@@ -26,8 +26,11 @@ typedef struct {
 
 void glo_addVar(const char name[], int type);
 int glo_lookup(const char name[]);
+int glo_get_addr(const char name[]);
 void glo_display_table();
 void loc_addVar(const char name[], int type);
 int loc_lookup(const char name[]);
+int loc_get_addr(const char name[]);
 void loc_display_table();
+void check_expected_type(int type_to_check, int type_expected);
 #endif
\ No newline at end of file
diff --git a/src/tpc.y b/src/tpc.y
index c575863..387b3ef 100644
--- a/src/tpc.y
+++ b/src/tpc.y
@@ -10,10 +10,10 @@
 extern int lineno;
 int yylex();
 void yyerror(char *);
-
 #define GLOBAL 0
 #define LOCAL 1
 static int status = GLOBAL;
+
 %}
 
 %union {
@@ -44,9 +44,9 @@ static int status = GLOBAL;
 %%
 Prog:{printf("extern printf\n");
                         printf("section .data\n");
-                        printf("format_int db \"%%d \\n\",10,0\n\n");
+                        printf("format_int db \"%%d\",10,0\n\n");
                         printf("section .bss\nsection .text\n\nglobal _start\n");
-                        printf("print: ;print needs an argument in rax\n\n");
+                        printf("print: ;print needs an argument in rax\n");
                         printf("push rbp\n");
                         printf("mov rbp, rsp\n");