mirror of
https://github.com/cve-search/cpe-guesser.git
synced 2024-11-14 19:08:27 +00:00
new: [server] basic HTTP server for the CPE guesser
This commit is contained in:
parent
f3d8ae9e6e
commit
56cfdd5852
1 changed files with 51 additions and 0 deletions
51
bin/server.py
Normal file
51
bin/server.py
Normal file
|
@ -0,0 +1,51 @@
|
||||||
|
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()
|
||||||
|
|
Loading…
Reference in a new issue