LockScreen.mm

   1 /* CydgetScript - open-source IntelliDial replacement
   2  * Copyright (C) 2009  Jay Freeman (saurik)
   3 */
   4
   5 /*
   6  *        Redistribution and use in source and binary
   7  * forms, with or without modification, are permitted
   8  * provided that the following conditions are met:
   9  *
  10  * 1. Redistributions of source code must retain the
  11  *    above copyright notice, this list of conditions
  12  *    and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the
  14  *    above copyright notice, this list of conditions
  15  *    and the following disclaimer in the documentation
  16  *    and/or other materials provided with the
  17  *    distribution.
  18  * 3. The name of the author may not be used to endorse
  19  *    or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
  23  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
  24  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  25  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
  27  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  28  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  29  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  30  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  31  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  32  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  33  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  34  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  35  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  36 */
  37
  38 #include <substrate.h>
  39 #include <sys/sysctl.h>
  40
  41 #import <GraphicsServices/GraphicsServices.h>
  42 #import <UIKit/UIKit.h>
  43 #import <AddressBook/AddressBook.h>
  44
  45 #import <SpringBoard/SBStatusBarController.h>
  46 #import <SpringBoardUI/SBAwayViewPluginController.h>
  47 #import <TelephonyUI/TPBottomLockBar.h>
  48
  49 #import <QuartzCore/CALayer.h>
  50 // XXX: fix the minimum requirement
  51 extern NSString * const kCAFilterNearest;
  52
  53 #include <WebKit/DOMCSSPrimitiveValue.h>
  54 #include <WebKit/DOMCSSStyleDeclaration.h>
  55 #include <WebKit/DOMDocument.h>
  56 #include <WebKit/DOMHTMLBodyElement.h>
  57 #include <WebKit/DOMNodeList.h>
  58 #include <WebKit/DOMRGBColor.h>
  59
  60 #include <WebKit/WebFrame.h>
  61 #include <WebKit/WebPolicyDelegate.h>
  62 #include <WebKit/WebPreferences.h>
  63 #include <WebKit/WebScriptObject.h>
  64
  65 #import <WebKit/WebView.h>
  66 #import <WebKit/WebView-WebPrivate.h>
  67
  68 #include <WebCore/Page.h>
  69 #include <WebCore/Settings.h>
  70
  71 #include <WebCore/WebCoreThread.h>
  72 #include <WebKit/WebPreferences-WebPrivate.h>
  73
  74 #include "JSGlobalData.h"
  75
  76 #include "SourceCode.h"
  77
  78 #include <apr-1/apr_pools.h>
  79 #include <pcre.h>
  80
  81 @interface WebView (UICaboodle)
  82 - (void) setScriptDebugDelegate:(id)delegate;
  83 - (void) _setFormDelegate:(id)delegate;
  84 - (void) _setUIKitDelegate:(id)delegate;
  85 - (void) setWebMailDelegate:(id)delegate;
  86 - (void) _setLayoutInterval:(float)interval;
  87 @end
  88
  89 #define _transient
  90 #define _forever for (;;)
  91
  92 _disused static unsigned trace_;
  93
  94 #define _trace() do { \
  95     NSLog(@"_trace(%u)@%s:%u[%s](%p)\n", \
  96         trace_++, __FILE__, __LINE__, __FUNCTION__, pthread_self() \
  97     ); \
  98 } while (false)
  99
 100 #define _assert(test) do \
 101     if (!(test)) { \
 102         fprintf(stderr, "_assert(%d:%s)@%s:%u[%s]\n", errno, #test, __FILE__, __LINE__, __FUNCTION__); \
 103         exit(-1); \
 104     } \
 105 while (false)
 106
 107 #define _syscall(expr) \
 108     do if ((long) (expr) != -1) \
 109         break; \
 110     else switch (errno) { \
 111         case EINTR: \
 112             continue; \
 113         default: \
 114             _assert(false); \
 115     } while (true)
 116
 117 @protocol CydgetController
 118 - (NSDictionary *) currentConfiguration;
 119 @end
 120
 121 static Class $CydgetController(objc_getClass("CydgetController"));
 122 static Class $UIFormAssistant(objc_getClass("UIFormAssistant"));
 123 //static Class $SBStatusBarController(objc_getClass("SBStatusBarController"));
 124
 125 static Class $UIWebBrowserView;
 126 static bool Wildcat_, iOS4;
 127
 128 @interface NSString (UIKit)
 129 - (NSString *) stringByAddingPercentEscapes;
 130 @end
 131
 132 @implementation UIWebDocumentView (WebCycript)
 133
 134 - (void) _setScrollerOffset:(CGPoint)offset {
 135     UIScroller *scroller([self _scroller]);
 136
 137     CGSize size([scroller contentSize]);
 138     CGSize bounds([scroller bounds].size);
 139
 140     CGPoint max;
 141     max.x = size.width - bounds.width;
 142     max.y = size.height - bounds.height;
 143
 144     // wtf Apple?!
 145     if (max.x < 0)
 146         max.x = 0;
 147     if (max.y < 0)
 148         max.y = 0;
 149
 150     offset.x = offset.x < 0 ? 0 : offset.x > max.x ? max.x : offset.x;
 151     offset.y = offset.y < 0 ? 0 : offset.y > max.y ? max.y : offset.y;
 152
 153     [scroller setOffset:offset];
 154 }
 155
 156 @end
 157
 158 /* Perl-Compatible RegEx {{{ */
 159 class Pcre {
 160   private:
 161     pcre *code_;
 162     pcre_extra *study_;
 163     int capture_;
 164     int *matches_;
 165     const char *data_;
 166
 167   public:
 168     Pcre(const char *regex, int options = 0) :
 169         study_(NULL)
 170     {
 171         const char *error;
 172         int offset;
 173         code_ = pcre_compile(regex, options, &error, &offset, NULL);
 174
 175         if (code_ == NULL)
 176             @throw [NSException exceptionWithName:NSInvalidArgumentException reason:[NSString stringWithFormat:@"*** Pcre(,): [%u] %s", offset, error] userInfo:nil];
 177
 178         pcre_fullinfo(code_, study_, PCRE_INFO_CAPTURECOUNT, &capture_);
 179         matches_ = new int[(capture_ + 1) * 3];
 180     }
 181
 182     ~Pcre() {
 183         pcre_free(code_);
 184         delete matches_;
 185     }
 186
 187     NSString *operator [](size_t match) {
 188         return [[[NSString alloc] initWithBytes:(data_ + matches_[match * 2]) length:(matches_[match * 2 + 1] - matches_[match * 2]) encoding:NSUTF8StringEncoding] autorelease];
 189     }
 190
 191     bool operator ()(NSString *data) {
 192         // XXX: length is for characters, not for bytes
 193         return operator ()([data UTF8String], [data length]);
 194     }
 195
 196     bool operator ()(const char *data, size_t size) {
 197         data_ = data;
 198         return pcre_exec(code_, study_, data, size, 0, 0, matches_, (capture_ + 1) * 3) >= 0;
 199     }
 200 };
 201 /* }}} */
 202 /* WebCycript Delegate {{{ */
 203 @interface WebCycriptDelegate : NSObject {
 204     _transient volatile id delegate_;
 205 }
 206
 207 - (void) setDelegate:(id)delegate;
 208 - (id) initWithDelegate:(id)delegate;
 209 @end
 210
 211 @implementation WebCycriptDelegate
 212
 213 - (void) setDelegate:(id)delegate {
 214     delegate_ = delegate;
 215 }
 216
 217 - (id) initWithDelegate:(id)delegate {
 218     delegate_ = delegate;
 219     return self;
 220 }
 221
 222 - (void) webView:(WebView *)sender didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
 223     if (delegate_ != nil)
 224         return [delegate_ webView:sender didClearWindowObject:window forFrame:frame];
 225 }
 226
 227 - (void) webView:(WebView *)sender didCommitLoadForFrame:(WebFrame *)frame {
 228     if (delegate_ != nil)
 229         return [delegate_ webView:sender didCommitLoadForFrame:frame];
 230 }
 231
 232 - (void) webView:(WebView *)sender didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 233     if (delegate_ != nil)
 234         return [delegate_ webView:sender didFailLoadWithError:error forFrame:frame];
 235 }
 236
 237 - (void) webView:(WebView *)sender didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 238     if (delegate_ != nil)
 239         return [delegate_ webView:sender didFailProvisionalLoadWithError:error forFrame:frame];
 240 }
 241
 242 - (void) webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame {
 243     if (delegate_ != nil)
 244         return [delegate_ webView:sender didFinishLoadForFrame:frame];
 245 }
 246
 247 /*- (void) webView:(WebView *)sender didReceiveTitle:(NSString *)title forFrame:(WebFrame *)frame {
 248     if (delegate_ != nil)
 249         return [delegate_ webView:sender didReceiveTitle:title forFrame:frame];
 250 }*/
 251
 252 - (void) webView:(WebView *)sender didStartProvisionalLoadForFrame:(WebFrame *)frame {
 253     if (delegate_ != nil)
 254         return [delegate_ webView:sender didStartProvisionalLoadForFrame:frame];
 255 }
 256
 257 /*- (void) webView:(WebView *)sender resource:(id)identifier didReceiveAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge fromDataSource:(WebDataSource *)source {
 258     if (delegate_ != nil)
 259         return [delegate_ webView:sender resource:identifier didReceiveAuthenticationChallenge:challenge fromDataSource:source];
 260 }*/
 261
 262 /*- (NSURLRequest *) webView:(WebView *)sender resource:(id)identifier willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse fromDataSource:(WebDataSource *)source {
 263     if (delegate_ != nil)
 264         return [delegate_ webView:sender resource:identifier willSendRequest:request redirectResponse:redirectResponse fromDataSource:source];
 265     return nil;
 266 }*/
 267
 268 - (IMP) methodForSelector:(SEL)sel {
 269     if (IMP method = [super methodForSelector:sel])
 270         return method;
 271     fprintf(stderr, "methodForSelector:[%s] == NULL\n", sel_getName(sel));
 272     return NULL;
 273 }
 274
 275 - (BOOL) respondsToSelector:(SEL)sel {
 276     if ([super respondsToSelector:sel])
 277         return YES;
 278     // XXX: WebThreadCreateNSInvocation returns nil
 279     //fprintf(stderr, "[%s]R?%s\n", class_getName(self->isa), sel_getName(sel));
 280     return delegate_ == nil ? NO : [delegate_ respondsToSelector:sel];
 281 }
 282
 283 - (NSMethodSignature *) methodSignatureForSelector:(SEL)sel {
 284     if (NSMethodSignature *method = [super methodSignatureForSelector:sel])
 285         return method;
 286     //fprintf(stderr, "[%s]S?%s\n", class_getName(self->isa), sel_getName(sel));
 287     if (delegate_ != nil)
 288         if (NSMethodSignature *sig = [delegate_ methodSignatureForSelector:sel])
 289             return sig;
 290     // XXX: I fucking hate Apple so very very bad
 291     return [NSMethodSignature signatureWithObjCTypes:"v@:"];
 292 }
 293
 294 - (void) forwardInvocation:(NSInvocation *)inv {
 295     SEL sel = [inv selector];
 296     if (delegate_ != nil && [delegate_ respondsToSelector:sel])
 297         [inv invokeWithTarget:delegate_];
 298 }
 299
 300 @end
 301 /* }}} */
 302
 303 @interface WebCydgetLockScreenView : UIView {
 304     WebCycriptDelegate *indirect_;
 305     UIProgressIndicator *indicator_;
 306     UIScroller *scroller_;
 307     UIWebDocumentView *document_;
 308
 309     NSString *cycript_;
 310     bool scrollable_;
 311
 312     float width_;
 313     CGSize size_;
 314     bool editing_;
 315
 316     NSNumber *confirm_;
 317
 318     NSMutableSet *loading_;
 319     bool error_;
 320     bool reloading_;
 321
 322     //UIKeyboard *keyboard_;
 323 }
 324
 325 @end
 326
 327 @implementation WebCydgetLockScreenView
 328
 329 //#include "UICaboodle/UCInternal.h"
 330
 331 - (void) dealloc {
 332     WebThreadLock();
 333
 334     WebView *webview([document_ webView]);
 335     [webview setFrameLoadDelegate:nil];
 336     [webview setResourceLoadDelegate:nil];
 337     [webview setUIDelegate:nil];
 338     [webview setScriptDebugDelegate:nil];
 339     [webview setPolicyDelegate:nil];
 340
 341     /* XXX: these are set by UIWebDocumentView
 342     [webview setDownloadDelegate:nil];
 343     [webview _setFormDelegate:nil];
 344     [webview _setUIKitDelegate:nil];
 345     [webview setEditingDelegate:nil];*/
 346
 347     /* XXX: no one sets this, ever
 348     [webview setWebMailDelegate:nil];*/
 349
 350     [document_ setDelegate:nil];
 351     [document_ setGestureDelegate:nil];
 352
 353     if ([document_ respondsToSelector:@selector(setFormEditingDelegate:)])
 354         [document_ setFormEditingDelegate:nil];
 355
 356     [document_ setInteractionDelegate:nil];
 357
 358     [indirect_ setDelegate:nil];
 359
 360     [webview close];
 361     [document_ release];
 362
 363     [indirect_ release];
 364
 365     WebThreadUnlock();
 366
 367     [scroller_ setDelegate:nil];
 368
 369     if (confirm_ != nil)
 370         [confirm_ release];
 371
 372     //[keyboard_ release];
 373
 374     [scroller_ release];
 375     [indicator_ release];
 376     [loading_ release];
 377     [super dealloc];
 378 }
 379
 380 + (float) defaultWidth {
 381     return 980;
 382 }
 383
 384 - (void) webView:(WebView *)sender didReceiveMessage:(NSDictionary *)dictionary {
 385 #if LogBrowser || ForSaurik
 386     NSLog(@"Console:%@\n", [dictionary description]);
 387 #endif
 388     if ([document_ respondsToSelector:@selector(webView:didReceiveMessage:)])
 389         [document_ webView:sender didReceiveMessage:dictionary];
 390 }
 391
 392 - (void) webView:(id)sender willCloseFrame:(id)frame {
 393     if ([document_ respondsToSelector:@selector(webView:willCloseFrame:)])
 394         [document_ webView:sender willCloseFrame:frame];
 395 }
 396
 397 - (void) webView:(id)sender didFinishDocumentLoadForFrame:(id)frame {
 398     if ([document_ respondsToSelector:@selector(webView:didFinishDocumentLoadForFrame:)])
 399         [document_ webView:sender didFinishDocumentLoadForFrame:frame];
 400 }
 401
 402 - (void) webView:(id)sender didFirstLayoutInFrame:(id)frame {
 403     if ([document_ respondsToSelector:@selector(webView:didFirstLayoutInFrame:)])
 404         [document_ webView:sender didFirstLayoutInFrame:frame];
 405 }
 406
 407 - (void) webViewFormEditedStatusHasChanged:(id)changed {
 408     if ([document_ respondsToSelector:@selector(webViewFormEditedStatusHasChanged:)])
 409         [document_ webViewFormEditedStatusHasChanged:changed];
 410 }
 411
 412 - (void) webView:(id)sender formStateDidFocusNode:(id)formState {
 413     if ([document_ respondsToSelector:@selector(webView:formStateDidFocusNode:)])
 414         [document_ webView:sender formStateDidFocusNode:formState];
 415 }
 416
 417 - (void) webView:(id)sender formStateDidBlurNode:(id)formState {
 418     if ([document_ respondsToSelector:@selector(webView:formStateDidBlurNode:)])
 419         [document_ webView:sender formStateDidBlurNode:formState];
 420 }
 421
 422 - (void) webViewDidLayout:(id)sender {
 423     [document_ webViewDidLayout:sender];
 424 }
 425
 426 - (void) webView:(id)sender didFirstVisuallyNonEmptyLayoutInFrame:(id)frame {
 427     [document_ webView:sender didFirstVisuallyNonEmptyLayoutInFrame:frame];
 428 }
 429
 430 - (void) webView:(id)sender saveStateToHistoryItem:(id)item forFrame:(id)frame {
 431     [document_ webView:sender saveStateToHistoryItem:item forFrame:frame];
 432 }
 433
 434 - (void) webView:(id)sender restoreStateFromHistoryItem:(id)item forFrame:(id)frame force:(BOOL)force {
 435     [document_ webView:sender restoreStateFromHistoryItem:item forFrame:frame force:force];
 436 }
 437
 438 - (void) webView:(id)sender attachRootLayer:(id)layer {
 439     [document_ webView:sender attachRootLayer:layer];
 440 }
 441
 442 - (id) webView:(id)sender plugInViewWithArguments:(id)arguments fromPlugInPackage:(id)package {
 443     return [document_ webView:sender plugInViewWithArguments:arguments fromPlugInPackage:package];
 444 }
 445
 446 - (void) webView:(id)sender willShowFullScreenForPlugInView:(id)view {
 447     [document_ webView:sender willShowFullScreenForPlugInView:view];
 448 }
 449
 450 - (void) webView:(id)sender didHideFullScreenForPlugInView:(id)view {
 451     [document_ webView:sender didHideFullScreenForPlugInView:view];
 452 }
 453
 454 - (void) webView:(id)sender willAddPlugInView:(id)view {
 455     [document_ webView:sender willAddPlugInView:view];
 456 }
 457
 458 - (void) webView:(id)sender didObserveDeferredContentChange:(int)change forFrame:(id)frame {
 459     [document_ webView:sender didObserveDeferredContentChange:change forFrame:frame];
 460 }
 461
 462 - (void) webViewDidPreventDefaultForEvent:(id)sender {
 463     [document_ webViewDidPreventDefaultForEvent:sender];
 464 }
 465
 466 - (void) _setTileDrawingEnabled:(BOOL)enabled {
 467     //[document_ setTileDrawingEnabled:enabled];
 468 }
 469
 470 - (void) willStartGesturesInView:(UIView *)view forEvent:(GSEventRef)event {
 471     [self _setTileDrawingEnabled:NO];
 472 }
 473
 474 - (void) didFinishGesturesInView:(UIView *)view forEvent:(GSEventRef)event {
 475     [self _setTileDrawingEnabled:YES];
 476     [document_ redrawScaledDocument];
 477 }
 478
 479 - (void) setViewportWidth:(float)width {
 480     width_ = width != 0 ? width : [[self class] defaultWidth];
 481     [document_ setViewportSize:CGSizeMake(width_, UIWebViewGrowsAndShrinksToFitHeight) forDocumentTypes:0x10];
 482 }
 483
 484 - (void) scrollerWillStartDragging:(UIScroller *)scroller {
 485     [self _setTileDrawingEnabled:NO];
 486 }
 487
 488 - (void) scrollerDidEndDragging:(UIScroller *)scroller willSmoothScroll:(BOOL)smooth {
 489     [self _setTileDrawingEnabled:YES];
 490 }
 491
 492 - (void) scrollerDidEndDragging:(UIScroller *)scroller {
 493     [self _setTileDrawingEnabled:YES];
 494 }
 495
 496 - (void) loadRequest:(NSURLRequest *)request {
 497     error_ = false;
 498
 499     WebThreadLock();
 500     [document_ loadRequest:request];
 501     WebThreadUnlock();
 502 }
 503
 504 - (void) loadURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {
 505     [self loadRequest:[NSURLRequest
 506         requestWithURL:url
 507         cachePolicy:policy
 508         timeoutInterval:30.0
 509     ]];
 510 }
 511
 512 - (void) loadURL:(NSURL *)url {
 513     [self loadURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy];
 514 }
 515
 516 - (id) init {
 517     CGRect frame = {{0, 0}, {320, 480}};
 518     frame.size.height -= 20; //[[[$SBStatusBarController sharedStatusBarController] statusBarView] frame].size.height;
 519
 520     if ((self = [super initWithFrame:frame]) != nil) {
 521         loading_ = [[NSMutableSet alloc] initWithCapacity:3];
 522
 523         struct CGRect bounds([self bounds]);
 524
 525         scroller_ = [[objc_getClass(Wildcat_ ? "UIScrollView" : "UIScroller") alloc] initWithFrame:bounds];
 526         [self addSubview:scroller_];
 527
 528         [scroller_ setFixedBackgroundPattern:YES];
 529         [scroller_ setBackgroundColor:[UIColor blackColor]];
 530
 531         [scroller_ setScrollingEnabled:YES];
 532         [scroller_ setClipsSubviews:YES];
 533
 534         if (!Wildcat_)
 535             [scroller_ setAllowsRubberBanding:YES];
 536
 537         [scroller_ setDelegate:self];
 538         [scroller_ setBounces:YES];
 539
 540         if (!Wildcat_) {
 541             [scroller_ setScrollHysteresis:8];
 542             [scroller_ setThumbDetectionEnabled:NO];
 543             [scroller_ setDirectionalScrolling:YES];
 544             //[scroller_ setScrollDecelerationFactor:0.99]; /* 0.989324 */
 545             [scroller_ setEventMode:YES];
 546         }
 547
 548         if (Wildcat_) {
 549             UIScrollView *scroller((UIScrollView *)scroller_);
 550             //[scroller setDirectionalLockEnabled:NO];
 551             [scroller setDelaysContentTouches:NO];
 552             //[scroller setScrollsToTop:NO];
 553             //[scroller setCanCancelContentTouches:NO];
 554         }
 555
 556         [scroller_ setShowBackgroundShadow:NO]; /* YES */
 557         //[scroller_ setAllowsRubberBanding:YES]; /* Vertical */
 558
 559         if (!Wildcat_)
 560             [scroller_ setAdjustForContentSizeChange:YES]; /* NO */
 561
 562         CGRect rect([scroller_ bounds]);
 563         //rect.size.height = 0;
 564
 565         WebThreadLock();
 566
 567         document_ = [[$UIWebBrowserView alloc] initWithFrame:rect];
 568         WebView *webview([document_ webView]);
 569
 570         [document_ setBackgroundColor:[UIColor blackColor]];
 571         if ([document_ respondsToSelector:@selector(setDrawsBackground:)])
 572             [document_ setDrawsBackground:NO];
 573         [webview setDrawsBackground:NO];
 574
 575         [webview setPreferencesIdentifier:@"WebCycript"];
 576
 577         [document_ setTileSize:CGSizeMake(rect.size.width, 500)];
 578
 579         if ([document_ respondsToSelector:@selector(enableReachability)])
 580             [document_ enableReachability];
 581         if ([document_ respondsToSelector:@selector(setAllowsMessaging:)])
 582             [document_ setAllowsMessaging:YES];
 583         if ([document_ respondsToSelector:@selector(useSelectionAssistantWithMode:)])
 584             [document_ useSelectionAssistantWithMode:0];
 585
 586         [document_ setTilingEnabled:YES];
 587         [document_ setDrawsGrid:NO];
 588         [document_ setLogsTilingChanges:NO];
 589         [document_ setTileMinificationFilter:kCAFilterNearest];
 590
 591         if ([document_ respondsToSelector:@selector(setDataDetectorTypes:)])
 592             /* XXX: abstractify */
 593             [document_ setDataDetectorTypes:0x80000000];
 594         else
 595             [document_ setDetectsPhoneNumbers:NO];
 596
 597         [document_ setAutoresizes:YES];
 598
 599         [document_ setMinimumScale:0.25f forDocumentTypes:0x10];
 600         [document_ setMaximumScale:5.00f forDocumentTypes:0x10];
 601         [document_ setInitialScale:UIWebViewScalesToFitScale forDocumentTypes:0x10];
 602         //[document_ setViewportSize:CGSizeMake(980, UIWebViewGrowsAndShrinksToFitHeight) forDocumentTypes:0x10];
 603
 604         [document_ setViewportSize:CGSizeMake(320, UIWebViewGrowsAndShrinksToFitHeight) forDocumentTypes:0x2];
 605
 606         [document_ setMinimumScale:1.00f forDocumentTypes:0x8];
 607         [document_ setInitialScale:UIWebViewScalesToFitScale forDocumentTypes:0x8];
 608         [document_ setViewportSize:CGSizeMake(320, UIWebViewGrowsAndShrinksToFitHeight) forDocumentTypes:0x8];
 609
 610         [document_ _setDocumentType:0x4];
 611
 612         if ([document_ respondsToSelector:@selector(setZoomsFocusedFormControl:)])
 613             [document_ setZoomsFocusedFormControl:YES];
 614         [document_ setContentsPosition:7];
 615         [document_ setEnabledGestures:0xa];
 616         [document_ setValue:[NSNumber numberWithBool:YES] forGestureAttribute:UIGestureAttributeIsZoomRubberBandEnabled];
 617         [document_ setValue:[NSNumber numberWithBool:YES] forGestureAttribute:UIGestureAttributeUpdatesScroller];
 618
 619         [document_ setSmoothsFonts:YES];
 620         [document_ setAllowsImageSheet:YES];
 621         [webview _setUsesLoaderCache:YES];
 622
 623         [webview setGroupName:@"CydgetGroup"];
 624
 625         WebPreferences *preferences([webview preferences]);
 626
 627         if ([webview respondsToSelector:@selector(_setLayoutInterval:)])
 628             [webview _setLayoutInterval:0];
 629         else
 630             [preferences _setLayoutInterval:0];
 631
 632         [self setViewportWidth:0];
 633
 634         [document_ setDelegate:self];
 635         [document_ setGestureDelegate:self];
 636
 637         if ([document_ respondsToSelector:@selector(setFormEditingDelegate:)])
 638             [document_ setFormEditingDelegate:self];
 639         [document_ setInteractionDelegate:self];
 640
 641         [scroller_ addSubview:document_];
 642
 643         //NSNotificationCenter *center = [NSNotificationCenter defaultCenter];
 644
 645         indirect_ = [[WebCycriptDelegate alloc] initWithDelegate:self];
 646
 647         [webview setFrameLoadDelegate:indirect_];
 648         [webview setPolicyDelegate:indirect_];
 649         [webview setResourceLoadDelegate:indirect_];
 650         [webview setUIDelegate:indirect_];
 651
 652         /* XXX: do not turn this on under penalty of extreme pain */
 653         [webview setScriptDebugDelegate:nil];
 654
 655         WebThreadUnlock();
 656
 657         CGSize indsize([UIProgressIndicator defaultSizeForStyle:UIProgressIndicatorStyleMediumWhite]);
 658         indicator_ = [[UIProgressIndicator alloc] initWithFrame:CGRectMake(281, 12, indsize.width, indsize.height)];
 659         [indicator_ setStyle:UIProgressIndicatorStyleMediumWhite];
 660
 661         [self setAutoresizingMask:UIViewAutoresizingFlexibleHeight];
 662         [scroller_ setAutoresizingMask:UIViewAutoresizingFlexibleHeight];
 663
 664         NSDictionary *configuration([$CydgetController currentConfiguration]);
 665
 666         cycript_ = [configuration objectForKey:@"CycriptURLs"];
 667
 668         scrollable_ = [[configuration objectForKey:@"Scrollable"] boolValue];
 669         [scroller_ setScrollingEnabled:scrollable_];
 670
 671         NSString *homepage([configuration objectForKey:@"Homepage"]);
 672         [self loadURL:[NSURL URLWithString:homepage]];
 673
 674         /*[UIKeyboard initImplementationNow];
 675         CGSize keysize = [UIKeyboard defaultSize];
 676         CGRect keyrect = {{0, [self bounds].size.height - 100}, keysize};
 677         keyboard_ = [[UIKeyboard alloc] initWithFrame:keyrect];
 678         [self addSubview:keyboard_];
 679
 680         [self addSubview:[[UITextView alloc] initWithFrame:CGRectMake(200, 0, 100, 100)]];
 681         [self addSubview:[[UITextView alloc] initWithFrame:CGRectMake(200, 150, 100, 100)]];
 682         [self addSubview:[[UITextView alloc] initWithFrame:CGRectMake(200, 300, 100, 100)]];*/
 683     } return self;
 684 }
 685
 686 - (void) webView:(WebView *)sender runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
 687     [self retain];
 688
 689     UIActionSheet *sheet = [[[UIActionSheet alloc]
 690         initWithTitle:nil
 691         buttons:[NSArray arrayWithObjects:@"OK", nil]
 692         defaultButtonIndex:0
 693         delegate:self
 694         context:@"alert"
 695     ] autorelease];
 696
 697     [sheet setBodyText:message];
 698     [sheet popupAlertAnimated:YES];
 699 }
 700
 701 - (BOOL) webView:(WebView *)sender runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
 702     [self retain];
 703
 704     UIActionSheet *sheet = [[[UIActionSheet alloc]
 705         initWithTitle:nil
 706         buttons:[NSArray arrayWithObjects:@"OK", @"CANCEL", nil]
 707         defaultButtonIndex:0
 708         delegate:indirect_
 709         context:@"confirm"
 710     ] autorelease];
 711
 712     [sheet setNumberOfRows:1];
 713     [sheet setBodyText:message];
 714     [sheet popupAlertAnimated:YES];
 715
 716     NSRunLoop *loop([NSRunLoop currentRunLoop]);
 717     NSDate *future([NSDate distantFuture]);
 718
 719     while (confirm_ == nil && [loop runMode:NSDefaultRunLoopMode beforeDate:future]);
 720
 721     NSNumber *confirm([confirm_ autorelease]);
 722     confirm_ = nil;
 723
 724     [self autorelease];
 725     return [confirm boolValue];
 726 }
 727
 728 /* XXX: WebThreadLock? */
 729 - (void) _fixScroller:(CGRect)bounds {
 730     float extra;
 731     if (!editing_ || $UIFormAssistant == nil)
 732         extra = 0;
 733     else {
 734         UIFormAssistant *assistant([$UIFormAssistant sharedFormAssistant]);
 735         CGRect peripheral([assistant peripheralFrame]);
 736         extra = peripheral.size.height;
 737     }
 738
 739     CGRect subrect([scroller_ frame]);
 740     subrect.size.height -= [TPBottomLockBar defaultHeight];
 741     subrect.size.height -= extra;
 742
 743     if ([scroller_ respondsToSelector:@selector(setScrollerIndicatorSubrect:)])
 744         [scroller_ setScrollerIndicatorSubrect:subrect];
 745
 746     [document_ setValue:[NSValue valueWithSize:NSMakeSize(subrect.size.width, subrect.size.height)] forGestureAttribute:UIGestureAttributeVisibleSize];
 747
 748     CGSize size(size_);
 749     size.height += extra;
 750     size.height += [TPBottomLockBar defaultHeight];
 751     [scroller_ setContentSize:size];
 752
 753     if ([scroller_ respondsToSelector:@selector(releaseRubberBandIfNecessary)])
 754         [scroller_ releaseRubberBandIfNecessary];
 755 }
 756
 757 - (void) fixScroller {
 758     CGRect bounds([document_ documentBounds]);
 759     [self _fixScroller:bounds];
 760 }
 761
 762 - (void) view:(UIView *)sender didSetFrame:(CGRect)frame {
 763     size_ = frame.size;
 764     [self _fixScroller:frame];
 765 }
 766
 767 - (void) view:(UIView *)sender didSetFrame:(CGRect)frame oldFrame:(CGRect)old {
 768     [self view:sender didSetFrame:frame];
 769 }
 770
 771 - (void) formAssistant:(id)sender didBeginEditingFormNode:(id)node {
 772 }
 773
 774 - (void) formAssistant:(id)sender didEndEditingFormNode:(id)node {
 775     [self fixScroller];
 776 }
 777
 778 - (void) webView:(WebView *)sender willBeginEditingFormElement:(id)element {
 779     editing_ = true;
 780 }
 781
 782 - (void) webView:(WebView *)sender didBeginEditingFormElement:(id)element {
 783     [self fixScroller];
 784 }
 785
 786 - (void) webViewDidEndEditingFormElements:(WebView *)sender {
 787     editing_ = false;
 788     [self fixScroller];
 789 }
 790
 791 - (void) webView:(WebView *)sender didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
 792     if (cycript_ != nil)
 793         if (NSString *href = [[[[frame dataSource] request] URL] absoluteString])
 794             if (Pcre([cycript_ UTF8String], 0 /*XXX:PCRE_UTF8*/)(href))
 795                 if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
 796                     if (void (*CYSetupContext)(JSGlobalContextRef) = reinterpret_cast<void (*)(JSGlobalContextRef)>(dlsym(handle, "CydgetSetupContext"))) {
 797                         WebView *webview([document_ webView]);
 798                         WebFrame *frame([webview mainFrame]);
 799                         JSGlobalContextRef context([frame globalContext]);
 800                         CYSetupContext(context);
 801                     }
 802 }
 803
 804 - (bool) isLoading {
 805     return [loading_ count] != 0;
 806 }
 807
 808 - (void) reloadButtons {
 809     if ([self isLoading]) {
 810         [UIApp setNetworkActivityIndicatorVisible:YES];
 811         [indicator_ startAnimation];
 812     } else {
 813         [UIApp setNetworkActivityIndicatorVisible:NO];
 814         [indicator_ stopAnimation];
 815     }
 816 }
 817
 818 - (void) _finishLoading {
 819     size_t count([loading_ count]);
 820     /*if (count == 0)
 821         [self autorelease];*/
 822     if (reloading_ || count != 0)
 823         return;
 824     [self reloadButtons];
 825 }
 826
 827 - (BOOL) webView:(WebView *)sender shouldScrollToPoint:(struct CGPoint)point forFrame:(WebFrame *)frame {
 828     return [document_ webView:sender shouldScrollToPoint:point forFrame:frame];
 829 }
 830
 831 - (void) webView:(WebView *)sender didReceiveViewportArguments:(id)arguments forFrame:(WebFrame *)frame {
 832     return [document_ webView:sender didReceiveViewportArguments:arguments forFrame:frame];
 833 }
 834
 835 - (void) webView:(WebView *)sender needsScrollNotifications:(id)notifications forFrame:(WebFrame *)frame {
 836     return [document_ webView:sender needsScrollNotifications:notifications forFrame:frame];
 837 }
 838
 839 - (void) webView:(WebView *)sender didCommitLoadForFrame:(WebFrame *)frame {
 840     return [document_ webView:sender didCommitLoadForFrame:frame];
 841 }
 842
 843 - (void) webView:(WebView *)sender didReceiveDocTypeForFrame:(WebFrame *)frame {
 844     return [document_ webView:sender didReceiveDocTypeForFrame:frame];
 845 }
 846
 847 - (void) webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame {
 848     [loading_ removeObject:[NSValue valueWithNonretainedObject:frame]];
 849     [self _finishLoading];
 850     return [document_ webView:sender didFinishLoadForFrame:frame];
 851 }
 852
 853 - (void) webView:(WebView *)sender didStartProvisionalLoadForFrame:(WebFrame *)frame {
 854     /*if ([loading_ count] == 0)
 855         [self retain];*/
 856     [loading_ addObject:[NSValue valueWithNonretainedObject:frame]];
 857
 858     if ([frame parentFrame] == nil) {
 859         [document_ resignFirstResponder];
 860
 861         reloading_ = false;
 862
 863         if (Wildcat_) {
 864             CGRect webrect = [scroller_ bounds];
 865             webrect.size.height = 1;
 866             [document_ setFrame:webrect];
 867         }
 868
 869         if ([scroller_ respondsToSelector:@selector(scrollPointVisibleAtTopLeft:)])
 870             [scroller_ scrollPointVisibleAtTopLeft:CGPointZero];
 871         else
 872             [scroller_ scrollRectToVisible:CGRectZero animated:NO];
 873
 874         if ([scroller_ respondsToSelector:@selector(setZoomScale:duration:)])
 875             [scroller_ setZoomScale:1 duration:0];
 876         else if ([scroller_ respondsToSelector:@selector(_setZoomScale:duration:)])
 877             [scroller_ _setZoomScale:1 duration:0];
 878         /*else if ([scroller_ respondsToSelector:@selector(setZoomScale:animated:)])
 879             [scroller_ setZoomScale:1 animated:NO];*/
 880
 881         if (!Wildcat_) {
 882             CGRect webrect = [scroller_ bounds];
 883             webrect.size.height = 1;
 884             [document_ setFrame:webrect];
 885         }
 886     }
 887
 888     [self reloadButtons];
 889 }
 890
 891 - (void) _didFailWithError:(NSError *)error forFrame:(WebFrame *)frame {
 892     /*if ([frame parentFrame] == nil)
 893         [self autorelease];*/
 894
 895     [loading_ removeObject:[NSValue valueWithNonretainedObject:frame]];
 896     [self _finishLoading];
 897
 898     if (reloading_)
 899         return;
 900
 901     if ([frame parentFrame] == nil) {
 902         [self loadURL:[NSURL URLWithString:[NSString stringWithFormat:@"%@?%@",
 903             [[NSURL fileURLWithPath:[[NSBundle mainBundle] pathForResource:@"error" ofType:@"html"]] absoluteString],
 904             [[error localizedDescription] stringByAddingPercentEscapes]
 905         ]]];
 906
 907         error_ = true;
 908     }
 909 }
 910
 911 - (void) webView:(WebView *)sender didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 912     [self _didFailWithError:error forFrame:frame];
 913     if ([document_ respondsToSelector:@selector(webView:didFailLoadWithError:forFrame:)])
 914         [document_ webView:sender didFailLoadWithError:error forFrame:frame];
 915 }
 916
 917 - (void) webView:(WebView *)sender didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 918     [self _didFailWithError:error forFrame:frame];
 919 }
 920
 921 - (void) webView:(WebView *)sender addMessageToConsole:(NSDictionary *)dictionary {
 922     NSLog(@"Console:%@\n", [dictionary description]);
 923 }
 924
 925 @end
 926
 927 @interface WebCycriptLockScreenController : SBAwayViewPluginController {
 928 }
 929
 930 @end
 931
 932 #include <string>
 933
 934 struct State {
 935     unsigned state;
 936 };
 937
 938 // State Machine {{{
 939 static bool cycript_;
 940
 941 MSHook(bool, _ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, const WebCore::String &mime) {
 942     _trace();
 943     if (mime != "text/cycript") {
 944         cycript_ = false;
 945         return __ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE(mime);
 946     }
 947     _trace();
 948
 949     static void *handle(dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL));
 950     if (handle == NULL)
 951         return false;
 952
 953     cycript_ = true;
 954     return true;
 955 }
 956 // }}}
 957 // Script Compiler {{{
 958 static void Log(const WebCore::String &string) {
 959     size_t length(string.length());
 960     UChar data[length + 1];
 961     data[length] = 0;
 962     memcpy(data, string.characters(), length * 2);
 963     NSLog(@"wtf %p:%S:", &string, data);
 964 }
 965
 966 static void Cycriptify(apr_pool_t *pool, const uint16_t *&data, size_t &size) {
 967     cycript_ = false;
 968
 969     _trace();
 970     if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
 971         if (void (*CYParseUChar)(apr_pool_t *, const uint16_t **, size_t *) = reinterpret_cast<void (*)(apr_pool_t *, const uint16_t **, size_t *)>(dlsym(handle, "CydgetPoolParse")))
 972             CYParseUChar(pool, &data, &size);
 973 }
 974
 975 static void (*_ZN7WebCore6String6appendEPKtj)(WebCore::String *, const UChar *, unsigned);
 976 static void (*_ZN7WebCore6String8truncateEj)(WebCore::String *, unsigned);
 977
 978 static void Cycriptify(const WebCore::String &source, int *psize = NULL) {
 979     if (!cycript_)
 980         return;
 981
 982     const uint16_t *data(source.characters());
 983     size_t length(source.length()), size(length);
 984
 985     apr_pool_t *pool;
 986     apr_pool_create(&pool, NULL);
 987
 988     Cycriptify(pool, data, size);
 989
 990     WebCore::String &script(const_cast<WebCore::String &>(source));
 991
 992     _ZN7WebCore6String8truncateEj(&script, 0);
 993     _ZN7WebCore6String6appendEPKtj(&script, data, size);
 994
 995     if (psize != NULL)
 996         *psize = size;
 997
 998     apr_pool_destroy(pool);
 999
1000     Log(source);
1001 }
1002 // }}}
1003
1004 extern "C" void *_ZN3JSC7UString3Rep14nullBaseStringE __attribute__((__weak_import__));
1005 extern "C" void *_ZN3JSC7UString3Rep7destroyEv __attribute__((__weak_import__));
1006 extern "C" void *_ZN3JSC7UStringC1EPKti __attribute__((__weak_import__));
1007 extern "C" void *_ZN3JSC7UStringC1EPKc __attribute__((__weak_import__));
1008 extern "C" void *_ZNK3JSC7UString6substrEii __attribute__((__weak_import__));
1009 extern "C" void *_ZN3WTF10fastMallocEm __attribute__((__weak_import__));
1010 extern "C" void WTFReportAssertionFailure(const char *, int, const char *, const char *) __attribute__((__weak_import__));
1011 extern "C" void *_ZN3WTF8fastFreeEPv __attribute__((__weak_import__));
1012
1013 bool CYWeakHell() {
1014     return
1015         &_ZN3JSC7UString3Rep14nullBaseStringE == NULL ||
1016         &_ZN3JSC7UString3Rep7destroyEv == NULL ||
1017         &_ZN3JSC7UStringC1EPKti == NULL ||
1018         &_ZN3JSC7UStringC1EPKc == NULL ||
1019         &_ZNK3JSC7UString6substrEii == NULL ||
1020         &_ZN3WTF10fastMallocEm == NULL ||
1021         &WTFReportAssertionFailure == NULL ||
1022         &_ZN3WTF8fastFreeEPv == NULL ||
1023     false;
1024 }
1025
1026 static WebCore::String *string;
1027
1028 MSHook(const WebCore::String &, _ZNK7WebCore20StringSourceProvider6sourceEv, void *_this) {
1029     _trace();
1030     const WebCore::String &source(__ZNK7WebCore20StringSourceProvider6sourceEv(_this));
1031     string = const_cast<WebCore::String *>(&source);
1032     return source;
1033 }
1034
1035 // iOS 2.x
1036 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, void *_this, const WebCore::String &string, State state, const WebCore::String &url, int line) {
1037     _trace();
1038     Cycriptify(string);
1039     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i(_this, string, state, url, line);
1040 }
1041
1042 // iOS 3.x
1043 MSHook(void, _ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, JSC::SourceCode **_this, JSC::JSGlobalData *global, int *line, JSC::UString *message) {
1044     if (cycript_) {
1045         JSC::SourceCode *source(*_this);
1046         const uint16_t *data(source->data());
1047         size_t size(source->length());
1048
1049         apr_pool_t *pool;
1050         apr_pool_create(&pool, NULL);
1051
1052         Cycriptify(pool, data, size);
1053         source->~SourceCode();
1054         // XXX: I actually don't have the original URL here: pants
1055         new (source) JSC::SourceCode(JSC::UStringSourceProvider::create(JSC::UString(data, size), "cycript://"), 1);
1056
1057         apr_pool_destroy(pool);
1058
1059     }
1060
1061     return __ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE(_this, global, line, message);
1062 }
1063
1064 // iOS 4.x cdata
1065 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, void *_this, const WebCore::String &source, const WebCore::KURL &url, int line) {
1066     _trace();
1067     Cycriptify(source);
1068     return __ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi(_this, source, url, line);
1069 }
1070
1071 // iOS 4.x @src=
1072 MSHook(const WebCore::String &, _ZN7WebCore12CachedScript6scriptEv, void *_this) {
1073     _trace();
1074     const WebCore::String &script(__ZN7WebCore12CachedScript6scriptEv(_this));
1075     string = const_cast<WebCore::String *>(&script);
1076     Log(script);
1077     return script;
1078 }
1079
1080 // iOS 4.x @src=
1081 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, void *_this, void *source, State state) {
1082     _trace();
1083     if (string != NULL) {
1084         if (iOS4)
1085             Cycriptify(*string, reinterpret_cast<int *>(source) + 3);
1086         else
1087             Cycriptify(*string);
1088     }
1089     string = NULL;
1090     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE(_this, source, state);
1091 }
1092
1093 /* Cydget:// Protocol {{{ */
1094 @interface CydgetURLProtocol : NSURLProtocol {
1095 }
1096
1097 @end
1098
1099 @implementation CydgetURLProtocol
1100
1101 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
1102     NSURL *url([request URL]);
1103     if (url == nil)
1104         return NO;
1105     NSString *scheme([[url scheme] lowercaseString]);
1106     if (scheme == nil || ![scheme isEqualToString:@"cydget"])
1107         return NO;
1108     return YES;
1109 }
1110
1111 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
1112     return request;
1113 }
1114
1115 - (void) _returnPNGWithImage:(UIImage *)icon forRequest:(NSURLRequest *)request {
1116     id<NSURLProtocolClient> client([self client]);
1117     if (icon == nil)
1118         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
1119     else {
1120         NSData *data(UIImagePNGRepresentation(icon));
1121
1122         NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:@"image/png" expectedContentLength:-1 textEncodingName:nil] autorelease]);
1123         [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1124         [client URLProtocol:self didLoadData:data];
1125         [client URLProtocolDidFinishLoading:self];
1126     }
1127 }
1128
1129 - (void) startLoading {
1130     id<NSURLProtocolClient> client([self client]);
1131     NSURLRequest *request([self request]);
1132
1133     NSURL *url([request URL]);
1134     NSString *href([url absoluteString]);
1135
1136     NSString *path([href substringFromIndex:9]);
1137     NSRange slash([path rangeOfString:@"/"]);
1138
1139     NSString *command;
1140     if (slash.location == NSNotFound) {
1141         command = path;
1142         path = nil;
1143     } else {
1144         command = [path substringToIndex:slash.location];
1145         path = [path substringFromIndex:(slash.location + 1)];
1146     }
1147
1148     if ([command isEqualToString:@"_UIImageWithName"]) {
1149         if (path == nil)
1150             goto fail;
1151         path = [path stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
1152         UIImage *icon(_UIImageWithName(path));
1153         [self _returnPNGWithImage:icon forRequest:request];
1154     } else fail: {
1155         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1156     }
1157 }
1158
1159 - (void) stopLoading {
1160 }
1161
1162 @end
1163 /* }}} */
1164 /* Cydget-CGI:// Protocol {{{ */
1165 @interface CydgetCGIURLProtocol : NSURLProtocol {
1166     pid_t pid_;
1167     CFHTTPMessageRef http_;
1168     NSFileHandle *handle_;
1169 }
1170
1171 @end
1172
1173 @implementation CydgetCGIURLProtocol
1174
1175 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
1176     NSURL *url([request URL]);
1177     if (url == nil)
1178         return NO;
1179     NSString *scheme([[url scheme] lowercaseString]);
1180     if (scheme == nil || ![scheme isEqualToString:@"cydget-cgi"])
1181         return NO;
1182     return YES;
1183 }
1184
1185 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
1186     return request;
1187 }
1188
1189 - (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)response client:(id<NSURLProtocolClient>)client {
1190     if ((self = [super initWithRequest:request cachedResponse:response client:client]) != nil) {
1191         pid_ = -1;
1192     } return self;
1193 }
1194
1195 - (void) startLoading {
1196     id<NSURLProtocolClient> client([self client]);
1197     NSURLRequest *request([self request]);
1198     NSURL *url([request URL]);
1199
1200     NSString *path([url path]);
1201     if (path == nil) {
1202         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1203         return;
1204     }
1205
1206     NSFileManager *manager([NSFileManager defaultManager]);
1207     if (![manager fileExistsAtPath:path]) {
1208         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
1209         return;
1210     }
1211
1212     int fds[2];
1213     _assert(pipe(fds) != -1);
1214
1215     _assert(pid_ == -1);
1216     pid_ = fork();
1217     if (pid_ == -1) {
1218         _assert(close(fds[0]) != -1);
1219         _assert(close(fds[1]) != -1);
1220         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1221         return;
1222     }
1223
1224     if (pid_ == 0) {
1225         const char *script([path UTF8String]);
1226
1227         setenv("GATEWAY_INTERFACE", "CGI/1.1", true);
1228         setenv("SCRIPT_FILENAME", script, true);
1229         NSString *query([url query]);
1230         if (query != nil)
1231             setenv("QUERY_STRING", [query UTF8String], true);
1232
1233         _assert(dup2(fds[1], 1) != -1);
1234         _assert(close(fds[0]) != -1);
1235         _assert(close(fds[1]) != -1);
1236
1237         execl(script, script, NULL);
1238         exit(1);
1239         _assert(false);
1240     }
1241
1242     _assert(close(fds[1]) != -1);
1243
1244     _assert(http_ == NULL);
1245     http_ = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, FALSE);
1246     CFHTTPMessageAppendBytes(http_, (const uint8_t *) "HTTP/1.1 200 OK\r\n", 17);
1247
1248     _assert(handle_ == nil);
1249     handle_ = [[NSFileHandle alloc] initWithFileDescriptor:fds[0] closeOnDealloc:YES];
1250
1251     [[NSNotificationCenter defaultCenter]
1252         addObserver:self
1253         selector:@selector(onRead:)
1254         name:@"NSFileHandleReadCompletionNotification"
1255         object:handle_
1256     ];
1257
1258     [handle_ readInBackgroundAndNotify];
1259 }
1260
1261 - (void) onRead:(NSNotification *)notification {
1262     NSFileHandle *handle([notification object]);
1263
1264     NSData *data([[notification userInfo] objectForKey:NSFileHandleNotificationDataItem]);
1265
1266     if (size_t length = [data length]) {
1267         CFHTTPMessageAppendBytes(http_, reinterpret_cast<const UInt8 *>([data bytes]), length);
1268         [handle readInBackgroundAndNotify];
1269     } else {
1270         id<NSURLProtocolClient> client([self client]);
1271
1272         CFStringRef mime(CFHTTPMessageCopyHeaderFieldValue(http_, CFSTR("Content-type")));
1273         if (mime == NULL)
1274             [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorBadServerResponse userInfo:nil]];
1275         else {
1276             NSURLRequest *request([self request]);
1277
1278             NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:(NSString *)mime expectedContentLength:-1 textEncodingName:nil] autorelease]);
1279             CFRelease(mime);
1280
1281             [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1282
1283             CFDataRef body(CFHTTPMessageCopyBody(http_));
1284             [client URLProtocol:self didLoadData:(NSData *)body];
1285             CFRelease(body);
1286
1287             [client URLProtocolDidFinishLoading:self];
1288         }
1289
1290         CFRelease(http_);
1291         http_ = NULL;
1292     }
1293 }
1294
1295     //[client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorNetworkConnectionLost userInfo:nil]];
1296
1297 - (void) stopLoading_ {
1298     [[NSNotificationCenter defaultCenter] removeObserver:self];
1299
1300     if (handle_ != nil) {
1301         [handle_ release];
1302         handle_ = nil;
1303     }
1304
1305     if (pid_ != -1) {
1306         kill(pid_, SIGTERM);
1307         int status;
1308         _syscall(waitpid(pid_, &status, 0));
1309         pid_ = -1;
1310     }
1311 }
1312
1313 - (void) stopLoading {
1314     [self
1315         performSelectorOnMainThread:@selector(stopLoading_)
1316         withObject:nil
1317         waitUntilDone:NO
1318     ];
1319 }
1320
1321 @end
1322 /* }}} */
1323
1324 template <typename Type_>
1325 static void nlset(Type_ &function, struct nlist *nl, size_t index) {
1326     struct nlist &name(nl[index]);
1327     uintptr_t value(name.n_value);
1328     if ((name.n_desc & N_ARM_THUMB_DEF) != 0)
1329         value |= 0x00000001;
1330     function = reinterpret_cast<Type_>(value);
1331 }
1332
1333 template <typename Type_>
1334 static void dlset(Type_ &function, const char *name) {
1335     function = reinterpret_cast<Type_>(dlsym(RTLD_DEFAULT, name));
1336 }
1337
1338 @implementation WebCycriptLockScreenController
1339
1340 + (void) initialize {
1341     iOS4 = kCFCoreFoundationVersionNumber >= 550.32;
1342
1343     $UIWebBrowserView = objc_getClass("UIWebBrowserView");
1344     if ($UIWebBrowserView == nil) {
1345         Wildcat_ = false;
1346         $UIWebBrowserView = objc_getClass("UIWebDocumentView");
1347     } else {
1348         Wildcat_ = true;
1349     }
1350
1351     int maxproc;
1352     size_t size(sizeof(maxproc));
1353     if (sysctlbyname("kern.maxproc", &maxproc, &size, NULL, 0) == -1)
1354         NSLog(@"sysctlbyname(\"kern.maxproc\", ?)");
1355     else if (maxproc < 72) {
1356         maxproc = 72;
1357         if (sysctlbyname("kern.maxproc", NULL, NULL, &maxproc, sizeof(maxproc)) == -1)
1358             NSLog(@"sysctlbyname(\"kern.maxproc\", #)");
1359     }
1360
1361     apr_initialize();
1362
1363     [NSURLProtocol registerClass:[CydgetURLProtocol class]];
1364     [NSURLProtocol registerClass:[CydgetCGIURLProtocol class]];
1365
1366     if (!iOS4) {
1367         void (*_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE)(JSC::SourceCode **, JSC::JSGlobalData *, int *, JSC::UString *);
1368         dlset(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, "_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE");
1369         if (_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE != NULL)
1370             MSHookFunction(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE, MSHake(_ZN3JSC6Parser5parseEPNS_12JSGlobalDataEPiPNS_7UStringE));
1371     }
1372
1373     struct nlist nl[9];
1374     memset(nl, 0, sizeof(nl));
1375
1376     nl[0].n_un.n_name = (char *) "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE";
1377
1378     nl[1].n_un.n_name = (char *) "__ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi";
1379
1380     nl[2].n_un.n_name = (char *) "__ZN7WebCore12CachedScript6scriptEv";
1381     nl[3].n_un.n_name = (char *) "__ZNK7WebCore20StringSourceProvider6sourceEv";
1382
1383     nl[4].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i";
1384     nl[5].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE";
1385
1386     nl[6].n_un.n_name = (char *) "__ZN7WebCore6String6appendEPKtj";
1387     nl[7].n_un.n_name = (char *) "__ZN7WebCore6String8truncateEj";
1388
1389     nlist("/System/Library/PrivateFrameworks/WebCore.framework/WebCore", nl);
1390
1391     bool (*_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE)(const WebCore::String &);
1392     nlset(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, nl, 0);
1393     if (_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE != NULL)
1394         MSHookFunction(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, MSHake(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE));
1395
1396     void (*_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi)(void *, const WebCore::String &, const WebCore::KURL &, int);
1397     nlset(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, nl, 1);
1398     if (_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi != NULL)
1399         MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi));
1400
1401     if (iOS4) {
1402         const WebCore::String &(*_ZN7WebCore12CachedScript6scriptEv)(void *);
1403         nlset(_ZN7WebCore12CachedScript6scriptEv, nl, 2);
1404         if (_ZN7WebCore12CachedScript6scriptEv != NULL)
1405             MSHookFunction(_ZN7WebCore12CachedScript6scriptEv, MSHake(_ZN7WebCore12CachedScript6scriptEv));
1406     }
1407
1408     /*const WebCore::String &(*_ZNK7WebCore20StringSourceProvider6sourceEv)(void *);
1409     nlset(_ZNK7WebCore20StringSourceProvider6sourceEv, nl, 3);
1410     MSHookFunction(_ZNK7WebCore20StringSourceProvider6sourceEv, MSHake(_ZNK7WebCore20StringSourceProvider6sourceEv));*/
1411
1412     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i)(void *, const WebCore::String &, State, const WebCore::String &, int);
1413     nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, nl, 4);
1414     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i != NULL)
1415         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i));
1416
1417     if (iOS4) {
1418         State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE)(void *, void *, State);
1419         nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, nl, 5);
1420         if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE != NULL)
1421             MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE));
1422     }
1423
1424     nlset(_ZN7WebCore6String6appendEPKtj, nl, 6);
1425     nlset(_ZN7WebCore6String8truncateEj, nl, 7);
1426 }
1427
1428 + (id) rootViewController {
1429     return [[[self alloc] init] autorelease];
1430 }
1431
1432 - (void) loadView {
1433     [self setView:[[[WebCydgetLockScreenView alloc] init] autorelease]];
1434 }
1435
1436 @end