From 2bcb72d82c0f85714af13324b5ca00f2c4fefc40 Mon Sep 17 00:00:00 2001
From: "Casper V. Kristensen" <c@spervk.com>
Date: Tue, 8 Oct 2019 13:07:59 +0200
Subject: [PATCH] Week7 done.

---
 __main__.py |   2 +-
 week7.py    | 118 ++++++++++++++++++++++++++++------------------------
 2 files changed, 64 insertions(+), 56 deletions(-)

diff --git a/__main__.py b/__main__.py
index 29d1f7a..4e1ca82 100644
--- a/__main__.py
+++ b/__main__.py
@@ -1,3 +1,3 @@
-from .week6 import main
+from .week7 import main
 
 main()
diff --git a/week7.py b/week7.py
index c8551e1..1a418d8 100644
--- a/week7.py
+++ b/week7.py
@@ -1,75 +1,83 @@
 from secrets import SystemRandom
 
+from .week1 import BloodType, blood_cell_compatibility_lookup
+
 random = SystemRandom()
 
-'''
-def find_big_odd(k):
-	
-	num = random.getrandbits(k)
-	while num % 2 == 0:
-		num = 
-'''
-
 
 def enc(m, pk):
-	S = 1000
-	c = m + sum(random.sample(pk, S))
-	return c
+    S = 1000
+    c = m + sum(random.sample(pk, S))
+    return c
+
+
+def dec(c, p):
+    m = (c % p) % 2
+    return m
 
 
 class Alice:
-	def __init__(self, ya, yb, yr):
-		self.ya = ya
-		self.yb = yb
-		self.yr = yr
-		# p is the secret key
-		self.p = random.getrandbits(2000)*2 + 1 # odd integer
-		n = 2000
-		bigInts = [random.getrandbits(10**7) for _ in range(n)]
-		smallInts = [random.getrandbits(60) for _ in range(n)]
+    def __init__(self, ra, rb, rs):
+        self.ra = ra
+        self.rb = rb
+        self.rs = rs
 
-		self.pk = [self.p*bigInts[i] + 2*smallInts[i] for i in range(n)]
-		self.cya = enc(ya, self.pk)
-		self.cyb = enc(yb, self.pk)
-		self.cyr = enc(yr, self.pk)
+        self.p = random.getrandbits(2000)
+        if self.p % 2 == 0:
+            self.p -= 1
 
-	def send(self):
-		return self.cya, self.cyb, self.cyr, self.pk
+        n = 2000
+        big_ints = [random.getrandbits(10**7) for _ in range(n)]
+        small_ints = [random.getrandbits(60) for _ in range(n)]
 
-	def dec(self, c):
-		m = (c % self.p) % 2
-		return m
+        self.pk = [self.p * big_ints[i] + 2 * small_ints[i] for i in range(n)]
+
+    def send(self):
+        cra = enc(self.ra, self.pk)
+        crb = enc(self.rb, self.pk)
+        crs = enc(self.rs, self.pk)
+        return self.pk, cra, crb, crs
+
+    def receive(self, result_cipher):
+        return dec(result_cipher, self.p)
 
 
 class Bob:
-	def __init__(self, xa, xb, xr):
-		self.xa = xa
-		self.xb = xb
-		self.xr = xr
+    def __init__(self, da, db, ds):
+        self.da = da
+        self.db = db
+        self.ds = ds
+        self.result_cipher = None
 
-	def receive(self, cya, cyb, cyr, pk):
-		self.pk = pk
+    def receive(self, pk, cra, crb, crs):
+        cda = enc(self.da, pk)
+        cdb = enc(self.db, pk)
+        cds = enc(self.ds, pk)
+        self.result_cipher = (1 + (cda * (1 + cra))) * (1 + (cdb * (1 + crb))) * (1 + (cds * (1 + crs)))
 
-		# these are ciphered values from alice
-		self.cya = cya
-		self.cyb = cyb
-		self.cyr = cyr
-
-		# these are ciphered values from bob
-		self.cxa = enc(self.xa, pk)
-		self.cxb = enc(self.xb, pk)
-		self.cxr = enc(self.xr, pk)
-
-	def compute_bloodcompatiblity(self):
+    def send(self):
+        return self.result_cipher
 
 
-		#f = (1 ^ (self.xa & (1 ^ self.cya))) & (1 ^ (self.xb & (1 ^ self.cyb))) & (1 ^ (self.xr & (1 ^ self.cyr)))
-		#z = enc(f, self.pk)
-		#return z
-		pass
+def run(da, db, ds, ra, rb, rs):
+    alice = Alice(ra, rb, rs)
+    bob = Bob(da, db, ds)
+    bob.receive(*alice.send())
+    z = alice.receive(bob.send())
+    return z
 
-def protocol(donor, recipient):
-	# Alice is recipient
-	alice = Alice(0,1,1)
-	bob = Bob(1,0,0)
-	pass
\ No newline at end of file
+
+def main():
+    green = 0
+    red = 0
+    for i, recipient in enumerate(BloodType):
+        for j, donor in enumerate(BloodType):
+            z = run(*donor.value, *recipient.value)
+            lookup = blood_cell_compatibility_lookup(recipient, donor)
+            if lookup == z:
+                green += 1
+            else:
+                print(f"'{BloodType(donor).name} -> {BloodType(recipient).name}' should be {lookup}.")
+                red += 1
+    print("Green:", green)
+    print("Red  :", red)