Improve JSC use of Strings after the UString->String change
[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 "RegExpObject.h"
34 #include "RegExp.h"
35 #include "RegExpCache.h"
36 #include "StringRecursionChecker.h"
37
38 namespace JSC {
39
40 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState*);
41 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState*);
42 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState*);
43 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState*);
44
45 }
46
47 #include "RegExpPrototype.lut.h"
48
49 namespace JSC {
50
51 const ClassInfo RegExpPrototype::s_info = { "RegExp", &RegExpObject::s_info, 0, ExecState::regExpPrototypeTable, CREATE_METHOD_TABLE(RegExpPrototype) };
52
53 /* Source for RegExpPrototype.lut.h
54 @begin regExpPrototypeTable
55   compile   regExpProtoFuncCompile      DontEnum|Function 2
56   exec      regExpProtoFuncExec         DontEnum|Function 1
57   test      regExpProtoFuncTest         DontEnum|Function 1
58   toString  regExpProtoFuncToString     DontEnum|Function 0
59 @end
60 */
61
62 ASSERT_CLASS_FITS_IN_CELL(RegExpPrototype);
63
64 RegExpPrototype::RegExpPrototype(JSGlobalObject* globalObject, Structure* structure, RegExp* regExp)
65     : RegExpObject(globalObject, structure, regExp)
66 {
67 }
68
69 bool RegExpPrototype::getOwnPropertySlot(JSCell* cell, ExecState* exec, PropertyName propertyName, PropertySlot &slot)
70 {
71     return getStaticFunctionSlot<RegExpObject>(exec, ExecState::regExpPrototypeTable(exec), jsCast<RegExpPrototype*>(cell), propertyName, slot);
72 }
73
74 bool RegExpPrototype::getOwnPropertyDescriptor(JSObject* object, ExecState* exec, PropertyName propertyName, PropertyDescriptor& descriptor)
75 {
76     return getStaticFunctionDescriptor<RegExpObject>(exec, ExecState::regExpPrototypeTable(exec), jsCast<RegExpPrototype*>(object), propertyName, descriptor);
77 }
78
79 // ------------------------------ Functions ---------------------------
80
81 EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState* exec)
82 {
83     JSValue thisValue = exec->hostThisValue();
84     if (!thisValue.inherits(&RegExpObject::s_info))
85         return throwVMTypeError(exec);
86     return JSValue::encode(jsBoolean(asRegExpObject(thisValue)->test(exec, exec->argument(0).toString(exec))));
87 }
88
89 EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState* exec)
90 {
91     JSValue thisValue = exec->hostThisValue();
92     if (!thisValue.inherits(&RegExpObject::s_info))
93         return throwVMTypeError(exec);
94     return JSValue::encode(asRegExpObject(thisValue)->exec(exec, exec->argument(0).toString(exec)));
95 }
96
97 EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState* exec)
98 {
99     JSValue thisValue = exec->hostThisValue();
100     if (!thisValue.inherits(&RegExpObject::s_info))
101         return throwVMTypeError(exec);
102
103     RegExp* regExp;
104     JSValue arg0 = exec->argument(0);
105     JSValue arg1 = exec->argument(1);
106     
107     if (arg0.inherits(&RegExpObject::s_info)) {
108         if (!arg1.isUndefined())
109             return throwVMError(exec, createTypeError(exec, ASCIILiteral("Cannot supply flags when constructing one RegExp from another.")));
110         regExp = asRegExpObject(arg0)->regExp();
111     } else {
112         String pattern = !exec->argumentCount() ? String("") : arg0.toString(exec)->value(exec);
113         if (exec->hadException())
114             return JSValue::encode(jsUndefined());
115
116         RegExpFlags flags = NoFlags;
117         if (!arg1.isUndefined()) {
118             flags = regExpFlags(arg1.toString(exec)->value(exec));
119             if (exec->hadException())
120                 return JSValue::encode(jsUndefined());
121             if (flags == InvalidFlags)
122                 return throwVMError(exec, createSyntaxError(exec, ASCIILiteral("Invalid flags supplied to RegExp constructor.")));
123         }
124         regExp = RegExp::create(exec->globalData(), pattern, flags);
125     }
126
127     if (!regExp->isValid())
128         return throwVMError(exec, createSyntaxError(exec, regExp->errorMessage()));
129
130     asRegExpObject(thisValue)->setRegExp(exec->globalData(), regExp);
131     asRegExpObject(thisValue)->setLastIndex(exec, 0);
132     return JSValue::encode(jsUndefined());
133 }
134
135 EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState* exec)
136 {
137     JSValue thisValue = exec->hostThisValue();
138     if (!thisValue.inherits(&RegExpObject::s_info))
139         return throwVMTypeError(exec);
140
141     RegExpObject* thisObject = asRegExpObject(thisValue);
142
143     StringRecursionChecker checker(exec, thisObject);
144     if (JSValue earlyReturnValue = checker.earlyReturnValue())
145         return JSValue::encode(earlyReturnValue);
146
147     char postfix[5] = { '/', 0, 0, 0, 0 };
148     int index = 1;
149     if (thisObject->get(exec, exec->propertyNames().global).toBoolean(exec))
150         postfix[index++] = 'g';
151     if (thisObject->get(exec, exec->propertyNames().ignoreCase).toBoolean(exec))
152         postfix[index++] = 'i';
153     if (thisObject->get(exec, exec->propertyNames().multiline).toBoolean(exec))
154         postfix[index] = 'm';
155     String source = thisObject->get(exec, exec->propertyNames().source).toString(exec)->value(exec);
156     // If source is empty, use "/(?:)/" to avoid colliding with comment syntax
157     return JSValue::encode(jsMakeNontrivialString(exec, "/", source, postfix));
158 }
159
160 } // namespace JSC