- CSRF全拼为Cross Site Request Forgery,译为跨站请求伪造。
- CSRF指攻击者盗用了你的身份,以你的名义发送恶意请求。
- 包括:以你名义发送邮件,发消息,盗取你的账号,甚至于购买商品,虚拟货币转账......
- 造成的问题:个人隐私泄露以及财产安全。
CSRF攻击示意图
-
客户端访问服务器时没有同服务器做安全验证
防止CSRF攻击
步骤
- 在客户端向后端请求界面数据的时候,后端会往响应中的cookie中设置csrf_token的值
- 在Form表单中添加一个隐藏的字段,值也是csrf_token
- 在用户点击提交的时候,会带上这两个值向后台发起请求
- 后端接收到请求,会以此啊几件事件:
- 从cookie中取出csrf_token的值
- 从表单数据中取出隐藏的csrf_token的值
- 进行对比
- 如果比较之后两个值一样,那么代表是正常的请求,如果没有取到或者比较不一样,代表不是正常的请求,不会执行下一步操作
代码演示
未进行 csrf 校验的 WebA
- 后端代码实现
from flask import Flask, render_template, make_response
from flask import redirect
from flask import request
from flask import url_for
app = Flask(__name__)
@app.route('/', methods=["POST", "GET"])
def index():
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('temp_login.html')
@app.route('/transfer', methods=["POST", "GET"])
def transfer():
# 从cookie中取到用户名
username = request.cookies.get('username', None)
# 如果没有取到,代表没有登录
if not username:
return redirect(url_for('index'))
if request.method == "POST":
to_account = request.form.get("to_account")
money = request.form.get("money")
print('假装执行转操作,将当前登录用户的钱转账到指定账户')
return '转账 %s 元到 %s 成功' % (money, to_account)
# 渲染转换页面
response = make_response(render_template('temp_transfer.html'))
return response
if __name__ == '__main__':
app.run(debug=True, port=9000)
- 前端登录页面代码
<!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>
- 前端代码转账代码
<!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="to_account" placeholder="请输入要转账的账户"><br/>
<label>金额:</label><input type="number" name="money" placeholder="请输入转账金额"><br/>
<input type="submit" value="转账">
</form>
</body>
</html>
攻击网站B代码
- 后端代码实现
from flask import Flask
from flask import render_template
app = Flask(__name__)
@app.route('/')
def index():
return render_template('temp_index.html')
if __name__ == '__main__':
app.run(debug=True, port=8000)
- 前端代码实现
<!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>
--运行测试,在用户登录网站A的情况下,点击网站B的按钮,可以实现伪造访问
在网站A中模拟实现csrf_token校验的流程
- 添加生成csrf_token的函数
# 生成csrf_token的函数
def generate_csrf():
return bytes.decode(base64.b64encode(os.urandom(48)))
- 在渲染转账页面的时候,做以下几件事情:
- 生成csrf_token的值
- 在返回转账页面的响应里面设置csrf_token到cookie中
- 将csrf_token保存到变单的隐藏字段中
@app.route('/transfer', methods=["POST", "GET"])
def transfer():
...
# 生成 csrf_token 的值
csrf_token = generate_csrf()
# 渲染转换页面,传入 csrf_token 到模板中
response = make_response(render_template('temp_transfer.html', csrf_token=csrf_token))
# 设置csrf_token到cookie中,用于提交校验
response.set_cookie('csrf_token', csrf_token)
return response
- 在转账模板表单中添加 csrf_token 隐藏字段
<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>
- 在执行转账逻辑之前进行 csrf_token 的校验
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)
- 运行测试,用户直接在网站A操作没有问题,在去网站B进行操作,发现转账不成功,因为网站B获取不到表单中的csrf_token的隐藏字段,而且浏览器有 同源策略,网站B是获取不到网站A的cookie的,所以就解决了 跨站请求伪造 的问题
在 Flask 项目中解决 CSRF 攻击
CSRF
- 包含请求体的请求都需要开启CSRF
# CSRF1. 在app创建的地方开启CSRFProtect保护
# 开启保护之后,程序会获取cookie中的随机值,以及从表单或者ajax中获取随机值,进行对比
# if 对比失败,则无法访问路由
# 后续需要设置随机值到cookie中,以及增加ajax的headers
from flask_wtf.csrf import CSRFProtect
...
app.config.from_object(Config)
...
CSRFProtect(app)
- CSRFProtect只做校验工作,cookie中的csrf_token和表单中的csrf_token需要我们自己实现
思路分析
一、在 前端发起的POST请求 中没有进行csrf_token校验,根据csrf_token校验原理,具体操作步骤有以下几步:
- 后端生成csrf_token的值,在前端请求界面的时候将值传给前端,传给前端的方式可能有以下两种方式:
- 在模版中的From表单中的添加隐藏的字段
- 将csrf_token使用cookie的方式传给前端
- 在前端 发起请求时,在表单或者请求头中带上指定的csrf_token
- 后端在接收请求之后,取到前端发送过来的csrf_token,与第一步生成的csrf_token值进行校验
- 如果校验对csrf_token一致,则代表是正常的请求,否则是伪造的请求,不予通过
二、在Flask中,CSRFProtect这个类专门只对指定的app进行csrf_token校验操作,所以开发者需要做以下的操作:
- 生成csrf_token的值
- 将csrf_token的值传给前端浏览器
完成代码逻辑
- 生成csrf_token的值
# CSRF2. 增加请求钩子,在请求之后设置cookie
# 我们无法判断用户第一次访问网页,是哪个网页,不能写死给那个路由的某个我网页
# 所有就需要对所有飞请求进行监听
# 导入生成 csrf_token 值的函数
from flask_wtf.csrf import generate_csrf
# 调用函数生成 csrf_token
csrf_token = generate_csrf()
- 将 csrf_token 的值传给前端浏览器
- 实现思路:可以在请求勾子函数中完成此逻辑
# CSRF3. generate_csrf()-->源码-->csrf_token会被缓存起来,多次调用,只会返回相同的token(被强制删除或者cookie过期,会生成新的token)
# 如果获取csrf_token,可以使用session['csrf_token']--->token会被自动扩展到session中
# 应该对所有的post/put/delete/请求增加ajax的headers或者表单的隐藏字段(对数据有修改/提交/设置)
@app.after_request
def after_request(response):
# 调用函数生成 csrf_token
csrf_token = generate_csrf()
# 通过 cookie 将值传给前端
response.set_cookie("csrf_token", csrf_token)
return response
- 在前端请求时带上 csrf_token 值
- 根据登录和注册的业务逻辑,当前采用的是 ajax 请求
- 所以在提交登录或者注册请求时,需要在请求头中添加 X-CSRFToken 的键值对
$.ajax({
url:"/passport/register",
type: "post",
headers: {
"X-CSRFToken": getCookie("csrf_token")
},
data: JSON.stringify(params),
contentType: "application/json",
success: function (resp) {
if (resp.errno == "0"){
// 刷新当前界面
location.reload()
}else {
$("#register-password-err").html(resp.errmsg)
$("#register-password-err").show()
}
}
})
打开 CSRFProtect 的代码,并运行测试 注:X-CSRFToken 这个 key 是 CSRFProtect 这个类里面指定的,具体请点击进入到此类查看源代码