vm: add isContext; prevent double-contextifying

Previously, calling `vm.createContext(o)` repeatedly on the same `o`
would cause new C++ `ContextifyContext`s to be created and stored on
`o`, while the previous resident went off into leaked-memory limbo.
Now, repeatedly trying to contextify a sandbox will do nothing after
the first time.

To detect this, an independently-useful `vm.isContext(sandbox)` export
was added.
pull/41362/head
Domenic Denicola 2013-08-24 15:45:02 -04:00 committed by Ben Noordhuis
parent a3bf3d10ef
commit 9c110d8027
4 changed files with 73 additions and 4 deletions

View File

@ -28,6 +28,7 @@ var util = require('util');
// - runInThisContext()
// - runInContext(sandbox, [timeout])
// - makeContext(sandbox)
// - isContext(sandbox)
// From this we build the entire documented API.
Script.prototype.runInNewContext = function(initSandbox, timeout, disp) {
@ -44,10 +45,11 @@ exports.createScript = function(code, filename, disp) {
exports.createContext = function(initSandbox) {
if (util.isUndefined(initSandbox)) {
initSandbox = {};
} else if (binding.isContext(initSandbox)) {
return initSandbox;
}
binding.makeContext(initSandbox);
return initSandbox;
};
@ -65,3 +67,5 @@ exports.runInThisContext = function(code, filename, timeout, disp) {
var script = exports.createScript(code, filename, disp);
return script.runInThisContext(timeout, disp);
};
exports.isContext = binding.isContext;

View File

@ -119,6 +119,7 @@ class ContextifyContext {
data_wrapper_ctor.Reset(node_isolate, function_template->GetFunction());
NODE_SET_METHOD(target, "makeContext", MakeContext);
NODE_SET_METHOD(target, "isContext", IsContext);
}
@ -130,14 +131,34 @@ class ContextifyContext {
}
Local<Object> sandbox = args[0].As<Object>();
ContextifyContext* context = new ContextifyContext(sandbox);
Local<External> hidden_context = External::New(context);
Local<String> hidden_name =
FIXED_ONE_BYTE_STRING(node_isolate, "_contextifyHidden");
// Don't allow contextifying a sandbox multiple times.
assert(sandbox->GetHiddenValue(hidden_name).IsEmpty());
ContextifyContext* context = new ContextifyContext(sandbox);
Local<External> hidden_context = External::New(context);
sandbox->SetHiddenValue(hidden_name, hidden_context);
}
static void IsContext(const FunctionCallbackInfo<Value>& args) {
HandleScope scope(node_isolate);
if (!args[0]->IsObject()) {
ThrowTypeError("sandbox must be an object");
return;
}
Local<Object> sandbox = args[0].As<Object>();
Local<String> hidden_name =
FIXED_ONE_BYTE_STRING(node_isolate, "_contextifyHidden");
args.GetReturnValue().Set(!sandbox->GetHiddenValue(hidden_name).IsEmpty());
}
static void SandboxFreeCallback(Isolate* isolate,
Persistent<Object>* target,
ContextifyContext* context) {

View File

@ -25,9 +25,15 @@ var vm = require('vm');
assert.throws(function() {
var ctx = vm.createContext('string is not supported');
});
}, TypeError);
assert.doesNotThrow(function() {
var ctx = vm.createContext({ a: 1 });
ctx = vm.createContext([0, 1, 2, 3]);
});
assert.doesNotThrow(function() {
var sandbox = {};
vm.createContext(sandbox);
vm.createContext(sandbox);
});

View File

@ -0,0 +1,38 @@
// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.
var common = require('../common');
var assert = require('assert');
var vm = require('vm');
assert.throws(function() {
vm.isContext('string is not supported');
}, TypeError);
assert.strictEqual(vm.isContext({}), false);
assert.strictEqual(vm.isContext([]), false);
assert.strictEqual(vm.isContext(vm.createContext()), true);
assert.strictEqual(vm.isContext(vm.createContext([])), true);
var sandbox = { foo: 'bar' };
vm.createContext(sandbox);
assert.strictEqual(vm.isContext(sandbox), true);