src/d2cs/server.c

/*
 * Copyright (C) 2000,2001      Onlyer  (onlyer@263.net)
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
#include "common/setup_before.h"
#include "setup.h"

#ifdef HAVE_STRING_H
# include <string.h>
#else
# ifdef HAVE_STRINGS_H
#  include <strings.h>
# endif
# ifdef HAVE_MEMORY_H
#  include <memory.h>
# endif
#endif
#ifdef STDC_HEADERS
# include <stdlib.h>
#else
# ifdef HAVE_MALLOC_H
#  include <malloc.h>
# endif
#endif
#include "compat/memset.h"
#include <errno.h>
#include "compat/strerror.h"
#ifdef TIME_WITH_SYS_TIME
# include <time.h>
# include <sys/time.h>
#else
# ifdef HAVE_SYS_TIME_H
#  include <sys/time.h>
# else
#  include <time.h>
# endif
#endif
#ifdef HAVE_FCNTL_H
# include <fcntl.h>
#else
# ifdef HAVE_SYS_FILE_H
#  include <sys/file.h>
# endif
#endif
#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif
#ifdef HAVE_SYS_SOCKET_H
# include <sys/socket.h>
#endif
#include "compat/psock.h"
#ifdef HAVE_SYS_TYPES_H
# include <sys/types.h>
#endif
#ifdef HAVE_NETINET_IN_H
# include <netinet/in.h>
#endif
#include "compat/netinet_in.h"
#ifdef HAVE_ARPA_INET_H
# include <arpa/inet.h>
#endif
#include "compat/inet_ntoa.h"
#include "compat/psock.h"

#include "d2gs.h"
#include "net.h"
#include "s2s.h"
#include "gamequeue.h"
#include "game.h"
#include "connection.h"
#include "serverqueue.h"
#include "common/fdwatch.h"
#include "server.h"
#include "prefs.h"
#include "d2ladder.h"
#ifndef WIN32
#include "handle_signal.h"
#endif
#include "common/addr.h"
#include "common/list.h"
#include "common/hashtable.h"
#include "common/eventlog.h"
#ifdef WIN32
# include <conio.h> /* for kbhit() and getch() */
#endif
#include "common/setup_after.h"

static int server_listen(void);
static int server_accept(int sock);
static int server_handle_timed_event(void);
static int server_handle_socket(void);
static int server_loop(void);
static int server_cleanup(void);

t_addrlist              * server_listen_addrs;
extern int g_ServiceStatus;

static int server_listen(void)
{
        t_addr          * curr_laddr;
        t_addr_data     laddr_data;
        int             sock;

        if (!(server_listen_addrs=addrlist_create(prefs_get_servaddrs(),INADDR_ANY,D2CS_SERVER_PORT))) {
                eventlog(eventlog_level_error,__FUNCTION__,"error create listening address list");
                return -1;
        }
        BEGIN_LIST_TRAVERSE_DATA(server_listen_addrs,curr_laddr)
        {
                sock=net_listen(addr_get_ip(curr_laddr),addr_get_port(curr_laddr),PSOCK_SOCK_STREAM);
                if (sock<0) {
                        eventlog(eventlog_level_error,__FUNCTION__,"error listen socket");
                        return -1;
                }

                if (psock_ctl(sock,PSOCK_NONBLOCK)<0) {
                        eventlog(eventlog_level_error,__FUNCTION__,"error set listen socket in non-blocking mode");
                }

                laddr_data.i = sock;
                addr_set_data(curr_laddr,laddr_data);

                if (fdwatch_add_fd(sock, fdwatch_type_read, d2cs_server_handle_accept, curr_laddr)<0) {
                    eventlog(eventlog_level_error,__FUNCTION__,"error adding socket %d to fdwatch pool (max sockets?)",sock);
                    psock_close(sock);
                    return -1;
                }

                eventlog(eventlog_level_info,__FUNCTION__,"listen on %s", addr_num_to_addr_str(addr_get_ip(curr_laddr),addr_get_port(curr_laddr)));
        }
        END_LIST_TRAVERSE_DATA()
        return 0;
}

static int server_accept(int sock)
{
        int                     csock;
        struct sockaddr_in      caddr, raddr;
        psock_t_socklen         caddr_len, raddr_len;
        int                     val;
        unsigned int            ip;
        unsigned short          port;
        t_connection           *cc;

        caddr_len=sizeof(caddr);
        memset(&caddr,0,sizeof(caddr));
        csock=psock_accept(sock,(struct sockaddr *)&caddr,&caddr_len);
        if (csock<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"error accept new connection");
                return -1;
        } else eventlog(eventlog_level_info,__FUNCTION__,"accept connection from %s",inet_ntoa(caddr.sin_addr));

        val=1;
        if (psock_setsockopt(csock, PSOCK_SOL_SOCKET, PSOCK_SO_KEEPALIVE, &val,sizeof(val))<0) {
                eventlog(eventlog_level_warn,__FUNCTION__,"error set sock option keep alive");
        }
        if (psock_ctl(csock, PSOCK_NONBLOCK)<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"error set socket to non-blocking mode");
                psock_close(csock);
                return -1;
        }

        raddr_len=sizeof(raddr);
        memset(&raddr,0,sizeof(raddr));
        ip=port=0;
        if (psock_getsockname(csock,(struct sockaddr *)&raddr,&raddr_len)<0) {
                eventlog(eventlog_level_warn,__FUNCTION__,"unable to get local socket info");
        } else {
                if (raddr.sin_family!=PSOCK_AF_INET) {
                        eventlog(eventlog_level_warn,__FUNCTION__,"got bad socket family %d",raddr.sin_family);
                } else {
                        ip=ntohl(raddr.sin_addr.s_addr);
                        port=ntohs(raddr.sin_port);
                }
        }
        if (!(cc = d2cs_conn_create(csock,ip,port,ntohl(caddr.sin_addr.s_addr),ntohs(caddr.sin_port)))) {
                eventlog(eventlog_level_error,__FUNCTION__,"error create new connection");
                psock_close(csock);
                return -1;
        }
        if (conn_add_fd(cc, fdwatch_type_read, d2cs_server_handle_tcp)<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"error adding socket %d to fdwatch pool (max sockets?)",csock);
                d2cs_conn_set_state(cc,conn_state_destroy);
                return -1;
        }

        return 0;
}

static int server_handle_timed_event(void)
{
        static  time_t  prev_list_purgetime=0;
        static  time_t  prev_gamequeue_checktime=0;
        static  time_t  prev_s2s_checktime=0;
        static  time_t  prev_sq_checktime=0;
        static  time_t  prev_d2ladder_refresh_time=0;
        static  time_t  prev_s2s_keepalive_time=0;
        static  time_t  prev_timeout_checktime;
        time_t          now;

        now=time(NULL);
        if (now-prev_list_purgetime>(signed)prefs_get_list_purgeinterval()) {
                hashtable_purge(d2cs_connlist());
                d2cs_gamelist_check_voidgame();
                prev_list_purgetime=now;
        }
        if (now-prev_gamequeue_checktime>(signed)prefs_get_gamequeue_checkinterval()) {
                gqlist_update_all_clients();
                prev_gamequeue_checktime=now;
        }
        if (now-prev_s2s_checktime>(signed)prefs_get_s2s_retryinterval()) {
                s2s_check();
                prev_s2s_checktime=now;
        }
        if (now-prev_sq_checktime>(signed)prefs_get_sq_checkinterval()) {
                sqlist_check_timeout();
                prev_sq_checktime=now;
        }
        if (now-prev_d2ladder_refresh_time>(signed)prefs_get_d2ladder_refresh_interval()) {
                d2ladder_refresh();
                prev_d2ladder_refresh_time=now;
        }
        if (now-prev_s2s_keepalive_time>(signed)prefs_get_s2s_keepalive_interval()) {
                d2gs_keepalive();
                prev_s2s_keepalive_time=now;
        }
        if (now-prev_timeout_checktime>(signed)prefs_get_timeout_checkinterval()) {
                connlist_check_timeout();
                prev_timeout_checktime=now;
        }
        return 0;
}

extern int d2cs_server_handle_accept(void *data, t_fdwatch_type rw)
{
    int sock;

    sock = addr_get_data((t_addr *)data).i;
    server_accept(sock);
    return 0;
}

extern int d2cs_server_handle_tcp(void *data, t_fdwatch_type rw)
{
    t_connection *c = (t_connection *)data;

    if (rw & fdwatch_type_read) conn_add_socket_flag(c,SOCKET_FLAG_READ);
    if (rw & fdwatch_type_write) conn_add_socket_flag(c,SOCKET_FLAG_WRITE);
    if (conn_handle_socket(c)<0)
        d2cs_conn_set_state(c, conn_state_destroy);

    return 0;
}

static int server_handle_socket(void)
{
        switch (fdwatch(BNETD_POLL_INTERVAL)) {
                case -1:
                        if (
#ifdef PSOCK_EINTR
                            psock_errno()!=PSOCK_EINTR &&
#endif
                            1) {
                                eventlog(eventlog_level_error,__FUNCTION__,"select failed (select: %s)",pstrerror(psock_errno()));
                                return -1;
                        }
                        /* fall through */
                case 0:
                        return 0;
                default:
                        break;
        }

        fdwatch_handle();

        return 0;
}

static int server_loop(void)
{
        unsigned int count;

        count=0;
        while (1) {

#ifdef WIN32
        if (g_ServiceStatus<0 && kbhit() && getch()=='q')
            signal_quit_wrapper();
        if (g_ServiceStatus == 0) signal_quit_wrapper();

        while (g_ServiceStatus == 2) Sleep(1000);
#endif

                if (handle_signal()<0) break;
                if (++count>=(1000/BNETD_POLL_INTERVAL)) {
                        server_handle_timed_event();
                        count=0;
                }
                server_handle_socket();
                d2cs_connlist_reap();
        }
        return 0;
}

static int server_cleanup(void)
{
        t_addr          * curr_laddr;
        int             sock;

        BEGIN_LIST_TRAVERSE_DATA(server_listen_addrs,curr_laddr)
        {
                sock=addr_get_data(curr_laddr).i;
                psock_close(sock);
        }
        END_LIST_TRAVERSE_DATA()
        addrlist_destroy(server_listen_addrs);
        return 0;
}

extern int d2cs_server_process(void)
{
#ifndef WIN32
        handle_signal_init();
#endif
        if (psock_init()<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"failed to init network");
                return -1;
        }
        eventlog(eventlog_level_info,__FUNCTION__,"network initialized");
        if (s2s_init()<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"failed to init s2s connection");
                return -1;
        }
        if (server_listen()<0) {
                eventlog(eventlog_level_error,__FUNCTION__,"failed to setup listen socket");
                return -1;
        }
        eventlog(eventlog_level_info,__FUNCTION__,"entering server loop");
        server_loop();
        eventlog(eventlog_level_info,__FUNCTION__,"exit from server loop,cleanup");
        server_cleanup();
        return 0;
}
1	/*
2	* Copyright (C) 2000,2001 Onlyer (onlyer@263.net)
3	*
4	* This program is free software; you can redistribute it and/or
5	* modify it under the terms of the GNU General Public License
6	* as published by the Free Software Foundation; either version 2
7	* of the License, or (at your option) any later version.
8	*
9	* This program is distributed in the hope that it will be useful,
10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12	* GNU General Public License for more details.
13	*
14	* You should have received a copy of the GNU General Public License
15	* along with this program; if not, write to the Free Software
16	* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17	*/
18	#include "common/setup_before.h"
19	#include "setup.h"
20
21	#ifdef HAVE_STRING_H
22	# include <string.h>
23	#else
24	# ifdef HAVE_STRINGS_H
25	# include <strings.h>
26	# endif
27	# ifdef HAVE_MEMORY_H
28	# include <memory.h>
29	# endif
30	#endif
31	#ifdef STDC_HEADERS
32	# include <stdlib.h>
33	#else
34	# ifdef HAVE_MALLOC_H
35	# include <malloc.h>
36	# endif
37	#endif
38	#include "compat/memset.h"
39	#include <errno.h>
40	#include "compat/strerror.h"
41	#ifdef TIME_WITH_SYS_TIME
42	# include <time.h>
43	# include <sys/time.h>
44	#else
45	# ifdef HAVE_SYS_TIME_H
46	# include <sys/time.h>
47	# else
48	# include <time.h>
49	# endif
50	#endif
51	#ifdef HAVE_FCNTL_H
52	# include <fcntl.h>
53	#else
54	# ifdef HAVE_SYS_FILE_H
55	# include <sys/file.h>
56	# endif
57	#endif
58	#ifdef HAVE_UNISTD_H
59	# include <unistd.h>
60	#endif
61	#ifdef HAVE_SYS_SOCKET_H
62	# include <sys/socket.h>
63	#endif
64	#include "compat/psock.h"
65	#ifdef HAVE_SYS_TYPES_H
66	# include <sys/types.h>
67	#endif
68	#ifdef HAVE_NETINET_IN_H
69	# include <netinet/in.h>
70	#endif
71	#include "compat/netinet_in.h"
72	#ifdef HAVE_ARPA_INET_H
73	# include <arpa/inet.h>
74	#endif
75	#include "compat/inet_ntoa.h"
76	#include "compat/psock.h"
77
78	#include "d2gs.h"
79	#include "net.h"
80	#include "s2s.h"
81	#include "gamequeue.h"
82	#include "game.h"
83	#include "connection.h"
84	#include "serverqueue.h"
85	#include "common/fdwatch.h"
86	#include "server.h"
87	#include "prefs.h"
88	#include "d2ladder.h"
89	#ifndef WIN32
90	#include "handle_signal.h"
91	#endif
92	#include "common/addr.h"
93	#include "common/list.h"
94	#include "common/hashtable.h"
95	#include "common/eventlog.h"
96	#ifdef WIN32
97	# include <conio.h> /* for kbhit() and getch() */
98	#endif
99	#include "common/setup_after.h"
100
101	static int server_listen(void);
102	static int server_accept(int sock);
103	static int server_handle_timed_event(void);
104	static int server_handle_socket(void);
105	static int server_loop(void);
106	static int server_cleanup(void);
107
108	t_addrlist * server_listen_addrs;
109	extern int g_ServiceStatus;
110
111	static int server_listen(void)
112	{
113	t_addr * curr_laddr;
114	t_addr_data laddr_data;
115	int sock;
116
117	if (!(server_listen_addrs=addrlist_create(prefs_get_servaddrs(),INADDR_ANY,D2CS_SERVER_PORT))) {
118	eventlog(eventlog_level_error,__FUNCTION__,"error create listening address list");
119	return -1;
120	}
121	BEGIN_LIST_TRAVERSE_DATA(server_listen_addrs,curr_laddr)
122	{
123	sock=net_listen(addr_get_ip(curr_laddr),addr_get_port(curr_laddr),PSOCK_SOCK_STREAM);
124	if (sock<0) {
125	eventlog(eventlog_level_error,__FUNCTION__,"error listen socket");
126	return -1;
127	}
128
129	if (psock_ctl(sock,PSOCK_NONBLOCK)<0) {
130	eventlog(eventlog_level_error,__FUNCTION__,"error set listen socket in non-blocking mode");
131	}
132
133	laddr_data.i = sock;
134	addr_set_data(curr_laddr,laddr_data);
135
136	if (fdwatch_add_fd(sock, fdwatch_type_read, d2cs_server_handle_accept, curr_laddr)<0) {
137	eventlog(eventlog_level_error,__FUNCTION__,"error adding socket %d to fdwatch pool (max sockets?)",sock);
138	psock_close(sock);
139	return -1;
140	}
141
142	eventlog(eventlog_level_info,__FUNCTION__,"listen on %s", addr_num_to_addr_str(addr_get_ip(curr_laddr),addr_get_port(curr_laddr)));
143	}
144	END_LIST_TRAVERSE_DATA()
145	return 0;
146	}
147
148	static int server_accept(int sock)
149	{
150	int csock;
151	struct sockaddr_in caddr, raddr;
152	psock_t_socklen caddr_len, raddr_len;
153	int val;
154	unsigned int ip;
155	unsigned short port;
156	t_connection *cc;
157
158	caddr_len=sizeof(caddr);
159	memset(&caddr,0,sizeof(caddr));
160	csock=psock_accept(sock,(struct sockaddr *)&caddr,&caddr_len);
161	if (csock<0) {
162	eventlog(eventlog_level_error,__FUNCTION__,"error accept new connection");
163	return -1;
164	} else eventlog(eventlog_level_info,__FUNCTION__,"accept connection from %s",inet_ntoa(caddr.sin_addr));
165
166	val=1;
167	if (psock_setsockopt(csock, PSOCK_SOL_SOCKET, PSOCK_SO_KEEPALIVE, &val,sizeof(val))<0) {
168	eventlog(eventlog_level_warn,__FUNCTION__,"error set sock option keep alive");
169	}
170	if (psock_ctl(csock, PSOCK_NONBLOCK)<0) {
171	eventlog(eventlog_level_error,__FUNCTION__,"error set socket to non-blocking mode");
172	psock_close(csock);
173	return -1;
174	}
175
176	raddr_len=sizeof(raddr);
177	memset(&raddr,0,sizeof(raddr));
178	ip=port=0;
179	if (psock_getsockname(csock,(struct sockaddr *)&raddr,&raddr_len)<0) {
180	eventlog(eventlog_level_warn,__FUNCTION__,"unable to get local socket info");
181	} else {
182	if (raddr.sin_family!=PSOCK_AF_INET) {
183	eventlog(eventlog_level_warn,__FUNCTION__,"got bad socket family %d",raddr.sin_family);
184	} else {
185	ip=ntohl(raddr.sin_addr.s_addr);
186	port=ntohs(raddr.sin_port);
187	}
188	}
189	if (!(cc = d2cs_conn_create(csock,ip,port,ntohl(caddr.sin_addr.s_addr),ntohs(caddr.sin_port)))) {
190	eventlog(eventlog_level_error,__FUNCTION__,"error create new connection");
191	psock_close(csock);
192	return -1;
193	}
194	if (conn_add_fd(cc, fdwatch_type_read, d2cs_server_handle_tcp)<0) {
195	eventlog(eventlog_level_error,__FUNCTION__,"error adding socket %d to fdwatch pool (max sockets?)",csock);
196	d2cs_conn_set_state(cc,conn_state_destroy);
197	return -1;
198	}
199
200	return 0;
201	}
202
203	static int server_handle_timed_event(void)
204	{
205	static time_t prev_list_purgetime=0;
206	static time_t prev_gamequeue_checktime=0;
207	static time_t prev_s2s_checktime=0;
208	static time_t prev_sq_checktime=0;
209	static time_t prev_d2ladder_refresh_time=0;
210	static time_t prev_s2s_keepalive_time=0;
211	static time_t prev_timeout_checktime;
212	time_t now;
213
214	now=time(NULL);
215	if (now-prev_list_purgetime>(signed)prefs_get_list_purgeinterval()) {
216	hashtable_purge(d2cs_connlist());
217	d2cs_gamelist_check_voidgame();
218	prev_list_purgetime=now;
219	}
220	if (now-prev_gamequeue_checktime>(signed)prefs_get_gamequeue_checkinterval()) {
221	gqlist_update_all_clients();
222	prev_gamequeue_checktime=now;
223	}
224	if (now-prev_s2s_checktime>(signed)prefs_get_s2s_retryinterval()) {
225	s2s_check();
226	prev_s2s_checktime=now;
227	}
228	if (now-prev_sq_checktime>(signed)prefs_get_sq_checkinterval()) {
229	sqlist_check_timeout();
230	prev_sq_checktime=now;
231	}
232	if (now-prev_d2ladder_refresh_time>(signed)prefs_get_d2ladder_refresh_interval()) {
233	d2ladder_refresh();
234	prev_d2ladder_refresh_time=now;
235	}
236	if (now-prev_s2s_keepalive_time>(signed)prefs_get_s2s_keepalive_interval()) {
237	d2gs_keepalive();
238	prev_s2s_keepalive_time=now;
239	}
240	if (now-prev_timeout_checktime>(signed)prefs_get_timeout_checkinterval()) {
241	connlist_check_timeout();
242	prev_timeout_checktime=now;
243	}
244	return 0;
245	}
246
247	extern int d2cs_server_handle_accept(void *data, t_fdwatch_type rw)
248	{
249	int sock;
250
251	sock = addr_get_data((t_addr *)data).i;
252	server_accept(sock);
253	return 0;
254	}
255
256	extern int d2cs_server_handle_tcp(void *data, t_fdwatch_type rw)
257	{
258	t_connection c = (t_connection )data;
259
260	if (rw & fdwatch_type_read) conn_add_socket_flag(c,SOCKET_FLAG_READ);
261	if (rw & fdwatch_type_write) conn_add_socket_flag(c,SOCKET_FLAG_WRITE);
262	if (conn_handle_socket(c)<0)
263	d2cs_conn_set_state(c, conn_state_destroy);
264
265	return 0;
266	}
267
268	static int server_handle_socket(void)
269	{
270	switch (fdwatch(BNETD_POLL_INTERVAL)) {
271	case -1:
272	if (
273	#ifdef PSOCK_EINTR
274	psock_errno()!=PSOCK_EINTR &&
275	#endif
276	1) {
277	eventlog(eventlog_level_error,__FUNCTION__,"select failed (select: %s)",pstrerror(psock_errno()));
278	return -1;
279	}
280	/* fall through */
281	case 0:
282	return 0;
283	default:
284	break;
285	}
286
287	fdwatch_handle();
288
289	return 0;
290	}
291
292	static int server_loop(void)
293	{
294	unsigned int count;
295
296	count=0;
297	while (1) {
298
299	#ifdef WIN32
300	if (g_ServiceStatus<0 && kbhit() && getch()=='q')
301	signal_quit_wrapper();
302	if (g_ServiceStatus == 0) signal_quit_wrapper();
303
304	while (g_ServiceStatus == 2) Sleep(1000);
305	#endif
306
307	if (handle_signal()<0) break;
308	if (++count>=(1000/BNETD_POLL_INTERVAL)) {
309	server_handle_timed_event();
310	count=0;
311	}
312	server_handle_socket();
313	d2cs_connlist_reap();
314	}
315	return 0;
316	}
317
318	static int server_cleanup(void)
319	{
320	t_addr * curr_laddr;
321	int sock;
322
323	BEGIN_LIST_TRAVERSE_DATA(server_listen_addrs,curr_laddr)
324	{
325	sock=addr_get_data(curr_laddr).i;
326	psock_close(sock);
327	}
328	END_LIST_TRAVERSE_DATA()
329	addrlist_destroy(server_listen_addrs);
330	return 0;
331	}
332
333	extern int d2cs_server_process(void)
334	{
335	#ifndef WIN32
336	handle_signal_init();
337	#endif
338	if (psock_init()<0) {
339	eventlog(eventlog_level_error,__FUNCTION__,"failed to init network");
340	return -1;
341	}
342	eventlog(eventlog_level_info,__FUNCTION__,"network initialized");
343	if (s2s_init()<0) {
344	eventlog(eventlog_level_error,__FUNCTION__,"failed to init s2s connection");
345	return -1;
346	}
347	if (server_listen()<0) {
348	eventlog(eventlog_level_error,__FUNCTION__,"failed to setup listen socket");
349	return -1;
350	}
351	eventlog(eventlog_level_info,__FUNCTION__,"entering server loop");
352	server_loop();
353	eventlog(eventlog_level_info,__FUNCTION__,"exit from server loop,cleanup");
354	server_cleanup();
355	return 0;
356	}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1