Fix personality function handling.

Author: maleadt

src/core/function.jl

@@ -12,8 +12,12 @@ Function(mod::Module, name::String, ft::FunctionType) =
 
 unsafe_delete!(::Module, f::Function) = API.LLVMDeleteFunction(ref(f))
 
-personality(f::Function) = Function(API.LLVMGetPersonalityFn(ref(f)))
-personality!(f::Function, persfn::Function) = API.LLVMSetPersonalityFn(ref(f), ref(persfn))
+function personality(f::Function)
+    has_personality = convert(Core.Bool, API.LLVMHasPersonalityFn(ref(f)))
+    return has_personality ? Function(API.LLVMGetPersonalityFn(ref(f))) : nothing
+end
+personality!(f::Function, persfn::Union{Nothing,Function}) =
+    API.LLVMSetPersonalityFn(ref(f), persfn===nothing ? C_NULL : ref(persfn))
 
 intrinsic_id(f::Function) = API.LLVMGetIntrinsicID(ref(f))
 

test/core.jl

@@ -669,7 +669,14 @@ LLVM.Module("SomeModule", ctx) do mod
 
     show(devnull, fn)
 
-    # @show personality(fn)
+    @test personality(fn) === nothing
+    pers_ft = LLVM.FunctionType(LLVM.Int32Type(ctx); vararg=true)
+    pers_fn = LLVM.Function(mod, "PersonalityFunction", ft)
+    personality!(fn, pers_fn)
+    @test personality(fn) == pers_fn
+    personality!(fn, nothing)
+    @test personality(fn) === nothing
+    unsafe_delete!(mod, pers_fn)
 
     @test intrinsic_id(fn) == 0