diff options
| author | GeogeFI | 2022-12-25 22:43:21 +0100 |
|---|---|---|
| committer | GeogeFI | 2022-12-25 22:43:21 +0100 |
| commit | 40cbd0572ec5ea7454145303fc8bf1d0c71d9682 (patch) | |
| tree | 79930723e9862f13ead0d1a7c882ec0870d1b22a /src | |
| parent | 33cc8b4200a197603354b1142d1f4a4e964bd88b (diff) | |
| download | sec-certs-40cbd0572ec5ea7454145303fc8bf1d0c71d9682.tar.gz sec-certs-40cbd0572ec5ea7454145303fc8bf1d0c71d9682.tar.zst sec-certs-40cbd0572ec5ea7454145303fc8bf1d0c71d9682.zip | |
chore: Formating, fixed tests
Diffstat (limited to 'src')
| -rw-r--r-- | src/sec_certs/dataset/cve.py | 2 | ||||
| -rw-r--r-- | src/sec_certs/sample/cve.py | 8 |
2 files changed, 7 insertions, 3 deletions
diff --git a/src/sec_certs/dataset/cve.py b/src/sec_certs/dataset/cve.py index 566bc2e4..9a36fa6b 100644 --- a/src/sec_certs/dataset/cve.py +++ b/src/sec_certs/dataset/cve.py @@ -84,7 +84,7 @@ class CVEDataset(JSONPathDataset, ComplexSerializableType): for cve in tqdm(self, desc="Building-up lookup dictionaries for fast CVE matching"): # See note above, we use matching_dict.get(cpe, []) instead of matching_dict[cpe] as would be expected if use_nist_mapping: - vulnerable_configurations = list( + vulnerable_configurations = set( itertools.chain.from_iterable(matching_dict.get(cpe, []) for cpe in cve.vulnerable_cpes) ) else: diff --git a/src/sec_certs/sample/cve.py b/src/sec_certs/sample/cve.py index caf2b4d7..9b2c2437 100644 --- a/src/sec_certs/sample/cve.py +++ b/src/sec_certs/sample/cve.py @@ -221,8 +221,12 @@ class CVE(PandasSerializableType, ComplexSerializableType): impact = cls.Impact.from_nist_dict(dct) cpe_and_cpe_configurations = get_vulnerable_cpes_from_nist_dict(dct) # There exist CVEs such as (CVE-2022-0177) which are rejected and do not contain any assinged CPEs - vulnerable_cpes = set(itertools.chain.from_iterable(cpe_and_cpe_configurations[0])) if cpe_and_cpe_configurations else set() - vulnerable_cpe_configurations = set(itertools.chain.from_iterable(cpe_and_cpe_configurations[1])) if cpe_and_cpe_configurations else set() + vulnerable_cpes = ( + set(itertools.chain.from_iterable(cpe_and_cpe_configurations[0])) if cpe_and_cpe_configurations else set() + ) + vulnerable_cpe_configurations = ( + set(itertools.chain.from_iterable(cpe_and_cpe_configurations[1])) if cpe_and_cpe_configurations else set() + ) published_date = dct["publishedDate"] cwe_ids = cls.parse_cwe_data(dct) |