Python - 无法识别的参数错误与警告消息

Question

我在run.py脚本中有以下内容。

import time 

NUM_PACKETS = 500 

import random 
import argparse 

import threading 
from scapy.all import sniff 
from scapy.all import Ether, IP, IPv6, TCP 

parser = argparse.ArgumentParser(description='run_test.py') 
parser.add_argument('--random-dport', 
        help='Use a random TCP dest port for each packet', 
        action="store_true", default=False) 
args = parser.parse_args() 

class PacketQueue: 
    def __init__(self): 
     self.pkts = [] 
     self.lock = threading.Lock() 
     self.ifaces = set() 

    def add_iface(self, iface): 
     self.ifaces.add(iface) 

    def get(self): 
     self.lock.acquire() 
     if not self.pkts: 
      self.lock.release() 
      return None, None 
     pkt = self.pkts.pop(0) 
     self.lock.release() 
     return pkt 

    def add(self, iface, pkt): 
     if iface not in self.ifaces: 
      return 
     self.lock.acquire() 
     self.pkts.append((iface, pkt)) 
     self.lock.release() 

queue = PacketQueue() 

def pkt_handler(pkt, iface): 
    if IPv6 in pkt: 
     return 
    queue.add(iface, pkt) 

class SnifferThread(threading.Thread): 
    def __init__(self, iface, handler = pkt_handler): 
     threading.Thread.__init__(self) 
     self.iface = iface 
     self.handler = handler 

    def run(self): 
     sniff(
      iface = self.iface, 
      prn = lambda x: self.handler(x, self.iface) 
     ) 

class PacketDelay: 
    def __init__(self, bsize, bdelay, imin, imax, num_pkts = 100): 
     self.bsize = bsize 
     self.bdelay = bdelay 
     self.imin = imin 
     self.imax = imax 
     self.num_pkts = num_pkts 
     self.current = 1 

    def __iter__(self): 
     return self 

    def next(self): 
     if self.num_pkts <= 0: 
      raise StopIteration 
     self.num_pkts -= 1 
     if self.current == self.bsize: 
      self.current = 1 
      return random.randint(self.imin, self.imax) 
     else: 
      self.current += 1 
      return self.bdelay 


pkt = Ether()/IP(dst='10.0.0.1', ttl=64)/TCP() 

port_map = { 
    1: "veth3", 
    2: "veth5", 
    3: "veth7" 
} 

iface_map = {} 
for p, i in port_map.items(): 
    iface_map[i] = p 

queue.add_iface("veth3") 
queue.add_iface("veth5") 

for p, iface in port_map.items(): 
    t = SnifferThread(iface) 
    t.daemon = True 
    t.start() 

import socket 

send_socket = socket.socket(socket.AF_PACKET, socket.SOCK_RAW, 
          socket.htons(0x03)) 
send_socket.bind((port_map[3], 0)) 

delays = PacketDelay(10, 5, 25, 100, NUM_PACKETS) 
ports = [] 
print "Sending", NUM_PACKETS, "packets ..." 
for d in delays: 
    # sendp is too slow... 
    # sendp(pkt, iface=port_map[3], verbose=0) 
    if args.random_dport: 
     pkt["TCP"].dport = random.randint(1025, 65535) 
    send_socket.send(str(pkt)) 
    time.sleep(d/1000.) 
time.sleep(1) 
iface, pkt = queue.get() 
while pkt: 
    ports.append(iface_map[iface]) 
    iface, pkt = queue.get() 
print ports 
print "DISTRIBUTION..." 
for p in port_map: 
    c = ports.count(p) 
    print "port {}: {:>3} [ {:>5}% ]".format(p, c, 100. * c/NUM_PACKETS) 

我尝试使用以下命令

./run_test.py '--random-dport' 2 

它投掷一个无法识别的参数误差并显示以下消息运行脚本。

ubuntu@workshop:~/p4lang/tutorials/workshop_/mp$ sudo ./run.py 2 
WARNING: No route found for IPv6 destination :: (no default route?) 
usage: run.py [-h] [--random-dport] 
run_test.py: error: unrecognized arguments: 2 
ubuntu@workshop:~/p4lang/tutorials/workshop/mp$ 

这里有什么问题。我假设其余的run.py代码很好，问题在于以上几行。如果需要，我可以添加其余的代码。我很长时间以来一直坚持这一点。任何投入将帮助我！

Answer 1

如果要指定一个特定的端口（例如2）

我不知道为什么你使用action="store_true"。

只是做（而不是你的add_argument，该代码的其余部分是相同的）：

parser.add_argument("--random-dport", type=int, 
       help='Use a random TCP dest port for each packet', 
       default=1) #default to port 1 

然后你就可以做访问端口：

./run_test.py '--random-dport' 2 
args.random-dport 
>> 2 

---

如果您想要选择一个随机端口

如果你真的想要一个随机端口的真/假标志，因为它看起来你的代码正在努力实现的，你可以使用你原来的代码，但后来经过2是没有意义的，因为存在random-dport标志将存储真正进入变量：

./run_test.py 
args.random-dport 
>>False 

这不是present->假

./run_test.py '--random-dport' 
args.random-dport 
>>True 

这是present->真

./run_test.py '--random-dport' 2 
>>error 

它有一个额外的值 - >错误。

这里2没有意义，你指定你想要一个RANDOM dport，这意味着你不能选择一个特定的。上一节让你选择一个特定的端口。）

---

无论哪种方式检查argparse tutorial。向你提供你想要做什么的信息是非常有帮助的。