fix DelegatedWormhole

This commit is contained in:
Brian Warner 2017-04-18 22:49:50 -04:00
parent 323d5a896f
commit ef56336eab

View File

@ -61,8 +61,8 @@ class _DelegatedWormhole(object):
def allocate_code(self, code_length=2): def allocate_code(self, code_length=2):
self._boss.allocate_code(code_length) self._boss.allocate_code(code_length)
def input_code(self, stdio): def input_code(self):
self._boss.input_code(stdio) return self._boss.input_code()
def set_code(self, code): def set_code(self, code):
self._boss.set_code(code) self._boss.set_code(code)
@ -97,7 +97,7 @@ class _DelegatedWormhole(object):
def got_code(self, code): def got_code(self, code):
self._delegate.wormhole_code(code) self._delegate.wormhole_code(code)
def got_key(self, key): def got_key(self, key):
self._delegate.wormhole_key() self._delegate.wormhole_key(key)
self._key = key # for derive_key() self._key = key # for derive_key()
def got_verifier(self, verifier): def got_verifier(self, verifier):
self._delegate.wormhole_verified(verifier) self._delegate.wormhole_verified(verifier)