TaaC AI下載TaaC AI源代碼下載

TaaC AI

其他源碼

1.0.0

下載

AI驅動的威脅建模AS-A代碼（TAAC-AI）

它利用AI來分析服務描述，確定安全威脅，可視化數據流並根據Stride框架和OWASP指南提出補救措施。該腳本生成深入的HTML報告，並包括手動風險評估的功能。

設計

它如何工作

YAML文件處理：加載和驗證包含服務詳細信息的YAML文件
AI威脅分析：如果提供了OpenAI API密鑰，則腳本使用AI來生成全面的威脅建模分析
AI威脅交叉驗證：各種LLMS執行驗證結果
數據流生成：自動生成服務中數據流的視覺表示
手動風險管理：用戶可以在生成報告中手動添加，修改或劃出風險
報告生成：生成詳細的HTML報告，包括AI生成和手動增加風險

支持模型

GPT-3.5
GPT-4
克勞德3 haiku
Mistral 7b（通過Ollama）
雙子座（計劃）

設置API鍵

將OpenAI API密鑰設置為環境變量OpenAI_KEY和ANTHROPIC_KEY在您的操作系統中

Openai 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桌面應用已關閉）

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生成.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]