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

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

它不存在 - ＆gt; false

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

它存在 - ＆gt; true

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

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

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

无论哪种方式都可以查看argparse tutorial。向您提供有关您想要做的事情的信息非常有用。

Python - 无法识别的参数错误，带有警告消息

1 个答案:

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

如果要选择随机端口