CVE-2024-8309 — 개념 증명(PoC)으로, Langchain의 GraphCypherQAChain에서 프롬프트 인젝션을 시연하여 Neo4j 데이터베이스의 SQL 인젝션으로 이어지는 것을 보여줍니다. 교육용 테스트를 위한 FastAPI 백엔드와 Streamlit 프론트엔드를 포함한 Docker 기반 설정이 포함되어 있습니다. | Kitploit

공유

Langchain CVE-2024–8309 취약점에 대한 개념 증명 (PoC)

개요

이 설정은 Neo4j 데이터베이스에서 SQL 인젝션을 허용하는 GraphCypherQAChain 클래스의 프롬프트 인젝션 취약점에 대한 개념 증명을 보여줍니다.

PoC CVE

구성 요소:

Neo4j 데이터베이스: Neo4j를 실행합니다.
백엔드 (FastAPI): Langchain을 사용하여 Neo4j와 상호 작용합니다.
프론트엔드 (Streamlit): 백엔드와 상호 작용할 수 있는 간단한 인터페이스입니다.

사용법

이 저장소를 클론합니다.

.env 파일을 구성합니다.

AZURE_API_KEY=
AZURE_CHAT_DEPLOYMENT=
AZURE_ENDPOINT=

OPENAI_API_KEY=
LLM_PROVIDER= # "azure, openai"

서비스를 시작합니다:
```
docker-compose build
docker-compose up
```
http://localhost:7474에서 Neo4j 데이터베이스에 접속합니다 (기본 사용자 이름: neo4j, 비밀번호: password).