Changeset 181 for etherws/trunk


Ignore:
Timestamp:
07/27/12 01:08:30 (12 years ago)
Author:
atzm
Message:
  • select client ssl handler per instance, not class
File:
1 edited

Legend:

Unmodified
Added
Removed
  • etherws/trunk/etherws.py

    r180 r181  
    349349 
    350350class EtherWebSocketClient(DebugMixIn): 
    351     def __init__(self, ioloop, switch, url, cred=None, debug=False): 
     351    def __init__(self, ioloop, switch, url, ssl_=None, cred=None, debug=False): 
    352352        self._ioloop = ioloop 
    353353        self._switch = switch 
    354354        self._url = url 
     355        self._ssl = ssl_ 
    355356        self._debug = debug 
    356357        self._sock = None 
     
    367368 
    368369    def open(self): 
     370        sslwrap = websocket._SSLSocketWrapper 
     371 
    369372        if not self.closed: 
    370373            raise websocket.WebSocketException('already opened') 
    371         self._sock = websocket.WebSocket() 
    372         self._sock.connect(self._url, **self._options) 
    373         self._switch.register_port(self) 
    374         self._ioloop.add_handler(self.fileno(), self, self._ioloop.READ) 
    375         self.dprintf('connected: %s\n', lambda: self._url) 
     374 
     375        if self._ssl: 
     376            websocket._SSLSocketWrapper = self._ssl 
     377 
     378        try: 
     379            self._sock = websocket.WebSocket() 
     380            self._sock.connect(self._url, **self._options) 
     381            self._switch.register_port(self) 
     382            self._ioloop.add_handler(self.fileno(), self, self._ioloop.READ) 
     383            self.dprintf('connected: %s\n', lambda: self._url) 
     384        finally: 
     385            websocket._SSLSocketWrapper = sslwrap 
    376386 
    377387    def close(self): 
     
    474484    ioloop = tornado.ioloop.IOLoop.instance() 
    475485    fdb = FDB(ageout=args.ageout, debug=args.debug) 
    476     switch = SwitchingHub(fdb, debug=args.debug) 
    477  
    478     harg = {'switch': switch, 'htpasswd': args.htpasswd, 'debug': args.debug} 
     486    sw = SwitchingHub(fdb, debug=args.debug) 
     487 
     488    harg = {'switch': sw, 'htpasswd': args.htpasswd, 'debug': args.debug} 
    479489    serv = (args.path, EtherWebSocketHandler, harg) 
    480490    app = tornado.web.Application([serv]) 
     
    483493 
    484494    for dev in args.device: 
    485         tap = TapHandler(ioloop, switch, dev, debug=args.debug) 
     495        tap = TapHandler(ioloop, sw, dev, debug=args.debug) 
    486496        tap.open() 
    487497 
     
    504514        args.passwd = getpass.getpass() 
    505515 
    506     websocket._SSLSocketWrapper = ssl_wrapper(args.insecure, args.cacerts) 
     516    ssl_ = ssl_wrapper(args.insecure, args.cacerts) 
    507517    cred = {'user': args.user, 'passwd': args.passwd} 
    508518    ioloop = tornado.ioloop.IOLoop.instance() 
    509519    fdb = FDB(ageout=args.ageout, debug=args.debug) 
    510     switch = SwitchingHub(fdb, debug=args.debug) 
     520    sw = SwitchingHub(fdb, debug=args.debug) 
    511521 
    512522    for uri in args.uri: 
    513         client = EtherWebSocketClient(ioloop, switch, uri, cred, args.debug) 
     523        client = EtherWebSocketClient(ioloop, sw, uri, ssl_, cred, args.debug) 
    514524        client.open() 
    515525 
    516526    for dev in args.device: 
    517         tap = TapHandler(ioloop, switch, dev, debug=args.debug) 
     527        tap = TapHandler(ioloop, sw, dev, debug=args.debug) 
    518528        tap.open() 
    519529 
Note: See TracChangeset for help on using the changeset viewer.