Support unnamed_addr for functions

llvm-hs-pure/src/LLVM/AST/Global.hs

@@ -52,6 +52,7 @@ data Global
         linkage :: L.Linkage,
         visibility :: V.Visibility,
         dllStorageClass :: Maybe DLL.StorageClass,
+        unnamedAddr :: Maybe UnnamedAddr,
         callingConvention :: CC.CallingConvention,
         returnAttributes :: [A.ParameterAttribute],
         returnType :: Type,
@@ -124,6 +125,7 @@ functionDefaults =
     linkage = L.External,
     visibility = V.Default,
     dllStorageClass = Nothing,
+    unnamedAddr = Nothing,
     callingConvention = CC.C,
     returnAttributes = [],
     returnType = error "function return type not defined",

llvm-hs/src/LLVM/Internal/Module.hs

@@ -338,12 +338,15 @@ createModuleFromAST context@(Context c) (A.Module moduleId sourceFileName dataLa
            setThreadLocalMode a' (A.G.threadLocalMode a)
            (liftIO . FFI.setAliasee a') =<< encodeM (A.G.aliasee a)
            return (FFI.upCast a')
-       (A.Function _ _ _ cc rAttrs resultType fName (args, isVarArgs) attrs _ _ _ gc prefix blocks personality metadata) -> do
+       fn@(A.Function _ _ _ _ cc rAttrs resultType fName (args, isVarArgs) attrs _ _ _ gc prefix blocks personality metadata) -> do
          typ <- encodeM $ A.FunctionType resultType [t | A.Parameter t _ _ <- args] isVarArgs
          f <- liftIO . withName fName $ \fName -> FFI.addFunction ffiMod fName typ
          defineGlobal fName f
          cc <- encodeM cc
-         liftIO $ FFI.setFunctionCallingConvention f cc
+         liftIO $ do
+           FFI.setFunctionCallingConvention f cc
+           hua <- encodeM (A.G.unnamedAddr fn)
+           FFI.setUnnamedAddr (FFI.upCast f) hua
          setFunctionAttributes f (AttributeList attrs rAttrs [pa | A.Parameter _ _ pa <- args])
          setPrefixData f prefix
          setSection f (A.G.section g)
@@ -487,6 +490,7 @@ decodeFunctions mod = do
           <$> getLinkage f
           <*> getVisibility f
           <*> getDLLStorageClass f
+          <*> (liftIO $ decodeM =<< FFI.getUnnamedAddr (FFI.upCast f))
           <*> (liftIO $ decodeM =<< FFI.getFunctionCallingConvention f)
           <*> return rAttrs
           <*> return returnType

llvm-hs/test/LLVM/Test/Global.hs

@@ -43,5 +43,29 @@ tests = testGroup "Global" [
           gn (G.Function {}) = "function"
           gn (G.GlobalVariable {}) = "variable"
           name = gn g <> ", align " <> show a <> (maybe "" (("  section " <>) . ByteString.unpack . fromShort) s)
+   ],
+   testGroup "UnnamedAddr" [
+    testCase name $ withContext $ \context -> do
+      let ast = Module "<string>" "<string>" Nothing Nothing [ GlobalDefinition g ]
+      ast' <- withModuleFromAST context ast moduleAST
+      ast' @?= ast
+    | a <- [Nothing, Just LocalAddr, Just GlobalAddr],
+      g <- [
+       globalVariableDefaults {
+        G.name = UnName 0,
+        G.type' = i32,
+        G.unnamedAddr = a
+        },
+       functionDefaults {
+         G.returnType = A.T.void,
+         G.name = UnName 0,
+         G.parameters = ([], False),
+         G.unnamedAddr = a
+       }
+       ],
+      let
+          gn (G.Function {}) = "function"
+          gn (G.GlobalVariable {}) = "variable"
+          name = gn g <> ", unnamedAddr " <> show a
    ]
  ]

llvm-hs/test/LLVM/Test/Optimization.hs

@@ -119,6 +119,7 @@ tests = testGroup "Optimization" [
          G.name = Name "foo",
          G.parameters = ([Parameter i32 (Name "x") []], False),
          G.functionAttributes = [Left (A.GroupID 0)],
+         G.unnamedAddr = Just LocalAddr,
          G.basicBlocks = [
            BasicBlock (Name "here") [
               ] (