fix(security): timing attack, rate limiter за прокси, хардкод ID в скриптах (20.09.2026)

This commit is contained in:
2026-09-20 17:49:58 +00:00
parent 67709cb6d2
commit e15a228f80
4 changed files with 87 additions and 23 deletions
+1 -1
View File
@@ -58,7 +58,7 @@ async def verify_telegram_initdata(init_data: str):
hashlib.sha256
).hexdigest()
if calculated_hash != received_hash:
if not hmac.compare_digest(calculated_hash, received_hash):
raise HTTPException(401, "Invalid initData signature")
# Извлекаем данные пользователя
+39 -11
View File
@@ -3,29 +3,57 @@
Очищает ВСЕ данные по играм, урокам, чат-учителю, достижениям.
Не трогает: users (остаётся), tts_cache (аудио).
Запуск: python3 full_reset.py
Запуск: python3 full_reset.py <USER_ID>
🔧 ФИКС (20.09.2026): убран хардкод USER_ID — теперь через аргумент + подтверждение.
Раньше: USER_ID = 6694989403 (мог случайно снести прогресс на проде).
Теперь: python3 full_reset.py 123456789 → запросит подтверждение "YES".
"""
import asyncio
import sys
import asyncpg
from config import DB_CONFIG
USER_ID = 6694989403
# ============ Проверка аргументов ============
if len(sys.argv) < 2:
print("❌ Использование: python3 full_reset.py <USER_ID>")
print(" Пример: python3 full_reset.py 6694989403")
sys.exit(1)
try:
USER_ID = int(sys.argv[1])
except ValueError:
print(f"❌ USER_ID должен быть числом, получено: {sys.argv[1]}")
sys.exit(1)
confirm = input(
f"⚠️ Удалить ВСЕ данные пользователя {USER_ID}? Напиши YES для подтверждения: "
)
if confirm != "YES":
print("❌ Отменено.")
sys.exit(0)
# ============ Основная логика ============
async def main():
pool = await asyncpg.create_pool(**DB_CONFIG)
async with pool.acquire() as conn:
# Проверяем пользователя
user_exists = await conn.fetchval("""
SELECT 1 FROM users WHERE telegram_id = $1
""", USER_ID)
if not user_exists:
print(f"❌ Пользователь {USER_ID} не найден!")
await pool.close()
return
# Список таблиц для очистки (по user_id)
user_tables = [
"user_progress", # Филворды
@@ -46,9 +74,9 @@ async def main():
"daily_logins", # Входы
"ai_user_context", # AI-контекст
]
total_deleted = 0
for table in user_tables:
try:
result = await conn.execute(f"""
@@ -59,14 +87,14 @@ async def main():
print(f"🗑️ {table}: удалено {affected}")
except Exception as e:
print(f"⚠️ {table}: {str(e)}")
# Сбрасываем last_completed_level
await conn.execute("""
UPDATE users SET last_completed_level = 0, current_level = 'A1'
UPDATE users SET last_completed_level = 0, current_level = 'A1'
WHERE telegram_id = $1
""", USER_ID)
print("✅ last_completed_level = 0, current_level = A1")
# Очищаем архив ответов
try:
result = await conn.execute("""
@@ -75,7 +103,7 @@ async def main():
print(f"🗑️ user_learning_answers_archive: удалено {result.split()[-1]}")
except Exception as e:
print(f"⚠️ archive: {str(e)}")
await pool.close()
print(f"\n🎉 Готово! Удалено {total_deleted} записей.")
print("Можно начинать тестирование с чистого листа!")
+9 -1
View File
@@ -128,8 +128,16 @@ async def rate_limit_middleware(request: Request, call_next):
if path in ["/", "/health"]:
return await call_next(request)
# 🔧 ФИКС (20.09.2026): за Cloudflare/nginx request.client.host — IP прокси.
# Берём реальный IP из X-Forwarded-For (первый адрес).
forwarded = request.headers.get("x-forwarded-for")
if forwarded:
client_ip = forwarded.split(",")[0].strip()
elif request.headers.get("cf-connecting-ip"):
client_ip = request.headers.get("cf-connecting-ip")
else:
client_ip = request.client.host if request.client else "unknown"
rate_key = f"{client_ip}:{path}"
rate_key = f"{client_ip}:{path}"
if not rate_limiter.is_allowed(rate_key, path):
retry_after = rate_limiter.get_retry_after(rate_key, path)
+38 -10
View File
@@ -3,30 +3,58 @@
Очищает: user_learning_answers, user_lesson_progress, user_topic_progress,
user_skill_mastery, learning_sessions, teacher_chat_history, teacher_ai_usage.
Запуск: python3 reset_my_progress.py
Запуск: python3 reset_my_progress.py <USER_ID>
🔧 ФИКС (20.09.2026): убран хардкод USER_ID — теперь через аргумент + подтверждение.
Раньше: USER_ID = 6694989403 (мог случайно снести прогресс на проде).
Теперь: python3 reset_my_progress.py 123456789 → запросит подтверждение "YES".
"""
import asyncio
import sys
import asyncpg
from config import DB_CONFIG
# ID пользователя для сброса (твой Telegram ID)
USER_ID = 6694989403
# ============ Проверка аргументов ============
if len(sys.argv) < 2:
print("❌ Использование: python3 reset_my_progress.py <USER_ID>")
print(" Пример: python3 reset_my_progress.py 6694989403")
sys.exit(1)
try:
USER_ID = int(sys.argv[1])
except ValueError:
print(f"❌ USER_ID должен быть числом, получено: {sys.argv[1]}")
sys.exit(1)
confirm = input(
f"⚠️ Сбросить прогресс «Учитель» и «Уроки» для пользователя {USER_ID}? "
f"Напиши YES для подтверждения: "
)
if confirm != "YES":
print("❌ Отменено.")
sys.exit(0)
# ============ Основная логика ============
async def main():
pool = await asyncpg.create_pool(**DB_CONFIG)
async with pool.acquire() as conn:
# Проверяем, что пользователь существует
user_exists = await conn.fetchval("""
SELECT 1 FROM users WHERE telegram_id = $1
""", USER_ID)
if not user_exists:
print(f"❌ Пользователь {USER_ID} не найден!")
await pool.close()
return
# Список таблиц для очистки
tables = [
"user_learning_answers",
@@ -37,26 +65,26 @@ async def main():
"teacher_chat_history",
"teacher_ai_usage",
]
for table in tables:
result = await conn.execute(f"""
DELETE FROM {table} WHERE user_id = $1
""", USER_ID)
affected = int(result.split()[-1])
print(f"🗑️ {table}: удалено {affected} записей")
# Также очищаем ai_user_context
result = await conn.execute("""
DELETE FROM ai_user_context WHERE user_id = $1
""", USER_ID)
print(f"🗑️ ai_user_context: удалено {result.split()[-1]} записей")
# Сбрасываем last_completed_level
await conn.execute("""
UPDATE users SET last_completed_level = 0 WHERE telegram_id = $1
""", USER_ID)
print("✅ last_completed_level сброшен на 0")
await pool.close()
print("\n🎉 Готово! Можно начинать заново!")