diff options
Diffstat (limited to 'Source/JavaScriptCore/jsc.cpp')
-rw-r--r-- | Source/JavaScriptCore/jsc.cpp | 568 |
1 files changed, 568 insertions, 0 deletions
diff --git a/Source/JavaScriptCore/jsc.cpp b/Source/JavaScriptCore/jsc.cpp new file mode 100644 index 0000000..dd4b677 --- /dev/null +++ b/Source/JavaScriptCore/jsc.cpp @@ -0,0 +1,568 @@ +/* + * Copyright (C) 1999-2000 Harri Porten (porten@kde.org) + * Copyright (C) 2004, 2005, 2006, 2007, 2008 Apple Inc. All rights reserved. + * Copyright (C) 2006 Bjoern Graf (bjoern.graf@gmail.com) + * + * 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 "BytecodeGenerator.h" +#include "Completion.h" +#include "CurrentTime.h" +#include "ExceptionHelpers.h" +#include "InitializeThreading.h" +#include "JSArray.h" +#include "JSFunction.h" +#include "JSLock.h" +#include "JSString.h" +#include "PrototypeFunction.h" +#include "SamplingTool.h" +#include <math.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#if !OS(WINDOWS) +#include <unistd.h> +#endif + +#if HAVE(READLINE) +#include <readline/history.h> +#include <readline/readline.h> +#endif + +#if HAVE(SYS_TIME_H) +#include <sys/time.h> +#endif + +#if HAVE(SIGNAL_H) +#include <signal.h> +#endif + +#if COMPILER(MSVC) && !OS(WINCE) +#include <crtdbg.h> +#include <mmsystem.h> +#include <windows.h> +#endif + +#if PLATFORM(QT) +#include <QCoreApplication> +#include <QDateTime> +#endif + +using namespace JSC; +using namespace WTF; + +static void cleanupGlobalData(JSGlobalData*); +static bool fillBufferWithContentsOfFile(const UString& fileName, Vector<char>& buffer); + +static EncodedJSValue JSC_HOST_CALL functionPrint(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionDebug(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionGC(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionVersion(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionRun(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionLoad(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionCheckSyntax(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionReadline(ExecState*); +static NO_RETURN_WITH_VALUE EncodedJSValue JSC_HOST_CALL functionQuit(ExecState*); + +#if ENABLE(SAMPLING_FLAGS) +static EncodedJSValue JSC_HOST_CALL functionSetSamplingFlags(ExecState*); +static EncodedJSValue JSC_HOST_CALL functionClearSamplingFlags(ExecState*); +#endif + +struct Script { + bool isFile; + char* argument; + + Script(bool isFile, char *argument) + : isFile(isFile) + , argument(argument) + { + } +}; + +struct Options { + Options() + : interactive(false) + , dump(false) + { + } + + bool interactive; + bool dump; + Vector<Script> scripts; + Vector<UString> arguments; +}; + +static const char interactivePrompt[] = "> "; +static const UString interpreterName("Interpreter"); + +class StopWatch { +public: + void start(); + void stop(); + long getElapsedMS(); // call stop() first + +private: + double m_startTime; + double m_stopTime; +}; + +void StopWatch::start() +{ + m_startTime = currentTime(); +} + +void StopWatch::stop() +{ + m_stopTime = currentTime(); +} + +long StopWatch::getElapsedMS() +{ + return static_cast<long>((m_stopTime - m_startTime) * 1000); +} + +class GlobalObject : public JSGlobalObject { +public: + GlobalObject(const Vector<UString>& arguments); + virtual UString className() const { return "global"; } +}; +COMPILE_ASSERT(!IsInteger<GlobalObject>::value, WTF_IsInteger_GlobalObject_false); +ASSERT_CLASS_FITS_IN_CELL(GlobalObject); + +GlobalObject::GlobalObject(const Vector<UString>& arguments) + : JSGlobalObject() +{ + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "debug"), functionDebug)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "print"), functionPrint)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 0, Identifier(globalExec(), "quit"), functionQuit)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 0, Identifier(globalExec(), "gc"), functionGC)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "version"), functionVersion)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "run"), functionRun)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "load"), functionLoad)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "checkSyntax"), functionCheckSyntax)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 0, Identifier(globalExec(), "readline"), functionReadline)); + +#if ENABLE(SAMPLING_FLAGS) + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "setSamplingFlags"), functionSetSamplingFlags)); + putDirectFunction(globalExec(), new (globalExec()) NativeFunctionWrapper(globalExec(), this, prototypeFunctionStructure(), 1, Identifier(globalExec(), "clearSamplingFlags"), functionClearSamplingFlags)); +#endif + + JSObject* array = constructEmptyArray(globalExec()); + for (size_t i = 0; i < arguments.size(); ++i) + array->put(globalExec(), i, jsString(globalExec(), arguments[i])); + putDirect(Identifier(globalExec(), "arguments"), array); +} + +EncodedJSValue JSC_HOST_CALL functionPrint(ExecState* exec) +{ + for (unsigned i = 0; i < exec->argumentCount(); ++i) { + if (i) + putchar(' '); + + printf("%s", exec->argument(i).toString(exec).utf8().data()); + } + + putchar('\n'); + fflush(stdout); + return JSValue::encode(jsUndefined()); +} + +EncodedJSValue JSC_HOST_CALL functionDebug(ExecState* exec) +{ + fprintf(stderr, "--> %s\n", exec->argument(0).toString(exec).utf8().data()); + return JSValue::encode(jsUndefined()); +} + +EncodedJSValue JSC_HOST_CALL functionGC(ExecState* exec) +{ + JSLock lock(SilenceAssertionsOnly); + exec->heap()->collectAllGarbage(); + return JSValue::encode(jsUndefined()); +} + +EncodedJSValue JSC_HOST_CALL functionVersion(ExecState*) +{ + // We need this function for compatibility with the Mozilla JS tests but for now + // we don't actually do any version-specific handling + return JSValue::encode(jsUndefined()); +} + +EncodedJSValue JSC_HOST_CALL functionRun(ExecState* exec) +{ + UString fileName = exec->argument(0).toString(exec); + Vector<char> script; + if (!fillBufferWithContentsOfFile(fileName, script)) + return JSValue::encode(throwError(exec, createError(exec, "Could not open file."))); + + GlobalObject* globalObject = new (&exec->globalData()) GlobalObject(Vector<UString>()); + + StopWatch stopWatch; + stopWatch.start(); + evaluate(globalObject->globalExec(), globalObject->globalScopeChain(), makeSource(script.data(), fileName)); + stopWatch.stop(); + + return JSValue::encode(jsNumber(stopWatch.getElapsedMS())); +} + +EncodedJSValue JSC_HOST_CALL functionLoad(ExecState* exec) +{ + UString fileName = exec->argument(0).toString(exec); + Vector<char> script; + if (!fillBufferWithContentsOfFile(fileName, script)) + return JSValue::encode(throwError(exec, createError(exec, "Could not open file."))); + + JSGlobalObject* globalObject = exec->lexicalGlobalObject(); + Completion result = evaluate(globalObject->globalExec(), globalObject->globalScopeChain(), makeSource(script.data(), fileName)); + if (result.complType() == Throw) + throwError(exec, result.value()); + return JSValue::encode(result.value()); +} + +EncodedJSValue JSC_HOST_CALL functionCheckSyntax(ExecState* exec) +{ + UString fileName = exec->argument(0).toString(exec); + Vector<char> script; + if (!fillBufferWithContentsOfFile(fileName, script)) + return JSValue::encode(throwError(exec, createError(exec, "Could not open file."))); + + JSGlobalObject* globalObject = exec->lexicalGlobalObject(); + + StopWatch stopWatch; + stopWatch.start(); + Completion result = checkSyntax(globalObject->globalExec(), makeSource(script.data(), fileName)); + stopWatch.stop(); + + if (result.complType() == Throw) + throwError(exec, result.value()); + return JSValue::encode(jsNumber(stopWatch.getElapsedMS())); +} + +#if ENABLE(SAMPLING_FLAGS) +EncodedJSValue JSC_HOST_CALL functionSetSamplingFlags(ExecState* exec) +{ + for (unsigned i = 0; i < exec->argumentCount(); ++i) { + unsigned flag = static_cast<unsigned>(exec->argument(i).toNumber(exec)); + if ((flag >= 1) && (flag <= 32)) + SamplingFlags::setFlag(flag); + } + return JSValue::encode(jsNull()); +} + +EncodedJSValue JSC_HOST_CALL functionClearSamplingFlags(ExecState* exec) +{ + for (unsigned i = 0; i < exec->argumentCount(); ++i) { + unsigned flag = static_cast<unsigned>(exec->argument(i).toNumber(exec)); + if ((flag >= 1) && (flag <= 32)) + SamplingFlags::clearFlag(flag); + } + return JSValue::encode(jsNull()); +} +#endif + +EncodedJSValue JSC_HOST_CALL functionReadline(ExecState* exec) +{ + Vector<char, 256> line; + int c; + while ((c = getchar()) != EOF) { + // FIXME: Should we also break on \r? + if (c == '\n') + break; + line.append(c); + } + line.append('\0'); + return JSValue::encode(jsString(exec, line.data())); +} + +EncodedJSValue JSC_HOST_CALL functionQuit(ExecState* exec) +{ + // Technically, destroying the heap in the middle of JS execution is a no-no, + // but we want to maintain compatibility with the Mozilla test suite, so + // we pretend that execution has terminated to avoid ASSERTs, then tear down the heap. + exec->globalData().dynamicGlobalObject = 0; + + cleanupGlobalData(&exec->globalData()); + exit(EXIT_SUCCESS); + +#if COMPILER(MSVC) && OS(WINCE) + // Without this, Visual Studio will complain that this method does not return a value. + return JSValue::encode(jsUndefined()); +#endif +} + +// Use SEH for Release builds only to get rid of the crash report dialog +// (luckily the same tests fail in Release and Debug builds so far). Need to +// be in a separate main function because the jscmain function requires object +// unwinding. + +#if COMPILER(MSVC) && !defined(_DEBUG) && !OS(WINCE) +#define TRY __try { +#define EXCEPT(x) } __except (EXCEPTION_EXECUTE_HANDLER) { x; } +#else +#define TRY +#define EXCEPT(x) +#endif + +int jscmain(int argc, char** argv, JSGlobalData*); + +int main(int argc, char** argv) +{ +#if defined(_DEBUG) && OS(WINDOWS) + _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR); + _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE); + _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR); + _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE); + _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR); + _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE); +#endif + +#if COMPILER(MSVC) && !OS(WINCE) + timeBeginPeriod(1); +#endif + +#if PLATFORM(QT) + QCoreApplication app(argc, argv); +#endif + + // Initialize JSC before getting JSGlobalData. + JSC::initializeThreading(); + + // We can't use destructors in the following code because it uses Windows + // Structured Exception Handling + int res = 0; + JSGlobalData* globalData = JSGlobalData::create(ThreadStackTypeLarge).leakRef(); + TRY + res = jscmain(argc, argv, globalData); + EXCEPT(res = 3) + + cleanupGlobalData(globalData); + return res; +} + +static void cleanupGlobalData(JSGlobalData* globalData) +{ + JSLock lock(SilenceAssertionsOnly); + globalData->heap.destroy(); + globalData->deref(); +} + +static bool runWithScripts(GlobalObject* globalObject, const Vector<Script>& scripts, bool dump) +{ + UString script; + UString fileName; + Vector<char> scriptBuffer; + + if (dump) + BytecodeGenerator::setDumpsGeneratedCode(true); + + JSGlobalData& globalData = globalObject->globalData(); + +#if ENABLE(SAMPLING_FLAGS) + SamplingFlags::start(); +#endif + + bool success = true; + for (size_t i = 0; i < scripts.size(); i++) { + if (scripts[i].isFile) { + fileName = scripts[i].argument; + if (!fillBufferWithContentsOfFile(fileName, scriptBuffer)) + return false; // fail early so we can catch missing files + script = scriptBuffer.data(); + } else { + script = scripts[i].argument; + fileName = "[Command Line]"; + } + + globalData.startSampling(); + + Completion completion = evaluate(globalObject->globalExec(), globalObject->globalScopeChain(), makeSource(script, fileName)); + success = success && completion.complType() != Throw; + if (dump) { + if (completion.complType() == Throw) + printf("Exception: %s\n", completion.value().toString(globalObject->globalExec()).utf8().data()); + else + printf("End: %s\n", completion.value().toString(globalObject->globalExec()).utf8().data()); + } + + globalData.stopSampling(); + globalObject->globalExec()->clearException(); + } + +#if ENABLE(SAMPLING_FLAGS) + SamplingFlags::stop(); +#endif + globalData.dumpSampleData(globalObject->globalExec()); +#if ENABLE(SAMPLING_COUNTERS) + AbstractSamplingCounter::dump(); +#endif +#if ENABLE(REGEXP_TRACING) + globalData.dumpRegExpTrace(); +#endif + return success; +} + +#define RUNNING_FROM_XCODE 0 + +static void runInteractive(GlobalObject* globalObject) +{ + while (true) { +#if HAVE(READLINE) && !RUNNING_FROM_XCODE + char* line = readline(interactivePrompt); + if (!line) + break; + if (line[0]) + add_history(line); + Completion completion = evaluate(globalObject->globalExec(), globalObject->globalScopeChain(), makeSource(line, interpreterName)); + free(line); +#else + printf("%s", interactivePrompt); + Vector<char, 256> line; + int c; + while ((c = getchar()) != EOF) { + // FIXME: Should we also break on \r? + if (c == '\n') + break; + line.append(c); + } + if (line.isEmpty()) + break; + line.append('\0'); + Completion completion = evaluate(globalObject->globalExec(), globalObject->globalScopeChain(), makeSource(line.data(), interpreterName)); +#endif + if (completion.complType() == Throw) + printf("Exception: %s\n", completion.value().toString(globalObject->globalExec()).utf8().data()); + else + printf("%s\n", completion.value().toString(globalObject->globalExec()).utf8().data()); + + globalObject->globalExec()->clearException(); + } + printf("\n"); +} + +static NO_RETURN void printUsageStatement(JSGlobalData* globalData, bool help = false) +{ + fprintf(stderr, "Usage: jsc [options] [files] [-- arguments]\n"); + fprintf(stderr, " -d Dumps bytecode (debug builds only)\n"); + fprintf(stderr, " -e Evaluate argument as script code\n"); + fprintf(stderr, " -f Specifies a source file (deprecated)\n"); + fprintf(stderr, " -h|--help Prints this help message\n"); + fprintf(stderr, " -i Enables interactive mode (default if no files are specified)\n"); +#if HAVE(SIGNAL_H) + fprintf(stderr, " -s Installs signal handlers that exit on a crash (Unix platforms only)\n"); +#endif + + cleanupGlobalData(globalData); + exit(help ? EXIT_SUCCESS : EXIT_FAILURE); +} + +static void parseArguments(int argc, char** argv, Options& options, JSGlobalData* globalData) +{ + int i = 1; + for (; i < argc; ++i) { + const char* arg = argv[i]; + if (!strcmp(arg, "-f")) { + if (++i == argc) + printUsageStatement(globalData); + options.scripts.append(Script(true, argv[i])); + continue; + } + if (!strcmp(arg, "-e")) { + if (++i == argc) + printUsageStatement(globalData); + options.scripts.append(Script(false, argv[i])); + continue; + } + if (!strcmp(arg, "-i")) { + options.interactive = true; + continue; + } + if (!strcmp(arg, "-d")) { + options.dump = true; + continue; + } + if (!strcmp(arg, "-s")) { +#if HAVE(SIGNAL_H) + signal(SIGILL, _exit); + signal(SIGFPE, _exit); + signal(SIGBUS, _exit); + signal(SIGSEGV, _exit); +#endif + continue; + } + if (!strcmp(arg, "--")) { + ++i; + break; + } + if (!strcmp(arg, "-h") || !strcmp(arg, "--help")) + printUsageStatement(globalData, true); + options.scripts.append(Script(true, argv[i])); + } + + if (options.scripts.isEmpty()) + options.interactive = true; + + for (; i < argc; ++i) + options.arguments.append(argv[i]); +} + +int jscmain(int argc, char** argv, JSGlobalData* globalData) +{ + JSLock lock(SilenceAssertionsOnly); + + Options options; + parseArguments(argc, argv, options, globalData); + + GlobalObject* globalObject = new (globalData) GlobalObject(options.arguments); + bool success = runWithScripts(globalObject, options.scripts, options.dump); + if (options.interactive && success) + runInteractive(globalObject); + + return success ? 0 : 3; +} + +static bool fillBufferWithContentsOfFile(const UString& fileName, Vector<char>& buffer) +{ + FILE* f = fopen(fileName.utf8().data(), "r"); + if (!f) { + fprintf(stderr, "Could not open file: %s\n", fileName.utf8().data()); + return false; + } + + size_t bufferSize = 0; + size_t bufferCapacity = 1024; + + buffer.resize(bufferCapacity); + + while (!feof(f) && !ferror(f)) { + bufferSize += fread(buffer.data() + bufferSize, 1, bufferCapacity - bufferSize, f); + if (bufferSize == bufferCapacity) { // guarantees space for trailing '\0' + bufferCapacity *= 2; + buffer.resize(bufferCapacity); + } + } + fclose(f); + buffer[bufferSize] = '\0'; + + if (buffer[0] == '#' && buffer[1] == '!') + buffer[0] = buffer[1] = '/'; + + return true; +} |