Implement Battle.Net Key Analyzer

pastepwn/analyzers/__init__.py

@@ -18,6 +18,7 @@
 from .phonenumberanalyzer import PhoneNumberAnalyzer
 from .steamkeyanalyzer import SteamKeyAnalyzer
 from .uplaykeyanalyzer import UplayKeyAnalyzer
+from .battlenetanalyzer import BattleNetKeyAnalyzer
 
 __all__ = (
     'AlwaysTrueAnalyzer',
@@ -38,5 +39,6 @@
     'PrivateKeyAnalyzer',
     'PhoneNumberAnalyzer',
     'SteamKeyAnalyzer',
-    'UplayKeyAnalyzer'
+    'UplayKeyAnalyzer',
+    'BattleNetKeyAnalyzer'
 )

pastepwn/analyzers/battlenetkeyanalyzer.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+from .regexanalyzer import RegexAnalyzer
+
+
+class BattleNetKeyAnalyzer(RegexAnalyzer):
+    """
+    Analyzer to match battle.net keys via regex
+    """
+    name = "BattleNetKeyAnalyzer"
+
+    def __init__(self, actions):
+        # battle.net activation page shows 4-4-4-4 format, but there are also codes in the 4-4-5-4-4 format which work
+        regex = r"\b([A-Z0-9]{4}\-[A-Z0-9]{4}\-[A-Z0-9]{4}\-[A-Z0-9]{4}|[A-Z0-9]{4}\-[A-Z0-9]{4}\-[A-Z0-9]{5}\-[A-Z0-9]{4}\-[A-Z0-9]{4})\b"
+        super().__init__(actions, regex)

pastepwn/analyzers/tests/battlenetkeyanalyzer_test.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+import unittest
+from unittest import mock
+
+from pastepwn.analyzers.battlenetkeyanalyzer import BattleNetKeyAnalyzer
+
+
+class TestBattleNetKeyAnalyzer(unittest.TestCase):
+    def setUp(self):
+        self.analyzer = BattleNetKeyAnalyzer(None)
+        self.paste = mock.Mock()
+
+    def test_match_positive(self):
+        """Test if positives are recognized"""
+        # working battlenet key in 4-4-5-4-4 format
+        self.paste.body = "NGM5-XB9P-8TLTR-V8BH-QFSW"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # key in 4-4-4-4 format
+        self.paste.body = "MB9C-LV3C-4RG8-FME8"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # part of a sentence
+        self.paste.body = "Hey, I have your key right here: MB9C-LV3C-4RG8-FME8!"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # Newline seperated keys
+        self.paste.body = "MB9C-LV3C-4RG8-FME8\nNGM5-XB9P-8TLTR-V8BH-QFSW"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+    def test_match_negative(self):
+        """Test if negatives are not recognized"""
+        self.paste.body = ""
+        self.assertFalse(self.analyzer.match(self.paste))
+
+        self.paste.body = None
+        self.assertFalse(self.analyzer.match(self.paste))
+
+        # Invalid length
+        self.paste.body = "MB9AC-LV3C-4RG8-FME8"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+        # Lower-case
+        self.paste.body = "mb9c-lv3c-4rg8-fme8"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+
+if __name__ == '__main__':
+    unittest.main()