scyther/gui/Scyther.py

134 lines
3.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 sys
import StringIO
import tempfile
#---------------------------------------------------------------------------
""" Import scyther components """
import XMLReader
from Misc import *
#---------------------------------------------------------------------------
class Scyther(object):
def __init__ ( self):
self.program = "scyther"
self.options = ""
self.spdl = None
self.inputfile = None
2006-08-02 13:59:57 +01:00
self.claims = None
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 verify(self):
2006-08-02 14:10:38 +01:00
# Run Scyther on temp file
self.cmd = "%s --dot-output --xml-output --plain %s" % (self.program,self.options)
2006-08-02 13:59:57 +01:00
if self.spdl:
if self.inputfile:
fp = open(self.inputfile,'r')
else:
# Write spdl to temp file
fp = tempfile.NamedTemporaryFile()
fp.write(self.spdl)
fp.flush()
self.cmd += " %s" % (fp.name)
2006-08-02 13:59:57 +01:00
2006-08-02 14:10:38 +01:00
# If we are on windows, we don't get stderr. Maybe we need a
# switch to enforce this.
if sys.platform.startswith('linux'):
cmdline = "%s 2>/dev/null" % (self.cmd)
else:
# Non-linux does not generate stderr anyway
cmdline = "%s" % (self.cmd)
2006-08-02 13:59:57 +01:00
print cmdline
confirm("Run this command?")
2006-08-02 14:10:38 +01:00
result = os.popen(cmdline)
xmlinput = result.read()
result.close()
2006-08-02 13:59:57 +01:00
print xmlinput
confirm("Is this the correct output?")
2006-08-02 14:10:38 +01:00
if self.spdl:
2006-08-02 13:59:57 +01:00
fp.close()
2006-08-02 14:10:38 +01:00
xmlfile = StringIO.StringIO(xmlinput)
reader = XMLReader.XMLReader()
self.claims = reader.readXML(xmlfile)
2006-08-02 13:59:57 +01:00
2006-08-02 14:10:38 +01:00
return self.claims
2006-08-02 13:59:57 +01:00
def __str__(self):
if self.claims:
s = ""
for cl in self.claims:
s += str(cl) + "\n"
return s
else:
return "Scyther has not been run yet."
def basicTest():
2006-08-02 13:59:57 +01:00
# Some basic testing
#if sys.platform.startswith('win'):
# print "Dir test"
# p = os.popen("dir")
# print p.read()
# print p.close()
# confirm("See the dir?")
2006-08-02 13:59:57 +01:00
# Scyther
x = Scyther()
if sys.platform.startswith('win'):
x.program = "Scyther.exe"
2006-08-02 13:59:57 +01:00
if not os.path.isfile(x.program):
print "I can't find the Scyther executable %s" % (x.program)
else:
p = os.popen("%s --help" % x.program)
print p.read()
confirm("Do you see the help?")
2006-08-02 13:59:57 +01:00
x.setFile("ns3.spdl")
x.verify()
2006-08-02 14:10:38 +01:00
print x
confirm("See the output?")
def simpleRun(args):
x = Scyther()
x.options = args
x.verify()
return x
2006-08-02 13:59:57 +01:00
if __name__ == '__main__':
2006-08-02 14:10:38 +01:00
pars = sys.argv[1:]
if len(pars) == 0:
basicTest()
2006-08-02 14:10:38 +01:00
else:
print simpleRun(" ".join(pars))
2006-08-02 13:59:57 +01:00