Python批量掃描服務器指定端口狀態

閑來無事用Python寫了一個簡陋的端口掃描腳本，其簡單的邏輯如下：

1. python DetectHostPort.py iplist.txt(存放着需要掃描的IP地址列表的文本，每行一個地址)

2. 輸入指定的掃描端口，掃描后完成后將生成日志文件。

 

源碼如下：

#!/usr/bin/env python

import sys
import time
import socket


def getaddresslist(addr):
    """
    getaddresslist(addr) -> IP address file

    IP address read from the file.
    """
    try:
        with open(addr, "r") as ip_list:
            lines = ip_list.readlines()
            address = [line.strip() for line in lines]
        return address
    except (IOError, IndexError), e:
        return str(e)


def scan(ip_list, port):
    """
    scan() -> getaddresslist()

    getaddresslist() function returns the IP address of the list.
    """
    if not isinstance(ip_list, list):
        sys.exit("Function getaddresslist() return error message: %s" % ip_list)

    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(1)
    f = open('E:\programming\public\python\scan.log', 'ab')
    for addr in ip_list:
        host = (addr, int(port))
        try:
            s.connect(host)
            f.write("Host %s:%s connection success. \n" % (host[0], host[1]))
        except Exception, e:
            f.write("Host %s:%s connection failure: %s. \n" % (host[0], host[1], e))
    f.close()
    s.close()
    return True


if __name__ == '__main__':

    addrs = sys.argv[1]
    isNone = True
    while isNone:
        scanport = raw_input("Enter the scan port: ")
        if scanport:
            isNone = False
        else:
            continue
    scan(getaddresslist(addrs), scanport)

運行時只能掃描一個端口，但是可以對代碼進行修改，擴展為掃描多個端口。