2011-01-18 Darin Adler <darin@apple.com>
[WebKit-https.git] / Source / JavaScriptCore / runtime / RegExpPrototype.cpp
1 /*
2  *  Copyright (C) 1999-2000 Harri Porten (porten@kde.org)
3  *  Copyright (C) 2003, 2007, 2008 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 "RegExpPrototype.h"
23
24 #include "ArrayPrototype.h"
25 #include "Error.h"
26 #include "JSArray.h"
27 #include "JSFunction.h"
28 #include "JSObject.h"
29 #include "JSString.h"
30 #include "JSStringBuilder.h"
31 #include "JSValue.h"
32 #include "ObjectPrototype.h"
33 #include "PrototypeFunction.h"
34 #include "RegExpObject.h"
35 #include "RegExp.h"
36 #include "RegExpCache.h"
37 #include "StringRecursionChecker.h"
38 #include "UStringConcatenate.h"
39
40 namespace JSC {
41
42 ASSERT_CLASS_FITS_IN_CELL(RegExpPrototype);
43
44 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState*);
45 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState*);
46 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState*);
47 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState*);
48
49 // ECMA 15.10.5
50
51 RegExpPrototype::RegExpPrototype(ExecState* exec, JSGlobalObject* globalObject, NonNullPassRefPtr<Structure> structure, Structure* prototypeFunctionStructure)
52     : RegExpObject(globalObject, structure, RegExp::create(&exec->globalData(), "", ""))
53 {
54     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().compile, regExpProtoFuncCompile), DontEnum);
55     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().exec, regExpProtoFuncExec), DontEnum);
56     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().test, regExpProtoFuncTest), DontEnum);
57     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().toString, regExpProtoFuncToString), DontEnum);
58 }
59
60 // ------------------------------ Functions ---------------------------
61
62 EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState* exec)
63 {
64     JSValue thisValue = exec->hostThisValue();
65     if (!thisValue.inherits(&RegExpObject::info))
66         return throwVMTypeError(exec);
67     return JSValue::encode(asRegExpObject(thisValue)->test(exec));
68 }
69
70 EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState* exec)
71 {
72     JSValue thisValue = exec->hostThisValue();
73     if (!thisValue.inherits(&RegExpObject::info))
74         return throwVMTypeError(exec);
75     return JSValue::encode(asRegExpObject(thisValue)->exec(exec));
76 }
77
78 EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState* exec)
79 {
80     JSValue thisValue = exec->hostThisValue();
81     if (!thisValue.inherits(&RegExpObject::info))
82         return throwVMTypeError(exec);
83
84     RefPtr<RegExp> regExp;
85     JSValue arg0 = exec->argument(0);
86     JSValue arg1 = exec->argument(1);
87     
88     if (arg0.inherits(&RegExpObject::info)) {
89         if (!arg1.isUndefined())
90             return throwVMError(exec, createTypeError(exec, "Cannot supply flags when constructing one RegExp from another."));
91         regExp = asRegExpObject(arg0)->regExp();
92     } else {
93         UString pattern = !exec->argumentCount() ? UString("") : arg0.toString(exec);
94         UString flags = arg1.isUndefined() ? UString("") : arg1.toString(exec);
95         regExp = exec->globalData().regExpCache()->lookupOrCreate(pattern, flags);
96     }
97
98     if (!regExp->isValid())
99         return throwVMError(exec, createSyntaxError(exec, regExp->errorMessage()));
100
101     asRegExpObject(thisValue)->setRegExp(regExp.release());
102     asRegExpObject(thisValue)->setLastIndex(0);
103     return JSValue::encode(jsUndefined());
104 }
105
106 EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState* exec)
107 {
108     JSValue thisValue = exec->hostThisValue();
109     if (!thisValue.inherits(&RegExpObject::info)) {
110         if (thisValue.inherits(&RegExpPrototype::info))
111             return JSValue::encode(jsNontrivialString(exec, "//"));
112         return throwVMTypeError(exec);
113     }
114
115     RegExpObject* thisObject = asRegExpObject(thisValue);
116
117     StringRecursionChecker checker(exec, thisObject);
118     if (EncodedJSValue earlyReturnValue = checker.earlyReturnValue())
119         return earlyReturnValue;
120
121     char postfix[5] = { '/', 0, 0, 0, 0 };
122     int index = 1;
123     if (thisObject->get(exec, exec->propertyNames().global).toBoolean(exec))
124         postfix[index++] = 'g';
125     if (thisObject->get(exec, exec->propertyNames().ignoreCase).toBoolean(exec))
126         postfix[index++] = 'i';
127     if (thisObject->get(exec, exec->propertyNames().multiline).toBoolean(exec))
128         postfix[index] = 'm';
129     UString source = thisObject->get(exec, exec->propertyNames().source).toString(exec);
130     // If source is empty, use "/(?:)/" to avoid colliding with comment syntax
131     return JSValue::encode(jsMakeNontrivialString(exec, "/", source.length() ? source : UString("(?:)"), postfix));
132 }
133
134 } // namespace JSC