michael@0: /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0:  * vim: set ts=8 sts=4 et sw=4 tw=99:
michael@0:  *
michael@0:  * Test script cloning.
michael@0:  */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "js/OldDebugAPI.h"
michael@0: #include "jsapi-tests/tests.h"
michael@0: 
michael@0: BEGIN_TEST(test_cloneScript)
michael@0: {
michael@0:     JS::RootedObject A(cx, createGlobal());
michael@0:     JS::RootedObject B(cx, createGlobal());
michael@0: 
michael@0:     CHECK(A);
michael@0:     CHECK(B);
michael@0: 
michael@0:     const char *source =
michael@0:         "var i = 0;\n"
michael@0:         "var sum = 0;\n"
michael@0:         "while (i < 10) {\n"
michael@0:         "    sum += i;\n"
michael@0:         "    ++i;\n"
michael@0:         "}\n"
michael@0:         "(sum);\n";
michael@0: 
michael@0:     JS::RootedObject obj(cx);
michael@0: 
michael@0:     // compile for A
michael@0:     {
michael@0:         JSAutoCompartment a(cx, A);
michael@0:         JSFunction *fun;
michael@0:         JS::CompileOptions options(cx);
michael@0:         options.setFileAndLine(__FILE__, 1);
michael@0:         CHECK(fun = JS_CompileFunction(cx, A, "f", 0, nullptr, source,
michael@0:                                        strlen(source), options));
michael@0:         CHECK(obj = JS_GetFunctionObject(fun));
michael@0:     }
michael@0: 
michael@0:     // clone into B
michael@0:     {
michael@0:         JSAutoCompartment b(cx, B);
michael@0:         CHECK(JS_CloneFunctionObject(cx, obj, B));
michael@0:     }
michael@0: 
michael@0:     return true;
michael@0: }
michael@0: END_TEST(test_cloneScript)
michael@0: 
michael@0: static void
michael@0: DestroyPrincipals(JSPrincipals *principals)
michael@0: {
michael@0:     delete principals;
michael@0: }
michael@0: 
michael@0: struct Principals : public JSPrincipals
michael@0: {
michael@0:   public:
michael@0:     Principals()
michael@0:     {
michael@0:         refcount = 0;
michael@0:     }
michael@0: };
michael@0: 
michael@0: class AutoDropPrincipals
michael@0: {
michael@0:     JSRuntime *rt;
michael@0:     JSPrincipals *principals;
michael@0: 
michael@0:   public:
michael@0:     AutoDropPrincipals(JSRuntime *rt, JSPrincipals *principals)
michael@0:       : rt(rt), principals(principals)
michael@0:     {
michael@0:         JS_HoldPrincipals(principals);
michael@0:     }
michael@0: 
michael@0:     ~AutoDropPrincipals()
michael@0:     {
michael@0:         JS_DropPrincipals(rt, principals);
michael@0:     }
michael@0: };
michael@0: 
michael@0: BEGIN_TEST(test_cloneScriptWithPrincipals)
michael@0: {
michael@0:     JS_InitDestroyPrincipalsCallback(rt, DestroyPrincipals);
michael@0: 
michael@0:     JSPrincipals *principalsA = new Principals();
michael@0:     AutoDropPrincipals dropA(rt, principalsA);
michael@0:     JSPrincipals *principalsB = new Principals();
michael@0:     AutoDropPrincipals dropB(rt, principalsB);
michael@0: 
michael@0:     JS::RootedObject A(cx, createGlobal(principalsA));
michael@0:     JS::RootedObject B(cx, createGlobal(principalsB));
michael@0: 
michael@0:     CHECK(A);
michael@0:     CHECK(B);
michael@0: 
michael@0:     const char *argnames[] = { "arg" };
michael@0:     const char *source = "return function() { return arg; }";
michael@0: 
michael@0:     JS::RootedObject obj(cx);
michael@0: 
michael@0:     // Compile in A
michael@0:     {
michael@0:         JSAutoCompartment a(cx, A);
michael@0:         JS::CompileOptions options(cx);
michael@0:         options.setFileAndLine(__FILE__, 1);
michael@0:         JS::RootedFunction fun(cx, JS_CompileFunction(cx, A, "f",
michael@0:                 mozilla::ArrayLength(argnames), argnames, source,
michael@0:                 strlen(source), options));
michael@0:         CHECK(fun);
michael@0: 
michael@0:         JSScript *script;
michael@0:         CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0: 
michael@0:         CHECK(JS_GetScriptPrincipals(script) == principalsA);
michael@0:         CHECK(obj = JS_GetFunctionObject(fun));
michael@0:     }
michael@0: 
michael@0:     // Clone into B
michael@0:     {
michael@0:         JSAutoCompartment b(cx, B);
michael@0:         JS::RootedObject cloned(cx);
michael@0:         CHECK(cloned = JS_CloneFunctionObject(cx, obj, B));
michael@0: 
michael@0:         JS::RootedFunction fun(cx);
michael@0:         JS::RootedValue clonedValue(cx, JS::ObjectValue(*cloned));
michael@0:         CHECK(fun = JS_ValueToFunction(cx, clonedValue));
michael@0: 
michael@0:         JSScript *script;
michael@0:         CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0: 
michael@0:         CHECK(JS_GetScriptPrincipals(script) == principalsB);
michael@0: 
michael@0:         JS::RootedValue v(cx);
michael@0:         JS::RootedValue arg(cx, JS::Int32Value(1));
michael@0:         CHECK(JS_CallFunctionValue(cx, B, clonedValue, arg, &v));
michael@0:         CHECK(v.isObject());
michael@0: 
michael@0:         JSObject *funobj = &v.toObject();
michael@0:         CHECK(JS_ObjectIsFunction(cx, funobj));
michael@0:         CHECK(fun = JS_ValueToFunction(cx, v));
michael@0:         CHECK(script = JS_GetFunctionScript(cx, fun));
michael@0:         CHECK(JS_GetScriptPrincipals(script) == principalsB);
michael@0:     }
michael@0: 
michael@0:     return true;
michael@0: }
michael@0: END_TEST(test_cloneScriptWithPrincipals)