-
Notifications
You must be signed in to change notification settings - Fork 5
/
harvester.py
58 lines (48 loc) · 1.93 KB
/
harvester.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
"""
Run theharvester save data and create images of the results
USAGE: python harvester.py <output_dir> <domain> [-s <screenshot directory>]
"""
import os
import logging
import argparse
from utils import utils, logging_config # noqa pylint: disable=unused-import
from utils import run_commands
LOG = logging.getLogger("ptscripts.harvester")
def main(args):
LOG.info("Running theHarvester")
os.makedirs(args.output, exist_ok=True)
output = os.path.join(args.output, "harvester.html")
command = """theHarvester -d {domain} -b all -l 100 -f {output}""".format(
domain=args.domain, output=output)
LOG.info("Running the command: {}".format(command))
file_name = "harvester_output.html"
html_path = os.path.join(args.output, file_name)
LOG.info("Saving output to: {}".format(html_path))
text_output = run_commands.bash_command(command)
html_output = run_commands.create_html_file(text_output, command, html_path)
if html_output and args.screenshot:
utils.selenium_image(html_output, args.screenshot)
if not html_output:
LOG.error("Didn't receive a response from running the command.")
def parse_args(args):
parser = argparse.ArgumentParser(
parents=[utils.parent_argparser()],
description='Capture theharvester data and image.',
)
parser.add_argument('domain', help="Domain to be tested.")
parser.add_argument('output', help="full path to where the results will be saved.")
parser.add_argument("-s", "--screenshot",
help="full path to where the screenshot will be saved.")
args = parser.parse_args(args)
logger = logging.getLogger("ptscripts")
if args.quiet:
logger.setLevel('ERROR')
elif args.verbose:
logger.setLevel('DEBUG')
logger.debug("Logger set to debug.")
else:
logger.setLevel('INFO')
return args
if __name__ == "__main__":
import sys
main(parse_args(sys.argv[1:]))