X-Git-Url: https://git.saurik.com/veency.git/blobdiff_plain/55e20baf092464e45be96ae8d8f9793b92791c4c..1c6eb1765c7dac2819bfe7fe48ef619edbc1b792:/Tweak.mm diff --git a/Tweak.mm b/Tweak.mm index 852ad7e..1b99b5b 100644 --- a/Tweak.mm +++ b/Tweak.mm @@ -1,42 +1,28 @@ /* Veency - VNC Remote Access Server for iPhoneOS - * Copyright (C) 2008-2010 Jay Freeman (saurik) + * Copyright (C) 2008-2012 Jay Freeman (saurik) */ +/* GNU Affero General Public License, Version 3 {{{ */ /* - * Redistribution and use in source and binary - * forms, with or without modification, are permitted - * provided that the following conditions are met: - * - * 1. Redistributions of source code must retain the - * above copyright notice, this list of conditions - * and the following disclaimer. - * 2. Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions - * and the following disclaimer in the documentation - * and/or other materials provided with the - * distribution. - * 3. The name of the author may not be used to endorse - * or promote products derived from this software - * without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' - * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, - * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF - * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE - * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, - * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT - * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR - * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS - * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF - * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR - * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN - * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF - * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -*/ + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . +**/ +/* }}} */ #define _trace() \ fprintf(stderr, "_trace()@%s:%u[%s]\n", __FILE__, __LINE__, __FUNCTION__) +#define _likely(expr) \ + __builtin_expect(expr, 1) #define _unlikely(expr) \ __builtin_expect(expr, 0) @@ -45,28 +31,144 @@ #include #include -#include +#include +#include -#import -#import +#include +#include +#undef assert + +#include #import #import #import -#import -#import #import -#import -#import -#import +#include +#include + +extern "C" { +#include "SpringBoardAccess.h" +} + +MSClassHook(BKAccessibility) +MSClassHook(UIApplication) + +@interface UIApplication (Apple) +- (void) addStatusBarImageNamed:(NSString *)name; +- (void) removeStatusBarImageNamed:(NSString *)name; +@end + +@interface CAWindowServerDisplay : NSObject +- (mach_port_t) clientPortAtPosition:(CGPoint)position; +- (unsigned) contextIdAtPosition:(CGPoint)position; +- (mach_port_t) taskPortOfContextId:(unsigned)context; +@end + +@interface CAWindowServer : NSObject ++ (CAWindowServer *) serverIfRunning; +- (NSArray *) displays; +@end + +@interface UIModalView : UIView +- (id) addButtonWithTitle:(NSString *)title; +- (void) setBodyText:(NSString *)text; +- (void) setDelegate:(id)delegate; +- (void) setTitle:(NSString *)title; +@end + +@interface SBAlertItem : NSObject +- (void) dismiss; +- (UIModalView *) alertSheet; +@end + +@interface SBAlertItemsController : NSObject ++ (SBAlertItemsController *) sharedInstance; +- (void) activateAlertItem:(SBAlertItem *)item; +@end + +@interface SBStatusBarController : NSObject ++ (SBStatusBarController *) sharedStatusBarController; +- (void) addStatusBarItem:(NSString *)item; +- (void) removeStatusBarItem:(NSString *)item; +@end + +@interface BKHIDClientConnectionManager : NSObject +- () clientForTaskPort:(mach_port_t)port; +@end + +@interface BKAccessibility : NSObject ++ (BKHIDClientConnectionManager *) _eventRoutingClientConnectionManager; +@end + +typedef void *CoreSurfaceBufferRef; + +extern CFStringRef kCoreSurfaceBufferGlobal; +extern CFStringRef kCoreSurfaceBufferMemoryRegion; +extern CFStringRef kCoreSurfaceBufferPitch; +extern CFStringRef kCoreSurfaceBufferWidth; +extern CFStringRef kCoreSurfaceBufferHeight; +extern CFStringRef kCoreSurfaceBufferPixelFormat; +extern CFStringRef kCoreSurfaceBufferAllocSize; + +extern "C" CoreSurfaceBufferRef CoreSurfaceBufferCreate(CFDictionaryRef dict); +extern "C" int CoreSurfaceBufferLock(CoreSurfaceBufferRef surface, unsigned int lockType); +extern "C" int CoreSurfaceBufferUnlock(CoreSurfaceBufferRef surface); +extern "C" void *CoreSurfaceBufferGetBaseAddress(CoreSurfaceBufferRef surface); + +extern "C" void CoreSurfaceBufferFlushProcessorCaches(CoreSurfaceBufferRef buffer); + +typedef void *CoreSurfaceAcceleratorRef; + +extern "C" int CoreSurfaceAcceleratorCreate(CFAllocatorRef allocator, void *type, CoreSurfaceAcceleratorRef *accel); +extern "C" unsigned int CoreSurfaceAcceleratorTransferSurface(CoreSurfaceAcceleratorRef accelerator, CoreSurfaceBufferRef dest, CoreSurfaceBufferRef src, CFDictionaryRef options/*, void *, void *, void **/); + +typedef void *IOMobileFramebufferRef; + +extern "C" kern_return_t IOMobileFramebufferSwapSetLayer( + IOMobileFramebufferRef fb, + int layer, + CoreSurfaceBufferRef buffer, + CGRect bounds, + CGRect frame, + int flags +); + +extern "C" void IOMobileFramebufferGetDisplaySize(IOMobileFramebufferRef connect, CGSize *size); +extern "C" void IOMobileFramebufferIsMainDisplay(IOMobileFramebufferRef connect, int *main); + +typedef CFTypeRef IOHIDEventRef; +typedef CFTypeRef IOHIDEventSystemClientRef; +typedef CFTypeRef IOHIDEventSystemConnectionRef; + +extern "C" { + IOHIDEventRef IOHIDEventCreateKeyboardEvent(CFAllocatorRef allocator, uint64_t time, uint16_t page, uint16_t usage, Boolean down, IOHIDEventOptionBits flags); + + IOHIDEventRef IOHIDEventCreateDigitizerEvent(CFAllocatorRef allocator, uint64_t timeStamp, IOHIDDigitizerTransducerType type, uint32_t index, uint32_t identity, uint32_t eventMask, uint32_t buttonMask, IOHIDFloat x, IOHIDFloat y, IOHIDFloat z, IOHIDFloat tipPressure, IOHIDFloat barrelPressure, Boolean range, Boolean touch, IOOptionBits options); + IOHIDEventRef IOHIDEventCreateDigitizerFingerEvent(CFAllocatorRef allocator, uint64_t timeStamp, uint32_t index, uint32_t identity, uint32_t eventMask, IOHIDFloat x, IOHIDFloat y, IOHIDFloat z, IOHIDFloat tipPressure, IOHIDFloat twist, Boolean range, Boolean touch, IOOptionBits options); -static size_t Width = 320; -static size_t Height = 480; + IOHIDEventSystemClientRef IOHIDEventSystemClientCreate(CFAllocatorRef allocator); + + void IOHIDEventAppendEvent(IOHIDEventRef parent, IOHIDEventRef child); + void IOHIDEventSetIntegerValue(IOHIDEventRef event, IOHIDEventField field, int value); + void IOHIDEventSetSenderID(IOHIDEventRef event, uint64_t sender); + + void IOHIDEventSystemClientDispatchEvent(IOHIDEventSystemClientRef client, IOHIDEventRef event); + void IOHIDEventSystemConnectionDispatchEvent(IOHIDEventSystemConnectionRef connection, IOHIDEventRef event); +} + +static size_t width_; +static size_t height_; +static NSUInteger ratio_ = 0; static const size_t BytesPerPixel = 4; static const size_t BitsPerSample = 8; +static CoreSurfaceAcceleratorRef accelerator_; +static CoreSurfaceBufferRef buffer_; +static CFDictionaryRef options_; + static NSMutableSet *handlers_; static rfbScreenInfoPtr screen_; static bool running_; @@ -78,10 +180,25 @@ static unsigned clients_; static CFMessagePortRef ashikase_; static bool cursor_; +static rfbPixel *black_; + +static void VNCBlack() { + if (_unlikely(black_ == NULL)) + black_ = reinterpret_cast(mmap(NULL, sizeof(rfbPixel) * width_ * height_, PROT_READ, MAP_ANON | MAP_PRIVATE | MAP_NOCACHE, VM_FLAGS_PURGABLE, 0)); + screen_->frameBuffer = reinterpret_cast(black_); +} + static bool Ashikase(bool always) { + if (!always && !cursor_) + return false; + if (ashikase_ == NULL) ashikase_ = CFMessagePortCreateRemote(kCFAllocatorDefault, CFSTR("jp.ashikase.mousesupport")); - return ashikase_ != NULL && (always || cursor_); + if (ashikase_ != NULL) + return true; + + cursor_ = false; + return false; } static CFDataRef cfTrue_; @@ -120,18 +237,45 @@ static void AshikaseSetEnabled(bool enabled, bool always) { AshikaseSendEvent(x_, y_); } +MSClassHook(SBAlertItem) MSClassHook(SBAlertItemsController) MSClassHook(SBStatusBarController) -@class VNCAlertItem; +@interface VNCAlertItem : SBAlertItem +@end + static Class $VNCAlertItem; -static rfbNewClientAction action_ = RFB_CLIENT_ON_HOLD; +static NSString *DialogTitle(@"Remote Access Request"); +static NSString *DialogFormat(@"Accept connection from\n%s?\n\nVeency VNC Server\nby Jay Freeman (saurik)\nsaurik@saurik.com\nhttp://www.saurik.com/\n\nSet a VNC password in Settings!"); +static NSString *DialogAccept(@"Accept"); +static NSString *DialogReject(@"Reject"); + +static volatile rfbNewClientAction action_ = RFB_CLIENT_ON_HOLD; static NSCondition *condition_; static NSLock *lock_; static rfbClientPtr client_; +static void VNCSetup(); +static void VNCEnabled(); + +float (*$GSMainScreenScaleFactor)(); + +static void OnUserNotification(CFUserNotificationRef notification, CFOptionFlags flags) { + [condition_ lock]; + + if ((flags & 0x3) == 1) + action_ = RFB_CLIENT_ACCEPT; + else + action_ = RFB_CLIENT_REFUSE; + + [condition_ signal]; + [condition_ unlock]; + + CFRelease(notification); +} + @interface VNCBridge : NSObject { } @@ -144,18 +288,73 @@ static rfbClientPtr client_; @implementation VNCBridge + (void) askForConnection { - [[$SBAlertItemsController sharedInstance] activateAlertItem:[[[$VNCAlertItem alloc] init] autorelease]]; + if ($VNCAlertItem != nil) { + [[$SBAlertItemsController sharedInstance] activateAlertItem:[[[$VNCAlertItem alloc] init] autorelease]]; + return; + } + + SInt32 error; + CFUserNotificationRef notification(CFUserNotificationCreate(kCFAllocatorDefault, 0, kCFUserNotificationPlainAlertLevel, &error, (CFDictionaryRef) [NSDictionary dictionaryWithObjectsAndKeys: + DialogTitle, kCFUserNotificationAlertHeaderKey, + [NSString stringWithFormat:DialogFormat, client_->host], kCFUserNotificationAlertMessageKey, + DialogAccept, kCFUserNotificationAlternateButtonTitleKey, + DialogReject, kCFUserNotificationDefaultButtonTitleKey, + nil])); + + if (error != 0) { + CFRelease(notification); + notification = NULL; + } + + if (notification == NULL) { + [condition_ lock]; + action_ = RFB_CLIENT_REFUSE; + [condition_ signal]; + [condition_ unlock]; + return; + } + + CFRunLoopSourceRef source(CFUserNotificationCreateRunLoopSource(kCFAllocatorDefault, notification, &OnUserNotification, 0)); + CFRunLoopAddSource(CFRunLoopGetCurrent(), source, kCFRunLoopDefaultMode); } + (void) removeStatusBarItem { AshikaseSetEnabled(false, false); - [[$SBStatusBarController sharedStatusBarController] removeStatusBarItem:@"Veency"]; + + if (SBA_available()) + SBA_removeStatusBarImage(const_cast("Veency")); + else if ($SBStatusBarController != nil) + [[$SBStatusBarController sharedStatusBarController] removeStatusBarItem:@"Veency"]; + else if (UIApplication *app = [$UIApplication sharedApplication]) + [app removeStatusBarImageNamed:@"Veency"]; } + (void) registerClient { + // XXX: this could find a better home + if (ratio_ == 0) { + if ($GSMainScreenScaleFactor == NULL) + ratio_ = 1.0f; + else + ratio_ = $GSMainScreenScaleFactor(); + } + ++clients_; AshikaseSetEnabled(true, false); - [[$SBStatusBarController sharedStatusBarController] addStatusBarItem:@"Veency"]; + + if (SBA_available()) + SBA_addStatusBarImage(const_cast("Veency")); + else if ($SBStatusBarController != nil) + [[$SBStatusBarController sharedStatusBarController] addStatusBarItem:@"Veency"]; + else if (UIApplication *app = [$UIApplication sharedApplication]) + [app addStatusBarImageNamed:@"Veency"]; +} + ++ (void) performSetup:(NSThread *)thread { + NSAutoreleasePool *pool([[NSAutoreleasePool alloc] init]); + [thread autorelease]; + VNCSetup(); + VNCEnabled(); + [pool release]; } @end @@ -185,10 +384,10 @@ MSInstanceMessage2(void, VNCAlertItem, alertSheet,buttonClicked, id, sheet, int, MSInstanceMessage2(void, VNCAlertItem, configure,requirePasscodeForActions, BOOL, configure, BOOL, require) { UIModalView *sheet([self alertSheet]); [sheet setDelegate:self]; - [sheet setTitle:@"Remote Access Request"]; - [sheet setBodyText:[NSString stringWithFormat:@"Accept connection from\n%s?\n\nVeency VNC Server\nby Jay Freeman (saurik)\nsaurik@saurik.com\nhttp://www.saurik.com/\n\nSet a VNC password in Settings!", client_->host]]; - [sheet addButtonWithTitle:@"Accept"]; - [sheet addButtonWithTitle:@"Reject"]; + [sheet setTitle:DialogTitle]; + [sheet setBodyText:[NSString stringWithFormat:DialogFormat, client_->host]]; + [sheet addButtonWithTitle:DialogAccept]; + [sheet addButtonWithTitle:DialogReject]; } MSInstanceMessage0(void, VNCAlertItem, performUnlockAction) { @@ -197,10 +396,10 @@ MSInstanceMessage0(void, VNCAlertItem, performUnlockAction) { static mach_port_t (*GSTakePurpleSystemEventPort)(void); static bool PurpleAllocated; -static bool Two_; +static int Level_; static void FixRecord(GSEventRecord *record) { - if (Two_) + if (Level_ < 1) memmove(&record->windowContextId, &record->windowContextId + 1, sizeof(*record) - (reinterpret_cast(&record->windowContextId + 1) - reinterpret_cast(record)) + record->size); } @@ -246,13 +445,48 @@ static void VNCNotifySettings( static rfbBool VNCCheck(rfbClientPtr client, const char *data, int size) { @synchronized (condition_) { if (NSString *password = reinterpret_cast(screen_->authPasswdData)) { + NSAutoreleasePool *pool([[NSAutoreleasePool alloc] init]); rfbEncryptBytes(client->authChallenge, const_cast([password UTF8String])); - return memcmp(client->authChallenge, data, size) == 0; + bool good(memcmp(client->authChallenge, data, size) == 0); + [pool release]; + return good; } return TRUE; } } +static bool iPad1_; + +struct VeencyEvent { + struct GSEventRecord record; + struct { + struct GSEventRecordInfo info; + struct GSPathInfo path; + } data; +}; + +static void VNCPointerOld(int buttons, int x, int y, CGPoint location, int diff, bool twas, bool tis); +static void VNCPointerNew(int buttons, int x, int y, CGPoint location, int diff, bool twas, bool tis); + static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { + if (ratio_ == 0) + return; + + CGPoint location = {x, y}; + + if (width_ > height_) { + int t(x); + x = height_ - 1 - y; + y = t; + + if (!iPad1_) { + x = height_ - 1 - x; + y = width_ - 1 - y; + } + } + + x /= ratio_; + y /= ratio_; + x_ = x; y_ = y; int diff = buttons_ ^ buttons; bool twas((buttons_ & 0x1) != 0); @@ -266,6 +500,13 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { return; } + if (kCFCoreFoundationVersionNumber >= 800) + return VNCPointerNew(buttons, x, y, location, diff, twas, tis); + else + return VNCPointerOld(buttons, x, y, location, diff, twas, tis); +} + +static void VNCPointerOld(int buttons, int x, int y, CGPoint location, int diff, bool twas, bool tis) { mach_port_t purple(0); if ((diff & 0x10) != 0) { @@ -273,7 +514,7 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { memset(&record, 0, sizeof(record)); - record.type = (buttons & 0x4) != 0 ? + record.type = (buttons & 0x10) != 0 ? GSEventTypeHeadsetButtonDown : GSEventTypeHeadsetButtonUp; @@ -288,7 +529,7 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { memset(&record, 0, sizeof(record)); - record.type = (buttons & 0x4) != 0 ? + record.type = (buttons & 0x04) != 0 ? GSEventTypeMenuButtonDown : GSEventTypeMenuButtonUp; @@ -303,7 +544,7 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { memset(&record, 0, sizeof(record)); - record.type = (buttons & 0x2) != 0 ? + record.type = (buttons & 0x02) != 0 ? GSEventTypeLockButtonDown : GSEventTypeLockButtonUp; @@ -314,13 +555,7 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { } if (twas != tis || tis) { - struct { - struct GSEventRecord record; - struct { - struct GSEventRecordInfo info; - struct GSPathInfo path; - } data; - } event; + struct VeencyEvent event; memset(&event, 0, sizeof(event)); @@ -339,7 +574,10 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { event.data.info.handInfo.x34 = 0x1; event.data.info.handInfo.x38 = tis ? 0x1 : 0x0; - event.data.info.pathPositions = 1; + if (Level_ < 3) + event.data.info.pathPositions = 1; + else + event.data.info.x52 = 1; event.data.path.x00 = 0x01; event.data.path.x01 = 0x02; @@ -352,7 +590,7 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { NSArray *displays([server displays]); if (displays != nil && [displays count] != 0) if (CAWindowServerDisplay *display = [displays objectAtIndex:0]) - port = [display clientPortAtPosition:event.record.locationInWindow]; + port = [display clientPortAtPosition:location]; } if (port == 0) { @@ -369,7 +607,169 @@ static void VNCPointer(int buttons, int x, int y, rfbClientPtr client) { mach_port_deallocate(mach_task_self(), purple); } +static void VNCSetSender(IOHIDEventRef event) { + IOHIDEventSetSenderID(event, 0xDEFACEDBEEFFECE5); +} + +static void VNCSendHIDEvent(IOHIDEventRef event) { + static IOHIDEventSystemClientRef client_(NULL); + if (client_ == NULL) + client_ = IOHIDEventSystemClientCreate(kCFAllocatorDefault); + + VNCSetSender(event); + IOHIDEventSystemClientDispatchEvent(client_, event); + CFRelease(event); +} + +static void VNCPointerNew(int buttons, int x, int y, CGPoint location, int diff, bool twas, bool tis) { + if ((diff & 0x10) != 0) + VNCSendHIDEvent(IOHIDEventCreateKeyboardEvent(kCFAllocatorDefault, mach_absolute_time(), kHIDPage_Telephony, kHIDUsage_Tfon_Flash, (buttons & 0x10) != 0, 0)); + if ((diff & 0x04) != 0) + VNCSendHIDEvent(IOHIDEventCreateKeyboardEvent(kCFAllocatorDefault, mach_absolute_time(), kHIDPage_Consumer, kHIDUsage_Csmr_Menu, (buttons & 0x04) != 0, 0)); + if ((diff & 0x02) != 0) + VNCSendHIDEvent(IOHIDEventCreateKeyboardEvent(kCFAllocatorDefault, mach_absolute_time(), kHIDPage_Consumer, kHIDUsage_Csmr_Power, (buttons & 0x02) != 0, 0)); + + uint32_t handm; + uint32_t fingerm; + + if (twas == 0 && tis == 1) { + handm = kIOHIDDigitizerEventRange | kIOHIDDigitizerEventTouch | kIOHIDDigitizerEventIdentity; + fingerm = kIOHIDDigitizerEventRange | kIOHIDDigitizerEventTouch; + } else if (twas == 1 && tis == 1) { + handm = kIOHIDDigitizerEventPosition; + fingerm = kIOHIDDigitizerEventPosition; + } else if (twas == 1 && tis == 0) { + handm = kIOHIDDigitizerEventRange | kIOHIDDigitizerEventTouch | kIOHIDDigitizerEventIdentity | kIOHIDDigitizerEventPosition; + fingerm = kIOHIDDigitizerEventRange | kIOHIDDigitizerEventTouch; + } else return; + + CAWindowServer *server([CAWindowServer serverIfRunning]); + if (server == nil) + return; + + CAWindowServerDisplay *display([[server displays] objectAtIndex:0]); + if (display == nil) + return; + + unsigned context([display contextIdAtPosition:CGPointMake(x, y)]); + mach_port_t port([display taskPortOfContextId:context]); + if (port == MACH_PORT_NULL) + return; + + IOHIDEventSystemConnectionRef connection([[$BKAccessibility _eventRoutingClientConnectionManager] clientForTaskPort:port]); + if (connection == NULL) + return; + + // XXX: I guess this isn't ambiguous, and it works + IOHIDFloat xf(x); + IOHIDFloat yf(y); + + IOHIDEventRef hand(IOHIDEventCreateDigitizerEvent(kCFAllocatorDefault, mach_absolute_time(), kIOHIDDigitizerTransducerTypeHand, 1<<22, 1, handm, 0, xf, yf, 0, 0, 0, 0, 0, 0)); + IOHIDEventSetIntegerValue(hand, kIOHIDEventFieldIsBuiltIn, true); + IOHIDEventSetIntegerValue(hand, kIOHIDEventFieldDigitizerIsDisplayIntegrated, true); + + IOHIDEventRef finger(IOHIDEventCreateDigitizerFingerEvent(kCFAllocatorDefault, mach_absolute_time(), 3, 2, fingerm, xf, yf, 0, 0, 0, tis, tis, 0)); + IOHIDEventAppendEvent(hand, finger); + CFRelease(finger); + + VNCSetSender(hand); + IOHIDEventSystemConnectionDispatchEvent(connection, hand); + CFRelease(hand); +} + +GSEventRef (*$GSEventCreateKeyEvent)(int, CGPoint, CFStringRef, CFStringRef, id, UniChar, short, short); +GSEventRef (*$GSCreateSyntheticKeyEvent)(UniChar, BOOL, BOOL); + +static void VNCKeyboardNew(rfbBool down, rfbKeySym key, rfbClientPtr client) { + //NSLog(@"VNC d:%u k:%04x", down, key); + + uint16_t usage; + + switch (key) { + case XK_exclam: case XK_1: usage = kHIDUsage_Keyboard1; break; + case XK_at: case XK_2: usage = kHIDUsage_Keyboard2; break; + case XK_numbersign: case XK_3: usage = kHIDUsage_Keyboard3; break; + case XK_dollar: case XK_4: usage = kHIDUsage_Keyboard4; break; + case XK_percent: case XK_5: usage = kHIDUsage_Keyboard5; break; + case XK_asciicircum: case XK_6: usage = kHIDUsage_Keyboard6; break; + case XK_ampersand: case XK_7: usage = kHIDUsage_Keyboard7; break; + case XK_asterisk: case XK_8: usage = kHIDUsage_Keyboard8; break; + case XK_parenleft: case XK_9: usage = kHIDUsage_Keyboard9; break; + case XK_parenright: case XK_0: usage = kHIDUsage_Keyboard0; break; + + case XK_A: case XK_a: usage = kHIDUsage_KeyboardA; break; + case XK_B: case XK_b: usage = kHIDUsage_KeyboardB; break; + case XK_C: case XK_c: usage = kHIDUsage_KeyboardC; break; + case XK_D: case XK_d: usage = kHIDUsage_KeyboardD; break; + case XK_E: case XK_e: usage = kHIDUsage_KeyboardE; break; + case XK_F: case XK_f: usage = kHIDUsage_KeyboardF; break; + case XK_G: case XK_g: usage = kHIDUsage_KeyboardG; break; + case XK_H: case XK_h: usage = kHIDUsage_KeyboardH; break; + case XK_I: case XK_i: usage = kHIDUsage_KeyboardI; break; + case XK_J: case XK_j: usage = kHIDUsage_KeyboardJ; break; + case XK_K: case XK_k: usage = kHIDUsage_KeyboardK; break; + case XK_L: case XK_l: usage = kHIDUsage_KeyboardL; break; + case XK_M: case XK_m: usage = kHIDUsage_KeyboardM; break; + case XK_N: case XK_n: usage = kHIDUsage_KeyboardN; break; + case XK_O: case XK_o: usage = kHIDUsage_KeyboardO; break; + case XK_P: case XK_p: usage = kHIDUsage_KeyboardP; break; + case XK_Q: case XK_q: usage = kHIDUsage_KeyboardQ; break; + case XK_R: case XK_r: usage = kHIDUsage_KeyboardR; break; + case XK_S: case XK_s: usage = kHIDUsage_KeyboardS; break; + case XK_T: case XK_t: usage = kHIDUsage_KeyboardT; break; + case XK_U: case XK_u: usage = kHIDUsage_KeyboardU; break; + case XK_V: case XK_v: usage = kHIDUsage_KeyboardV; break; + case XK_W: case XK_w: usage = kHIDUsage_KeyboardW; break; + case XK_X: case XK_x: usage = kHIDUsage_KeyboardX; break; + case XK_Y: case XK_y: usage = kHIDUsage_KeyboardY; break; + case XK_Z: case XK_z: usage = kHIDUsage_KeyboardZ; break; + + case XK_underscore: case XK_minus: usage = kHIDUsage_KeyboardHyphen; break; + case XK_plus: case XK_equal: usage = kHIDUsage_KeyboardEqualSign; break; + case XK_braceleft: case XK_bracketleft: usage = kHIDUsage_KeyboardOpenBracket; break; + case XK_braceright: case XK_bracketright: usage = kHIDUsage_KeyboardCloseBracket; break; + case XK_bar: case XK_backslash: usage = kHIDUsage_KeyboardBackslash; break; + case XK_colon: case XK_semicolon: usage = kHIDUsage_KeyboardSemicolon; break; + case XK_quotedbl: case XK_apostrophe: usage = kHIDUsage_KeyboardQuote; break; + case XK_asciitilde: case XK_grave: usage = kHIDUsage_KeyboardGraveAccentAndTilde; break; + case XK_less: case XK_comma: usage = kHIDUsage_KeyboardComma; break; + case XK_greater: case XK_period: usage = kHIDUsage_KeyboardPeriod; break; + case XK_question: case XK_slash: usage = kHIDUsage_KeyboardSlash; break; + + case XK_Return: usage = kHIDUsage_KeyboardReturnOrEnter; break; + case XK_BackSpace: usage = kHIDUsage_KeyboardDeleteOrBackspace; break; + case XK_Tab: usage = kHIDUsage_KeyboardTab; break; + case XK_space: usage = kHIDUsage_KeyboardSpacebar; break; + + case XK_Shift_L: usage = kHIDUsage_KeyboardLeftShift; break; + case XK_Shift_R: usage = kHIDUsage_KeyboardRightShift; break; + case XK_Control_L: usage = kHIDUsage_KeyboardLeftControl; break; + case XK_Control_R: usage = kHIDUsage_KeyboardRightControl; break; + case XK_Meta_L: usage = kHIDUsage_KeyboardLeftAlt; break; + case XK_Meta_R: usage = kHIDUsage_KeyboardRightAlt; break; + case XK_Alt_L: usage = kHIDUsage_KeyboardLeftGUI; break; + case XK_Alt_R: usage = kHIDUsage_KeyboardRightGUI; break; + + case XK_Up: usage = kHIDUsage_KeyboardUpArrow; break; + case XK_Down: usage = kHIDUsage_KeyboardDownArrow; break; + case XK_Left: usage = kHIDUsage_KeyboardLeftArrow; break; + case XK_Right: usage = kHIDUsage_KeyboardRightArrow; break; + + case XK_Home: case XK_Begin: usage = kHIDUsage_KeyboardHome; break; + case XK_End: usage = kHIDUsage_KeyboardEnd; break; + case XK_Page_Up: usage = kHIDUsage_KeyboardPageUp; break; + case XK_Page_Down: usage = kHIDUsage_KeyboardPageDown; break; + + default: return; + } + + VNCSendHIDEvent(IOHIDEventCreateKeyboardEvent(kCFAllocatorDefault, mach_absolute_time(), kHIDPage_KeyboardOrKeypad, usage, down, 0)); +} + static void VNCKeyboard(rfbBool down, rfbKeySym key, rfbClientPtr client) { + if (kCFCoreFoundationVersionNumber >= 800) + return VNCKeyboardNew(down, key, client); + if (!down) return; @@ -381,9 +781,21 @@ static void VNCKeyboard(rfbBool down, rfbKeySym key, rfbClientPtr client) { if (key > 0xfff) return; - GSEventRef event(_GSCreateSyntheticKeyEvent(key, YES, YES)); - GSEventRecord *record(_GSEventGetGSEventRecord(event)); - record->type = GSEventTypeKeyDown; + CGPoint point(CGPointMake(x_, y_)); + + UniChar unicode(key); + CFStringRef string(NULL); + + GSEventRef event0, event1(NULL); + if ($GSEventCreateKeyEvent != NULL) { + string = CFStringCreateWithCharacters(kCFAllocatorDefault, &unicode, 1); + event0 = (*$GSEventCreateKeyEvent)(10, point, string, string, nil, 0, 0, 1); + event1 = (*$GSEventCreateKeyEvent)(11, point, string, string, nil, 0, 0, 1); + } else if ($GSCreateSyntheticKeyEvent != NULL) { + event0 = (*$GSCreateSyntheticKeyEvent)(unicode, YES, YES); + GSEventRecord *record(_GSEventGetGSEventRecord(event0)); + record->type = GSEventTypeKeyDown; + } else return; mach_port_t port(0); @@ -391,7 +803,7 @@ static void VNCKeyboard(rfbBool down, rfbKeySym key, rfbClientPtr client) { NSArray *displays([server displays]); if (displays != nil && [displays count] != 0) if (CAWindowServerDisplay *display = [displays objectAtIndex:0]) - port = [display clientPortAtPosition:CGPointMake(x_, y_)]; + port = [display clientPortAtPosition:point]; } mach_port_t purple(0); @@ -402,13 +814,20 @@ static void VNCKeyboard(rfbBool down, rfbKeySym key, rfbClientPtr client) { port = purple; } - if (port != 0) - GSSendEvent(record, port); + if (port != 0) { + GSSendEvent(_GSEventGetGSEventRecord(event0), port); + if (event1 != NULL) + GSSendEvent(_GSEventGetGSEventRecord(event1), port); + } if (purple != 0 && PurpleAllocated) mach_port_deallocate(mach_task_self(), purple); - CFRelease(event); + CFRelease(event0); + if (event1 != NULL) + CFRelease(event1); + if (string != NULL) + CFRelease(string); } static void VNCDisconnect(rfbClientPtr client) { @@ -441,14 +860,20 @@ static rfbNewClientAction VNCClient(rfbClientPtr client) { return action; } -static rfbPixel black_[320][480]; +extern "C" bool GSSystemHasCapability(NSString *); + +static CFTypeRef (*$GSSystemCopyCapability)(CFStringRef); +static CFTypeRef (*$GSSystemGetCapability)(CFStringRef); +static BOOL (*$MGGetBoolAnswer)(CFStringRef); static void VNCSetup() { + rfbLogEnable(false); + @synchronized (condition_) { int argc(1); char *arg0(strdup("VNCServer")); char *argv[] = {arg0, NULL}; - screen_ = rfbGetScreen(&argc, argv, Width, Height, BitsPerSample, 3, BytesPerPixel); + screen_ = rfbGetScreen(&argc, argv, width_, height_, BitsPerSample, 3, BytesPerPixel); free(arg0); VNCSettings(); @@ -458,13 +883,56 @@ static void VNCSetup() { screen_->alwaysShared = TRUE; screen_->handleEventsEagerly = TRUE; - screen_->deferUpdateTime = 5; + screen_->deferUpdateTime = 1000 / 25; screen_->serverFormat.redShift = BitsPerSample * 2; screen_->serverFormat.greenShift = BitsPerSample * 1; screen_->serverFormat.blueShift = BitsPerSample * 0; - screen_->frameBuffer = reinterpret_cast(black_); + $GSSystemCopyCapability = reinterpret_cast(dlsym(RTLD_DEFAULT, "GSSystemCopyCapability")); + $GSSystemGetCapability = reinterpret_cast(dlsym(RTLD_DEFAULT, "GSSystemGetCapability")); + $MGGetBoolAnswer = reinterpret_cast(dlsym(RTLD_DEFAULT, "MGGetBoolAnswer")); + + CFTypeRef opengles2; + + if ($GSSystemCopyCapability != NULL) { + opengles2 = (*$GSSystemCopyCapability)(CFSTR("opengles-2")); + } else if ($GSSystemGetCapability != NULL) { + opengles2 = (*$GSSystemGetCapability)(CFSTR("opengles-2")); + if (opengles2 != NULL) + CFRetain(opengles2); + } else if ($MGGetBoolAnswer != NULL) { + opengles2 = $MGGetBoolAnswer(CFSTR("opengles-2")) ? kCFBooleanTrue : kCFBooleanFalse; + CFRetain(opengles2); + } else + opengles2 = NULL; + + bool accelerated(opengles2 != NULL && [(NSNumber *)opengles2 boolValue]); + + if (accelerated) + CoreSurfaceAcceleratorCreate(NULL, NULL, &accelerator_); + + if (opengles2 != NULL) + CFRelease(opengles2); + + if (accelerator_ != NULL) + buffer_ = CoreSurfaceBufferCreate((CFDictionaryRef) [NSDictionary dictionaryWithObjectsAndKeys: + @"PurpleEDRAM", kCoreSurfaceBufferMemoryRegion, + [NSNumber numberWithBool:YES], kCoreSurfaceBufferGlobal, + [NSNumber numberWithInt:(width_ * BytesPerPixel)], kCoreSurfaceBufferPitch, + [NSNumber numberWithInt:width_], kCoreSurfaceBufferWidth, + [NSNumber numberWithInt:height_], kCoreSurfaceBufferHeight, + [NSNumber numberWithInt:'BGRA'], kCoreSurfaceBufferPixelFormat, + [NSNumber numberWithInt:(width_ * height_ * BytesPerPixel)], kCoreSurfaceBufferAllocSize, + nil]); + else + VNCBlack(); + + //screen_->frameBuffer = reinterpret_cast(mmap(NULL, sizeof(rfbPixel) * width_ * height_, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE | MAP_NOCACHE, VM_FLAGS_PURGABLE, 0)); + + CoreSurfaceBufferLock(buffer_, 3); + screen_->frameBuffer = reinterpret_cast(CoreSurfaceBufferGetBaseAddress(buffer_)); + CoreSurfaceBufferUnlock(buffer_); screen_->kbdAddEvent = &VNCKeyboard; screen_->ptrAddEvent = &VNCPointer; @@ -472,12 +940,13 @@ static void VNCSetup() { screen_->newClientHook = &VNCClient; screen_->passwordCheck = &VNCCheck; - char data[0], mask[0]; - rfbCursorPtr cursor = rfbMakeXCursor(0, 0, data, mask); - rfbSetCursor(screen_, cursor); + screen_->cursor = NULL; } static void VNCEnabled() { + if (screen_ == NULL) + return; + [lock_ lock]; bool enabled(true); @@ -509,6 +978,61 @@ static void VNCNotifyEnabled( VNCEnabled(); } +void (*$IOMobileFramebufferIsMainDisplay)(IOMobileFramebufferRef, int *); + +static IOMobileFramebufferRef main_; +static CoreSurfaceBufferRef layer_; + +static void OnLayer(IOMobileFramebufferRef fb, CoreSurfaceBufferRef layer) { + if (_unlikely(width_ == 0 || height_ == 0)) { + CGSize size; + IOMobileFramebufferGetDisplaySize(fb, &size); + + width_ = size.width; + height_ = size.height; + + if (width_ == 0 || height_ == 0) + return; + + NSThread *thread([NSThread alloc]); + + [thread + initWithTarget:[VNCBridge class] + selector:@selector(performSetup:) + object:thread + ]; + + [thread start]; + } else if (_unlikely(clients_ != 0)) { + if (layer == NULL) { + if (accelerator_ != NULL) + memset(screen_->frameBuffer, 0, sizeof(rfbPixel) * width_ * height_); + else + VNCBlack(); + } else { + if (accelerator_ != NULL) + CoreSurfaceAcceleratorTransferSurface(accelerator_, layer, buffer_, options_); + else { + CoreSurfaceBufferLock(layer, 2); + rfbPixel *data(reinterpret_cast(CoreSurfaceBufferGetBaseAddress(layer))); + + CoreSurfaceBufferFlushProcessorCaches(layer); + + /*rfbPixel corner(data[0]); + data[0] = 0; + data[0] = corner;*/ + + screen_->frameBuffer = const_cast(reinterpret_cast(data)); + CoreSurfaceBufferUnlock(layer); + } + } + + rfbMarkRectAsModified(screen_, 0, 0, width_, height_); + } +} + +static bool wait_ = false; + MSHook(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, @@ -517,25 +1041,38 @@ MSHook(kern_return_t, IOMobileFramebufferSwapSetLayer, CGRect frame, int flags ) { - if (_unlikely(screen_ == NULL)) { - VNCSetup(); - VNCEnabled(); - } else if (_unlikely(clients_ != 0)) { - if (buffer == NULL) - screen_->frameBuffer = reinterpret_cast(black_); - else { - CoreSurfaceBufferLock(buffer, 2); - rfbPixel (*data)[480] = reinterpret_cast(CoreSurfaceBufferGetBaseAddress(buffer)); - screen_->frameBuffer = const_cast(reinterpret_cast(data)); - CoreSurfaceBufferUnlock(buffer); - } - - rfbMarkRectAsModified(screen_, 0, 0, Width, Height); + int main(false); + + if (_unlikely(buffer == NULL)) + main = fb == main_; + else if (_unlikely(fb == NULL)) + main = false; + else if ($IOMobileFramebufferIsMainDisplay == NULL) + main = true; + else + (*$IOMobileFramebufferIsMainDisplay)(fb, &main); + + if (_likely(main)) { + main_ = fb; + if (wait_) + layer_ = buffer; + else + OnLayer(fb, buffer); } return _IOMobileFramebufferSwapSetLayer(fb, layer, buffer, bounds, frame, flags); } +// XXX: beg rpetrich for the type of this function +extern "C" void *IOMobileFramebufferSwapWait(IOMobileFramebufferRef, void *, unsigned); + +MSHook(void *, IOMobileFramebufferSwapWait, IOMobileFramebufferRef fb, void *arg1, unsigned flags) { + void *value(_IOMobileFramebufferSwapWait(fb, arg1, flags)); + if (fb == main_) + OnLayer(fb, layer_); + return value; +} + MSHook(void, rfbRegisterSecurityHandler, rfbSecurityHandler *handler) { NSAutoreleasePool *pool([[NSAutoreleasePool alloc] init]); @@ -547,25 +1084,53 @@ MSHook(void, rfbRegisterSecurityHandler, rfbSecurityHandler *handler) { [pool release]; } +template +static void dlset(Type_ &function, const char *name) { + function = reinterpret_cast(dlsym(RTLD_DEFAULT, name)); +} + MSInitialize { NSAutoreleasePool *pool([[NSAutoreleasePool alloc] init]); - MSHookSymbol(GSTakePurpleSystemEventPort, "GSGetPurpleSystemEventPort"); + MSHookSymbol(GSTakePurpleSystemEventPort, "_GSGetPurpleSystemEventPort"); if (GSTakePurpleSystemEventPort == NULL) { - MSHookSymbol(GSTakePurpleSystemEventPort, "GSCopyPurpleSystemEventPort"); + MSHookSymbol(GSTakePurpleSystemEventPort, "_GSCopyPurpleSystemEventPort"); PurpleAllocated = true; } - Two_ = dlsym(RTLD_DEFAULT, "GSEventGetWindowContextId") == NULL; + if (dlsym(RTLD_DEFAULT, "GSLibraryCopyGenerationInfoValueForKey") != NULL) + Level_ = 3; + else if (dlsym(RTLD_DEFAULT, "GSKeyboardCreate") != NULL) + Level_ = 2; + else if (dlsym(RTLD_DEFAULT, "GSEventGetWindowContextId") != NULL) + Level_ = 1; + else + Level_ = 0; + + size_t size; + sysctlbyname("hw.machine", NULL, &size, NULL, 0); + char machine[size]; + sysctlbyname("hw.machine", machine, &size, NULL, 0); + iPad1_ = strcmp(machine, "iPad1,1") == 0; + + dlset($GSMainScreenScaleFactor, "GSMainScreenScaleFactor"); + dlset($GSEventCreateKeyEvent, "GSEventCreateKeyEvent"); + dlset($GSCreateSyntheticKeyEvent, "_GSCreateSyntheticKeyEvent"); + dlset($IOMobileFramebufferIsMainDisplay, "IOMobileFramebufferIsMainDisplay"); MSHookFunction(&IOMobileFramebufferSwapSetLayer, MSHake(IOMobileFramebufferSwapSetLayer)); MSHookFunction(&rfbRegisterSecurityHandler, MSHake(rfbRegisterSecurityHandler)); - $VNCAlertItem = objc_allocateClassPair(objc_getClass("SBAlertItem"), "VNCAlertItem", 0); - MSAddMessage2(VNCAlertItem, "v@:@i", alertSheet,buttonClicked); - MSAddMessage2(VNCAlertItem, "v@:cc", configure,requirePasscodeForActions); - MSAddMessage0(VNCAlertItem, "v@:", performUnlockAction); - objc_registerClassPair($VNCAlertItem); + if (wait_) + MSHookFunction(&IOMobileFramebufferSwapWait, MSHake(IOMobileFramebufferSwapWait)); + + if ($SBAlertItem != nil) { + $VNCAlertItem = objc_allocateClassPair($SBAlertItem, "VNCAlertItem", 0); + MSAddMessage2(VNCAlertItem, "v@:@i", alertSheet,buttonClicked); + MSAddMessage2(VNCAlertItem, "v@:cc", configure,requirePasscodeForActions); + MSAddMessage0(VNCAlertItem, "v@:", performUnlockAction); + objc_registerClassPair($VNCAlertItem); + } CFNotificationCenterAddObserver( CFNotificationCenterGetDarwinNotifyCenter(), @@ -581,8 +1146,6 @@ MSInitialize { lock_ = [[NSLock alloc] init]; handlers_ = [[NSMutableSet alloc] init]; - [pool release]; - bool value; value = true; @@ -592,4 +1155,9 @@ MSInitialize { cfFalse_ = CFDataCreate(kCFAllocatorDefault, reinterpret_cast(&value), sizeof(value)); cfEvent_ = CFDataCreateWithBytesNoCopy(kCFAllocatorDefault, reinterpret_cast(&event_), sizeof(event_), kCFAllocatorNull); + + options_ = (CFDictionaryRef) [[NSDictionary dictionaryWithObjectsAndKeys: + nil] retain]; + + [pool release]; }