hashlookup-server/bin/import_NSRLOS.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

29 lines
868 B
Python

import sys
import redis
rdb = redis.Redis(host='127.0.0.1', port='6666', decode_responses=True)
lines = open('../data/NSRLOS.txt', 'r')
ln = 0
#rdb.delete("stat:NSRLOS-import")
maxvalue = 500000000
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['OpSystemCode'])
rdb.sadd('s:OpSystemCode', drecords['OpSystemCode'])
rdb.hmset("h-OpSystemCode:{}".format(drecords['OpSystemCode']), drecords)
rdb.incrby("stat:NSRLOS-import")
if ln == maxvalue:
sys.exit(1)
ln = ln + 1