diff options
| author | Rusty Wagner <rusty@vector35.com> | 2019-02-26 23:09:38 -0500 |
|---|---|---|
| committer | Rusty Wagner <rusty@vector35.com> | 2019-02-26 23:09:38 -0500 |
| commit | 49f6de04c7106e6ab8ce31fac6299f372a665698 (patch) | |
| tree | 1a0711864b3c113c273d4453f14773445037295e /python | |
| parent | 2f84b5c600f08101a6a7ee08738a3e3a32686f46 (diff) | |
Fix some flow graph and disassembly API issues
Diffstat (limited to 'python')
| -rw-r--r-- | python/flowgraph.py | 9 | ||||
| -rw-r--r-- | python/function.py | 5 |
2 files changed, 11 insertions, 3 deletions
diff --git a/python/flowgraph.py b/python/flowgraph.py index 103eed69..4d2bd3ba 100644 --- a/python/flowgraph.py +++ b/python/flowgraph.py @@ -78,7 +78,7 @@ class FlowGraphNode(object): @property def basic_block(self): - """Basic block associated with this part of the flow graph (read-only)""" + """Basic block associated with this part of the flow graph""" block = core.BNGetFlowGraphBasicBlock(self.handle) if not block: return None @@ -100,6 +100,13 @@ class FlowGraphNode(object): block = basicblock.BasicBlock(block, view) return block + @basic_block.setter + def basic_block(self, block): + if block is None: + core.BNSetFlowGraphBasicBlock(self.handle, None) + else: + core.BNSetFlowGraphBasicBlock(self.handle, block.handle) + @property def x(self): """Flow graph block X (read-only)""" diff --git a/python/function.py b/python/function.py index 7471d639..065211c9 100644 --- a/python/function.py +++ b/python/function.py @@ -1905,11 +1905,12 @@ class DisassemblyTextRenderer(object): def basic_block(self, block): if block is not None: core.BNSetDisassemblyTextRendererBasicBlock(self.handle, block.handle) - core.BNSetDisassemblyTextRendererBasicBlock(self.handle, None) + else: + core.BNSetDisassemblyTextRendererBasicBlock(self.handle, None) @property def arch(self): - return binaryninja.architecture.Architecture(handle = core.BNSetDisassemblyTextRendererArchitecture(self.handle)) + return binaryninja.architecture.CoreArchitecture(handle = core.BNGetDisassemblyTextRendererArchitecture(self.handle)) @arch.setter def arch(self, arch): |