1 #!/usr/bin env python

  2 #-*-coding:utf-8-*-

  3 import socket,time,threading

  4 import signal,os,argparse,re

  5 #---------------------------------------------信号侦听模块------------------------------------------

  6 def signal_handler(signal, frame):

  7     # print(success)

  8     if success:

  9         for ip in success:

 10             print()

 11             print('扫描结束,主机%s,开放的端口有%s'%(ip,success[ip]))

 12     os._exit(0)

 13 signal.signal(signal.SIGINT, signal_handler)

 14 #----------------------------------------------扫描函数-------------------------------------------------

 15 socket.setdefaulttimeout(0.1)

 16 success = {}

 17 port_li = []

 18 def scan(HOST,port,q):

 19     startport,endport = port

 20     for port in range (startport,endport):

 21         s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

 22         try:

 23             s.connect((HOST,port))

 24             print('%s-%d端口开启'%(HOST,port))

 25             port_li.append(port)

 26             success[HOST]=port_li

 27         except:

 28             pass

 29 def thread(ip,q,startport,maxport):

 30     li = []

 31     startport = 1

 32     endport = startport+200

 33     if endport > maxport:

 34         endport = maxport

 35     dup=[(startport,endport)]

 36     while endport < maxport:

 37         startport,endport=endport,endport+200

 38         if endport >= maxport:

 39             endport = maxport

 40         dup.append((startport,endport))

 41     for port in dup:

 42         t1 = threading.Thread(target=scan,args=(ip,port,q))

 43         t1.start()

 44         li.append(t1)

 45     for t in li:

 46         t.join()

 47     if success:

 48         q.put(success)

 49

 50

 51 if __name__=='__main__':

 52     from multiprocessing import Process,Queue

 53     #接收参数

 54     parser = argparse.ArgumentParser(description='ip_scan')

 55     parser.add_argument('-t',action = 'store',dest='tip')

 56     parser.add_argument('-p',action='store',dest='port')

 57     args= parser.parse_args()

 58     try:

 59         ip = args.tip

 60         port = args.port

 61         dip = re.findall('\d+',ip)

 62         startport,maxport = port.split('-')

 63     except:

 64         print('-----------------example--------------')

 65         print('tcpportstatus.py -t 10.0.0.0/24 -p 0-65535')

 66         exit(0)

 67     q = Queue()

 68     li = []

 69     ip_list = []

 70     if int(dip[4]) ==0 or int(dip[4])>32:

 71         print('----------------掩码错误----------------------')

 72         exit(0)

 73     #计算地址数量

 74     subnet = int(dip[4])%8

 75     if subnet !=0:

 76         subnet_count = 2**(8-subnet)

 77     else:

 78         subnet_count = 2**subnet

 79     #创建ip地址列表

 80     if 32 >= int(dip[4])>24:

 81         #验证子网划分是否错误

 82         if int(dip[3])%subnet_count != 0:

 83             print('-----------ip格式错误-----------')

 84             exit(0)

 85         startip = int(dip[3])

 86         endip = startip + subnet_count

 87         dip[3] = '%s'

 88         dip = '.'.join(dip[0:4])

 89         for d in range(startip,endip):

 90             ip_list.append(dip%d)

 91     elif int(dip[4])>16:

 92         if int(dip[2])%subnet_count != 0:

 93             print('-----------ip格式错误-----------')

 94             exit(0)

 95         startip = int(dip[2])

 96         endip = startip + subnet_count

 97         dip[2] = '%s'

 98         dip[3] = '%s'

 99         dip = '.'.join(dip[0:4])

100         for c in range(startip,endip):

101             dip_c = dip%(c,'%s')

102             for d in range(1,256):

103                 dip_d = dip_c%d

104                 ip_list.append(dip_d)

105     elif int(dip[4])>8:

106         if int(dip[1])%subnet_count != 0:

107             print('-----------ip格式错误-----------')

108             exit(0)

109         startip = int(dip[1])

110         endip = startip + subnet_count

111         dip[1] = '%s'

112         dip[2] = '%s'

113         dip[3] = '%s'

114         dip = '.'.join(dip[0:4])

115         for b in range(startip,endip):

116             dip_b = dip%(b,'%s','%s')

117             for c in range(0,256):

118                 dip_c = dip_b%(c,'%s')

119                 for d in range(1,256):

120                     dip_d = dip_c%d

121                     ip_list.append(dip_d)

122     elif int(dip[4])>0:

123         if int(dip[1])%subnet_count != 0:

124             print('-----------ip格式错误-----------')

125             exit(0)

126         startip = int(dip[0])

127         endip = startip + subnet_count

128         dip[0] = '%s'

129         dip[1] = '%s'

130         dip[2] = '%s'

131         dip[3] = '%s'

132         dip = '.'.join(dip[0:4])

133         for a in range(startip,endip):

134             dip_a = dip%(a,'%s','%s','%s')

135             for b in range(0,256):

136                 dip_b = dip_a%(b,'%s','%s')

137                 for c in range(0,256):

138                     dip_c = dip_b%(c,'%s')

139                     for d in range(0,256):

140                         dip_d = dip_c%d

141                         ip_list.append(dip_d)

142     print('-----------------------------------开始扫描------------------------------------')

143     starttime = time.time()

144     print('扫描地址数量%s，端口%s-%s'%(len(ip_list),int(startport),int(maxport)))

145

146     for address in ip_list:

147         t1 = Process(target=thread,args=(address,q,int(startport),int(maxport)))

148         t1.start()

149         li.append(t1)

150     for t in li:

151         t.join()

152     usetime = time.time()-starttime

153     print('--------------------------扫描结束用时%s秒----------------------'%int(usetime))

154     if not q.empty():

155         while not q.empty():

156             q_dic = q.get()

157             for ip in q_dic:

158                 print('扫描结束,此次扫描的主机%s,开放的端口有%s'%(ip,q_dic[ip]))

159     else:

160         print()

161         print('扫描结束,此次扫描的主机,没有开放的TCP端口')