safexmlrpc.py

   1 """Leverage curl to make XMLRPC requests that check the server's credentials."""
   2
   3 import curlwrapper
   4 import xmlrpclib
   5
   6
   7 class CertificateCheckingSafeTransport(xmlrpclib.Transport):
   8     def __init__(self, cacert, timeout):
   9         self.cacert = cacert
  10         self.timeout = timeout
  11
  12     def request(self, host, handler, request_body, verbose=0):
  13         self.verbose = verbose
  14         try:
  15             contents = curlwrapper.retrieve('https://%s%s' % (host, handler),
  16                                             cacert = self.cacert,
  17                                             postdata = request_body,
  18                                             timeout = self.timeout)
  19             return xmlrpclib.loads(contents)[0]
  20         except curlwrapper.CurlException, e:
  21             raise xmlrpclib.ProtocolError(host + handler, -1, str(e), '')
  22
  23 class ServerProxy(xmlrpclib.ServerProxy):
  24     def __init__(self, uri, cacert, timeout = 300, **kwds):
  25         xmlrpclib.ServerProxy.__init__(self, uri,
  26                                        CertificateCheckingSafeTransport(cacert, timeout),
  27                                        **kwds)