docker compose bugfix

services/phew/src/Pailier.py

@@ -0,0 +1,47 @@
+from Crypto.Util.number import *
+from math import lcm
+import random
+
+class pailier:
+    def __init__(self):
+        self.primes = [getPrime(512) for _ in range(2)]
+        self.n = 1
+        self.phi = 1
+        self.mul = 1
+        for i in range(2):
+            self.n  *= self.primes[i]
+            self.phi *= (self.primes[i] - 1)  
+        self.n2 = self.n * self.n
+        self.g = [pow(random.randrange(1, self.n2), self.primes[i], self.n2) for i in range(2)]
+        for x in self.g:
+            self.mul = (self.mul * x) % self.n2
+        self.miu = inverse(self.L(pow(self.mul, self.phi, self.n2)), self.n)
+        self.alpha = [None, None]
+        for idx in range(2):
+            while True:
+                a = random.randrange(2, self.n - 1)
+                if GCD(a, self.n) == 1:
+                    self.alpha[idx] = a
+                    break
+        self.beta  = [random.randrange(0, self.n), random.randrange(0, self.n)]
+
+    def L(self, val):
+        return (val - 1) // self.n
+
+    def pubkey(self):
+        return (self.n, self.g)
+
+    def encrypt(self, msg: int) -> int:
+        r  = random.randrange(0, self.n - 1)
+        gb = self.g[random.randrange(0, 2)]
+        gm = pow(gb, msg, self.n2)
+        rn = pow(r, self.n, self.n2)
+        return (gm * rn) % self.n2
+
+    def decrypt(self, ct: int) -> int:
+        raw = self.L(pow(ct, self.phi, self.n2)) % self.n
+        raw = (raw * self.miu) % self.n
+        t = pow(ct % self.n, (self.n - 1) // 2, self.n) if (self.n % 2 == 1) else 0
+        idx = 0 if t == 1 else 1
+        return (self.alpha[idx] * raw + self.beta[idx]) % self.n
+

services/phew/src/chall.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+from Pailier import *
+from Crypto.Util.number import *
+with open("/flag.txt", "rb") as f:
+    flag = f.read()
+flag = bytes_to_long(flag)
+cipher = pailier()
+while True:
+    print("1. encrypt")
+    print("2. bingo")
+    print("3. decrypt")
+    print("4. exit")
+    inp = int(input("> "))
+    if inp==1:
+        print("pt (hex)")
+        inp = input("> ")
+        ct = cipher.encrypt(int(inp,16))
+        print('ct : ','{0:x}'.format(ct))
+    elif inp==2:
+        ct = cipher.encrypt(flag)
+        print('ct : ','{0:x}'.format(ct))
+    elif inp==3:
+        print("ct (hex)")
+        inp = input("> ")
+        pt = cipher.decrypt(int(inp,16))
+        print('pt : ','{0:x}'.format(pt))
+    else:
+        exit()

services/phew/src/run.sh

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+exec socat tcp-l:8000,reuseaddr,fork exec:"python3 ./chall.py"