A vulnerability in the PyTorch's torch.distributed.rpc framework, specifically in versions prior to 2.2.2, allows for remote code execution (RCE). The framework, which is used in distributed training scenarios, does not properly verify the functions being called during RPC (Remote Procedure Call) operations. This oversight permits attackers to execute arbitrary commands by leveraging built-in Python functions such as eval during multi-cpu RPC communication. The vulnerability arises from the lack of restriction on function calls when a worker node serializes and sends a PythonUDF (User Defined Function) to the master node, which then deserializes and executes the function without validation. This flaw can be exploited to compromise master nodes initiating distributed training, potentially leading to the theft of sensitive AI-related data.
CVE-2024-5480 - Critical Severity Vulnerability
Vulnerable Libraries - torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl, torch-1.11.0-cp37-cp37m-manylinux1_x86_64.whl
torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Library home page: https://files.pythonhosted.org/packages/dd/b9/824df420f6abf551e41bbaacbaa0be8321dc104f9f3803051513844dc310/torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl
Path to dependency file: /jupyter-pytorch/cuda-requirements.txt
Path to vulnerable library: /jupyter-pytorch/cuda-requirements.txt,/jupyter-pytorch/cuda-requirements.txt
Dependency Hierarchy: - :x: **torch-1.8.1-cp36-cp36m-manylinux1_x86_64.whl** (Vulnerable Library)
torch-1.11.0-cp37-cp37m-manylinux1_x86_64.whl
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Library home page: https://files.pythonhosted.org/packages/94/32/96a0955e5d6ed8a837eda5ca095dd2694c4617cfa70ca599660cd5ff7447/torch-1.11.0-cp37-cp37m-manylinux1_x86_64.whl
Path to dependency file: /jupyter-pytorch-full/requirements.txt
Path to vulnerable library: /jupyter-pytorch-full/requirements.txt,/jupyter-pytorch-full/requirements.txt
Dependency Hierarchy: - :x: **torch-1.11.0-cp37-cp37m-manylinux1_x86_64.whl** (Vulnerable Library)
Found in HEAD commit: 513157c5d3f5743d53e228da1ec8289e92c92836
Found in base branch: master
Vulnerability Details
A vulnerability in the PyTorch's torch.distributed.rpc framework, specifically in versions prior to 2.2.2, allows for remote code execution (RCE). The framework, which is used in distributed training scenarios, does not properly verify the functions being called during RPC (Remote Procedure Call) operations. This oversight permits attackers to execute arbitrary commands by leveraging built-in Python functions such as eval during multi-cpu RPC communication. The vulnerability arises from the lack of restriction on function calls when a worker node serializes and sends a PythonUDF (User Defined Function) to the master node, which then deserializes and executes the function without validation. This flaw can be exploited to compromise master nodes initiating distributed training, potentially leading to the theft of sensitive AI-related data.
Publish Date: 2024-06-06
URL: CVE-2024-5480
CVSS 3 Score Details (10.0)
Base Score Metrics: - Exploitability Metrics: - Attack Vector: Network - Attack Complexity: Low - Privileges Required: None - User Interaction: None - Scope: Changed - Impact Metrics: - Confidentiality Impact: High - Integrity Impact: High - Availability Impact: High
For more information on CVSS3 Scores, click here.Step up your Open Source Security Game with Mend here