X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Fclient%2Fsfaserverproxy.py;h=457a1d252072b6aefd9cd66d33317d88b356a051;hb=04acd3228e5911d36d0cd58dc35b9319fc558e17;hp=18b77131a956bc0a4aead85ad36fc2f5d906172e;hpb=935de64c2ed3a68566da471c68447a1ac384455d;p=sfa.git diff --git a/sfa/client/sfaserverproxy.py b/sfa/client/sfaserverproxy.py index 18b77131..457a1d25 100644 --- a/sfa/client/sfaserverproxy.py +++ b/sfa/client/sfaserverproxy.py @@ -1,13 +1,20 @@ # XMLRPC-specific code for SFA Client -import xmlrpclib -from httplib import HTTPS, HTTPSConnection +# starting with 2.7.9 we need to turn off server verification +import ssl +try: + turn_off_server_verify = {'context': ssl._create_unverified_context()} +except: + turn_off_server_verify = {} + +import xmlrpc.client +import http.client try: from sfa.util.sfalogging import logger except: import logging - logger=logging.getLogger('sfaserverproxy') + logger = logging.getLogger('sfaserverproxy') ## # ServerException, ExceptionUnmarshaller @@ -15,14 +22,17 @@ except: # Used to convert server exception strings back to an exception. # from usenet, Raghuram Devarakonda + class ServerException(Exception): pass -class ExceptionUnmarshaller(xmlrpclib.Unmarshaller): + +class ExceptionUnmarshaller(xmlrpc.client.Unmarshaller): + def close(self): try: - return xmlrpclib.Unmarshaller.close(self) - except xmlrpclib.Fault, e: + return xmlrpc.client.Unmarshaller.close(self) + except xmlrpc.client.Fault as e: raise ServerException(e.faultString) ## @@ -30,41 +40,33 @@ class ExceptionUnmarshaller(xmlrpclib.Unmarshaller): # # A transport for XMLRPC that works on top of HTTPS -# python 2.7 xmlrpclib has changed its internal code -# it now calls 'getresponse' on the obj returned by make_connection -# while it used to call 'getreply' -# regardless of the version, httplib.HTTPS does implement getreply, -# while httplib.HTTPSConnection has getresponse -# so we create a dummy instance to check what's expected -need_HTTPSConnection=hasattr(xmlrpclib.Transport().make_connection('localhost'),'getresponse') +# targetting only python-2.7 we can get rid of some older code + + +class XMLRPCTransport(xmlrpc.client.Transport): -class XMLRPCTransport(xmlrpclib.Transport): - def __init__(self, key_file=None, cert_file=None, timeout=None): - xmlrpclib.Transport.__init__(self) - self.timeout=timeout + xmlrpc.client.Transport.__init__(self) + self.timeout = timeout self.key_file = key_file self.cert_file = cert_file - + def make_connection(self, host): # create a HTTPS connection object from a host descriptor # host may be a string, or a (host, x509-dict) tuple host, extra_headers, x509 = self.get_host_info(host) - if need_HTTPSConnection: - #conn = HTTPSConnection(host, None, key_file=self.key_file, cert_file=self.cert_file, timeout=self.timeout) #**(x509 or {})) - conn = HTTPSConnection(host, None, key_file=self.key_file, cert_file=self.cert_file) #**(x509 or {})) - else: - #conn = HTTPS(host, None, key_file=self.key_file, cert_file=self.cert_file, timeout=self.timeout) #**(x509 or {})) - conn = HTTPS(host, None, key_file=self.key_file, cert_file=self.cert_file) #**(x509 or {})) - - if hasattr(conn, 'set_timeout'): - conn.set_timeout(self.timeout) + conn = http.client.HTTPSConnection(host, None, key_file=self.key_file, + cert_file=self.cert_file, + **turn_off_server_verify) # Some logic to deal with timeouts. It appears that some (or all) versions # of python don't set the timeout after the socket is created. We'll do it # ourselves by forcing the connection to connect, finding the socket, and # calling settimeout() on it. (tested with python 2.6) if self.timeout: + if hasattr(conn, 'set_timeout'): + conn.set_timeout(self.timeout) + if hasattr(conn, "_conn"): # HTTPS is a wrapper around HTTPSConnection real_conn = conn._conn @@ -78,33 +80,41 @@ class XMLRPCTransport(xmlrpclib.Transport): def getparser(self): unmarshaller = ExceptionUnmarshaller() - parser = xmlrpclib.ExpatParser(unmarshaller) + parser = xmlrpc.client.ExpatParser(unmarshaller) return parser, unmarshaller -class XMLRPCServerProxy(xmlrpclib.ServerProxy): + +class XMLRPCServerProxy(xmlrpc.client.ServerProxy): + def __init__(self, url, transport, allow_none=True, verbose=False): # remember url for GetVersion - self.url=url - xmlrpclib.ServerProxy.__init__(self, url, transport, allow_none=allow_none, verbose=verbose) + # xxx not sure this is still needed as SfaServerProxy has this too + self.url = url + xmlrpc.client.ServerProxy.__init__(self, url, transport, allow_none=allow_none, + verbose=verbose, + **turn_off_server_verify) def __getattr__(self, attr): - logger.debug ("xml-rpc %s method:%s"%(self.url,attr)) - return xmlrpclib.ServerProxy.__getattr__(self, attr) + logger.debug("xml-rpc %s method:%s" % (self.url, attr)) + return xmlrpc.client.ServerProxy.__getattr__(self, attr) + +# the object on which we can send methods that get sent over xmlrpc + -########## the object on which we can send methods that get sent over xmlrpc class SfaServerProxy: - def __init__ (self, url, keyfile, certfile, verbose=False, timeout=None): - self.url=url - self.keyfile=keyfile - self.certfile=certfile - self.verbose=verbose - self.timeout=timeout - # an instance of xmlrpclib.ServerProxy + def __init__(self, url, keyfile, certfile, verbose=False, timeout=None): + self.url = url + self.keyfile = keyfile + self.certfile = certfile + self.verbose = verbose + self.timeout = timeout + # an instance of xmlrpc.client.ServerProxy transport = XMLRPCTransport(keyfile, certfile, timeout) - self.serverproxy = XMLRPCServerProxy(url, transport, allow_none=True, verbose=verbose) + self.serverproxy = XMLRPCServerProxy( + url, transport, allow_none=True, verbose=verbose) - # this is python magic to return the code to run when + # this is python magic to return the code to run when # SfaServerProxy receives a method call # so essentially we send the same method with identical arguments # to the server_proxy object @@ -112,4 +122,3 @@ class SfaServerProxy: def func(*args, **kwds): return getattr(self.serverproxy, name)(*args, **kwds) return func -