| import os, re, time, subprocess, sys |
| from autotest_lib.client.bin import test, utils |
| from autotest_lib.client.common_lib import error |
| |
| |
| class parallel_dd(test.test): |
| version = 2 |
| |
| def initialize(self, fs, fstype = 'ext2', megabytes = 1000, streams = 2): |
| self.megabytes = megabytes |
| self.blocks = megabytes * 256 |
| self.blocks_per_file = self.blocks / streams |
| self.fs = fs |
| self.fstype = fstype |
| self.streams = streams |
| |
| self.old_fstype = self._device_to_fstype('/etc/mtab') |
| if not self.old_fstype: |
| self.old_fstpye = self._device_to_fstype('/etc/fstab') |
| if not self.old_fstype: |
| self.old_fstype = self.fstype |
| |
| print 'Dumping %d megabytes across %d streams' % (megabytes, streams) |
| |
| |
| def raw_write(self): |
| print "Timing raw write of %d megabytes" % self.megabytes |
| sys.stdout.flush() |
| dd = 'dd if=/dev/zero of=%s bs=4k count=%d' % \ |
| (self.fs.device, self.blocks) |
| print dd |
| utils.system(dd + ' > /dev/null') |
| |
| |
| def raw_read(self): |
| print "Timing raw read of %d megabytes" % self.megabytes |
| sys.stdout.flush() |
| dd = 'dd if=%s of=/dev/null bs=4k count=%d' % \ |
| (self.fs.device, self.blocks) |
| print dd |
| utils.system(dd + ' > /dev/null') |
| |
| |
| def fs_write(self): |
| p = [] |
| # Write out 'streams' files in parallel background tasks |
| for i in range(self.streams): |
| file = 'poo%d' % (i+1) |
| file = os.path.join(self.job.tmpdir, file) |
| dd = 'dd if=/dev/zero of=%s bs=4k count=%d' % \ |
| (file, self.blocks_per_file) |
| print dd |
| p.append(subprocess.Popen(dd + ' > /dev/null', shell=True)) |
| print "Waiting for %d streams" % self.streams |
| # Wait for everyone to complete |
| for i in range(self.streams): |
| print "Waiting for %d" % p[i].pid |
| sys.stdout.flush() |
| os.waitpid(p[i].pid, 0) |
| sys.stdout.flush() |
| sys.stderr.flush() |
| |
| |
| def fs_read(self): |
| for i in range(self.streams): |
| file = os.path.join(self.job.tmpdir, 'poo%d' % (i+1)) |
| dd = 'dd if=%s of=/dev/null bs=4k count=%d' % \ |
| (file, self.blocks_per_file) |
| utils.system(dd + ' > /dev/null') |
| |
| |
| def _device_to_fstype(self, file): |
| device = self.fs.device |
| try: |
| line = utils.system_output('egrep ^%s %s' % (device, file)) |
| print line |
| fstype = line.split()[2] |
| print 'Found %s is type %s from %s' % (device, fstype, file) |
| return fstype |
| except error.CmdError, e: |
| print 'No %s found in %s' % (device, file) |
| return None |
| |
| |
| def run_once(self): |
| try: |
| self.fs.unmount() |
| except error.CmdError, e: |
| pass |
| |
| print '----------------- Timing raw operations ----------------------' |
| start = time.time() |
| self.raw_write() |
| self.raw_write_rate = self.megabytes / (time.time() - start) |
| |
| start = time.time() |
| self.raw_read() |
| self.raw_read_rate = self.megabytes / (time.time() - start) |
| |
| # Set up the filesystem |
| self.fs.mkfs(self.fstype) |
| self.fs.mount() |
| |
| print '----------------- Timing fs operations ----------------------' |
| start = time.time() |
| self.fs_write() |
| self.fs_write_rate = self.megabytes / (time.time() - start) |
| self.fs.unmount() |
| |
| self.fs.mount() |
| start = time.time() |
| self.fs_read() |
| self.fs_read_rate = self.megabytes / (time.time() - start) |
| |
| self.write_perf_keyval({ |
| 'raw_write' : self.raw_write_rate, |
| 'raw_read' : self.raw_read_rate, |
| 'fs_write' : self.fs_write_rate, |
| 'fs_read' : self.fs_read_rate }) |
| |
| |
| def cleanup(self): |
| try: |
| self.fs.unmount() |
| except error.CmdError, e: |
| pass |
| print '\nFormatting %s back to type %s\n' % (self.fs, self.old_fstype) |
| self.fs.mkfs(self.old_fstype) |
| self.fs.mount() |