修复最大内存限制数字可能溢出的问题

将int改为了long
QingdaoU · Mar 22, 2016 · 5304fe4 · 5304fe4
1 parent 62f6e7a
commit 5304fe4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/judger.c b/judger.c
@@ -14,7 +14,7 @@ static PyObject *judger_run(PyObject *self, PyObject *args, PyObject *kwargs) {
                                   "max_memory", "args", "env", "use_sandbox", "use_nobody", NULL};
 
     config.path = config.in_file = config.out_file = NULL;
-    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "sssii|OOOO", kwargs_list, &(config.path), &(config.in_file),
+    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "sssil|OOOO", kwargs_list, &(config.path), &(config.in_file),
                                      &(config.out_file), &(config.max_cpu_time), &(config.max_memory),
                                      &args_list, &env_list, &use_sandbox, &use_nobody)) {
         PyErr_SetString(PyExc_ValueError, "Invalid args and kwargs");

diff --git a/runner.h b/runner.h
@@ -36,7 +36,7 @@ struct result {
 
 struct config {
     int max_cpu_time;
-    int max_memory;
+    long max_memory;
     char *path;
     char *in_file;
     char *out_file;