368 lines
13 KiB
C++
368 lines
13 KiB
C++
//////////////////////////////////////////////////////////////////////////////////////
|
|
// akashi - a server for Attorney Online 2 //
|
|
// Copyright (C) 2020 scatterflower //
|
|
// //
|
|
// This program is free software: you can redistribute it and/or modify //
|
|
// it under the terms of the GNU Affero General Public License as //
|
|
// published by the Free Software Foundation, either version 3 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 Affero General Public License for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU Affero General Public License //
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>. //
|
|
//////////////////////////////////////////////////////////////////////////////////////
|
|
#include "include/aoclient.h"
|
|
|
|
// Be sure to register the command in the header before adding it here!
|
|
|
|
void AOClient::cmdDefault(int argc, QStringList argv)
|
|
{
|
|
sendServerMessage("Invalid command.");
|
|
return;
|
|
}
|
|
|
|
void AOClient::cmdLogin(int argc, QStringList argv)
|
|
{
|
|
QSettings config("config/config.ini", QSettings::IniFormat);
|
|
config.beginGroup("Options");
|
|
QString modpass = config.value("modpass", "default").toString();
|
|
QString auth_type = config.value("auth", "simple").toString();
|
|
|
|
if (authenticated) {
|
|
sendServerMessage("You are already logged in!");
|
|
return;
|
|
}
|
|
|
|
// TODO: tell the user if no modpass is set
|
|
if (auth_type == "simple") {
|
|
if(argv[0] == modpass) {
|
|
sendServerMessage("Logged in as a moderator."); // This string has to be exactly this, because it is hardcoded in the client
|
|
authenticated = true;
|
|
} else {
|
|
sendServerMessage("Incorrect password.");
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
if (argc < 2) {
|
|
sendServerMessage("You must specify a username and a password");
|
|
return;
|
|
}
|
|
QString username = argv[0];
|
|
QString password = argv[1];
|
|
if (server->db_manager->authenticate(username, password)) {
|
|
moderator_name = username;
|
|
authenticated = true;
|
|
sendServerMessage("Logged in as " + username);
|
|
}
|
|
else {
|
|
sendServerMessage("Incorrect password.");
|
|
}
|
|
}
|
|
}
|
|
|
|
void AOClient::cmdGetAreas(int argc, QStringList argv)
|
|
{
|
|
QStringList entries;
|
|
entries.append("== Area List ==");
|
|
for (int i = 0; i < server->area_names.length(); i++) {
|
|
QStringList cur_area_lines = buildAreaList(i);
|
|
entries.append(cur_area_lines);
|
|
}
|
|
sendServerMessage(entries.join("\n"));
|
|
}
|
|
|
|
void AOClient::cmdGetArea(int argc, QStringList argv)
|
|
{
|
|
QStringList entries = buildAreaList(current_area);
|
|
sendServerMessage(entries.join("\n"));
|
|
}
|
|
|
|
void AOClient::cmdBan(int argc, QStringList argv)
|
|
{
|
|
QString target_ipid = argv[0];
|
|
QHostAddress ip;
|
|
QString hdid;
|
|
unsigned long time = QDateTime::currentDateTime().toTime_t();
|
|
QString reason = argv[1];
|
|
bool ban_logged = false;
|
|
|
|
if (argc > 2) {
|
|
for (int i = 2; i < argv.length(); i++) {
|
|
reason += " " + argv[i];
|
|
}
|
|
}
|
|
|
|
for (AOClient* client : server->clients) {
|
|
if (client->getIpid() == target_ipid) {
|
|
if (!ban_logged) {
|
|
ip = client->remote_ip;
|
|
hdid = client->hwid;
|
|
server->db_manager->addBan(target_ipid, ip, hdid, time, reason);
|
|
sendServerMessage("Banned user with ipid " + target_ipid + " for reason: " + reason);
|
|
ban_logged = true;
|
|
}
|
|
client->sendPacket("KB", {reason});
|
|
client->socket->close();
|
|
}
|
|
}
|
|
|
|
if (!ban_logged)
|
|
sendServerMessage("User with ipid not found!");
|
|
}
|
|
|
|
void AOClient::cmdKick(int argc, QStringList argv)
|
|
{
|
|
QString target_ipid = argv[0];
|
|
QString reason = argv[1];
|
|
bool did_kick = false;
|
|
|
|
if (argc > 2) {
|
|
for (int i = 2; i < argv.length(); i++) {
|
|
reason += " " + argv[i];
|
|
}
|
|
}
|
|
|
|
for (AOClient* client : server->clients) {
|
|
if (client->getIpid() == target_ipid) {
|
|
client->sendPacket("KK", {reason});
|
|
client->socket->close();
|
|
did_kick = true;
|
|
}
|
|
}
|
|
|
|
if (did_kick)
|
|
sendServerMessage("Kicked user with ipid " + target_ipid + " for reason: " + reason);
|
|
else
|
|
sendServerMessage("User with ipid not found!");
|
|
}
|
|
|
|
void AOClient::cmdChangeAuth(int argc, QStringList argv)
|
|
{
|
|
QSettings settings("config/config.ini", QSettings::IniFormat);
|
|
settings.beginGroup("Options");
|
|
QString auth_type = settings.value("auth", "simple").toString();
|
|
|
|
if (auth_type == "simple") {
|
|
change_auth_started = true;
|
|
sendServerMessage("WARNING!\nThis command will change how logging in as a moderator works.\nOnly proceed if you know what you are doing\nUse the command /rootpass to set the password for your root account.");
|
|
}
|
|
}
|
|
|
|
void AOClient::cmdSetRootPass(int argc, QStringList argv)
|
|
{
|
|
if (!change_auth_started)
|
|
return;
|
|
|
|
sendServerMessage("Changing auth type and setting root password.\nLogin again with /login root [password]");
|
|
authenticated = false;
|
|
QSettings settings("config/config.ini", QSettings::IniFormat);
|
|
settings.beginGroup("Options");
|
|
settings.setValue("auth", "advanced");
|
|
|
|
quint64 salt_number = QRandomGenerator::system()->generate64();
|
|
QString salt = QStringLiteral("%1").arg(salt_number, 16, 16, QLatin1Char('0'));
|
|
|
|
server->db_manager->createUser("root", salt, argv[0], ACLFlags.value("SUPER"));
|
|
}
|
|
|
|
void AOClient::cmdSetBackground(int argc, QStringList argv)
|
|
{
|
|
// TODO: area locks in areas.ini
|
|
// TODO: sendServerMessage but area broadcast
|
|
AreaData* area = server->areas[current_area];
|
|
if (authenticated || !area->bg_locked) {
|
|
if (server->backgrounds.contains(argv[0])) {
|
|
area->background = argv[0];
|
|
server->broadcast(AOPacket("BN", {argv[0]}), current_area);
|
|
server->broadcast(AOPacket("CT", {"Server", current_char + " changed the background to " + argv[0], "1"}), current_area);
|
|
}
|
|
else {
|
|
sendServerMessage("Invalid background name.");
|
|
}
|
|
}
|
|
else {
|
|
sendServerMessage("This area's background is locked.");
|
|
}
|
|
}
|
|
|
|
void AOClient::cmdBgLock(int argc, QStringList argv)
|
|
{
|
|
AreaData* area = server->areas[current_area];
|
|
area->bg_locked = true;
|
|
server->broadcast(AOPacket("CT", {"Server", current_char + " locked the background.", "1"}), current_area);
|
|
}
|
|
|
|
void AOClient::cmdBgUnlock(int argc, QStringList argv)
|
|
{
|
|
AreaData* area = server->areas[current_area];
|
|
area->bg_locked = false;
|
|
server->broadcast(AOPacket("CT", {"Server", current_char + " unlocked the background.", "1"}), current_area);
|
|
}
|
|
|
|
void AOClient::cmdAddUser(int argc, QStringList argv)
|
|
{
|
|
quint64 salt_number = QRandomGenerator::system()->generate64();
|
|
QString salt = QStringLiteral("%1").arg(salt_number, 16, 16, QLatin1Char('0'));
|
|
|
|
if (server->db_manager->createUser(argv[0], salt, argv[1], ACLFlags.value("NONE")))
|
|
sendServerMessage("Created user " + argv[0] + ".\nUse /addperm to modify their permissions.");
|
|
else
|
|
sendServerMessage("Unable to create user " + argv[0] + ".\nDoes a user with that name already exist?");
|
|
}
|
|
|
|
void AOClient::cmdListPerms(int argc, QStringList argv)
|
|
{
|
|
unsigned long long user_acl = server->db_manager->getACL(moderator_name);
|
|
QStringList message;
|
|
if (argc == 0) {
|
|
// Just print out all permissions available to the user.
|
|
message.append("You have been given the following permissions:");
|
|
for (QString perm : ACLFlags.keys()) {
|
|
if (perm == "NONE"); // don't need to list this one
|
|
else if (perm == "SUPER") {
|
|
if (user_acl == ACLFlags.value("SUPER")) // This has to be checked separately, because SUPER & anything will always be truthy
|
|
message.append("SUPER (Be careful! This grants the user all permissions.)");
|
|
}
|
|
else if ((ACLFlags.value(perm) & user_acl) == 0); // user doesn't have this permission, don't print it
|
|
else
|
|
message.append(perm);
|
|
}
|
|
}
|
|
else {
|
|
if ((user_acl & ACLFlags.value("MODIFY_USERS")) == 0) {
|
|
sendServerMessage("You do not have permission to view other users' permissions.");
|
|
return;
|
|
}
|
|
|
|
message.append("User " + argv[0] + " has the following permissions:");
|
|
unsigned long long acl = server->db_manager->getACL(argv[0]);
|
|
if (acl == 0) {
|
|
sendServerMessage("This user either doesn't exist, or has no permissions set.");
|
|
return;
|
|
}
|
|
|
|
for (QString perm : ACLFlags.keys()) {
|
|
if ((ACLFlags.value(perm) & acl) != 0 && perm != "SUPER") {
|
|
message.append(perm);
|
|
}
|
|
}
|
|
}
|
|
sendServerMessage(message.join("\n"));
|
|
}
|
|
|
|
void AOClient::cmdAddPerms(int argc, QStringList argv)
|
|
{
|
|
unsigned long long user_acl = server->db_manager->getACL(moderator_name);
|
|
argv[1] = argv[1].toUpper();
|
|
|
|
if (!ACLFlags.keys().contains(argv[1])) {
|
|
sendServerMessage("That permission doesn't exist!");
|
|
return;
|
|
}
|
|
|
|
if (argv[1] == "SUPER") {
|
|
if (user_acl != ACLFlags.value("SUPER")) {
|
|
// This has to be checked separately, because SUPER & anything will always be truthy
|
|
sendServerMessage("You aren't allowed to add that permission!");
|
|
return;
|
|
}
|
|
}
|
|
if (argv[1] == "NONE") {
|
|
sendServerMessage("Added no permissions!");
|
|
return;
|
|
}
|
|
|
|
unsigned long long newperm = ACLFlags.value(argv[1]);
|
|
if ((newperm & user_acl) != 0) {
|
|
if (server->db_manager->updateACL(argv[0], newperm, true))
|
|
sendServerMessage("Successfully added permission " + argv[1] + " to user " + argv[0]);
|
|
else
|
|
sendServerMessage(argv[0] + " wasn't found!");
|
|
return;
|
|
}
|
|
|
|
sendServerMessage("You aren't allowed to add that permission!");
|
|
}
|
|
|
|
void AOClient::cmdRemovePerms(int argc, QStringList argv)
|
|
{
|
|
unsigned long long user_acl = server->db_manager->getACL(moderator_name);
|
|
argv[1] = argv[1].toUpper();
|
|
|
|
if (!ACLFlags.keys().contains(argv[1])) {
|
|
sendServerMessage("That permission doesn't exist!");
|
|
return;
|
|
}
|
|
|
|
if (argv[0] == "root") {
|
|
sendServerMessage("You cannot change the permissions of the root account!");
|
|
return;
|
|
}
|
|
|
|
if (argv[1] == "SUPER") {
|
|
if (user_acl != ACLFlags.value("SUPER")) {
|
|
// This has to be checked separately, because SUPER & anything will always be truthy
|
|
sendServerMessage("You aren't allowed to remove that permission!");
|
|
return;
|
|
}
|
|
}
|
|
if (argv[1] == "NONE") {
|
|
sendServerMessage("Removed no permissions!");
|
|
return;
|
|
}
|
|
|
|
unsigned long long newperm = ACLFlags.value(argv[1]);
|
|
if ((newperm & user_acl) != 0) {
|
|
if (server->db_manager->updateACL(argv[0], newperm, false))
|
|
sendServerMessage("Successfully removed permission " + argv[1] + " from user " + argv[0]);
|
|
else
|
|
sendServerMessage(argv[0] + " wasn't found!");
|
|
return;
|
|
}
|
|
|
|
sendServerMessage("You aren't allowed to remove that permission!");
|
|
}
|
|
|
|
void AOClient::cmdListUsers(int argc, QStringList argv)
|
|
{
|
|
QStringList users = server->db_manager->getUsers();
|
|
sendServerMessage("All users:\n" + users.join("\n"));
|
|
}
|
|
|
|
void AOClient::cmdLogout(int argc, QStringList argv)
|
|
{
|
|
if (!authenticated) {
|
|
sendServerMessage("You are not logged in!");
|
|
return;
|
|
}
|
|
authenticated = false;
|
|
moderator_name = "";
|
|
sendServerMessage("You have been logged out.");
|
|
return;
|
|
}
|
|
|
|
QStringList AOClient::buildAreaList(int area_idx)
|
|
{
|
|
QStringList entries;
|
|
QString area_name = server->area_names[area_idx];
|
|
AreaData* area = server->areas[area_idx];
|
|
entries.append("=== " + area_name + " ===");
|
|
entries.append("[" + QString::number(area->player_count) + " users][" + area->status + "]");
|
|
for (AOClient* client : server->clients) {
|
|
if (client->current_area == area_idx && client->joined) {
|
|
QString char_entry = client->current_char;
|
|
if (char_entry == "")
|
|
char_entry = "Spectator";
|
|
if (authenticated)
|
|
char_entry += " (" + client->getIpid() + "): " + client->ooc_name;
|
|
entries.append(char_entry);
|
|
}
|
|
}
|
|
return entries;
|
|
}
|