Ejemplo n.º 1
0
def do_ingrep(self, cmd):
    """
    usage: ingrep <task> <pattern> <filepath>

    Tries to grep through the specified ASCII file using .incbin directive.
    """
    brutejudge.cheats.cheating(self)
    cmd = shlex.split(cmd)
    if len(cmd) != 3:
        return self.do_help('ingrep')
    task, pattern, file = cmd
    tasks = task_list(self.url, self.cookie)
    try:
        task_id = tasks.index(task)
    except ValueError:
        raise BruteError("No such task.")
    out = io.StringIO()
    incat(self, task, task_id, file, out, '''\
istringstream sin(included_s);
string s2;
string l;
while (getline(sin, l))
{
    if (l.find("'''+pattern.replace('\\', '\\\\').replace('"', '\\"')+'''") != l.npos)
    {
        s2 += l;
        s2 += '\\n';
    }
}
included_s = s2;
''')
    print(out.getvalue())
Ejemplo n.º 2
0
def format_scoreboard(aug=lambda y, x: x):
    url, cookie = force_session()
    with bj.may_cache(url, cookie):
        try:
            data = bj.scoreboard(url, cookie)
        except BruteError as e:
            bottle.response.status = 500
            return aug(True, '<pre>' + html.escape(str(e)) + '</pre>')
        except:
            bottle.response.status = 500
            return aug(True, '<pre>Internal server error</pre>')
        tasks = bj.task_list(url, cookie)
    task = pkgutil.get_data('ejui', 'scoreboard_task.html').decode('utf-8')
    tasks = ''.join(task.format(name=html.escape(i)) for i in tasks)
    contestant = pkgutil.get_data('ejui', 'contestant.html').decode('utf-8')
    contestant_task = pkgutil.get_data('ejui',
                                       'contestant_task.html').decode('utf-8')
    contestants = []
    for index, (data, scores) in enumerate(data):
        cur_tasks = ''.join(
            '<td></td>' if i == None else contestant_task.
            format(kind=(('ok' if i['attempts'] >= 0 else 'fail'
                          ) if 'attempts' in i else 'unknown'),
                   score=sb_format_single(i)) for i in scores)
        contestants.append(
            contestant.format(index=index + 1,
                              nickname=data['name'],
                              tasks=cur_tasks))
    return aug(
        False,
        pkgutil.get_data('ejui', 'scoreboard.html').decode('utf-8').format(
            tasks=tasks, contestants=''.join(contestants)))
Ejemplo n.º 3
0
def format_page(page, text, tl=None, subms=None, clars=None):
    url, cookie = force_session()
    if tl == None: tl = bj.task_list(url, cookie)
    if clars == None: clars = list(zip(*bj.clars(url, cookie)))
    if subms == None: subms = format_submissions(None)[2]
    data = [('main', '', '/', '<b>ejui</b>')]
    for i, j in enumerate(tl):
        data.append(('task%d' % i, '', '/task/%d' % i, html.escape(j)))
    data2 = [('error', '', '', '<div id="error_btn">!</div>'),
             ('subms', '', '/submissions', 'Submissions'),
             ('scoreboard', '', '/scoreboard', 'Scoreboard'),
             ('logout', 'toolbar_icon', '/logout',
              '<img src="/logout.png" alt="Log out" />')]
    if clars or page == 'clars':
        data2.insert(3, ('clars', 'toolbar_icon', '/clars',
                         '<img src="/mail.png" alt="Clarifications" />'))
    head = ''
    for a, b, c, d in data:
        if a == page:
            b += ' selected'
        head += '<a id="toolbar_item_' + a + '" class="' + b + '" href="' + c + '" onclick="ajax_load(this); return false"'
        head += '>' + d + '</a>'
    head += '</td><td align=right>'
    for a, b, c, d in data2:
        head += '<a id="' + a + '" class="' + b + '" href="' + c + '" onclick="ajax_load(this); return false"'
        if a == page:
            head += ' class=selected'
        head += '>' + d + '</a>'
    head += ''
    if page.startswith('task'):
        curr_task = repr(tl[int(page[4:])])
    else:
        curr_task = 'null'
    return pkgutil.get_data('ejui', 'skel.html').decode('utf-8').format(
        task=curr_task, subm_preload=json.dumps(subms), head=head, body=text)
Ejemplo n.º 4
0
def task_page(task):
    task = int(task)
    url, cookie = force_session()
    with bj.may_cache(url, cookie):
        tl = bj.task_list(url, cookie)
        td = task_data(task, tl=tl)
        t1 = pkgutil.get_data('ejui', 'task.html').decode('utf-8')
        t2 = pkgutil.get_data('ejui', 'compiler.html').decode('utf-8')
        subms, any_subms, json_subms = format_submissions(td['name'])
        if any_subms:
            subms = pkgutil.get_data(
                'ejui', 'task_subms.html').decode('utf-8').format(subms=subms)
        else:
            subms = ''
        compilers = ''
        for a, b, c in td['compilers']:
            compilers += t2.format(id=a,
                                   short_name=html.escape(b),
                                   long_name=html.escape(c))
        if td['compilers']:
            compilers = pkgutil.get_data(
                'ejui',
                'compilers.html').decode('utf-8').format(data=compilers)
        return format_page('task%d' % task,
                           t1.format(id=task,
                                     name=html.escape(td['name']),
                                     subms=subms,
                                     compilers=compilers),
                           tl=tl,
                           subms=json_subms)
Ejemplo n.º 5
0
def clar_form(task):
    task = int(task)
    url, cookie = force_session()
    tl = bj.task_list(url, cookie)
    if task not in range(len(tl)):
        abort(404)
    name = tl[task]
    return format_page(
        'clars',
        pkgutil.get_data('ejui', 'submit_clar.html').decode('utf-8').format(
            id=task, name=html.escape(name)))
Ejemplo n.º 6
0
def easy_incat(self, task, filepath, custom_include=None, binary=True):
    try:
        task_id = task_list(self.url, self.cookie).index(task)
    except ValueError:
        raise BruteError("No such task.")
    x = io.StringIO()
    incat(self,
          task,
          task_id,
          filepath,
          x,
          filter=(base64_filter if binary else ''),
          custom_include=custom_include)
    x.seek(0)
    ans = x.read()
    if binary: ans = base64.b64decode(ans.encode('ascii'))
    return ans
Ejemplo n.º 7
0
def do_brute(self, cmd):
    """
    usage: brute <task> <reader> [dump_path] [max_count]

    Tries to retrieve the test cases using binary search.
    """
    brutejudge.cheats.cheating(self)
    sp = cmd.split()
    if len(sp) not in range(2, 5):
        return self.do_help('brute')
    tasks = task_list(self.url, self.cookie)
    try:
        task_id = tasks.index(sp[0])
    except ValueError:
        raise BruteError("No such task.")
    try:
        with open(sp[1]) as file:
            code = file.read()
    except FileNotFoundError:
        raise BruteError("File not found.")
    srch = Searcher(self.url,
                    self.cookie,
                    task_id,
                    input_file=getattr(self, 'input_file', None),
                    output_file=getattr(self, 'output_file', None))
    max_tests = -1
    if len(sp) == 4:
        try:
            max_tests = int(sp[3])
        except ValueError:
            raise BruteError("max_tests must be a number")
    srch.execute(code, max_tests)
    if len(sp) == 2 or sp[2] == '-':
        from sys import stdout as file
        do_close = False
    else:
        try:
            file = open(sp[2], "a")
        except IOError as e:
            raise BruteError("Error creating output file: " + str(e))
        do_close = True
    for i, t in enumerate(srch.tests):
        print("---------- test #", i, sep='', file=file)
        print(t, file=file)
Ejemplo n.º 8
0
def task_data(task, tl=None):
    task = int(task)
    url, cookie = force_session()
    with bj.may_cache(url, cookie):
        if tl == None: tl = bj.task_list(url, cookie)
        if task not in range(len(tl)):
            abort(404)
        name = tl[task]
        tids = bj.task_ids(url, cookie)
        try:
            tid = tids[task]
        except IndexError:
            compilers = []
        else:
            try:
                compilers = bj.compiler_list(url, cookie, tid)
            except BruteError:
                compilers = []
        return {'name': name, 'compilers': compilers}
Ejemplo n.º 9
0
def format_page(page,
                text,
                tl=None,
                subms=None,
                clars=None,
                status=None,
                scores=None,
                jjs_devmode=None):
    url, cookie = force_session()
    with bj.may_cache(url, cookie):
        if tl == None: tl = bj.task_list(url, cookie)
        if clars == None:
            try:
                clars = list(zip(*bj.clars(url, cookie)))
            except:
                clars = []
        if status == None:
            try:
                status = bj.status(url, cookie)
            except:
                status = {}
        if scores == None:
            try:
                scores = bj.scores(url, cookie)
            except:
                scores = {}
        if subms == None: subms = format_submissions(None)[2]
        if jjs_devmode == None:
            jjs_devmode = get_contest_info(url, cookie)[2].get(
                'jjs_devmode', False)
    data = [('main', '', '/', '<b>ejui</b>', '')]
    dyn_style = ''
    for i, j in enumerate(tl):
        if status.get(j, None) != None or scores.get(j, None) != None:
            st = status.get(j, 'True')
            sc = scores.get(j, 0)
            c1 = get_submission_color(st, sc, 1)
            c2 = get_submission_color(st, sc, 2)
            c = 'data-color1="%s" data-color2="%s" data-taskid="%s"' % (
                c1, c2, html.escape(j))
            dyn_style += '#toolbar_item_task%d\n{\n    background: %s !important;\n}\n\n' % (
                i, c2 if page == 'task%d' % i else c1)
        else:
            c = 'data-taskid="%s"' % html.escape(j)
        data.append(('task%d' % i, '', '/task/%d' % i, html.escape(j), c))
    if dyn_style:
        dyn_style = '<style id="dyn_style">\n' + dyn_style.strip(
        ) + '\n</style>'
    data2 = [('error', '', '', '<div id="error_btn">!</div>', ''),
             ('subms', '', '/submissions', 'Submissions', ''),
             ('scoreboard', '', '/scoreboard', 'Scoreboard', ''),
             ('logout', 'toolbar_icon', '/logout',
              '<img src="/logout.png" alt="Log out" />', '')]
    if clars or page == 'clars':
        data2.insert(3, ('clars', 'toolbar_icon', '/clars',
                         '<img src="/mail.png" alt="Clarifications" />', ''))
    head = ''
    for a, b, c, d, e in data:
        if a == page:
            b += ' selected'
        head += '<a id="toolbar_item_' + a + '" class="' + b + '" href="' + c + '" onclick="ajax_load(this); return false"'
        if e:
            head += ' ' + e
        head += '>' + d + '</a>'
    head += '</td><td align=right>'
    for a, b, c, d, e in data2:
        head += '<a id="toolbar_item_' + a + '" class="' + b + '" href="' + c + '" onclick="ajax_load(this); return false"'
        if a == page:
            head += ' class=selected'
        if e:
            head += ' ' + e
        head += '>' + d + '</a>'
    head += ''
    if page.startswith('task'):
        curr_task = repr(tl[int(page[4:])])
    else:
        curr_task = 'null'
    return pkgutil.get_data('ejui', 'skel.html').decode('utf-8').format(
        body_style=(' style="background: url(/jjs_devmode.svg)"'
                    if jjs_devmode else ''),
        task=curr_task,
        subm_preload=json.dumps(subms),
        dynamic_styles=dyn_style,
        head=head,
        body=text)
Ejemplo n.º 10
0
def do_score100(self, cmd):
    """
    usage: score100 <task> <test_cnt> || score100 <subm_id> || score100 <task> <start-end>

    Submit a solution that will get a score of 100.
    If `task` and `test_cnt` are supplied, will submit a solution for task `task` that will pass `test_cnt` tests.
    If `subm_id` is supplied, `task` and `test_cnt` will be taken from that.
    If `start` and `end` are supplied, the task is assumed to have from `start` to `end` tests, the exact number will be retrieved using binary search.
    """
    brutejudge.cheats.cheating(self)
    sp = cmd.split()
    if len(sp) not in (1, 2):
        return self.do_help('score100')
    if not isinstance(self.url, Ejudge):
        raise BruteError("score100 only works on ejudge.")
    lx, ly = submission_list(self.url, self.cookie)
    if len(sp) == 1:
        if not sp[0].isnumeric():
            raise BruteError("Submission ID must be a number.")
        subm_id = int(sp[0])
        try: sp[0] = ly[lx.index(subm_id)].strip()
        except (ValueError, IndexError):
            raise BruteError("No such submission.")
        x, y = submission_results(self.url, self.cookie, subm_id)
        sp.append(str(len(x)))
    tasks = task_list(self.url, self.cookie)
    if sp[0] not in tasks:
        raise BruteError("No such task.")
    task_id = tasks.index(sp[0])
    try:
        if '-' in sp[1]:
            spa, spb = sp[1].split('-', 1)
            test_cnt_low = int(spa)
            test_cnt_high = int(spb) + 1
        else:
            test_cnt_low = test_cnt_high = int(sp[1])
            test_cnt_low -= 1
            test_cnt_high += 1
    except ValueError:
        raise BruteError("test_cnt must be a number or a pair of numbers. (e.g. 1 or 1-2)")
    contest_id = self.url.contest_id
    code0 = ('//random: %r\n#define EXECUTE_FLAGS 08\n#include "'+'../'*16+'home/judges/%06d/problems/%s/Makefile"\n')%(random.random(), contest_id, sp[0])
    lang_id = get_possible_lang_id(self, ('g++', 'g++-32', 'gcc', 'gcc-32'), task_id)
    submit(self.url, self.cookie, task_id, lang_id, code0)
    lx2, ly2 = submission_list(self.url, self.cookie)
    if len(lx2) == len(lx):
        raise BruteError("Error while sending.")
    while still_running(submission_status(self.url, self.cookie, lx2[0])): pass
    tailcode = get_tailcode(self, lx2[0], 'Makefile')
    tp = '%02d'
    cp = '%02d.a'
    for l in tailcode.split('\n'):
        if l.startswith('EXECUTE_FLAGS = '):
            tp = l.split('--test-pattern=', 1)[1].split()[0]
            cp = l.split('--corr-pattern=', 1)[1].split()[0]
    while test_cnt_high - test_cnt_low > 1:
        test_cnt = (test_cnt_high + test_cnt_low) // 2
        code = ''
        for i in range(test_cnt):
            code += ('asm("test_%d:\\n.incbin \\"'+'../'*16+'home/judges/%06d/problems/%s/tests/'+tp+'\\"\\n.byte 0");\n')%(i, contest_id, sp[0], i + 1)
            code += ('asm("ans_%d:\\n.incbin \\"'+'../'*16+'home/judges/%06d/problems/%s/tests/'+cp+'\\"\\n.byte 0");\n')%(i, contest_id, sp[0], i + 1)
            code += 'extern char test_%d[];\n'%i
            code += 'extern char ans_%d[];\n'%i
        code += '\n//random: %r\n\n#include <stdio.h>\n#include <stdlib.h>\n#include <string.h>\n\nint main()\n{'%random.random()+r'''
    int buf_sz = 1;
    void* buf = (void*)0;
    while (1) {
        buf = realloc(buf, buf_sz);
        int bytes_read = fread((void*)(((char*)buf)+buf_sz/2), 1, (buf_sz+1)/2, stdin);
        if (bytes_read < (buf_sz+1)/2) {
            *(((char*)buf)+buf_sz/2+bytes_read) = 0;
            break;
        }
        buf_sz *= 2;
    }
    fprintf(stderr, "%s", (char*)buf);
'''
        for i in range(test_cnt):
            code += '    if (!strcmp((char*)buf, test_%d)) {\n'%i
            code += '        printf("%%s", ans_%d);\n'%i
            code += '        return 0;\n    }\n'
        code += '    printf("Sorry, I don\'t know the answer!\\n");\n'
        code += '    return 0;\n}\n'
        submit(self.url, self.cookie, task_id, lang_id, code)
        lx3, ly3 = submission_list(self.url, self.cookie)
        if len(lx2) == len(lx3):
            raise BruteError("Error while sending.")
        lx2, ly2 = lx3, ly3
        while True:
            status = submission_status(self.url, self.cookie, lx2[0])
            if not still_running(status): break
        if status == 'OK': return
        elif status == ('Compilation error', 'Compiler failed'): test_cnt_high = test_cnt
        else: test_cnt_low = test_cnt
    raise BruteError("Unknown failure, probably an interactive task.")