1 files changed, 22 insertions, 3 deletions
diff --git a/lib/CodeGen/StringHandler.cc b/lib/CodeGen/StringHandler.cc
index e5971d0..56b5aea 100644
--- a/lib/CodeGen/StringHandler.cc
+++ b/lib/CodeGen/StringHandler.cc
@@ -16,9 +16,6 @@ void StringHandler::setupExternalFunctions() {
     FunctionType *MallocType = FunctionType::get(PointerType::getUnqual(Context), {Type::getInt64Ty(Context)}, false);
     MallocFunc = Module.getOrInsertFunction("malloc", MallocType);
 
-    FunctionType *FreeType = FunctionType::get(Type::getVoidTy(Context), {PointerType::getUnqual(Context)}, false);
-    FreeFunc = Module.getOrInsertFunction("free", FreeType);
-
     FunctionType *StrLenType = FunctionType::get(Type::getInt64Ty(Context), {PointerType::getUnqual(Context)}, false);
     StrLenFunc = Module.getOrInsertFunction("strlen", StrLenType);
 
@@ -26,6 +23,12 @@ void StringHandler::setupExternalFunctions() {
     FunctionType *MemCpyType = FunctionType::get(PointerType::getUnqual(Context), MemCpyArgs, false);
     MemCpyFunc = Module.getOrInsertFunction("memcpy", MemCpyType);
 
+    FunctionType *StrCpyType = FunctionType::get(PointerType::getUnqual(Context),
+                                                 {PointerType::getUnqual(Context), PointerType::getUnqual(Context)},
+                                                 false);
+    StrCpyFunc = Module.getOrInsertFunction("strcpy", StrCpyType);
+    StrCatFunc = Module.getOrInsertFunction("strcat", StrCpyType);
+
     FunctionType *CharCaseType = FunctionType::get(Type::getInt32Ty(Context), {Type::getInt32Ty(Context)}, false);
     ToUpperFunc = Module.getOrInsertFunction("toupper", CharCaseType);
     ToLowerFunc = Module.getOrInsertFunction("tolower", CharCaseType);
@@ -50,6 +53,22 @@ Value *StringHandler::emitLength(Value *Str) {
     return Builder.CreateCall(StrLenFunc, Str, "len");
 }
 
+Value *StringHandler::emitConcat(Value *LHS, Value *RHS) {
+    if (!LHS || !RHS) return nullptr;
+
+    Value *LLen = Builder.CreateCall(StrLenFunc, {LHS}, "llen");
+    Value *RLen = Builder.CreateCall(StrLenFunc, {RHS}, "rlen");
+    Value *TotalLen = Builder.CreateAdd(LLen, RLen, "totallen");
+    Value *AllocSize = Builder.CreateAdd(TotalLen, ConstantInt::get(Type::getInt64Ty(Context), 1), "allocsize");
+
+    Value *NewStr = Builder.CreateCall(MallocFunc, {AllocSize}, "concat_str");
+
+    Builder.CreateCall(StrCpyFunc, {NewStr, LHS});
+    Builder.CreateCall(StrCatFunc, {NewStr, RHS});
+
+    return NewStr;
+}
+
 Value *StringHandler::emitMid(Value *Str, Value *Start, Value *Len) {
     
     Value *FullLen = emitLength(Str);