From f35fb4209684f4a2502284ad2f9e831209f27196 Mon Sep 17 00:00:00 2001 From: Seth Schoen Date: Thu, 31 Oct 2013 12:37:44 -0700 Subject: [PATCH] in demo, only do POP challenge if the challenge type is POP challenge --- popchallenge_demo.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/popchallenge_demo.py b/popchallenge_demo.py index b8bab9965..14ac53454 100644 --- a/popchallenge_demo.py +++ b/popchallenge_demo.py @@ -12,13 +12,15 @@ skid, snonce = make_challenge(open("/tmp/pub.pem")) m.challenge.add(name="what-is-the-frequency", type=m.ProofOfPossession, data=(skid, snonce)) for challenge in m.challenge: - pcr = POPChallengeResponder(*challenge.data) - pcr.find_priv(["/tmp/pub.pem", "/tmp/decoy.pem", "/tmp/falkdjaslkdj", "/tmp/priv.pem"]) - assert pcr.privkey - cnonce, sig = pcr.respond_challenge() - r.completedchallenge.add(name=challenge.name, type=r.ProofOfPossession, data=(cnonce,sig)) + if challenge.type == m.ProofOfPossession: + pcr = POPChallengeResponder(*challenge.data) + pcr.find_priv(["/tmp/pub.pem", "/tmp/decoy.pem", "/tmp/falkdjaslkdj", "/tmp/priv.pem"]) + assert pcr.privkey + cnonce, sig = pcr.respond_challenge() + r.completedchallenge.add(name=challenge.name, type=r.ProofOfPossession, data=(cnonce,sig)) for completedchallenge in r.completedchallenge: # If there's actually more than one then we'd need to store and use # multiple different values of snonce. - print verify_challenge_response(open("/tmp/pub.pem").read(), snonce, *completedchallenge.data) + if completedchallenge.type == r.ProofOfPossession: + print verify_challenge_response(open("/tmp/pub.pem").read(), snonce, *completedchallenge.data)