mostm/screendump
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Complete VNC Preview for phone View

Browse Source
This commit is contained in:
Sharat M R
2019-04-24 19:52:53 +05:30
parent 71f4114ce3
commit 84176580d6
149 changed files with 54567 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

171
Tweak.xm
View File

@@ -1,8 +1,29 @@
static BOOL CCSisEnabled = YES;
#include <errno.h>
#include <substrate.h>
#include <rfb/rfb.h>
static bool CCSisEnabled = true;
static rfbScreenInfoPtr screen;
static bool isVNCRunning;
// static NSCondition *condition_;
static NSLock *lock;
static size_t width;
static size_t height;
static size_t byte_per_pixel;
static CFTypeRef (*$GSSystemCopyCapability)(CFStringRef);
static CFTypeRef (*$GSSystemGetCapability)(CFStringRef);
static BOOL (*$MGGetBoolAnswer)(CFStringRef);
typedef void *IOMobileFramebufferRef;
typedef void *IOSurfaceAcceleratorRef;
extern CFStringRef kIOSurfaceMemoryRegion;
extern const CFStringRef kIOSurfaceIsGlobal;
extern "C" void IOMobileFramebufferGetDisplaySize(IOMobileFramebufferRef connect, CGSize *size);
extern "C" int IOSurfaceAcceleratorCreate(CFAllocatorRef allocator, void *type, IOSurfaceAcceleratorRef *accel);
extern "C" unsigned int IOSurfaceAcceleratorTransferSurface(IOSurfaceAcceleratorRef accelerator, IOSurfaceRef dest, IOSurfaceRef src, void *, void *, void *, void *);
extern "C" kern_return_t IOMobileFramebufferSwapSetLayer(
IOMobileFramebufferRef fb,
@@ -16,6 +37,10 @@ extern "C" kern_return_t IOMobileFramebufferSwapSetLayer(
extern "C" void IOSurfaceFlushProcessorCaches(IOSurfaceRef buffer);
extern "C" int IOSurfaceLock(IOSurfaceRef surface, uint32_t options, uint32_t *seed);
extern "C" int IOSurfaceUnlock(IOSurfaceRef surface, uint32_t options, uint32_t *seed);
static void VNCUpdateRunState(bool shouldStart);
static IOSurfaceAcceleratorRef accelerator;
static IOSurfaceRef static_buffer;
int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filename) {
// unsigned char header[54] = {
@@ -77,40 +102,135 @@ int write_to_file(const void *image, size_t xsize, size_t ysize, size_t pixel_si
return 0;
}
%hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) {
CGSize size;
static void VNCSetup() {
int argc(1);
char *arg0(strdup("ScreenDumpVNC"));
char *argv[] = {arg0, NULL};
screen = rfbGetScreen(&argc, argv, width, height, 8, 3, byte_per_pixel);
screen->frameBuffer = (char *)malloc(width*height*byte_per_pixel);
free(arg0);
}
static void VNCBlack() {
screen->frameBuffer = (char *)malloc(width*height*byte_per_pixel);
}
static void initialBuffer() {
$GSSystemCopyCapability = reinterpret_cast<CFTypeRef (*)(CFStringRef)>(dlsym(RTLD_DEFAULT, "GSSystemCopyCapability"));
$GSSystemGetCapability = reinterpret_cast<CFTypeRef (*)(CFStringRef)>(dlsym(RTLD_DEFAULT, "GSSystemGetCapability"));
$MGGetBoolAnswer = reinterpret_cast<BOOL (*)(CFStringRef)>(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 isAccelerated(opengles2 != NULL && [(NSNumber *)opengles2 boolValue]);
if (isAccelerated) {
IOSurfaceAcceleratorCreate(NULL, NULL, &accelerator);
}
if (opengles2 != NULL) {
CFRelease(opengles2);
}
if (accelerator == NULL) {
VNCBlack();
} else {
static_buffer = IOSurfaceCreate((CFDictionaryRef) [NSDictionary dictionaryWithObjectsAndKeys:
@"PurpleEDRAM", kIOSurfaceMemoryRegion,
[NSNumber numberWithBool:YES], kIOSurfaceIsGlobal,
[NSNumber numberWithInt:(width * byte_per_pixel)], kIOSurfaceBytesPerRow,
[NSNumber numberWithInt:width], kIOSurfaceWidth,
[NSNumber numberWithInt:height], kIOSurfaceHeight,
[NSNumber numberWithInt:'BGRA'], kIOSurfacePixelFormat,
[NSNumber numberWithInt:(width * height * byte_per_pixel)], kIOSurfaceAllocSize,
nil]);
screen->frameBuffer = reinterpret_cast<char *>(IOSurfaceGetBaseAddress(static_buffer));
}
}
static void VNCUpdateRunState(bool shouldStart) {
if(screen == NULL) {
return;
}
if(shouldStart == isVNCRunning) {
return;
}
if(shouldStart) {
rfbInitServer(screen);
rfbRunEventLoop(screen, -1, true);
} else {
rfbShutdownServer(screen, true);
}
isVNCRunning = shouldStart;
}
static void OnFrameUpdate(IOMobileFramebufferRef fb, IOSurfaceRef buffer) {
size_t width_;
size_t height_;
if(!CCSisEnabled) {
return;
}
CGSize size;
IOMobileFramebufferGetDisplaySize(fb, &size);
width_ = size.width;
height_ = size.height;
size_t width = IOSurfaceGetWidth(buffer);
size_t height = IOSurfaceGetHeight(buffer);
size_t byte_per_pixel = IOSurfaceGetBytesPerElement(buffer);
NSLog(@"sharat %ld, %ld, %ld, %ld, %ld", width_, height_, width, height, byte_per_pixel);
// NSString *path = @"/tmp/test.bmp";
// void *bytes = IOSurfaceGetBaseAddress(buffer);
// if(width) {
// int ret;
// ret = bmp_write(bytes, width, height, [path UTF8String]);
// NSLog(@"sharat %d", ret);
// }
if(width == 0 || height == 0) {
width = IOSurfaceGetWidth(buffer);
height = IOSurfaceGetHeight(buffer);
byte_per_pixel = IOSurfaceGetBytesPerElement(buffer);
if(width == 0 || height == 0) {
return;
}
VNCSetup();
initialBuffer();
}
// IOSurfaceLock(buffer, kIOSurfaceLockReadOnly, NULL);
// IOSurfaceFlushProcessorCaches(buffer);
// write_to_file(bytes, width, height, byte_per_pixel, [path UTF8String]);
// IOSurfaceUnlock(buffer, kIOSurfaceLockReadOnly, NULL);
NSLog(@"sharat %d", CCSisEnabled);
return %orig;
NSLog(@"sharat %ld, %ld, %ld, %ld, %ld", width_, height_, width, height, byte_per_pixel);
NSLog(@"sharat accerated %ld", accelerator == NULL);
if(screen == NULL) {
return;
}
if (accelerator != NULL) {
IOSurfaceAcceleratorTransferSurface(accelerator, buffer, static_buffer, NULL, NULL, NULL, NULL);
NSLog(@"sharat accerated transfer");
} else {
IOSurfaceLock(buffer, kIOSurfaceLockReadOnly, NULL);
void *bytes = IOSurfaceGetBaseAddress(buffer);
IOSurfaceFlushProcessorCaches(buffer);
screen->frameBuffer = reinterpret_cast<char *> (bytes);
IOSurfaceUnlock(buffer, kIOSurfaceLockReadOnly, NULL);
}
rfbMarkRectAsModified(screen, 0, 0, width, height);
}
static void loadPrefs()
{
NSMutableDictionary *prefs = [[NSMutableDictionary alloc] initWithContentsOfFile:@"/var/mobile/Library/Preferences/com.cosmosgenius.screendumpprefs.plist"];
if(prefs) {
CCSisEnabled = ([prefs objectForKey:@"CCSisEnabled"]) ? [[prefs objectForKey:@"CCSisEnabled"] boolValue] : CCSisEnabled;
CFPreferencesAppSynchronize(CFSTR("com.cosmosgenius.screendump"));
Boolean valid;
bool enabled(CFPreferencesGetAppBooleanValue(CFSTR("CCSisEnabled"), CFSTR("com.cosmosgenius.screendump"), &valid));
CCSisEnabled = enabled;
@synchronized (lock) {
VNCUpdateRunState(CCSisEnabled);
}
[prefs release];
}
%hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) {
OnFrameUpdate(fb, buffer);
return %orig;
}
%ctor
@@ -118,7 +238,8 @@ static void loadPrefs()
CFNotificationCenterAddObserver(
CFNotificationCenterGetDarwinNotifyCenter(),
NULL, (CFNotificationCallback)loadPrefs,
CFSTR("com.cosmosgenius.screendumpprefs/settingschanged"),
CFSTR("com.cosmosgenius.screendump/preferences.changed"),
NULL, CFNotificationSuspensionBehaviorCoalesce);
lock = [[NSLock alloc] init];
loadPrefs();
}