2 files changed, 45 insertions, 0 deletions
diff --git a/include/cps/CodeGen.h b/include/cps/CodeGen.h
index 8f84004..94aacb7 100644
--- a/include/cps/CodeGen.h
+++ b/include/cps/CodeGen.h
@@ -22,6 +22,7 @@ namespace cps {
 
 class ArrayHandler;
 class FileHandler;
+class MemoryHandler;
 class ExceptionHandler;
 class CastExprAST;
 
@@ -52,6 +53,7 @@ class CodeGen {
     std::unique_ptr<StringHandler> StrHandler;
     std::unique_ptr<StringConversionHandler> StrConvHandler;
     std::unique_ptr<FileHandler> Files;
+    std::unique_ptr<MemoryHandler> Memory;
 
     llvm::FunctionCallee PrintfFunc;
     llvm::FunctionCallee ScanfFunc;
@@ -106,6 +108,7 @@ class CodeGen {
     bool emitLValue(ExprAST *E, LValueInfo &Out);
     // The rvalue read: load + the STRING null->"" guard.
     llvm::Value *loadFromLValue(const LValueInfo &LV);
+    llvm::Value *emitAllocationSize(ExprAST *Expr);
     llvm::Value *emitCall(const std::string &Callee,
                           const std::vector<std::unique_ptr<ExprAST>> &ArgExprs, int Line);
     bool marshalCallArgs(const std::string &Callee,
diff --git a/include/cps/MemoryHandler.h b/include/cps/MemoryHandler.h
new file mode 100644
index 0000000..fd2c93e
--- /dev/null
+++ b/include/cps/MemoryHandler.h
@@ -0,0 +1,42 @@
+#pragma once
+#include "llvm/IR/IRBuilder.h"
+#include "llvm/IR/Module.h"
+
+namespace cps {
+
+class TypeSystem;
+class RuntimeCheck;
+class ExceptionHandler;
+
+// Only explicit memory builtins own entries in this runtime's allocation list.
+// Internal STRING/ARRAY storage retains its existing lifetime rules.
+class MemoryHandler {
+    llvm::LLVMContext &Context;
+    llvm::IRBuilder<> &Builder;
+    llvm::Module &Module;
+    TypeSystem &Types;
+    RuntimeCheck &Checker;
+    ExceptionHandler &Exceptions;
+    llvm::IntegerType *SizeTy = nullptr;
+    llvm::StructType *NodeTy = nullptr;
+    llvm::GlobalVariable *Head = nullptr;
+    llvm::FunctionCallee Malloc, Free;
+    llvm::Function *AllocateFn = nullptr, *ReleaseFn = nullptr, *FindFn = nullptr;
+    llvm::Value *AllocationError = nullptr, *TrackingError = nullptr, *InvalidAddress = nullptr;
+
+    void ensureRuntime();
+    llvm::Function *createHelper(const char *Name, llvm::ArrayRef<llvm::Type *> Args);
+    void buildFind();
+    void buildAllocate();
+    void buildRelease();
+
+public:
+    MemoryHandler(llvm::LLVMContext &C, llvm::IRBuilder<> &B, llvm::Module &M,
+                  TypeSystem &TS, RuntimeCheck &RC, ExceptionHandler &EH)
+        : Context(C), Builder(B), Module(M), Types(TS), Checker(RC), Exceptions(EH) {}
+
+    llvm::Value *emitAllocate(llvm::Value *Size, int Line);
+    llvm::Value *emitRelease(llvm::Value *Address, int Line);
+};
+
+} // namespace cps