Deserialization of Untrusted Data Affecting pytorch-lightning package, versions [,1.6.0rc0)
Threat Intelligence
Do your applications use this vulnerable package?
In a few clicks we can analyze your entire application and see what components are vulnerable in your application, and suggest you quick fixes.
Test your applications- Snyk ID SNYK-PYTHON-PYTORCHLIGHTNING-2325279
- published 24 Dec 2021
- disclosed 24 Dec 2021
- credit oivrip
Introduced: 24 Dec 2021
CVE-2021-4118 Open this link in a new tabHow to fix?
Upgrade pytorch-lightning to version 1.6.0rc0 or higher.
Overview
pytorch-lightning is a lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
Affected versions of this package are vulnerable to Deserialization of Untrusted Data via saving.py functionality which is calling yaml.UnsafeLoader from pyyaml Python library which is not a secure method. Because of that, maliciously crafted yaml config file can cause code execution on the victim's machine.
Details
Serialization is a process of converting an object into a sequence of bytes which can be persisted to a disk or database or can be sent through streams. The reverse process of creating object from sequence of bytes is called deserialization. Serialization is commonly used for communication (sharing objects between multiple hosts) and persistence (store the object state in a file or a database). It is an integral part of popular protocols like Remote Method Invocation (RMI), Java Management Extension (JMX), Java Messaging System (JMS), Action Message Format (AMF), Java Server Faces (JSF) ViewState, etc.
Deserialization of untrusted data (CWE-502) is when the application deserializes untrusted data without sufficiently verifying that the resulting data will be valid, thus allowing the attacker to control the state or the flow of the execution.