client/tests/netperf2/netperf2.py

import os
from autotest_lib.client.bin import test, autotest_utils
from autotest_lib.client.common_lib import utils, error


class netperf2(test.test):
	version = 1

	# ftp://ftp.netperf.org/netperf/netperf-2.4.1.tar.gz
	def setup(self, tarball = 'netperf-2.4.1.tar.gz'):
		tarball = utils.unmap_url(self.bindir, tarball,
		                                   self.tmpdir)
		autotest_utils.extract_tarball_to_dir(tarball, self.srcdir)
		os.chdir(self.srcdir)

		utils.system('./configure')
		utils.system('make')


	def initialize(self):
		# netserver doesn't detach properly from the console. When 
		# it is run from ssh, this causes the ssh command not to 
		# return even though netserver meant to be backgrounded.
		# This behavior is remedied by redirecting fd 0, 1 & 2
		self.server_path = ('%s &>/dev/null </dev/null' 
			% os.path.join(self.srcdir, 'src/netserver'))
		self.client_path = os.path.join(self.srcdir, 'src/netperf')


	def execute(self, server_ip, client_ip, role, 
					script='snapshot_script', args=''):
		server_tag = server_ip + '#netperf-server'
		client_tag = client_ip + '#netperf-client'
		all = [server_tag, client_tag]
		job = self.job
		if (role == 'server'):
			self.server_start()
			try:
				job.barrier(server_tag, 'start',
							600).rendevous(*all)
				job.barrier(server_tag, 'stop',
							3600).rendevous(*all)
			finally:
				self.server_stop()
		elif (role == 'client'):
			os.environ['NETPERF_CMD'] = self.client_path
			job.barrier(client_tag, 'start', 600).rendevous(*all)
			self.client(script, server_ip, args)
			job.barrier(client_tag, 'stop',  30).rendevous(*all)
		else:
			raise error.UnhandledError('invalid role specified')


	def server_start(self):
		# we should really record the pid we forked off, but there
		# was no obvious way to run the daemon in the foreground.
		# Hacked it for now
		system('killall netserver', ignore_status=True)
		system(self.server_path)


	def server_stop(self):
		# this should really just kill the pid I forked, but ...
		system('killall netserver')


	def client(self, script, server_ip, args = 'CPU'):
		# run some client stuff
		stdout_path = os.path.join(self.resultsdir, script + '.stdout')
		stderr_path = os.path.join(self.resultsdir, script + '.stderr')
		self.job.stdout.tee_redirect(stdout_path)
		self.job.stderr.tee_redirect(stderr_path)

		script_path = os.path.join(self.srcdir, 'doc/examples', script)
		system('%s %s %s' % (script_path, server_ip, args))

		self.job.stdout.restore()
		self.job.stderr.restore()