You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.2 KiB
102 lines
2.2 KiB
/*--------------------------------------------------------------------------------------------- |
|
* Copyright (c) Microsoft Corporation. All rights reserved. |
|
* Licensed under the MIT License. See License.txt in the project root for license information. |
|
*--------------------------------------------------------------------------------------------*/ |
|
using System; |
|
using System.IO; |
|
|
|
namespace Microsoft.Unity.VisualStudio.Editor |
|
{ |
|
public sealed class Image : IDisposable |
|
{ |
|
|
|
long position; |
|
Stream stream; |
|
|
|
Image(Stream stream) |
|
{ |
|
this.stream = stream; |
|
this.position = stream.Position; |
|
this.stream.Position = 0; |
|
} |
|
|
|
bool Advance(int length) |
|
{ |
|
if (stream.Position + length >= stream.Length) |
|
return false; |
|
|
|
stream.Seek(length, SeekOrigin.Current); |
|
return true; |
|
} |
|
|
|
bool MoveTo(uint position) |
|
{ |
|
if (position >= stream.Length) |
|
return false; |
|
|
|
stream.Position = position; |
|
return true; |
|
} |
|
|
|
void IDisposable.Dispose() |
|
{ |
|
stream.Position = position; |
|
} |
|
|
|
ushort ReadUInt16() |
|
{ |
|
return (ushort)(stream.ReadByte() |
|
| (stream.ReadByte() << 8)); |
|
} |
|
|
|
uint ReadUInt32() |
|
{ |
|
return (uint)(stream.ReadByte() |
|
| (stream.ReadByte() << 8) |
|
| (stream.ReadByte() << 16) |
|
| (stream.ReadByte() << 24)); |
|
} |
|
|
|
bool IsManagedAssembly() |
|
{ |
|
if (stream.Length < 318) |
|
return false; |
|
if (ReadUInt16() != 0x5a4d) |
|
return false; |
|
if (!Advance(58)) |
|
return false; |
|
if (!MoveTo(ReadUInt32())) |
|
return false; |
|
if (ReadUInt32() != 0x00004550) |
|
return false; |
|
if (!Advance(20)) |
|
return false; |
|
if (!Advance(ReadUInt16() == 0x20b ? 222 : 206)) |
|
return false; |
|
|
|
return ReadUInt32() != 0; |
|
} |
|
|
|
public static bool IsAssembly(string file) |
|
{ |
|
if (file == null) |
|
throw new ArgumentNullException("file"); |
|
|
|
using (var stream = new FileStream(file, FileMode.Open, FileAccess.Read, FileShare.Read)) |
|
return IsAssembly(stream); |
|
} |
|
|
|
public static bool IsAssembly(Stream stream) |
|
{ |
|
if (stream == null) |
|
throw new ArgumentNullException(nameof(stream)); |
|
if (!stream.CanRead) |
|
throw new ArgumentException(nameof(stream)); |
|
if (!stream.CanSeek) |
|
throw new ArgumentException(nameof(stream)); |
|
|
|
using (var image = new Image(stream)) |
|
return image.IsManagedAssembly(); |
|
} |
|
} |
|
}
|
|
|