#!/usr/bin/python #--------------------------------------------------------------------------- """ Import externals """ import wx import os import sys import re import threading import StringIO # Python Imaging library? usePIL = True try: import Image except ImportError: usePIL = False #--------------------------------------------------------------------------- """ Import scyther components """ import XMLReader """ Import scyther-gui components """ import Tempfile import Claim import Preference import Scyther import Attackwindow import Icon #--------------------------------------------------------------------------- #--------------------------------------------------------------------------- class ScytherThread(threading.Thread): """ The reason this is a thread is because we might to decide to abort it. However, apparently Python has no good support for killing threads yet :( """ # Override Thread's __init__ method to accept the parameters needed: def __init__ ( self, parent ): self.parent = parent parent.verified = False parent.claims = [] threading.Thread.__init__ ( self ) def run(self): self.claimResults() # Results are done (claimstatus can be reported) # Shoot down the verification window and let the RunScyther function handle the rest self.parent.verified = True self.parent.verifywin.Close() def claimResults(self): """ Convert spdl to result (using Scyther) """ self.parent.scyther = scyther = Scyther.Scyther() scyther.options = self.parent.options scyther.setInput(self.parent.spdl) self.parent.claims = scyther.verify() self.parent.summary = str(scyther) #--------------------------------------------------------------------------- class AttackThread(threading.Thread): """ This is a thread because it computes images from stuff in the background """ # Override Thread's __init__ method to accept the parameters needed: def __init__ ( self, parent, resultwin, callbackclaim=None,callbackattack=None ): self.parent = parent self.resultwin = resultwin self.callbackclaim = callbackclaim self.callbackattack = callbackattack self.totalattacks = 0 for cl in self.parent.claims: for attack in cl.attacks: self.totalattacks += 1 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 self.makeImages() def makeImages(self): """ create images """ done = 0 for cl in self.parent.claims: for attack in cl.attacks: self.makeImage(attack) done += 1 if self.callbackattack: self.callbackattack(attack,self.totalattacks,done) if self.callbackclaim: self.callbackclaim(cl) def makeImage(self,attack): """ create image for this particular attack """ global usePIL if 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') cmd = "dot -T%s" % (type) # execute command cin,cout = os.popen2(cmd) cin.write(attack.scytherDot) cin.close() for l in cout.read(): f.write(l) cout.close() f.flush() f.close() # if this is done, store and report attack.filetype = type attack.file = fpname2 # this is where the file name is stored #--------------------------------------------------------------------------- class VerificationWindow(wx.Dialog): def __init__( self, parent, title, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE ): wx.Dialog.__init__(self,parent,-1,title,pos,size,style) sizer = wx.BoxSizer(wx.VERTICAL) label = wx.StaticText(self, -1, "Verifying protocol description") 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() 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, style=wx.DEFAULT_DIALOG_STYLE,errors=[] ): wx.Dialog.__init__(self,parent,-1,title,pos,size,style) 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) label = wx.StaticText(self, -1, "\n".join(errors)) sizer.Add(label, 0, wx.ALIGN_LEFT|wx.ALL, 5) 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) self.SetSizer(sizer) sizer.Fit(self) #--------------------------------------------------------------------------- class ResultWindow(wx.Frame): """ Displays the claims status and contains buttons to show the actual attack graphs TODO: this really should have a statusbar that works. TODO: on windows, it updates really slow, and the background is the wrong colour. Basically, it inhales air. Hard. """ def __init__( self, parent, parentwindow, title, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE ): wx.Frame.__init__(self,parentwindow,-1,title,pos,size,style) Icon.ScytherIcon(self) self.parent = parent self.thread = None self.Bind(wx.EVT_CLOSE, self.onCloseWindow) self.CreateStatusBar() self.BuildTable() def onViewButton(self,evt): btn = evt.GetEventObject() w = Attackwindow.AttackWindow(btn.claim) w.Show(True) def onCloseWindow(self,evt): """ TODO we should kill self.thread """ # Clean up self.parent.claims = None self.Destroy() def BuildTable(self): # Now continue with the normal construction of the dialog # contents sizer = wx.BoxSizer(wx.VERTICAL) # For these claims... claims = self.parent.claims # set up grid self.grid = grid = wx.GridBagSizer(7,1+len(claims)) def titlebar(x,title,width=1): txt = wx.StaticText(self,-1,title) font = wx.Font(14,wx.NORMAL,wx.NORMAL,wx.NORMAL) txt.SetFont(font) grid.Add(txt,(0,x),(1,width)) titlebar(0,"Claim",5) if len(claims) > 0: sn = claims[0].stateName(2) resulttxt = sn[0].upper() + sn[1:] else: resulttxt = "Results" titlebar(5,resulttxt,2) self.lastprot = None self.lastrole = None for index in range(0,len(claims)): self.BuildClaim(grid,claims[index],index+1) sizer.Add(grid, 0,wx.ALIGN_CENTRE|wx.ALL,5) self.SetSizer(sizer) sizer.Fit(self) def BuildClaim(self,grid,cl,ypos): # a support function def addtxt(txt,column): grid.Add(wx.StaticText(self,-1,txt),(ypos,column),(1,1),wx.ALIGN_CENTER_VERTICAL) # button for ok/fail 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,0),(1,1),wx.ALIGN_CENTER_VERTICAL) # protocol, role, label prot = str(cl.protocol) showPR = False if prot != self.lastprot: self.lastprot = prot showPR = True role = str(cl.role) if role != self.lastrole: self.lastrole = role showPR = True if showPR: addtxt(prot,1) addtxt(role,2) addtxt(str(cl.shortlabel),3) claimdetails = str(cl.claimtype) if cl.parameter: claimdetails += " %s" % (cl.parameter) addtxt(claimdetails + " ",4) # add view button (if needed) n = len(cl.attacks) cl.button = wx.Button(self,-1,"%i %s" % (n,cl.stateName(n))) cl.button.claim = cl grid.Add(cl.button,(ypos,5),(1,1),wx.ALIGN_CENTER_VERTICAL) cl.button.Disable() if n > 0: # Aha, something to show self.Bind(wx.EVT_BUTTON, self.onViewButton,cl.button) # remark something about completeness remark = "" if not cl.complete: if n == 0: # no attacks, no states within bounds remark = "within bounds" else: # some attacks/states within bounds remark = "at least, maybe more" else: if n == 0: # no attacks, no states remark = "none" else: # there exist n states/attacks (within any number of runs) remark = "exactly" addtxt(" (%s)" % remark,6) #--------------------------------------------------------------------------- class ScytherRun(object): def __init__(self,mainwin,mode): self.mainwin = mainwin self.mode = mode self.spdl = mainwin.control.GetValue() # Verification window self.verifywin = verifywin = VerificationWindow(mainwin,"Running Scyther %s process" % mode) verifywin.CenterOnScreen() verifywin.Show(True) # start the thread self.options = mainwin.settings.ScytherArguments(mode) self.verified = False verifywin.SetCursor(wx.StockCursor(wx.CURSOR_WAIT)) t = ScytherThread(self) t.start() # 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 = verifywin.ShowModal() # Cursor back to normal verifywin.SetCursor(wx.StockCursor(wx.CURSOR_ARROW)) if self.verified: # Scyther program is done (the alternative is that it was # cancelled) if self.scyther.errorcount == 0: # Great, we verified stuff, progress to the claim report title = "Scyther results : %s" % mode self.resultwin = resultwin = ResultWindow(self,mainwin,title) resultwin.Show(True) def attackDone(attack,total,done): if done < total: txt = "Generating attack graphs (%i of %i done)." % (done,total) else: txt = "Done." resultwin.SetStatusText(txt) def claimDone(claim): if claim.button and len(claim.attacks) > 0: claim.button.Enable() t = AttackThread(self,resultwin,claimDone,attackDone) t.start() resultwin.thread = t resultwin.CenterOnScreen() resultwin.Show(True) else: # Darn, some errors. report. title = "Scyther errors : %s" % mode errorwin = ErrorWindow(mainwin,title,errors=self.scyther.errors) errorwin.Show(True) errorwin.CenterOnScreen() val = errorwin.ShowModal() #---------------------------------------------------------------------------