Aprimorando a Segurança com Hashing de Senhas
Em nossas implementações anteriores, armazenamos senhas em texto simples, o que representa um risco de segurança significativo. Nesta etapa, aprimoraremos nosso sistema de autenticação implementando o hashing de senhas usando o algoritmo bcrypt.
Por que Fazer Hash de Senhas?
Armazenar senhas em texto simples apresenta vários riscos de segurança:
- Se seu banco de dados for comprometido, os invasores podem usar as senhas imediatamente
- Os usuários geralmente reutilizam senhas em vários sites, portanto, uma violação pode afetar outros serviços
- Viola as melhores práticas de segurança e pode violar os regulamentos de proteção de dados
O hashing de senhas resolve esses problemas ao:
- Converter senhas em strings de comprimento fixo que não podem ser revertidas
- Adicionar um "sal" para impedir que invasores usem tabelas de pesquisa pré-computadas (tabelas rainbow)
- Garantir que, mesmo que dois usuários tenham a mesma senha, seus hashes serão diferentes
Instalando Pacotes Necessários
Primeiro, precisamos instalar o pacote bcrypt:
pip install bcrypt
Você deve ver uma saída confirmando a instalação:
Collecting bcrypt
Downloading bcrypt-4.0.1-cp36-abi3-manylinux_2_28_x86_64.whl (593 kB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 593.8/593.8 KB 10.5 MB/s eta 0:00:00
Installing collected packages: bcrypt
Successfully installed bcrypt-4.0.1
Criando um Sistema de Registro de Usuário
Vamos aprimorar nosso servidor para incluir o registro de usuário com hashing de senha. Atualize o arquivo server.py:
code ~/project/auth_demo/server.py
Substitua o código existente por:
from flask import Flask, request, jsonify
import jwt
import datetime
import os
import bcrypt
app = Flask(__name__)
## Secret key for signing JWT tokens
SECRET_KEY = os.urandom(24)
## Store users as a dictionary with username as key and a dict of hashed password and roles as value
users = {}
def hash_password(password):
"""Hash a password using bcrypt"""
## Convert password to bytes if it's a string
if isinstance(password, str):
password = password.encode('utf-8')
## Generate a salt and hash the password
salt = bcrypt.gensalt()
hashed = bcrypt.hashpw(password, salt)
return hashed
def check_password(password, hashed):
"""Verify a password against its hash"""
## Convert password to bytes if it's a string
if isinstance(password, str):
password = password.encode('utf-8')
## Check if the password matches the hash
return bcrypt.checkpw(password, hashed)
def generate_token(username, role):
"""Generate a JWT token for the authenticated user"""
## Token expires after 30 minutes
expiration = datetime.datetime.utcnow() + datetime.timedelta(minutes=30)
payload = {
'username': username,
'role': role,
'exp': expiration
}
## Create the JWT token
token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')
return token
def verify_token(token):
"""Verify the JWT token"""
try:
## Decode and verify the token
payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])
return payload
except jwt.ExpiredSignatureError:
## Token has expired
return None
except jwt.InvalidTokenError:
## Invalid token
return None
@app.route('/register', methods=['POST'])
def register():
data = request.get_json()
## Extract registration data
username = data.get('username')
password = data.get('password')
## Basic validation
if not username or not password:
return jsonify({"status": "error", "message": "Username and password are required"}), 400
## Check if the username already exists
if username in users:
return jsonify({"status": "error", "message": "Username already exists"}), 409
## Hash the password and store the user
hashed_password = hash_password(password)
users[username] = {
'password': hashed_password,
'role': 'user' ## Default role
}
return jsonify({
"status": "success",
"message": f"User {username} registered successfully"
})
@app.route('/login', methods=['POST'])
def login():
data = request.get_json()
## Extract username and password from the request
username = data.get('username')
password = data.get('password')
## Check if the user exists
if username not in users:
return jsonify({"status": "error", "message": "Invalid credentials"}), 401
## Check if the password is correct
if check_password(password, users[username]['password']):
## Generate a JWT token
token = generate_token(username, users[username]['role'])
return jsonify({
"status": "success",
"message": f"Welcome {username}!",
"token": token
})
else:
return jsonify({"status": "error", "message": "Invalid credentials"}), 401
@app.route('/protected', methods=['GET'])
def protected():
## Get the token from the Authorization header
auth_header = request.headers.get('Authorization')
if not auth_header or not auth_header.startswith('Bearer '):
return jsonify({"status": "error", "message": "Missing or invalid token"}), 401
## Extract the token
token = auth_header.split(' ')[1]
## Verify the token
payload = verify_token(token)
if payload:
return jsonify({
"status": "success",
"message": f"You are viewing protected content, {payload['username']}!",
"role": payload['role']
})
else:
return jsonify({"status": "error", "message": "Invalid or expired token"}), 401
@app.route('/admin', methods=['GET'])
def admin():
## Get the token from the Authorization header
auth_header = request.headers.get('Authorization')
if not auth_header or not auth_header.startswith('Bearer '):
return jsonify({"status": "error", "message": "Missing or invalid token"}), 401
## Extract the token
token = auth_header.split(' ')[1]
## Verify the token
payload = verify_token(token)
if not payload:
return jsonify({"status": "error", "message": "Invalid or expired token"}), 401
## Check if user has admin role
if payload['role'] != 'admin':
return jsonify({"status": "error", "message": "Admin access required"}), 403
return jsonify({
"status": "success",
"message": f"Welcome to the admin panel, {payload['username']}!"
})
## Add an admin user for testing
admin_password = hash_password("admin123")
users["admin"] = {
'password': admin_password,
'role': 'admin'
}
if __name__ == '__main__':
app.run(host='0.0.0.0', port=5000, debug=True)
Este servidor aprimorado:
- Usa bcrypt para hashing seguro de senha
- Inclui um endpoint de registro de usuário
- Adiciona controle de acesso baseado em função (admin vs usuários regulares)
- Pré-cria um usuário admin para teste
Atualizando o Cliente
Agora, vamos atualizar nosso cliente para testar esses novos recursos. Atualize o arquivo client.py:
code ~/project/auth_demo/client.py
Substitua o código existente por:
import requests
import json
## Server URL
BASE_URL = "http://localhost:5000"
def register(username, password):
"""Register a new user"""
response = requests.post(
f"{BASE_URL}/register",
json={"username": username, "password": password}
)
print(f"Registration response: {response.status_code}")
print(response.json())
return response.status_code == 200
def login(username, password):
"""Authenticate with the server using username and password"""
response = requests.post(
f"{BASE_URL}/login",
json={"username": username, "password": password}
)
print(f"Login response: {response.status_code}")
data = response.json()
print(data)
## Return the token if login was successful
return data.get('token') if response.status_code == 200 else None
def access_protected(token=None):
"""Access a protected resource using JWT token"""
headers = {}
if token:
headers['Authorization'] = f'Bearer {token}'
response = requests.get(f"{BASE_URL}/protected", headers=headers)
print(f"Protected resource response: {response.status_code}")
print(response.json())
return response
def access_admin(token=None):
"""Access the admin panel using JWT token"""
headers = {}
if token:
headers['Authorization'] = f'Bearer {token}'
response = requests.get(f"{BASE_URL}/admin", headers=headers)
print(f"Admin panel response: {response.status_code}")
print(response.json())
return response
if __name__ == "__main__":
## Test user registration
print("=== Registering a new user ===")
register("testuser", "testpass123")
## Try to register with the same username (should fail)
print("\n=== Registering with existing username ===")
register("testuser", "differentpass")
## Test regular user login and access
print("\n=== Regular user login ===")
user_token = login("testuser", "testpass123")
if user_token:
print("\n=== Regular user accessing protected resource ===")
access_protected(user_token)
print("\n=== Regular user trying to access admin panel ===")
access_admin(user_token)
## Test admin login and access
print("\n=== Admin login ===")
admin_token = login("admin", "admin123")
if admin_token:
print("\n=== Admin accessing protected resource ===")
access_protected(admin_token)
print("\n=== Admin accessing admin panel ===")
access_admin(admin_token)
Este cliente atualizado:
- Testa a funcionalidade de registro de usuário
- Demonstra o controle de acesso baseado em função
- Testa a autenticação de usuário regular e admin
Executando o Sistema de Autenticação Aprimorado
Agora, vamos executar nosso sistema de autenticação aprimorado. Primeiro, pare quaisquer processos de servidor em execução com Ctrl+C, depois inicie o servidor atualizado:
python ~/project/auth_demo/server.py
Abra uma nova aba de terminal ou use a segunda aba existente para executar o cliente:
cd ~/project/auth_demo
python client.py
Você deve ver uma saída demonstrando:
- Registro de usuário bem-sucedido
- Registro falhado com um nome de usuário existente
- Login de usuário regular e acesso a recursos protegidos
- Tentativa falhada por um usuário regular de acessar o painel de administração
- Login de administrador e acesso a recursos protegidos e ao painel de administração
Esta saída confirma que nosso sistema de autenticação aprimorado com hashing de senha e controle de acesso baseado em função está funcionando corretamente.
Vantagens de Segurança
Nosso sistema aprimorado agora fornece:
- Armazenamento Seguro de Senhas: As senhas são hashadas usando bcrypt, uma função de hashing lenta projetada para resistir a ataques de força bruta.
- Controle de Acesso Baseado em Função: Diferentes usuários podem ter diferentes níveis de acesso com base em suas funções.
- Autenticação Baseada em Token: Continua a fornecer os benefícios dos tokens JWT que implementamos anteriormente.
- Registro de Usuário: Permite a criação dinâmica de usuários com armazenamento seguro de senhas.
Essas melhorias tornam nosso sistema de autenticação muito mais robusto e adequado para aplicações do mundo real.