From 612c320425871e27d66e3fa97159e63499ec4bfd Mon Sep 17 00:00:00 2001
From: doronz88 <doron88@gmail.com>
Date: Mon, 14 Oct 2024 17:22:15 +0300
Subject: [PATCH] symbol: guard pb2 import

---
 src/rpcclient/rpcclient/symbol.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/rpcclient/rpcclient/symbol.py b/src/rpcclient/rpcclient/symbol.py
index 8fdb4c1e..b88d0ee9 100644
--- a/src/rpcclient/rpcclient/symbol.py
+++ b/src/rpcclient/rpcclient/symbol.py
@@ -6,9 +6,12 @@
 from capstone import CS_ARCH_ARM64, CS_ARCH_X86, CS_MODE_64, CS_MODE_LITTLE_ENDIAN, Cs, CsInsn
 from construct import Container
 
-from rpcclient.protos.rpc_pb2 import ARCH_ARM64
 from rpcclient.structs.generic import Dl_info
 
+# make sure imports from the *_pb2 modules don't depend on the locally installed protobuf version
+os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'python'
+from rpcclient.protos.rpc_pb2 import ARCH_ARM64  # noqa: E402
+
 ADDRESS_SIZE_TO_STRUCT_FORMAT = {1: 'B', 2: 'H', 4: 'I', 8: 'Q'}
 RETVAL_BIT_COUNT = 64