Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /**
- * The Forgotten Server - a free and open-source MMORPG server emulator
- * Copyright (C) 2017 Mark Samman <mark.samman@gmail.com>
- *
- * 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.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
- #include "otpch.h"
- #include "iologindata.h"
- #include "configmanager.h"
- #include "game.h"
- extern ConfigManager g_config;
- extern Game g_game;
- Account IOLoginData::loadAccount(uint32_t accno)
- {
- Account account;
- std::ostringstream query;
- query << "SELECT `id`, `name`, `password`, `type`, `premdays`, `lastday` FROM `accounts` WHERE `id` = " << accno;
- DBResult_ptr result = Database::getInstance().storeQuery(query.str());
- if (!result) {
- return account;
- }
- account.id = result->getNumber<uint32_t>("id");
- account.name = result->getString("name");
- account.accountType = static_cast<AccountType_t>(result->getNumber<int32_t>("type"));
- account.premiumDays = result->getNumber<uint16_t>("premdays");
- account.lastDay = result->getNumber<time_t>("lastday");
- return account;
- }
- bool IOLoginData::saveAccount(const Account& acc)
- {
- std::ostringstream query;
- query << "UPDATE `accounts` SET `premdays` = " << acc.premiumDays << ", `lastday` = " << acc.lastDay << " WHERE `id` = " << acc.id;
- return Database::getInstance().executeQuery(query.str());
- }
- std::string decodeSecret(const std::string& secret)
- {
- // simple base32 decoding
- std::string key;
- key.reserve(10);
- uint32_t buffer = 0, left = 0;
- for (const auto& ch : secret) {
- buffer <<= 5;
- if (ch >= 'A' && ch <= 'Z') {
- buffer |= (ch & 0x1F) - 1;
- } else if (ch >= '2' && ch <= '7') {
- buffer |= ch - 24;
- } else {
- // if a key is broken, return empty and the comparison
- // will always be false since the token must not be empty
- return {};
- }
- left += 5;
- if (left >= 8) {
- left -= 8;
- key.push_back(static_cast<char>(buffer >> left));
- }
- }
- return key;
- }
- bool IOLoginData::loginserverAuthentication(const std::string& name, const std::string& password, Account& account)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id`, `name`, `password`, `secret`, `type`, `premdays`, `lastday` FROM `accounts` WHERE `name` = " << db.escapeString(name);
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return false;
- }
- if (transformToSHA1(password) != result->getString("password")) {
- return false;
- }
- account.id = result->getNumber<uint32_t>("id");
- account.name = result->getString("name");
- account.accountType = static_cast<AccountType_t>(result->getNumber<int32_t>("type"));
- account.premiumDays = result->getNumber<uint16_t>("premdays");
- account.lastDay = result->getNumber<time_t>("lastday");
- query.str(std::string());
- query << "SELECT `name`, `deletion` FROM `players` WHERE `account_id` = " << account.id;
- result = db.storeQuery(query.str());
- if (result) {
- do {
- if (result->getNumber<uint64_t>("deletion") == 0) {
- account.characters.push_back(result->getString("name"));
- }
- } while (result->next());
- std::sort(account.characters.begin(), account.characters.end());
- }
- return true;
- }
- uint32_t IOLoginData::gameworldAuthentication(const std::string& accountName, const std::string& password, std::string& characterName, std::string& token, uint32_t tokenTime)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id`, `password`, `secret` FROM `accounts` WHERE `name` = " << db.escapeString(accountName);
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return 0;
- }
- std::string secret = decodeSecret(result->getString("secret"));
- if (!secret.empty()) {
- if (token.empty()) {
- return 0;
- }
- bool tokenValid = token == generateToken(secret, tokenTime) || token == generateToken(secret, tokenTime - 1) || token == generateToken(secret, tokenTime + 1);
- if (!tokenValid) {
- return 0;
- }
- }
- if (transformToSHA1(password) != result->getString("password")) {
- return 0;
- }
- uint32_t accountId = result->getNumber<uint32_t>("id");
- query.str(std::string());
- query << "SELECT `account_id`, `name`, `deletion` FROM `players` WHERE `name` = " << db.escapeString(characterName);
- result = db.storeQuery(query.str());
- if (!result) {
- return 0;
- }
- if (result->getNumber<uint32_t>("account_id") != accountId || result->getNumber<uint64_t>("deletion") != 0) {
- return 0;
- }
- characterName = result->getString("name");
- return accountId;
- }
- AccountType_t IOLoginData::getAccountType(uint32_t accountId)
- {
- std::ostringstream query;
- query << "SELECT `type` FROM `accounts` WHERE `id` = " << accountId;
- DBResult_ptr result = Database::getInstance().storeQuery(query.str());
- if (!result) {
- return ACCOUNT_TYPE_NORMAL;
- }
- return static_cast<AccountType_t>(result->getNumber<uint16_t>("type"));
- }
- void IOLoginData::setAccountType(uint32_t accountId, AccountType_t accountType)
- {
- std::ostringstream query;
- query << "UPDATE `accounts` SET `type` = " << static_cast<uint16_t>(accountType) << " WHERE `id` = " << accountId;
- Database::getInstance().executeQuery(query.str());
- }
- void IOLoginData::updateOnlineStatus(uint32_t guid, bool login)
- {
- if (g_config.getBoolean(ConfigManager::ALLOW_CLONES)) {
- return;
- }
- std::ostringstream query;
- if (login) {
- query << "INSERT INTO `players_online` VALUES (" << guid << ')';
- } else {
- query << "DELETE FROM `players_online` WHERE `player_id` = " << guid;
- }
- Database::getInstance().executeQuery(query.str());
- }
- bool IOLoginData::preloadPlayer(Player* player, const std::string& name)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id`, `account_id`, `group_id`, `deletion`, (SELECT `type` FROM `accounts` WHERE `accounts`.`id` = `account_id`) AS `account_type`";
- if (!g_config.getBoolean(ConfigManager::FREE_PREMIUM)) {
- query << ", (SELECT `premdays` FROM `accounts` WHERE `accounts`.`id` = `account_id`) AS `premium_days`";
- }
- query << " FROM `players` WHERE `name` = " << db.escapeString(name);
- DBResult_ptr result = db.storeQuery(query.str());
- if (result) {
- if (result->getNumber<uint64_t>("deletion") != 0) {
- return false;
- }
- player->setGUID(result->getNumber<uint32_t>("id"));
- Group* group = g_game.groups.getGroup(result->getNumber<uint16_t>("group_id"));
- if (!group) {
- std::cout << "[Error - IOLoginData::preloadPlayer] " << player->name << " has Group ID " << result->getNumber<uint16_t>("group_id") << " which doesn't exist." << std::endl;
- return false;
- }
- player->setGroup(group);
- player->accountNumber = result->getNumber<uint32_t>("account_id");
- player->accountType = static_cast<AccountType_t>(result->getNumber<uint16_t>("account_type"));
- if (!g_config.getBoolean(ConfigManager::FREE_PREMIUM)) {
- player->premiumDays = result->getNumber<uint16_t>("premium_days");
- } else {
- player->premiumDays = std::numeric_limits<uint16_t>::max();
- }
- }
- query.str(std::string());
- query << "SELECT * FROM `account_vip` WHERE `id` = " << result->getNumber<uint32_t>("account_id");
- result = db.storeQuery(query.str());
- if (result) {
- player->vipTime = result->getNumber<uint64_t>("time");
- }
- return true;
- }
- bool IOLoginData::loadPlayerById(Player* player, uint32_t id)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id`, `name`, `account_id`, `group_id`, `sex`, `vocation`, `experience`, `level`, `maglevel`, `health`, `healthmax`, `blessings`, `mana`, `manamax`, `manaspent`, `soul`, `lookbody`, `lookfeet`, `lookhead`, `looklegs`, `looktype`, `lookaddons`, `posx`, `posy`, `posz`, `cap`, `lastlogin`, `lastlogout`, `lastip`, `conditions`, `skulltime`, `skull`, `town_id`, `balance`, `offlinetraining_time`, `offlinetraining_skill`, `stamina`, `skill_fist`, `skill_fist_tries`, `skill_club`, `skill_club_tries`, `skill_sword`, `skill_sword_tries`, `skill_axe`, `skill_axe_tries`, `skill_dist`, `skill_dist_tries`, `skill_shielding`, `skill_shielding_tries`, `skill_fishing`, `skill_fishing_tries` FROM `players` WHERE `id` = " << id;
- return loadPlayer(player, db.storeQuery(query.str()));
- }
- bool IOLoginData::loadPlayerByName(Player* player, const std::string& name)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id`, `name`, `account_id`, `group_id`, `sex`, `vocation`, `experience`, `level`, `maglevel`, `health`, `healthmax`, `blessings`, `mana`, `manamax`, `manaspent`, `soul`, `lookbody`, `lookfeet`, `lookhead`, `looklegs`, `looktype`, `lookaddons`, `posx`, `posy`, `posz`, `cap`, `lastlogin`, `lastlogout`, `lastip`, `conditions`, `skulltime`, `skull`, `town_id`, `balance`, `offlinetraining_time`, `offlinetraining_skill`, `stamina`, `skill_fist`, `skill_fist_tries`, `skill_club`, `skill_club_tries`, `skill_sword`, `skill_sword_tries`, `skill_axe`, `skill_axe_tries`, `skill_dist`, `skill_dist_tries`, `skill_shielding`, `skill_shielding_tries`, `skill_fishing`, `skill_fishing_tries` FROM `players` WHERE `name` = " << db.escapeString(name);
- return loadPlayer(player, db.storeQuery(query.str()));
- }
- bool IOLoginData::loadPlayer(Player* player, DBResult_ptr result)
- {
- if (!result) {
- return false;
- }
- Database& db = Database::getInstance();
- uint32_t accno = result->getNumber<uint32_t>("account_id");
- Account acc = loadAccount(accno);
- player->setGUID(result->getNumber<uint32_t>("id"));
- player->name = result->getString("name");
- player->accountNumber = accno;
- player->accountType = acc.accountType;
- if (g_config.getBoolean(ConfigManager::FREE_PREMIUM)) {
- player->premiumDays = std::numeric_limits<uint16_t>::max();
- } else {
- player->premiumDays = acc.premiumDays;
- }
- Group* group = g_game.groups.getGroup(result->getNumber<uint16_t>("group_id"));
- if (!group) {
- std::cout << "[Error - IOLoginData::loadPlayer] " << player->name << " has Group ID " << result->getNumber<uint16_t>("group_id") << " which doesn't exist" << std::endl;
- return false;
- }
- player->setGroup(group);
- player->bankBalance = result->getNumber<uint64_t>("balance");
- player->setSex(static_cast<PlayerSex_t>(result->getNumber<uint16_t>("sex")));
- player->level = std::max<uint32_t>(1, result->getNumber<uint32_t>("level"));
- uint64_t experience = result->getNumber<uint64_t>("experience");
- uint64_t currExpCount = Player::getExpForLevel(player->level);
- uint64_t nextExpCount = Player::getExpForLevel(player->level + 1);
- if (experience < currExpCount || experience > nextExpCount) {
- experience = currExpCount;
- }
- player->experience = experience;
- if (currExpCount < nextExpCount) {
- player->levelPercent = Player::getPercentLevel(player->experience - currExpCount, nextExpCount - currExpCount);
- } else {
- player->levelPercent = 0;
- }
- player->soul = result->getNumber<uint16_t>("soul");
- player->capacity = result->getNumber<uint32_t>("cap") * 100;
- player->blessings = result->getNumber<uint16_t>("blessings");
- unsigned long conditionsSize;
- const char* conditions = result->getStream("conditions", conditionsSize);
- PropStream propStream;
- propStream.init(conditions, conditionsSize);
- Condition* condition = Condition::createCondition(propStream);
- while (condition) {
- if (condition->unserialize(propStream)) {
- player->storedConditionList.push_front(condition);
- } else {
- delete condition;
- }
- condition = Condition::createCondition(propStream);
- }
- if (!player->setVocation(result->getNumber<uint16_t>("vocation"))) {
- std::cout << "[Error - IOLoginData::loadPlayer] " << player->name << " has Vocation ID " << result->getNumber<uint16_t>("vocation") << " which doesn't exist" << std::endl;
- return false;
- }
- player->mana = result->getNumber<uint32_t>("mana");
- player->manaMax = result->getNumber<uint32_t>("manamax");
- player->magLevel = result->getNumber<uint32_t>("maglevel");
- uint64_t nextManaCount = player->vocation->getReqMana(player->magLevel + 1);
- uint64_t manaSpent = result->getNumber<uint64_t>("manaspent");
- if (manaSpent > nextManaCount) {
- manaSpent = 0;
- }
- player->manaSpent = manaSpent;
- player->magLevelPercent = Player::getPercentLevel(player->manaSpent, nextManaCount);
- player->health = result->getNumber<int32_t>("health");
- player->healthMax = result->getNumber<int32_t>("healthmax");
- player->defaultOutfit.lookType = result->getNumber<uint16_t>("looktype");
- player->defaultOutfit.lookHead = result->getNumber<uint16_t>("lookhead");
- player->defaultOutfit.lookBody = result->getNumber<uint16_t>("lookbody");
- player->defaultOutfit.lookLegs = result->getNumber<uint16_t>("looklegs");
- player->defaultOutfit.lookFeet = result->getNumber<uint16_t>("lookfeet");
- player->defaultOutfit.lookAddons = result->getNumber<uint16_t>("lookaddons");
- player->currentOutfit = player->defaultOutfit;
- if (g_game.getWorldType() != WORLD_TYPE_PVP_ENFORCED) {
- const time_t skullSeconds = result->getNumber<time_t>("skulltime") - time(nullptr);
- if (skullSeconds > 0) {
- //ensure that we round up the number of ticks
- player->skullTicks = (skullSeconds + 2) * 1000;
- uint16_t skull = result->getNumber<uint16_t>("skull");
- if (skull == SKULL_RED) {
- player->skull = SKULL_RED;
- } else if (skull == SKULL_BLACK) {
- player->skull = SKULL_BLACK;
- }
- }
- }
- player->loginPosition.x = result->getNumber<uint16_t>("posx");
- player->loginPosition.y = result->getNumber<uint16_t>("posy");
- player->loginPosition.z = result->getNumber<uint16_t>("posz");
- player->lastLoginSaved = result->getNumber<time_t>("lastlogin");
- player->lastLogout = result->getNumber<time_t>("lastlogout");
- player->offlineTrainingTime = result->getNumber<int32_t>("offlinetraining_time") * 1000;
- player->offlineTrainingSkill = result->getNumber<int32_t>("offlinetraining_skill");
- Town* town = g_game.map.towns.getTown(result->getNumber<uint32_t>("town_id"));
- if (!town) {
- std::cout << "[Error - IOLoginData::loadPlayer] " << player->name << " has Town ID " << result->getNumber<uint32_t>("town_id") << " which doesn't exist" << std::endl;
- return false;
- }
- player->town = town;
- const Position& loginPos = player->loginPosition;
- if (loginPos.x == 0 && loginPos.y == 0 && loginPos.z == 0) {
- player->loginPosition = player->getTemplePosition();
- }
- player->staminaMinutes = result->getNumber<uint16_t>("stamina");
- static const std::string skillNames[] = {"skill_fist", "skill_club", "skill_sword", "skill_axe", "skill_dist", "skill_shielding", "skill_fishing"};
- static const std::string skillNameTries[] = {"skill_fist_tries", "skill_club_tries", "skill_sword_tries", "skill_axe_tries", "skill_dist_tries", "skill_shielding_tries", "skill_fishing_tries"};
- static constexpr size_t size = sizeof(skillNames) / sizeof(std::string);
- for (uint8_t i = 0; i < size; ++i) {
- uint16_t skillLevel = result->getNumber<uint16_t>(skillNames[i]);
- uint64_t skillTries = result->getNumber<uint64_t>(skillNameTries[i]);
- uint64_t nextSkillTries = player->vocation->getReqSkillTries(i, skillLevel + 1);
- if (skillTries > nextSkillTries) {
- skillTries = 0;
- }
- player->skills[i].level = skillLevel;
- player->skills[i].tries = skillTries;
- player->skills[i].percent = Player::getPercentLevel(skillTries, nextSkillTries);
- }
- std::ostringstream query;
- query << "SELECT `guild_id`, `rank_id`, `nick` FROM `guild_membership` WHERE `player_id` = " << player->getGUID();
- if ((result = db.storeQuery(query.str()))) {
- uint32_t guildId = result->getNumber<uint32_t>("guild_id");
- uint32_t playerRankId = result->getNumber<uint32_t>("rank_id");
- player->guildNick = result->getString("nick");
- Guild* guild = g_game.getGuild(guildId);
- if (!guild) {
- guild = IOGuild::loadGuild(guildId);
- g_game.addGuild(guild);
- }
- if (guild) {
- player->guild = guild;
- const GuildRank* rank = guild->getRankById(playerRankId);
- if (!rank) {
- query.str(std::string());
- query << "SELECT `id`, `name`, `level` FROM `guild_ranks` WHERE `id` = " << playerRankId;
- if ((result = db.storeQuery(query.str()))) {
- guild->addRank(result->getNumber<uint32_t>("id"), result->getString("name"), result->getNumber<uint16_t>("level"));
- }
- rank = guild->getRankById(playerRankId);
- if (!rank) {
- player->guild = nullptr;
- }
- }
- player->guildRank = rank;
- IOGuild::getWarList(guildId, player->guildWarVector);
- query.str(std::string());
- query << "SELECT COUNT(*) AS `members` FROM `guild_membership` WHERE `guild_id` = " << guildId;
- if ((result = db.storeQuery(query.str()))) {
- guild->setMemberCount(result->getNumber<uint32_t>("members"));
- }
- }
- }
- query.str(std::string());
- query << "SELECT `player_id`, `name` FROM `player_spells` WHERE `player_id` = " << player->getGUID();
- if ((result = db.storeQuery(query.str()))) {
- do {
- player->learnedInstantSpellList.emplace_front(result->getString("name"));
- } while (result->next());
- }
- //load inventory items
- ItemMap itemMap;
- query.str(std::string());
- query << "SELECT `pid`, `sid`, `itemtype`, `count`, `attributes` FROM `player_items` WHERE `player_id` = " << player->getGUID() << " ORDER BY `sid` DESC";
- if ((result = db.storeQuery(query.str()))) {
- loadItems(itemMap, result);
- for (ItemMap::const_reverse_iterator it = itemMap.rbegin(), end = itemMap.rend(); it != end; ++it) {
- const std::pair<Item*, int32_t>& pair = it->second;
- Item* item = pair.first;
- int32_t pid = pair.second;
- if (pid >= 1 && pid <= 11) {
- player->internalAddThing(pid, item);
- } else {
- ItemMap::const_iterator it2 = itemMap.find(pid);
- if (it2 == itemMap.end()) {
- continue;
- }
- Container* container = it2->second.first->getContainer();
- if (container) {
- container->internalAddThing(item);
- }
- }
- }
- }
- //store inbox
- if (!player->inventory[CONST_SLOT_STORE_INBOX]) {
- player->internalAddThing(CONST_SLOT_STORE_INBOX, Item::CreateItem(ITEM_STORE_INBOX));
- }
- //load depot items
- itemMap.clear();
- query.str(std::string());
- query << "SELECT `pid`, `sid`, `itemtype`, `count`, `attributes` FROM `player_depotitems` WHERE `player_id` = " << player->getGUID() << " ORDER BY `sid` DESC";
- if ((result = db.storeQuery(query.str()))) {
- loadItems(itemMap, result);
- for (ItemMap::const_reverse_iterator it = itemMap.rbegin(), end = itemMap.rend(); it != end; ++it) {
- const std::pair<Item*, int32_t>& pair = it->second;
- Item* item = pair.first;
- int32_t pid = pair.second;
- if (pid >= 0 && pid < 100) {
- DepotChest* depotChest = player->getDepotChest(pid, true);
- if (depotChest) {
- depotChest->internalAddThing(item);
- }
- } else {
- ItemMap::const_iterator it2 = itemMap.find(pid);
- if (it2 == itemMap.end()) {
- continue;
- }
- Container* container = it2->second.first->getContainer();
- if (container) {
- container->internalAddThing(item);
- }
- }
- }
- }
- //load inbox items
- itemMap.clear();
- query.str(std::string());
- query << "SELECT `pid`, `sid`, `itemtype`, `count`, `attributes` FROM `player_inboxitems` WHERE `player_id` = " << player->getGUID() << " ORDER BY `sid` DESC";
- if ((result = db.storeQuery(query.str()))) {
- loadItems(itemMap, result);
- for (ItemMap::const_reverse_iterator it = itemMap.rbegin(), end = itemMap.rend(); it != end; ++it) {
- const std::pair<Item*, int32_t>& pair = it->second;
- Item* item = pair.first;
- int32_t pid = pair.second;
- if (pid >= 0 && pid < 100) {
- player->getInbox()->internalAddThing(item);
- } else {
- ItemMap::const_iterator it2 = itemMap.find(pid);
- if (it2 == itemMap.end()) {
- continue;
- }
- Container* container = it2->second.first->getContainer();
- if (container) {
- container->internalAddThing(item);
- }
- }
- }
- }
- //load storage map
- query.str(std::string());
- query << "SELECT `key`, `value` FROM `player_storage` WHERE `player_id` = " << player->getGUID();
- if ((result = db.storeQuery(query.str()))) {
- do {
- player->addStorageValue(result->getNumber<uint32_t>("key"), result->getNumber<int32_t>("value"), true);
- } while (result->next());
- }
- //load vip
- query.str(std::string());
- query << "SELECT `player_id` FROM `account_viplist` WHERE `account_id` = " << player->getAccount();
- if ((result = db.storeQuery(query.str()))) {
- do {
- player->addVIPInternal(result->getNumber<uint32_t>("player_id"));
- } while (result->next());
- }
- //bestiary
- if (!player->hasFlag(PlayerFlag_CannotBeAttacked))
- {
- player->updateAllBestiary(true);
- query.str(std::string());
- query << "SELECT `player_id`, `name`, `kills`, `items` FROM `player_bestiary` WHERE `player_id` = " << player->getGUID();
- if ((result = db.storeQuery(query.str()))) {
- do {
- player->addBestiaryPosting(result->getString("name"), result->getNumber<uint32_t>("kills"), result->getString("items"));
- } while (result->next());
- }
- }
- //load auto loot items
- query.str(std::string());
- query << "SELECT * FROM `player_autolootlist` WHERE `player_id` = " << player->getGUID();
- if ((result = db.storeQuery(query.str()))) {
- unsigned long autoLootListSize;
- const char* autoLootList = result->getStream("items", autoLootListSize);
- propStream.init(autoLootList, autoLootListSize);
- std::string item;
- while (propStream.readString(item) && item != "0") {
- player->autoLootList.insert(item);
- }
- }
- player->updateBaseSpeed();
- player->updateInventoryWeight();
- player->updateItemsLight(true);
- return true;
- }
- bool IOLoginData::saveItems(const Player* player, const ItemBlockList& itemList, DBInsert& query_insert, PropWriteStream& propWriteStream)
- {
- std::ostringstream ss;
- using ContainerBlock = std::pair<Container*, int32_t>;
- std::list<ContainerBlock> queue;
- int32_t runningId = 100;
- Database& db = Database::getInstance();
- for (const auto& it : itemList) {
- int32_t pid = it.first;
- Item* item = it.second;
- ++runningId;
- propWriteStream.clear();
- item->serializeAttr(propWriteStream);
- size_t attributesSize;
- const char* attributes = propWriteStream.getStream(attributesSize);
- ss << player->getGUID() << ',' << pid << ',' << runningId << ',' << item->getID() << ',' << item->getSubType() << ',' << db.escapeBlob(attributes, attributesSize);
- if (!query_insert.addRow(ss)) {
- return false;
- }
- if (Container* container = item->getContainer()) {
- queue.emplace_back(container, runningId);
- }
- }
- while (!queue.empty()) {
- const ContainerBlock& cb = queue.front();
- Container* container = cb.first;
- int32_t parentId = cb.second;
- queue.pop_front();
- for (Item* item : container->getItemList()) {
- ++runningId;
- Container* subContainer = item->getContainer();
- if (subContainer) {
- queue.emplace_back(subContainer, runningId);
- }
- propWriteStream.clear();
- item->serializeAttr(propWriteStream);
- size_t attributesSize;
- const char* attributes = propWriteStream.getStream(attributesSize);
- ss << player->getGUID() << ',' << parentId << ',' << runningId << ',' << item->getID() << ',' << item->getSubType() << ',' << db.escapeBlob(attributes, attributesSize);
- if (!query_insert.addRow(ss)) {
- return false;
- }
- }
- }
- return query_insert.execute();
- }
- bool IOLoginData::savePlayer(Player* player)
- {
- if (player->getHealth() <= 0) {
- player->changeHealth(1);
- }
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `save` FROM `players` WHERE `id` = " << player->getGUID();
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return false;
- }
- if (result->getNumber<uint16_t>("save") == 0) {
- query.str(std::string());
- query << "UPDATE `players` SET `lastlogin` = " << player->lastLoginSaved << ", `lastip` = " << player->lastIP << " WHERE `id` = " << player->getGUID();
- return db.executeQuery(query.str());
- }
- //serialize conditions
- PropWriteStream propWriteStream;
- for (Condition* condition : player->conditions) {
- if (condition->isPersistent()) {
- condition->serialize(propWriteStream);
- propWriteStream.write<uint8_t>(CONDITIONATTR_END);
- }
- }
- size_t conditionsSize;
- const char* conditions = propWriteStream.getStream(conditionsSize);
- //First, an UPDATE query to write the player itself
- query.str(std::string());
- query << "UPDATE `players` SET ";
- query << "`level` = " << player->level << ',';
- query << "`group_id` = " << player->group->id << ',';
- query << "`vocation` = " << player->getVocationId() << ',';
- query << "`health` = " << player->health << ',';
- query << "`healthmax` = " << player->healthMax << ',';
- query << "`experience` = " << player->experience << ',';
- query << "`lookbody` = " << static_cast<uint32_t>(player->defaultOutfit.lookBody) << ',';
- query << "`lookfeet` = " << static_cast<uint32_t>(player->defaultOutfit.lookFeet) << ',';
- query << "`lookhead` = " << static_cast<uint32_t>(player->defaultOutfit.lookHead) << ',';
- query << "`looklegs` = " << static_cast<uint32_t>(player->defaultOutfit.lookLegs) << ',';
- query << "`looktype` = " << player->defaultOutfit.lookType << ',';
- query << "`lookaddons` = " << static_cast<uint32_t>(player->defaultOutfit.lookAddons) << ',';
- query << "`maglevel` = " << player->magLevel << ',';
- query << "`mana` = " << player->mana << ',';
- query << "`manamax` = " << player->manaMax << ',';
- query << "`manaspent` = " << player->manaSpent << ',';
- query << "`soul` = " << static_cast<uint16_t>(player->soul) << ',';
- query << "`town_id` = " << player->town->getID() << ',';
- const Position& loginPosition = player->getLoginPosition();
- query << "`posx` = " << loginPosition.getX() << ',';
- query << "`posy` = " << loginPosition.getY() << ',';
- query << "`posz` = " << loginPosition.getZ() << ',';
- query << "`cap` = " << (player->capacity / 100) << ',';
- query << "`sex` = " << player->sex << ',';
- if (player->lastLoginSaved != 0) {
- query << "`lastlogin` = " << player->lastLoginSaved << ',';
- }
- if (player->lastIP != 0) {
- query << "`lastip` = " << player->lastIP << ',';
- }
- query << "`conditions` = " << db.escapeBlob(conditions, conditionsSize) << ',';
- if (g_game.getWorldType() != WORLD_TYPE_PVP_ENFORCED) {
- int32_t skullTime = 0;
- if (player->skullTicks > 0) {
- skullTime = time(nullptr) + player->skullTicks / 1000;
- }
- query << "`skulltime` = " << skullTime << ',';
- Skulls_t skull = SKULL_NONE;
- if (player->skull == SKULL_RED) {
- skull = SKULL_RED;
- } else if (player->skull == SKULL_BLACK) {
- skull = SKULL_BLACK;
- }
- query << "`skull` = " << static_cast<uint32_t>(skull) << ',';
- }
- query << "`lastlogout` = " << player->getLastLogout() << ',';
- query << "`balance` = " << player->bankBalance << ',';
- query << "`offlinetraining_time` = " << player->getOfflineTrainingTime() / 1000 << ',';
- query << "`offlinetraining_skill` = " << player->getOfflineTrainingSkill() << ',';
- query << "`stamina` = " << player->getStaminaMinutes() << ',';
- query << "`skill_fist` = " << player->skills[SKILL_FIST].level << ',';
- query << "`skill_fist_tries` = " << player->skills[SKILL_FIST].tries << ',';
- query << "`skill_club` = " << player->skills[SKILL_CLUB].level << ',';
- query << "`skill_club_tries` = " << player->skills[SKILL_CLUB].tries << ',';
- query << "`skill_sword` = " << player->skills[SKILL_SWORD].level << ',';
- query << "`skill_sword_tries` = " << player->skills[SKILL_SWORD].tries << ',';
- query << "`skill_axe` = " << player->skills[SKILL_AXE].level << ',';
- query << "`skill_axe_tries` = " << player->skills[SKILL_AXE].tries << ',';
- query << "`skill_dist` = " << player->skills[SKILL_DISTANCE].level << ',';
- query << "`skill_dist_tries` = " << player->skills[SKILL_DISTANCE].tries << ',';
- query << "`skill_shielding` = " << player->skills[SKILL_SHIELD].level << ',';
- query << "`skill_shielding_tries` = " << player->skills[SKILL_SHIELD].tries << ',';
- query << "`skill_fishing` = " << player->skills[SKILL_FISHING].level << ',';
- query << "`skill_fishing_tries` = " << player->skills[SKILL_FISHING].tries << ',';
- if (!player->isOffline()) {
- query << "`onlinetime` = `onlinetime` + " << (time(nullptr) - player->lastLoginSaved) << ',';
- }
- query << "`blessings` = " << static_cast<uint32_t>(player->blessings);
- query << " WHERE `id` = " << player->getGUID();
- DBTransaction transaction;
- if (!transaction.begin()) {
- return false;
- }
- if (!db.executeQuery(query.str())) {
- return false;
- }
- // learned spells
- query.str(std::string());
- query << "DELETE FROM `player_spells` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- query.str(std::string());
- DBInsert spellsQuery("INSERT INTO `player_spells` (`player_id`, `name` ) VALUES ");
- for (const std::string& spellName : player->learnedInstantSpellList) {
- query << player->getGUID() << ',' << db.escapeString(spellName);
- if (!spellsQuery.addRow(query)) {
- return false;
- }
- }
- if (!spellsQuery.execute()) {
- return false;
- }
- //item saving
- query << "DELETE FROM `player_items` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- DBInsert itemsQuery("INSERT INTO `player_items` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
- ItemBlockList itemList;
- for (int32_t slotId = 1; slotId <= 11; ++slotId) {
- Item* item = player->inventory[slotId];
- if (item) {
- itemList.emplace_back(slotId, item);
- }
- }
- if (!saveItems(player, itemList, itemsQuery, propWriteStream)) {
- return false;
- }
- if (player->lastDepotId != -1) {
- //save depot items
- query.str(std::string());
- query << "DELETE FROM `player_depotitems` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- DBInsert depotQuery("INSERT INTO `player_depotitems` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
- itemList.clear();
- for (const auto& it : player->depotChests) {
- DepotChest* depotChest = it.second;
- for (Item* item : depotChest->getItemList()) {
- itemList.emplace_back(it.first, item);
- }
- }
- if (!saveItems(player, itemList, depotQuery, propWriteStream)) {
- return false;
- }
- }
- //save auto loot items
- propWriteStream.clear();
- for (const auto& it : player->autoLootList) {
- propWriteStream.writeString(it);
- }
- propWriteStream.writeString("0");
- size_t autoLootItemsSize;
- const char* autoLootItems = propWriteStream.getStream(autoLootItemsSize);
- query.str(std::string());
- query << "DELETE FROM `player_autolootlist` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- query.str(std::string());
- query << "INSERT INTO `player_autolootlist` (`player_id`, `items`) VALUES (" << player->getGUID() << ", " << db.escapeBlob(autoLootItems, autoLootItemsSize) << ')';
- if (!db.executeQuery(query.str())) {
- return false;
- }
- //save inbox items
- query.str(std::string());
- query << "DELETE FROM `player_inboxitems` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- DBInsert inboxQuery("INSERT INTO `player_inboxitems` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
- itemList.clear();
- for (Item* item : player->getInbox()->getItemList()) {
- itemList.emplace_back(0, item);
- }
- if (!saveItems(player, itemList, inboxQuery, propWriteStream)) {
- return false;
- }
- //bestiary
- if (!player->hasFlag(PlayerFlag_CannotBeAttacked))
- {
- query.str(std::string());
- query << "DELETE FROM `player_bestiary` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- BestiaryMap bestiaryMap = player->getBestiary();
- if (!bestiaryMap.empty())
- {
- query.str(std::string());
- DBInsert bestiaryQuery("INSERT INTO `player_bestiary` (`player_id`, `name`, `kills`, `items`) VALUES ");
- for (auto it = bestiaryMap.begin(); it != bestiaryMap.end(); it++)
- {
- if (it->second.kills == 0)
- continue;
- query << player->getGUID() << "," << db.escapeString(it->first).c_str() << "," << it->second.kills << ",";
- if (!it->second.lootItemList.empty())
- {
- std::stringstream s;
- for (LootItemList::const_iterator _it = it->second.lootItemList.begin(); _it != it->second.lootItemList.end(); ++_it)
- {
- if (_it->second != 0)
- s << (s.str().empty() ? "" : ";") << _it->first << "-" << _it->second;
- }
- query << db.escapeString(s.str()).c_str();
- }
- else
- query << "''";
- if (!bestiaryQuery.addRow(query)) {
- return false;
- }
- }
- if (!bestiaryQuery.execute()) {
- return false;
- }
- }
- }
- query.str(std::string());
- query << "DELETE FROM `player_storage` WHERE `player_id` = " << player->getGUID();
- if (!db.executeQuery(query.str())) {
- return false;
- }
- query.str(std::string());
- DBInsert storageQuery("INSERT INTO `player_storage` (`player_id`, `key`, `value`) VALUES ");
- player->genReservedStorageRange();
- for (const auto& it : player->storageMap) {
- query << player->getGUID() << ',' << it.first << ',' << it.second;
- if (!storageQuery.addRow(query)) {
- return false;
- }
- }
- if (!storageQuery.execute()) {
- return false;
- }
- //End the transaction
- return transaction.commit();
- }
- std::string IOLoginData::getNameByGuid(uint32_t guid)
- {
- std::ostringstream query;
- query << "SELECT `name` FROM `players` WHERE `id` = " << guid;
- DBResult_ptr result = Database::getInstance().storeQuery(query.str());
- if (!result) {
- return std::string();
- }
- return result->getString("name");
- }
- uint32_t IOLoginData::getGuidByName(const std::string& name)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id` FROM `players` WHERE `name` = " << db.escapeString(name);
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return 0;
- }
- return result->getNumber<uint32_t>("id");
- }
- bool IOLoginData::getGuidByNameEx(uint32_t& guid, bool& specialVip, std::string& name)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `name`, `id`, `group_id`, `account_id` FROM `players` WHERE `name` = " << db.escapeString(name);
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return false;
- }
- name = result->getString("name");
- guid = result->getNumber<uint32_t>("id");
- Group* group = g_game.groups.getGroup(result->getNumber<uint16_t>("group_id"));
- uint64_t flags;
- if (group) {
- flags = group->flags;
- } else {
- flags = 0;
- }
- specialVip = (flags & PlayerFlag_SpecialVIP) != 0;
- return true;
- }
- bool IOLoginData::formatPlayerName(std::string& name)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `name` FROM `players` WHERE `name` = " << db.escapeString(name);
- DBResult_ptr result = db.storeQuery(query.str());
- if (!result) {
- return false;
- }
- name = result->getString("name");
- return true;
- }
- void IOLoginData::loadItems(ItemMap& itemMap, DBResult_ptr result)
- {
- do {
- uint32_t sid = result->getNumber<uint32_t>("sid");
- uint32_t pid = result->getNumber<uint32_t>("pid");
- uint16_t type = result->getNumber<uint16_t>("itemtype");
- uint16_t count = result->getNumber<uint16_t>("count");
- unsigned long attrSize;
- const char* attr = result->getStream("attributes", attrSize);
- PropStream propStream;
- propStream.init(attr, attrSize);
- Item* item = Item::CreateItem(type, count);
- if (item) {
- if (!item->unserializeAttr(propStream)) {
- std::cout << "WARNING: Serialize error in IOLoginData::loadItems" << std::endl;
- }
- std::pair<Item*, uint32_t> pair(item, pid);
- itemMap[sid] = pair;
- }
- } while (result->next());
- }
- void IOLoginData::increaseBankBalance(uint32_t guid, uint64_t bankBalance)
- {
- std::ostringstream query;
- query << "UPDATE `players` SET `balance` = `balance` + " << bankBalance << " WHERE `id` = " << guid;
- Database::getInstance().executeQuery(query.str());
- }
- bool IOLoginData::hasBiddedOnHouse(uint32_t guid)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "SELECT `id` FROM `houses` WHERE `highest_bidder` = " << guid << " LIMIT 1";
- return db.storeQuery(query.str()).get() != nullptr;
- }
- std::forward_list<VIPEntry> IOLoginData::getVIPEntries(uint32_t accountId)
- {
- std::forward_list<VIPEntry> entries;
- std::ostringstream query;
- query << "SELECT `player_id`, (SELECT `name` FROM `players` WHERE `id` = `player_id`) AS `name`, `description`, `icon`, `notify` FROM `account_viplist` WHERE `account_id` = " << accountId;
- DBResult_ptr result = Database::getInstance().storeQuery(query.str());
- if (result) {
- do {
- entries.emplace_front(
- result->getNumber<uint32_t>("player_id"),
- result->getString("name"),
- result->getString("description"),
- result->getNumber<uint32_t>("icon"),
- result->getNumber<uint16_t>("notify") != 0
- );
- } while (result->next());
- }
- return entries;
- }
- void IOLoginData::addVIPEntry(uint32_t accountId, uint32_t guid, const std::string& description, uint32_t icon, bool notify)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "INSERT INTO `account_viplist` (`account_id`, `player_id`, `description`, `icon`, `notify`) VALUES (" << accountId << ',' << guid << ',' << db.escapeString(description) << ',' << icon << ',' << notify << ')';
- db.executeQuery(query.str());
- }
- void IOLoginData::editVIPEntry(uint32_t accountId, uint32_t guid, const std::string& description, uint32_t icon, bool notify)
- {
- Database& db = Database::getInstance();
- std::ostringstream query;
- query << "UPDATE `account_viplist` SET `description` = " << db.escapeString(description) << ", `icon` = " << icon << ", `notify` = " << notify << " WHERE `account_id` = " << accountId << " AND `player_id` = " << guid;
- db.executeQuery(query.str());
- }
- void IOLoginData::removeVIPEntry(uint32_t accountId, uint32_t guid)
- {
- std::ostringstream query;
- query << "DELETE FROM `account_viplist` WHERE `account_id` = " << accountId << " AND `player_id` = " << guid;
- Database::getInstance().executeQuery(query.str());
- }
- void IOLoginData::addPremiumDays(uint32_t accountId, int32_t addDays)
- {
- std::ostringstream query;
- query << "UPDATE `accounts` SET `premdays` = `premdays` + " << addDays << " WHERE `id` = " << accountId;
- Database::getInstance().executeQuery(query.str());
- }
- void IOLoginData::removePremiumDays(uint32_t accountId, int32_t removeDays)
- {
- std::ostringstream query;
- query << "UPDATE `accounts` SET `premdays` = `premdays` - " << removeDays << " WHERE `id` = " << accountId;
- Database::getInstance().executeQuery(query.str());
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement