/
ban_cache.dm
84 lines (67 loc) · 2.48 KB
/
ban_cache.dm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/// Subsystem that batches a ban cache list for clients on initialize
/// This way we don't need to do ban checks in series later in the code
SUBSYSTEM_DEF(ban_cache)
/datum/controller/subsystem/ban_cache
name = "Ban Cache"
init_order = INIT_ORDER_BAN_CACHE
flags = SS_NO_FIRE
var/query_started = FALSE
/datum/controller/subsystem/ban_cache/Initialize()
generate_queries()
return SS_INIT_SUCCESS
/// Generates ban caches for any logged in clients. This ensures the amount of in-series ban checking we have to do that actually involves sleeps is VERY low
/datum/controller/subsystem/ban_cache/proc/generate_queries()
query_started = TRUE
if(!SSdbcore.Connect())
return
var/current_time = REALTIMEOFDAY
var/list/look_for = list()
var/list/query_args = list()
var/list/query_arg_keys = list()
var/num_keys = 0
for(var/ckey in GLOB.directory)
var/client/lad = GLOB.directory[ckey]
// If they've already got a ban cached, or a request goin, don't do it
if(lad.ban_cache || lad.ban_cache_start)
continue
look_for += ckey
lad.ban_cache_start = current_time
query_args += list("key[num_keys]" = ckey)
query_arg_keys += ":key[num_keys]"
num_keys++
// We're gonna try and make a query for clients
var/datum/db_query/query_batch_ban_cache = SSdbcore.NewQuery(
"SELECT ckey, role, applies_to_admins FROM [format_table_name("ban")] WHERE ckey IN ([query_arg_keys.Join(",")]) AND unbanned_datetime IS NULL AND (expiration_time IS NULL OR expiration_time > NOW())",
query_args
)
var/succeeded = query_batch_ban_cache.Execute()
for(var/ckey in look_for)
var/client/lad = GLOB.directory[ckey]
if(!lad || lad.ban_cache_start != current_time)
continue
lad.ban_cache_start = 0
if(!succeeded)
qdel(query_batch_ban_cache)
return
var/list/ckey_to_bans = list()
// Runs after the check for safety, don't want to override anything
for(var/ckey in look_for)
ckey_to_bans[ckey] = list()
while(query_batch_ban_cache.NextRow())
var/ckey = query_batch_ban_cache.item[1]
var/role = query_batch_ban_cache.item[2]
var/hits_admins = query_batch_ban_cache.item[3]
var/list/bans = ckey_to_bans[ckey]
if(!bans)
continue
// Yes I know this is slightly unoptimal, no I do not care
var/is_admin = GLOB.admin_datums[ckey] || GLOB.deadmins[ckey]
if(is_admin && !text2num(hits_admins))
continue
bans[role] = TRUE
for(var/ckey in ckey_to_bans)
var/client/lad = GLOB.directory[ckey]
if(!lad)
continue
lad.ban_cache = ckey_to_bans[ckey]
qdel(query_batch_ban_cache)