From 5461f91a393e78bd901487d1762b93637dad9f95 Mon Sep 17 00:00:00 2001
From: Seohyun Lee <studioseorii@gmail.com>
Date: Wed, 29 Dec 2021 21:05:23 +0900
Subject: [PATCH] fix: build source with user permission

---
 Dockerfile          |   2 +-
 res/test.html       |   2 +-
 src/runner/cpp.ts   | 114 +++++++++++++++++++-------------------------
 src/runner/index.ts |  10 ++--
 4 files changed, 54 insertions(+), 74 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 073b8bc..97e1941 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,6 @@
 FROM node:16-alpine
 EXPOSE 80
-RUN apk add --update alpine-sdk runuser
+RUN apk add --update alpine-sdk
 RUN apk update
 COPY dist/ /HANA/dist/
 COPY res/ /HANA/res/
diff --git a/res/test.html b/res/test.html
index 5124fc4..4a5ff2e 100644
--- a/res/test.html
+++ b/res/test.html
@@ -275,7 +275,7 @@
             <div class="form">
                 <label
                     class="mdc-text-field mdc-text-field--outlined mdc-text-field--textarea mdc-text-field--no-label"
-                    style="width: calc(100vw - 20px); margin: 10px 0"
+                    style="width: calc(100vw - 40px); margin: 10px 0"
                 >
                     <span class="mdc-notched-outline">
                         <span class="mdc-notched-outline__leading"></span>
diff --git a/src/runner/cpp.ts b/src/runner/cpp.ts
index 0397321..b391599 100644
--- a/src/runner/cpp.ts
+++ b/src/runner/cpp.ts
@@ -17,7 +17,7 @@ export default function (
         CommonDataSet
     >
 ) {
-    return new Promise<JudgeResult>((resolve) => {
+    return new Promise<JudgeResult>(async (resolve) => {
         let match = Array(data.dataSet.data.length).fill(false)
         const tmpPath = '/tmp/' + data.uid
         const exePath = tmpPath + '/main'
@@ -28,78 +28,62 @@ export default function (
             reason: 'CP',
         })
 
+        execSync(`adduser --disabled-password --no-create-home p-${data.uid}`)
+
         fs.mkdirSync(tmpPath)
+        execSync(`chmod 777 ${tmpPath}`)
 
         for (const i of data.source)
             fs.writeFileSync(tmpPath + '/' + i.name, i.source)
 
-        const cp = spawn('g++', [
-            `${tmpPath}/main.cpp`,
-            '-o',
-            exePath,
-            '-O2',
-            '-Wall',
-            '-lm',
-            '--static',
-            '-pipe',
-            '-std=c++17',
-        ])
-        let stderr = ''
+        const result = await execute(
+            `p-${data.uid}`,
+            `g++ ${tmpPath}/main.cpp -o ${exePath} -O2 -Wall -lm --static -pipe -std=c++17`,
+            ''
+        )
 
-        cp.stderr.on('data', (data) => {
-            if (data) stderr += data.toString()
-        })
+        if (result.code !== 0) {
+            fs.rmSync(tmpPath, { recursive: true })
+            resolve({
+                uid: data.uid,
+                result: match,
+                reason: 'CE',
+                time: 0,
+                memory: 0,
+                message: result.stderr,
+            })
+            return
+        }
 
-        cp.on('exit', async (code) => {
-            if (code !== 0) {
-                fs.rmdirSync(tmpPath, { recursive: true })
-                resolve({
-                    uid: data.uid,
-                    result: match,
-                    reason: 'CE',
-                    time: 0,
-                    memory: 0,
-                    message: stderr,
-                })
-                return
-            } else {
-                execSync(
-                    `adduser --disabled-password --no-create-home p-${data.uid}`
-                )
-                sendMessage(WebSocketResponseType.JUDGE_PROGRESS, {
-                    uid: data.uid,
-                    progress: 0,
-                    reason: 'RUN',
-                })
+        sendMessage(WebSocketResponseType.JUDGE_PROGRESS, {
+            uid: data.uid,
+            progress: 0,
+            reason: 'RUN',
+        })
 
-                for (const i in data.dataSet.data) {
-                    const result = await execute(
-                        `p-${data.uid}`,
-                        exePath,
-                        data.dataSet.data[i].input
-                    )
-                    if (isSame(result.stdout, data.dataSet.data[i].output))
-                        match[i] = true
-                    sendMessage(WebSocketResponseType.JUDGE_PROGRESS, {
-                        uid: data.uid,
-                        progress:
-                            (i as unknown as number) / data.dataSet.data.length,
-                        reason: 'RUN',
-                    })
-                }
-                fs.rmSync(tmpPath, { recursive: true })
-                resolve({
-                    uid: data.uid,
-                    result: match,
-                    reason:
-                        match.reduce((a, b) => a + b, 0) === match.length
-                            ? 'AC'
-                            : 'WA',
-                    time: 0,
-                    memory: 0,
-                })
-                execSync(`deluser p-${data.uid}`)
-            }
+        for (const i in data.dataSet.data) {
+            const result = await execute(
+                `p-${data.uid}`,
+                exePath,
+                data.dataSet.data[i].input
+            )
+            if (isSame(result.stdout, data.dataSet.data[i].output))
+                match[i] = true
+            sendMessage(WebSocketResponseType.JUDGE_PROGRESS, {
+                uid: data.uid,
+                progress: (i as unknown as number) / data.dataSet.data.length,
+                reason: 'RUN',
+            })
+        }
+        fs.rmSync(tmpPath, { recursive: true })
+        execSync(`deluser p-${data.uid}`)
+        resolve({
+            uid: data.uid,
+            result: match,
+            reason:
+                match.reduce((a, b) => a + b, 0) === match.length ? 'AC' : 'WA',
+            time: 0,
+            memory: 0,
         })
     })
 }
diff --git a/src/runner/index.ts b/src/runner/index.ts
index 1163558..0104983 100644
--- a/src/runner/index.ts
+++ b/src/runner/index.ts
@@ -14,13 +14,9 @@ import { spawn } from 'child_process'
 export function execute(userName: string, exePath: string, input: string) {
     return new Promise<{ code: number; stdout: string; stderr: string }>(
         (resolve, reject) => {
-            const child = spawn(
-                `runuser`,
-                ['-l', userName, '-c', `'${exePath}'`],
-                {
-                    stdio: ['pipe', 'pipe', 'pipe'],
-                }
-            )
+            const child = spawn(`su`, [userName, '-c', exePath], {
+                stdio: ['pipe', 'pipe', 'pipe'],
+            })
 
             child.stdin.write(input)
             child.stdin.end()