def __init__(self):
self.server_vxlan_values = VXLANGPE()
self.server_base_values = BASEHEADER()
self.server_ctx_values = CONTEXTHEADER()
self.server_ethernet_values = ETHHEADER()
self.server_trace_values = TRACEREQHEADER()
self.inner_header = InnerHeader()
def __init__(self, loop, encapsulate_type, encapsulate_header_values,
base_header_values, ctx_header_values, dest_addr, dest_port):
self.transport = None
self.loop = loop
self.encapsulate_header_values = encapsulate_header_values
self.base_header_values = base_header_values
self.ctx_header_values = ctx_header_values
# Need to implement GRE decode in decode.py
# self.server_gre_values = GREHEADER()
self.server_base_values = BASEHEADER()
self.server_ctx_values = CONTEXTHEADER()
self.dest_addr = dest_addr
# self.remote_sff_port = remote_sff_port
self.encapsulate_type = encapsulate_type
def __init__(self, loop, vxlan_header_values, base_header_values,
ctx_header_values, trace_req_header_values, dest_addr,
dest_port, num_trace_hops):
self.transport = None
self.loop = loop
self.vxlan_header_values = vxlan_header_values
self.base_header_values = base_header_values
self.ctx_header_values = ctx_header_values
self.trace_req_header_values = trace_req_header_values
self.server_trace_values = TRACEREQHEADER()
self.server_ctx_values = CONTEXTHEADER()
self.server_vxlan_values = VXLANGPE()
self.server_base_values = BASEHEADER()
self.dest_addr = dest_addr
self.dest_port = dest_port
self.hop_count = 0
self.num_trace_hops = num_trace_hops
def main(argv):
"""
Example:
python3.4 sff_client.py --remote-sff-ip 10.0.1.41 --remote-sff-port 4789 --sfp-id 1 --sfp-index 255
--encapsulate gpe-nsh-ipv4
python3.4 sff_client.py --remote-sff-ip 192.168.1.27 --local-port 6633 --remote-sff-port 4790 --sfp-id 3
--sfp-index 255 --encapsulate vxlan-nsh-ethernet-legacy --inner-src-ip 192.168.0.1 --inner-dest-ip 192.168.1.13
--inner-src-port 10000 --inner-dest-port 20000 --ctx1 192.168.1.7 --inner-src-eth 3c:15:c2:c9:4f:bc
--inner-dest-eth 08:00:27:b6:b0:58
Trace Example:
python3.4 sff_client.py --remote-sff-ip 10.0.1.41 --remote-sff-port 4789 --sfp-id 1 --sfp-index 255 \
--trace-req --num-trace-hops 1
:param argv:
:return:
"""
def handler(signum=None, frame=None):
print("Signal handler called with signal {}".format(signum))
loop.call_soon_threadsafe(loop.stop)
time.sleep(1)
loop.call_soon_threadsafe(loop.close)
time.sleep(1) # here check if process is done
print("Wait done")
sys.exit(0)
# global base_values
# Some Good defaults
remote_sff_port = 4790
local_port = 4790
remote_sff_ip = "127.0.0.1"
local_ip = "0.0.0.0"
sfp_id = 0x000001
sfp_index = 3
trace_req = False
num_trace_hops = 254
inner_src_ip = "192.168.0.1"
inner_dest_ip = "192.168.0.2"
inner_src_port = 10000
inner_dest_port = 20000
encapsulate = 'gpe-nsh-ipv4' # make vxlan-gpe encapsulation default
ctx1 = ctx2 = ctx3 = ctx4 = 0
inner_src_eth_list = [0x3c, 0x15, 0xc2, 0xc9, 0x4f, 0xbc]
inner_dest_eth_list = [0x08, 0x00, 0x27, 0xb6, 0xb0, 0x58]
try:
logging.basicConfig(level=logging.INFO)
opt, args = getopt.getopt(argv, "h", [
"help", "local-port=", "local-ip=", "inner-src-ip=",
"inner-dest-ip=", "inner-src-port=", "inner-dest-port=",
"remote-sff-ip=", "remote-sff-port=", "sfp-id=", "sfp-index=",
"trace-req", "num-trace-hops=", "encapsulate=", "ctx1=", "ctx2=",
"ctx3=", "ctx4=", "inner-src-eth=", "inner-dest-eth="
])
except getopt.GetoptError:
print(
"sff_client --help | --local-port | --local-ip | --inner-src-ip | --inner-dest-ip | --inner-src-port | "
"--inner-dest-port | --remote-sff-ip | --ctx1 | --ctx2 | --ctx3 | --ctx4"
"--remote-sff-port | --sfp-id | --sfp-index | --trace-req | --num-trace-hops | --encapsulate"
"--inner-src-eth | --inner-dest-eth")
sys.exit(2)
for opt, arg in opt:
if opt == "--remote-sff-port":
remote_sff_port = arg
continue
if opt in ('-h', '--help'):
print(
"sff_client \n --remote-sff-ip=<IP address of remote SFF> \n "
"--remote-sff-port=<UDP port of remote SFF> \n "
"--sfp-id=<Service Function Path id> \n --sfp-index<SFP starting index> \n "
"--encapsulate=<gpe-nsh-ethernet|gre|gpe-nsh-ipv4|vxlan-nsh-ethernet-legacy> \n "
"--inner-src-ip=<source IP of inner packet> \n --inner-dest-ip=<destination IP of inner packet> \n "
"--ctx1=<context header 1> \n --ctx2=<context header 2> \n --ctx3=<context header 3> \n "
"--ctx4=<context header 4> \n --local-port=<source port> \n --local-ip=<source IP> \n"
"--inner-src-eth=<inner src ethernet address> \n --inner-dest-eth=<inner dest ethernet address>"
)
sys.exit()
if opt == "--inner-src-eth":
inner_src_eth_list = arg.split(':')
if len(inner_src_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_src_eth_list):
inner_src_eth_list[i] = int(val, 16)
else:
logger.error(
"Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--inner-dest-eth":
inner_dest_eth_list = arg.split(':')
if len(inner_dest_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_dest_eth_list):
inner_dest_eth_list[i] = int(val, 16)
else:
logger.error(
"Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--remote-sff-ip":
remote_sff_ip = arg
continue
if opt == "--sfp-id":
sfp_id = arg
continue
if opt == "--sfp-index":
sfp_index = arg
continue
if opt == "--trace-req":
trace_req = True
continue
if opt == "--num-trace-hops":
num_trace_hops = arg
continue
if opt == "--encapsulate":
encapsulate = arg
continue
if opt == "--local-port":
local_port = int(arg)
continue
if opt == "--local-ip":
local_ip = arg
continue
if opt == "--inner-dest-port":
inner_dest_port = int(arg)
continue
if opt == "--inner-dest-ip":
inner_dest_ip = arg
continue
if opt == "--inner-src-port":
inner_src_port = int(arg)
continue
if opt == "--inner-src-ip":
inner_src_ip = arg
continue
if opt == "--ctx1":
ctx1 = arg
continue
if opt == "--ctx2":
ctx2 = arg
continue
if opt == "--ctx3":
ctx3 = arg
continue
if opt == "--ctx4":
ctx4 = arg
continue
loop = asyncio.get_event_loop()
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGABRT]:
signal.signal(sig, handler)
# Can not install SIGHUP in Windows
if platform.system() in ["Linux", "Darwin"]:
signal.signal(signal.SIGHUP, handler)
signal.signal(signal.SIGQUIT, handler)
# # create a raw socket
# euid = os.geteuid()
# sock_raw = None
# if euid != 0:
# print("Script not started as root. Running sudo...")
# args = ['sudo', sys.executable] + sys.argv + [os.environ]
# # the next line replaces the currently-running process with the sudo
# os.execlpe('sudo', *args)
#
# if platform.system() == "Darwin":
# try:
# sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_UDP)
# # sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_RAW)
# # sock_raw.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)
# except socket.error as msg:
# print("Socket could not be created. Error Code : {}".format(msg))
# sys.exit()
# else:
# try:
# sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_RAW)
# except socket.error as msg:
# print("Socket could not be created. Error Code : {}".format(msg))
# sys.exit()
#
# udp_packet = build_udp_packet("", "100.0.3.2", 10000, 4790, "test".encode('utf-8'))
# try:
# sock_raw.sendto(udp_packet, ("100.0.3.2", 0))
# #sock_raw.sendto(udp_packet, int.from_bytes(socket.inet_aton("10.0.1.1"), byteorder='big'))
# except socket.error as msg:
# print(msg)
# sys.exit()
if trace_req:
# MD-type 0x1, OAM set
vxlan_header_values = VXLANGPE(int('00000100', 2), 0, 0x894F,
int('111111111111111111111111', 2), 64)
base_header_values = BASEHEADER(service_path=int(sfp_id),
service_index=int(sfp_index))
ctx_header_values = CONTEXTHEADER()
trace_req_header_values = build_trace_req_header(
OAM_TRACE_REQ_TYPE, 254, remote_sff_ip, 55555)
traceclient = MyTraceClient(loop, vxlan_header_values,
base_header_values, ctx_header_values,
trace_req_header_values, remote_sff_ip,
int(remote_sff_port), int(num_trace_hops))
start_client(
loop,
(str(ipaddress.IPv4Address(trace_req_header_values.ip_4)), 55555),
(remote_sff_ip, int(remote_sff_port)), traceclient)
else:
# Common initializations for all encapsulations types
base_header_values = BASEHEADER(service_path=int(sfp_id),
service_index=int(sfp_index),
proto=NSH_NEXT_PROTO_ETH)
context_headers = process_context_headers(ctx1, ctx2, ctx3, ctx4)
ctx_values = CONTEXTHEADER(context_headers[0], context_headers[1],
context_headers[2], context_headers[3])
inner_header = InnerHeader(inner_src_ip, inner_dest_ip, inner_src_port,
inner_dest_port)
if encapsulate == 'gpe-nsh-ipv4':
# NSH type 1
vxlan_header_values = VXLANGPE()
# override encap type
base_header_values.next_protocol = NSH_NEXT_PROTO_IPV4
udpclient = MyVxlanGpeNshIpClient(loop, vxlan_header_values,
base_header_values, ctx_values,
remote_sff_ip,
int(remote_sff_port),
inner_header)
start_client(loop, (local_ip, local_port),
(remote_sff_ip, remote_sff_port), udpclient)
elif encapsulate == 'gre-nsh-ethernet':
# add GRE header
gre_header_values = GREHEADER(int('0', 2), int('000000000000', 2),
int('000', 2), 0x894F,
int('0000000000000000', 2), 0)
greclient = MyGreNshEthClient(loop, 'GRE/NSH/Ethernet',
gre_header_values,
base_header_values, ctx_values,
remote_sff_ip, int(remote_sff_port))
greclient.send_gre_nsh()
elif encapsulate == 'gpe-nsh-ethernet':
# NSH type 3
vxlan_header_values = VXLANGPE()
ethernet_header_values = ETHHEADER(
inner_src_eth_list[0], inner_src_eth_list[1],
inner_src_eth_list[2], inner_src_eth_list[3],
inner_src_eth_list[4], inner_src_eth_list[5],
inner_dest_eth_list[0], inner_dest_eth_list[1],
inner_dest_eth_list[2], inner_dest_eth_list[3],
inner_dest_eth_list[4], inner_dest_eth_list[5], 0x08, 0x00)
udpclient = MyVxlanGpeNshEthClient(loop, ethernet_header_values,
vxlan_header_values,
base_header_values, ctx_values,
remote_sff_ip,
int(remote_sff_port),
inner_header)
start_client(loop, (local_ip, local_port),
(remote_sff_ip, remote_sff_port), udpclient)
elif encapsulate == 'vxlan-nsh-ethernet-legacy':
# NSH type 3
vxlan_header_values = VXLAN()
ethernet_header_values = ETHHEADER(
inner_src_eth_list[0], inner_src_eth_list[1],
inner_src_eth_list[2], inner_src_eth_list[3],
inner_src_eth_list[4], inner_src_eth_list[5],
inner_dest_eth_list[0], inner_dest_eth_list[1],
inner_dest_eth_list[2], inner_dest_eth_list[3],
inner_dest_eth_list[4], inner_dest_eth_list[5], 0x08, 0x00)
udpclient = MyVxlanNshEthClient(loop, ethernet_header_values,
vxlan_header_values,
base_header_values,
ctx_values, remote_sff_ip,
int(remote_sff_port), inner_header)
start_client(loop, (local_ip, local_port),
(remote_sff_ip, remote_sff_port), udpclient)
else:
logger.error(
"Encapsulation must be specified, e.g. gpe-nsh-ethernet")
def sff_client(argv, message='ping!'):
# global base_values
# Some Good defaults
remote_sff_port = 4790
local_port = 4790
remote_sff_ip = "127.0.0.1"
local_ip = "0.0.0.0"
sfp_id = 0x000001
sfp_index = 3
trace_req = False
num_trace_hops = 254
inner_src_ip = "192.168.0.1"
inner_dest_ip = "192.168.0.2"
inner_src_port = 10000
inner_dest_port = 20000
encapsulate = 'gpe-nsh-ipv4' # make vxlan-gpe encapsulation default
ctx1 = ctx2 = ctx3 = ctx4 = 0
inner_src_eth_list = [0x3c, 0x15, 0xc2, 0xc9, 0x4f, 0xbc]
inner_dest_eth_list = [0x08, 0x00, 0x27, 0xb6, 0xb0, 0x58]
try:
logging.basicConfig(level=logging.INFO)
opt, args = getopt.getopt(argv, "h", [
"help", "local-port=", "local-ip=", "inner-src-ip=",
"inner-dest-ip=", "inner-src-port=", "inner-dest-port=",
"remote-sff-ip=", "remote-sff-port=", "sfp-id=", "sfp-index=",
"trace-req", "num-trace-hops=", "encapsulate=", "ctx1=", "ctx2=",
"ctx3=", "ctx4=", "inner-src-eth=", "inner-dest-eth="
])
except getopt.GetoptError:
print(
"sff_client --help | --local-port | --local-ip | --inner-src-ip | --inner-dest-ip | --inner-src-port | "
"--inner-dest-port | --remote-sff-ip | --ctx1 | --ctx2 | --ctx3 | --ctx4"
"--remote-sff-port | --sfp-id | --sfp-index | --trace-req | --num-trace-hops | --encapsulate"
"--inner-src-eth | --inner-dest-eth")
sys.exit(2)
for opt, arg in opt:
if opt == "--remote-sff-port":
remote_sff_port = arg
continue
if opt in ('-h', '--help'):
print(
"sff_client \n --remote-sff-ip=<IP address of remote SFF> \n "
"--remote-sff-port=<UDP port of remote SFF> \n "
"--sfp-id=<Service Function Path id> \n --sfp-index<SFP starting index> \n "
"--encapsulate=<gpe-nsh-ethernet|gre|gpe-nsh-ipv4|vxlan-nsh-ethernet-legacy> \n "
"--inner-src-ip=<source IP of inner packet> \n --inner-dest-ip=<destination IP of inner packet> \n "
"--ctx1=<context header 1> \n --ctx2=<context header 2> \n --ctx3=<context header 3> \n "
"--ctx4=<context header 4> \n --local-port=<source port> \n --local-ip=<source IP> \n"
"--inner-src-eth=<inner src ethernet address> \n --inner-dest-eth=<inner dest ethernet address>"
)
sys.exit()
if opt == "--inner-src-eth":
inner_src_eth_list = arg.split(':')
if len(inner_src_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_src_eth_list):
inner_src_eth_list[i] = int(val, 16)
else:
logger.error(
"Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--inner-dest-eth":
inner_dest_eth_list = arg.split(':')
if len(inner_dest_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_dest_eth_list):
inner_dest_eth_list[i] = int(val, 16)
else:
logger.error(
"Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--remote-sff-ip":
remote_sff_ip = arg
continue
if opt == "--sfp-id":
sfp_id = arg
continue
if opt == "--sfp-index":
sfp_index = arg
continue
if opt == "--trace-req":
trace_req = True
continue
if opt == "--num-trace-hops":
num_trace_hops = arg
continue
if opt == "--encapsulate":
encapsulate = arg
continue
if opt == "--local-port":
local_port = int(arg)
continue
if opt == "--local-ip":
local_ip = arg
continue
if opt == "--inner-dest-port":
inner_dest_port = int(arg)
continue
if opt == "--inner-dest-ip":
inner_dest_ip = arg
continue
if opt == "--inner-src-port":
inner_src_port = int(arg)
continue
if opt == "--inner-src-ip":
inner_src_ip = arg
continue
if opt == "--ctx1":
ctx1 = arg
continue
if opt == "--ctx2":
ctx2 = arg
continue
if opt == "--ctx3":
ctx3 = arg
continue
if opt == "--ctx4":
ctx4 = arg
continue
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
loop = asyncio.get_event_loop()
# Common initializations for all encapsulations types
base_header_values = BASEHEADER(service_path=int(sfp_id),
service_index=int(sfp_index),
proto=NSH_NEXT_PROTO_ETH)
context_headers = process_context_headers(ctx1, ctx2, ctx3, ctx4)
ctx_values = CONTEXTHEADER(context_headers[0], context_headers[1],
context_headers[2], context_headers[3])
inner_header = InnerHeader(inner_src_ip, inner_dest_ip, inner_src_port,
inner_dest_port)
# NSH type 1
vxlan_header_values = VXLANGPE()
# override encap type
base_header_values.next_protocol = NSH_NEXT_PROTO_IPV4
# udpclient = MyVxlanGpeNshIpClient(loop, vxlan_header_values, base_header_values,
# ctx_values, remote_sff_ip, int(remote_sff_port), inner_header,
# message=message)
# start_client(loop, (local_ip, local_port), (remote_sff_ip, remote_sff_port), udpclient)
packet = build_nsh_header(vxlan_header_values, base_header_values,
ctx_values)
udp_inner_packet = build_udp_packet(inner_header.inner_src_ip,
inner_header.inner_dest_ip,
inner_src_port, inner_dest_port,
message)
transport = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
logger.debug("Client sending packet to %s",
(remote_sff_ip, int(remote_sff_port)))
try:
transport.sendto(packet + udp_inner_packet,
(remote_sff_ip, int(remote_sff_port)))
transport.close()
except socket.error as msg:
logger.error('Failed to send packet. Error Code : ' + str(msg))
transport.close()
sys.exit()
except Exception as e:
logger.error("Error processing client: %s" % str(e))
transport.close()
def main(argv):
"""
Example:
python3.4 sff_client.py --remote-sff-ip 10.0.1.41 --remote-sff-port 4789 --sfp-id 1 --sfp-index 255
--encapsulate gpe-nsh-ipv4
python3.4 sff_client.py --remote-sff-ip 192.168.1.27 --local-port 6633 --remote-sff-port 4790 --sfp-id 3
--sfp-index 255 --encapsulate vxlan-nsh-ethernet-legacy --inner-src-ip 192.168.0.1 --inner-dest-ip 192.168.1.13
--inner-src-port 10000 --inner-dest-port 20000 --ctx1 192.168.1.7 --inner-src-eth 3c:15:c2:c9:4f:bc
--inner-dest-eth 08:00:27:b6:b0:58
Trace Example:
python3.4 sff_client.py --remote-sff-ip 10.0.1.41 --remote-sff-port 4789 --sfp-id 1 --sfp-index 255 \
--trace-req --num-trace-hops 1
:param argv:
:return:
"""
def handler(signum=None, frame=None):
print("Signal handler called with signal {}".format(signum))
loop.call_soon_threadsafe(loop.stop)
time.sleep(1)
loop.call_soon_threadsafe(loop.close)
time.sleep(1) # here check if process is done
print("Wait done")
sys.exit(0)
# global base_values
# Some Good defaults
remote_sff_port = 4790
local_port = 4790
remote_sff_ip = "127.0.0.1"
local_ip = "0.0.0.0"
sfp_id = 0x000001
sfp_index = 3
trace_req = False
num_trace_hops = 254
inner_src_ip = "192.168.0.1"
inner_dest_ip = "192.168.0.2"
inner_src_port = 10000
inner_dest_port = 20000
encapsulate = 'gpe-nsh-ipv4' # make vxlan-gpe encapsulation default
ctx1 = ctx2 = ctx3 = ctx4 = 0
inner_src_eth_list = [0x3c, 0x15, 0xc2, 0xc9, 0x4f, 0xbc]
inner_dest_eth_list = [0x08, 0x00, 0x27, 0xb6, 0xb0, 0x58]
try:
logging.basicConfig(level=logging.INFO)
opt, args = getopt.getopt(argv, "h",
["help", "local-port=", "local-ip=", "inner-src-ip=", "inner-dest-ip=",
"inner-src-port=", "inner-dest-port=", "remote-sff-ip=",
"remote-sff-port=", "sfp-id=", "sfp-index=", "trace-req", "num-trace-hops=",
"encapsulate=", "ctx1=", "ctx2=", "ctx3=", "ctx4=", "inner-src-eth=",
"inner-dest-eth="])
except getopt.GetoptError:
print(
"sff_client --help | --local-port | --local-ip | --inner-src-ip | --inner-dest-ip | --inner-src-port | "
"--inner-dest-port | --remote-sff-ip | --ctx1 | --ctx2 | --ctx3 | --ctx4"
"--remote-sff-port | --sfp-id | --sfp-index | --trace-req | --num-trace-hops | --encapsulate"
"--inner-src-eth | --inner-dest-eth")
sys.exit(2)
for opt, arg in opt:
if opt == "--remote-sff-port":
remote_sff_port = arg
continue
if opt in ('-h', '--help'):
print("sff_client \n --remote-sff-ip=<IP address of remote SFF> \n "
"--remote-sff-port=<UDP port of remote SFF> \n "
"--sfp-id=<Service Function Path id> \n --sfp-index<SFP starting index> \n "
"--encapsulate=<gpe-nsh-ethernet|gre|gpe-nsh-ipv4|vxlan-nsh-ethernet-legacy> \n "
"--inner-src-ip=<source IP of inner packet> \n --inner-dest-ip=<destination IP of inner packet> \n "
"--ctx1=<context header 1> \n --ctx2=<context header 2> \n --ctx3=<context header 3> \n "
"--ctx4=<context header 4> \n --local-port=<source port> \n --local-ip=<source IP> \n"
"--inner-src-eth=<inner src ethernet address> \n --inner-dest-eth=<inner dest ethernet address>")
sys.exit()
if opt == "--inner-src-eth":
inner_src_eth_list = arg.split(':')
if len(inner_src_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_src_eth_list):
inner_src_eth_list[i] = int(val, 16)
else:
logger.error("Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--inner-dest-eth":
inner_dest_eth_list = arg.split(':')
if len(inner_dest_eth_list) == ETHERNET_ADDR_SIZE:
for i, val in enumerate(inner_dest_eth_list):
inner_dest_eth_list[i] = int(val, 16)
else:
logger.error("Ethernet address must be in the form aa:bb:cc:dd:ee:ff")
sys.exit(2)
continue
if opt == "--remote-sff-ip":
remote_sff_ip = arg
continue
if opt == "--sfp-id":
sfp_id = arg
continue
if opt == "--sfp-index":
sfp_index = arg
continue
if opt == "--trace-req":
trace_req = True
continue
if opt == "--num-trace-hops":
num_trace_hops = arg
continue
if opt == "--encapsulate":
encapsulate = arg
continue
if opt == "--local-port":
local_port = int(arg)
continue
if opt == "--local-ip":
local_ip = arg
continue
if opt == "--inner-dest-port":
inner_dest_port = int(arg)
continue
if opt == "--inner-dest-ip":
inner_dest_ip = arg
continue
if opt == "--inner-src-port":
inner_src_port = int(arg)
continue
if opt == "--inner-src-ip":
inner_src_ip = arg
continue
if opt == "--ctx1":
ctx1 = arg
continue
if opt == "--ctx2":
ctx2 = arg
continue
if opt == "--ctx3":
ctx3 = arg
continue
if opt == "--ctx4":
ctx4 = arg
continue
loop = asyncio.get_event_loop()
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGABRT]:
signal.signal(sig, handler)
# Can not install SIGHUP in Windows
if platform.system() in ["Linux", "Darwin"]:
signal.signal(signal.SIGHUP, handler)
signal.signal(signal.SIGQUIT, handler)
# # create a raw socket
# euid = os.geteuid()
# sock_raw = None
# if euid != 0:
# print("Script not started as root. Running sudo...")
# args = ['sudo', sys.executable] + sys.argv + [os.environ]
# # the next line replaces the currently-running process with the sudo
# os.execlpe('sudo', *args)
#
# if platform.system() == "Darwin":
# try:
# sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_UDP)
# # sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_RAW)
# # sock_raw.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)
# except socket.error as msg:
# print("Socket could not be created. Error Code : {}".format(msg))
# sys.exit()
# else:
# try:
# sock_raw = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_RAW)
# except socket.error as msg:
# print("Socket could not be created. Error Code : {}".format(msg))
# sys.exit()
#
# udp_packet = build_udp_packet("", "100.0.3.2", 10000, 4790, "test".encode('utf-8'))
# try:
# sock_raw.sendto(udp_packet, ("100.0.3.2", 0))
# #sock_raw.sendto(udp_packet, int.from_bytes(socket.inet_aton("10.0.1.1"), byteorder='big'))
# except socket.error as msg:
# print(msg)
# sys.exit()
if trace_req:
# MD-type 0x1, OAM set
vxlan_header_values = VXLANGPE(int('00000100', 2), 0, 0x894F, int('111111111111111111111111', 2), 64)
base_header_values = BASEHEADER(service_path=int(sfp_id), service_index=int(sfp_index))
ctx_header_values = CONTEXTHEADER()
trace_req_header_values = build_trace_req_header(OAM_TRACE_REQ_TYPE, 254,
remote_sff_ip, 55555)
traceclient = MyTraceClient(loop, vxlan_header_values, base_header_values,
ctx_header_values, trace_req_header_values, remote_sff_ip, int(remote_sff_port),
int(num_trace_hops))
start_client(loop, (str(ipaddress.IPv4Address(trace_req_header_values.ip_4)), 55555),
(remote_sff_ip, int(remote_sff_port)), traceclient)
else:
# Common initializations for all encapsulations types
base_header_values = BASEHEADER(service_path=int(sfp_id), service_index=int(sfp_index),
proto=NSH_NEXT_PROTO_ETH)
context_headers = process_context_headers(ctx1, ctx2, ctx3, ctx4)
ctx_values = CONTEXTHEADER(context_headers[0], context_headers[1], context_headers[2], context_headers[3])
inner_header = InnerHeader(inner_src_ip, inner_dest_ip, inner_src_port, inner_dest_port)
if encapsulate == 'gpe-nsh-ipv4':
# NSH type 1
vxlan_header_values = VXLANGPE()
# override encap type
base_header_values.next_protocol = NSH_NEXT_PROTO_IPV4
udpclient = MyVxlanGpeNshIpClient(loop, vxlan_header_values, base_header_values,
ctx_values, remote_sff_ip, int(remote_sff_port), inner_header)
start_client(loop, (local_ip, local_port), (remote_sff_ip, remote_sff_port), udpclient)
elif encapsulate == 'gre-nsh-ethernet':
# add GRE header
gre_header_values = GREHEADER(int('0', 2), int('000000000000', 2), int('000', 2), 0x894F,
int('0000000000000000', 2), 0)
greclient = MyGreNshEthClient(loop, 'GRE/NSH/Ethernet', gre_header_values, base_header_values, ctx_values,
remote_sff_ip, int(remote_sff_port))
greclient.send_gre_nsh()
elif encapsulate == 'gpe-nsh-ethernet':
# NSH type 3
vxlan_header_values = VXLANGPE()
ethernet_header_values = ETHHEADER(inner_src_eth_list[0], inner_src_eth_list[1], inner_src_eth_list[2],
inner_src_eth_list[3], inner_src_eth_list[4], inner_src_eth_list[5],
inner_dest_eth_list[0], inner_dest_eth_list[1], inner_dest_eth_list[2],
inner_dest_eth_list[3], inner_dest_eth_list[4], inner_dest_eth_list[5],
0x08, 0x00)
udpclient = MyVxlanGpeNshEthClient(loop, ethernet_header_values,
vxlan_header_values, base_header_values,
ctx_values, remote_sff_ip,
int(remote_sff_port), inner_header)
start_client(loop, (local_ip, local_port), (remote_sff_ip, remote_sff_port), udpclient)
elif encapsulate == 'vxlan-nsh-ethernet-legacy':
# NSH type 3
vxlan_header_values = VXLAN()
ethernet_header_values = ETHHEADER(inner_src_eth_list[0], inner_src_eth_list[1], inner_src_eth_list[2],
inner_src_eth_list[3], inner_src_eth_list[4], inner_src_eth_list[5],
inner_dest_eth_list[0], inner_dest_eth_list[1], inner_dest_eth_list[2],
inner_dest_eth_list[3], inner_dest_eth_list[4], inner_dest_eth_list[5],
0x08, 0x00)
udpclient = MyVxlanNshEthClient(loop, ethernet_header_values,
vxlan_header_values, base_header_values,
ctx_values, remote_sff_ip,
int(remote_sff_port), inner_header)
start_client(loop, (local_ip, local_port), (remote_sff_ip, remote_sff_port), udpclient)
else:
logger.error("Encapsulation must be specified, e.g. gpe-nsh-ethernet")