避免命令注入的关键在于严格控制用户输入，确保其无法影响到系统命令的执行。这并非易事，需要从代码编写、安全审计到安全意识培养多方面入手。
我曾经参与过一个项目，其中一个功能允许用户输入文件名来下载文件。看似简单的功能，却差点酿成大祸。开发人员直接将用户输入拼接进系统命令 cat filename | gzip > output.gz 中。一个恶意用户只需输入 filename; rm -rf /，就能轻易删除服务器上的所有文件。 这给我敲响了警钟，也让我深刻理解了命令注入的危害。
避免命令注入，最有效的方法是永远不要直接将用户输入拼接进系统命令。 这听起来简单，但实际操作中却容易忽略。 举个例子，假设你需要根据用户输入的文件名读取文件内容。错误的做法是：
filename = request.GET['filename']
os.system(f'cat {filename}')正确的做法是使用操作系统提供的安全函数，例如Python的 subprocess 模块，并对用户输入进行严格的验证和过滤。 正确的代码片段如下：
Python之模块学习 中文WORD版
下载
本文档主要讲述的是Python之模块学习；python是由一系列的模块组成的，每个模块就是一个py为后缀的文件，同时模块也是一个命名空间，从而避免了变量名称冲突的问题。模块我们就可以理解为lib库，如果需要使用某个模块中的函数或对象，则要导入这个模块才可以使用，除了系统默认的模块（内置函数）不需要导入外。希望本文档会给有需要的朋友带来帮助；感兴趣的朋友可以过来看看
import subprocess
import os
filename = request.GET['filename']
# 验证文件名，只允许字母数字和下划线
if not filename.isalnum() and '_' not in filename:
raise ValueError("Invalid filename")
# 构建绝对路径，避免路径穿越攻击
filepath = os.path.join('/safe/directory/', filename) # 指定安全目录
try:
with open(filepath, 'r') as f:
# 使用subprocess安全地执行命令
process = subprocess.Popen(['cat', filepath], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = process.communicate()
if stderr:
raise RuntimeError(f"Error reading file: {stderr.decode()}")
print(stdout.decode())
except FileNotFoundError:
print("File not found.")
except (ValueError, RuntimeError) as e:
print(f"Error: {e}")
这段代码做了几件事：
- 验证文件名: 只允许字母数字和下划线，防止用户输入恶意字符。 实际应用中，需要根据具体需求调整验证规则。
-
使用绝对路径: 避免路径穿越攻击，确保只访问指定目录下的文件。
/safe/directory/应该是一个事先创建好的，只有程序有读写权限的目录。 -
使用
subprocess模块: 安全地执行系统命令，避免直接使用os.system。 - 错误处理: 处理可能出现的异常，例如文件不存在或读取失败。
除了代码层面的安全措施，还需要进行定期的安全审计，检查代码中是否存在潜在的命令注入漏洞。 更重要的是，培养开发人员的安全意识，让他们时刻警惕命令注入的风险。 只有多方面共同努力，才能有效避免命令注入带来的安全隐患。 记住，安全无小事，任何一个细节都可能成为攻击的突破口。
