Commit f4e24bc8 authored by Alexandre Dulaunoy's avatar Alexandre Dulaunoy Committed by GitHub
Browse files

Merge pull request #199 from sec9/fix-utf8-encoding-issue 

Fix UTF-8 encoding issue when parsing CWE and ExploitDB Files
parents 7fb70ebd 739e210e

sbin/db_mgmt_cwe.py

+1 −1
Original line number Diff line number Diff line
@@ -100,7 +100,7 @@ cwezip.close() 
with zipfile.ZipFile(tmpfile.name) as z:

    z.extractall(tmpdir)

    z.close()

f = open(os.path.join(tmpdir, 'cwec_v2.8.xml'))

f = open(os.path.join(tmpdir, 'cwec_v2.8.xml'), encoding='utf-8')

# parse xml and store in database

parser.parse(f)

cweList=[]

sbin/db_mgmt_exploitdb.py

+1 −1
Original line number Diff line number Diff line
@@ -50,7 +50,7 @@ with open(csvfile, 'wb') as fp: 
fp.close()



exploits=[]

with open(csvfile, newline='') as csvtoparse:

with open(csvfile, newline='', encoding='utf-8') as csvtoparse:

    exploitcsv = csv.DictReader(csvtoparse, delimiter=',')

    for row in exploitcsv:

        exploits.append(row)

CRA Git | Maintained and supported by SUSTech CRA and CCSE