mirror of
https://github.com/GaloisInc/macaw.git
synced 2024-12-18 19:41:44 +03:00
51 lines
1.9 KiB
Haskell
51 lines
1.9 KiB
Haskell
{-# LANGUAGE DataKinds #-}
|
|
{-# LANGUAGE TypeApplications #-}
|
|
module PPC64InstructionCoverage (
|
|
ppc64InstructionCoverageTests
|
|
) where
|
|
|
|
import Control.Lens ( (^.) )
|
|
import qualified Data.Map as M
|
|
import Data.Maybe ( fromJust, mapMaybe )
|
|
import Data.Proxy ( Proxy(..) )
|
|
import qualified Data.Set as S
|
|
import Data.Word ( Word64 )
|
|
import qualified Test.Tasty as T
|
|
import qualified Test.Tasty.HUnit as T
|
|
|
|
import qualified Data.ElfEdit as E
|
|
|
|
import qualified Data.Parameterized.Some as PU
|
|
import qualified Data.Macaw.Memory as MM
|
|
import qualified Data.Macaw.Discovery as MD
|
|
import qualified Data.Macaw.PPC as RO
|
|
import qualified Data.Macaw.PPC.BinaryFormat.ELF as E
|
|
import qualified SemMC.Architecture.PPC64 as PPC64
|
|
|
|
import Shared
|
|
|
|
ppc64InstructionCoverageTests :: [FilePath] -> T.TestTree
|
|
ppc64InstructionCoverageTests = T.testGroup "PPCCoverage" . map mkTest
|
|
|
|
mkTest :: FilePath -> T.TestTree
|
|
mkTest fp = T.testCase fp (withELF fp testMacaw)
|
|
|
|
testMacaw :: E.Elf 64 -> IO ()
|
|
testMacaw elf =
|
|
withMemory MM.Addr64 elf $ \mem -> do
|
|
let Just entryPoint = MM.asSegmentOff mem (findEntryPoint64 elf mem)
|
|
let tocBase = RO.tocBaseForELF (Proxy @PPC64.PPC) elf
|
|
let otherEntryAddrs :: [MM.MemAddr 64]
|
|
otherEntryAddrs = E.tocEntryAddrsForElf (Proxy @PPC64.PPC) elf
|
|
let otherEntries = mapMaybe (MM.asSegmentOff mem) otherEntryAddrs
|
|
let di = MD.cfgFromAddrs (RO.ppc64_linux_info tocBase) mem M.empty (entryPoint:otherEntries) []
|
|
let allFoundBlockAddrs :: S.Set Word64
|
|
allFoundBlockAddrs =
|
|
S.fromList [ fromIntegral (fromJust (MM.asAbsoluteAddr (MM.relativeSegmentAddr (MD.pblockAddr pbr))))
|
|
| PU.Some dfi <- M.elems (di ^. MD.funInfo)
|
|
, pbr <- M.elems (dfi ^. MD.parsedBlocks)
|
|
]
|
|
T.assertBool "No blocks found" (not (S.null allFoundBlockAddrs))
|
|
|
|
|