1  /*
  2   * Copyright (C) 2013-2017 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 "WeakMapConstructor.h"
 28  
 29  #include "IteratorOperations.h"
 30  #include "JSCInlines.h"
 31  #include "JSWeakMap.h"
 32  #include "WeakMapPrototype.h"
 33  
 34  namespace JSC {
 35  
 36  const ClassInfo WeakMapConstructor::s_info = { "Function", &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(WeakMapConstructor) };
 37  
 38  void WeakMapConstructor::finishCreation(VM& vm, WeakMapPrototype* prototype)
 39  {
 40      Base::finishCreation(vm, 0, "WeakMap"_s, PropertyAdditionMode::WithoutStructureTransition);
 41      putDirectWithoutTransition(vm, vm.propertyNames->prototype, prototype, PropertyAttribute::DontEnum | PropertyAttribute::DontDelete | PropertyAttribute::ReadOnly);
 42  }
 43  
 44  static JSC_DECLARE_HOST_FUNCTION(callWeakMap);
 45  static JSC_DECLARE_HOST_FUNCTION(constructWeakMap);
 46  
 47  WeakMapConstructor::WeakMapConstructor(VM& vm, Structure* structure)
 48      : Base(vm, structure, callWeakMap, constructWeakMap)
 49  {
 50  }
 51  
 52  JSC_DEFINE_HOST_FUNCTION(callWeakMap, (JSGlobalObject* globalObject, CallFrame*))
 53  {
 54      VM& vm = globalObject->vm();
 55      auto scope = DECLARE_THROW_SCOPE(vm);
 56      return JSValue::encode(throwConstructorCannotBeCalledAsFunctionTypeError(globalObject, scope, "WeakMap"));
 57  }
 58  
 59  JSC_DEFINE_HOST_FUNCTION(constructWeakMap, (JSGlobalObject* globalObject, CallFrame* callFrame))
 60  {
 61      VM& vm = globalObject->vm();
 62      auto scope = DECLARE_THROW_SCOPE(vm);
 63  
 64      JSObject* newTarget = asObject(callFrame->newTarget());
 65      Structure* weakMapStructure = newTarget == callFrame->jsCallee()
 66          ? globalObject->weakMapStructure()
 67          : InternalFunction::createSubclassStructure(globalObject, newTarget, getFunctionRealm(vm, newTarget)->weakMapStructure());
 68      RETURN_IF_EXCEPTION(scope, { });
 69  
 70      JSWeakMap* weakMap = JSWeakMap::create(vm, weakMapStructure);
 71      JSValue iterable = callFrame->argument(0);
 72      if (iterable.isUndefinedOrNull())
 73          return JSValue::encode(weakMap);
 74  
 75      JSValue adderFunction = weakMap->JSObject::get(globalObject, vm.propertyNames->set);
 76      RETURN_IF_EXCEPTION(scope, encodedJSValue());
 77  
 78      auto adderFunctionCallData = getCallData(vm, adderFunction);
 79      if (adderFunctionCallData.type == CallData::Type::None)
 80          return JSValue::encode(throwTypeError(globalObject, scope));
 81  
 82      scope.release();
 83      forEachInIterable(globalObject, iterable, [&](VM& vm, JSGlobalObject* globalObject, JSValue nextItem) {
 84          auto scope = DECLARE_THROW_SCOPE(vm);
 85          if (!nextItem.isObject()) {
 86              throwTypeError(globalObject, scope);
 87              return;
 88          }
 89  
 90          JSValue key = nextItem.get(globalObject, static_cast<unsigned>(0));
 91          RETURN_IF_EXCEPTION(scope, void());
 92  
 93          JSValue value = nextItem.get(globalObject, static_cast<unsigned>(1));
 94          RETURN_IF_EXCEPTION(scope, void());
 95  
 96          MarkedArgumentBuffer arguments;
 97          arguments.append(key);
 98          arguments.append(value);
 99          ASSERT(!arguments.hasOverflowed());
100          scope.release();
101          call(globalObject, adderFunction, adderFunctionCallData, weakMap, arguments);
102      });
103  
104      return JSValue::encode(weakMap);
105  }
106  
107  }