GTPS-HTTPS/main.cpp at main · Leomjson/GTPS-HTTPS · GitHub

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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include "httplib.hpp"
#include "json.hpp"
#include <future>
#include <filesystem>
using namespace std;
httplib::SSLServer server("cert.pem", "key.pem");
vector<thread> backtasks;
const char* server_data =
R"(server|127.0.0.1
port|17091
type|1
meta|cmeta
RTENDMARKERBS1001)";
const char* gov =
R"(<style>h1 { text-align: center; }
</style>
<body>
<h1><b><big>Unauthorized Access</big></b></h1>
</body>)";
class connection {
public:
	string ip = "";
	int attempts = 0;
}; map<string, connection> connection_data;

void SaveConnectionData(connection data, string ip) {
	nlohmann::json j;
	ofstream w("connection/" + ip);
	j.dump(1);
	j["ip"] = data.ip;
	j["attempts"] = data.attempts;
	w << setw(2) << j;
}
connection LoadConnectionData(string ip) {
	nlohmann::json j;
	ifstream r("connection/" + ip); r >> j;
	connection data;
	data.ip = j["ip"];
	data.attempts = j["attempts"];
	return data;
}
void append_reset(string ip) {
	while (true) {
		for (auto it = connection_data.begin(); it not_eq connection_data.end(); ++it)
			if (it->first == ip)
				if (it->second.attempts > 3) {
					this_thread::sleep_for(4s);
					it->second.attempts -= 3;
					connection data = LoadConnectionData(ip);
					data.attempts = it->second.attempts;
					SaveConnectionData(data, it->second.ip);
				}
				else {
					if (it->second.attempts <= 0) continue; // skip
					this_thread::sleep_for(1800ms);
					it->second.attempts -= 1;
					connection data = LoadConnectionData(ip);
					data.attempts = it->second.attempts;
					SaveConnectionData(data, it->second.ip);
				}
		this_thread::sleep_for(5ms);
	}
}
bool request(const httplib::Request req)
{
	ifstream r("connection/" + req.remote_addr);
	if (not r.is_open()) {
		connection new_data;
		new_data.ip = req.remote_addr;
		new_data.attempts = 1;
		SaveConnectionData(new_data, req.remote_addr);
		connection_data.emplace(req.remote_addr, new_data);
	}
	else for (auto it = connection_data.begin(); it not_eq connection_data.end(); ++it) if (it->first == req.remote_addr)
	{
		it->second.attempts++;
		if (it->second.attempts > 3) {
			backtasks.emplace_back(append_reset, req.remote_addr);
			return false;
		}
		connection data = LoadConnectionData(req.remote_addr);
		data.attempts = it->second.attempts;
		SaveConnectionData(data, it->second.ip);
	}
	return true;
}
int main()
{
	for (const auto& i : filesystem::directory_iterator("./connection/")) {
		if (i.path().filename().string() == "tmp" or filesystem::is_directory(i.path())) continue;
		connection_data.emplace(i.path().filename().string(), LoadConnectionData(i.path().filename().string()));
	}
	server.Post(("/growtopia/server_data.php"), [&](const auto& req, auto& res) {
		future<bool> Request = async(request, req); Request.wait();
	if (not Request.get()) {
		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), 12);
		cerr << req.remote_addr << " was rejected" << endl;
		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), 7);
		return;
	}
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), 10);
	cerr << req.remote_addr << " has connected" << endl;
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), 7);
	res.set_content(server_data, "text/plain");
		});
	server.Get(("/growtopia/server_data.php"), [&](const auto& req, auto& res) { res.set_content(gov, "text/html"); });
	server.listen("0.0.0.0", 443);
	return 0;
}