aboutsummaryrefslogtreecommitdiffhomepage
path: root/worker/php/lib.mjs
blob: a5f10abec11a2470b10d81c2eb97e7c5fb111c5b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
const PRELUDE = `
  define('STDIN', fopen('php://stdin', 'r'));
  define('STDOUT', fopen('php://stdout', 'r'));
  define('STDERR', fopen('php://stderr', 'r'));

  error_reporting(E_ALL & ~E_WARNING & ~E_NOTICE & ~E_DEPRECATED);

  `;

const BUFFER_MAX = 10 * 1024;

const FORBIDDEN_CONSTRUCTS = [/\beval\b/i];

export function validateCode(code) {
  for (const pattern of FORBIDDEN_CONSTRUCTS) {
    if (pattern.test(code)) {
      return `Forbidden: eval() is not allowed`;
    }
  }
  return null;
}

export function preprocessCode(originalCode) {
  if (originalCode.startsWith("<?php")) {
    return PRELUDE + originalCode.slice(5);
  }
  if (originalCode.startsWith("<?")) {
    return PRELUDE + originalCode.slice(2);
  }
  return PRELUDE + originalCode;
}

export function createIOCallbacks(input) {
  let stdinPos = 0;
  const stdinBuf = Buffer.from(input);
  let stdoutPos = 0;
  const stdoutBuf = Buffer.alloc(BUFFER_MAX);
  let stderrPos = 0;
  const stderrBuf = Buffer.alloc(BUFFER_MAX);

  return {
    stdin: () => {
      if (stdinBuf.length <= stdinPos) {
        return null;
      }
      return stdinBuf.readUInt8(stdinPos++);
    },
    stdout: (asciiCode) => {
      if (asciiCode === null) {
        return;
      }
      if (BUFFER_MAX <= stdoutPos) {
        return;
      }
      stdoutBuf.writeUInt8(
        asciiCode < 0 ? asciiCode + 256 : asciiCode,
        stdoutPos++,
      );
    },
    stderr: (asciiCode) => {
      if (asciiCode === null) {
        return;
      }
      if (BUFFER_MAX <= stderrPos) {
        return;
      }
      stderrBuf.writeUInt8(
        asciiCode < 0 ? asciiCode + 256 : asciiCode,
        stderrPos++,
      );
    },
    getStdout: () => stdoutBuf.subarray(0, stdoutPos).toString(),
    getStderr: () => stderrBuf.subarray(0, stderrPos).toString(),
  };
}

export function buildResult(err, ccallResult, getStdout, getStderr) {
  if (err) {
    return {
      status: "runtime_error",
      stdout: getStdout(),
      stderr: `${getStderr()}\n${err.toString()}`,
    };
  }
  return {
    status: ccallResult === 0 ? "success" : "runtime_error",
    stdout: getStdout(),
    stderr: getStderr(),
  };
}