freewvs.git

1) import unittest
2) import glob
3) import json
4) import sys
5) import difflib

6) import re

7) 
8) 

9) def versioncompare(safe_version, find_version):
10)     safe_version_tup = [int(x) for x in safe_version.split(".")]
11)     find_version_tup = [int(x) for x in find_version.split(".")]
12)     return find_version_tup < safe_version_tup
13) 
14) 

15) class TestJsonLint(unittest.TestCase):

16)     @unittest.skipIf(
17)         sys.version_info < (3, 6, 0), "json.dumps force-sorts on python 3.5"
18)     )

19)     def test_json_lint(self):
20)         valid = True
21)         for f in glob.glob("freewvsdb/*.json"):

22)             fp = open(f, encoding="ascii")

23)             orig = fp.read()
24)             fp.close()
25)             tmp = json.loads(orig)

26)             new = json.dumps(tmp, indent=2) + "\n"

27)             if orig != new:

28)                 print(f"json {f} not valid")

29)                 sys.stdout.writelines(difflib.unified_diff(orig, new))
30)                 valid = False

31)         self.assertTrue(valid)

32) 

33)     def test_json_values(self):
34)         jconfig = []

35)         for cfile in glob.glob("freewvsdb/*.json"):

36)             with open(cfile, encoding="ascii") as json_file:

37)                 jconfig += json.load(json_file)
38) 

39)         mkeys = {"name", "url", "safe", "vuln", "detection"}

40)         for item in jconfig:
41) 
42)             # check for all mandatory keys

43)             self.assertEqual(
44)                 mkeys.intersection(item.keys()),
45)                 mkeys,
46)                 msg=f"Missing key in {item['name']}",
47)             )

48) 
49)             # check we have at least one detection

50)             self.assertTrue(
51)                 len(item["detection"]) >= 1, msg=f"No detection in {item['name']}"
52)             )

53) 
54)             # vuln needs to be CVE or HTTPS URL

55)             self.assertTrue(
56)                 re.match("^CVE-[0-9]*-[0-9]*$", item["vuln"])
57)                 or item["vuln"].startswith("https://"),
58)                 msg=f"{item['name']}: Invalid vuln {item['vuln']}",
59)             )

60) 
61)             # make sure old_safe is properly sorted

62)             if "old_safe" in item:
63)                 old_safe = item["old_safe"].split(",")

64)                 for i in range(1, len(old_safe)):

65)                     self.assertTrue(
66)                         versioncompare(old_safe[i - 1], old_safe[i]),
67)                         msg=f"{item['name']}: Invalid old_safe"
68)                         " ordering {item['old_safe']}",
69)                     )

70) 

71)             # make sure latest is not outdated

72)             if "latest" in item and item["safe"] != "":
73)                 self.assertTrue(
74)                     not versioncompare(item["safe"], item["latest"]),
75)                     msg=f"{item['name']}: Safe version "
76)                     "{item['safe']} newer than latest"
77)                     " {item['latest']}",
78)                 )

79) 

80)             # subdir needs to be integer

81)             for det in item["detection"]:
82)                 self.assertTrue(
83)                     isinstance(det["subdir"], int),
84)                     msg=f"{item['name']}: subdir not int",
85)                 )

86) 

87) 

88) if __name__ == "__main__":