monocul.us/de4dot-cex
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add some more methods

Browse Source
This commit is contained in:
de4dot 2011-12-20 20:11:32 +01:00
parent c65b76583c
commit b60d53ea77
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
de4dot.code/PE/PeImage.cs
View File

@ -35,6 +35,10 @@ namespace de4dot.code.PE {
get { return reader; } get { return reader; }
} }
public uint ImageLength {
get { return (uint)reader.BaseStream.Length; }
}
internal Cor20Header Cor20Header { internal Cor20Header Cor20Header {
get { return cor20Header; } get { return cor20Header; }
} }
@ -123,9 +127,8 @@ namespace de4dot.code.PE {
return intersect(offset, length, location.Offset, location.Length); 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) { if (cor20Header != null) {
uint offset = rvaToOffset(rva);
uint length = (uint)data.Length; uint length = (uint)data.Length;
if (!dotNetSection.isInside(offset, length)) if (!dotNetSection.isInside(offset, length))
@ -136,10 +139,14 @@ namespace de4dot.code.PE {
return false; return false;
} }
write(rva, data); offsetWrite(offset, data);
return true; return true;
} }
public bool dotNetSafeWrite(uint rva, byte[] data) {
return dotNetSafeWriteOffset(rvaToOffset(rva), data);
}
public void write(uint rva, byte[] data) { public void write(uint rva, byte[] data) {
seekRva(rva); seekRva(rva);
writer.Write(data); writer.Write(data);
@ -180,6 +187,11 @@ namespace de4dot.code.PE {
return reader.ReadBytes(size); return reader.ReadBytes(size);
} }
public void offsetWrite(uint offset, byte[] data) {
seek(offset);
writer.Write(data);
}
public byte[] offsetReadBytes(uint offset, int size) { public byte[] offsetReadBytes(uint offset, int size) {
seek(offset); seek(offset);
return reader.ReadBytes(size); return reader.ReadBytes(size);