5 #include <enet/types.h>
7 #define MS_VERSION "0.2"
14 #define NET_MAXCLIENTS 64
16 #define NET_BUFSIZE 65536
18 #define NET_MSG_ADD 200
19 #define NET_MSG_RM 201
20 #define NET_MSG_LIST 202
22 #define LC_MS_INIT "D2DF master server starting on port %d...\n"
23 #define LC_MS_ADD "\nAdded server in slot #%d:\n%s:%d\n%s\n%s (%d)\n%d/%d plrs\nproto: %d pw?: %d\n"
24 #define LC_MS_UPD "\nUpdated server #%d (%s:%d):\n%s\n%s (%d)\n%d/%d plrs\nproto: %d pw?: %d\n"
25 #define LC_MS_RM "\nRemoved server #%d (%s:%d) by request.\n"
26 #define LC_MS_TIME "\nServer #%d (%s:%d) timed out.\n"
27 #define LC_MS_LIST "\nSent server list to %x:%u (ver. %s).\n"
28 #define LC_MS_DIE "\nD2DF master server shutting down...\n"
29 #define LC_MS_CONN "\nIncoming connection from %x:%u...\n"
41 enet_uint8 s_protocol
;
46 typedef struct ms_server_s ms_server
;
48 const char ms_game_ver
[] = "0.63";
50 int ms_timeout
= 100000;
55 enet_uint8 b_send
[NET_BUFSIZE
];
57 ENetHost
*ms_host
= NULL
;
58 ENetPeer
*ms_peers
[NET_MAXCLIENTS
];
59 ms_server ms_srv
[MS_MAXSRVS
];
60 enet_uint8 ms_count
= 0;
62 // fake servers to show on old versions of the game
63 static const ms_server ms_fake_srv
[] = {
67 .s_name
= "! \xc2\xc0\xd8\xc0 \xca\xce\xcf\xc8\xdf \xc8\xc3\xd0\xdb "
68 "\xd3\xd1\xd2\xc0\xd0\xc5\xcb\xc0! "
69 "\xd1\xca\xc0\xd7\xc0\xc9\xd2\xc5 \xcd\xce\xc2\xd3\xde C "
71 .s_map
= "! Your game is outdated. "
72 "Get latest version at doom2d.org !",
78 .s_name
= "! \xcf\xd0\xce\xc1\xd0\xce\xd1\xdcTE \xcf\xce\xd0\xd2\xdb "
79 "25666 \xc8 57133 HA CEPBEPE \xcf\xc5\xd0\xc5\xc4 \xc8\xc3\xd0\xce\xc9 !",
80 .s_map
= "! Forward ports 25666 and 57133 before hosting !",
85 #define MS_FAKESRVS (sizeof(ms_fake_srv) / sizeof(ms_fake_srv[0]))
88 printf("Usage: d2df_master -p port_number [-t timeout_seconds]\n");
94 printf("Doom 2D Forever master server v%s\n", MS_VERSION
);
99 void d_error (const char *msg
, int fatal
) {
101 fprintf(stderr
, "FATAL ERROR: %s\n", msg
);
104 fprintf(stderr
, "ERROR: %s\n", msg
);
109 void d_getargs (int argc
, char *argv
[]) {
116 for (int i
= 1; i
< argc
; ++i
) {
117 if (!strcmp(argv
[i
], "-v")) {
121 } else if (!strcmp(argv
[i
], "-p")) {
123 d_error("Specify a port value!", 1);
126 ms_port
= atoi(argv
[++i
]);
128 } else if (!strcmp(argv
[i
], "-t") & (i
+ 1 < argc
)) {
129 ms_timeout
= atoi(argv
[++i
]) * 1000;
135 enet_uint8
b_read_uint8 (enet_uint8 buf
[], size_t *pos
) {
136 return buf
[(*pos
)++];
140 enet_uint16
b_read_uint16 (enet_uint8 buf
[], size_t *pos
) {
143 ret
= *(enet_uint16
*)(buf
+ *pos
);
144 *pos
+= sizeof(enet_uint16
);
150 char* b_read_dstring (enet_uint8 buf
[], size_t *pos
) {
153 size_t len
= b_read_uint8(buf
, pos
);
155 ret
= malloc(len
+ 1);
157 memmove(ret
, (char*)(buf
+ *pos
), len
);
165 void b_write_uint8 (enet_uint8 buf
[], size_t *pos
, enet_uint8 val
) {
170 void b_write_uint16 (enet_uint8 buf
[], size_t *pos
, enet_uint16 val
) {
171 *(enet_uint16
*)(buf
+ *pos
) = val
;
172 *pos
+= sizeof(enet_uint16
);
176 void b_write_dstring (enet_uint8 buf
[], size_t *pos
, const char* val
) {
177 enet_uint8 len
= strlen(val
);
178 b_write_uint8(buf
, pos
, len
);
180 memmove((char*)(buf
+ *pos
), val
, len
);
185 void b_write_server (enet_uint8 buf
[], size_t *pos
, ms_server s
) {
186 b_write_dstring(b_send
, pos
, s
.s_ip
);
187 b_write_uint16 (b_send
, pos
, s
.s_port
);
188 b_write_dstring(b_send
, pos
, s
.s_name
);
189 b_write_dstring(b_send
, pos
, s
.s_map
);
190 b_write_uint8 (b_send
, pos
, s
.s_mode
);
191 b_write_uint8 (b_send
, pos
, s
.s_plrs
);
192 b_write_uint8 (b_send
, pos
, s
.s_maxplrs
);
193 b_write_uint8 (b_send
, pos
, s
.s_protocol
);
194 b_write_uint8 (b_send
, pos
, s
.s_pw
);
198 int main (int argc
, char *argv
[]) {
199 d_getargs(argc
, argv
);
201 if (enet_initialize()) {
202 d_error("Could not init ENet!", 1);
206 printf(LC_MS_INIT
, ms_port
);
208 for (int i
= 0; i
< NET_MAXCLIENTS
; ++i
) ms_peers
[i
] = NULL
;
210 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
212 ms_srv
[i
].s_ip
[0] = '\0';
213 ms_srv
[i
].s_name
[0] = '\0';
214 ms_srv
[i
].s_map
[0] = '\0';
219 addr
.host
= ENET_HOST_ANY
;
222 ms_host
= enet_host_create(&addr
, NET_MAXCLIENTS
, NET_CHANS
, 0, 0);
224 d_error("Could not create host on specified port!", 1);
228 atexit(enet_deinitialize
);
232 enet_uint8 msg
= 255;
235 enet_uint16 port
= 0;
239 char *clientver
= NULL
;
244 enet_uint8 proto
= 0;
247 while (enet_host_service(ms_host
, &event
, 1) > 0) {
248 switch (event
.type
) {
249 case ENET_EVENT_TYPE_CONNECT
:
250 printf(LC_MS_CONN
, event
.peer
->address
.host
, event
.peer
->address
.port
);
252 case ENET_EVENT_TYPE_RECEIVE
:
253 if (!event
.peer
) continue;
255 msg
= b_read_uint8(event
.packet
->data
, &b_read
);
259 if (!event
.peer
) continue;
261 enet_address_get_host_ip(&(event
.peer
->address
), ip
, 17);
262 port
= b_read_uint16(event
.packet
->data
, &b_read
);
264 name
= b_read_dstring(event
.packet
->data
, &b_read
);
265 map
= b_read_dstring(event
.packet
->data
, &b_read
);
266 gm
= b_read_uint8(event
.packet
->data
, &b_read
);
268 pl
= b_read_uint8(event
.packet
->data
, &b_read
);
269 mpl
= b_read_uint8(event
.packet
->data
, &b_read
);
271 proto
= b_read_uint8(event
.packet
->data
, &b_read
);
272 pw
= b_read_uint8(event
.packet
->data
, &b_read
);
274 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
275 if (ms_srv
[i
].used
) {
276 if ((strncmp(ip
, ms_srv
[i
].s_ip
, 16) == 0) && (ms_srv
[i
].s_port
== port
)) {
277 strncpy(ms_srv
[i
].s_map
, map
, sizeof(ms_srv
[i
].s_map
));
278 strncpy(ms_srv
[i
].s_name
, name
, sizeof(ms_srv
[i
].s_name
));
279 ms_srv
[i
].s_plrs
= pl
;
280 ms_srv
[i
].s_maxplrs
= mpl
;
282 ms_srv
[i
].s_mode
= gm
;
284 ms_srv
[i
].ttl
= ms_timeout
;
286 printf(LC_MS_UPD
, i
, ip
, port
, name
, map
, gm
, pl
, mpl
, proto
, pw
);
290 strncpy(ms_srv
[i
].s_ip
, ip
, sizeof(ms_srv
[i
].s_ip
));
291 strncpy(ms_srv
[i
].s_map
, map
, sizeof(ms_srv
[i
].s_map
));
292 strncpy(ms_srv
[i
].s_name
, name
, sizeof(ms_srv
[i
].s_name
));
293 ms_srv
[i
].s_port
= port
;
294 ms_srv
[i
].s_plrs
= pl
;
295 ms_srv
[i
].s_maxplrs
= mpl
;
297 ms_srv
[i
].s_mode
= gm
;
298 ms_srv
[i
].s_protocol
= proto
;
299 ms_srv
[i
].ttl
= ms_timeout
;
303 printf(LC_MS_ADD
, i
, ip
, port
, name
, map
, gm
, pl
, mpl
, proto
, pw
);
313 enet_address_get_host_ip(&(event
.peer
->address
), ip
, 17);
314 port
= b_read_uint16(event
.packet
->data
, &b_read
);
315 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
316 if (ms_srv
[i
].used
) {
317 if ((strncmp(ip
, ms_srv
[i
].s_ip
, 16) == 0) && (ms_srv
[i
].s_port
== port
)) {
319 printf(LC_MS_RM
, i
, ip
, port
);
326 if (!event
.peer
) continue;
329 b_write_uint8(b_send
, &b_write
, NET_MSG_LIST
);
331 if (event
.packet
->dataLength
> 2) {
332 // holy shit a fresh client
333 clientver
= b_read_dstring(event
.packet
->data
, &b_read
);
334 b_write_uint8(b_send
, &b_write
, ms_count
);
336 // old client, feed them bullshit first
337 b_write_uint8(b_send
, &b_write
, ms_count
+ 2);
338 for (int i
= 0; i
< MS_FAKESRVS
; ++i
)
339 b_write_server(b_send
, &b_write
, ms_fake_srv
[i
]);
342 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
343 if (ms_srv
[i
].used
) b_write_server(b_send
, &b_write
, ms_srv
[i
]);
347 // TODO: check if this client is outdated (?) and send back new verstring
348 // for now just write the same shit back
349 b_write_dstring(b_send
, &b_write
, clientver
);
352 ENetPacket
*p
= enet_packet_create(b_send
, b_write
, ENET_PACKET_FLAG_RELIABLE
);
353 enet_peer_send(event
.peer
, NET_CH_MAIN
, p
);
354 enet_host_flush(ms_host
);
356 printf(LC_MS_LIST
, event
.peer
->address
.host
, event
.peer
->address
.port
, clientver
? clientver
: "<old>");
362 enet_packet_destroy(event
.packet
);
370 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
371 if (ms_srv
[i
].used
) {
372 if (--(ms_srv
[i
].ttl
) == 0) {
374 printf(LC_MS_TIME
, i
, ms_srv
[i
].s_ip
, ms_srv
[i
].s_port
);