flask中安全策略简要说明

PT、小小马2024-09-16 10:06

问题：针对服务安全基础策略

策略一、接口入参长度限制

策略二、接口入参不接收特殊字符

解决方案：

一、flask对象实例化后，提供有config配置

复制代码

app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 设置为16MB

二、flask有提供参数校验前的函数，可以以装饰器的方式调用实现接收参数前的处理

from flask import Flask, request, jsonify

app = Flask(name)

定义一个函数来校验参数中是否包含特殊字符

def validate_parameters(params):

for key, value in params.items():

if not isinstance(value, str):

continue

if any(char in value for char in $'\<', '\>', '\&', '"', "'", ';', '\\\\'$ ):

return False

return True

@app.before_request

def before_request():

if request.method in $'POST', 'PUT'$ :

if 'application/json' in request.headers.get('Content-Type', ''):

params = request.get_json()

else:

params = request.form

if not validate_parameters(params):

return jsonify({'error': 'Parameters contain special characters'}), 400

@app.route('/your-endpoint', methods= $'POST'$ )

def your_endpoint():

这里可以安全地使用参数，因为它们已经被校验过不包含特殊字符

data = request.get_json() if request.is_json else request.form

return jsonify({'message': 'Success'}), 200

if name == 'main':

app.run(debug=True)

小结：提供前置函数，这个思想非常值得学习。本来还在考虑，按正常操作，会需要给每一个路由都添加指定的方法来实现逻辑筛选，这种常规思维的方式会让工作量变的很大。做许多没有意义的事情。后续这个思想会需要继续进行深化