scyther/gui/Gui/Scytherthread.py

580 lines
18 KiB
Python
Raw Normal View History

2006-08-02 13:59:57 +01:00
#!/usr/bin/python
#---------------------------------------------------------------------------
""" Import externals """
import wx
import os
import sys
import re
import threading
import StringIO
#---------------------------------------------------------------------------
""" Import scyther components """
2006-12-14 14:06:50 +00:00
import Scyther.Scyther
import Scyther.Error
from Scyther.Misc import *
2006-08-02 13:59:57 +01:00
""" Import scyther-gui components """
import Tempfile
import Preference
2006-08-03 14:40:39 +01:00
import Attackwindow
2006-08-06 19:06:26 +01:00
import Icon
2006-12-14 14:06:50 +00:00
import Error
2006-08-02 13:59:57 +01:00
#---------------------------------------------------------------------------
if Preference.usePIL():
import Image
2006-08-02 13:59:57 +01:00
#---------------------------------------------------------------------------
class ScytherThread(threading.Thread):
"""
Apply Scyther algorithm to input and retrieve results
"""
2006-08-04 23:00:22 +01:00
# Override Thread's __init__ method to accept the parameters needed:
def __init__ ( self, spdl, options="", callback=None, mode=None ):
2006-08-02 13:59:57 +01:00
self.spdl = spdl
self.options = options
self.callback = callback
self.mode = mode
2006-08-02 13:59:57 +01:00
threading.Thread.__init__ ( self )
def run(self):
(scyther, claims, summary) = self.claimResults()
2006-08-02 13:59:57 +01:00
# Results are done (claimstatus can be reported)
if self.callback:
wx.CallAfter(self.callback, scyther, claims, summary)
def claimFixViewOne(self,claims):
if claims:
for cl in claims:
if len(cl.attacks) > 1:
# Fix it such that by default, only the best attack is
# shown, unless we are in characterize or check mode
# TODO [X] [CC] make switch-dependant.
if not self.mode in ["characterize","check"]:
cl.attacks = [cl.attacks[-1]]
return claims
2006-08-02 13:59:57 +01:00
def claimResults(self):
""" Convert spdl to result (using Scyther)
"""
2006-08-02 13:59:57 +01:00
2006-12-14 14:06:50 +00:00
scyther = Scyther.Scyther.Scyther()
scyther.options = self.options
scyther.setInput(self.spdl)
2006-08-09 12:54:37 +01:00
# verification start
try:
claims = scyther.verify()
except Scyther.Error.ScytherError, el:
claims = None
pass
summary = str(scyther)
2006-08-09 12:54:37 +01:00
claims = self.claimFixViewOne(claims)
return (scyther, claims, summary)
2006-08-02 13:59:57 +01:00
2006-08-04 23:00:22 +01:00
#---------------------------------------------------------------------------
2006-08-03 13:06:43 +01:00
class AttackThread(threading.Thread):
2006-08-04 23:00:22 +01:00
""" This is a thread because it computes images from stuff in the
background """
2006-08-03 13:06:43 +01:00
# Override Thread's __init__ method to accept the parameters needed:
2006-08-09 10:46:45 +01:00
def __init__ ( self, parent, resultwin, callbackclaim=None,callbackattack=None,callbackdone=None ):
2006-08-03 13:06:43 +01:00
2006-08-04 23:00:22 +01:00
self.parent = parent
2006-08-03 13:06:43 +01:00
self.resultwin = resultwin
self.callbackclaim = callbackclaim
self.callbackattack = callbackattack
2006-08-09 10:46:45 +01:00
self.callbackdone = callbackdone
self.totalattacks = 0
for cl in self.parent.claims:
for attack in cl.attacks:
self.totalattacks += 1
2006-08-03 13:06:43 +01:00
threading.Thread.__init__ ( self )
def run(self):
# create the images in the background
# when the images of a claim are done, callback is called with
# the claim
2006-08-03 13:06:43 +01:00
self.makeImages()
2006-08-02 13:59:57 +01:00
def makeImages(self):
""" create images """
done = 0
2006-08-04 23:00:22 +01:00
for cl in self.parent.claims:
for attack in cl.attacks:
self.makeImage(attack)
done += 1
if self.callbackattack:
2006-08-08 22:49:37 +01:00
wx.CallAfter(self.callbackattack,attack,self.totalattacks,done)
if self.callbackclaim:
2006-08-08 22:49:37 +01:00
wx.CallAfter(self.callbackclaim,cl)
2006-08-09 10:46:45 +01:00
if self.callbackdone:
wx.CallAfter(self.callbackdone)
2006-08-10 15:42:51 +01:00
def writeGraph(self,txt,fp):
2006-08-11 09:34:22 +01:00
EDGE = 0
NODE = 1
DEFAULT = 2
ALL = 3
2006-08-10 15:42:51 +01:00
def graphLine(txt):
fp.write("\t%s;\n" % (txt))
2006-08-11 09:34:22 +01:00
def setAttr(atxt,EdgeNodeDefAll=ALL):
if EdgeNodeDefAll == ALL:
setAttr(atxt,EDGE)
setAttr(atxt,NODE)
setAttr(atxt,DEFAULT)
2006-08-10 15:42:51 +01:00
else:
2006-08-11 09:34:22 +01:00
if EdgeNodeDefAll == EDGE:
2006-08-10 15:42:51 +01:00
edge = "edge"
2006-08-11 09:34:22 +01:00
elif EdgeNodeDefAll == NODE:
2006-08-10 15:42:51 +01:00
edge = "node"
else:
graphLine("%s" % atxt)
return
graphLine("%s [%s]" % (edge,atxt))
if sys.platform.startswith("darwin"):
self.fontname = "Helvetica"
elif sys.platform.startswith("win"):
self.fontname = "Courier"
else:
2007-05-19 14:43:57 +01:00
#font = wx.Font(9,wx.SWISS,wx.NORMAL,wx.NORMAL)
#self.fontname = font.GetFaceName()
self.fontname = "\"Helvetica\""
2006-08-11 09:34:22 +01:00
# write all graph lines but add layout modifiers
2006-08-10 15:42:51 +01:00
for l in txt.splitlines():
2006-08-11 09:34:22 +01:00
fp.write(l)
if l.startswith("digraph"):
# Write additional stuff for this graph
#
# [CC][x] This dpi setting messed up quite a bit
#graphLine("dpi=96")
2006-08-11 09:34:22 +01:00
graphLine("rankdir=TB")
2007-01-10 14:30:52 +00:00
#graphLine("nodesep=0.1")
#graphLine("ranksep=0.001")
#graphLine("mindist=0.1")
# Set fontname
if self.fontname:
fontstring = "fontname=%s" % (self.fontname)
setAttr(fontstring)
# Stupid Mac <> Graphviz bug fix
if (sys.platform.startswith("mac")) or (sys.platform.startswith("darwin")):
# Note that dot on Mac cannot find the fonts by default,
# and we have to set them accordingly.
os.environ["DOTFONTPATH"]="~/Library/Fonts:/Library/Fonts:/System/Library/Fonts"
# Select font size
2006-08-11 11:43:28 +01:00
if self.parent and self.parent.mainwin:
fontsize = self.parent.mainwin.settings.fontsize
setAttr("fontsize=%s" % fontsize)
2007-01-10 14:30:52 +00:00
#setAttr("height=\"0.1\"",NODE)
#setAttr("width=\"1.0\"",NODE)
#setAttr("margin=\"0.3,0.03\"",NODE)
2006-08-10 15:42:51 +01:00
def makeImage(self,attack):
""" create image for this particular attack """
if Preference.usePIL():
# If we have the PIL library, we can do postscript! great
# stuff.
type = "ps"
ext = ".ps"
else:
# Ye olde pnge file
type = "png"
ext = ".png"
# command to write to temporary file
(fd2,fpname2) = Tempfile.tempcleaned(ext)
f = os.fdopen(fd2,'w')
(fd3,fpname3) = Tempfile.tempcleaned(ext)
dotfile = os.fdopen(fd3,'w')
self.writeGraph(attack.scytherDot,dotfile)
dotfile.flush()
dotfile.seek(0)
2006-08-10 15:42:51 +01:00
cmd = "dot -T%s -o%s %s" % (type,fpname2,fpname3)
# execute command
# Start the process
safeCommand(cmd)
# Print
#print fpname2
#raw_input()
# if this is done, store and report
attack.filetype = type
attack.file = fpname2 # this is where the file name is stored
# Maybe we should remove the temporary file... TODO
#---------------------------------------------------------------------------
2006-08-03 13:06:43 +01:00
class VerificationWindow(wx.Dialog):
def __init__(
self, parent, title, pos=wx.DefaultPosition, size=wx.DefaultSize,
2006-08-03 13:06:43 +01:00
style=wx.DEFAULT_DIALOG_STYLE
):
wx.Dialog.__init__(self,parent,-1,title,pos,size,style)
2006-08-03 13:06:43 +01:00
sizer = wx.BoxSizer(wx.VERTICAL)
label = wx.StaticText(self, -1, "Verifying protocol description")
2006-08-03 13:06:43 +01:00
sizer.Add(label, 0, wx.ALIGN_CENTRE|wx.ALL, 5)
line = wx.StaticLine(self, -1, size=(20,-1), style=wx.LI_HORIZONTAL)
sizer.Add(line, 0, wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.RIGHT|wx.TOP, 5)
btnsizer = wx.StdDialogButtonSizer()
btn = wx.Button(self, wx.ID_CANCEL)
btnsizer.AddButton(btn)
btnsizer.Realize()
2006-08-07 10:31:49 +01:00
sizer.Add(btnsizer, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL|wx.ALIGN_CENTER, 5)
self.SetSizer(sizer)
sizer.Fit(self)
#---------------------------------------------------------------------------
class ErrorWindow(wx.Dialog):
def __init__(
self, parent, title, pos=wx.DefaultPosition, size=wx.DefaultSize,
2006-08-07 10:31:49 +01:00
style=wx.DEFAULT_DIALOG_STYLE,errors=[]
):
wx.Dialog.__init__(self,parent,-1,title,pos,size,style)
2006-08-07 10:31:49 +01:00
sizer = wx.BoxSizer(wx.VERTICAL)
label = wx.StaticText(self, -1, "Errors")
sizer.Add(label, 0, wx.ALIGN_LEFT|wx.ALL, 5)
line = wx.StaticLine(self, -1, size=(20,-1), style=wx.LI_HORIZONTAL)
sizer.Add(line, 0, wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.RIGHT|wx.TOP, 5)
2007-01-29 15:17:11 +00:00
etxt = ""
prefix = "error: "
for er in errors:
if er.startswith(prefix):
er = er[len(prefix):]
etxt = etxt + "%s\n" % (er)
label = wx.StaticText(self, -1, etxt)
2006-08-07 10:31:49 +01:00
sizer.Add(label, 0, wx.ALIGN_LEFT|wx.ALL, 5)
2006-08-07 17:36:54 +01:00
line = wx.StaticLine(self, -1, size=(20,-1), style=wx.LI_HORIZONTAL)
2006-08-07 10:31:49 +01:00
sizer.Add(line, 0, wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.RIGHT|wx.TOP, 5)
btnsizer = wx.StdDialogButtonSizer()
btn = wx.Button(self, wx.ID_OK)
btnsizer.AddButton(btn)
btnsizer.Realize()
sizer.Add(btnsizer, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL|wx.ALIGN_CENTER, 5)
2006-08-03 13:06:43 +01:00
self.SetSizer(sizer)
sizer.Fit(self)
#---------------------------------------------------------------------------
2006-08-03 14:40:39 +01:00
class ResultWindow(wx.Frame):
2006-08-04 23:29:51 +01:00
"""
Displays the claims status and contains buttons to show the actual
attack graphs
"""
2006-08-03 13:06:43 +01:00
def __init__(
2006-08-04 23:00:22 +01:00
self, parent, parentwindow, title, pos=wx.DefaultPosition, size=wx.DefaultSize,
2006-08-04 22:19:38 +01:00
style=wx.DEFAULT_DIALOG_STYLE
2006-08-03 13:06:43 +01:00
):
2006-08-02 13:59:57 +01:00
2006-08-04 23:00:22 +01:00
wx.Frame.__init__(self,parentwindow,-1,title,pos,size,style)
2007-01-10 14:30:52 +00:00
self.SetBackgroundColour('Default')
2006-08-06 19:06:26 +01:00
Icon.ScytherIcon(self)
2006-08-03 14:40:39 +01:00
2006-08-06 19:06:26 +01:00
self.parent = parent
self.thread = None
self.Bind(wx.EVT_CLOSE, self.onCloseWindow)
2006-08-03 13:06:43 +01:00
2006-08-06 19:06:26 +01:00
self.CreateStatusBar()
2006-08-04 22:28:58 +01:00
self.BuildTable()
def onViewButton(self,evt):
btn = evt.GetEventObject()
2006-08-04 23:08:00 +01:00
w = Attackwindow.AttackWindow(btn.claim)
w.Show(True)
2006-08-04 22:28:58 +01:00
def onCloseWindow(self,evt):
2006-08-04 23:29:51 +01:00
""" TODO we should kill self.thread """
2006-08-04 23:00:22 +01:00
# Clean up
self.parent.claims = None
2006-08-04 22:28:58 +01:00
self.Destroy()
def BuildTable(self):
2006-08-03 13:06:43 +01:00
# Now continue with the normal construction of the dialog
# contents
2006-08-04 22:28:58 +01:00
# For these claims...
2006-08-04 23:00:22 +01:00
claims = self.parent.claims
2006-08-04 22:28:58 +01:00
2006-08-03 14:40:39 +01:00
# set up grid
2006-08-09 08:59:57 +01:00
self.grid = grid = wx.GridBagSizer(0,0)
#self.grid = grid = wx.GridBagSizer(7,1+len(claims))
2006-08-03 14:40:39 +01:00
def titlebar(x,title,width=1):
txt = wx.StaticText(self,-1,title)
font = wx.Font(14,wx.NORMAL,wx.NORMAL,wx.BOLD)
2006-08-03 15:14:38 +01:00
txt.SetFont(font)
2006-08-09 08:59:57 +01:00
grid.Add(txt,(0,x),(1,width),wx.ALL,10)
2006-08-03 15:14:38 +01:00
titlebar(0,"Claim",4)
titlebar(4,"Status",2)
2006-08-09 12:36:33 +01:00
titlebar(6,"Comments",1)
2006-08-03 14:40:39 +01:00
2006-08-04 22:28:58 +01:00
self.lastprot = None
self.lastrole = None
views = 0
for index in range(0,len(claims)):
views += self.BuildClaim(grid,claims[index],index+1)
if views > 0:
titlebar(7,"Patterns",1)
self.SetSizer(grid)
self.Fit()
2006-08-03 13:06:43 +01:00
2006-08-04 22:28:58 +01:00
def BuildClaim(self,grid,cl,ypos):
# a support function
def addtxt(txt,column):
2006-08-09 08:59:57 +01:00
grid.Add(wx.StaticText(self,-1,txt),(ypos,column),(1,1),wx.ALIGN_CENTER_VERTICAL|wx.ALL,10)
n = len(cl.attacks)
xpos = 0
2006-08-04 22:28:58 +01:00
# protocol, role, label
prot = str(cl.protocol)
2006-08-09 12:36:33 +01:00
showP = False
showR = False
2006-08-04 22:28:58 +01:00
if prot != self.lastprot:
self.lastprot = prot
2006-08-09 12:36:33 +01:00
showP = True
showR = True
2006-08-04 22:28:58 +01:00
role = str(cl.role)
if role != self.lastrole:
self.lastrole = role
2006-08-09 12:36:33 +01:00
showR = True
if showP:
addtxt(prot,xpos)
2006-08-09 12:36:33 +01:00
if showR:
addtxt(role,xpos+1)
xpos += 2
2006-08-04 22:28:58 +01:00
# claim id
addtxt(str(cl.id),xpos)
xpos += 1
2006-08-04 22:28:58 +01:00
# claim parameters
2006-08-04 22:28:58 +01:00
claimdetails = str(cl.claimtype)
if cl.parameter:
2006-08-06 16:35:22 +01:00
claimdetails += " %s" % (cl.parameter)
addtxt(claimdetails + " ",xpos)
xpos += 1
2006-08-04 22:28:58 +01:00
# button for ok/fail
if None:
# old style buttons (but they looked ugly on windows)
tsize = (16,16)
if cl.okay:
bmp = wx.ArtProvider_GetBitmap(wx.ART_TICK_MARK,wx.ART_CMN_DIALOG,tsize)
else:
bmp = wx.ArtProvider_GetBitmap(wx.ART_CROSS_MARK,wx.ART_CMN_DIALOG,tsize)
if not bmp.Ok():
bmp = wx.EmptyBitmap(tsize)
bmpfield = wx.StaticBitmap(self,-1,bmp)
grid.Add(bmpfield,(ypos,xpos),(1,1),wx.ALIGN_CENTER_VERTICAL|wx.ALL,10)
else:
# new style text control Ok/Fail
rankc = cl.getColour()
rankt = cl.getOkay()
txt = wx.StaticText(self,-1,rankt)
font = wx.Font(11,wx.NORMAL,wx.NORMAL,wx.BOLD)
txt.SetFont(font)
txt.SetForegroundColour(rankc)
grid.Add(txt,(ypos,xpos),(1,1),wx.ALL,10)
xpos += 1
2006-08-04 22:28:58 +01:00
# verified?
vt = cl.getVerified()
if vt:
addtxt(vt,xpos)
xpos += 1
2006-08-09 12:28:01 +01:00
# remark something
addtxt(cl.getComment(),xpos)
xpos += 1
2006-08-04 22:28:58 +01:00
# add view button (enabled later if needed)
if n > 0:
cl.button = wx.Button(self,-1,"%i %s" % (n,cl.stateName(n)))
cl.button.claim = cl
grid.Add(cl.button,(ypos,xpos),(1,1),wx.ALIGN_CENTER_VERTICAL|wx.ALL,5)
cl.button.Disable()
if n > 0:
# Aha, something to show
self.Bind(wx.EVT_BUTTON, self.onViewButton,cl.button)
else:
cl.button = None
xpos += 1
# Return 1 if there is a view possible
if n > 0:
return 1
else:
return 0
2006-08-03 14:40:39 +01:00
2006-08-03 13:06:43 +01:00
#---------------------------------------------------------------------------
2006-08-04 23:00:22 +01:00
class ScytherRun(object):
2006-12-14 20:32:24 +00:00
2007-01-31 15:45:05 +00:00
def __init__(self,mainwin,mode,spdl,errorcallback=None):
2006-08-02 13:59:57 +01:00
2006-08-04 23:00:22 +01:00
self.mainwin = mainwin
self.mode = mode
2007-01-29 16:32:35 +00:00
self.spdl = spdl
self.verified = False
self.options = mainwin.settings.ScytherArguments(mode)
2007-01-31 15:45:05 +00:00
self.errorcallback=errorcallback
self.main()
2006-08-02 13:59:57 +01:00
2006-12-14 20:32:24 +00:00
def main(self):
"""
Start process
"""
title = "Running Scyther %s process" % self.mode
# start the window and show until something happens
# if it terminates, this is a cancel, and should also kill the thread. (what happens to a spawned Scyther in that case?)
# if the thread terminames, it should close the window normally, and we end up here as well.
#val = self.verifywin.ShowModal()
self.verifywin = VerificationWindow(self.mainwin,title)
self.verifywin.Center()
self.verifywin.Show(True)
# Check sanity of Scyther thing here (as opposed to the thread)
# which makes error reporting somewhat easier
try:
Scyther.Scyther.Check()
except Scyther.Error.BinaryError, e:
# e.file is the supposed location of the binary
text = "Could not find Scyther binary at\n%s" % (e.file)
Error.ShowAndExit(text)
# start the thread
self.verifywin.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
t = ScytherThread(self.spdl, self.options, self.verificationDone, self.mode)
2006-12-14 20:32:24 +00:00
t.start()
# after verification, we proceed to the callback below...
def verificationDone(self, scyther, claims, summary):
2006-12-14 20:32:24 +00:00
"""
This is where we end up after a callback from the thread, stating that verification succeeded.
"""
2006-08-03 13:06:43 +01:00
self.scyther = scyther
self.claims = claims
self.summary = summary
2006-08-03 13:06:43 +01:00
self.verified = True
self.verifywin.Close()
2006-08-02 13:59:57 +01:00
# Process the claims
if self.scyther.errorcount == 0:
self.verificationOkay()
else:
self.verificationErrors()
2006-08-03 13:06:43 +01:00
def verificationOkay(self):
2006-12-14 20:32:24 +00:00
# Great, we verified stuff, progress to the claim report
title = "Scyther results : %s" % self.mode
self.resultwin = resultwin = ResultWindow(self,self.mainwin,title)
2006-08-03 13:06:43 +01:00
def attackDone(attack,total,done):
if resultwin:
txt = "Generating attack graphs (%i of %i done)." % (done,total)
resultwin.SetStatusText(txt)
#resultwin.Refresh()
2006-08-07 10:31:49 +01:00
def claimDone(claim):
if resultwin:
if claim.button and len(claim.attacks) > 0:
claim.button.Enable()
def allDone():
if resultwin:
resultwin.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
resultwin.SetStatusText("Done.")
resultwin.Center()
resultwin.Show(True)
resultwin.SetCursor(wx.StockCursor(wx.CURSOR_ARROWWAIT))
2006-08-09 10:46:45 +01:00
wx.Yield()
2006-08-09 10:46:45 +01:00
t = AttackThread(self,resultwin,claimDone,attackDone,allDone)
t.start()
resultwin.thread = t
2006-08-07 10:31:49 +01:00
def verificationErrors(self):
"""
Verification process generated errors. Show them.
"""
2007-01-31 15:45:05 +00:00
if self.errorcallback:
self.errorcallback(self.scyther.errors)
title = "Scyther errors : %s" % self.mode
errorwin = ErrorWindow(self.mainwin,title,errors=self.scyther.errors)
errorwin.Center()
val = errorwin.ShowModal()
2006-08-04 23:00:22 +01:00
#---------------------------------------------------------------------------
2006-12-14 20:32:24 +00:00
# vim: set ts=4 sw=4 et list lcs=tab\:>-: