LockScreen.mm

   1 /* Cydget - open-source AwayView plugin multiplexer
   2  * Copyright (C) 2009-2014  Jay Freeman (saurik)
   3 */
   4
   5 /* GNU General Public License, Version 3 {{{ */
   6 /*
   7  * Cydia is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published
   9  * by the Free Software Foundation, either version 3 of the License,
  10  * or (at your option) any later version.
  11  *
  12  * Cydia is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Cydia.  If not, see <http://www.gnu.org/licenses/>.
  19 **/
  20 /* }}} */
  21
  22 #include <CydiaSubstrate/CydiaSubstrate.h>
  23 #include <UIKit/UIKit.h>
  24
  25 #include <sys/sysctl.h>
  26 #include <pthread.h>
  27
  28 #include <SpringBoardUI/SBAwayViewPluginController.h>
  29
  30 #include <WebKit/WebFrame.h>
  31 #include <WebKit/WebView.h>
  32 #include <WebKit/WebPreferences-WebPrivate.h>
  33
  34 #include "yieldToSelector.h"
  35
  36 #ifdef USE_ICU_REGEX
  37 #include <unicode/uregex.h>
  38 #else
  39 #include <pcre.h>
  40 #endif
  41
  42 #define _transient
  43 #define _forever for (;;)
  44
  45 _disused static unsigned trace_;
  46
  47 #define _trace() do { \
  48     NSLog(@"_trace(%u)@%s:%u[%s](%p)\n", \
  49         trace_++, __FILE__, __LINE__, __FUNCTION__, pthread_self() \
  50     ); \
  51 } while (false)
  52
  53 #define _assert(test) do \
  54     if (!(test)) { \
  55         NSLog(@"_assert(%d:%s)@%s:%u[%s]\n", errno, #test, __FILE__, __LINE__, __FUNCTION__); \
  56         exit(-1); \
  57     } \
  58 while (false)
  59
  60 #define _syscall(expr) \
  61     do if ((long) (expr) != -1) \
  62         break; \
  63     else switch (errno) { \
  64         case EINTR: \
  65             continue; \
  66         default: \
  67             _assert(false); \
  68     } while (true)
  69
  70 extern "C" UIImage *_UIImageWithName(NSString *name);
  71
  72 typedef uint16_t UChar;
  73
  74 @interface TPBottomLockBar
  75 - (CGFloat) defaultHeight;
  76 @end
  77
  78 @interface UIApplication (Apple)
  79 - (void) applicationOpenURL:(NSURL *)url;
  80 @end
  81
  82 @interface UIScroller : UIView
  83 - (CGSize) contentSize;
  84 - (void) setDirectionalScrolling:(BOOL)directional;
  85 - (void) setOffset:(CGPoint)offset;
  86 - (void) setScrollDecelerationFactor:(CGFloat)factor;
  87 - (void) setScrollHysteresis:(CGFloat)hysteresis;
  88 - (void) setThumbDetectionEnabled:(BOOL)enabled;
  89 @end
  90
  91 @interface UIWebDocumentView : UIView
  92 - (CGRect) documentBounds;
  93 - (void) enableReachability;
  94 - (void) loadRequest:(NSURLRequest *)request;
  95 - (void) redrawScaledDocument;
  96 - (void) setAllowsImageSheet:(BOOL)allows;
  97 - (void) setAllowsMessaging:(BOOL)allows;
  98 - (void) setAutoresizes:(BOOL)autoresizes;
  99 - (void) setContentsPosition:(NSInteger)position;
 100 - (void) setDrawsBackground:(BOOL)draws;
 101 - (void) _setDocumentType:(NSInteger)type;
 102 - (void) setDrawsGrid:(BOOL)draws;
 103 - (void) setInitialScale:(float)scale forDocumentTypes:(NSInteger)types;
 104 - (void) setLogsTilingChanges:(BOOL)logs;
 105 - (void) setMinimumScale:(float)scale forDocumentTypes:(NSInteger)types;
 106 - (void) setMinimumSize:(CGSize)size;
 107 - (void) setMaximumScale:(float)scale forDocumentTypes:(NSInteger)tpyes;
 108 - (void) setSmoothsFonts:(BOOL)smooths;
 109 - (void) setTileMinificationFilter:(NSString *)filter;
 110 - (void) setTileSize:(CGSize)size;
 111 - (void) setTilingEnabled:(BOOL)enabled;
 112 - (void) setViewportSize:(CGSize)size forDocumentTypes:(NSInteger)types;
 113 - (void) setZoomsFocusedFormControl:(BOOL)zooms;
 114 - (void) useSelectionAssistantWithMode:(NSInteger)mode;
 115 - (WebView *) webView;
 116 @end
 117
 118 @interface UIView (Apple)
 119 - (UIScroller *) _scroller;
 120 - (void) setClipsSubviews:(BOOL)clips;
 121 - (void) setEnabledGestures:(NSInteger)gestures;
 122 - (void) setFixedBackgroundPattern:(BOOL)fixed;
 123 - (void) setGestureDelegate:(id)delegate;
 124 - (void) setNeedsDisplayOnBoundsChange:(BOOL)needs;
 125 - (void) setValue:(NSValue *)value forGestureAttribute:(NSInteger)attribute;
 126 - (void) setZoomScale:(float)scale duration:(double)duration;
 127 - (void) _setZoomScale:(float)scale duration:(double)duration;
 128 @end
 129
 130 @interface SBLockScreenView : UIView
 131 - (BOOL) mediaControlsHidden;
 132 @end
 133
 134 @interface SBLockScreenNotificationListController : NSObject
 135 - (BOOL) hasAnyContent;
 136 @end
 137
 138 @interface SBLockScreenViewController : UIViewController
 139 - (SBLockScreenView *) lockScreenView;
 140 - (BOOL) isShowingMediaControls;
 141 - (void) _setMediaControlsVisible:(BOOL)visible;
 142 - (SBLockScreenNotificationListController *) _notificationController;
 143 @end
 144
 145 @interface SBLockScreenManager : NSObject
 146 + (SBLockScreenManager *) sharedInstance;
 147 - (SBLockScreenViewController *) lockScreenViewController;
 148 @end
 149
 150 @protocol CydgetController
 151 - (NSDictionary *) currentConfiguration;
 152 - (NSString *) currentPath;
 153 @end
 154
 155 static Class $CydgetController(objc_getClass("CydgetController"));
 156
 157 static bool iOS32, iOS4;
 158
 159 @interface NSString (UIKit)
 160 - (NSString *) stringByAddingPercentEscapes;
 161 @end
 162
 163 @implementation UIWebDocumentView (WebCycript)
 164
 165 - (void) _setScrollerOffset:(CGPoint)offset {
 166     UIScroller *scroller([self _scroller]);
 167
 168     CGSize size([scroller contentSize]);
 169     CGSize bounds([scroller bounds].size);
 170
 171     CGPoint max;
 172     max.x = size.width - bounds.width;
 173     max.y = size.height - bounds.height;
 174
 175     // wtf Apple?!
 176     if (max.x < 0)
 177         max.x = 0;
 178     if (max.y < 0)
 179         max.y = 0;
 180
 181     offset.x = offset.x < 0 ? 0 : offset.x > max.x ? max.x : offset.x;
 182     offset.y = offset.y < 0 ? 0 : offset.y > max.y ? max.y : offset.y;
 183
 184     [scroller setOffset:offset];
 185 }
 186
 187 @end
 188
 189 #ifdef USE_ICU_REGEX
 190 /* ICU Regular Expression {{{ */
 191 class RegEx {
 192   private:
 193     URegularExpression *regex_;
 194
 195   public:
 196     RegEx(const char *regex) {
 197         UParseError error;
 198         UErrorCode status(U_ZERO_ERROR);
 199         regex_ = uregex_openC(regex, 0, &error, &status);
 200         if (U_FAILURE(status))
 201             @throw [NSException exceptionWithName:NSInvalidArgumentException reason:[NSString stringWithFormat:@"*** RegEx(): [%u] %s", error.offset, u_errorName(status)] userInfo:nil];
 202     }
 203
 204     ~RegEx() {
 205         uregex_close(regex_);
 206     }
 207
 208     bool operator ()(NSString *string) {
 209         return operator ()(reinterpret_cast<const uint16_t *>([string cStringUsingEncoding:NSUTF16StringEncoding]), [string length]);
 210     }
 211
 212     bool operator ()(const UChar *data, size_t size) {
 213         UErrorCode status(U_ZERO_ERROR);
 214         uregex_setText(regex_, data, size, &status);
 215         _assert(U_SUCCESS(status));
 216         bool matches(uregex_matches(regex_, 0, &status));
 217         _assert(U_SUCCESS(status));
 218         return matches;
 219     }
 220 };
 221 /* }}} */
 222 #else
 223 /* Perl-Compatible RegEx {{{ */
 224 class RegEx {
 225   private:
 226     pcre *code_;
 227     pcre_extra *study_;
 228     int capture_;
 229     int *matches_;
 230     const char *data_;
 231
 232   public:
 233     RegEx(const char *regex) :
 234         study_(NULL)
 235     {
 236         const char *error;
 237         int offset;
 238         code_ = pcre_compile(regex, 0, &error, &offset, NULL);
 239
 240         if (code_ == NULL)
 241             @throw [NSException exceptionWithName:NSInvalidArgumentException reason:[NSString stringWithFormat:@"*** RegEx(): [%u] %s", offset, error] userInfo:nil];
 242
 243         pcre_fullinfo(code_, study_, PCRE_INFO_CAPTURECOUNT, &capture_);
 244         matches_ = new int[(capture_ + 1) * 3];
 245     }
 246
 247     ~RegEx() {
 248         pcre_free(code_);
 249         delete matches_;
 250     }
 251
 252     bool operator ()(NSString *data) {
 253         // XXX: length is for characters, not for bytes
 254         return operator ()([data UTF8String], [data length]);
 255     }
 256
 257     bool operator ()(const char *data, size_t size) {
 258         data_ = data;
 259         return pcre_exec(code_, study_, data, size, 0, 0, matches_, (capture_ + 1) * 3) >= 0;
 260     }
 261 };
 262 /* }}} */
 263 #endif
 264
 265 @interface DOMCSSStyleSheet : NSObject
 266 - (int) addRule:(NSString *)rule style:(NSString *)style index:(unsigned)index;
 267 - (void) deleteRule:(unsigned)index;
 268 @end
 269
 270 @interface DOMStyleSheetList : NSObject
 271 - (DOMCSSStyleSheet *) item:(unsigned)index;
 272 @end
 273
 274 @interface DOMDocument : NSObject
 275 - (DOMStyleSheetList *) styleSheets;
 276 @end
 277
 278 static float CYScrollViewDecelerationRateNormal;
 279
 280 @interface NSURL (Apple)
 281 - (BOOL) isSpringboardHandledURL;
 282 @end
 283
 284 @interface UIScrollView (Apple)
 285 - (void) setDecelerationRate:(CGFloat)value;
 286 - (void) setScrollingEnabled:(BOOL)enabled;
 287 - (void) setShowBackgroundShadow:(BOOL)show;
 288 @end
 289
 290 @interface UIWebView (Apple)
 291 - (UIWebDocumentView *) _documentView;
 292 - (void) setDataDetectorTypes:(NSInteger)types;
 293 - (void) _setDrawInWebThread:(BOOL)draw;
 294 - (UIScrollView *) _scrollView;
 295 - (UIScroller *) _scroller;
 296 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message;
 297 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame;
 298 @end
 299
 300 @interface WebView (Apple)
 301 - (void) _setLayoutInterval:(float)interval;
 302 - (void) _setAllowsMessaging:(BOOL)allows;
 303 - (void) setShouldUpdateWhileOffscreen:(BOOL)update;
 304 @end
 305
 306 @protocol CydgetWebViewDelegate <UIWebViewDelegate>
 307 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame;
 308 @end
 309
 310 @class UIWebViewWebViewDelegate;
 311
 312 @interface CydgetWebView : UIWebView {
 313 }
 314
 315 - (void) updateStyles;
 316
 317 @end
 318
 319 MSClassHook(UIApplication)
 320 MSClassHook(SBLockScreenManager)
 321
 322 MSInstanceMessageHook1(void, UIApplication, openURL, NSURL *, url) {
 323     [self applicationOpenURL:url];
 324 }
 325
 326 @implementation NSURL (Cydget)
 327
 328 - (NSNumber *) cydget$isSpringboardHandledURL {
 329     return [NSNumber numberWithBool:[self isSpringboardHandledURL]];
 330 }
 331
 332 @end
 333
 334 MSClassHook(NSURL)
 335
 336 MSInstanceMessageHook0(BOOL, NSURL, isSpringboardHandledURL) {
 337     if (![NSThread isMainThread])
 338         return MSOldCall();
 339
 340     return [[self cydget$yieldToSelector:@selector(cydget$isSpringboardHandledURL)] boolValue];
 341 }
 342
 343 @implementation CydgetWebView
 344
 345 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
 346     NSObject<CydgetWebViewDelegate> *delegate((NSObject<CydgetWebViewDelegate> *) [self delegate]);
 347     if ([delegate respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
 348         [delegate webView:view didClearWindowObject:window forFrame:frame];
 349     if ([UIWebView instancesRespondToSelector:@selector(webView:didClearWindowObject:forFrame:)])
 350         [super webView:view didClearWindowObject:window forFrame:frame];
 351 }
 352
 353 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message {
 354     NSLog(@"addMessageToConsole:%@", message);
 355
 356     if ([UIWebView instancesRespondToSelector:@selector(webView:addMessageToConsole:)])
 357         [super webView:view addMessageToConsole:message];
 358 }
 359
 360 - (void) updateStyles {
 361     DOMCSSStyleSheet *sheet([[[[[[self _documentView] webView] mainFrame] DOMDocument] styleSheets] item:0]);
 362     [sheet addRule:@"cydget" style:@"color: black" index:0];
 363     [sheet deleteRule:0];
 364 }
 365
 366 @end
 367
 368 @interface WebCydgetLockScreenView : UIView <UIWebViewDelegate> {
 369     CydgetWebView *webview_;
 370     UIScrollView *scroller_;
 371     NSString *cycript_;
 372 }
 373
 374 - (void) updateStyles;
 375
 376 @end
 377
 378 @implementation WebCydgetLockScreenView
 379
 380 //#include "UICaboodle/UCInternal.h"
 381
 382 - (void) dealloc {
 383     [[NSNotificationCenter defaultCenter] removeObserver:self];
 384     [webview_ setDelegate:nil];
 385     [webview_ release];
 386     [super dealloc];
 387 }
 388
 389 - (void) loadRequest:(NSURLRequest *)request {
 390     [webview_ loadRequest:request];
 391 }
 392
 393 - (void) loadURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {
 394     [self loadRequest:[NSURLRequest
 395         requestWithURL:url
 396         cachePolicy:policy
 397         timeoutInterval:30.0
 398     ]];
 399 }
 400
 401 - (void) loadURL:(NSURL *)url {
 402     [self loadURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy];
 403 }
 404
 405 - (id) initWithURL:(NSURL *)url {
 406     CGRect frame = [[UIScreen mainScreen] bounds];
 407     if (kCFCoreFoundationVersionNumber < 800)
 408         frame.size.height -= 20; //[[[$SBStatusBarController sharedStatusBarController] statusBarView] frame].size.height;
 409
 410     if ((self = [super initWithFrame:frame]) != nil) {
 411         CGRect bounds([self bounds]);
 412         if (kCFCoreFoundationVersionNumber < 800)
 413             bounds.size.height -= [objc_getClass("TPBottomLockBar") defaultHeight];
 414
 415         webview_ = [[CydgetWebView alloc] initWithFrame:bounds];
 416         [webview_ setDelegate:self];
 417         [self addSubview:webview_];
 418
 419         if ([webview_ respondsToSelector:@selector(setDataDetectorTypes:)])
 420             [webview_ setDataDetectorTypes:0x80000000];
 421         else
 422             [webview_ setDetectsPhoneNumbers:NO];
 423
 424         [webview_ setScalesPageToFit:YES];
 425
 426         if (kCFCoreFoundationVersionNumber < 478.61)
 427             if ([webview_ respondsToSelector:@selector(_setDrawInWebThread:)])
 428                 [webview_ _setDrawInWebThread:NO];
 429
 430         UIWebDocumentView *document([webview_ _documentView]);
 431         WebView *webview([document webView]);
 432         WebPreferences *preferences([webview preferences]);
 433
 434         [document setTileSize:CGSizeMake(bounds.size.width, 500)];
 435
 436         [document setBackgroundColor:[UIColor clearColor]];
 437         [document setDrawsBackground:NO];
 438
 439         [webview setPreferencesIdentifier:@"WebCycript"];
 440
 441         if ([webview respondsToSelector:@selector(_setLayoutInterval:)])
 442             [webview _setLayoutInterval:0];
 443         else
 444             [preferences _setLayoutInterval:0];
 445
 446         [preferences setCacheModel:WebCacheModelDocumentViewer];
 447         [preferences setJavaScriptCanOpenWindowsAutomatically:YES];
 448         [preferences setOfflineWebApplicationCacheEnabled:YES];
 449
 450         if ([webview respondsToSelector:@selector(setShouldUpdateWhileOffscreen:)])
 451             [webview setShouldUpdateWhileOffscreen:NO];
 452
 453         if ([document respondsToSelector:@selector(setAllowsMessaging:)])
 454             [document setAllowsMessaging:YES];
 455         if ([webview respondsToSelector:@selector(_setAllowsMessaging:)])
 456             [webview _setAllowsMessaging:YES];
 457
 458         if ([webview_ respondsToSelector:@selector(_scrollView)]) {
 459             scroller_ = [webview_ _scrollView];
 460
 461             [scroller_ setDirectionalLockEnabled:YES];
 462             [scroller_ setDecelerationRate:CYScrollViewDecelerationRateNormal];
 463             [scroller_ setDelaysContentTouches:NO];
 464
 465             [scroller_ setCanCancelContentTouches:YES];
 466
 467             [scroller_ setAlwaysBounceVertical:NO];
 468         } else if ([webview_ respondsToSelector:@selector(_scroller)]) {
 469             UIScroller *scroller([webview_ _scroller]);
 470             scroller_ = (UIScrollView *) scroller;
 471
 472             [scroller setDirectionalScrolling:YES];
 473             [scroller setScrollDecelerationFactor:CYScrollViewDecelerationRateNormal]; /* 0.989324 */
 474             [scroller setScrollHysteresis:0]; /* 8 */
 475
 476             [scroller setThumbDetectionEnabled:NO];
 477         }
 478
 479         [webview_ setOpaque:NO];
 480         [webview_ setBackgroundColor:[UIColor clearColor]];
 481
 482         [scroller_ setFixedBackgroundPattern:YES];
 483         [scroller_ setBackgroundColor:[UIColor clearColor]];
 484         [scroller_ setClipsSubviews:NO];
 485
 486         [scroller_ setBounces:YES];
 487         [scroller_ setShowBackgroundShadow:NO]; /* YES */
 488
 489         [self setAutoresizingMask:(UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight)];
 490         [webview_ setAutoresizingMask:(UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight)];
 491
 492         NSDictionary *configuration([$CydgetController currentConfiguration]);
 493         cycript_ = [configuration objectForKey:@"CycriptURLs"];
 494
 495         [scroller_ setScrollingEnabled:[[configuration objectForKey:@"Scrollable"] boolValue]];
 496
 497         [self loadURL:url];
 498
 499         [[NSNotificationCenter defaultCenter]
 500             addObserver:self
 501             selector:@selector(mediaControlsDidSomething:)
 502             name:@"SBLockScreenViewControllerMediaControlsDidShow"
 503             object:nil
 504         ];
 505
 506         [[NSNotificationCenter defaultCenter]
 507             addObserver:self
 508             selector:@selector(mediaControlsDidSomething:)
 509             name:@"SBLockScreenViewControllerMediaControlsDidHide"
 510             object:nil
 511         ];
 512     } return self;
 513 }
 514
 515 - (void) mediaControlsDidSomething:(NSNotification *)notification {
 516     [self updateStyles];
 517 }
 518
 519 - (void) webView:(WebView *)webview didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
 520     if (cycript_ != nil)
 521         if (NSString *href = [[[[frame dataSource] request] URL] absoluteString])
 522             if (RegEx([cycript_ UTF8String])(href))
 523                 if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
 524                     if (void (*CYSetupContext)(JSGlobalContextRef) = reinterpret_cast<void (*)(JSGlobalContextRef)>(dlsym(handle, "CydgetSetupContext"))) {
 525                         WebFrame *frame([webview mainFrame]);
 526                         JSGlobalContextRef context([frame globalContext]);
 527                         @try {
 528                             CYSetupContext(context);
 529                         } @catch (NSException *e) {
 530                             NSLog(@"*** CydgetSetupContext => %@", e);
 531                         }
 532                     }
 533 }
 534
 535 - (void) updateStyles {
 536     [webview_ updateStyles];
 537 }
 538
 539 @end
 540
 541 @interface WebCycriptLockScreenController : SBAwayViewPluginController {
 542     NSDictionary *configuration_;
 543     bool legacy_;
 544     bool media_;
 545     bool items_;
 546     WebCydgetLockScreenView *background_;
 547     WebCydgetLockScreenView *foreground_;
 548 }
 549
 550 @end
 551
 552 #include <string>
 553
 554 struct State {
 555     unsigned state;
 556 };
 557
 558 namespace JSC {
 559     class JSGlobalData;
 560     class UString;
 561 }
 562
 563 namespace WebCore {
 564     class KURL;
 565 }
 566
 567 namespace WebCore {
 568 struct String {
 569     void *impl_;
 570 }; }
 571
 572 namespace JSC {
 573 struct SourceCode {
 574     void *provider_;
 575     int start_;
 576     int end_;
 577     int line_;
 578 }; }
 579
 580 namespace JSC {
 581 union ScriptSourceCode {
 582     struct {
 583         JSC::SourceCode source_;
 584     } Old;
 585     struct {
 586         void *provider_;
 587         JSC::SourceCode source_;
 588     } New;
 589 }; }
 590
 591 class CFStringStruct {
 592   private:
 593     CFStringRef value_;
 594
 595   public:
 596     CFStringStruct() :
 597         value_(NULL)
 598     {
 599     }
 600
 601     ~CFStringStruct() {
 602     }
 603
 604     operator CFStringRef() const {
 605         return value_;
 606     }
 607
 608     operator NSString *() const {
 609         return (NSString *) value_;
 610     }
 611 };
 612
 613 // String Helpers {{{
 614 static const UChar *(*_ZNK7WebCore6String10charactersEv)(const WebCore::String *);
 615 static const UChar *(*_ZN7WebCore6String29charactersWithNullTerminationEv)(const WebCore::String *);
 616 static CFStringStruct (*_ZNK3WTF6String14createCFStringEv)(const WebCore::String *);
 617 static unsigned (*_ZNK7WebCore6String6lengthEv)(const WebCore::String *);
 618
 619 static bool StringGet(const WebCore::String &string, const UChar *&data, size_t &length) {
 620     bool terminated;
 621
 622     if (false) {
 623     } else if (_ZNK7WebCore6String10charactersEv != NULL) {
 624         data = (*_ZNK7WebCore6String10charactersEv)(&string);
 625         terminated = false;
 626     } else if (_ZN7WebCore6String29charactersWithNullTerminationEv != NULL) {
 627         data = (*_ZN7WebCore6String29charactersWithNullTerminationEv)(&string);
 628         terminated = true;
 629     } else if (_ZNK3WTF6String14createCFStringEv != NULL) {
 630         CFStringStruct cf((*_ZNK3WTF6String14createCFStringEv)(&string));
 631         data = (const UChar *) [cf cStringUsingEncoding:NSUTF16StringEncoding];
 632         length = CFStringGetLength(cf);
 633         [cf autorelease];
 634     } else return false;
 635
 636     if (data == NULL)
 637         return false;
 638
 639     if (_ZNK7WebCore6String6lengthEv != NULL)
 640         length = (*_ZNK7WebCore6String6lengthEv)(&string);
 641     else if (terminated)
 642         for (length = 0; data[length] != 0; ++length);
 643     else return false;
 644
 645     return true;
 646 }
 647
 648 static bool StringEquals(const WebCore::String &string, const char *value) {
 649     const UChar *data;
 650     size_t size;
 651     if (!StringGet(string, data, size))
 652         return false;
 653
 654     size_t length(strlen(value));
 655     if (size != length)
 656         return false;
 657
 658     for (size_t index(0); index != length; ++index)
 659         if (data[index] != value[index])
 660             return false;
 661
 662     return true;
 663 }
 664 // }}}
 665 // State Machine {{{
 666 static bool cycript_;
 667
 668 MSHook(bool, _ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, const WebCore::String &mime) {
 669     if (!StringEquals(mime, "text/cycript")) {
 670         cycript_ = false;
 671         return __ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE(mime);
 672     }
 673
 674     static void *handle(dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL));
 675     if (handle == NULL)
 676         return false;
 677
 678     cycript_ = true;
 679     return true;
 680 }
 681 // }}}
 682 // Script Compiler {{{
 683 static void Log(const WebCore::String &string) {
 684 #if 0
 685     const UChar *data;
 686     size_t length;
 687     if (!StringGet(string, data, length))
 688         return;
 689
 690     UChar terminated[length + 1];
 691     terminated[length] = 0;
 692     memcpy(terminated, data, length * 2);
 693     NSLog(@"wtf %p:%zu:%S:", &string, length, terminated);
 694 #endif
 695 }
 696
 697 static bool Cycriptify(const uint16_t *&data, size_t &size) {
 698     cycript_ = false;
 699
 700     if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
 701         if (void (*CydgetMemoryParse)(const uint16_t **, size_t *) = reinterpret_cast<void (*)(const uint16_t **, size_t *)>(dlsym(handle, "CydgetMemoryParse"))) @try {
 702             CydgetMemoryParse(&data, &size);
 703             return true;
 704         } @catch (NSException *e) {
 705             NSLog(@"*** CydgetMemoryParse => %@", e);
 706         }
 707     return false;
 708 }
 709
 710 static void (*_ZN7WebCore6String6appendEPKtj)(WebCore::String *, const UChar *, unsigned);
 711 static void (*_ZN7WebCore6String8truncateEj)(WebCore::String *, unsigned);
 712
 713 static void Cycriptify(const WebCore::String &source, int *psize = NULL) {
 714     if (!cycript_)
 715         return;
 716     cycript_ = false;
 717
 718     const UChar *data;
 719     size_t length;
 720     if (!StringGet(source, data, length))
 721         return;
 722
 723     size_t size(length);
 724     if (!Cycriptify(data, size))
 725         return;
 726
 727     WebCore::String &script(const_cast<WebCore::String &>(source));
 728     _ZN7WebCore6String8truncateEj(&script, 0);
 729     _ZN7WebCore6String6appendEPKtj(&script, data, size);
 730
 731     if (psize != NULL)
 732         *psize = size;
 733
 734     free((void *) data);
 735
 736     Log(source);
 737 }
 738 // }}}
 739
 740 static WebCore::String *string;
 741
 742 // iOS 2.x
 743 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, void *_this, const WebCore::String &string, State state, const WebCore::String &url, int line) {
 744     Cycriptify(string);
 745     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i(_this, string, state, url, line);
 746 }
 747
 748 // iOS 3.x
 749 MSHook(void, _ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, JSC::SourceCode **_this, JSC::JSGlobalData *global, int *line, JSC::UString *message) {
 750     /*if (cycript_) {
 751         JSC::SourceCode *source(_this[iOS32 ? 6 : 0]);
 752         const uint16_t *data(source->data());
 753         size_t size(source->length());
 754
 755         if (Cycriptify(data, size)) {
 756             source->~SourceCode();
 757             // XXX: I actually don't have the original URL here: pants
 758             new (source) JSC::SourceCode(JSC::UStringSourceProvider::create(JSC::UString(data, size), "cycript://"), 1);
 759             free((void *) data);
 760         }
 761     }*/
 762
 763     return __ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE(_this, global, line, message);
 764 }
 765
 766 // iOS 3.x cdata
 767 MSHook(const WebCore::String &, _ZNK7WebCore4Node11textContentEb, void *_this, bool convert) {
 768     const WebCore::String &code(__ZNK7WebCore4Node11textContentEb(_this, convert));
 769     string = const_cast<WebCore::String *>(&code);
 770     Cycriptify(code);
 771     return code;
 772 }
 773
 774 // iOS 4.x cdata
 775 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, void *_this, const WebCore::String &source, const WebCore::KURL &url, int line) {
 776     Cycriptify(source);
 777     return __ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi(_this, source, url, line);
 778 }
 779
 780 // iOS 4.x+5.0 @src=
 781 MSHook(const WebCore::String &, _ZN7WebCore12CachedScript6scriptEv, void *_this) {
 782     const WebCore::String &script(__ZN7WebCore12CachedScript6scriptEv(_this));
 783     string = const_cast<WebCore::String *>(&script);
 784     return script;
 785 }
 786
 787 // iOS 4.x @src=
 788 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, void *_this, JSC::ScriptSourceCode &script, State state) {
 789     if (string != NULL) {
 790         JSC::SourceCode *source(iOS4 ? &script.New.source_ : &script.Old.source_);
 791         Cycriptify(*string, &source->end_);
 792         string = NULL;
 793     }
 794
 795     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE(_this, script, state);
 796 }
 797
 798 // iOS 5.0 cdata
 799 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, void *_this, const WebCore::String &source, const WebCore::KURL &url, void *position) {
 800     Cycriptify(source);
 801     return __ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE(_this, source, url, position);
 802 }
 803
 804 // iOS 5.0 @src=
 805 MSHook(void, _ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, void *_this, void *position, int legacy) {
 806     string = NULL;
 807     return __ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE(_this, position, legacy);
 808 }
 809
 810 void (*$_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE)(void *_this, int legacy);
 811
 812 // iOS 5.0 @src=
 813 MSHook(void, _ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, void *_this, JSC::ScriptSourceCode &script) {
 814     if (string != NULL) {
 815         JSC::SourceCode *source(&script.New.source_);
 816         $_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE(_this, 0);
 817         Cycriptify(*string, &source->end_);
 818         string = NULL;
 819     }
 820
 821     return __ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE(_this, script);
 822 }
 823
 824 // iOS 6.0 cdata
 825 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, void *_this, const WebCore::String &source, const WebCore::KURL &url, void *position) {
 826     Cycriptify(source);
 827     return __ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE(_this, source, url, position);
 828 }
 829
 830 // iOS 6.0 @src=
 831 MSHook(void, _ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, void *_this, void *position, int legacy) {
 832     string = NULL;
 833     return __ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE(_this, position, legacy);
 834 }
 835
 836 /* Cydget:// Protocol {{{ */
 837 @interface CydgetURLProtocol : NSURLProtocol {
 838 }
 839
 840 @end
 841
 842 @implementation CydgetURLProtocol
 843
 844 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
 845     NSURL *url([request URL]);
 846     if (url == nil)
 847         return NO;
 848     NSString *scheme([[url scheme] lowercaseString]);
 849     if (scheme == nil || ![scheme isEqualToString:@"cydget"])
 850         return NO;
 851     return YES;
 852 }
 853
 854 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
 855     return request;
 856 }
 857
 858 - (void) _returnPNGWithImage:(UIImage *)icon forRequest:(NSURLRequest *)request {
 859     id<NSURLProtocolClient> client([self client]);
 860     if (icon == nil)
 861         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
 862     else {
 863         NSData *data(UIImagePNGRepresentation(icon));
 864
 865         NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:@"image/png" expectedContentLength:-1 textEncodingName:nil] autorelease]);
 866         [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
 867         [client URLProtocol:self didLoadData:data];
 868         [client URLProtocolDidFinishLoading:self];
 869     }
 870 }
 871
 872 - (void) startLoading {
 873     id<NSURLProtocolClient> client([self client]);
 874     NSURLRequest *request([self request]);
 875
 876     NSURL *url([request URL]);
 877     NSString *href([url absoluteString]);
 878
 879     NSString *path([href substringFromIndex:9]);
 880     NSRange slash([path rangeOfString:@"/"]);
 881
 882     NSString *command;
 883     if (slash.location == NSNotFound) {
 884         command = path;
 885         path = nil;
 886     } else {
 887         command = [path substringToIndex:slash.location];
 888         path = [path substringFromIndex:(slash.location + 1)];
 889     }
 890
 891     if ([command isEqualToString:@"_UIImageWithName"]) {
 892         if (path == nil)
 893             goto fail;
 894         path = [path stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
 895         UIImage *icon(_UIImageWithName(path));
 896         [self _returnPNGWithImage:icon forRequest:request];
 897     } else fail: {
 898         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 899     }
 900 }
 901
 902 - (void) stopLoading {
 903 }
 904
 905 @end
 906 /* }}} */
 907 /* Cydget-CGI:// Protocol {{{ */
 908 @interface CydgetCGIURLProtocol : NSURLProtocol {
 909     pid_t pid_;
 910     CFHTTPMessageRef http_;
 911     NSFileHandle *handle_;
 912 }
 913
 914 @end
 915
 916 @implementation CydgetCGIURLProtocol
 917
 918 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
 919     NSURL *url([request URL]);
 920     if (url == nil)
 921         return NO;
 922     NSString *scheme([[url scheme] lowercaseString]);
 923     if (scheme == nil || ![scheme isEqualToString:@"cydget-cgi"])
 924         return NO;
 925     return YES;
 926 }
 927
 928 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
 929     return request;
 930 }
 931
 932 - (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)response client:(id<NSURLProtocolClient>)client {
 933     if ((self = [super initWithRequest:request cachedResponse:response client:client]) != nil) {
 934         pid_ = -1;
 935     } return self;
 936 }
 937
 938 - (void) startLoading {
 939     id<NSURLProtocolClient> client([self client]);
 940     NSURLRequest *request([self request]);
 941     NSURL *url([request URL]);
 942
 943     NSString *path([url path]);
 944     if (path == nil) {
 945         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 946         return;
 947     }
 948
 949     NSFileManager *manager([NSFileManager defaultManager]);
 950     if (![manager fileExistsAtPath:path]) {
 951         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
 952         return;
 953     }
 954
 955     int fds[2];
 956     _assert(pipe(fds) != -1);
 957
 958     _assert(pid_ == -1);
 959     pid_ = fork();
 960     if (pid_ == -1) {
 961         _assert(close(fds[0]) != -1);
 962         _assert(close(fds[1]) != -1);
 963         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 964         return;
 965     }
 966
 967     if (pid_ == 0) {
 968         const char *script([path UTF8String]);
 969
 970         setenv("GATEWAY_INTERFACE", "CGI/1.1", true);
 971         setenv("SCRIPT_FILENAME", script, true);
 972         NSString *query([url query]);
 973         if (query != nil)
 974             setenv("QUERY_STRING", [query UTF8String], true);
 975
 976         _assert(dup2(fds[1], 1) != -1);
 977         _assert(close(fds[0]) != -1);
 978         _assert(close(fds[1]) != -1);
 979
 980         execl(script, script, NULL);
 981         exit(1);
 982         _assert(false);
 983     }
 984
 985     _assert(close(fds[1]) != -1);
 986
 987     _assert(http_ == NULL);
 988     http_ = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, FALSE);
 989     CFHTTPMessageAppendBytes(http_, (const uint8_t *) "HTTP/1.1 200 OK\r\n", 17);
 990
 991     _assert(handle_ == nil);
 992     handle_ = [[NSFileHandle alloc] initWithFileDescriptor:fds[0] closeOnDealloc:YES];
 993
 994     [[NSNotificationCenter defaultCenter]
 995         addObserver:self
 996         selector:@selector(onRead:)
 997         name:@"NSFileHandleReadCompletionNotification"
 998         object:handle_
 999     ];
1000
1001     [handle_ readInBackgroundAndNotify];
1002 }
1003
1004 - (void) onRead:(NSNotification *)notification {
1005     NSFileHandle *handle([notification object]);
1006
1007     NSData *data([[notification userInfo] objectForKey:NSFileHandleNotificationDataItem]);
1008
1009     if (size_t length = [data length]) {
1010         CFHTTPMessageAppendBytes(http_, reinterpret_cast<const UInt8 *>([data bytes]), length);
1011         [handle readInBackgroundAndNotify];
1012     } else {
1013         id<NSURLProtocolClient> client([self client]);
1014
1015         CFStringRef mime(CFHTTPMessageCopyHeaderFieldValue(http_, CFSTR("Content-type")));
1016         if (mime == NULL)
1017             [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorBadServerResponse userInfo:nil]];
1018         else {
1019             NSURLRequest *request([self request]);
1020
1021             NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:(NSString *)mime expectedContentLength:-1 textEncodingName:nil] autorelease]);
1022             CFRelease(mime);
1023
1024             [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1025
1026             CFDataRef body(CFHTTPMessageCopyBody(http_));
1027             [client URLProtocol:self didLoadData:(NSData *)body];
1028             CFRelease(body);
1029
1030             [client URLProtocolDidFinishLoading:self];
1031         }
1032
1033         CFRelease(http_);
1034         http_ = NULL;
1035     }
1036 }
1037
1038     //[client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorNetworkConnectionLost userInfo:nil]];
1039
1040 - (void) stopLoading_ {
1041     [[NSNotificationCenter defaultCenter] removeObserver:self];
1042
1043     if (handle_ != nil) {
1044         [handle_ release];
1045         handle_ = nil;
1046     }
1047
1048     if (pid_ != -1) {
1049         kill(pid_, SIGTERM);
1050         int status;
1051         _syscall(waitpid(pid_, &status, 0));
1052         pid_ = -1;
1053     }
1054 }
1055
1056 - (void) stopLoading {
1057     [self
1058         performSelectorOnMainThread:@selector(stopLoading_)
1059         withObject:nil
1060         waitUntilDone:NO
1061     ];
1062 }
1063
1064 @end
1065 /* }}} */
1066
1067 namespace WebCore {
1068     class MediaQueryEvaluator;
1069     class CSSParserValueList;
1070 }
1071
1072 namespace WebCore {
1073 struct MediaQueryExp {
1074     String feature_;
1075     void *value_;
1076     bool valid_;
1077     String cache_;
1078 }; }
1079
1080 bool CYHaveMediaControls() {
1081     SBLockScreenView *view([[[$SBLockScreenManager sharedInstance] lockScreenViewController] lockScreenView]);
1082     return view != nil && ![view mediaControlsHidden];
1083     //return [[[$SBLockScreenManager sharedInstance] lockScreenViewController] isShowingMediaControls];
1084 }
1085
1086 bool CYHaveNotificationList() {
1087     SBLockScreenNotificationListController *controller([[[$SBLockScreenManager sharedInstance] lockScreenViewController] _notificationController]);
1088     return controller != nil && [controller hasAnyContent];
1089 }
1090
1091 MSHook(bool, _ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, WebCore::MediaQueryEvaluator *_this, WebCore::String &query) {
1092     Log(query); if (false);
1093     else if (kCFCoreFoundationVersionNumber >= 800 && StringEquals(query, "-cydget-media-controls"))
1094         return CYHaveMediaControls();
1095     else if (kCFCoreFoundationVersionNumber >= 800 && StringEquals(query, "-cydget-notification-list"))
1096         return CYHaveNotificationList();
1097     else
1098         return __ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE(_this, query);
1099 }
1100
1101 MSHook(void *, _ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, WebCore::MediaQueryExp *_this, WebCore::String &query, WebCore::CSSParserValueList *values) {
1102     Log(query);
1103     void *value(__ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE(_this, query, values));
1104     if (!_this->valid_) if (
1105         StringEquals(query, "-cydget-media-controls") ||
1106         StringEquals(query, "-cydget-notification-list") ||
1107     false) _this->valid_ = true;
1108     return value;
1109 }
1110
1111 template <typename Type_>
1112 static void dlset(Type_ &function, const char *name) {
1113     function = reinterpret_cast<Type_>(dlsym(RTLD_DEFAULT, name));
1114 }
1115
1116 #define msset(function, handle) \
1117     MSHookSymbol(function, "_" #function, handle)
1118
1119 @implementation WebCycriptLockScreenController
1120
1121 static void $UIWebViewWebViewDelegate$webView$addMessageToConsole$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *message) {
1122     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
1123     if ([uiWebView respondsToSelector:@selector(webView:addMessageToConsole:)])
1124         [uiWebView webView:view addMessageToConsole:message];
1125 }
1126
1127 static void $UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebScriptObject *window, WebFrame *frame) {
1128     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
1129     if ([uiWebView respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
1130         [uiWebView webView:view didClearWindowObject:window forFrame:frame];
1131 }
1132
1133 + (void) initialize {
1134     if (Class $UIWebViewWebViewDelegate = objc_getClass("UIWebViewWebViewDelegate")) {
1135         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:addMessageToConsole:), (IMP) &$UIWebViewWebViewDelegate$webView$addMessageToConsole$, "v16@0:4@8@12");
1136         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didClearWindowObject:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$, "v20@0:4@8@12@16");
1137     }
1138
1139     if (CGFloat *_UIScrollViewDecelerationRateNormal = reinterpret_cast<CGFloat *>(dlsym(RTLD_DEFAULT, "UIScrollViewDecelerationRateNormal")))
1140         CYScrollViewDecelerationRateNormal = *_UIScrollViewDecelerationRateNormal;
1141     else // XXX: this actually might be fast on some older systems: we should look into this
1142         CYScrollViewDecelerationRateNormal = 0.998;
1143
1144     iOS4 = kCFCoreFoundationVersionNumber >= 550.32;
1145     iOS32 = !iOS4 && kCFCoreFoundationVersionNumber >= 478.61;
1146
1147     int maxproc;
1148     size_t size(sizeof(maxproc));
1149     if (sysctlbyname("kern.maxproc", &maxproc, &size, NULL, 0) == -1)
1150         NSLog(@"sysctlbyname(\"kern.maxproc\", ?)");
1151     else if (maxproc < 72) {
1152         maxproc = 72;
1153         if (sysctlbyname("kern.maxproc", NULL, NULL, &maxproc, sizeof(maxproc)) == -1)
1154             NSLog(@"sysctlbyname(\"kern.maxproc\", #)");
1155     }
1156
1157     [NSURLProtocol registerClass:[CydgetURLProtocol class]];
1158     [WebView registerURLSchemeAsLocal:@"cydget"];
1159
1160     [NSURLProtocol registerClass:[CydgetCGIURLProtocol class]];
1161     [WebView registerURLSchemeAsLocal:@"cydget-cgi"];
1162
1163     MSImageRef JavaScriptCore(NULL);
1164     if (JavaScriptCore == NULL)
1165         JavaScriptCore = MSGetImageByName("/System/Library/Frameworks/JavaScriptCore.framework/JavaScriptCore");
1166     if (JavaScriptCore == NULL)
1167         JavaScriptCore = MSGetImageByName("/System/Library/PrivateFrameworks/JavaScriptCore.framework/JavaScriptCore");
1168
1169     MSImageRef WebCore(MSGetImageByName("/System/Library/PrivateFrameworks/WebCore.framework/WebCore"));
1170
1171     if (!iOS4) {
1172         void (*_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE)(JSC::SourceCode **, JSC::JSGlobalData *, int *, JSC::UString *);
1173         dlset(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, "_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE");
1174         if (_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE != NULL)
1175             MSHookFunction(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, MSHake(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE));
1176     }
1177
1178     bool (*_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE)(const WebCore::String &) = NULL;
1179     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE == NULL)
1180         MSHookSymbol(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE", WebCore);
1181     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE == NULL)
1182         MSHookSymbol(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKN3WTF6StringE", WebCore);
1183     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE != NULL)
1184         MSHookFunction(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, MSHake(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE));
1185
1186     void (*_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi)(void *, const WebCore::String &, const WebCore::KURL &, int) = NULL;
1187     if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi == NULL)
1188         MSHookSymbol(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, "__ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi", WebCore);
1189     if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi != NULL)
1190         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi));
1191
1192     if (!iOS4) {
1193         const WebCore::String &(*_ZNK7WebCore4Node11textContentEb)(void *, bool) = NULL;
1194         if (_ZNK7WebCore4Node11textContentEb == NULL)
1195             MSHookSymbol(_ZNK7WebCore4Node11textContentEb, "__ZNK7WebCore4Node11textContentEb", WebCore);
1196         if (_ZNK7WebCore4Node11textContentEb != NULL)
1197             MSHookFunction(_ZNK7WebCore4Node11textContentEb, MSHake(_ZNK7WebCore4Node11textContentEb));
1198     }
1199
1200     const WebCore::String &(*_ZN7WebCore12CachedScript6scriptEv)(void *) = NULL;
1201     if (_ZN7WebCore12CachedScript6scriptEv == NULL)
1202         MSHookSymbol(_ZN7WebCore12CachedScript6scriptEv, "__ZN7WebCore12CachedScript6scriptEv", WebCore);
1203     if (_ZN7WebCore12CachedScript6scriptEv != NULL)
1204         MSHookFunction(_ZN7WebCore12CachedScript6scriptEv, MSHake(_ZN7WebCore12CachedScript6scriptEv));
1205
1206     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i)(void *, const WebCore::String &, State, const WebCore::String &, int) = NULL;
1207     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i == NULL)
1208         MSHookSymbol(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i", WebCore);
1209     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i != NULL)
1210         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i));
1211
1212     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE)(void *, JSC::ScriptSourceCode &, State) = NULL;
1213     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE == NULL)
1214         MSHookSymbol(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE", WebCore);
1215     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE != NULL)
1216         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE));
1217
1218     void (*_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE)(void *, const WebCore::String &, const WebCore::KURL &, void *);
1219     msset(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, WebCore);
1220     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE != NULL)
1221         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE));
1222
1223     void (*_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE)(void *, const WebCore::String &, const WebCore::KURL &, void *);
1224     msset(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, WebCore);
1225     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE == NULL)
1226         MSHookSymbol(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, "__ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_3URLERKNS1_12TextPositionE", WebCore);
1227     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE != NULL)
1228         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE));
1229
1230     void (*_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE)(void *, void *, int);
1231     msset(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, WebCore);
1232     if (_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE != NULL)
1233         MSHookFunction(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, MSHake(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE));
1234
1235     void (*_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE)(void *, void *, int);
1236     msset(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, WebCore);
1237     if (_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE != NULL)
1238         MSHookFunction(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, MSHake(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE));
1239
1240     MSHookSymbol($_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE, "__ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE", WebCore);
1241
1242     void (*_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE)(void *, JSC::ScriptSourceCode &);
1243     msset(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, WebCore);
1244     if (_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE != NULL)
1245         MSHookFunction(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, MSHake(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE));
1246
1247     if (_ZN7WebCore6String6appendEPKtj == NULL)
1248         MSHookSymbol(_ZN7WebCore6String6appendEPKtj, "__ZN7WebCore6String6appendEPKtj", WebCore);
1249     if (_ZN7WebCore6String6appendEPKtj == NULL)
1250         msset(_ZN7WebCore6String6appendEPKtj, JavaScriptCore);
1251     if (_ZN7WebCore6String6appendEPKtj == NULL)
1252         MSHookSymbol(_ZN7WebCore6String6appendEPKtj, "__ZN3WTF6String6appendEPKtj", JavaScriptCore);
1253
1254     if (_ZN7WebCore6String8truncateEj == NULL)
1255         MSHookSymbol(_ZN7WebCore6String8truncateEj, "__ZN7WebCore6String8truncateEj", WebCore);
1256     if (_ZN7WebCore6String8truncateEj == NULL)
1257         msset(_ZN7WebCore6String8truncateEj, JavaScriptCore);
1258     if (_ZN7WebCore6String8truncateEj == NULL)
1259         MSHookSymbol(_ZN7WebCore6String8truncateEj, "__ZN3WTF6String8truncateEj", JavaScriptCore);
1260
1261     msset(_ZNK7WebCore6String10charactersEv, WebCore);
1262
1263     msset(_ZN7WebCore6String29charactersWithNullTerminationEv, JavaScriptCore);
1264     if (_ZN7WebCore6String29charactersWithNullTerminationEv == NULL)
1265         MSHookSymbol(_ZN7WebCore6String29charactersWithNullTerminationEv, "__ZN3WTF6String29charactersWithNullTerminationEv", JavaScriptCore);
1266
1267     msset(_ZNK3WTF6String14createCFStringEv, JavaScriptCore);
1268
1269     msset(_ZNK7WebCore6String6lengthEv, WebCore);
1270
1271     bool (*_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE)(WebCore::MediaQueryEvaluator *, WebCore::String &);
1272     msset(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, WebCore);
1273     if (_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE != NULL)
1274         MSHookFunction(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, MSHake(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE));
1275
1276     void *(*_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE)(WebCore::MediaQueryExp *, WebCore::String &, WebCore::CSSParserValueList *);
1277     msset(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, WebCore);
1278     if (_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE != NULL)
1279         MSHookFunction(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, MSHake(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE));
1280 }
1281
1282 + (id) rootViewController {
1283     return [[[self alloc] init] autorelease];
1284 }
1285
1286 - (void) dealloc {
1287     [configuration_ release];
1288     [background_ release];
1289     [foreground_ release];
1290     [super dealloc];
1291 }
1292
1293 - (id) init {
1294     if ((self = [super init]) != nil) {
1295         configuration_ = [[$CydgetController currentConfiguration] retain];
1296         legacy_ = [configuration_ objectForKey:@"Background"] == nil;
1297         media_ = [[configuration_ objectForKey:@"MediaControls"] boolValue];
1298         items_ = [[configuration_ objectForKey:@"NotificationList"] boolValue];
1299     } return self;
1300 }
1301
1302 - (void) loadView {
1303     NSURL *base([NSURL fileURLWithPath:[$CydgetController currentPath]]);
1304
1305     if (NSString *background = [configuration_ objectForKey:@"Background"])
1306         background_ = [[WebCydgetLockScreenView alloc] initWithURL:[NSURL URLWithString:background relativeToURL:base]];
1307
1308     if (NSString *homepage = [configuration_ objectForKey:@"Homepage"]) {
1309         foreground_ = [[WebCydgetLockScreenView alloc] initWithURL:[NSURL URLWithString:homepage relativeToURL:base]];
1310         [self setView:foreground_];
1311     }
1312 }
1313
1314 - (void) purgeView {
1315     [background_ removeFromSuperview];
1316     [background_ release];
1317     background_ = nil;
1318     [foreground_ removeFromSuperview];
1319     [foreground_ release];
1320     foreground_ = nil;
1321     [super purgeView];
1322 }
1323
1324 - (void) updateStyles {
1325     [foreground_ updateStyles];
1326     [background_ updateStyles];
1327 }
1328
1329 - (UIView *) backgroundView {
1330     return background_;
1331 }
1332
1333 - (BOOL) showAwayItems {
1334     return YES;
1335 }
1336
1337 - (BOOL) updateHidden {
1338     if (foreground_ == nil)
1339         return true;
1340     bool media(CYHaveMediaControls());
1341     [foreground_ setHidden:media];
1342     return media;
1343 }
1344
1345 - (BOOL) showDateView {
1346     if (kCFCoreFoundationVersionNumber < 800)
1347         return false;
1348     [self updateStyles];
1349     if (!legacy_ && foreground_ == nil)
1350         return true;
1351     if (!items_ && CYHaveNotificationList())
1352         return true;
1353     return false;
1354 }
1355
1356 - (BOOL) allowsLockScreenMediaControls {
1357     if (kCFCoreFoundationVersionNumber < 800)
1358         return true;
1359     return media_;
1360 }
1361
1362 /*- (BOOL) showHeaderView {
1363     return YES;
1364 }*/
1365
1366 // 0: view is rendered above head
1367 // 1: view moves as one with head
1368 // 2: view moves up and down only
1369 // 3: view simply never does move
1370 - (NSUInteger) presentationStyle {
1371     return 1;
1372 }
1373
1374 // 1: light blur
1375 // 2: heavy blur
1376 // 3: just black
1377 // 4: legibility
1378 // 5: no overlay
1379 - (NSUInteger) overlayStyle {
1380     return legacy_ ? 1 : 4;
1381 }
1382
1383 // 1: blur -> view -> list
1384 // 2: view -> blur -> list
1385 // 3: view. unblur below?!
1386 // 4: blur -> list -> view
1387 - (NSUInteger) notificationBehavior {
1388     return items_ ? 1 : 2;
1389 }
1390
1391 - (BOOL) viewWantsFullscreenLayout {
1392     return kCFCoreFoundationVersionNumber >= 800;
1393 }
1394
1395 /*- (BOOL) viewWantsOverlayLayout {
1396     return kCFCoreFoundationVersionNumber >= 800;
1397 }*/
1398
1399 - (BOOL) shouldDisableOnUnlock {
1400     return YES;
1401 }
1402
1403 - (BOOL) canBeAlwaysFullscreen {
1404     return YES;
1405 }
1406
1407 /*- (BOOL) wantsSwipeGestureRecognizer {
1408     return YES;
1409 }
1410
1411 - (BOOL) handleGesture:(int)arg1 fingerCount:(NSUInteger)fingers {
1412     return NO;
1413     return YES;
1414 }*/
1415
1416 // - (void) lockScreenMediaControlsShown:(BOOL)shown;
1417
1418 - (BOOL) handleMenuButtonDoubleTap {
1419     if (kCFCoreFoundationVersionNumber >= 800) {
1420         SBLockScreenViewController *controller([[$SBLockScreenManager sharedInstance] lockScreenViewController]);
1421         [controller _setMediaControlsVisible:![controller isShowingMediaControls]];
1422     }
1423
1424     return [super handleMenuButtonDoubleTap];
1425 }
1426
1427 @end
1428
1429 MSClassHook(WebView)
1430 MSMetaClassHook(WebView)
1431
1432 MSClassMessageHook0(void, WebView, enableWebThread) {
1433     if (kCFCoreFoundationVersionNumber >= 478.61)
1434         return MSOldCall();
1435
1436     NSLog(@"-[WebView enableWebThread]");
1437 }