1  /*
  2   * Copyright (C) 2015-2019 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 JSC_DECLARE_HOST_FUNCTION(protoFuncWeakSetDelete);
 37  static JSC_DECLARE_HOST_FUNCTION(protoFuncWeakSetHas);
 38  static JSC_DECLARE_HOST_FUNCTION(protoFuncWeakSetAdd);
 39  
 40  void WeakSetPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject)
 41  {
 42      Base::finishCreation(vm);
 43      ASSERT(inherits(vm, info()));
 44  
 45      JSC_NATIVE_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->deleteKeyword, protoFuncWeakSetDelete, static_cast<unsigned>(PropertyAttribute::DontEnum), 1);
 46      JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->has, protoFuncWeakSetHas, static_cast<unsigned>(PropertyAttribute::DontEnum), 1, JSWeakSetHasIntrinsic);
 47      JSC_NATIVE_INTRINSIC_FUNCTION_WITHOUT_TRANSITION(vm.propertyNames->add, protoFuncWeakSetAdd, static_cast<unsigned>(PropertyAttribute::DontEnum), 1, JSWeakSetAddIntrinsic);
 48  
 49      JSC_TO_STRING_TAG_WITHOUT_TRANSITION();
 50  }
 51  
 52  ALWAYS_INLINE static JSWeakSet* getWeakSet(JSGlobalObject* globalObject, JSValue value)
 53  {
 54      VM& vm = globalObject->vm();
 55      auto scope = DECLARE_THROW_SCOPE(vm);
 56  
 57      if (UNLIKELY(!value.isObject())) {
 58          throwTypeError(globalObject, scope, "Called WeakSet function on non-object"_s);
 59          return nullptr;
 60      }
 61  
 62      auto* set = jsDynamicCast<JSWeakSet*>(vm, asObject(value));
 63      if (LIKELY(set))
 64          return set;
 65  
 66      throwTypeError(globalObject, scope, "Called WeakSet function on a non-WeakSet object"_s);
 67      return nullptr;
 68  }
 69  
 70  JSC_DEFINE_HOST_FUNCTION(protoFuncWeakSetDelete, (JSGlobalObject* globalObject, CallFrame* callFrame))
 71  {
 72      auto* set = getWeakSet(globalObject, 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  JSC_DEFINE_HOST_FUNCTION(protoFuncWeakSetHas, (JSGlobalObject* globalObject, CallFrame* callFrame))
 80  {
 81      auto* set = getWeakSet(globalObject, 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  JSC_DEFINE_HOST_FUNCTION(protoFuncWeakSetAdd, (JSGlobalObject* globalObject, CallFrame* callFrame))
 89  {
 90      VM& vm = globalObject->vm();
 91      auto scope = DECLARE_THROW_SCOPE(vm);
 92  
 93      auto* set = getWeakSet(globalObject, 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(globalObject, scope, "Attempted to add a non-object key to a WeakSet"_s));
100      set->add(vm, asObject(key));
101      return JSValue::encode(callFrame->thisValue());
102  }
103  
104  }