runtime/WeakMapConstructor.cpp

   1 /*
   2  * Copyright (C) 2013 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 "Error.h"
  30 #include "IteratorOperations.h"
  31 #include "JSCJSValueInlines.h"
  32 #include "JSCellInlines.h"
  33 #include "JSGlobalObject.h"
  34 #include "JSWeakMap.h"
  35 #include "StructureInlines.h"
  36 #include "WeakMapPrototype.h"
  37
  38 namespace JSC {
  39
  40 const ClassInfo WeakMapConstructor::s_info = { "Function", &Base::s_info, 0, CREATE_METHOD_TABLE(WeakMapConstructor) };
  41
  42 void WeakMapConstructor::finishCreation(VM& vm, WeakMapPrototype* prototype)
  43 {
  44     Base::finishCreation(vm, prototype->classInfo()->className);
  45     putDirectWithoutTransition(vm, vm.propertyNames->prototype, prototype, DontEnum | DontDelete | ReadOnly);
  46     putDirectWithoutTransition(vm, vm.propertyNames->length, jsNumber(0), ReadOnly | DontEnum | DontDelete);
  47 }
  48
  49 static EncodedJSValue JSC_HOST_CALL callWeakMap(ExecState* exec)
  50 {
  51     return JSValue::encode(throwTypeError(exec, ASCIILiteral("WeakMap cannot be called as a function")));
  52 }
  53
  54 static EncodedJSValue JSC_HOST_CALL constructWeakMap(ExecState* exec)
  55 {
  56     JSGlobalObject* globalObject = asInternalFunction(exec->callee())->globalObject();
  57     Structure* weakMapStructure = globalObject->weakMapStructure();
  58     JSWeakMap* weakMap = JSWeakMap::create(exec, weakMapStructure);
  59     JSValue iterable = exec->argument(0);
  60     if (iterable.isUndefinedOrNull())
  61         return JSValue::encode(weakMap);
  62
  63     JSValue adderFunction = weakMap->JSObject::get(exec, exec->propertyNames().set);
  64     if (exec->hadException())
  65         return JSValue::encode(jsUndefined());
  66
  67     CallData adderFunctionCallData;
  68     CallType adderFunctionCallType = getCallData(adderFunction, adderFunctionCallData);
  69     if (adderFunctionCallType == CallTypeNone)
  70         return JSValue::encode(throwTypeError(exec));
  71
  72     JSValue iteratorFunction = iterable.get(exec, exec->propertyNames().iteratorSymbol);
  73     if (exec->hadException())
  74         return JSValue::encode(jsUndefined());
  75
  76     CallData iteratorFunctionCallData;
  77     CallType iteratorFunctionCallType = getCallData(iteratorFunction, iteratorFunctionCallData);
  78     if (iteratorFunctionCallType == CallTypeNone)
  79         return JSValue::encode(throwTypeError(exec));
  80
  81     ArgList iteratorFunctionArguments;
  82     JSValue iterator = call(exec, iteratorFunction, iteratorFunctionCallType, iteratorFunctionCallData, iterable, iteratorFunctionArguments);
  83     if (exec->hadException())
  84         return JSValue::encode(jsUndefined());
  85
  86     if (!iterator.isObject())
  87         return JSValue::encode(throwTypeError(exec));
  88
  89     while (true) {
  90         JSValue next = iteratorStep(exec, iterator);
  91         if (exec->hadException())
  92             return JSValue::encode(jsUndefined());
  93
  94         if (next.isFalse())
  95             return JSValue::encode(weakMap);
  96
  97         JSValue nextItem = iteratorValue(exec, next);
  98         if (exec->hadException())
  99             return JSValue::encode(jsUndefined());
 100
 101         if (!nextItem.isObject()) {
 102             throwTypeError(exec);
 103             iteratorClose(exec, iterator);
 104             return JSValue::encode(jsUndefined());
 105         }
 106
 107         JSValue key = nextItem.get(exec, 0);
 108         if (exec->hadException()) {
 109             iteratorClose(exec, iterator);
 110             return JSValue::encode(jsUndefined());
 111         }
 112
 113         JSValue value = nextItem.get(exec, 1);
 114         if (exec->hadException()) {
 115             iteratorClose(exec, iterator);
 116             return JSValue::encode(jsUndefined());
 117         }
 118
 119         MarkedArgumentBuffer arguments;
 120         arguments.append(key);
 121         arguments.append(value);
 122         call(exec, adderFunction, adderFunctionCallType, adderFunctionCallData, weakMap, arguments);
 123         if (exec->hadException()) {
 124             iteratorClose(exec, iterator);
 125             return JSValue::encode(jsUndefined());
 126         }
 127     }
 128     RELEASE_ASSERT_NOT_REACHED();
 129     return JSValue::encode(weakMap);
 130 }
 131
 132 ConstructType WeakMapConstructor::getConstructData(JSCell*, ConstructData& constructData)
 133 {
 134     constructData.native.function = constructWeakMap;
 135     return ConstructTypeHost;
 136 }
 137
 138 CallType WeakMapConstructor::getCallData(JSCell*, CallData& callData)
 139 {
 140     callData.native.function = callWeakMap;
 141     return CallTypeHost;
 142 }
 143
 144 }