new file mode 100755
@@ -0,0 +1,37 @@
+#!/usr/bin/python
+import socket, struct, os, signal, sys
+# -*- coding: utf-8 -*-
+
+"""
+Script used to join machine into multicast groups.
+
+@author Amos Kong <akong@redhat.com>
+"""
+
+if __name__ == "__main__":
+ if len(sys.argv) < 4:
+ print """%s [mgroup_count] [prefix] [suffix]
+ mgroup_count: count of multicast addresses
+ prefix: multicast address prefix
+ suffix: multicast address suffix""" % sys.argv[0]
+ sys.exit()
+
+ mgroup_count = int(sys.argv[1])
+ prefix = sys.argv[2]
+ suffix = int(sys.argv[3])
+
+ s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+ for i in range(mgroup_count):
+ mcast = prefix + "." + str(suffix + i)
+ try:
+ mreq = struct.pack("4sl", socket.inet_aton(mcast),
+ socket.INADDR_ANY)
+ s.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
+ except:
+ s.close()
+ print "Could not join multicast: %s" % mcast
+ raise
+
+ print "join_mcast_pid:%s" % os.getpid()
+ os.kill(os.getpid(), signal.SIGSTOP)
+ s.close()
new file mode 100644
@@ -0,0 +1,91 @@
+import logging, os, re
+from autotest_lib.client.common_lib import error
+from autotest_lib.client.bin import utils
+import kvm_test_utils
+
+
+def run_multicast(test, params, env):
+ """
+ Test multicast function of nic (rtl8139/e1000/virtio)
+
+ 1) Create a VM.
+ 2) Join guest into multicast groups.
+ 3) Ping multicast addresses on host.
+ 4) Flood ping test with different size of packets.
+ 5) Final ping test and check if lose packet.
+
+ @param test: KVM test object.
+ @param params: Dictionary with the test parameters.
+ @param env: Dictionary with test environment.
+ """
+ vm = kvm_test_utils.get_living_vm(env, params.get("main_vm"))
+ session = kvm_test_utils.wait_for_login(vm,
+ timeout=int(params.get("login_timeout", 360)))
+
+ def run_guest(cmd):
+ s, o = session.get_command_status_output(cmd)
+ if s:
+ logging.warning('Command %s executed in guest returned exit code '
+ '%s, output: %s', cmd, s, o.strip())
+
+ def run_host_guest(cmd):
+ run_guest(cmd)
+ utils.system(cmd, ignore_status=True)
+
+ # flush the firewall rules
+ cmd_flush = "iptables -F"
+ cmd_selinux = ("if [ -e /selinux/enforce ]; then setenforce 0; "
+ "else echo 'no /selinux/enforce file present'; fi")
+ run_host_guest(cmd_flush)
+ run_host_guest(cmd_selinux)
+ # make sure guest replies to broadcasts
+ cmd_broadcast = "echo 0 > /proc/sys/net/ipv4/icmp_echo_ignore"
+ cmd_broadcast_2 = "echo 0 > /proc/sys/net/ipv4/icmp_echo_ignore_all"
+ run_guest(cmd_broadcast)
+ run_guest(cmd_broadcast_2)
+
+ # base multicast address
+ mcast = params.get("mcast", "225.0.0.1")
+ # count of multicast addresses, less than 20
+ mgroup_count = int(params.get("mgroup_count", 5))
+ flood_minutes = float(params.get("flood_minutes", 10))
+ ifname = vm.get_ifname()
+ prefix = re.findall("\d+.\d+.\d+", mcast)[0]
+ suffix = int(re.findall("\d+", mcast)[-1])
+ # copy python script to guest for joining guest to multicast groups
+ mcast_path = os.path.join(test.bindir, "scripts/join_mcast.py")
+ if not vm.copy_files_to(mcast_path, "/tmp"):
+ raise error.TestError("Fail to copy %s to guest" % mcast_path)
+ output = session.get_command_output("python /tmp/join_mcast.py %d %s %d" %
+ (mgroup_count, prefix, suffix))
+
+ # if success to join multicast, the process will be paused, and return PID.
+ try:
+ pid = re.findall("join_mcast_pid:(\d+)", output)[0]
+ except IndexError:
+ raise error.TestFail("Can't join multicast groups,output:%s" % output)
+
+ try:
+ for i in range(mgroup_count):
+ new_suffix = suffix + i
+ mcast = "%s.%d" % (prefix, new_suffix)
+
+ logging.info("Initial ping test, mcast: %s", mcast)
+ s, o = kvm_test_utils.ping(mcast, 10, interface=ifname, timeout=20)
+ if s != 0:
+ raise error.TestFail(" Ping return non-zero value %s" % o)
+
+ logging.info("Flood ping test, mcast: %s", mcast)
+ kvm_test_utils.ping(mcast, None, interface=ifname, flood=True,
+ output_func=None, timeout=flood_minutes*60)
+
+ logging.info("Final ping test, mcast: %s", mcast)
+ s, o = kvm_test_utils.ping(mcast, 10, interface=ifname, timeout=20)
+ if s != 0:
+ raise error.TestFail("Ping failed, status: %s, output: %s" %
+ (s, o))
+
+ finally:
+ logging.debug(session.get_command_output("ipmaddr show"))
+ session.get_command_output("kill -s SIGCONT %s" % pid)
+ session.close()
@@ -493,6 +493,13 @@ variants:
type = nic_promisc
file_size = 1, 1460, 65000, 100000000
+ - multicast: install setup unattended_install.cdrom
+ type = multicast
+ nic_mode = tap
+ mcast = 225.0.0.1
+ mgroup_count = 20
+ flood_minutes = 1
+
- physical_resources_check: install setup unattended_install.cdrom
type = physical_resources_check
catch_uuid_cmd = dmidecode | awk -F: '/UUID/ {print $2}'
@@ -1267,7 +1274,7 @@ variants:
# Windows section
- @Windows:
- no autotest linux_s3 vlan_tag ioquit unattended_install.(url|nfs|remote_ks) jumbo file_transfer nicdriver_unload nic_promisc
+ no autotest linux_s3 vlan_tag ioquit unattended_install.(url|nfs|remote_ks) jumbo file_transfer nicdriver_unload nic_promisc multicast
shutdown_command = shutdown /s /f /t 0
reboot_command = shutdown /r /f /t 0
status_test_command = echo %errorlevel%