source: trunk/CrypPlugins/KeySearcher/KeySearcher.cs @ 716

Last change on this file since 716 was 716, checked in by Sven Rech, 12 years ago

added ProgressChange to KeySearcher

File size: 13.1 KB
Line 
1using System;
2using System.Linq;
3using System.Text;
4using Cryptool.PluginBase.Analysis;
5using Cryptool.PluginBase;
6using System.Windows.Controls;
7using System.ComponentModel;
8using Cryptool.PluginBase.Control;
9using System.Collections;
10using System.Collections.Generic;
11
12namespace KeySearcher
13{
14    public class KeyPattern
15    {
16        private class Wildcard
17        {
18            private char[] values = new char[256];
19            private int length;
20            private int counter;           
21
22            public Wildcard(string valuePattern)
23            {               
24                counter = 0;
25                length = 0;
26                int i = 1;
27                while (valuePattern[i] != ']')
28                {
29                    if (valuePattern[i + 1] == '-')
30                    {
31                        for (char c = valuePattern[i]; c <= valuePattern[i + 2]; c++)
32                            values[length++] = c;
33                        i += 2;
34                    }
35                    else
36                        values[length++] = valuePattern[i];
37                    i++;
38                }
39            }
40
41            public char getChar()
42            {               
43                return (char)values[counter];
44            }
45
46            public bool succ()
47            {
48                counter++;
49                if (counter >= length)
50                {
51                    counter = 0;
52                    return true;
53                }
54                return false;
55            }
56
57            public int Size()
58            {
59                return length;
60            }
61
62        }
63
64        private string pattern;
65        private string key;       
66        private ArrayList wildcardList;
67
68        public KeyPattern(string pattern)
69        {
70            this.pattern = pattern;
71        }
72
73        public string giveWildcardKey()
74        {
75            string res = "";
76            int i = 0;
77            while (i < pattern.Length)
78            {
79                if (pattern[i] != '[')
80                    res += pattern[i];
81                else
82                {
83                    res += '*';
84                    while (pattern[i] != ']')
85                        i++;
86                }
87                i++;
88            }
89            return res;
90        }
91
92        public bool testKey(string key)
93        {
94            int kcount = 0;
95            int pcount = 0;
96            while (kcount < key.Length && pcount < pattern.Length)
97            {
98                if (pattern[pcount] != '[')
99                {
100                    if (key[kcount] != '*' && pattern[pcount] != key[kcount])
101                        return false;
102                    kcount++;
103                    pcount++;
104                }
105                else
106                {
107                    bool contains = false;
108                    pcount++;
109                    while (pattern[pcount] != ']')
110                    {
111                        if (key[kcount] != '*')
112                        {
113                            if (pattern[pcount + 1] == '-')
114                            {
115                                if (key[kcount] >= pattern[pcount] && key[kcount] <= pattern[pcount + 2])
116                                    contains = true;
117                                pcount += 2;
118                            }
119                            else
120                                if (pattern[pcount] == key[kcount])
121                                    contains = true;
122                        }
123                        pcount++;
124                    }
125                    if (!contains && !(key[kcount] == '*'))
126                        return false;
127                    kcount++;
128                    pcount++;
129                }               
130            }
131            if (pcount != pattern.Length || kcount != key.Length)
132                return false;
133            return true;
134        }
135
136        public int initKeyIteration(string key)
137        {
138            int counter = 0;
139            this.key = key;
140            int pcount = 0;
141            wildcardList = new ArrayList();
142            for (int i = 0; i < key.Length; i++)
143            {
144                if (key[i] == '*')
145                {
146                    Wildcard wc = new Wildcard(pattern.Substring(pcount, pattern.IndexOf(']', pcount) + 1 - pcount));
147                    wildcardList.Add(wc);
148                    counter += wc.Size();
149                }
150
151                if (pattern[pcount] == '[')
152                    while (pattern[pcount] != ']')
153                        pcount++;
154                pcount++;
155            }
156            return counter;
157        }
158
159        public bool nextKey()
160        {
161            int wildcardCount = wildcardList.Count-1;
162            bool overflow = ((Wildcard)wildcardList[wildcardCount]).succ();
163            wildcardCount--;
164            while (overflow && (wildcardCount >= 0))
165                overflow = ((Wildcard)wildcardList[wildcardCount--]).succ();
166            return !overflow;
167        }
168
169        public string getKey()
170        {
171            string res = "";
172            int wildcardCount = 0;
173            for (int i = 0; i < key.Length; i++)
174            {
175                if (key[i] != '*')
176                    res += key[i];
177                else
178                {
179                    Wildcard wc = (Wildcard)wildcardList[wildcardCount++];
180                    res += wc.getChar();
181                }
182            }
183            return res;
184        }
185    }
186   
187    [Author("Thomas Schmid", "thomas.schmid@cryptool.org", "Uni Siegen", "http://www.uni-siegen.de")]
188    //[Author("Sven Rech", "rech@cryptool.org", "Uni Duisburg-Essen", "http://www.uni-due.de")]
189    [PluginInfo(true, "KeySearcher", "Bruteforces a decryption algorithm.", null, "KeySearcher/Images/icon.png")]
190    public class KeySearcher : IAnalysisMisc
191    {
192        private KeyPattern pattern = null;
193        public KeyPattern Pattern
194        {
195            get
196            {
197                return pattern;
198            }
199            set
200            {
201                pattern = value;
202                if ((settings.Key == null) ||((settings.Key != null) && !pattern.testKey(settings.Key)))
203                    settings.Key = pattern.giveWildcardKey();
204            }
205        }
206
207        private bool stop;
208
209        #region IPlugin Members
210
211        public event StatusChangedEventHandler OnPluginStatusChanged;
212
213        public event GuiLogNotificationEventHandler OnGuiLogNotificationOccured;
214
215        public event PluginProgressChangedEventHandler OnPluginProgressChanged;
216
217        private KeySearcherSettings settings;       
218
219        public KeySearcher()
220        {
221            settings = new KeySearcherSettings(this);
222        }
223
224        public ISettings Settings
225        {
226            get { return settings; }
227        }
228
229        public UserControl Presentation
230        {
231            get { return null; }
232        }
233
234        public UserControl QuickWatchPresentation
235        {
236            get { return null; }
237        }
238
239        public void PreExecution()
240        {
241        }
242
243        public void Execute()
244        {
245        }
246
247        public void process(IControlEncryption sender)
248        {
249            int counter = 0;
250            if (sender != null && costMaster != null)
251            {
252                int maxInList = 10;
253
254                LinkedList<ValueKey> costList = new LinkedList<ValueKey>();
255
256                stop = false;
257                if (!Pattern.testKey(settings.Key))
258                {
259                    GuiLogMessage("Wrong key pattern!", NotificationLevel.Error);
260                    return;
261                }
262                int size = Pattern.initKeyIteration(settings.Key);
263                string key;
264                do
265                {
266                    key = Pattern.getKey();
267                    GuiLogMessage("Try key " + key, NotificationLevel.Debug);
268                    byte[] decryption = sender.Decrypt(ControlMaster.getKeyFromString(key));
269
270                    ValueKey valueKey = new ValueKey();
271                    valueKey.value = CostMaster.calculateCost(decryption);
272                    valueKey.key = key;
273
274                    if (this.costMaster.getRelationOperator() == RelationOperator.LargerThen)
275                    {
276
277                        LinkedListNode<ValueKey> node = costList.First;
278
279                        while (node != null)
280                        {
281
282                            if (valueKey.value > node.Value.value)
283                            {
284                                costList.AddBefore(node, valueKey);
285                                break;
286                            }
287                            node = node.Next;
288                        }
289                    }
290                    else
291                    {
292                        LinkedListNode<ValueKey> node = costList.First;
293
294                        while (node != null)
295                        {
296
297                            if (valueKey.value < node.Value.value)
298                            {
299                                costList.AddBefore(node, valueKey);
300                                break;
301                            }
302                            node = node.Next;
303                        }
304                    }
305                    if (costList.Count > maxInList)
306                    {
307                        costList.RemoveLast();
308                    }
309
310                    counter++;
311                    ProgressChanged(counter, size);
312                } while (Pattern.nextKey() && !stop);
313
314                GuiLogMessage("Calculated value/key - list:", NotificationLevel.Info);
315                LinkedListNode<ValueKey> n = costList.First;
316                while (n != null)
317                {
318                    GuiLogMessage(n.Value.value + " = " + n.Value.key, NotificationLevel.Info);
319                    n = n.Next;
320                }
321
322            }//end if
323        }
324
325        public void PostExecution()
326        {
327        }
328
329        public void Pause()
330        {
331        }
332
333        public void Stop()
334        {
335            stop = true;
336        }
337
338        public void Initialize()
339        {
340        }
341
342        public void Dispose()
343        {
344        }
345
346        #endregion
347
348        #region INotifyPropertyChanged Members
349
350        public event PropertyChangedEventHandler PropertyChanged;
351
352        public void OnPropertyChanged(string name)
353        {
354            if (PropertyChanged != null)
355            {
356                PropertyChanged(this, new PropertyChangedEventArgs(name));
357            }
358        }
359
360        #endregion
361
362        private void keyPatternChanged()
363        {
364            Pattern = new KeyPattern(controlMaster.getKeyPattern());
365        }
366
367        private void onStatusChanged(IControl sender, bool readyForExecution)
368        {
369            if (readyForExecution)
370            {
371                this.process((IControlEncryption)sender);
372            }
373        }
374
375        #region IControlEncryption Members
376
377        private IControlEncryption controlMaster;
378        [PropertyInfo(Direction.ControlMaster, "Control Master", "Used for bruteforcing", "", DisplayLevel.Beginner)]
379        public IControlEncryption ControlMaster
380        {
381            get { return controlMaster; }
382            set
383            {
384                if (controlMaster != null)
385                {
386                    controlMaster.keyPatternChanged -= keyPatternChanged;
387                    controlMaster.OnStatusChanged -= onStatusChanged;
388                }
389                if (value != null)
390                {
391                    Pattern = new KeyPattern(value.getKeyPattern());
392                    value.keyPatternChanged += keyPatternChanged;
393                    value.OnStatusChanged += onStatusChanged;
394                    controlMaster = value;
395                    OnPropertyChanged("ControlMaster");
396                   
397                }
398                else
399                    controlMaster = null;
400            }
401        }
402
403        #endregion
404
405        #region IControlCost Members
406
407        private IControlCost costMaster;
408        [PropertyInfo(Direction.ControlMaster, "Cost Master", "Used for cost calculation", "", DisplayLevel.Beginner)]
409        public IControlCost CostMaster
410        {
411            get { return costMaster; }
412            set
413            {
414                costMaster = value;
415            }
416        }
417
418        #endregion
419
420        public void GuiLogMessage(string message, NotificationLevel loglevel)
421        {
422            if (OnGuiLogNotificationOccured != null)
423                OnGuiLogNotificationOccured(this, new GuiLogEventArgs(message, this, loglevel));
424        }
425
426        public void ProgressChanged(double value, double max)
427        {
428            if (OnPluginProgressChanged != null)
429            {
430                OnPluginProgressChanged(this, new PluginProgressEventArgs(value, max));
431            }
432        }
433
434        private struct ValueKey
435        {
436            public double value;
437            public String key;
438        };
439    }
440}
Note: See TracBrowser for help on using the repository browser.