Parse target triples

GaloisInc · Sep 28, 2022 · e2dfe38 · e2dfe38
1 parent d605c01
commit e2dfe38
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/src/Data/LLVM/BitCode/IR/Module.hs b/src/Data/LLVM/BitCode/IR/Module.hs
@@ -16,6 +16,8 @@ import Data.LLVM.BitCode.Match
 import Data.LLVM.BitCode.Parse
 import Data.LLVM.BitCode.Record
 import Text.LLVM.AST
+import Text.LLVM.Triple.AST (TargetTriple)
+import Text.LLVM.Triple.Parse (parseTriple)
 
 import qualified Codec.Binary.UTF8.String as UTF8 (decode)
 import Control.Monad (foldM,guard,when,forM_)
@@ -34,6 +36,7 @@ data PartialModule = PartialModule
   , partialGlobals    :: GlobalList
   , partialDefines    :: DefineList
   , partialDeclares   :: DeclareList
+  , partialTriple     :: TargetTriple
   , partialDataLayout :: DataLayout
   , partialInlineAsm  :: InlineAsm
   , partialComdat     :: !(Seq (String,SelectionKind))
@@ -51,6 +54,7 @@ emptyPartialModule  = PartialModule
   , partialGlobals    = mempty
   , partialDefines    = mempty
   , partialDeclares   = mempty
+  , partialTriple     = mempty
   , partialDataLayout = mempty
   , partialInlineAsm  = mempty
   , partialAliasIx    = 0
@@ -75,6 +79,7 @@ finalizeModule pm = label "finalizeModule" $ do
   defines <- T.mapM (finalizePartialDefine lkp) (partialDefines pm)
   return emptyModule
     { modSourceName = partialSourceName pm
+    , modTriple     = partialTriple pm
     , modDataLayout = partialDataLayout pm
     , modNamedMd    = F.toList (partialNamedMd pm)
     , modUnnamedMd  = sortOn umIndex (F.toList unnamed)
@@ -167,9 +172,10 @@ parseModuleBlockEntry pm (valueSymtabBlockId -> Just _es) = do
   -- MODULE_BLOCK
   return pm
 
-parseModuleBlockEntry pm (moduleCodeTriple -> Just _) = do
+parseModuleBlockEntry pm (moduleCodeTriple -> Just r) = do
   -- MODULE_CODE_TRIPLE
-  return pm
+  triple <- UTF8.decode <$> parseFields r 0 char
+  return (pm { partialTriple = parseTriple triple })
 
 parseModuleBlockEntry pm (moduleCodeDatalayout -> Just r) = do
   -- MODULE_CODE_DATALAYOUT

diff --git a/unit-test/Tests/Instances.hs b/unit-test/Tests/Instances.hs
@@ -9,10 +9,19 @@ import Test.QuickCheck.Arbitrary (Arbitrary(..))
 
 import Data.LLVM.Internal
 import Text.LLVM.AST
+import qualified Text.LLVM.Triple.AST as Triple
 
 -------------------------------------------------------------------------
 -- ** llvm-pretty
 
+instance Arbitrary Triple.Arch where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.Environment where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.OS where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.ObjectFormat where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.SubArch where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.TargetTriple where arbitrary = genericArbitrary uniform
+instance Arbitrary Triple.Vendor where arbitrary = genericArbitrary uniform
+
 instance Arbitrary Module                                             where arbitrary = genericArbitrary uniform
 instance Arbitrary NamedMd                                            where arbitrary = genericArbitrary uniform
 instance Arbitrary UnnamedMd                                          where arbitrary = genericArbitrary uniform