From 49174b15862c07bd654b318b9d4e57f5af1dc97b Mon Sep 17 00:00:00 2001
From: Lauri Kasanen <cand@gmx.com>
Date: Mon, 6 Dec 2021 13:41:07 +0200
Subject: [PATCH] Threaded context

---
 common/network/Blacklist.cxx | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/common/network/Blacklist.cxx b/common/network/Blacklist.cxx
index a5caeea..d1f7ff2 100644
--- a/common/network/Blacklist.cxx
+++ b/common/network/Blacklist.cxx
@@ -24,6 +24,8 @@
 #include <errno.h>
 #include <netinet/tcp.h>
 #include <netdb.h>
+#include <pthread.h>
+#include <stdlib.h>
 #include <time.h>
 
 #include <map>
@@ -35,6 +37,9 @@
 static std::map<std::string, unsigned> hits;
 static std::map<std::string, time_t> blacklist;
 
+static pthread_mutex_t hitmutex = PTHREAD_MUTEX_INITIALIZER;
+static pthread_mutex_t blmutex = PTHREAD_MUTEX_INITIALIZER;
+
 unsigned char bl_isBlacklisted(const char *addr) {
 	const unsigned char count = blacklist.count(addr);
 	if (!count)
@@ -43,19 +48,35 @@ unsigned char bl_isBlacklisted(const char *addr) {
 	const time_t now = time(NULL);
 	const unsigned timeout = rfb::Blacklist::initialTimeout;
 
+	if (pthread_mutex_lock(&blmutex))
+		abort();
+
 	if (now - timeout > blacklist[addr]) {
 		blacklist.erase(addr);
+		pthread_mutex_unlock(&blmutex);
+
+		if (pthread_mutex_lock(&hitmutex))
+			abort();
 		hits.erase(addr);
+		pthread_mutex_unlock(&hitmutex);
 		return 0;
 	} else {
 		blacklist[addr] = now;
+		pthread_mutex_unlock(&blmutex);
 		return 1;
 	}
 }
 
 void bl_addFailure(const char *addr) {
+	if (pthread_mutex_lock(&hitmutex))
+		abort();
 	const unsigned num = ++hits[addr];
+	pthread_mutex_unlock(&hitmutex);
+
 	if (num >= (unsigned) rfb::Blacklist::threshold) {
+		if (pthread_mutex_lock(&blmutex))
+			abort();
 		blacklist[addr] = time(NULL);
+		pthread_mutex_unlock(&blmutex);
 	}
 }