1 files changed, 25 insertions, 3 deletions
diff --git a/include/program.h b/include/program.h
index 6603c63..543b38d 100644
--- a/include/program.h
+++ b/include/program.h
@@ -2,9 +2,12 @@
 #define AMAL_PROGRAM_H
 
 #include "std/buffer.h"
+#include "std/hash_map.h"
+#include "std/arena_allocator.h"
 #include "bytecode/bytecode.h"
 #include "../executor/executor.h"
 
+/* TODO: Remove all these errors and move program decoding and execution to another process (crash sandbox) */
 #define AMAL_PROGRAM_OK 0
 #define AMAL_PROGRAM_ERR -1
 #define AMAL_PROGRAM_INVALID_HEADER -16
@@ -14,7 +17,7 @@
 #define AMAL_PROGRAM_INVALID_INTERMEDIATES_SIZE -5
 #define AMAL_PROGRAM_INVALID_STRINGS -6
 #define AMAL_PROGRAM_INVALID_STRINGS_SIZE -7
-#define AMAL_PROGRAM_STRING_ALLOC_FAILURE -8
+#define AMAL_PROGRAM_ALLOC_FAILURE -8
 #define AMAL_PROGRAM_INVALID_INSTRUCTIONS_SIZE -9
 #define AMAL_PROGRAM_INVALID_INSTRUCTION -10
 #define AMAL_PROGRAM_INSTRUCTION_INVALID_INTERMEDIATE_INDEX -11
@@ -23,6 +26,10 @@
 #define AMAL_PROGRAM_INSTRUCTION_STACK_OOM -14
 #define AMAL_PROGRAM_INSTRUCTION_ILLEGAL_JUMP_TARGET -15
 #define AMAL_PROGRAM_INVALID_FUNCTIONS -16
+#define AMAL_PROGRAM_INVALID_EXTERNAL_FUNCTIONS -17
+#define AMAL_PROGRAM_INVALID_EXTERNAL_FUNCTIONS_SIZE -18
+#define AMAL_PROGRAM_INSTRUCTION_INVALID_EXTERN_FUNC_INDEX -19
+#define AMAL_PROGRAM_NO_SUCH_EXTERNAL_FUNCTION -20
 
 #define AMAL_PROGRAM_MAGIC_NUMBER (u32)0xdec05eba
 #define AMAL_PROGRAM_MAJOR_VERSION 1
@@ -31,21 +38,36 @@
 
 #define AMAL_PROGRAM_NUM_REGISTERS 256
 
+#define AMAL_PROGRAM_ARGS_SIZE_VARARGS -1
+
+typedef struct {
+    void *func;
+    int args_byte_size; /* -1 if varargs (AMAL_PROGRAM_ARGS_SIZE_VARARGS) */
+} ProgramExternFunc;
+
 typedef struct {
     Buffer/*<...>*/ data;
     u32 *string_indices;
-    char *intermediates_start;  /* Reference inside @data */
-    char *strings_start;        /* Reference inside @data */
+    u32 *extern_func_indices;
+    u8 *intermediates_start;  /* Reference inside @data */
+    u8 *strings_start;        /* Reference inside @data */
+    u8 *extern_funcs_start;   /* Reference inside @data */
     usize read_index;
 
     u16 num_intermediates;
     u16 num_strings;
     u16 num_functions;
+    u16 num_extern_functions;
+
+    ArenaAllocator allocator; /* Owned. Used by @extern_funcs_map */
+    HashMapType(BufferView, ProgramExternFunc) extern_funcs_map;
 } amal_program;
 
 CHECK_RESULT int amal_program_init(amal_program *self);
 void amal_program_deinit(amal_program *self);
 
+CHECK_RESULT int amal_program_add_extern_func(amal_program *self, BufferView name, void *func_ptr, int args_byte_size);
+
 CHECK_RESULT int amal_program_append_bytecode(amal_program *self, Bytecode *bytecode);
 CHECK_RESULT int amal_program_run(amal_program *self);
 CHECK_RESULT int amal_program_save(amal_program *self, const char *filepath);