blob: 19b1cbfcda769342a4a01f853c7eb459d3ed19ab (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#!/usr/bin/python3
import os
import os.path
import string
import sys
def setup_paths():
check_file = 'lib/python/debian_support.py'
path = os.getcwd()
while 1:
if os.path.exists("%s/%s" % (path, check_file)):
sys.path = [path + '/lib/python'] + sys.path
return path
idx = string.rfind(path, '/')
if idx == -1:
raise ImportError("could not setup paths")
path = path[0:idx]
setup_paths()
import nvd
import security_db
base = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
db_file = os.path.join(base, 'data/security.db')
db = security_db.DB(db_file)
incremental = False
data = []
for name in sys.argv[1:]:
if name == '-i':
incremental = True
continue
f = open(name)
data += nvd.parse(f)
f.close()
# For some reason, NVD adds duplicates, so we need to get rid of them.
# Sort afterwords to increase locality in the insert process.
deduplicate = {}
for x in data:
deduplicate[x[0]] = x
data = list(deduplicate.values())
data.sort()
cursor = db.writeTxn()
db.updateNVD(cursor, data, incremental)
db.commit(cursor)
|
