X-Git-Url: https://review.openocd.org/gitweb?a=blobdiff_plain;f=contrib%2Frpc_examples%2Focd_rpc_example.py;h=9d17e769571c530055b5b245a4dcf530714df07d;hb=223c28f9b9f685c98fb6882e1657b6b53379a1a6;hp=4b1516ad900599ee9d65e3422f728ea4e452d6ab;hpb=6568d29cc1d0d94daafec5bdb73de7d4f17da257;p=openocd.git

diff --git a/contrib/rpc_examples/ocd_rpc_example.py b/contrib/rpc_examples/ocd_rpc_example.py
index 4b1516ad90..9d17e76957 100755
--- a/contrib/rpc_examples/ocd_rpc_example.py
+++ b/contrib/rpc_examples/ocd_rpc_example.py
@@ -49,10 +49,16 @@ class OpenOcd:
         self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
     def __enter__(self):
-        self.sock.connect((self.tclRpcIp, self.tclRpcPort))
+        self.connect()
         return self
 
     def __exit__(self, type, value, traceback):
+        self.disconnect()
+
+    def connect(self):
+        self.sock.connect((self.tclRpcIp, self.tclRpcPort))
+
+    def disconnect(self):
         try:
             self.send("exit")
         finally: