使用Python编写CSRF攻击和防御工具

import requests

# 使用requests库来发送带有CSRF攻击payload的POST要求
def csrf_attack():
    url = 'http://example.com'
    payload = {'username': 'username', 'password': 'password', 'csrf_token': 'attacker_token'}

    # 在要求头中添加Referer字段，告知服务器要求源地址
    headers = {'Referer': 'http://attacker.com'}

    # 发送POST要求
    response = requests.post(url, data=payload, headers=headers)

    # 打印响应结果
    print(response)
    print(response.text)

import random
import string

# 生成随机的CSRF令牌
def generate_csrf_token():
    characters = string.ascii_letters + string.digits
    token = ''.join(random.choice(characters) for i in range(32))
    return token

# 将生成的CSRF令牌添加到要求中
def add_csrf_token_to_request():
    # 在要求头中添加CSRF令牌
    headers = {'X-CSRF-TOKEN': generate_csrf_token()}

    # 发送POST要求
    response = requests.post(url, data=payload, headers=headers)

    # 打印响应结果
    print(response)
    print(response.text)