栈空间大小上报

Author: undefined-moe

examples/langs.yaml

@@ -1,3 +1,8 @@
+ccWithoutO2:
+    type: compiler
+    compile: /usr/bin/g++ -Wall -o ${name} foo.cc
+    code_file: foo.cc
+    execute: ${dir}/${name}
 c:
     type: compiler
     compile: /usr/bin/gcc -O2 -Wall -std=c99 -o ${name} foo.c -lm

judger/compile.js

@@ -10,7 +10,7 @@ const { LANGS_FILE, LANGS } = require('./config');
 let _langs = {};
 try {
     if (LANGS) _langs = LANGS;
-    _langs = yaml.safeLoad(fs.readFileSync(LANGS_FILE));
+    else _langs = yaml.safeLoad(fs.readFileSync(LANGS_FILE));
 } catch (e) {
     log.error('Invalidate Language file %s', LANGS_FILE);
     log.error(e);

judger/hosts/hydro.js

@@ -27,11 +27,13 @@ class JudgeTask {
     async handle() {
         try {
             this.stat.handle = new Date();
-            this.pid = this.request.pid;
-            this.rid = this.request.rid;
+            this.pid = (this.request.pid || 'unknown').toString();
+            this.rid = this.request.rid.toString();
+            this.domainId = this.request.domainId;
             this.lang = this.request.lang;
             this.code = this.request.code;
             this.data = this.request.data;
+            this.config = this.request.config;
             this.next = this.getNext(this);
             this.end = this.getEnd(this.session, this.rid);
             this.tmpdir = path.resolve(TEMP_DIR, 'tmp', this.host, this.rid);
@@ -124,7 +126,7 @@ class JudgeTask {
     }
 }
 
-module.exports = class AxiosInstance {
+class Hydro {
     constructor(config) {
         this.config = config;
         this.config.detail = this.config.detail || true;
@@ -249,4 +251,7 @@ module.exports = class AxiosInstance {
         fs.writeFileSync(path.join(filePath, 'version'), version);
         return filePath;
     }
-};
+}
+
+Hydro.JudgeTask = JudgeTask;
+module.exports = Hydro;

judger/sysinfo.js

@@ -1,4 +1,9 @@
+const path = require('path');
 const systeminformation = require('systeminformation');
+const { judge } = require('./judger/run');
+const { TEMP_DIR } = require('./config');
+const { mkdirp, rmdir } = require('./utils');
+const tmpfs = require('./tmpfs');
 
 function size(s, base = 1) {
     s *= base;
@@ -13,15 +18,58 @@ function size(s, base = 1) {
 
 const cache = {};
 
+async function stackSize() {
+    let output = '';
+    const context = {
+        lang: 'ccWithoutO2',
+        code: `
+#include <iostream>
+using namespace std;
+int i=1;
+int main(){
+    char a[1048576]={'1'};
+    cout<<" "<<i<<flush;
+    i++;
+    if (i>256) return 0;
+    main();
+}`,
+        config: {
+            time: 3000,
+            memory: 256,
+        },
+        stat: {},
+        clean: [],
+        next: () => { },
+        end: (data) => {
+            if (data.stdout) output = data.stdout;
+        },
+    };
+    context.tmpdir = path.resolve(TEMP_DIR, 'tmp', 'sysinfo');
+    mkdirp(context.tmpdir);
+    tmpfs.mount(context.tmpdir, '64m');
+    await judge(context).catch((e) => console.error(e));
+    // eslint-disable-next-line no-await-in-loop
+    for (const clean of context.clean) await clean().catch();
+    tmpfs.umount(context.tmpdir);
+    await rmdir(context.tmpdir);
+    const a = output.split(' ');
+    return parseInt(a[a.length - 1]);
+}
+
 async function get() {
-    const [Cpu, Memory, OsInfo, CurrentLoad, CpuFlags, CpuTemp, Battery] = await Promise.all([
+    const [
+        Cpu, Memory, OsInfo,
+        CurrentLoad, CpuFlags, CpuTemp,
+        Battery, stack,
+    ] = await Promise.all([
         systeminformation.cpu(),
         systeminformation.mem(),
         systeminformation.osInfo(),
         systeminformation.currentLoad(),
         systeminformation.cpuFlags(),
         systeminformation.cpuTemperature(),
         systeminformation.battery(),
+        stackSize(),
     ]);
     const cpu = `${Cpu.manufacturer} ${Cpu.brand}`;
     const memory = `${size(Memory.active)}/${size(Memory.total)}`;
@@ -31,13 +79,14 @@ async function get() {
     let battery;
     if (!Battery.hasbattery) battery = 'No battery';
     else battery = `${Battery.type} ${Battery.model} ${Battery.percent}%${Battery.ischarging ? ' Charging' : ''}`;
-    const _id = OsInfo.serial;
+    const mid = OsInfo.serial;
     cache.cpu = cpu;
     cache.osinfo = osinfo;
     cache.flags = flags;
-    cache._id = _id;
+    cache.mid = mid;
+    cache.stack = stack;
     return {
-        _id, cpu, memory, osinfo, load, flags, CpuTemp, battery,
+        mid, cpu, memory, osinfo, load, flags, CpuTemp, battery, stack,
     };
 }
 
@@ -49,20 +98,20 @@ async function update() {
         systeminformation.battery(),
     ]);
     const {
-        _id, cpu, osinfo, flags,
+        mid, cpu, osinfo, flags, stack,
     } = cache;
     const memory = `${size(Memory.active)}/${size(Memory.total)}`;
     const load = `${CurrentLoad.avgload}`;
     let battery;
     if (!Battery.hasbattery) battery = 'No battery';
     else battery = `${Battery.type} ${Battery.model} ${Battery.percent}%${Battery.ischarging ? ' Charging' : ''}`;
     return [
-        _id,
+        mid,
         {
             memory, load, battery, CpuTemp,
         },
         {
-            _id, cpu, memory, osinfo, load, flags, battery, CpuTemp,
+            mid, cpu, memory, osinfo, load, flags, battery, CpuTemp, stack,
         },
     ];
 }

module/service.js

@@ -15,10 +15,18 @@ async function postInit() {
     const { STATUS_COMPILE_ERROR, STATUS_SYSTEM_ERROR } = require('../judger/status');
     const readCases = require('../judger/cases');
     const judger = require('../judger/judger');
+    const sysinfo = require('../judger/sysinfo');
 
     const fsp = fs.promises;
     const { problem, file, task } = global.Hydro.model;
-    const { judge } = global.Hydro.handler;
+    const { judge, misc } = global.Hydro.handler;
+
+    const info = await sysinfo.get();
+    misc.updateStatus(info);
+    setInterval(async () => {
+        const [mid, info] = await sysinfo.update();
+        misc.updateStatus({ mid, ...info });
+    }, 1200000);
 
     async function processData(folder) {
         let files = await fsp.readdir(folder);