-
Notifications
You must be signed in to change notification settings - Fork 33
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
ae259d9
commit 222e05d
Showing
11 changed files
with
507 additions
and
176 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,159 @@ | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
| using System.Security.Cryptography; | ||
| using System.Xml; | ||
| using static HuaweiUnlocker.LangProc; | ||
|
|
||
| namespace HuaweiUnlocker.DIAGNOS | ||
| { | ||
| public class Bootloader | ||
| { | ||
| public class Image | ||
| { | ||
| private bool? valid = null; | ||
| private int? size = null; | ||
|
|
||
| public string Path { get; } | ||
| public string Role { get; } | ||
| public uint Address { get; } | ||
| private string Hash { get; } | ||
| public bool IsValid { get => valid ?? Validate(); } | ||
| public int Size { get => size ?? GetSize(); } | ||
|
|
||
| private bool Validate() | ||
| { | ||
| if (Hash == null) | ||
| { | ||
| return true; | ||
| } | ||
|
|
||
| using (var stream = File.OpenRead(Path)) { | ||
| using (var sha1 = SHA1.Create()) | ||
| { | ||
| stream.Position = 0; | ||
| byte[] hash = sha1.ComputeHash(stream); | ||
| stream.Close(); | ||
| valid = BitConverter.ToString(hash).Replace("-", "").ToLower() == Hash; | ||
| } | ||
| } | ||
|
|
||
| return valid.Value; | ||
| } | ||
|
|
||
| private int GetSize() | ||
| { | ||
| size = (int)new FileInfo(Path).Length; | ||
|
|
||
| return size.Value; | ||
| } | ||
|
|
||
| public Image(string path, string role, uint address, string hash = null) | ||
| { | ||
| Path = path; | ||
| Role = role; | ||
| Address = address; | ||
| Hash = hash; | ||
| } | ||
| } | ||
|
|
||
| public Image[] Images { get; } | ||
| public string Title { get; } | ||
| public string Name { get; } | ||
|
|
||
| public Bootloader(string name, string title, Image[] images) | ||
| { | ||
| Title = title; | ||
| Name = name; | ||
| Images = images; | ||
| } | ||
|
|
||
| private static uint ParseAddress(string str) => Convert.ToUInt32(str, str.StartsWith("0x") ? 16 : 10); | ||
|
|
||
| private static XmlElement GetRootFromFile(string filename) | ||
| { | ||
| var xml = new XmlDocument(); | ||
| xml.Load(filename); | ||
|
|
||
| var root = xml.DocumentElement; | ||
|
|
||
| if (root.Name != "bootloader") | ||
| { | ||
| throw new Exception("XML root name is invalid."); | ||
| } | ||
|
|
||
| return root; | ||
| } | ||
|
|
||
| private readonly static string[] requiredStrings = new[] { "path", "role", "hash", "address" }; | ||
|
|
||
| public static Bootloader ParseBootloader(string filename) | ||
| { | ||
| var root = GetRootFromFile(filename); | ||
| var dir = Path.GetDirectoryName(filename); | ||
|
|
||
| var title = root.GetAttribute("name"); | ||
|
|
||
| if (string.IsNullOrEmpty(title)) | ||
| { | ||
| LOG(2, "Name attribute is invalid!"); | ||
| title = "Unknown bootloader"; | ||
| } | ||
|
|
||
| var images = new List<Image>(); | ||
|
|
||
| foreach (XmlNode node in root) | ||
| { | ||
| bool isBad = node.Name != "image"; | ||
|
|
||
| foreach (var key in requiredStrings) | ||
| { | ||
| var item = node.Attributes.GetNamedItem(key); | ||
| isBad |= item == null || string.IsNullOrWhiteSpace(item.Value); | ||
| } | ||
|
|
||
| if (isBad) | ||
| { | ||
| throw new Exception("Failed to parse image"); | ||
| } | ||
|
|
||
| images.Add(new Image( | ||
| Path.Combine(dir, node.Attributes.GetNamedItem("path").Value), | ||
| node.Attributes.GetNamedItem("role").Value, | ||
| ParseAddress(node.Attributes.GetNamedItem("address").Value), | ||
| node.Attributes.GetNamedItem("hash").Value | ||
| )); | ||
| } | ||
|
|
||
| return new Bootloader(Path.GetFileName(dir), title, images.ToArray()); | ||
| } | ||
|
|
||
| public static Bootloader[] GetBootloaders() | ||
| { | ||
| var bootloaders = new List<Bootloader>(); | ||
| var root = Path.Combine(Environment.CurrentDirectory, "bootloaders"); | ||
|
|
||
| if (!Directory.Exists(root)) | ||
| { | ||
| return new Bootloader[] { }; | ||
| } | ||
|
|
||
| var dirs = Directory.GetDirectories(root); | ||
|
|
||
| foreach (var dir in dirs) | ||
| { | ||
| var manifest = Path.Combine(dir, "manifest.xml"); | ||
|
|
||
| if (!File.Exists(manifest)) | ||
| { | ||
| LOG(2, $"{Path.GetFileName(dir)}: Manifest not found"); | ||
| continue; | ||
| } | ||
|
|
||
| bootloaders.Add(ParseBootloader(manifest)); | ||
| } | ||
|
|
||
| return bootloaders.ToArray(); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.