JavaScriptCore/kjs/testkjs.cpp

   1 // -*- c-basic-offset: 2 -*-
   2 /*
   3  *  This file is part of the KDE libraries
   4  *  Copyright (C) 1999-2000 Harri Porten (porten@kde.org)
   5  *  Copyright (C) 2004-2006 Apple Computer, Inc.
   6  *  Copyright (C) 2006 Bjoern Graf (bjoern.graf@gmail.com)
   7  *
   8  *  This library is free software; you can redistribute it and/or
   9  *  modify it under the terms of the GNU Library General Public
  10  *  License as published by the Free Software Foundation; either
  11  *  version 2 of the License, or (at your option) any later version.
  12  *
  13  *  This library is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  *  Library General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU Library General Public License
  19  *  along with this library; see the file COPYING.LIB.  If not, write to
  20  *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  21  *  Boston, MA 02110-1301, USA.
  22  *
  23  */
  24
  25 #include "config.h"
  26 #include "collector.h"
  27
  28 #include <wtf/HashTraits.h>
  29 #include "JSLock.h"
  30 #include "object.h"
  31 #include "Parser.h"
  32
  33 #include <math.h>
  34 #include <stdio.h>
  35
  36 #include <string.h>
  37 #if HAVE(SYS_TIME_H)
  38 #include <sys/time.h>
  39 #endif
  40
  41 #include "protect.h"
  42
  43 #if PLATFORM(WIN_OS)
  44 #include <windows.h>
  45 #include <crtdbg.h>
  46 #endif
  47
  48 #if PLATFORM(QT)
  49 #include <QDateTime>
  50 #endif
  51
  52 using namespace KJS;
  53 using namespace WTF;
  54
  55 static void testIsInteger();
  56 static char* createStringWithContentsOfFile(const char* fileName);
  57
  58 class StopWatch
  59 {
  60 public:
  61     void start();
  62     void stop();
  63     long getElapsedMS(); // call stop() first
  64
  65 private:
  66 #if PLATFORM(QT)
  67     uint m_startTime;
  68     uint m_stopTime;
  69 #elif PLATFORM(WIN_OS)
  70     DWORD m_startTime;
  71     DWORD m_stopTime;
  72 #else
  73     // Windows does not have timeval, disabling this class for now (bug 7399)
  74     timeval m_startTime;
  75     timeval m_stopTime;
  76 #endif
  77 };
  78
  79 void StopWatch::start()
  80 {
  81 #if PLATFORM(QT)
  82     QDateTime t = QDateTime::currentDateTime();
  83     m_startTime = t.toTime_t() * 1000 + t.time().msec();
  84 #elif PLATFORM(WIN_OS)
  85     m_startTime = timeGetTime();
  86 #else
  87     gettimeofday(&m_startTime, 0);
  88 #endif
  89 }
  90
  91 void StopWatch::stop()
  92 {
  93 #if PLATFORM(QT)
  94     QDateTime t = QDateTime::currentDateTime();
  95     m_stopTime = t.toTime_t() * 1000 + t.time().msec();
  96 #elif PLATFORM(WIN_OS)
  97     m_stopTime = timeGetTime();
  98 #else
  99     gettimeofday(&m_stopTime, 0);
 100 #endif
 101 }
 102
 103 long StopWatch::getElapsedMS()
 104 {
 105 #if PLATFORM(WIN_OS) || PLATFORM(QT)
 106     return m_stopTime - m_startTime;
 107 #else
 108     timeval elapsedTime;
 109     timersub(&m_stopTime, &m_startTime, &elapsedTime);
 110
 111     return elapsedTime.tv_sec * 1000 + lroundf(elapsedTime.tv_usec / 1000.0f);
 112 #endif
 113 }
 114
 115 class GlobalImp : public JSObject {
 116 public:
 117   virtual UString className() const { return "global"; }
 118 };
 119
 120 class TestFunctionImp : public JSObject {
 121 public:
 122   TestFunctionImp(int i, int length);
 123   virtual bool implementsCall() const { return true; }
 124   virtual JSValue* callAsFunction(ExecState* exec, JSObject* thisObj, const List &args);
 125
 126   enum { Print, Debug, Quit, GC, Version, Run };
 127
 128 private:
 129   int id;
 130 };
 131
 132 TestFunctionImp::TestFunctionImp(int i, int length) : JSObject(), id(i)
 133 {
 134   putDirect(Identifier("length"), length, DontDelete | ReadOnly | DontEnum);
 135 }
 136
 137 JSValue* TestFunctionImp::callAsFunction(ExecState* exec, JSObject*, const List &args)
 138 {
 139   switch (id) {
 140     case Print:
 141       printf("--> %s\n", args[0]->toString(exec).UTF8String().c_str());
 142       return jsUndefined();
 143     case Debug:
 144       fprintf(stderr, "--> %s\n", args[0]->toString(exec).UTF8String().c_str());
 145       return jsUndefined();
 146     case GC:
 147     {
 148       JSLock lock;
 149       Collector::collect();
 150       return jsUndefined();
 151     }
 152     case Version:
 153       // We need this function for compatibility with the Mozilla JS tests but for now
 154       // we don't actually do any version-specific handling
 155       return jsUndefined();
 156     case Run:
 157     {
 158       StopWatch stopWatch;
 159       char* fileName = strdup(args[0]->toString(exec).UTF8String().c_str());
 160       char* script = createStringWithContentsOfFile(fileName);
 161       if (!script)
 162         return throwError(exec, GeneralError, "Could not open file.");
 163
 164       stopWatch.start();
 165       exec->dynamicInterpreter()->evaluate(fileName, 0, script);
 166       stopWatch.stop();
 167
 168       free(script);
 169       free(fileName);
 170
 171       return jsNumber(stopWatch.getElapsedMS());
 172     }
 173     case Quit:
 174       exit(0);
 175     default:
 176       abort();
 177   }
 178   return 0;
 179 }
 180
 181 #if PLATFORM(WIN_OS)
 182
 183 // Use SEH for Release builds only to get rid of the crash report dialog
 184 // (luckyly the same tests fail in Release and Debug builds so far). Need to
 185 // be in a separate main function because the kjsmain function requires object
 186 // unwinding.
 187
 188 #if defined(_DEBUG)
 189 #define TRY
 190 #define EXCEPT(x)
 191 #else
 192 #define TRY       __try {
 193 #define EXCEPT(x) } __except (EXCEPTION_EXECUTE_HANDLER) { x; }
 194 #endif
 195
 196 #else
 197
 198 #define TRY
 199 #define EXCEPT(x)
 200
 201 #endif
 202
 203 int kjsmain(int argc, char** argv);
 204
 205 int main(int argc, char** argv)
 206 {
 207 #if defined(_DEBUG) && PLATFORM(WIN_OS)
 208     _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);
 209     _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
 210     _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
 211     _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
 212     _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR);
 213     _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE);
 214 #endif
 215
 216     int res = 0;
 217     TRY
 218         res = kjsmain(argc, argv);
 219     EXCEPT(res = 3)
 220     return res;
 221 }
 222
 223
 224 bool doIt(int argc, char** argv)
 225 {
 226   bool success = true;
 227   bool prettyPrint = false;
 228   GlobalImp* global = new GlobalImp();
 229
 230   // create interpreter
 231   RefPtr<Interpreter> interp = new Interpreter(global);
 232   // add debug() function
 233   global->put(interp->globalExec(), "debug", new TestFunctionImp(TestFunctionImp::Debug, 1));
 234   // add "print" for compatibility with the mozilla js shell
 235   global->put(interp->globalExec(), "print", new TestFunctionImp(TestFunctionImp::Print, 1));
 236   // add "quit" for compatibility with the mozilla js shell
 237   global->put(interp->globalExec(), "quit", new TestFunctionImp(TestFunctionImp::Quit, 0));
 238   // add "gc" for compatibility with the mozilla js shell
 239   global->put(interp->globalExec(), "gc", new TestFunctionImp(TestFunctionImp::GC, 0));
 240   // add "version" for compatibility with the mozilla js shell
 241   global->put(interp->globalExec(), "version", new TestFunctionImp(TestFunctionImp::Version, 1));
 242   global->put(interp->globalExec(), "run", new TestFunctionImp(TestFunctionImp::Run, 1));
 243
 244   Interpreter::setShouldPrintExceptions(true);
 245
 246   for (int i = 1; i < argc; i++) {
 247     const char* fileName = argv[i];
 248     if (strcmp(fileName, "-f") == 0) // mozilla test driver script uses "-f" prefix for files
 249       continue;
 250     if (strcmp(fileName, "-p") == 0) {
 251       prettyPrint = true;
 252       continue;
 253     }
 254
 255     char* script = createStringWithContentsOfFile(fileName);
 256     if (!script) {
 257       success = false;
 258       break; // fail early so we can catch missing files
 259     }
 260
 261     if (prettyPrint) {
 262       int errLine = 0;
 263       UString errMsg;
 264       UString s = Parser::prettyPrint(script, &errLine, &errMsg);
 265       if (s.isNull()) {
 266         fprintf(stderr, "%s:%d: %s.\n", fileName, errLine, errMsg.UTF8String().c_str());
 267         success = false;
 268         free(script);
 269         break;
 270       }
 271
 272       printf("%s\n", s.UTF8String().c_str());
 273
 274     } else {
 275       Completion completion = interp->evaluate(fileName, 0, script);
 276       success = success && completion.complType() != Throw;
 277     }
 278
 279     free(script);
 280   }
 281
 282   return success;
 283 }
 284
 285
 286 int kjsmain(int argc, char** argv)
 287 {
 288   if (argc < 2) {
 289     fprintf(stderr, "Usage: testkjs file1 [file2...]\n");
 290     return -1;
 291   }
 292
 293   testIsInteger();
 294
 295   JSLock lock;
 296
 297   bool success = doIt(argc, argv);
 298
 299 #ifndef NDEBUG
 300   Collector::collect();
 301 #endif
 302
 303   if (success)
 304     fprintf(stderr, "OK.\n");
 305
 306 #ifdef KJS_DEBUG_MEM
 307   Interpreter::finalCheck();
 308 #endif
 309   return success ? 0 : 3;
 310 }
 311
 312 static void testIsInteger()
 313 {
 314   // Unit tests for WTF::IsInteger. Don't have a better place for them now.
 315   // FIXME: move these once we create a unit test directory for WTF.
 316
 317   assert(IsInteger<bool>::value);
 318   assert(IsInteger<char>::value);
 319   assert(IsInteger<signed char>::value);
 320   assert(IsInteger<unsigned char>::value);
 321   assert(IsInteger<short>::value);
 322   assert(IsInteger<unsigned short>::value);
 323   assert(IsInteger<int>::value);
 324   assert(IsInteger<unsigned int>::value);
 325   assert(IsInteger<long>::value);
 326   assert(IsInteger<unsigned long>::value);
 327   assert(IsInteger<long long>::value);
 328   assert(IsInteger<unsigned long long>::value);
 329
 330   assert(!IsInteger<char*>::value);
 331   assert(!IsInteger<const char* >::value);
 332   assert(!IsInteger<volatile char* >::value);
 333   assert(!IsInteger<double>::value);
 334   assert(!IsInteger<float>::value);
 335   assert(!IsInteger<GlobalImp>::value);
 336 }
 337
 338 static char* createStringWithContentsOfFile(const char* fileName)
 339 {
 340   char* buffer;
 341
 342   size_t buffer_size = 0;
 343   size_t buffer_capacity = 1024;
 344   buffer = (char*)malloc(buffer_capacity);
 345
 346   FILE* f = fopen(fileName, "r");
 347   if (!f) {
 348     fprintf(stderr, "Could not open file: %s\n", fileName);
 349     free(buffer);
 350     return 0;
 351   }
 352
 353   while (!feof(f) && !ferror(f)) {
 354     buffer_size += fread(buffer + buffer_size, 1, buffer_capacity - buffer_size, f);
 355     if (buffer_size == buffer_capacity) { // guarantees space for trailing '\0'
 356       buffer_capacity *= 2;
 357       buffer = (char*)realloc(buffer, buffer_capacity);
 358       assert(buffer);
 359     }
 360
 361     assert(buffer_size < buffer_capacity);
 362   }
 363   fclose(f);
 364   buffer[buffer_size] = '\0';
 365
 366   return buffer;
 367 }