Veil.py

#!/usr/bin/env python3

"""
"""

import argparse
import sys
from lib.common import helpers
from lib.common import messages
from lib.common import orchestra
sys.dont_write_bytecode = True


if __name__ == "__main__":
    import os
    os.chdir(os.path.dirname(os.path.realpath(__file__)))

    parser = argparse.ArgumentParser(
        add_help=False, description="Veil is a framework containing multiple\
         tools.")
    parser.add_argument(
        '-h', '-?', '--h', '-help', '--help', action="store_true",
        help=argparse.SUPPRESS)

    veilframework = parser.add_argument_group('[*] Veil Options')
    veilframework.add_argument(
        '--list-tools', action="store_true", default=False,
        help='List Veil\'s tools')
    veilframework.add_argument(
        '-t', '--tool', metavar='TOOL', default=False,
        help='Specify Veil tool to use (Evasion, Ordnance etc.)')
    veilframework.add_argument(
        '--update', action='store_true', help='Update the Veil framework')
    veilframework.add_argument(
        '--setup', action='store_true', help='Run the Veil framework setup file & regenerate the configuration')
    veilframework.add_argument(
        '--config', action='store_true', help='Regenerate the Veil framework configuration file')
    veilframework.add_argument(
        '--version', action="store_true", help='Displays version and quits')

    callback_args = parser.add_argument_group('[*] Callback Settings')
    callback_args.add_argument(
        "--ip", "--domain", metavar="IP", default=None,
        help='IP address to connect back to')
    callback_args.add_argument(
        '--port', metavar="PORT", default=8675, type=int,
        help='Port number to connect to')

    payload_args = parser.add_argument_group('[*] Payload Settings')
    payload_args.add_argument(
        '--list-payloads', default=False, action='store_true',
        help='Lists all available payloads for that tool')

    veilevasion = parser.add_argument_group('[*] Veil-Evasion Options')
    veilevasion.add_argument(
        '-p', metavar="PAYLOAD", nargs='?', const="list",
        help='Payload to generate')
    veilevasion.add_argument(
        '-o', metavar="OUTPUT-NAME", default="payload",
        help='Output file base name for source and compiled binaries')
    veilevasion.add_argument(
        '-c', metavar='OPTION=value', nargs='*',
        default=None, help='Custom payload module options')
    veilevasion.add_argument(
        '--msfoptions', metavar="OPTION=value", nargs='*',
        help='Options for the specified metasploit payload')
    veilevasion.add_argument(
        '--msfvenom', metavar="", nargs='?',
        default='windows/meterpreter/reverse_tcp', help='Metasploit shellcode to generate (e.g. windows/meterpreter/reverse_tcp etc.)')
    veilevasion.add_argument(
        '--compiler', metavar="pyinstaller", default='pyinstaller',
        help='Compiler option for payload (currently only needed for Python)')
    veilevasion.add_argument(
        '--clean', action='store_true',
        help='Clean out payload folders')

    ordnance_shellcode = parser.add_argument_group('[*] Veil-Ordnance Shellcode Options')
    ordnance_shellcode.add_argument(
        "--ordnance-payload", metavar="PAYLOAD", default=None,
        help='Payload type (bind_tcp, rev_tcp, etc.)')

    ordnance_encoder = parser.add_argument_group('[*] Veil-Ordnance Encoder Options')
    ordnance_encoder.add_argument(
        '--list-encoders', default=False, action='store_true',
        help='Lists all available encoders')
    ordnance_encoder.add_argument(
        "-e", "--encoder", metavar="ENCODER", default=None,
        help='Name of shellcode encoder to use')
    ordnance_encoder.add_argument(
        "-b", "--bad-chars", metavar="\\x00\\x0a..", default=None,
        help='Bad characters to avoid')
    ordnance_encoder.add_argument(
        '--print-stats', default=False, action='store_true',
        help='Print information about the encoded shellcode')

    args = parser.parse_args()

    the_conductor = orchestra.Conductor(args)

    # --help
    if args.h:
        parser.print_help()
        sys.exit()

    # --version
    if args.version:
        messages.title_screen()
        sys.exit()

    # --update
    if args.update:
        the_conductor.update_veil()
        sys.exit()

    # --setup
    if args.setup:
        the_conductor.setup_veil()
        sys.exit()

    # --config
    if args.config:
        the_conductor.config_veil()
        sys.exit()

    # --list-tools
    if args.list_tools:
        the_conductor.list_tools()
        sys.exit()

    # --clean
    if args.clean:
        helpers.clean_payloads()
        sys.exit()

    # Anything else that isn't defined
    if not args.tool:
        the_conductor.main_menu()
        sys.exit()

    # This should hit if trying to use the CLI
    else:
        the_conductor.command_line_use()