def __init__(self, builtin_cv, common_input=None):
Protocol.__init__(self, None)
verifier_states = { 2:self.verifier_state2, 4:self.verifier_state4, 6:self.verifier_state6 }
prover_states = { 1:self.prover_state1, 3:self.prover_state3, 5:self.prover_state5 }
verifier_trans = { 2:4, 4:[2,6] }
prover_trans = { 1:3, 3:5, 5:1 }
# describe the parties involved and the valid transitions
Protocol.addPartyType(self, VERIFIER, verifier_states, verifier_trans)
Protocol.addPartyType(self, PROVER, prover_states, prover_trans, True)
self.group = ECGroup(builtin_cv)
#db = {}
Protocol.setSubclassVars(self, self.group) #, db)
class SchnorrZK(Protocol):
def __init__(self, builtin_cv, common_input=None):
Protocol.__init__(self, None)
verifier_states = { 2:self.verifier_state2, 4:self.verifier_state4, 6:self.verifier_state6 }
prover_states = { 1:self.prover_state1, 3:self.prover_state3, 5:self.prover_state5 }
verifier_trans = { 2:4, 4:[2,6] }
prover_trans = { 1:3, 3:5, 5:1 }
# describe the parties involved and the valid transitions
Protocol.addPartyType(self, VERIFIER, verifier_states, verifier_trans)
Protocol.addPartyType(self, PROVER, prover_states, prover_trans, True)
self.group = ECGroup(builtin_cv)
#db = {}
Protocol.setSubclassVars(self, self.group) #, db)
# PROVER states
def prover_state1(self):
x = self.group.random()
r, g = self.group.random(), self.group.random(G)
t = g ** r
print('prover: ',"hello to verifier.")
Protocol.store(self, ('r',r), ('x',x))
Protocol.setState(self, 3)
return {'t':t, 'g':g, 'y':g ** x } # output goes to the next state.
def prover_state3( self, input):
print("state3 => ", input)
(r, x, c) = Protocol.get(self, ['r', 'x', 'c'])
s = r + c * x
Protocol.setState(self, 5)
return {'s':s}
def prover_state5( self, input ):
print("state5 => ", input)
result = input.split(':')[1]
if result == 'ACCEPTED': Protocol.setState(self, None)
else: Protocol.setState(self, 1); return 'REPEAT'
return None
# VERIFIER states
def verifier_state2(self, input):
#print("state2 received => ", input)
# compute challenge c and send to prover
c = self.group.random()
print("state2 generate c :=", c)
Protocol.store(self, ('c',c))
Protocol.setState(self, 4)
return {'c':c}
def verifier_state4( self, input ):
(t,g,y,c,s) = Protocol.get(self, ['t','g','y','c','s'])
print("state4: s :=", s)
if (g ** s == t * (y ** c)):
print("SUCCESSFUL VERIFICATION!!!")
output = "verifier : ACCEPTED!"
else:
print("FAILED TO VERIFY!!!")
output = "verifier : FAILED!"
Protocol.setState(self, 6)
return output
def verifier_state6(self, input ):
print("state6: => ", input)
Protocol.setState(self, None)
return None
