允许在Web端指定动态文件

Author: qwertycxz

src/Config.ts

@@ -100,6 +100,15 @@ export class JudgeFactoryConfig {
     remoteFileCacheBytes!: number;
 }
 export class FpgaConfig {
+    @IsString()
+    @IsNotEmpty()
+    constraints!: string;
+    @IsString()
+    @IsNotEmpty()
+    program!: string;
+    @IsString()
+    @IsNotEmpty()
+    package!: string;
     @ArrayNotEmpty()
     @ArrayUnique()
     @IsHexadecimal({

src/Spawn/Language/Verilog.ts

@@ -1,9 +1,16 @@
+import { DynamicFile } from "heng-protocol";
 import { join } from "path";
 import { getConfig } from "../../Config";
 import { Language, RunOption, RunType } from "./decl";
+import { getPrj, getXst } from "./XilinxSynthesisTechnology";
 
 export class Verilog extends Language {
-    srcFileName = "main.v";
+    static srcFileName = "main.v";
+    static modifyDynamicFile(files: DynamicFile[] = []) {
+        return [getPrj(this.srcFileName, files), getXst(), ...files];
+    }
+
+    srcFileName = Verilog.srcFileName;
 
     [RunType.Synthesis] = {
         cacheable: true,
@@ -12,7 +19,7 @@ export class Verilog extends Language {
             return {
                 skip: false,
                 command: getConfig().language.xst,
-                args: ["-ifn", "xc6slx9-2-ftg256.verilog.xst"],
+                args: ["-ifn", "main.xst"],
             };
         },
     };

src/Spawn/Language/XilinxSynthesisTechnology.ts

@@ -0,0 +1,47 @@
+import { DynamicFile } from "heng-protocol";
+import { getConfig } from "../../Config";
+
+export function getXst(
+    input = "main.prj",
+    output = "main",
+    fpga = getConfig().fpga.package,
+    top = "main",
+    name = "main.xst"
+): DynamicFile {
+    return {
+        file: {
+            content: `run -ifn ${input} -ofn ${output} -p ${fpga} -top ${top}`,
+            type: "direct",
+        },
+        name,
+        type: "remote",
+    };
+}
+
+function getPrjLine(name: string) {
+    if (name.endsWith(".v")) {
+        return `verilog work ${name}`;
+    }
+    if (name.endsWith(".vhd")) {
+        return `vhdl work ${name}`;
+    }
+    return "";
+}
+
+export function getPrj(
+    srcFileName: string,
+    files: DynamicFile[] = [],
+    name = "main.prj"
+): DynamicFile {
+    return {
+        file: {
+            content: files.reduce(
+                (content, { name }) => `${content}\n${getPrjLine(name)}`,
+                getPrjLine(srcFileName)
+            ),
+            type: "direct",
+        },
+        name,
+        type: "remote",
+    };
+}

src/Spawn/Language/decl.ts

@@ -1,4 +1,4 @@
-import { Executable } from "heng-protocol";
+import { DynamicFile, Executable } from "heng-protocol";
 import { HengSpawnOption } from "..";
 
 export enum RunType {
@@ -55,6 +55,9 @@ export abstract class Language {
         this.runDir = option.runDir;
     }
     abstract get srcFileName(): string;
+    static modifyDynamicFile(files: DynamicFile[] = []) {
+        return files;
+    }
     abstract [RunType.Synthesis]: compileOption;
     abstract [RunType.Translate]: compileOption;
     abstract [RunType.Map]: compileOption;

src/Spawn/Language/index.ts

@@ -19,3 +19,15 @@ export function getConfiguredLanguage(
             break;
     }
 }
+
+export function getLanguage(lang: string): typeof Language {
+    lang = lang.toLowerCase();
+    switch (lang) {
+        case "verilog":
+            return Verilog;
+        case "plaintext":
+            return PlainText;
+        default:
+            throw new Error("Unrecognized language");
+    }
+}

src/Utilities/Judge.ts

@@ -20,6 +20,7 @@ import { SerialPort } from "serialport";
 import { getConfig } from "../Config";
 import { Controller } from "../controller";
 import { Tests } from "../SelfTest";
+import { getLanguage } from "../Spawn/Language";
 import { ExecType, RunType } from "../Spawn/Language/decl";
 import { MeterResult } from "../Spawn/Meter";
 import { ExecutableAgent, runLogName } from "./ExecutableAgent";
@@ -238,24 +239,11 @@ export abstract class JudgeAgent {
         }
     ): Promise<[ExecutableAgent, JudgeResult | undefined]> {
         this.checkInit();
-        const executableAgent = new ExecutableAgent(execType, executable, [
-            {
-                name: "ax309.ucf",
-                type: "builtin",
-            },
-            {
-                name: "main.cmd",
-                type: "builtin",
-            },
-            {
-                name: "verilog.prj",
-                type: "builtin",
-            },
-            {
-                name: "xc6slx9-2-ftg256.verilog.xst",
-                type: "builtin",
-            },
-        ]);
+        const executableAgent = new ExecutableAgent(
+            execType,
+            executable,
+            this.judge.dynamicFiles
+        );
         this.ExecutableAgents.push(executableAgent);
         await executableAgent.init();
         await executableAgent.releaseFile();
@@ -286,14 +274,14 @@ export abstract class JudgeAgent {
             statistics.tick(this.judge.id);
             await this.init();
             const ret = await this.getResult();
-            // await this.clean();
+            await this.clean();
             statistics.finish(this.judge.id);
             return ret;
         } catch (err) {
             this.logger.fatal(err);
-            // await this.clean().catch((error) => {
-            //     this.logger.fatal(error);
-            // });
+            await this.clean().catch((error) => {
+                this.logger.fatal(error);
+            });
             statistics.finish(this.judge.id);
             const e = {
                 kind: JudgeResultKind.SystemError,
@@ -568,6 +556,19 @@ export class JudgeFactory {
     ) {}
 
     getJudgerAgent(judgeInfo: CreateJudgeArgs): JudgeAgent {
+        judgeInfo.dynamicFiles = [
+            {
+                type: "builtin",
+                name: getConfig().fpga.constraints,
+            },
+            {
+                type: "builtin",
+                name: getConfig().fpga.program,
+            },
+            ...getLanguage(
+                judgeInfo.judge.user.environment.language
+            ).modifyDynamicFile(judgeInfo.dynamicFiles),
+        ];
         judgeInfo.judge.user.limit.compiler.cpuTime = Math.ceil(
             judgeInfo.judge.user.limit.compiler.cpuTime / this.timeRatio
         );