TaaC AIダウンロード-TAACAI TaaC AIコードのダウンロード

TaaC AI

その他のソースコード

1.0.0

ダウンロード

AI駆動型の脅威モデリング - コード（TAAC-AI）

AIを活用して、サービスの説明を分析し、セキュリティの脅威を特定し、データフローを視覚化し、StrideフレームワークとOWASPガイドラインに基づいて修正を提案します。このスクリプトは、詳細なHTMLレポートを生成し、手動リスク評価の機能が含まれています。

デザイン

それがどのように機能するか

YAMLファイル処理：サービスの詳細を含むYAMLファイルを読み込んで検証します
AI脅威分析： OpenAI APIキーが提供されている場合、スクリプトはAIを使用して包括的な脅威モデリング分析を生成します
AIの脅威クロス検証：さまざまなLLMSが検証結果を実行します
データフローの生成：サービス内のデータフローの視覚的表現を自動的に生成します
マニュアルリスク管理：ユーザーは、生成されたレポートでリスクを手動で追加、変更、または削除できます
レポート生成： AIの生成と手動で追加されたリスクの両方を含む詳細なHTMLレポートを作成します

サポートされているモデル

GPT-3.5
GPT-4
クロード3ハイク
ミストラル7b（オラマを通じて）
ジェミニ（計画）

APIキーを設定します

OpenAI APIキーを環境変数として設定しますopenai_keyおよびanthropic_keyオペレーティングシステムで

Openai API人類API

Linux/Mac

 export OPENAI_KEY=sk-ApiKeyExample

 export ANTHROPIC_KEY=sk-ant-api03-ApiKeyExample

Windows

 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を使用して、1つを生成するプロセスをガイドします

yaml_generator！

スクリプトを実行します（GPT-3.5はデフォルトで使用されます）

python3 TaaC-AI.py < path_to_yaml_file >

--modelオプションを指定することにより、脅威を特定するためのモデルとしてgpt-4 、 claudeまたはmistral使用してください

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 >

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]