python source code of conftest

Project: MadMax (GitHub Link)

MadMax-master
- src
  - opcodes.py
  - tac_cfg.py
  - default_config.ini
  - blockparse.py
  - exporter.py
  - lattice.py
  - memtypes.py
  - function.py
  - patterns.py
  - evm_cfg.py
  - settings.py
  - cfg.py
  - dataflow.py
- Makefile
- examples
  - basic_optimized.dasm
  - dao_hack.dasm
  - basic.hex
  - basic_optimized.hex
  - empty.hex
  - slides.dasm
  - dao_hack.hex
  - README.md
  - slides.hex
  - empty.dasm
  - basic.dasm
- .gitmodules
- LICENSE
- test
  - bats
    - test_helper.bash
    - disassemble-examples.generated.bats.sh
    - bats
    - helpers
      - bats-support
      - assertions
    - decompile.bats
    - disassemble.bats
    - decompile-examples.generated.bats.sh
    - decompile-regressions.bats
    - environment.bash
    - sphinx-docs.bats
    - decompile-disassemble-integration.bats
    - generate.sh
  - test_cfg.py
  - data
    - hex
      - multisig.hex
      - checked.hex
      - mem_leak.hex
      - basic.hex
      - asmtest.hex
      - byzantium-basic.hex
      - basic_optimized.hex
      - empty.hex
      - byzantium-revert.hex
      - withdraw_dao.hex
      - recursion.hex
      - dao_hack.hex
      - README.md
      - slides.hex
      - mutual_recursion.hex
      - invalid_ops.hex
      - basic_example.hex
      - example_two.hex
    - regression
      - private_func_no_body.hex
      - long_running.hex
    - expected
      - disassemble_pretty_slides.hex.output
      - disassemble_pretty_empty.hex.output
      - decompile_slides.hex.output
      - disassemble_invalid_ops.hex.output
      - disassemble_empty.hex.output
      - disassemble_slides.hex.output
      - disassemble_pretty_withdraw_dao.hex.output
      - disassemble_pretty_mem_leak.hex.output
      - disassemble_basic_example.hex.output
      - decompile_byzantium-basic.hex.output
      - decompile_invalid_ops.hex.output
      - decompile_mem_leak.hex.output
      - disassemble_pretty_byzantium-revert.hex.output
      - decompile_basic_optimized.dasm.output
      - decompile_checked.hex.output
      - disassemble_pretty_dao_hack.hex.output
      - decompile_checked.dasm.output
      - decompile_empty.dasm.output
      - disassemble_recursion.hex.output
      - decompile_example_two.hex.output
      - disassemble_multisig.hex.output
      - decompile_byzantium-revert.hex.output
      - disassemble_pretty_basic_example.hex.output
      - disassemble_basic_optimized.hex.output
      - decompile_dao_hack.hex.output
      - disassemble_byzantium-revert.hex.output
      - disassemble_pretty_mutual_recursion.hex.output
      - disassemble_asmtest.hex.output
      - disassemble_withdraw_dao.hex.output
      - disassemble_pretty_asmtest.hex.output
      - disassemble_pretty_basic.hex.output
      - decompile_basic_example.hex.output
      - disassemble_mutual_recursion.hex.output
      - decompile_withdraw_dao.hex.output
      - decompile_basic.hex.output
      - decompile_recursion.hex.output
      - disassemble_pretty_basic_optimized.hex.output
      - disassemble_mem_leak.hex.output
      - disassemble_checked.hex.output
      - decompile_asmtest.hex.output
      - decompile_withdraw_dao.dasm.output
      - disassemble_example_two.hex.output
      - decompile_basic.dasm.output
      - disassemble_pretty_checked.hex.output
      - disassemble_basic.hex.output
      - decompile_dao_hack.dasm.output
      - disassemble_pretty_example_two.hex.output
      - disassemble_pretty_recursion.hex.output
      - disassemble_dao_hack.hex.output
      - disassemble_pretty_invalid_ops.hex.output
      - disassemble_byzantium-basic.hex.output
      - decompile_slides.dasm.output
      - disassemble_pretty_byzantium-basic.hex.output
      - decompile_mem_leak.dasm.output
      - decompile_basic_optimized.hex.output
      - decompile_empty.hex.output
      - disassemble_pretty_multisig.hex.output
      - decompile_mutual_recursion.hex.output
      - decompile_multisig.hex.output
    - dasm
      - basic_optimized.dasm
      - dao_hack.dasm
      - mem_leak.dasm
      - slides.dasm
      - checked.dasm
      - withdraw_dao.dasm
      - README.md
      - empty.dasm
      - basic.dasm
  - test_function.py
  - conftest.py
- .editorconfig
- .travis.yml
- README.md
- bin
  - disassemble
  - decompile
  - config.ini
- tools
  - interface_checker
    - check_interfaces.py
    - contracts
      - erc20sigs.hex
      - erc20token.hex
      - noncompliant.hex
      - erc20optsigs.hex
    - README.md
    - requirements.txt
    - interfaces
      - ERC20.interface
      - ERC20Optional.interface
      - ERC20Complete.json
  - bulk_analyser
    - analyse.py
    - filterresults.py
    - oldclient.dl
    - find_contract.py
    - vandal-analytics.dl
    - unstructured_loop_detector.dl
    - analyse_opcodes.py
    - tac_instructions.dl
    - README.md
    - spec.dl
    - assertions.dl
  - creationiser
    - convert.py
    - contract_script.js
    - README.md
  - analyser
    - run_all.sh
    - generate_batches.py
    - README.md
    - run_batch.sh
    - analyse.rb
- requirements.txt
- CODEOWNERS
- .gitignore
- doc
  - Makefile
  - source
    - index.rst
    - conf.py
    - _static
      - .dummyfile
  - make.bat
  - sphinxext
    - custom_skip.py
    - __init__.py
  - templates
    - autosummary
      - module.rst
      - class.rst
      - base.rst
  - README.md

# BSD 3-Clause License
#
# Copyright (c) 2016, 2017, The University of Sydney. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# * Neither the name of the copyright holder nor the names of its
#   contributors may be used to endorse or promote products derived from
#   this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import fileinput
import glob
import sys
from os.path import dirname, join, abspath

import pytest

src_path = join(dirname(abspath(__file__)), "..")
sys.path.insert(0, src_path)

DISASM_TEST_PROGS = "data/dasm/*.dasm"
DISASM_TEST_PROGS = join(dirname(abspath(__file__)), DISASM_TEST_PROGS)


@pytest.fixture(params=list(glob.glob(DISASM_TEST_PROGS)))
def cfg(request):
    from src.tac_cfg import TACGraph
    import src.optimise as optimise  # TODO: Fix this
    with open(request.param) as f:
        disasm = f.read()
    cfg = TACGraph.from_dasm(fileinput.input())
    optimise.fold_constants(cfg)
    cfg.hook_up_jumps()
    yield cfg