1 | /* |
2 | * Copyright (C) 2013, 2016 Apple, Inc. All rights reserved. |
3 | * |
4 | * Redistribution and use in source and binary forms, with or without |
5 | * modification, are permitted provided that the following conditions |
6 | * are met: |
7 | * 1. Redistributions of source code must retain the above copyright |
8 | * notice, this list of conditions and the following disclaimer. |
9 | * 2. Redistributions in binary form must reproduce the above copyright |
10 | * notice, this list of conditions and the following disclaimer in the |
11 | * documentation and/or other materials provided with the distribution. |
12 | * |
13 | * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY |
14 | * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
15 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
16 | * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR |
17 | * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
18 | * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
19 | * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
20 | * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY |
21 | * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
22 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
23 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
24 | */ |
25 | |
26 | #include "config.h" |
27 | #include "WeakMapPrototype.h" |
28 | |
29 | #include "JSCInlines.h" |
30 | #include "JSWeakMap.h" |
31 | |
32 | namespace JSC { |
33 | |
34 | const ClassInfo WeakMapPrototype::s_info = { "WeakMap" , &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(WeakMapPrototype) }; |
35 | |
36 | static EncodedJSValue JSC_HOST_CALL protoFuncWeakMapDelete(ExecState*); |
37 | static EncodedJSValue JSC_HOST_CALL protoFuncWeakMapGet(ExecState*); |
38 | static EncodedJSValue JSC_HOST_CALL protoFuncWeakMapHas(ExecState*); |
39 | static EncodedJSValue JSC_HOST_CALL protoFuncWeakMapSet(ExecState*); |
40 | |
41 | void WeakMapPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject) |
42 | { |
43 | Base::finishCreation(vm); |
44 | ASSERT(inherits(vm, info())); |
45 | |
46 | JSC_NATIVE_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->deleteKeyword, protoFuncWeakMapDelete, static_cast<unsigned>(PropertyAttribute::DontEnum), 1); |
47 | JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->get, protoFuncWeakMapGet, static_cast<unsigned>(PropertyAttribute::DontEnum), 1, JSWeakMapGetIntrinsic); |
48 | JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->has, protoFuncWeakMapHas, static_cast<unsigned>(PropertyAttribute::DontEnum), 1, JSWeakMapHasIntrinsic); |
49 | JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->set, protoFuncWeakMapSet, static_cast<unsigned>(PropertyAttribute::DontEnum), 2, JSWeakMapSetIntrinsic); |
50 | |
51 | putDirectWithoutTransition(vm, vm.propertyNames->toStringTagSymbol, jsString(&vm, "WeakMap" ), PropertyAttribute::DontEnum | PropertyAttribute::ReadOnly); |
52 | } |
53 | |
54 | ALWAYS_INLINE static JSWeakMap* getWeakMap(CallFrame* callFrame, JSValue value) |
55 | { |
56 | VM& vm = callFrame->vm(); |
57 | auto scope = DECLARE_THROW_SCOPE(vm); |
58 | |
59 | if (UNLIKELY(!value.isObject())) { |
60 | throwTypeError(callFrame, scope, "Called WeakMap function on non-object"_s ); |
61 | return nullptr; |
62 | } |
63 | |
64 | auto* map = jsDynamicCast<JSWeakMap*>(vm, asObject(value)); |
65 | if (LIKELY(map)) |
66 | return map; |
67 | |
68 | throwTypeError(callFrame, scope, "Called WeakMap function on a non-WeakMap object"_s ); |
69 | return nullptr; |
70 | } |
71 | |
72 | EncodedJSValue JSC_HOST_CALL protoFuncWeakMapDelete(CallFrame* callFrame) |
73 | { |
74 | auto* map = getWeakMap(callFrame, callFrame->thisValue()); |
75 | if (!map) |
76 | return JSValue::encode(jsUndefined()); |
77 | JSValue key = callFrame->argument(0); |
78 | return JSValue::encode(jsBoolean(key.isObject() && map->remove(asObject(key)))); |
79 | } |
80 | |
81 | EncodedJSValue JSC_HOST_CALL protoFuncWeakMapGet(CallFrame* callFrame) |
82 | { |
83 | auto* map = getWeakMap(callFrame, callFrame->thisValue()); |
84 | if (!map) |
85 | return JSValue::encode(jsUndefined()); |
86 | JSValue key = callFrame->argument(0); |
87 | if (!key.isObject()) |
88 | return JSValue::encode(jsUndefined()); |
89 | return JSValue::encode(map->get(asObject(key))); |
90 | } |
91 | |
92 | EncodedJSValue JSC_HOST_CALL protoFuncWeakMapHas(CallFrame* callFrame) |
93 | { |
94 | auto* map = getWeakMap(callFrame, callFrame->thisValue()); |
95 | if (!map) |
96 | return JSValue::encode(jsUndefined()); |
97 | JSValue key = callFrame->argument(0); |
98 | return JSValue::encode(jsBoolean(key.isObject() && map->has(asObject(key)))); |
99 | } |
100 | |
101 | EncodedJSValue JSC_HOST_CALL protoFuncWeakMapSet(CallFrame* callFrame) |
102 | { |
103 | VM& vm = callFrame->vm(); |
104 | auto scope = DECLARE_THROW_SCOPE(vm); |
105 | |
106 | auto* map = getWeakMap(callFrame, callFrame->thisValue()); |
107 | EXCEPTION_ASSERT(!!scope.exception() == !map); |
108 | if (!map) |
109 | return JSValue::encode(jsUndefined()); |
110 | JSValue key = callFrame->argument(0); |
111 | if (!key.isObject()) |
112 | return JSValue::encode(throwTypeError(callFrame, scope, "Attempted to set a non-object key in a WeakMap"_s )); |
113 | map->set(vm, asObject(key), callFrame->argument(1)); |
114 | return JSValue::encode(callFrame->thisValue()); |
115 | } |
116 | |
117 | } |
118 | |