mirror of
https://github.com/openmax-server/server.git
synced 2026-07-13 07:20:13 +03:00
W.I.P.: OpenMAX 0.1.0 (#28)
* Различные фиксы (#14)
* Исключаем только ошибку валидации
* Небольшие правки вебсокета тамтама
* Теперь номера брутить чутка сложнее
* Авторизация теперь проверяется для некоторых команд
* Теперь проверяется orign у вебсокета тамтама
* Дополнил пример env
* Починил немного сокет тамтама
* Теперь все должно работать исправно
* Документация: описал про SESSION_INIT
* Update deviceId description in tamtam_ws.md
* feat: 16 опкод для обновления профиля
* feat: 23 опкод для регистрации, смс шлюз, докер
* SQLite: Ошибка
И вообще не юзайте sqlite пожалуйста, он баганный
* Серверный конфиг снова сломался
* Починил изменение профиля
* История сообщений (вроде как даже работает)
* Дополнил немного 48 опкод, и починил отдачу истории в избранном (избранное все равно не работает)
* Доделал авторизацию в ТамТаме
* Швырнул архитектуру, чтобы позже объединить контроллеры веба и сокета в одно, а также разделить процессоры
* Удалил tamtam.proto, поскольку нигде больше не используется
* Поделил процессоры в таме
* Коды стран
* Генерируем айди, за вместо того, чтобы писать его попорядку
* Вынес список участников чата в отдельную таблицу
* Теперь мы не должны отправлять 128 опкод самому себе. вроде
* ТамТам: реализация 32 опкода. Надеюсь, будет работать
* ТамТам: обработчик 5 и 1 опкода
* ТамТам: починил 32 опкод и добавил серверный конфиг. Избранное больше не добавляется в список чатов
MAX: починил 32 опкод
* TT: oops
* MAX: заглушка в 158 опкоде
* Common: поправил 1 прикол
* TamTam: websocket transport for web version
* TamTam && MAX: история (в мохе она вроде теперь получше работает)
* Правка контроллеров небольшая
* MAX: Разделил процессоры
* TG Bot: переписал на aiogram
* Common: фикс года
* Начальная реализация транспорта ws для max web и прочие улучшения
* Common: добавил сжатие lz4
* Исправление конфликтов
* TT: фикс вебсокета
* Fix OpenMAX mobile compatibility and Telegram auth fallback (#30)
* Fix OpenMAX mobile compatibility and Telegram auth fallback
* Common: Убрал скобку в конфиге
---------
Co-authored-by: Alexey Polyakov <starwear3000@mail.ru>
* MAX: заглушка для баннеров, правка пакета со списком жалоб, отдача контактов и прочие улучшения
* oops
* MAX: Рабочие баннеры
* MAX: обновление настроек приватности
* MAX: пуши через firebase (особо не тестил, вроде работает)
* MAX: Рефактор папок
* MAX: фикс версий < 25.8.0
* MAX: отдача черного списка
* Теперь ID пользователей идут по порядку, как и раньше
* MAX: создание папок
* MAX: oops
* MAX: добавление/удаление контактов, статусы
* MAX & TT: теперь в качестве страны локации используется настоящая страна пользователя, а также зафиксировал версии библиотек в зависимостях
* TT: ну вроде шире поддержка, а вообще обратная совместимость с максом клас
* oops
* TG Bot: возможность включения белого списка
* Решение проблемы обработки sigterm
* nothing
* fix
* TG Bot: fix
* TG Bot: fix username
* MAX && TT: общение в таме, и корректировки под веб морду
* MAX: Почистил серверный конфиг
* nothing
* MAX: web fix
* MAX: web fix 2
* oops
* nothing
* MAX: теперь для избранного не сравниваем айди с нулём
* Вынес sqlite в отдельный модуль
* MAX: блокировка и разблокировка контакта
* oops
* TT: тоже самое что в макс
* MAX && TT: обновление контакта
* MAX: добавление контакта по номеру телефона
* MAX: исправление уязвимости с избранными
* Генерируем айди пользователей рандомно (опять, да)
* update sql scheme
* Фикс сборки избранного
* Попытка починить историю (спойлер, нихуя не получилось)
* MAX: история таки заработала — cid/link/reactionInfo обязательны в схеме
Десктопный MAX подключается через TCP (mobile-протокол) и парсит
msgpack по фиксированной схеме. Если в сообщении выпадает любое из
полей — клиент молча обрывает соединение. После 87cfc19 как раз
такие условные `if elements: ...` / `if link: ...` (а link и
reaction_info там всегда были `{}`, то есть falsy) вырезали поля
из ответа CHAT_HISTORY и MSG_SEND, чем и сломали историю.
- src/common/tools.py: новый build_message_dict() — единая сборка
тела сообщения, где все поля (id, cid, time, type, sender, text,
attaches, elements, reactionInfo, link) присутствуют ВСЕГДА.
get_last_message переписан через него.
- src/oneme/processors/history.py: chat_history использует
build_message_dict вместо ручной логики с условными if-ками.
- src/oneme/processors/messages.py: msg_send.bodyMessage теперь
отдает cid / reactionInfo / link даже пустыми и приводит id
к int для mobile, str для web.
Цепная польза: auth.py LOGIN bootstrap (через generate_chats →
get_last_message) и search.py тоже теперь шлют корректную схему.
* MAX: bootstrap-история в LOGIN — клиент перестал думать что всё уже синканулось
В ответе LOGIN сервер слал messages: {} и chatMarker: 0. Десктопный
клиент в этом случае считает, что локальная история уже синхронизирована
со старого запуска, и НЕ отправляет CHAT_HISTORY (49) при открытии чата.
В окне видно только lastMessage из chats[], а вся реальная переписка —
ничерта.
- src/common/tools.py: collect_bootstrap_history(chatIds, ...) —
собирает карту {chatId: [последние N сообщений]}, в т.ч. избранное
под клиентским id = senderId ^ senderId.
- src/oneme/processors/auth.py: подсовываем эту карту в
payload.messages, chatMarker = текущее время вместо 0.
* fix(chat history): фикс
* fix(chat history): фикс
* fix(chat history): фикс
* fix(chat history): фикс
* fix(chat history): фикс
* fix(chat history): фикс
* fix(chat history): фикс
* MAX & TT: теперь полноценный чёрный список
* TT: регистрация через клиент
* Вроде как нормальные заглушки под ассеты
* maybe fixes
---------
Co-authored-by: WowInceptionGood <143893762+WowInceptionGood@users.noreply.github.com>
Co-authored-by: zavolo <me@kirillka.ru>
Co-authored-by: Aleksandr Kosachev <38947781+gohryt@users.noreply.github.com>
This commit is contained in:
Alexey Polyakov
@@ -0,0 +1,496 @@
|
||||
import asyncio
|
||||
import logging
|
||||
import time
|
||||
import traceback
|
||||
|
||||
from common.opcodes import Opcodes
|
||||
from common.proto_tcp import MobileProto
|
||||
from common.rate_limiter import RateLimiter
|
||||
from common.tools import Tools
|
||||
from oneme.processors import Processors
|
||||
|
||||
|
||||
class OnemeMobile:
|
||||
def __init__(
|
||||
self, host, port, ssl_context, db_pool, clients, send_event, telegram_bot
|
||||
):
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.ssl_context = ssl_context
|
||||
self.server = None
|
||||
self.logger = logging.getLogger(__name__)
|
||||
self.db_pool = db_pool
|
||||
self.clients = clients
|
||||
|
||||
self.proto = MobileProto()
|
||||
self.auth_required = Tools().auth_required
|
||||
self.processors = Processors(
|
||||
db_pool=db_pool,
|
||||
clients=clients,
|
||||
send_event=send_event,
|
||||
telegram_bot=telegram_bot,
|
||||
)
|
||||
self.opcodes = Opcodes()
|
||||
|
||||
# rate limiter anti ddos brute force protection
|
||||
self.auth_rate_limiter = RateLimiter(max_attempts=15, window_seconds=60)
|
||||
|
||||
self.read_timeout = 300 # Таймаут чтения из сокета (секунды)
|
||||
self.max_read_size = 65536 # Максимальный размер данных из сокета
|
||||
|
||||
async def handle_client(self, reader, writer):
|
||||
"""Функция для обработки подключений"""
|
||||
# IP-адрес клиента
|
||||
address = writer.get_extra_info("peername")
|
||||
self.logger.info(f"Работаю с клиентом {address[0]}:{address[1]}")
|
||||
|
||||
deviceType = None
|
||||
deviceName = None
|
||||
appVersion = None
|
||||
|
||||
userPhone = None
|
||||
userId = None
|
||||
hashedToken = None
|
||||
|
||||
try:
|
||||
while True:
|
||||
# Читаем новые данные из сокета с таймаутом
|
||||
try:
|
||||
data = await asyncio.wait_for(
|
||||
reader.read(self.max_read_size), timeout=self.read_timeout
|
||||
)
|
||||
except asyncio.TimeoutError:
|
||||
self.logger.info(
|
||||
f"Таймаут соединения для {address[0]}:{address[1]}"
|
||||
)
|
||||
break
|
||||
|
||||
# Если сокет закрыт - выходим из цикла
|
||||
if not data:
|
||||
break
|
||||
|
||||
if len(data) > self.max_read_size:
|
||||
self.logger.warning(
|
||||
f"Пакет от {address[0]}:{address[1]} превышает лимит ({len(data)} байт)"
|
||||
)
|
||||
break
|
||||
|
||||
# Распаковываем данные
|
||||
packet = self.proto.unpack_packet(data)
|
||||
|
||||
# Скип если пакет невалидный
|
||||
if packet is None:
|
||||
self.logger.warning(
|
||||
f"Невалидный пакет от {address[0]}:{address[1]}"
|
||||
)
|
||||
continue
|
||||
|
||||
opcode = packet.get("opcode")
|
||||
seq = packet.get("seq")
|
||||
payload = packet.get("payload")
|
||||
|
||||
match opcode:
|
||||
case self.opcodes.SESSION_INIT:
|
||||
deviceType, deviceName, appVersion = await self.processors.session_init(
|
||||
payload, seq, writer
|
||||
)
|
||||
case self.opcodes.AUTH_REQUEST:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(
|
||||
seq,
|
||||
self.opcodes.AUTH_REQUEST,
|
||||
self.processors.error_types.RATE_LIMITED,
|
||||
writer,
|
||||
)
|
||||
else:
|
||||
await self.processors.auth_request(payload, seq, writer)
|
||||
case self.opcodes.AUTH:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(
|
||||
seq,
|
||||
self.opcodes.AUTH,
|
||||
self.processors.error_types.RATE_LIMITED,
|
||||
writer,
|
||||
)
|
||||
else:
|
||||
await self.processors.auth(
|
||||
payload, seq, writer, deviceType, deviceName, appVersion, address[0]
|
||||
)
|
||||
case self.opcodes.AUTH_CONFIRM:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(
|
||||
seq,
|
||||
self.opcodes.AUTH_CONFIRM,
|
||||
self.processors.error_types.RATE_LIMITED,
|
||||
writer,
|
||||
)
|
||||
elif payload and payload.get("tokenType") == "REGISTER":
|
||||
await self.processors.auth_confirm(
|
||||
payload, seq, writer, deviceType, deviceName, appVersion, address[0]
|
||||
)
|
||||
else:
|
||||
self.logger.warning(
|
||||
f"AUTH_CONFIRM с неизвестным tokenType: {payload}"
|
||||
)
|
||||
case self.opcodes.LOGIN:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(
|
||||
seq,
|
||||
self.opcodes.LOGIN,
|
||||
self.processors.error_types.RATE_LIMITED,
|
||||
writer,
|
||||
)
|
||||
else:
|
||||
(
|
||||
userPhone,
|
||||
userId,
|
||||
hashedToken,
|
||||
) = await self.processors.login(payload, seq, writer, appVersion)
|
||||
|
||||
if userPhone:
|
||||
await self._finish_auth(
|
||||
writer, address, userPhone, userId
|
||||
)
|
||||
case self.opcodes.LOGOUT:
|
||||
await self.processors.logout(
|
||||
seq, writer, hashedToken=hashedToken
|
||||
)
|
||||
break
|
||||
case self.opcodes.PING:
|
||||
await self.processors.ping(payload, seq, writer, userId)
|
||||
case self.opcodes.LOG:
|
||||
await self.processors.log(payload, seq, writer)
|
||||
case self.opcodes.ASSETS_UPDATE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_update,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_GET:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_get,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_GET_BY_IDS:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_get_by_ids,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_ADD:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_add,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_REMOVE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_remove,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_MOVE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_move,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.ASSETS_LIST_MODIFY:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.assets_list_modify,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.VIDEO_CHAT_HISTORY:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.video_chat_history,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.MSG_SEND:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.msg_send,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
self.db_pool,
|
||||
)
|
||||
case self.opcodes.FOLDERS_GET:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.folders_get,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userPhone,
|
||||
)
|
||||
case self.opcodes.FOLDERS_UPDATE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.folders_update,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userPhone,
|
||||
)
|
||||
case self.opcodes.SESSIONS_INFO:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.sessions_info,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userPhone,
|
||||
hashedToken,
|
||||
)
|
||||
case self.opcodes.CHAT_INFO:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.chat_info,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CHAT_HISTORY:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.chat_history,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CONTACT_INFO_BY_PHONE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_info_by_phone,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.OK_TOKEN:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.ok_token, payload, seq, writer
|
||||
)
|
||||
case self.opcodes.MSG_TYPING:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.msg_typing,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CONTACT_INFO:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_info,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CONTACT_LIST:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_list,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.COMPLAIN_REASONS_GET:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.complain_reasons_get,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.PROFILE:
|
||||
await self.processors.profile(
|
||||
payload, seq, writer, userId=userId
|
||||
)
|
||||
case self.opcodes.CHAT_SUBSCRIBE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.chat_subscribe,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
)
|
||||
case self.opcodes.CONFIG:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.update_config,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userPhone,
|
||||
hashedToken,
|
||||
)
|
||||
case self.opcodes.CONTACT_UPDATE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_update,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CONTACT_ADD_BY_PHONE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_add_by_phone,
|
||||
payload,
|
||||
seq,
|
||||
writer,
|
||||
userId,
|
||||
)
|
||||
case self.opcodes.CONTACT_PRESENCE:
|
||||
await self.auth_required(
|
||||
userPhone,
|
||||
self.processors.contact_presence,
|
||||
payload,
|
||||
seq,
|
||||
writer
|
||||
)
|
||||
case _:
|
||||
self.logger.warning(f"Неизвестный опкод {opcode}")
|
||||
except Exception as e:
|
||||
self.logger.error(
|
||||
f"Произошла ошибка при работе с клиентом {address[0]}:{address[1]}: {e}"
|
||||
)
|
||||
traceback.print_exc()
|
||||
|
||||
# Удаляем клиента из словаря
|
||||
if userPhone:
|
||||
await self._end_session(userId, address[0], address[1])
|
||||
|
||||
writer.close()
|
||||
self.logger.info(
|
||||
f"Прекратил работать работать с клиентом {address[0]}:{address[1]}"
|
||||
)
|
||||
|
||||
async def _finish_auth(self, writer, addr, phone, id):
|
||||
"""Завершение открытия сессии"""
|
||||
# Ищем пользователя в словаре
|
||||
user = self.clients.get(id)
|
||||
|
||||
# Добавляем новое подключение в словарь
|
||||
if user:
|
||||
user["clients"].append(
|
||||
{"writer": writer, "ip": addr[0], "port": addr[1], "protocol": "oneme"}
|
||||
)
|
||||
else:
|
||||
self.clients[id] = {
|
||||
"phone": phone,
|
||||
"id": id,
|
||||
"status": 2,
|
||||
"last_seen": 0,
|
||||
"clients": [
|
||||
{
|
||||
"writer": writer,
|
||||
"ip": addr[0],
|
||||
"port": addr[1],
|
||||
"protocol": "oneme",
|
||||
}
|
||||
],
|
||||
}
|
||||
|
||||
await self._broadcast_presence(id, True)
|
||||
|
||||
async def _broadcast_presence(self, userId, online):
|
||||
now = int(time.time())
|
||||
now_ms = int(time.time() * 1000)
|
||||
|
||||
if online:
|
||||
presence_data = {"on": "ON", "seen": now, "status": 1}
|
||||
else:
|
||||
presence_data = {"seen": now}
|
||||
|
||||
async with self.db_pool.acquire() as conn:
|
||||
async with conn.cursor() as cursor:
|
||||
await cursor.execute(
|
||||
"SELECT owner_id FROM contacts WHERE contact_id = %s",
|
||||
(userId,)
|
||||
)
|
||||
contact_owners = await cursor.fetchall()
|
||||
|
||||
for row in contact_owners:
|
||||
owner_id = int(row.get("owner_id"))
|
||||
if owner_id in self.clients:
|
||||
await self.processors.event(
|
||||
owner_id,
|
||||
{
|
||||
"eventType": "presence",
|
||||
"userId": userId,
|
||||
"presence": presence_data,
|
||||
"time": now_ms,
|
||||
}
|
||||
)
|
||||
|
||||
async def _end_session(self, id, ip, port):
|
||||
"""Завершение сессии"""
|
||||
# Получаем пользователя в списке
|
||||
user = self.clients.get(id)
|
||||
if not user:
|
||||
return
|
||||
|
||||
# Получаем подключения пользователя
|
||||
clients = user.get("clients", [])
|
||||
|
||||
# Удаляем нужное подключение из словаря
|
||||
for i, client in enumerate(clients):
|
||||
if (client.get("ip"), client.get("port")) == (ip, port):
|
||||
clients.pop(i)
|
||||
|
||||
if not clients:
|
||||
now = int(time.time())
|
||||
user["status"] = 0
|
||||
user["last_seen"] = now
|
||||
|
||||
async with self.db_pool.acquire() as conn:
|
||||
async with conn.cursor() as cursor:
|
||||
await cursor.execute(
|
||||
"UPDATE users SET lastseen = %s WHERE id = %s",
|
||||
(str(now), id)
|
||||
)
|
||||
|
||||
await self._broadcast_presence(id, False)
|
||||
|
||||
async def start(self):
|
||||
"""Функция для запуска сервера"""
|
||||
self.server = await asyncio.start_server(
|
||||
self.handle_client, self.host, self.port, ssl=self.ssl_context
|
||||
)
|
||||
|
||||
self.logger.info(f"Сокет запущен на порту {self.port}")
|
||||
|
||||
try:
|
||||
async with self.server:
|
||||
await self.server.serve_forever()
|
||||
except asyncio.CancelledError:
|
||||
self.server.close()
|
||||
await self.server.wait_closed()
|
||||
Reference in New Issue
Block a user