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]