initial

1 files changed, 67 insertions, 0 deletions

diff --git a/other/Kermit/test/test_patch.cpp b/other/Kermit/test/test_patch.cpp
new file mode 100644
index 0000000..c5bbd94
--- /dev/null
+++ b/other/Kermit/test/test_patch.cpp
@@ -0,0 +1,67 @@
+#include <Patch.hpp>
+#include <SymbolTable.hpp>
+#include <rwKernel.hpp>
+#include <string>
+
+void bla_foo (char *n, bool a)
+{
+  if (a == false)
+    {
+      cout << n << " Failed" << endl;
+      abort ();
+    }
+}
+
+
+int main ()
+{
+  bool foo = false;
+  Patch *mall = NULL;
+  SymbolTable *tab = NULL;
+  Addr2AddrList *x = NULL;
+  rwKernel *y = NULL;
+  unsigned char *tmp = NULL;
+  int w = 0;
+
+  unsigned char sys_malloc[] =
+        "\x55\x89\xe5\x83\xec\x14\x53\xc7\x45\xfc\xee\xee"
+        "\xf8\xf8\xc7\x45\xf8\x00\x00\x00\x00\x83\xc4\xf8"
+        "\x6a\x15\x8b\x45\x08\x50\x8b\x5d\xfc\xff\xd3\x83"
+        "\xc4\x10\x89\xc0\x89\x45\xf8\x8b\x55\xf8\x89\xd0"
+        "\xeb\x06\x8d\xb6\x00\x00\x00\x00\x8b\x5d\xe8\x89"
+        "\xec\x5d\xc3\x90";
+
+  y = new rwKernel ();
+  genDummyValMap ();
+  tab = new SymbolTable (y);
+
+
+/* look up all needed symbols */
+  foo = tab->findSymbol (string ("init"));
+  bla_foo ("init", foo);
+  foo = tab->findSymbol (string ("kmalloc"));
+  bla_foo ("kmalloc", foo);
+
+
+  cout << "Found all needed symbols, proceeding .." << endl;
+  x = genReplaceValMap (tab);
+  if (x == NULL)
+    {
+      cout << "Foo!" << endl;
+      abort ();
+    }
+
+/* insert malloc function [overwritting init] */
+  cout << "Creating patch" << endl;
+  mall = new Patch (sys_malloc, 72, tab->getSymbol ("init"));
+
+  cout << "linking" << endl;
+  mall->link (x);
+  foo = mall->wasChanged ();
+  bla_foo ("link", foo);
+  cout << "appling now ..." << endl;
+  mall->apply (y);
+  cout << "done." << endl;
+  return 0;
+}
+