hashlookup-server/bin/import.py
Alexandre Dulaunoy a4b3b7ba60
new: [hashlookup-server] initial import of the code
- This includes a simple HTTP server for doing bulk and lookup of hashes.
- A simple DNS server to do lookup via DNS
- Various import script for NSRL

This works on a test instance.

TODO:

- Automatic script for NSRL download and import
- Bloomfilter export
- Improved documentation
2021-07-15 17:49:52 +02:00

30 lines
876 B
Python

import sys
import redis
rdb = redis.Redis(host='127.0.0.1', port='6666', decode_responses=True)
lines = open('../data/NSRLFile.txt', 'r')
ln = 0
rdb.delete("stat:NSRLAndroid")
rdb.set("stat:NSRLAndroid", 0)
maxvalue = 5000000000
for l in lines:
if ln == 0:
headers = l.rstrip().replace("\"","").split(",")
print (headers)
else:
records = l.rstrip().replace("\"","").split(",")
drecords = {}
for index, value in enumerate(records):
try:
drecords[headers[index]] = value
except:
continue
print(drecords)
print(drecords['SHA-1'])
rdb.set("l:{}".format(drecords['MD5']), drecords['SHA-1'])
rdb.hmset("h:{}".format(drecords['SHA-1']), drecords)
rdb.incrby("stat:NSRLAndroid")
if ln == maxvalue:
sys.exit(1)
ln = ln + 1