[python] python扫描网段IP

python 网段IP扫描

1. 首先知道自己的IP，然后构造这个网段内的IP，使用list来构建。
2. 使用前需要注意2.7中对dict和list的处理。list是没有keys的
3. 需要注意对ip的split操作来判断，split后来判断输入的Ip范围，可以用掩码的形式来进行范围确认。
4. 传入参数的处理，这个还不完善，需要对输入的ip段和端口段进行判断，然后进行扫描。
5. 超时时间的处理
   因为在测试过程中有的设备没有在线设备，需要设置socket去connect的timeout时间，而且这个设置必须在connect之前进行，否则不生效。

#!/usr/bin/env python
#coding=utf-8
from socket import *
import os
import sys

def getIPs(IP_prefix):
    list=[]
    k = 0
    sepIps = IP_prefix.split('.')
    if(len(sepIps)==4):
        list.insert(k,IP_prefix)
        return list
    elif(len(sepIps)==3):
        for i in range (1,254):
            list.insert(k,IP_prefix+'.'+str(i))
        return list

    for i in range (1,254):
        for j in range (1,254):
            list.insert(k,IP_prefix+'.'+str(i)+'.'+str(j))
            k += 1
    return list

def getPorts():
    list = []
    k = 0
    for i in range(1,65535):
        list.insert(k,i)
        k += 1
    return list

def getSpecifiedPorts(givenPorts):
    if(len(givenPorts)==0):
        return [22,23,3128,8192]
    return givenPorts

fileObj = open("results.txt",'w')


if __name__ == "__main__":
    """judge the parameters, need more careful design here.
        in """
    ips = getIPs(sys.argv[1])
    if(len(sys.argv)==3):
        ports = getSpecifiedPorts(sys.argv[2].split(','))
    if(len(sys.argv)==2):
        ports = getPorts()

    print len(ports)
    flag = 0 #indict if result should be output to files
    for i in range(len(ips)):
        result = str(ips[i]) #record results
        for p in range(len(ports)):
            try:
                tcpCliSock = socket(AF_INET,SOCK_STREAM)
                print str(ips[i])+"......."+ports[p]
                tcpCliSock.settimeout(3)
                tcpCliSock.connect((ips[i],int(ports[p]))) #connect to ip,port
                tcpCliSock.close() 
                del tcpCliSock
                result = result+" "+ports[p]
                flag = 1
            except error:
                print str(ips[i])+"failed ports "+ports[p]

        if(flag == 1):
            fileObj.writelines(result)
            flag=0
    fileObj.close()