Add WebKitInitializer and use it in DumpRenderTree and testkjs
[WebKit-https.git] / 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 <WebKitInitializer/WebKitInitializer.h>
45 #include <windows.h>
46 #include <crtdbg.h>
47 #endif
48
49 #if PLATFORM(QT)
50 #include <QDateTime>
51 #endif
52
53 using namespace KJS;
54 using namespace WTF;
55
56 static void testIsInteger();
57 static char* createStringWithContentsOfFile(const char* fileName);
58
59 class StopWatch
60 {
61 public:
62     void start();
63     void stop();
64     long getElapsedMS(); // call stop() first
65     
66 private:
67 #if PLATFORM(QT)
68     uint m_startTime;
69     uint m_stopTime;
70 #elif PLATFORM(WIN_OS)
71     DWORD m_startTime;
72     DWORD m_stopTime;
73 #else
74     // Windows does not have timeval, disabling this class for now (bug 7399)
75     timeval m_startTime;
76     timeval m_stopTime;
77 #endif
78 };
79
80 void StopWatch::start()
81 {
82 #if PLATFORM(QT)
83     QDateTime t = QDateTime::currentDateTime();
84     m_startTime = t.toTime_t() * 1000 + t.time().msec();
85 #elif PLATFORM(WIN_OS)
86     m_startTime = timeGetTime();
87 #else
88     gettimeofday(&m_startTime, 0);
89 #endif
90 }
91
92 void StopWatch::stop()
93 {
94 #if PLATFORM(QT)
95     QDateTime t = QDateTime::currentDateTime();
96     m_stopTime = t.toTime_t() * 1000 + t.time().msec();
97 #elif PLATFORM(WIN_OS)
98     m_stopTime = timeGetTime();
99 #else
100     gettimeofday(&m_stopTime, 0);
101 #endif
102 }
103
104 long StopWatch::getElapsedMS()
105 {
106 #if PLATFORM(WIN_OS) || PLATFORM(QT)
107     return m_stopTime - m_startTime;
108 #else
109     timeval elapsedTime;
110     timersub(&m_stopTime, &m_startTime, &elapsedTime);
111     
112     return elapsedTime.tv_sec * 1000 + lroundf(elapsedTime.tv_usec / 1000.0f);
113 #endif
114 }
115
116 class GlobalImp : public JSObject {
117 public:
118   virtual UString className() const { return "global"; }
119 };
120
121 class TestFunctionImp : public JSObject {
122 public:
123   TestFunctionImp(int i, int length);
124   virtual bool implementsCall() const { return true; }
125   virtual JSValue* callAsFunction(ExecState* exec, JSObject* thisObj, const List &args);
126
127   enum { Print, Debug, Quit, GC, Version, Run };
128
129 private:
130   int id;
131 };
132
133 TestFunctionImp::TestFunctionImp(int i, int length) : JSObject(), id(i)
134 {
135   putDirect(Identifier("length"), length, DontDelete | ReadOnly | DontEnum);
136 }
137
138 JSValue* TestFunctionImp::callAsFunction(ExecState* exec, JSObject*, const List &args)
139 {
140   switch (id) {
141     case Print:
142       printf("--> %s\n", args[0]->toString(exec).UTF8String().c_str());
143       return jsUndefined();
144     case Debug:
145       fprintf(stderr, "--> %s\n", args[0]->toString(exec).UTF8String().c_str());
146       return jsUndefined();
147     case GC:
148     {
149       JSLock lock;
150       Collector::collect();
151       return jsUndefined();
152     }
153     case Version:
154       // We need this function for compatibility with the Mozilla JS tests but for now
155       // we don't actually do any version-specific handling
156       return jsUndefined();
157     case Run:
158     {
159       StopWatch stopWatch;
160       char* fileName = strdup(args[0]->toString(exec).UTF8String().c_str());
161       char* script = createStringWithContentsOfFile(fileName);
162       if (!script)
163         return throwError(exec, GeneralError, "Could not open file.");
164
165       stopWatch.start();
166       exec->dynamicInterpreter()->evaluate(fileName, 0, script);
167       stopWatch.stop();
168
169       free(script);
170       free(fileName);
171       
172       return jsNumber(stopWatch.getElapsedMS());
173     }
174     case Quit:
175       exit(0);
176     default:
177       abort();
178   }
179   return 0;
180 }
181
182 #if PLATFORM(WIN_OS)
183
184 // Use SEH for Release builds only to get rid of the crash report dialog
185 // (luckyly the same tests fail in Release and Debug builds so far). Need to
186 // be in a separate main function because the kjsmain function requires object
187 // unwinding.
188
189 #if defined(_DEBUG)
190 #define TRY
191 #define EXCEPT(x)
192 #else
193 #define TRY       __try {
194 #define EXCEPT(x) } __except (EXCEPTION_EXECUTE_HANDLER) { x; }
195 #endif
196
197 #else
198
199 #define TRY
200 #define EXCEPT(x)
201
202 #endif
203
204 int kjsmain(int argc, char** argv);
205
206 int main(int argc, char** argv)
207 {
208 #if PLATFORM(WIN_OS)
209     if (!initializeWebKit()) {
210         fprintf(stderr, "Failed to initialize WebKit\n");
211         abort();
212     }
213 #endif
214 #if defined(_DEBUG) && PLATFORM(WIN_OS)
215     _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);
216     _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
217     _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
218     _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
219     _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR);
220     _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE);
221 #endif
222
223     int res = 0;
224     TRY
225         res = kjsmain(argc, argv);
226     EXCEPT(res = 3)
227     return res;
228 }
229
230
231 bool doIt(int argc, char** argv)
232 {
233   bool success = true;
234   bool prettyPrint = false;
235   GlobalImp* global = new GlobalImp();
236
237   // create interpreter
238   RefPtr<Interpreter> interp = new Interpreter(global);
239   // add debug() function
240   global->put(interp->globalExec(), "debug", new TestFunctionImp(TestFunctionImp::Debug, 1));
241   // add "print" for compatibility with the mozilla js shell
242   global->put(interp->globalExec(), "print", new TestFunctionImp(TestFunctionImp::Print, 1));
243   // add "quit" for compatibility with the mozilla js shell
244   global->put(interp->globalExec(), "quit", new TestFunctionImp(TestFunctionImp::Quit, 0));
245   // add "gc" for compatibility with the mozilla js shell
246   global->put(interp->globalExec(), "gc", new TestFunctionImp(TestFunctionImp::GC, 0));
247   // add "version" for compatibility with the mozilla js shell 
248   global->put(interp->globalExec(), "version", new TestFunctionImp(TestFunctionImp::Version, 1));
249   global->put(interp->globalExec(), "run", new TestFunctionImp(TestFunctionImp::Run, 1));
250   
251   Interpreter::setShouldPrintExceptions(true);
252   
253   for (int i = 1; i < argc; i++) {
254     const char* fileName = argv[i];
255     if (strcmp(fileName, "-f") == 0) // mozilla test driver script uses "-f" prefix for files
256       continue;
257     if (strcmp(fileName, "-p") == 0) {
258       prettyPrint = true;
259       continue;
260     }
261     
262     char* script = createStringWithContentsOfFile(fileName);
263     if (!script) {
264       success = false;
265       break; // fail early so we can catch missing files
266     }
267     
268     if (prettyPrint) {
269       int errLine = 0;
270       UString errMsg;
271       UString s = Parser::prettyPrint(script, &errLine, &errMsg);
272       if (s.isNull()) {
273         fprintf(stderr, "%s:%d: %s.\n", fileName, errLine, errMsg.UTF8String().c_str());
274         success = false;
275         free(script);
276         break;
277       }
278       
279       printf("%s\n", s.UTF8String().c_str());
280       
281     } else {
282       Completion completion = interp->evaluate(fileName, 0, script);
283       success = success && completion.complType() != Throw;
284     }
285     
286     free(script);
287   }
288
289   return success;
290 }
291
292
293 int kjsmain(int argc, char** argv)
294 {
295   if (argc < 2) {
296     fprintf(stderr, "Usage: testkjs file1 [file2...]\n");
297     return -1;
298   }
299
300   testIsInteger();
301
302   JSLock lock;
303
304   bool success = doIt(argc, argv);
305
306 #ifndef NDEBUG
307   Collector::collect();
308 #endif
309
310   if (success)
311     fprintf(stderr, "OK.\n");
312   
313 #ifdef KJS_DEBUG_MEM
314   Interpreter::finalCheck();
315 #endif
316   return success ? 0 : 3;
317 }
318
319 static void testIsInteger()
320 {
321   // Unit tests for WTF::IsInteger. Don't have a better place for them now.
322   // FIXME: move these once we create a unit test directory for WTF.
323
324   assert(IsInteger<bool>::value);
325   assert(IsInteger<char>::value);
326   assert(IsInteger<signed char>::value);
327   assert(IsInteger<unsigned char>::value);
328   assert(IsInteger<short>::value);
329   assert(IsInteger<unsigned short>::value);
330   assert(IsInteger<int>::value);
331   assert(IsInteger<unsigned int>::value);
332   assert(IsInteger<long>::value);
333   assert(IsInteger<unsigned long>::value);
334   assert(IsInteger<long long>::value);
335   assert(IsInteger<unsigned long long>::value);
336
337   assert(!IsInteger<char*>::value);
338   assert(!IsInteger<const char* >::value);
339   assert(!IsInteger<volatile char* >::value);
340   assert(!IsInteger<double>::value);
341   assert(!IsInteger<float>::value);
342   assert(!IsInteger<GlobalImp>::value);
343 }
344
345 static char* createStringWithContentsOfFile(const char* fileName)
346 {
347   char* buffer;
348   
349   size_t buffer_size = 0;
350   size_t buffer_capacity = 1024;
351   buffer = (char*)malloc(buffer_capacity);
352   
353   FILE* f = fopen(fileName, "r");
354   if (!f) {
355     fprintf(stderr, "Could not open file: %s\n", fileName);
356     free(buffer);
357     return 0;
358   }
359   
360   while (!feof(f) && !ferror(f)) {
361     buffer_size += fread(buffer + buffer_size, 1, buffer_capacity - buffer_size, f);
362     if (buffer_size == buffer_capacity) { // guarantees space for trailing '\0'
363       buffer_capacity *= 2;
364       buffer = (char*)realloc(buffer, buffer_capacity);
365       assert(buffer);
366     }
367     
368     assert(buffer_size < buffer_capacity);
369   }
370   fclose(f);
371   buffer[buffer_size] = '\0';
372   
373   return buffer;
374 }