ethereum-analyzer-2.0.0: test/Ethereum/Analyzer/UtilSpec.hs
{-# LANGUAGE OverloadedStrings, NoImplicitPrelude #-}
module Ethereum.Analyzer.UtilSpec
( spec
) where
import Protolude hiding (show)
import Data.Text
import Ethereum.Analyzer
import Ethereum.Analyzer.CfgAugmentPass
import Ethereum.Analyzer.TestData.Basic
import Ethereum.Analyzer.Util
import Test.Hspec
spec :: Spec
spec =
describe "toDotText" $ do
it "shows HplBody's dot graph" $ do
let disasmd = disasm hexstring2
(unpack . unWordLabelMapM)
(toDotText . bodyOf . ctorOf <$> evmOps2HplContract disasmd) `shouldBe`
toS expectedHexString2RawDot
it "shows HplBody after CfgAugmentPass" $ do
let disasmd@((_, _):_) = disasm hexstring2
result =
(unpack . unWordLabelMapM) $ do
contract <- evmOps2HplContract disasmd
toDotText . bodyOf . ctorOf <$> doCfgAugmentPass contract
result `shouldBe` toS expectedHexString2AugDot
expectedHexString2RawDot :: Text
expectedHexString2RawDot =
"digraph {\n" <> " 1 [label=\"CO: L1\\n" <> "OO: 0: PUSH [96]\\n" <>
"OO: 2: PUSH [64]\\n" <>
"OO: 4: MSTORE\\n" <>
"OO: 5: CALLDATASIZE\\n" <>
"OO: 6: ISZERO\\n" <>
"OO: 7: PUSH [39]\\n" <>
"OC: 9: JUMPI -> [L2]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 2 [label=\"CO: L2\\n" <>
"OO: 10: PUSH [224]\\n" <>
"OO: 12: PUSH [2]\\n" <>
"OO: 14: EXP\\n" <>
"OO: 15: PUSH [0]\\n" <>
"OO: 17: CALLDATALOAD\\n" <>
"OO: 18: DIV\\n" <>
"OO: 19: PUSH [65,192,225,181]\\n" <>
"OO: 24: DUP2\\n" <>
"OO: 25: EQ\\n" <>
"OO: 26: PUSH [110]\\n" <>
"OC: 28: JUMPI -> [L3]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 3 [label=\"CO: L3\\n" <>
"OO: 29: DUP1\\n" <>
"OO: 30: PUSH [229,34,83,129]\\n" <>
"OO: 35: EQ\\n" <>
"OO: 36: PUSH [150]\\n" <>
"OC: 38: JUMPI -> [L4]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 4 [label=\"CO: L4\\n" <>
"OO: 39: JUMPDEST\\n" <>
"OO: 40: PUSH [213]\\n" <>
"OO: 42: PUSH [0]\\n" <>
"OO: 44: CALLVALUE\\n" <>
"OO: 45: GT\\n" <>
"OO: 46: ISZERO\\n" <>
"OO: 47: PUSH [108]\\n" <>
"OC: 49: JUMPI -> [L5]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 5 [label=\"CO: L5\\n" <>
"OO: 50: CALLVALUE\\n" <>
"OO: 51: PUSH [96]\\n" <>
"OO: 53: SWAP1\\n" <>
"OO: 54: DUP2\\n" <>
"OO: 55: MSTORE\\n" <>
"OO: 56: PUSH [88]\\n" <>
"OO: 58: SWAP1\\n" <>
"OO: 59: PUSH [1]\\n" <>
"OO: 61: PUSH [160]\\n" <>
"OO: 63: PUSH [2]\\n" <>
"OO: 65: EXP\\n" <>
"OO: 66: SUB\\n" <>
"OO: 67: CALLER\\n" <>
"OO: 68: AND\\n" <>
"OO: 69: SWAP1\\n" <>
"OO: 70: PUSH [144,137,8,9,198,84,241,29,110,114,162,143,166,1,73,119,10,13,17,236,108,146,49,157,108,235,43,176,164,234,26,21]\\n" <>
"OO: 103: SWAP1\\n" <>
"OO: 104: PUSH [32]\\n" <>
"OO: 106: SWAP1\\n" <>
"OC: 107: LOG3 -> [L6]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 6 [label=\"CO: L6\\n" <>
"OO: 108: JUMPDEST\\n" <>
"OC: 109: JUMP -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 7 [label=\"CO: L7\\n" <>
"OO: 110: JUMPDEST\\n" <>
"OO: 111: PUSH [213]\\n" <>
"OO: 113: PUSH [0]\\n" <>
"OO: 115: SLOAD\\n" <>
"OO: 116: PUSH [1]\\n" <>
"OO: 118: PUSH [160]\\n" <>
"OO: 120: PUSH [2]\\n" <>
"OO: 122: EXP\\n" <>
"OO: 123: SUB\\n" <>
"OO: 124: SWAP1\\n" <>
"OO: 125: DUP2\\n" <>
"OO: 126: AND\\n" <>
"OO: 127: CALLER\\n" <>
"OO: 128: SWAP2\\n" <>
"OO: 129: SWAP1\\n" <>
"OO: 130: SWAP2\\n" <>
"OO: 131: AND\\n" <>
"OO: 132: EQ\\n" <>
"OO: 133: ISZERO\\n" <>
"OO: 134: PUSH [108]\\n" <>
"OC: 136: JUMPI -> [L8]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 8 [label=\"CO: L8\\n" <>
"OO: 137: PUSH [0]\\n" <>
"OO: 139: SLOAD\\n" <>
"OO: 140: PUSH [1]\\n" <>
"OO: 142: PUSH [160]\\n" <>
"OO: 144: PUSH [2]\\n" <>
"OO: 146: EXP\\n" <>
"OO: 147: SUB\\n" <>
"OO: 148: AND\\n" <>
"OC: 149: SUICIDE -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 9 [label=\"CO: L9\\n" <>
"OO: 150: JUMPDEST\\n" <>
"OO: 151: PUSH [213]\\n" <>
"OO: 153: PUSH [0]\\n" <>
"OO: 155: SLOAD\\n" <>
"OO: 156: PUSH [1]\\n" <>
"OO: 158: PUSH [160]\\n" <>
"OO: 160: PUSH [2]\\n" <>
"OO: 162: EXP\\n" <>
"OO: 163: SUB\\n" <>
"OO: 164: SWAP1\\n" <>
"OO: 165: DUP2\\n" <>
"OO: 166: AND\\n" <>
"OO: 167: CALLER\\n" <>
"OO: 168: SWAP2\\n" <>
"OO: 169: SWAP1\\n" <>
"OO: 170: SWAP2\\n" <>
"OO: 171: AND\\n" <>
"OO: 172: EQ\\n" <>
"OO: 173: ISZERO\\n" <>
"OO: 174: PUSH [108]\\n" <>
"OC: 176: JUMPI -> [L10]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 10 [label=\"CO: L10\\n" <>
"OO: 177: PUSH [0]\\n" <>
"OO: 179: DUP1\\n" <>
"OO: 180: SLOAD\\n" <>
"OO: 181: PUSH [1]\\n" <>
"OO: 183: PUSH [160]\\n" <>
"OO: 185: PUSH [2]\\n" <>
"OO: 187: EXP\\n" <>
"OO: 188: SUB\\n" <>
"OO: 189: SWAP1\\n" <>
"OO: 190: DUP2\\n" <>
"OO: 191: AND\\n" <>
"OO: 192: SWAP2\\n" <>
"OO: 193: SWAP1\\n" <>
"OO: 194: ADDRESS\\n" <>
"OO: 195: AND\\n" <>
"OO: 196: BALANCE\\n" <>
"OO: 197: PUSH [96]\\n" <>
"OO: 199: DUP3\\n" <>
"OO: 200: DUP2\\n" <>
"OO: 201: DUP2\\n" <>
"OO: 202: DUP2\\n" <>
"OO: 203: DUP6\\n" <>
"OO: 204: DUP9\\n" <>
"OO: 205: DUP4\\n" <>
"OC: 206: CALL -> [L11]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 11 [label=\"CO: L11\\n" <>
"OO: 207: POP\\n" <>
"OO: 208: POP\\n" <>
"OO: 209: POP\\n" <>
"OO: 210: POP\\n" <>
"OO: 211: POP\\n" <>
"OC: 212: JUMP -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 12 [label=\"CO: L12\\n" <>
"OO: 213: JUMPDEST\\n" <>
"OC: 214: STOP -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 1 -> 2;\n" <>
" 2 -> 3;\n" <>
" 3 -> 4;\n" <>
" 4 -> 5;\n" <>
" 5 -> 6;\n" <>
" 7 -> 8;\n" <>
" 9 -> 10;\n" <>
" 10 -> 11;\n" <>
"}"
expectedHexString2AugDot :: Text
expectedHexString2AugDot =
"digraph {\n" <> " 1 [label=\"CO: L1\\n" <> "OO: 0: PUSH [96]\\n" <>
"OO: 2: PUSH [64]\\n" <>
"OO: 4: MSTORE\\n" <>
"OO: 5: CALLDATASIZE\\n" <>
"OO: 6: ISZERO\\n" <>
"OO: 7: PUSH [39]\\n" <>
"OC: 9: JUMPI -> [L2,L4]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 2 [label=\"CO: L2\\n" <>
"OO: 10: PUSH [224]\\n" <>
"OO: 12: PUSH [2]\\n" <>
"OO: 14: EXP\\n" <>
"OO: 15: PUSH [0]\\n" <>
"OO: 17: CALLDATALOAD\\n" <>
"OO: 18: DIV\\n" <>
"OO: 19: PUSH [65,192,225,181]\\n" <>
"OO: 24: DUP2\\n" <>
"OO: 25: EQ\\n" <>
"OO: 26: PUSH [110]\\n" <>
"OC: 28: JUMPI -> [L3,L7]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 3 [label=\"CO: L3\\n" <>
"OO: 29: DUP1\\n" <>
"OO: 30: PUSH [229,34,83,129]\\n" <>
"OO: 35: EQ\\n" <>
"OO: 36: PUSH [150]\\n" <>
"OC: 38: JUMPI -> [L4,L9]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 4 [label=\"CO: L4\\n" <>
"OO: 39: JUMPDEST\\n" <>
"OO: 40: PUSH [213]\\n" <>
"OO: 42: PUSH [0]\\n" <>
"OO: 44: CALLVALUE\\n" <>
"OO: 45: GT\\n" <>
"OO: 46: ISZERO\\n" <>
"OO: 47: PUSH [108]\\n" <>
"OC: 49: JUMPI -> [L5,L6]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 5 [label=\"CO: L5\\n" <>
"OO: 50: CALLVALUE\\n" <>
"OO: 51: PUSH [96]\\n" <>
"OO: 53: SWAP1\\n" <>
"OO: 54: DUP2\\n" <>
"OO: 55: MSTORE\\n" <>
"OO: 56: PUSH [88]\\n" <>
"OO: 58: SWAP1\\n" <>
"OO: 59: PUSH [1]\\n" <>
"OO: 61: PUSH [160]\\n" <>
"OO: 63: PUSH [2]\\n" <>
"OO: 65: EXP\\n" <>
"OO: 66: SUB\\n" <>
"OO: 67: CALLER\\n" <>
"OO: 68: AND\\n" <>
"OO: 69: SWAP1\\n" <>
"OO: 70: PUSH [144,137,8,9,198,84,241,29,110,114,162,143,166,1,73,119,10,13,17,236,108,146,49,157,108,235,43,176,164,234,26,21]\\n" <>
"OO: 103: SWAP1\\n" <>
"OO: 104: PUSH [32]\\n" <>
"OO: 106: SWAP1\\n" <>
"OC: 107: LOG3 -> [L6]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 6 [label=\"CO: L6\\n" <>
"OO: 108: JUMPDEST\\n" <>
"OC: 109: JUMP -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 7 [label=\"CO: L7\\n" <>
"OO: 110: JUMPDEST\\n" <>
"OO: 111: PUSH [213]\\n" <>
"OO: 113: PUSH [0]\\n" <>
"OO: 115: SLOAD\\n" <>
"OO: 116: PUSH [1]\\n" <>
"OO: 118: PUSH [160]\\n" <>
"OO: 120: PUSH [2]\\n" <>
"OO: 122: EXP\\n" <>
"OO: 123: SUB\\n" <>
"OO: 124: SWAP1\\n" <>
"OO: 125: DUP2\\n" <>
"OO: 126: AND\\n" <>
"OO: 127: CALLER\\n" <>
"OO: 128: SWAP2\\n" <>
"OO: 129: SWAP1\\n" <>
"OO: 130: SWAP2\\n" <>
"OO: 131: AND\\n" <>
"OO: 132: EQ\\n" <>
"OO: 133: ISZERO\\n" <>
"OO: 134: PUSH [108]\\n" <>
"OC: 136: JUMPI -> [L6,L8]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 8 [label=\"CO: L8\\n" <>
"OO: 137: PUSH [0]\\n" <>
"OO: 139: SLOAD\\n" <>
"OO: 140: PUSH [1]\\n" <>
"OO: 142: PUSH [160]\\n" <>
"OO: 144: PUSH [2]\\n" <>
"OO: 146: EXP\\n" <>
"OO: 147: SUB\\n" <>
"OO: 148: AND\\n" <>
"OC: 149: SUICIDE -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 9 [label=\"CO: L9\\n" <>
"OO: 150: JUMPDEST\\n" <>
"OO: 151: PUSH [213]\\n" <>
"OO: 153: PUSH [0]\\n" <>
"OO: 155: SLOAD\\n" <>
"OO: 156: PUSH [1]\\n" <>
"OO: 158: PUSH [160]\\n" <>
"OO: 160: PUSH [2]\\n" <>
"OO: 162: EXP\\n" <>
"OO: 163: SUB\\n" <>
"OO: 164: SWAP1\\n" <>
"OO: 165: DUP2\\n" <>
"OO: 166: AND\\n" <>
"OO: 167: CALLER\\n" <>
"OO: 168: SWAP2\\n" <>
"OO: 169: SWAP1\\n" <>
"OO: 170: SWAP2\\n" <>
"OO: 171: AND\\n" <>
"OO: 172: EQ\\n" <>
"OO: 173: ISZERO\\n" <>
"OO: 174: PUSH [108]\\n" <>
"OC: 176: JUMPI -> [L6,L10]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 10 [label=\"CO: L10\\n" <>
"OO: 177: PUSH [0]\\n" <>
"OO: 179: DUP1\\n" <>
"OO: 180: SLOAD\\n" <>
"OO: 181: PUSH [1]\\n" <>
"OO: 183: PUSH [160]\\n" <>
"OO: 185: PUSH [2]\\n" <>
"OO: 187: EXP\\n" <>
"OO: 188: SUB\\n" <>
"OO: 189: SWAP1\\n" <>
"OO: 190: DUP2\\n" <>
"OO: 191: AND\\n" <>
"OO: 192: SWAP2\\n" <>
"OO: 193: SWAP1\\n" <>
"OO: 194: ADDRESS\\n" <>
"OO: 195: AND\\n" <>
"OO: 196: BALANCE\\n" <>
"OO: 197: PUSH [96]\\n" <>
"OO: 199: DUP3\\n" <>
"OO: 200: DUP2\\n" <>
"OO: 201: DUP2\\n" <>
"OO: 202: DUP2\\n" <>
"OO: 203: DUP6\\n" <>
"OO: 204: DUP9\\n" <>
"OO: 205: DUP4\\n" <>
"OC: 206: CALL -> [L11]\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 11 [label=\"CO: L11\\n" <>
"OO: 207: POP\\n" <>
"OO: 208: POP\\n" <>
"OO: 209: POP\\n" <>
"OO: 210: POP\\n" <>
"OO: 211: POP\\n" <>
"OC: 212: JUMP -> []\\n" <>
"\"\n" <>
" ,shape=box];\n" <>
" 1 -> 2;\n" <>
" 1 -> 4;\n" <>
" 2 -> 3;\n" <>
" 2 -> 7;\n" <>
" 3 -> 4;\n" <>
" 3 -> 9;\n" <>
" 4 -> 5;\n" <>
" 4 -> 6;\n" <>
" 5 -> 6;\n" <>
" 7 -> 6;\n" <>
" 7 -> 8;\n" <>
" 9 -> 6;\n" <>
" 9 -> 10;\n" <>
" 10 -> 11;\n" <>
"}"