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     Log(code);
 771     Cycriptify(code);
 772     return code;
 773 }
 774
 775 // iOS 4.x cdata
 776 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, void *_this, const WebCore::String &source, const WebCore::KURL &url, int line) {
 777     Cycriptify(source);
 778     return __ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi(_this, source, url, line);
 779 }
 780
 781 // iOS 4.x+5.0 @src=
 782 MSHook(const WebCore::String &, _ZN7WebCore12CachedScript6scriptEv, void *_this) {
 783     const WebCore::String &script(__ZN7WebCore12CachedScript6scriptEv(_this));
 784     string = const_cast<WebCore::String *>(&script);
 785     Log(script);
 786     return script;
 787 }
 788
 789 // iOS 4.x @src=
 790 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, void *_this, JSC::ScriptSourceCode &script, State state) {
 791     if (string != NULL) {
 792         JSC::SourceCode *source(iOS4 ? &script.New.source_ : &script.Old.source_);
 793         Cycriptify(*string, &source->end_);
 794         string = NULL;
 795     }
 796
 797     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE(_this, script, state);
 798 }
 799
 800 // iOS 5.0 cdata
 801 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, void *_this, const WebCore::String &source, const WebCore::KURL &url, void *position) {
 802     Cycriptify(source);
 803     return __ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE(_this, source, url, position);
 804 }
 805
 806 // iOS 5.0 @src=
 807 MSHook(void, _ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, void *_this, void *position, int legacy) {
 808     string = NULL;
 809     return __ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE(_this, position, legacy);
 810 }
 811
 812 void (*$_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE)(void *_this, int legacy);
 813
 814 // iOS 5.0 @src=
 815 MSHook(void, _ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, void *_this, JSC::ScriptSourceCode &script) {
 816     if (string != NULL) {
 817         JSC::SourceCode *source(&script.New.source_);
 818         $_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE(_this, 0);
 819         Cycriptify(*string, &source->end_);
 820         string = NULL;
 821     }
 822
 823     return __ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE(_this, script);
 824 }
 825
 826 // iOS 6.0 cdata
 827 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, void *_this, const WebCore::String &source, const WebCore::KURL &url, void *position) {
 828     Cycriptify(source);
 829     return __ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE(_this, source, url, position);
 830 }
 831
 832 // iOS 6.0 @src=
 833 MSHook(void, _ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, void *_this, void *position, int legacy) {
 834     string = NULL;
 835     return __ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE(_this, position, legacy);
 836 }
 837
 838 /* Cydget:// Protocol {{{ */
 839 @interface CydgetURLProtocol : NSURLProtocol {
 840 }
 841
 842 @end
 843
 844 @implementation CydgetURLProtocol
 845
 846 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
 847     NSURL *url([request URL]);
 848     if (url == nil)
 849         return NO;
 850     NSString *scheme([[url scheme] lowercaseString]);
 851     if (scheme == nil || ![scheme isEqualToString:@"cydget"])
 852         return NO;
 853     return YES;
 854 }
 855
 856 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
 857     return request;
 858 }
 859
 860 - (void) _returnPNGWithImage:(UIImage *)icon forRequest:(NSURLRequest *)request {
 861     id<NSURLProtocolClient> client([self client]);
 862     if (icon == nil)
 863         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
 864     else {
 865         NSData *data(UIImagePNGRepresentation(icon));
 866
 867         NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:@"image/png" expectedContentLength:-1 textEncodingName:nil] autorelease]);
 868         [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
 869         [client URLProtocol:self didLoadData:data];
 870         [client URLProtocolDidFinishLoading:self];
 871     }
 872 }
 873
 874 - (void) startLoading {
 875     id<NSURLProtocolClient> client([self client]);
 876     NSURLRequest *request([self request]);
 877
 878     NSURL *url([request URL]);
 879     NSString *href([url absoluteString]);
 880
 881     NSString *path([href substringFromIndex:9]);
 882     NSRange slash([path rangeOfString:@"/"]);
 883
 884     NSString *command;
 885     if (slash.location == NSNotFound) {
 886         command = path;
 887         path = nil;
 888     } else {
 889         command = [path substringToIndex:slash.location];
 890         path = [path substringFromIndex:(slash.location + 1)];
 891     }
 892
 893     if ([command isEqualToString:@"_UIImageWithName"]) {
 894         if (path == nil)
 895             goto fail;
 896         path = [path stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
 897         UIImage *icon(_UIImageWithName(path));
 898         [self _returnPNGWithImage:icon forRequest:request];
 899     } else fail: {
 900         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 901     }
 902 }
 903
 904 - (void) stopLoading {
 905 }
 906
 907 @end
 908 /* }}} */
 909 /* Cydget-CGI:// Protocol {{{ */
 910 @interface CydgetCGIURLProtocol : NSURLProtocol {
 911     pid_t pid_;
 912     CFHTTPMessageRef http_;
 913     NSFileHandle *handle_;
 914 }
 915
 916 @end
 917
 918 @implementation CydgetCGIURLProtocol
 919
 920 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
 921     NSURL *url([request URL]);
 922     if (url == nil)
 923         return NO;
 924     NSString *scheme([[url scheme] lowercaseString]);
 925     if (scheme == nil || ![scheme isEqualToString:@"cydget-cgi"])
 926         return NO;
 927     return YES;
 928 }
 929
 930 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
 931     return request;
 932 }
 933
 934 - (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)response client:(id<NSURLProtocolClient>)client {
 935     if ((self = [super initWithRequest:request cachedResponse:response client:client]) != nil) {
 936         pid_ = -1;
 937     } return self;
 938 }
 939
 940 - (void) startLoading {
 941     id<NSURLProtocolClient> client([self client]);
 942     NSURLRequest *request([self request]);
 943     NSURL *url([request URL]);
 944
 945     NSString *path([url path]);
 946     if (path == nil) {
 947         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 948         return;
 949     }
 950
 951     NSFileManager *manager([NSFileManager defaultManager]);
 952     if (![manager fileExistsAtPath:path]) {
 953         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
 954         return;
 955     }
 956
 957     int fds[2];
 958     _assert(pipe(fds) != -1);
 959
 960     _assert(pid_ == -1);
 961     pid_ = fork();
 962     if (pid_ == -1) {
 963         _assert(close(fds[0]) != -1);
 964         _assert(close(fds[1]) != -1);
 965         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
 966         return;
 967     }
 968
 969     if (pid_ == 0) {
 970         const char *script([path UTF8String]);
 971
 972         setenv("GATEWAY_INTERFACE", "CGI/1.1", true);
 973         setenv("SCRIPT_FILENAME", script, true);
 974         NSString *query([url query]);
 975         if (query != nil)
 976             setenv("QUERY_STRING", [query UTF8String], true);
 977
 978         _assert(dup2(fds[1], 1) != -1);
 979         _assert(close(fds[0]) != -1);
 980         _assert(close(fds[1]) != -1);
 981
 982         execl(script, script, NULL);
 983         exit(1);
 984         _assert(false);
 985     }
 986
 987     _assert(close(fds[1]) != -1);
 988
 989     _assert(http_ == NULL);
 990     http_ = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, FALSE);
 991     CFHTTPMessageAppendBytes(http_, (const uint8_t *) "HTTP/1.1 200 OK\r\n", 17);
 992
 993     _assert(handle_ == nil);
 994     handle_ = [[NSFileHandle alloc] initWithFileDescriptor:fds[0] closeOnDealloc:YES];
 995
 996     [[NSNotificationCenter defaultCenter]
 997         addObserver:self
 998         selector:@selector(onRead:)
 999         name:@"NSFileHandleReadCompletionNotification"
1000         object:handle_
1001     ];
1002
1003     [handle_ readInBackgroundAndNotify];
1004 }
1005
1006 - (void) onRead:(NSNotification *)notification {
1007     NSFileHandle *handle([notification object]);
1008
1009     NSData *data([[notification userInfo] objectForKey:NSFileHandleNotificationDataItem]);
1010
1011     if (size_t length = [data length]) {
1012         CFHTTPMessageAppendBytes(http_, reinterpret_cast<const UInt8 *>([data bytes]), length);
1013         [handle readInBackgroundAndNotify];
1014     } else {
1015         id<NSURLProtocolClient> client([self client]);
1016
1017         CFStringRef mime(CFHTTPMessageCopyHeaderFieldValue(http_, CFSTR("Content-type")));
1018         if (mime == NULL)
1019             [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorBadServerResponse userInfo:nil]];
1020         else {
1021             NSURLRequest *request([self request]);
1022
1023             NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:(NSString *)mime expectedContentLength:-1 textEncodingName:nil] autorelease]);
1024             CFRelease(mime);
1025
1026             [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1027
1028             CFDataRef body(CFHTTPMessageCopyBody(http_));
1029             [client URLProtocol:self didLoadData:(NSData *)body];
1030             CFRelease(body);
1031
1032             [client URLProtocolDidFinishLoading:self];
1033         }
1034
1035         CFRelease(http_);
1036         http_ = NULL;
1037     }
1038 }
1039
1040     //[client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorNetworkConnectionLost userInfo:nil]];
1041
1042 - (void) stopLoading_ {
1043     [[NSNotificationCenter defaultCenter] removeObserver:self];
1044
1045     if (handle_ != nil) {
1046         [handle_ release];
1047         handle_ = nil;
1048     }
1049
1050     if (pid_ != -1) {
1051         kill(pid_, SIGTERM);
1052         int status;
1053         _syscall(waitpid(pid_, &status, 0));
1054         pid_ = -1;
1055     }
1056 }
1057
1058 - (void) stopLoading {
1059     [self
1060         performSelectorOnMainThread:@selector(stopLoading_)
1061         withObject:nil
1062         waitUntilDone:NO
1063     ];
1064 }
1065
1066 @end
1067 /* }}} */
1068
1069 namespace WebCore {
1070     class MediaQueryEvaluator;
1071     class CSSParserValueList;
1072 }
1073
1074 namespace WebCore {
1075 struct MediaQueryExp {
1076     String feature_;
1077     void *value_;
1078     bool valid_;
1079     String cache_;
1080 }; }
1081
1082 bool CYHaveMediaControls() {
1083     SBLockScreenView *view([[[$SBLockScreenManager sharedInstance] lockScreenViewController] lockScreenView]);
1084     return view != nil && ![view mediaControlsHidden];
1085     //return [[[$SBLockScreenManager sharedInstance] lockScreenViewController] isShowingMediaControls];
1086 }
1087
1088 bool CYHaveNotificationList() {
1089     SBLockScreenNotificationListController *controller([[[$SBLockScreenManager sharedInstance] lockScreenViewController] _notificationController]);
1090     return controller != nil && [controller hasAnyContent];
1091 }
1092
1093 MSHook(bool, _ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, WebCore::MediaQueryEvaluator *_this, WebCore::String &query) {
1094     Log(query); if (false);
1095     else if (kCFCoreFoundationVersionNumber >= 800 && StringEquals(query, "-cydget-media-controls"))
1096         return CYHaveMediaControls();
1097     else if (kCFCoreFoundationVersionNumber >= 800 && StringEquals(query, "-cydget-notification-list"))
1098         return CYHaveNotificationList();
1099     else
1100         return __ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE(_this, query);
1101 }
1102
1103 MSHook(void, _ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, WebCore::MediaQueryExp *_this, WebCore::String &query, WebCore::CSSParserValueList *values) {
1104     Log(query);
1105     __ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE(_this, query, values);
1106     if (!_this->valid_) if (
1107         StringEquals(query, "-cydget-media-controls") ||
1108         StringEquals(query, "-cydget-notification-list") ||
1109     false) _this->valid_ = true;
1110 }
1111
1112 template <typename Type_>
1113 static void dlset(Type_ &function, const char *name) {
1114     function = reinterpret_cast<Type_>(dlsym(RTLD_DEFAULT, name));
1115 }
1116
1117 #define msset(function, handle) \
1118     MSHookSymbol(function, "_" #function, handle)
1119
1120 @implementation WebCycriptLockScreenController
1121
1122 static void $UIWebViewWebViewDelegate$webView$addMessageToConsole$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *message) {
1123     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
1124     if ([uiWebView respondsToSelector:@selector(webView:addMessageToConsole:)])
1125         [uiWebView webView:view addMessageToConsole:message];
1126 }
1127
1128 static void $UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebScriptObject *window, WebFrame *frame) {
1129     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
1130     if ([uiWebView respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
1131         [uiWebView webView:view didClearWindowObject:window forFrame:frame];
1132 }
1133
1134 + (void) initialize {
1135     if (Class $UIWebViewWebViewDelegate = objc_getClass("UIWebViewWebViewDelegate")) {
1136         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:addMessageToConsole:), (IMP) &$UIWebViewWebViewDelegate$webView$addMessageToConsole$, "v16@0:4@8@12");
1137         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didClearWindowObject:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$, "v20@0:4@8@12@16");
1138     }
1139
1140     if (CGFloat *_UIScrollViewDecelerationRateNormal = reinterpret_cast<CGFloat *>(dlsym(RTLD_DEFAULT, "UIScrollViewDecelerationRateNormal")))
1141         CYScrollViewDecelerationRateNormal = *_UIScrollViewDecelerationRateNormal;
1142     else // XXX: this actually might be fast on some older systems: we should look into this
1143         CYScrollViewDecelerationRateNormal = 0.998;
1144
1145     iOS4 = kCFCoreFoundationVersionNumber >= 550.32;
1146     iOS32 = !iOS4 && kCFCoreFoundationVersionNumber >= 478.61;
1147
1148     int maxproc;
1149     size_t size(sizeof(maxproc));
1150     if (sysctlbyname("kern.maxproc", &maxproc, &size, NULL, 0) == -1)
1151         NSLog(@"sysctlbyname(\"kern.maxproc\", ?)");
1152     else if (maxproc < 72) {
1153         maxproc = 72;
1154         if (sysctlbyname("kern.maxproc", NULL, NULL, &maxproc, sizeof(maxproc)) == -1)
1155             NSLog(@"sysctlbyname(\"kern.maxproc\", #)");
1156     }
1157
1158     [NSURLProtocol registerClass:[CydgetURLProtocol class]];
1159     [WebView registerURLSchemeAsLocal:@"cydget"];
1160
1161     [NSURLProtocol registerClass:[CydgetCGIURLProtocol class]];
1162     [WebView registerURLSchemeAsLocal:@"cydget-cgi"];
1163
1164     MSImageRef JavaScriptCore(NULL);
1165     if (JavaScriptCore == NULL)
1166         JavaScriptCore = MSGetImageByName("/System/Library/Frameworks/JavaScriptCore.framework/JavaScriptCore");
1167     if (JavaScriptCore == NULL)
1168         JavaScriptCore = MSGetImageByName("/System/Library/PrivateFrameworks/JavaScriptCore.framework/JavaScriptCore");
1169
1170     MSImageRef WebCore(MSGetImageByName("/System/Library/PrivateFrameworks/WebCore.framework/WebCore"));
1171
1172     if (!iOS4) {
1173         void (*_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE)(JSC::SourceCode **, JSC::JSGlobalData *, int *, JSC::UString *);
1174         dlset(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, "_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE");
1175         if (_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE != NULL)
1176             MSHookFunction(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, MSHake(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE));
1177     }
1178
1179     bool (*_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE)(const WebCore::String &) = NULL;
1180     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE == NULL)
1181         MSHookSymbol(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE", WebCore);
1182     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE == NULL)
1183         MSHookSymbol(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKN3WTF6StringE", WebCore);
1184     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE != NULL)
1185         MSHookFunction(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, MSHake(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE));
1186
1187     void (*_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi)(void *, const WebCore::String &, const WebCore::KURL &, int) = NULL;
1188     if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi == NULL)
1189         MSHookSymbol(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, "__ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi", WebCore);
1190     if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi != NULL)
1191         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi));
1192
1193     if (!iOS4) {
1194         const WebCore::String &(*_ZNK7WebCore4Node11textContentEb)(void *, bool) = NULL;
1195         if (_ZNK7WebCore4Node11textContentEb == NULL)
1196             MSHookSymbol(_ZNK7WebCore4Node11textContentEb, "__ZNK7WebCore4Node11textContentEb", WebCore);
1197         if (_ZNK7WebCore4Node11textContentEb != NULL)
1198             MSHookFunction(_ZNK7WebCore4Node11textContentEb, MSHake(_ZNK7WebCore4Node11textContentEb));
1199     }
1200
1201     const WebCore::String &(*_ZN7WebCore12CachedScript6scriptEv)(void *) = NULL;
1202     if (_ZN7WebCore12CachedScript6scriptEv == NULL)
1203         MSHookSymbol(_ZN7WebCore12CachedScript6scriptEv, "__ZN7WebCore12CachedScript6scriptEv", WebCore);
1204     if (_ZN7WebCore12CachedScript6scriptEv != NULL)
1205         MSHookFunction(_ZN7WebCore12CachedScript6scriptEv, MSHake(_ZN7WebCore12CachedScript6scriptEv));
1206
1207     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i)(void *, const WebCore::String &, State, const WebCore::String &, int) = NULL;
1208     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i == NULL)
1209         MSHookSymbol(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i", WebCore);
1210     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i != NULL)
1211         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i));
1212
1213     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE)(void *, JSC::ScriptSourceCode &, State) = NULL;
1214     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE == NULL)
1215         MSHookSymbol(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE", WebCore);
1216     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE != NULL)
1217         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE));
1218
1219     void (*_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE)(void *, const WebCore::String &, const WebCore::KURL &, void *);
1220     msset(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, WebCore);
1221     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE != NULL)
1222         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionINS1_14OneBasedNumberEEE));
1223
1224     void (*_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE)(void *, const WebCore::String &, const WebCore::KURL &, void *);
1225     msset(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, WebCore);
1226     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE == NULL)
1227         MSHookSymbol(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, "__ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_3URLERKNS1_12TextPositionE", WebCore);
1228     if (_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE != NULL)
1229         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKN3WTF6StringERKNS_4KURLERKNS1_12TextPositionE));
1230
1231     void (*_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE)(void *, void *, int);
1232     msset(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, WebCore);
1233     if (_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE != NULL)
1234         MSHookFunction(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE, MSHake(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionINS1_14OneBasedNumberEEENS0_17LegacyTypeSupportE));
1235
1236     void (*_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE)(void *, void *, int);
1237     msset(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, WebCore);
1238     if (_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE != NULL)
1239         MSHookFunction(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE, MSHake(_ZN7WebCore13ScriptElement13prepareScriptERKN3WTF12TextPositionENS0_17LegacyTypeSupportE));
1240
1241     MSHookSymbol($_ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE, "__ZNK7WebCore13ScriptElement21isScriptTypeSupportedENS0_17LegacyTypeSupportE", WebCore);
1242
1243     void (*_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE)(void *, JSC::ScriptSourceCode &);
1244     msset(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, WebCore);
1245     if (_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE != NULL)
1246         MSHookFunction(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE, MSHake(_ZN7WebCore13ScriptElement13executeScriptERKNS_16ScriptSourceCodeE));
1247
1248     if (_ZN7WebCore6String6appendEPKtj == NULL)
1249         MSHookSymbol(_ZN7WebCore6String6appendEPKtj, "__ZN7WebCore6String6appendEPKtj", WebCore);
1250     if (_ZN7WebCore6String6appendEPKtj == NULL)
1251         msset(_ZN7WebCore6String6appendEPKtj, JavaScriptCore);
1252     if (_ZN7WebCore6String6appendEPKtj == NULL)
1253         MSHookSymbol(_ZN7WebCore6String6appendEPKtj, "__ZN3WTF6String6appendEPKtj", JavaScriptCore);
1254
1255     if (_ZN7WebCore6String8truncateEj == NULL)
1256         MSHookSymbol(_ZN7WebCore6String8truncateEj, "__ZN7WebCore6String8truncateEj", WebCore);
1257     if (_ZN7WebCore6String8truncateEj == NULL)
1258         msset(_ZN7WebCore6String8truncateEj, JavaScriptCore);
1259     if (_ZN7WebCore6String8truncateEj == NULL)
1260         MSHookSymbol(_ZN7WebCore6String8truncateEj, "__ZN3WTF6String8truncateEj", JavaScriptCore);
1261
1262     msset(_ZNK7WebCore6String10charactersEv, WebCore);
1263
1264     msset(_ZN7WebCore6String29charactersWithNullTerminationEv, JavaScriptCore);
1265     if (_ZN7WebCore6String29charactersWithNullTerminationEv == NULL)
1266         MSHookSymbol(_ZN7WebCore6String29charactersWithNullTerminationEv, "__ZN3WTF6String29charactersWithNullTerminationEv", JavaScriptCore);
1267
1268     msset(_ZNK3WTF6String14createCFStringEv, JavaScriptCore);
1269
1270     msset(_ZNK7WebCore6String6lengthEv, WebCore);
1271
1272     bool (*_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE)(WebCore::MediaQueryEvaluator *, WebCore::String &);
1273     msset(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, WebCore);
1274     if (_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE != NULL)
1275         MSHookFunction(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE, MSHake(_ZNK7WebCore19MediaQueryEvaluator4evalEPKNS_13MediaQueryExpE));
1276
1277     void (*_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE)(WebCore::MediaQueryExp *, WebCore::String &, WebCore::CSSParserValueList *);
1278     msset(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, WebCore);
1279     if (_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE != NULL)
1280         MSHookFunction(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE, MSHake(_ZN7WebCore13MediaQueryExpC2ERKN3WTF12AtomicStringEPNS_18CSSParserValueListE));
1281 }
1282
1283 + (id) rootViewController {
1284     return [[[self alloc] init] autorelease];
1285 }
1286
1287 - (void) dealloc {
1288     [configuration_ release];
1289     [background_ release];
1290     [foreground_ release];
1291     [super dealloc];
1292 }
1293
1294 - (id) init {
1295     if ((self = [super init]) != nil) {
1296         configuration_ = [[$CydgetController currentConfiguration] retain];
1297         legacy_ = [configuration_ objectForKey:@"Background"] == nil;
1298         media_ = [[configuration_ objectForKey:@"MediaControls"] boolValue];
1299         items_ = [[configuration_ objectForKey:@"NotificationList"] boolValue];
1300     } return self;
1301 }
1302
1303 - (void) loadView {
1304     NSURL *base([NSURL fileURLWithPath:[$CydgetController currentPath]]);
1305
1306     if (NSString *background = [configuration_ objectForKey:@"Background"])
1307         background_ = [[WebCydgetLockScreenView alloc] initWithURL:[NSURL URLWithString:background relativeToURL:base]];
1308
1309     if (NSString *homepage = [configuration_ objectForKey:@"Homepage"]) {
1310         foreground_ = [[WebCydgetLockScreenView alloc] initWithURL:[NSURL URLWithString:homepage relativeToURL:base]];
1311         [self setView:foreground_];
1312     }
1313 }
1314
1315 - (void) purgeView {
1316     [background_ removeFromSuperview];
1317     [background_ release];
1318     background_ = nil;
1319     [foreground_ removeFromSuperview];
1320     [foreground_ release];
1321     foreground_ = nil;
1322     [super purgeView];
1323 }
1324
1325 - (void) updateStyles {
1326     [foreground_ updateStyles];
1327     [background_ updateStyles];
1328 }
1329
1330 - (UIView *) backgroundView {
1331     return background_;
1332 }
1333
1334 - (BOOL) showAwayItems {
1335     return YES;
1336 }
1337
1338 - (BOOL) updateHidden {
1339     if (foreground_ == nil)
1340         return true;
1341     bool media(CYHaveMediaControls());
1342     [foreground_ setHidden:media];
1343     return media;
1344 }
1345
1346 - (BOOL) showDateView {
1347     if (kCFCoreFoundationVersionNumber < 800)
1348         return false;
1349     [self updateStyles];
1350     if (!legacy_ && foreground_ == nil)
1351         return true;
1352     if (!items_ && CYHaveNotificationList())
1353         return true;
1354     return false;
1355 }
1356
1357 - (BOOL) allowsLockScreenMediaControls {
1358     if (kCFCoreFoundationVersionNumber < 800)
1359         return true;
1360     return media_;
1361 }
1362
1363 /*- (BOOL) showHeaderView {
1364     return YES;
1365 }*/
1366
1367 // 0: view is rendered above head
1368 // 1: view moves as one with head
1369 // 2: view moves up and down only
1370 // 3: view simply never does move
1371 - (NSUInteger) presentationStyle {
1372     return 1;
1373 }
1374
1375 // 1: light blur
1376 // 2: heavy blur
1377 // 3: just black
1378 // 4: legibility
1379 // 5: no overlay
1380 - (NSUInteger) overlayStyle {
1381     return legacy_ ? 1 : 4;
1382 }
1383
1384 // 1: blur -> view -> list
1385 // 2: view -> blur -> list
1386 // 3: view. unblur below?!
1387 // 4: blur -> list -> view
1388 - (NSUInteger) notificationBehavior {
1389     return items_ ? 1 : 2;
1390 }
1391
1392 - (BOOL) viewWantsFullscreenLayout {
1393     return kCFCoreFoundationVersionNumber >= 800;
1394 }
1395
1396 /*- (BOOL) viewWantsOverlayLayout {
1397     return kCFCoreFoundationVersionNumber >= 800;
1398 }*/
1399
1400 - (BOOL) shouldDisableOnUnlock {
1401     return YES;
1402 }
1403
1404 - (BOOL) canBeAlwaysFullscreen {
1405     return YES;
1406 }
1407
1408 /*- (BOOL) wantsSwipeGestureRecognizer {
1409     return YES;
1410 }
1411
1412 - (BOOL) handleGesture:(int)arg1 fingerCount:(NSUInteger)fingers {
1413     return NO;
1414     return YES;
1415 }*/
1416
1417 // - (void) lockScreenMediaControlsShown:(BOOL)shown;
1418
1419 - (BOOL) handleMenuButtonDoubleTap {
1420     if (kCFCoreFoundationVersionNumber >= 800) {
1421         SBLockScreenViewController *controller([[$SBLockScreenManager sharedInstance] lockScreenViewController]);
1422         [controller _setMediaControlsVisible:![controller isShowingMediaControls]];
1423     }
1424
1425     return [super handleMenuButtonDoubleTap];
1426 }
1427
1428 @end
1429
1430 MSClassHook(WebView)
1431 MSMetaClassHook(WebView)
1432
1433 MSClassMessageHook0(void, WebView, enableWebThread) {
1434     if (kCFCoreFoundationVersionNumber >= 478.61)
1435         return MSOldCall();
1436
1437     NSLog(@"-[WebView enableWebThread]");
1438 }