mirror of https://github.com/crytic/slither
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.9 KiB
52 lines
1.9 KiB
6 years ago
|
import os
|
||
|
|
import argparse
|
||
|
|
from slither import Slither
|
||
|
|
from slither.utils.colors import red
|
||
|
|
import logging
|
||
|
|
from .slither_format import slither_format
|
||
|
|
|
||
|
|
logging.basicConfig()
|
||
|
|
logging.getLogger("Slither").setLevel(logging.INFO)
|
||
|
|
|
||
|
|
available_detectors = ["external-function"]
|
||
|
|
detectors_to_run = []
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
"""
|
||
|
|
Parse the underlying arguments for the program.
|
||
|
|
:return: Returns the arguments for the program.
|
||
|
|
"""
|
||
|
|
parser = argparse.ArgumentParser(description='slither_format',
|
||
|
|
usage='slither_format filename')
|
||
|
|
|
||
|
|
parser.add_argument('filename', help='The filename of the contract or truffle directory to analyze.')
|
||
|
|
parser.add_argument('--solc', help='solc path', default='solc')
|
||
|
|
parser.add_argument('--verbose', '-v', help='verbose mode output',action='store_true',default=False)
|
||
|
|
|
||
|
|
group_detector = parser.add_argument_group('Detectors')
|
||
|
|
group_detector.add_argument('--detect',
|
||
|
|
help='Comma-separated list of detectors, defaults to all, '
|
||
|
|
'available detectors: {}'.format(
|
||
|
|
', '.join(d for d in available_detectors)),
|
||
|
|
action='store',
|
||
|
|
dest='detectors_to_run',
|
||
|
|
default='all')
|
||
|
|
return parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# ------------------------------
|
||
|
|
# Usage: python3 -m slither_format filename
|
||
|
|
# Example: python3 -m slither_format contract.sol
|
||
|
|
# ------------------------------
|
||
|
|
# Parse all arguments
|
||
|
|
args = parse_args()
|
||
|
|
|
||
|
|
# Perform slither analysis on the given filename
|
||
|
|
slither = Slither(args.filename, is_truffle=os.path.isdir(args.filename), solc=args.solc, disable_solc_warnings=True)
|
||
|
|
|
||
|
|
# Format the input files based on slither analysis
|
||
|
|
slither_format(args, slither)
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|