/
white.sheikh
/
mangoszero-server
Обзор
Документация
Войти
/
white.sheikh
/
mangoszero-server
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
src/game/Server/WorldSocket.cpp
955 строк
25 KB
Niam5
[Eluna] Update Eluna to latest and reenable Eluna tests (#204)
30 авг 2024, 03:01
Не верифицирован
30 авг 2024, 03:01
e64de89
Код
Авторство
О чём код?
/** * MaNGOS is a full featured server for World of Warcraft, supporting * the following clients: 1.12.x, 2.4.3, 3.3.5a, 4.3.4a and 5.4.8 * * Copyright (C) 2005-2023 MaNGOS <https://getmangos.eu> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * * World of Warcraft, and all World of Warcraft or Warcraft art, images, * and lore are copyrighted by Blizzard Entertainment, Inc. */ #include <ace/Message_Block.h> #include <ace/OS_NS_string.h> #include <ace/OS_NS_unistd.h> #include <ace/os_include/arpa/os_inet.h> #include <ace/os_include/netinet/os_tcp.h> #include <ace/os_include/sys/os_types.h> #include <ace/os_include/sys/os_socket.h> #include <ace/OS_NS_string.h> #include <ace/Reactor.h> #include <ace/Auto_Ptr.h> #include "WorldSocket.h" #include "Common.h" #include "Util.h" #include "World.h" #include "WorldPacket.h" #include "SharedDefines.h" #include "ByteBuffer.h" #include "AddonHandler.h" #include "Opcodes.h" #include "Database/DatabaseEnv.h" #include "Auth/BigNumber.h" #include "Auth/Sha1.h" #include "WorldSession.h" #include "WorldSocketMgr.h" #include "Log.h" #include "DBCStores.h" #ifdef ENABLE_ELUNA #include "LuaEngine.h" #endif /* ENABLE_ELUNA */ #if defined( __GNUC__ ) #pragma pack(1) #else #pragma pack(push,1) #endif struct ServerPktHeader { uint16 size; uint16 cmd; }; struct ClientPktHeader { uint16 size; uint32 cmd; }; #if defined( __GNUC__ ) #pragma pack() #else #pragma pack(pop) #endif WorldSocket::WorldSocket(void) : WorldHandler(), m_LastPingTime(ACE_Time_Value::zero), m_OverSpeedPings(0), m_Session(0), m_RecvWPct(0), m_RecvPct(), m_Header(sizeof(ClientPktHeader)), m_OutBufferLock(), m_OutBuffer(0), m_OutBufferSize(65536), m_Seed(rand32()) { reference_counting_policy().value(ACE_Event_Handler::Reference_Counting_Policy::ENABLED); } WorldSocket::~WorldSocket(void) { delete m_RecvWPct; if (m_OutBuffer) { m_OutBuffer->release(); } closing_ = true; peer().close(); WorldPacket* pct; while (m_PacketQueue.dequeue_head(pct) == 0) { delete pct; } } bool WorldSocket::IsClosed(void) const { return closing_; } void WorldSocket::CloseSocket(void) { ACE_GUARD(LockType, Guard, m_OutBufferLock); if (closing_) { return; } closing_ = true; peer().close_writer(); m_Session = NULL; } const std::string& WorldSocket::GetRemoteAddress(void) const { return m_Address; } int WorldSocket::SendPacket(const WorldPacket& pkt) { ACE_GUARD_RETURN(LockType, Guard, m_OutBufferLock, -1); if (closing_) { return -1; } WorldPacket pct = pkt; if (iSendPacket(pct) == -1) { WorldPacket* npct; ACE_NEW_RETURN(npct, WorldPacket(pct), -1); // NOTE maybe check of the size of the queue can be good ? // to make it bounded instead of unbounded if (m_PacketQueue.enqueue_tail(npct) == -1) { delete npct; sLog.outError("WorldSocket::SendPacket: m_PacketQueue.enqueue_tail failed"); return -1; } } if (reactor()->schedule_wakeup(this, ACE_Event_Handler::WRITE_MASK) == -1) { sLog.outError("SendPacket failed setting WRITE mask, peer = %s", GetRemoteAddress().c_str()); return -1; } return 0; } long WorldSocket::AddReference(void) { return static_cast<long>(add_reference()); } long WorldSocket::RemoveReference(void) { return static_cast<long>(remove_reference()); } int WorldSocket::open(void* a) { ACE_UNUSED_ARG(a); // Prevent double call to this func. if (m_OutBuffer) { return -1; } // Hook for the manager. if (sWorldSocketMgr->OnSocketOpen(this) == -1) { return -1; } // Allocate the buffer. ACE_NEW_RETURN(m_OutBuffer, ACE_Message_Block(m_OutBufferSize), -1); // Store peer address. ACE_INET_Addr remote_addr; if (peer().get_remote_addr(remote_addr) == -1) { sLog.outError("WorldSocket::open: peer ().get_remote_addr errno = %s", ACE_OS::strerror(errno)); return -1; } m_Address = remote_addr.get_host_addr(); // Register with ACE Reactor if (reactor()->register_handler(this, ACE_Event_Handler::READ_MASK) == -1) { sLog.outError("WorldSocket::open: unable to register client handler errno = %s", ACE_OS::strerror(errno)); return -1; } // reactor takes care of the socket from now on remove_reference(); // Send startup packet. WorldPacket packet(SMSG_AUTH_CHALLENGE, 4); packet << m_Seed; return SendPacket(packet); } int WorldSocket::close(u_long) { shutdown(); closing_ = true; remove_reference(); return 0; } int WorldSocket::handle_input(ACE_HANDLE) { if (closing_) { return -1; } switch (handle_input_missing_data()) { case -1 : { if ((errno == EWOULDBLOCK) || (errno == EAGAIN)) { return 0; } } case 0: { errno = ECONNRESET; return -1; } default: return 0; } ACE_NOTREACHED(return -1); } int WorldSocket::handle_output(ACE_HANDLE) { ACE_GUARD_RETURN(LockType, Guard, m_OutBufferLock, -1); if (closing_) { return -1; } const size_t send_len = m_OutBuffer->length(); if (send_len == 0) { reactor()->cancel_wakeup(this, ACE_Event_Handler::WRITE_MASK); return 0; } #ifdef MSG_NOSIGNAL ssize_t n = peer().send(m_OutBuffer->rd_ptr(), send_len, MSG_NOSIGNAL); #else ssize_t n = peer().send(m_OutBuffer->rd_ptr(), send_len); #endif // MSG_NOSIGNAL if (n == 0) { return -1; } else if (n == -1) { if (errno == EWOULDBLOCK || errno == EAGAIN) { return 0; } return -1; } else if (n < (ssize_t)send_len) // now n > 0 { m_OutBuffer->rd_ptr(static_cast<size_t>(n)); // move the data to the base of the buffer m_OutBuffer->crunch(); return 0; } else // now n == send_len { m_OutBuffer->reset(); if(!iFlushPacketQueue()) //no more packets in queue { reactor()->cancel_wakeup(this, ACE_Event_Handler::WRITE_MASK); } return 0; } ACE_NOTREACHED(return 0); } int WorldSocket::handle_close(ACE_HANDLE h, ACE_Reactor_Mask) { { ACE_GUARD_RETURN(LockType, Guard, m_OutBufferLock, -1); closing_ = true; if (h == ACE_INVALID_HANDLE) { peer().close_writer(); } } m_Session = NULL; reactor()->remove_handler(this, ACE_Event_Handler::DONT_CALL | ACE_Event_Handler::ALL_EVENTS_MASK); return 0; } int WorldSocket::handle_input_header(void) { MANGOS_ASSERT(m_RecvWPct == NULL); MANGOS_ASSERT(m_Header.length() == sizeof(ClientPktHeader)); m_Crypt.DecryptRecv((uint8*) m_Header.rd_ptr(), sizeof(ClientPktHeader)); ClientPktHeader& header = *((ClientPktHeader*) m_Header.rd_ptr()); EndianConvertReverse(header.size); EndianConvert(header.cmd); if ((header.size < 4) || (header.size > 10240) || (header.cmd > 10240)) { sLog.outError("WorldSocket::handle_input_header: client sent malformed packet size = %d , cmd = %d", header.size, header.cmd); errno = EINVAL; return -1; } header.size -= 4; ACE_NEW_RETURN(m_RecvWPct, WorldPacket((uint16)header.cmd, header.size), -1); if (header.size > 0) { m_RecvWPct->resize(header.size); m_RecvPct.base((char*) m_RecvWPct->contents(), m_RecvWPct->size()); } else { MANGOS_ASSERT(m_RecvPct.space() == 0); } return 0; } int WorldSocket::handle_input_payload(void) { // set errno properly here on error !!! // now have a header and payload MANGOS_ASSERT(m_RecvPct.space() == 0); MANGOS_ASSERT(m_Header.space() == 0); MANGOS_ASSERT(m_RecvWPct != NULL); const int ret = ProcessIncoming(m_RecvWPct); m_RecvPct.base(NULL, 0); m_RecvPct.reset(); m_RecvWPct = NULL; m_Header.reset(); if (ret == -1) { errno = EINVAL; } return ret; } int WorldSocket::handle_input_missing_data(void) { char buf [4096]; ACE_Data_Block db(sizeof(buf), ACE_Message_Block::MB_DATA, buf, 0, 0, ACE_Message_Block::DONT_DELETE, 0); ACE_Message_Block message_block(&db, ACE_Message_Block::DONT_DELETE, 0); const size_t recv_size = message_block.space(); const ssize_t n = peer().recv(message_block.wr_ptr(), recv_size); if (n <= 0) { return (int)n; } message_block.wr_ptr(n); while (message_block.length() > 0) { if (m_Header.space() > 0) { // need to receive the header const size_t to_header = (message_block.length() > m_Header.space() ? m_Header.space() : message_block.length()); m_Header.copy(message_block.rd_ptr(), to_header); message_block.rd_ptr(to_header); if (m_Header.space() > 0) { // Couldn't receive the whole header this time. MANGOS_ASSERT(message_block.length() == 0); errno = EWOULDBLOCK; return -1; } // We just received nice new header if (handle_input_header() == -1) { MANGOS_ASSERT((errno != EWOULDBLOCK) && (errno != EAGAIN)); return -1; } } // Its possible on some error situations that this happens // for example on closing when epoll receives more chunked data and stuff // hope this is not hack ,as proper m_RecvWPct is asserted around if (!m_RecvWPct) { sLog.outError("Forcing close on input m_RecvWPct = NULL"); errno = EINVAL; return -1; } // We have full read header, now check the data payload if (m_RecvPct.space() > 0) { // need more data in the payload const size_t to_data = (message_block.length() > m_RecvPct.space() ? m_RecvPct.space() : message_block.length()); m_RecvPct.copy(message_block.rd_ptr(), to_data); message_block.rd_ptr(to_data); if (m_RecvPct.space() > 0) { // Couldn't receive the whole data this time. MANGOS_ASSERT(message_block.length() == 0); errno = EWOULDBLOCK; return -1; } } // just received fresh new payload if (handle_input_payload() == -1) { MANGOS_ASSERT((errno != EWOULDBLOCK) && (errno != EAGAIN)); return -1; } } return size_t(n) == recv_size ? 1 : 2; } int WorldSocket::ProcessIncoming(WorldPacket* new_pct) { MANGOS_ASSERT(new_pct); // manage memory ;) ACE_Auto_Ptr<WorldPacket> aptr(new_pct); const ACE_UINT16 opcode = new_pct->GetOpcode(); if (opcode >= NUM_MSG_TYPES) { sLog.outError("SESSION: received nonexistent opcode 0x%.4X", opcode); return -1; } if (closing_) { return -1; } // Dump received packet. sLog.outWorldPacketDump(uint32(get_handle()), new_pct->GetOpcode(), new_pct->GetOpcodeName(), new_pct, true); try { switch (opcode) { case CMSG_PING: return HandlePing(*new_pct); case CMSG_AUTH_SESSION: if (m_Session) { sLog.outError("WorldSocket::ProcessIncoming: Player send CMSG_AUTH_SESSION again"); return -1; } #ifdef ENABLE_ELUNA if (Eluna* e = sWorld.GetEluna()) { if (!e->OnPacketReceive(m_Session, *new_pct)) { return 0; } } #endif /* ENABLE_ELUNA */ return HandleAuthSession(*new_pct); case CMSG_KEEP_ALIVE: DEBUG_LOG("CMSG_KEEP_ALIVE ,size: %zu ", new_pct->size()); #ifdef ENABLE_ELUNA if (Eluna* e = sWorld.GetEluna()) { e->OnPacketReceive(m_Session, *new_pct); } #endif /* ENABLE_ELUNA */ return 0; default: { if (m_Session != NULL) { // OK ,give the packet to WorldSession aptr.release(); m_Session->QueuePacket(new_pct); return 0; } else { sLog.outError("WorldSocket::ProcessIncoming: Client not authed opcode = %u", uint32(opcode)); return -1; } } } } catch (ByteBufferException&) { sLog.outError("WorldSocket::ProcessIncoming ByteBufferException occured while parsing an instant handled packet (opcode: %u) from client %s, accountid=%i.", opcode, GetRemoteAddress().c_str(), m_Session ? m_Session->GetAccountId() : -1); if (sLog.HasLogLevelOrHigher(LOG_LVL_DEBUG)) { DEBUG_LOG("Dumping error-causing packet:"); new_pct->hexlike(); } if (sWorld.getConfig(CONFIG_BOOL_KICK_PLAYER_ON_BAD_PACKET)) { DETAIL_LOG("Disconnecting session [account id %i / address %s] for badly formatted packet.", m_Session ? m_Session->GetAccountId() : -1, GetRemoteAddress().c_str()); return -1; } else { return 0; } } ACE_NOTREACHED(return 0); } int WorldSocket::HandleAuthSession(WorldPacket& recvPacket) { // NOTE: ATM the socket is singlethread, have this in mind ... uint8 digest[SHA_DIGEST_LENGTH]; uint32 clientSeed; uint32 unk2; uint32 BuiltNumberClient; std::string account; BigNumber v, s, g, N, K; const bool wardenActive = (sWorld.getConfig(CONFIG_BOOL_WARDEN_WIN_ENABLED) || sWorld.getConfig(CONFIG_BOOL_WARDEN_OSX_ENABLED)); // Read the content of the packet recvPacket >> BuiltNumberClient; recvPacket >> unk2; recvPacket >> account; recvPacket >> clientSeed; recvPacket.read(digest, SHA_DIGEST_LENGTH); DEBUG_LOG("WorldSocket::HandleAuthSession: client %u, unk2 %u, account %s, clientseed %u", BuiltNumberClient, unk2, account.c_str(), clientSeed); // Check the version of client trying to connect if (!IsAcceptableClientBuild(BuiltNumberClient)) { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_VERSION_MISMATCH); SendPacket(packet); sLog.outError("WorldSocket::HandleAuthSession: Sent Auth Response (version mismatch)."); return -1; } // Get the account information from the realmd database std::string safe_account = account; // Duplicate, else will screw the SHA hash verification below LoginDatabase.escape_string(safe_account); // No SQL injection, username escaped. QueryResult* result = LoginDatabase.PQuery("SELECT " "`id`, " // 0 "`gmlevel`, " // 1 "`sessionkey`, " // 2 "`last_ip`, " // 3 "`locked`, " // 4 "`v`, " // 5 "`s`, " // 6 "`mutetime`, " // 7 "`locale`, " // 8 "`os` " // 9 "FROM `account` " "WHERE `username` = '%s'", safe_account.c_str()); // Stop if the account is not found if (!result) { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_UNKNOWN_ACCOUNT); SendPacket(packet); sLog.outError("WorldSocket::HandleAuthSession: Sent Auth Response (unknown account)."); return -1; } const Field* fields = result->Fetch(); N.SetHexStr("894B645E89E1535BBDAD5B8B290650530801B18EBFBF5E8FAB3C82872A3E9BB7"); g.SetDword(7); v.SetHexStr(fields[5].GetString()); s.SetHexStr(fields[6].GetString()); const char* sStr = s.AsHexStr(); // Must be freed by OPENSSL_free() const char* vStr = v.AsHexStr(); // Must be freed by OPENSSL_free() DEBUG_LOG("WorldSocket::HandleAuthSession: (s,v) check s: %s v: %s", sStr, vStr); OPENSSL_free((void*) sStr); OPENSSL_free((void*) vStr); ///- Re-check ip locking (same check as in realmd). if (fields[4].GetBool()) { if (strcmp(fields[3].GetString(), GetRemoteAddress().c_str())) { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_FAILED); SendPacket(packet); delete result; BASIC_LOG("WorldSocket::HandleAuthSession: Sent Auth Response (Account IP differs)."); return -1; } } uint32 id = fields[0].GetUInt32(); uint32 security = fields[1].GetUInt16(); if (security > SEC_ADMINISTRATOR) // prevent invalid security settings in DB { security = SEC_ADMINISTRATOR; } K.SetHexStr(fields[2].GetString()); time_t mutetime = time_t (fields[7].GetUInt64()); uint8 tmpLoc = fields[8].GetUInt8(); LocaleConstant locale = tmpLoc >= MAX_LOCALE ? LOCALE_enUS : LocaleConstant(tmpLoc); std::string os = fields[9].GetString(); delete result; // Re-check account ban (same check as in realmd) QueryResult* banresult = LoginDatabase.PQuery("SELECT 1 FROM `account_banned` WHERE `id` = %u AND `active` = 1 AND (`unbandate` > UNIX_TIMESTAMP() OR `unbandate` = `bandate`)" "UNION " "SELECT 1 FROM `ip_banned` WHERE (`unbandate` = `bandate` OR `unbandate` > UNIX_TIMESTAMP()) AND `ip` = '%s'", id, GetRemoteAddress().c_str()); if (banresult) // if account banned { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_BANNED); SendPacket(packet); delete banresult; sLog.outError("WorldSocket::HandleAuthSession: Sent Auth Response (Account banned)."); return -1; } // Check locked state for server AccountTypes allowedAccountType = sWorld.GetPlayerSecurityLimit(); if (allowedAccountType > SEC_PLAYER && AccountTypes(security) < allowedAccountType) { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_UNAVAILABLE); SendPacket(packet); BASIC_LOG("WorldSocket::HandleAuthSession: User tries to login but his security level is not enough"); return -1; } // Must be done before WorldSession is created if (wardenActive && os != "Win" && os != "OSX") { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_REJECT); SendPacket(packet); BASIC_LOG("WorldSocket::HandleAuthSession: Client %s attempted to log in using invalid client OS (%s).", GetRemoteAddress().c_str(), os.c_str()); return -1; } // Check that Key and account name are the same on client and server uint8 t[4]{ 0 }; uint32 seed = m_Seed; Sha1Hash sha; sha.UpdateData(account); sha.UpdateData((uint8*) & t, 4); sha.UpdateData((uint8*) & clientSeed, 4); sha.UpdateData((uint8*) & seed, 4); sha.UpdateBigNumbers(&K, nullptr); sha.Finalize(); if (memcmp(sha.GetDigest(), digest, SHA_DIGEST_LENGTH)) { WorldPacket packet(SMSG_AUTH_RESPONSE, 1); packet << uint8(AUTH_FAILED); SendPacket(packet); sLog.outError("WorldSocket::HandleAuthSession: Sent Auth Response (authentification failed)."); return -1; } std::string address = GetRemoteAddress(); DEBUG_LOG("WorldSocket::HandleAuthSession: Client '%s' authenticated successfully from %s.", account.c_str(), address.c_str()); // Update the last_ip in the database // No SQL injection, username escaped. static SqlStatementID updAccount; SqlStatement stmt = LoginDatabase.CreateStatement(updAccount, "UPDATE `account` SET `last_ip` = ? WHERE `username` = ?"); stmt.PExecute(address.c_str(), account.c_str()); // NOTE ATM the socket is single-threaded, have this in mind ... ACE_NEW_RETURN(m_Session, WorldSession(id, this, AccountTypes(security), mutetime, locale), -1); m_Crypt.SetKey(K.AsByteArray(), 40); m_Crypt.Init(); m_Session->LoadTutorialsData(); // In case needed sometime the second arg is in microseconds 1 000 000 = 1 sec ACE_OS::sleep(ACE_Time_Value(0, 10000)); // Initialize Warden system only if it is enabled by config if (wardenActive) { m_Session->InitWarden(uint16(BuiltNumberClient), &K, os); } sWorld.AddSession(m_Session); // Create and send the Addon packet WorldPacket SendAddonPacked; if (sAddOnHandler.BuildAddonPacket(&recvPacket, &SendAddonPacked)) { SendPacket(SendAddonPacked); } return 0; } int WorldSocket::HandlePing(WorldPacket& recvPacket) { uint32 ping; uint32 latency; // Get the ping packet content recvPacket >> ping; recvPacket >> latency; if (m_LastPingTime == ACE_Time_Value::zero) { m_LastPingTime = ACE_OS::gettimeofday(); // for 1st ping } else { ACE_Time_Value cur_time = ACE_OS::gettimeofday(); ACE_Time_Value diff_time(cur_time); diff_time -= m_LastPingTime; m_LastPingTime = cur_time; if (diff_time < ACE_Time_Value(27)) { ++m_OverSpeedPings; uint32 max_count = sWorld.getConfig(CONFIG_UINT32_MAX_OVERSPEED_PINGS); if (max_count && m_OverSpeedPings > max_count) { if (m_Session && m_Session->GetSecurity() == SEC_PLAYER) { sLog.outError("WorldSocket::HandlePing: Player kicked for " "overspeeded pings address = %s", GetRemoteAddress().c_str()); return -1; } } } else { m_OverSpeedPings = 0; } } if (m_Session) { m_Session->SetLatency(latency); m_Session->SetClientTimeDelay(0); // recalculated on next movement packet } else { sLog.outError("WorldSocket::HandlePing: peer sent CMSG_PING, " "but is not authenticated or got recently kicked," " address = %s", GetRemoteAddress().c_str()); return -1; } WorldPacket packet(SMSG_PONG, 4); packet << ping; return SendPacket(packet); } int WorldSocket::iSendPacket(const WorldPacket& pct) { if (m_OutBuffer->space() < pct.size() + sizeof(ServerPktHeader)) { errno = ENOBUFS; return -1; } ServerPktHeader header; header.cmd = pct.GetOpcode(); header.size = (uint16) pct.size() + 2; EndianConvertReverse(header.size); EndianConvert(header.cmd); m_Crypt.EncryptSend((uint8*) & header, sizeof(header)); if (m_OutBuffer->copy((char*) & header, sizeof(header)) == -1) { ACE_ASSERT(false); } if (!pct.empty()) if (m_OutBuffer->copy((char*) pct.contents(), pct.size()) == -1) { ACE_ASSERT(false); } return 0; } bool WorldSocket::iFlushPacketQueue() { WorldPacket* pct; bool haveone = false; while (m_PacketQueue.dequeue_head(pct) == 0) { if (iSendPacket(*pct) == -1) { if (m_PacketQueue.enqueue_head(pct) == -1) { delete pct; sLog.outError("WorldSocket::iFlushPacketQueue m_PacketQueue->enqueue_head"); return false; } break; } else { haveone = true; delete pct; } } return haveone; }