X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=src%2Fscripts%2Fvnet%2Furi%2Fdummy_app.py;h=7fab2d766ad9ab7f63bc03faa13bd91132046f55;hb=d9b0c6fbf7aa5bd9af84264105b39c82028a4a29;hp=b80fbb2816334a91acd828b1f7787cdf33a11c7f;hpb=e04c29942af6a130591059679531c9ffa3d7237a;p=vpp.git diff --git a/src/scripts/vnet/uri/dummy_app.py b/src/scripts/vnet/uri/dummy_app.py old mode 100644 new mode 100755 index b80fbb28163..7fab2d766ad --- a/src/scripts/vnet/uri/dummy_app.py +++ b/src/scripts/vnet/uri/dummy_app.py @@ -1,65 +1,163 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 import socket import sys -import bitstring +import time +import argparse -# action can be reflect or drop +# action can be reflect or drop action = "drop" +test = 0 -def handle_connection (connection, client_address): + +def test_data(data, n_rcvd): + n_read = len(data) + for i in range(n_read): + expected = (n_rcvd + i) & 0xFF + byte_got = ord(data[i]) + if byte_got != expected: + print( + "Difference at byte {}. Expected {} got {}".format( + n_rcvd + i, expected, byte_got + ) + ) + return n_read + + +def handle_connection(connection, client_address): print("Received connection from {}".format(repr(client_address))) + n_rcvd = 0 try: while True: data = connection.recv(4096) if not data: - break; - if (action != "drop"): + break + if test == 1: + n_rcvd += test_data(data, n_rcvd) + if action != "drop": connection.sendall(data) finally: connection.close() - -def run_server(ip, port): - print("Starting server {}:{}".format(repr(ip), repr(port))) + + +def run_tcp_server(ip, port): + print("Starting TCP server {}:{}".format(repr(ip), repr(port))) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) server_address = (ip, int(port)) sock.bind(server_address) sock.listen(1) - while True: connection, client_address = sock.accept() - handle_connection (connection, client_address) + handle_connection(connection, client_address) + + +def run_udp_server(ip, port): + print("Starting UDP server {}:{}".format(repr(ip), repr(port))) + sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + server_address = (ip, int(port)) + sock.bind(server_address) + while True: + data, addr = sock.recvfrom(4096) + if action != "drop": + # snd_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + sock.sendto(data, addr) + + +def run_server(ip, port, proto): + if proto == "tcp": + run_tcp_server(ip, port) + elif proto == "udp": + run_udp_server(ip, port) -def prepare_data(): + +def prepare_data(power): buf = [] - for i in range (0, pow(2, 16)): - buf.append(i & 0xff) + for i in range(0, pow(2, power)): + buf.append(i & 0xFF) return bytearray(buf) -def run_client(ip, port): - print("Starting client {}:{}".format(repr(ip), repr(port))) + +def run_tcp_client(ip, port): + print("Starting TCP client {}:{}".format(repr(ip), repr(port))) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - server_address = ("6.0.1.1", 1234) + server_address = (ip, int(port)) sock.connect(server_address) - - data = prepare_data() + + data = prepare_data(16) + n_rcvd = 0 + n_sent = len(data) try: sock.sendall(data) + + timeout = time.time() + 2 + while n_rcvd < n_sent and time.time() < timeout: + tmp = sock.recv(1500) + tmp = bytearray(tmp) + n_read = len(tmp) + for i in range(n_read): + if data[n_rcvd + i] != tmp[i]: + print( + "Difference at byte {}. Sent {} got {}".format( + n_rcvd + i, data[n_rcvd + i], tmp[i] + ) + ) + n_rcvd += n_read + + if n_rcvd < n_sent or n_rcvd > n_sent: + print("Sent {} and got back {}".format(n_sent, n_rcvd)) + else: + print("Got back what we've sent!!") + finally: sock.close() - -def run(mode, ip, port): - if (mode == "server"): - run_server (ip, port) - elif (mode == "client"): - run_client (ip, port) + + +def run_udp_client(ip, port): + print("Starting UDP client {}:{}".format(repr(ip), repr(port))) + n_packets = 100 + sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + server_address = (ip, int(port)) + data = prepare_data(10) + try: + for i in range(0, n_packets): + sock.sendto(data, server_address) + finally: + sock.close() + + +def run_client(ip, port, proto): + if proto == "tcp": + run_tcp_client(ip, port) + elif proto == "udp": + run_udp_client(ip, port) + + +def run(mode, ip, port, proto): + if mode == "server": + run_server(ip, port, proto) + elif mode == "client": + run_client(ip, port, proto) else: raise Exception("Unknown mode. Only client and server supported") -if __name__ == "__main__": - if (len(sys.argv)) < 4: - raise Exception("Usage: ./dummy_app []") - if (len(sys.argv) == 5): - action = sys.argv[4] - run (sys.argv[1], sys.argv[2], sys.argv[3]) +if __name__ == "__main__": + parser = argparse.ArgumentParser() + parser.add_argument("-m", action="store", dest="mode") + parser.add_argument("-i", action="store", dest="ip") + parser.add_argument("-p", action="store", dest="port") + parser.add_argument("-proto", action="store", dest="proto") + parser.add_argument("-a", action="store", dest="action") + parser.add_argument("-t", action="store", dest="test") + results = parser.parse_args() + action = results.action + test = results.test + run(results.mode, results.ip, results.port, results.proto) + # if (len(sys.argv)) < 4: + # raise Exception("Usage: ./dummy_app [ ]") + # if (len(sys.argv) == 6): + # action = sys.argv[4] + # test = int(sys.argv[5]) + # run (sys.argv[1], sys.argv[2], int(sys.argv[3]))