Win32/Sojaner.MemoryScanner/MemoryScanner.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Diagnostics;
using System.Threading;
using System.Runtime.InteropServices;
using RomCheater.Logging;
using System.IO;
using Sojaner.MemoryScanner.MemoryProviers;

namespace Sojaner.MemoryScanner
{
    // code borrowed from: http://www.codeproject.com/KB/cs/sojaner_memory_scanner.aspx    
    #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
    internal class ProcessMemoryReader : IMemoryReader, IMemoryWriter, IFileWriter
    {

        public ProcessMemoryReader()
        {
        }

        /// <summary>   
        /// Process from which to read          
        /// </summary>
        public Process ReadProcess
        {
            get
            {
                return m_ReadProcess;
            }
            set
            {
                m_ReadProcess = value;
            }
        }

        private Process m_ReadProcess = null;

        private static IntPtr m_hProcess = IntPtr.Zero;

        public void OpenProcess()
        {
            //                  m_hProcess = ProcessMemoryReaderApi.OpenProcess(ProcessMemoryReaderApi.PROCESS_VM_READ, 1, (uint)m_ReadProcess.Id);
            ProcessMemoryReaderApi.ProcessAccessType access;
            access = ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_READ
                | ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_WRITE
                | ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_OPERATION;
            m_hProcess = ProcessMemoryReaderApi.OpenProcess((uint)access, 1, (uint)m_ReadProcess.Id);
        }

        public void CloseHandle()
        {
            try
            {
                int iRetValue;
                iRetValue = ProcessMemoryReaderApi.CloseHandle(m_hProcess);
                if (iRetValue == 0)
                {
                    throw new Exception("CloseHandle failed");
                }
            }
            catch (Exception ex)
            {
                //System.Windows.Forms.MessageBox.Show(ex.Message, "error", System.Windows.Forms.MessageBoxButtons.OK, System.Windows.Forms.MessageBoxIcon.Warning);
                throw ex;
            }
        }


        #region RamDumper
        //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() { } 
            #region IRamDumper members
            #region DumpMemoryToByteArray
            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;
                // get common init parameters
                //InitMemoryDump(out byte_alignment);
                uint 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)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();
                        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));
                }
                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());
                }
            }
            #endregion
            #endregion
        }
        #endregion
        /// <summary>
        /// ProcessMemoryReader is a class that enables direct reading a process memory
        /// </summary>
        private class ProcessMemoryReaderApi
        {
            // constants information can be found in <winnt.h>
            [Flags]
            public enum ProcessAccessType
            {
                PROCESS_TERMINATE = (0x0001),
                PROCESS_CREATE_THREAD = (0x0002),
                PROCESS_SET_SESSIONID = (0x0004),
                PROCESS_VM_OPERATION = (0x0008),
                PROCESS_VM_READ = (0x0010),
                PROCESS_VM_WRITE = (0x0020),
                PROCESS_DUP_HANDLE = (0x0040),
                PROCESS_CREATE_PROCESS = (0x0080),
                PROCESS_SET_QUOTA = (0x0100),
                PROCESS_SET_INFORMATION = (0x0200),
                PROCESS_QUERY_INFORMATION = (0x0400)
            }

            // function declarations are found in the MSDN and in <winbase.h> 

            //          HANDLE OpenProcess(
            //                  DWORD dwDesiredAccess,  // access flag
            //                  BOOL bInheritHandle,    // handle inheritance option
            //                  DWORD dwProcessId       // process identifier
            //                  );
            [DllImport("kernel32.dll")]
            public static extern IntPtr OpenProcess(UInt32 dwDesiredAccess, Int32 bInheritHandle, UInt32 dwProcessId);

            //          BOOL CloseHandle(
            //                  HANDLE hObject   // handle to object
            //                  );
            [DllImport("kernel32.dll")]
            public static extern Int32 CloseHandle(IntPtr hObject);

            //          BOOL ReadProcessMemory(
            //                  HANDLE hProcess,              // handle to the process
            //                  LPCVOID lpBaseAddress,        // base of memory area
            //                  LPVOID lpBuffer,              // data buffer
            //                  SIZE_T nSize,                 // number of bytes to read
            //                  SIZE_T * lpNumberOfBytesRead  // number of bytes read
            //                  );
            [DllImport("kernel32.dll")]
            public static extern Int32 ReadProcessMemory(IntPtr hProcess, UIntPtr lpBaseAddress, [In, Out] byte[] buffer, UInt32 size, out IntPtr lpNumberOfBytesRead);

            //          BOOL WriteProcessMemory(
            //                  HANDLE hProcess,                // handle to process
            //                  LPVOID lpBaseAddress,           // base of memory area
            //                  LPCVOID lpBuffer,               // data buffer
            //                  SIZE_T nSize,                   // count of bytes to write
            //                  SIZE_T * lpNumberOfBytesWritten // count of bytes written
            //                  );
            [DllImport("kernel32.dll")]
            public static extern Int32 WriteProcessMemory(IntPtr hProcess, UIntPtr lpBaseAddress, [In, Out] byte[] buffer, UInt32 size, out IntPtr lpNumberOfBytesWritten);


        }

        #region IMemoryReader Members
        public bool ReadFirstNonZeroByte(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];
                    }
                    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, uint bytesToRead, out int bytesRead, out byte[] data)
        {
            byte[] buffer = new byte[] { };
            IntPtr ptrBytesRead;
            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, 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 = 102400;
            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];
                        }
                        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
}
Revision:	235
Committed:	Sun Jun 3 12:44:26 2012 UTC (11 years, 5 months ago) by william
File size:	17657 byte(s)
Log Message:	change int in method to uint
#	Content
1	using System;
2	using System.Collections.Generic;
3	using System.Text;
4	using System.Diagnostics;
5	using System.Threading;
6	using System.Runtime.InteropServices;
7	using RomCheater.Logging;
8	using System.IO;
9	using Sojaner.MemoryScanner.MemoryProviers;
10
11	namespace Sojaner.MemoryScanner
12	{
13	// code borrowed from: http://www.codeproject.com/KB/cs/sojaner_memory_scanner.aspx
14	#region ProcessMemoryReader class
15	//Thanks goes to Arik Poznanski for P/Invokes and methods needed to read and write the Memory
16	//For more information refer to "Minesweeper, Behind the scenes" article by Arik Poznanski at Codeproject.com
17	internal class ProcessMemoryReader : IMemoryReader, IMemoryWriter, IFileWriter
18	{
19
20	public ProcessMemoryReader()
21	{
22	}
23
24	/// <summary>
25	/// Process from which to read
26	/// </summary>
27	public Process ReadProcess
28	{
29	get
30	{
31	return m_ReadProcess;
32	}
33	set
34	{
35	m_ReadProcess = value;
36	}
37	}
38
39	private Process m_ReadProcess = null;
40
41	private static IntPtr m_hProcess = IntPtr.Zero;
42
43	public void OpenProcess()
44	{
45	// m_hProcess = ProcessMemoryReaderApi.OpenProcess(ProcessMemoryReaderApi.PROCESS_VM_READ, 1, (uint)m_ReadProcess.Id);
46	ProcessMemoryReaderApi.ProcessAccessType access;
47	access = ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_READ
48	\| ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_WRITE
49	\| ProcessMemoryReaderApi.ProcessAccessType.PROCESS_VM_OPERATION;
50	m_hProcess = ProcessMemoryReaderApi.OpenProcess((uint)access, 1, (uint)m_ReadProcess.Id);
51	}
52
53	public void CloseHandle()
54	{
55	try
56	{
57	int iRetValue;
58	iRetValue = ProcessMemoryReaderApi.CloseHandle(m_hProcess);
59	if (iRetValue == 0)
60	{
61	throw new Exception("CloseHandle failed");
62	}
63	}
64	catch (Exception ex)
65	{
66	//System.Windows.Forms.MessageBox.Show(ex.Message, "error", System.Windows.Forms.MessageBoxButtons.OK, System.Windows.Forms.MessageBoxIcon.Warning);
67	throw ex;
68	}
69	}
70
71
72
73	#region RamDumper
74	//private interface IRamDumper
75	//{
76	// bool DumpMemoryToFile(Process ppid, string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead);
77	// void DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead, out byte[] data);
78	//}
79	private class RamDumper
80	{
81	public RamDumper() { }
82	#region IRamDumper members
83	#region DumpMemoryToByteArray
84	public void DumpMemoryToByteArray(Process ppid, uint MemoryAddress, uint bytesToRead, out int bytesRead, out byte[] data)
85	{
86	data = new byte[] { };
87	//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));
88	bytesRead = 0;
89	uint byte_alignment = 1;
90	// get common init parameters
91	//InitMemoryDump(out byte_alignment);
92	uint address = MemoryAddress;
93	uint _bytesToRead = bytesToRead;
94	byte[] buffer = new byte[] { };
95	try
96	{
97	using (MemoryStream ms = new MemoryStream())
98	{
99	BinaryWriter bw = new BinaryWriter(ms);
100	//foreach (byte b in data) { bw.Write(b); }
101
102	for (uint i = 0; i <= bytesToRead; )
103	{
104	if (_bytesToRead < byte_alignment)
105	{
106	_bytesToRead = bytesToRead;
107	buffer = new byte[_bytesToRead];
108	}
109	else
110	{
111	_bytesToRead = byte_alignment;
112	buffer = new byte[byte_alignment];
113	}
114	IntPtr ptrBytesRead;
115	ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)address, buffer, _bytesToRead, out ptrBytesRead);
116	bytesRead = ptrBytesRead.ToInt32();
117	bw.Write(buffer);
118	bw.Flush();
119
120	if (_bytesToRead < byte_alignment)
121	{
122	i += _bytesToRead;
123	address += _bytesToRead;
124	}
125	else
126	{
127	i += byte_alignment;
128	address += byte_alignment;
129	}
130
131
132	}
133	bw.Close();
134	data = ms.ToArray();
135	}
136	//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));
137	}
138	catch (OutOfMemoryException ex)
139	{
140	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));
141	logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
142	logger.Error.WriteLine(ex.ToString());
143	}
144	catch (Exception ex)
145	{
146	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));
147	logger.Error.WriteLine("DumpMemory(): Exception");
148	logger.Error.WriteLine(ex.ToString());
149	}
150	}
151	#endregion
152	#endregion
153	}
154	#endregion
155	/// <summary>
156	/// ProcessMemoryReader is a class that enables direct reading a process memory
157	/// </summary>
158	private class ProcessMemoryReaderApi
159	{
160	// constants information can be found in <winnt.h>
161	[Flags]
162	public enum ProcessAccessType
163	{
164	PROCESS_TERMINATE = (0x0001),
165	PROCESS_CREATE_THREAD = (0x0002),
166	PROCESS_SET_SESSIONID = (0x0004),
167	PROCESS_VM_OPERATION = (0x0008),
168	PROCESS_VM_READ = (0x0010),
169	PROCESS_VM_WRITE = (0x0020),
170	PROCESS_DUP_HANDLE = (0x0040),
171	PROCESS_CREATE_PROCESS = (0x0080),
172	PROCESS_SET_QUOTA = (0x0100),
173	PROCESS_SET_INFORMATION = (0x0200),
174	PROCESS_QUERY_INFORMATION = (0x0400)
175	}
176
177	// function declarations are found in the MSDN and in <winbase.h>
178
179	// HANDLE OpenProcess(
180	// DWORD dwDesiredAccess, // access flag
181	// BOOL bInheritHandle, // handle inheritance option
182	// DWORD dwProcessId // process identifier
183	// );
184	[DllImport("kernel32.dll")]
185	public static extern IntPtr OpenProcess(UInt32 dwDesiredAccess, Int32 bInheritHandle, UInt32 dwProcessId);
186
187	// BOOL CloseHandle(
188	// HANDLE hObject // handle to object
189	// );
190	[DllImport("kernel32.dll")]
191	public static extern Int32 CloseHandle(IntPtr hObject);
192
193	// BOOL ReadProcessMemory(
194	// HANDLE hProcess, // handle to the process
195	// LPCVOID lpBaseAddress, // base of memory area
196	// LPVOID lpBuffer, // data buffer
197	// SIZE_T nSize, // number of bytes to read
198	// SIZE_T * lpNumberOfBytesRead // number of bytes read
199	// );
200	[DllImport("kernel32.dll")]
201	public static extern Int32 ReadProcessMemory(IntPtr hProcess, UIntPtr lpBaseAddress, [In, Out] byte[] buffer, UInt32 size, out IntPtr lpNumberOfBytesRead);
202
203	// BOOL WriteProcessMemory(
204	// HANDLE hProcess, // handle to process
205	// LPVOID lpBaseAddress, // base of memory area
206	// LPCVOID lpBuffer, // data buffer
207	// SIZE_T nSize, // count of bytes to write
208	// SIZE_T * lpNumberOfBytesWritten // count of bytes written
209	// );
210	[DllImport("kernel32.dll")]
211	public static extern Int32 WriteProcessMemory(IntPtr hProcess, UIntPtr lpBaseAddress, [In, Out] byte[] buffer, UInt32 size, out IntPtr lpNumberOfBytesWritten);
212
213
214	}
215
216	#region IMemoryReader Members
217	public bool ReadFirstNonZeroByte(uint MemoryAddress, uint bytesToRead, out uint address)
218	{
219	//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));
220	address = 0;
221	uint byte_alignment = 1;
222	// get common init parameters
223	//InitMemoryDump(out byte_alignment);
224	uint mem_address = MemoryAddress;
225	uint _bytesToRead = bytesToRead;
226	byte[] buffer = new byte[] { };
227	try
228	{
229	//using (MemoryStream ms = new MemoryStream())
230	//{
231	// //BinaryWriter bw = new BinaryWriter(ms);
232	// //foreach (byte b in data) { bw.Write(b); }
233	for (uint i = 0; i <= bytesToRead; )
234	{
235	if (_bytesToRead < byte_alignment)
236	{
237	_bytesToRead = bytesToRead;
238	buffer = new byte[_bytesToRead];
239	}
240	else
241	{
242	_bytesToRead = byte_alignment;
243	buffer = new byte[byte_alignment];
244	}
245	int bytesRead = 0;
246	ReadProcessMemory(mem_address, _bytesToRead, out bytesRead, out buffer);
247	//bw.Write(buffer);
248	//bw.Flush();
249	if (_bytesToRead < byte_alignment)
250	{
251	i += _bytesToRead;
252	mem_address += (uint)_bytesToRead;
253	}
254	else
255	{
256	i += byte_alignment;
257	mem_address += (uint)byte_alignment;
258	}
259	for (uint j = 0; j < buffer.Length; j++)
260	{
261	if (buffer[j] != 0)
262	{
263	address = mem_address;
264	break;
265	}
266	}
267	if (address != 0)
268	break;
269	}
270	// bw.Close();
271	//}
272	//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));
273	return true;
274	}
275	catch (OutOfMemoryException ex)
276	{
277	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));
278	logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
279	logger.Error.WriteLine(ex.ToString());
280	}
281	catch (Exception ex)
282	{
283	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));
284	logger.Error.WriteLine("DumpMemory(): Exception");
285	logger.Error.WriteLine(ex.ToString());
286	}
287	return false;
288	}
289	public void ReadProcessMemory(uint MemoryAddress, uint bytesToRead, out int bytesRead, out byte[] data)
290	{
291	byte[] buffer = new byte[] { };
292	IntPtr ptrBytesRead;
293	ProcessMemoryReader.ProcessMemoryReaderApi.ReadProcessMemory(m_hProcess, (UIntPtr)MemoryAddress, buffer, bytesToRead, out ptrBytesRead);
294	bytesRead = ptrBytesRead.ToInt32();
295	data = buffer;
296	}
297	#endregion
298
299	#region IMemoryWriter Members
300	public void WriteProcessMemory(uint MemoryAddress, byte byteToWrite, out int bytesWritten)
301	{
302	WriteProcessMemory(MemoryAddress, new byte[] { byteToWrite }, out bytesWritten);
303	}
304	public void WriteProcessMemory(uint MemoryAddress, byte[] bytesToWrite, out int bytesWritten)
305	{
306	IntPtr ptrBytesWritten;
307	ProcessMemoryReaderApi.WriteProcessMemory(m_hProcess, (UIntPtr)MemoryAddress, bytesToWrite, (uint)bytesToWrite.Length, out ptrBytesWritten);
308	bytesWritten = ptrBytesWritten.ToInt32();
309	}
310	#endregion
311
312	#region IFileWriter Members
313
314	public bool WriteProcessMemoryToFile(string filename, uint MemoryAddress, uint bytesToRead, out int bytesRead)
315	{
316	//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));
317	bytesRead = 0;
318	uint byte_alignment = 102400;
319	uint address = MemoryAddress;
320	uint _bytesToRead = bytesToRead;
321	byte[] buffer = new byte[] { };
322	try
323	{
324	FileInfo fi = new FileInfo(filename);
325	if (fi.Exists)
326	fi.Delete();
327	using (FileStream fs = new FileStream(filename, FileMode.CreateNew, FileAccess.ReadWrite, FileShare.ReadWrite))
328	{
329	BinaryWriter bw = new BinaryWriter(fs);
330	//foreach (byte b in data) { bw.Write(b); }
331
332	for (uint i = 0; i <= bytesToRead; )
333	{
334	if (_bytesToRead < byte_alignment)
335	{
336	_bytesToRead = bytesToRead;
337	buffer = new byte[_bytesToRead];
338	}
339	else
340	{
341	_bytesToRead = byte_alignment;
342	buffer = new byte[byte_alignment];
343	}
344	ReadProcessMemory(address, _bytesToRead, out bytesRead, out buffer);
345	bw.Write(buffer);
346	bw.Flush();
347
348	if (_bytesToRead < byte_alignment)
349	{
350	i += _bytesToRead;
351	address += (uint)_bytesToRead;
352	}
353	else
354	{
355	i += byte_alignment;
356	address += (uint)byte_alignment;
357	}
358
359
360	}
361	bw.Close();
362	}
363	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));
364	return true;
365	}
366	catch (OutOfMemoryException ex)
367	{
368	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));
369	logger.Error.WriteLine("DumpMemory(): OutOfMemoryException");
370	logger.Error.WriteLine(ex.ToString());
371	}
372	catch (Exception ex)
373	{
374	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));
375	logger.Error.WriteLine("DumpMemory(): Exception");
376	logger.Error.WriteLine(ex.ToString());
377	}
378	return false;
379	}
380
381	#endregion
382	}
383	#endregion
384	}