Browse Source

[enh] add plugin converting strings into hash digests

Venca24 7 years ago
parent
commit
c9593c8ffd
4 changed files with 104 additions and 1 deletions
  1. 1 1
      AUTHORS.rst
  2. 2 0
      searx/plugins/__init__.py
  3. 54 0
      searx/plugins/hash_plugin.py
  4. 47 0
      tests/unit/test_plugins.py

+ 1 - 1
AUTHORS.rst

@@ -96,7 +96,7 @@ generally made searx better:
 - Alice Ferrazzi @aliceinwire
 - @LiquidLemon
 - @dadosch
-- @Venca24
+- Václav Zouzalík @Venca24
 - @ZEROF
 - Ivan Skytte Jørgensen @isj-privacore
 - @miicha

+ 2 - 0
searx/plugins/__init__.py

@@ -28,6 +28,7 @@ from searx import logger, settings, static_path
 logger = logger.getChild('plugins')
 
 from searx.plugins import (oa_doi_rewrite,
+                           hash_plugin,
                            https_rewrite,
                            infinite_scroll,
                            self_info,
@@ -163,6 +164,7 @@ def sha_sum(filename):
 
 plugins = PluginStore()
 plugins.register(oa_doi_rewrite)
+plugins.register(hash_plugin)
 plugins.register(https_rewrite)
 plugins.register(infinite_scroll)
 plugins.register(self_info)

+ 54 - 0
searx/plugins/hash_plugin.py

@@ -0,0 +1,54 @@
+'''
+searx is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+searx is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with searx. If not, see < http://www.gnu.org/licenses/ >.
+
+(C) 2015 by Adam Tauber, <asciimoo@gmail.com>
+(C) 2018 by Vaclav Zouzalik
+'''
+
+from flask_babel import gettext
+import hashlib
+import re
+
+name = "Hash plugin"
+description = gettext("Converts strings to different hash digests.")
+default_on = True
+
+parser_re = re.compile(b'(md5|sha1|sha224|sha256|sha384|sha512) (.*)', re.I)
+
+def post_search(request, search):
+    # process only on first page
+    if search.search_query.pageno > 1:
+        return True
+    m = parser_re.match(search.search_query.query)
+    if not m:
+        # wrong query
+        return True
+
+    # end if the string is empty
+    function, string = m.groups()
+    if string.strip().__len__() == 0:
+        return True
+
+    # select hash function
+    f = hashlib.new(function.lower())
+    #f = hash_function[function.lower()]
+
+    # make digest from the given string
+    f.update(string.strip())
+    digest = f.hexdigest()
+
+    # print result
+    search.result_container.answers.clear()
+    search.result_container.answers.add(function + " " + gettext('hash function') + ": " + digest)
+    return True

+ 47 - 0
tests/unit/test_plugins.py

@@ -83,3 +83,50 @@ class SelfIPTest(SearxTestCase):
         search = get_search_mock(query='What is my User-Agent?', pageno=2)
         store.call(store.plugins, 'post_search', request, search)
         self.assertFalse('user-agent' in search.result_container.answers)
+
+class HashPluginTest(SearxTestCase):
+
+    def test_PluginStore_init(self):
+        store = plugins.PluginStore()
+        store.register(plugins.hash_plugin)
+
+        self.assertTrue(len(store.plugins) == 1)
+
+        request = Mock(remote_addr='127.0.0.1')
+        request.headers.getlist.return_value = []
+
+        # MD5
+        search = get_search_mock(query=b'md5 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('md5 hash function: 098f6bcd4621d373cade4e832627b4f6' in search.result_container.answers)
+
+        search = get_search_mock(query=b'md5 test', pageno=2)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertFalse('md5 hash function: 098f6bcd4621d373cade4e832627b4f6' in search.result_container.answers)
+
+        # SHA1
+        search = get_search_mock(query=b'sha1 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('sha1 hash function: a94a8fe5ccb19ba61c4c0873d391e987982fbbd3' in search.result_container.answers)
+
+        # SHA224
+        search = get_search_mock(query=b'sha224 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('sha224 hash function: 90a3ed9e32b2aaf4c61c410eb925426119e1a9dc53d4286ade99a809' in search.result_container.answers)
+
+        # SHA256
+        search = get_search_mock(query=b'sha256 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('sha256 hash function: 9f86d081884c7d659a2feaa0c55ad015a3bf4f1b2b0b822cd15d6c15b0f00a08' in search.result_container.answers)
+
+        # SHA384
+        search = get_search_mock(query=b'sha384 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('sha384 hash function: 768412320f7b0aa5812fce428dc4706b3cae50e02a64caa16a782249bfe8efc4b' \
+          '7ef1ccb126255d196047dfedf17a0a9' in search.result_container.answers)
+
+        # SHA512
+        search = get_search_mock(query=b'sha512 test', pageno=1)
+        store.call(store.plugins, 'post_search', request, search)
+        self.assertTrue('sha512 hash function: ee26b0dd4af7e749aa1a8ee3c10ae9923f618980772e473f8819a5d4940e0db27a' \
+'c185f8a0e1d5f84f88bc887fd67b143732c304cc5fa9ad8e6f57f50028a8ff' in search.result_container.answers)