From 4760d91ccd6be2f87b89745539e1ad43c71f4685 Mon Sep 17 00:00:00 2001 From: Kamil Date: Tue, 17 Nov 2020 01:00:10 +0100 Subject: [PATCH] refactored giant switch to smaller functions --- src/CNLoginServer.cpp | 667 +++++++++++++++++++++++------------------- src/CNLoginServer.hpp | 11 + 2 files changed, 370 insertions(+), 308 deletions(-) diff --git a/src/CNLoginServer.cpp b/src/CNLoginServer.cpp index 710a749..82543e7 100644 --- a/src/CNLoginServer.cpp +++ b/src/CNLoginServer.cpp @@ -21,123 +21,7 @@ void CNLoginServer::handlePacket(CNSocket* sock, CNPacketData* data) { switch (data->type) { case P_CL2LS_REQ_LOGIN: { - if (data->size != sizeof(sP_CL2LS_REQ_LOGIN)) - return; // ignore the malformed packet - - sP_CL2LS_REQ_LOGIN* login = (sP_CL2LS_REQ_LOGIN*)data->buf; - // TODO: implement better way of sending credentials - std::string userLogin((char*)login->szCookie_TEGid); - std::string userPassword((char*)login->szCookie_authid); - - /* - * Sometimes the client sends garbage cookie data. - * Validate it as normal credentials instead of using a length check before falling back. - */ - if (!CNLoginServer::isLoginDataGood(userLogin, userPassword)) { - /* - * The std::string -> char* -> std::string maneuver should remove any - * trailing garbage after the null terminator. - */ - userLogin = std::string(U16toU8(login->szID).c_str()); - userPassword = std::string(U16toU8(login->szPassword).c_str()); - } - - bool success = false; - int errorCode = 0; - - // checking regex - if (!CNLoginServer::isLoginDataGood(userLogin, userPassword)) { - errorCode = (int)LoginError::LOGIN_ERROR; - } else { - std::unique_ptr findUser = Database::findAccount(userLogin); - // if account not found, make new one - if (findUser == nullptr) { - loginSessions[sock] = CNLoginData(); - loginSessions[sock].userID = Database::addAccount(userLogin, userPassword); - loginSessions[sock].slot = 1; - loginSessions[sock].lastHeartbeat = getTime(); - success = true; - - // if user exists, check if password is correct - } else if (CNLoginServer::isPasswordCorrect(findUser->Password, userPassword)) { - /*calling this here to timestamp login attempt, - * in order to make duplicate exit sanity check work*/ - Database::updateSelected(findUser->AccountID, findUser->Selected); - // check if account isn't currently in use - if (CNLoginServer::isAccountInUse(findUser->AccountID)) { - errorCode = (int)LoginError::ID_ALREADY_IN_USE; - } else { // if not, login success - loginSessions[sock] = CNLoginData(); - loginSessions[sock].userID = findUser->AccountID; - loginSessions[sock].slot = findUser->Selected; - loginSessions[sock].lastHeartbeat = getTime(); - success = true; - } - } else { - errorCode = (int)LoginError::ID_AND_PASSWORD_DO_NOT_MATCH; - } - } - - - if (success) { - std::vector characters = Database::getCharacters(loginSessions[sock].userID); - int charCount = characters.size(); - - INITSTRUCT(sP_LS2CL_REP_LOGIN_SUCC, resp); - // set username in resp packet - memcpy(resp.szID, login->szID, sizeof(char16_t) * 33); - - resp.iCharCount = charCount; - resp.iSlotNum = loginSessions[sock].slot; - resp.iPaymentFlag = 1; - resp.iOpenBetaFlag = 0; - resp.uiSvrTime = getTime(); - - // send the resp in with original key - sock->sendPacket((void*)&resp, P_LS2CL_REP_LOGIN_SUCC, sizeof(sP_LS2CL_REP_LOGIN_SUCC)); - - // update keys - sock->setEKey(CNSocketEncryption::createNewKey(resp.uiSvrTime, resp.iCharCount + 1, resp.iSlotNum + 1)); - sock->setFEKey(CNSocketEncryption::createNewKey((uint64_t)(*(uint64_t*)&CNSocketEncryption::defaultKey[0]), login->iClientVerC, 1)); - - // now send the characters :) - std::vector::iterator it; - for (it = characters.begin(); it != characters.end(); it++) { - sP_LS2CL_REP_CHAR_INFO charInfo = sP_LS2CL_REP_CHAR_INFO(); - - charInfo.iSlot = (int8_t)it->slot; - charInfo.iLevel = (int16_t)it->level; - charInfo.sPC_Style = it->PCStyle; - charInfo.sPC_Style2 = it->PCStyle2; - - // position - charInfo.iX = it->x; - charInfo.iY = it->y; - charInfo.iZ = it->z; - - // save character in session (for char select) - int UID = it->iID; - loginSessions[sock].characters[UID] = Player(*it); - loginSessions[sock].characters[UID].FEKey = sock->getFEKey(); - - // Equip info - for (int i = 0; i < AEQUIP_COUNT; i++) { - charInfo.aEquip[i] = it->Equip[i]; - } - - // set default to the first character - if (it == characters.begin()) - loginSessions[sock].selectedChar = UID; - - sock->sendPacket((void*)&charInfo, P_LS2CL_REP_CHAR_INFO, sizeof(sP_LS2CL_REP_CHAR_INFO)); - } - } else { - INITSTRUCT(sP_LS2CL_REP_LOGIN_FAIL, resp); - U8toU16(userLogin, resp.szID, sizeof(resp.szID)); - resp.iErrorCode = errorCode; - sock->sendPacket((void*)&resp, P_LS2CL_REP_LOGIN_FAIL, sizeof(sP_LS2CL_REP_LOGIN_FAIL)); - } - + login(sock, data); break; } case P_CL2LS_REP_LIVE_CHECK: { @@ -145,220 +29,39 @@ void CNLoginServer::handlePacket(CNSocket* sock, CNPacketData* data) { break; } case P_CL2LS_REQ_CHECK_CHAR_NAME: { - if (data->size != sizeof(sP_CL2LS_REQ_CHECK_CHAR_NAME)) - return; - - sP_CL2LS_REQ_CHECK_CHAR_NAME* nameCheck = (sP_CL2LS_REQ_CHECK_CHAR_NAME*)data->buf; - bool success = true; - int errorcode = 0; - - // check regex - if (!CNLoginServer::isCharacterNameGood(U16toU8(nameCheck->szFirstName), U16toU8(nameCheck->szLastName))) { - success = false; - errorcode = 4; - } else if (!Database::isNameFree(nameCheck)){ // check if name isn't already occupied - success = false; - errorcode = 1; - } - - loginSessions[sock].lastHeartbeat = getTime(); - - if (success) { - INITSTRUCT(sP_LS2CL_REP_CHECK_CHAR_NAME_SUCC, resp); - - DEBUGLOG( - std::cout << "P_CL2LS_REQ_CHECK_CHAR_NAME:" << std::endl; - std::cout << "\tFirstName: " << U16toU8(nameCheck->szFirstName) << " LastName: " << U16toU8(nameCheck->szLastName) << std::endl; - ) - - memcpy(resp.szFirstName, nameCheck->szFirstName, sizeof(char16_t) * 9); - memcpy(resp.szLastName, nameCheck->szLastName, sizeof(char16_t) * 17); - - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHECK_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_CHECK_CHAR_NAME_SUCC)); - } else { - INITSTRUCT(sP_LS2CL_REP_CHECK_CHAR_NAME_FAIL, resp); - resp.iErrorCode = errorcode; - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHECK_CHAR_NAME_FAIL, sizeof(sP_LS2CL_REP_CHECK_CHAR_NAME_FAIL)); - } + nameCheck(sock, data); break; } case P_CL2LS_REQ_SAVE_CHAR_NAME: { - if (data->size != sizeof(sP_CL2LS_REQ_SAVE_CHAR_NAME)) - return; - - sP_CL2LS_REQ_SAVE_CHAR_NAME* save = (sP_CL2LS_REQ_SAVE_CHAR_NAME*)data->buf; - INITSTRUCT(sP_LS2CL_REP_SAVE_CHAR_NAME_SUCC, resp); - - DEBUGLOG( - std::cout << "P_CL2LS_REQ_SAVE_CHAR_NAME:" << std::endl; - std::cout << "\tSlot: " << (int)save->iSlotNum << std::endl; - std::cout << "\tName: " << U16toU8(save->szFirstName) << " " << U16toU8(save->szLastName) << std::endl; - ) - - resp.iSlotNum = save->iSlotNum; - resp.iGender = save->iGender; - resp.iPC_UID = Database::createCharacter(save, loginSessions[sock].userID); - memcpy(resp.szFirstName, save->szFirstName, sizeof(char16_t) * 9); - memcpy(resp.szLastName, save->szLastName, sizeof(char16_t) * 17); - - loginSessions[sock].lastHeartbeat = getTime(); - - sock->sendPacket((void*)&resp, P_LS2CL_REP_SAVE_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_SAVE_CHAR_NAME_SUCC)); - - Database::updateSelected(loginSessions[sock].userID, save->iSlotNum); + nameSave(sock, data); break; } case P_CL2LS_REQ_CHAR_CREATE: { - if (data->size != sizeof(sP_CL2LS_REQ_CHAR_CREATE)) - return; - - sP_CL2LS_REQ_CHAR_CREATE* character = (sP_CL2LS_REQ_CHAR_CREATE*)data->buf; - Database::finishCharacter(character); - - DEBUGLOG( - std::cout << "P_CL2LS_REQ_CHAR_CREATE:" << std::endl; - std::cout << "\tPC_UID: " << character->PCStyle.iPC_UID << std::endl; - std::cout << "\tNameCheck: " << (int)character->PCStyle.iNameCheck << std::endl; - std::cout << "\tName: " << U16toU8(character->PCStyle.szFirstName) << " " << U16toU8(character->PCStyle.szLastName) << std::endl; - std::cout << "\tGender: " << (int)character->PCStyle.iGender << std::endl; - std::cout << "\tFace: " << (int)character->PCStyle.iFaceStyle << std::endl; - std::cout << "\tHair: " << (int)character->PCStyle.iHairStyle << std::endl; - std::cout << "\tHair Color: " << (int)character->PCStyle.iHairColor << std::endl; - std::cout << "\tSkin Color: " << (int)character->PCStyle.iSkinColor << std::endl; - std::cout << "\tEye Color: " << (int)character->PCStyle.iEyeColor << std::endl; - std::cout << "\tHeight: " << (int)character->PCStyle.iHeight << std::endl; - std::cout << "\tBody: " << (int)character->PCStyle.iBody << std::endl; - std::cout << "\tClass: " << (int)character->PCStyle.iClass << std::endl; - std::cout << "\tiEquipUBID: " << (int)character->sOn_Item.iEquipUBID << std::endl; - std::cout << "\tiEquipLBID: " << (int)character->sOn_Item.iEquipLBID << std::endl; - std::cout << "\tiEquipFootID: " << (int)character->sOn_Item.iEquipFootID << std::endl; - ) - - Player player = Database::getPlayer(character->PCStyle.iPC_UID); - int64_t UID = player.iID; - - INITSTRUCT(sP_LS2CL_REP_CHAR_CREATE_SUCC, resp); - resp.sPC_Style = player.PCStyle; - resp.sPC_Style2 = player.PCStyle2; - resp.iLevel = player.level; - resp.sOn_Item = character->sOn_Item; - - // save player in session - loginSessions[sock].characters[UID] = Player(player); - loginSessions[sock].characters[UID].FEKey = sock->getFEKey(); - - loginSessions[sock].lastHeartbeat = getTime(); - - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_CREATE_SUCC, sizeof(sP_LS2CL_REP_CHAR_CREATE_SUCC)); - Database::updateSelected(loginSessions[sock].userID, player.slot); + characterCreate(sock, data); break; } case P_CL2LS_REQ_CHAR_DELETE: { - if (data->size != sizeof(sP_CL2LS_REQ_CHAR_DELETE)) - return; - - sP_CL2LS_REQ_CHAR_DELETE* del = (sP_CL2LS_REQ_CHAR_DELETE*)data->buf; - int operationResult = Database::deleteCharacter(del->iPC_UID, loginSessions[sock].userID); - - INITSTRUCT(sP_LS2CL_REP_CHAR_DELETE_SUCC, resp); - resp.iSlotNum = operationResult; - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_DELETE_SUCC, sizeof(sP_LS2CL_REP_CHAR_DELETE_SUCC)); - loginSessions[sock].lastHeartbeat = getTime(); + characterDelete(sock, data); break; } case P_CL2LS_REQ_CHAR_SELECT: { - if (data->size != sizeof(sP_CL2LS_REQ_CHAR_SELECT)) - return; - - // character selected - sP_CL2LS_REQ_CHAR_SELECT* chararacter = (sP_CL2LS_REQ_CHAR_SELECT*)data->buf; - INITSTRUCT(sP_LS2CL_REP_CHAR_SELECT_SUCC, resp); - - DEBUGLOG( - std::cout << "P_CL2LS_REQ_CHAR_SELECT:" << std::endl; - std::cout << "\tPC_UID: " << chararacter->iPC_UID << std::endl; - ) - - loginSessions[sock].lastHeartbeat = getTime(); - - loginSessions[sock].selectedChar = chararacter->iPC_UID; - Database::updateSelected(loginSessions[sock].userID, loginSessions[sock].characters[chararacter->iPC_UID].slot); - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_SELECT_SUCC, sizeof(sP_LS2CL_REP_CHAR_SELECT_SUCC)); + characterSelect(sock, data); break; } case P_CL2LS_REQ_SHARD_SELECT: { - if (data->size != sizeof(sP_CL2LS_REQ_SHARD_SELECT)) - return; - - // tell client to connect to the shard server - sP_CL2LS_REQ_SHARD_SELECT* shard = (sP_CL2LS_REQ_SHARD_SELECT*)data->buf; - INITSTRUCT(sP_LS2CL_REP_SHARD_SELECT_SUCC, resp); - - DEBUGLOG( - std::cout << "P_CL2LS_REQ_SHARD_SELECT:" << std::endl; - std::cout << "\tShard: " << (int)shard->ShardNum << std::endl; - ) - - const char* SHARD_IP = settings::SHARDSERVERIP.c_str(); - resp.iEnterSerialKey = rand(); - resp.g_FE_ServerPort = settings::SHARDPORT; - - // copy IP to resp (this struct uses ASCII encoding so we don't have to goof around converting encodings) - memcpy(resp.g_FE_ServerIP, SHARD_IP, strlen(SHARD_IP)); - resp.g_FE_ServerIP[strlen(SHARD_IP)] = '\0'; - - // pass player to CNSharedData - CNSharedData::setPlayer(resp.iEnterSerialKey, loginSessions[sock].characters[loginSessions[sock].selectedChar]); - - sock->sendPacket((void*)&resp, P_LS2CL_REP_SHARD_SELECT_SUCC, sizeof(sP_LS2CL_REP_SHARD_SELECT_SUCC)); + shardSelect(sock, data); break; } case P_CL2LS_REQ_SAVE_CHAR_TUTOR: { - if (data->size != sizeof(sP_CL2LS_REQ_SAVE_CHAR_TUTOR)) - return; - sP_CL2LS_REQ_SAVE_CHAR_TUTOR* save = (sP_CL2LS_REQ_SAVE_CHAR_TUTOR*)data->buf; - Database::finishTutorial(save->iPC_UID); - // update character in session - auto key = loginSessions[sock].characters[save->iPC_UID].FEKey; - loginSessions[sock].characters[save->iPC_UID] = Player(Database::getPlayer(save->iPC_UID)); - loginSessions[sock].characters[save->iPC_UID].FEKey = key; - - loginSessions[sock].lastHeartbeat = getTime(); - // no response here + finishTutorial(sock, data); break; } case P_CL2LS_REQ_CHANGE_CHAR_NAME: { - if (data->size != sizeof(sP_CL2LS_REQ_CHANGE_CHAR_NAME)) - return; - - sP_CL2LS_REQ_CHANGE_CHAR_NAME* save = (sP_CL2LS_REQ_CHANGE_CHAR_NAME*)data->buf; - Database::changeName(save); - - INITSTRUCT(sP_LS2CL_REP_CHANGE_CHAR_NAME_SUCC, resp); - resp.iPC_UID = save->iPCUID; - memcpy(resp.szFirstName, save->szFirstName, sizeof(char16_t)*9); - memcpy(resp.szLastName, save->szLastName, sizeof(char16_t) * 17); - resp.iSlotNum = save->iSlotNum; - - loginSessions[sock].lastHeartbeat = getTime(); - - sock->sendPacket((void*)&resp, P_LS2CL_REP_CHANGE_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_CHANGE_CHAR_NAME_SUCC)); + changeName(sock, data); break; } case P_CL2LS_REQ_PC_EXIT_DUPLICATE:{ - if (data->size != sizeof(sP_CL2LS_REQ_PC_EXIT_DUPLICATE)) - return; - - sP_CL2LS_REQ_PC_EXIT_DUPLICATE* exit = (sP_CL2LS_REQ_PC_EXIT_DUPLICATE*)data->buf; - auto account = Database::findAccount(U16toU8(exit->szID)); - - // sanity check - if (account == nullptr) { - std::cout << "[WARN] P_CL2LS_REQ_PC_EXIT_DUPLICATE submitted unknown username: " << exit->szID << std::endl; - break; - } - - exitDuplicate(account->AccountID); + duplicateExit(sock, data); break; } default: @@ -374,6 +77,352 @@ void CNLoginServer::handlePacket(CNSocket* sock, CNPacketData* data) { } } +#pragma region packets +void CNLoginServer::login(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_LOGIN)) + return; // ignore the malformed packet + + sP_CL2LS_REQ_LOGIN* login = (sP_CL2LS_REQ_LOGIN*)data->buf; + // TODO: implement better way of sending credentials + std::string userLogin((char*)login->szCookie_TEGid); + std::string userPassword((char*)login->szCookie_authid); + + /* + * Sometimes the client sends garbage cookie data. + * Validate it as normal credentials instead of using a length check before falling back. + */ + if (!CNLoginServer::isLoginDataGood(userLogin, userPassword)) { + /* + * The std::string -> char* -> std::string maneuver should remove any + * trailing garbage after the null terminator. + */ + userLogin = std::string(U16toU8(login->szID).c_str()); + userPassword = std::string(U16toU8(login->szPassword).c_str()); + } + + bool success = false; + int errorCode = 0; + + // checking regex + if (!CNLoginServer::isLoginDataGood(userLogin, userPassword)) { + errorCode = (int)LoginError::LOGIN_ERROR; + } + else { + std::unique_ptr findUser = Database::findAccount(userLogin); + // if account not found, make new one + if (findUser == nullptr) { + loginSessions[sock] = CNLoginData(); + loginSessions[sock].userID = Database::addAccount(userLogin, userPassword); + loginSessions[sock].slot = 1; + loginSessions[sock].lastHeartbeat = getTime(); + success = true; + + // if user exists, check if password is correct + } + else if (CNLoginServer::isPasswordCorrect(findUser->Password, userPassword)) { + /*calling this here to timestamp login attempt, + * in order to make duplicate exit sanity check work*/ + Database::updateSelected(findUser->AccountID, findUser->Selected); + // check if account isn't currently in use + if (CNLoginServer::isAccountInUse(findUser->AccountID)) { + errorCode = (int)LoginError::ID_ALREADY_IN_USE; + } + else { // if not, login success + loginSessions[sock] = CNLoginData(); + loginSessions[sock].userID = findUser->AccountID; + loginSessions[sock].slot = findUser->Selected; + loginSessions[sock].lastHeartbeat = getTime(); + success = true; + } + } + else { + errorCode = (int)LoginError::ID_AND_PASSWORD_DO_NOT_MATCH; + } + } + + + if (success) { + std::vector characters = Database::getCharacters(loginSessions[sock].userID); + int charCount = characters.size(); + + INITSTRUCT(sP_LS2CL_REP_LOGIN_SUCC, resp); + // set username in resp packet + memcpy(resp.szID, login->szID, sizeof(char16_t) * 33); + + resp.iCharCount = charCount; + resp.iSlotNum = loginSessions[sock].slot; + resp.iPaymentFlag = 1; + resp.iOpenBetaFlag = 0; + resp.uiSvrTime = getTime(); + + // send the resp in with original key + sock->sendPacket((void*)&resp, P_LS2CL_REP_LOGIN_SUCC, sizeof(sP_LS2CL_REP_LOGIN_SUCC)); + + // update keys + sock->setEKey(CNSocketEncryption::createNewKey(resp.uiSvrTime, resp.iCharCount + 1, resp.iSlotNum + 1)); + sock->setFEKey(CNSocketEncryption::createNewKey((uint64_t)(*(uint64_t*)&CNSocketEncryption::defaultKey[0]), login->iClientVerC, 1)); + + // now send the characters :) + std::vector::iterator it; + for (it = characters.begin(); it != characters.end(); it++) { + sP_LS2CL_REP_CHAR_INFO charInfo = sP_LS2CL_REP_CHAR_INFO(); + + charInfo.iSlot = (int8_t)it->slot; + charInfo.iLevel = (int16_t)it->level; + charInfo.sPC_Style = it->PCStyle; + charInfo.sPC_Style2 = it->PCStyle2; + + // position + charInfo.iX = it->x; + charInfo.iY = it->y; + charInfo.iZ = it->z; + + // save character in session (for char select) + int UID = it->iID; + loginSessions[sock].characters[UID] = Player(*it); + loginSessions[sock].characters[UID].FEKey = sock->getFEKey(); + + // Equip info + for (int i = 0; i < AEQUIP_COUNT; i++) { + charInfo.aEquip[i] = it->Equip[i]; + } + + // set default to the first character + if (it == characters.begin()) + loginSessions[sock].selectedChar = UID; + + sock->sendPacket((void*)&charInfo, P_LS2CL_REP_CHAR_INFO, sizeof(sP_LS2CL_REP_CHAR_INFO)); + } + } + else { + INITSTRUCT(sP_LS2CL_REP_LOGIN_FAIL, resp); + U8toU16(userLogin, resp.szID, sizeof(resp.szID)); + resp.iErrorCode = errorCode; + sock->sendPacket((void*)&resp, P_LS2CL_REP_LOGIN_FAIL, sizeof(sP_LS2CL_REP_LOGIN_FAIL)); + } +} + +void CNLoginServer::nameCheck(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_CHECK_CHAR_NAME)) + return; + + sP_CL2LS_REQ_CHECK_CHAR_NAME* nameCheck = (sP_CL2LS_REQ_CHECK_CHAR_NAME*)data->buf; + bool success = true; + int errorcode = 0; + + // check regex + if (!CNLoginServer::isCharacterNameGood(U16toU8(nameCheck->szFirstName), U16toU8(nameCheck->szLastName))) { + success = false; + errorcode = 4; + } + else if (!Database::isNameFree(nameCheck)) { // check if name isn't already occupied + success = false; + errorcode = 1; + } + + loginSessions[sock].lastHeartbeat = getTime(); + + if (success) { + INITSTRUCT(sP_LS2CL_REP_CHECK_CHAR_NAME_SUCC, resp); + + DEBUGLOG( + std::cout << "P_CL2LS_REQ_CHECK_CHAR_NAME:" << std::endl; + std::cout << "\tFirstName: " << U16toU8(nameCheck->szFirstName) << " LastName: " << U16toU8(nameCheck->szLastName) << std::endl; + ) + + memcpy(resp.szFirstName, nameCheck->szFirstName, sizeof(char16_t) * 9); + memcpy(resp.szLastName, nameCheck->szLastName, sizeof(char16_t) * 17); + + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHECK_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_CHECK_CHAR_NAME_SUCC)); + } + else { + INITSTRUCT(sP_LS2CL_REP_CHECK_CHAR_NAME_FAIL, resp); + resp.iErrorCode = errorcode; + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHECK_CHAR_NAME_FAIL, sizeof(sP_LS2CL_REP_CHECK_CHAR_NAME_FAIL)); + } +} + +void CNLoginServer::nameSave(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_SAVE_CHAR_NAME)) + return; + + sP_CL2LS_REQ_SAVE_CHAR_NAME* save = (sP_CL2LS_REQ_SAVE_CHAR_NAME*)data->buf; + INITSTRUCT(sP_LS2CL_REP_SAVE_CHAR_NAME_SUCC, resp); + + DEBUGLOG( + std::cout << "P_CL2LS_REQ_SAVE_CHAR_NAME:" << std::endl; + std::cout << "\tSlot: " << (int)save->iSlotNum << std::endl; + std::cout << "\tName: " << U16toU8(save->szFirstName) << " " << U16toU8(save->szLastName) << std::endl; + ) + + resp.iSlotNum = save->iSlotNum; + resp.iGender = save->iGender; + resp.iPC_UID = Database::createCharacter(save, loginSessions[sock].userID); + memcpy(resp.szFirstName, save->szFirstName, sizeof(char16_t) * 9); + memcpy(resp.szLastName, save->szLastName, sizeof(char16_t) * 17); + + loginSessions[sock].lastHeartbeat = getTime(); + + sock->sendPacket((void*)&resp, P_LS2CL_REP_SAVE_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_SAVE_CHAR_NAME_SUCC)); + + Database::updateSelected(loginSessions[sock].userID, save->iSlotNum); +} + +void CNLoginServer::characterCreate(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_CHAR_CREATE)) + return; + + sP_CL2LS_REQ_CHAR_CREATE* character = (sP_CL2LS_REQ_CHAR_CREATE*)data->buf; + Database::finishCharacter(character); + + DEBUGLOG( + std::cout << "P_CL2LS_REQ_CHAR_CREATE:" << std::endl; + std::cout << "\tPC_UID: " << character->PCStyle.iPC_UID << std::endl; + std::cout << "\tNameCheck: " << (int)character->PCStyle.iNameCheck << std::endl; + std::cout << "\tName: " << U16toU8(character->PCStyle.szFirstName) << " " << U16toU8(character->PCStyle.szLastName) << std::endl; + std::cout << "\tGender: " << (int)character->PCStyle.iGender << std::endl; + std::cout << "\tFace: " << (int)character->PCStyle.iFaceStyle << std::endl; + std::cout << "\tHair: " << (int)character->PCStyle.iHairStyle << std::endl; + std::cout << "\tHair Color: " << (int)character->PCStyle.iHairColor << std::endl; + std::cout << "\tSkin Color: " << (int)character->PCStyle.iSkinColor << std::endl; + std::cout << "\tEye Color: " << (int)character->PCStyle.iEyeColor << std::endl; + std::cout << "\tHeight: " << (int)character->PCStyle.iHeight << std::endl; + std::cout << "\tBody: " << (int)character->PCStyle.iBody << std::endl; + std::cout << "\tClass: " << (int)character->PCStyle.iClass << std::endl; + std::cout << "\tiEquipUBID: " << (int)character->sOn_Item.iEquipUBID << std::endl; + std::cout << "\tiEquipLBID: " << (int)character->sOn_Item.iEquipLBID << std::endl; + std::cout << "\tiEquipFootID: " << (int)character->sOn_Item.iEquipFootID << std::endl; + ) + + Player player = Database::getPlayer(character->PCStyle.iPC_UID); + int64_t UID = player.iID; + + INITSTRUCT(sP_LS2CL_REP_CHAR_CREATE_SUCC, resp); + resp.sPC_Style = player.PCStyle; + resp.sPC_Style2 = player.PCStyle2; + resp.iLevel = player.level; + resp.sOn_Item = character->sOn_Item; + + // save player in session + loginSessions[sock].characters[UID] = Player(player); + loginSessions[sock].characters[UID].FEKey = sock->getFEKey(); + + loginSessions[sock].lastHeartbeat = getTime(); + + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_CREATE_SUCC, sizeof(sP_LS2CL_REP_CHAR_CREATE_SUCC)); + Database::updateSelected(loginSessions[sock].userID, player.slot); +} + +void CNLoginServer::characterDelete(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_CHAR_DELETE)) + return; + + sP_CL2LS_REQ_CHAR_DELETE* del = (sP_CL2LS_REQ_CHAR_DELETE*)data->buf; + int operationResult = Database::deleteCharacter(del->iPC_UID, loginSessions[sock].userID); + + INITSTRUCT(sP_LS2CL_REP_CHAR_DELETE_SUCC, resp); + resp.iSlotNum = operationResult; + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_DELETE_SUCC, sizeof(sP_LS2CL_REP_CHAR_DELETE_SUCC)); + loginSessions[sock].lastHeartbeat = getTime(); +} + +void CNLoginServer::characterSelect(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_CHAR_SELECT)) + return; + + // character selected + sP_CL2LS_REQ_CHAR_SELECT* chararacter = (sP_CL2LS_REQ_CHAR_SELECT*)data->buf; + INITSTRUCT(sP_LS2CL_REP_CHAR_SELECT_SUCC, resp); + + DEBUGLOG( + std::cout << "P_CL2LS_REQ_CHAR_SELECT:" << std::endl; + std::cout << "\tPC_UID: " << chararacter->iPC_UID << std::endl; + ) + + loginSessions[sock].lastHeartbeat = getTime(); + + loginSessions[sock].selectedChar = chararacter->iPC_UID; + Database::updateSelected(loginSessions[sock].userID, loginSessions[sock].characters[chararacter->iPC_UID].slot); + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHAR_SELECT_SUCC, sizeof(sP_LS2CL_REP_CHAR_SELECT_SUCC)); +} + +void CNLoginServer::shardSelect(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_SHARD_SELECT)) + return; + + // tell client to connect to the shard server + sP_CL2LS_REQ_SHARD_SELECT* shard = (sP_CL2LS_REQ_SHARD_SELECT*)data->buf; + INITSTRUCT(sP_LS2CL_REP_SHARD_SELECT_SUCC, resp); + + DEBUGLOG( + std::cout << "P_CL2LS_REQ_SHARD_SELECT:" << std::endl; + std::cout << "\tShard: " << (int)shard->ShardNum << std::endl; + ) + + const char* SHARD_IP = settings::SHARDSERVERIP.c_str(); + resp.iEnterSerialKey = rand(); + resp.g_FE_ServerPort = settings::SHARDPORT; + + // copy IP to resp (this struct uses ASCII encoding so we don't have to goof around converting encodings) + memcpy(resp.g_FE_ServerIP, SHARD_IP, strlen(SHARD_IP)); + resp.g_FE_ServerIP[strlen(SHARD_IP)] = '\0'; + + // pass player to CNSharedData + CNSharedData::setPlayer(resp.iEnterSerialKey, loginSessions[sock].characters[loginSessions[sock].selectedChar]); + + sock->sendPacket((void*)&resp, P_LS2CL_REP_SHARD_SELECT_SUCC, sizeof(sP_LS2CL_REP_SHARD_SELECT_SUCC)); +} + +void CNLoginServer::finishTutorial(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_SAVE_CHAR_TUTOR)) + return; + sP_CL2LS_REQ_SAVE_CHAR_TUTOR* save = (sP_CL2LS_REQ_SAVE_CHAR_TUTOR*)data->buf; + Database::finishTutorial(save->iPC_UID); + // update character in session + auto key = loginSessions[sock].characters[save->iPC_UID].FEKey; + loginSessions[sock].characters[save->iPC_UID] = Player(Database::getPlayer(save->iPC_UID)); + loginSessions[sock].characters[save->iPC_UID].FEKey = key; + + loginSessions[sock].lastHeartbeat = getTime(); + // no response here +} + +void CNLoginServer::changeName(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_CHANGE_CHAR_NAME)) + return; + + sP_CL2LS_REQ_CHANGE_CHAR_NAME* save = (sP_CL2LS_REQ_CHANGE_CHAR_NAME*)data->buf; + Database::changeName(save); + + INITSTRUCT(sP_LS2CL_REP_CHANGE_CHAR_NAME_SUCC, resp); + resp.iPC_UID = save->iPCUID; + memcpy(resp.szFirstName, save->szFirstName, sizeof(char16_t) * 9); + memcpy(resp.szLastName, save->szLastName, sizeof(char16_t) * 17); + resp.iSlotNum = save->iSlotNum; + + loginSessions[sock].lastHeartbeat = getTime(); + + sock->sendPacket((void*)&resp, P_LS2CL_REP_CHANGE_CHAR_NAME_SUCC, sizeof(sP_LS2CL_REP_CHANGE_CHAR_NAME_SUCC)); +} + +void CNLoginServer::duplicateExit(CNSocket* sock, CNPacketData* data) { + if (data->size != sizeof(sP_CL2LS_REQ_PC_EXIT_DUPLICATE)) + return; + + sP_CL2LS_REQ_PC_EXIT_DUPLICATE* exit = (sP_CL2LS_REQ_PC_EXIT_DUPLICATE*)data->buf; + auto account = Database::findAccount(U16toU8(exit->szID)); + + // sanity check + if (account == nullptr) { + std::cout << "[WARN] P_CL2LS_REQ_PC_EXIT_DUPLICATE submitted unknown username: " << exit->szID << std::endl; + return; + } + + exitDuplicate(account->AccountID); +} +#pragma endregion + +#pragma region connections void CNLoginServer::newConnection(CNSocket* cns) { cns->setActiveKey(SOCKETKEY_E); // by default they accept keys encrypted with the default key } @@ -401,6 +450,8 @@ void CNLoginServer::onStep() { } } +#pragma endregion + #pragma region helperMethods bool CNLoginServer::isAccountInUse(int accountId) { std::map::iterator it; @@ -443,4 +494,4 @@ bool CNLoginServer::isCharacterNameGood(std::string Firstname, std::string Lastn std::regex lastnamecheck(R"(((?! )(?!\.)[a-zA-Z0-9]*\.{0,1}(?!\.+ +)[a-zA-Z0-9]* {0,1}(?! +))*$)"); return (std::regex_match(Firstname, firstnamecheck) && std::regex_match(Lastname, lastnamecheck)); } -#pragma endregion helperMethods +#pragma endregion diff --git a/src/CNLoginServer.hpp b/src/CNLoginServer.hpp index 6727e98..1ab05f3 100644 --- a/src/CNLoginServer.hpp +++ b/src/CNLoginServer.hpp @@ -31,6 +31,17 @@ private: static void handlePacket(CNSocket* sock, CNPacketData* data); static std::map loginSessions; + static void login(CNSocket* sock, CNPacketData* data); + static void nameCheck(CNSocket* sock, CNPacketData* data); + static void nameSave(CNSocket* sock, CNPacketData* data); + static void characterCreate(CNSocket* sock, CNPacketData* data); + static void characterDelete(CNSocket* sock, CNPacketData* data); + static void characterSelect(CNSocket* sock, CNPacketData* data); + static void shardSelect(CNSocket* sock, CNPacketData* data); + static void finishTutorial(CNSocket* sock, CNPacketData* data); + static void changeName(CNSocket* sock, CNPacketData* data); + static void duplicateExit(CNSocket* sock, CNPacketData* data); + static bool isLoginDataGood(std::string login, std::string password); static bool isPasswordCorrect(std::string actualPassword, std::string tryPassword); static bool isAccountInUse(int accountId);