From b60d53ea778c9e576ac41bb3ae200384ffccd3d6 Mon Sep 17 00:00:00 2001
From: de4dot <de4dot@gmail.com>
Date: Tue, 20 Dec 2011 20:11:32 +0100
Subject: [PATCH] Add some more methods

---
 de4dot.code/PE/PeImage.cs | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/de4dot.code/PE/PeImage.cs b/de4dot.code/PE/PeImage.cs
index 959c32cf..0a94ceab 100644
--- a/de4dot.code/PE/PeImage.cs
+++ b/de4dot.code/PE/PeImage.cs
@@ -35,6 +35,10 @@ namespace de4dot.code.PE {
 			get { return reader; }
 		}
 
+		public uint ImageLength {
+			get { return (uint)reader.BaseStream.Length; }
+		}
+
 		internal Cor20Header Cor20Header {
 			get { return cor20Header; }
 		}
@@ -123,9 +127,8 @@ namespace de4dot.code.PE {
 			return intersect(offset, length, location.Offset, location.Length);
 		}
 
-		public bool dotNetSafeWrite(uint rva, byte[] data) {
+		public bool dotNetSafeWriteOffset(uint offset, byte[] data) {
 			if (cor20Header != null) {
-				uint offset = rvaToOffset(rva);
 				uint length = (uint)data.Length;
 
 				if (!dotNetSection.isInside(offset, length))
@@ -136,10 +139,14 @@ namespace de4dot.code.PE {
 					return false;
 			}
 
-			write(rva, data);
+			offsetWrite(offset, data);
 			return true;
 		}
 
+		public bool dotNetSafeWrite(uint rva, byte[] data) {
+			return dotNetSafeWriteOffset(rvaToOffset(rva), data);
+		}
+
 		public void write(uint rva, byte[] data) {
 			seekRva(rva);
 			writer.Write(data);
@@ -180,6 +187,11 @@ namespace de4dot.code.PE {
 			return reader.ReadBytes(size);
 		}
 
+		public void offsetWrite(uint offset, byte[] data) {
+			seek(offset);
+			writer.Write(data);
+		}
+
 		public byte[] offsetReadBytes(uint offset, int size) {
 			seek(offset);
 			return reader.ReadBytes(size);