adulau/MalwareClassifier
1
0
Fork 0
mirror of https://github.com/adulau/MalwareClassifier.git synced 2024-12-22 08:46:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

Workshop - 5th step added

Browse source 
Mapping between malware MD5 and element MD5 added in Redis
This commit is contained in:
Alexandre Dulaunoy 2015-01-10 19:32:24 +01:00
parent 3b1ca45d06
commit 56a42333fa
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
bin/import.py
View file

@ -1,6 +1,7 @@
import argparse import argparse
import sys import sys
import redis import redis
import hashlib
argParser = argparse.ArgumentParser(description='Malware classifier') argParser = argparse.ArgumentParser(description='Malware classifier')
argParser.add_argument('-f', action='append', help='Filename') argParser.add_argument('-f', action='append', help='Filename')
@ -24,6 +25,12 @@ if args.f is not None:
for element in elements: for element in elements:
try: try:
r.sadd('e:'+fields[i], element) r.sadd('e:'+fields[i], element)
#
ehash = hashlib.md5()
ehash.update(element.encode('utf-8'))
ehhex = ehash.hexdigest()
if element is not "":
r.sadd('v:'+ehhex, md5)
except IndexError: except IndexError:
print("Empty fields") print("Empty fields")
i = i+1 i = i+1