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

Update control

This commit is contained in:
Sharat M R 2019-04-22 18:07:38 +05:30
parent 5c4850659b
commit 0ec8d1a621
No known key found for this signature in database
GPG Key ID: E450A6E45DAE1B94
3 changed files with 60 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Makefile
View File

@ -1,4 +1,5 @@
TARGET = iphone:11.2:10.0 TARGET = iphone:11.2:10.0
ARCHS := arm64
include $(THEOS)/makefiles/common.mk include $(THEOS)/makefiles/common.mk

86
Tweak.xm
View File

@ -47,31 +47,35 @@ extern "C" kern_return_t IOMobileFramebufferSwapSetLayer(
int flags int flags
); );
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);
int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filename) { int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filename) {
unsigned char header[54] = { // unsigned char header[54] = {
0x42, 0x4d, 0, 0, 0, 0, 0, 0, 0, 0, // 0x42, 0x4d, 0, 0, 0, 0, 0, 0, 0, 0,
54, 0, 0, 0, 40, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 32, 0, // 54, 0, 0, 0, 40, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 32, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0 // 0, 0, 0, 0
}; // };
long file_size = (long)xsize * (long)ysize * 4 + 54; // long file_size = (long)xsize * (long)ysize * 4 + 54;
header[2] = (unsigned char)(file_size &0x000000ff); // header[2] = (unsigned char)(file_size &0x000000ff);
header[3] = (file_size >> 8) & 0x000000ff; // header[3] = (file_size >> 8) & 0x000000ff;
header[4] = (file_size >> 16) & 0x000000ff; // header[4] = (file_size >> 16) & 0x000000ff;
header[5] = (file_size >> 24) & 0x000000ff; // header[5] = (file_size >> 24) & 0x000000ff;
long width = xsize; // long width = xsize;
header[18] = width & 0x000000ff; // header[18] = width & 0x000000ff;
header[19] = (width >> 8) &0x000000ff; // header[19] = (width >> 8) &0x000000ff;
header[20] = (width >> 16) &0x000000ff; // header[20] = (width >> 16) &0x000000ff;
header[21] = (width >> 24) &0x000000ff; // header[21] = (width >> 24) &0x000000ff;
long height = ysize; // long height = ysize;
header[22] = height &0x000000ff; // header[22] = height &0x000000ff;
header[23] = (height >> 8) &0x000000ff; // header[23] = (height >> 8) &0x000000ff;
header[24] = (height >> 16) &0x000000ff; // header[24] = (height >> 16) &0x000000ff;
header[25] = (height >> 24) &0x000000ff; // header[25] = (height >> 24) &0x000000ff;
char fname_bmp[128]; char fname_bmp[128];
sprintf(fname_bmp, "%s", filename); sprintf(fname_bmp, "%s", filename);
@ -82,13 +86,31 @@ int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filenam
return -1; return -1;
} }
fwrite(header, sizeof(unsigned char), 54, fp); // fwrite(header, sizeof(unsigned char), 54, fp);
fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * 4, fp); fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * 4, fp);
fclose(fp); fclose(fp);
return 0; return 0;
} }
int write_to_file(const void *image, size_t xsize, size_t ysize, size_t pixel_size, const char *filename) {
char fname_bmp[128];
sprintf(fname_bmp, "%s", filename);
if( access( fname_bmp, F_OK ) != -1 ) {
return 0;
} else {
FILE *fp;
if (!(fp = fopen(fname_bmp, "wb"))) {
NSLog(@"Error no is : %s, %d", fname_bmp, errno);
return -1;
}
NSLog(@"sharat write to file");
fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * pixel_size, fp);
fclose(fp);
}
return 0;
}
%hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) { %hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) {
CGSize size; CGSize size;
size_t width_; size_t width_;
@ -98,13 +120,21 @@ int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filenam
height_ = size.height; height_ = size.height;
size_t width = IOSurfaceGetWidth(buffer); size_t width = IOSurfaceGetWidth(buffer);
size_t height = IOSurfaceGetHeight(buffer); size_t height = IOSurfaceGetHeight(buffer);
NSLog(@"sharat %ld, %ld, %ld, %ld", width_, height_, width, height); 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"; NSString *path = @"/tmp/test.bmp";
void *bytes = IOSurfaceGetBaseAddress(buffer); void *bytes = IOSurfaceGetBaseAddress(buffer);
if(width) { // if(width) {
int ret; // int ret;
ret = bmp_write(bytes, width, height, [path UTF8String]); // ret = bmp_write(bytes, width, height, [path UTF8String]);
NSLog(@"sharat %d", ret); // NSLog(@"sharat %d", ret);
} // }
IOSurfaceLock(buffer, kIOSurfaceLockReadOnly, NULL);
IOSurfaceFlushProcessorCaches(buffer);
write_to_file(bytes, width, height, byte_per_pixel, [path UTF8String]);
IOSurfaceUnlock(buffer, kIOSurfaceLockReadOnly, NULL);
return %orig; return %orig;
} }

2
control
View File

@ -1,4 +1,4 @@
Package: com.appknox.screendump Package: com.cosmosgenius.screendump
Name: screendump Name: screendump
Depends: mobilesubstrate Depends: mobilesubstrate
Version: 0.0.1 Version: 0.0.1