mirror of
https://github.com/adulau/MalwareClassifier.git
synced 2024-12-22 08:46:04 +00:00
21 lines
569 B
Python
21 lines
569 B
Python
import argparse
|
|
import sys
|
|
import redis
|
|
|
|
argParser = argparse.ArgumentParser(description='Malware classifier')
|
|
argParser.add_argument('-f', action='append', help='Filename')
|
|
args = argParser.parse_args()
|
|
|
|
r = redis.StrictRedis(host='localhost', port=6379, db=0)
|
|
|
|
if args.f is not None:
|
|
md5 = args.f[0].split(".")[0]
|
|
r.sadd('processed', md5)
|
|
lnumber = 0
|
|
for line in sys.stdin:
|
|
if lnumber == 0:
|
|
for field in line.rstrip().split(","):
|
|
r.sadd('type', field)
|
|
lnumber = lnumber + 1
|
|
else:
|
|
argParser.print_help()
|