import os import time import shutil import subprocess import sys cache_size = 300 # in MiB toolset = '' if len(sys.argv) > 1: toolset = sys.argv[1] ret = os.system('cd ../examples && bjam profile statistics=on %s stage_client_test' % toolset) if ret != 0: print 'ERROR: build failed: %d' % ret sys.exit(1) ret = os.system('cd ../examples && bjam release %s stage_connection_tester' % toolset) if ret != 0: print 'ERROR: build failed: %d' % ret sys.exit(1) try: os.remove('.ses_state') except: pass try: shutils.rmtree('.resume') except: pass if not os.path.exists('cpu_benchmark.torrent'): ret = os.system('../examples/connection_tester gen-torrent -s 10000 -n 15 -t cpu_benchmark.torrent') if ret != 0: print 'ERROR: connection_tester failed: %d' % ret sys.exit(1) try: shutil.rmtree('t') except: pass def run_test(name, test_cmd, client_arg, num_peers): output_dir = 'logs_%s' % name try: os.mkdir(output_dir) except: pass port = (int(time.time()) % 50000) + 2000 try: shutil.rmtree('session_stats') except: pass try: shutil.rmtree('session_stats_report') except: pass start = time.time(); client_cmd = '../examples/client_test -p %d cpu_benchmark.torrent -k -z -H -X -q 120 %s -h -c %d -T %d -C %d -f %s/events.log' \ % (port, client_arg, num_peers*2, num_peers*2, cache_size * 16, output_dir) test_cmd = '../examples/connection_tester %s -c %d -d 127.0.0.1 -p %d -t cpu_benchmark.torrent' % (test_cmd, num_peers, port) client_out = open('%s/client.out' % output_dir, 'w+') test_out = open('%s/test.out' % output_dir, 'w+') print client_cmd c = subprocess.Popen(client_cmd.split(' '), stdout=client_out, stderr=client_out, stdin=subprocess.PIPE) time.sleep(2) print test_cmd t = subprocess.Popen(test_cmd.split(' '), stdout=test_out, stderr=test_out) t.wait() end = time.time(); c.communicate('q') c.wait() print 'runtime %d seconds' % (end - start) print 'analyzing proile...' os.system('gprof ../examples/client_test >%s/gprof.out' % output_dir) print 'generating profile graph...' os.system('python gprof2dot.py --strip <%s/gprof.out | dot -Tpng -o %s/cpu_profile.png' % (output_dir, output_dir)) os.system('python parse_session_stats.py session_stats/*.log') try: shutil.move('session_stats_report', '%s/session_stats_report' % output_dir) except: pass try: shutil.move('session_stats', '%s/session_stats' % output_dir) except: pass run_test('download', 'upload', '', 50) run_test('upload', 'download', '-G', 20)