65 lines
2.5 KiB
Python
Executable File
65 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
import binascii
|
|
import configparser
|
|
import sys
|
|
|
|
from pybloom import BloomFilter
|
|
|
|
def main():
|
|
default_config_file='/nsrl/nsrl.conf'
|
|
config = configparser.ConfigParser()
|
|
config.read(default_config_file)
|
|
#add commandline options
|
|
hash_type=config.get('config','hashfile_type')
|
|
|
|
parser = argparse.ArgumentParser(prog='nsrl')
|
|
parser.add_argument("-v", "--verbose", help="Display verbose output message", action="store_true", required=False)
|
|
parser.add_argument("-0", "--no-hits", help="Suppress Output of matching hashes", action="store_true", required=False)
|
|
parser.add_argument("-1", "--no-misses", help="Suppress Output of mismatching hashes", action="store_true", required=False)
|
|
inputs = parser.add_mutually_exclusive_group(required=True)
|
|
inputs.add_argument('hash', metavar='<{}>'.format(hash_type), type=str, nargs='*', default=[], help='{} hash to search for.'.format(hash_type))
|
|
inputs.add_argument('-s','--stdin',help="Read hashes from stdin", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
if args.verbose:
|
|
print("Version INFO: {}".format(config.get('config',"rds_version")))
|
|
print("Error Rate: {}".format(config.get('config',"error_rate")))
|
|
print("Build Date: {}".format(config.get('config',"build_date")))
|
|
print("Filename: {}".format(config.get('config',"hashfile_name")))
|
|
print("Hashcount: {}".format(config.get('config',"hash_count")))
|
|
|
|
|
|
|
|
with open('nsrl.bloom', 'rb') as nb:
|
|
bf = BloomFilter.fromfile(nb)
|
|
|
|
if args.stdin:
|
|
hashlist=[hash.strip() for hash in sys.stdin.readlines()]
|
|
else:
|
|
hashlist=args.hash
|
|
for hash_hex in hashlist:
|
|
hash = binascii.unhexlify(hash_hex)
|
|
output=""
|
|
|
|
# only print output if for mismatches if selected
|
|
hash_is_a_match=(hash in bf)
|
|
if (hash_is_a_match and not args.no_hits) or (not hash_is_a_match and not args.no_misses):
|
|
#output
|
|
if args.verbose:
|
|
output = "{}:{}".format(hash_hex,hash_is_a_match)
|
|
elif args.no_hits != args.no_misses :
|
|
output = "{}".format(hash_hex)
|
|
else:
|
|
output = "{}:{}".format("+"if hash_is_a_match else "-",hash_hex)
|
|
print(output)
|
|
return
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except Exception as e:
|
|
print("Error: %s" % e)
|