1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
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;
}
}
}
|
