Python中的CSRF攻击与加密技术

from flask import Flask, request, render_template, make_response
import secrets

app = Flask(__name__)

# 生成随机Token
def gen_token():
    return secrets.token_hex(16)

# 登录页面
@app.route('/')
def login():
    # 生成Token
    token = gen_token()
    # 将Token存储在Cookie中
    resp = make_response(render_template('login.html', token=token))
    resp.set_cookie('csrf_token', token)
    return resp

# 提交评论
@app.route('/comment', methods=['POST'])
def comment():
    # 获得表单数据和Token
    comment = request.form['comment']
    token = request.form['csrf_token']
    # 检查Token会不会合法
    if token == request.cookies.get('csrf_token'):
        # Token合法，处理评论逻辑
        return '评论成功：{}'.format(comment)
    else:
        # Token不合法，返回毛病提示
        return 'CSRF攻击！'

if __name__ == '__main__':
    app.run()