wiz r1 relese
[ginge.git] / loader / patches.c
index ccbaadf..0504909 100644 (file)
@@ -1,86 +1,84 @@
 // vim:shiftwidth=2:expandtab
 #include <stdio.h>
-#include <string.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include <sys/mman.h>
-#include <unistd.h>
 
 #include "header.h"
 #include "sys_cacheflush.h"
 
-#define strace(...)
+#include "override.c"
+
+// note: first mask must be always full for search algo
+static const unsigned int sig_mask_all[] = {
+  0xffffffff, 0xffffffff, 0xffffffff, 0xffffffff,
+  0xffffffff, 0xffffffff, 0xffffffff, 0xffffffff
+};
 
 static const unsigned int sig_open[] = {
   0xe59cc000, 0xe33c0000, 0x1a000003, 0xef900005
 };
+#define sig_mask_open sig_mask_all
 
 static const unsigned int sig_mmap[] = {
   0xe92d000f, 0xe1a0000d, 0xef90005a, 0xe28dd010
 };
+#define sig_mask_mmap sig_mask_all
 
-static const unsigned int sig_mmap_[] = {
+static const unsigned int sig_mmap2[] = {
   0xe52d5004, 0xe59d5008, 0xe52d4004, 0xe59d4008,
   0xe1b0ca05, 0x1a000006, 0xe1a05625, 0xef9000c0
 };
+#define sig_mask_mmap2 sig_mask_all
 
 static const unsigned int sig_read[] = {
   0xe59fc080, 0xe59cc000, 0xe33c0000, 0x1a000003, 0xef900003
 };
+#define sig_mask_read sig_mask_all
 
-#define FAKE_DEVMEM_DEVICE      10001
-#define FAKE_DEVGPIO_DEVICE     10002
+static const unsigned int sig_ioctl[] = {
+  0xef900036, 0xe3700a01, 0x312fff1e
+};
+#define sig_mask_ioctl sig_mask_all
 
-static int w_open(const char *pathname, int flags, mode_t mode)
-{
-  int ret;
-  if      (strcmp(pathname, "/dev/mem") == 0)
-    ret = FAKE_DEVMEM_DEVICE;
-  else if (strcmp(pathname, "/dev/GPIO") == 0)
-    ret = FAKE_DEVGPIO_DEVICE;
-  else
-    ret = open(pathname, flags, mode);
-
-  strace("open(%s) = %d\n", pathname, ret);
-  return ret;
-}
+static const unsigned int sig_sigaction[] = {
+  0xe59f300c, 0xe3530000, 0x0a000000, 0xea000000, 0xea000000
+};
+static const unsigned int sig_mask_sigaction[] = {
+  0xffffffff, 0xffffffff, 0xffffffff, 0xff000000, 0xff000000
+};
 
-static void *w_mmap(void *addr, size_t length, int prot, int flags, int fd, off_t offset)
-{
-  void *ret;
-  if (fd != FAKE_DEVMEM_DEVICE)
-    ret = mmap(addr, length, prot, flags, fd, offset);
-  else
-    ret = emu_mmap_dev(length, prot, flags, offset);
-
-  strace("mmap(%p, %x, %x, %x, %d, %lx) = %p\n", addr, length, prot, flags, fd, (long)offset, ret);
-  return ret;
-}
-#define w_mmap_ w_mmap
+static const unsigned int sig_execve[] = {
+  0xef90000b, 0xe1a04000, 0xe3700a01
+};
+#define sig_mask_execve sig_mask_all
 
-ssize_t w_read(int fd, void *buf, size_t count)
-{
-  ssize_t ret;
-  if (fd != FAKE_DEVGPIO_DEVICE)
-    return read(fd, buf, count);
+static const unsigned int sig_execve2[] = {
+  0xef90000b, 0xe3700a01, 0xe1a04000
+};
+#define sig_mask_execve2 sig_mask_all
 
-  ret = emu_read_gpiodev(buf, count);
-  //printf("read(%d, %p, %d) = %d\n", fd, buf, count, ret);
-  return ret;
-}
+static const unsigned int sig_chdir[] = {
+  0xef90000c, 0xe3700a01, 0x312fff1e, 0xea0004bb
+};
+static const unsigned int sig_mask_chdir[] = {
+  0xffffffff, 0xffffffff, 0xffffffff, 0xff000000
+};
 
-#define PATCH(f) { sig_##f, ARRAY_SIZE(sig_##f), w_##f }
+#define PATCH_(f,p) { sig_##p, sig_mask_##p, ARRAY_SIZE(sig_##p), w_##f }
+#define PATCH(f) PATCH_(f,f)
 
 static const struct {
   const unsigned int *sig;
+  const unsigned int *sig_mask;
   size_t sig_cnt;
   void *func;
 } patches[] = {
   PATCH(open),
   PATCH(mmap),
-  PATCH(mmap_), // mmap using mmap2 syscall
+  PATCH(mmap2), // mmap2 syscall
   PATCH(read),
+  PATCH(ioctl),
+  PATCH(sigaction),
+//  PATCH_(execve, execve2), // hangs
+  PATCH(chdir),
 };
 
 void do_patches(void *ptr, unsigned int size)
@@ -89,6 +87,7 @@ void do_patches(void *ptr, unsigned int size)
 
   for (i = 0; i < ARRAY_SIZE(patches); i++) {
     const unsigned int *sig = patches[i].sig;
+    const unsigned int *sig_mask = patches[i].sig_mask;
     unsigned int *seg = (void *)(((long)ptr + 3) & ~3);
     unsigned int *seg_end = seg + size / 4;
     unsigned int sig0 = sig[0];
@@ -98,7 +97,7 @@ void do_patches(void *ptr, unsigned int size)
         continue;
 
       for (s = 1; s < patches[i].sig_cnt; s++)
-        if (seg[s] != sig[s])
+        if ((seg[s] ^ sig[s]) & sig_mask[s])
           break;
 
       if (s == patches[i].sig_cnt)
@@ -107,7 +106,7 @@ void do_patches(void *ptr, unsigned int size)
     continue;
 
 found:
-    printf("  patch #%i @ %08x\n", i, (int)seg);
+    dbg("  patch #%i @ %08x\n", i, (int)seg);
     seg[0] = 0xe59ff000; // ldr pc, [pc]
     seg[1] = 0;
     seg[2] = (unsigned int)patches[i].func;