michael@0: /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* vim: set ts=8 sts=2 et sw=2 tw=80: */
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 "xpcprivate.h"
michael@0: #include "XPCWrapper.h"
michael@0: #include "WrapperFactory.h"
michael@0: #include "AccessCheck.h"
michael@0: 
michael@0: using namespace xpc;
michael@0: using namespace mozilla;
michael@0: 
michael@0: namespace XPCNativeWrapper {
michael@0: 
michael@0: static inline
michael@0: bool
michael@0: ThrowException(nsresult ex, JSContext *cx)
michael@0: {
michael@0:   XPCThrower::Throw(ex, cx);
michael@0: 
michael@0:   return false;
michael@0: }
michael@0: 
michael@0: static bool
michael@0: UnwrapNW(JSContext *cx, unsigned argc, jsval *vp)
michael@0: {
michael@0:   JS::CallArgs args = JS::CallArgsFromVp(argc, vp);
michael@0:   if (args.length() != 1) {
michael@0:     return ThrowException(NS_ERROR_XPC_NOT_ENOUGH_ARGS, cx);
michael@0:   }
michael@0: 
michael@0:   JS::RootedValue v(cx, args[0]);
michael@0:   if (!v.isObject() || !js::IsWrapper(&v.toObject())) {
michael@0:     args.rval().set(v);
michael@0:     return true;
michael@0:   }
michael@0: 
michael@0:   if (AccessCheck::wrapperSubsumes(&v.toObject())) {
michael@0:     bool ok = xpc::WrapperFactory::WaiveXrayAndWrap(cx, &v);
michael@0:     NS_ENSURE_TRUE(ok, false);
michael@0:   }
michael@0: 
michael@0:   args.rval().set(v);
michael@0:   return true;
michael@0: }
michael@0: 
michael@0: static bool
michael@0: XrayWrapperConstructor(JSContext *cx, unsigned argc, jsval *vp)
michael@0: {
michael@0:   JS::CallArgs args = CallArgsFromVp(argc, vp);
michael@0:   if (args.length() == 0) {
michael@0:     return ThrowException(NS_ERROR_XPC_NOT_ENOUGH_ARGS, cx);
michael@0:   }
michael@0: 
michael@0:   if (!args[0].isObject()) {
michael@0:     args.rval().set(args[0]);
michael@0:     return true;
michael@0:   }
michael@0: 
michael@0:   args.rval().setObject(*js::UncheckedUnwrap(&args[0].toObject()));
michael@0:   return JS_WrapValue(cx, args.rval());
michael@0: }
michael@0: // static
michael@0: bool
michael@0: AttachNewConstructorObject(JSContext *aCx, JS::HandleObject aGlobalObject)
michael@0: {
michael@0:   // Pushing a JSContext calls ActivateDebugger which calls this function, so
michael@0:   // we can't use an AutoJSContext here until JSD is gone.
michael@0:   JSAutoCompartment ac(aCx, aGlobalObject);
michael@0:   JSFunction *xpcnativewrapper =
michael@0:     JS_DefineFunction(aCx, aGlobalObject, "XPCNativeWrapper",
michael@0:                       XrayWrapperConstructor, 1,
michael@0:                       JSPROP_READONLY | JSPROP_PERMANENT | JSFUN_STUB_GSOPS | JSFUN_CONSTRUCTOR);
michael@0:   if (!xpcnativewrapper) {
michael@0:     return false;
michael@0:   }
michael@0:   JS::RootedObject obj(aCx, JS_GetFunctionObject(xpcnativewrapper));
michael@0:   return JS_DefineFunction(aCx, obj, "unwrap", UnwrapNW, 1,
michael@0:                            JSPROP_READONLY | JSPROP_PERMANENT) != nullptr;
michael@0: }
michael@0: 
michael@0: } // namespace XPCNativeWrapper
michael@0: 
michael@0: namespace XPCWrapper {
michael@0: 
michael@0: JSObject *
michael@0: UnsafeUnwrapSecurityWrapper(JSObject *obj)
michael@0: {
michael@0:   if (js::IsProxy(obj)) {
michael@0:     return js::UncheckedUnwrap(obj);
michael@0:   }
michael@0: 
michael@0:   return obj;
michael@0: }
michael@0: 
michael@0: } // namespace XPCWrapper