runtime/WeakSetConstructor.cpp

   1 /*
   2  * Copyright (C) 2015 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 "WeakSetConstructor.h"
  28
  29 #include "Error.h"
  30 #include "IteratorOperations.h"
  31 #include "JSCJSValueInlines.h"
  32 #include "JSCellInlines.h"
  33 #include "JSGlobalObject.h"
  34 #include "JSWeakSet.h"
  35 #include "StructureInlines.h"
  36 #include "WeakSetPrototype.h"
  37
  38 namespace JSC {
  39
  40 const ClassInfo WeakSetConstructor::s_info = { "Function", &Base::s_info, 0, CREATE_METHOD_TABLE(WeakSetConstructor) };
  41
  42 void WeakSetConstructor::finishCreation(VM& vm, WeakSetPrototype* 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 callWeakSet(ExecState* exec)
  50 {
  51     return JSValue::encode(throwTypeError(exec, ASCIILiteral("WeakSet cannot be called as a function")));
  52 }
  53
  54 static EncodedJSValue JSC_HOST_CALL constructWeakSet(ExecState* exec)
  55 {
  56     JSGlobalObject* globalObject = asInternalFunction(exec->callee())->globalObject();
  57     Structure* weakSetStructure = globalObject->weakSetStructure();
  58     JSWeakSet* weakSet = JSWeakSet::create(exec, weakSetStructure);
  59     JSValue iterable = exec->argument(0);
  60     if (iterable.isUndefinedOrNull())
  61         return JSValue::encode(weakSet);
  62
  63     JSValue adderFunction = weakSet->JSObject::get(exec, exec->propertyNames().add);
  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(weakSet);
  96
  97         JSValue nextValue = iteratorValue(exec, next);
  98         if (exec->hadException())
  99             return JSValue::encode(jsUndefined());
 100
 101         MarkedArgumentBuffer arguments;
 102         arguments.append(nextValue);
 103         call(exec, adderFunction, adderFunctionCallType, adderFunctionCallData, weakSet, arguments);
 104         if (exec->hadException()) {
 105             iteratorClose(exec, iterator);
 106             return JSValue::encode(jsUndefined());
 107         }
 108     }
 109     RELEASE_ASSERT_NOT_REACHED();
 110     return JSValue::encode(weakSet);
 111 }
 112
 113 ConstructType WeakSetConstructor::getConstructData(JSCell*, ConstructData& constructData)
 114 {
 115     constructData.native.function = constructWeakSet;
 116     return ConstructTypeHost;
 117 }
 118
 119 CallType WeakSetConstructor::getCallData(JSCell*, CallData& callData)
 120 {
 121     callData.native.function = callWeakSet;
 122     return CallTypeHost;
 123 }
 124
 125 }