def __init__(self, name=None):
GraphWidget.__init__(self)
self.getParams().addChildren([
{'name':'Draw Type', 'type':'list', 'key':'drawtype', 'values':['Fastest', 'Normal', 'Detailed'],
'value':'Normal',
'help': '%namehdr%'+
"Draw types:\n"
" * Fast: Group traces into a min/max area;\n"
" * Normal: Plot all traces continuously;\n"
" * Detailed: Plot all traces individually.\n\n"
"Only highlighted traces can be selected in fast/normal modes.\n"
},
])
self.setObjectName(self.getName())
self.bselection = QToolBar()
self.layout().addWidget(self.bselection)
self.doRedraw = True
self.enabledbytes = []
AttackObserver.__init__(self)
self.initUI(True)
# Setup the redrawPlot() to be delayed when pressing the Key buttons
self.delayedRedrawPlot = Timer()
self.delayedRedrawPlot.timeout.connect(self.redrawPlot)
self.delayedRedrawPlot.setSingleShot(True)
self.delayedRedrawPlot.setInterval(1000)
def __init__(self, name=None):
GraphWidget.__init__(self)
self.getParams().addChildren([
{
'name':
'Draw Type',
'type':
'list',
'key':
'drawtype',
'values': ['Fastest', 'Normal', 'Detailed'],
'value':
'Normal',
'help':
'%namehdr%' + "Draw types:\n"
" * Fast: Group traces into a min/max area;\n"
" * Normal: Plot all traces continuously;\n"
" * Detailed: Plot all traces individually.\n\n"
"Only highlighted traces can be selected in fast/normal modes.\n"
},
])
self.setObjectName(self.getName())
self.bselection = QToolBar()
self.layout().addWidget(self.bselection)
self.doRedraw = True
self.enabledbytes = []
AttackObserver.__init__(self)
self.initUI(True)
# Setup the redrawPlot() to be delayed when pressing the Key buttons
self.delayedRedrawPlot = Timer()
self.delayedRedrawPlot.timeout.connect(self.redrawPlot)
self.delayedRedrawPlot.setSingleShot(True)
self.delayedRedrawPlot.setInterval(1000)
class AttackResultPlot(GraphWidget, ResultsBase, AttackObserver):
"""
Generic data plotting stuff. Adds ability to highlight certain guesses, used in plotting for example the
correlation over all data points, or the most likely correlation over number of traces
"""
def __init__(self, name=None):
GraphWidget.__init__(self)
self.getParams().addChildren([
{'name':'Draw Type', 'type':'list', 'key':'drawtype', 'values':['Fastest', 'Normal', 'Detailed'],
'value':'Normal',
'help': '%namehdr%'+
"Draw types:\n"
" * Fast: Group traces into a min/max area;\n"
" * Normal: Plot all traces continuously;\n"
" * Detailed: Plot all traces individually.\n\n"
"Only highlighted traces can be selected in fast/normal modes.\n"
},
])
self.setObjectName(self.getName())
self.bselection = QToolBar()
self.layout().addWidget(self.bselection)
self.doRedraw = True
self.enabledbytes = []
AttackObserver.__init__(self)
self.initUI(True)
# Setup the redrawPlot() to be delayed when pressing the Key buttons
self.delayedRedrawPlot = Timer()
self.delayedRedrawPlot.timeout.connect(self.redrawPlot)
self.delayedRedrawPlot.setSingleShot(True)
self.delayedRedrawPlot.setInterval(1000)
def initUI(self, firstTime=False):
if firstTime or self._numKeys() != len(self.enabledbytes):
self.enabledbytes = [False]*self._numKeys()
self.bselection.clear()
self.byteNumAct=[]
if self._analysisSource:
for i in range(0, self._numKeys()):
newAct = QAction('%d' % i, self)
newAct.triggered[bool].connect(partial(self.setBytePlot, i))
newAct.setCheckable(True)
self.bselection.addAction(newAct)
self.byteNumAct.append(newAct)
byteNumAllOn = QAction('All On', self)
byteNumAllOn.triggered.connect(partial(self.setByteAll, False))
self.bselection.addAction(byteNumAllOn)
byteNumAllOff = QAction('All Off', self)
byteNumAllOff.triggered.connect(partial(self.setByteAll, True))
self.bselection.addAction(byteNumAllOff)
def analysisStarted(self):
self.initUI()
for i in range(0, self._numKeys()):
if i in self._analysisSource.getTargetSubkeys():
self.byteNumAct[i].setVisible(True)
else:
self.byteNumAct[i].setVisible(False)
def setBytePlot(self, num, sel):
"""Set which bytes to plot"""
self.enabledbytes[num] = sel
if self.doRedraw:
self.delayedRedrawPlot.start()
def setByteAll(self, status):
"""Enable all bytes in plot"""
self.doRedraw = False
for t in self.byteNumAct:
t.setChecked(status)
t.trigger()
self.doRedraw = True
self.redrawPlot()
def redrawPlot(self):
pass
def drawData(self, progress, xdatalst, ydatalst, enabledBytes=None):
"""Redraw the plot"""
progress.setMaximum(len(enabledBytes))
progress.setStatusMask("Clearing previous plotting...")
self.clearPushed()
progress.setStatusMask("Plotting...")
drawtype = self.findParam('drawtype').getValue().lower()
pvalue = 0
top = bottom = None
xdataptr = None
highlightedKeys = self._highlightedKeys()
for bnum in enabledBytes:
if not xdatalst[bnum] or len(xdatalst[bnum])==0:
break
if bnum != -1:
ydataptr = ydatalst[bnum]
xdataptr = xdatalst[bnum]
else:
ydataptr = ydatalst
xdataptr = xdatalst
pointargsg = {}
if len(ydataptr) == 0 or ydataptr[0] is None:
continue
if not hasattr(ydataptr[0], '__iter__'):
ydataptr = [[t] for t in ydataptr]
pointargsg = {'symbol':'t', 'symbolPen':'b', 'symbolBrush':'g'}
if drawtype.startswith('fast'):
newdiff = np.array(ydataptr)
if highlightedKeys[bnum] is not None and bnum < len(highlightedKeys):
newdiff = np.delete(newdiff, highlightedKeys[bnum], 0)
if top is not None:
top = np.maximum.reduce([top, np.amax(newdiff, 0)])
bottom = np.minimum.reduce([bottom, np.amin(newdiff, 0)])
else:
top = np.amax(newdiff, 0)
bottom = np.amin(newdiff, 0)
elif drawtype.startswith('norm'):
tlisttst = []
maxlisttst = []
if len(pointargsg) == 0:
tlist_fixed = [xdataptr[-1], xdataptr[0]]
tlist_fixed[:0] = xdataptr
else:
tlist_fixed = xdataptr
for i in range(0, self._numPerms(bnum)):
if bnum < len(highlightedKeys) and i == highlightedKeys[bnum]:
continue
tlisttst.extend(tlist_fixed)
if len(pointargsg) == 0:
newmax = [0, 0]
newmax[:0] = ydataptr[i]
else:
newmax = ydataptr[i]
maxlisttst.extend(newmax)
self.setupPlot(self.pw.plot(tlisttst, maxlisttst, pen=self.traceColor, **pointargsg), 0, False, str(bnum) + ":All")
elif drawtype.startswith('detail'):
for i in range(0, self._numPerms(bnum)):
self.setupPlot(self.pw.plot(xdataptr, ydataptr[i], pen=QColor(*self.traceColor), **pointargsg), 0, True, str(bnum) + ":%02X" % i)
# Plot the highlighted byte(s) on top
pointargsr = {}
if not hasattr(ydataptr[0], '__iter__'):
ydataptr = [[t] for t in ydataptr]
pointargsr = {'symbol':'o', 'symbolPen':'b', 'symbolBrush':'r'}
if bnum < len(highlightedKeys):
b = highlightedKeys[bnum]
if b >=0 and b < len(ydataptr):
self.setupPlot(self.pw.plot(xdataptr, ydataptr[b], pen=QColor(*self.highlightedKeyColor), **pointargsr), 1, True, str(bnum) + ":%02X" % b)
pvalue += 1
progress.updateStatus(pvalue)
if progress.wasAborted():
break
if drawtype.startswith('fast') and xdataptr:
p1 = self.setupPlot(self.pw.plot(x=xdataptr, y=top), -1, True, "Maxes")
p2 = self.setupPlot(self.pw.plot(x=xdataptr, y=bottom), -1, True, "Mins")
try:
p3 = pg.FillBetweenItem(p1, p2, brush=self.traceColor)
p3.setZValue(-1)
self.pw.addItem(p3)
except AttributeError:
logging.debug("fast draw type but failed to fill - probably old version of pyqtgraph?")
def processAnalysis(self):
self.redrawPlot()
def getWidget(self):
return self
class AttackResultPlot(GraphWidget, ResultsBase, AttackObserver):
"""
Generic data plotting stuff. Adds ability to highlight certain guesses, used in plotting for example the
correlation over all data points, or the most likely correlation over number of traces
"""
def __init__(self, name=None):
GraphWidget.__init__(self)
self.getParams().addChildren([
{
'name':
'Draw Type',
'type':
'list',
'key':
'drawtype',
'values': ['Fastest', 'Normal', 'Detailed'],
'value':
'Normal',
'help':
'%namehdr%' + "Draw types:\n"
" * Fast: Group traces into a min/max area;\n"
" * Normal: Plot all traces continuously;\n"
" * Detailed: Plot all traces individually.\n\n"
"Only highlighted traces can be selected in fast/normal modes.\n"
},
])
self.setObjectName(self.getName())
self.bselection = QToolBar()
self.layout().addWidget(self.bselection)
self.doRedraw = True
self.enabledbytes = []
AttackObserver.__init__(self)
self.initUI(True)
# Setup the redrawPlot() to be delayed when pressing the Key buttons
self.delayedRedrawPlot = Timer()
self.delayedRedrawPlot.timeout.connect(self.redrawPlot)
self.delayedRedrawPlot.setSingleShot(True)
self.delayedRedrawPlot.setInterval(1000)
def initUI(self, firstTime=False):
if firstTime or self._numKeys() != len(self.enabledbytes):
self.enabledbytes = [False] * self._numKeys()
self.bselection.clear()
self.byteNumAct = []
if self._analysisSource:
for i in range(0, self._numKeys()):
newAct = QAction('%d' % i, self)
newAct.triggered[bool].connect(partial(
self.setBytePlot, i))
newAct.setCheckable(True)
self.bselection.addAction(newAct)
self.byteNumAct.append(newAct)
byteNumAllOn = QAction('All On', self)
byteNumAllOn.triggered.connect(partial(self.setByteAll, False))
self.bselection.addAction(byteNumAllOn)
byteNumAllOff = QAction('All Off', self)
byteNumAllOff.triggered.connect(partial(self.setByteAll, True))
self.bselection.addAction(byteNumAllOff)
def analysisStarted(self):
self.initUI()
for i in range(0, self._numKeys()):
if i in self._analysisSource.getTargetSubkeys():
self.byteNumAct[i].setVisible(True)
else:
self.byteNumAct[i].setVisible(False)
def setBytePlot(self, num, sel):
"""Set which bytes to plot"""
self.enabledbytes[num] = sel
if self.doRedraw:
self.delayedRedrawPlot.start()
def setByteAll(self, status):
"""Enable all bytes in plot"""
self.doRedraw = False
for t in self.byteNumAct:
t.setChecked(status)
t.trigger()
self.doRedraw = True
self.redrawPlot()
def redrawPlot(self):
pass
def drawData(self, progress, xdatalst, ydatalst, enabledBytes=None):
"""Redraw the plot"""
progress.setMaximum(len(enabledBytes))
progress.setStatusMask("Clearing previous plotting...")
self.clearPushed()
progress.setStatusMask("Plotting...")
drawtype = self.findParam('drawtype').getValue().lower()
pvalue = 0
top = bottom = None
xdataptr = None
highlightedKeys = self._highlightedKeys()
for bnum in enabledBytes:
if not xdatalst[bnum] or len(xdatalst[bnum]) == 0:
break
if bnum != -1:
ydataptr = ydatalst[bnum]
xdataptr = xdatalst[bnum]
else:
ydataptr = ydatalst
xdataptr = xdatalst
pointargsg = {}
if len(ydataptr) == 0 or ydataptr[0] is None:
continue
if not hasattr(ydataptr[0], '__iter__'):
ydataptr = [[t] for t in ydataptr]
pointargsg = {
'symbol': 't',
'symbolPen': 'b',
'symbolBrush': 'g'
}
if drawtype.startswith('fast'):
newdiff = np.array(ydataptr)
if highlightedKeys[bnum] is not None and bnum < len(
highlightedKeys):
newdiff = np.delete(newdiff, highlightedKeys[bnum], 0)
if top is not None:
top = np.maximum.reduce([top, np.amax(newdiff, 0)])
bottom = np.minimum.reduce([bottom, np.amin(newdiff, 0)])
else:
top = np.amax(newdiff, 0)
bottom = np.amin(newdiff, 0)
elif drawtype.startswith('norm'):
tlisttst = []
maxlisttst = []
if len(pointargsg) == 0:
tlist_fixed = [xdataptr[-1], xdataptr[0]]
tlist_fixed[:0] = xdataptr
else:
tlist_fixed = xdataptr
for i in range(0, self._numPerms(bnum)):
if bnum < len(
highlightedKeys) and i == highlightedKeys[bnum]:
continue
tlisttst.extend(tlist_fixed)
if len(pointargsg) == 0:
newmax = [0, 0]
newmax[:0] = ydataptr[i]
else:
newmax = ydataptr[i]
maxlisttst.extend(newmax)
self.setupPlot(
self.pw.plot(tlisttst,
maxlisttst,
pen=self.traceColor,
**pointargsg), 0, False,
str(bnum) + ":All")
elif drawtype.startswith('detail'):
for i in range(0, self._numPerms(bnum)):
self.setupPlot(
self.pw.plot(xdataptr,
ydataptr[i],
pen=QColor(*self.traceColor),
**pointargsg), 0, True,
str(bnum) + ":%02X" % i)
# Plot the highlighted byte(s) on top
pointargsr = {}
if not hasattr(ydataptr[0], '__iter__'):
ydataptr = [[t] for t in ydataptr]
pointargsr = {
'symbol': 'o',
'symbolPen': 'b',
'symbolBrush': 'r'
}
if bnum < len(highlightedKeys):
b = highlightedKeys[bnum]
if b >= 0 and b < len(ydataptr):
self.setupPlot(
self.pw.plot(xdataptr,
ydataptr[b],
pen=QColor(*self.highlightedKeyColor),
**pointargsr), 1, True,
str(bnum) + ":%02X" % b)
pvalue += 1
progress.updateStatus(pvalue)
if progress.wasAborted():
break
if drawtype.startswith('fast') and xdataptr:
p1 = self.setupPlot(self.pw.plot(x=xdataptr, y=top), -1, True,
"Maxes")
p2 = self.setupPlot(self.pw.plot(x=xdataptr, y=bottom), -1, True,
"Mins")
try:
p3 = pg.FillBetweenItem(p1, p2, brush=self.traceColor)
p3.setZValue(-1)
self.pw.addItem(p3)
except AttributeError:
logging.debug(
"fast draw type but failed to fill - probably old version of pyqtgraph?"
)
def processAnalysis(self):
self.redrawPlot()
def getWidget(self):
return self
