Determined (crypto)

"It is my experience that proofs involving matrices can be shortened by 50% if one throws the matrices out." Emil Artin

2KB

server.py

211B

gen.py

We are given the main code running on the server in server.py, and gen.py which generates an encrypted message using RSA. We are given the RSA modulus n, exponent e and ciphertext c.

# server.py
from Crypto.Util.number import bytes_to_long, long_to_bytes
from itertools import permutations
from SECRET import FLAG, p, q, r

def inputs():
    print("[DET] First things first, gimme some numbers:")
    M = [
        [0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0],
    ]
    try:
        M[0][0] = p
        M[0][2] = int(input("[DET] M[0][2] = "))
        M[0][4] = int(input("[DET] M[0][4] = "))

        M[1][1] = int(input("[DET] M[1][1] = "))
        M[1][3] = int(input("[DET] M[1][3] = "))

        M[2][0] = int(input("[DET] M[2][0] = "))
        M[2][2] = int(input("[DET] M[2][2] = "))
        M[2][4] = int(input("[DET] M[2][4] = "))

        M[3][1] = q
        M[3][3] = r

        M[4][0] = int(input("[DET] M[4][0] = "))
        M[4][2] = int(input("[DET] M[4][2] = "))
        """
    M = [
        [p, 0, x, 0, x],
        [0, x, 0, x, 0],
        [x, 0, x, 0, x],
        [0, q, 0, r, 0],
        [x, 0, x, 0, 0],
    ]
        """
    except:
        return None
    return M

def handler(signum, frame):
    raise Exception("[DET] You're trying too hard, try something simpler")

def fun(M):
    def sign(sigma):
        l = 0
        for i in range(5):
            for j in range(i + 1, 5):
                if sigma[i] > sigma[j]:
                    l += 1
        return (-1)**l

    res = 0
    for sigma in permutations([0,1,2,3,4]):
        curr = 1
        for i in range(5):
            curr *= M[sigma[i]][i]
        res += sign(sigma) * curr
    return res

def main():
    print("[DET] Welcome")
    M = inputs()
    if M is None:
        print("[DET] You tried something weird...")
        return

    res = fun(M)
    print(f"[DET] Have fun: {res}")

if __name__ == "__main__":
    main()

# gen.py
from SECRET import FLAG, p, q, r
from Crypto.Util.number import bytes_to_long
n = p * q
e = 65535
m = bytes_to_long(FLAG)
c = pow(m, e, n)
# printed to gen.txt
print(f"{n = }")
print(f"{e = }")
print(f"{c = }")

server.py imports the RSA primes p and q , along with r (but r is useless). main() first calls inputs(), which prompts us for entries to a matrix M. After inputting our values, M looks like this:

$$M=\begin{bmatrix} p & 0 & x_0 & 0 & x_1\\ 0 & x_2 & 0 & x_3 & 0\\ x_4 & 0 & x_5 & 0 & x_6 \\ 0 & q & 0 & r & 0 \\ x_7 & 0 & x_8 & 0 & 0 \end{bmatrix}$$

where all the $x$ values are our controlled inputs.

fun(M) is then printed. fun(M) is essentially using a permutation method to compute the determinant of a matrix. Let's compute the determinant of M using cofactor expansion:

Along the fourth row:

$$\begin{vmatrix} p & 0 & x_0 & 0 & x_1\\ 0 & x_2 & 0 & x_3 & 0\\ x_4 & 0 & x_5 & 0 & x_6 \\ 0 & q & 0 & r & 0 \\ x_7 & 0 & x_8 & 0 & 0 \end{vmatrix}=q\begin{vmatrix} p & x_0 & 0 & x_1\\ 0 & 0 & x_3 & 0 \\ x_4 & x_5 & 0 & x_6\\ x_7 & x_8 & 0 & 0 \end{vmatrix} + r\begin{vmatrix} p & 0 & x_0 & x_1\\ 0 & x_2 & 0 & 0 \\ x_4 & 0 & x_5 & x_6 \\ x_7 & 0 & x_8 & 0 \end{vmatrix}$$

Along second row for both:

$$=-qx_3\begin{vmatrix} p & x_0 & x_1\\ x_4 & x_5 & x_6\\ x_7 & x_8 & 0 \end{vmatrix} + rx_2\begin{vmatrix} p & x_0 & x_1\\ x_4 & x_5 & x_6\\ x_7 & x_8 & 0 \end{vmatrix}$$

$$=(rx_2-qx_3)(x_0x_6x_7+x_1x_4x_8-x_1x_5x_7-px_6x_8)$$

Observing this expression closely, by controlling the different $x$ values, we can obtain $q$ alone. We can do this by setting $x_0=x_2=x_4=x_6=x_8=0$ and $x_1=x_3=x_5=x_7=1$.

Providing these values to the server, we get that q = 12538849920148192679761652092105069246209474199128389797086660026385076472391166777813291228233723977872612370392782047693930516418386543325438897742835129. With q, we know n hence we can find p. We can then find phi and hence d. We can then find the plaintext m.

from Crypto.Util.number import long_to_bytes

n = 158794636700752922781275926476194117856757725604680390949164778150869764326023702391967976086363365534718230514141547968577753309521188288428236024251993839560087229636799779157903650823700424848036276986652311165197569877428810358366358203174595667453056843209344115949077094799081260298678936223331932826351
"""
    M = [
        [p, 0, x, 0, x],
        [0, x, 0, x, 0],
        [x, 0, x, 0, x],
        [0, q, 0, r, 0],
        [x, 0, x, 0, 0],
    ]
"""
q = 12538849920148192679761652092105069246209474199128389797086660026385076472391166777813291228233723977872612370392782047693930516418386543325438897742835129

assert n % q == 0
p = n // q
tot = (p - 1) * (q - 1)
e = 65535
c = 72186625991702159773441286864850566837138114624570350089877959520356759693054091827950124758916323653021925443200239303328819702117245200182521971965172749321771266746783797202515535351816124885833031875091162736190721470393029924557370228547165074694258453101355875242872797209141366404264775972151904835111
d = pow(e, -1, tot)
m = pow(c, d, n)
print(long_to_bytes(m)) # uiuctf{h4rd_w0rk_&&_d3t3rm1n4t10n}