Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for Calculating Risk in VulnerableCode #1593

Merged
merged 24 commits into from
Nov 7, 2024

Conversation

ziadhany
Copy link
Collaborator

@ziadhany ziadhany commented Sep 17, 2024

issue: #1543

image

image

Screenshot from 2024-10-28 18-33-27

Screenshot from 2024-10-29 15-48-52

@TG1999 TG1999 added the 2-next label Oct 15, 2024
@DennisClark
Copy link
Member

it would be really great to see a review of this PR soon!

Set data_source as the header for the exploit table.
Squash the migration files into a single file.
Add test for exploit-db , metasploit
Add a missing migration file
Rename resources_and_notes to notes
Fix Api test
Refactor metasploit , exploitdb , kev improver
Rename Kev tab to exploit tab
Add support for exploitdb , metasploit, kev

Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Refactor the error handling logic in the code.

Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Add pipeline_id for ( kev, metasploit, exploit-db )

Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Set data_source as the header for the exploit table.
Squash the migration files into a single file.
Add test for exploit-db , metasploit
Add a missing migration file
Rename resources_and_notes to notes
Fix Api test
Refactor metasploit , exploitdb , kev improver
Rename Kev tab to exploit tab
Add support for exploitdb , metasploit, kev

Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Refactor the error handling logic in the code.

Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
Signed-off-by: ziadhany <[email protected]>
Signed-off-by: ziad hany <[email protected]>
uncomment all importers

Signed-off-by: ziad hany <[email protected]>
@ziadhany ziadhany requested a review from TG1999 October 22, 2024 13:40
@TG1999
Copy link
Contributor

TG1999 commented Oct 28, 2024

@ziadhany this generally looks good to me, can we have some logs for the pipeline and some UI and API screenshots so we can merge it.

Copy link
Member

@keshav-space keshav-space left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @ziadhany, some nits for your consideration.
Also, please use LoopProgress while iterating over large records, as we do here

progress = LoopProgress(total_iterations=fetched_exploit_count, logger=self.log)

vulnerabilities/pipelines/risk_package.py Outdated Show resolved Hide resolved
vulnerabilities/pipelines/risk_package.py Outdated Show resolved Hide resolved
vulnerabilities/risk.py Outdated Show resolved Hide resolved
vulnerabilities/pipelines/risk_package.py Outdated Show resolved Hide resolved
vulnerabilities/pipelines/risk_package.py Outdated Show resolved Hide resolved
vulnerabilities/pipelines/risk_package.py Outdated Show resolved Hide resolved
vulnerabilities/risk.py Outdated Show resolved Hide resolved
vulnerabilities/models.py Outdated Show resolved Hide resolved
vulnerabilities/migrations/0074_package_risk.py Outdated Show resolved Hide resolved
vulnerabilities/risk.py Outdated Show resolved Hide resolved
…skPackagePipeline to ComputePackageRiskPipeline. Add a tooltip for risk, and remove any unused imports in the view.

Signed-off-by: ziad hany <[email protected]>
…ore and remove any extra whitespace in views.py.

Signed-off-by: ziad hany <[email protected]>
@keshav-space
Copy link
Member

And please let me know if anyone has any other ideas, and feel free to edit the weight configuration list as needed so I can use it in the code.

@ziadhany this looks good to start.

@keshav-space
Copy link
Member

@ziadhany please update the changelog. Also, IMO we should use weight_config.yml instead of JSON file for configuration, like we do in our other projects.

@DennisClark
Copy link
Member

Re: "IMO we should use weight_config.yml instead of JSON file for configuration, like we do in our other projects." I am fine with that alternative.

@pombredanne
Copy link
Member

Re: "IMO we should use weight_config.yml instead of JSON file for configuration, like we do in our other projects." I am fine with that alternative.

But why having a config file? IMHO this should just be some Python data in a .py file. This is not meant to be updated by anyone but us for now.

@DennisClark
Copy link
Member

I think that someone running their own installation of VulnerableCode should be able to adjust the weight assignments based on their own experience. The idea of a config file has been in the design document for quite a while now and we need to move things forward to get this done.

@DennisClark
Copy link
Member

ok, let's go with .py if that is the simplest

@mjherzog
Copy link
Member

mjherzog commented Nov 7, 2024

We should keep local configuration in the plan but custom configuration would add unnecessary complexity at this early stage.

@keshav-space keshav-space linked an issue Nov 7, 2024 that may be closed by this pull request
@keshav-space
Copy link
Member

Thanks @ziadhany, merging this now!

@keshav-space keshav-space merged commit 0e13b55 into aboutcode-org:main Nov 7, 2024
5 checks passed
@DennisClark
Copy link
Member

@ziadhany or @keshav-space could you please post some screenshots here that show the display of the new values in VCIO? Thanks! Also please confirm that all 3 values will be available in the V2 API.

  • Weighted Severity
  • Exploitability
  • Risk

@ziadhany
Copy link
Collaborator Author

ziadhany commented Nov 8, 2024

@DennisClark We store the value of risk_score in a column and currently display it in the API. However, Exploitability and Weighted Severity are still missing. We already have the logic to calculate these metrics (Exploitability, Weighted Severity, and Risk), and we can easily implement it. However, we need to decide whether to store exploitability and weighted_severity in the database or calculate them dynamically with each API call. If we choose to store them, we must also determine whether to store them at the Vulnerability level or the Package level.

Screenshot from 2024-11-08 00-53-08

@DennisClark
Copy link
Member

@ziadhany I think your should store the values of exploitability and weighted_severity at the same level as risk_score.

@tdruez
Copy link
Contributor

tdruez commented Nov 8, 2024

@ziadhany @DennisClark

If we choose to store them, we must also determine whether to store them at the Vulnerability level or the Package level.

Looking at https://github.com/aboutcode-org/vulnerablecode/pull/1593/files

I see 4 values being computed:

  • get_weighted_severity
  • get_exploitability_level
  • compute_vulnerability_risk
  • compute_package_risk

We need to clarify if those values are computed/defined for a Vulnerability entry of for a Package (combination of Vulnerabilities).

This matters a lot in the presentation of the data, depending on the context, one may view it through 1 Package, or through 1 vulnerability that affects many Packages.

At the moment, only the Package.risk_score is stored and displayed (Package model and API).
I would suggest that the Vulnerability.risk_score should also be stored and available in the API so the value can be used when looking at the list of Vulnerabilities.

Now, for weighted_severity and exploitability_level, looking at the code or at the design document, it is not clear if those values are qualifying a Package, a Vulnerability, or both.

@ziadhany
Copy link
Collaborator Author

ziadhany commented Nov 8, 2024

@DennisClark @tdruez

Yes, we can store a vulnerability risk. However, weighted_severity and exploitability_level are specific to each vulnerability and are computed using the exploits, references, and severities associated with it.

If we plan to store the risk_score for each vulnerability while also storing and displaying the weighted_severity and exploitability_level, it will lead to data duplication.

I suggest storing only the weighted_severity and exploitability_level for each vulnerability. This way, calculating the vulnerability risk and package risk becomes a straightforward mathematical operation.

However, if we don't mind some data duplication in exchange for better performance, we could store the following:
exploitability_level, weighted_severity, and vulnerability risk in the vulnerability model
package risk in the package model (already exists).

Please let me know exactly what needs to be done so we can quickly create a pull request for it.

@tdruez
Copy link
Contributor

tdruez commented Nov 8, 2024

If we plan to store the risk_score for each vulnerability while also storing and displaying the weighted_severity and exploitability_level, it will lead to data duplication.

@ziadhany could you clarify the "data duplication", maybe with an example? I'm not sure to get what would be duplicated here.

@ziadhany
Copy link
Collaborator Author

ziadhany commented Nov 8, 2024

@tdruez
Let's say we have two vulnerabilities, both affecting the package.

We are going to calculate the weighted_severity and exploitability_level for Vulnerability 1. We get values of 2 and 3, respectively.
We did the same for Vulnerability 2 and get 4 , 2.

The risk for Vulnerability 1: ( 2 * 3 ) = 6
The risk for Vulnerability 2: ( 4 * 2 ) = 8

The package risk is the maximum risk of the vulnerabilities affecting this package, which is 8.

If we store the weighted_severity and exploitability_level, there’s no need to store the vulnerability risk (as it would be redundant data). Similarly, there’s no need to store the package risk since it can be easily calculated as the maximum of the vulnerability risks.

@tdruez
Copy link
Contributor

tdruez commented Nov 8, 2024

If we plan to store the risk_score for each vulnerability while also storing and displaying the weighted_severity and exploitability_level, it will lead to data duplication.

I still don't see the duplication storing the risk_score for each vulnerability.
The risk for Vulnerability 1: ( 2 * 3 ) = 6 and The risk for Vulnerability 2: ( 4 * 2 ) = 8 in your example: both have a different risk_score that is useful to order and filter a Vulnerability listing outside a Package context.

If we store the weighted_severity and exploitability_level, there’s no need to store the vulnerability risk (as it would be redundant data).

Could you clarify this? Why is it redundant?

@ziadhany
Copy link
Collaborator Author

ziadhany commented Nov 8, 2024

@tdruez I mean, if we create two columns for weighted_severity and exploitability_level, there’s no need to have a vulnerability_risk column because it is the product of the two columns.

and there might be no need for a package risk column since it represents the maximum of the vulnerability risk value that affect this package .

But, as you mentioned, it can be useful for ordering and filtering a vulnerability listing, and it can lead to better performance.

@tdruez
Copy link
Contributor

tdruez commented Nov 8, 2024

@ziadhany Alright, we should start by adding weighted_severity and exploitability_level as fields of the Vulnerability model in any case., right?

@ziadhany
Copy link
Collaborator Author

ziadhany commented Nov 8, 2024

@ziadhany Alright, we should start by adding weighted_severity and exploitability_level as fields of the Vulnerability model in any case., right?

Alright, I'll create a quick PR for this.

@DennisClark
Copy link
Member

@ziadhany I think that the suggestions from @tdruez make more sense than my previous remarks about putting all 3 fields at the same level. Thanks for all the analysis.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

CRAVEX: Calculate Package Vulnerability Risk
7 participants