常见的Python安全漏洞及防范方法

2023-04-17 00:00:00 安全漏洞 防范 常见
  1. SQL注入漏洞

SQL注入是指攻击者通过在应用程序的输入框中输入SQL语句来获取或篡改数据库中的数据。在Python应用程序中，这种漏洞通常由于没有对用户输入进行正确的过滤或转义而产生。

例如，假设我们有一个简单的登录表单，其中有一个用户名输入框和一个密码输入框：

import sqlite3

def login(username, password):
    conn = sqlite3.connect('users.db')
    cur = conn.cursor()
    query = "SELECT * FROM users WHERE username = '{}' AND password = '{}'".format(username, password)
    cur.execute(query)
    result = cur.fetchone()
    cur.close()
    conn.close()
    return result

这个简单的登录功能有一个SQL注入漏洞，攻击者可以通过在用户名输入框中输入一个SQL语句来执行任意的数据库操作。例如，如果攻击者在用户名输入框中输入“' or 1=1;--”，那么这个查询语句就会变成：

SELECT * FROM users WHERE username = '' or 1=1;--' AND password = ''

这个语句会返回所有的用户信息，因为1=1永远是成立的。为了防止SQL注入攻击，我们需要使用参数化查询，而不是在查询语句中直接拼接用户输入的值，可以使用Python内置的sqlite3模块，这个模块会自动进行参数化查询：

def login(username, password):
    conn = sqlite3.connect('users.db')
    cur = conn.cursor()
    query = "SELECT * FROM users WHERE username = ? AND password = ?"
    cur.execute(query, (username, password))
    result = cur.fetchone()
    cur.close()
    conn.close()
    return result

在这个例子中，我们使用了参数化查询，将用户名和密码作为参数传递给cur.execute()函数，而不是直接拼接到查询语句中。这样，无论用户输入了什么内容，我们都可以保证查询语句是合法的。

  1. XSS漏洞

XSS漏洞是一种跨站脚本攻击，攻击者通过在应用程序中注入恶意脚本来获取用户的登录凭证，篡改页面内容等等。在Python应用程序中，XSS漏洞通常是由于没有对输出的数据进行正确的转义和过滤而产生。

例如，假设我们有一个简单的留言板功能，在这个功能中，用户可以发布评论，并且这些评论会在页面上显示：

from flask import Flask, request

app = Flask(__name__)

@app.route('/comment', methods=['POST'])
def comment():
    message = request.form['message']
    return '<p>{}</p>'.format(message)

这个例子也有XSS漏洞，攻击者可以在评论框中输入一些恶意的HTML代码，这些代码将会被原样输出到页面上，例如：

<script>
  alert('pidancode.com');
</script>

为了防止XSS攻击，我们需要对输出的内容进行过滤和转义，将特殊字符进行转义，这样就可以防止攻击者在应用程序中注入任意的HTML/JavaScript代码。

可以使用Python内置的html模块进行转义：

import html

@app.route('/comment', methods=['POST'])
def comment():
    message = request.form['message']
    return '<p>{}</p>'.format(html.escape(message))

在这个例子中，我们使用了html.escape()函数将特殊字符进行转义，这样就可以避免输出恶意的HTML/JavaScript代码。

  1. 文件包含漏洞

文件包含漏洞是一种安全漏洞，攻击者可以利用这个漏洞在服务器上执行任意的代码。在Python应用程序中，常见的文件包含漏洞通常是由于没有对用户输入进行正确的过滤或者路径遍历攻击而产生。

例如，假设我们有一个简单的图像加载功能，用户可以通过URL参数加载图像：

from flask import Flask, request, send_file

app = Flask(__name__)

@app.route('/image')
def image():
    filename = request.args.get('filename', '')
    return send_file(filename, mimetype='image/jpeg')

这个例子有一个文件包含漏洞，攻击者可以通过URL参数加载任意的文件，例如，可以加载一些Python代码并在服务器上执行：

http://example.com/image?filename=/var/www/html/app.py

为了防止文件包含漏洞，我们需要对用户输入的路径进行过滤和限制，只允许访问指定的目录或者使用绝对路径，避免使用相对路径或者使用../这样的路径遍历攻击。

import os
from flask import abort, send_file

@app.route('/image')
def image():
    filename = request.args.get('filename', '')
    if not filename.startswith('/photos/'):
        abort(403)
    fullpath = os.path.abspath('.' + filename)
    if not os.path.exists(fullpath):
        abort(404)
    return send_file(fullpath, mimetype='image/jpeg')

在这个例子中，我们通过限制用户只能访问指定的目录来阻止了路径遍历攻击，并且使用os.path.abspath()函数将相对路径转换为绝对路径，可以避免一些普通的路径漏洞。如果请求的文件不存在或者路径不合法，我们会返回相应的错误码，而不是继续执行可能会带来安全隐患的代码。

  1. CSRF漏洞

CSRF漏洞是一种跨站请求伪造攻击，攻击者利用用户已经登录的凭证来发起一些恶意的请求。在Python应用程序中，常见的CSRF漏洞通常是由于没有正确的使用CSRF令牌进行防范。

例如，假设我们有一个简单的转账功能，用户可以通过一个POST请求将钱转到另外一个账户：

from flask import Flask, request

app = Flask(__name__)

@app.route('/transfer', methods=['POST'])
def transfer():
    from_account = request.form['from_account']
    to_account = request.form['to_account']
    amount = request.form['amount']
    if amount > 1000:
        return 'Transaction limit exceeded'
    # TODO: Transfer the money
    return 'Money transferred'

这个例子有一个CSRF漏洞，攻击者可以在另外一个页面中插入一些HTML代码，伪造一个POST请求，而这个请求会携带用户的登录凭证发送过来，从而让服务器误以为是用户在执行操作。

为了防范CSRF漏洞，我们可以使用CSRF令牌来进行防范，例如，可以在HTML页面中插入一个名为_csrf_token的隐藏字段，将该字段的值设置为一个随机数，服务器在接收到POST请求时，检查_csrf_token字段的值是否与之前生成的值一样即可。

import uuid
from flask import Flask, request, render_template_string, session

app = Flask(__name__)
app.secret_key = str(uuid.uuid4())

@app.before_request
def csrf_protect():
    if request.method == "POST":
        token = session.pop('_csrf_token', None)
        if not token or token != request.form.get('_csrf_token'):
            abort(403)

@app.route('/transfer', methods=['GET', 'POST'])
def transfer():
    if request.method == 'GET':
        session['_csrf_token'] = str(uuid.uuid4())
        return render_template_string('''
            <form method="POST">
                <input type="hidden" name="_csrf_token" value="{{ csrf_token() }}">
                <p><input type="text" name="from_account"></p>
                <p><input type="text" name="to_account"></p>
                <p><input type="text" name="amount"></p>
                <p><input type="submit" value="Transfer"></p>
            </form>
        ''')
    else:
        from_account = request.form['from_account']
        to_account = request.form['to_account']
        amount = request.form['amount']
        if amount > 1000:
            return 'Transaction limit exceeded'
        # TODO: Transfer the money
        return 'Money transferred'

在这个例子中，我们使用了一个名为_csrf_token的隐藏字段，将其值设置为一个随机数，在接收到POST请求时，我们首先从session中取出令牌的值，与请求中的值进行比较，如果不一样就返回403错误，防止了CSRF攻击。同时，我们还使用了Flask的模板引擎来创建HTML页面，确保了_csrf_token字段的正确显示。

相关文章