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

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

some KeySearcher changes

File size: 13.2 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                int blocksize = CostMaster.getBlocksize();
264                string key;
265                do
266                {
267                    key = Pattern.getKey();
268                    GuiLogMessage("Try key " + key, NotificationLevel.Debug);
269                    byte[] decryption = sender.Decrypt(ControlMaster.getKeyFromString(key), blocksize);
270
271                    ValueKey valueKey = new ValueKey();
272                    valueKey.value = CostMaster.calculateCost(decryption);
273                    valueKey.key = key;
274
275                    if (this.costMaster.getRelationOperator() == RelationOperator.LargerThen)
276                    {
277
278                        LinkedListNode<ValueKey> node = costList.First;
279
280                        while (node != null)
281                        {
282
283                            if (valueKey.value > node.Value.value)
284                            {
285                                costList.AddBefore(node, valueKey);
286                                break;
287                            }
288                            node = node.Next;
289                        }
290                    }
291                    else
292                    {
293                        LinkedListNode<ValueKey> node = costList.First;
294
295                        while (node != null)
296                        {
297
298                            if (valueKey.value < node.Value.value)
299                            {
300                                costList.AddBefore(node, valueKey);
301                                break;
302                            }
303                            node = node.Next;
304                        }
305                    }
306                    if (costList.Count > maxInList)
307                    {
308                        costList.RemoveLast();
309                    }
310
311                    counter++;
312                    ProgressChanged(counter, size);
313                } while (Pattern.nextKey() && !stop);
314
315                GuiLogMessage("Calculated value/key - list:", NotificationLevel.Info);
316                LinkedListNode<ValueKey> n = costList.First;
317                while (n != null)
318                {
319                    GuiLogMessage(n.Value.value + " = " + n.Value.key, NotificationLevel.Info);
320                    n = n.Next;
321                }
322
323            }//end if
324        }
325
326        public void PostExecution()
327        {
328        }
329
330        public void Pause()
331        {
332        }
333
334        public void Stop()
335        {
336            stop = true;
337        }
338
339        public void Initialize()
340        {
341        }
342
343        public void Dispose()
344        {
345        }
346
347        #endregion
348
349        #region INotifyPropertyChanged Members
350
351        public event PropertyChangedEventHandler PropertyChanged;
352
353        public void OnPropertyChanged(string name)
354        {
355            if (PropertyChanged != null)
356            {
357                PropertyChanged(this, new PropertyChangedEventArgs(name));
358            }
359        }
360
361        #endregion
362
363        private void keyPatternChanged()
364        {
365            Pattern = new KeyPattern(controlMaster.getKeyPattern());
366        }
367
368        private void onStatusChanged(IControl sender, bool readyForExecution)
369        {
370            if (readyForExecution)
371            {
372                this.process((IControlEncryption)sender);
373            }
374        }
375
376        #region IControlEncryption Members
377
378        private IControlEncryption controlMaster;
379        [PropertyInfo(Direction.ControlMaster, "Control Master", "Used for bruteforcing", "", DisplayLevel.Beginner)]
380        public IControlEncryption ControlMaster
381        {
382            get { return controlMaster; }
383            set
384            {
385                if (controlMaster != null)
386                {
387                    controlMaster.keyPatternChanged -= keyPatternChanged;
388                    controlMaster.OnStatusChanged -= onStatusChanged;
389                }
390                if (value != null)
391                {
392                    Pattern = new KeyPattern(value.getKeyPattern());
393                    value.keyPatternChanged += keyPatternChanged;
394                    value.OnStatusChanged += onStatusChanged;
395                    controlMaster = value;
396                    OnPropertyChanged("ControlMaster");
397                   
398                }
399                else
400                    controlMaster = null;
401            }
402        }
403
404        #endregion
405
406        #region IControlCost Members
407
408        private IControlCost costMaster;
409        [PropertyInfo(Direction.ControlMaster, "Cost Master", "Used for cost calculation", "", DisplayLevel.Beginner)]
410        public IControlCost CostMaster
411        {
412            get { return costMaster; }
413            set
414            {
415                costMaster = value;
416            }
417        }
418
419        #endregion
420
421        public void GuiLogMessage(string message, NotificationLevel loglevel)
422        {
423            if (OnGuiLogNotificationOccured != null)
424                OnGuiLogNotificationOccured(this, new GuiLogEventArgs(message, this, loglevel));
425        }
426
427        public void ProgressChanged(double value, double max)
428        {
429            if (OnPluginProgressChanged != null)
430            {
431                OnPluginProgressChanged(this, new PluginProgressEventArgs(value, max));
432            }
433        }
434
435        private struct ValueKey
436        {
437            public double value;
438            public String key;
439        };
440    }
441}
Note: See TracBrowser for help on using the repository browser.