From 5158c005cf229b3d2a3ea17705666e805ea74d18 Mon Sep 17 00:00:00 2001 From: doronz88 Date: Tue, 15 Oct 2024 19:49:05 +0300 Subject: [PATCH] protobuf_bridge: expose `CmdCustom` --- src/rpcclient/rpcclient/protobuf_bridge.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/rpcclient/rpcclient/protobuf_bridge.py b/src/rpcclient/rpcclient/protobuf_bridge.py index 720b02a..3306ca2 100644 --- 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']