scyther/gui/Scyther/Scyther.py

150 lines
4.2 KiB
Python
Raw Normal View History

2006-08-02 13:59:57 +01:00
#!/usr/bin/python
#
# Scyther interface
#
#---------------------------------------------------------------------------
""" Import externals """
import os
import os.path
2006-08-02 13:59:57 +01:00
import sys
import StringIO
#---------------------------------------------------------------------------
""" Import scyther components """
import XMLReader
from Misc import *
#---------------------------------------------------------------------------
class Scyther(object):
def __init__ ( self):
2006-08-04 23:00:22 +01:00
# Where is my executable?
if sys.platform.startswith('win'):
""" Windows """
# TODO hardcoded for now, bad
self.program = os.path.join("bin","Scyther.exe")
if not os.path.isfile(self.program):
print "I can't find the Scyther executable at %s" % (self.program)
2006-08-04 23:00:22 +01:00
else:
""" Non-windows """
self.program = os.path.join("bin","scyther")
2006-08-04 23:00:22 +01:00
2006-08-08 16:54:00 +01:00
# Init
2006-08-02 13:59:57 +01:00
self.spdl = None
self.inputfile = None
2006-08-08 17:16:28 +01:00
self.options = ""
2006-08-02 13:59:57 +01:00
self.claims = None
2006-08-06 20:52:07 +01:00
self.errors = None
2006-08-06 22:16:14 +01:00
self.errorcount = 0
2006-08-07 11:52:48 +01:00
self.run = False
2006-08-08 16:54:00 +01:00
self.output = None
# defaults
self.xml = True # this results in a claim end, otherwise we simply get the output
2006-08-02 13:59:57 +01:00
def setInput(self,spdl):
self.spdl = spdl
self.inputfile = None
2006-08-02 13:59:57 +01:00
def setFile(self,filename):
self.inputfile = filename
2006-08-02 13:59:57 +01:00
self.spdl = ""
fp = open(filename,"r")
for l in fp.readlines():
self.spdl += l
fp.close()
def addFile(self,filename):
self.inputfile = None
if not self.spdl:
self.spdl = ""
fp = open(filename,"r")
for l in fp.readlines():
self.spdl += l
fp.close()
2006-08-02 13:59:57 +01:00
def verify(self):
2006-08-02 14:10:38 +01:00
# Run Scyther on temp file
2006-08-08 16:54:00 +01:00
self.cmd = self.program
if self.xml:
self.cmd += " --dot-output --xml-output --plain"
self.cmd += " " + self.options
2006-08-02 14:10:38 +01:00
2006-08-06 20:52:07 +01:00
(stdin,stdout,stderr) = os.popen3(self.cmd)
2006-08-02 14:10:38 +01:00
if self.spdl:
2006-08-06 20:52:07 +01:00
stdin.write(self.spdl)
stdin.close()
# In the order below, or stuff breaks (hangs), as described at
# http://mail.python.org/pipermail/python-dev/2000-September/009460.html
#
# TODO this is annoying: we would like to determine progress
# from the error output (maybe this can also be done by flushing
# the XML at certain points...)
2006-08-08 16:54:00 +01:00
output = stdout.read()
2006-08-07 12:02:14 +01:00
errlines = stderr.readlines()
2006-08-06 22:16:14 +01:00
# filter out any non-errors (say maybe only claim etc) and count
# them.
2006-08-07 12:02:14 +01:00
self.errors = []
for l in errlines:
if not l.startswith("claim\t"):
self.errors.append(l.strip())
2006-08-06 22:16:14 +01:00
self.errorcount = len(self.errors)
2006-08-06 20:52:07 +01:00
# close
stdout.close()
stderr.close()
2006-08-08 16:54:00 +01:00
if self.xml:
2006-08-08 17:16:28 +01:00
self.validxml = False
2006-08-08 16:54:00 +01:00
if len(output) > 0:
2006-08-08 17:16:28 +01:00
if output.startswith("<scyther>"):
self.validxml = True
if self.validxml:
2006-08-08 16:54:00 +01:00
xmlfile = StringIO.StringIO(output)
reader = XMLReader.XMLReader()
self.claims = reader.readXML(xmlfile)
else:
2006-08-08 17:16:28 +01:00
# no xml output... store whatever comes out
2006-08-08 16:54:00 +01:00
self.claims = []
2006-08-08 17:16:28 +01:00
self.output = output
2006-08-08 16:54:00 +01:00
result = self.claims
else:
2006-08-08 16:54:00 +01:00
self.output = output
result = self.output
2006-08-02 13:59:57 +01:00
2006-08-07 11:52:48 +01:00
self.run = True
2006-08-08 16:54:00 +01:00
return result
2006-08-02 13:59:57 +01:00
def getClaim(self,claimid):
if self.claims:
for cl in self.claims:
if cl.id == claimid:
return cl
return None
2006-08-02 13:59:57 +01:00
def __str__(self):
2006-08-07 11:52:48 +01:00
if self.run:
if self.errorcount > 0:
2006-08-07 17:40:46 +01:00
return "%i errors:\n%s" % (self.errorcount, "\n".join(self.errors))
2006-08-07 11:52:48 +01:00
else:
2006-08-08 17:16:28 +01:00
if self.xml and self.validxml:
s = "Verification results:\n"
2006-08-08 16:54:00 +01:00
for cl in self.claims:
s += str(cl) + "\n"
return s
else:
return self.output
2006-08-02 13:59:57 +01:00
else:
return "Scyther has not been run yet."