18 changed files with 1090 additions and 254 deletions
Binary file not shown.
Binary file not shown.
@ -0,0 +1,197 @@ |
|||||
|
/**
|
||||
|
* Copyright (C) 2016 Jessica James. |
||||
|
* |
||||
|
* Permission to use, copy, modify, and/or distribute this software for any |
||||
|
* purpose with or without fee is hereby granted, provided that the above |
||||
|
* copyright notice and this permission notice appear in all copies. |
||||
|
* |
||||
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
||||
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
||||
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY |
||||
|
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
||||
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION |
||||
|
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN |
||||
|
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
||||
|
* |
||||
|
* Written by Jessica James <jessica.aj@outlook.com> |
||||
|
*/ |
||||
|
|
||||
|
#include <cstdio> |
||||
|
#include "Jupiter/IRC_Client.h" |
||||
|
#include "Jupiter/INIFile.h" |
||||
|
#include "RenX_PlayerInfo.h" |
||||
|
#include "RenX_ExemptionDatabase.h" |
||||
|
#include "RenX_Core.h" |
||||
|
#include "RenX_Plugin.h" |
||||
|
|
||||
|
using namespace Jupiter::literals; |
||||
|
|
||||
|
RenX::ExemptionDatabase _exemptionDatabase; |
||||
|
RenX::ExemptionDatabase *RenX::exemptionDatabase = &_exemptionDatabase; |
||||
|
RenX::ExemptionDatabase &RenX::defaultExemptionDatabase = _exemptionDatabase; |
||||
|
|
||||
|
void RenX::ExemptionDatabase::process_data(Jupiter::DataBuffer &buffer, FILE *file, fpos_t pos) |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::Entry *entry = new RenX::ExemptionDatabase::Entry(); |
||||
|
entry->pos = pos; |
||||
|
|
||||
|
// Read data from buffer to entry
|
||||
|
entry->flags = buffer.pop<uint8_t>(); |
||||
|
entry->timestamp = std::chrono::system_clock::time_point(std::chrono::seconds(buffer.pop<uint64_t>())); |
||||
|
entry->length = std::chrono::seconds(buffer.pop<int64_t>()); |
||||
|
entry->steamid = buffer.pop<uint64_t>(); |
||||
|
entry->ip = buffer.pop<uint32_t>(); |
||||
|
entry->prefix_length = buffer.pop<uint8_t>(); |
||||
|
entry->setter = buffer.pop<Jupiter::String_Strict, char>(); |
||||
|
|
||||
|
RenX::ExemptionDatabase::entries.add(entry); |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::process_header(FILE *file) |
||||
|
{ |
||||
|
int chr = fgetc(file); |
||||
|
if (chr != EOF) |
||||
|
RenX::ExemptionDatabase::read_version = chr; |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::create_header(FILE *file) |
||||
|
{ |
||||
|
fputc(RenX::ExemptionDatabase::write_version, file); |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::process_file_finish(FILE *file) |
||||
|
{ |
||||
|
fgetpos(file, std::addressof(RenX::ExemptionDatabase::eof)); |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::upgrade_database() |
||||
|
{ |
||||
|
FILE *file = fopen(RenX::ExemptionDatabase::filename.c_str(), "wb"); |
||||
|
if (file != nullptr) |
||||
|
{ |
||||
|
this->create_header(file); |
||||
|
for (size_t index = 0; RenX::ExemptionDatabase::entries.size(); ++index) |
||||
|
RenX::ExemptionDatabase::write(RenX::ExemptionDatabase::entries.get(index), file); |
||||
|
|
||||
|
fclose(file); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::write(RenX::ExemptionDatabase::Entry *entry) |
||||
|
{ |
||||
|
FILE *file = fopen(filename.c_str(), "r+b"); |
||||
|
fsetpos(file, std::addressof(RenX::ExemptionDatabase::eof)); |
||||
|
if (file != nullptr) |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::write(entry, file); |
||||
|
fclose(file); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::write(RenX::ExemptionDatabase::Entry *entry, FILE *file) |
||||
|
{ |
||||
|
Jupiter::DataBuffer buffer; |
||||
|
fgetpos(file, &entry->pos); |
||||
|
|
||||
|
// push data from entry to buffer
|
||||
|
buffer.push(entry->flags); |
||||
|
buffer.push(static_cast<int64_t>(std::chrono::duration_cast<std::chrono::seconds>(entry->timestamp.time_since_epoch()).count())); |
||||
|
buffer.push(static_cast<int64_t>(entry->length.count())); |
||||
|
buffer.push(entry->steamid); |
||||
|
buffer.push(entry->ip); |
||||
|
buffer.push(entry->prefix_length); |
||||
|
buffer.push(entry->setter); |
||||
|
|
||||
|
// push buffer to file
|
||||
|
buffer.push_to(file); |
||||
|
fgetpos(file, std::addressof(RenX::ExemptionDatabase::eof)); |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::add(RenX::Server *server, const RenX::PlayerInfo *player, const Jupiter::ReadableString &setter, std::chrono::seconds length, uint8_t flags) |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::add(player->ip32, 32U, player->steamid, setter, length, flags); |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::add(uint32_t ip, uint8_t prefix_length, uint64_t steamid, const Jupiter::ReadableString &setter, std::chrono::seconds length, uint8_t flags) |
||||
|
{ |
||||
|
Entry *entry = new Entry(); |
||||
|
entry->set_active(); |
||||
|
entry->flags |= flags; |
||||
|
entry->timestamp = std::chrono::system_clock::now(); |
||||
|
entry->length = length; |
||||
|
entry->steamid = steamid; |
||||
|
entry->ip = ip; |
||||
|
entry->prefix_length = prefix_length; |
||||
|
entry->setter = setter; |
||||
|
|
||||
|
entries.add(entry); |
||||
|
RenX::ExemptionDatabase::write(entry); |
||||
|
} |
||||
|
|
||||
|
bool RenX::ExemptionDatabase::deactivate(size_t index) |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::Entry *entry = RenX::ExemptionDatabase::entries.get(index); |
||||
|
if (entry->is_active()) |
||||
|
{ |
||||
|
entry->unset_active(); |
||||
|
FILE *file = fopen(RenX::ExemptionDatabase::filename.c_str(), "r+b"); |
||||
|
if (file != nullptr) |
||||
|
{ |
||||
|
fsetpos(file, &entry->pos); |
||||
|
fseek(file, sizeof(size_t), SEEK_CUR); |
||||
|
fwrite(std::addressof(entry->flags), sizeof(entry->flags), 1, file); |
||||
|
fclose(file); |
||||
|
} |
||||
|
return true; |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
void RenX::ExemptionDatabase::exemption_check(RenX::PlayerInfo *player) |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::Entry *entry; |
||||
|
uint32_t netmask; |
||||
|
size_t index = RenX::ExemptionDatabase::entries.size(); |
||||
|
while (index != 0) |
||||
|
{ |
||||
|
entry = RenX::ExemptionDatabase::entries.get(--index); |
||||
|
if (entry->is_active()) |
||||
|
{ |
||||
|
if (entry->length == std::chrono::seconds::zero() || entry->timestamp + entry->length < std::chrono::system_clock::now()) |
||||
|
{ |
||||
|
netmask = Jupiter_prefix_length_to_netmask(entry->prefix_length); |
||||
|
if ((player->steamid != 0 && entry->steamid == player->steamid) // SteamID exemption
|
||||
|
|| (player->ip32 != 0U && (player->ip32 & netmask) == (entry->ip & netmask))) // IP address exemption
|
||||
|
player->exemption_flags |= entry->flags; |
||||
|
} |
||||
|
else |
||||
|
RenX::ExemptionDatabase::deactivate(index); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
uint8_t RenX::ExemptionDatabase::getVersion() const |
||||
|
{ |
||||
|
return RenX::ExemptionDatabase::write_version; |
||||
|
} |
||||
|
|
||||
|
const Jupiter::ReadableString &RenX::ExemptionDatabase::getFileName() const |
||||
|
{ |
||||
|
return RenX::ExemptionDatabase::filename; |
||||
|
} |
||||
|
|
||||
|
const Jupiter::ArrayList<RenX::ExemptionDatabase::Entry> &RenX::ExemptionDatabase::getEntries() const |
||||
|
{ |
||||
|
return RenX::ExemptionDatabase::entries; |
||||
|
} |
||||
|
|
||||
|
RenX::ExemptionDatabase::ExemptionDatabase() |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::filename = Jupiter::IRC::Client::Config->get(STRING_LITERAL_AS_REFERENCE("RenX"), STRING_LITERAL_AS_REFERENCE("ExemptionDB"), STRING_LITERAL_AS_REFERENCE("Exemptions.db")); |
||||
|
this->process_file(filename); |
||||
|
} |
||||
|
|
||||
|
RenX::ExemptionDatabase::~ExemptionDatabase() |
||||
|
{ |
||||
|
RenX::ExemptionDatabase::entries.emptyAndDelete(); |
||||
|
} |
||||
@ -0,0 +1,202 @@ |
|||||
|
/**
|
||||
|
* Copyright (C) 2016 Jessica James. |
||||
|
* |
||||
|
* Permission to use, copy, modify, and/or distribute this software for any |
||||
|
* purpose with or without fee is hereby granted, provided that the above |
||||
|
* copyright notice and this permission notice appear in all copies. |
||||
|
* |
||||
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
||||
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
||||
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY |
||||
|
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
||||
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION |
||||
|
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN |
||||
|
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
||||
|
* |
||||
|
* Written by Jessica James <jessica.aj@outlook.com> |
||||
|
*/ |
||||
|
|
||||
|
#if !defined _RENX_EXEMPTIONDATABASE_H_HEADER |
||||
|
#define _RENX_EXEMPTIONDATABASE_H_HEADER |
||||
|
|
||||
|
#include <cstdint> |
||||
|
#include <chrono> |
||||
|
#include "Jupiter/Database.h" |
||||
|
#include "Jupiter/String.h" |
||||
|
#include "Jupiter/CString.h" |
||||
|
#include "Jupiter/ArrayList.h" |
||||
|
#include "RenX.h" |
||||
|
|
||||
|
/** DLL Linkage Nagging */ |
||||
|
#if defined _MSC_VER |
||||
|
#pragma warning(push) |
||||
|
#pragma warning(disable: 4251) |
||||
|
#endif |
||||
|
|
||||
|
namespace RenX |
||||
|
{ |
||||
|
struct PlayerInfo; |
||||
|
class Server; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Represents the local exemption database. |
||||
|
*/ |
||||
|
class RENX_API ExemptionDatabase : public Jupiter::Database |
||||
|
{ |
||||
|
public: // Jupiter::Database
|
||||
|
/**
|
||||
|
* @brief Processes a chunk of data in a database. |
||||
|
* |
||||
|
* @param buffer Buffer to process |
||||
|
* @param file File being processed |
||||
|
* @param pos position that the buffer starts at in the file |
||||
|
*/ |
||||
|
void process_data(Jupiter::DataBuffer &buffer, FILE *file, fpos_t pos) override; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Processes the header for a database. |
||||
|
* |
||||
|
* @param file File being processed |
||||
|
*/ |
||||
|
void process_header(FILE *file) override; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Generates a header for a database. |
||||
|
* |
||||
|
* @param file File being created |
||||
|
*/ |
||||
|
void create_header(FILE *file) override; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Called when process_file() is successfully completed. |
||||
|
* |
||||
|
* @param file File being processed |
||||
|
*/ |
||||
|
void process_file_finish(FILE *file) override; |
||||
|
|
||||
|
struct RENX_API Entry |
||||
|
{ |
||||
|
fpos_t pos; /** Position of the entry in the database */ |
||||
|
uint8_t flags /** Flags affecting this exemption entry (See below for flags) */ = 0x00; |
||||
|
std::chrono::system_clock::time_point timestamp /** Time the exemption was created */; |
||||
|
std::chrono::seconds length /** Duration of the exemption; 0 if permanent */; |
||||
|
uint64_t steamid /** SteamID of the exempted player */; |
||||
|
uint32_t ip /** IPv4 address of the exempted player */; |
||||
|
uint8_t prefix_length /** Prefix length for the IPv4 address block */; |
||||
|
Jupiter::StringS setter /** Name of the user who added the exemption */; |
||||
|
|
||||
|
static const uint8_t FLAG_ACTIVE = 0x80; |
||||
|
static const uint8_t FLAG_USE_IP = 0x40; |
||||
|
static const uint8_t FLAG_TYPE_KICK = 0x04; |
||||
|
static const uint8_t FLAG_TYPE_BAN = 0x02; |
||||
|
|
||||
|
inline bool is_active() { return (flags & FLAG_ACTIVE) != 0; }; |
||||
|
inline bool is_ip_exemption() { return (flags & FLAG_USE_IP) != 0; }; |
||||
|
inline bool is_type_kick() { return (flags & FLAG_TYPE_KICK) != 0; }; |
||||
|
inline bool is_type_ban() { return (flags & FLAG_TYPE_BAN) != 0; }; |
||||
|
|
||||
|
inline void set_active() { flags |= FLAG_ACTIVE; }; |
||||
|
inline void set_ip_exemption() { flags |= FLAG_USE_IP; }; |
||||
|
inline void set_type_kick() { flags |= FLAG_TYPE_KICK; }; |
||||
|
inline void set_type_ban() { flags |= FLAG_TYPE_BAN; }; |
||||
|
|
||||
|
inline void unset_active() { flags &= ~FLAG_ACTIVE; }; |
||||
|
inline void unset_ip_exemption() { flags &= ~FLAG_USE_IP; }; |
||||
|
inline void unset_type_kick() { flags &= ~FLAG_TYPE_KICK; }; |
||||
|
inline void unset_type_ban() { flags &= ~FLAG_TYPE_BAN; }; |
||||
|
}; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Adds an exemption entry for a player and immediately writes it to the database. |
||||
|
* |
||||
|
* @param server Server the player is playing in |
||||
|
* @param player Data of the player to be exempted |
||||
|
* @param length Duration of the exempt |
||||
|
*/ |
||||
|
void add(RenX::Server *server, const RenX::PlayerInfo *player, const Jupiter::ReadableString &setter, std::chrono::seconds length, uint8_t flags); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Adds an exemption entry for a set of player information and immediately writes it to the database. |
||||
|
* |
||||
|
* @param name Name of the player to exempt |
||||
|
* @param ip IPv4 address of the player to exempt |
||||
|
* @param steamid SteamID of the player to exempt |
||||
|
* @param setter Person implementing the exempt |
||||
|
* @param length Duration of the exemption |
||||
|
*/ |
||||
|
void add(uint32_t ip, uint8_t prefix_length, uint64_t steamid, const Jupiter::ReadableString &setter, std::chrono::seconds length, uint8_t flags); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Upgrades the exemption database to the current write_version. |
||||
|
*/ |
||||
|
void upgrade_database(); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Writes an exemption entry to the database. |
||||
|
* |
||||
|
* @param entry Entry to write to the database. |
||||
|
*/ |
||||
|
void write(Entry *entry); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Writes an exemption entry to the database. |
||||
|
* |
||||
|
* @param entry Entry to write to the database. |
||||
|
* @param file FILE stream to write to. |
||||
|
*/ |
||||
|
void write(Entry *entry, FILE *file); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Deactivates an exemption entry. |
||||
|
* |
||||
|
* @param index Index of the entry to deactivate. |
||||
|
* @param True if the entry was active and is now inactive, false otherwise. |
||||
|
*/ |
||||
|
bool deactivate(size_t index); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Checks a player for any relevant ban exemptions, and assigns their exemption_flags |
||||
|
* |
||||
|
* @param player Player to check exemption flags for |
||||
|
*/ |
||||
|
void exemption_check(RenX::PlayerInfo *player); |
||||
|
|
||||
|
/**
|
||||
|
* @brief Fetches the version of the database file. |
||||
|
* |
||||
|
* @return Database version |
||||
|
*/ |
||||
|
uint8_t getVersion() const; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Fetches the name of the database file. |
||||
|
* |
||||
|
* @return Database file name |
||||
|
*/ |
||||
|
const Jupiter::ReadableString &getFileName() const; |
||||
|
|
||||
|
/**
|
||||
|
* @brief Fetches the list of exemption entries. |
||||
|
* |
||||
|
* @return List of entries |
||||
|
*/ |
||||
|
const Jupiter::ArrayList<RenX::ExemptionDatabase::Entry> &getEntries() const; |
||||
|
|
||||
|
ExemptionDatabase(); |
||||
|
~ExemptionDatabase(); |
||||
|
|
||||
|
private: |
||||
|
/** Database version */ |
||||
|
const uint8_t write_version = 0U; |
||||
|
uint8_t read_version = write_version; |
||||
|
fpos_t eof; |
||||
|
|
||||
|
Jupiter::CStringS filename; |
||||
|
Jupiter::ArrayList<RenX::ExemptionDatabase::Entry> entries; |
||||
|
}; |
||||
|
|
||||
|
RENX_API extern RenX::ExemptionDatabase *exemptionDatabase; |
||||
|
RENX_API extern RenX::ExemptionDatabase &defaultExemptionDatabase; |
||||
|
} |
||||
|
|
||||
|
#endif // _RENX_EXEMPTIONDATABASE_H_HEADER
|
||||
Loading…
Reference in new issue