Merge pull request #190 from PabloLec/main

Add TOTP URIs

Author: bee-san

fixtures/file

@@ -91,4 +91,6 @@ nano_1c46rz7xnk98ozhzdjq7thwty844sgnqxk9496yysit1bnio1rcdzshc5ymn
 
 b2ced6f5-2542-4f7d-b131-e3ada95d8b75
 5fc7c33a7ef88b139122a38a
-01ERJ58HMWDN3VTRRHZQV2T5R5
+01ERJ58HMWDN3VTRRHZQV2T5R5
+
+otpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&issuer=Example

pywhat/Data/regex.json

@@ -971,6 +971,19 @@
          "Nano"
       ]
    },
+   {
+    "Name": "Time-Based One-Time Password (TOTP) URI",
+    "Regex": "^([^?\\n]+:\/\/totp\/[^\\n]+secret=[A-Z2-7=]+[^\\n]+)$",
+    "plural_name": false,
+    "Description": null,
+    "Rarity": 1,
+    "URL": null,
+    "Tags": [
+         "Credentials",
+         "TOTP",
+         "Bug Bounty"
+    ]
+   },
    {
       "Name": "Google Cloud Platform API Key",
       "Regex": "(?i)^([0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12})$",

tests/test_click.py

@@ -618,3 +618,10 @@ def test_file_fixture_ulid():
     result = runner.invoke(main, ["fixtures/file"])
     assert result.exit_code == 0
     assert re.findall("ULID", str(result.output))
+
+
+def test_file_fixture_totp_URI():
+    runner = CliRunner()
+    result = runner.invoke(main, ["fixtures/file"])
+    assert result.exit_code == 0
+    assert re.findall("Time-Based One-Time Password [(]TOTP[)] URI", str(result.output))

tests/test_regex_identifier.py

@@ -1013,3 +1013,21 @@ def test_objectid():
 def test_ulid():
     res = r.check(["01ERJ58HMWDN3VTRRHZQV2T5R5"])
     assert "ULID" in str(res)
+
+
+def test_totp_URI():
+    res = r.check(
+        [
+            "otpauth://totp/Example:alice@google.com?secret=JBSWY3DPEHPK3PXP&issuer=Example"
+        ]
+    )
+    _assert_match_first_item("Time-Based One-Time Password (TOTP) URI", res)
+
+
+def test_complex_totp_URI():
+    res = r.check(
+        [
+            "otpauth://totp/My_Ex0T1c-L00king.name?issuer=ArgsShouldBeCommutable&secret=JBSWY3DPEHPK3PXP&digits=6&period=30"
+        ]
+    )
+    _assert_match_first_item("Time-Based One-Time Password (TOTP) URI", res)