[JSC] Implement optimized WeakMap and WeakSet
[WebKit-https.git] / Source / JavaScriptCore / runtime / WeakSetPrototype.cpp
1 /*
2  * Copyright (C) 2015-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 "WeakSetPrototype.h"
28
29 #include "JSCInlines.h"
30 #include "JSWeakSet.h"
31
32 namespace JSC {
33
34 const ClassInfo WeakSetPrototype::s_info = { "WeakSet", &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(WeakSetPrototype) };
35
36 static EncodedJSValue JSC_HOST_CALL protoFuncWeakSetDelete(ExecState*);
37 static EncodedJSValue JSC_HOST_CALL protoFuncWeakSetHas(ExecState*);
38 static EncodedJSValue JSC_HOST_CALL protoFuncWeakSetAdd(ExecState*);
39
40 void WeakSetPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject)
41 {
42     Base::finishCreation(vm);
43     ASSERT(inherits(vm, info()));
44     didBecomePrototype();
45
46     JSC_NATIVE_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->deleteKeyword, protoFuncWeakSetDelete, static_cast<unsigned>(PropertyAttribute::DontEnum), 1);
47     JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->has, protoFuncWeakSetHas, static_cast<unsigned>(PropertyAttribute::DontEnum), 1, JSWeakSetHasIntrinsic);
48     JSC_NATIVE_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->add, protoFuncWeakSetAdd, static_cast<unsigned>(PropertyAttribute::DontEnum), 1);
49
50     putDirectWithoutTransition(vm, vm.propertyNames->toStringTagSymbol, jsString(&vm, "WeakSet"), PropertyAttribute::DontEnum | PropertyAttribute::ReadOnly);
51 }
52
53 ALWAYS_INLINE static JSWeakSet* getWeakSet(CallFrame* callFrame, JSValue value)
54 {
55     VM& vm = callFrame->vm();
56     auto scope = DECLARE_THROW_SCOPE(vm);
57
58     if (UNLIKELY(!value.isObject())) {
59         throwTypeError(callFrame, scope, WTF::ASCIILiteral("Called WeakSet function on non-object"));
60         return nullptr;
61     }
62
63     if (LIKELY(isJSWeakSet(asObject(value))))
64         return jsCast<JSWeakSet*>(value);
65
66     throwTypeError(callFrame, scope, WTF::ASCIILiteral("Called WeakSet function on a non-WeakSet object"));
67     return nullptr;
68 }
69
70 EncodedJSValue JSC_HOST_CALL protoFuncWeakSetDelete(CallFrame* callFrame)
71 {
72     auto* set = getWeakSet(callFrame, callFrame->thisValue());
73     if (!set)
74         return JSValue::encode(jsUndefined());
75     JSValue key = callFrame->argument(0);
76     return JSValue::encode(jsBoolean(key.isObject() && set->remove(asObject(key))));
77 }
78
79 EncodedJSValue JSC_HOST_CALL protoFuncWeakSetHas(CallFrame* callFrame)
80 {
81     auto* set = getWeakSet(callFrame, callFrame->thisValue());
82     if (!set)
83         return JSValue::encode(jsUndefined());
84     JSValue key = callFrame->argument(0);
85     return JSValue::encode(jsBoolean(key.isObject() && set->has(asObject(key))));
86 }
87
88 EncodedJSValue JSC_HOST_CALL protoFuncWeakSetAdd(CallFrame* callFrame)
89 {
90     VM& vm = callFrame->vm();
91     auto scope = DECLARE_THROW_SCOPE(vm);
92
93     auto* set = getWeakSet(callFrame, callFrame->thisValue());
94     EXCEPTION_ASSERT(!!scope.exception() == !set);
95     if (!set)
96         return JSValue::encode(jsUndefined());
97     JSValue key = callFrame->argument(0);
98     if (!key.isObject())
99         return JSValue::encode(throwTypeError(callFrame, scope, WTF::ASCIILiteral("Attempted to add a non-object key to a WeakSet")));
100     set->add(vm, asObject(key));
101     return JSValue::encode(callFrame->thisValue());
102 }
103
104 }