Trap gRPC errors

GaloisInc · May 7, 2024 · dfd59cb · dfd59cb
1 parent 84fa27f
commit dfd59cb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pate_binja/mcad/PateMcad.py b/pate_binja/mcad/PateMcad.py
@@ -134,7 +134,11 @@ def request_cycle_counts(self, instructions: list[bytes]) -> List[CycleCount]:
         if not self.isRunning():
             return []
         pbInstructions = map(lambda b: binja_pb2.BinjaInstructions.Instruction(opcode=b), instructions)
-        pbCycleCounts = self.stub.RequestCycleCounts(binja_pb2.BinjaInstructions(instruction=pbInstructions))
+        try:
+            pbCycleCounts = self.stub.RequestCycleCounts(binja_pb2.BinjaInstructions(instruction=pbInstructions))
+        except grpc.RpcError as rpc_error:
+            print("gRPC Call failure: %s", rpc_error)
+            return []
         return list(map(lambda cc: CycleCount(cc.ready, cc.executed, cc.is_under_pressure), pbCycleCounts.cycle_count))