Merge pull request #4 from codingo/codingo-production-polish

Added --no-cide support
This commit is contained in:
Michael Skelton
2019-01-06 12:17:05 +10:00
committed by GitHub
5 changed files with 25 additions and 3 deletions

Binary file not shown.

View File

@@ -1,2 +1,2 @@
__version__ = '0.01'
__version__ = '1.0'

View File

@@ -18,16 +18,30 @@ class InputHelper(object):
return arg
@staticmethod
def expand_targets(targets, dont_expand_cidr):
if not dont_expand_cidr:
# expand CIDR from net addr
pass
# expand comma notation
# return list of unique hosts
return targets
@staticmethod
def process_targets(arguments):
targets = set()
# build list of targets from file/input
if arguments.target:
targets.add(arguments.target)
else:
for target in arguments.target_list:
targets.add(target.strip())
# take list of targets and expand CIDR / comma notation
targets = set(arguments.expand_targets(targets, arguments.nocidr))
return targets
@staticmethod
@@ -57,7 +71,8 @@ class InputParser(object):
targets.add_argument(
'-t', dest='target', required=False,
help='Specify a target or domain name.'
help='Specify a target or domain name either in comma format, '
'CIDR notation, or a single target.'
)
targets.add_argument(
@@ -96,6 +111,12 @@ class InputParser(object):
)
parser.add_argument(
'--no-cidr', dest='nocidr', action='store_true', default=False,
help='If set then CIDR notation in a target file will not be automatically '
'be expanded into individual hosts.'
)
parser.add_argument(
'--no-color', dest='nocolor', action='store_true', default=False,
help='If set then any foreground or background colours will be '

View File

@@ -58,6 +58,7 @@ class Pool(object):
for thread in threads:
thread.join()
# test harness
if __name__ == "__main__":
tasks = ["sleep 1",
"sleep 2",

View File

@@ -7,7 +7,7 @@ A threading management application that allows controlled execution of multiple
| Argument | Description |
|------------|--------------------------------------------------------------------------------------------------------------|
| -t | Specify a target or domain name |
| -t | Specify a target or domain name either in comma format, CIDR notation, or as an individual host. |
| -tL | Specify a list of targets or domain names |
| -threads | Specify the maximum number of threads to run at any one time (DEFAULT:5) |
| -timeout | Specify a timeout value in seconds for any one thread (DEFAULT:600) |