Compare commits
7 Commits
d5ea45cb96
...
11b2e2748d
| Author | SHA1 | Date |
|---|---|---|
|
|
11b2e2748d | |
|
|
6c05b5f1b5 | |
|
|
2dab853569 | |
|
|
c7eace4648 | |
|
|
09c1f8b6f6 | |
|
|
9c3038bc84 | |
|
|
91453d7173 |
|
|
@ -0,0 +1,156 @@
|
|||
class Opcodes:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
PING = 1
|
||||
DEBUG = 2
|
||||
RECONNECT = 3
|
||||
LOG = 5
|
||||
SESSION_INIT = 6
|
||||
PROFILE = 16
|
||||
AUTH_REQUEST = 17
|
||||
AUTH = 18
|
||||
LOGIN = 19
|
||||
LOGOUT = 20
|
||||
SYNC = 21
|
||||
CONFIG = 22
|
||||
AUTH_CONFIRM = 23
|
||||
AUTH_CREATE_TRACK = 112
|
||||
AUTH_CHECK_PASSWORD = 113
|
||||
AUTH_LOGIN_CHECK_PASSWORD = 115
|
||||
AUTH_LOGIN_PROFILE_DELETE = 116
|
||||
AUTH_LOGIN_RESTORE_PASSWORD = 101
|
||||
AUTH_VALIDATE_PASSWORD = 107
|
||||
AUTH_VALIDATE_HINT = 108
|
||||
AUTH_VERIFY_EMAIL = 109
|
||||
AUTH_CHECK_EMAIL = 110
|
||||
AUTH_SET_2FA = 111
|
||||
AUTH_2FA_DETAILS = 104
|
||||
ASSETS_GET = 26
|
||||
ASSETS_UPDATE = 27
|
||||
ASSETS_GET_BY_IDS = 28
|
||||
ASSETS_LIST_MODIFY = 261
|
||||
ASSETS_REMOVE = 259
|
||||
ASSETS_MOVE = 260
|
||||
ASSETS_ADD = 29
|
||||
PRESET_AVATARS = 25
|
||||
CONTACT_INFO = 32
|
||||
CONTACT_INFO_BY_PHONE = 46
|
||||
CONTACT_ADD = 33
|
||||
CONTACT_UPDATE = 34
|
||||
CONTACT_PRESENCE = 35
|
||||
CONTACT_LIST = 36
|
||||
CONTACT_SEARCH = 37
|
||||
CONTACT_MUTUAL = 38
|
||||
CONTACT_PHOTOS = 39
|
||||
CONTACT_SORT = 40
|
||||
CONTACT_VERIFY = 42
|
||||
REMOVE_CONTACT_PHOTO = 43
|
||||
CHAT_INFO = 48
|
||||
CHAT_HISTORY = 49
|
||||
CHAT_MARK = 50
|
||||
CHAT_MEDIA = 51
|
||||
CHAT_DELETE = 52
|
||||
CHATS_LIST = 53
|
||||
CHAT_CLEAR = 54
|
||||
CHAT_UPDATE = 55
|
||||
CHAT_CHECK_LINK = 56
|
||||
CHAT_JOIN = 57
|
||||
CHAT_LEAVE = 58
|
||||
CHAT_MEMBERS = 59
|
||||
PUBLIC_SEARCH = 60
|
||||
CHAT_PERSONAL_CONFIG = 61
|
||||
CHAT_CREATE = 63
|
||||
REACTIONS_SETTINGS_GET_BY_CHAT_ID = 258
|
||||
CHAT_REACTIONS_SETTINGS_SET = 257
|
||||
MSG_SEND = 64
|
||||
MSG_TYPING = 65
|
||||
MSG_DELETE = 66
|
||||
MSG_EDIT = 67
|
||||
MSG_DELETE_RANGE = 92
|
||||
MSG_REACTION = 178
|
||||
MSG_CANCEL_REACTION = 179
|
||||
MSG_GET_REACTIONS = 180
|
||||
MSG_GET_DETAILED_REACTIONS = 181
|
||||
CHAT_SEARCH = 68
|
||||
MSG_SHARE_PREVIEW = 70
|
||||
MSG_GET = 71
|
||||
MSG_SEARCH_TOUCH = 72
|
||||
MSG_SEARCH = 73
|
||||
MSG_GET_STAT = 74
|
||||
CHAT_SUBSCRIBE = 75
|
||||
VIDEO_CHAT_START = 76
|
||||
VIDEO_CHAT_START_ACTIVE = 78
|
||||
CHAT_MEMBERS_UPDATE = 77
|
||||
VIDEO_CHAT_HISTORY = 79
|
||||
PHOTO_UPLOAD = 80
|
||||
STICKER_UPLOAD = 81
|
||||
VIDEO_UPLOAD = 82
|
||||
VIDEO_PLAY = 83
|
||||
VIDEO_CHAT_CREATE_JOIN_LINK = 84
|
||||
CHAT_PIN_SET_VISIBILITY = 86
|
||||
FILE_UPLOAD = 87
|
||||
FILE_DOWNLOAD = 88
|
||||
LINK_INFO = 89
|
||||
SESSIONS_INFO = 96
|
||||
SESSIONS_CLOSE = 97
|
||||
PHONE_BIND_REQUEST = 98
|
||||
PHONE_BIND_CONFIRM = 99
|
||||
GET_INBOUND_CALLS = 103
|
||||
EXTERNAL_CALLBACK = 105
|
||||
OK_TOKEN = 158
|
||||
CHAT_COMPLAIN = 117
|
||||
MSG_SEND_CALLBACK = 118
|
||||
SUSPEND_BOT = 119
|
||||
LOCATION_STOP = 124
|
||||
GET_LAST_MENTIONS = 127
|
||||
STICKER_CREATE = 193
|
||||
STICKER_SUGGEST = 194
|
||||
VIDEO_CHAT_MEMBERS = 195
|
||||
NOTIF_MESSAGE = 128
|
||||
NOTIF_TYPING = 129
|
||||
NOTIF_MARK = 130
|
||||
NOTIF_CONTACT = 131
|
||||
NOTIF_PRESENCE = 132
|
||||
NOTIF_CONFIG = 134
|
||||
NOTIF_CHAT = 135
|
||||
NOTIF_ATTACH = 136
|
||||
NOTIF_CALL_START = 137
|
||||
NOTIF_CONTACT_SORT = 139
|
||||
NOTIF_MSG_DELETE_RANGE = 140
|
||||
NOTIF_MSG_DELETE = 142
|
||||
NOTIF_MSG_REACTIONS_CHANGED = 155
|
||||
NOTIF_MSG_YOU_REACTED = 156
|
||||
NOTIF_CALLBACK_ANSWER = 143
|
||||
CHAT_BOT_COMMANDS = 144
|
||||
BOT_INFO = 145
|
||||
NOTIF_LOCATION = 147
|
||||
NOTIF_LOCATION_REQUEST = 148
|
||||
NOTIF_ASSETS_UPDATE = 150
|
||||
NOTIF_DRAFT = 152
|
||||
NOTIF_DRAFT_DISCARD = 153
|
||||
DRAFT_SAVE = 176
|
||||
DRAFT_DISCARD = 177
|
||||
CHAT_HIDE = 196
|
||||
CHAT_SEARCH_COMMON_PARTICIPANTS = 198
|
||||
NOTIF_MSG_DELAYED = 154
|
||||
NOTIF_PROFILE = 159
|
||||
PROFILE_DELETE = 199
|
||||
PROFILE_DELETE_TIME = 200
|
||||
WEB_APP_INIT_DATA = 160
|
||||
COMPLAIN = 161
|
||||
COMPLAIN_REASONS_GET = 162
|
||||
FOLDERS_GET = 272
|
||||
FOLDERS_GET_BY_ID = 273
|
||||
FOLDERS_UPDATE = 274
|
||||
FOLDERS_REORDER = 275
|
||||
FOLDERS_DELETE = 276
|
||||
NOTIF_FOLDERS = 277
|
||||
|
||||
AUTH_QR_APPROVE = 290
|
||||
NOTIF_BANNERS = 292
|
||||
CHAT_SUGGEST = 300
|
||||
AUDIO_PLAY = 301
|
||||
SEND_VOTE = 304
|
||||
VOTERS_LIST_BY_ANSWER = 305
|
||||
GET_POLL_UPDATES = 306
|
||||
|
|
@ -1,6 +1,6 @@
|
|||
import lz4.block, msgpack, logging, json
|
||||
|
||||
class Proto:
|
||||
class MobileProto:
|
||||
def __init__(self) -> None:
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
|
|
@ -95,19 +95,3 @@ class Proto:
|
|||
CMD_NOF = 0x200
|
||||
CMD_ERR = 0x300
|
||||
PROTO_VER = 10
|
||||
|
||||
HELLO = 6
|
||||
REQUEST_CODE = 17
|
||||
VERIFY_CODE = 18
|
||||
FINAL_AUTH = 23
|
||||
LOGIN = 19
|
||||
PING = 1
|
||||
TELEMETRY = 5
|
||||
GET_ASSETS = 27
|
||||
GET_CALL_HISTORY = 79
|
||||
SEND_MESSAGE = 64
|
||||
GET_FOLDERS = 272
|
||||
GET_SESSIONS = 96
|
||||
LOGOUT = 20
|
||||
SEARCH_CHATS = 48
|
||||
SEARCH_BY_PHONE = 46
|
||||
|
|
@ -0,0 +1,48 @@
|
|||
import json
|
||||
|
||||
class WebProto:
|
||||
def pack_packet(self, ver=10, cmd=1, seq=0, opcode=1, payload=None):
|
||||
# а разве не надо в жсон запаковывать ещё
|
||||
# о всё
|
||||
return json.dumps({
|
||||
"ver": ver,
|
||||
"cmd": cmd,
|
||||
"seq": seq,
|
||||
"opcode": opcode,
|
||||
"payload": payload
|
||||
})
|
||||
|
||||
MAX_PACKET_SIZE = 65536 # 64 KB, заглушка, нужно узнать реальные лимиты и поменять, хотя кто будет это делать...
|
||||
|
||||
def unpack_packet(self, packet):
|
||||
# try catch чтобы не сыпалось всё при неверных пакетах
|
||||
if isinstance(packet, (str, bytes)) and len(packet) > self.MAX_PACKET_SIZE:
|
||||
return {}
|
||||
|
||||
try:
|
||||
parsed_packet = json.loads(packet)
|
||||
except (json.JSONDecodeError, TypeError, ValueError):
|
||||
return {}
|
||||
|
||||
return parsed_packet
|
||||
# мне кажется долго вручную всё писать
|
||||
# а как еще
|
||||
# ну вставить сюда целиком и потом через multiline cursor удалить лишнее
|
||||
# ну ты удалишь тогда. я на тачпаде
|
||||
# ладно щас другим способом удалю
|
||||
# всё нахуй
|
||||
# TAMTAM SOURCE LEAK 2026
|
||||
# так ну че делать будем
|
||||
# так ну
|
||||
|
||||
# 19 опкод сделан?
|
||||
# нет сэр пошли библиотеку тамы смотреть
|
||||
# мб найдем че. она без обфускации
|
||||
# а ты ее видишь?
|
||||
# пошли
|
||||
|
||||
### Констаты протокола
|
||||
CMD_OK = 1
|
||||
CMD_NOF = 2
|
||||
CMD_ERR = 3
|
||||
PROTO_VER = 10
|
||||
|
|
@ -79,12 +79,16 @@ class Tools:
|
|||
|
||||
return contact
|
||||
|
||||
def generate_chat(self, id, owner, type, participants, lastMessage, lastEventTime):
|
||||
def generate_chat(self, id, owner, type, participants, lastMessage, lastEventTime, prevMessageId=0):
|
||||
"""Генерация чата"""
|
||||
# Генерируем список участников
|
||||
result_participants = {
|
||||
str(participant): 0 for participant in participants
|
||||
}
|
||||
if isinstance(participants, dict):
|
||||
result_participants = {str(k): v for k, v in participants.items()}
|
||||
else:
|
||||
# assume list
|
||||
result_participants = {
|
||||
str(participant): 0 for participant in participants
|
||||
}
|
||||
|
||||
result = None
|
||||
|
||||
|
|
@ -101,6 +105,7 @@ class Tools:
|
|||
"lastDelayedUpdateTime": 0,
|
||||
"lastFireDelayedErrorTime": 0,
|
||||
"created": 1,
|
||||
"prevMessageId": prevMessageId,
|
||||
"joinTime": 1,
|
||||
"modified": lastEventTime
|
||||
}
|
||||
|
|
@ -127,11 +132,14 @@ class Tools:
|
|||
chatId, db_pool
|
||||
)
|
||||
|
||||
# Формируем список участников
|
||||
participants = {
|
||||
str(participant): 0 for participant in row.get("participants")
|
||||
}
|
||||
# Формируем список участников с временем последней активности
|
||||
participant_ids = json.loads(row.get("participants"))
|
||||
participants = await self.get_participant_last_activity(
|
||||
chatId, participant_ids, db_pool
|
||||
)
|
||||
|
||||
# Получаем ID предыдущего сообщения
|
||||
prevMessageId = await self.get_previous_message_id(chatId, db_pool)
|
||||
# Выносим результат в лист
|
||||
chats.append(
|
||||
self.generate_chat(
|
||||
|
|
@ -140,7 +148,8 @@ class Tools:
|
|||
row.get("type"),
|
||||
participants,
|
||||
message,
|
||||
messageTime
|
||||
messageTime,
|
||||
prevMessageId
|
||||
)
|
||||
)
|
||||
|
||||
|
|
@ -152,15 +161,23 @@ class Tools:
|
|||
# ID избранного
|
||||
chatId = senderId ^ senderId
|
||||
|
||||
# Получаем последнюю активность участника (отправителя) в избранном
|
||||
participants = await self.get_participant_last_activity(
|
||||
senderId, [senderId], db_pool
|
||||
)
|
||||
|
||||
# Получаем ID предыдущего сообщения для избранного (чат ID = senderId)
|
||||
prevMessageId = await self.get_previous_message_id(senderId, db_pool)
|
||||
# Хардкодим в лист чатов избранное
|
||||
chats.append(
|
||||
self.generate_chat(
|
||||
chatId,
|
||||
senderId,
|
||||
"DIALOG",
|
||||
[senderId],
|
||||
participants,
|
||||
message,
|
||||
messageTime
|
||||
messageTime,
|
||||
prevMessageId
|
||||
)
|
||||
)
|
||||
|
||||
|
|
@ -216,6 +233,54 @@ class Tools:
|
|||
# Возвращаем
|
||||
return message, int(row.get("time"))
|
||||
|
||||
async def get_previous_message_id(self, chatId, db_pool):
|
||||
"""Получение ID предыдущего сообщения (второго с конца) в чате."""
|
||||
async with db_pool.acquire() as db_connection:
|
||||
async with db_connection.cursor() as cursor:
|
||||
await cursor.execute(
|
||||
"SELECT id FROM `messages` WHERE chat_id = %s ORDER BY time DESC LIMIT 1 OFFSET 1",
|
||||
(chatId,)
|
||||
)
|
||||
row = await cursor.fetchone()
|
||||
|
||||
# Если результат есть, возвращаем его
|
||||
if row:
|
||||
return int(row.get("id"))
|
||||
|
||||
# В ином случае возвращаем 0
|
||||
return 0
|
||||
|
||||
async def get_participant_last_activity(self, chatId, participant_ids, db_pool):
|
||||
"""Возвращает словарь {participant_id: last_activity_time} для участников чата."""
|
||||
if not participant_ids:
|
||||
return {}
|
||||
|
||||
async with db_pool.acquire() as db_connection:
|
||||
async with db_connection.cursor() as cursor:
|
||||
# Собираем всех участников
|
||||
placeholders = ','.join(['%s'] * len(participant_ids))
|
||||
query = f"""
|
||||
SELECT sender, MAX(time) as last_time
|
||||
FROM messages
|
||||
WHERE chat_id = %s AND sender IN ({placeholders})
|
||||
GROUP BY sender
|
||||
"""
|
||||
params = (chatId,) + tuple(participant_ids)
|
||||
await cursor.execute(query, params)
|
||||
rows = await cursor.fetchall()
|
||||
|
||||
# Собираем список участников без времени последней активности в чате
|
||||
result = {str(pid): 0 for pid in participant_ids}
|
||||
|
||||
# Обновляем для каждого участника время последней активности в чате
|
||||
for row in rows:
|
||||
sender = str(row["sender"])
|
||||
last_time = row["last_time"]
|
||||
if last_time is not None:
|
||||
result[sender] = int(last_time)
|
||||
|
||||
return result
|
||||
|
||||
async def auth_required(self, userPhone, coro, *args):
|
||||
if userPhone:
|
||||
await coro(*args)
|
||||
|
|
|
|||
|
|
@ -1,10 +1,9 @@
|
|||
# Импортирование библиотек
|
||||
import ssl, logging, asyncio
|
||||
from common.config import ServerConfig
|
||||
from oneme_tcp.controller import OnemeMobileController
|
||||
from oneme.controller import OnemeMobileController
|
||||
from telegrambot.controller import TelegramBotController
|
||||
from tamtam_tcp.controller import TTMobileController
|
||||
from tamtam_ws.controller import TTWSController
|
||||
from tamtam.controller import TTMobileController
|
||||
# Конфиг сервера
|
||||
server_config = ServerConfig()
|
||||
|
||||
|
|
@ -75,7 +74,6 @@ async def main():
|
|||
controllers = {
|
||||
"oneme_mobile": OnemeMobileController(),
|
||||
"tamtam_mobile": TTMobileController(),
|
||||
"tamtam_ws": TTWSController(),
|
||||
"telegrambot": TelegramBotController()
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,7 @@
|
|||
class OnemeConfig:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
SERVER_CONFIG = {
|
||||
|
||||
}
|
||||
|
|
@ -1,13 +1,16 @@
|
|||
import asyncio
|
||||
from oneme_tcp.server import OnemeMobileServer
|
||||
from oneme_tcp.proto import Proto
|
||||
from oneme.socket import OnemeMobileServer
|
||||
from common.proto_tcp import MobileProto
|
||||
from common.proto_web import WebProto
|
||||
from classes.controllerbase import ControllerBase
|
||||
from common.config import ServerConfig
|
||||
from common.opcodes import Opcodes
|
||||
|
||||
class OnemeMobileController(ControllerBase):
|
||||
def __init__(self):
|
||||
self.config = ServerConfig()
|
||||
self.proto = Proto()
|
||||
self.proto = MobileProto()
|
||||
self.opcodes = Opcodes()
|
||||
|
||||
async def event(self, target, client, eventData):
|
||||
# Извлекаем тип события и врайтер
|
||||
|
|
@ -34,7 +37,7 @@ class OnemeMobileController(ControllerBase):
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=0, seq=1, opcode=self.proto.NOTIF_MESSAGE, payload=payload
|
||||
cmd=0, seq=1, opcode=self.opcodes.NOTIF_MESSAGE, payload=payload
|
||||
)
|
||||
elif eventType == "typing":
|
||||
# Данные события
|
||||
|
|
@ -51,7 +54,20 @@ class OnemeMobileController(ControllerBase):
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=0, seq=1, opcode=self.proto.NOTIF_TYPING, payload=payload
|
||||
cmd=0, seq=1, opcode=self.opcodes.NOTIF_TYPING, payload=payload
|
||||
)
|
||||
elif eventType == "profile_updated":
|
||||
# Данные события
|
||||
profile = eventData.get("profile")
|
||||
|
||||
# Данные пакета
|
||||
payload = {
|
||||
"profile": profile
|
||||
}
|
||||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=0, seq=1, opcode=self.opcodes.NOTIF_PROFILE, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем пакет
|
||||
|
|
@ -122,4 +122,8 @@ class AuthConfirmRegisterPayloadModel(pydantic.BaseModel):
|
|||
v = v.strip()
|
||||
if len(v) > 59:
|
||||
raise ValueError('lastName too long')
|
||||
return v
|
||||
return v
|
||||
|
||||
class ChatHistoryPayloadModel(pydantic.BaseModel):
|
||||
chatId: int
|
||||
backward: int
|
||||
|
|
@ -1,18 +1,24 @@
|
|||
import json, secrets, hashlib, time, logging
|
||||
from oneme_tcp.models import *
|
||||
from oneme_tcp.proto import Proto
|
||||
from oneme_tcp.config import OnemeConfig
|
||||
import json
|
||||
import secrets
|
||||
import hashlib
|
||||
import time
|
||||
import logging
|
||||
from oneme.models import *
|
||||
from common.proto_tcp import MobileProto
|
||||
from common.proto_web import WebProto
|
||||
from common.opcodes import Opcodes
|
||||
from oneme.config import OnemeConfig
|
||||
from common.tools import Tools
|
||||
from common.config import ServerConfig
|
||||
from common.static import Static
|
||||
from common.sms import send_sms_code
|
||||
|
||||
class Processors:
|
||||
def __init__(self, db_pool=None, clients={}, send_event=None, telegram_bot=None):
|
||||
self.proto = Proto()
|
||||
def __init__(self, db_pool=None, clients={}, send_event=None, telegram_bot=None, type="socket"):
|
||||
self.tools = Tools()
|
||||
self.config = ServerConfig()
|
||||
self.static = Static()
|
||||
self.opcodes = Opcodes()
|
||||
self.server_config = OnemeConfig().SERVER_CONFIG
|
||||
self.error_types = self.static.ErrorTypes()
|
||||
self.chat_types = self.static.ChatTypes()
|
||||
|
|
@ -23,6 +29,11 @@ class Processors:
|
|||
self.telegram_bot = telegram_bot
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
if type == "socket":
|
||||
self.proto = MobileProto()
|
||||
elif type == "web":
|
||||
self.proto = WebProto()
|
||||
|
||||
async def _send(self, writer, packet):
|
||||
try:
|
||||
writer.write(packet)
|
||||
|
|
@ -44,14 +55,14 @@ class Processors:
|
|||
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_hello(self, payload, seq, writer):
|
||||
async def session_init(self, payload, seq, writer):
|
||||
"""Обработчик приветствия"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
HelloPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.SESSION_INIT, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.SESSION_INIT, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return None, None
|
||||
|
||||
# Получаем данные из пакета
|
||||
|
|
@ -76,50 +87,50 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.SESSION_INIT, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.SESSION_INIT, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
return deviceType, deviceName
|
||||
|
||||
async def process_ping(self, payload, seq, writer):
|
||||
async def ping(self, payload, seq, writer):
|
||||
"""Обработчик пинга"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
PingPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.PING, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.PING, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Собираем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.PING, payload=None
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.PING, payload=None
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_telemetry(self, payload, seq, writer):
|
||||
async def log(self, payload, seq, writer):
|
||||
"""Обработчик телеметрии"""
|
||||
# TODO: можно было бы реализовать валидацию телеметрии, но сейчас это не особо важно
|
||||
|
||||
# Собираем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.LOG, payload=None
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.LOG, payload=None
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_request_code(self, payload, seq, writer):
|
||||
async def auth_request(self, payload, seq, writer):
|
||||
"""Обработчик запроса кода"""
|
||||
try:
|
||||
RequestCodePayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.AUTH_REQUEST, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_REQUEST, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем телефон из пакета
|
||||
|
|
@ -143,9 +154,9 @@ class Processors:
|
|||
# Получаем код через SMS шлюз или генерируем локально (безопасность прежде всего)
|
||||
if self.config.sms_gateway_url:
|
||||
code = await send_sms_code(self.config.sms_gateway_url, phone)
|
||||
|
||||
if code is None:
|
||||
await self._send_error(seq, self.proto.AUTH_REQUEST, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
code = str(secrets.randbelow(900000) + 100000)
|
||||
else:
|
||||
code = str(secrets.randbelow(900000) + 100000)
|
||||
|
||||
|
|
@ -185,20 +196,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.AUTH_REQUEST, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH_REQUEST, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
self.logger.debug(f"Код для {phone}: {code} (существующий={user_exists})")
|
||||
|
||||
async def process_verify_code(self, payload, seq, writer, deviceType, deviceName):
|
||||
async def auth(self, payload, seq, writer, deviceType, deviceName):
|
||||
"""Обработчик проверки кода"""
|
||||
try:
|
||||
VerifyCodePayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.AUTH, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -225,12 +236,12 @@ class Processors:
|
|||
|
||||
# Если токен просрочен, или его нет - отправляем ошибку
|
||||
if stored_token is None:
|
||||
await self._send_error(seq, self.proto.AUTH, self.error_types.CODE_EXPIRED, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.CODE_EXPIRED, writer)
|
||||
return
|
||||
|
||||
# Проверяем код
|
||||
if stored_token.get("code_hash") != hashed_code:
|
||||
await self._send_error(seq, self.proto.AUTH, self.error_types.INVALID_CODE, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.INVALID_CODE, writer)
|
||||
return
|
||||
|
||||
# Если это новый пользователь - переводим токен в state='verified'
|
||||
|
|
@ -241,7 +252,7 @@ class Processors:
|
|||
("verified", hashed_token,)
|
||||
)
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.AUTH,
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH,
|
||||
payload={
|
||||
"tokenAttrs": {
|
||||
"REGISTER": {
|
||||
|
|
@ -299,20 +310,20 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.AUTH, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_auth_confirm(self, payload, seq, writer, deviceType, deviceName):
|
||||
async def auth_confirm(self, payload, seq, writer, deviceType, deviceName):
|
||||
"""Обработчик подтверждения регистрации нового пользователя"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
AuthConfirmRegisterPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.AUTH_CONFIRM, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -338,7 +349,7 @@ class Processors:
|
|||
|
||||
# Если токен не найден или просрочен - отправляем ошибку
|
||||
if stored_token is None:
|
||||
await self._send_error(seq, self.proto.AUTH_CONFIRM, self.error_types.CODE_EXPIRED, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.CODE_EXPIRED, writer)
|
||||
return
|
||||
|
||||
phone = stored_token.get("phone")
|
||||
|
|
@ -346,7 +357,7 @@ class Processors:
|
|||
# Проверяем что пользователь с таким телефоном ещё не существует
|
||||
await cursor.execute("SELECT id FROM users WHERE phone = %s", (phone,))
|
||||
if await cursor.fetchone():
|
||||
await self._send_error(seq, self.proto.AUTH_CONFIRM, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
now_ms = int(time.time() * 1000)
|
||||
|
|
@ -421,21 +432,21 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.AUTH_CONFIRM, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH_CONFIRM, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
self.logger.info(f"Новый пользователь зарегистрирован: phone={phone} id={user_id} name={first_name} {last_name}")
|
||||
|
||||
async def process_login(self, payload, seq, writer):
|
||||
async def login(self, payload, seq, writer):
|
||||
"""Обработчик авторизации клиента на сервере"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
LoginPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.LOGIN, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.LOGIN, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Получаем данные из пакета
|
||||
|
|
@ -452,7 +463,7 @@ class Processors:
|
|||
|
||||
# Если токен не найден, отправляем ошибку
|
||||
if token_data is None:
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.INVALID_TOKEN, writer)
|
||||
await self._send_error(seq, self.opcodes.LOGIN, self.error_types.INVALID_TOKEN, writer)
|
||||
return
|
||||
|
||||
# Ищем аккаунт пользователя в бд
|
||||
|
|
@ -509,14 +520,14 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.LOGIN, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.LOGIN, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
return int(user.get("phone")), int(user.get("id")), hashed_token
|
||||
|
||||
async def process_logout(self, seq, writer, hashedToken):
|
||||
async def logout(self, seq, writer, hashedToken):
|
||||
"""Обработчик завершения сессии"""
|
||||
# Удаляем токен из бд
|
||||
async with self.db_pool.acquire() as conn:
|
||||
|
|
@ -525,20 +536,20 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.LOGOUT, payload=None
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.LOGOUT, payload=None
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_get_assets(self, payload, seq, writer):
|
||||
async def assets_update(self, payload, seq, writer):
|
||||
"""Обработчик запроса ассетов клиента на сервере"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
AssetsPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.ASSETS_UPDATE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.ASSETS_UPDATE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# TODO: сейчас это заглушка, а попозже нужно сделать полноценную реализацию
|
||||
|
|
@ -551,20 +562,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.ASSETS_UPDATE, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.ASSETS_UPDATE, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_get_call_history(self, payload, seq, writer):
|
||||
async def video_chat_history(self, payload, seq, writer):
|
||||
"""Обработчик получения истории звонков"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
GetCallHistoryPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.VIDEO_CHAT_HISTORY, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.VIDEO_CHAT_HISTORY, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# TODO: сейчас это заглушка, а попозже нужно сделать полноценную реализацию
|
||||
|
|
@ -579,20 +590,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.VIDEO_CHAT_HISTORY, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.VIDEO_CHAT_HISTORY, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_send_message(self, payload, seq, writer, senderId, db_pool):
|
||||
async def msg_send(self, payload, seq, writer, senderId, db_pool):
|
||||
"""Функция отправки сообщения"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
SendMessagePayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.MSG_SEND, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_SEND, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -629,7 +640,7 @@ class Processors:
|
|||
|
||||
# Если нет такого чата - выбрасываем ошибку
|
||||
if not chat:
|
||||
await self._send_error(seq, self.proto.MSG_SEND, self.error_types.CHAT_NOT_FOUND, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_SEND, self.error_types.CHAT_NOT_FOUND, writer)
|
||||
return
|
||||
|
||||
# Список участников
|
||||
|
|
@ -637,7 +648,7 @@ class Processors:
|
|||
|
||||
# Проверяем, является ли отправитель участником чата
|
||||
if int(senderId) not in participants:
|
||||
await self._send_error(seq, self.proto.MSG_SEND, self.error_types.CHAT_NOT_ACCESS, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_SEND, self.error_types.CHAT_NOT_ACCESS, writer)
|
||||
return
|
||||
|
||||
# Добавляем сообщение в историю
|
||||
|
|
@ -687,20 +698,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.MSG_SEND, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.MSG_SEND, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_get_folders(self, payload, seq, writer, senderPhone):
|
||||
async def folders_get(self, payload, seq, writer, senderPhone):
|
||||
"""Синхронизация папок с сервером"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
SyncFoldersPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.FOLDERS_GET, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.FOLDERS_GET, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Ищем папки в бд
|
||||
|
|
@ -720,13 +731,13 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.FOLDERS_GET, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.FOLDERS_GET, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_get_sessions(self, payload, seq, writer, senderPhone, hashedToken):
|
||||
async def sessions_info(self, payload, seq, writer, senderPhone, hashedToken):
|
||||
"""Получение активных сессий на аккаунте"""
|
||||
# Готовый список сессий
|
||||
sessions = []
|
||||
|
|
@ -756,20 +767,20 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.SESSIONS_INFO, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.SESSIONS_INFO, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_search_users(self, payload, seq, writer):
|
||||
async def contact_info(self, payload, seq, writer):
|
||||
"""Поиск пользователей по ID"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
SearchUsersPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.CONTACT_INFO, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.CONTACT_INFO, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Итоговый список пользователей
|
||||
|
|
@ -818,20 +829,20 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
response = self.proto.pack_packet(
|
||||
seq=seq, opcode=self.proto.CONTACT_INFO, payload=payload
|
||||
seq=seq, opcode=self.opcodes.CONTACT_INFO, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_search_chats(self, payload, seq, writer, senderId):
|
||||
async def chat_info(self, payload, seq, writer, senderId):
|
||||
"""Поиск чатов по ID"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
SearchChatsPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.CHAT_INFO, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.CHAT_INFO, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Итоговый список чатов
|
||||
|
|
@ -894,20 +905,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.CHAT_INFO, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.CHAT_INFO, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_search_by_phone(self, payload, seq, writer, senderId):
|
||||
async def contact_info_by_phone(self, payload, seq, writer, senderId):
|
||||
"""Поиск по номеру телефона"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
SearchByPhonePayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.CONTACT_INFO_BY_PHONE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.CONTACT_INFO_BY_PHONE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Ищем пользователя в бд
|
||||
|
|
@ -918,7 +929,7 @@ class Processors:
|
|||
|
||||
# Если пользователь не найден, отправляем ошибку
|
||||
if not user:
|
||||
await self._send_error(seq, self.proto.CONTACT_INFO_BY_PHONE, self.error_types.USER_NOT_FOUND, writer)
|
||||
await self._send_error(seq, self.opcodes.CONTACT_INFO_BY_PHONE, self.error_types.USER_NOT_FOUND, writer)
|
||||
return
|
||||
|
||||
# ID чата
|
||||
|
|
@ -964,34 +975,34 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
response = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.CONTACT_INFO_BY_PHONE, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.CONTACT_INFO_BY_PHONE, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, response)
|
||||
|
||||
async def process_get_call_token(self, payload, seq, writer):
|
||||
async def ok_token(self, payload, seq, writer):
|
||||
"""Получение токена для звонка"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
GetCallTokenPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.OK_TOKEN, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.OK_TOKEN, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# TODO: когда-то взяться за звонки
|
||||
|
||||
await self._send_error(seq, self.proto.OK_TOKEN, self.error_types.NOT_IMPLEMENTED, writer)
|
||||
await self._send_error(seq, self.opcodes.OK_TOKEN, self.error_types.NOT_IMPLEMENTED, writer)
|
||||
|
||||
async def process_typing(self, payload, seq, writer, senderId):
|
||||
async def msg_typing(self, payload, seq, writer, senderId):
|
||||
"""Обработчик события печатания"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
TypingPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.MSG_TYPING, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_TYPING, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -1006,7 +1017,7 @@ class Processors:
|
|||
|
||||
# Если чат не найден, отправляем ошибку
|
||||
if not chat:
|
||||
await self._send_error(seq, self.proto.MSG_TYPING, self.error_types.CHAT_NOT_FOUND, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_TYPING, self.error_types.CHAT_NOT_FOUND, writer)
|
||||
return
|
||||
|
||||
# Участники чата
|
||||
|
|
@ -1014,7 +1025,7 @@ class Processors:
|
|||
|
||||
# Проверяем, является ли отправитель участником чата
|
||||
if int(senderId) not in participants:
|
||||
await self._send_error(seq, self.proto.MSG_TYPING, self.error_types.CHAT_NOT_ACCESS, writer)
|
||||
await self._send_error(seq, self.opcodes.MSG_TYPING, self.error_types.CHAT_NOT_ACCESS, writer)
|
||||
return
|
||||
|
||||
# Рассылаем событие участникам чата
|
||||
|
|
@ -1033,20 +1044,20 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
seq=seq, opcode=self.proto.MSG_TYPING
|
||||
seq=seq, opcode=self.opcodes.MSG_TYPING
|
||||
)
|
||||
|
||||
# Отправляем пакет
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_complain_reasons_get(self, payload, seq, writer):
|
||||
async def complain_reasons_get(self, payload, seq, writer):
|
||||
"""Обработчик получения причин жалоб"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
ComplainReasonsGetPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.COMPLAIN_REASONS_GET, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.COMPLAIN_REASONS_GET, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Собираем данные пакета
|
||||
|
|
@ -1057,18 +1068,116 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
seq=seq, opcode=self.proto.COMPLAIN_REASONS_GET, payload=payload
|
||||
seq=seq, opcode=self.opcodes.COMPLAIN_REASONS_GET, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем пакет
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_update_profile(self, payload, seq, writer, userId, userPhone):
|
||||
async def chat_history(self, payload, seq, writer, senderId):
|
||||
"""Обработчик получения истории чата"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
ChatHistoryPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.opcodes.CHAT_HISTORY, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
chatId = payload.get("chatId")
|
||||
forward = payload.get("forward", 0)
|
||||
backward = payload.get("backward", 0)
|
||||
from_time = payload.get("from", 0)
|
||||
getMessages = payload.get("getMessages", True)
|
||||
messages = []
|
||||
|
||||
# Если пользователь хочет получить историю из избранного,
|
||||
# то выставляем в качестве ID чата его ID
|
||||
if chatId == 0:
|
||||
chatId = senderId
|
||||
|
||||
# Проверяем, существует ли чат
|
||||
async with self.db_pool.acquire() as conn:
|
||||
async with conn.cursor() as cursor:
|
||||
# Проверяем состоит ли пользователь в чате,
|
||||
# только в случае того, если это не избранное
|
||||
if chatId != senderId:
|
||||
await cursor.execute("SELECT * FROM chats WHERE id = %s", (chatId,))
|
||||
chat = await cursor.fetchone()
|
||||
|
||||
# Выбрасываем ошибку, если чата нет
|
||||
if not chat:
|
||||
await self._send_error(seq, self.opcodes.CHAT_HISTORY, self.error_types.CHAT_NOT_FOUND, writer)
|
||||
return
|
||||
|
||||
# Проверяем, является ли пользователь участником чата
|
||||
participants = json.loads(chat.get("participants"))
|
||||
if int(senderId) not in participants:
|
||||
await self._send_error(seq, self.opcodes.CHAT_HISTORY, self.error_types.CHAT_NOT_ACCESS, writer)
|
||||
return
|
||||
|
||||
# Если запрошены сообщения
|
||||
if getMessages:
|
||||
if backward > 0:
|
||||
await cursor.execute(
|
||||
"SELECT * FROM messages WHERE chat_id = %s AND time < %s ORDER BY id DESC LIMIT %s",
|
||||
(chatId, from_time, backward)
|
||||
)
|
||||
|
||||
result = await cursor.fetchall()
|
||||
|
||||
for row in result:
|
||||
messages.append({
|
||||
"id": row.get("id"),
|
||||
"time": int(row.get("time")),
|
||||
"type": row.get("type"),
|
||||
"sender": row.get("sender"),
|
||||
"text": row.get("text"),
|
||||
"attaches": json.loads(row.get("attaches")),
|
||||
"elements": json.loads(row.get("elements")),
|
||||
"reactionInfo": {}
|
||||
})
|
||||
|
||||
if forward > 0:
|
||||
await cursor.execute(
|
||||
"SELECT * FROM messages WHERE chat_id = %s AND time > %s ORDER BY id ASC LIMIT %s",
|
||||
(chatId, from_time, forward)
|
||||
)
|
||||
|
||||
result = await cursor.fetchall()
|
||||
|
||||
for row in result:
|
||||
messages.append({
|
||||
"id": row.get("id"),
|
||||
"time": int(row.get("time")),
|
||||
"type": row.get("type"),
|
||||
"sender": row.get("sender"),
|
||||
"text": row.get("text"),
|
||||
"attaches": json.loads(row.get("attaches")),
|
||||
"elements": json.loads(row.get("elements")),
|
||||
"reactionInfo": {}
|
||||
})
|
||||
|
||||
# Формируем ответ
|
||||
payload = {
|
||||
"messages": messages
|
||||
}
|
||||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.CHAT_HISTORY, payload=payload
|
||||
)
|
||||
|
||||
# Отправялем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def profile(self, payload, seq, writer, userId, userPhone):
|
||||
# Валидируем входные данные
|
||||
try:
|
||||
UpdateProfilePayloadModel.model_validate(payload)
|
||||
except Exception as e:
|
||||
await self._send_error(seq, self.proto.PROFILE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.PROFILE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем поля из пакета (каждое может быть None)
|
||||
|
|
@ -1121,14 +1230,22 @@ class Processors:
|
|||
username=user.get("username")
|
||||
)
|
||||
|
||||
# Данные пакета
|
||||
payload = {
|
||||
"profile": profile
|
||||
}
|
||||
|
||||
# Отправляем ответ на запрос (CMD_OK)
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.PROFILE, payload=profile
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.PROFILE, payload=payload
|
||||
)
|
||||
await self._send(writer, packet)
|
||||
|
||||
# Отправляем уведомление об изменении профиля (CMD_NOF)
|
||||
notif_packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_NOF, seq=0, opcode=self.proto.NOTIF_PROFILE, payload=profile
|
||||
)
|
||||
await self._send(writer, notif_packet)
|
||||
# Отправляем всем сессиям о изменении профиля
|
||||
await self.event(
|
||||
user.get('id'),
|
||||
{
|
||||
"eventType": "profile_updated",
|
||||
"profile": profile
|
||||
}
|
||||
)
|
||||
|
|
@ -1,8 +1,9 @@
|
|||
import asyncio, logging, traceback
|
||||
from oneme_tcp.proto import Proto
|
||||
from oneme_tcp.processors import Processors
|
||||
from common.proto_tcp import MobileProto
|
||||
from oneme.processors import Processors
|
||||
from common.rate_limiter import RateLimiter
|
||||
from common.tools import Tools
|
||||
from common.opcodes import Opcodes
|
||||
|
||||
class OnemeMobileServer:
|
||||
def __init__(self, host="0.0.0.0", port=443, ssl_context=None, db_pool=None, clients={}, send_event=None, telegram_bot=None):
|
||||
|
|
@ -14,9 +15,10 @@ class OnemeMobileServer:
|
|||
self.db_pool = db_pool
|
||||
self.clients = clients
|
||||
|
||||
self.proto = Proto()
|
||||
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=5, window_seconds=60)
|
||||
|
|
@ -71,86 +73,90 @@ class OnemeMobileServer:
|
|||
payload = packet.get("payload")
|
||||
|
||||
match opcode:
|
||||
case self.proto.SESSION_INIT:
|
||||
deviceType, deviceName = await self.processors.process_hello(payload, seq, writer)
|
||||
case self.proto.AUTH_REQUEST:
|
||||
case self.opcodes.SESSION_INIT:
|
||||
deviceType, deviceName = 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.proto.AUTH_REQUEST, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.AUTH_REQUEST, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
await self.processors.process_request_code(payload, seq, writer)
|
||||
case self.proto.AUTH:
|
||||
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.proto.AUTH, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.AUTH, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
await self.processors.process_verify_code(payload, seq, writer, deviceType, deviceName)
|
||||
case self.proto.AUTH_CONFIRM:
|
||||
await self.processors.auth(payload, seq, writer, deviceType, deviceName)
|
||||
case self.opcodes.AUTH_CONFIRM:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(seq, self.proto.AUTH_CONFIRM, self.processors.error_types.RATE_LIMITED, writer)
|
||||
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.process_auth_confirm(payload, seq, writer, deviceType, deviceName)
|
||||
await self.processors.auth_confirm(payload, seq, writer, deviceType, deviceName)
|
||||
else:
|
||||
self.logger.warning(f"AUTH_CONFIRM с неизвестным tokenType: {payload}")
|
||||
case self.proto.LOGIN:
|
||||
case self.opcodes.LOGIN:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(seq, self.proto.LOGIN, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.LOGIN, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
userPhone, userId, hashedToken = await self.processors.process_login(payload, seq, writer)
|
||||
userPhone, userId, hashedToken = await self.processors.login(payload, seq, writer)
|
||||
|
||||
if userPhone:
|
||||
await self._finish_auth(writer, address, userPhone, userId)
|
||||
case self.proto.LOGOUT:
|
||||
await self.processors.process_logout(seq, writer, hashedToken=hashedToken)
|
||||
case self.opcodes.LOGOUT:
|
||||
await self.processors.logout(seq, writer, hashedToken=hashedToken)
|
||||
break
|
||||
case self.proto.PING:
|
||||
await self.processors.process_ping(payload, seq, writer)
|
||||
case self.proto.LOG:
|
||||
await self.processors.process_telemetry(payload, seq, writer)
|
||||
case self.proto.ASSETS_UPDATE:
|
||||
case self.opcodes.PING:
|
||||
await self.processors.ping(payload, seq, writer)
|
||||
case self.opcodes.LOG:
|
||||
await self.processors.log(payload, seq, writer)
|
||||
case self.opcodes.ASSETS_UPDATE:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_get_assets, payload, seq, writer
|
||||
userPhone, self.processors.assets_update, payload, seq, writer
|
||||
)
|
||||
case self.proto.VIDEO_CHAT_HISTORY:
|
||||
case self.opcodes.VIDEO_CHAT_HISTORY:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_get_call_history, payload, seq, writer
|
||||
userPhone, self.processors.video_chat_history, payload, seq, writer
|
||||
)
|
||||
case self.proto.MSG_SEND:
|
||||
case self.opcodes.MSG_SEND:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_send_message, payload, seq, writer, userId, self.db_pool
|
||||
userPhone, self.processors.msg_send, payload, seq, writer, userId, self.db_pool
|
||||
)
|
||||
case self.proto.FOLDERS_GET:
|
||||
case self.opcodes.FOLDERS_GET:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_get_folders, payload, seq, writer, userPhone
|
||||
userPhone, self.processors.folders_get, payload, seq, writer, userPhone
|
||||
)
|
||||
case self.proto.SESSIONS_INFO:
|
||||
case self.opcodes.SESSIONS_INFO:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_get_sessions, payload, seq, writer, userPhone, hashedToken
|
||||
userPhone, self.processors.sessions_info, payload, seq, writer, userPhone, hashedToken
|
||||
)
|
||||
case self.proto.CHAT_INFO:
|
||||
case self.opcodes.CHAT_INFO:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_search_chats, payload, seq, writer, userId
|
||||
userPhone, self.processors.chat_info, payload, seq, writer, userId
|
||||
)
|
||||
case self.proto.CONTACT_INFO_BY_PHONE:
|
||||
case self.opcodes.CHAT_HISTORY:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_search_by_phone, payload, seq, writer, userId
|
||||
userPhone, self.processors.chat_history, payload, seq, writer, userId
|
||||
)
|
||||
case self.proto.OK_TOKEN:
|
||||
case self.opcodes.CONTACT_INFO_BY_PHONE:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_get_call_token, payload, seq, writer
|
||||
userPhone, self.processors.contact_info_by_phone, payload, seq, writer, userId
|
||||
)
|
||||
case self.proto.MSG_TYPING:
|
||||
case self.opcodes.OK_TOKEN:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_typing, payload, seq, writer, userId
|
||||
userPhone, self.processors.ok_token, payload, seq, writer
|
||||
)
|
||||
case self.proto.CONTACT_INFO:
|
||||
case self.opcodes.MSG_TYPING:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_search_users, payload, seq, writer
|
||||
userPhone, self.processors.msg_typing, payload, seq, writer, userId
|
||||
)
|
||||
case self.proto.COMPLAIN_REASONS_GET:
|
||||
case self.opcodes.CONTACT_INFO:
|
||||
await self.auth_required(
|
||||
userPhone, self.processors.process_complain_reasons_get, payload, seq, writer
|
||||
userPhone, self.processors.contact_info, payload, seq, writer
|
||||
)
|
||||
case self.proto.PROFILE:
|
||||
await self.processors.process_update_profile(
|
||||
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, userPhone=userPhone
|
||||
)
|
||||
case _:
|
||||
|
|
@ -1,383 +0,0 @@
|
|||
class OnemeConfig:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
# TODO: почистить вообще надо, и настройки потыкать
|
||||
SERVER_CONFIG = {
|
||||
"account-nickname-enabled": False,
|
||||
"account-removal-enabled": False,
|
||||
"anr-config": {
|
||||
"enabled": True,
|
||||
"timeout": {
|
||||
"low": 5000,
|
||||
"avg": 5000,
|
||||
"high": 5000
|
||||
}
|
||||
},
|
||||
"appearance-multi-theme-screen-enabled": True,
|
||||
"audio-transcription-locales": [],
|
||||
"available-complaints": [
|
||||
"FAKE",
|
||||
"SPAM",
|
||||
"PORNO",
|
||||
"EXTREMISM",
|
||||
"THREAT",
|
||||
"OTHER"
|
||||
],
|
||||
"avatars-screen-enabled": True,
|
||||
"bad-networ-indicator-config": {
|
||||
"signalingConfig": {
|
||||
"dcReportNetworkStatEnabled": False
|
||||
}
|
||||
},
|
||||
"bots-channel-adding": True,
|
||||
"cache-msg-preprocess": True,
|
||||
"call-incoming-ab": 2,
|
||||
"call-permissions-interval": 259200,
|
||||
"call-pinch-to-zoom": True,
|
||||
"call-rate": {
|
||||
"limit": 3,
|
||||
"sdk-limit": 2,
|
||||
"duration": 10,
|
||||
"delay": 86400
|
||||
},
|
||||
"callDontUseVpnForRtp": False,
|
||||
"callEnableIceRenomination": False,
|
||||
"calls-endpoint": "https://calls.okcdn.ru/",
|
||||
"calls-sdk-am-speaker-fix": True,
|
||||
"calls-sdk-audio-dynamic-redundancy": {
|
||||
"mab": 16,
|
||||
"dsb": 64,
|
||||
"nl": True,
|
||||
"df": True,
|
||||
"dlb": True
|
||||
},
|
||||
"calls-sdk-enable-nohost": True,
|
||||
"calls-sdk-incall-stat": False,
|
||||
"calls-sdk-linear-opus-bwe": True,
|
||||
"calls-sdk-mapping": {
|
||||
"off": True
|
||||
},
|
||||
"calls-sdk-remove-nonopus-audiocodecs": True,
|
||||
"calls-use-call-end-reason-fix": True,
|
||||
"calls-use-ws-url-validation": True,
|
||||
"cfs": True,
|
||||
"channels-complaint-enabled": True,
|
||||
"channels-enabled": True,
|
||||
"channels-search-subscribers-visible": True,
|
||||
"chat-complaint-enabled": False,
|
||||
"chat-gif-autoplay-enabled": True,
|
||||
"chat-history-notif-msg-strategy": 1,
|
||||
"chat-history-persist": False,
|
||||
"chat-history-warm-opts": 0,
|
||||
"chat-invite-link-permissions-enabled": True,
|
||||
"chat-media-scrollable-caption-enabled": True,
|
||||
"chat-video-autoplay-enabled": True,
|
||||
"chat-video-call-button": True,
|
||||
"chatlist-subtitle-ver": 1,
|
||||
"chats-folder-enabled": True,
|
||||
"chats-page-size": 50,
|
||||
"chats-preload-period": 15,
|
||||
"cis-enabled": True,
|
||||
"contact-add-bottom-sheet": True,
|
||||
"creation-2fa-config": {
|
||||
"pass_min_len": 6,
|
||||
"pass_max_len": 64,
|
||||
"hint_max_len": 30,
|
||||
"enabled": True
|
||||
},
|
||||
"debug-profile-info": False,
|
||||
"default-reactions-settings": {
|
||||
"isActive": True,
|
||||
"count": 8,
|
||||
"included": False,
|
||||
"reactionIds": []
|
||||
},
|
||||
"delete-msg-fys-large-chat-disabled": True,
|
||||
"devnull": {
|
||||
"opcode": True,
|
||||
"upload_hang": True
|
||||
},
|
||||
"disconnect-timeout": 300,
|
||||
"double-tap-reaction": "👍",
|
||||
"double-tap-reaction-enabled": True,
|
||||
"drafts-sync-enabled": False,
|
||||
"edit-chat-type-screen-enabled": False,
|
||||
"edit-timeout": 604800,
|
||||
"enable-filters-for-folders": True,
|
||||
"enable-unknown-contact-bottom-sheet": 2,
|
||||
"fake-chats": True,
|
||||
"family-protection-botid": 67804175,
|
||||
"february-23-26-theme": True,
|
||||
"file-preview": True,
|
||||
"file-upload-enabled": True,
|
||||
"file-upload-max-size": 4294967296,
|
||||
"file-upload-unsupported-types": [
|
||||
"exe"
|
||||
],
|
||||
"force-play-embed": True,
|
||||
"gc-from-p2p": True,
|
||||
"gce": False,
|
||||
"group-call-part-limit": 100,
|
||||
"grse": False,
|
||||
"gsse": True,
|
||||
"hide-incoming-call-notif": True,
|
||||
"host-reachability": True,
|
||||
"image-height": 1920,
|
||||
"image-quality": 0.800000011920929,
|
||||
"image-size": 40000000,
|
||||
"image-width": 1920,
|
||||
"in-app-review-triggers": 255,
|
||||
"informer-enabled": True,
|
||||
"inline-ev-player": True,
|
||||
"invalidate-db-msg-exception": True,
|
||||
"invite-friends-sheet-frequency": [
|
||||
2,
|
||||
7
|
||||
],
|
||||
"invite-link": "https://t.me/openmax_alerts",
|
||||
"invite-long": "Я пользуюсь OpenMAX. Присоединяйся! https://t.me/openmax_alerts",
|
||||
"invite-short": "Я пользуюсь OpenMAX. Присоединяйся! https://t.me/openmax_alerts",
|
||||
"join-requests": True,
|
||||
"js-download-delegate": False,
|
||||
"keep-connection": 2,
|
||||
"lebedev-theme-enabled": True,
|
||||
"lgce": True,
|
||||
"markdown-enabled": True,
|
||||
"markdown-menu": 0,
|
||||
"max-audio-length": 3600,
|
||||
"max-description-length": 400,
|
||||
"max-favorite-chats": 5,
|
||||
"max-favorite-sticker-sets": 100,
|
||||
"max-favorite-stickers": 100,
|
||||
"max-msg-length": 4000,
|
||||
"max-participants": 20000,
|
||||
"max-readmarks": 100,
|
||||
"max-theme-length": 200,
|
||||
"max-video-duration-download": 1200,
|
||||
"max-video-message-length": 60,
|
||||
"media-order": 1,
|
||||
"media-playlist-enabled": True,
|
||||
"media-transform": {
|
||||
"enabled": True,
|
||||
"hdr_enabled": False,
|
||||
"hevc_enabled": True,
|
||||
"max_enc_frames": {
|
||||
"low": 1,
|
||||
"avg": 1,
|
||||
"high": 2
|
||||
}
|
||||
},
|
||||
"media-viewer-rotation-enabled": True,
|
||||
"media-viewer-video-collage-enabled": True,
|
||||
"mentions-enabled": True,
|
||||
"mentions_entity_names_limit": 3,
|
||||
"migrate-unsafe-warn": True,
|
||||
"min-image-side-size": 64,
|
||||
"miui-menu-enabled": True,
|
||||
"money-transfer-botid": 1134691,
|
||||
"moscow-theme-enabled": True,
|
||||
"msg-get-reactions-page-size": 40,
|
||||
"music-files-enabled": False,
|
||||
"mytracker-enabled": False,
|
||||
"net-client-dns-enabled": True,
|
||||
"net-session-suppress-bad-disconnected-state": True,
|
||||
"net-stat-config": [
|
||||
64,
|
||||
48,
|
||||
128,
|
||||
135
|
||||
],
|
||||
"new-admin-permissions": True,
|
||||
"new-logout-logic": False,
|
||||
"new-media-upload-ui": True,
|
||||
"new-media-viewer-enabled": True,
|
||||
"new-settings-storage-screen-enabled": False,
|
||||
"new-width-text-bubbles-mob": True,
|
||||
"new-year-theme-2026": False,
|
||||
"nick-max-length": 60,
|
||||
"nick-min-length": 7,
|
||||
"official-org": True,
|
||||
"one-video-failover": True,
|
||||
"one-video-player": True,
|
||||
"one-video-uploader": True,
|
||||
"one-video-uploader-audio": True,
|
||||
"one-video-uploader-progress-fix": True,
|
||||
"perf-events": {
|
||||
"startup_report": 2,
|
||||
"web_app": 2
|
||||
},
|
||||
"player-load-control": {
|
||||
"mp_autoplay_enabled": False,
|
||||
"time_over_size": False,
|
||||
"buffer_after_rebuffer_ms": 3000,
|
||||
"buffer_ms": 500,
|
||||
"max_buffer_ms": 13000,
|
||||
"min_buffer_ms": 5000,
|
||||
"use_min_size_lc": True,
|
||||
"min_size_lc_fmt_mis_sf": 4
|
||||
},
|
||||
"progress-diff-for-notify": 1,
|
||||
"push-delivery": True,
|
||||
"qr-auth-enabled": True,
|
||||
"quotes-enabled": True,
|
||||
"react-errors": [
|
||||
"error.comment.chat.access",
|
||||
"error.comment.invalid",
|
||||
"error.message.invalid",
|
||||
"error.message.chat.access",
|
||||
"error.message.like.unknown.like",
|
||||
"error.message.like.unknown.reaction",
|
||||
"error.too-many-unlikes-dialog",
|
||||
"error.too-many-unlikes-chat",
|
||||
"error.too-many-likes",
|
||||
"error.reactions.not.allowed"
|
||||
],
|
||||
"react-permission": 2,
|
||||
"reactions-enabled": True,
|
||||
"reactions-max": 8,
|
||||
"reactions-menu": [
|
||||
"👍",
|
||||
"❤️",
|
||||
"🤣",
|
||||
"🔥",
|
||||
"😭",
|
||||
"💯",
|
||||
"💩",
|
||||
"😡"
|
||||
],
|
||||
"reactions-settings-enabled": True,
|
||||
"reconnect-call-ringtone": True,
|
||||
"ringtone-am-mode": True,
|
||||
"saved-messages-aliases": [
|
||||
"избранное",
|
||||
"saved",
|
||||
"favourite",
|
||||
"favorite",
|
||||
"личное",
|
||||
"моё",
|
||||
"мои",
|
||||
"мой",
|
||||
"моя",
|
||||
"любимое",
|
||||
"сохраненные",
|
||||
"сохраненное",
|
||||
"заметки",
|
||||
"закладки"
|
||||
],
|
||||
"scheduled-messages-enabled": True,
|
||||
"scheduled-posts-enabled": True,
|
||||
"search-webapps-showcase": {
|
||||
"items": [
|
||||
{
|
||||
"id": 4479862,
|
||||
"icon": "https://st.max.ru/icons/icon_channel_square.webp",
|
||||
"title": "Каналы"
|
||||
}
|
||||
]
|
||||
},
|
||||
"send-location-enabled": True,
|
||||
"send-logs-interval-sec": 900,
|
||||
"server-side-complains-enabled": True,
|
||||
"set-audio-device": False,
|
||||
"set-unread-timeout": 31536000,
|
||||
"settings-entry-banners": [
|
||||
{
|
||||
"id": 1,
|
||||
"logo": "https://st.max.ru/icons/epgu_white_111125.png",
|
||||
"align": 2,
|
||||
"items": [
|
||||
{
|
||||
"icon": "https://st.max.ru/icons/digital_id_new_40_3x.png",
|
||||
"title": "Цифровой ID",
|
||||
"appid": 8250447
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"id": 2,
|
||||
"items": [
|
||||
{
|
||||
"icon": "https://st.max.ru/icons/sferum_with_padding_120.png",
|
||||
"title": "Войти в Cферум",
|
||||
"appid": 2340831
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"show-reactions-on-multiselect": True,
|
||||
"show-warning-links": True,
|
||||
"speedy-upload": True,
|
||||
"speedy-voice-messages": True,
|
||||
"sse": True,
|
||||
"stat-session-background-threshold": 60000,
|
||||
"sticker-suggestion": [
|
||||
"RECENT",
|
||||
"NEW",
|
||||
"TOP"
|
||||
],
|
||||
"stickers-controller-suspend": True,
|
||||
"stickers-db-batch": True,
|
||||
"streamable-mp4": True,
|
||||
"stub": "stub2",
|
||||
"suspend-video-converter": True,
|
||||
"system-default-ringtone-opt": True,
|
||||
"transfer-botid": 1134691,
|
||||
"typing-enabled-FILE": True,
|
||||
"unique-favorites": True,
|
||||
"unsafe-files-alert": True,
|
||||
"upload-reusability": True,
|
||||
"upload-rx-no-blocking": True,
|
||||
"user-debug-report": 2340932,
|
||||
"video-msg-channels-enabled": True,
|
||||
"video-msg-config": {
|
||||
"duration": 60,
|
||||
"quality": 480,
|
||||
"min_frame_rate": 30,
|
||||
"max_frame_rate": 30
|
||||
},
|
||||
"video-msg-enabled": True,
|
||||
"video-transcoding-class": [
|
||||
2,
|
||||
3
|
||||
],
|
||||
"views-count-enabled": True,
|
||||
"watchdog-config": {
|
||||
"enabled": True,
|
||||
"stuck": 10,
|
||||
"hang": 60
|
||||
},
|
||||
"webapp-exc": [
|
||||
63602953,
|
||||
8250447
|
||||
],
|
||||
"webapp-push-open": True,
|
||||
"webview-cache-enabled": False,
|
||||
"welcome-sticker-ids": [
|
||||
272821,
|
||||
295349,
|
||||
13571,
|
||||
546741,
|
||||
476341
|
||||
],
|
||||
"white-list-links": [
|
||||
"max.ru",
|
||||
"vk.com",
|
||||
"vk.ru",
|
||||
"gosuslugi.ru",
|
||||
"mail.ru",
|
||||
"vk.ru",
|
||||
"vkvideo.ru"
|
||||
],
|
||||
"wm-analytics-enabled": True,
|
||||
"wm-workers-limit": 80,
|
||||
"wud": False,
|
||||
"y-map": {
|
||||
"tile": "34c7fd82-723d-4b23-8abb-33376729a893",
|
||||
"geocoder": "34c7fd82-723d-4b23-8abb-33376729a893",
|
||||
"static": "34c7fd82-723d-4b23-8abb-33376729a893",
|
||||
"logoLight": "https://st.max.ru/icons/ya_maps_logo_light.webp",
|
||||
"logoDark": "https://st.max.ru/icons/ya_maps_logo_dark.webp"
|
||||
},
|
||||
"has-phone": True
|
||||
}
|
||||
|
|
@ -1,251 +0,0 @@
|
|||
import lz4.block, msgpack, logging, json
|
||||
|
||||
class Proto:
|
||||
def __init__(self) -> None:
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
# TODO узнать какие должны быть лимиты и поменять,
|
||||
# сейчас это больше заглушка
|
||||
MAX_PAYLOAD_SIZE = 1048576 # 1 MB
|
||||
MAX_DECOMPRESSED_SIZE = 1048576 # 1 MB
|
||||
HEADER_SIZE = 10 # 1+2+1+2+4
|
||||
|
||||
### Работа с протоколом
|
||||
def unpack_packet(self, data: bytes) -> dict | None:
|
||||
# Проверяем минимальный размер пакета
|
||||
if len(data) < self.HEADER_SIZE:
|
||||
self.logger.warning(f"Пакет слишком маленький: {len(data)} байт")
|
||||
return None
|
||||
|
||||
# Распаковываем заголовок
|
||||
ver = int.from_bytes(data[0:1], "big")
|
||||
cmd = int.from_bytes(data[1:3], "big")
|
||||
seq = int.from_bytes(data[3:4], "big")
|
||||
opcode = int.from_bytes(data[4:6], "big")
|
||||
packed_len = int.from_bytes(data[6:10], "big")
|
||||
|
||||
# Флаг упаковки
|
||||
comp_flag = packed_len >> 24
|
||||
|
||||
# Парсим данные пакета
|
||||
payload_length = packed_len & 0xFFFFFF
|
||||
|
||||
# Проверяем размер payload
|
||||
if payload_length > self.MAX_PAYLOAD_SIZE:
|
||||
self.logger.warning(f"Payload слишком большой: {payload_length} B (лимит {self.MAX_PAYLOAD_SIZE})")
|
||||
return None
|
||||
|
||||
# Проверяем длину пакета
|
||||
if len(data) < self.HEADER_SIZE + payload_length:
|
||||
self.logger.warning(f"Пакет неполный: требуется {self.HEADER_SIZE + payload_length} B, получено {len(data)}")
|
||||
return None
|
||||
|
||||
payload_bytes = data[10 : 10 + payload_length]
|
||||
payload = None
|
||||
|
||||
# Декодируем данные пакета
|
||||
if payload_bytes:
|
||||
# Разжимаем данные пакета, если требуется
|
||||
if comp_flag != 0:
|
||||
compressed_data = payload_bytes
|
||||
try:
|
||||
payload_bytes = lz4.block.decompress(
|
||||
compressed_data,
|
||||
uncompressed_size=self.MAX_DECOMPRESSED_SIZE,
|
||||
)
|
||||
except lz4.block.LZ4BlockError:
|
||||
self.logger.warning("Ошибка декомпрессии LZ4")
|
||||
return None
|
||||
|
||||
# Распаковываем msgpack
|
||||
payload = msgpack.unpackb(payload_bytes, raw=False, strict_map_key=False)
|
||||
|
||||
self.logger.debug(f"Распаковал - ver={ver} cmd={cmd} seq={seq} opcode={opcode} payload={payload}")
|
||||
|
||||
# Возвращаем
|
||||
return {
|
||||
"ver": ver,
|
||||
"cmd": cmd,
|
||||
"seq": seq,
|
||||
"opcode": opcode,
|
||||
"payload": payload,
|
||||
}
|
||||
|
||||
def pack_packet(self, ver: int = 10, cmd: int = 1, seq: int = 1, opcode: int = 6, payload: dict = None) -> bytes:
|
||||
# Запаковываем заголовок
|
||||
ver_b = ver.to_bytes(1, "big")
|
||||
cmd_b = cmd.to_bytes(2, "big")
|
||||
seq_b = seq.to_bytes(1, "big")
|
||||
opcode_b = opcode.to_bytes(2, "big")
|
||||
|
||||
# Запаковываем данные пакета
|
||||
payload_bytes: bytes | None = msgpack.packb(payload)
|
||||
if payload_bytes is None:
|
||||
payload_bytes = b""
|
||||
payload_len = len(payload_bytes) & 0xFFFFFF
|
||||
payload_len_b = payload_len.to_bytes(4, 'big')
|
||||
|
||||
self.logger.debug(f"Упаковал - ver={ver} cmd={cmd} seq={seq} opcode={opcode} payload={payload}")
|
||||
|
||||
# Возвращаем пакет
|
||||
return ver_b + cmd_b + seq_b + opcode_b + payload_len_b + payload_bytes
|
||||
|
||||
### Констаты протокола
|
||||
CMD_OK = 0x100
|
||||
CMD_NOF = 0x200
|
||||
CMD_ERR = 0x300
|
||||
PROTO_VER = 10
|
||||
|
||||
### Команды
|
||||
PING = 1
|
||||
DEBUG = 2
|
||||
RECONNECT = 3
|
||||
LOG = 5
|
||||
SESSION_INIT = 6
|
||||
PROFILE = 16
|
||||
AUTH_REQUEST = 17
|
||||
AUTH = 18
|
||||
LOGIN = 19
|
||||
LOGOUT = 20
|
||||
SYNC = 21
|
||||
CONFIG = 22
|
||||
AUTH_CONFIRM = 23
|
||||
AUTH_CREATE_TRACK = 112
|
||||
AUTH_CHECK_PASSWORD = 113
|
||||
AUTH_LOGIN_CHECK_PASSWORD = 115
|
||||
AUTH_LOGIN_PROFILE_DELETE = 116
|
||||
AUTH_LOGIN_RESTORE_PASSWORD = 101
|
||||
AUTH_VALIDATE_PASSWORD = 107
|
||||
AUTH_VALIDATE_HINT = 108
|
||||
AUTH_VERIFY_EMAIL = 109
|
||||
AUTH_CHECK_EMAIL = 110
|
||||
AUTH_SET_2FA = 111
|
||||
AUTH_2FA_DETAILS = 104
|
||||
ASSETS_GET = 26
|
||||
ASSETS_UPDATE = 27
|
||||
ASSETS_GET_BY_IDS = 28
|
||||
ASSETS_LIST_MODIFY = 261
|
||||
ASSETS_REMOVE = 259
|
||||
ASSETS_MOVE = 260
|
||||
ASSETS_ADD = 29
|
||||
PRESET_AVATARS = 25
|
||||
CONTACT_INFO = 32
|
||||
CONTACT_INFO_BY_PHONE = 46
|
||||
CONTACT_ADD = 33
|
||||
CONTACT_UPDATE = 34
|
||||
CONTACT_PRESENCE = 35
|
||||
CONTACT_LIST = 36
|
||||
CONTACT_SEARCH = 37
|
||||
CONTACT_MUTUAL = 38
|
||||
CONTACT_PHOTOS = 39
|
||||
CONTACT_SORT = 40
|
||||
CONTACT_VERIFY = 42
|
||||
REMOVE_CONTACT_PHOTO = 43
|
||||
CHAT_INFO = 48
|
||||
CHAT_HISTORY = 49
|
||||
CHAT_MARK = 50
|
||||
CHAT_MEDIA = 51
|
||||
CHAT_DELETE = 52
|
||||
CHATS_LIST = 53
|
||||
CHAT_CLEAR = 54
|
||||
CHAT_UPDATE = 55
|
||||
CHAT_CHECK_LINK = 56
|
||||
CHAT_JOIN = 57
|
||||
CHAT_LEAVE = 58
|
||||
CHAT_MEMBERS = 59
|
||||
PUBLIC_SEARCH = 60
|
||||
CHAT_PERSONAL_CONFIG = 61
|
||||
CHAT_CREATE = 63
|
||||
REACTIONS_SETTINGS_GET_BY_CHAT_ID = 258
|
||||
CHAT_REACTIONS_SETTINGS_SET = 257
|
||||
MSG_SEND = 64
|
||||
MSG_TYPING = 65
|
||||
MSG_DELETE = 66
|
||||
MSG_EDIT = 67
|
||||
MSG_DELETE_RANGE = 92
|
||||
MSG_REACTION = 178
|
||||
MSG_CANCEL_REACTION = 179
|
||||
MSG_GET_REACTIONS = 180
|
||||
MSG_GET_DETAILED_REACTIONS = 181
|
||||
CHAT_SEARCH = 68
|
||||
MSG_SHARE_PREVIEW = 70
|
||||
MSG_GET = 71
|
||||
MSG_SEARCH_TOUCH = 72
|
||||
MSG_SEARCH = 73
|
||||
MSG_GET_STAT = 74
|
||||
CHAT_SUBSCRIBE = 75
|
||||
VIDEO_CHAT_START = 76
|
||||
VIDEO_CHAT_START_ACTIVE = 78
|
||||
CHAT_MEMBERS_UPDATE = 77
|
||||
VIDEO_CHAT_HISTORY = 79
|
||||
PHOTO_UPLOAD = 80
|
||||
STICKER_UPLOAD = 81
|
||||
VIDEO_UPLOAD = 82
|
||||
VIDEO_PLAY = 83
|
||||
VIDEO_CHAT_CREATE_JOIN_LINK = 84
|
||||
CHAT_PIN_SET_VISIBILITY = 86
|
||||
FILE_UPLOAD = 87
|
||||
FILE_DOWNLOAD = 88
|
||||
LINK_INFO = 89
|
||||
SESSIONS_INFO = 96
|
||||
SESSIONS_CLOSE = 97
|
||||
PHONE_BIND_REQUEST = 98
|
||||
PHONE_BIND_CONFIRM = 99
|
||||
GET_INBOUND_CALLS = 103
|
||||
EXTERNAL_CALLBACK = 105
|
||||
OK_TOKEN = 158
|
||||
CHAT_COMPLAIN = 117
|
||||
MSG_SEND_CALLBACK = 118
|
||||
SUSPEND_BOT = 119
|
||||
LOCATION_STOP = 124
|
||||
GET_LAST_MENTIONS = 127
|
||||
STICKER_CREATE = 193
|
||||
STICKER_SUGGEST = 194
|
||||
VIDEO_CHAT_MEMBERS = 195
|
||||
NOTIF_MESSAGE = 128
|
||||
NOTIF_TYPING = 129
|
||||
NOTIF_MARK = 130
|
||||
NOTIF_CONTACT = 131
|
||||
NOTIF_PRESENCE = 132
|
||||
NOTIF_CONFIG = 134
|
||||
NOTIF_CHAT = 135
|
||||
NOTIF_ATTACH = 136
|
||||
NOTIF_CALL_START = 137
|
||||
NOTIF_CONTACT_SORT = 139
|
||||
NOTIF_MSG_DELETE_RANGE = 140
|
||||
NOTIF_MSG_DELETE = 142
|
||||
NOTIF_MSG_REACTIONS_CHANGED = 155
|
||||
NOTIF_MSG_YOU_REACTED = 156
|
||||
NOTIF_CALLBACK_ANSWER = 143
|
||||
CHAT_BOT_COMMANDS = 144
|
||||
BOT_INFO = 145
|
||||
NOTIF_LOCATION = 147
|
||||
NOTIF_LOCATION_REQUEST = 148
|
||||
NOTIF_ASSETS_UPDATE = 150
|
||||
NOTIF_DRAFT = 152
|
||||
NOTIF_DRAFT_DISCARD = 153
|
||||
DRAFT_SAVE = 176
|
||||
DRAFT_DISCARD = 177
|
||||
CHAT_HIDE = 196
|
||||
CHAT_SEARCH_COMMON_PARTICIPANTS = 198
|
||||
NOTIF_MSG_DELAYED = 154
|
||||
NOTIF_PROFILE = 159
|
||||
PROFILE_DELETE = 199
|
||||
PROFILE_DELETE_TIME = 200
|
||||
WEB_APP_INIT_DATA = 160
|
||||
COMPLAIN = 161
|
||||
COMPLAIN_REASONS_GET = 162
|
||||
FOLDERS_GET = 272
|
||||
FOLDERS_GET_BY_ID = 273
|
||||
FOLDERS_UPDATE = 274
|
||||
FOLDERS_REORDER = 275
|
||||
FOLDERS_DELETE = 276
|
||||
NOTIF_FOLDERS = 277
|
||||
|
||||
AUTH_QR_APPROVE = 290
|
||||
NOTIF_BANNERS = 292
|
||||
CHAT_SUGGEST = 300
|
||||
AUDIO_PLAY = 301
|
||||
SEND_VOTE = 304
|
||||
VOTERS_LIST_BY_ANSWER = 305
|
||||
GET_POLL_UPDATES = 306
|
||||
|
|
@ -1,5 +1,5 @@
|
|||
import asyncio
|
||||
from tamtam_tcp.server import TTMobileServer
|
||||
from tamtam.socket import TTMobileServer
|
||||
from classes.controllerbase import ControllerBase
|
||||
from common.config import ServerConfig
|
||||
|
||||
|
|
@ -27,4 +27,8 @@ class FinalAuthPayloadModel(pydantic.BaseModel):
|
|||
deviceType: str
|
||||
tokenType: str
|
||||
deviceId: str
|
||||
token: str
|
||||
|
||||
class LoginPayloadModel(pydantic.BaseModel):
|
||||
interactive: bool
|
||||
token: str
|
||||
|
|
@ -6,21 +6,30 @@ import json
|
|||
import re
|
||||
from common.static import Static
|
||||
from common.tools import Tools
|
||||
from tamtam_tcp.proto import Proto
|
||||
from tamtam_tcp.models import *
|
||||
|
||||
from common.proto_tcp import MobileProto
|
||||
from common.proto_web import WebProto
|
||||
from common.opcodes import Opcodes
|
||||
|
||||
from tamtam.models import *
|
||||
|
||||
|
||||
class Processors:
|
||||
def __init__(self, db_pool=None, clients=None, send_event=None):
|
||||
def __init__(self, db_pool=None, clients=None, send_event=None, type="socket"):
|
||||
if clients is None:
|
||||
clients = {} # Более правильная логика
|
||||
self.static = Static()
|
||||
self.proto = Proto()
|
||||
self.tools = Tools()
|
||||
self.opcodes = Opcodes()
|
||||
self.error_types = self.static.ErrorTypes()
|
||||
self.db_pool = db_pool
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
if type == "socket":
|
||||
self.proto = MobileProto()
|
||||
elif type == "web":
|
||||
self.proto = WebProto()
|
||||
|
||||
async def _send(self, writer, packet):
|
||||
try:
|
||||
writer.write(packet)
|
||||
|
|
@ -42,13 +51,13 @@ class Processors:
|
|||
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_hello(self, payload, seq, writer):
|
||||
async def session_init(self, payload, seq, writer):
|
||||
"""Обработчик приветствия"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
HelloPayloadModel.model_validate(payload)
|
||||
except Exception as e:
|
||||
await self._send_error(seq, self.proto.HELLO, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.SESSION_INIT, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return None, None
|
||||
|
||||
# Получаем данные из пакета
|
||||
|
|
@ -67,20 +76,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.HELLO, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.SESSION_INIT, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
return device_type, device_name
|
||||
|
||||
async def process_request_code(self, payload, seq, writer):
|
||||
async def auth_request(self, payload, seq, writer):
|
||||
"""Обработчик запроса кода"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
RequestCodePayloadModel.model_validate(payload)
|
||||
except Exception as e:
|
||||
await self._send_error(seq, self.proto.REQUEST_CODE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_REQUEST, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем телефон из пакета
|
||||
|
|
@ -120,21 +129,20 @@ class Processors:
|
|||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.REQUEST_CODE, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH_REQUEST, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
self.logger.debug(f"Код для {phone}: {code}")
|
||||
|
||||
async def process_verify_code(self, payload, seq, writer):
|
||||
async def auth(self, payload, seq, writer):
|
||||
"""Обработчик проверки кода"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
VerifyCodePayloadModel.model_validate(payload)
|
||||
except Exception as e:
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -154,12 +162,12 @@ class Processors:
|
|||
stored_token = await cursor.fetchone()
|
||||
|
||||
if not stored_token:
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.CODE_EXPIRED, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.CODE_EXPIRED, writer)
|
||||
return
|
||||
|
||||
# Проверяем код
|
||||
if stored_token.get("code_hash") != hashed_code:
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.INVALID_CODE, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH, self.error_types.INVALID_CODE, writer)
|
||||
return
|
||||
|
||||
# Ищем аккаунт
|
||||
|
|
@ -200,18 +208,18 @@ class Processors:
|
|||
}
|
||||
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.VERIFY_CODE, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH, payload=payload
|
||||
)
|
||||
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_final_auth(self, payload, seq, writer, deviceType, deviceName):
|
||||
async def auth_confirm(self, payload, seq, writer, deviceType, deviceName):
|
||||
"""Обработчик финальной аутентификации"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
FinalAuthPayloadModel.model_validate(payload)
|
||||
except Exception as e:
|
||||
await self._send_error(seq, self.proto.FINAL_AUTH, self.error_types.INVALID_PAYLOAD, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
|
|
@ -236,12 +244,12 @@ class Processors:
|
|||
stored_token = await cursor.fetchone()
|
||||
|
||||
if stored_token is None:
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.INVALID_TOKEN, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.INVALID_TOKEN, writer)
|
||||
return
|
||||
|
||||
# Если авторизация только началась - отдаем ошибку
|
||||
if stored_token.get("state") == "started":
|
||||
await self._send_error(seq, self.proto.VERIFY_CODE, self.error_types.INVALID_TOKEN, writer)
|
||||
await self._send_error(seq, self.opcodes.AUTH_CONFIRM, self.error_types.INVALID_TOKEN, writer)
|
||||
return
|
||||
|
||||
# Ищем аккаунт
|
||||
|
|
@ -284,8 +292,109 @@ class Processors:
|
|||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.proto.FINAL_AUTH, payload=payload
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.AUTH_CONFIRM, payload=payload
|
||||
)
|
||||
|
||||
# Отправялем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def login(self, payload, seq, writer):
|
||||
"""Обработчик авторизации клиента на сервере"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
LoginPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.opcodes.LOGIN, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return
|
||||
|
||||
# Получаем данные из пакета
|
||||
token = payload.get("token")
|
||||
|
||||
# Хешируем токен
|
||||
hashed_token = hashlib.sha256(token.encode()).hexdigest()
|
||||
|
||||
# Ищем токен в бд
|
||||
async with self.db_pool.acquire() as conn:
|
||||
async with conn.cursor() as cursor:
|
||||
await cursor.execute("SELECT * FROM tokens WHERE token_hash = %s", (hashed_token,))
|
||||
token_data = await cursor.fetchone()
|
||||
|
||||
# Если токен не найден, отправляем ошибку
|
||||
if token_data is None:
|
||||
await self._send_error(seq, self.opcodes.LOGIN, self.error_types.INVALID_TOKEN, writer)
|
||||
return
|
||||
|
||||
# Ищем аккаунт пользователя в бд
|
||||
await cursor.execute("SELECT * FROM users WHERE phone = %s", (token_data.get("phone"),))
|
||||
user = await cursor.fetchone()
|
||||
|
||||
# Ищем данные пользователя в бд
|
||||
await cursor.execute("SELECT * FROM user_data WHERE phone = %s", (token_data.get("phone"),))
|
||||
user_data = await cursor.fetchone()
|
||||
|
||||
# Аватарка с биографией
|
||||
photo_id = None if not user.get("avatar_id") else int(user.get("avatar_id"))
|
||||
avatar_url = None if not photo_id else self.config.avatar_base_url + photo_id
|
||||
description = None if not user.get("description") else user.get("description")
|
||||
|
||||
# Генерируем профиль
|
||||
profile = self.tools.generate_profile_tt(
|
||||
id=user.get("id"),
|
||||
phone=int(user.get("phone")),
|
||||
avatarUrl=avatar_url,
|
||||
photoId=photo_id,
|
||||
updateTime=int(user.get("updatetime")),
|
||||
firstName=user.get("firstname"),
|
||||
lastName=user.get("lastname"),
|
||||
options=json.loads(user.get("options")),
|
||||
description=description,
|
||||
username=user.get("username")
|
||||
)
|
||||
|
||||
chats = await self.tools.generate_chats(
|
||||
json.loads(user_data.get("chats")),
|
||||
self.db_pool, user.get("id")
|
||||
)
|
||||
|
||||
# Формируем данные пакета
|
||||
payload = {
|
||||
"profile": profile,
|
||||
"chats": chats,
|
||||
"chatMarker": 0,
|
||||
"messages": {},
|
||||
"contacts": [],
|
||||
"presence": {},
|
||||
"config": {
|
||||
"hash": "0",
|
||||
"server": {},
|
||||
"user": json.loads(user_data.get("user_config")),
|
||||
"chatFolders": {
|
||||
"FOLDERS": [],
|
||||
"ALL_FILTER_EXCLUDE": []
|
||||
}
|
||||
},
|
||||
"token": token,
|
||||
"calls": [],
|
||||
"videoChatHistory": False,
|
||||
"drafts": {
|
||||
"chats": {
|
||||
"discarded": {},
|
||||
"saved": {}
|
||||
},
|
||||
"users": {
|
||||
"discarded": {},
|
||||
"saved": {}
|
||||
}
|
||||
},
|
||||
"time": int(time.time() * 1000)
|
||||
}
|
||||
|
||||
# Собираем пакет
|
||||
packet = self.proto.pack_packet(
|
||||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.LOGIN, payload=payload
|
||||
)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
return int(user.get("phone")), int(user.get("id")), hashed_token
|
||||
|
|
@ -1,7 +1,8 @@
|
|||
import asyncio, logging, traceback
|
||||
from tamtam_tcp.proto import Proto
|
||||
from tamtam_tcp.processors import Processors
|
||||
from common.proto_tcp import MobileProto
|
||||
from tamtam.processors import Processors
|
||||
from common.rate_limiter import RateLimiter
|
||||
from common.opcodes import Opcodes
|
||||
|
||||
class TTMobileServer:
|
||||
def __init__(self, host="0.0.0.0", port=443, ssl_context=None, db_pool=None, clients={}, send_event=None):
|
||||
|
|
@ -13,7 +14,9 @@ class TTMobileServer:
|
|||
self.db_pool = db_pool
|
||||
self.clients = clients
|
||||
|
||||
self.proto = Proto()
|
||||
self.opcodes = Opcodes()
|
||||
|
||||
self.proto = MobileProto()
|
||||
self.processors = Processors(db_pool=db_pool, clients=clients, send_event=send_event)
|
||||
|
||||
# rate limiter
|
||||
|
|
@ -69,23 +72,31 @@ class TTMobileServer:
|
|||
payload = packet.get("payload")
|
||||
|
||||
match opcode:
|
||||
case self.proto.HELLO:
|
||||
deviceType, deviceName = await self.processors.process_hello(payload, seq, writer)
|
||||
case self.proto.REQUEST_CODE:
|
||||
case self.opcodes.SESSION_INIT:
|
||||
deviceType, deviceName = 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.proto.REQUEST_CODE, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.AUTH_REQUEST, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
await self.processors.process_request_code(payload, seq, writer)
|
||||
case self.proto.VERIFY_CODE:
|
||||
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.proto.VERIFY_CODE, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.AUTH, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
await self.processors.process_verify_code(payload, seq, writer)
|
||||
case self.proto.FINAL_AUTH:
|
||||
await self.processors.auth(payload, seq, writer)
|
||||
case self.opcodes.AUTH_CONFIRM:
|
||||
if not self.auth_rate_limiter.is_allowed(address[0]):
|
||||
await self.processors._send_error(seq, self.proto.FINAL_AUTH, self.processors.error_types.RATE_LIMITED, writer)
|
||||
await self.processors._send_error(seq, self.opcodes.AUTH_CONFIRM, self.processors.error_types.RATE_LIMITED, writer)
|
||||
else:
|
||||
await self.processors.process_final_auth(payload, seq, writer, deviceType, deviceName)
|
||||
await self.processors.auth_confirm(payload, seq, writer, deviceType, deviceName)
|
||||
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)
|
||||
|
||||
if userPhone:
|
||||
await self._finish_auth(writer, address, userPhone, userId)
|
||||
case _:
|
||||
self.logger.warning(f"Неизвестный опкод {opcode}")
|
||||
except Exception as e:
|
||||
|
|
@ -95,6 +106,50 @@ class TTMobileServer:
|
|||
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_mobile"
|
||||
}
|
||||
)
|
||||
else:
|
||||
self.clients[id] = {
|
||||
"phone": phone,
|
||||
"id": id,
|
||||
"clients": [
|
||||
{
|
||||
"writer": writer,
|
||||
"ip": addr[0],
|
||||
"port": addr[1],
|
||||
"protocol": "oneme_mobile"
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
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)
|
||||
|
||||
async def start(self):
|
||||
"""Функция для запуска сервера"""
|
||||
self.server = await asyncio.start_server(
|
||||
|
|
@ -1,22 +0,0 @@
|
|||
import asyncio
|
||||
from classes.controllerbase import ControllerBase
|
||||
from common.config import ServerConfig
|
||||
from tamtam_ws.server import TTWSServer
|
||||
|
||||
class TTWSController(ControllerBase):
|
||||
def __init__(self):
|
||||
self.config = ServerConfig()
|
||||
|
||||
def launch(self, api):
|
||||
async def _start_all():
|
||||
await asyncio.gather(
|
||||
TTWSServer(
|
||||
host=self.config.host,
|
||||
port=self.config.tamtam_ws_port,
|
||||
db_pool=api['db'],
|
||||
clients=api['clients'],
|
||||
send_event=api['event']
|
||||
).start()
|
||||
)
|
||||
|
||||
return _start_all()
|
||||
|
|
@ -1,27 +0,0 @@
|
|||
import pydantic
|
||||
|
||||
class MessageModel(pydantic.BaseModel):
|
||||
ver: int
|
||||
cmd: int
|
||||
seq: int
|
||||
opcode: int
|
||||
payload: dict = None
|
||||
|
||||
class UserAgentModel(pydantic.BaseModel):
|
||||
deviceType: str
|
||||
appVersion: str
|
||||
osVersion: str
|
||||
locale: str
|
||||
deviceLocale: str
|
||||
deviceName: str
|
||||
screen: str
|
||||
headerUserAgent: str
|
||||
timezone: str
|
||||
|
||||
class HelloPayloadModel(pydantic.BaseModel):
|
||||
userAgent: UserAgentModel
|
||||
deviceId: str
|
||||
|
||||
class RequestCodePayloadModel(pydantic.BaseModel):
|
||||
phone: str
|
||||
requestType: str
|
||||
|
|
@ -1,80 +0,0 @@
|
|||
import hashlib, secrets, random, time, logging, json
|
||||
from common.static import Static
|
||||
from common.tools import Tools
|
||||
from tamtam_ws.proto import Proto
|
||||
from tamtam_ws.models import *
|
||||
|
||||
class Processors:
|
||||
def __init__(self, db_pool=None, clients={}, send_event=None):
|
||||
self.static = Static()
|
||||
self.tools = Tools()
|
||||
self.proto = Proto()
|
||||
self.error_types = self.static.ErrorTypes()
|
||||
self.db_pool = db_pool
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
async def _send(self, writer, packet):
|
||||
"""Отправка пакета"""
|
||||
try:
|
||||
await writer.send(packet)
|
||||
except Exception as error:
|
||||
self.logger.error(f"Ошибка при отправке пакета - {error}")
|
||||
|
||||
async def _send_error(self, seq, opcode, type, writer):
|
||||
payload = self.static.ERROR_TYPES.get(type, {
|
||||
"localizedMessage": "Неизвестная ошибка",
|
||||
"error": "unknown.error",
|
||||
"message": "Unknown error",
|
||||
"title": "Неизвестная ошибка"
|
||||
})
|
||||
|
||||
packet = self.proto.pack_packet(
|
||||
seq=seq, opcode=opcode, payload=payload
|
||||
)
|
||||
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_hello(self, payload, seq, writer):
|
||||
"""Обработчик приветствия"""
|
||||
# Валидируем данные пакета
|
||||
try:
|
||||
HelloPayloadModel.model_validate(payload)
|
||||
except pydantic.ValidationError as error:
|
||||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||||
await self._send_error(seq, self.proto.SESSION_INIT, self.error_types.INVALID_PAYLOAD, writer)
|
||||
return None, None
|
||||
|
||||
# Получаем данные из пакета
|
||||
deviceType = payload.get("userAgent").get("deviceType")
|
||||
deviceName = payload.get("userAgent").get("deviceName")
|
||||
|
||||
# Собираем данные ответа
|
||||
payload = {
|
||||
"proxy": "",
|
||||
"logs-enabled": False,
|
||||
"proxy-domains": [],
|
||||
"location": "RU"
|
||||
}
|
||||
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(seq=seq, opcode=self.proto.SESSION_INIT, payload=payload)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
return deviceType, deviceName
|
||||
|
||||
async def process_ping(self, payload, seq, writer):
|
||||
"""Обработчик пинга"""
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(seq=seq, opcode=self.proto.PING)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
||||
async def process_telemetry(self, payload, seq, writer):
|
||||
"""Обработчик телеметрии"""
|
||||
# Создаем пакет
|
||||
packet = self.proto.pack_packet(seq=seq, opcode=self.proto.LOG)
|
||||
|
||||
# Отправляем
|
||||
await self._send(writer, packet)
|
||||
|
|
@ -1,165 +0,0 @@
|
|||
import json
|
||||
|
||||
class Proto:
|
||||
def pack_packet(self, ver=10, cmd=1, seq=0, opcode=1, payload=None):
|
||||
# а разве не надо в жсон запаковывать ещё
|
||||
# о всё
|
||||
return json.dumps({
|
||||
"ver": ver,
|
||||
"cmd": cmd,
|
||||
"seq": seq,
|
||||
"opcode": opcode,
|
||||
"payload": payload
|
||||
})
|
||||
|
||||
MAX_PACKET_SIZE = 65536 # 64 KB, заглушка, нужно узнать реальные лимиты и поменять, хотя кто будет это делать...
|
||||
|
||||
def unpack_packet(self, packet):
|
||||
# try catch чтобы не сыпалось всё при неверных пакетах
|
||||
if isinstance(packet, (str, bytes)) and len(packet) > self.MAX_PACKET_SIZE:
|
||||
return {}
|
||||
|
||||
try:
|
||||
parsed_packet = json.loads(packet)
|
||||
except (json.JSONDecodeError, TypeError, ValueError):
|
||||
return {}
|
||||
|
||||
return parsed_packet
|
||||
# мне кажется долго вручную всё писать
|
||||
# а как еще
|
||||
# ну вставить сюда целиком и потом через multiline cursor удалить лишнее
|
||||
# ну ты удалишь тогда. я на тачпаде
|
||||
# ладно щас другим способом удалю
|
||||
# всё нахуй
|
||||
# TAMTAM SOURCE LEAK 2026
|
||||
# так ну че делать будем
|
||||
# так ну
|
||||
|
||||
# 19 опкод сделан?
|
||||
# нет сэр пошли библиотеку тамы смотреть
|
||||
# мб найдем че. она без обфускации
|
||||
# а ты ее видишь?
|
||||
# пошли
|
||||
|
||||
### Констаты протокола
|
||||
CMD_OK = 1
|
||||
CMD_NOF = 2
|
||||
CMD_ERR = 3
|
||||
PROTO_VER = 10
|
||||
|
||||
### Команды
|
||||
PING = 1
|
||||
LOG = 5
|
||||
SESSION_INIT = 6
|
||||
PROFILE = 16
|
||||
AUTH_REQUEST = 17
|
||||
AUTH_CHECK_SCENARIO = 263
|
||||
AUTH = 18
|
||||
LOGIN = 19
|
||||
LOGOUT = 20
|
||||
SYNC = 21
|
||||
CONFIG = 22
|
||||
AUTH_CONFIRM = 23
|
||||
ASSETS_GET = 26
|
||||
ASSETS_UPDATE = 27
|
||||
ASSETS_GET_BY_IDS = 28
|
||||
ASSETS_ADD = 29
|
||||
ASSETS_REMOVE = 259
|
||||
ASSETS_MOVE = 260
|
||||
ASSETS_LIST_MODIFY = 261
|
||||
CONTACT_INFO = 32
|
||||
CONTACT_UPDATE = 34
|
||||
CONTACT_PRESENCE = 35
|
||||
CONTACT_LIST = 36
|
||||
CONTACT_PHOTOS = 39
|
||||
CONTACT_CREATE = 41
|
||||
REMOVE_CONTACT_PHOTO = 43
|
||||
OWN_CONTACT_SEARCH = 44
|
||||
CHAT_INFO = 48
|
||||
CHAT_HISTORY = 49
|
||||
CHAT_MARK = 50
|
||||
CHAT_MEDIA = 51
|
||||
CHAT_DELETE = 52
|
||||
CHAT_LIST = 53
|
||||
CHAT_CLEAR = 54
|
||||
CHAT_UPDATE = 55
|
||||
CHAT_CHECK_LINK = 56
|
||||
CHAT_JOIN = 57
|
||||
CHAT_LEAVE = 58
|
||||
CHAT_MEMBERS = 59
|
||||
CHAT_CLOSE = 61
|
||||
CHAT_BOT_COMMANDS = 144
|
||||
CHAT_SUBSCRIBE = 75
|
||||
PUBLIC_SEARCH = 60
|
||||
CHAT_CREATE = 63
|
||||
MSG_SEND = 64
|
||||
MSG_TYPING = 65
|
||||
MSG_DELETE = 66
|
||||
MSG_EDIT = 67
|
||||
CHAT_SEARCH = 68
|
||||
MSG_SHARE_PREVIEW = 70
|
||||
MSG_SEARCH_TOUCH = 72
|
||||
MSG_SEARCH = 73
|
||||
MSG_GET_STAT = 74
|
||||
MSG_GET = 71
|
||||
VIDEO_CHAT_START = 76
|
||||
VIDEO_CHAT_JOIN = 102
|
||||
VIDEO_CHAT_COMMAND = 78
|
||||
VIDEO_CHAT_MEMBERS = 195
|
||||
CHAT_MEMBERS_UPDATE = 77
|
||||
PHOTO_UPLOAD = 80
|
||||
STICKER_UPLOAD = 81
|
||||
VIDEO_UPLOAD = 82
|
||||
VIDEO_PLAY = 83
|
||||
MUSIC_PLAY = 84
|
||||
MUSIC_PLAY30 = 85
|
||||
FILE_UPLOAD = 87
|
||||
FILE_DOWNLOAD = 88
|
||||
CHAT_PIN_SET_VISIBILITY = 86
|
||||
LINK_INFO = 89
|
||||
MESSAGE_LINK = 90
|
||||
MSG_CONSTRUCT = 94
|
||||
SESSIONS_INFO = 96
|
||||
SESSIONS_CLOSE = 97
|
||||
PHONE_BIND_REQUEST = 98
|
||||
PHONE_BIND_CONFIRM = 99
|
||||
UNBIND_OK_PROFILE = 100
|
||||
CHAT_COMPLAIN = 117
|
||||
MSG_SEND_CALLBACK = 118
|
||||
SUSPEND_BOT = 119
|
||||
MSG_REACT = 178
|
||||
MSG_CANCEL_REACTION = 179
|
||||
MSG_GET_REACTIONS = 180
|
||||
MSG_GET_DETAILED_REACTIONS = 181
|
||||
LOCATION_STOP = 124
|
||||
LOCATION_SEND = 125
|
||||
LOCATION_REQUEST = 126
|
||||
NOTIF_MESSAGE = 128
|
||||
NOTIF_TYPING = 129
|
||||
NOTIF_MARK = 130
|
||||
NOTIF_CONTACT = 131
|
||||
NOTIF_PRESENCE = 132
|
||||
NOTIF_CONFIG = 134
|
||||
NOTIF_CHAT = 135
|
||||
NOTIF_ATTACH = 136
|
||||
NOTIF_VIDEO_CHAT_START = 137
|
||||
NOTIF_VIDEO_CHAT_COMMAND = 138
|
||||
NOTIF_CALLBACK_ANSWER = 143
|
||||
NOTIF_MSG_CONSTRUCT = 146
|
||||
NOTIF_LOCATION = 147
|
||||
NOTIF_LOCATION_REQUEST = 148
|
||||
NOTIF_ASSETS_UPDATE = 150
|
||||
NOTIF_MSG_REACTIONS_CHANGED = 155
|
||||
NOTIF_MSG_YOU_REACTED = 156
|
||||
NOTIF_DRAFT = 152
|
||||
NOTIF_DRAFT_DISCARD = 153
|
||||
NOTIF_MSG_DELAYED = 154
|
||||
AUTH_CALL_INFO = 256
|
||||
CONTACT_INFO_EXTERNAL = 45
|
||||
DRAFT_SAVE = 176
|
||||
DRAFT_DISCARD = 177
|
||||
STICKER_CREATE = 193
|
||||
STICKER_SUGGEST = 194
|
||||
CHAT_SEARCH_COUNT_MSG = 197
|
||||
CHAT_SEARCH_COMMON_PARTICIPANTS = 198
|
||||
GET_USER_SCORE = 201
|
||||
|
|
@ -1,66 +0,0 @@
|
|||
import asyncio, logging, json
|
||||
from websockets.asyncio.server import serve
|
||||
from tamtam_ws.models import *
|
||||
from pydantic import ValidationError
|
||||
from tamtam_ws.proto import Proto
|
||||
from tamtam_ws.processors import Processors
|
||||
|
||||
class TTWSServer:
|
||||
def __init__(self, host, port, db_pool=None, clients={}, send_event=None, origins=None):
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.proto = Proto()
|
||||
self.processors = Processors(db_pool=db_pool, clients=clients, send_event=send_event)
|
||||
self.logger = logging.getLogger(__name__)
|
||||
self.origins = origins
|
||||
|
||||
async def handle_client(self, websocket):
|
||||
deviceType = None
|
||||
deviceName = None
|
||||
|
||||
async for message in websocket:
|
||||
# Распаковываем пакет
|
||||
packet = self.proto.unpack_packet(message)
|
||||
|
||||
if not packet:
|
||||
self.logger.warning("Невалидный пакет от ws клиента")
|
||||
continue
|
||||
|
||||
# Валидируем структуру пакета
|
||||
try:
|
||||
MessageModel.model_validate(packet)
|
||||
except ValidationError as e:
|
||||
self.logger.warning(f"Ошибка валидации пакета: {e}")
|
||||
continue
|
||||
|
||||
# Извлекаем данные из пакета
|
||||
seq = packet['seq']
|
||||
opcode = packet['opcode']
|
||||
payload = packet['payload']
|
||||
|
||||
match opcode:
|
||||
case self.proto.SESSION_INIT:
|
||||
# ПРИВЕТ АНДРЕЙ МАЛАХОВ
|
||||
# не не удаляй этот коммент. пусть останется на релизе аххахаха
|
||||
deviceType, deviceName = await self.processors.process_hello(payload, seq, websocket)
|
||||
case self.proto.PING:
|
||||
await self.processors.process_ping(payload, seq, websocket)
|
||||
case self.proto.LOG:
|
||||
# телеметрия аааа слежка цру фсб фбр
|
||||
# УДАЛЯЕМ MYTRACKER ИЗ TAMTAM ТАМ ВИРУС
|
||||
# майтрекер отправляет все ваши сообщения на сервер барака обамы. немедленно удаляем!!!
|
||||
await self.processors.process_telemetry(payload, seq, websocket)
|
||||
|
||||
# лан я пойду. пока
|
||||
# а ок
|
||||
|
||||
async def start(self):
|
||||
self.logger.info(f"Вебсокет запущен на порту {self.port}")
|
||||
|
||||
async with serve(
|
||||
self.handle_client, self.host, self.port,
|
||||
max_size=65536,
|
||||
open_timeout=10,
|
||||
close_timeout=10,
|
||||
):
|
||||
await asyncio.Future()
|
||||
Loading…
Reference in New Issue