from tempfile import NamedTemporaryFile, TemporaryDirectory
from os import urandom, system, mkdir
from string import printable
from pathlib import Path
from shutil import copy
blacklist = [
"cInclude",
"embedFile",
]
character_whitelist = printable
def err(code: int, msg: str):
print(msg)
exit(code)
def check(leftover: str):
for disallow in blacklist:
if leftover.startswith(disallow):
err(1, "blacklisted directive found")
def round():
with TemporaryDirectory() as cache:
cache = Path(cache)
with NamedTemporaryFile("w", suffix=".zig", dir=cache) as zig:
zig.write(src)
zig.flush()
cmd = f"/app/zig/zig build-exe -fno-emit-bin -fsingle-threaded -fno-lto -fno-PIC --no-gc-sections --cache-dir {cache} --global-cache-dir /tmp {zig.name}"
system(cmd)
limit = 1337
src = ""
print("code: ")
while True:
line = input()
assert all([ch in character_whitelist for ch in line]), "character not in whitelist"
if line.startswith("EOF"):
break
src += line + "\n"
overflow = len(src) - limit
if overflow > 0:
suffix = "s" if overflow > 1 else ""
err(1, f"input too long by {overflow} byte{suffix}")
for i in range(len(src)):
if src[i] == '@':
check(src[i+1:])
round()
