zoukankan      html  css  js  c++  java
  • Flask基础(17)-->防止 CSRF 攻击

    CSRF

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

    CSRF攻击示意图

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

    防止 CSRF 攻击

    步骤

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

    那么下面我们用一个代码案例进行演示一下:防止CSRF攻击的代码示例(如果想测试没有CSRF预防的情况,可以在代码里直接更改即可:即去掉转账视图crsf_token验证功能即可)

    CRSF_test.py(视图页面)

    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(模板页)

    <!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>

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

  • 相关阅读:
    js 字符串转化成数字
    web项目中各种路径的获取
    个人作业——软件工程实践总结作业
    Beta 答辩总结
    Beta 冲刺 (7/7)
    Beta 冲刺 (6/7)
    Beta 冲刺 (5/7)
    Beta 冲刺 (4/7)
    Beta 冲刺 (3/7)
    软件产品案例分析(团队)
  • 原文地址:https://www.cnblogs.com/888888CN/p/9489345.html
Copyright © 2011-2022 走看看