我不明白为什么我的 Port Scanner Python 程序出错

Question

我最近开始学习 Python，并尝试使用 Python 制作端口扫描器，但遇到了一些错误。有人可以帮助我了解我哪里出错了吗？

#!/usr/bin/python3

from socket import *
import optparse
from threading import *

def connScan(tgHost,tgPort):
        try:
                sock=socket(AF_NET,SOCK_STREAM)
                sock.connect((tgHost,tgPort))
                print('[+]%d/tcp Open'%tgPort)
        except:
                print('[-]%d/tcp Closed'%tgPort)
        finally:
                sock.close()

def portscan(tgHost,tgPort):
        try:
                tgIP=gethostbyname(tgHost)
        except:
                print('[!] Unkonown Host %s'%tgHost)
        try:
                tgName=gethostbyaddr(tgIP)
                print('[+] Scan Result :' + tgName[0])
        except:
                print('[+]Scan Reults for:' + tgIP)
        setdefaulttimeout(0)
        for port in tgPort:
                t=Thread(target=connScan,args=(tgHost,int(port)))
                t.start()

def main():
        parser=optparse.OptionParser("Usage of Program :" + " -H <target host> -p <target port> ")
        parser.add_option('-H',dest='tgHost',type='string',help='specify target host')
        parser.add_option('-p',dest='tgPort',type='string',help='specify target port separated by comma')
        (options,args)=parser.parse_args()
        tgHost=options.tgHost
        tgPort=str(options.tgPort).split(',')

        if (tgHost==None) | (tgPort[0]==None):
                print (parser.usage)
                exit(0)

        portscan(tgHost,tgPort)

if __name__=='__main__':
        main()

示例输入：

python portscan.py -H 192.168.0.102 -p 80

样本输出：

[+]Scan Reults for:192.168.0.102 
    [-]80/tcp Closed
    Exception in thread Thread-1:
    Traceback (most recent call last):
      File "C:\python3.9.4\lib\threading.py", line 954, in _bootstrap_inner
        self.run()
      File "C:\python3.9.4\lib\threading.py", line 892, in run
        self._target(*self._args, **self._kwargs)
      File "C:\Users\DEBGANDHAR\pytools\adv.py", line 15, in connScan
        sock.close()
    UnboundLocalError: local variable 'sock' referenced before assignment

Answer 1

错误信息应该是不言而喻的；您尝试在finally: 块中使用的变量未定义，因为try: 块中的错误发生在分配变量之前。

即使您刚刚发现无法解析 IP 地址，您也会尝试使用 tgIP。

一般来说，您应该尽可能避免使用import *；您的代码仅在少数地方使用socket，因此将其拼写出来会带来一些不便，但它会使您的代码更清晰。 threading 同上。

同样，您应该避免覆盖except: 块，并且只捕获您真正知道如何处理的异常。

optparse 模块已过时，在新代码中通常应替换为argparse；但是您的参数首先不是可选的，所以也许不要为这个琐碎的任务使用单独的模块。

#!/usr/bin/python3

import socket as s
import threading

def connScan(tgHost, tgPort):
        sock = None
        try:
                # Notice typo fix AF_INET
                sock = s.socket(AF_INET, SOCK_STREAM)
                sock.connect((tgHost,tgPort))
                print('[+]%d/tcp Open'%tgPort)
        except s.error:
                print('[-]%d/tcp Closed'%tgPort)
        finally:
                if sock is not None:
                    sock.close()

def portscan(tgHost,tgPorts):
        try:
                tgIP = s.gethostbyname(tgHost)
        except s.gaierror:
                print('[!] Unknown Host %s'%tgHost)
                # Give up
                return None
        try:
                tgName = s.gethostbyaddr(tgIP)
                print('[+] Scan Result :' + tgName[0])
        except s.gaierror:
                print('[+]Scan Results for:' + tgIP)
        s.setdefaulttimeout(0)
        for port in tgPorts:
                t = threading.Thread(target=connScan,args=(tgHost,int(port)))
                t.start()

def main():
    from sys import argv
    import logging
    if len(argv) < 3:
        logging.error("Usage of Program: %s <target host> <target port> ...", argv[0].split('/')[-1])
        exit(2)
    portscan(argv[1], argv[2:])

if __name__=='__main__':
        main()

请注意端口列表现在是如何以空格分隔的。 （还要注意“分离”的正确拼写。）