Merge pull request #369 from doronz88/feature/cmd-custom

protobuf_bridge: expose `CmdCustom`
doronz88 · Oct 15, 2024 · 4bd3ac9 · 4bd3ac9
2 parents 510dcab + 5158c00
commit 4bd3ac9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/rpcclient/rpcclient/protobuf_bridge.py b/src/rpcclient/rpcclient/protobuf_bridge.py
@@ -6,10 +6,10 @@
 os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'python'
 os.environ['TEMORARILY_DISABLE_PROTOBUF_VERSION_CHECK'] = 'true'
 
-from rpcclient.protos.rpc_pb2 import ARCH_ARM64, Argument, CmdCall, CmdClose, CmdDlclose, CmdDlopen, CmdDlsym, \
-    CmdDummyBlock, CmdExec, CmdGetClassList, CmdListDir, CmdPeek, CmdPoke, CmdShowClass, CmdShowObject, Command, \
-    Handshake, Response  # noqa: E402
+from rpcclient.protos.rpc_pb2 import ARCH_ARM64, Argument, CmdCall, CmdClose, CmdCustom, CmdDlclose, CmdDlopen, \
+    CmdDlsym, CmdDummyBlock, CmdExec, CmdGetClassList, CmdListDir, CmdPeek, CmdPoke, CmdShowClass, CmdShowObject, \
+    Command, Handshake, Response, ResponseCustom  # noqa: E402
 
 __all__ = ['Argument', 'CmdCall', 'CmdClose', 'CmdDlclose', 'CmdDlopen', 'CmdDlsym', 'CmdDummyBlock', 'CmdExec',
            'CmdGetClassList', 'CmdListDir', 'CmdPeek', 'CmdPoke', 'CmdShowClass', 'CmdShowObject', 'Command',
-           'Handshake', 'Response', 'ARCH_ARM64']
+           'Handshake', 'Response', 'ARCH_ARM64', 'CmdCustom', 'ResponseCustom']