
Cloud Vulnerability DB
A community-led vulnerabilities database
A deserialization vulnerability (CVE-2024-37058) affects the MLflow platform versions 2.5.0 and newer. The vulnerability exists within the mlflow/langchain/utils.py file, specifically in the _load_from_pickle function, which is called during model loading operations. This vulnerability enables a maliciously uploaded Langchain AgentExecutor model to execute arbitrary code on an end user's system when interacted with (HiddenLayer Advisory, NVD).
The vulnerability stems from the insecure use of cloudpickle.load() function within the _load_from_pickle function in mlflow/langchain/utils.py. When the mlflow.langchain.load_model function is called, it processes the model through this vulnerable function. The vulnerability has been assigned a CVSS v3.1 base score of 8.8 (HIGH) with the vector string AV:N/AC:L/PR:N/UI:R/S:U/C:H/I:H/A:H, indicating a high-severity issue with potential for remote exploitation. The vulnerability is classified as CWE-502: Deserialization of Untrusted Data (HiddenLayer Advisory).
Successful exploitation of this vulnerability allows attackers to execute arbitrary code on the target system when a maliciously crafted model is loaded. This can lead to complete system compromise, as the code executes with the permissions of the process loading the model. The impact is particularly severe in environments where MLflow is used to share and load models across different users or systems (HiddenLayer Advisory).
The vulnerability can be exploited by crafting a malicious AgentExecutor with specially designed Tools that trigger the vulnerable code path in the _save_base_lcs function. When such a model is loaded by a victim using mlflow.langchain.load_model(), the malicious code embedded in the model is executed on their system. The exploitation requires the attacker to have the ability to upload models to the MLflow server and requires user interaction to load the malicious model (HiddenLayer Advisory).
Organizations using MLflow should upgrade to a version that contains fixes for this vulnerability. Additionally, it's recommended to implement strict access controls for model uploads and validate all models before loading them into production environments. Organizations should also consider implementing additional security controls around model loading operations (NVD).
Source: This report was generated using AI
Free Vulnerability Assessment
Evaluate your cloud security practices across 9 security domains to benchmark your risk level and identify gaps in your defenses.
Get a personalized demo
"Best User Experience I have ever seen, provides full visibility to cloud workloads."
"Wiz provides a single pane of glass to see what is going on in our cloud environments."
"We know that if Wiz identifies something as critical, it actually is."