1.1 --- a/core/smn_banning.cpp Fri Mar 28 06:21:34 2008 +0000
1.2 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000
1.3 @@ -1,396 +0,0 @@
1.4 -/**
1.5 - * vim: set ts=4 :
1.6 - * =============================================================================
1.7 - * SourceMod
1.8 - * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved.
1.9 - * =============================================================================
1.10 - *
1.11 - * This program is free software; you can redistribute it and/or modify it under
1.12 - * the terms of the GNU General Public License, version 3.0, as published by the
1.13 - * Free Software Foundation.
1.14 - *
1.15 - * This program is distributed in the hope that it will be useful, but WITHOUT
1.16 - * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.17 - * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.18 - * details.
1.19 - *
1.20 - * You should have received a copy of the GNU General Public License along with
1.21 - * this program. If not, see <http://www.gnu.org/licenses/>.
1.22 - *
1.23 - * As a special exception, AlliedModders LLC gives you permission to link the
1.24 - * code of this program (as well as its derivative works) to "Half-Life 2," the
1.25 - * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.26 - * by the Valve Corporation. You must obey the GNU General Public License in
1.27 - * all respects for all other code used. Additionally, AlliedModders LLC grants
1.28 - * this exception to all derivative works. AlliedModders LLC defines further
1.29 - * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.30 - * or <http://www.sourcemod.net/license.php>.
1.31 - *
1.32 - * Version: $Id$
1.33 - */
1.34 -
1.35 -#include "sm_globals.h"
1.36 -#include "sm_stringutil.h"
1.37 -#include "HalfLife2.h"
1.38 -#include "PlayerManager.h"
1.39 -#include "ForwardSys.h"
1.40 -
1.41 -#define BANFLAG_AUTO (1<<0) /**< Auto-detects whether to ban by steamid or IP */
1.42 -#define BANFLAG_IP (1<<1) /**< Always ban by IP address */
1.43 -#define BANFLAG_AUTHID (1<<2) /**< Ban by SteamID */
1.44 -#define BANFLAG_NOKICK (1<<3) /**< Does not kick the client */
1.45 -#define BANFLAG_NOWRITE (1<<4) /**< Ban is not written to SourceDS's files if permanent */
1.46 -
1.47 -IForward *g_pOnBanClient = NULL;
1.48 -IForward *g_pOnBanIdentity = NULL;
1.49 -IForward *g_pOnRemoveBan = NULL;
1.50 -
1.51 -class BanNativeHelpers : public SMGlobalClass
1.52 -{
1.53 -public:
1.54 - void OnSourceModAllInitialized()
1.55 - {
1.56 - g_pOnBanClient = g_Forwards.CreateForward(
1.57 - "OnBanClient",
1.58 - ET_Event,
1.59 - 7,
1.60 - NULL,
1.61 - Param_Cell,
1.62 - Param_Cell,
1.63 - Param_Cell,
1.64 - Param_String,
1.65 - Param_String,
1.66 - Param_String,
1.67 - Param_Cell);
1.68 - g_pOnBanIdentity = g_Forwards.CreateForward(
1.69 - "OnBanIdentity",
1.70 - ET_Event,
1.71 - 6,
1.72 - NULL,
1.73 - Param_String,
1.74 - Param_Cell,
1.75 - Param_Cell,
1.76 - Param_String,
1.77 - Param_String,
1.78 - Param_Cell);
1.79 - g_pOnRemoveBan = g_Forwards.CreateForward(
1.80 - "OnRemoveBan",
1.81 - ET_Event,
1.82 - 4,
1.83 - NULL,
1.84 - Param_String,
1.85 - Param_Cell,
1.86 - Param_String,
1.87 - Param_Cell);
1.88 - }
1.89 - void OnSourceModShutdown()
1.90 - {
1.91 - g_Forwards.ReleaseForward(g_pOnBanClient);
1.92 - g_Forwards.ReleaseForward(g_pOnBanIdentity);
1.93 - g_Forwards.ReleaseForward(g_pOnRemoveBan);
1.94 -
1.95 - g_pOnBanClient = NULL;
1.96 - g_pOnBanIdentity = NULL;
1.97 - g_pOnRemoveBan = NULL;
1.98 - }
1.99 -} s_BanNativeHelpers;
1.100 -
1.101 -
1.102 -static cell_t BanIdentity(IPluginContext *pContext, const cell_t *params)
1.103 -{
1.104 - char *r_identity, *ban_reason, *ban_cmd;
1.105 - int ban_time, ban_flags, ban_source;
1.106 -
1.107 - pContext->LocalToString(params[1], &r_identity);
1.108 - pContext->LocalToString(params[4], &ban_reason);
1.109 - pContext->LocalToString(params[5], &ban_cmd);
1.110 - ban_time = params[2];
1.111 - ban_flags = params[3];
1.112 - ban_source = params[6];
1.113 -
1.114 - /* Make sure we can ban by one of the two methods! */
1.115 - bool ban_by_ip = ((ban_flags & BANFLAG_IP) == BANFLAG_IP);
1.116 - if (!ban_by_ip && ((ban_flags & BANFLAG_AUTHID) != BANFLAG_AUTHID))
1.117 - {
1.118 - return pContext->ThrowNativeError("No valid ban flags specified");
1.119 - }
1.120 -
1.121 - /* Sanitize the input */
1.122 - char identity[64];
1.123 - strncopy(identity, r_identity, sizeof(identity));
1.124 - UTIL_ReplaceAll(identity, sizeof(identity), ";", "");
1.125 -
1.126 - cell_t handled = 0;
1.127 - if (ban_cmd[0] != '\0' && g_pOnBanIdentity->GetFunctionCount() > 0)
1.128 - {
1.129 - g_pOnBanIdentity->PushString(identity);
1.130 - g_pOnBanIdentity->PushCell(ban_time);
1.131 - g_pOnBanIdentity->PushCell(ban_flags);
1.132 - g_pOnBanIdentity->PushString(ban_reason);
1.133 - g_pOnBanIdentity->PushString(ban_cmd);
1.134 - g_pOnBanIdentity->PushCell(ban_source);
1.135 - g_pOnBanIdentity->Execute(&handled);
1.136 - }
1.137 -
1.138 - if (!handled)
1.139 - {
1.140 - bool write_ban = ((ban_flags & BANFLAG_NOWRITE) != BANFLAG_NOWRITE);
1.141 -
1.142 - char command[256];
1.143 - if (ban_by_ip)
1.144 - {
1.145 - UTIL_Format(
1.146 - command,
1.147 - sizeof(command),
1.148 - "addip %d %s\n",
1.149 - ban_time,
1.150 - identity);
1.151 - engine->ServerCommand(command);
1.152 -
1.153 - if (write_ban && ban_time == 0)
1.154 - {
1.155 - engine->ServerCommand("writeip\n");
1.156 - }
1.157 - }
1.158 - else if (!g_HL2.IsLANServer())
1.159 - {
1.160 - UTIL_Format(
1.161 - command,
1.162 - sizeof(command),
1.163 - "banid %d %s\n",
1.164 - ban_time,
1.165 - identity);
1.166 - engine->ServerCommand(command);
1.167 -
1.168 - if (write_ban && ban_time == 0)
1.169 - {
1.170 - engine->ServerCommand("writeid\n");
1.171 - }
1.172 - }
1.173 - else
1.174 - {
1.175 - return 0;
1.176 - }
1.177 - }
1.178 -
1.179 - return 1;
1.180 -}
1.181 -
1.182 -static cell_t RemoveBan(IPluginContext *pContext, const cell_t *params)
1.183 -{
1.184 - char *r_identity, *ban_cmd;
1.185 - int ban_flags, ban_source;
1.186 -
1.187 - pContext->LocalToString(params[1], &r_identity);
1.188 - pContext->LocalToString(params[3], &ban_cmd);
1.189 - ban_flags = params[2];
1.190 - ban_source = params[4];
1.191 -
1.192 - /* Make sure we can ban by one of the two methods! */
1.193 - bool ban_by_ip = ((ban_flags & BANFLAG_IP) == BANFLAG_IP);
1.194 - if (!ban_by_ip && ((ban_flags & BANFLAG_AUTHID) != BANFLAG_AUTHID))
1.195 - {
1.196 - return pContext->ThrowNativeError("No valid ban flags specified");
1.197 - }
1.198 -
1.199 - char identity[64];
1.200 - strncopy(identity, r_identity, sizeof(identity));
1.201 - UTIL_ReplaceAll(identity, sizeof(identity), ";", "");
1.202 -
1.203 - cell_t handled = 0;
1.204 - if (ban_cmd[0] != '\0' && g_pOnRemoveBan->GetFunctionCount() > 0)
1.205 - {
1.206 - g_pOnRemoveBan->PushString(identity);
1.207 - g_pOnRemoveBan->PushCell(ban_flags);
1.208 - g_pOnRemoveBan->PushString(ban_cmd);
1.209 - g_pOnRemoveBan->PushCell(ban_source);
1.210 - g_pOnRemoveBan->Execute(&handled);
1.211 - }
1.212 -
1.213 - char command[256];
1.214 - if (ban_by_ip)
1.215 - {
1.216 - if (!handled)
1.217 - {
1.218 - UTIL_Format(
1.219 - command,
1.220 - sizeof(command),
1.221 - "removeip %s\n",
1.222 - identity);
1.223 - engine->ServerCommand(command);
1.224 - engine->ServerCommand("writeip\n");
1.225 - }
1.226 - }
1.227 - else if (!g_HL2.IsLANServer())
1.228 - {
1.229 - if (!handled)
1.230 - {
1.231 - UTIL_Format(
1.232 - command,
1.233 - sizeof(command),
1.234 - "removeid %s\n",
1.235 - identity);
1.236 - engine->ServerCommand(command);
1.237 - engine->ServerCommand("writeid\n");
1.238 - }
1.239 - }
1.240 - else
1.241 - {
1.242 - return 0;
1.243 - }
1.244 -
1.245 - return 1;
1.246 -}
1.247 -
1.248 -static cell_t BanClient(IPluginContext *pContext, const cell_t *params)
1.249 -{
1.250 - const char *kick_message;
1.251 - char *ban_reason, *ban_cmd;
1.252 - int client, ban_flags, ban_source, ban_time;
1.253 -
1.254 - client = params[1];
1.255 -
1.256 - CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
1.257 - if (!pPlayer || !pPlayer->IsConnected())
1.258 - {
1.259 - return pContext->ThrowNativeError("Client index %d is invalid", client);
1.260 - }
1.261 -
1.262 - pContext->LocalToString(params[4], &ban_reason);
1.263 - pContext->LocalToString(params[5], (char **)&kick_message);
1.264 - pContext->LocalToString(params[6], &ban_cmd);
1.265 - ban_time = params[2];
1.266 - ban_flags = params[3];
1.267 - ban_source = params[7];
1.268 -
1.269 - /* Check how we should ban the player */
1.270 - if ((ban_flags & BANFLAG_AUTO) == BANFLAG_AUTO)
1.271 - {
1.272 - if (g_HL2.IsLANServer() || !pPlayer->IsAuthorized())
1.273 - {
1.274 - ban_flags |= BANFLAG_IP;
1.275 - ban_flags &= ~BANFLAG_AUTHID;
1.276 - }
1.277 - else
1.278 - {
1.279 - ban_flags |= BANFLAG_AUTHID;
1.280 - ban_flags &= ~BANFLAG_IP;
1.281 - }
1.282 - }
1.283 - else if ((ban_flags & BANFLAG_IP) == BANFLAG_IP)
1.284 - {
1.285 - ban_flags |= BANFLAG_IP;
1.286 - ban_flags &= ~BANFLAG_AUTHID;
1.287 - }
1.288 - else if ((ban_flags & BANFLAG_AUTHID) == BANFLAG_AUTHID)
1.289 - {
1.290 - if (pPlayer->IsAuthorized())
1.291 - {
1.292 - ban_flags |= BANFLAG_AUTHID;
1.293 - ban_flags &= ~BANFLAG_IP;
1.294 - }
1.295 - else
1.296 - {
1.297 - return 0;
1.298 - }
1.299 - }
1.300 - else
1.301 - {
1.302 - return pContext->ThrowNativeError("No valid ban method flags specified");
1.303 - }
1.304 -
1.305 - cell_t handled = 0;
1.306 - if (ban_cmd[0] != '\0' && g_pOnBanClient->GetFunctionCount() > 0)
1.307 - {
1.308 - g_pOnBanClient->PushCell(client);
1.309 - g_pOnBanClient->PushCell(ban_time);
1.310 - g_pOnBanClient->PushCell(ban_flags);
1.311 - g_pOnBanClient->PushString(ban_reason);
1.312 - g_pOnBanClient->PushString(kick_message);
1.313 - g_pOnBanClient->PushString(ban_cmd);
1.314 - g_pOnBanClient->PushCell(ban_source);
1.315 - g_pOnBanClient->Execute(&handled);
1.316 - }
1.317 -
1.318 - /* Sanitize the kick message */
1.319 - if (kick_message[0] == '\0')
1.320 - {
1.321 - kick_message = "Kicked";
1.322 - }
1.323 -
1.324 - if (!handled)
1.325 - {
1.326 - if ((ban_flags & BANFLAG_IP) == BANFLAG_IP)
1.327 - {
1.328 - /* Get the IP address and strip the port */
1.329 - char ip[24], *ptr;
1.330 - strncopy(ip, pPlayer->GetIPAddress(), sizeof(ip));
1.331 - if ((ptr = strchr(ip, ':')) != NULL)
1.332 - {
1.333 - *ptr = '\0';
1.334 - }
1.335 -
1.336 - /* Tell the server to ban the ip */
1.337 - char command[256];
1.338 - UTIL_Format(
1.339 - command,
1.340 - sizeof(command),
1.341 - "addip %d %s\n",
1.342 - ban_time,
1.343 - ip);
1.344 -
1.345 - /* Kick, then ban */
1.346 - if ((ban_flags & BANFLAG_NOKICK) != BANFLAG_NOKICK)
1.347 - {
1.348 - pPlayer->Kick(kick_message);
1.349 - }
1.350 - engine->ServerCommand(command);
1.351 -
1.352 - /* Physically write the ban */
1.353 - if ((ban_time == 0) && ((ban_flags & BANFLAG_NOWRITE) != BANFLAG_NOWRITE))
1.354 - {
1.355 - engine->ServerCommand("writeip\n");
1.356 - }
1.357 - }
1.358 - else if ((ban_flags & BANFLAG_AUTHID) == BANFLAG_AUTHID)
1.359 - {
1.360 - /* Tell the server to ban the auth string */
1.361 - char command[256];
1.362 - UTIL_Format(
1.363 - command,
1.364 - sizeof(command),
1.365 - "banid %d %s\n",
1.366 - ban_time,
1.367 - pPlayer->GetAuthString());
1.368 -
1.369 - /* Kick, then ban */
1.370 - if ((ban_flags & BANFLAG_NOKICK) != BANFLAG_NOKICK)
1.371 - {
1.372 - pPlayer->Kick(kick_message);
1.373 - }
1.374 - engine->ServerCommand(command);
1.375 -
1.376 - /* Physically write the ban if it's permanent */
1.377 - if ((ban_time == 0) && ((ban_flags & BANFLAG_NOWRITE) != BANFLAG_NOWRITE))
1.378 - {
1.379 - engine->ServerCommand("writeid\n");
1.380 - }
1.381 - }
1.382 - }
1.383 - else if ((ban_flags & BANFLAG_NOKICK) != BANFLAG_NOKICK)
1.384 - {
1.385 - pPlayer->Kick(kick_message);
1.386 - }
1.387 -
1.388 -
1.389 - return 1;
1.390 -}
1.391 -
1.392 -REGISTER_NATIVES(banNatives)
1.393 -{
1.394 - {"BanClient", BanClient},
1.395 - {"BanIdentity", BanIdentity},
1.396 - {"RemoveBan", RemoveBan},
1.397 - {NULL, NULL}
1.398 -};
1.399 -
