X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=safexmlrpc.py;fp=safexmlrpc.py;h=96865dfd8518d08132fec505ad22ded603319c87;hb=b0deb4f91ea673181a2db5ee432f8d9c36fc7312;hp=0000000000000000000000000000000000000000;hpb=2bdabc3d01aac59e8000093482f369f81990a924;p=nodemanager.git

diff --git a/safexmlrpc.py b/safexmlrpc.py
new file mode 100644
index 0000000..96865df
--- /dev/null
+++ b/safexmlrpc.py
@@ -0,0 +1,24 @@
+"""Leverage curl to make XMLRPC requests that check the server's credentials."""
+
+from subprocess import PIPE, Popen
+import xmlrpclib
+
+
+CURL = '/usr/bin/curl'
+
+class CertificateCheckingSafeTransport(xmlrpclib.Transport):
+    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())