Python 带 arg 解析器的端口扫描器不能正确地接受多个参数

Question

我正在尝试从一本名为 "Violent Python." 的很棒但过时的书中更新一些代码，其中许多示例使用了已弃用的 optparse 库，并且在大多数示例中我很容易切换到 argparse，但是，我修改后的代码没有' 如果我使用多个端口参数，则为这个特定的脚本工作。我变得非常沮丧，我试图使用原始示例，但该代码根本不起作用。

什么有效...

python portScan.py -H ipaddress -p 21

[+] Scan results for: (banner)
[-]21/tcp closed for port

什么不起作用...

python portScan.py -H ipaddress -p 21, 22

usage: usage%prog -H  <target host> -p <target port> [-h] [-H TGTHOST]
                                                 [-p [TGTPORT [TGTPORT ...]]]
usage%prog -H  <target host> -p <target port>: error: argument -p/--tgtPort:   invalid int value: '21,'

我意识到这是因为 argparse 为一个参数取多个值并将它们变成一个列表，但我已经尝试了多种方法将列表映射为整数值。他们都失败了。

下面是我的代码...

import argparse
from socket import *
from threading import *

screenLock = Semaphore(value=1)
def connScan(tgtHost, tgtPort):
    try:
        connSkt = socket(AF_INET, SOCK_STREAM)
        connSkt.connect((tgtHost, tgtPort))
        connSkt.send('ViolentPython\r\n')
        results = connSkt.recv(100)
        screenLock.acquire()
        print '[+]%d/tcp open for port '% tgtPort #+ str(tgtPort)
        print '[+] ' + str(results)
    except:
        screenLock.acquire()
        print '[-]%d/tcp closed for port '% tgtPort #+ str(tgtPort)
    finally:
        screenLock.release()
        connSkt.close()
def portScan(tgtHost, tgtPorts):
    try:
        tgtIP = gethostbyname(tgtHost)
    except:
        print "[-] Cannot resolve '%s': Unknown host"%tgtHost
        return
    try:
        tgtName = gethostbyaddr(tgtIP)
        print '\n[+] Scan results for: ' + tgtName[0]
    except:
        print '\n[+] Scan Results for: ' + tgtIP
        setdefaulttimeout(1)
    for tgtPort in tgtPorts:
        t = Thread(target=connScan, args=(tgtHost, tgtPort))
        t.start() 
def main():
    parser = argparse.ArgumentParser("usage%prog " "-H  <target host> -p <target port>")
    parser.add_argument('-H', '--tgtHost', help='specify target host')
    parser.add_argument('-p', '--tgtPort', nargs= '*', type=int, help='specify target port')
    args = parser.parse_args()
    tgtHost = args.tgtHost
    tgtPorts = args.tgtPort #str(args.tgtPort).split(',')
    if (tgtHost == None) | (tgtPorts == None):
        print parser.usage
        exit(0)
    portScan(tgtHost, tgtPorts)
if __name__ == '__main__':
    main()

Answer 1

我试过你的代码。你知道大多数时候我们不使用 , 来分隔 arg 值。相反，我们使用 space:

>python port_scanner.py -H 127.0.0.1 -p 21 22

[+] Scan results for: localhost
[-]21/tcp closed for port
[-]22/tcp closed for port

Python 带 arg 解析器的端口扫描器不能正确地接受多个参数

Python Port Scanner with arg parser doesn't properly take multiple arguments

python-2.7

argparse