]> git.saurik.com Git - cydget.git/blob - LockScreen.mm
We need a more recent build of Cycript now.
[cydget.git] / LockScreen.mm
1 /* Cydget - open-source AwayView plugin multiplexer
2 * Copyright (C) 2009-2011 Jay Freeman (saurik)
3 */
4
5 /*
6 * Redistribution and use in source and binary
7 * forms, with or without modification, are permitted
8 * provided that the following conditions are met:
9 *
10 * 1. Redistributions of source code must retain the
11 * above copyright notice, this list of conditions
12 * and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the
14 * above copyright notice, this list of conditions
15 * and the following disclaimer in the documentation
16 * and/or other materials provided with the
17 * distribution.
18 * 3. The name of the author may not be used to endorse
19 * or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
23 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
24 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
25 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
27 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
28 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
29 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
30 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
31 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
32 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
33 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
34 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
35 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 */
37
38 #include <substrate.h>
39 #include <sys/sysctl.h>
40
41 #import <GraphicsServices/GraphicsServices.h>
42 #import <UIKit/UIKit.h>
43 #import <AddressBook/AddressBook.h>
44
45 #import <SpringBoard/SBStatusBarController.h>
46 #import <SpringBoardUI/SBAwayViewPluginController.h>
47 #import <TelephonyUI/TPBottomLockBar.h>
48
49 #import <QuartzCore/CALayer.h>
50 // XXX: fix the minimum requirement
51 extern NSString * const kCAFilterNearest;
52
53 #include <WebKit/DOMCSSPrimitiveValue.h>
54 #include <WebKit/DOMCSSStyleDeclaration.h>
55 #include <WebKit/DOMDocument.h>
56 #include <WebKit/DOMHTMLBodyElement.h>
57 #include <WebKit/DOMNodeList.h>
58 #include <WebKit/DOMRGBColor.h>
59
60 #include <WebKit/WebFrame.h>
61 #include <WebKit/WebPolicyDelegate.h>
62 #include <WebKit/WebPreferences.h>
63 #include <WebKit/WebScriptObject.h>
64
65 #import <WebKit/WebView.h>
66 #import <WebKit/WebView-WebPrivate.h>
67
68 #include <WebCore/Page.h>
69 #include <WebCore/Settings.h>
70
71 #include <WebCore/WebCoreThread.h>
72 #include <WebKit/WebPreferences-WebPrivate.h>
73
74 #include "JSGlobalData.h"
75
76 #include "SourceCode.h"
77
78 #include "yieldToSelector.h"
79
80 #include <apr-1/apr_pools.h>
81 #include <pcre.h>
82
83 #define _transient
84 #define _forever for (;;)
85
86 _disused static unsigned trace_;
87
88 #define _trace() do { \
89 NSLog(@"_trace(%u)@%s:%u[%s](%p)\n", \
90 trace_++, __FILE__, __LINE__, __FUNCTION__, pthread_self() \
91 ); \
92 } while (false)
93
94 #define _assert(test) do \
95 if (!(test)) { \
96 fprintf(stderr, "_assert(%d:%s)@%s:%u[%s]\n", errno, #test, __FILE__, __LINE__, __FUNCTION__); \
97 exit(-1); \
98 } \
99 while (false)
100
101 #define _syscall(expr) \
102 do if ((long) (expr) != -1) \
103 break; \
104 else switch (errno) { \
105 case EINTR: \
106 continue; \
107 default: \
108 _assert(false); \
109 } while (true)
110
111 @protocol CydgetController
112 - (NSDictionary *) currentConfiguration;
113 @end
114
115 static Class $CydgetController(objc_getClass("CydgetController"));
116
117 static bool iOS32, iOS4;
118
119 @interface NSString (UIKit)
120 - (NSString *) stringByAddingPercentEscapes;
121 @end
122
123 @implementation UIWebDocumentView (WebCycript)
124
125 - (void) _setScrollerOffset:(CGPoint)offset {
126 UIScroller *scroller([self _scroller]);
127
128 CGSize size([scroller contentSize]);
129 CGSize bounds([scroller bounds].size);
130
131 CGPoint max;
132 max.x = size.width - bounds.width;
133 max.y = size.height - bounds.height;
134
135 // wtf Apple?!
136 if (max.x < 0)
137 max.x = 0;
138 if (max.y < 0)
139 max.y = 0;
140
141 offset.x = offset.x < 0 ? 0 : offset.x > max.x ? max.x : offset.x;
142 offset.y = offset.y < 0 ? 0 : offset.y > max.y ? max.y : offset.y;
143
144 [scroller setOffset:offset];
145 }
146
147 @end
148
149 /* Perl-Compatible RegEx {{{ */
150 class Pcre {
151 private:
152 pcre *code_;
153 pcre_extra *study_;
154 int capture_;
155 int *matches_;
156 const char *data_;
157
158 public:
159 Pcre(const char *regex, int options = 0) :
160 study_(NULL)
161 {
162 const char *error;
163 int offset;
164 code_ = pcre_compile(regex, options, &error, &offset, NULL);
165
166 if (code_ == NULL)
167 @throw [NSException exceptionWithName:NSInvalidArgumentException reason:[NSString stringWithFormat:@"*** Pcre(,): [%u] %s", offset, error] userInfo:nil];
168
169 pcre_fullinfo(code_, study_, PCRE_INFO_CAPTURECOUNT, &capture_);
170 matches_ = new int[(capture_ + 1) * 3];
171 }
172
173 ~Pcre() {
174 pcre_free(code_);
175 delete matches_;
176 }
177
178 NSString *operator [](size_t match) {
179 return [[[NSString alloc] initWithBytes:(data_ + matches_[match * 2]) length:(matches_[match * 2 + 1] - matches_[match * 2]) encoding:NSUTF8StringEncoding] autorelease];
180 }
181
182 bool operator ()(NSString *data) {
183 // XXX: length is for characters, not for bytes
184 return operator ()([data UTF8String], [data length]);
185 }
186
187 bool operator ()(const char *data, size_t size) {
188 data_ = data;
189 return pcre_exec(code_, study_, data, size, 0, 0, matches_, (capture_ + 1) * 3) >= 0;
190 }
191 };
192 /* }}} */
193
194 static float CYScrollViewDecelerationRateNormal;
195
196 @interface NSURL (Apple)
197 - (BOOL) isSpringboardHandledURL;
198 @end
199
200 @interface UIScrollView (Apple)
201 - (void) setDecelerationRate:(float)value;
202 - (void) setScrollingEnabled:(BOOL)enabled;
203 @end
204
205 @interface UIWebView (Apple)
206 - (void) setDataDetectorTypes:(int)types;
207 - (void) _setDrawInWebThread:(BOOL)draw;
208 - (UIScrollView *) _scrollView;
209 - (UIScroller *) _scroller;
210 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message;
211 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame;
212 @end
213
214 @interface WebView (Apple)
215 - (void) _setLayoutInterval:(float)interval;
216 - (void) _setAllowsMessaging:(BOOL)allows;
217 - (void) setShouldUpdateWhileOffscreen:(BOOL)update;
218 @end
219
220 @protocol CydgetWebViewDelegate //<UIWebViewDelegate>
221 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame;
222 @end
223
224 @class UIWebViewWebViewDelegate;
225
226 @interface CydgetWebView : UIWebView {
227 }
228
229 @end
230
231 MSClassHook(UIApplication)
232
233 MSInstanceMessageHook1(void, UIApplication, openURL, NSURL *, url) {
234 [self applicationOpenURL:url];
235 }
236
237 @implementation NSURL (Cydget)
238
239 - (NSNumber *) cydget$isSpringboardHandledURL {
240 return [NSNumber numberWithBool:[self isSpringboardHandledURL]];
241 }
242
243 @end
244
245 MSClassHook(NSURL)
246
247 MSInstanceMessageHook0(BOOL, NSURL, isSpringboardHandledURL) {
248 if (![NSThread isMainThread])
249 return MSOldCall();
250
251 return [[self cydget$yieldToSelector:@selector(cydget$isSpringboardHandledURL)] boolValue];
252 }
253
254 @implementation CydgetWebView
255
256 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
257 NSObject<CydgetWebViewDelegate> *delegate([self delegate]);
258 if ([delegate respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
259 [delegate webView:view didClearWindowObject:window forFrame:frame];
260 if ([UIWebView instancesRespondToSelector:@selector(webView:didClearWindowObject:forFrame:)])
261 [super webView:view didClearWindowObject:window forFrame:frame];
262 }
263
264 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message {
265 NSLog(@"addMessageToConsole:%@", message);
266
267 if ([UIWebView instancesRespondToSelector:@selector(webView:addMessageToConsole:)])
268 [super webView:view addMessageToConsole:message];
269 }
270
271 @end
272
273 @interface WebCydgetLockScreenView : UIView {
274 CydgetWebView *webview_;
275 UIScrollView *scroller_;
276 NSString *cycript_;
277 }
278
279 @end
280
281 @implementation WebCydgetLockScreenView
282
283 //#include "UICaboodle/UCInternal.h"
284
285 - (void) dealloc {
286 [webview_ setDelegate:nil];
287 [webview_ release];
288 [super dealloc];
289 }
290
291 - (void) loadRequest:(NSURLRequest *)request {
292 [webview_ loadRequest:request];
293 }
294
295 - (void) loadURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {
296 [self loadRequest:[NSURLRequest
297 requestWithURL:url
298 cachePolicy:policy
299 timeoutInterval:30.0
300 ]];
301 }
302
303 - (void) loadURL:(NSURL *)url {
304 [self loadURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy];
305 }
306
307 - (id) init {
308 CGRect frame = {{0, 0}, {320, 480}};
309 frame.size.height -= 20; //[[[$SBStatusBarController sharedStatusBarController] statusBarView] frame].size.height;
310
311 if ((self = [super initWithFrame:frame]) != nil) {
312 CGRect bounds([self bounds]);
313 bounds.size.height -= [TPBottomLockBar defaultHeight];
314
315 webview_ = [[CydgetWebView alloc] initWithFrame:bounds];
316 [webview_ setDelegate:self];
317 [self addSubview:webview_];
318
319 if ([webview_ respondsToSelector:@selector(setDataDetectorTypes:)])
320 [webview_ setDataDetectorTypes:0x80000000];
321 else
322 [webview_ setDetectsPhoneNumbers:NO];
323
324 [webview_ setScalesPageToFit:YES];
325
326 if (kCFCoreFoundationVersionNumber < 478.61)
327 if ([webview_ respondsToSelector:@selector(_setDrawInWebThread:)])
328 [webview_ _setDrawInWebThread:NO];
329
330 UIWebDocumentView *document([webview_ _documentView]);
331 WebView *webview([document webView]);
332 WebPreferences *preferences([webview preferences]);
333
334 [document setTileSize:CGSizeMake(bounds.size.width, 500)];
335
336 [document setBackgroundColor:[UIColor blackColor]];
337 [document setDrawsBackground:NO];
338
339 [webview setPreferencesIdentifier:@"WebCycript"];
340
341 if ([webview respondsToSelector:@selector(_setLayoutInterval:)])
342 [webview _setLayoutInterval:0];
343 else
344 [preferences _setLayoutInterval:0];
345
346 [preferences setCacheModel:WebCacheModelDocumentViewer];
347 [preferences setJavaScriptCanOpenWindowsAutomatically:YES];
348 [preferences setOfflineWebApplicationCacheEnabled:YES];
349
350 if ([webview respondsToSelector:@selector(setShouldUpdateWhileOffscreen:)])
351 [webview setShouldUpdateWhileOffscreen:NO];
352
353 if ([document respondsToSelector:@selector(setAllowsMessaging:)])
354 [document setAllowsMessaging:YES];
355 if ([webview respondsToSelector:@selector(_setAllowsMessaging:)])
356 [webview _setAllowsMessaging:YES];
357
358 if ([webview_ respondsToSelector:@selector(_scrollView)]) {
359 scroller_ = [webview_ _scrollView];
360
361 [scroller_ setDirectionalLockEnabled:YES];
362 [scroller_ setDecelerationRate:CYScrollViewDecelerationRateNormal];
363 [scroller_ setDelaysContentTouches:NO];
364
365 [scroller_ setCanCancelContentTouches:YES];
366
367 [scroller_ setAlwaysBounceVertical:NO];
368 } else if ([webview_ respondsToSelector:@selector(_scroller)]) {
369 UIScroller *scroller([webview_ _scroller]);
370 scroller_ = (UIScrollView *) scroller;
371
372 [scroller setDirectionalScrolling:YES];
373 [scroller setScrollDecelerationFactor:CYScrollViewDecelerationRateNormal]; /* 0.989324 */
374 [scroller setScrollHysteresis:0]; /* 8 */
375
376 [scroller setThumbDetectionEnabled:NO];
377 }
378
379 [scroller_ setFixedBackgroundPattern:YES];
380 [scroller_ setBackgroundColor:[UIColor blackColor]];
381 [scroller_ setClipsSubviews:NO];
382
383 [scroller_ setBounces:YES];
384 [scroller_ setShowBackgroundShadow:NO]; /* YES */
385
386 [self setAutoresizingMask:(UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight)];
387 [webview_ setAutoresizingMask:(UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight)];
388
389 NSDictionary *configuration([$CydgetController currentConfiguration]);
390
391 cycript_ = [configuration objectForKey:@"CycriptURLs"];
392
393 [scroller_ setScrollingEnabled:[[configuration objectForKey:@"Scrollable"] boolValue]];
394
395 NSString *homepage([configuration objectForKey:@"Homepage"]);
396 [self loadURL:[NSURL URLWithString:homepage]];
397 } return self;
398 }
399
400 - (void) webView:(WebView *)webview didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
401 if (cycript_ != nil)
402 if (NSString *href = [[[[frame dataSource] request] URL] absoluteString])
403 if (Pcre([cycript_ UTF8String], 0 /*XXX:PCRE_UTF8*/)(href))
404 if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
405 if (void (*CYSetupContext)(JSGlobalContextRef) = reinterpret_cast<void (*)(JSGlobalContextRef)>(dlsym(handle, "CydgetSetupContext"))) {
406 WebFrame *frame([webview mainFrame]);
407 JSGlobalContextRef context([frame globalContext]);
408 CYSetupContext(context);
409 }
410 }
411
412 @end
413
414 @interface WebCycriptLockScreenController : SBAwayViewPluginController {
415 }
416
417 @end
418
419 #include <string>
420
421 struct State {
422 unsigned state;
423 };
424
425 // String Helpers {{{
426 static const UChar *(*_ZNK7WebCore6String10charactersEv)(const WebCore::String *);
427 static const UChar *(*_ZN7WebCore6String29charactersWithNullTerminationEv)(const WebCore::String *);
428 static unsigned (*_ZNK7WebCore6String6lengthEv)(const WebCore::String *);
429
430 static bool StringGet(const WebCore::String &string, const UChar *&data, size_t &length) {
431 bool terminated;
432
433 if (_ZNK7WebCore6String10charactersEv != NULL) {
434 data = (*_ZNK7WebCore6String10charactersEv)(&string);
435 terminated = false;
436 } else if (_ZN7WebCore6String29charactersWithNullTerminationEv != NULL) {
437 data = (*_ZN7WebCore6String29charactersWithNullTerminationEv)(&string);
438 terminated = true;
439 } else return false;
440
441 if (data == NULL)
442 return false;
443
444 if (_ZNK7WebCore6String6lengthEv != NULL)
445 length = (*_ZNK7WebCore6String6lengthEv)(&string);
446 else if (terminated)
447 for (length = 0; data[length] != 0; ++length);
448 else return false;
449
450 return true;
451 }
452
453 static bool StringEquals(const WebCore::String &string, const char *value) {
454 const UChar *data;
455 size_t size;
456 if (!StringGet(string, data, size))
457 return false;
458
459 size_t length(strlen(value));
460 if (size != length)
461 return false;
462
463 for (size_t index(0); index != length; ++index)
464 if (data[index] != value[index])
465 return false;
466
467 return true;
468 }
469 // }}}
470 // State Machine {{{
471 static bool cycript_;
472
473 MSHook(bool, _ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, const WebCore::String &mime) {
474 if (!StringEquals(mime, "text/cycript")) {
475 cycript_ = false;
476 return __ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE(mime);
477 }
478
479 static void *handle(dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL));
480 if (handle == NULL)
481 return false;
482
483 cycript_ = true;
484 return true;
485 }
486 // }}}
487 // Script Compiler {{{
488 static void Log(const WebCore::String &string) {
489 #if 0
490 const UChar *data;
491 size_t length;
492 if (!StringGet(string, data, length))
493 return;
494
495 UChar terminated[length + 1];
496 terminated[length] = 0;
497 memcpy(terminated, data, length * 2);
498 NSLog(@"wtf %p:%zu:%S:", &string, length, terminated);
499 #endif
500 }
501
502 static void Cycriptify(apr_pool_t *pool, const uint16_t *&data, size_t &size) {
503 cycript_ = false;
504
505 if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
506 if (void (*CydgetPoolParse)(apr_pool_t *, const uint16_t **, size_t *) = reinterpret_cast<void (*)(apr_pool_t *, const uint16_t **, size_t *)>(dlsym(handle, "CydgetPoolParse")))
507 CydgetPoolParse(pool, &data, &size);
508 }
509
510 static void (*_ZN7WebCore6String6appendEPKtj)(WebCore::String *, const UChar *, unsigned);
511 static void (*_ZN7WebCore6String8truncateEj)(WebCore::String *, unsigned);
512
513 static void Cycriptify(const WebCore::String &source, int *psize = NULL) {
514 if (!cycript_)
515 return;
516 cycript_ = false;
517
518 const UChar *data;
519 size_t length;
520
521 if (!StringGet(source, data, length)) {
522 return;
523 }
524
525 size_t size(length);
526
527 apr_pool_t *pool;
528 apr_pool_create(&pool, NULL);
529
530 Cycriptify(pool, data, size);
531
532 WebCore::String &script(const_cast<WebCore::String &>(source));
533
534 _ZN7WebCore6String8truncateEj(&script, 0);
535 _ZN7WebCore6String6appendEPKtj(&script, data, size);
536
537 if (psize != NULL)
538 *psize = size;
539
540 apr_pool_destroy(pool);
541
542 Log(source);
543 }
544 // }}}
545
546 static WebCore::String *string;
547
548 // iOS 2.x
549 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, void *_this, const WebCore::String &string, State state, const WebCore::String &url, int line) {
550 Cycriptify(string);
551 return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i(_this, string, state, url, line);
552 }
553
554 // iOS 3.x
555 MSHook(void, _ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, JSC::SourceCode **_this, JSC::JSGlobalData *global, int *line, JSC::UString *message) {
556 if (cycript_) {
557 JSC::SourceCode *source(_this[iOS32 ? 6 : 0]);
558 const uint16_t *data(source->data());
559 size_t size(source->length());
560
561 apr_pool_t *pool;
562 apr_pool_create(&pool, NULL);
563
564 Cycriptify(pool, data, size);
565 source->~SourceCode();
566 // XXX: I actually don't have the original URL here: pants
567 new (source) JSC::SourceCode(JSC::UStringSourceProvider::create(JSC::UString(data, size), "cycript://"), 1);
568
569 apr_pool_destroy(pool);
570
571 }
572
573 return __ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE(_this, global, line, message);
574 }
575
576 // iOS 4.x cdata
577 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, void *_this, const WebCore::String &source, const WebCore::KURL &url, int line) {
578 Cycriptify(source);
579 return __ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi(_this, source, url, line);
580 }
581
582 // iOS 4.x+5.0 @src=
583 MSHook(const WebCore::String &, _ZN7WebCore12CachedScript6scriptEv, void *_this) {
584 const WebCore::String &script(__ZN7WebCore12CachedScript6scriptEv(_this));
585 string = const_cast<WebCore::String *>(&script);
586 Log(script);
587 return script;
588 }
589
590 // iOS 4.x @src=
591 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, void *_this, void *source, State state) {
592 if (string != NULL) {
593 if (iOS4)
594 Cycriptify(*string, reinterpret_cast<int *>(source) + 3);
595 else
596 Cycriptify(*string);
597 }
598 string = NULL;
599 return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE(_this, source, state);
600 }
601
602 // iOS 5.0 cdata
603 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, void *_this, const WebCore::String &source, const WebCore::KURL &url, void *position) {
604 Cycriptify(source);
605 return __ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE(_this, source, url, position);
606 }
607
608 // iOS 5.0 @src=
609 MSHook(void, _ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, void *_this, void *position, int legacy) {
610 string = NULL;
611 return __ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE(_this, position, legacy);
612 }
613
614 void (*_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE)(void *_this, int legacy);
615
616 // iOS 5.0 @src=
617 MSHook(void, _ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, void *_this, void *source) {
618 if (string != NULL) {
619 _ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE(_this, 0);
620 Cycriptify(*string, reinterpret_cast<int *>(source) + 3);
621 string = NULL;
622 }
623
624 return __ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE(_this, source);
625 }
626
627 /* Cydget:// Protocol {{{ */
628 @interface CydgetURLProtocol : NSURLProtocol {
629 }
630
631 @end
632
633 @implementation CydgetURLProtocol
634
635 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
636 NSURL *url([request URL]);
637 if (url == nil)
638 return NO;
639 NSString *scheme([[url scheme] lowercaseString]);
640 if (scheme == nil || ![scheme isEqualToString:@"cydget"])
641 return NO;
642 return YES;
643 }
644
645 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
646 return request;
647 }
648
649 - (void) _returnPNGWithImage:(UIImage *)icon forRequest:(NSURLRequest *)request {
650 id<NSURLProtocolClient> client([self client]);
651 if (icon == nil)
652 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
653 else {
654 NSData *data(UIImagePNGRepresentation(icon));
655
656 NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:@"image/png" expectedContentLength:-1 textEncodingName:nil] autorelease]);
657 [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
658 [client URLProtocol:self didLoadData:data];
659 [client URLProtocolDidFinishLoading:self];
660 }
661 }
662
663 - (void) startLoading {
664 id<NSURLProtocolClient> client([self client]);
665 NSURLRequest *request([self request]);
666
667 NSURL *url([request URL]);
668 NSString *href([url absoluteString]);
669
670 NSString *path([href substringFromIndex:9]);
671 NSRange slash([path rangeOfString:@"/"]);
672
673 NSString *command;
674 if (slash.location == NSNotFound) {
675 command = path;
676 path = nil;
677 } else {
678 command = [path substringToIndex:slash.location];
679 path = [path substringFromIndex:(slash.location + 1)];
680 }
681
682 if ([command isEqualToString:@"_UIImageWithName"]) {
683 if (path == nil)
684 goto fail;
685 path = [path stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
686 UIImage *icon(_UIImageWithName(path));
687 [self _returnPNGWithImage:icon forRequest:request];
688 } else fail: {
689 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
690 }
691 }
692
693 - (void) stopLoading {
694 }
695
696 @end
697 /* }}} */
698 /* Cydget-CGI:// Protocol {{{ */
699 @interface CydgetCGIURLProtocol : NSURLProtocol {
700 pid_t pid_;
701 CFHTTPMessageRef http_;
702 NSFileHandle *handle_;
703 }
704
705 @end
706
707 @implementation CydgetCGIURLProtocol
708
709 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
710 NSURL *url([request URL]);
711 if (url == nil)
712 return NO;
713 NSString *scheme([[url scheme] lowercaseString]);
714 if (scheme == nil || ![scheme isEqualToString:@"cydget-cgi"])
715 return NO;
716 return YES;
717 }
718
719 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
720 return request;
721 }
722
723 - (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)response client:(id<NSURLProtocolClient>)client {
724 if ((self = [super initWithRequest:request cachedResponse:response client:client]) != nil) {
725 pid_ = -1;
726 } return self;
727 }
728
729 - (void) startLoading {
730 id<NSURLProtocolClient> client([self client]);
731 NSURLRequest *request([self request]);
732 NSURL *url([request URL]);
733
734 NSString *path([url path]);
735 if (path == nil) {
736 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
737 return;
738 }
739
740 NSFileManager *manager([NSFileManager defaultManager]);
741 if (![manager fileExistsAtPath:path]) {
742 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
743 return;
744 }
745
746 int fds[2];
747 _assert(pipe(fds) != -1);
748
749 _assert(pid_ == -1);
750 pid_ = fork();
751 if (pid_ == -1) {
752 _assert(close(fds[0]) != -1);
753 _assert(close(fds[1]) != -1);
754 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
755 return;
756 }
757
758 if (pid_ == 0) {
759 const char *script([path UTF8String]);
760
761 setenv("GATEWAY_INTERFACE", "CGI/1.1", true);
762 setenv("SCRIPT_FILENAME", script, true);
763 NSString *query([url query]);
764 if (query != nil)
765 setenv("QUERY_STRING", [query UTF8String], true);
766
767 _assert(dup2(fds[1], 1) != -1);
768 _assert(close(fds[0]) != -1);
769 _assert(close(fds[1]) != -1);
770
771 execl(script, script, NULL);
772 exit(1);
773 _assert(false);
774 }
775
776 _assert(close(fds[1]) != -1);
777
778 _assert(http_ == NULL);
779 http_ = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, FALSE);
780 CFHTTPMessageAppendBytes(http_, (const uint8_t *) "HTTP/1.1 200 OK\r\n", 17);
781
782 _assert(handle_ == nil);
783 handle_ = [[NSFileHandle alloc] initWithFileDescriptor:fds[0] closeOnDealloc:YES];
784
785 [[NSNotificationCenter defaultCenter]
786 addObserver:self
787 selector:@selector(onRead:)
788 name:@"NSFileHandleReadCompletionNotification"
789 object:handle_
790 ];
791
792 [handle_ readInBackgroundAndNotify];
793 }
794
795 - (void) onRead:(NSNotification *)notification {
796 NSFileHandle *handle([notification object]);
797
798 NSData *data([[notification userInfo] objectForKey:NSFileHandleNotificationDataItem]);
799
800 if (size_t length = [data length]) {
801 CFHTTPMessageAppendBytes(http_, reinterpret_cast<const UInt8 *>([data bytes]), length);
802 [handle readInBackgroundAndNotify];
803 } else {
804 id<NSURLProtocolClient> client([self client]);
805
806 CFStringRef mime(CFHTTPMessageCopyHeaderFieldValue(http_, CFSTR("Content-type")));
807 if (mime == NULL)
808 [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorBadServerResponse userInfo:nil]];
809 else {
810 NSURLRequest *request([self request]);
811
812 NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:(NSString *)mime expectedContentLength:-1 textEncodingName:nil] autorelease]);
813 CFRelease(mime);
814
815 [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
816
817 CFDataRef body(CFHTTPMessageCopyBody(http_));
818 [client URLProtocol:self didLoadData:(NSData *)body];
819 CFRelease(body);
820
821 [client URLProtocolDidFinishLoading:self];
822 }
823
824 CFRelease(http_);
825 http_ = NULL;
826 }
827 }
828
829 //[client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorNetworkConnectionLost userInfo:nil]];
830
831 - (void) stopLoading_ {
832 [[NSNotificationCenter defaultCenter] removeObserver:self];
833
834 if (handle_ != nil) {
835 [handle_ release];
836 handle_ = nil;
837 }
838
839 if (pid_ != -1) {
840 kill(pid_, SIGTERM);
841 int status;
842 _syscall(waitpid(pid_, &status, 0));
843 pid_ = -1;
844 }
845 }
846
847 - (void) stopLoading {
848 [self
849 performSelectorOnMainThread:@selector(stopLoading_)
850 withObject:nil
851 waitUntilDone:NO
852 ];
853 }
854
855 @end
856 /* }}} */
857
858 template <typename Type_>
859 static void nlset(Type_ &function, struct nlist *nl, size_t index) {
860 struct nlist &name(nl[index]);
861 uintptr_t value(name.n_value);
862 if ((name.n_desc & N_ARM_THUMB_DEF) != 0)
863 value |= 0x00000001;
864 function = reinterpret_cast<Type_>(value);
865 }
866
867 template <typename Type_>
868 static void dlset(Type_ &function, const char *name) {
869 function = reinterpret_cast<Type_>(dlsym(RTLD_DEFAULT, name));
870 }
871
872 template <typename Type_>
873 static void msset_(Type_ &function, const char *name, MSImageRef handle) {
874 function = reinterpret_cast<Type_>(MSFindSymbol(handle, name));
875 }
876
877 #define msset(function, handle) \
878 msset_(function, "_" #function, handle)
879
880 @implementation WebCycriptLockScreenController
881
882 static void $UIWebViewWebViewDelegate$webView$addMessageToConsole$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *message) {
883 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
884 if ([uiWebView respondsToSelector:@selector(webView:addMessageToConsole:)])
885 [uiWebView webView:view addMessageToConsole:message];
886 }
887
888 static void $UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebScriptObject *window, WebFrame *frame) {
889 UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
890 if ([uiWebView respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
891 [uiWebView webView:view didClearWindowObject:window forFrame:frame];
892 }
893
894 + (void) initialize {
895 if (Class $UIWebViewWebViewDelegate = objc_getClass("UIWebViewWebViewDelegate")) {
896 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:addMessageToConsole:), (IMP) &$UIWebViewWebViewDelegate$webView$addMessageToConsole$, "v16@0:4@8@12");
897 class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didClearWindowObject:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$, "v20@0:4@8@12@16");
898 }
899
900 if (float *_UIScrollViewDecelerationRateNormal = reinterpret_cast<float *>(dlsym(RTLD_DEFAULT, "UIScrollViewDecelerationRateNormal")))
901 CYScrollViewDecelerationRateNormal = *_UIScrollViewDecelerationRateNormal;
902 else // XXX: this actually might be fast on some older systems: we should look into this
903 CYScrollViewDecelerationRateNormal = 0.998;
904
905 iOS4 = kCFCoreFoundationVersionNumber >= 550.32;
906 iOS32 = !iOS4 && kCFCoreFoundationVersionNumber >= 478.61;
907
908 int maxproc;
909 size_t size(sizeof(maxproc));
910 if (sysctlbyname("kern.maxproc", &maxproc, &size, NULL, 0) == -1)
911 NSLog(@"sysctlbyname(\"kern.maxproc\", ?)");
912 else if (maxproc < 72) {
913 maxproc = 72;
914 if (sysctlbyname("kern.maxproc", NULL, NULL, &maxproc, sizeof(maxproc)) == -1)
915 NSLog(@"sysctlbyname(\"kern.maxproc\", #)");
916 }
917
918 apr_initialize();
919
920 [NSURLProtocol registerClass:[CydgetURLProtocol class]];
921 [WebView registerURLSchemeAsLocal:@"cydget"];
922
923 [NSURLProtocol registerClass:[CydgetCGIURLProtocol class]];
924 [WebView registerURLSchemeAsLocal:@"cydget-cgi"];
925
926 MSImageRef JavaScriptCore(MSGetImageByName("/System/Library/PrivateFrameworks/JavaScriptCore.framework/JavaScriptCore"));
927 MSImageRef WebCore(MSGetImageByName("/System/Library/PrivateFrameworks/WebCore.framework/WebCore"));
928
929 if (!iOS4) {
930 void (*_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE)(JSC::SourceCode **, JSC::JSGlobalData *, int *, JSC::UString *);
931 dlset(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, "_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE");
932 if (_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE != NULL)
933 MSHookFunction(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, MSHake(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE));
934 }
935
936 struct nlist nl[9];
937 memset(nl, 0, sizeof(nl));
938
939 nl[0].n_un.n_name = (char *) "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE";
940
941 nl[1].n_un.n_name = (char *) "__ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi";
942
943 nl[2].n_un.n_name = (char *) "__ZN7WebCore12CachedScript6scriptEv";
944 nl[3].n_un.n_name = (char *) "__ZNK7WebCore20StringSourceProvider6sourceEv";
945
946 nl[4].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i";
947 nl[5].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE";
948
949 nl[6].n_un.n_name = (char *) "__ZN7WebCore6String6appendEPKtj";
950 nl[7].n_un.n_name = (char *) "__ZN7WebCore6String8truncateEj";
951
952 nlist("/System/Library/PrivateFrameworks/WebCore.framework/WebCore", nl);
953
954 bool (*_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE)(const WebCore::String &);
955 nlset(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, nl, 0);
956 if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE == NULL)
957 MSHookSymbol(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKN3WTF6StringE", WebCore);
958 if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE != NULL)
959 MSHookFunction(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, MSHake(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE));
960
961 void (*_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi)(void *, const WebCore::String &, const WebCore::KURL &, int);
962 nlset(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, nl, 1);
963 if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi != NULL)
964 MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi));
965
966 if (iOS4) {
967 const WebCore::String &(*_ZN7WebCore12CachedScript6scriptEv)(void *);
968 nlset(_ZN7WebCore12CachedScript6scriptEv, nl, 2);
969 if (_ZN7WebCore12CachedScript6scriptEv != NULL)
970 MSHookFunction(_ZN7WebCore12CachedScript6scriptEv, MSHake(_ZN7WebCore12CachedScript6scriptEv));
971 }
972
973 State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i)(void *, const WebCore::String &, State, const WebCore::String &, int);
974 nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, nl, 4);
975 if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i != NULL)
976 MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i));
977
978 if (iOS4) {
979 State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE)(void *, void *, State);
980 nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, nl, 5);
981 if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE != NULL)
982 MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE));
983 }
984
985 nlset(_ZN7WebCore6String6appendEPKtj, nl, 6);
986 nlset(_ZN7WebCore6String8truncateEj, nl, 7);
987
988 void (*_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE)(void *, const WebCore::String &, const WebCore::KURL &, void *);
989 msset(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, WebCore);
990 if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE != NULL)
991 MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE));
992
993 void (*_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE)(void *, void *, int);
994 msset(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, WebCore);
995 if (_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE != NULL)
996 MSHookFunction(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, MSHake(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE));
997
998 msset(_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE, WebCore);
999
1000 void (*_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE)(void *, void *);
1001 msset(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, WebCore);
1002 if (_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE != NULL)
1003 MSHookFunction(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, MSHake(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE));
1004
1005 if (_ZN7WebCore6String6appendEPKtj == NULL)
1006 msset(_ZN7WebCore6String6appendEPKtj, JavaScriptCore);
1007 if (_ZN7WebCore6String6appendEPKtj == NULL)
1008 MSHookSymbol(_ZN7WebCore6String6appendEPKtj, "__ZN3WTF6String6appendEPKtj", JavaScriptCore);
1009
1010 if (_ZN7WebCore6String8truncateEj == NULL)
1011 msset(_ZN7WebCore6String8truncateEj, JavaScriptCore);
1012 if (_ZN7WebCore6String8truncateEj == NULL)
1013 MSHookSymbol(_ZN7WebCore6String8truncateEj, "__ZN3WTF6String8truncateEj", JavaScriptCore);
1014
1015 msset(_ZNK7WebCore6String10charactersEv, WebCore);
1016
1017 msset(_ZN7WebCore6String29charactersWithNullTerminationEv, JavaScriptCore);
1018 if (_ZN7WebCore6String29charactersWithNullTerminationEv == NULL)
1019 MSHookSymbol(_ZN7WebCore6String29charactersWithNullTerminationEv, "__ZN3WTF6String29charactersWithNullTerminationEv", JavaScriptCore);
1020
1021 msset(_ZNK7WebCore6String6lengthEv, WebCore);
1022 }
1023
1024 + (id) rootViewController {
1025 return [[[self alloc] init] autorelease];
1026 }
1027
1028 - (void) loadView {
1029 [self setView:[[[WebCydgetLockScreenView alloc] init] autorelease]];
1030 }
1031
1032 - (BOOL) showAwayItems {
1033 return NO;
1034 }
1035
1036 @end
1037
1038 MSClassHook(WebView)
1039 MSMetaClassHook(WebView)
1040
1041 MSClassMessageHook0(void, WebView, enableWebThread) {
1042 if (kCFCoreFoundationVersionNumber >= 478.61)
1043 return MSOldCall();
1044
1045 NSLog(@"-[WebView enableWebThread]");
1046 }