test: add tests about writes with ctypes

Author: MrIndeciso

test/binaries/ctypes_test

@@ -17,6 +17,7 @@ def test_all_types(self):
         d.run()
 
         bp = d.breakpoint("leak")
+        check = d.breakpoint("correct")
 
         class provola(struct):
             a: c_bool
@@ -78,5 +79,34 @@ class provola(struct):
         self.assertEqual(obj.y.value, 0xbeefbeefbeef)
         self.assertEqual(obj.z.value, 0xd00dd00dd00dd00d)
 
+        obj.a.value = False
+        obj.b.value = 1234
+        obj.c.value = b'\x23'
+        obj.e.value = 1234.5678
+        obj.f.value = 234.567
+        obj.g.value = 23456
+        obj.h.value = 32767
+        obj.i.value = 0xadbeef
+        obj.j.value = 0xadbeefdeadbeef
+        obj.k.value = 98
+        obj.l.value = 0xadbeefd00dbeef
+        obj.m.value = 345.678
+        obj.n.value = 0xedbeefdeadbeef
+        obj.o.value = 8765
+        obj.p.value = 0xeadbeefdeadbeef
+        obj.r.value = 0x234567
+        obj.s.value = 0x2345
+        obj.t.value = 0xeadbeef
+        obj.u.value = 0x2345
+        obj.v.value = 0xbadbeef
+        obj.w.value = 0xbadbeefbadbeef
+        obj.x.value = 0x28
+        obj.y.value = 0xbadbadbadbad
+        obj.z.value = 0xbadd00dbadd00d
+
+        d.cont()
+
+        self.assertTrue(check.hit_on(d))
+
         d.kill()
         d.terminate()

test/scripts/ctypes_test.py

@@ -10,6 +10,7 @@
 #include <stdint.h>
 #include <string.h>
 #include <time.h>
+#include <math.h>
 
 #pragma pack(push, 1)
 struct provola {
@@ -48,6 +49,11 @@ void leak(struct provola *ptr)
 
 }
 
+void correct()
+{
+
+}
+
 int main()
 {
     struct provola object;
@@ -79,5 +85,79 @@ int main()
 
     leak(&object);
 
+    if (object.a != 0)
+        return 0;
+
+    if (object.b != 1234)
+        return 0;
+
+    if (object.c != 0x23)
+        return 0;
+
+    if (object.e != 1234.5678)
+        return 0;
+
+    if (object.f < 234.5 || object.f > 234.6)
+        return 0;
+
+    if (object.g != 23456)
+        return 0;
+
+    if (object.h != 32767)
+        return 0;
+
+    if (object.i != 0xadbeef)
+        return 0;
+
+    if (object.j != 0xadbeefdeadbeef)
+        return 0;
+
+    if (object.k != 98)
+        return 0;
+
+    if (object.l != 0xadbeefd00dbeef)
+        return 0;
+
+    if (object.m != 345.678)
+        return 0;
+
+    if (object.n != 0xedbeefdeadbeef)
+        return 0;
+
+    if (object.o != 8765)
+        return 0;
+
+    if (object.p != 0xeadbeefdeadbeef)
+        return 0;
+
+    if (object.r != 0x234567)
+        return 0;
+
+    if (object.s != (void *) 0x2345)
+        return 0;
+
+    if (object.t != 0xeadbeef)
+        return 0;
+
+    if (object.u != 0x2345)
+        return 0;
+
+    if (object.v != 0xbadbeef)
+        return 0;
+
+    if (object.w != 0xbadbeefbadbeef)
+        return 0;
+
+    if (object.x != 0x28)
+        return 0;
+
+    if (object.y != 0xbadbadbadbad)
+        return 0;
+
+    if (object.z != 0xbadd00dbadd00d)
+        return 0;
+
+    correct();
+
     return 0;
 }