mirror of
https://github.com/cve-search/cpe-guesser.git
synced 2024-11-14 19:08:27 +00:00
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import falcon
|
|
from wsgiref.simple_server import make_server
|
|
import requests
|
|
import redis
|
|
from datetime import datetime
|
|
import json
|
|
|
|
rdb = redis.Redis(host='127.0.0.1', port=6379, db=8, decode_responses=True)
|
|
|
|
|
|
class Search():
|
|
def on_post(self, req, resp):
|
|
ret = []
|
|
data_post = req.bounded_stream.read()
|
|
js = data_post.decode('utf-8')
|
|
q = json.loads(js)
|
|
|
|
if 'query' in q:
|
|
pass
|
|
else:
|
|
resp.status = falcon.HTTP_500
|
|
resp.media = "Missing query array or incorrect JSON format"
|
|
return
|
|
|
|
k=[]
|
|
for keyword in q['query']:
|
|
k.append('w:{}'.format(keyword.lower()))
|
|
|
|
maxinter = len(k)
|
|
cpes = []
|
|
for x in reversed(range(maxinter)):
|
|
ret = rdb.sinter(k[x])
|
|
cpes.append(list(ret))
|
|
result = set(cpes[0]).intersection(*cpes)
|
|
|
|
ranked = []
|
|
|
|
for cpe in result:
|
|
rank = rdb.zrank('rank:cpe', cpe)
|
|
ranked.append((rank, cpe))
|
|
|
|
resp.media=sorted(ranked)
|
|
|
|
app = falcon.App()
|
|
app.add_route('/search', Search())
|
|
|
|
if __name__ == '__main__':
|
|
with make_server('', 8000, app) as httpd:
|
|
print('Serving on port 8000...')
|
|
httpd.serve_forever()
|
|
|