/[RomCheater]/trunk/Win32/Sojaner.MemoryScanner/MemoryScanner.cs
ViewVC logotype

Diff of /trunk/Win32/Sojaner.MemoryScanner/MemoryScanner.cs

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

--- trunk/Win32/Sojaner.MemoryScanner/MemoryScanner.cs	2012/05/31 09:08:24	204
+++ trunk/Win32/Sojaner.MemoryScanner/MemoryScanner.cs	2012/06/03 12:36:47	231
@@ -5,8 +5,8 @@
 using System.Threading;
 using System.Runtime.InteropServices;
 using RomCheater.Logging;
-using RomCheater.Core;
 using System.IO;
+using Sojaner.MemoryScanner.MemoryProviers;
 
 namespace Sojaner.MemoryScanner
 {
@@ -14,7 +14,7 @@
     #region ProcessMemoryReader class
     //Thanks goes to Arik Poznanski for P/Invokes and methods needed to read and write the Memory
     //For more information refer to "Minesweeper, Behind the scenes" article by Arik Poznanski at Codeproject.com
-    public class ProcessMemoryReader
+    internal class ProcessMemoryReader : IMemoryReader, IMemoryWriter, IFileWriter
     {
 
         public ProcessMemoryReader()
@@ -68,132 +68,22 @@
             }
         }
 
-        #region WriteProcessMemoryToFile
-        public bool WriteProcessMemoryToFile(string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead)
-        {
-            RamDumper dumper = new RamDumper();
-            return dumper.DumpMemoryToFile(ReadProcess, filename, MemoryAddress, bytesToRead, out bytesRead);
-        }
-        #endregion
 
-        #region ReadProcessMemory
-        public byte[] ReadProcessMemory(uint MemoryAddress, uint bytesToRead, out int bytesRead)
-        {
-            RamDumper dumper = new RamDumper();
-            return dumper.DumpMemoryToByteArray(ReadProcess, MemoryAddress, bytesToRead, out bytesRead);
-        }
-        #endregion
-
-        #region ReadProcessMemory
-        public bool ReadFirstNonZeroByte(uint MemoryAddress, uint bytesToRead, out uint address)
-        {
-            RamDumper dumper = new RamDumper();
-            return dumper.ReadFirstNonZeroByte(ReadProcess, MemoryAddress, bytesToRead, out address);
-        }
-        #endregion
-        #region WriteProcessMemory
-        public void WriteProcessMemory(UIntPtr MemoryAddress, byte byteToWrite, out int bytesWritten)
-        {
-            WriteProcessMemory(MemoryAddress, new byte[] { byteToWrite }, out bytesWritten);
-        }
-        public void WriteProcessMemory(UIntPtr MemoryAddress, byte[] bytesToWrite, out int bytesWritten)
-        {
-            IntPtr ptrBytesWritten;
-            ProcessMemoryReaderApi.WriteProcessMemory(m_hProcess, MemoryAddress, bytesToWrite, (uint)bytesToWrite.Length, out ptrBytesWritten);
-            bytesWritten = ptrBytesWritten.ToInt32();
-        }
-        #endregion
 
         #region RamDumper
-        private interface IRamDumper
-        {
-           bool DumpMemoryToFile(Process ppid, string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead);
-           byte[] DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead);
-        }
-        private class RamDumper : IRamDumper
+        //private interface IRamDumper
+        //{
+        //    bool DumpMemoryToFile(Process ppid, string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead);
+        //    void DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead, out byte[] data);
+        //}
+        private class RamDumper
         {
-            public RamDumper() { }
-            private void InitMemoryDump(out uint byte_alignment)
-            {
-                byte_alignment = 102400; // get memory in 100mb chunks
-            }
+            public RamDumper() { } 
             #region IRamDumper members
-            #region DumpMemoryToFile
-            public bool DumpMemoryToFile(Process ppid, string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead)
-            {
-                //logger.Info.WriteLine("Dumping memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                bytesRead = 0;
-                uint byte_alignment = 0;
-                // get common init parameters
-                InitMemoryDump(out byte_alignment);
-                uint address = MemoryAddress;
-                uint _bytesToRead = bytesToRead;
-                byte[] buffer = new byte[] { };
-                try
-                {
-                    FileInfo fi = new FileInfo(filename);
-                    if (fi.Exists)
-                        fi.Delete();
-                    using (FileStream fs = new FileStream(filename, FileMode.CreateNew, FileAccess.ReadWrite, FileShare.ReadWrite))
-                    {
-                        BinaryWriter bw = new BinaryWriter(fs);
-                        //foreach (byte b in data) { bw.Write(b); }
-
-                        for (uint i = 0; i <= bytesToRead; )
-                        {
-                            if (_bytesToRead < byte_alignment)
-                            {
-                                _bytesToRead = bytesToRead;
-                                buffer = new byte[_bytesToRead];
-                            }
-                            else
-                            {
-                                _bytesToRead = byte_alignment;
-                                buffer = new byte[byte_alignment];
-                            }
-                            IntPtr ptrBytesRead;
-
-                            ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)address, buffer, _bytesToRead, out ptrBytesRead);
-                            bytesRead = ptrBytesRead.ToInt32();
-                            bw.Write(buffer);
-                            bw.Flush();
-
-                            if (_bytesToRead < byte_alignment)
-                            {
-                                i += _bytesToRead;
-                                address += _bytesToRead;
-                            }
-                            else
-                            {
-                                i += byte_alignment;
-                                address += byte_alignment;
-                            }
-
-
-                        }
-                        bw.Close();
-                    }
-                    logger.Info.WriteLine("Succefully dumped memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    return true;
-                }
-                catch (OutOfMemoryException ex)
-                {
-                    logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
-                    logger.Error.WriteLine(ex.ToString());
-                }
-                catch (Exception ex)
-                {
-                    logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    logger.Error.WriteLine("DumpMemory(): Exception");
-                    logger.Error.WriteLine(ex.ToString());
-                }
-                return false;
-            }
-            #endregion
             #region DumpMemoryToByteArray
-            public byte[] DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead)
+            public void DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead, out byte[] data)
             {
+                data = new byte[] { };
                 //logger.Info.WriteLine("Dumping memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
                 bytesRead = 0;
                 uint byte_alignment = 1;
@@ -222,7 +112,6 @@
                                 buffer = new byte[byte_alignment];
                             }
                             IntPtr ptrBytesRead;
-
                             ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)address, buffer, _bytesToRead, out ptrBytesRead);
                             bytesRead = ptrBytesRead.ToInt32();
                             bw.Write(buffer);
@@ -242,7 +131,7 @@
 
                         }
                         bw.Close();
-                        return ms.ToArray();
+                        data = ms.ToArray();
                     }
                     //logger.Info.WriteLine("Succefully dumped memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
                 }
@@ -258,91 +147,15 @@
                     logger.Error.WriteLine("DumpMemory(): Exception");
                     logger.Error.WriteLine(ex.ToString());
                 }
-                return new byte[]{};
             }
             #endregion
             #endregion
-
-            #region ReadFirstNonZeroByte
-            public bool ReadFirstNonZeroByte(Process ppid, uint MemoryAddress, uint bytesToRead, out uint address)
-            {
-                //logger.Info.WriteLine("Dumping memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                address = 0;
-                uint byte_alignment = 1;
-                // get common init parameters
-                //InitMemoryDump(out byte_alignment);
-                uint mem_address = MemoryAddress;
-                uint _bytesToRead = bytesToRead;
-                byte[] buffer = new byte[] { };
-                try
-                {
-                    //using (MemoryStream ms = new MemoryStream())
-                    //{
-                    //    //BinaryWriter bw = new BinaryWriter(ms);
-                    //    //foreach (byte b in data) { bw.Write(b); }
-                        for (uint i = 0; i <= bytesToRead; )
-                        {
-                            if (_bytesToRead < byte_alignment)
-                            {
-                                _bytesToRead = bytesToRead;
-                                buffer = new byte[_bytesToRead];
-                            }
-                            else
-                            {
-                                _bytesToRead = byte_alignment;
-                                buffer = new byte[byte_alignment];
-                            }
-                            IntPtr ptrBytesRead;
-                            ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)mem_address, buffer, _bytesToRead, out ptrBytesRead);
-                            //bw.Write(buffer);
-                            //bw.Flush();
-                            if (_bytesToRead < byte_alignment)
-                            {
-                                i += _bytesToRead;
-                                mem_address += _bytesToRead;
-                            }
-                            else
-                            {
-                                i += byte_alignment;
-                                mem_address += byte_alignment;
-                            }
-                            for (uint j = 0; j < buffer.Length; j++)
-                            {
-                                if (buffer[j] != 0)
-                                {
-                                    address = mem_address;
-                                    break;
-                                }
-                            }
-                            if (address != 0)
-                                break;
-                        }
-                    //    bw.Close();
-                    //}
-                    //logger.Info.WriteLine("Succefully dumped memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    return true;
-                }
-                catch (OutOfMemoryException ex)
-                {
-                    logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
-                    logger.Error.WriteLine(ex.ToString());
-                }
-                catch (Exception ex)
-                {
-                    logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
-                    logger.Error.WriteLine("DumpMemory(): Exception");
-                    logger.Error.WriteLine(ex.ToString());
-                }
-                return false;
-            }
-            #endregion
         }
         #endregion
         /// <summary>
         /// ProcessMemoryReader is a class that enables direct reading a process memory
         /// </summary>
-        public class ProcessMemoryReaderApi
+        private class ProcessMemoryReaderApi
         {
             // constants information can be found in <winnt.h>
             [Flags]
@@ -399,6 +212,174 @@
 
 
         }
+
+        #region IMemoryReader Members
+        public bool ReadFirstNonZeroByte(uint MemoryAddress, int bytesToRead, out uint address)
+        {
+            //logger.Info.WriteLine("Dumping memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
+            address = 0;
+            int byte_alignment = 1;
+            // get common init parameters
+            //InitMemoryDump(out byte_alignment);
+            uint mem_address = MemoryAddress;
+            int _bytesToRead = bytesToRead;
+            byte[] buffer = new byte[] { };
+            try
+            {
+                //using (MemoryStream ms = new MemoryStream())
+                //{
+                //    //BinaryWriter bw = new BinaryWriter(ms);
+                //    //foreach (byte b in data) { bw.Write(b); }
+                for (int i = 0; i <= bytesToRead; )
+                {
+                    if (_bytesToRead < byte_alignment)
+                    {
+                        _bytesToRead = bytesToRead;
+                        buffer = new byte[_bytesToRead];
+                    }
+                    else
+                    {
+                        _bytesToRead = byte_alignment;
+                        buffer = new byte[byte_alignment];
+                    }
+                    int bytesRead = 0;
+                    ReadProcessMemory(mem_address, _bytesToRead, out bytesRead, out buffer);
+                    //bw.Write(buffer);
+                    //bw.Flush();
+                    if (_bytesToRead < byte_alignment)
+                    {
+                        i += _bytesToRead;
+                        mem_address += (uint)_bytesToRead;
+                    }
+                    else
+                    {
+                        i += byte_alignment;
+                        mem_address += (uint)byte_alignment;
+                    }
+                    for (uint j = 0; j < buffer.Length; j++)
+                    {
+                        if (buffer[j] != 0)
+                        {
+                            address = mem_address;
+                            break;
+                        }
+                    }
+                    if (address != 0)
+                        break;
+                }
+                //    bw.Close();
+                //}
+                //logger.Info.WriteLine("Succefully dumped memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
+                return true;
+            }
+            catch (OutOfMemoryException ex)
+            {
+                logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ReadProcess.Id, ReadProcess.ProcessName));
+                logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
+                logger.Error.WriteLine(ex.ToString());
+            }
+            catch (Exception ex)
+            {
+                logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({2})", MemoryAddress, MemoryAddress + bytesToRead, string.Format("0x{0:x4} {1}.exe", ReadProcess.Id, ReadProcess.ProcessName));
+                logger.Error.WriteLine("DumpMemory(): Exception");
+                logger.Error.WriteLine(ex.ToString());
+            }
+            return false;
+        }
+        public void ReadProcessMemory(uint MemoryAddress, int bytesToRead, out int bytesRead, out byte[] data)
+        {
+            byte[] buffer = new byte[] { };
+            IntPtr ptrBytesRead;
+            uint _bytesToRead = (uint)bytesToRead;
+            ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)MemoryAddress, buffer, _bytesToRead, out ptrBytesRead);
+            bytesRead = ptrBytesRead.ToInt32();
+            data = buffer;
+        }
+        #endregion
+
+        #region IMemoryWriter Members
+        public void WriteProcessMemory(uint MemoryAddress, byte byteToWrite, out int bytesWritten)
+        {
+            WriteProcessMemory(MemoryAddress, new byte[] { byteToWrite }, out bytesWritten);
+        }
+        public void WriteProcessMemory(uint MemoryAddress, byte[] bytesToWrite, out int bytesWritten)
+        {
+            IntPtr ptrBytesWritten;
+            ProcessMemoryReaderApi.WriteProcessMemory(m_hProcess, (UIntPtr)MemoryAddress, bytesToWrite, (uint)bytesToWrite.Length, out ptrBytesWritten);
+            bytesWritten = ptrBytesWritten.ToInt32();
+        }
+        #endregion
+
+        #region IFileWriter Members
+
+        public bool WriteProcessMemoryToFile(string filename, uint MemoryAddress, int bytesToRead, out int bytesRead)
+        {
+            //logger.Info.WriteLine("Dumping memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ppid.Id, ppid.ProcessName));
+            bytesRead = 0;
+            int byte_alignment = 102400;
+            uint address = MemoryAddress;
+            int _bytesToRead = bytesToRead;
+            byte[] buffer = new byte[] { };
+            try
+            {
+                FileInfo fi = new FileInfo(filename);
+                if (fi.Exists)
+                    fi.Delete();
+                using (FileStream fs = new FileStream(filename, FileMode.CreateNew, FileAccess.ReadWrite, FileShare.ReadWrite))
+                {
+                    BinaryWriter bw = new BinaryWriter(fs);
+                    //foreach (byte b in data) { bw.Write(b); }
+
+                    for (int i = 0; i <= bytesToRead; )
+                    {
+                        if (_bytesToRead < byte_alignment)
+                        {
+                            _bytesToRead = bytesToRead;
+                            buffer = new byte[_bytesToRead];
+                        }
+                        else
+                        {
+                            _bytesToRead = byte_alignment;
+                            buffer = new byte[byte_alignment];
+                        }
+                        ReadProcessMemory(address, _bytesToRead, out bytesRead, out buffer);
+                        bw.Write(buffer);
+                        bw.Flush();
+
+                        if (_bytesToRead < byte_alignment)
+                        {
+                            i += _bytesToRead;
+                            address += (uint)_bytesToRead;
+                        }
+                        else
+                        {
+                            i += byte_alignment;
+                            address += (uint)byte_alignment;
+                        }
+
+
+                    }
+                    bw.Close();
+                }
+                logger.Info.WriteLine("Succefully dumped memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ReadProcess.Id, ReadProcess.ProcessName));
+                return true;
+            }
+            catch (OutOfMemoryException ex)
+            {
+                logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ReadProcess.Id, ReadProcess.ProcessName));
+                logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
+                logger.Error.WriteLine(ex.ToString());
+            }
+            catch (Exception ex)
+            {
+                logger.Error.WriteLine("Failed to dump memory (0x{0:x8}-0x{1:x8}) from pid=({3}) to file {2}", MemoryAddress, MemoryAddress + bytesToRead, filename, string.Format("0x{0:x4} {1}.exe", ReadProcess.Id, ReadProcess.ProcessName));
+                logger.Error.WriteLine("DumpMemory(): Exception");
+                logger.Error.WriteLine(ex.ToString());
+            }
+            return false;
+        }
+
+        #endregion
     }
     #endregion
 }

 

  ViewVC Help
Powered by ViewVC 1.1.22