125 lines
5.4 KiB
Python
125 lines
5.4 KiB
Python
from .Challenge import Challenge
|
|
|
|
from fastecdsa.curve import Curve
|
|
from fastecdsa.point import Point
|
|
import requests
|
|
import time
|
|
import os
|
|
import json
|
|
|
|
|
|
class Burvesigner(Challenge):
|
|
flag_location = 'flags/burvesigner.txt'
|
|
history_location = 'history/burvesigner.txt'
|
|
priv_location = 'files/burvesigner.priv'
|
|
|
|
def distribute(self, flag):
|
|
try:
|
|
with open(self.flag_location, 'w') as f:
|
|
f.write(flag)
|
|
|
|
with open(self.priv_location, 'wb') as f:
|
|
f.write(os.urandom(256))
|
|
|
|
with open(self.history_location, 'a') as f:
|
|
f.write(flag + '\n')
|
|
|
|
self.logger.info(f'Flag {flag} written to {self.flag_location}')
|
|
return True
|
|
|
|
except Exception as e:
|
|
self.logger.error(f'Could not write flag to {self.flag_location}: {e}')
|
|
return False
|
|
|
|
def check(self):
|
|
try:
|
|
url = f'http://localhost:{self.port}'
|
|
flag = open(self.flag_location).read()
|
|
|
|
# C1: login guest success
|
|
guest_data = {
|
|
'username': 'guest',
|
|
'password': 'guest',
|
|
}
|
|
response = requests.post(url, data=guest_data, timeout=5)
|
|
guest_token = response.cookies["token"]
|
|
assert "Welcome, guest!" in response.text, "Guest cannot login"
|
|
assert "flashes" not in response.text, "Guest cannot login"
|
|
self.logger.info(f'C1 success for burvesigner')
|
|
|
|
# C2: login admin success
|
|
admin_data = {
|
|
'username': 'merricx_number_1_fans',
|
|
'password': getattr(self.settings, f'PASSWORD_{self.port}'),
|
|
}
|
|
response = requests.post(url, data=admin_data, timeout=5)
|
|
admin_token = response.cookies["token"]
|
|
assert "Welcome, merricx_number_1_fans!" in response.text, "Admin cannot login"
|
|
assert flag in response.text, "Flag is missing in admin page"
|
|
assert "flashes" not in response.text, "Admin cannot login"
|
|
self.logger.info(f'C2 success for burvesigner')
|
|
|
|
# C3: login guest fail
|
|
guest_data = {
|
|
'username': 'guest',
|
|
'password': 'Guest',
|
|
}
|
|
response = requests.post(url, data=guest_data, timeout=5).text
|
|
assert "Welcome, guest!" not in response, "Guest with wrong credential can login"
|
|
assert "flashes" in response, "Guest with wrong credential can login"
|
|
self.logger.info(f'C3 success for burvesigner')
|
|
|
|
# C4: login admin fail
|
|
admin_data = {
|
|
'username': 'merricx_number_1_fans',
|
|
'password': 'password',
|
|
}
|
|
response = requests.post(url, data=admin_data, timeout=5).text
|
|
assert "Welcome, merricx_number_1_fans!" not in response, "Admin with wrong credential can login"
|
|
assert flag not in response, "Flag is showing in admin page with wrong credential"
|
|
assert "flashes" in response, "Admin with wrong credential can login"
|
|
self.logger.info(f'C4 success for burvesigner')
|
|
|
|
self.logger.info('calling sleep(5)')
|
|
time.sleep(5)
|
|
|
|
# C5: cek apakah token guest di C1 sudah expired atau belum
|
|
response = requests.get(url, cookies={"token": guest_token}, timeout=5)
|
|
assert "Welcome, guest!" in response.text, "Guest token expires early"
|
|
assert "flashes" not in response.text, "Guest token expires early"
|
|
self.logger.info(f'C5 success for burvesigner')
|
|
|
|
# C6: cek apakah token admin di C2 sudah expired atau belum
|
|
response = requests.get(url, cookies={"token": admin_token}, timeout=5)
|
|
assert "Welcome, merricx_number_1_fans!" in response.text, "Admin token expires early"
|
|
assert flag in response.text, "Admin token expires early"
|
|
assert "flashes" not in response.text, "Admin token expires early"
|
|
self.logger.info(f'C6 success for burvesigner')
|
|
|
|
# C7: cek endpoint /params
|
|
response = requests.get(url + "/params", timeout=5).text
|
|
response = response.replace("<pre>", "").replace("</pre>", "")
|
|
params = json.loads(response)
|
|
assert params["p"] and params["a"] and params["b"] and params["n"], "Missing p, a, b and/or n parameter(s)"
|
|
assert params["G"][0] and params["G"][1] and params["Y"][0] and params["Y"][1], "Missing G and/or Y point(s)"
|
|
self.logger.info(f'C7 success for burvesigner')
|
|
|
|
# C8: cek apakah curve C valid dan point G di C
|
|
C = Curve("burvesigner", params["p"], params["a"], params["b"], params["n"], params["G"][0], params["G"][1])
|
|
assert C.G == Point(params["G"][0], params["G"][1], C), "Point G is not valid"
|
|
self.logger.info(f'C8 success for burvesigner')
|
|
|
|
# C9: cek apakah point G * priv = Y
|
|
t = params["p"].bit_length() // 8
|
|
priv = open(self.priv_location, "rb").read()[:t]
|
|
x = int.from_bytes(priv, "little")
|
|
Y = Point(params["Y"][0], params["Y"][1], C)
|
|
assert C.G * x == Y, "Point Y is not valid"
|
|
self.logger.info(f'C9 success for burvesigner')
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
self.logger.error(f'Could not check burvesigner: {e}')
|
|
return False
|