diff options
Diffstat (limited to 'utils/ipn_tool/Watermarks.cs')
| -rw-r--r-- | utils/ipn_tool/Watermarks.cs | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/utils/ipn_tool/Watermarks.cs b/utils/ipn_tool/Watermarks.cs new file mode 100644 index 0000000..3ed47f5 --- /dev/null +++ b/utils/ipn_tool/Watermarks.cs @@ -0,0 +1,111 @@ +using System; +using System.Collections.Generic; +using System.IO; +using System.Linq; +using System.Text.RegularExpressions; +using System.Xml; + +namespace ipn_tool +{ + internal static class Watermarks + { + internal static int CheckIfPresent(string xmlDb, string id, string binFile) + { + try + { + var wm = LoadWm(xmlDb, id); + return FindWm(wm, binFile) ? 0 : 1; + } + catch (Exception ex) + { + Console.Error.WriteLine(ex); + return 1; + } + } + + private static string LoadWm(string xmlDb, string id) + { + var s = new FileStream(xmlDb, FileMode.Open, FileAccess.Read); + using (var x = new XmlTextReader(s)) + { + while (x.Read()) + { + if (x.NodeType == XmlNodeType.Element && x.Name == "Watermark" && x.GetAttribute("Name") == id) + { + if(x.Read() && x.NodeType == XmlNodeType.Text && !string.IsNullOrEmpty(x.Value)) + return x.Value; + } + } + } + throw new KeyNotFoundException("WM.id=" + id); + } + + private static bool FindWm(string wm, string binFile) + { + var wmna = WmStringToNibbleArray(wm.ToUpper()); + var filesToCheck = new List<string>(); + if (Directory.Exists(binFile)) + { + filesToCheck.AddRange(Directory.EnumerateFiles(binFile)); + } + else + { + filesToCheck.Add(binFile); + } + return filesToCheck.All(s => CheckFile(s, wmna)); + } + + private static bool CheckFile(string binFile, int[] wmna) + { + var filena = new List<int>(wmna.Length + 1); + using (var s = new FileStream(binFile, FileMode.Open, FileAccess.Read)) + { + while (true) + { + var nextByte = s.ReadByte(); + if (nextByte == -1) + { + Console.Error.WriteLine("Watermark was not found in {0}", binFile); + return false; + } + filena.Add(nextByte >> 4); + filena.Add(nextByte & 15); + while (filena.Count >= wmna.Length) + { + if (MatchWm(wmna, filena)) + { + Console.WriteLine("Watermark in {0}: FOUND", binFile); + return true; + } + filena.RemoveAt(0); + } + } + } + } + + private static bool MatchWm(IEnumerable<int> wmna, IList<int> filena) + { + return !wmna.Where((t, i) => t != -1 && t != filena[i]).Any(); + } + + private static int[] WmStringToNibbleArray(string wm) + { + if(!Regex.IsMatch(wm, @"[A-Z0-9\?]+")) + throw new ArgumentException("Watermark contains bad symbols: " + wm, "wm"); + + var ret = new int[wm.Length]; + var idx = 0; + foreach (var ch in wm) + { + if (ch == '?') + ret[idx++] = -1; + else if (ch <= '9') + ret[idx++] = ch - '0'; + else + ret[idx++] = ch - 'A' + 10; + } + return ret; + } + + } +}
\ No newline at end of file |