Update playerlist.cpp
This commit is contained in:
BenCat07
GitHub
parent
9bffc05cd3
commit
d505af9213
@ -1,225 +1,227 @@
|
||||
/*
|
||||
* playerlist.cpp
|
||||
*
|
||||
* Created on: Apr 11, 2017
|
||||
* Author: nullifiedcat
|
||||
*/
|
||||
/*
|
||||
* playerlist.cpp
|
||||
*
|
||||
* Created on: Apr 11, 2017
|
||||
* Author: nullifiedcat
|
||||
*/
|
||||
|
||||
#include "playerlist.hpp"
|
||||
#include "common.hpp"
|
||||
#include "playerlist.hpp"
|
||||
#include "common.hpp"
|
||||
|
||||
#include <stdint.h>
|
||||
#include <dirent.h>
|
||||
#include <sys/stat.h>
|
||||
#include <stdint.h>
|
||||
#include <dirent.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
namespace playerlist
|
||||
{
|
||||
namespace playerlist
|
||||
{
|
||||
|
||||
std::unordered_map<unsigned, userdata> data{};
|
||||
std::unordered_map<unsigned, userdata> data{};
|
||||
|
||||
const userdata null_data{};
|
||||
const userdata null_data{};
|
||||
|
||||
rgba_t k_Colors[] = { colors::empty, colors::FromRGBA8(99, 226, 161, 255),
|
||||
colors::FromRGBA8(226, 204, 99, 255),
|
||||
colors::FromRGBA8(232, 134, 6, 255), colors::empty };
|
||||
rgba_t k_Colors[] = { colors::empty, colors::FromRGBA8(99, 226, 161, 255),
|
||||
colors::FromRGBA8(226, 204, 99, 255),
|
||||
colors::FromRGBA8(232, 134, 6, 255), colors::empty };
|
||||
|
||||
bool ShouldSave(const userdata &data)
|
||||
{
|
||||
return data.color || (data.state != k_EState::DEFAULT);
|
||||
}
|
||||
bool ShouldSave(const userdata &data)
|
||||
{
|
||||
return data.color || (data.state != k_EState::DEFAULT);
|
||||
}
|
||||
|
||||
void Save()
|
||||
{
|
||||
DIR *cathook_directory = opendir(DATA_PATH);
|
||||
if (!cathook_directory)
|
||||
{
|
||||
logging::Info("[ERROR] cathook data directory doesn't exist! How did "
|
||||
"the cheat even get injected?");
|
||||
return;
|
||||
}
|
||||
else
|
||||
closedir(cathook_directory);
|
||||
try
|
||||
{
|
||||
std::ofstream file(DATA_PATH "/plist",
|
||||
std::ios::out | std::ios::binary);
|
||||
file.write(reinterpret_cast<const char *>(&SERIALIZE_VERSION),
|
||||
sizeof(SERIALIZE_VERSION));
|
||||
int size = 0;
|
||||
for (const auto &item : data)
|
||||
{
|
||||
if (ShouldSave(item.second))
|
||||
size++;
|
||||
}
|
||||
file.write(reinterpret_cast<const char *>(&size), sizeof(size));
|
||||
for (const auto &item : data)
|
||||
{
|
||||
if (!ShouldSave(item.second))
|
||||
continue;
|
||||
file.write(reinterpret_cast<const char *>(&item.first),
|
||||
sizeof(item.first));
|
||||
file.write(reinterpret_cast<const char *>(&item.second),
|
||||
sizeof(item.second));
|
||||
}
|
||||
file.close();
|
||||
logging::Info("Writing successful");
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logging::Info("Writing unsuccessful: %s", e.what());
|
||||
}
|
||||
}
|
||||
void Save()
|
||||
{
|
||||
DIR *cathook_directory = opendir(DATA_PATH);
|
||||
if (!cathook_directory)
|
||||
{
|
||||
logging::Info("[ERROR] cathook data directory doesn't exist! How did "
|
||||
"the cheat even get injected?");
|
||||
return;
|
||||
}
|
||||
else
|
||||
closedir(cathook_directory);
|
||||
try
|
||||
{
|
||||
std::ofstream file(DATA_PATH "/plist",
|
||||
std::ios::out | std::ios::binary);
|
||||
file.write(reinterpret_cast<const char *>(&SERIALIZE_VERSION),
|
||||
sizeof(SERIALIZE_VERSION));
|
||||
int size = 0;
|
||||
for (const auto &item : data)
|
||||
{
|
||||
if (ShouldSave(item.second))
|
||||
size++;
|
||||
}
|
||||
file.write(reinterpret_cast<const char *>(&size), sizeof(size));
|
||||
for (const auto &item : data)
|
||||
{
|
||||
if (!ShouldSave(item.second))
|
||||
continue;
|
||||
file.write(reinterpret_cast<const char *>(&item.first),
|
||||
sizeof(item.first));
|
||||
file.write(reinterpret_cast<const char *>(&item.second),
|
||||
sizeof(item.second));
|
||||
}
|
||||
file.close();
|
||||
logging::Info("Writing successful");
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logging::Info("Writing unsuccessful: %s", e.what());
|
||||
}
|
||||
}
|
||||
|
||||
void Load()
|
||||
{
|
||||
data.clear();
|
||||
DIR *cathook_directory = opendir(DATA_PATH);
|
||||
if (!cathook_directory)
|
||||
{
|
||||
logging::Info("[ERROR] cathook data directory doesn't exist! How did "
|
||||
"the cheat even get injected?");
|
||||
return;
|
||||
}
|
||||
else
|
||||
closedir(cathook_directory);
|
||||
try
|
||||
{
|
||||
std::ifstream file(DATA_PATH "/plist", std::ios::in | std::ios::binary);
|
||||
int file_serialize = 0;
|
||||
file.read(reinterpret_cast<char *>(&file_serialize),
|
||||
sizeof(file_serialize));
|
||||
if (file_serialize != SERIALIZE_VERSION)
|
||||
{
|
||||
logging::Info(
|
||||
"Outdated/corrupted playerlist file! Cannot load this.");
|
||||
file.close();
|
||||
return;
|
||||
}
|
||||
int count = 0;
|
||||
file.read(reinterpret_cast<char *>(&count), sizeof(count));
|
||||
logging::Info("Reading %i entries...", count);
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
int steamid;
|
||||
userdata udata;
|
||||
file.read(reinterpret_cast<char *>(&steamid), sizeof(steamid));
|
||||
file.read(reinterpret_cast<char *>(&udata), sizeof(udata));
|
||||
data.emplace(steamid, udata);
|
||||
}
|
||||
file.close();
|
||||
logging::Info("Reading successful!");
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logging::Info("Reading unsuccessful: %s", e.what());
|
||||
}
|
||||
}
|
||||
void Load()
|
||||
{
|
||||
data.clear();
|
||||
DIR *cathook_directory = opendir(DATA_PATH);
|
||||
if (!cathook_directory)
|
||||
{
|
||||
logging::Info("[ERROR] cathook data directory doesn't exist! How did "
|
||||
"the cheat even get injected?");
|
||||
return;
|
||||
}
|
||||
else
|
||||
closedir(cathook_directory);
|
||||
try
|
||||
{
|
||||
std::ifstream file(DATA_PATH "/plist", std::ios::in | std::ios::binary);
|
||||
int file_serialize = 0;
|
||||
file.read(reinterpret_cast<char *>(&file_serialize),
|
||||
sizeof(file_serialize));
|
||||
if (file_serialize != SERIALIZE_VERSION)
|
||||
{
|
||||
logging::Info(
|
||||
"Outdated/corrupted playerlist file! Cannot load this.");
|
||||
file.close();
|
||||
return;
|
||||
}
|
||||
int count = 0;
|
||||
file.read(reinterpret_cast<char *>(&count), sizeof(count));
|
||||
logging::Info("Reading %i entries...", count);
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
int steamid;
|
||||
userdata udata;
|
||||
file.read(reinterpret_cast<char *>(&steamid), sizeof(steamid));
|
||||
file.read(reinterpret_cast<char *>(&udata), sizeof(udata));
|
||||
data.emplace(steamid, udata);
|
||||
}
|
||||
file.close();
|
||||
logging::Info("Reading successful!");
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
logging::Info("Reading unsuccessful: %s", e.what());
|
||||
}
|
||||
}
|
||||
|
||||
rgba_t Color(unsigned steamid)
|
||||
{
|
||||
if (AccessData(steamid).state == k_EState::DEVELOPER)
|
||||
return colors::RainbowCurrent();
|
||||
if (AccessData(steamid).color.a)
|
||||
{
|
||||
return AccessData(steamid).color;
|
||||
}
|
||||
else
|
||||
{
|
||||
return k_Colors[static_cast<int>(AccessData(steamid).state)];
|
||||
}
|
||||
}
|
||||
rgba_t Color(unsigned steamid)
|
||||
{
|
||||
if (AccessData(steamid).state == k_EState::DEVELOPER)
|
||||
return colors::RainbowCurrent();
|
||||
if (AccessData(steamid).state == k_EState::CAT)
|
||||
return colors::RainbowCurrent();
|
||||
if (AccessData(steamid).color.a)
|
||||
{
|
||||
return AccessData(steamid).color;
|
||||
}
|
||||
else
|
||||
{
|
||||
return k_Colors[static_cast<int>(AccessData(steamid).state)];
|
||||
}
|
||||
}
|
||||
|
||||
rgba_t Color(CachedEntity *player)
|
||||
{
|
||||
if (CE_GOOD(player))
|
||||
return Color(player->player_info.friendsID);
|
||||
return colors::empty;
|
||||
}
|
||||
rgba_t Color(CachedEntity *player)
|
||||
{
|
||||
if (CE_GOOD(player))
|
||||
return Color(player->player_info.friendsID);
|
||||
return colors::empty;
|
||||
}
|
||||
|
||||
userdata &AccessData(unsigned steamid)
|
||||
{
|
||||
try
|
||||
{
|
||||
return data.at(steamid);
|
||||
}
|
||||
catch (std::out_of_range &oor)
|
||||
{
|
||||
data.emplace(steamid, userdata{});
|
||||
return data.at(steamid);
|
||||
}
|
||||
}
|
||||
userdata &AccessData(unsigned steamid)
|
||||
{
|
||||
try
|
||||
{
|
||||
return data.at(steamid);
|
||||
}
|
||||
catch (std::out_of_range &oor)
|
||||
{
|
||||
data.emplace(steamid, userdata{});
|
||||
return data.at(steamid);
|
||||
}
|
||||
}
|
||||
|
||||
// Assume player is non-null
|
||||
userdata &AccessData(CachedEntity *player)
|
||||
{
|
||||
if (CE_GOOD(player))
|
||||
return AccessData(player->player_info.friendsID);
|
||||
return AccessData(0U);
|
||||
}
|
||||
// Assume player is non-null
|
||||
userdata &AccessData(CachedEntity *player)
|
||||
{
|
||||
if (CE_GOOD(player))
|
||||
return AccessData(player->player_info.friendsID);
|
||||
return AccessData(0U);
|
||||
}
|
||||
|
||||
bool IsDefault(unsigned steamid)
|
||||
{
|
||||
const userdata &data = AccessData(steamid);
|
||||
return data.state == k_EState::DEFAULT && !data.color.a;
|
||||
}
|
||||
bool IsDefault(unsigned steamid)
|
||||
{
|
||||
const userdata &data = AccessData(steamid);
|
||||
return data.state == k_EState::DEFAULT && !data.color.a;
|
||||
}
|
||||
|
||||
bool IsDefault(CachedEntity *entity)
|
||||
{
|
||||
if (CE_GOOD(entity))
|
||||
return IsDefault(entity->player_info.friendsID);
|
||||
return true;
|
||||
}
|
||||
bool IsDefault(CachedEntity *entity)
|
||||
{
|
||||
if (CE_GOOD(entity))
|
||||
return IsDefault(entity->player_info.friendsID);
|
||||
return true;
|
||||
}
|
||||
|
||||
CatCommand pl_save("pl_save", "Save playerlist", Save);
|
||||
CatCommand pl_load("pl_load", "Load playerlist", Load);
|
||||
CatCommand pl_save("pl_save", "Save playerlist", Save);
|
||||
CatCommand pl_load("pl_load", "Load playerlist", Load);
|
||||
|
||||
CatCommand pl_set_state(
|
||||
"pl_set_state",
|
||||
"pl_set_state uniqueid state\nfor example pl_set_state 306902159 0",
|
||||
[](const CCommand &args) {
|
||||
if (args.ArgC() < 3)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid = strtoul(args.Arg(1), nullptr, 10);
|
||||
k_EState state =
|
||||
static_cast<k_EState>(strtol(args.Arg(2), nullptr, 10));
|
||||
if (state < k_EState::DEFAULT || state > k_EState::STATE_LAST)
|
||||
state = k_EState::DEFAULT;
|
||||
AccessData(steamid).state = state;
|
||||
logging::Info("Set %d to %d", steamid, state);
|
||||
});
|
||||
CatCommand pl_set_state(
|
||||
"pl_set_state",
|
||||
"pl_set_state uniqueid state\nfor example pl_set_state 306902159 0",
|
||||
[](const CCommand &args) {
|
||||
if (args.ArgC() < 3)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid = strtoul(args.Arg(1), nullptr, 10);
|
||||
k_EState state =
|
||||
static_cast<k_EState>(strtol(args.Arg(2), nullptr, 10));
|
||||
if (state < k_EState::DEFAULT || state > k_EState::STATE_LAST)
|
||||
state = k_EState::DEFAULT;
|
||||
AccessData(steamid).state = state;
|
||||
logging::Info("Set %d to %d", steamid, state);
|
||||
});
|
||||
|
||||
CatCommand pl_set_color("pl_set_color", "pl_set_color uniqueid r g b",
|
||||
[](const CCommand &args) {
|
||||
if (args.ArgC() < 5)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid =
|
||||
strtoul(args.Arg(1), nullptr, 10);
|
||||
int r = strtol(args.Arg(2), nullptr, 10);
|
||||
int g = strtol(args.Arg(3), nullptr, 10);
|
||||
int b = strtol(args.Arg(4), nullptr, 10);
|
||||
rgba_t color = colors::FromRGBA8(r, g, b, 255);
|
||||
AccessData(steamid).color = color;
|
||||
logging::Info("Changed %d's color", steamid);
|
||||
});
|
||||
CatCommand pl_set_color("pl_set_color", "pl_set_color uniqueid r g b",
|
||||
[](const CCommand &args) {
|
||||
if (args.ArgC() < 5)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid =
|
||||
strtoul(args.Arg(1), nullptr, 10);
|
||||
int r = strtol(args.Arg(2), nullptr, 10);
|
||||
int g = strtol(args.Arg(3), nullptr, 10);
|
||||
int b = strtol(args.Arg(4), nullptr, 10);
|
||||
rgba_t color = colors::FromRGBA8(r, g, b, 255);
|
||||
AccessData(steamid).color = color;
|
||||
logging::Info("Changed %d's color", steamid);
|
||||
});
|
||||
|
||||
CatCommand pl_info("pl_info", "pl_info uniqueid", [](const CCommand &args) {
|
||||
if (args.ArgC() < 2)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid = strtoul(args.Arg(1), nullptr, 10);
|
||||
logging::Info("Data for %i: ", steamid);
|
||||
logging::Info(" State: %i", AccessData(steamid).state);
|
||||
/*int clr = AccessData(steamid).color;
|
||||
if (clr) {
|
||||
ConColorMsg(*reinterpret_cast<::Color*>(&clr), "[CUSTOM COLOR]\n");
|
||||
}*/
|
||||
});
|
||||
}
|
||||
CatCommand pl_info("pl_info", "pl_info uniqueid", [](const CCommand &args) {
|
||||
if (args.ArgC() < 2)
|
||||
{
|
||||
logging::Info("Invalid call");
|
||||
return;
|
||||
}
|
||||
unsigned steamid = strtoul(args.Arg(1), nullptr, 10);
|
||||
logging::Info("Data for %i: ", steamid);
|
||||
logging::Info(" State: %i", AccessData(steamid).state);
|
||||
/*int clr = AccessData(steamid).color;
|
||||
if (clr) {
|
||||
ConColorMsg(*reinterpret_cast<::Color*>(&clr), "[CUSTOM COLOR]\n");
|
||||
}*/
|
||||
});
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user