95 lines
2.1 KiB
Python
95 lines
2.1 KiB
Python
#
|
|
# Scyther caching mechanism
|
|
#
|
|
# Uses md5 hashes to store previously calculated results.
|
|
#
|
|
# (c)2005 Cas Cremers
|
|
#
|
|
#
|
|
# TODO:
|
|
#
|
|
# - Maybe it is an idea to time the output. If Scyther takes less
|
|
# than a second, we don't need to cache the output. That would
|
|
# reduce the required cache size significantly.
|
|
#
|
|
|
|
import md5
|
|
import commands
|
|
import os
|
|
import sys
|
|
from tempfile import NamedTemporaryFile, gettempdir
|
|
|
|
# scyther should reside in $PATH
|
|
def scythercall (argumentstring, inputfile):
|
|
clstring = "scyther " + argumentstring + " " + inputfile
|
|
(status,scout) = commands.getstatusoutput(clstring)
|
|
return (status,scout)
|
|
|
|
# cached results
|
|
# input: a large string (consisting of read input files)
|
|
# argstring: a smaller string
|
|
def eval (argumentstring, inputstring):
|
|
|
|
def cacheid():
|
|
m = md5.new()
|
|
|
|
# # Determine scyther version
|
|
# (status, scout) = scythercall ("--version", "")
|
|
# if status == 1 or status < 0:
|
|
# # some problem
|
|
# print "Problem with determining scyther version!"
|
|
# os.exit()
|
|
# # Add version to hash
|
|
# m.update (scout)
|
|
|
|
# Add inputfile to hash
|
|
m.update (inputstring)
|
|
|
|
# Add arguments to hash
|
|
m.update (argumentstring)
|
|
|
|
# Return a readable ID (good for a filename)
|
|
return m.hexdigest()
|
|
|
|
# Ensure directory
|
|
def ensureDirectory (path):
|
|
if not os.path.exists(path):
|
|
os.mkdir(path)
|
|
|
|
# Ensure directories for a file
|
|
def ensureDirectories (filename):
|
|
for i in range(1,len(filename)):
|
|
if filename[i] == '/':
|
|
np = i+1
|
|
ensureDirectory(filename[:np])
|
|
|
|
# Determine the unique filename for this test
|
|
id = cacheid()
|
|
filename = "scyther/cache-" + id[:3] + "/res-" + id[3:] + ".txt"
|
|
cachefile = gettempdir() + "/" + filename
|
|
ensureDirectories(cachefile)
|
|
|
|
# Does it already exist?
|
|
if os.path.exists(cachefile):
|
|
# Great: return the cached results
|
|
f = open(cachefile,'r')
|
|
res = f.read()
|
|
f.close()
|
|
return (0,res)
|
|
else:
|
|
# Hmm, we need to compute this result
|
|
h = NamedTemporaryFile()
|
|
h.write(inputstring)
|
|
h.flush()
|
|
(status, scout) = scythercall (argumentstring, h.name)
|
|
h.close()
|
|
|
|
# Write cache file even if it's wrong
|
|
f = open(cachefile,'w')
|
|
f.write(scout)
|
|
f.close()
|
|
|
|
return (status,scout)
|
|
|
|
|