CIRCL
/
AIL-framework
mirror of https://github.com/CIRCL/AIL-framework
2
1
Fork 0
Code Issues Releases Wiki Activity

chg: [import_dir] fix is_gzip test, use magic number

pull/486/head
Terrtia 2020-02-27 14:21:32 +01:00
parent 998f8cc8e1
commit 873797d87f
No known key found for this signature in database
GPG Key ID: 1E1B1F50D84613D0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bin/import_dir.py
View File

@ -7,9 +7,9 @@ from io import StringIO
import datetime import datetime
import gzip import gzip
import argparse import argparse
import binascii
import os import os
import time, datetime import time, datetime
import magic
import re import re
''' '''
@ -37,6 +37,8 @@ import re
' '
''' '''
def is_gzip_file(magic_nuber):
return binascii.hexlify(magic_nuber) == b'1f8b'
def is_hierachy_valid(path): def is_hierachy_valid(path):
var = path.split('/') var = path.split('/')
@ -82,7 +84,7 @@ if __name__ == "__main__":
messagedata = f.read() messagedata = f.read()
#verify that the data is gzipEncoded. if not compress it #verify that the data is gzipEncoded. if not compress it
if 'text/plain' in str(magic.from_buffer(messagedata, mime=True)): if not is_gzip_file(messagedata[0:2]):
messagedata = gzip.compress(messagedata) messagedata = gzip.compress(messagedata)
complete_path += '.gz' complete_path += '.gz'