Resolve symbol types in symbol table

Author: danpoe

src/json-symtab-language/json_symtab_language.cpp

@@ -11,6 +11,7 @@ Author: Chris Smowton, [email protected]
 #include "json_symtab_language.h"
 #include <json/json_parser.h>
 #include <util/json_symbol_table.h>
+#include <util/namespace.h>
 
 bool json_symtab_languaget::parse(
   std::istream &instream,
@@ -30,6 +31,7 @@ bool json_symtab_languaget::typecheck(
   try
   {
     symbol_table_from_json(parsed_json_file, symbol_table);
+    follow_type_symbols(symbol_table);
     return false;
   }
   catch(const std::string &str)
@@ -39,6 +41,57 @@ bool json_symtab_languaget::typecheck(
   }
 }
 
+void json_symtab_languaget::follow_type_symbols(
+  irept &irep,
+  const namespacet &ns)
+{
+  ns.follow_type_symbol(irep);
+
+  for(irept &sub : irep.get_sub())
+  {
+    follow_type_symbols(sub, ns);
+  }
+
+  for(auto &entry : irep.get_named_sub())
+  {
+    irept &sub = entry.second;
+
+    follow_type_symbols(sub, ns);
+  }
+}
+
+void json_symtab_languaget::follow_type_symbols(symbol_tablet &symbol_table)
+{
+  const namespacet ns(symbol_table);
+
+  typedef symbol_tablet::symbolst symbolst;
+  symbolst &symbols = symbol_table.symbols;
+
+  for(auto &entry : symbols)
+  {
+    symbolt &symbol = entry.second;
+
+    // Modify entries in place
+    follow_type_symbols(symbol.type, ns);
+    follow_type_symbols(symbol.value, ns);
+  }
+
+  // Remove type entries from the symbol table
+  for(symbolst::iterator it = symbols.begin(); it != symbols.end();)
+  {
+    const symbolt &symbol = it->second;
+
+    if(symbol.is_type)
+    {
+      it = symbols.erase(it);
+    }
+    else
+    {
+      it++;
+    }
+  }
+}
+
 void json_symtab_languaget::show_parse(std::ostream &out)
 {
   parsed_json_file.output(out);

src/json-symtab-language/json_symtab_language.h

@@ -60,6 +60,9 @@ class json_symtab_languaget:public languaget
   }
 
  protected:
+  void follow_type_symbols(symbol_tablet &symbol_table);
+  void follow_type_symbols(irept &irep, const namespacet &ns);
+
   jsont parsed_json_file;
 };
 

src/util/namespace.cpp

@@ -67,6 +67,23 @@ void namespace_baset::follow_symbol(irept &irep) const
   }
 }
 
+void namespace_baset::follow_type_symbol(irept &irep) const
+{
+  while(irep.id() == ID_symbol)
+  {
+    const symbolt &symbol = lookup(irep);
+
+    if(symbol.is_type && !symbol.type.is_nil())
+    {
+      irep = symbol.type;
+    }
+    else
+    {
+      break;
+    }
+  }
+}
+
 const typet &namespace_baset::follow(const typet &src) const
 {
   if(src.id()!=ID_symbol)

src/util/namespace.h

@@ -43,6 +43,7 @@ class namespace_baset
   virtual ~namespace_baset();
 
   void follow_symbol(irept &irep) const;
+  void follow_type_symbol(irept &irep) const;
   void follow_macros(exprt &expr) const;
   const typet &follow(const typet &src) const;