ce98cfcd0a6e0772edf32b8a0f3155bc3573b291
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"
30 #define LC_MS_MOTD "\nMOTD: %s\n"
31 #define LC_MS_URGENT "\nURGENT: %s\n"
33 #define MS_URGENT_FILE "urgent.txt"
34 #define MS_MOTD_FILE "motd.txt"
45 enet_uint8 s_protocol
;
50 typedef struct ms_server_s ms_server
;
52 const char ms_game_ver
[] = "0.63";
53 char ms_motd
[256] = "";
54 char ms_urgent
[256] = "";
57 int ms_timeout
= 100000;
62 enet_uint8 b_send
[NET_BUFSIZE
];
64 ENetHost
*ms_host
= NULL
;
65 ENetPeer
*ms_peers
[NET_MAXCLIENTS
];
66 ms_server ms_srv
[MS_MAXSRVS
];
67 enet_uint8 ms_count
= 0;
69 // fake servers to show on old versions of the game
70 static const ms_server ms_fake_srv
[] = {
74 .s_name
= "! \xc2\xc0\xd8\xc0 \xca\xce\xcf\xc8\xdf \xc8\xc3\xd0\xdb "
75 "\xd3\xd1\xd2\xc0\xd0\xc5\xcb\xc0! "
76 "\xd1\xca\xc0\xd7\xc0\xc9\xd2\xc5 \xcd\xce\xc2\xd3\xde C "
78 .s_map
= "! Your game is outdated. "
79 "Get latest version at doom2d.org !",
85 .s_name
= "! \xcf\xd0\xce\xc1\xd0\xce\xd1\xdcTE \xcf\xce\xd0\xd2\xdb "
86 "25666 \xc8 57133 HA CEPBEPE \xcf\xc5\xd0\xc5\xc4 \xc8\xc3\xd0\xce\xc9 !",
87 .s_map
= "! Forward ports 25666 and 57133 before hosting !",
92 #define MS_FAKESRVS (sizeof(ms_fake_srv) / sizeof(ms_fake_srv[0]))
95 printf("Usage: d2df_master -p port_number [-t timeout_seconds]\n");
101 printf("Doom 2D Forever master server v%s\n", MS_VERSION
);
106 void d_error (const char *msg
, int fatal
) {
108 fprintf(stderr
, "FATAL ERROR: %s\n", msg
);
111 fprintf(stderr
, "ERROR: %s\n", msg
);
116 void d_getargs (int argc
, char *argv
[]) {
123 for (int i
= 1; i
< argc
; ++i
) {
124 if (!strcmp(argv
[i
], "-v")) {
128 } else if (!strcmp(argv
[i
], "-p")) {
130 d_error("Specify a port value!", 1);
133 ms_port
= atoi(argv
[++i
]);
135 } else if (!strcmp(argv
[i
], "-t") & (i
+ 1 < argc
)) {
136 ms_timeout
= atoi(argv
[++i
]) * 1000;
142 int d_readtextfile (const char *fname
, char *buf
, size_t max
) {
143 FILE *f
= fopen(fname
, "r");
145 char line
[256] = { 0 };
146 while(fgets(buffer
, sizeof(line
), (FILE*) fp
))
147 printf("%s\n", buffer
);
155 enet_uint8
b_read_uint8 (enet_uint8 buf
[], size_t *pos
) {
156 return buf
[(*pos
)++];
160 enet_uint16
b_read_uint16 (enet_uint8 buf
[], size_t *pos
) {
163 ret
= *(enet_uint16
*)(buf
+ *pos
);
164 *pos
+= sizeof(enet_uint16
);
170 char* b_read_dstring (enet_uint8 buf
[], size_t *pos
) {
173 size_t len
= b_read_uint8(buf
, pos
);
175 ret
= malloc(len
+ 1);
177 memmove(ret
, (char*)(buf
+ *pos
), len
);
185 void b_write_uint8 (enet_uint8 buf
[], size_t *pos
, enet_uint8 val
) {
190 void b_write_uint16 (enet_uint8 buf
[], size_t *pos
, enet_uint16 val
) {
191 *(enet_uint16
*)(buf
+ *pos
) = val
;
192 *pos
+= sizeof(enet_uint16
);
196 void b_write_dstring (enet_uint8 buf
[], size_t *pos
, const char* val
) {
197 enet_uint8 len
= strlen(val
);
198 b_write_uint8(buf
, pos
, len
);
200 memmove((char*)(buf
+ *pos
), val
, len
);
205 void b_write_server (enet_uint8 buf
[], size_t *pos
, ms_server s
) {
206 b_write_dstring(b_send
, pos
, s
.s_ip
);
207 b_write_uint16 (b_send
, pos
, s
.s_port
);
208 b_write_dstring(b_send
, pos
, s
.s_name
);
209 b_write_dstring(b_send
, pos
, s
.s_map
);
210 b_write_uint8 (b_send
, pos
, s
.s_mode
);
211 b_write_uint8 (b_send
, pos
, s
.s_plrs
);
212 b_write_uint8 (b_send
, pos
, s
.s_maxplrs
);
213 b_write_uint8 (b_send
, pos
, s
.s_protocol
);
214 b_write_uint8 (b_send
, pos
, s
.s_pw
);
218 int main (int argc
, char *argv
[]) {
219 d_getargs(argc
, argv
);
221 if (enet_initialize()) {
222 d_error("Could not init ENet!", 1);
226 printf(LC_MS_INIT
, ms_port
);
228 d_readtextfile(MS_MOTD_FILE
, ms_motd
, sizeof(ms_motd
));
229 d_readtextfile(MS_URGENT_FILE
, ms_urgent
, sizeof(ms_urgent
));
231 if (ms_motd
[0]) printf(LC_MS_MOTD
, ms_motd
);
232 if (ms_urgent
[0]) printf(LC_MS_URGENT
, ms_urgent
);
234 for (int i
= 0; i
< NET_MAXCLIENTS
; ++i
) ms_peers
[i
] = NULL
;
236 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
238 ms_srv
[i
].s_ip
[0] = '\0';
239 ms_srv
[i
].s_name
[0] = '\0';
240 ms_srv
[i
].s_map
[0] = '\0';
245 addr
.host
= ENET_HOST_ANY
;
248 ms_host
= enet_host_create(&addr
, NET_MAXCLIENTS
, NET_CHANS
, 0, 0);
250 d_error("Could not create host on specified port!", 1);
254 atexit(enet_deinitialize
);
258 enet_uint8 msg
= 255;
261 enet_uint16 port
= 0;
265 char *clientver
= NULL
;
270 enet_uint8 proto
= 0;
273 while (enet_host_service(ms_host
, &event
, 1) > 0) {
274 switch (event
.type
) {
275 case ENET_EVENT_TYPE_CONNECT
:
276 printf(LC_MS_CONN
, event
.peer
->address
.host
, event
.peer
->address
.port
);
278 case ENET_EVENT_TYPE_RECEIVE
:
279 if (!event
.peer
) continue;
281 msg
= b_read_uint8(event
.packet
->data
, &b_read
);
285 if (!event
.peer
) continue;
287 enet_address_get_host_ip(&(event
.peer
->address
), ip
, 17);
288 port
= b_read_uint16(event
.packet
->data
, &b_read
);
290 name
= b_read_dstring(event
.packet
->data
, &b_read
);
291 map
= b_read_dstring(event
.packet
->data
, &b_read
);
292 gm
= b_read_uint8(event
.packet
->data
, &b_read
);
294 pl
= b_read_uint8(event
.packet
->data
, &b_read
);
295 mpl
= b_read_uint8(event
.packet
->data
, &b_read
);
297 proto
= b_read_uint8(event
.packet
->data
, &b_read
);
298 pw
= b_read_uint8(event
.packet
->data
, &b_read
);
300 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
301 if (ms_srv
[i
].used
) {
302 if ((strncmp(ip
, ms_srv
[i
].s_ip
, 16) == 0) && (ms_srv
[i
].s_port
== port
)) {
303 strncpy(ms_srv
[i
].s_map
, map
, sizeof(ms_srv
[i
].s_map
));
304 strncpy(ms_srv
[i
].s_name
, name
, sizeof(ms_srv
[i
].s_name
));
305 ms_srv
[i
].s_plrs
= pl
;
306 ms_srv
[i
].s_maxplrs
= mpl
;
308 ms_srv
[i
].s_mode
= gm
;
310 ms_srv
[i
].ttl
= ms_timeout
;
312 printf(LC_MS_UPD
, i
, ip
, port
, name
, map
, gm
, pl
, mpl
, proto
, pw
);
316 strncpy(ms_srv
[i
].s_ip
, ip
, sizeof(ms_srv
[i
].s_ip
));
317 strncpy(ms_srv
[i
].s_map
, map
, sizeof(ms_srv
[i
].s_map
));
318 strncpy(ms_srv
[i
].s_name
, name
, sizeof(ms_srv
[i
].s_name
));
319 ms_srv
[i
].s_port
= port
;
320 ms_srv
[i
].s_plrs
= pl
;
321 ms_srv
[i
].s_maxplrs
= mpl
;
323 ms_srv
[i
].s_mode
= gm
;
324 ms_srv
[i
].s_protocol
= proto
;
325 ms_srv
[i
].ttl
= ms_timeout
;
329 printf(LC_MS_ADD
, i
, ip
, port
, name
, map
, gm
, pl
, mpl
, proto
, pw
);
339 enet_address_get_host_ip(&(event
.peer
->address
), ip
, 17);
340 port
= b_read_uint16(event
.packet
->data
, &b_read
);
341 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
342 if (ms_srv
[i
].used
) {
343 if ((strncmp(ip
, ms_srv
[i
].s_ip
, 16) == 0) && (ms_srv
[i
].s_port
== port
)) {
345 printf(LC_MS_RM
, i
, ip
, port
);
352 if (!event
.peer
) continue;
355 b_write_uint8(b_send
, &b_write
, NET_MSG_LIST
);
357 if (event
.packet
->dataLength
> 2) {
358 // holy shit a fresh client
359 clientver
= b_read_dstring(event
.packet
->data
, &b_read
);
360 b_write_uint8(b_send
, &b_write
, ms_count
);
362 // old client, feed them bullshit first
363 b_write_uint8(b_send
, &b_write
, ms_count
+ 2);
364 for (int i
= 0; i
< MS_FAKESRVS
; ++i
)
365 b_write_server(b_send
, &b_write
, ms_fake_srv
[i
]);
368 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
369 if (ms_srv
[i
].used
) b_write_server(b_send
, &b_write
, ms_srv
[i
]);
373 // TODO: check if this client is outdated (?) and send back new verstring
374 // for now just write the same shit back
375 b_write_dstring(b_send
, &b_write
, clientver
);
376 // write the motd and urgent message
377 b_write_dstring(b_send
, &b_write
, ms_motd
);
378 b_write_dstring(b_send
, &b_write
, ms_urgent
);
381 ENetPacket
*p
= enet_packet_create(b_send
, b_write
, ENET_PACKET_FLAG_RELIABLE
);
382 enet_peer_send(event
.peer
, NET_CH_MAIN
, p
);
383 enet_host_flush(ms_host
);
385 printf(LC_MS_LIST
, event
.peer
->address
.host
, event
.peer
->address
.port
, clientver
? clientver
: "<old>");
391 enet_packet_destroy(event
.packet
);
399 for (int i
= 0; i
< MS_MAXSRVS
; ++i
) {
400 if (ms_srv
[i
].used
) {
401 if (--(ms_srv
[i
].ttl
) == 0) {
403 printf(LC_MS_TIME
, i
, ms_srv
[i
].s_ip
, ms_srv
[i
].s_port
);