js/src/jsapi-tests/testCloneScript.cpp

Wed, 31 Dec 2014 06:09:35 +0100

author
Michael Schloh von Bennewitz <michael@schloh.com>
date
Wed, 31 Dec 2014 06:09:35 +0100
changeset 0
6474c204b198
permissions
-rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

michael@0 1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0 2 * vim: set ts=8 sts=4 et sw=4 tw=99:
michael@0 3 *
michael@0 4 * Test script cloning.
michael@0 5 */
michael@0 6 /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0 7 * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0 8 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0 9
michael@0 10 #include "js/OldDebugAPI.h"
michael@0 11 #include "jsapi-tests/tests.h"
michael@0 12
michael@0 13 BEGIN_TEST(test_cloneScript)
michael@0 14 {
michael@0 15 JS::RootedObject A(cx, createGlobal());
michael@0 16 JS::RootedObject B(cx, createGlobal());
michael@0 17
michael@0 18 CHECK(A);
michael@0 19 CHECK(B);
michael@0 20
michael@0 21 const char *source =
michael@0 22 "var i = 0;\n"
michael@0 23 "var sum = 0;\n"
michael@0 24 "while (i < 10) {\n"
michael@0 25 " sum += i;\n"
michael@0 26 " ++i;\n"
michael@0 27 "}\n"
michael@0 28 "(sum);\n";
michael@0 29
michael@0 30 JS::RootedObject obj(cx);
michael@0 31
michael@0 32 // compile for A
michael@0 33 {
michael@0 34 JSAutoCompartment a(cx, A);
michael@0 35 JSFunction *fun;
michael@0 36 JS::CompileOptions options(cx);
michael@0 37 options.setFileAndLine(__FILE__, 1);
michael@0 38 CHECK(fun = JS_CompileFunction(cx, A, "f", 0, nullptr, source,
michael@0 39 strlen(source), options));
michael@0 40 CHECK(obj = JS_GetFunctionObject(fun));
michael@0 41 }
michael@0 42
michael@0 43 // clone into B
michael@0 44 {
michael@0 45 JSAutoCompartment b(cx, B);
michael@0 46 CHECK(JS_CloneFunctionObject(cx, obj, B));
michael@0 47 }
michael@0 48
michael@0 49 return true;
michael@0 50 }
michael@0 51 END_TEST(test_cloneScript)
michael@0 52
michael@0 53 static void
michael@0 54 DestroyPrincipals(JSPrincipals *principals)
michael@0 55 {
michael@0 56 delete principals;
michael@0 57 }
michael@0 58
michael@0 59 struct Principals : public JSPrincipals
michael@0 60 {
michael@0 61 public:
michael@0 62 Principals()
michael@0 63 {
michael@0 64 refcount = 0;
michael@0 65 }
michael@0 66 };
michael@0 67
michael@0 68 class AutoDropPrincipals
michael@0 69 {
michael@0 70 JSRuntime *rt;
michael@0 71 JSPrincipals *principals;
michael@0 72
michael@0 73 public:
michael@0 74 AutoDropPrincipals(JSRuntime *rt, JSPrincipals *principals)
michael@0 75 : rt(rt), principals(principals)
michael@0 76 {
michael@0 77 JS_HoldPrincipals(principals);
michael@0 78 }
michael@0 79
michael@0 80 ~AutoDropPrincipals()
michael@0 81 {
michael@0 82 JS_DropPrincipals(rt, principals);
michael@0 83 }
michael@0 84 };
michael@0 85
michael@0 86 BEGIN_TEST(test_cloneScriptWithPrincipals)
michael@0 87 {
michael@0 88 JS_InitDestroyPrincipalsCallback(rt, DestroyPrincipals);
michael@0 89
michael@0 90 JSPrincipals *principalsA = new Principals();
michael@0 91 AutoDropPrincipals dropA(rt, principalsA);
michael@0 92 JSPrincipals *principalsB = new Principals();
michael@0 93 AutoDropPrincipals dropB(rt, principalsB);
michael@0 94
michael@0 95 JS::RootedObject A(cx, createGlobal(principalsA));
michael@0 96 JS::RootedObject B(cx, createGlobal(principalsB));
michael@0 97
michael@0 98 CHECK(A);
michael@0 99 CHECK(B);
michael@0 100
michael@0 101 const char *argnames[] = { "arg" };
michael@0 102 const char *source = "return function() { return arg; }";
michael@0 103
michael@0 104 JS::RootedObject obj(cx);
michael@0 105
michael@0 106 // Compile in A
michael@0 107 {
michael@0 108 JSAutoCompartment a(cx, A);
michael@0 109 JS::CompileOptions options(cx);
michael@0 110 options.setFileAndLine(__FILE__, 1);
michael@0 111 JS::RootedFunction fun(cx, JS_CompileFunction(cx, A, "f",
michael@0 112 mozilla::ArrayLength(argnames), argnames, source,
michael@0 113 strlen(source), options));
michael@0 114 CHECK(fun);
michael@0 115
michael@0 116 JSScript *script;
michael@0 117 CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0 118
michael@0 119 CHECK(JS_GetScriptPrincipals(script) == principalsA);
michael@0 120 CHECK(obj = JS_GetFunctionObject(fun));
michael@0 121 }
michael@0 122
michael@0 123 // Clone into B
michael@0 124 {
michael@0 125 JSAutoCompartment b(cx, B);
michael@0 126 JS::RootedObject cloned(cx);
michael@0 127 CHECK(cloned = JS_CloneFunctionObject(cx, obj, B));
michael@0 128
michael@0 129 JS::RootedFunction fun(cx);
michael@0 130 JS::RootedValue clonedValue(cx, JS::ObjectValue(*cloned));
michael@0 131 CHECK(fun = JS_ValueToFunction(cx, clonedValue));
michael@0 132
michael@0 133 JSScript *script;
michael@0 134 CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0 135
michael@0 136 CHECK(JS_GetScriptPrincipals(script) == principalsB);
michael@0 137
michael@0 138 JS::RootedValue v(cx);
michael@0 139 JS::RootedValue arg(cx, JS::Int32Value(1));
michael@0 140 CHECK(JS_CallFunctionValue(cx, B, clonedValue, arg, &v));
michael@0 141 CHECK(v.isObject());
michael@0 142
michael@0 143 JSObject *funobj = &v.toObject();
michael@0 144 CHECK(JS_ObjectIsFunction(cx, funobj));
michael@0 145 CHECK(fun = JS_ValueToFunction(cx, v));
michael@0 146 CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0 147 CHECK(JS_GetScriptPrincipals(script) == principalsB);
michael@0 148 }
michael@0 149
michael@0 150 return true;
michael@0 151 }
michael@0 152 END_TEST(test_cloneScriptWithPrincipals)

mercurial