cc: analyzer: store function return type

atirut-w · Jul 29, 2024 · 0f5b771 · 0f5b771
1 parent b4de0bd
commit 0f5b771
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/cc/analyzer.cpp b/src/cc/analyzer.cpp
@@ -1,4 +1,5 @@
 #include "CParser.h"
+#include "types.hpp"
 #include <analyzer.hpp>
 #include <any>
 
@@ -14,12 +15,14 @@ any Analyzer::visitCompilationUnit(CParser::CompilationUnitContext *ctx)
 any Analyzer::visitFunctionDefinition(CParser::FunctionDefinitionContext *ctx)
 {
     string name = ctx->declarator()->directDeclarator()->directDeclarator()->Identifier()->getText();
+    string return_type = ctx->declarationSpecifiers()->declarationSpecifier()[0]->typeSpecifier()->getText();
 
     if (meta.functions.find(name) == meta.functions.end())
     {
         meta.functions[name] = FunctionMeta();
     }
     current_function = &meta.functions[name];
+    current_function->return_type = &primitives[return_type];
 
     if (auto *itemlist_ctx = ctx->compoundStatement()->blockItemList())
     {

diff --git a/src/cc/include/analyzer.hpp b/src/cc/include/analyzer.hpp
@@ -2,6 +2,7 @@
 #include <CBaseVisitor.h>
 #include <map>
 #include <string>
+#include <types.hpp>
 
 struct SymbolMeta
 {
@@ -17,6 +18,7 @@ struct LocalMeta
 
 struct FunctionMeta
 {
+    Type *return_type;
     // Offset into the local frame for local variables
     std::map<std::string, LocalMeta> variables;
     int local_alloc = 0;