def test4_extractPublicKey(self):
# Create UniformDH authentication message.
sharedSecret = "A" * const.SHARED_SECRET_LENGTH
realEpoch = util.getEpoch
# Try three valid and one invalid epoch value.
for epoch in util.expandedEpoch() + ["000000"]:
udh = uniformdh.new(sharedSecret, True)
util.getEpoch = lambda: epoch
authMsg = udh.createHandshake()
util.getEpoch = realEpoch
buf = obfs_buf.Buffer()
buf.write(authMsg)
if epoch == "000000":
self.assertFalse(udh.extractPublicKey(buf))
else:
self.assertTrue(udh.extractPublicKey(buf))
def test4_extractPublicKey(self):
# Create UniformDH authentication message.
sharedSecret = "A" * const.SHARED_SECRET_LENGTH
realEpoch = util.getEpoch
# Try three valid and one invalid epoch value.
for epoch in util.expandedEpoch() + ["000000"]:
udh = uniformdh.new(sharedSecret, True)
util.getEpoch = lambda: epoch
authMsg = udh.createHandshake()
util.getEpoch = realEpoch
buf = obfs_buf.Buffer()
buf.write(authMsg)
if epoch == "000000":
self.assertFalse(udh.extractPublicKey(buf))
else:
self.assertTrue(udh.extractPublicKey(buf))
def setUp(self):
weAreServer = True
self.udh = uniformdh.new("A" * const.SHARED_SECRET_LENGTH, weAreServer)
def setUp(self):
weAreServer = True
self.udh = uniformdh.new("A" * const.SHARED_SECRET_LENGTH, weAreServer)