ดาวน์โหลด TaaC AI - ดาวน์โหลดซอร์สโค้ด TaaC AI

TaaC AI

ซอร์สโค้ดอื่น ๆ

1.0.0

ดาวน์โหลด

การสร้างแบบจำลองภัยคุกคามที่ขับเคลื่อนด้วย AI-as-a-code (TAAC-AI)

มันใช้ประโยชน์จาก AI ในการวิเคราะห์คำอธิบายบริการระบุภัยคุกคามความปลอดภัยแสดงภาพการไหลของข้อมูลและแนะนำการแก้ไขตามแนวทางการก้าวย่างและแนวทาง OWASP สคริปต์สร้างรายงาน HTML เชิงลึกและรวมถึงคุณสมบัติสำหรับการประเมินความเสี่ยงด้วยตนเอง

ออกแบบ

มันทำงานอย่างไร

การประมวลผลไฟล์ YAML: โหลดและตรวจสอบไฟล์ YAML ที่มีรายละเอียดบริการ
การวิเคราะห์ภัยคุกคาม AI: หากมีการจัดเตรียมคีย์ OpenAI API สคริปต์ใช้ AI เพื่อสร้างการวิเคราะห์การสร้างแบบจำลองภัยคุกคามที่ครอบคลุม
AI ภัยคุกคามข้ามการตรวจสอบความถูกต้อง: LLM ต่างๆดำเนินการตรวจสอบความถูกต้อง
การสร้างกระแสข้อมูล: สร้างการแสดงภาพของการไหลของข้อมูลภายในบริการโดยอัตโนมัติ
การจัดการความเสี่ยงด้วยตนเอง: ผู้ใช้สามารถเพิ่มแก้ไขหรือข้ามความเสี่ยงในรายงานที่สร้างขึ้นด้วยตนเอง
การสร้างรายงาน: สร้างรายงาน HTML โดยละเอียดรวมถึงความเสี่ยงทั้ง AI-Generated และเพิ่มด้วยตนเอง

รองรับแบบจำลอง

GPT-3.5
GPT-4
Claude 3 Haiku
MISTRAL 7B (ผ่าน Ollama)
ราศีเมถุน (วางแผน)

ตั้งคีย์ API

ตั้งค่าคีย์ OpenAI API เป็นตัวแปรสภาพแวดล้อม openai_key และ anthropic_key ในระบบปฏิบัติการของคุณ

Openai API API API

Linux/Mac

 export OPENAI_KEY=sk-ApiKeyExample

 export ANTHROPIC_KEY=sk-ant-api03-ApiKeyExample

หน้าต่าง

 set OPENAI_KEY=sk-ApiKeyExample

 set ANTHROPIC_KEY=sk-ant-api03-ApiKeyExample

ติดตั้งการพึ่งพา

pip3 install -r requirements.txt

ติดตั้ง Mistral และ Ollama

ดาวน์โหลดและติดตั้ง ollama ollama.ai
ติดตั้ง Mistral

ollama pull mistral

เริ่มบริการ Ollama (ตรวจสอบให้แน่ใจว่าแอพ Ollama Desktop ปิดอยู่)

ollama serve

ตอนนี้คุณสามารถใช้ Mistral เป็นรุ่นหลัก:

python3 TaaC-AI.py --model mistral < path_to_yaml_file >

หรือสำหรับการตรวจสอบข้าม

python3 TaaC-AI.py --model claude --cross-validation mistral < path_to_yaml_file >

วิธีใช้❓

สร้างคำอธิบายบริการที่ถูกต้องโดยใช้แนวทางเหล่านี้หรือใช้ taac_yaml_generator.py ที่จะแนะนำคุณตลอดกระบวนการสร้าง

yaml_generator!

เรียกใช้สคริปต์ (GPT-3.5 ถูกใช้โดยค่าเริ่มต้น)

python3 TaaC-AI.py < path_to_yaml_file >

ใช้ gpt-4 , claude หรือ mistral เป็นแบบจำลองเพื่อระบุภัยคุกคามโดยการระบุตัวเลือก --model

python3 TaaC-AI.py --model gpt-4 < path_to_yaml_file >

ในการดำเนินการตรวจสอบผลการคุกคามโดยการใช้ LLM อื่น-ตัวเลือก --cross-validation

python3 TaaC-AI.py --model claude --cross-validation claude < path_to_yaml_file >

Open Generate .html รายงาน
ทบทวน/แก้ไขตารางการวิเคราะห์การสร้างแบบจำลองภัยคุกคามที่ขับเคลื่อนด้วย AI และสำหรับข้อดีที่ผิดพลาดหรือปัญหาที่ได้รับการแก้ไขทำเครื่องหมายช่องทำเครื่องหมาย 'สถานะ'
เพิ่มภัยคุกคามที่ระบุด้วยตนเองในตาราง (ไม่บังคับ)
ดาวน์โหลดรายงานผ่านปุ่มดาวน์โหลดรายงาน

ตัวอย่างการใช้งาน?

ตัวอย่างคำอธิบายบริการที่ถูกต้อง

 Version : ' 1.0 '
Date : 14.11.2023

# Authentication Service Description
Description :
  Name :  AuthService
  Type : Service
  Criticality : Tier1

# Service Functionality
Functionality : Handles user authentication, including login and token generation.

# Data Processing Details
DataProcessed : 
  Type : Confidential
  DataCategory : Auth
  EncryptionAtRest : Yes

# Components Used by the Service
Components :
  Internal : 
    Exist : Yes
    Source : Private
    Note : Scoped Package Access
  External : 
    Exist : Yes
    PackageManager : NPM

# Pipeline Configuration
Pipeline :
  Type : GithubActions
  CODEOWNERS : Yes
  BranchProtection : Yes
  SignCommits : Yes
  PinActions : Yes
  
# Network Information
Network :
  Access : Private

# Authentication Service Data Flow
dataFlow :  # Removed the dash here
  - name : UserAuthenticationFlow
    description : Authenticates users and issues tokens.
    source : UserLoginInterface
    EncryptionTransit : Yes
    Authentication :
      Exist : Yes
      Type : JWT
    Authorization : read-write
    Protocol : HTTPS
    Communication :
      Type : RESTful API
    interactions :
      - from : UserLoginInterface
        to : AuthService
        method : RESTful API
        protocol : HTTPS
      - from : AuthService
        to : UserDatabase
        method : Query
        protocol : JDBC
    servicesInvolved : [UserLoginInterface, AuthService, UserDatabase]