Probe_spider/ipaddress.py at master · Aravindha1234u/Probe_spider · 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
96
97
98
import requests
import gmplot
from src.api import ipstack
import webbrowser
import re
from src.api import gmap
from ipaddress import *
from webosint.who.whois import *

R = '\033[31m' # red
G = '\033[32m' # green
C = '\033[36m' # cyan
W = '\033[0m'  # white

api_key = ipstack()
if api_key == "":
    print("Add you ipstack api key to src/api.py")
    exit()

def IPHeatmap():
    print(C+'''Choose
    1) Trace single IP
    2) Trace multiple IPs''')
    choice = input(C+"root@osint:"+W+"~/IPaddress# ")

    if choice == '1':
        IP = input(C+"Enter the IP : "+W)
        read_single_IP(IP)
    elif choice == '2':
        IP_file = input(C+"Enter the IP File Location : "+W)
        read_multiple_IP(IP_file)
    else:
        print(R+"\nError: Please choose an appropriate option")

def read_single_IP(IP):
    print ( W + '[+]' + G + "Processing IP: %s ..." %IP + '\n')
    if not re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$",IP):
        print("Invalid IP Address")
        IPHeatmap()
    lats = []
    lons = []
    r = requests.get("http://api.IPstack.com/" + IP + "?access_key=" + api_key)
    response = r.json()
    print('')
    print(C+"IP :"+W+response['ip'])
    print(C+"Location : " +W+ response['region_name'])
    print(C+"Country : " +W+response['country_name'])
    print(C+"Latitude :"+W+" {latitude}".format(**response))
    print(C+"Longitude :"+W+" {longitude}".format(**response))
    if input(C+"Want More Whois Details (Y/N): "+W):
        whois_more(IP)
    if response['latitude'] and response['longitude']:
        lats = response['latitude']
        lons = response['longitude']
    maps_url = "https://maps.google.com/maps?q=%s,+%s" % (lats, lons)
    print("")
    openWeb = input("Open GPS location in web broser? (Y/N) ")
    if openWeb.upper() == 'Y':
        webbrowser.open(maps_url, new=2)
    else:
        pass

def read_multiple_IP(IP_file):
    lats = []
    lons = []
    try:
        f = open(IP_file, "r")
        f1 = f.readlines()
        print ( W + '[+]' + G + "Processing....." + '\n')
        for line in f1:
            IP=re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$",line)
            IP=IP.group()
            r = requests.get("http://api.IPstack.com/" + IP + "?access_key=" + api_key)
            response = r.json()
            if response['latitude'] and response['longitude']:
                lats.append(response['latitude'])
                lons.append(response['longitude'])
        heat_map(lats,lons)
    except IOError:
        print(R+"ERROR : File Does not Exist\n")
        IPHeatmap()


def heat_map(lats,lons):
    gmap3 = gmplot.GoogleMapPlotter(20.5937, 78.9629, 5)
    gmap3.heatmap(lats,lons)
    gmap3.scatter(lats,lons, '#FF0000', size=50, marker=False)
    gmap3.plot(lats,lons, 'cornflowerblue', edge_width = 3.0)
    save_location = input(C+"Enter the location to save file : "+W)
    gmap3.apikey = gmap()
    location = save_location + "/heatmap.html"
    gmap3.draw(location)
    print(G+"Heatmap saved at " + location)
    openWeb = input(C+"Open Heatmap in web broser? (Y/N) : "+W)
    if openWeb.upper() == 'Y':
        webbrowser.open(url=("file:///"+location))
    else:
        pass