Merge pull request #80 from nagisa/builtins

Remind LLVM about alloc function names in Rust

Author: TimNN

include/llvm/Analysis/TargetLibraryInfo.def

@@ -316,6 +316,27 @@ TLI_DEFINE_STRING_INTERNAL("__powf_finite")
 /// long double __powl_finite(long double x, long double y);
 TLI_DEFINE_ENUM_INTERNAL(powl_finite)
 TLI_DEFINE_STRING_INTERNAL("__powl_finite")
+
+/// uint8_t *__rust_allocate(size_t size, size_t align)
+TLI_DEFINE_ENUM_INTERNAL(rust_allocate)
+TLI_DEFINE_STRING_INTERNAL("__rust_allocate")
+
+/// uint8_t *__rust_allocate_zeroed(size_t size, size_t align)
+TLI_DEFINE_ENUM_INTERNAL(rust_allocate_zeroed)
+TLI_DEFINE_STRING_INTERNAL("__rust_allocate_zeroed")
+
+/// void __rust_deallocate(uint8_t *ptr, size_t size, size_t align)
+TLI_DEFINE_ENUM_INTERNAL(rust_deallocate)
+TLI_DEFINE_STRING_INTERNAL("__rust_deallocate")
+
+/// uint8_t *__rust_reallocate(uint8_t *ptr, size_t oldsz, size_t newsz, size_t align)
+TLI_DEFINE_ENUM_INTERNAL(rust_reallocate)
+TLI_DEFINE_STRING_INTERNAL("__rust_reallocate")
+
+/// uint8_t *__rust_reallocate_inplace(uint8_t *ptr, size_t oldsz, size_t newsz, size_t align)
+TLI_DEFINE_ENUM_INTERNAL(rust_reallocate_inplace)
+TLI_DEFINE_STRING_INTERNAL("__rust_reallocate_inplace")
+
 /// double __sincospi_stret(double x);
 TLI_DEFINE_ENUM_INTERNAL(sincospi_stret)
 TLI_DEFINE_STRING_INTERNAL("__sincospi_stret")

lib/Analysis/MemoryBuiltins.cpp

@@ -74,7 +74,12 @@ static const std::pair<LibFunc, AllocFnsTy> AllocationFnData[] = {
   {LibFunc_realloc,             {ReallocLike, 2, 1,  -1}},
   {LibFunc_reallocf,            {ReallocLike, 2, 1,  -1}},
   {LibFunc_strdup,              {StrDupLike,  1, -1, -1}},
-  {LibFunc_strndup,             {StrDupLike,  2, 1,  -1}}
+  {LibFunc_strndup,             {StrDupLike,  2, 1,  -1}},
+
+  {LibFunc_rust_allocate,       {MallocLike,  2, 0,  -1}},
+  {LibFunc_rust_allocate_zeroed, {MallocLike,  2, 0,  -1}},
+  {LibFunc_rust_reallocate,     {ReallocLike,  4, 2,  -1}},
+  {LibFunc_rust_reallocate_inplace, {ReallocLike,  4, 2,  -1}}
   // TODO: Handle "int posix_memalign(void **, size_t, size_t)"
 };
 
@@ -370,6 +375,8 @@ const CallInst *llvm::isFreeCall(const Value *I, const TargetLibraryInfo *TLI) {
            TLIFn == LibFunc_msvc_delete_array_ptr32_nothrow || // delete[](void*, nothrow)
            TLIFn == LibFunc_msvc_delete_array_ptr64_nothrow)   // delete[](void*, nothrow)
     ExpectedNumParams = 2;
+  else if (TLIFn == LibFunc_rust_deallocate)
+    ExpectedNumParams = 3;
   else
     return nullptr;