Merge 03c2b76 into bb51e3e

pastepwn/analyzers/__init__.py

@@ -28,6 +28,7 @@
 from .adobekeyanalyzer import AdobeKeyAnalyzer
 from .facebookaccesstokenanalyzer import FacebookAccessTokenAnalyzer
 from .base64analyzer import Base64Analyzer
+from .stripeapikeyanalyzer import StripeApiKeyAnalyzer
 
 __all__ = (
     'AlwaysTrueAnalyzer',
@@ -57,5 +58,6 @@
     'PrivateKeyAnalyzer',
     'EmailPasswordPairAnalyzer',
     'FacebookAccessTokenAnalyzer',
-    'Base64Analyzer'
+    'Base64Analyzer',
+    'StripeApiKeyAnalyzer'
 )

pastepwn/analyzers/stripeapikeyanalyzer.py

@@ -0,0 +1,8 @@
+from .regexanalyzer import RegexAnalyzer
+
+
+class StripeApiKeyAnalyzer(RegexAnalyzer):
+
+    def __init__(self, actions):
+        regex = r'(?:r|s)k_(live|test)_[0-9a-zA-Z]{24}'
+        super().__init__(actions, regex)

pastepwn/analyzers/tests/stripeapikeyanalyzer_test.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+import unittest
+from unittest import mock
+
+from pastepwn.analyzers.stripeapikeyanalyzer import StripeApiKeyAnalyzer
+
+
+class TestStripeApiKeyAnalyzer(unittest.TestCase):
+
+    def setUp(self):
+        self.analyzer = StripeApiKeyAnalyzer(None)
+        self.paste = mock.Mock()
+
+    def test_match_positive(self):
+        """Test if positives are recognized"""
+        # steam key dump
+        self.paste.body = "sk_test_4fsrdffsdf345345dfgfg34i"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # steam key dump
+        self.paste.body = "rk_test_4fsrdffsdf345345dfgfg34i"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # steam key dump
+        self.paste.body = "sk_live_4fsrdffsdf345345dfgfg34i"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # steam key dump
+        self.paste.body = "rk_live_4fsrdffsdf345345dfgfg34i"
+        self.assertTrue(self.analyzer.match(self.paste))
+
+        # steam key dump
+        self.paste.body = "sk_test_YUTGF76uyh876Tyg87T786Tu"
+        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 = "sk_test_YUTGF76uyh876"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+        # Invalid first letter
+        self.paste.body = "pk_test_YUTGF76uyh876Tyg87T786Tu"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+        # Invalid state
+        self.paste.body = "sk_staging_YUTGF76uyh876Tyg87T786Tu"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+
+if __name__ == '__main__':
+    unittest.main()