2011-01-27 Oliver Hunt <oliver@apple.com>
[WebKit-https.git] / Source / JavaScriptCore / runtime / FunctionPrototype.cpp
1 /*
2  *  Copyright (C) 1999-2001 Harri Porten (porten@kde.org)
3  *  Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reserved.
4  *
5  *  This library is free software; you can redistribute it and/or
6  *  modify it under the terms of the GNU Lesser General Public
7  *  License as published by the Free Software Foundation; either
8  *  version 2 of the License, or (at your option) any later version.
9  *
10  *  This library is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  *  Lesser General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Lesser General Public
16  *  License along with this library; if not, write to the Free Software
17  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
18  *
19  */
20
21 #include "config.h"
22 #include "FunctionPrototype.h"
23
24 #include "Arguments.h"
25 #include "JSArray.h"
26 #include "JSFunction.h"
27 #include "JSString.h"
28 #include "JSStringBuilder.h"
29 #include "Interpreter.h"
30 #include "Lexer.h"
31 #include "PrototypeFunction.h"
32
33 namespace JSC {
34
35 ASSERT_CLASS_FITS_IN_CELL(FunctionPrototype);
36
37 static EncodedJSValue JSC_HOST_CALL functionProtoFuncToString(ExecState*);
38 static EncodedJSValue JSC_HOST_CALL functionProtoFuncApply(ExecState*);
39 static EncodedJSValue JSC_HOST_CALL functionProtoFuncCall(ExecState*);
40
41 FunctionPrototype::FunctionPrototype(ExecState* exec, JSGlobalObject* globalObject, NonNullPassRefPtr<Structure> structure)
42     : InternalFunction(&exec->globalData(), globalObject, structure, exec->propertyNames().nullIdentifier)
43 {
44     putDirectWithoutTransition(exec->globalData(), exec->propertyNames().length, jsNumber(0), DontDelete | ReadOnly | DontEnum);
45 }
46
47 void FunctionPrototype::addFunctionProperties(ExecState* exec, JSGlobalObject* globalObject, Structure* prototypeFunctionStructure, NativeFunctionWrapper** callFunction, NativeFunctionWrapper** applyFunction)
48 {
49     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().toString, functionProtoFuncToString), DontEnum);
50     *applyFunction = new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 2, exec->propertyNames().apply, functionProtoFuncApply);
51     putDirectFunctionWithoutTransition(exec, *applyFunction, DontEnum);
52     *callFunction = new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 1, exec->propertyNames().call, functionProtoFuncCall);
53     putDirectFunctionWithoutTransition(exec, *callFunction, DontEnum);
54 }
55
56 static EncodedJSValue JSC_HOST_CALL callFunctionPrototype(ExecState*)
57 {
58     return JSValue::encode(jsUndefined());
59 }
60
61 // ECMA 15.3.4
62 CallType FunctionPrototype::getCallData(CallData& callData)
63 {
64     callData.native.function = callFunctionPrototype;
65     return CallTypeHost;
66 }
67
68 // Functions
69
70 // Compatibility hack for the Optimost JavaScript library. (See <rdar://problem/6595040>.)
71 static inline void insertSemicolonIfNeeded(UString& functionBody)
72 {
73     ASSERT(functionBody[0] == '{');
74     ASSERT(functionBody[functionBody.length() - 1] == '}');
75
76     for (size_t i = functionBody.length() - 2; i > 0; --i) {
77         UChar ch = functionBody[i];
78         if (!Lexer::isWhiteSpace(ch) && !Lexer::isLineTerminator(ch)) {
79             if (ch != ';' && ch != '}')
80                 functionBody = makeUString(functionBody.substringSharingImpl(0, i + 1), ";", functionBody.substringSharingImpl(i + 1, functionBody.length() - (i + 1)));
81             return;
82         }
83     }
84 }
85
86 EncodedJSValue JSC_HOST_CALL functionProtoFuncToString(ExecState* exec)
87 {
88     JSValue thisValue = exec->hostThisValue();
89     if (thisValue.inherits(&JSFunction::info)) {
90         JSFunction* function = asFunction(thisValue);
91         if (function->isHostFunction())
92             return JSValue::encode(jsMakeNontrivialString(exec, "function ", function->name(exec), "() {\n    [native code]\n}"));
93         FunctionExecutable* executable = function->jsExecutable();
94         UString sourceString = executable->source().toString();
95         insertSemicolonIfNeeded(sourceString);
96         return JSValue::encode(jsMakeNontrivialString(exec, "function ", function->name(exec), "(", executable->paramString(), ") ", sourceString));
97     }
98
99     if (thisValue.inherits(&InternalFunction::info)) {
100         InternalFunction* function = asInternalFunction(thisValue);
101         return JSValue::encode(jsMakeNontrivialString(exec, "function ", function->name(exec), "() {\n    [native code]\n}"));
102     }
103
104     return throwVMTypeError(exec);
105 }
106
107 EncodedJSValue JSC_HOST_CALL functionProtoFuncApply(ExecState* exec)
108 {
109     JSValue thisValue = exec->hostThisValue();
110     CallData callData;
111     CallType callType = getCallData(thisValue, callData);
112     if (callType == CallTypeNone)
113         return throwVMTypeError(exec);
114
115     JSValue array = exec->argument(1);
116
117     MarkedArgumentBuffer applyArgs;
118     if (!array.isUndefinedOrNull()) {
119         if (!array.isObject())
120             return throwVMTypeError(exec);
121         if (asObject(array)->classInfo() == &Arguments::info)
122             asArguments(array)->fillArgList(exec, applyArgs);
123         else if (isJSArray(&exec->globalData(), array))
124             asArray(array)->fillArgList(exec, applyArgs);
125         else if (asObject(array)->inherits(&JSArray::info)) {
126             unsigned length = asArray(array)->get(exec, exec->propertyNames().length).toUInt32(exec);
127             for (unsigned i = 0; i < length; ++i)
128                 applyArgs.append(asArray(array)->get(exec, i));
129         } else
130             return throwVMTypeError(exec);
131     }
132
133     return JSValue::encode(call(exec, thisValue, callType, callData, exec->argument(0), applyArgs));
134 }
135
136 EncodedJSValue JSC_HOST_CALL functionProtoFuncCall(ExecState* exec)
137 {
138     JSValue thisValue = exec->hostThisValue();
139     CallData callData;
140     CallType callType = getCallData(thisValue, callData);
141     if (callType == CallTypeNone)
142         return throwVMTypeError(exec);
143
144     ArgList args(exec);
145     ArgList callArgs;
146     args.getSlice(1, callArgs);
147     return JSValue::encode(call(exec, thisValue, callType, callData, exec->argument(0), callArgs));
148 }
149
150 } // namespace JSC