d84179e42f
* Remove unused python imports These changes have been auto-generated with: pip3 install isort autoflake isort -rc -sl . autoflake --remove-all-unused-imports -i -r . isort -rc -m 3 . Signed-off-by: Radostin Stoyanov <rstoyanov@fedoraproject.org> * python: remove redundant parenthesis Signed-off-by: Radostin Stoyanov <rstoyanov@fedoraproject.org>
86 lines
2.1 KiB
Python
Executable File
86 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import socket
|
|
import sys
|
|
from time import sleep
|
|
|
|
from scapy.all import (
|
|
IP,
|
|
UDP,
|
|
Ether,
|
|
FieldLenField,
|
|
IntField,
|
|
IPOption,
|
|
Packet,
|
|
PacketListField,
|
|
ShortField,
|
|
get_if_hwaddr,
|
|
get_if_list,
|
|
sendp
|
|
)
|
|
from scapy.layers.inet import _IPOption_HDR
|
|
|
|
|
|
def get_if():
|
|
ifs=get_if_list()
|
|
iface=None # "h1-eth0"
|
|
for i in get_if_list():
|
|
if "eth0" in i:
|
|
iface=i
|
|
break;
|
|
if not iface:
|
|
print("Cannot find eth0 interface")
|
|
exit(1)
|
|
return iface
|
|
|
|
class SwitchTrace(Packet):
|
|
fields_desc = [ IntField("swid", 0),
|
|
IntField("qdepth", 0)]
|
|
def extract_padding(self, p):
|
|
return "", p
|
|
|
|
class IPOption_MRI(IPOption):
|
|
name = "MRI"
|
|
option = 31
|
|
fields_desc = [ _IPOption_HDR,
|
|
FieldLenField("length", None, fmt="B",
|
|
length_of="swtraces",
|
|
adjust=lambda pkt,l:l*2+4),
|
|
ShortField("count", 0),
|
|
PacketListField("swtraces",
|
|
[],
|
|
SwitchTrace,
|
|
count_from=lambda pkt:(pkt.count*1)) ]
|
|
|
|
|
|
def main():
|
|
|
|
if len(sys.argv)<3:
|
|
print('pass 2 arguments: <destination> "<message>"')
|
|
exit(1)
|
|
|
|
addr = socket.gethostbyname(sys.argv[1])
|
|
iface = get_if()
|
|
|
|
pkt = Ether(src=get_if_hwaddr(iface), dst="ff:ff:ff:ff:ff:ff") / IP(
|
|
dst=addr, options = IPOption_MRI(count=0,
|
|
swtraces=[])) / UDP(
|
|
dport=4321, sport=1234) / sys.argv[2]
|
|
|
|
# pkt = Ether(src=get_if_hwaddr(iface), dst="ff:ff:ff:ff:ff:ff") / IP(
|
|
# dst=addr, options = IPOption_MRI(count=2,
|
|
# swtraces=[SwitchTrace(swid=0,qdepth=0), SwitchTrace(swid=1,qdepth=0)])) / UDP(
|
|
# dport=4321, sport=1234) / sys.argv[2]
|
|
pkt.show2()
|
|
#hexdump(pkt)
|
|
try:
|
|
for i in range(int(sys.argv[3])):
|
|
sendp(pkt, iface=iface)
|
|
sleep(1)
|
|
except KeyboardInterrupt:
|
|
raise
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|