From 5c4850659b843a3bf2fc5556b429a162280fdfdd Mon Sep 17 00:00:00 2001
From: Sharat M R <cosmosgenius@gmail.com>
Date: Fri, 21 Sep 2018 18:53:55 +0530
Subject: [PATCH] Initial commit

---
 .gitignore       |   3 ++
 Makefile         |  13 ++++++
 Tweak.xm         | 110 +++++++++++++++++++++++++++++++++++++++++++++++
 control          |   9 ++++
 screendump.plist |   1 +
 5 files changed, 136 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 Makefile
 create mode 100644 Tweak.xm
 create mode 100644 control
 create mode 100644 screendump.plist

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..031616e
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+.theos
+obj
+packages
diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..23e3e93
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,13 @@
+TARGET = iphone:11.2:10.0
+
+include $(THEOS)/makefiles/common.mk
+
+TWEAK_NAME = screendump
+screendump_FILES = Tweak.xm
+screendump_FRAMEWORKS := IOSurface
+screendump_PRIVATE_FRAMEWORKS := IOMobileFramebuffer
+
+include $(THEOS_MAKE_PATH)/tweak.mk
+
+after-install::
+	install.exec "killall -9 backboardd"
diff --git a/Tweak.xm b/Tweak.xm
new file mode 100644
index 0000000..1728ad0
--- /dev/null
+++ b/Tweak.xm
@@ -0,0 +1,110 @@
+/* How to Hook with Logos
+Hooks are written with syntax similar to that of an Objective-C @implementation.
+You don't need to #include <substrate.h>, it will be done automatically, as will
+the generation of a class list and an automatic constructor.
+
+%hook ClassName
+
+// Hooking a class method
++ (id)sharedInstance {
+	return %orig;
+}
+
+// Hooking an instance method with an argument.
+- (void)messageName:(int)argument {
+	%log; // Write a message about this call, including its class, name and arguments, to the system log.
+
+	%orig; // Call through to the original function with its original arguments.
+	%orig(nil); // Call through to the original function with a custom argument.
+
+	// If you use %orig(), you MUST supply all arguments (except for self and _cmd, the automatically generated ones.)
+}
+
+// Hooking an instance method with no arguments.
+- (id)noArguments {
+	%log;
+	id awesome = %orig;
+	[awesome doSomethingElse];
+
+	return awesome;
+}
+
+// Always make sure you clean up after yourself; Not doing so could have grave consequences!
+%end
+*/
+
+#include <errno.h>
+
+typedef void *IOMobileFramebufferRef;
+extern "C" void IOMobileFramebufferGetDisplaySize(IOMobileFramebufferRef connect, CGSize *size);
+
+extern "C" kern_return_t IOMobileFramebufferSwapSetLayer(
+    IOMobileFramebufferRef fb,
+    int layer,
+    IOSurfaceRef buffer,
+    CGRect bounds,
+    CGRect frame,
+    int flags
+);
+
+int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filename) {
+    unsigned char header[54] = {
+        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,
+        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;
+    header[2] = (unsigned char)(file_size &0x000000ff);
+    header[3] = (file_size >> 8) & 0x000000ff;
+    header[4] = (file_size >> 16) & 0x000000ff;
+    header[5] = (file_size >> 24) & 0x000000ff;
+
+    long width = xsize;
+    header[18] = width & 0x000000ff;
+    header[19] = (width >> 8) &0x000000ff;
+    header[20] = (width >> 16) &0x000000ff;
+    header[21] = (width >> 24) &0x000000ff;
+
+    long height = ysize;
+    header[22] = height &0x000000ff;
+    header[23] = (height >> 8) &0x000000ff;
+    header[24] = (height >> 16) &0x000000ff;
+    header[25] = (height >> 24) &0x000000ff;
+
+    char fname_bmp[128];
+    sprintf(fname_bmp, "%s", filename);
+
+    FILE *fp;
+    if (!(fp = fopen(fname_bmp, "wb"))) {
+        NSLog(@"Error no is : %s, %d", fname_bmp, errno);
+        return -1;
+    }
+
+    fwrite(header, sizeof(unsigned char), 54, fp);
+    fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * 4, fp);
+
+    fclose(fp);
+    return 0;
+}
+
+%hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) {
+    CGSize size;
+    size_t width_;
+    size_t height_;
+    IOMobileFramebufferGetDisplaySize(fb, &size);
+    width_ = size.width;
+    height_ = size.height;
+    size_t width = IOSurfaceGetWidth(buffer);
+    size_t height = IOSurfaceGetHeight(buffer);
+    NSLog(@"sharat %ld, %ld, %ld, %ld", width_, height_, width, height);
+    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);
+    }
+    return %orig;
+}
diff --git a/control b/control
new file mode 100644
index 0000000..b24cf02
--- /dev/null
+++ b/control
@@ -0,0 +1,9 @@
+Package: com.appknox.screendump
+Name: screendump
+Depends: mobilesubstrate
+Version: 0.0.1
+Architecture: iphoneos-arm
+Description: An awesome MobileSubstrate tweak!
+Maintainer: Sharat M R
+Author: Sharat M R
+Section: Tweaks
diff --git a/screendump.plist b/screendump.plist
new file mode 100644
index 0000000..812057f
--- /dev/null
+++ b/screendump.plist
@@ -0,0 +1 @@
+{ Filter = { Bundles = ( "com.apple.IOMobileFramebuffer" ); }; }