X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=tests%2Fdata_plane%2Fvpp_lite_topo%2Fscripts%2Fport_flood.py;fp=tests%2Fdata_plane%2Fvpp_lite_topo%2Fscripts%2Fport_flood.py;h=4921df1365c2d1b233d8d42438337d7d9b49a348;hb=25531ee0d1350f554314d0cbf914b4f461d57828;hp=0000000000000000000000000000000000000000;hpb=9325071dfe6f832c5be80e1206bc6338335cf977;p=one.git

diff --git a/tests/data_plane/vpp_lite_topo/scripts/port_flood.py b/tests/data_plane/vpp_lite_topo/scripts/port_flood.py
new file mode 100755
index 0000000..4921df1
--- /dev/null
+++ b/tests/data_plane/vpp_lite_topo/scripts/port_flood.py
@@ -0,0 +1,28 @@
+#/usr/bin/env python
+
+import sys
+import socket
+
+
+def do_flood(host, num):
+  try:
+    socket.inet_aton(host)
+    is_ip4 = True
+  except socket.error:
+    try:
+      socket.inet_pton(socket.AF_INET6, host)
+      is_ip4 = False
+    except socket.error:
+        raise Exception('Invlid ip4/6 address!')
+
+  family = socket.AF_INET if is_ip4 else socket.AF_INET6
+
+  for port in range(num):
+    sock = socket.socket(family, socket.SOCK_DGRAM)
+    sock.sendto('test', (host, port + 1))
+
+
+if __name__ == '__main__':
+  if len(sys.argv) < 2:
+    raise Exception('IP and packet count expected!')
+  do_flood(sys.argv[1], int(sys.argv[2]))