SEEC-Code-Judge/lorun/cext/convert.c

/**
 * Loco program runner core
 * Copyright (C) 2011  Lodevil(Du Jiong)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "convert.h"

int initCalls(PyObject *li, u_char calls[]) {
    PyObject *t;
    Py_ssize_t len, i;

    memset(calls, 0, sizeof(u_char) * CALLS_MAX);

    len = PyList_Size(li);
    for (i = 0; i < len; i++) {
        t = PyList_GetItem(li, i);
        if (t == NULL) {return -1;}

        #ifdef IS_PY3
        if (!PyLong_Check(t))
            RAISE1("calls must be a list of numbers.");
        calls[PyLong_AsLong(t)] = 1;
        #else
        if (!PyInt_Check(t) && !PyLong_Check(t))
            RAISE1("calls must be a list of numbers.");
        calls[PyLong_AsLong(t)] = 1;
        #endif
    }

    return 0;
}

PyObject *genResult(struct Result *rst) {
    PyObject *rst_obj, *j, *t, *m;
    if ((rst_obj = PyDict_New()) == NULL)
        RAISE0("new dict failure");

    j = PyLong_FromLong(rst->judge_result);
    t = PyLong_FromLong(rst->time_used);
    m = PyLong_FromLong(rst->memory_used);

    if (!j || !t || !m)
        RAISE0("set item falure(1)");

    if (PyDict_SetItemString(rst_obj, "result", j)) {
        RAISE0("set item falure");
    }

    if (PyDict_SetItemString(rst_obj, "timeused", t)
            || PyDict_SetItemString(rst_obj, "memoryused", m)) {
        RAISE0("set item failure");
    }

    if (rst->re_signum) {
        PyDict_SetItemString(rst_obj, "re_signum",
                PyLong_FromLong(rst->re_signum));
    }
    if (rst->re_call != -1) {
        PyDict_SetItemString(rst_obj, "re_call", PyLong_FromLong(rst->re_call));
    }
    if (rst->re_file) {
        #ifdef IS_PY3
        PyObject *re_file = PyUnicode_FromString(rst->re_file);
        #else
        PyObject *re_file = PyString_FromString(rst->re_file);
        #endif
        PyDict_SetItemString(rst_obj, "re_file", re_file);
        PyDict_SetItemString(rst_obj, "re_file_flag",
            PyLong_FromLong(rst->re_file_flag));
    }

    return rst_obj;
}

char * const * genRunArgs(PyObject *args_obj) { //generate the argsments for exec*
    PyObject *arg;
    const char **args;
    int len, i;

    if (!PyList_Check(args_obj))
        RAISE0("args must be a list")

    len = PyList_GET_SIZE(args_obj);
    args = (const char**) malloc(sizeof(char*) * (len + 1));

    for (i = 0; i < len; i++) {
        #ifdef IS_PY3
        PyObject *arg_bytes;
        #endif

        if ((arg = PyList_GetItem(args_obj, i)) == NULL) {
            free(args);
            return NULL;
        }

        #ifdef IS_PY3
        arg_bytes = PyUnicode_AsUTF8String(arg);
        if (arg_bytes == NULL) {return NULL;}
        args[i] = PyBytes_AsString(arg_bytes);
        #else
        args[i] = PyString_AsString(arg);
        #endif

        if (args[i] == NULL) {return NULL;}
    }
    args[i] = NULL;

    return (char * const *) args;
}