Flask基础(17)-->防止 CSRF 攻击
CSRF
CSRF全拼为Cross Site Request Forgery，译为跨站请求伪造。CSRF指攻击者盗用了你的身份，以你的名义发送恶意请求。- 包括：以你名义发送邮件，发消息，盗取你的账号，甚至于购买商品，虚拟货币转账......
- 造成的问题：个人隐私泄露以及财产安全。
CSRF攻击示意图
- 客户端访问服务器时没有同服务器做安全验证
防止 CSRF 攻击
步骤
- 在客户端向后端请求界面数据的时候，后端会往响应中的 cookie 中设置 csrf_token 的值
- 在 Form 表单中添加一个隐藏的的字段，值也是 csrf_token
- 在用户点击提交的时候，会带上这两个值向后台发起请求
- 后端接受到请求，以会以下几件事件：
- 从 cookie中取出 csrf_token
- 从 表单数据中取出来隐藏的 csrf_token 的值
- 进行对比
- 如果比较之后两值一样，那么代表是正常的请求，如果没取到或者比较不一样，代表不是正常的请求，不执行下一步操作
那么下面我们用一个代码案例进行演示一下：防止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>
以上代码亲测有效，如有问题或者建议，欢迎在下面留言☺☺☺
