درس ۸ سطح متوسط: امنیت در پایتون – رمزنگاری، اعتبارسنجی، جلوگیری از حملات و نوشتن کد امن

import secrets
import string

# توکن امن ۳۲ بایتی
token = secrets.token_hex(32)
print(token)

# رمزعبور امن
alphabet = string.ascii_letters + string.digits + "!@#$%"
password = ''.join(secrets.choice(alphabet) for i in range(16))
print(password)

import bcrypt

# ذخیره رمزعبور
password = "MySecurePass123"
hashed = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
print(hashed)

# بررسی رمزعبور
def verify_password(plain_password, hashed):
    return bcrypt.checkpw(plain_password.encode('utf-8'), hashed)

print(verify_password("MySecurePass123", hashed))  # True
print(verify_password("wrong", hashed))            # False

import jwt
import datetime

secret_key = "your-super-secret-key-256-bit"

# ساخت توکن
payload = {
    "user_id": 123,
    "exp": datetime.datetime.utcnow() + datetime.timedelta(hours=1),
    "iat": datetime.datetime.utcnow()
}
token = jwt.encode(payload, secret_key, algorithm="HS256")
print(token)

# اعتبارسنجی توکن
try:
    decoded = jwt.decode(token, secret_key, algorithms=["HS256"])
    print("توکن معتبر:", decoded)
except jwt.ExpiredSignatureError:
    print("توکن منقضی شده")
except jwt.InvalidTokenError:
    print("توکن نامعتبر")

import sqlite3

# خطرناک - هرگز این کار رو نکن!
# cursor.execute(f"SELECT * FROM users WHERE name = '{user_input}'")

# امن - همیشه از پارامتر استفاده کن
safe_query = "SELECT * FROM users WHERE name = ? AND active = ?"
cursor.execute(safe_query, (user_input, 1))

import re
from html import escape

def validate_email(email):
    pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
    return re.match(pattern, email) is not None

def sanitize_input(user_input):
    # جلوگیری از XSS
    return escape(user_input.strip())

# استفاده
if validate_email(user_email):
    safe_name = sanitize_input(user_name)
else:
    raise ValueError("ایمیل نامعتبر")

from collections import defaultdict
import time

login_attempts = defaultdict(list)

def is_rate_limited(ip, max_attempts=5, window=300):
    now = time.time()
    attempts = login_attempts[ip]
    # حذف تلاش‌های قدیمی
    attempts = [t for t in attempts if now - t < window] login_attempts[ip] = attempts if len(attempts) >= max_attempts:
        return True
    attempts.append(now)
    return False

import bcrypt
import jwt
import secrets
from datetime import datetime, timedelta

class SecureAuth:
    def __init__(self):
        self.secret_key = secrets.token_hex(32)
        self.users = {}  # در پروژه واقعی از دیتابیس استفاده کن
    
    def register(self, username, password):
        if username in self.users:
            raise ValueError("کاربر وجود دارد")
        hashed = bcrypt.hashpw(password.encode(), bcrypt.gensalt())
        self.users[username] = hashed
        return "ثبت‌نام موفق"
    
    def login(self, username, password):
        if username not in self.users:
            return None
        if bcrypt.checkpw(password.encode(), self.users[username]):
            payload = {
                "username": username,
                "exp": datetime.utcnow() + timedelta(hours=2),
                "iat": datetime.utcnow()
            }
            token = jwt.encode(payload, self.secret_key, algorithm="HS256")
            return token
        return None

auth = SecureAuth()
auth.register("sajjad", "MySuperSecurePass123!")
token = auth.login("sajjad", "MySuperSecurePass123!")
print("توکن امن:", token)