2c8fa150bd0688c88fbf69a510cca25fd18445cd
[WebKit-https.git] / Source / JavaScriptCore / runtime / ArrayIteratorPrototype.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 "ArrayIteratorPrototype.h"
28
29 #include "JSArrayIterator.h"
30 #include "JSCJSValueInlines.h"
31 #include "JSCellInlines.h"
32 #include "JSGlobalObject.h"
33 #include "ObjectConstructor.h"
34
35 namespace JSC {
36
37 const ClassInfo ArrayIteratorPrototype::s_info = { "Array Iterator", &Base::s_info, 0, 0, CREATE_METHOD_TABLE(ArrayIteratorPrototype) };
38
39 static EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeNext(ExecState*);
40
41 void ArrayIteratorPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject)
42 {
43     Base::finishCreation(vm);
44     ASSERT(inherits(info()));
45     vm.prototypeMap.addPrototype(this);
46
47     JSC_NATIVE_FUNCTION(vm.propertyNames->next, arrayIteratorPrototypeNext, DontEnum, 0);
48 }
49
50 static EncodedJSValue createIteratorResult(CallFrame* callFrame, ArrayIterationKind kind, size_t index, JSValue result, bool done)
51 {
52     JSGlobalObject* globalObject = callFrame->callee()->globalObject();
53     JSObject* resultObject = constructEmptyObject(callFrame);
54     resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().done, jsBoolean(done));
55     switch (kind & ~ArrayIterateSparseTag) {
56     case ArrayIterateKey:
57         resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, done ? jsUndefined() : jsNumber(index));
58         break;
59     case ArrayIterateValue:
60         resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, done ? jsUndefined() : result);
61         break;
62     case ArrayIterateKeyValue: {
63         if (!done) {
64             MarkedArgumentBuffer args;
65             args.append(jsNumber(index));
66             args.append(result);
67             JSArray* resultArray = constructArray(callFrame, 0, globalObject, args);
68             resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, resultArray);
69         } else
70             resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, jsUndefined());
71
72         break;
73     }
74     default:
75         RELEASE_ASSERT_NOT_REACHED();
76     }
77     return JSValue::encode(resultObject);
78 }
79     
80 EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeNext(CallFrame* callFrame)
81 {
82     JSArrayIterator* iterator = jsDynamicCast<JSArrayIterator*>(callFrame->thisValue());
83     if (!iterator)
84         throwTypeError(callFrame, ASCIILiteral("Cannot call ArrayIterator.next() on a non-ArrayIterator object"));
85     JSObject* iteratedObject = iterator->iteratedObject();
86     size_t index = iterator->nextIndex();
87     ArrayIterationKind kind = iterator->iterationKind();
88     JSValue jsLength = JSValue(iteratedObject).get(callFrame, callFrame->propertyNames().length);
89     if (callFrame->hadException())
90         return JSValue::encode(jsNull());
91
92     size_t length = jsLength.toUInt32(callFrame);
93     if (callFrame->hadException())
94         return JSValue::encode(jsNull());
95
96     if (index >= length) {
97         iterator->finish();
98         return createIteratorResult(callFrame, kind, index, jsUndefined(), true);
99     }
100     if (JSValue result = iteratedObject->tryGetIndexQuickly(index)) {
101         iterator->setNextIndex(index + 1);
102         return createIteratorResult(callFrame, kind, index, result, false);
103     }
104     
105     JSValue result = jsUndefined();
106     PropertySlot slot(iteratedObject);
107     if (kind > ArrayIterateSparseTag) {
108         // We assume that the indexed property will be an own property so cache the getOwnProperty
109         // method locally
110         auto getOwnPropertySlotByIndex = iteratedObject->methodTable()->getOwnPropertySlotByIndex;
111         while (index < length) {
112             if (getOwnPropertySlotByIndex(iteratedObject, callFrame, index, slot)) {
113                 result = slot.getValue(callFrame, index);
114                 break;
115             }
116             if (iteratedObject->getPropertySlot(callFrame, index, slot)) {
117                 result = slot.getValue(callFrame, index);
118                 break;
119             }
120             index++;
121         }
122     } else if (iteratedObject->getPropertySlot(callFrame, index, slot))
123         result = slot.getValue(callFrame, index);
124
125     if (index == length)
126         iterator->finish();
127     else
128         iterator->setNextIndex(index + 1);
129     return createIteratorResult(callFrame, kind, index, jsUndefined(), index == length);
130 }
131
132 }