Update example

tshemsedinov · tshemsedinov · commit 94a9ebb89fa8 · 2018-02-23T22:53:27.000+02:00
diff --git a/JavaScript/application.js b/JavaScript/application.js
@@ -7,7 +7,10 @@
 // Print from the global context of application module
 console.log('From application global context');
 
-module.exports = function() {
+const fs = require('fs');
+console.dir({ fs });
+
+module.exports = () => {
   // Print from the exported function context
   console.log('From application exported function');
 };
diff --git a/JavaScript/framework.js b/JavaScript/framework.js
@@ -4,13 +4,26 @@
 // appication runtime, load an application code and passes a sandbox into app
 // as a global context and receives exported application interface
 
+const PARSING_TIMEOUT = 1000;
+const EXECUTION_TIMEOUT = 5000;
+
 // The framework can require core libraries
 const fs = require('fs');
 const vm = require('vm');
 
 // Create a hash and turn it into the sandboxed context which will be
 // the global context of an application
-const context = { module: {}, console };
+const context = {
+  module: {}, console,
+  require: (name) => {
+    if (name === 'fs') {
+      console.log('Module fs is restricted');
+      return null;
+    }
+    return require(name);
+  }
+};
+
 context.global = context;
 const sandbox = vm.createContext(context);
 
@@ -20,13 +33,27 @@ fs.readFile(fileName, (err, src) => {
   // We need to handle errors here
 
   // Run an application in sandboxed context
-  const script = vm.createScript(src, fileName);
-  script.runInNewContext(sandbox);
+  let script;
+  try {
+    script = new vm.Script(src, { timeout: PARSING_TIMEOUT });
+  } catch (e) {
+    console.log('Parsing timeout');
+    process.exit(1);
+  }
+
+  try {
+    script.runInNewContext(sandbox, { timeout: EXECUTION_TIMEOUT });
+    const exported = sandbox.module.exports;
+    console.dir({ exported });
+  } catch (e) {
+    console.log('Execution timeout');
+    process.exit(1);
+  }
 
   // We can access a link to exported interface from sandbox.module.exports
   // to execute, save to the cache, print to console, etc.
 });
 
 process.on('uncaughtException', (err) => {
-  console.log('>>>' + err.stack);
+  console.log('Unhandled exception: ' + err);
 });
