Flask模拟实现CSRF攻击的方法_Python

csrf

csrf全拼为cross site request forgery，译为跨站请求伪造。

csrf指攻击者盗用了你的身份，以你的名义发送恶意请求。

包括：以你名义发送邮件，发消息，盗取你的账号，甚至于购买商品，虚拟货币转账......

造成的问题：个人隐私泄露以及财产安全。

csrf攻击示意图

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

Flask模拟实现CSRF攻击的方法

防止 csrf

1.在客户端向后端请求界面数据的时候，后端会往响应中的 cookie 中设置 csrf_token 的值

2.在 form 表单中添加一个隐藏的的字段，值也是 csrf_token

3.在用户点击提交的时候，会带上这两个值向后台发起请求

4.后端接受到请求，以会以下几件事件： •从 cookie中取出 csrf_token

从表单数据中取出来隐藏的 csrf_token 的值
进行对比

5.如果比较之后两值一样，那么代表是正常的请求，如果没取到或者比较不一样，代表不是正常的请求，不执行下一步操作

代码演示

未进行 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>

运行测试，进入到转账页面之后，查看 cookie 和 html 源代码

Flask模拟实现CSRF攻击的方法

在执行转账逻辑之前进行 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 攻击

在 flask 中， flask-wtf 扩展有一套完善的 csrf 防护体系，对于我们开发者来说，使用起来非常简单

在 flaskform 中实现校验

设置应用程序的 secret_key

用于加密生成的 csrf_token 的值

1	`app.secret_key` `=` `"#此处可以写随机字符串#"`

在模板的表单中添加以下代码

									<form method="post">

									  {{ form.csrf_token() }}

									  {{ form.username.label }} {{ form.username }}<br/>

									  {{ form.password.label }} {{ form.password }}<br/>

									  {{ form.password2.label }} {{ form.password2 }}<br/>

									  {{ form.submit }}

									</form>

渲染出来的前端页面为：

Flask模拟实现CSRF攻击的方法

设置完毕，cookie 中的 csrf_token 不需要我们关心，会自动帮我们设置

单独使用

设置应用程序的 secret_key

用于加密生成的 csrf_token 的值

1	`app.secret_key` `=` `"#此处可以写随机字符串#"`

导入 flask_wtf.csrf 中的 csrfprotect 类，进行初始化，并在初始化的时候关联 app

1 2	`from` `flask.ext.wtf` `import` `csrfprotect` `csrfprotect(app)`

如果模板中有表单，不需要做任何事。与之前一样:

									<form method="post">

									  {{ form.csrf_token }}

									  ...

									</form>

但如果模板中没有表单，你仍需要 csrf 令牌:

									<form method="post" action="/">

									  <input type="hidden" name="csrf_token" value="{{ csrf_token() }}" />

									</form>

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。　

原文链接：https://www.cnblogs.com/wf-skylark/p/9306203.html

Flask模拟实现CSRF攻击的方法

延伸 · 阅读

Windows下Python的Django框架环境部署及应用编写入门

Python实现敏感词过滤的4种方法

分析并输出Python代码依赖的库的实现代码

python中使用smtplib和email模块发送邮件实例

python访问类中docstring注释的实现方法

keras.layer.input()用法说明

Django自带的用户验证系统实现

pandas将DataFrame的列变成行索引的方法

PyCharm设置SSH远程调试的方法

Python安装图文教程 Pycharm安装教程

python是什么意思？python有什么用？

使用Python抓取模板之家的CSS模板

Python 列表(List)操作方法详解