diff options
Diffstat (limited to 'python')
| -rwxr-xr-x | python/examples/export_svg.py | 117 |
1 files changed, 85 insertions, 32 deletions
diff --git a/python/examples/export_svg.py b/python/examples/export_svg.py index d0885524..4ff2106f 100755 --- a/python/examples/export_svg.py +++ b/python/examples/export_svg.py @@ -8,9 +8,10 @@ from urllib.request import pathname2url from binaryninja.interaction import get_save_filename_input, show_message_box, TextLineField, ChoiceField, SaveFileNameField, get_form_input from binaryninja.settings import Settings -from binaryninja.enums import MessageBoxButtonSet, MessageBoxIcon, MessageBoxButtonResult, InstructionTextTokenType, BranchType, DisassemblyOption, FunctionGraphType +from binaryninja.enums import MessageBoxButtonSet, MessageBoxIcon, MessageBoxButtonResult, InstructionTextTokenType, BranchType, DisassemblyOption, FunctionGraphType, ThemeColor from binaryninja.function import DisassemblySettings from binaryninja.plugin import PluginCommand +from binaryninjaui import getThemeColor, getTokenColor, UIContext colors = { 'green': [162, 217, 175], 'red': [222, 143, 151], 'blue': [128, 198, 233], 'cyan': [142, 230, 237], @@ -92,6 +93,23 @@ def instruction_data_flow(function, address): padded = ' '.join([hex[i:i + 2] for i in range(0, len(hex), 2)]) return f'Opcode: {padded}' +def rgbStr(tokenType): + '''Given a token string name, look up the theme color for it and return as rbg(x,y,z) str''' + try: + color = eval(f'getThemeColor(ThemeColor.{tokenType})') + except: + color = None + if (not color): + try: + ctx = UIContext.activeContext() + view_frame = ctx.getCurrentViewFrame() + color = eval(f'getTokenColor(view_frame, InstructionTextTokenType.{tokenType})') + except: + return 'rgb(224, 224, 224)' + r = color.getRgb()[0] + g = color.getRgb()[1] + b = color.getRgb()[2] + return f"rgb({r}, {g}, {b})" def render_svg(function, offset, mode, form, showOpcodes, showAddresses, origname): settings = DisassemblySettings() @@ -130,21 +148,23 @@ def render_svg(function, offset, mode, form, showOpcodes, showAddresses, orignam <style type="text/css"> @import url(https://fonts.googleapis.com/css?family=Source+Code+Pro); body {{ + /* These colors are only for the bottom section, can tweak later */ background-color: rgb(42, 42, 42); - color: rgb(220, 220, 220); - font-family: "Source Code Pro", "Lucida Console", "Consolas", monospace; + color: rgb(220, 220, 220); + font-family: "Source Code Pro", "Lucida Console", "Consolas", monospace; + }} + a, a:visited {{ + color: rgb(200, 200, 200); + font-weight: bold; }} - a, a:visited {{ - color: rgb(200, 200, 200); - font-weight: bold; - }} svg {{ - background-color: rgb(42, 42, 42); + background-color: {rgbStr('GraphBackgroundDarkColor')}; display: block; margin: 0 auto; }} .basicblock {{ - stroke: rgb(224, 224, 224); + stroke: {rgbStr('GraphNodeOutlineColor')}; + fill: {rgbStr('GraphNodeDarkColor')}; }} .edge {{ fill: none; @@ -155,16 +175,16 @@ def render_svg(function, offset, mode, form, showOpcodes, showAddresses, orignam stroke-width: 2px; }} .UnconditionalBranch, .IndirectBranch {{ - stroke: rgb(128, 198, 233); - color: rgb(128, 198, 233); + stroke: {rgbStr('UnconditionalBranchColor')}; + color: {rgbStr('UnconditionalBranchColor')}; }} .FalseBranch {{ - stroke: rgb(222, 143, 151); - color: rgb(222, 143, 151); + stroke: {rgbStr('FalseBranchColor')}; + color: {rgbStr('FalseBranchColor')}; }} .TrueBranch {{ - stroke: rgb(162, 217, 175); - color: rgb(162, 217, 175); + stroke: {rgbStr('TrueBranchColor')}; + color: {rgbStr('TrueBranchColor')}; }} .arrow {{ stroke-width: 1; @@ -173,34 +193,67 @@ def render_svg(function, offset, mode, form, showOpcodes, showAddresses, orignam text {{ font-family: "Source Code Pro", "Lucida Console", "Consolas", monospace; font-size: 9pt; - fill: rgb(224, 224, 224); + fill: {rgbStr('TextToken')}; + }} + .RegisterToken {{ + fill: {rgbStr('RegisterColor')}; + }} + .CodeRelativeAddressToken, .PossibleAddressToken, .IntegerToken, .FloatingPointToken, .ArrayIndexToken {{ + fill: {rgbStr('NumberColor')}; }} .CodeSymbolToken {{ - fill: rgb(128, 198, 223); + fill: {rgbStr('CodeSymbolColor')}; }} .DataSymbolToken {{ - fill: rgb(142, 230, 237); - }} - .TextToken, .InstructionToken, .BeginMemoryOperandToken, .EndMemoryOperandToken {{ - fill: rgb(224, 224, 224); + fill: {rgbStr('DataSymbolColor')}; }} - .CodeRelativeAddressToken, .PossibleAddressToken, .IntegerToken, .AddressDisplayToken {{ - fill: rgb(162, 217, 175); + .LocalVariableToken, .ArgumentNameToken {{ + fill: {rgbStr('StackVariableColor')}; }} - .RegisterToken {{ - fill: rgb(237, 223, 179); + .IndirectImportToken, .ImportToken, .ExternalSymbolToken {{ + fill: {rgbStr('ImportColor')}; }} .AnnotationToken {{ - fill: rgb(218, 196, 209); + fill: {rgbStr('AnnotationColor')}; }} - .IndirectImportToken, .ImportToken, .ExternalSymbolToken {{ - fill: rgb(237, 189, 129); + .CommentToken {{ + fill: {rgbStr('CommentColor')}; + }} + .AddressDisplayToken {{ + fill: {rgbStr('AddressColor')}; + }} + .UnknownMemoryToken, .OpcodeToken {{ + fill: {rgbStr('OpcodeColor')}; + }} + .StringToken, .CharacterConstantToken {{ + fill: {rgbStr('StringColor')}; + }} + .TypeNameToken {{ + fill: {rgbStr('TypeNameColor')}; + }} + .FieldNameToken, .StructOffsetToken {{ + fill: {rgbStr('FieldNameColor')}; }} - .LocalVariableToken, .StackVariableToken {{ - fill: rgb(193, 220, 199); + .KeywordToken, .EnumerationMemberToken {{ + fill: {rgbStr('KeywordColor')}; }} - .OpcodeToken {{ - fill: rgb(144, 144, 144); + .NamespaceToken {{ + fill: {rgbStr('NameSpaceColor')}; + }} + .NamespaceSeparatorToken {{ + fill: {rgbStr('NameSpaceSeparatorColor')}; + }} + .GotoLabelToken {{ + fill: {rgbStr('GotoLabelColor')}; + }} + .OperationToken {{ + fill: {rgbStr('OperationColor')}; + }} + .BaseStructureNameToken, .BaseStructureSeparatorToken {{ + fill: {rgbStr('BaseStructureNameColor')}; + }} + .TextToken, .InstructionToken, .BeginMemoryOperandToken, .EndMemoryOperandToken {{ + fill: {rgbStr('TextToken')}; }} </style> <defs> |