From ef59d79b1acc6ce4478ace497c4ca02af1089c24 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sun, 3 Apr 2022 15:16:29 +0800
Subject: [PATCH] judge: code wrapper for sysinfo

---
 packages/hydrojudge/src/service.ts | 1 +
 packages/hydrojudge/src/sysinfo.ts | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/hydrojudge/src/service.ts b/packages/hydrojudge/src/service.ts
index a78bc38e1..cace7bd8f 100644
--- a/packages/hydrojudge/src/service.ts
+++ b/packages/hydrojudge/src/service.ts
@@ -232,6 +232,7 @@ async function postInit() {
             this.folder = await cacheOpen(this.source, this.data);
             if ((this.code as any).content.startsWith('@@hydro_submission_file@@')) {
                 const id = (this.code as any).content.split('@@hydro_submission_file@@')[1]?.split('#')?.[0];
+                if (!id) throw new SystemError('Submission File Not Found');
                 const target = await fetchCodeFile(id);
                 this.code = { src: target };
                 this.clean.push(() => fs.remove(target));
diff --git a/packages/hydrojudge/src/sysinfo.ts b/packages/hydrojudge/src/sysinfo.ts
index d5a6a6e45..fb76dd4f9 100644
--- a/packages/hydrojudge/src/sysinfo.ts
+++ b/packages/hydrojudge/src/sysinfo.ts
@@ -14,7 +14,8 @@ async function stackSize() {
     try {
         const context: Context = {
             lang: 'cc',
-            code: `#include <iostream>
+            code: {
+                content: `#include <iostream>
 using namespace std;
 int i=1;
 int main(){
@@ -24,6 +25,7 @@ int main(){
     if (i>256) return 0;
     main();
 }`,
+            },
             config: {
                 time: '3s',
                 memory: '256m',