X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=safexmlrpc.py;h=71bf4d62d478d17516d177f3952328cb785c4abd;hb=1926e74e4865032256cf5cebfe977ac9c3dd149a;hp=96865dfd8518d08132fec505ad22ded603319c87;hpb=b0deb4f91ea673181a2db5ee432f8d9c36fc7312;p=nodemanager.git

diff --git a/safexmlrpc.py b/safexmlrpc.py
index 96865df..71bf4d6 100644
--- a/safexmlrpc.py
+++ b/safexmlrpc.py
@@ -1,24 +1,29 @@
 """Leverage curl to make XMLRPC requests that check the server's credentials."""
 
-from subprocess import PIPE, Popen
-import xmlrpclib
+import xmlrpc.client
 
+import curlwrapper
 
-CURL = '/usr/bin/curl'
 
-class CertificateCheckingSafeTransport(xmlrpclib.Transport):
+class CertificateCheckingSafeTransport (xmlrpc.client.Transport):
+
+    def __init__(self, cacert, timeout):
+        self.cacert = cacert
+        self.timeout = timeout
+
     def request(self, host, handler, request_body, verbose=0):
         self.verbose = verbose
-        p = Popen((CURL, '--cacert', '/usr/boot/cacert.pem', '--data', '@-', 'https://%s%s' % (host, handler)), stdin=PIPE, stdout=PIPE, stderr=PIPE)
-        p.stdin.write(request_body)
-        p.stdin.close()
-        contents = p.stdout.read()
-        p.stdout.close()
-        error = p.stderr.read()
-        p.stderr.close()
-        rc = p.wait()
-        if rc != 0: raise xmlrpclib.ProtocolError(host + handler, rc, error, '')
-        return xmlrpclib.loads(contents)[0]
-
-class ServerProxy(xmlrpclib.ServerProxy):
-    def __init__(self, handler, *args, **kw_args): xmlrpclib.ServerProxy.__init__(self, handler, CertificateCheckingSafeTransport())
+        url='https://%s%s' % (host, handler)
+        # this might raise an xmlrpclib.Protocolerror exception
+        contents = curlwrapper.retrieve(url,
+                                        cacert = self.cacert,
+                                        postdata = request_body,
+                                        timeout = self.timeout)
+        return xmlrpc.client.loads(contents)[0]
+
+class ServerProxy(xmlrpc.client.ServerProxy):
+
+    def __init__(self, uri, cacert, timeout = 300, **kwds):
+        xmlrpc.client.ServerProxy.__init__(self, uri,
+                                       CertificateCheckingSafeTransport(cacert, timeout),
+                                       **kwds)