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。向你提供你想要做什麼的信息是非常有幫助的。