/*

 *  Copyright (C) 1999-2002 Harri Porten (porten@kde.org)

 *  Copyright (C) 2001 Peter Kelly (pmk@post.com)

 *  Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reserved.

 *  Copyright (C) 2007 Cameron Zwarich (cwzwarich@uwaterloo.ca)

 *  Copyright (C) 2007 Maks Orlovich

 *

 *  This library is free software; you can redistribute it and/or

 *  modify it under the terms of the GNU Library 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

 *  Library General Public License for more details.

 *

 *  You should have received a copy of the GNU Library General Public License

 *  along with this library; see the file COPYING.LIB.  If not, write to

 *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

 *  Boston, MA 02110-1301, USA.

 *

 */

#include "config.h"

#include "JSFunction.h"

#include "CodeBlock.h"

#include "CommonIdentifiers.h"

#include "CallFrame.h"

#include "ExceptionHelpers.h"

#include "FunctionPrototype.h"

#include "JSGlobalObject.h"

#include "JSNotAnObject.h"

#include "Interpreter.h"

#include "ObjectPrototype.h"

#include "Parser.h"

#include "PropertyNameArray.h"

#include "ScopeChainMark.h"

using namespace WTF;

using namespace Unicode;

namespace JSC {

#if ENABLE(JIT)

EncodedJSValue JSC_HOST_CALL callHostFunctionAsConstructor(ExecState* exec)

{

    CodeBlock* codeBlock = exec->callerFrame()->codeBlock();

    unsigned vPCIndex = codeBlock->bytecodeOffset(exec, exec->returnPC());

    return throwVMError(exec, createNotAConstructorError(exec, exec->callee(), vPCIndex, codeBlock));

}

#endif

ASSERT_CLASS_FITS_IN_CELL(JSFunction);

const ClassInfo JSFunction::info = { "Function", 0, 0, 0 };

bool JSFunction::isHostFunctionNonInline() const

{

    return isHostFunction();

}

JSFunction::JSFunction(NonNullPassRefPtr<Structure> structure)

    : Base(structure)

    , m_executable(adoptRef(new VPtrHackExecutable()))

    , m_scopeChain(NoScopeChain())

{

}

#if ENABLE(JIT)

JSFunction::JSFunction(ExecState* exec, JSGlobalObject* globalObject, NonNullPassRefPtr<Structure> structure, int length, const Identifier& name, PassRefPtr<NativeExecutable> thunk)

    : Base(globalObject, structure)

    , m_executable(thunk)

    , m_scopeChain(globalObject->globalScopeChain())

{

    putDirect(exec->globalData().propertyNames->name, jsString(exec, name.isNull() ? "" : name.ustring()), DontDelete | ReadOnly | DontEnum);

    putDirect(exec->propertyNames().length, jsNumber(exec, length), DontDelete | ReadOnly | DontEnum);

}

#endif

JSFunction::JSFunction(ExecState* exec, JSGlobalObject* globalObject, NonNullPassRefPtr<Structure> structure, int length, const Identifier& name, NativeFunction func)

    : Base(globalObject, structure)

#if ENABLE(JIT)

    , m_executable(exec->globalData().getHostFunction(func))

#endif

    , m_scopeChain(globalObject->globalScopeChain())

{

    putDirect(exec->globalData().propertyNames->name, jsString(exec, name.isNull() ? "" : name.ustring()), DontDelete | ReadOnly | DontEnum);

#if ENABLE(JIT)

    putDirect(exec->propertyNames().length, jsNumber(exec, length), DontDelete | ReadOnly | DontEnum);

#else

    UNUSED_PARAM(length);

    UNUSED_PARAM(func);

    ASSERT_NOT_REACHED();

#endif

}

JSFunction::JSFunction(ExecState* exec, NonNullPassRefPtr<FunctionExecutable> executable, ScopeChainNode* scopeChainNode)

    : Base(scopeChainNode->globalObject, scopeChainNode->globalObject->functionStructure())

    , m_executable(executable)

    , m_scopeChain(scopeChainNode)

{

    const Identifier& name = static_cast<FunctionExecutable*>(m_executable.get())->name();

    putDirect(exec->globalData().propertyNames->name, jsString(exec, name.isNull() ? "" : name.ustring()), DontDelete | ReadOnly | DontEnum);

}

JSFunction::~JSFunction()

{

    ASSERT(vptr() == JSGlobalData::jsFunctionVPtr);

    // JIT code for other functions may have had calls linked directly to the code for this function; these links

    // are based on a check for the this pointer value for this JSFunction - which will no longer be valid once

    // this memory is freed and may be reused (potentially for another, different JSFunction).

    if (!isHostFunction()) {

#if ENABLE(JIT_OPTIMIZE_CALL)

        ASSERT(m_executable);

        if (jsExecutable()->isGeneratedForCall())

            jsExecutable()->generatedBytecodeForCall().unlinkCallers();

        if (jsExecutable()->isGeneratedForConstruct())

            jsExecutable()->generatedBytecodeForConstruct().unlinkCallers();

#endif

    }

}

const UString& JSFunction::name(ExecState* exec)

{

    return asString(getDirect(exec->globalData().propertyNames->name))->tryGetValue();

}

const UString JSFunction::displayName(ExecState* exec)

{

    JSValue displayName = getDirect(exec->globalData().propertyNames->displayName);

    if (displayName && isJSString(&exec->globalData(), displayName))

        return asString(displayName)->tryGetValue();

    return UString::null();

}

const UString JSFunction::calculatedDisplayName(ExecState* exec)

{

    const UString explicitName = displayName(exec);

    if (!explicitName.isEmpty())

        return explicitName;

    return name(exec);

}

void JSFunction::markChildren(MarkStack& markStack)

{

    Base::markChildren(markStack);

    if (!isHostFunction()) {

        jsExecutable()->markAggregate(markStack);

        scope().markAggregate(markStack);

    }

}

CallType JSFunction::getCallData(CallData& callData)

{

#if ENABLE(JIT)

    if (isHostFunction()) {

        callData.native.function = nativeFunction();

        return CallTypeHost;

    }

#endif

    callData.js.functionExecutable = jsExecutable();

    callData.js.scopeChain = scope().node();

    return CallTypeJS;

}

JSValue JSFunction::argumentsGetter(ExecState* exec, JSValue slotBase, const Identifier&)

{

    JSFunction* thisObj = asFunction(slotBase);

    ASSERT(!thisObj->isHostFunction());

    return exec->interpreter()->retrieveArguments(exec, thisObj);

}

JSValue JSFunction::callerGetter(ExecState* exec, JSValue slotBase, const Identifier&)

{

    JSFunction* thisObj = asFunction(slotBase);

    ASSERT(!thisObj->isHostFunction());

    return exec->interpreter()->retrieveCaller(exec, thisObj);

}

JSValue JSFunction::lengthGetter(ExecState* exec, JSValue slotBase, const Identifier&)

{

    JSFunction* thisObj = asFunction(slotBase);

    ASSERT(!thisObj->isHostFunction());

    return jsNumber(exec, thisObj->jsExecutable()->parameterCount());

}

bool JSFunction::getOwnPropertySlot(ExecState* exec, const Identifier& propertyName, PropertySlot& slot)

{

    if (isHostFunction())

        return Base::getOwnPropertySlot(exec, propertyName, slot);

    if (propertyName == exec->propertyNames().prototype) {

        JSValue* location = getDirectLocation(propertyName);

        if (!location) {

            JSObject* prototype = new (exec) JSObject(scope().globalObject()->emptyObjectStructure());

            prototype->putDirect(exec->propertyNames().constructor, this, DontEnum);

            putDirect(exec->propertyNames().prototype, prototype, DontDelete);

            location = getDirectLocation(propertyName);

        }

        slot.setValueSlot(this, location, offsetForLocation(location));

    }

    if (propertyName == exec->propertyNames().arguments) {

        slot.setCacheableCustom(this, argumentsGetter);

        return true;

    }

    if (propertyName == exec->propertyNames().length) {

        slot.setCacheableCustom(this, lengthGetter);

        return true;

    }

    if (propertyName == exec->propertyNames().caller) {

        slot.setCacheableCustom(this, callerGetter);

        return true;

    }

    return Base::getOwnPropertySlot(exec, propertyName, slot);

}

    bool JSFunction::getOwnPropertyDescriptor(ExecState* exec, const Identifier& propertyName, PropertyDescriptor& descriptor)

    {

        if (isHostFunction())

            return Base::getOwnPropertyDescriptor(exec, propertyName, descriptor);

        if (propertyName == exec->propertyNames().prototype) {

            PropertySlot slot;

            getOwnPropertySlot(exec, propertyName, slot);

            return Base::getOwnPropertyDescriptor(exec, propertyName, descriptor);

        }

        if (propertyName == exec->propertyNames().arguments) {

            descriptor.setDescriptor(exec->interpreter()->retrieveArguments(exec, this), ReadOnly | DontEnum | DontDelete);

            return true;

        }

        if (propertyName == exec->propertyNames().length) {

            descriptor.setDescriptor(jsNumber(exec, jsExecutable()->parameterCount()), ReadOnly | DontEnum | DontDelete);

            return true;

        }

        if (propertyName == exec->propertyNames().caller) {

            descriptor.setDescriptor(exec->interpreter()->retrieveCaller(exec, this), ReadOnly | DontEnum | DontDelete);

            return true;

        }

        return Base::getOwnPropertyDescriptor(exec, propertyName, descriptor);

    }

void JSFunction::getOwnPropertyNames(ExecState* exec, PropertyNameArray& propertyNames, EnumerationMode mode)

{

    if (!isHostFunction() && (mode == IncludeDontEnumProperties)) {

        propertyNames.add(exec->propertyNames().arguments);

        propertyNames.add(exec->propertyNames().callee);

        propertyNames.add(exec->propertyNames().caller);

        propertyNames.add(exec->propertyNames().length);

    }

    Base::getOwnPropertyNames(exec, propertyNames, mode);

}

void JSFunction::put(ExecState* exec, const Identifier& propertyName, JSValue value, PutPropertySlot& slot)

{

    if (isHostFunction()) {

        Base::put(exec, propertyName, value, slot);

        return;

    }

    if (propertyName == exec->propertyNames().arguments || propertyName == exec->propertyNames().length)

        return;

    Base::put(exec, propertyName, value, slot);

}

bool JSFunction::deleteProperty(ExecState* exec, const Identifier& propertyName)

{

    if (isHostFunction())

        return Base::deleteProperty(exec, propertyName);

    if (propertyName == exec->propertyNames().arguments || propertyName == exec->propertyNames().length)

        return false;

    return Base::deleteProperty(exec, propertyName);

}

// ECMA 13.2.2 [[Construct]]

ConstructType JSFunction::getConstructData(ConstructData& constructData)

{

    if (isHostFunction())

        return ConstructTypeNone;

    constructData.js.functionExecutable = jsExecutable();

    constructData.js.scopeChain = scope().node();

    return ConstructTypeJS;

}

} // namespace JSC