Exploração de Vulnerabilidade de Desserialização Pickle no Transformers

Introdução ao Experimento

Apresentação do mecanismo de desserialização pickle e seu princípio de ataque reduce, reprodução da vulnerabilidade de desserialização CVE-2024-3568 na biblioteca Hugging Face Transformers, análise dos riscos de segurança na biblioteca Transformers e proposta de soluções eficazes para reforço de segurança. Para detalhes, consulte o documento PDF do experimento.

(O código do autor do PoC está detalhado em CVE-2024-3568 ou na pasta author_inf)

Informações Chave

1. Localização da Vulnerabilidade Método TFAutoModel.load_repo_checkpoint() (Transformers ≤ v4.37.2)

2. Princípio do Ataque O ficheiro extra_data.pickle em checkpoint malicioso aciona RCE ao ser desserializado via pickle.load()

3. Âmbito de Impacto

• Biblioteca Transformers ≤ v4.37.2
• Aplicações de IA que usam funcionalidade de recuperação de checkpoint

Configuração do Ambiente

# Consulte o PDF do experimento para instruções detalhadas de configuração
pip install transformers==4.37.2 tensorflow
git clone https://github.com/llm-sec/transformer-hacker

Mecanismo de Desserialização Pickle

Consulte o documento PDF para mais detalhes.

Os ficheiros test.py e test2.py envolvem análise do princípio (assistido por AI Wen Xiaobai, agradecendo o aumento de produtividade trazido pela inovação tecnológica).

Reprodução da Vulnerabilidade

1. Geração de Checkpoint Malicioso (Comandos específicos no documento PDF)

2. Acionamento da Vulnerabilidade (Exemplos de código no documento PDF)

3. Verificação do Ataque (Método de verificação no documento PDF)

Plano de Defesa

1. Atualizar Transformers para ≥v4.38.0
2. Desativar o método load_repo_checkpoint()
3. Implementar validação de ficheiros de checkpoint

Análise Técnica Detalhada: Consulte o documento PDF do experimento para o princípio completo da vulnerabilidade, processo de exploração e estratégias de defesa. Implementação do PoC: O código do autor do PoC está detalhado em CVE-2024-3568 ou na pasta author_inf.