diff --git a/cli.py b/cli.py index 45dad55..f486e40 100644 --- a/cli.py +++ b/cli.py @@ -64,7 +64,7 @@ def security_scan_with_justifications( # Call the individual scan functions from SaferPickle to sets of results. # Picklemagic Scan - picklemagic_results = safer_pickle.picklemagic_scan(pickle_bytes) + picklemagic_results = saferpickle.picklemagic_scan(pickle_bytes) safe_results.update(picklemagic_results.safe_results) unsafe_results.update(picklemagic_results.unsafe_results) @@ -72,7 +72,7 @@ def security_scan_with_justifications( unknown_results.update(picklemagic_results.unknown_results) # Genops Scan - genops_results = safer_pickle.genops_scan( + genops_results = saferpickle.genops_scan( pickle_bytes, pickle_file_path=file_path ) safe_results.update(genops_results.safe_results) @@ -97,7 +97,7 @@ def security_scan_with_justifications( num_unsafe, num_suspicious, _, # The unknown_score is not used for classification, only reporting - ) = safer_pickle.score_results( + ) = saferpickle.score_results( final_safe_results, final_unsafe_results, final_suspicious_results, @@ -105,7 +105,7 @@ def security_scan_with_justifications( ) # Check for safety and return the results with justifications. - if safer_pickle.is_unsafe(num_safe, num_unsafe, num_suspicious): + if saferpickle.is_unsafe(num_safe, num_unsafe, num_suspicious): if num_unsafe > num_suspicious: classification = "unsafe" all_results = [] diff --git a/lib/constants.py b/lib/constants.py index c43f8db..bdd2d9d 100644 --- a/lib/constants.py +++ b/lib/constants.py @@ -134,7 +134,8 @@ "read", "requests", "runpy", - "safer_pickle_hook", + "safer_pickle", + "saferpickle", "socket", "ssl", "stdin", diff --git a/lib/utils.py b/lib/utils.py index c8ff86f..02343b1 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Utility functions for safer_pickle.""" +"""Utility functions for saferpickle.""" import ast import bz2