Initial commit

Commands/BanIpCommand.cpp

@@ -0,0 +1,97 @@
+#include "stdafx.h"
+#include "BanIpCommand.h"
+#include "ServerCommand.h"
+#include "../../Minecraft.Client/ConsoleInputSource.h"
+#include "../../Minecraft.Client/MinecraftServer.h"
+#include "../../Minecraft.Client/PlayerList.h"
+#include "../../Minecraft.Client/ServerPlayer.h"
+#include "../../Minecraft.Client/PlayerConnection.h"
+#include "../../Minecraft.World/DisconnectPacket.h"
+#include "../../Minecraft.World/Connection.h"
+#include "../../Minecraft.World/Socket.h"
+#include "../../Minecraft.Client/Windows64/Network/WinsockNetLayer.h"
+#include "../Core/ServerLists.h"
+#include "ServerTextList.h"
+
+#include <sstream>
+
+static bool isValidIP(const wstring& str)
+{
+	int dots = 0;
+	for (size_t i = 0; i < str.size(); i++)
+	{
+		if (str[i] == L'.') dots++;
+		else if (str[i] < L'0' || str[i] > L'9') return false;
+	}
+	return dots == 3 && str.size() >= 7;
+}
+
+static wstring narrowToWide(const std::string& s)
+{
+	return wstring(s.begin(), s.end());
+}
+
+void BanIpCommand::execute(const wstring& args, ConsoleInputSource *src, MinecraftServer *server)
+{
+	wstringstream ss(args);
+	wstring arg;
+	if (!(ss >> arg))
+	{
+		src->warn(L"Usage: " + getUsage());
+		return;
+	}
+
+	wstring ip;
+	if (isValidIP(arg))
+	{
+		ip = arg;
+	}
+	else
+	{
+		if (!server || !server->getPlayers())
+		{
+			src->warn(L"Cannot resolve player name, no players online");
+			return;
+		}
+		shared_ptr<ServerPlayer> target = server->getPlayers()->getPlayer(arg);
+		if (!target || !target->connection)
+		{
+			src->warn(L"Cannot find player \"" + arg + L"\"");
+			return;
+		}
+		Connection *conn = target->connection->connection;
+		if (!conn || !conn->getSocket())
+		{
+			src->warn(L"Cannot determine IP for \"" + arg + L"\"");
+			return;
+		}
+		BYTE smallId = conn->getSocket()->getSmallId();
+		std::string ipStr = WinsockNetLayer::GetIPForSmallId(smallId);
+		if (ipStr.empty())
+		{
+			src->warn(L"Cannot determine IP for \"" + arg + L"\"");
+			return;
+		}
+		ip = narrowToWide(ipStr);
+	}
+
+	ServerLists_GetBannedIPs()->add(ip);
+	ServerCommand::notifyAdmins(src, server, L"Banned IP address " + ip);
+
+	if (server && server->getPlayers())
+	{
+		vector<shared_ptr<ServerPlayer>>& pl = server->getPlayers()->players;
+		for (unsigned int i = 0; i < pl.size(); i++)
+		{
+			if (!pl[i] || !pl[i]->connection) continue;
+			Connection *conn = pl[i]->connection->connection;
+			if (!conn || !conn->getSocket()) continue;
+			BYTE sid = conn->getSocket()->getSmallId();
+			std::string pIp = WinsockNetLayer::GetIPForSmallId(sid);
+			if (narrowToWide(pIp) == ip)
+			{
+				server->getPlayers()->kickPlayerByShortId(sid);
+			}
+		}
+	}
+}