Module to manipulate function headers.

2 files changed, 79 insertions, 0 deletions

diff --git a/func.c b/func.c
new file mode 100644
index 00000000..601513d1
--- /dev/null
+++ b/func.c
@@ -0,0 +1,59 @@
+#include <stdio.h>
+#include "table.h"
+#include "mem.h"
+#include "func.h"
+
+static TFunc *function_root = NULL;
+
+
+/*
+** Insert function in list for GC
+*/
+void luaI_insertfunction (TFunc *f)
+{
+  lua_pack();
+  f->next = function_root;
+  function_root = f;
+  f->marked = 0;
+}
+
+
+/*
+** Free function
+*/
+static void freefunc (TFunc *f)
+{
+  luaI_free (f->code);
+  luaI_free (f);
+}
+
+/*
+** Garbage collection function.
+** This function traverse the function list freeing unindexed functions
+*/
+Long luaI_funccollector (void)
+{
+  TFunc *curr = function_root;
+  TFunc *prev = NULL;
+  Long counter = 0;
+  while (curr)
+  {
+    TFunc *next = curr->next;
+    if (!curr->marked)
+    {
+      if (prev == NULL)
+        function_root = next;
+      else
+        prev->next = next;
+      freefunc (curr);
+      ++counter;
+    }
+    else
+    {
+      curr->marked = 0;
+      prev = curr;
+    } 
+    curr = next;
+  }
+  return counter;
+}
diff --git a/func.h b/func.h
new file mode 100644
index 00000000..8d2d124d
--- /dev/null
+++ b/func.h
@@ -0,0 +1,20 @@
+#ifndef func_h
+#define func_h
+
+#include "types.h"
+
+/*
+** Header para funcoes.
+*/
+typedef struct TFunc
+{
+  struct TFunc  *next;
+  char          marked;
+  int           size;
+  Byte          *code;
+} TFunc;
+
+Long luaI_funccollector (void);
+void luaI_insertfunction (TFunc *f);
+
+#endif