Merge 67bda70 into 0555d02

pastepwn/analyzers/__init__.py

@@ -12,6 +12,7 @@
 from .urlanalyzer import URLAnalyzer
 from .wordanalyzer import WordAnalyzer
 from .ibananalyzer import IBANAnalyzer
+from .dbconnstringanalyzer import DBConnAnalyzer
 from .privatekeyanalyzer import PrivateKeyAnalyzer
 
 __all__ = (
@@ -28,5 +29,6 @@
     'URLAnalyzer',
     'WordAnalyzer',
     'IBANAnalyzer',
+    'DBConnAnalyzer'
     'PrivateKeyAnalyzer'
 )

pastepwn/analyzers/dbconnstringanalyzer.py

@@ -0,0 +1,12 @@
+# -*- coding: utf-8 -*-
+from .regexanalyzer import RegexAnalyzer
+
+
+class DBConnAnalyzer(RegexAnalyzer):
+    """Analyzer to match on email addresses via regex"""
+    name = "DBConnAnalyzer"
+
+    def __init__(self, actions):
+        # https:// should be ignored
+        regex = r"(\b(?!http(s)?\b)\w+[a-zA-Z]+://[a-zA-z0-9.:,]+)"
+        super().__init__(actions, regex)

pastepwn/analyzers/tests/dbconnstringanalyzer_test.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+import unittest
+from unittest import mock
+
+from pastepwn.analyzers.dbconnstringanalyzer import DBConnAnalyzer
+
+
+class TestDBConnAnalyzer(unittest.TestCase):
+    def setUp(self):
+        self.analyzer = DBConnAnalyzer(None)
+        self.paste = mock.Mock()
+
+    def test_match_positive(self):
+        """Test if positives are recognized"""
+        self.paste.body = "cassandra://194.3.5.2"
+        self.assertTrue(self.analyzer.match(self.paste))
+        self.paste.body = "postgresql://user:secret@localhost"
+        self.assertTrue(self.analyzer.match(self.paste))
+        self.paste.body = "mongodb://router1.example.com:27017"
+        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))
+        self.paste.body = "https://www.google.com"
+        self.assertFalse(self.analyzer.match(self.paste))
+        self.paste.body = "http://www.google.com"
+        self.assertFalse(self.analyzer.match(self.paste))
+
+
+if __name__ == '__main__':
+    unittest.main()