mirror of
https://github.com/zardus/ctf-tools
synced 2024-12-13 14:32:34 +00:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import binpacking, sys, pprint
|
||
|
|
|
||
|
|
MAXBINDURATION = 2000 # seconds
|
||
|
|
|
||
|
|
def parseOutput(fn):
|
||
|
|
lines = [l.strip() for l in open(fn).readlines()]
|
||
|
|
out = {}
|
||
|
|
|
||
|
|
for l in lines:
|
||
|
|
[distro, tool, success, duration] = l.split(" ")
|
||
|
|
if not distro in out:
|
||
|
|
out[distro] = {}
|
||
|
|
out[distro][tool] = {
|
||
|
|
"success": success == "SUCCEEDED",
|
||
|
|
"duration": int(duration)
|
||
|
|
}
|
||
|
|
return out
|
||
|
|
|
||
|
|
def printBins(timingdata, distro, expectfail):
|
||
|
|
inputs = dict([(t, v["duration"]) for (t, v) in timingdata[distro].items() if v["success"] != expectfail])
|
||
|
|
bins = binpacking.to_constant_volume(inputs, MAXBINDURATION)
|
||
|
|
|
||
|
|
for b in bins:
|
||
|
|
tools = " ".join(sorted(b.keys()))
|
||
|
|
duration = sum(b.values())
|
||
|
|
if expectfail:
|
||
|
|
print("- DISTRO='{}' EXPECTFAIL=1 TOOL='{}' # estimated {} seconds".format(distro, tools, duration))
|
||
|
|
else:
|
||
|
|
print("- DISTRO='{}' TOOL='{}' # estimated {} seconds".format(distro, tools, duration))
|
||
|
|
|
||
|
|
def getToolsFromTimingdata(timingdata):
|
||
|
|
out = {}
|
||
|
|
for d, dd in timingdata.items():
|
||
|
|
for t, td in dd.items():
|
||
|
|
out[t] = 1
|
||
|
|
return out.keys()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
timingdata = parseOutput(sys.argv[1])
|
||
|
|
distros = sorted(timingdata.keys()) # all distros seen during previous build
|
||
|
|
tools = sorted(getToolsFromTimingdata(timingdata)) # all tools seen during previous build
|
||
|
|
|
||
|
|
for distro in distros:
|
||
|
|
printBins(timingdata, distro, False)
|
||
|
|
printBins(timingdata, distro, True)
|
||
|
|
|
||
|
|
# no timing data, assume the build took too long for this tool on this distro
|
||
|
|
nodata = [t for t in tools if t not in timingdata[distro]]
|
||
|
|
for tool in nodata:
|
||
|
|
print("# - DISTRO='{}' TOOL='{}' # unknown duration...".format(distro, tool))
|
||
|
|
|