1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/plugins/admin-sql-prefetch.sp Sun Mar 30 07:00:22 2008 +0000
1.3 @@ -0,0 +1,386 @@
1.4 +/**
1.5 + * vim: set ts=4 :
1.6 + * =============================================================================
1.7 + * SourceMod SQL Admins Plugin (Prefetch)
1.8 + * Prefetches admins from an SQL database without threading.
1.9 + *
1.10 + * SourceMod (C)2004-2008 AlliedModders LLC. All rights reserved.
1.11 + * =============================================================================
1.12 + *
1.13 + * This program is free software; you can redistribute it and/or modify it under
1.14 + * the terms of the GNU General Public License, version 3.0, as published by the
1.15 + * Free Software Foundation.
1.16 + *
1.17 + * This program is distributed in the hope that it will be useful, but WITHOUT
1.18 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.19 + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.20 + * details.
1.21 + *
1.22 + * You should have received a copy of the GNU General Public License along with
1.23 + * this program. If not, see <http://www.gnu.org/licenses/>.
1.24 + *
1.25 + * As a special exception, AlliedModders LLC gives you permission to link the
1.26 + * code of this program (as well as its derivative works) to "Half-Life 2," the
1.27 + * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.28 + * by the Valve Corporation. You must obey the GNU General Public License in
1.29 + * all respects for all other code used. Additionally, AlliedModders LLC grants
1.30 + * this exception to all derivative works. AlliedModders LLC defines further
1.31 + * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.32 + * or <http://www.sourcemod.net/license.php>.
1.33 + *
1.34 + * Version: $Id$
1.35 + */
1.36 +
1.37 +/* We like semicolons */
1.38 +#pragma semicolon 1
1.39 +
1.40 +#include <sourcemod>
1.41 +
1.42 +public Plugin:myinfo =
1.43 +{
1.44 + name = "SQL Admins (Prefetch)",
1.45 + author = "AlliedModders LLC",
1.46 + description = "Reads all admins from SQL",
1.47 + version = SOURCEMOD_VERSION,
1.48 + url = "http://www.sourcemod.net/"
1.49 +};
1.50 +
1.51 +public OnRebuildAdminCache(AdminCachePart:part)
1.52 +{
1.53 + /* First try to get a database connection */
1.54 + decl String:error[255];
1.55 + new Handle:db;
1.56 +
1.57 + if (SQL_CheckConfig("admins"))
1.58 + {
1.59 + db = SQL_Connect("admins", true, error, sizeof(error));
1.60 + } else {
1.61 + db = SQL_Connect("default", true, error, sizeof(error));
1.62 + }
1.63 +
1.64 + if (db == INVALID_HANDLE)
1.65 + {
1.66 + LogError("Could not connect to database \"default\": %s", error);
1.67 + return;
1.68 + }
1.69 +
1.70 + if (part == AdminCache_Overrides)
1.71 + {
1.72 + FetchOverrides(db);
1.73 + } else if (part == AdminCache_Groups) {
1.74 + FetchGroups(db);
1.75 + } else if (part == AdminCache_Admins) {
1.76 + FetchUsers(db);
1.77 + }
1.78 +
1.79 + CloseHandle(db);
1.80 +}
1.81 +
1.82 +FetchUserGroups(Handle:hQuery, AdminId:adm, id)
1.83 +{
1.84 + SQL_BindParamInt(hQuery, 0, id);
1.85 + if (!SQL_Execute(hQuery))
1.86 + {
1.87 + decl String:error[255];
1.88 + SQL_GetError(hQuery, error, sizeof(error));
1.89 + LogError("FetchUserGroups() statement execute failed: %s", error);
1.90 + return;
1.91 + }
1.92 +
1.93 + decl String:name[80];
1.94 + new GroupId:gid;
1.95 + while (SQL_FetchRow(hQuery))
1.96 + {
1.97 + SQL_FetchString(hQuery, 0, name, sizeof(name));
1.98 +
1.99 + if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
1.100 + {
1.101 + /* Group wasn't found, don't bother with it. */
1.102 + continue;
1.103 + }
1.104 +
1.105 + #if defined _DEBUG
1.106 + PrintToServer("Found admin group (%d, %d, %d)", id, gid, adm);
1.107 + #endif
1.108 +
1.109 + AdminInheritGroup(adm, gid);
1.110 + }
1.111 +}
1.112 +
1.113 +FetchUsers(Handle:db)
1.114 +{
1.115 + decl String:query[255], String:error[255];
1.116 + new Handle:hQuery, Handle:hGroupQuery;
1.117 +
1.118 + Format(query, sizeof(query), "SELECT id, authtype, identity, password, flags, name, immunity FROM sm_admins");
1.119 + if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
1.120 + {
1.121 + SQL_GetError(db, error, sizeof(error));
1.122 + LogError("FetchUsers() query failed: %s", query);
1.123 + LogError("Query error: %s", error);
1.124 + return;
1.125 + }
1.126 +
1.127 + /* If this fails it's non-fatal */
1.128 + Format(query,
1.129 + sizeof(query),
1.130 + "SELECT g.name FROM sm_admins_groups ag JOIN sm_groups g ON ag.group_id = g.id WHERE ag.admin_id = ? ORDER BY inherit_order ASC");
1.131 + if ((hGroupQuery = SQL_PrepareQuery(db, query, error, sizeof(error))) == INVALID_HANDLE)
1.132 + {
1.133 + LogError("FetchUsers() group query preparation failed: %s", query);
1.134 + LogError("Query error: %s", error);
1.135 + }
1.136 +
1.137 + decl String:authtype[16];
1.138 + decl String:identity[80];
1.139 + decl String:password[80];
1.140 + decl String:flags[32];
1.141 + decl String:name[80];
1.142 + new immunity;
1.143 + new AdminId:adm, id;
1.144 +
1.145 + while (SQL_FetchRow(hQuery))
1.146 + {
1.147 + id = SQL_FetchInt(hQuery, 0);
1.148 + SQL_FetchString(hQuery, 1, authtype, sizeof(authtype));
1.149 + SQL_FetchString(hQuery, 2, identity, sizeof(identity));
1.150 + SQL_FetchString(hQuery, 3, password, sizeof(password));
1.151 + SQL_FetchString(hQuery, 4, flags, sizeof(flags));
1.152 + SQL_FetchString(hQuery, 5, name, sizeof(name));
1.153 + immunity = SQL_FetchInt(hQuery, 6);
1.154 +
1.155 + /* Use a pre-existing admin if we can */
1.156 + if ((adm = FindAdminByIdentity(authtype, identity)) == INVALID_ADMIN_ID)
1.157 + {
1.158 + adm = CreateAdmin(name);
1.159 + if (!BindAdminIdentity(adm, authtype, identity))
1.160 + {
1.161 + LogError("Could not bind prefetched SQL admin (authtype \"%s\") (identity \"%s\")", authtype, identity);
1.162 + continue;
1.163 + }
1.164 + }
1.165 +
1.166 + #if defined _DEBUG
1.167 + PrintToServer("Found SQL admin (%d,%s,%s,%s,%s,%s,%d):%d", id, authtype, identity, password, flags, name, immunity, adm);
1.168 + #endif
1.169 +
1.170 + /* See if this admin wants a password */
1.171 + if (password[0] != '\0')
1.172 + {
1.173 + SetAdminPassword(adm, password);
1.174 + }
1.175 +
1.176 + /* Apply each flag */
1.177 + new len = strlen(flags);
1.178 + new AdminFlag:flag;
1.179 + for (new i=0; i<len; i++)
1.180 + {
1.181 + if (!FindFlagByChar(flags[i], flag))
1.182 + {
1.183 + continue;
1.184 + }
1.185 + SetAdminFlag(adm, flag, true);
1.186 + }
1.187 +
1.188 + SetAdminImmunityLevel(adm, immunity);
1.189 +
1.190 + /* Look up groups */
1.191 + if (hGroupQuery != INVALID_HANDLE)
1.192 + {
1.193 + FetchUserGroups(hGroupQuery, adm, id);
1.194 + }
1.195 + }
1.196 +
1.197 + CloseHandle(hGroupQuery);
1.198 + CloseHandle(hQuery);
1.199 +}
1.200 +
1.201 +FetchGroups(Handle:db)
1.202 +{
1.203 + decl String:query[255];
1.204 + new Handle:hQuery;
1.205 +
1.206 + Format(query, sizeof(query), "SELECT flags, name, immunity_level FROM sm_groups");
1.207 +
1.208 + if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
1.209 + {
1.210 + decl String:error[255];
1.211 + SQL_GetError(db, error, sizeof(error));
1.212 + LogError("FetchGroups() query failed: %s", query);
1.213 + LogError("Query error: %s", error);
1.214 + return;
1.215 + }
1.216 +
1.217 + /* Now start fetching groups */
1.218 + decl String:flags[32];
1.219 + decl String:name[128];
1.220 + new immunity;
1.221 + while (SQL_FetchRow(hQuery))
1.222 + {
1.223 + SQL_FetchString(hQuery, 0, flags, sizeof(flags));
1.224 + SQL_FetchString(hQuery, 1, name, sizeof(name));
1.225 + immunity = SQL_FetchInt(hQuery, 2);
1.226 +
1.227 +#if defined _DEBUG
1.228 + PrintToServer("Adding group (%d, %s, %s)", immunity, flags, name);
1.229 +#endif
1.230 +
1.231 + /* Find or create the group */
1.232 + new GroupId:gid;
1.233 + if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
1.234 + {
1.235 + gid = CreateAdmGroup(name);
1.236 + }
1.237 +
1.238 + /* Add flags from the database to the group */
1.239 + new num_flag_chars = strlen(flags);
1.240 + for (new i=0; i<num_flag_chars; i++)
1.241 + {
1.242 + decl AdminFlag:flag;
1.243 + if (!FindFlagByChar(flags[i], flag))
1.244 + {
1.245 + continue;
1.246 + }
1.247 + SetAdmGroupAddFlag(gid, flag, true);
1.248 + }
1.249 +
1.250 + /* Set the immunity level this group has */
1.251 + SetAdmGroupImmunityLevel(gid, immunity);
1.252 + }
1.253 +
1.254 + CloseHandle(hQuery);
1.255 +
1.256 + /**
1.257 + * Get immunity in a big lump. This is a nasty query but it gets the job done.
1.258 + */
1.259 + new len = 0;
1.260 + len += Format(query[len], sizeof(query)-len, "SELECT g1.name, g2.name FROM sm_group_immunity gi");
1.261 + len += Format(query[len], sizeof(query)-len, " LEFT JOIN sm_groups g1 ON g1.id = gi.group_id ");
1.262 + len += Format(query[len], sizeof(query)-len, " LEFT JOIN sm_groups g2 ON g2.id = gi.other_id");
1.263 +
1.264 + if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
1.265 + {
1.266 + decl String:error[255];
1.267 + SQL_GetError(db, error, sizeof(error));
1.268 + LogError("FetchGroups() query failed: %s", query);
1.269 + LogError("Query error: %s", error);
1.270 + return;
1.271 + }
1.272 +
1.273 + while (SQL_FetchRow(hQuery))
1.274 + {
1.275 + decl String:group1[80];
1.276 + decl String:group2[80];
1.277 + new GroupId:gid1, GroupId:gid2;
1.278 +
1.279 + SQL_FetchString(hQuery, 0, group1, sizeof(group1));
1.280 + SQL_FetchString(hQuery, 1, group2, sizeof(group2));
1.281 +
1.282 + if (((gid1 = FindAdmGroup(group1)) == INVALID_GROUP_ID)
1.283 + || (gid2 = FindAdmGroup(group2)) == INVALID_GROUP_ID)
1.284 + {
1.285 + continue;
1.286 + }
1.287 +
1.288 + SetAdmGroupImmuneFrom(gid1, gid2);
1.289 +#if defined _DEBUG
1.290 + PrintToServer("SetAdmGroupImmuneFrom(%d, %d)", gid1, gid2);
1.291 +#endif
1.292 + }
1.293 +
1.294 + CloseHandle(hQuery);
1.295 +
1.296 + /**
1.297 + * Fetch overrides in a lump query.
1.298 + */
1.299 + Format(query, sizeof(query), "SELECT g.name, go.type, go.name, go.access FROM sm_group_overrides go LEFT JOIN sm_groups g ON go.group_id = g.id");
1.300 +
1.301 + if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
1.302 + {
1.303 + decl String:error[255];
1.304 + SQL_GetError(db, error, sizeof(error));
1.305 + LogError("FetchGroups() query failed: %s", query);
1.306 + LogError("Query error: %s", error);
1.307 + return;
1.308 + }
1.309 +
1.310 + decl String:type[16];
1.311 + decl String:cmd[64];
1.312 + decl String:access[16];
1.313 + while (SQL_FetchRow(hQuery))
1.314 + {
1.315 + SQL_FetchString(hQuery, 0, name, sizeof(name));
1.316 + SQL_FetchString(hQuery, 1, type, sizeof(type));
1.317 + SQL_FetchString(hQuery, 2, cmd, sizeof(cmd));
1.318 + SQL_FetchString(hQuery, 3, access, sizeof(access));
1.319 +
1.320 + new GroupId:gid;
1.321 + if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
1.322 + {
1.323 + continue;
1.324 + }
1.325 +
1.326 + new OverrideType:o_type = Override_Command;
1.327 + if (StrEqual(type, "group"))
1.328 + {
1.329 + o_type = Override_CommandGroup;
1.330 + }
1.331 +
1.332 + new OverrideRule:o_rule = Command_Deny;
1.333 + if (StrEqual(access, "allow"))
1.334 + {
1.335 + o_rule = Command_Allow;
1.336 + }
1.337 +
1.338 + #if defined _DEBUG
1.339 + PrintToServer("AddAdmGroupCmdOverride(%d, %s, %d, %d)", gid, cmd, o_type, o_rule);
1.340 + #endif
1.341 +
1.342 + AddAdmGroupCmdOverride(gid, cmd, o_type, o_rule);
1.343 + }
1.344 +
1.345 + CloseHandle(hQuery);
1.346 +}
1.347 +
1.348 +FetchOverrides(Handle:db)
1.349 +{
1.350 + decl String:query[255];
1.351 + new Handle:hQuery;
1.352 +
1.353 + Format(query, sizeof(query), "SELECT type, name, flags FROM sm_overrides");
1.354 +
1.355 + if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
1.356 + {
1.357 + decl String:error[255];
1.358 + SQL_GetError(db, error, sizeof(error));
1.359 + LogError("FetchOverrides() query failed: %s", query);
1.360 + LogError("Query error: %s", error);
1.361 + return;
1.362 + }
1.363 +
1.364 + decl String:type[64];
1.365 + decl String:name[64];
1.366 + decl String:flags[32];
1.367 + new flag_bits;
1.368 + while (SQL_FetchRow(hQuery))
1.369 + {
1.370 + SQL_FetchString(hQuery, 0, type, sizeof(type));
1.371 + SQL_FetchString(hQuery, 1, name, sizeof(name));
1.372 + SQL_FetchString(hQuery, 2, flags, sizeof(flags));
1.373 +
1.374 +#if defined _DEBUG
1.375 + PrintToServer("Adding override (%s, %s, %s)", type, name, flags);
1.376 +#endif
1.377 +
1.378 + flag_bits = ReadFlagString(flags);
1.379 + if (StrEqual(type, "command"))
1.380 + {
1.381 + AddCommandOverride(name, Override_Command, flag_bits);
1.382 + } else if (StrEqual(type, "group")) {
1.383 + AddCommandOverride(name, Override_CommandGroup, flag_bits);
1.384 + }
1.385 + }
1.386 +
1.387 + CloseHandle(hQuery);
1.388 +}
1.389 +
