From eb77d3ee77545e6120340b72ef61fa460a2e552c Mon Sep 17 00:00:00 2001 From: MatToufoutu Date: Sat, 9 Jul 2016 17:58:10 +0200 Subject: make script python3 compatible --- php-malware-finder/generate_whitelist.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/php-malware-finder/generate_whitelist.py b/php-malware-finder/generate_whitelist.py index 4a8a6c5..04bcc91 100644 --- a/php-malware-finder/generate_whitelist.py +++ b/php-malware-finder/generate_whitelist.py @@ -8,15 +8,15 @@ import sys try: import yara except ImportError: - print 'Please install python-yara' + print('Please install python-yara') sys.exit(1) if len(sys.argv) != 3: - print 'Usage: %s name_of_the_rule_and_version folder_to_scan' % sys.argv[0] + print('Usage: %s name_of_the_rule_and_version folder_to_scan' % sys.argv[0]) sys.exit(1) if not os.path.isdir(sys.argv[2]): - print '%s is not a folder !' % sys.argv[2] + print('%s is not a folder !' % sys.argv[2]) sys.exit(1) rules = yara.compile('./php.yar', includes=True, error_on_warning=True) @@ -30,11 +30,11 @@ for cpt, (root, dirnames, filenames) in enumerate(os.walk(sys.argv[2])): matches = rules.match(os.path.join(root, filename), fast=True) if matches: matches = matches.pop() # only one match, since we're scaning files - output_list.append('hash.sha1(0, filename) == %s or // %s' % (hashlib.sha1(fname).hexdigest(), fname)) + output_list.append('hash.sha1(0, filename) == %s or // %s' % (hashlib.sha1(fname.encode('utf-8')).hexdigest(), fname)) output_rule = 'private rule %s\n{\n\tcondition:\n\t\t/* %s */\n\t\t' % (sys.argv[1].split(' ')[0], sys.argv[1]) output_list.append(output_list.pop().replace(' or ', ' ')) output_rule += '\n\t\t'.join(output_list) output_rule += '\n}' -print output_rule +print(output_rule) -- cgit v1.3