Flask基础(17)-->防止 CSRF 攻击

时间 2019-11-06

标签 flask 基础防止 csrf 攻击栏目 Python 繁體版

原文原文链接

CSRF

CSRF全拼为Cross Site Request Forgery，译为跨站请求伪造。
CSRF指攻击者盗用了你的身份，以你的名义发送恶意请求。
- 包括：以你名义发送邮件，发消息，盗取你的帐号，甚至于购买商品，虚拟货币转帐......
形成的问题：我的隐私泄露以及财产安全。

CSRF攻击示意图

客户端访问服务器时没有同服务器作安全验证

防止 CSRF 攻击

步骤

在客户端向后端请求界面数据的时候，后端会往响应中的 cookie 中设置 csrf_token 的值
在 Form 表单中添加一个隐藏的的字段，值也是 csrf_token
在用户点击提交的时候，会带上这两个值向后台发起请求
后端接受到请求，以会如下几件事件：
- 从 cookie中取出 csrf_token
- 从表单数据中取出来隐藏的 csrf_token 的值
- 进行对比
若是比较以后两值同样，那么表明是正常的请求，若是没取到或者比较不同，表明不是正常的请求，不执行下一步操做

那么下面咱们用一个代码案例进行演示一下：防止CSRF攻击的代码示例（若是想测试没有CSRF预防的状况，能够在代码里直接更改便可：即去掉转帐视图crsf_token验证功能便可）html

CRSF_test.py(视图页面)flask

from flask import Flask, render_template, make_response
from flask import redirect
from flask import request
from flask import url_for
import os,base64

app = Flask(__name__)


# 生成 csrf_token 函数
def generate_csrf():
    return bytes.decode(base64.b64encode(os.urandom(48)))

@app.route('/login', methods=["POST", "GET"])
def login():
    if request.method == "POST":
        # 取到表单中提交上来的参数
        username = request.form.get("username")
        password = request.form.get("password")

        if not all([username, password]):
            print('用户名或密码错误')
        else:
            print(username, password)
            if username == 'laowang' and password == '1234':
                # 状态保持，设置用户名到cookie中表示登陆成功
                response = redirect(url_for('transfer'))
                response.set_cookie('username', username)
                return response
            else:
                print('密码错误')

    return render_template('login_test.html')


@app.route('/transfer', methods=["POST", "GET"])
def transfer():
    # 从cookie中取到用户名
    username = request.cookies.get('username', None)
    # 若是没有取到，表明没有登陆
    if not username:
        return redirect(url_for('login'))

    if request.method == "POST":
        to_account = request.form.get("to_account")
        money = request.form.get("money")
        # 取出表单中的 csrf_token
        form_csrf_token = request.form.get("csrf_token")
        # 取出 cookie 中的 csrf_token
        cookie_csrf_token = request.cookies.get("csrf_token")
        # 进行对比
        if cookie_csrf_token != form_csrf_token:
            return 'token校验失败，非法操做'
        print('伪装执行转操做，将当前登陆用户的钱转帐到指定帐户')
        return '转帐 %s 元到 %s 成功' % (money, to_account)

    # 渲染转换页面
        # 生成 csrf_token 的值
    csrf_token = generate_csrf()

    # 渲染转换页面，传入 csrf_token 到模板中
    response = make_response(render_template('transfer_test.html', csrf_token=csrf_token))
    # 设置csrf_token到cookie中，用于提交校验
    response.set_cookie('csrf_token', csrf_token)
    return response

if __name__ == '__main__':
    app.run(debug=True, port=9000)

login.test.py(登陆模板页面)后端

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>登陆</title>
</head>
<body>

<h1>我是网站A，登陆页面</h1>

<form method="post">
    <label>用户名：</label><input type="text" name="username" placeholder="请输入用户名"><br/>
    <label>密码：</label><input type="password" name="password" placeholder="请输入密码"><br/>
    <input type="submit" value="登陆">
</form>

</body>
</html>

transfer_test.py(转帐模板页面)安全

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>转帐</title>
</head>
<body>
<h1>我是网站A，转帐页面</h1>

<form method="post">
    <input type="hidden" name="csrf_token" value="{{ csrf_token }}">
    <label>帐户：</label><input type="text" name="to_account" placeholder="请输入要转帐的帐户"><br/>
    <label>金额：</label><input type="number" name="money" placeholder="请输入转帐金额"><br/>
    <input type="submit" value="转帐">
</form>

</body>
</html>

下面是进行CSRF攻击的恶意网站(俗称钓鱼网站)：hack_test.py(视图页)服务器

from flask import Flask
from flask import render_template

app = Flask(__name__)

@app.route('/hongbao')
def index():
    return render_template('hack_test.html')

if __name__ == '__main__':
    app.run(debug=True, port=8000)

hack_test.py(模板页)cookie

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>

<h1>我是网站B</h1>

<form method="post" action="http://127.0.0.1:9000/transfer">
    <input type="hidden" name="to_account" value="999999">
    <input type="hidden" name="money" value="190000" hidden>
    <input type="submit" value="点击领取优惠券">
</form>

</body>
</html>

以上代码亲测有效，若有问题或者建议，欢迎在下面留言☺☺☺app