Ticket #47: colors.patch

Sources/Controller/ScanController.m

@@ -257 +257 @@
 
 - (id) tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *) aTableColumn row:(int) rowIndex {
     WaveNet *net = [_container netAtIndex:rowIndex];
-    
+    if ([[aTableColumn identifier] isEqualToString: @"wep"]) {
+        NSLog(@"WEP!! ");
+        switch([net crackProb]) {
+            case 0: 
+                [[aTableColumn dataCellForRow: rowIndex] setTextColor:[NSColor redColor]];//hahaha
+                break;
+            case 1:
+                [[aTableColumn dataCellForRow: rowIndex] setTextColor:[NSColor yellowColor]];//possible
+                break;
+            case 2:
+                [[aTableColumn dataCellForRow: rowIndex] setTextColor:[NSColor greenColor]];//good chance
+                break;
+            case 3:
+                [[aTableColumn dataCellForRow: rowIndex] setTextColor:[NSColor blueColor]];//cracked
+                break;
+        }
+    }
         return [[net cache] objectForKey: [aTableColumn identifier]];
 }
 

Sources/Core/WaveNet.mm

@@ -1240 +1240 @@
     return _password != nil;
 }
 
+- (int)crackProb {
+    if (_isWep <= 1 || _password != nil) {
+        NSLog(@"Blue!!");
+        return 3;
+    }
+    if ((_isWep == 2 || _isWep == 3) && [self uniqueIVs] > 8) {
+        return 1;
+    }
+    if ((_isWep == 2 || _isWep == 3) && [self uniqueIVs] > 250000) {
+        return 2;
+    }
+    if (_isWep == 4 && [self capturedEAPOLKeys] != 0 ) {
+        return 2;
+    }
+    
+    return 0;
+}
+
 #pragma mark -
 
 - (NSDictionary*)cache {

Sources/Core/WaveNet.h

@@ -156 +156 @@
 - (NSDictionary*)coordinates;
 - (WaveWeakContainer **)ivData;
 - (BOOL)passwordAvailable;
+- (int)crackProb; // 0=not possible, 1=possible but not likely, 2=you might have a chance 
 
 - (NSDictionary*)cache;