summaryrefslogtreecommitdiff
path: root/src/kill_proxy.cpp
blob: a2a2e27fa09cf1c09d6faf9269ef7f7deb29815e (plain)
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
#include <arpa/inet.h>
#include <bitset>
#include <concepts>
#include <cstdint>
#include <fcntl.h>
#include <format>
#include <iostream>
#include <netdb.h>
#include <netinet/in.h>
#include <ostream>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/uio.h>
#include <sys/wait.h>
#include <thread>
#include <unistd.h>
#include <utility>

#include "net/client.hpp"
#include "net/conn.hpp"
#include "net/server.hpp"
#include "net/types.hpp"

using net::packet::operator""_ns;

template <typename T, T... ints>
void forward_packets(net::conn &dst, net::conn &src, std::integer_sequence<T, ints...> int_seq) {
	(src.reg_handler<net::packet::raw<ints>>([&dst](auto &a) {
		dst.send_packet(a);
		return true;
	}), ...);
}

int main(int argc, char* argv[])
{
	auto server = net::client("localhost", 7777);
	auto proxy = net::server("localhost", 8888);

	proxy.bind();
	proxy.listen(1);

	auto server_conn = server.connect();
	auto proxy_conn = proxy.accept();

	uint8_t slot;
	server_conn.reg_handler<net::packet::accept>([=, &slot](auto& a) {
		slot = a.client_id;
		return false;
	});

	proxy_conn.reg_handler<net::packet::damage_player>([&slot, &server_conn](auto &dmg) {
		if (dmg.client_id != slot) {
			dmg.damage *= 100;
			dmg.reason.reasons.at(0) = 0;

			dmg.reason.reasons[net::packet::death_reason::reason::Custom] = true;
			dmg.reason.custom = "killed by server insecurity";
			
			server_conn.send_packet(dmg);
			return true;
		}
		return false;
	});

	forward_packets(server_conn, proxy_conn, std::make_integer_sequence<uint8_t, 255>{});
	forward_packets(proxy_conn, server_conn, std::make_integer_sequence<uint8_t, 255>{});

	auto handle_loop = [] (net::conn &c) {
		for(;;) {
			c.handle();
		}
	};

	std::thread server_thread(handle_loop, std::ref(server_conn));
	std::thread proxy_thread(handle_loop, std::ref(proxy_conn));

	return 0;
}