Python中的CSRF攻击与安全头部设置

2023-04-17 00:00:00 设置攻击头部

CSRF（Cross-site Request Forgery）攻击是一种利用用户在已登录的状态下对指定网站发起误导性请求的网络攻击方式。攻击者会诱导用户点击恶意链接或者访问恶意网站，在用户不知情的情况下向指定网站发送请求，达到非法盗取用户信息、操纵用户数据等目的。

为了防止CSRF攻击，我们需要在服务器端设置安全头部，使用以下方法：

1.设置X-CSRF-TOKEN头部：

在服务器端，为每个表单页面生成一个随机的CSRF Token，并将其存储在会话中，然后将这个Token作为X-CSRF-TOKEN的值添加到HTTP请求头部中，在验证请求时比较请求头部中的Token和会话中的Token是否一致，以此防止CSRF攻击。

示例代码：

# 生成CSRF Token并存储到会话中
import secrets
from flask import session

@app.route('/')
def index():
    token = secrets.token_hex(16)
    session['csrf_token'] = token
    return render_template('index.html', csrf_token=token)

# 验证请求头部中的Token和会话中的Token是否一致
from flask import request

@app.route('/submit', methods=['POST'])
def submit():
    if request.headers.get('X-CSRF-TOKEN') != session['csrf_token']:
        return 'Invalid CSRF Token'
    # 业务逻辑处理
    return 'Submit Succeeded'

2.设置Referer头部：

在服务器端，通过比较请求头部中的Referer字段和请求URL中的域名是否一致，以此判断请求来源是否合法，以此防止CSRF攻击。

示例代码：

from flask import request

@app.route('/submit', methods=['POST'])
def submit():
    referer = request.headers.get('Referer')
    if not referer or 'pidancode.com' not in referer:
        return 'Invalid Referer'
    # 业务逻辑处理
    return 'Submit Succeeded'

以上两种方法都可以有效地防止CSRF攻击，具体应该选择哪一种方法，取决于具体的业务场景，可根据实际情况选择最为适合的方法。

相关文章