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         return __ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE(mime);
 945     _trace();
 946
 947     static void *handle(dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL));
 948     if (handle == NULL)
 949         return false;
 950
 951     cycript_ = true;
 952     return true;
 953 }
 954 // }}}
 955 // Script Compiler {{{
 956 static void Log(const WebCore::String &string) {
 957     size_t length(string.length());
 958     UChar data[length + 1];
 959     data[length] = 0;
 960     memcpy(data, string.characters(), length * 2);
 961     NSLog(@"wtf %p:%S:", &string, data);
 962 }
 963
 964 static void Cycriptify(apr_pool_t *pool, const uint16_t *&data, size_t &size) {
 965     _trace();
 966     if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_GLOBAL))
 967         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")))
 968             CYParseUChar(pool, &data, &size);
 969 }
 970
 971 static void (*_ZN7WebCore6String6appendEPKtj)(WebCore::String *, const UChar *, unsigned);
 972 static void (*_ZN7WebCore6String8truncateEj)(WebCore::String *, unsigned);
 973
 974 static void Cycriptify(const WebCore::String &source, int *psize = NULL) {
 975     if (!cycript_)
 976         return;
 977     cycript_ = false;
 978
 979     const uint16_t *data(source.characters());
 980     size_t length(source.length()), size(length);
 981
 982     apr_pool_t *pool;
 983     apr_pool_create(&pool, NULL);
 984
 985     Cycriptify(pool, data, size);
 986
 987     WebCore::String &script(const_cast<WebCore::String &>(source));
 988
 989     _ZN7WebCore6String8truncateEj(&script, 0);
 990     _ZN7WebCore6String6appendEPKtj(&script, data, size);
 991
 992     if (psize != NULL)
 993         *psize = size;
 994
 995     apr_pool_destroy(pool);
 996
 997     Log(source);
 998 }
 999 // }}}
1000
1001 extern "C" void *_ZN3JSC7UString3Rep14nullBaseStringE __attribute__((__weak_import__));
1002 extern "C" void *_ZN3JSC7UString3Rep7destroyEv __attribute__((__weak_import__));
1003 extern "C" void *_ZN3JSC7UStringC1EPKti __attribute__((__weak_import__));
1004 extern "C" void *_ZN3JSC7UStringC1EPKc __attribute__((__weak_import__));
1005 extern "C" void *_ZNK3JSC7UString6substrEii __attribute__((__weak_import__));
1006 extern "C" void *_ZN3WTF10fastMallocEm __attribute__((__weak_import__));
1007 extern "C" void WTFReportAssertionFailure(const char *, int, const char *, const char *) __attribute__((__weak_import__));
1008 extern "C" void *_ZN3WTF8fastFreeEPv __attribute__((__weak_import__));
1009
1010 bool CYWeakHell() {
1011     return
1012         &_ZN3JSC7UString3Rep14nullBaseStringE == NULL ||
1013         &_ZN3JSC7UString3Rep7destroyEv == NULL ||
1014         &_ZN3JSC7UStringC1EPKti == NULL ||
1015         &_ZN3JSC7UStringC1EPKc == NULL ||
1016         &_ZNK3JSC7UString6substrEii == NULL ||
1017         &_ZN3WTF10fastMallocEm == NULL ||
1018         &WTFReportAssertionFailure == NULL ||
1019         &_ZN3WTF8fastFreeEPv == NULL ||
1020     false;
1021 }
1022
1023 static WebCore::String *string;
1024
1025 MSHook(const WebCore::String &, _ZNK7WebCore20StringSourceProvider6sourceEv, void *_this) {
1026     _trace();
1027     const WebCore::String &source(__ZNK7WebCore20StringSourceProvider6sourceEv(_this));
1028     string = const_cast<WebCore::String *>(&source);
1029     return source;
1030 }
1031
1032 // iOS 2.x
1033 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, void *_this, const WebCore::String &string, State state, const WebCore::String &url, int line) {
1034     _trace();
1035     Cycriptify(string);
1036     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i(_this, string, state, url, line);
1037 }
1038
1039 // iOS 4.x cdata
1040 MSHook(void, _ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, void *_this, const WebCore::String &source, const WebCore::KURL &url, int line) {
1041     _trace();
1042     Cycriptify(source);
1043     return __ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi(_this, source, url, line);
1044 }
1045
1046 // iOS 4.x @src=
1047 MSHook(const WebCore::String &, _ZN7WebCore12CachedScript6scriptEv, void *_this) {
1048     _trace();
1049     const WebCore::String &script(__ZN7WebCore12CachedScript6scriptEv(_this));
1050     string = const_cast<WebCore::String *>(&script);
1051     Log(script);
1052     return script;
1053 }
1054
1055 // iOS 4.x @src=
1056 MSHook(State, _ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, void *_this, void *source, State state) {
1057     _trace();
1058     if (string != NULL) {
1059         if (iOS4)
1060             Cycriptify(*string, reinterpret_cast<int *>(source) + 3);
1061         else
1062             Cycriptify(*string);
1063     }
1064     string = NULL;
1065     return __ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE(_this, source, state);
1066 }
1067
1068 /* Cydget:// Protocol {{{ */
1069 @interface CydgetURLProtocol : NSURLProtocol {
1070 }
1071
1072 @end
1073
1074 @implementation CydgetURLProtocol
1075
1076 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
1077     NSURL *url([request URL]);
1078     if (url == nil)
1079         return NO;
1080     NSString *scheme([[url scheme] lowercaseString]);
1081     if (scheme == nil || ![scheme isEqualToString:@"cydget"])
1082         return NO;
1083     return YES;
1084 }
1085
1086 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
1087     return request;
1088 }
1089
1090 - (void) _returnPNGWithImage:(UIImage *)icon forRequest:(NSURLRequest *)request {
1091     id<NSURLProtocolClient> client([self client]);
1092     if (icon == nil)
1093         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
1094     else {
1095         NSData *data(UIImagePNGRepresentation(icon));
1096
1097         NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:@"image/png" expectedContentLength:-1 textEncodingName:nil] autorelease]);
1098         [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1099         [client URLProtocol:self didLoadData:data];
1100         [client URLProtocolDidFinishLoading:self];
1101     }
1102 }
1103
1104 - (void) startLoading {
1105     id<NSURLProtocolClient> client([self client]);
1106     NSURLRequest *request([self request]);
1107
1108     NSURL *url([request URL]);
1109     NSString *href([url absoluteString]);
1110
1111     NSString *path([href substringFromIndex:9]);
1112     NSRange slash([path rangeOfString:@"/"]);
1113
1114     NSString *command;
1115     if (slash.location == NSNotFound) {
1116         command = path;
1117         path = nil;
1118     } else {
1119         command = [path substringToIndex:slash.location];
1120         path = [path substringFromIndex:(slash.location + 1)];
1121     }
1122
1123     if ([command isEqualToString:@"_UIImageWithName"]) {
1124         if (path == nil)
1125             goto fail;
1126         path = [path stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
1127         UIImage *icon(_UIImageWithName(path));
1128         [self _returnPNGWithImage:icon forRequest:request];
1129     } else fail: {
1130         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1131     }
1132 }
1133
1134 - (void) stopLoading {
1135 }
1136
1137 @end
1138 /* }}} */
1139 /* Cydget-CGI:// Protocol {{{ */
1140 @interface CydgetCGIURLProtocol : NSURLProtocol {
1141     pid_t pid_;
1142     CFHTTPMessageRef http_;
1143     NSFileHandle *handle_;
1144 }
1145
1146 @end
1147
1148 @implementation CydgetCGIURLProtocol
1149
1150 + (BOOL) canInitWithRequest:(NSURLRequest *)request {
1151     NSURL *url([request URL]);
1152     if (url == nil)
1153         return NO;
1154     NSString *scheme([[url scheme] lowercaseString]);
1155     if (scheme == nil || ![scheme isEqualToString:@"cydget-cgi"])
1156         return NO;
1157     return YES;
1158 }
1159
1160 + (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
1161     return request;
1162 }
1163
1164 - (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)response client:(id<NSURLProtocolClient>)client {
1165     if ((self = [super initWithRequest:request cachedResponse:response client:client]) != nil) {
1166         pid_ = -1;
1167     } return self;
1168 }
1169
1170 - (void) startLoading {
1171     id<NSURLProtocolClient> client([self client]);
1172     NSURLRequest *request([self request]);
1173     NSURL *url([request URL]);
1174
1175     NSString *path([url path]);
1176     if (path == nil) {
1177         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1178         return;
1179     }
1180
1181     NSFileManager *manager([NSFileManager defaultManager]);
1182     if (![manager fileExistsAtPath:path]) {
1183         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil]];
1184         return;
1185     }
1186
1187     int fds[2];
1188     _assert(pipe(fds) != -1);
1189
1190     _assert(pid_ == -1);
1191     pid_ = fork();
1192     if (pid_ == -1) {
1193         _assert(close(fds[0]) != -1);
1194         _assert(close(fds[1]) != -1);
1195         [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorResourceUnavailable userInfo:nil]];
1196         return;
1197     }
1198
1199     if (pid_ == 0) {
1200         const char *script([path UTF8String]);
1201
1202         setenv("GATEWAY_INTERFACE", "CGI/1.1", true);
1203         setenv("SCRIPT_FILENAME", script, true);
1204         NSString *query([url query]);
1205         if (query != nil)
1206             setenv("QUERY_STRING", [query UTF8String], true);
1207
1208         _assert(dup2(fds[1], 1) != -1);
1209         _assert(close(fds[0]) != -1);
1210         _assert(close(fds[1]) != -1);
1211
1212         execl(script, script, NULL);
1213         exit(1);
1214         _assert(false);
1215     }
1216
1217     _assert(close(fds[1]) != -1);
1218
1219     _assert(http_ == NULL);
1220     http_ = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, FALSE);
1221     CFHTTPMessageAppendBytes(http_, (const uint8_t *) "HTTP/1.1 200 OK\r\n", 17);
1222
1223     _assert(handle_ == nil);
1224     handle_ = [[NSFileHandle alloc] initWithFileDescriptor:fds[0] closeOnDealloc:YES];
1225
1226     [[NSNotificationCenter defaultCenter]
1227         addObserver:self
1228         selector:@selector(onRead:)
1229         name:@"NSFileHandleReadCompletionNotification"
1230         object:handle_
1231     ];
1232
1233     [handle_ readInBackgroundAndNotify];
1234 }
1235
1236 - (void) onRead:(NSNotification *)notification {
1237     NSFileHandle *handle([notification object]);
1238
1239     NSData *data([[notification userInfo] objectForKey:NSFileHandleNotificationDataItem]);
1240
1241     if (size_t length = [data length]) {
1242         CFHTTPMessageAppendBytes(http_, reinterpret_cast<const UInt8 *>([data bytes]), length);
1243         [handle readInBackgroundAndNotify];
1244     } else {
1245         id<NSURLProtocolClient> client([self client]);
1246
1247         CFStringRef mime(CFHTTPMessageCopyHeaderFieldValue(http_, CFSTR("Content-type")));
1248         if (mime == NULL)
1249             [client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorBadServerResponse userInfo:nil]];
1250         else {
1251             NSURLRequest *request([self request]);
1252
1253             NSURLResponse *response([[[NSURLResponse alloc] initWithURL:[request URL] MIMEType:(NSString *)mime expectedContentLength:-1 textEncodingName:nil] autorelease]);
1254             CFRelease(mime);
1255
1256             [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
1257
1258             CFDataRef body(CFHTTPMessageCopyBody(http_));
1259             [client URLProtocol:self didLoadData:(NSData *)body];
1260             CFRelease(body);
1261
1262             [client URLProtocolDidFinishLoading:self];
1263         }
1264
1265         CFRelease(http_);
1266         http_ = NULL;
1267     }
1268 }
1269
1270     //[client URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorNetworkConnectionLost userInfo:nil]];
1271
1272 - (void) stopLoading_ {
1273     [[NSNotificationCenter defaultCenter] removeObserver:self];
1274
1275     if (handle_ != nil) {
1276         [handle_ release];
1277         handle_ = nil;
1278     }
1279
1280     if (pid_ != -1) {
1281         kill(pid_, SIGTERM);
1282         int status;
1283         _syscall(waitpid(pid_, &status, 0));
1284         pid_ = -1;
1285     }
1286 }
1287
1288 - (void) stopLoading {
1289     [self
1290         performSelectorOnMainThread:@selector(stopLoading_)
1291         withObject:nil
1292         waitUntilDone:NO
1293     ];
1294 }
1295
1296 @end
1297 /* }}} */
1298
1299 template <typename Type_>
1300 static void nlset(Type_ &function, struct nlist *nl, size_t index) {
1301     struct nlist &name(nl[index]);
1302     uintptr_t value(name.n_value);
1303     if ((name.n_desc & N_ARM_THUMB_DEF) != 0)
1304         value |= 0x00000001;
1305     function = reinterpret_cast<Type_>(value);
1306 }
1307
1308 template <typename Type_>
1309 static void dlset(Type_ &function, const char *name) {
1310     function = reinterpret_cast<Type_>(dlsym(RTLD_DEFAULT, name));
1311 }
1312
1313 @implementation WebCycriptLockScreenController
1314
1315 + (void) initialize {
1316     iOS4 = kCFCoreFoundationVersionNumber >= 550.32;
1317
1318     $UIWebBrowserView = objc_getClass("UIWebBrowserView");
1319     if ($UIWebBrowserView == nil) {
1320         Wildcat_ = false;
1321         $UIWebBrowserView = objc_getClass("UIWebDocumentView");
1322     } else {
1323         Wildcat_ = true;
1324     }
1325
1326     int maxproc;
1327     size_t size(sizeof(maxproc));
1328     if (sysctlbyname("kern.maxproc", &maxproc, &size, NULL, 0) == -1)
1329         NSLog(@"sysctlbyname(\"kern.maxproc\", ?)");
1330     else if (maxproc < 72) {
1331         maxproc = 72;
1332         if (sysctlbyname("kern.maxproc", NULL, NULL, &maxproc, sizeof(maxproc)) == -1)
1333             NSLog(@"sysctlbyname(\"kern.maxproc\", #)");
1334     }
1335
1336     apr_initialize();
1337
1338     [NSURLProtocol registerClass:[CydgetURLProtocol class]];
1339     [NSURLProtocol registerClass:[CydgetCGIURLProtocol class]];
1340
1341     struct nlist nl[9];
1342     memset(nl, 0, sizeof(nl));
1343
1344     nl[0].n_un.n_name = (char *) "__ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE";
1345
1346     nl[1].n_un.n_name = (char *) "__ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi";
1347
1348     nl[2].n_un.n_name = (char *) "__ZN7WebCore12CachedScript6scriptEv";
1349     nl[3].n_un.n_name = (char *) "__ZNK7WebCore20StringSourceProvider6sourceEv";
1350
1351     nl[4].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i";
1352     nl[5].n_un.n_name = (char *) "__ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE";
1353
1354     nl[6].n_un.n_name = (char *) "__ZN7WebCore6String6appendEPKtj";
1355     nl[7].n_un.n_name = (char *) "__ZN7WebCore6String8truncateEj";
1356
1357     nlist("/System/Library/PrivateFrameworks/WebCore.framework/WebCore", nl);
1358
1359     bool (*_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE)(const WebCore::String &);
1360     nlset(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, nl, 0);
1361     MSHookFunction(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE, MSHake(_ZN7WebCore16MIMETypeRegistry29isSupportedJavaScriptMIMETypeERKNS_6StringE));
1362
1363     void (*_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi)(void *, const WebCore::String &, const WebCore::KURL &, int);
1364     nlset(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, nl, 1);
1365     MSHookFunction(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi, MSHake(_ZN7WebCore16ScriptSourceCodeC2ERKNS_6StringERKNS_4KURLEi));
1366
1367     const WebCore::String &(*_ZN7WebCore12CachedScript6scriptEv)(void *);
1368     nlset(_ZN7WebCore12CachedScript6scriptEv, nl, 2);
1369     MSHookFunction(_ZN7WebCore12CachedScript6scriptEv, MSHake(_ZN7WebCore12CachedScript6scriptEv));
1370
1371     /*const WebCore::String &(*_ZNK7WebCore20StringSourceProvider6sourceEv)(void *);
1372     nlset(_ZNK7WebCore20StringSourceProvider6sourceEv, nl, 3);
1373     MSHookFunction(_ZNK7WebCore20StringSourceProvider6sourceEv, MSHake(_ZNK7WebCore20StringSourceProvider6sourceEv));*/
1374
1375     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i)(void *, const WebCore::String &, State, const WebCore::String &, int);
1376     nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, nl, 4);
1377     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i != NULL)
1378         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_6StringENS0_5StateES3_i));
1379
1380     State (*_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE)(void *, void *, State);
1381     nlset(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, nl, 5);
1382     if (_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE != NULL)
1383         MSHookFunction(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE, MSHake(_ZN7WebCore13HTMLTokenizer15scriptExecutionERKNS_16ScriptSourceCodeENS0_5StateE));
1384
1385     nlset(_ZN7WebCore6String6appendEPKtj, nl, 6);
1386     nlset(_ZN7WebCore6String8truncateEj, nl, 7);
1387 }
1388
1389 + (id) rootViewController {
1390     return [[[self alloc] init] autorelease];
1391 }
1392
1393 - (void) loadView {
1394     [self setView:[[[WebCydgetLockScreenView alloc] init] autorelease]];
1395 }
1396
1397 @end