Made code a little cleaner and easier to use. Especially when dealing with modules.

Author: jonatino

.idea/compiler.xml

@@ -25,55 +25,82 @@
 package com.beaudoin.jmm.process;
 
 
+import com.beaudoin.jmm.misc.Cacheable;
 import com.beaudoin.jmm.misc.MemoryBuffer;
+import com.beaudoin.jmm.natives.win32.Kernel32;
+import com.beaudoin.jmm.process.impl.win32.Win32Process;
+import com.sun.jna.Native;
 import com.sun.jna.Pointer;
+import com.sun.jna.platform.win32.Win32Exception;
 
-public final class Module {
-
-	private final NativeProcess process;
-	private final String name;
-	private final long address;
-	private final int size;
-	private final Pointer pointer;
-	private MemoryBuffer data;
-
-	public Module(NativeProcess process, String name, Pointer pointer, long size) {
-		this.process = process;
-		this.name = name;
-		this.address = Pointer.nativeValue(pointer);
-		this.size = (int) size;
-		this.pointer = pointer;
-	}
-
-	public NativeProcess process() {
-		return process;
-	}
-
-	public Pointer pointer() {
-		return pointer;
-	}
-
-	public String name() {
-		return name;
-	}
-
-	public int size() {
-		return size;
-	}
-
-	public long address() {
-		return address;
-	}
-
-	public MemoryBuffer data() {
-		return data(false);
-	}
-
-	public MemoryBuffer data(boolean forceNew) {
-		if (forceNew || data == null) {
-			data = process().read(pointer(), size());
-		}
-		return data;
-	}
+public final class Module implements ReadableRegion {
+
+    private final NativeProcess process;
+    private final String name;
+    private final long address;
+    private final int size;
+    private final Pointer pointer;
+    private MemoryBuffer data;
+
+    public Module(NativeProcess process, String name, Pointer pointer, long size) {
+        this.process = process;
+        this.name = name;
+        this.address = Pointer.nativeValue(pointer);
+        this.size = (int) size;
+        this.pointer = pointer;
+    }
+
+    public NativeProcess process() {
+        return process;
+    }
+
+    public Pointer pointer() {
+        return pointer;
+    }
+
+    public String name() {
+        return name;
+    }
+
+    public int size() {
+        return size;
+    }
+
+    public long address() {
+        return address;
+    }
+
+    public MemoryBuffer data() {
+        return data(false);
+    }
+
+    public MemoryBuffer data(boolean forceNew) {
+        if (forceNew || data == null) {
+            data = process().read(pointer(), size());
+        }
+        return data;
+    }
+
+    @Override
+    public MemoryBuffer read(Pointer offset, int size) {
+        MemoryBuffer buffer = Cacheable.buffer(size);
+        if (Kernel32.ReadProcessMemory(((Win32Process) process()).pointer(), Cacheable.pointer(address() + Pointer.nativeValue(offset)), buffer, size, 0) == 0) {
+            throw new Win32Exception(Native.getLastError());
+        }
+        return buffer;
+    }
+
+    @Override
+    public NativeProcess write(Pointer offset, MemoryBuffer buffer) {
+        if (Kernel32.WriteProcessMemory(((Win32Process) process()).pointer(), Cacheable.pointer(address() + Pointer.nativeValue(offset)), buffer, buffer.size(), 0) == 0) {
+            throw new Win32Exception(Native.getLastError());
+        }
+        return process();
+    }
+
+    @Override
+    public boolean canRead(Pointer offset, int size) {
+        return Kernel32.ReadProcessMemory(((Win32Process) process()).pointer(), Cacheable.pointer(address() + Pointer.nativeValue(offset)), Cacheable.buffer(size), size, 0) != 0;
+    }
 
 }

.idea/misc.xml

@@ -24,13 +24,9 @@
 
 package com.beaudoin.jmm.process;
 
-import com.beaudoin.jmm.misc.Cacheable;
-import com.beaudoin.jmm.misc.MemoryBuffer;
-import com.beaudoin.jmm.misc.Strings;
 import com.beaudoin.jmm.misc.Utils;
 import com.beaudoin.jmm.natives.mac.mac;
 import com.beaudoin.jmm.natives.unix.libc;
-import com.beaudoin.jmm.natives.unix.unix;
 import com.beaudoin.jmm.natives.win32.Kernel32;
 import com.beaudoin.jmm.process.impl.mac.MacProcess;
 import com.beaudoin.jmm.process.impl.unix.UnixProcess;
@@ -41,12 +37,10 @@
 import com.sun.jna.platform.win32.Tlhelp32;
 import com.sun.jna.ptr.IntByReference;
 
-import static com.beaudoin.jmm.misc.Cacheable.buffer;
-
 /**
  * Created by Jonathan on 12/12/15.
  */
-public interface NativeProcess {
+public interface NativeProcess extends ReadableRegion {
 
     static NativeProcess byName(String name) {
         if (Platform.isWindows()) {
@@ -63,11 +57,11 @@ static NativeProcess byName(String name) {
                 Kernel32.CloseHandle(snapshot);
             }
         } else if (Platform.isMac() || Platform.isLinux()) {
-            return byId(Utils.exec("bash", "-c", "ps -A | grep -m1 \""+name+"\" | awk '{print $1}'"));
+            return byId(Utils.exec("bash", "-c", "ps -A | grep -m1 \"" + name + "\" | awk '{print $1}'"));
         } else {
             throw new UnsupportedOperationException("Unknown operating system! (" + System.getProperty("os.name") + ")");
         }
-        throw new IllegalStateException("Process " + name + " was not found. Are you sure its running?");
+        return null;
     }
 
     static NativeProcess byId(int id) {
@@ -99,75 +93,4 @@ static boolean checkSudo() {
 
     Module findModule(String moduleName);
 
-    MemoryBuffer read(Pointer address, int size);
-
-    NativeProcess write(Pointer address, MemoryBuffer buffer);
-
-    boolean canRead(Pointer address, int size);
-
-    default int readByte(long address) {
-        return read(address, 1).getByte();
-    }
-
-    default int readShort(long address) {
-        return read(address, 2).getShort();
-    }
-
-    default int readInt(long address) {
-        return read(address, 4).getInt();
-    }
-
-    default long readLong(long address) {
-        return read(address, 8).getLong();
-    }
-
-    default float readFloat(long address) {
-        return read(address, 4).getFloat();
-    }
-
-    default double readDouble(long address) {
-        return read(address, 8).getDouble();
-    }
-
-    default String readString(long address, int length) {
-        byte[] bytes = new byte[length];
-        read(address, bytes.length).get(bytes);
-        return Strings.transform(bytes);
-    }
-
-    default MemoryBuffer read(long address, int size) {
-        return read(Cacheable.pointer(address), size);
-    }
-
-    default NativeProcess writeBoolean(long address, boolean value) {
-        return write(Cacheable.pointer(address), buffer(1).putBoolean(value));
-    }
-
-    default NativeProcess writeByte(long address, int value) {
-        return write(Cacheable.pointer(address), buffer(1).putByte(value));
-    }
-
-    default NativeProcess writeShort(long address, int value) {
-        return write(Cacheable.pointer(address), buffer(2).putShort(value));
-    }
-
-    default NativeProcess writeInt(long address, int value) {
-        return write(Cacheable.pointer(address), buffer(4).putInt(value));
-    }
-
-    default NativeProcess writeLong(long address, long value) {
-        return write(Cacheable.pointer(address), buffer(8).putLong(value));
-    }
-
-    default NativeProcess writeFloat(long address, float value) {
-        return write(Cacheable.pointer(address), buffer(4).putFloat(value));
-    }
-
-    default NativeProcess writeDouble(long address, double value) {
-        return write(Cacheable.pointer(address), buffer(8).putDouble(value));
-    }
-
-    default boolean canRead(long address, int size) {
-        return canRead(Cacheable.pointer(address), size);
-    }
-}
+}