ipTrace/iptrace.py at main · Konstantinos-Papanagnou/ipTrace · 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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/python3

import Resolver
import ArgumentParser
import ResponseParser
import webbrowser
import sys
import visualcreator
from bannermagic import printBannerPadding

def main():
    args = ArgumentParser.ArgumentParser()
    parser = Resolver.Resolver(args.ipaddr, args.verbosity)

    resolved_ips = parser.getResolvedIps()
    if args.output:
        outfile = args.output
    else:
        outfile = False

    try:

        if(isinstance(resolved_ips, list)):

            if not resolved_ips:
                sys.stderr.write('[-] Could not resolve any ips... Exiting...\n\n')
                exit()
            ipsTraced = []
            locations = []
            lattitudes = []
            longtitudes = []
            orgs = []
            print('\n\n[*] Resolved Ips: {0}'.format(len(resolved_ips)))
            for ip in resolved_ips:
                loc, lat, lon, org = printResults(ip, args.verbosity, args.accurate, outfile)
                if args.visual:
                    ipsTraced.append(ip)
                    locations.append(loc)
                    lattitudes.append(lat)
                    longtitudes.append(lon)
                    orgs.append(org)
            if args.visual:
                visualcreator.create(ipsTraced, locations, lattitudes, longtitudes, orgs, args.visual, args.mobile)

        else:
            if not resolved_ips:
                sys.stderr.write('[-] Could not resolve any ips... Exiting...\n\n')
                exit()

            print('\n\nResolved Ip: {0}'.format(resolved_ips))
            loc, lat, lon, org = printResults(resolved_ips, args.verbosity, args.accurate, outfile)
            if args.visual:
                visualcreator.create(resolved_ips, loc, lat, lon, org, args.visual, args.mobile)

        print('\n\n')
        printBannerPadding()
        print('\n\n')
        print('[+]  Trace Completed Successfully!')
    finally:
        if outfile:
            outfile.close()

def printResults(ip, verbose, accurate, output):
    if accurate:
        webbrowser.get('firefox').open('https://whatismyipaddress.com/ip/{0}'.format(ip))
        if output:
            sys.stderr.write('[-] Cannot write results when working in high accuracy mode\n')
        return

    rp = ResponseParser.ResponseParser(verbose)
    data = rp.GetParsedData(ip)

    print('\n\n')
    printBannerPadding()
    print('\n\n')
    print('[*] Tracing {0}'.format(ip))
    if output:
        output.write('================================================================\n\n')
        output.write('  Tracing {0}\n\n'.format(ip))
    print('\n[*] Trace')

    genkeys=data.keys()
    for key in genkeys:
        print('[*] {0}  :   {1}'.format(key, data[key]))
        if output:
            output.write('{0}   :   {1}\n'.format(key, data[key]))

    if 'org' in data.keys():
        return data['city'], data['lat'], data['lon'], data['org']
    else:
        return data['city'], data['lat'], data['lon'], 'Unknown'


if __name__ == "__main__":
    main()