Added support for passing arguments to lambda_wrapper and added corou…

.github/workflows/ci-all-samples.yml

@@ -9,6 +9,7 @@ jobs:
         uses: actions/checkout@v2
         with:
           submodules: recursive
+      - run: sudo apt-get install g++
       - run: make -C ${{ github.workspace }} -j$(nproc) run
       - name: Install libunwind-dev
         run: sudo apt-get install libunwind-dev libdwarf-dev=20200114-1 libdwarf1=20200114-1

Makefile

@@ -10,6 +10,8 @@ all: executables
 
 CHECK_CONFIG=1
 CONFIG_STR=DEBUG=$(DEBUG) RECOVER_VAR_NAMES=$(RECOVER_VAR_NAMES) TRACER_USE_LIBUNWIND=$(TRACER_USE_LIBUNWIND)
+CONFIG_STR+=EXTRA_CFLAGS=$(EXTRA_CFLAGS)
+
 
 # Create a scratch directory where the files are stored
 $(shell mkdir -p $(BASE_DIR)/scratch)

include/builder/builder_context.h

@@ -37,17 +37,18 @@ class tag_map {
 	std::unordered_map<std::string, block::stmt_block::Ptr> map;
 };
 
-void lambda_wrapper(void);
+void lambda_wrapper(std::function<void(void)>);
 void lambda_wrapper_close(void);
-void lambda_wrapper_impl(void);
+
+void coroutine_wrapper(std::function<void(void)>);
+void coroutine_wrapper_close(void);
 
 class builder_context {
 public:
 	static builder_context *current_builder_context;
 	static int debug_creation_counter;
 
 	std::function<void(void)> internal_stored_lambda;
-	std::function<void(void)> current_function;
 
 	std::list<block::block::Ptr> uncommitted_sequence;
 	block::stmt::Ptr ast;

make/setvars.mk

@@ -9,10 +9,11 @@ endif
 DEBUG=1
 endif
 
+EXTRA_CFLAGS?=
 
 # Create CFLAGS, LINKER_FLAGS, CFLAGS_INTERNAL and INCLUDE_FLAGS based on config
 CFLAGS_INTERNAL=-std=c++11 -Wall -Wextra -Wno-unused-parameter -Wno-missing-field-initializers -Wmissing-declarations 
-CFLAGS_INTERNAL+=-Woverloaded-virtual -Wno-deprecated -Wdelete-non-virtual-dtor -Werror -Wno-vla -pedantic-errors
+CFLAGS_INTERNAL+=-Woverloaded-virtual -Wno-deprecated -Wdelete-non-virtual-dtor -Werror -Wno-vla -pedantic-errors 
 CFLAGS=
 LINKER_FLAGS=-L$(BUILD_DIR)/ -l$(LIBRARY_NAME)
 INCLUDE_FLAGS=-I$(INCLUDE_DIR) -I$(BUILD_DIR)/gen_headers/
@@ -49,4 +50,5 @@ endif
 
 
 LINKER_FLAGS+=-ldl
+CFLAGS+=$(EXTRA_CFLAGS)
 # --- flags are all ready

src/builder/builder_context.cpp

@@ -328,7 +328,7 @@ block::stmt::Ptr builder_context::extract_ast_from_function_internal(std::vector
 	try {
 		current_builder_context = this;
 		// function();
-		lambda_wrapper();
+		lambda_wrapper(internal_stored_lambda);
 		commit_uncommitted();
 		ret_ast = ast;
 		current_builder_context = nullptr;
@@ -463,8 +463,5 @@ block::stmt::Ptr builder_context::extract_ast_from_function_internal(std::vector
 	return ret_ast;
 }
 
-void lambda_wrapper_impl(void) {
-	builder_context::current_builder_context->internal_stored_lambda();
-}
 
 } // namespace builder

src/builder/builder_context_support.cpp

@@ -1,14 +1,22 @@
-
+#include <functional>
 namespace builder {
-void lambda_wrapper_impl(void);
-void lambda_wrapper(void);
+void lambda_wrapper(std::function<void(void)>);
 void lambda_wrapper_close(void);
+void coroutine_wrapper(std::function<void(void)>);
+void coroutine_wrapper_close(void);
 
 int tail_call_guard;
 
-void lambda_wrapper(void) {
-	lambda_wrapper_impl();
+void lambda_wrapper(std::function<void(void)> f) {
+	f();
 	tail_call_guard += 1;
 }
 void lambda_wrapper_close(void) {}
+
+
+void coroutine_wrapper(std::function<void(void)> f) {
+	f();
+	tail_call_guard +=1;
+}
+void coroutine_wrapper_close(void) {}
 } // namespace builder

src/util/tracer.cpp

@@ -8,7 +8,7 @@
 #endif
 
 namespace builder {
-extern void lambda_wrapper(void);
+extern void lambda_wrapper(std::function<void(void)>);
 extern void lambda_wrapper_close(void);
 } // namespace builder
 namespace tracer {