JSArray.cpp 40.3 KB
Newer Older
darin's avatar
darin committed
1 2
/*
 *  Copyright (C) 1999-2000 Harri Porten (porten@kde.org)
3
 *  Copyright (C) 2003, 2007, 2008, 2009, 2012 Apple Inc. All rights reserved.
darin's avatar
darin committed
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 *  Copyright (C) 2003 Peter Kelly (pmk@post.com)
 *  Copyright (C) 2006 Alexey Proskuryakov (ap@nypop.com)
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#include "config.h"
darin@apple.com's avatar
darin@apple.com committed
24
#include "JSArray.h"
darin's avatar
darin committed
25

darin@apple.com's avatar
darin@apple.com committed
26
#include "ArrayPrototype.h"
27
#include "ButterflyInlineMethods.h"
mhahnenberg@apple.com's avatar
mhahnenberg@apple.com committed
28 29
#include "CopiedSpace.h"
#include "CopiedSpaceInlineMethods.h"
30
#include "CachedCall.h"
31
#include "Error.h"
32
#include "Executable.h"
33
#include "GetterSetter.h"
34
#include "IndexingHeaderInlineMethods.h"
darin's avatar
darin committed
35
#include "PropertyNameArray.h"
36
#include "Reject.h"
ap@webkit.org's avatar
ap@webkit.org committed
37
#include <wtf/AVLTree.h>
38
#include <wtf/Assertions.h>
39
#include <wtf/OwnPtr.h>
40
#include <Operations.h>
darin's avatar
darin committed
41

42
using namespace std;
ap@webkit.org's avatar
ap@webkit.org committed
43
using namespace WTF;
bdash's avatar
bdash committed
44

45
namespace JSC {
darin's avatar
darin committed
46

47

ggaren@apple.com's avatar
ggaren@apple.com committed
48
ASSERT_CLASS_FITS_IN_CELL(JSArray);
49
ASSERT_HAS_TRIVIAL_DESTRUCTOR(JSArray);
ggaren@apple.com's avatar
ggaren@apple.com committed
50

51
const ClassInfo JSArray::s_info = {"Array", &JSNonFinalObject::s_info, 0, 0, CREATE_METHOD_TABLE(JSArray)};
darin's avatar
darin committed
52

53
Butterfly* createArrayButterflyInDictionaryIndexingMode(JSGlobalData& globalData, unsigned initialLength)
darin's avatar
darin committed
54
{
55 56 57 58 59 60 61 62 63
    Butterfly* butterfly = Butterfly::create(
        globalData, 0, 0, true, IndexingHeader(), ArrayStorage::sizeFor(0));
    ArrayStorage* storage = butterfly->arrayStorage();
    storage->setLength(initialLength);
    storage->setVectorLength(0);
    storage->m_indexBias = 0;
    storage->m_sparseMap.clear();
    storage->m_numValuesInVector = 0;
    return butterfly;
64 65 66 67 68 69 70 71
}

void JSArray::setLengthWritable(ExecState* exec, bool writable)
{
    ASSERT(isLengthWritable() || !writable);
    if (!isLengthWritable() || writable)
        return;

72
    enterDictionaryIndexingMode(exec->globalData());
73

74
    SparseArrayValueMap* map = arrayStorage()->m_sparseMap.get();
75 76 77 78 79
    ASSERT(map);
    map->setLengthIsReadOnly();
}

// Defined in ES5.1 15.4.5.1
80
bool JSArray::defineOwnProperty(JSObject* object, ExecState* exec, PropertyName propertyName, PropertyDescriptor& descriptor, bool throwException)
81
{
82
    JSArray* array = jsCast<JSArray*>(object);
83 84 85 86 87

    // 3. If P is "length", then
    if (propertyName == exec->propertyNames().length) {
        // All paths through length definition call the default [[DefineOwnProperty]], hence:
        // from ES5.1 8.12.9 7.a.
88
        if (descriptor.configurablePresent() && descriptor.configurable())
89 90
            return reject(exec, throwException, "Attempting to change configurable attribute of unconfigurable property.");
        // from ES5.1 8.12.9 7.b.
91
        if (descriptor.enumerablePresent() && descriptor.enumerable())
92 93 94 95 96 97 98
            return reject(exec, throwException, "Attempting to change enumerable attribute of unconfigurable property.");

        // a. If the [[Value]] field of Desc is absent, then
        // a.i. Return the result of calling the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length", Desc, and Throw as arguments.
        if (descriptor.isAccessorDescriptor())
            return reject(exec, throwException, "Attempting to change access mechanism for an unconfigurable property.");
        // from ES5.1 8.12.9 10.a.
99
        if (!array->isLengthWritable() && descriptor.writablePresent() && descriptor.writable())
100 101 102
            return reject(exec, throwException, "Attempting to change writable attribute of unconfigurable property.");
        // This descriptor is either just making length read-only, or changing nothing!
        if (!descriptor.value()) {
103 104
            if (descriptor.writablePresent())
                array->setLengthWritable(exec, descriptor.writable());
105 106 107 108 109 110 111 112 113 114 115 116 117 118
            return true;
        }
        
        // b. Let newLenDesc be a copy of Desc.
        // c. Let newLen be ToUint32(Desc.[[Value]]).
        unsigned newLen = descriptor.value().toUInt32(exec);
        // d. If newLen is not equal to ToNumber( Desc.[[Value]]), throw a RangeError exception.
        if (newLen != descriptor.value().toNumber(exec)) {
            throwError(exec, createRangeError(exec, "Invalid array length"));
            return false;
        }

        // Based on SameValue check in 8.12.9, this is always okay.
        if (newLen == array->length()) {
119 120
            if (descriptor.writablePresent())
                array->setLengthWritable(exec, descriptor.writable());
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
            return true;
        }

        // e. Set newLenDesc.[[Value] to newLen.
        // f. If newLen >= oldLen, then
        // f.i. Return the result of calling the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length", newLenDesc, and Throw as arguments.
        // g. Reject if oldLenDesc.[[Writable]] is false.
        if (!array->isLengthWritable())
            return reject(exec, throwException, "Attempting to change value of a readonly property.");
        
        // h. If newLenDesc.[[Writable]] is absent or has the value true, let newWritable be true.
        // i. Else,
        // i.i. Need to defer setting the [[Writable]] attribute to false in case any elements cannot be deleted.
        // i.ii. Let newWritable be false.
        // i.iii. Set newLenDesc.[[Writable] to true.
        // j. Let succeeded be the result of calling the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length", newLenDesc, and Throw as arguments.
        // k. If succeeded is false, return false.
        // l. While newLen < oldLen repeat,
        // l.i. Set oldLen to oldLen – 1.
        // l.ii. Let deleteSucceeded be the result of calling the [[Delete]] internal method of A passing ToString(oldLen) and false as arguments.
        // l.iii. If deleteSucceeded is false, then
142
        if (!array->setLength(exec, newLen, throwException)) {
143 144 145 146
            // 1. Set newLenDesc.[[Value] to oldLen+1.
            // 2. If newWritable is false, set newLenDesc.[[Writable] to false.
            // 3. Call the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length", newLenDesc, and false as arguments.
            // 4. Reject.
147 148
            if (descriptor.writablePresent())
                array->setLengthWritable(exec, descriptor.writable());
149 150 151 152
            return false;
        }

        // m. If newWritable is false, then
153 154 155 156 157
        // i. Call the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length",
        //    Property Descriptor{[[Writable]]: false}, and false as arguments. This call will always
        //    return true.
        if (descriptor.writablePresent())
            array->setLengthWritable(exec, descriptor.writable());
158 159 160 161 162 163
        // n. Return true.
        return true;
    }

    // 4. Else if P is an array index (15.4), then
    // a. Let index be ToUint32(P).
164 165
    unsigned index = propertyName.asIndex();
    if (index != PropertyName::NotAnIndex) {
166 167 168 169 170 171 172 173 174
        // b. Reject if index >= oldLen and oldLenDesc.[[Writable]] is false.
        if (index >= array->length() && !array->isLengthWritable())
            return reject(exec, throwException, "Attempting to define numeric property on array with non-writable length property.");
        // c. Let succeeded be the result of calling the default [[DefineOwnProperty]] internal method (8.12.9) on A passing P, Desc, and false as arguments.
        // d. Reject if succeeded is false.
        // e. If index >= oldLen
        // e.i. Set oldLenDesc.[[Value]] to index + 1.
        // e.ii. Call the default [[DefineOwnProperty]] internal method (8.12.9) on A passing "length", oldLenDesc, and false as arguments. This call will always return true.
        // f. Return true.
175
        return array->defineOwnIndexedProperty(exec, index, descriptor, throwException);
darin's avatar
darin committed
176 177
    }

178
    return array->JSObject::defineOwnNonIndexProperty(exec, propertyName, descriptor, throwException);
darin's avatar
darin committed
179 180
}

181
bool JSArray::getOwnPropertySlot(JSCell* cell, ExecState* exec, PropertyName propertyName, PropertySlot& slot)
182
{
183
    JSArray* thisObject = jsCast<JSArray*>(cell);
darin's avatar
darin committed
184
    if (propertyName == exec->propertyNames().length) {
185
        slot.setValue(jsNumber(thisObject->length()));
darin's avatar
darin committed
186 187 188
        return true;
    }

189
    return JSObject::getOwnPropertySlot(thisObject, exec, propertyName, slot);
darin's avatar
darin committed
190 191
}

192
bool JSArray::getOwnPropertyDescriptor(JSObject* object, ExecState* exec, PropertyName propertyName, PropertyDescriptor& descriptor)
193
{
194
    JSArray* thisObject = jsCast<JSArray*>(object);
195
    if (propertyName == exec->propertyNames().length) {
196
        descriptor.setDescriptor(jsNumber(thisObject->length()), thisObject->isLengthWritable() ? DontDelete | DontEnum : DontDelete | DontEnum | ReadOnly);
197 198
        return true;
    }
199

200
    return JSObject::getOwnPropertyDescriptor(thisObject, exec, propertyName, descriptor);
201 202
}

203
// ECMA 15.4.5.1
204
void JSArray::put(JSCell* cell, ExecState* exec, PropertyName propertyName, JSValue value, PutPropertySlot& slot)
205
{
206
    JSArray* thisObject = jsCast<JSArray*>(cell);
darin's avatar
darin committed
207 208

    if (propertyName == exec->propertyNames().length) {
weinig@apple.com's avatar
weinig@apple.com committed
209 210
        unsigned newLength = value.toUInt32(exec);
        if (value.toNumber(exec) != static_cast<double>(newLength)) {
211
            throwError(exec, createRangeError(exec, ASCIILiteral("Invalid array length")));
darin's avatar
darin committed
212 213
            return;
        }
214
        thisObject->setLength(exec, newLength, slot.isStrictMode());
darin's avatar
darin committed
215 216 217
        return;
    }

218
    JSObject::put(thisObject, exec, propertyName, value, slot);
darin's avatar
darin committed
219 220
}

221
bool JSArray::deleteProperty(JSCell* cell, ExecState* exec, PropertyName propertyName)
222
{
223
    JSArray* thisObject = jsCast<JSArray*>(cell);
darin's avatar
darin committed
224 225 226 227

    if (propertyName == exec->propertyNames().length)
        return false;

228
    return JSObject::deleteProperty(thisObject, exec, propertyName);
darin's avatar
darin committed
229 230
}

231 232
static int compareKeysForQSort(const void* a, const void* b)
{
233 234
    unsigned da = *static_cast<const unsigned*>(a);
    unsigned db = *static_cast<const unsigned*>(b);
235
    return (da > db) - (da < db);
darin's avatar
darin committed
236 237
}

238
void JSArray::getOwnNonIndexPropertyNames(JSObject* object, ExecState* exec, PropertyNameArray& propertyNames, EnumerationMode mode)
darin's avatar
darin committed
239
{
240
    JSArray* thisObject = jsCast<JSArray*>(object);
241

242 243 244
    if (mode == IncludeDontEnumProperties)
        propertyNames.add(exec->propertyNames().length);

245
    JSObject::getOwnNonIndexPropertyNames(thisObject, exec, propertyNames, mode);
246
}
darin's avatar
darin committed
247

248
// This method makes room in the vector, but leaves the new space uncleared.
249
bool JSArray::unshiftCountSlowCase(JSGlobalData& globalData, unsigned count)
250
{
251 252 253 254
    ArrayStorage* storage = ensureArrayStorage(globalData);
    Butterfly* butterfly = storage->butterfly();
    unsigned propertyCapacity = structure()->outOfLineCapacity();
    unsigned propertySize = structure()->outOfLineSize();
255

256 257
    // If not, we should have handled this on the fast path.
    ASSERT(count > storage->m_indexBias);
258

259 260 261 262 263 264 265
    // Step 1:
    // Gather 4 key metrics:
    //  * usedVectorLength - how many entries are currently in the vector (conservative estimate - fewer may be in use in sparse vectors).
    //  * requiredVectorLength - how many entries are will there be in the vector, after allocating space for 'count' more.
    //  * currentCapacity - what is the current size of the vector, including any pre-capacity.
    //  * desiredCapacity - how large should we like to grow the vector to - based on 2x requiredVectorLength.

266 267
    unsigned length = storage->length();
    unsigned usedVectorLength = min(storage->vectorLength(), length);
268 269 270
    ASSERT(usedVectorLength <= MAX_STORAGE_VECTOR_LENGTH);
    // Check that required vector length is possible, in an overflow-safe fashion.
    if (count > MAX_STORAGE_VECTOR_LENGTH - usedVectorLength)
271
        return false;
272 273 274
    unsigned requiredVectorLength = usedVectorLength + count;
    ASSERT(requiredVectorLength <= MAX_STORAGE_VECTOR_LENGTH);
    // The sum of m_vectorLength and m_indexBias will never exceed MAX_STORAGE_VECTOR_LENGTH.
275 276
    ASSERT(storage->vectorLength() <= MAX_STORAGE_VECTOR_LENGTH && (MAX_STORAGE_VECTOR_LENGTH - storage->vectorLength()) >= storage->m_indexBias);
    unsigned currentCapacity = storage->vectorLength() + storage->m_indexBias;
277 278 279 280 281 282
    // The calculation of desiredCapacity won't overflow, due to the range of MAX_STORAGE_VECTOR_LENGTH.
    unsigned desiredCapacity = min(MAX_STORAGE_VECTOR_LENGTH, max(BASE_VECTOR_LEN, requiredVectorLength) << 1);

    // Step 2:
    // We're either going to choose to allocate a new ArrayStorage, or we're going to reuse the existing on.

283
    void* newAllocBase = 0;
284 285 286
    unsigned newStorageCapacity;
    // If the current storage array is sufficiently large (but not too large!) then just keep using it.
    if (currentCapacity > desiredCapacity && isDenseEnoughForVector(currentCapacity, requiredVectorLength)) {
287
        newAllocBase = butterfly->base(structure());
288 289
        newStorageCapacity = currentCapacity;
    } else {
290 291
        size_t newSize = Butterfly::totalSize(0, propertyCapacity, true, ArrayStorage::sizeFor(desiredCapacity));
        if (!globalData.heap.tryAllocateStorage(newSize, &newAllocBase))
292 293 294 295 296 297 298 299 300 301 302 303
            return false;
        newStorageCapacity = desiredCapacity;
    }

    // Step 3:
    // Work out where we're going to move things to.

    // Determine how much of the vector to use as pre-capacity, and how much as post-capacity.
    // If the vector had no free post-capacity (length >= m_vectorLength), don't give it any.
    // If it did, we calculate the amount that will remain based on an atomic decay - leave the
    // vector with half the post-capacity it had previously.
    unsigned postCapacity = 0;
304
    if (length < storage->vectorLength()) {
305
        // Atomic decay, + the post-capacity cannot be greater than what is available.
306 307 308
        postCapacity = min((storage->vectorLength() - length) >> 1, newStorageCapacity - requiredVectorLength);
        // If we're moving contents within the same allocation, the post-capacity is being reduced.
        ASSERT(newAllocBase != butterfly->base(structure()) || postCapacity < storage->vectorLength() - length);
309
    }
310 311 312 313 314 315 316 317 318 319 320 321 322
    
    unsigned newVectorLength = requiredVectorLength + postCapacity;
    unsigned newIndexBias = newStorageCapacity - newVectorLength;

    Butterfly* newButterfly = Butterfly::fromBase(newAllocBase, newIndexBias, propertyCapacity);
    
    memmove(newButterfly->arrayStorage()->m_vector + count, storage->m_vector, sizeof(JSValue) * usedVectorLength);
    memmove(newButterfly->propertyStorage() - propertySize, butterfly->propertyStorage() - propertySize, sizeof(JSValue) * propertySize + sizeof(IndexingHeader) + ArrayStorage::sizeFor(0));
    
    newButterfly->arrayStorage()->setVectorLength(newVectorLength);
    newButterfly->arrayStorage()->m_indexBias = newIndexBias;
    
    m_butterfly = newButterfly;
323

ap@webkit.org's avatar
ap@webkit.org committed
324
    return true;
darin's avatar
darin committed
325 326
}

327
bool JSArray::setLength(ExecState* exec, unsigned newLength, bool throwException)
darin's avatar
darin committed
328
{
329 330
    ArrayStorage* storage = ensureArrayStorage(exec->globalData());
    unsigned length = storage->length();
darin's avatar
darin committed
331

332
    // If the length is read only then we enter sparse mode, so should enter the following 'if'.
333
    ASSERT(isLengthWritable() || storage->m_sparseMap);
334

335
    if (SparseArrayValueMap* map = storage->m_sparseMap.get()) {
336 337 338 339 340 341 342 343 344 345
        // Fail if the length is not writable.
        if (map->lengthIsReadOnly())
            return reject(exec, throwException, StrictModeReadonlyPropertyWriteError);

        if (newLength < length) {
            // Copy any keys we might be interested in into a vector.
            Vector<unsigned> keys;
            keys.reserveCapacity(min(map->size(), static_cast<size_t>(length - newLength)));
            SparseArrayValueMap::const_iterator end = map->end();
            for (SparseArrayValueMap::const_iterator it = map->begin(); it != end; ++it) {
346
                unsigned index = static_cast<unsigned>(it->first);
347 348 349 350 351 352 353 354 355 356 357 358 359 360
                if (index < length && index >= newLength)
                    keys.append(index);
            }

            // Check if the array is in sparse mode. If so there may be non-configurable
            // properties, so we have to perform deletion with caution, if not we can
            // delete values in any order.
            if (map->sparseMode()) {
                qsort(keys.begin(), keys.size(), sizeof(unsigned), compareKeysForQSort);
                unsigned i = keys.size();
                while (i) {
                    unsigned index = keys[--i];
                    SparseArrayValueMap::iterator it = map->find(index);
                    ASSERT(it != map->notFound());
361
                    if (it->second.attributes & DontDelete) {
362
                        storage->setLength(index + 1);
363 364 365 366 367 368 369
                        return reject(exec, throwException, "Unable to delete property.");
                    }
                    map->remove(it);
                }
            } else {
                for (unsigned i = 0; i < keys.size(); ++i)
                    map->remove(keys[i]);
370
                if (map->isEmpty())
371
                    deallocateSparseIndexMap();
372 373 374 375
            }
        }
    }

darin's avatar
darin committed
376
    if (newLength < length) {
377
        // Delete properties from the vector.
378
        unsigned usedVectorLength = min(length, storage->vectorLength());
darin's avatar
darin committed
379
        for (unsigned i = newLength; i < usedVectorLength; ++i) {
380
            WriteBarrier<Unknown>& valueSlot = storage->m_vector[i];
darin's avatar
darin committed
381
            bool hadValue = valueSlot;
382
            valueSlot.clear();
darin's avatar
darin committed
383 384 385
            storage->m_numValuesInVector -= hadValue;
        }
    }
aroben@apple.com's avatar
aroben@apple.com committed
386

387
    storage->setLength(newLength);
388

389
    return true;
darin's avatar
darin committed
390 391
}

392
JSValue JSArray::pop(ExecState* exec)
393
{
394
    switch (structure()->indexingType()) {
395
    case ArrayClass:
396
        return jsUndefined();
397
        
398
    case ARRAY_WITH_ARRAY_STORAGE_INDEXING_TYPES: {
399 400 401 402 403 404 405 406
        ArrayStorage* storage = m_butterfly->arrayStorage();
    
        unsigned length = storage->length();
        if (!length) {
            if (!isLengthWritable())
                throwTypeError(exec, StrictModeReadonlyPropertyWriteError);
            return jsUndefined();
        }
407

408 409 410 411 412 413 414
        unsigned index = length - 1;
        if (index < storage->vectorLength()) {
            WriteBarrier<Unknown>& valueSlot = storage->m_vector[index];
            if (valueSlot) {
                --storage->m_numValuesInVector;
                JSValue element = valueSlot.get();
                valueSlot.clear();
415
            
416 417 418 419
                ASSERT(isLengthWritable());
                storage->setLength(index);
                return element;
            }
420 421
        }

422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
        // Let element be the result of calling the [[Get]] internal method of O with argument indx.
        JSValue element = get(exec, index);
        if (exec->hadException())
            return jsUndefined();
        // Call the [[Delete]] internal method of O with arguments indx and true.
        if (!deletePropertyByIndex(this, exec, index)) {
            throwTypeError(exec, "Unable to delete property.");
            return jsUndefined();
        }
        // Call the [[Put]] internal method of O with arguments "length", indx, and true.
        setLength(exec, index, true);
        // Return element.
        return element;
    }
        
    default:
        ASSERT_NOT_REACHED();
        return JSValue();
440
    }
441 442
}

443 444 445
// Push & putIndex are almost identical, with two small differences.
//  - we always are writing beyond the current array bounds, so it is always necessary to update m_length & m_numValuesInVector.
//  - pushing to an array of length 2^32-1 stores the property, but throws a range error.
ggaren@apple.com's avatar
ggaren@apple.com committed
446
void JSArray::push(ExecState* exec, JSValue value)
447
{
448
    switch (structure()->indexingType()) {
449
    case ArrayClass: {
450 451
        putByIndexBeyondVectorLengthWithArrayStorage(exec, 0, value, true, createInitialArrayStorage(exec->globalData()));
        break;
452
    }
453
        
454 455 456 457 458 459 460 461 462 463
    case ArrayWithSlowPutArrayStorage: {
        unsigned oldLength = length();
        if (attemptToInterceptPutByIndexOnHole(exec, oldLength, value, true)) {
            if (!exec->hadException() && oldLength < 0xFFFFFFFFu)
                setLength(exec, oldLength + 1, true);
            return;
        }
        // Fall through.
    }
        
464 465 466 467 468 469 470 471 472 473 474
    case ArrayWithArrayStorage: {
        ArrayStorage* storage = m_butterfly->arrayStorage();

        // Fast case - push within vector, always update m_length & m_numValuesInVector.
        unsigned length = storage->length();
        if (length < storage->vectorLength()) {
            storage->m_vector[length].set(exec->globalData(), this, value);
            storage->setLength(length + 1);
            ++storage->m_numValuesInVector;
            return;
        }
475

476 477
        // Pushing to an array of invalid length (2^31-1) stores the property, but throws a range error.
        if (storage->length() > MAX_ARRAY_INDEX) {
478 479 480 481 482 483
            methodTable()->putByIndex(this, exec, storage->length(), value, true);
            // Per ES5.1 15.4.4.7 step 6 & 15.4.5.1 step 3.d.
            if (!exec->hadException())
                throwError(exec, createRangeError(exec, "Invalid array length"));
            return;
        }
484

485 486 487 488 489 490 491 492
        // Handled the same as putIndex.
        putByIndexBeyondVectorLengthWithArrayStorage(exec, storage->length(), value, true, storage);
        break;
    }
        
    default:
        ASSERT_NOT_REACHED();
    }
493 494
}

495
bool JSArray::shiftCount(ExecState* exec, unsigned count)
496 497 498
{
    ASSERT(count > 0);
    
499
    ArrayStorage* storage = ensureArrayStorage(exec->globalData());
500
    
501
    unsigned oldLength = storage->length();
502
    
503 504
    // If the array contains holes or is otherwise in an abnormal state,
    // use the generic algorithm in ArrayPrototype.
505
    if (oldLength != storage->m_numValuesInVector || inSparseIndexingMode())
506
        return false;
507

508 509
    if (!oldLength)
        return true;
510
    
511
    storage->m_numValuesInVector -= count;
512
    storage->setLength(oldLength - count);
513
    
514 515 516
    unsigned vectorLength = storage->vectorLength();
    if (vectorLength) {
        count = min(vectorLength, (unsigned)count);
517
        
518 519
        vectorLength -= count;
        storage->setVectorLength(vectorLength);
520
        
521 522 523 524
        if (vectorLength) {
            m_butterfly = m_butterfly->shift(structure(), count);
            storage = m_butterfly->arrayStorage();
            storage->m_indexBias += count;
525 526
        }
    }
527
    return true;
528
}
529 530 531

// Returns true if the unshift can be handled, false to fallback.    
bool JSArray::unshiftCount(ExecState* exec, unsigned count)
532
{
533 534
    ArrayStorage* storage = ensureArrayStorage(exec->globalData());
    unsigned length = storage->length();
535

536 537
    // If the array contains holes or is otherwise in an abnormal state,
    // use the generic algorithm in ArrayPrototype.
538
    if (length != storage->m_numValuesInVector || storage->inSparseMode())
539 540
        return false;

541 542 543 544 545
    if (storage->m_indexBias >= count) {
        m_butterfly = storage->butterfly()->unshift(structure(), count);
        storage = m_butterfly->arrayStorage();
        storage->m_indexBias -= count;
        storage->setVectorLength(storage->vectorLength() + count);
546 547 548
    } else if (unshiftCountSlowCase(exec->globalData(), count))
        storage = arrayStorage();
    else {
549
        throwOutOfMemoryError(exec);
550
        return true;
551
    }
552

553
    WriteBarrier<Unknown>* vector = storage->m_vector;
554
    for (unsigned i = 0; i < count; i++)
555
        vector[i].clear();
556
    return true;
557 558
}

ggaren@apple.com's avatar
ggaren@apple.com committed
559 560
static int compareNumbersForQSort(const void* a, const void* b)
{
ggaren@apple.com's avatar
ggaren@apple.com committed
561 562
    double da = static_cast<const JSValue*>(a)->asNumber();
    double db = static_cast<const JSValue*>(b)->asNumber();
ggaren@apple.com's avatar
ggaren@apple.com committed
563 564 565
    return (da > db) - (da < db);
}

566 567
static int compareByStringPairForQSort(const void* a, const void* b)
{
ggaren@apple.com's avatar
ggaren@apple.com committed
568 569
    const ValueStringPair* va = static_cast<const ValueStringPair*>(a);
    const ValueStringPair* vb = static_cast<const ValueStringPair*>(b);
570
    return codePointCompare(va->second, vb->second);
571
}
darin's avatar
darin committed
572

ggaren@apple.com's avatar
ggaren@apple.com committed
573
void JSArray::sortNumeric(ExecState* exec, JSValue compareFunction, CallType callType, const CallData& callData)
ggaren@apple.com's avatar
ggaren@apple.com committed
574
{
575
    ASSERT(!inSparseIndexingMode());
ggaren@apple.com's avatar
ggaren@apple.com committed
576

577
    switch (structure()->indexingType()) {
578
    case ArrayClass:
ggaren@apple.com's avatar
ggaren@apple.com committed
579 580
        return;
        
581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
    case ArrayWithArrayStorage: {
        unsigned lengthNotIncludingUndefined = compactForSorting(exec->globalData());
        ArrayStorage* storage = m_butterfly->arrayStorage();
        
        if (storage->m_sparseMap.get()) {
            throwOutOfMemoryError(exec);
            return;
        }
        
        if (!lengthNotIncludingUndefined)
            return;
        
        bool allValuesAreNumbers = true;
        size_t size = storage->m_numValuesInVector;
        for (size_t i = 0; i < size; ++i) {
            if (!storage->m_vector[i].isNumber()) {
                allValuesAreNumbers = false;
                break;
            }
ggaren@apple.com's avatar
ggaren@apple.com committed
600
        }
601 602 603 604 605 606 607 608 609 610 611 612 613 614
        
        if (!allValuesAreNumbers)
            return sort(exec, compareFunction, callType, callData);
        
        // For numeric comparison, which is fast, qsort is faster than mergesort. We
        // also don't require mergesort's stability, since there's no user visible
        // side-effect from swapping the order of equal primitive values.
        qsort(storage->m_vector, size, sizeof(WriteBarrier<Unknown>), compareNumbersForQSort);
        
        return;
    }
        
    default:
        ASSERT_NOT_REACHED();
ggaren@apple.com's avatar
ggaren@apple.com committed
615 616 617
    }
}

darin@apple.com's avatar
darin@apple.com committed
618
void JSArray::sort(ExecState* exec)
darin's avatar
darin committed
619
{
620
    ASSERT(!inSparseIndexingMode());
621
    
622
    switch (structure()->indexingType()) {
623
    case ArrayClass:
ap@webkit.org's avatar
ap@webkit.org committed
624
        return;
625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
        
    case ArrayWithArrayStorage: {
        unsigned lengthNotIncludingUndefined = compactForSorting(exec->globalData());
        ArrayStorage* storage = m_butterfly->arrayStorage();
        if (storage->m_sparseMap.get()) {
            throwOutOfMemoryError(exec);
            return;
        }
        
        if (!lengthNotIncludingUndefined)
            return;
        
        // Converting JavaScript values to strings can be expensive, so we do it once up front and sort based on that.
        // This is a considerable improvement over doing it twice per comparison, though it requires a large temporary
        // buffer. Besides, this protects us from crashing if some objects have custom toString methods that return
        // random or otherwise changing results, effectively making compare function inconsistent.
        
        Vector<ValueStringPair> values(lengthNotIncludingUndefined);
        if (!values.begin()) {
            throwOutOfMemoryError(exec);
            return;
        }
        
        Heap::heap(this)->pushTempSortVector(&values);
        
        bool isSortingPrimitiveValues = true;
        for (size_t i = 0; i < lengthNotIncludingUndefined; i++) {
            JSValue value = storage->m_vector[i].get();
            ASSERT(!value.isUndefined());
            values[i].first = value;
            isSortingPrimitiveValues = isSortingPrimitiveValues && value.isPrimitive();
        }
        
        // FIXME: The following loop continues to call toString on subsequent values even after
        // a toString call raises an exception.
        
        for (size_t i = 0; i < lengthNotIncludingUndefined; i++)
            values[i].second = values[i].first.toWTFStringInline(exec);
        
        if (exec->hadException()) {
            Heap::heap(this)->popTempSortVector(&values);
            return;
        }
        
        // FIXME: Since we sort by string value, a fast algorithm might be to use a radix sort. That would be O(N) rather
        // than O(N log N).
        
672
#if HAVE(MERGESORT)
673 674 675 676
        if (isSortingPrimitiveValues)
            qsort(values.begin(), values.size(), sizeof(ValueStringPair), compareByStringPairForQSort);
        else
            mergesort(values.begin(), values.size(), sizeof(ValueStringPair), compareByStringPairForQSort);
677
#else
678 679 680
        // FIXME: The qsort library function is likely to not be a stable sort.
        // ECMAScript-262 does not specify a stable sort, but in practice, browsers perform a stable sort.
        qsort(values.begin(), values.size(), sizeof(ValueStringPair), compareByStringPairForQSort);
681
#endif
682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703
        
        // If the toString function changed the length of the array or vector storage,
        // increase the length to handle the orignal number of actual values.
        if (storage->vectorLength() < lengthNotIncludingUndefined) {
            increaseVectorLength(exec->globalData(), lengthNotIncludingUndefined);
            storage = m_butterfly->arrayStorage();
        }
        if (storage->length() < lengthNotIncludingUndefined)
            storage->setLength(lengthNotIncludingUndefined);
        
        JSGlobalData& globalData = exec->globalData();
        for (size_t i = 0; i < lengthNotIncludingUndefined; i++)
            storage->m_vector[i].set(globalData, this, values[i].first);
        
        Heap::heap(this)->popTempSortVector(&values);
        
        return;
    }
        
    default:
        ASSERT_NOT_REACHED();
    }
darin's avatar
darin committed
704 705
}

ap@webkit.org's avatar
ap@webkit.org committed
706
struct AVLTreeNodeForArrayCompare {
ggaren@apple.com's avatar
ggaren@apple.com committed
707
    JSValue value;
ap@webkit.org's avatar
ap@webkit.org committed
708 709 710 711 712 713 714 715 716 717

    // Child pointers.  The high bit of gt is robbed and used as the
    // balance factor sign.  The high bit of lt is robbed and used as
    // the magnitude of the balance factor.
    int32_t gt;
    int32_t lt;
};

struct AVLTreeAbstractorForArrayCompare {
    typedef int32_t handle; // Handle is an index into m_nodes vector.
ggaren@apple.com's avatar
ggaren@apple.com committed
718
    typedef JSValue key;
ap@webkit.org's avatar
ap@webkit.org committed
719 720 721 722
    typedef int32_t size;

    Vector<AVLTreeNodeForArrayCompare> m_nodes;
    ExecState* m_exec;
ggaren@apple.com's avatar
ggaren@apple.com committed
723
    JSValue m_compareFunction;
darin@apple.com's avatar
darin@apple.com committed
724 725
    CallType m_compareCallType;
    const CallData* m_compareCallData;
726
    OwnPtr<CachedCall> m_cachedCall;
ap@webkit.org's avatar
ap@webkit.org committed
727 728 729 730 731 732 733 734 735 736 737 738 739 740

    handle get_less(handle h) { return m_nodes[h].lt & 0x7FFFFFFF; }
    void set_less(handle h, handle lh) { m_nodes[h].lt &= 0x80000000; m_nodes[h].lt |= lh; }
    handle get_greater(handle h) { return m_nodes[h].gt & 0x7FFFFFFF; }
    void set_greater(handle h, handle gh) { m_nodes[h].gt &= 0x80000000; m_nodes[h].gt |= gh; }

    int get_balance_factor(handle h)
    {
        if (m_nodes[h].gt & 0x80000000)
            return -1;
        return static_cast<unsigned>(m_nodes[h].lt) >> 31;
    }

    void set_balance_factor(handle h, int bf)
darin's avatar
darin committed
741
    {
ap@webkit.org's avatar
ap@webkit.org committed
742 743 744 745 746 747 748
        if (bf == 0) {
            m_nodes[h].lt &= 0x7FFFFFFF;
            m_nodes[h].gt &= 0x7FFFFFFF;
        } else {
            m_nodes[h].lt |= 0x80000000;
            if (bf < 0)
                m_nodes[h].gt |= 0x80000000;
ap@webkit.org's avatar
ap@webkit.org committed
749 750
            else
                m_nodes[h].gt &= 0x7FFFFFFF;
ap@webkit.org's avatar
ap@webkit.org committed
751
        }
darin's avatar
darin committed
752 753
    }

ap@webkit.org's avatar
ap@webkit.org committed
754
    int compare_key_key(key va, key vb)
ap@webkit.org's avatar
ap@webkit.org committed
755
    {
weinig@apple.com's avatar
weinig@apple.com committed
756 757
        ASSERT(!va.isUndefined());
        ASSERT(!vb.isUndefined());
darin's avatar
darin committed
758

ggaren@apple.com's avatar
ggaren@apple.com committed
759 760 761
        if (m_exec->hadException())
            return 1;

762 763
        double compareResult;
        if (m_cachedCall) {
764
            m_cachedCall->setThis(jsUndefined());
765 766
            m_cachedCall->setArgument(0, va);
            m_cachedCall->setArgument(1, vb);
767
            compareResult = m_cachedCall->call().toNumber(m_cachedCall->newCallFrame(m_exec));
768
        } else {
769
            MarkedArgumentBuffer arguments;
770 771
            arguments.append(va);
            arguments.append(vb);
772
            compareResult = call(m_exec, m_compareFunction, m_compareCallType, *m_compareCallData, jsUndefined(), arguments).toNumber(m_exec);
773
        }
ap@webkit.org's avatar
ap@webkit.org committed
774
        return (compareResult < 0) ? -1 : 1; // Not passing equality through, because we need to store all values, even if equivalent.
ap@webkit.org's avatar
ap@webkit.org committed
775
    }
darin's avatar
darin committed
776

ap@webkit.org's avatar
ap@webkit.org committed
777 778 779 780
    int compare_key_node(key k, handle h) { return compare_key_key(k, m_nodes[h].value); }
    int compare_node_node(handle h1, handle h2) { return compare_key_key(m_nodes[h1].value, m_nodes[h2].value); }

    static handle null() { return 0x7FFFFFFF; }
ap@webkit.org's avatar
ap@webkit.org committed
781
};
darin's avatar
darin committed
782

ggaren@apple.com's avatar
ggaren@apple.com committed
783
void JSArray::sort(ExecState* exec, JSValue compareFunction, CallType callType, const CallData& callData)
darin's avatar
darin committed
784
{
785 786 787
    ASSERT(!inSparseIndexingMode());
    
    switch (structure()->indexingType()) {
788
    case ArrayClass:
ap@webkit.org's avatar
ap@webkit.org committed
789
        return;
790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846
        
    case ArrayWithArrayStorage: {
        ArrayStorage* storage = m_butterfly->arrayStorage();
        
        // FIXME: This ignores exceptions raised in the compare function or in toNumber.
        
        // The maximum tree depth is compiled in - but the caller is clearly up to no good
        // if a larger array is passed.
        ASSERT(storage->length() <= static_cast<unsigned>(std::numeric_limits<int>::max()));
        if (storage->length() > static_cast<unsigned>(std::numeric_limits<int>::max()))
            return;
        
        unsigned usedVectorLength = min(storage->length(), storage->vectorLength());
        unsigned nodeCount = usedVectorLength + (storage->m_sparseMap ? storage->m_sparseMap->size() : 0);
        
        if (!nodeCount)
            return;
        
        AVLTree<AVLTreeAbstractorForArrayCompare, 44> tree; // Depth 44 is enough for 2^31 items
        tree.abstractor().m_exec = exec;
        tree.abstractor().m_compareFunction = compareFunction;
        tree.abstractor().m_compareCallType = callType;
        tree.abstractor().m_compareCallData = &callData;
        tree.abstractor().m_nodes.grow(nodeCount);
        
        if (callType == CallTypeJS)
            tree.abstractor().m_cachedCall = adoptPtr(new CachedCall(exec, jsCast<JSFunction*>(compareFunction), 2));
        
        if (!tree.abstractor().m_nodes.begin()) {
            throwOutOfMemoryError(exec);
            return;
        }
        
        // FIXME: If the compare function modifies the array, the vector, map, etc. could be modified
        // right out from under us while we're building the tree here.
        
        unsigned numDefined = 0;
        unsigned numUndefined = 0;
        
        // Iterate over the array, ignoring missing values, counting undefined ones, and inserting all other ones into the tree.
        for (; numDefined < usedVectorLength; ++numDefined) {
            JSValue v = storage->m_vector[numDefined].get();
            if (!v || v.isUndefined())
                break;
            tree.abstractor().m_nodes[numDefined].value = v;
            tree.insert(numDefined);
        }
        for (unsigned i = numDefined; i < usedVectorLength; ++i) {
            JSValue v = storage->m_vector[i].get();
            if (v) {
                if (v.isUndefined())
                    ++numUndefined;
                else {
                    tree.abstractor().m_nodes[numDefined].value = v;
                    tree.insert(numDefined);
                    ++numDefined;
                }
ap@webkit.org's avatar
ap@webkit.org committed
847 848
            }
        }
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867
        
        unsigned newUsedVectorLength = numDefined + numUndefined;
        
        if (SparseArrayValueMap* map = storage->m_sparseMap.get()) {
            newUsedVectorLength += map->size();
            if (newUsedVectorLength > storage->vectorLength()) {
                // Check that it is possible to allocate an array large enough to hold all the entries.
                if ((newUsedVectorLength > MAX_STORAGE_VECTOR_LENGTH) || !increaseVectorLength(exec->globalData(), newUsedVectorLength)) {
                    throwOutOfMemoryError(exec);
                    return;
                }
                storage = m_butterfly->arrayStorage();
            }
            
            SparseArrayValueMap::const_iterator end = map->end();
            for (SparseArrayValueMap::const_iterator it = map->begin(); it != end; ++it) {
                tree.abstractor().m_nodes[numDefined].value = it->second.getNonSparseMode();
                tree.insert(numDefined);
                ++numDefined;
ap@webkit.org's avatar
ap@webkit.org committed
868
            }
869 870
            
            deallocateSparseIndexMap();
ap@webkit.org's avatar
ap@webkit.org committed
871
        }
872 873 874 875 876 877 878 879 880 881 882 883 884
        
        ASSERT(tree.abstractor().m_nodes.size() >= numDefined);
        
        // FIXME: If the compare function changed the length of the array, the following might be
        // modifying the vector incorrectly.
        
        // Copy the values back into m_storage.
        AVLTree<AVLTreeAbstractorForArrayCompare, 44>::Iterator iter;
        iter.start_iter_least(tree);
        JSGlobalData& globalData = exec->globalData();
        for (unsigned i = 0; i < numDefined; ++i) {
            storage->m_vector[i].set(globalData, this, tree.abstractor().m_nodes[*iter].value);
            ++iter;
ap@webkit.org's avatar
ap@webkit.org committed
885
        }
886 887 888 889 890 891 892 893 894 895 896 897
        
        // Put undefined values back in.
        for (unsigned i = numDefined; i < newUsedVectorLength; ++i)
            storage->m_vector[i].setUndefined();
        
        // Ensure that unused values in the vector are zeroed out.
        for (unsigned i = newUsedVectorLength; i < usedVectorLength; ++i)
            storage->m_vector[i].clear();
        
        storage->m_numValuesInVector = newUsedVectorLength;
        
        return;
ap@webkit.org's avatar
ap@webkit.org committed
898
    }
899 900 901
        
    default:
        ASSERT_NOT_REACHED();
ap@webkit.org's avatar
ap@webkit.org committed
902
    }
darin's avatar
darin committed
903 904
}

905
void JSArray::fillArgList(ExecState* exec, MarkedArgumentBuffer& args)
weinig@apple.com's avatar
weinig@apple.com committed
906
{
907
    switch (structure()->indexingType()) {
908
    case ArrayClass:
909 910
        return;
    
911
    case ARRAY_WITH_ARRAY_STORAGE_INDEXING_TYPES: {
912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930
        ArrayStorage* storage = m_butterfly->arrayStorage();
        
        WriteBarrier<Unknown>* vector = storage->m_vector;
        unsigned vectorEnd = min(storage->length(), storage->vectorLength());
        unsigned i = 0;
        for (; i < vectorEnd; ++i) {
            WriteBarrier<Unknown>& v = vector[i];
            if (!v)
                break;
            args.append(v.get());
        }
        
        for (; i < storage->length(); ++i)
            args.append(get(exec, i));
        return;
    }
        
    default:
        ASSERT_NOT_REACHED();
931
    }
weinig@apple.com's avatar
weinig@apple.com committed
932 933
}

934
void JSArray::copyToArguments(ExecState* exec, CallFrame* callFrame, uint32_t length)
935
{
936
    ASSERT(length == this->length());
937
    switch (structure()->indexingType()) {
938
    case ArrayClass:
939 940
        return;
        
941
    case ARRAY_WITH_ARRAY_STORAGE_INDEXING_TYPES: {
942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959
        ArrayStorage* storage = m_butterfly->arrayStorage();
        unsigned i = 0;
        WriteBarrier<Unknown>* vector = storage->m_vector;
        unsigned vectorEnd = min(length, storage->vectorLength());
        for (; i < vectorEnd; ++i) {
            WriteBarrier<Unknown>& v = vector[i];
            if (!v)
                break;
            callFrame->setArgument(i, v.get());
        }
        
        for (; i < length; ++i)
            callFrame->setArgument(i, get(exec, i));
        return;
    }
        
    default:
        ASSERT_NOT_REACHED();
960
    }
961 962
}

963
unsigned JSArray::compactForSorting(JSGlobalData& globalData)
darin's avatar
darin committed
964
{
965
    ASSERT(!inSparseIndexingMode());
darin's avatar
darin committed
966

967
    switch (structure()->indexingType()) {
968
    case ArrayClass:
969
        return 0;
970

971 972 973 974 975 976 977 978 979 980 981 982
    case ArrayWithArrayStorage: {
        ArrayStorage* storage = m_butterfly->arrayStorage();
        
        unsigned usedVectorLength = min(storage->length(), storage->vectorLength());
        
        unsigned numDefined = 0;
        unsigned numUndefined = 0;
        
        for (; numDefined < usedVectorLength; ++numDefined) {
            JSValue v = storage->m_vector[numDefined].get();
            if (!v || v.isUndefined())
                break;
darin's avatar
darin committed
983
        }
984 985 986 987 988 989 990 991 992
        
        for (unsigned i = numDefined; i < usedVectorLength; ++i) {
            JSValue v = storage->m_vector[i].get();
            if (v) {
                if (v.isUndefined())
                    ++numUndefined;
                else
                    storage->m_vector[numDefined++].setWithoutWriteBarrier(v);
            }
darin's avatar
darin committed
993
        }
994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012
        
        unsigned newUsedVectorLength = numDefined + numUndefined;
        
        if (SparseArrayValueMap* map = storage->m_sparseMap.get()) {
            newUsedVectorLength += map->size();
            if (newUsedVectorLength > storage->vectorLength()) {
                // Check that it is possible to allocate an array large enough to hold all the entries - if not,
                // exception is thrown by caller.
                if ((newUsedVectorLength > MAX_STORAGE_VECTOR_LENGTH) || !increaseVectorLength(globalData, newUsedVectorLength))
                    return 0;
                
                storage = m_butterfly->arrayStorage();
            }
            
            SparseArrayValueMap::const_iterator end = map->end();
            for (SparseArrayValueMap::const_iterator it = map->begin(); it != end; ++it)
                storage->m_vector[numDefined++].setWithoutWriteBarrier(it->second.getNonSparseMode());
            
            deallocateSparseIndexMap();
1013
        }
1014 1015 1016 1017 1018 1019 1020 1021 1022
        
        for (unsigned i = numDefined; i < newUsedVectorLength; ++i)
            storage->m_vector[i].setUndefined();
        for (unsigned i = newUsedVectorLength; i < usedVectorLength; ++i)
            storage->m_vector[i].clear();
        
        storage->m_numValuesInVector = newUsedVectorLength;
        
        return numDefined;
1023
    }
1024 1025 1026 1027
        
    default:
        ASSERT_NOT_REACHED();
        return 0;
1028 1029 1030 1031
    }
}


1032
} // namespace JSC