lol
This commit is contained in:
parent
4b7459ad2a
commit
26439afd1a
119
week4.py
Normal file
119
week4.py
Normal file
|
@ -0,0 +1,119 @@
|
||||||
|
# Concept: Create 8 PKs where each represent a bloodtype. Let 7 of them be created by OGen and 1 of them by KeyGen.
|
||||||
|
# The one represents our bloodtype. Bob will then encrypt 8 values using these PKs, where each value repredents
|
||||||
|
# A truth value, thus either true or false, s.t. each cipher is an entry in the bloodtype comptability matrix.
|
||||||
|
|
||||||
|
import secrets
|
||||||
|
import numpy as np
|
||||||
|
from crypto.week1 import BloodType, convert_from_string_to_enum, blood_cell_compatibility_lookup
|
||||||
|
|
||||||
|
convert_bloodtype_to_index = {
|
||||||
|
BloodType.O_NEGATIVE: 0,
|
||||||
|
BloodType.O_POSITIVE: 1,
|
||||||
|
BloodType.A_NEGATIVE: 2,
|
||||||
|
BloodType.A_POSITIVE: 3,
|
||||||
|
BloodType.B_NEGATIVE: 4,
|
||||||
|
BloodType.B_POSITIVE: 5,
|
||||||
|
BloodType.AB_NEGATIVE: 6,
|
||||||
|
BloodType.AB_POSITIVE: 7,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class Group:
|
||||||
|
def __init__(self, q):
|
||||||
|
self.order = q
|
||||||
|
|
||||||
|
def gen_group_stuff(self):
|
||||||
|
my_group = []
|
||||||
|
for i in range(1, self.order):
|
||||||
|
if np.gcd(i, self.order) == 1:
|
||||||
|
my_group.append(i)
|
||||||
|
for group_ele in my_group:
|
||||||
|
if group_ele**self.order == 1 % self.order:
|
||||||
|
gen = group_ele
|
||||||
|
return gen, self.order, my_group
|
||||||
|
|
||||||
|
class ElGamal:
|
||||||
|
def __init__(self, G, g, q):
|
||||||
|
self.group = G
|
||||||
|
self.gen_ = g
|
||||||
|
self.order = q
|
||||||
|
self.pk = None
|
||||||
|
self.sk = None
|
||||||
|
|
||||||
|
def gen(self, sk):
|
||||||
|
h = self.gen_**sk
|
||||||
|
self.sk = sk
|
||||||
|
self.pk = (self.gen_, h)
|
||||||
|
return self.pk
|
||||||
|
|
||||||
|
def enc(self, m, pk):
|
||||||
|
# sample random r \in Zq
|
||||||
|
r = secrets.SystemRandom().randint(1, self.order-1)
|
||||||
|
g, h = pk
|
||||||
|
c = (g ** r, m * h**r)
|
||||||
|
return c
|
||||||
|
|
||||||
|
def dec(self, c):
|
||||||
|
c1, c2 = c
|
||||||
|
m = c2 * c1**(-self.sk) # % self.order
|
||||||
|
return m
|
||||||
|
|
||||||
|
def ogen(self, r):
|
||||||
|
# Here, q = 2p+1, thus we actually need to use the p here, instead of
|
||||||
|
# self.order, but as we do not know p yet, .e we
|
||||||
|
# TODO: Use p instead of self.order, s.t. self.order = 2p+1
|
||||||
|
s = secrets.SystemRandom().randint(1, self.order)
|
||||||
|
h = s**2 % self.order
|
||||||
|
return self.gen_, h
|
||||||
|
|
||||||
|
|
||||||
|
class Alice:
|
||||||
|
def __init__(self, bloodtype, elgamal):
|
||||||
|
self.elgamal = elgamal
|
||||||
|
self.gen_ = 9
|
||||||
|
self.order = 453
|
||||||
|
self.b = convert_bloodtype_to_index[convert_from_string_to_enum[bloodtype]]
|
||||||
|
self.sk = secrets.SystemRandom().randint(1, self.order)
|
||||||
|
|
||||||
|
self.pk = self.elgamal.gen(self.sk)
|
||||||
|
self.fake_pks = [self.elgamal.ogen(secrets.SystemRandom().randint(0, self.order))
|
||||||
|
for _ in range(7)]
|
||||||
|
|
||||||
|
def send_pks(self):
|
||||||
|
all_pks = self.fake_pks
|
||||||
|
all_pks.insert(self.b, self.pk)
|
||||||
|
return all_pks
|
||||||
|
|
||||||
|
def retrieve(self, ciphers):
|
||||||
|
mb = self.elgamal.dec(ciphers[self.b])
|
||||||
|
return mb
|
||||||
|
|
||||||
|
|
||||||
|
class Bob:
|
||||||
|
def __init__(self, bloodtype, elgamal):
|
||||||
|
self.bloodtype = convert_from_string_to_enum[bloodtype]
|
||||||
|
self.truth_vals = []
|
||||||
|
self.elgamal = elgamal
|
||||||
|
self.pks = None
|
||||||
|
for donor in BloodType:
|
||||||
|
truth_val = blood_cell_compatibility_lookup(self.bloodtype, donor)
|
||||||
|
self.truth_vals.append(truth_val)
|
||||||
|
|
||||||
|
def receive_pks(self, pks):
|
||||||
|
self.pks = pks
|
||||||
|
|
||||||
|
def transfer_messages(self):
|
||||||
|
ciphers = []
|
||||||
|
for idx, truth_val in enumerate(self.truth_vals):
|
||||||
|
pk = self.pks[idx]
|
||||||
|
c = self.elgamal.enc(truth_val, pk)
|
||||||
|
ciphers.append(c)
|
||||||
|
return ciphers
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
group = Group(11)
|
||||||
|
gen, order, my_group = group.gen_group_stuff()
|
||||||
|
|
||||||
|
print(gen, order, my_group)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user