repl: Mock basic initialization and usage of tower on mlir module c…

…reation.
trailofbits · Jun 20, 2024 · 7fc2184 · 7fc2184
1 parent 2c85be8
commit 7fc2184
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/include/vast/repl/state.hpp b/include/vast/repl/state.hpp
@@ -28,6 +28,11 @@ namespace vast::repl {
         // mlir module and context
         //
         mcontext_t &ctx;
+
+        //
+        // Tower related state
+        //
+        tw::location_info li;
         std::optional< tw::default_tower > tower;
 
         //
@@ -44,6 +49,9 @@ namespace vast::repl {
         // verbosity flags
         //
         bool verbose_pipeline = true;
+
+        void raise_tower(owning_module_ref mod);
+        vast_module current_module();
     };
 
 } // namespace vast::repl
diff --git a/tools/vast-repl/CMakeLists.txt b/tools/vast-repl/CMakeLists.txt
@@ -4,6 +4,7 @@ add_vast_executable(vast-repl
     codegen.cpp
     command.cpp
     config.cpp
+    state.cpp
 
     LINK_LIBS
       ${CLANG_LIBS}

diff --git a/tools/vast-repl/command.cpp b/tools/vast-repl/command.cpp
@@ -31,12 +31,9 @@ namespace cmd {
     }
 
     void check_and_emit_module(state_t &state) {
-        if (!state.tower) {
-            check_source(state);
-            auto mod    = codegen::emit_module(state.source.value(), state.ctx);
-            auto [t, _] = tw::default_tower::get(state.ctx, std::move(mod));
-            state.tower = std::move(t);
-        }
+        check_source(state);
+        auto mod = codegen::emit_module(state.source.value(), state.ctx);
+        state.raise_tower(std::move(mod));
     }
 
     //
@@ -77,7 +74,7 @@ namespace cmd {
 
     void show_module(state_t &state) {
         check_and_emit_module(state);
-        llvm::outs() << state.tower->top().mod << "\n";
+        llvm::outs() << state.current_module() << "\n";
     }
 
     void show_symbols(state_t &state) {