-
-
Notifications
You must be signed in to change notification settings - Fork 606
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
Pytorch Load / Save Plugin #1114
Conversation
This plugin checks for the use of `torch.load` and `torch.save`. Using `torch.load` with untrusted data can lead to arbitrary code execution, and improper use of `torch.save` might expose sensitive data or lead to data corruption. Signed-off-by: Luke Hinds <luke@stacklok.com>
Bit of cleaning up to do, will get onto next week |
Closing and reopening to trigger precommit ci to auto fix this for you |
for more information, see https://pre-commit.ci
Signed-off-by: Luke Hinds <luke@stacklok.com>
Signed-off-by: Luke Hinds <luke@stacklok.com>
Signed-off-by: Luke Hinds <luke@stacklok.com>
@sigmavirus24 / @ericwb I think I have reviewed most of the points now, fancy taking a second sweep? |
Signed-off-by: Luke Hinds <luke@stacklok.com>
for more information, see https://pre-commit.ci
Co-authored-by: Eric Brown <ericwb@users.noreply.github.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Recently the trojansource plugin was merged which conflicts with this plugin ID of B613. Please change to B614
Co-authored-by: Eric Brown <ericwb@users.noreply.github.com>
Co-authored-by: Eric Brown <ericwb@users.noreply.github.com>
Co-authored-by: Eric Brown <ericwb@users.noreply.github.com>
sorry for late action, changes accepted @ericwb |
* Pytorch Load / Save Plugin This plugin checks for the use of `torch.load` and `torch.save`. Using `torch.load` with untrusted data can lead to arbitrary code execution, and improper use of `torch.save` might expose sensitive data or lead to data corruption. Signed-off-by: Luke Hinds <luke@stacklok.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add missing save check Signed-off-by: Luke Hinds <luke@stacklok.com> * Review fixes from 8b92a02 Signed-off-by: Luke Hinds <luke@stacklok.com> * Fix tox issues Signed-off-by: Luke Hinds <luke@stacklok.com> * Review fixes Signed-off-by: Luke Hinds <luke@stacklok.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update test_functional.py * Update bandit/plugins/pytorch_load_save.py Co-authored-by: Eric Brown <ericwb@users.noreply.github.com> * Update bandit/plugins/pytorch_load_save.py Co-authored-by: Eric Brown <ericwb@users.noreply.github.com> * Update doc/source/plugins/b704_pytorch_load_save.rst Co-authored-by: Eric Brown <ericwb@users.noreply.github.com> * Update bandit/plugins/pytorch_load_save.py Co-authored-by: Eric Brown <ericwb@users.noreply.github.com> --------- Signed-off-by: Luke Hinds <luke@stacklok.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Eric Brown <ericwb@users.noreply.github.com> Signed-off-by: Eric Brown <eric_wade_brown@yahoo.com>
This plugin checks for the use of
torch.load
andtorch.save
.Using
torch.load
with untrusted data can lead to arbitrary code execution,and improper use of
torch.save
might expose sensitive data or lead to datacorruption.