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 "UStringConcatenate.h"
  38
  39 namespace JSC {
  40
  41 ASSERT_CLASS_FITS_IN_CELL(RegExpPrototype);
  42
  43 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState*);
  44 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState*);
  45 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState*);
  46 static EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState*);
  47
  48 // ECMA 15.10.5
  49
  50 RegExpPrototype::RegExpPrototype(ExecState* exec, JSGlobalObject* globalObject, NonNullPassRefPtr<Structure> structure, Structure* prototypeFunctionStructure)
  51     : RegExpObject(globalObject, structure, RegExp::create(&exec->globalData(), "", ""))
  52 {
  53     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().compile, regExpProtoFuncCompile), DontEnum);
  54     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().exec, regExpProtoFuncExec), DontEnum);
  55     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().test, regExpProtoFuncTest), DontEnum);
  56     putDirectFunctionWithoutTransition(exec, new (exec) NativeFunctionWrapper(exec, globalObject, prototypeFunctionStructure, 0, exec->propertyNames().toString, regExpProtoFuncToString), DontEnum);
  57 }
  58
  59 // ------------------------------ Functions ---------------------------
  60
  61 EncodedJSValue JSC_HOST_CALL regExpProtoFuncTest(ExecState* exec)
  62 {
  63     JSValue thisValue = exec->hostThisValue();
  64     if (!thisValue.inherits(&RegExpObject::info))
  65         return throwVMTypeError(exec);
  66     return JSValue::encode(asRegExpObject(thisValue)->test(exec));
  67 }
  68
  69 EncodedJSValue JSC_HOST_CALL regExpProtoFuncExec(ExecState* exec)
  70 {
  71     JSValue thisValue = exec->hostThisValue();
  72     if (!thisValue.inherits(&RegExpObject::info))
  73         return throwVMTypeError(exec);
  74     return JSValue::encode(asRegExpObject(thisValue)->exec(exec));
  75 }
  76
  77 EncodedJSValue JSC_HOST_CALL regExpProtoFuncCompile(ExecState* exec)
  78 {
  79     JSValue thisValue = exec->hostThisValue();
  80     if (!thisValue.inherits(&RegExpObject::info))
  81         return throwVMTypeError(exec);
  82
  83     RefPtr<RegExp> regExp;
  84     JSValue arg0 = exec->argument(0);
  85     JSValue arg1 = exec->argument(1);
  86
  87     if (arg0.inherits(&RegExpObject::info)) {
  88         if (!arg1.isUndefined())
  89             return throwVMError(exec, createTypeError(exec, "Cannot supply flags when constructing one RegExp from another."));
  90         regExp = asRegExpObject(arg0)->regExp();
  91     } else {
  92         UString pattern = !exec->argumentCount() ? UString("") : arg0.toString(exec);
  93         UString flags = arg1.isUndefined() ? UString("") : arg1.toString(exec);
  94         regExp = exec->globalData().regExpCache()->lookupOrCreate(pattern, flags);
  95     }
  96
  97     if (!regExp->isValid())
  98         return throwVMError(exec, createSyntaxError(exec, regExp->errorMessage()));
  99
 100     asRegExpObject(thisValue)->setRegExp(regExp.release());
 101     asRegExpObject(thisValue)->setLastIndex(0);
 102     return JSValue::encode(jsUndefined());
 103 }
 104
 105 EncodedJSValue JSC_HOST_CALL regExpProtoFuncToString(ExecState* exec)
 106 {
 107     JSValue thisValue = exec->hostThisValue();
 108     if (!thisValue.inherits(&RegExpObject::info)) {
 109         if (thisValue.inherits(&RegExpPrototype::info))
 110             return JSValue::encode(jsNontrivialString(exec, "//"));
 111         return throwVMTypeError(exec);
 112     }
 113
 114     char postfix[5] = { '/', 0, 0, 0, 0 };
 115     int index = 1;
 116     if (asRegExpObject(thisValue)->get(exec, exec->propertyNames().global).toBoolean(exec))
 117         postfix[index++] = 'g';
 118     if (asRegExpObject(thisValue)->get(exec, exec->propertyNames().ignoreCase).toBoolean(exec))
 119         postfix[index++] = 'i';
 120     if (asRegExpObject(thisValue)->get(exec, exec->propertyNames().multiline).toBoolean(exec))
 121         postfix[index] = 'm';
 122     UString source = asRegExpObject(thisValue)->get(exec, exec->propertyNames().source).toString(exec);
 123     // If source is empty, use "/(?:)/" to avoid colliding with comment syntax
 124     return JSValue::encode(jsMakeNontrivialString(exec, "/", source.length() ? source : UString("(?:)"), postfix));
 125 }
 126
 127 } // namespace JSC