JetBrains RiderでSnykプラグインを使って脆弱性チェックしてみた

#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdlib>
#include <string>

// 1. バッファオーバーフロー
void VulnerableBuffer(const char* input) {
    char buffer[8];
    // バッファサイズをチェックせずコピー（バッファオーバーフロー）
    strcpy(buffer, input);
    std::cout << "Buffer: " << buffer << std::endl;
}

// 2. パストラバーサル
void SaveData(const std::string& filename, const std::string& data) {
    // 入力値をそのままファイル名に使用（パストラバーサル脆弱性）
    std::ofstream ofs(filename);
    ofs << data;
    ofs.close();
    std::cout << "Saved data to " << filename << std::endl;
}

// 3. コマンドインジェクション
void RunCommand(const std::string& userInput) {
    // 入力値をそのままコマンドに（コマンドインジェクション）
    std::string cmd = "echo " + userInput;
    system(cmd.c_str());
}

int main() {
    std::cout << "脆弱性デモ開始\n";

    // 1. バッファオーバーフロー
    std::cout << "[1] バッファオーバーフローの例\n";
    char userInput[256];
    std::cout << "何か入力してください（長い文字列を入力すると危険！）: ";
    std::cin.getline(userInput, 256);
    VulnerableBuffer(userInput);

    // 2. パストラバーサル
    std::cout << "\n[2] ファイル保存の例（パストラバーサル脆弱性）\n";
    std::string filename;
    std::cout << "保存するファイル名を入力してください: ";
    std::getline(std::cin, filename);
    SaveData(filename, "これはテストデータです");

    // 3. コマンドインジェクション
    std::cout << "\n[3] コマンド実行の例（コマンドインジェクション）\n";
    std::string cmdInput;
    std::cout << "コマンドに渡す文字列を入力してください: ";
    std::getline(std::cin, cmdInput);
    RunCommand(cmdInput);

    std::cout << "\n脆弱性デモ終了\n";
    return 0;
}

void VulnerableBuffer(const char* input) {
    char buffer[8];
    // バッファサイズをチェックせずコピー（バッファオーバーフロー）
    strncpy(buffer, input, sizeof(buffer) - 1);  // Copy up to sizeof(buffer) - 1 characters
    std::cout << "Buffer: " << buffer << std::endl;
    buffer[sizeof(buffer) - 1] = '\0';  // Ensure null termination
}

#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdlib>
#include <string>
#include <regex>

// 1. バッファオーバーフロー対策
void SafeBuffer(const char* input) {
    char buffer[8];
    // strncpyを使い、終端も保証する
    std::strncpy(buffer, input, sizeof(buffer) - 1);
    buffer[sizeof(buffer) - 1] = '\0'; // 念のため終端
    std::cout << "Buffer: " << buffer << std::endl;
}

// 2. パストラバーサル対策
bool IsSafeFilename(const std::string& filename) {
    // ファイル名に危険な文字列が含まれていないかチェック
    // 例: ../ や ..\ などのパストラバーサルや、絶対パス（C:\や/）を禁止
    if (filename.find("..") != std::string::npos) return false;
    if (filename.find('/') != std::string::npos) return false;
    if (filename.find('\\') != std::string::npos) return false;
    if (filename.empty()) return false;
    // ファイル名は英数字と_と.のみ許可（例として）
    std::regex safePattern("^[A-Za-z0-9_.]+$");
    return std::regex_match(filename, safePattern);
}

void SafeSaveData(const std::string& filename, const std::string& data) {
    if (!IsSafeFilename(filename)) {
        std::cout << "不正なファイル名です: " << filename << std::endl;
        return;
    }
    std::ofstream ofs(filename);
    if (!ofs) {
        std::cout << "ファイルを開けませんでした: " << filename << std::endl;
        return;
    }
    ofs << data;
    ofs.close();
    std::cout << "Saved data to " << filename << std::endl;
}

// 3. コマンドインジェクション対策
void SafeRunCommand(const std::string& userInput) {
    // 入力値に危険な文字が含まれていないかチェック
    // 例: 英数字とスペースのみ許可
    std::regex safePattern("^[A-Za-z0-9 ]*$");
    if (!std::regex_match(userInput, safePattern)) {
        std::cout << "不正な文字が含まれています。コマンド実行を中止します。" << std::endl;
        return;
    }
    // 安全な場合のみコマンドを実行
    std::string cmd = "echo " + userInput;
    system(cmd.c_str());
}

...以下略