96 lines · plain
1#!/usr/bin/env python32# This script is used to bisect skip and count arguments for --debug-counter.3# It is similar to bisect, except it understands how to increase skip and decrease count4#5# Typical usage:6#7# bisect-skip-count bisect-command.sh "%(skip)d" "%(count)d" 2>&1 | tee bisect.out8#9# bisect-command.sh is something like this:10# #! /bin/bash11#12# skip=$113# count=$214#15# opt -debug-counter=my-counter-skip=${skip},my-counter-count=${count}16# ... Test output of opt and exit zero for pass, non-zero for fail17#18# Examine bisect.out to look for "Last good skip" and "Last good19# count" to find the values of the counter that produce a passing20# result. Incrementing the last good count by one or decrementing the21# last good skip by one should produce a failure.22#23from __future__ import print_function24import os25import sys26import argparse27# This is for timeout support. Use the recommended way of import.28# We do timeouts because when doing, execution testing, we have a habit29# of finding variants that infinite loop30if os.name == 'posix' and sys.version_info[0] < 3:31 import subprocess32 as subprocess32else:33 import subprocess34parser = argparse.ArgumentParser()35 36parser.add_argument('--skipstart', type=int, default=0)37parser.add_argument('--skipend', type=int, default=(1 << 32))38parser.add_argument('--countstart', type=int, default=0)39parser.add_argument('--countend', type=int, default=(1 << 32))40parser.add_argument('--timeout', type=int, default=None)41# Use shell support if you need to use complex shell expressions in your command42parser.add_argument('--shell', type=bool, default=False)43parser.add_argument('command', nargs='+')44 45args = parser.parse_args()46 47start = args.skipstart48end = args.skipend49 50print("Bisect of Skip starting!")51print("Start: %d" % start)52print("End: %d" % end)53 54last = None55while start != end and start != end-1:56 count = start + (end - start)//257 print("Visiting Skip: %d with (Start, End) = (%d,%d)" % (count, start, end))58 cmd = [x % {'skip':count, 'count':-1} for x in args.command]59 print(cmd)60 try:61 result = subprocess.call(cmd, shell=args.shell, timeout=args.timeout)62 if result == 0:63 print(" PASSES! Setting end to count")64 end = count65 else:66 print(" FAILS! Setting start to count")67 start = count68 except:69 print(" TIMEOUT, setting end to count")70 end = count71firstcount = start72print("Last good skip: %d" % start)73start = args.countstart74end = args.countend75print("Bisect of Count starting!")76print("Start: %d" % start)77print("End: %d" % end)78while start != end and start != end-1:79 count = start + (end - start)//280 print("Visiting Count: %d with (Start, End) = (%d,%d)" % (count, start, end))81 cmd = [x % {'count':count, 'skip':firstcount } for x in args.command]82 print(cmd)83 try:84 result = subprocess.call(cmd, shell=args.shell, timeout=args.timeout)85 if result == 0:86 print(" PASSES! Setting start to count")87 start = count88 else:89 print(" FAILS! Setting end to count")90 end = count91 except:92 print(" TIMEOUT, setting start to count")93 start = count94 95print("Last good count: %d" % start)96