43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
|
import requests
|
||
|
|
from base64 import b64decode
|
||
|
|
|
||
|
|
from .Challenge import Challenge
|
||
|
|
|
||
|
|
|
||
|
|
class Hirnfick(Challenge):
|
||
|
|
flag_location = 'flags/hirnfick.txt'
|
||
|
|
history_location = 'history/hirnfick.txt'
|
||
|
|
|
||
|
|
def distribute(self, flag):
|
||
|
|
try:
|
||
|
|
with open(self.flag_location, 'w') as f:
|
||
|
|
f.write(flag)
|
||
|
|
|
||
|
|
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:
|
||
|
|
res = requests.post(
|
||
|
|
f"http://localhost:{self.port}/api/run",
|
||
|
|
timeout=5,
|
||
|
|
json={
|
||
|
|
"code":
|
||
|
|
"+[-->-[>>+>-----<<]<--<---]>-.>>>+.>>..+++[.>]<<<<.+++.------.<<-.>>>>+."
|
||
|
|
})
|
||
|
|
|
||
|
|
assert b64decode(res.json()["output"]) == b"HirnFick 1.0\nHello, World!"
|
||
|
|
|
||
|
|
return True
|
||
|
|
except Exception as e:
|
||
|
|
self.logger.error(f'Could not check hirnfick: {e}')
|
||
|
|
return False
|