Bug 2291225 (CVE-2024-5480) - CVE-2024-5480 python-pytorch: Remoce Code Execution vulnerability in torch.distributed.rpc framework
Summary: CVE-2024-5480 python-pytorch: Remoce Code Execution vulnerability in torch.di...
Keywords:
Status: NEW
Alias: CVE-2024-5480
Product: Security Response
Classification: Other
Component: vulnerability
Version: unspecified
Hardware: All
OS: Linux
high
high
Target Milestone: ---
Assignee: Product Security
QA Contact:
URL:
Whiteboard:
Depends On: 2291226
Blocks:
TreeView+ depends on / blocked
 
Reported: 2024-06-10 21:02 UTC by Marco Benatto
Modified: 2024-07-24 02:47 UTC (History)
0 users

Fixed In Version: pytorch torch.distributed.rpc 2.2.2
Clone Of:
Environment:
Last Closed:
Embargoed:


Attachments (Terms of Use)

Description Marco Benatto 2024-06-10 21:02:34 UTC
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.

https://huntr.com/bounties/39811836-c5b3-4999-831e-46fee8fcade3

Comment 1 Marco Benatto 2024-06-10 21:02:46 UTC
Created python-torch tracking bugs for this issue:

Affects: fedora-40 [bug 2291226]


Note You need to log in before you can comment on or make changes to this bug.