changeme.py

"""
These are ctf dependent functions.
They have to be aligned.
"""
from random import randint


MESSAGES = [
        "SUCCESS",
        "INVALID",
        "ALREADY_SUBMITTED",
        "ERROR"
]


# TODO: Change these values
# list of team ids
all_teams = [i for i in range(20)]
# Has to be byte string since wre looking through byte output (STDOUT)
flag_regex = b"flag{[A-Za-z0-9_.]+}"

exploit_dir = "exploits"
wait_between_runs = 10
# Number of exploit workers
proc_num = 4
# DB file name
db_file_path = "flags.db"


def submission_logic(flags):
    # TODO: Add actual submission logic
    return [randint(0, len(MESSAGES) - 1) for _ in range(len(flags))]


def team_id_to_ip(team_id):
    """
    Somehow map team id to ip adress
    """
    return "10.13.37.{}".format(team_id)