Support DIArgLists (added in LLVM 13) (GaloisInc#103)

A part of a fix for GaloisInc/llvm-pretty-bc-parser#174.
Ptival · Mar 21, 2023 · 4318862 · 4318862
1 parent a272e38
commit 4318862
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/src/Text/LLVM/AST.hs b/src/Text/LLVM/AST.hs
@@ -1130,6 +1130,7 @@ data DebugInfo' lab
   | DebugInfoTemplateValueParameter (DITemplateValueParameter' lab)
   | DebugInfoImportedEntity (DIImportedEntity' lab)
   | DebugInfoLabel (DILabel' lab)
+  | DebugInfoArgList (DIArgList' lab)
     deriving (Data, Eq, Functor, Generic, Generic1, Ord, Show, Typeable)
 
 type DebugInfo = DebugInfo' BlockLabel
@@ -1363,6 +1364,13 @@ data DISubroutineType' lab = DISubroutineType
 
 type DISubroutineType = DISubroutineType' BlockLabel
 
+-- | See <https://releases.llvm.org/13.0.0/docs/LangRef.html#diarglist>.
+newtype DIArgList' lab = DIArgList
+  { dialArgs :: [ValMd' lab]
+  } deriving (Data, Eq, Functor, Generic, Generic1, Ord, Show, Typeable)
+
+type DIArgList = DIArgList' BlockLabel
+
 -- Aggregate Utilities ---------------------------------------------------------
 
 data IndexResult

diff --git a/src/Text/LLVM/Labels.hs b/src/Text/LLVM/Labels.hs
@@ -139,6 +139,7 @@ instance HasLabel DINameSpace'                where relabel = $(generateRelabel
 instance HasLabel DITemplateTypeParameter'    where relabel = $(generateRelabel 'relabel ''DITemplateTypeParameter')
 instance HasLabel DITemplateValueParameter'   where relabel = $(generateRelabel 'relabel ''DITemplateValueParameter')
 instance HasLabel DIImportedEntity'           where relabel = $(generateRelabel 'relabel ''DIImportedEntity')
+instance HasLabel DIArgList'                  where relabel = $(generateRelabel 'relabel ''DIArgList')
 
 -- | Clever instance that actually uses the block name
 instance HasLabel ConstExpr' where

diff --git a/src/Text/LLVM/Lens.hs b/src/Text/LLVM/Lens.hs
@@ -43,6 +43,7 @@ concat <$> mapM (makeLensesWith (lensRules & lensField .~ (\_ _ n -> [TopName $
     , ''DIDerivedType'
     , ''DILexicalBlock'
     , ''DILexicalBlockFile'
+    , ''DIArgList'
     , ''Instr'
     , ''ValMd'
     , ''ConvOp

diff --git a/src/Text/LLVM/PP.hs b/src/Text/LLVM/PP.hs
@@ -899,6 +899,7 @@ ppDebugInfo' pp di = case di of
   DebugInfoTemplateValueParameter dtvp -> ppDITemplateValueParameter' pp dtvp
   DebugInfoImportedEntity diip         -> ppDIImportedEntity' pp diip
   DebugInfoLabel dil            -> ppDILabel' pp dil
+  DebugInfoArgList args         -> ppDIArgList' pp args
 
 ppDebugInfo :: LLVM => DebugInfo -> Doc
 ppDebugInfo = ppDebugInfo' ppLabel
@@ -1174,6 +1175,13 @@ ppDISubroutineType' pp st = "!DISubroutineType"
 ppDISubroutineType :: LLVM => DISubroutineType -> Doc
 ppDISubroutineType = ppDISubroutineType' ppLabel
 
+ppDIArgList' :: LLVM => (i -> Doc) -> DIArgList' i -> Doc
+ppDIArgList' pp args = "!DIArgList"
+  <> parens (commas (map (ppValMd' pp) (dialArgs args)))
+
+ppDIArgList :: LLVM => DIArgList -> Doc
+ppDIArgList = ppDIArgList' ppLabel
+
 -- Utilities -------------------------------------------------------------------
 
 ppBool :: Bool -> Doc