| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 1 | import test |
| 2 | from autotest_utils import * |
| 3 | |
| 4 | class netperf2(test.test): |
| 5 | version = 1 |
| 6 | |
| 7 | # ftp://ftp.netperf.org/netperf/netperf-2.4.1.tar.gz |
| 8 | def setup(self, tarball = 'netperf-2.4.1.tar.gz'): |
| 9 | tarball = unmap_url(self.bindir, tarball, self.tmpdir) |
| 10 | extract_tarball_to_dir(tarball, self.srcdir) |
| 11 | os.chdir(self.srcdir) |
| 12 | |
| 13 | system('./configure') |
| 14 | system('make') |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 15 | |
| 16 | |
| 17 | def initialize(self): |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 18 | self.server_path = os.path.join(self.srcdir, 'src/netserver') |
| 19 | self.client_path = os.path.join(self.srcdir, 'src/netperf') |
| 20 | |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 21 | |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 22 | def execute(self, server_ip, client_ip, role, |
| 23 | script='snapshot_script', args=''): |
| 24 | server_tag = server_ip + '#netperf-server' |
| 25 | client_tag = client_ip + '#netperf-client' |
| 26 | all = [server_tag, client_tag] |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 27 | job = self.job |
| 28 | if (role == 'server'): |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 29 | self.server_start() |
| apw | da2a62f | 2006-11-06 17:39:25 +0000 | [diff] [blame] | 30 | try: |
| 31 | job.barrier(server_tag, 'start', |
| 32 | 30).rendevous(*all) |
| 33 | job.barrier(server_tag, 'stop', |
| 34 | 3600).rendevous(*all) |
| 35 | finally: |
| 36 | self.server_stop() |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 37 | elif (role == 'client'): |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 38 | os.environ['NETPERF_CMD'] = self.client_path |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 39 | job.barrier(client_tag, 'start', 30).rendevous(*all) |
| 40 | self.client(script, server_ip, args) |
| 41 | job.barrier(client_tag, 'stop', 30).rendevous(*all) |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 42 | else: |
| 43 | raise UnhandledError('invalid role specified') |
| 44 | |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 45 | |
| 46 | def server_start(self): |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 47 | # we should really record the pid we forked off, but there |
| 48 | # was no obvious way to run the daemon in the foreground. |
| 49 | # Hacked it for now |
| mbligh | 0d4a032 | 2006-09-25 14:17:55 +0000 | [diff] [blame] | 50 | system('killall netserver', ignorestatus=1) |
| mbligh | a8cc6e0 | 2006-09-23 05:00:33 +0000 | [diff] [blame] | 51 | system(self.server_path) |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 52 | |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 53 | |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 54 | def server_stop(self): |
| 55 | # this should really just kill the pid I forked, but ... |
| 56 | system('killall netserver') |
| 57 | |
| mbligh | 54e5684 | 2006-09-23 06:02:54 +0000 | [diff] [blame] | 58 | |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 59 | def client(self, script, server_ip, args = 'CPU'): |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 60 | # run some client stuff |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 61 | stdout_path = os.path.join(self.resultsdir, script + '.stdout') |
| 62 | stderr_path = os.path.join(self.resultsdir, script + '.stderr') |
| 63 | self.job.stdout.tee_redirect(stdout_path) |
| 64 | self.job.stderr.tee_redirect(stderr_path) |
| mbligh | 56a91f0 | 2006-09-14 17:55:19 +0000 | [diff] [blame] | 65 | |
| mbligh | e8fa3af | 2006-09-28 23:14:56 +0000 | [diff] [blame] | 66 | script_path = os.path.join(self.srcdir, 'doc/examples', script) |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 67 | system('%s %s %s' % (script_path, server_ip, args)) |
| mbligh | e8fa3af | 2006-09-28 23:14:56 +0000 | [diff] [blame] | 68 | |
| mbligh | c6b01ed | 2006-10-13 17:03:26 +0000 | [diff] [blame] | 69 | self.job.stdout.restore() |
| 70 | self.job.stderr.restore() |