#!/usr/bin/env python
# -*- coding: utf-8 -*-

import argparse
import binascii
import sys

from pybloom import BloomFilter

def main():
    parser = argparse.ArgumentParser(prog='nsrl')
    parser.add_argument("-v", "--verbose", help="Display verbose output message", action="store_true", required=False)
    parser.add_argument("-m", "--mismatched", help="Echo only mismatched Hashvalues", action="store_true", required=False)
    inputs = parser.add_mutually_exclusive_group(required=True)
    inputs.add_argument('hash', metavar='<MD5>', type=str, nargs='*', default=[], help='md5 hash to search for.')
    inputs.add_argument('-s','--stdin',help="Read hashes from stdin", action="store_true")
    args = parser.parse_args()

    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
            if ((not hash in bf) or (not args.mismatched)):
                if args.verbose:
                    output = "{}:{}".format(hash_hex,hash in bf)
                else:
                    if args.mismatched:
                        output = "{}".format(hash_hex)
                    else:
                        output = "{}".format(hash in bf)

                print output
    return

if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        print "Error: %s" % e