117 lines
2.9 KiB
Python
117 lines
2.9 KiB
Python
#!/usr/bin/python
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
""" Import externals """
|
|
import wx
|
|
import wx.lib.newevent
|
|
import os
|
|
import sys
|
|
import re
|
|
import threading
|
|
import StringIO
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
""" Import scyther components """
|
|
import XMLReader
|
|
|
|
""" Import scyther-gui components """
|
|
import Tempfile
|
|
import Claim
|
|
import Preference
|
|
import Scyther
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
""" Global declaration """
|
|
(UpdateAttackEvent, EVT_UPDATE_ATTACK) = wx.lib.newevent.NewEvent()
|
|
busy = threading.Semaphore()
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
class ScytherThread(threading.Thread):
|
|
# Override Thread's __init__ method to accept the parameters needed:
|
|
def __init__ ( self, win, spdl, details ):
|
|
|
|
self.win = win
|
|
self.spdl = spdl
|
|
self.details = details
|
|
|
|
self.claims = []
|
|
|
|
threading.Thread.__init__ ( self )
|
|
|
|
def run(self):
|
|
|
|
global busy
|
|
|
|
evt = UpdateAttackEvent(status="Running Scyther...")
|
|
wx.PostEvent(self.win, evt)
|
|
|
|
self.claimResults()
|
|
|
|
# Results are done (claimstatus can be reported)
|
|
evt = UpdateAttackEvent(status="Done.")
|
|
wx.PostEvent(self.win, evt)
|
|
|
|
# And create the images in the background
|
|
self.makeImages()
|
|
|
|
# Cursor back to normal
|
|
self.win.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
|
|
|
|
busy.release()
|
|
|
|
def claimResults(self):
|
|
""" Convert spdl to result (using Scyther)
|
|
|
|
The list of claim goes back to self.win.claims, which is a
|
|
property of the main window
|
|
"""
|
|
|
|
scyther = Scyther.Scyther()
|
|
if sys.platform.startswith('win'):
|
|
scyther.program = "c:\\Scyther.exe"
|
|
if not os.path.isfile(scyther.program):
|
|
print "I can't find the Scyther executable %s" % (scyther.program)
|
|
|
|
scyther.setInput(self.spdl)
|
|
self.win.claims = scyther.verify()
|
|
self.summary = str(scyther)
|
|
|
|
self.win.errors.update(self.summary)
|
|
|
|
def makeImages(self):
|
|
""" create images """
|
|
for cl in self.win.claims:
|
|
for attack in cl.attacks:
|
|
self.makeImage(attack)
|
|
|
|
def makeImage(self,attack):
|
|
""" create image for this particular attack """
|
|
|
|
(fd2,fpname2) = Tempfile.tempcleaned(".png")
|
|
pw,pr = os.popen2("dot -Tpng -o%s" % (fpname2))
|
|
pw.write(attack.scytherDot)
|
|
pw.close()
|
|
attack.pngfile = fpname2 # this is where the file name is stored
|
|
|
|
|
|
def RunScyther(win,mode):
|
|
|
|
global busy
|
|
|
|
if (busy.acquire(False)):
|
|
|
|
# start the thread
|
|
win.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
|
|
|
|
win.settings.mode = mode
|
|
t = ScytherThread(win,win.control.GetValue(),"")
|
|
t.start()
|
|
win.threads = [ t ]
|
|
|
|
|