From bb46d39030b25543b06a0a9c455e2dcf12626a2a Mon Sep 17 00:00:00 2001
From: Narayan Desai <desai@mcs.anl.gov>
Date: Tue, 20 Mar 2007 16:14:08 +0000
Subject: Set tlslite xmlrpc timeout up to 90 sec

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@2950 ce84e21b-d406-0410-9b95-82705330c041
---
 src/lib/Client/Proxy.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/lib/Client/Proxy.py b/src/lib/Client/Proxy.py
index 6140d1f90..2e6d50ae8 100644
--- a/src/lib/Client/Proxy.py
+++ b/src/lib/Client/Proxy.py
@@ -1,12 +1,46 @@
 '''Cobalt proxy provides client access to cobalt components'''
 __revision__ = '$Revision$'
 
-import logging, socket, time, xmlrpclib, ConfigParser
+import logging, socket, time, xmlrpclib, ConfigParser, httplib
 from Bcfg2.tlslite.integration.XMLRPCTransport import XMLRPCTransport
+from Bcfg2.tlslite.integration.HTTPTLSConnection import HTTPTLSConnection
+from Bcfg2.tlslite.TLSConnection import TLSConnection
 import Bcfg2.tlslite.errors
 
 #FIXME need to reimplement _binadaddress support for XMLRPCTransport
 
+class MyHTTPTLSConnection(HTTPTLSConnection):
+    def connect(self):
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        if hasattr(sock, 'settimeout'):
+            sock.settimeout(90)
+        sock.connect((self.host, self.port))
+
+        #Use a TLSConnection to emulate a socket
+        self.sock = TLSConnection(sock)
+
+        #When httplib closes this, close the socket
+        self.sock.closeSocket = True
+        self._handshake(self.sock)
+
+class MyXMLRPCTransport(XMLRPCTransport):
+    def make_connection(self, host):
+        # create a HTTPS connection object from a host descriptor
+        host, extra_headers, x509 = self.get_host_info(host)
+        http = MyHTTPTLSConnection(host, None,
+                                   self.username, self.password,
+                                   self.sharedKey,
+                                   self.certChain, self.privateKey,
+                                   self.checker.cryptoID,
+                                   self.checker.protocol,
+                                   self.checker.x509Fingerprint,
+                                   self.checker.x509TrustList,
+                                   self.checker.x509CommonName,
+                                   self.settings)
+        http2 = httplib.HTTP()
+        http2._setup(http)
+        return http2
+
 class CobaltComponentError(Exception):
     pass
 
-- 
cgit v1.2.3-1-g7c22