Chameleon/chameleon.py at master · MarcOverIP/Chameleon · GitHub

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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import argparse
import sys
from modules import *


class Chameleon:

    def __init__(self):
        pass

    def validate_args(self):
        parser = argparse.ArgumentParser(description="")
        parser.add_argument("--proxy", metavar="<proxy>", dest="proxy", default=None,
                            help="Proxy type: a = all, b = bluecoat, m = mcafee, i = IBM Xforce")
        parser.add_argument("--check", action='store_true', help="Perform check on current category")
        parser.add_argument("--submit", action='store_true', help="Submit new category")
        parser.add_argument("--domain", metavar="<domain>", dest="domain", default=None, help="Domain to validate")
        args = parser.parse_args()

        if not args.proxy:
            print("[-] Missing --proxy argument")
            sys.exit(-1)
        if not args.domain:
            print("[-] Missing --domain argument")
            sys.exit(-1)
        if not args.check and not args.submit:
            print("[-] Missing --check or --submit argument")
            sys.exit(-1)
        return args

    def show_banner(self):
        with open('banner.txt', 'r') as f:
            data = f.read()
            print("\033[92m%s\033[0;0m" % data)

    def run(self, args):
        if args.proxy == 'm' or args.proxy == 'a':
            print("\033[1;34m[-] Targeting McAfee Trustedsource\033[0;0m")
            ts = trustedsource.TrustedSource(args.domain)
            if args.check:
                ts.check_category(False)
            elif args.submit:
                ts.check_category(True)

        if args.proxy == 'b' or args.proxy == 'a':
            print("\033[1;34m[-] Targeting Bluecoat WebPulse\033[0;0m")
            if args.check:
                b = bluecoat.Bluecoat(args.domain, 'https://www.bankofamerica.com')
                b.check_category()
            elif args.submit:
                print(
                    "\033[1;31m[-] WARNING: This module must be run from the webserver you want to categorise\033[0;0m")
                print("\033[1;31m[-] Proceed: Y/N\033[0;0m")
                while True:
                    choice = input().lower()
                    if choice == 'Y' or choice == 'y':
                        b = bluecoat.Bluecoat(args.domain, 'https://www.bankofamerica.com')
                        b.run()
                        break
                    elif choice == 'N' or choice == 'n':
                        break

        if args.proxy == 'i' or args.proxy == 'a':
            print("\033[1;34m[-] Targeting IBM Xforce\033[0;0m")
            xf = ibmxforce.IBMXforce(args.domain)
            if args.check:
                xf.checkIBMxForce()
            elif args.submit:
                xf.submit_category()


if __name__ == "__main__":
    c = Chameleon()
    c.show_banner()
    args = c.validate_args()
    c.run(args)