ScanDeviceInfoInLan/HostListLAN.py at master · sujayy1983/ScanDeviceInfoInLan · 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
"""
Description: Scan ipaddresses in your Home LAN. Get OS and currently up configuration
"""

__author__ = 'Sujayyendhiren Srinivasamurthi'
__email__ = 'sujayy1983@gmail.com'

import os
import nmap
import socket
import traceback
import subprocess
from netaddr import *

class LANNetworkInfo(object):
    """ Use nmap and get network data."""

    def __init__(self):
        """ Initialize object."""

        #Following parameters are candidate for config files
        self.networkRange = '192.168.0.0/28'
        self.devicesUp = []
        self.portRange = '22-80'
        self.output = None
        self.count = 0
        self.separator = '\n==========================\n'

    def GetHostListInNetwork(self):
        """ Get a list of IPs in a network. Copied this function from internet."""

        try:
            ipNet = IPNetwork(self.networkRange)
            hosts = list(ipNet)
            if len(hosts) > 2:
                hosts.remove(ipNet.broadcast)
                hosts.remove(ipNet.network)
            hostList = [str(host) for host in hosts]

            nm = nmap.PortScanner()

            for host in hostList:
                nm.scan(host, self.portRange)
                hosts_list = [(x, nm[x]['status']['state']) for x in nm.all_hosts()]

                for host, status in hosts_list:
                    self.devicesUp.append(host)
                    print('{0}:{1}'.format(host, status))
        except:
            exception = traceback.print_exc()
            print exception

    def logOutput(self, buffer, name=None):

        fileLog = None
        if name == None:
            #Check if file exists and create the one with new one.
            fileLog = open( './logs/' + str(name), 'w')
        else:
            fileLog = open( './logs/' + name, 'w')

        fileLog.write(buffer)
        fileLog.close()

    def identifyOS(self):
        try:
            print 'In progres ... ',
            for device in self.devicesUp:
                print '.',
                cmd = 'sudo nmap -PN -O ' + device
                command = cmd.split(' ')
                process = subprocess.Popen( command, stdout=subprocess.PIPE)
                out, err = process.communicate()
                print '..',
                self.output = self.separator + "Output " + device + self.separator + str(out) + "\nError:\n" + str(err) + "\n"
                self.logOutput(self.output, 'Out'+device)
        except:
            exception = traceback.print_exc()
            print exception

if __name__ == '__main__':

    objLANInfo = LANNetworkInfo()
    objLANInfo.GetHostListInNetwork()
    objLANInfo.identifyOS()