Port_Scan/scan_4_1.py at master · windard/Port_Scan · 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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
# -*- coding: utf-8 -*-

import socket
import argparse
import threading
from Queue import Queue


ip2num = lambda x: sum(
    [256 ** j * int(i) for j, i in enumerate(x.split('.')[::-1])])
num2ip = lambda x: '.'.join(
    [str(x / (256 ** i) % 256) for i in range(3, -1, -1)])


def scan(host, port, show):
    s = socket.socket()
    protocolname = 'tcp'
    s.settimeout(0.1)
    if s.connect_ex((host, port)) == 0:
        try:
            print "%s:%4d open => service name: %s" % (
                host, port, socket.getservbyport(port, protocolname))
        except:
            print '%s:%4d open => service name: No Found' % (host, port)
    elif show:
        print port, 'Close'
    s.close()


def udp_scan(host, port, show):
    udpsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    protocolname = 'udp'
    udpsock.settimeout(0.6)
    freq = 0
    for j in xrange(3):
        udpsock.sendto("", (host, port))
        try:
            data, addr = udpsock.recvfrom(1024)
        except socket.timeout:
            freq += 1
        except Exception, e:
            if e.errno == 10054:
                pass
            else:
                print tuple(e)
    if freq == 3:
        try:
            print "%s:%4d open => udp service name: %s" % (
                host, port, socket.getservbyport(port, protocolname))
        except:
            print "%s:%4d open => udp service name: %s" % (
                host, port, "No Found")
    elif show:
        print port, 'Close'
    udpsock.close()


def writeQ(queue, host=None, host_start=None, host_end=None, port_start=None,
           port_end=None):
    if host:
        for port in xrange(port_start, port_end):
            queue.put((num2ip(host), port))
    else:
        for host in xrange(ip2num(host_start), ip2num(host_end)):
            for port in xrange(port_start, port_end):
                queue.put((num2ip(host), port))


def readQ(queue, show, udp):
    while not queue.empty():
        try:
            host, port = queue.get()
            if udp:
                udp_scan(host, port, show)
            else:
                scan(host, port, show)
        finally:
            queue.task_done()


def port_scan(host, host_start, host_end, port, port_start, port_end,
              thread_num, show, udp):
    q = Queue(500)
    if port != 0:
        if host != '127.0.0.1':
            threading.Thread(target=writeQ, args=(
                q, host, None, None, port, port + 1)).start()
        else:
            threading.Thread(target=writeQ, args=(
                q, None, host_start, host_end, port, port + 1)).start()
    else:
        if host != '127.0.0.1':
            threading.Thread(target=writeQ, args=(
                q, host, None, None, port_start, port_end)).start()
        else:
            threading.Thread(target=writeQ, args=(
                q, None, host_start, host_end, port_start, port_end)).start()

    for thread in xrange(thread_num):
        threading.Thread(target=readQ, args=(q, show, udp)).start()
    q.join()


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="input your host and port")
    parser.add_argument("-o", "--on", help="show close", action="store_true")
    parser.add_argument("-u", "--udp", help="UDP scan", action="store_true")
    parser.add_argument("--host", help="chose host", action="store",
                        default='127.0.0.1', dest="host")
    parser.add_argument("--host_start", help="chose host_start",
                        action="store", default='127.0.0.1', dest="host_start")
    parser.add_argument("--host_end", help="chose host_end", action="store",
                        default='127.0.0.2', dest="host_end")
    parser.add_argument("--port", help="chose port", action="store",
                        default=0, type=int, dest="port")
    parser.add_argument("--port_start", help="chose port port_start",
                        action="store", type=int, default=0, dest="port_start")
    parser.add_argument("--port_end", help="chose port port_end",
                        action="store", type=int, default=512, dest="port_end")
    parser.add_argument("--thread", help="how much thread", action="store",
                        type=int, default=4, dest="thread")
    args = parser.parse_args()
    host = args.host
    host_start = args.host_start
    host_end = args.host_end
    port = args.port
    port_start = args.port_start
    port_end = args.port_end
    thread_num = args.thread
    show = args.on
    udp = args.udp
    port_scan(host, host_start, host_end, port, port_start, port_end,
              thread_num, show, udp)