/*

 * Copyright (C) 2013 Apple Inc. All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR

 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

 */

#include "config.h"

#include "ArityCheckFailReturnThunks.h"

#if ENABLE(JIT)

#include "AssemblyHelpers.h"

namespace JSC {

ArityCheckFailReturnThunks::ArityCheckFailReturnThunks() { }

ArityCheckFailReturnThunks::~ArityCheckFailReturnThunks() { }

CodeLocationLabel* ArityCheckFailReturnThunks::returnPCsFor(

    VM& vm, unsigned numExpectedArgumentsIncludingThis)

{

    ASSERT(numExpectedArgumentsIncudingThis >= 1);

    HashMap<unsigned, std::unique_ptr<Data>>::iterator iter =

        m_map.find(numExpectedArgumentsIncludingThis);

    if (iter != m_map.end())

        return iter->value.m_returnPCs.get();

    AssemblyHelpers jit(&vm, 0);

    Vector<AssemblyHelpers::Label> returnLabels;

    for (unsigned i = 0; i < numExpectedArgumentsIncludingThis - 1; ++i) {

        returnLabels.append(jit.label());

        jit.loadPtr(

            AssemblyHelpers::Address(

                AssemblyHelpers::stackPointerRegister,

                (JSStack::CallFrameHeaderSize - JSStack::CallerFrameAndPCSize + numExpectedArgumentsIncludingThis) * sizeof(Register)),

            GPRInfo::regT1);

        jit.addPtr(

            AssemblyHelpers::TrustedImm32(

                (numExpectedArgumentsIncludingThis - i + 1) * sizeof(Register)),

            AssemblyHelpers::stackPointerRegister);

        jit.jump(GPRInfo::regT1);

    }

    LinkBuffer linkBuffer(vm, &jit, GLOBAL_THUNK_ID);

    std::unique_ptr<Data> data = std::make_unique<Data>();

    data->m_returnPCs = std::make_unique<CodeLocationLabel[]>(returnLabels.size());

    for (unsigned i = returnLabels.size(); i--;)

        data->m_returnPCs[i] = linkBuffer.locationOf(returnLabels[i]);

    data->m_ref = FINALIZE_CODE(linkBuffer, ("Arity check fail return thunk for numArgs = %u", numExpectedArgumentsIncludingThis));

    CodeLocationLabel* result = data->m_returnPCs.get();

    m_map.add(numExpectedArgumentsIncludingThis, std::move(data));

    return result;

}

} // namespace JSC

#endif // ENABLE(JIT)

/*

 * Copyright (C) 2013 Apple Inc. All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR

 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

 */

#ifndef ArityCheckFailReturnThunks_h

#define ArityCheckFailReturnThunks_h

#if ENABLE(JIT)

#include "CodeLocation.h"

#include <wtf/HashMap.h>

namespace JSC {

class ArityCheckFailReturnThunks {

public:

    ArityCheckFailReturnThunks();

    ~ArityCheckFailReturnThunks();

    CodeLocationLabel* returnPCsFor(

        VM&, unsigned numExpectedArgumentsIncludingThis);

private:

    struct Data {

        std::unique_ptr<CodeLocationLabel[]> m_returnPCs;

        MacroAssemblerCodeRef m_ref;

    };

    HashMap<unsigned, std::unique_ptr<Data>> m_map;

};

} // namespace JSC

#endif // ENABLE(JIT)

#endif // ArityCheckFailReturnThunks_h

    arityCheckFailReturnThunks = std::make_unique<ArityCheckFailReturnThunks>();

    class ArityCheckFailReturnThunks;

        std::unique_ptr<ArityCheckFailReturnThunks> arityCheckFailReturnThunks;