[ViewVC] Diff of: cvs/deliantra/server/socket/lowlevel.C

Comparing deliantra/server/socket/lowlevel.C (file contents):
Revision 1.20 by root, Thu Dec 14 21:46:34 2006 UTC vs.
Revision 1.25 by root, Sun Dec 17 19:14:00 2006 UTC

…		…
33	*/	33	*/
34		34
35	using namespace std;	35	using namespace std;
36		36
37	#include <global.h>	37	#include <global.h>
38	#include <newclient.h>
39	#include <sproto.h>	38	#include <sproto.h>
40	#include <cstdarg>	39	#include <cstdarg>
41		40
42	#ifdef __linux__	41	#ifdef __linux__
43	# include <sys/types.h>	42	# include <sys/types.h>
…		…
51	// easily die in 20 seconds...	50	// easily die in 20 seconds...
52	#define SOCKET_TIMEOUT1 10	51	#define SOCKET_TIMEOUT1 10
53	#define SOCKET_TIMEOUT2 20	52	#define SOCKET_TIMEOUT2 20
54		53
55	void	54	void
56	client_socket::flush ()	55	client::flush ()
57	{	56	{
58	#ifdef __linux__	57	#ifdef __linux__
59	// check time of last ack, and, if too old, kill connection	58	// check time of last ack, and, if too old, kill connection
60	struct tcp_info tcpi;	59	struct tcp_info tcpi;
61	socklen_t len = sizeof (tcpi);	60	socklen_t len = sizeof (tcpi);
62		61
63	if (!getsockopt (fd, IPPROTO_TCP, TCP_INFO, &tcpi, &len) && len == sizeof (tcpi))	62	if (!getsockopt (fd, IPPROTO_TCP, TCP_INFO, &tcpi, &len) && len == sizeof (tcpi))
64	{	63	{
65	unsigned int diff = tcpi.tcpi_last_ack_recv - tcpi.tcpi_last_data_sent;	64	unsigned int diff = tcpi.tcpi_last_ack_recv - tcpi.tcpi_last_data_sent;
66		65
		66	rtt = tcpi.tcpi_rtt;
		67	rttvar = tcpi.tcpi_rttvar;
		68
67	if (tcpi.tcpi_unacked && SOCKET_TIMEOUT1 * TCP_HZ < diff && diff < 0x80000000UL // ack delayed for 20s	69	if (tcpi.tcpi_unacked && SOCKET_TIMEOUT1 * TCP_HZ < diff && diff < 0x80000000UL // ack delayed for 20s
68	&& SOCKET_TIMEOUT2 * TCP_HZ < tcpi.tcpi_last_data_sent) // no data sent for 10s	70	&& SOCKET_TIMEOUT2 * TCP_HZ < tcpi.tcpi_last_data_sent) // no data sent for 10s
69	{	71	{
70	LOG (llevDebug, "Connection on fd %d closed due to ack timeout (%u/%u/%u)\n", fd,	72	LOG (llevDebug, "Connection on fd %d closed due to ack timeout (%u/%u/%u)\n", fd,
71	(unsigned) tcpi.tcpi_last_ack_recv, (unsigned) tcpi.tcpi_last_data_sent, (unsigned) tcpi.tcpi_unacked);	73	(unsigned) tcpi.tcpi_last_ack_recv, (unsigned) tcpi.tcpi_last_data_sent, (unsigned) tcpi.tcpi_unacked);
72	status = Ns_Dead;	74	destroy ();
73	}	75	}
74	}	76	}
75	#endif	77	#endif
76		78
77	/**	79	/**
…		…
86		88
87	write_outputbuffer ();	89	write_outputbuffer ();
88	}	90	}
89		91
90	void	92	void
91	client_socket::write_outputbuffer ()	93	client::write_outputbuffer ()
92	{	94	{
93	while (outputbuffer.len)	95	while (outputbuffer.len)
94	{	96	{
95	int res = write (fd, outputbuffer.data + outputbuffer.start,	97	int res = write (fd, outputbuffer.data + outputbuffer.start,
96	min (outputbuffer.len, SOCKETBUFSIZE - outputbuffer.start));	98	min (outputbuffer.len, SOCKETBUFSIZE - outputbuffer.start));
…		…
109	#endif	111	#endif
110	}	112	}
111	else if (res == 0)	113	else if (res == 0)
112	{	114	{
113	LOG (llevError, "socket write failed, connection closed.\n");	115	LOG (llevError, "socket write failed, connection closed.\n");
114	status = Ns_Dead;	116	destroy ();
115	return;	117	return;
116	}	118	}
117	else if (errno == EINTR)	119	else if (errno == EINTR)
118	{	120	{
119	// just retry	121	// just retry
…		…
126	return;	128	return;
127	}	129	}
128	else	130	else
129	{	131	{
130	LOG (llevError, "socket write failed: %s\n", strerror (errno));	132	LOG (llevError, "socket write failed: %s\n", strerror (errno));
131	status = Ns_Dead;	133	destroy ();
132	return;	134	return;
133	}	135	}
134	}	136	}
135		137
136	socket_ev.poll (socket_ev.poll () & ~PE_W);	138	socket_ev.poll (socket_ev.poll () & ~PE_W);
137	}	139	}
138		140
		141	/******************************************************************************
		142	*
		143	* Start of read routines.
		144	*
		145	******************************************************************************/
		146
		147	int
		148	client::next_packet ()
		149	{
		150	if (inbuf_len >= 2)
		151	{
		152	int pkt_len = (inbuf [0] << 8) \| inbuf [1];
		153
		154	if (inbuf_len >= 2 + pkt_len)
		155	return 2 + pkt_len;
		156
		157	if (inbuf_len == sizeof (inbuf))
		158	{
		159	send_packet_printf ("drawinfo %d input buffer overflow - closing connection.", NDI_RED);
		160	destroy ();
		161	return -1;
		162	}
		163	}
		164
		165	return 0;
		166	}
		167
		168	void
		169	client::skip_packet (int len)
		170	{
		171	inbuf_len -= len;
		172	memmove (inbuf, inbuf + len, inbuf_len);
		173	}
		174
139	/***********************************************************************	175	/*****************************************************************************
140	*	176	* Start of command dispatch area.
141	* packet functions/utilities	177	* The commands here are protocol commands.
142	*
143	**********************************************************************/	178	****************************************************************************/
144		179
145	packet &packet::operator <<(const data &v)	180	// SocketCommand, PlayingCommand, should not exist with those ugly casts
		181	#define SC(cb) (void )static_cast<void ()(char , int, client )>(cb),
		182	#define PC(cb) (void )static_cast<void ()(char , int, player )>(cb), PF_PLAYER \|
		183
		184	/**
		185	* Dispatch table for the server.
		186	*/
		187	static struct packet_type packets[] = {
		188	{"ncom", PC(NewPlayerCmd) PF_PLAYING },
		189	{"command", PC(PlayerCmd) PF_PLAYING },
		190
		191	{"examine", PC(ExamineCmd) PF_PLAYING },
		192	{"apply", PC(ApplyCmd) PF_PLAYING },
		193	{"reply", PC(ReplyCmd) PF_IMMEDIATE },
		194	{"lookat", PC(LookAt) PF_PLAYING },
		195	{"lock", PC(LockItem) PF_PLAYING },
		196	{"mark", PC(MarkItem) PF_PLAYING },
		197	{"move", PC(MoveCmd) PF_PLAYING },
		198	{"exti", PC(ExtCmd) PF_IMMEDIATE }, /* CF+ */
		199	{"ext", PC(ExtCmd) 0 }, /* CF+ */
		200	{"mapredraw", PC(MapRedrawCmd) PF_IMMEDIATE }, /* Added: phil */
		201	{"mapinfo", PC(MapInfoCmd) 0 }, /* CF+ */
		202
		203	{"addme", SC(AddMeCmd) PF_IMMEDIATE },
		204	{"askface", SC(SendFaceCmd) PF_IMMEDIATE }, /* Added: phil */
		205	{"requestinfo", SC(RequestInfo) PF_IMMEDIATE },
		206	{"setfacemode", SC(SetFaceMode) PF_IMMEDIATE },
		207	{"setsound", SC(SetSound) PF_IMMEDIATE },
		208	{"setup", SC(SetUp) PF_IMMEDIATE },
		209	{"version", SC(VersionCmd) PF_IMMEDIATE },
		210	{"toggleextendedinfos", SC(ToggleExtendedInfos) PF_IMMEDIATE }, /Added: tchize /
		211	{"toggleextendedtext", SC(ToggleExtendedText) PF_IMMEDIATE }, /Added: tchize /
		212	{"asksmooth", SC(AskSmooth) PF_IMMEDIATE }, /Added: tchize (smoothing technologies) /
		213	};
		214
		215	bool
		216	client::may_execute (const packet_type *pkt) const
146	{	217	{
147	if (room () < v.len)	218	return (!(pkt->flags & PF_PLAYER) \|\| pl)
148	reset ();	219	&& (!(pkt->flags & PF_PLAYING) \|\| (pl && pl->state == ST_PLAYING));
		220	}
		221
		222	void
		223	client::execute (const packet_type pkt, char data, int datalen)
		224	{
		225	if (may_execute (pkt))
		226	{
		227	//TODO: only one format
		228	if (pkt->flags & PF_PLAYER)
		229	((void ()(char , int, player ))pkt->cb)((char )data, datalen, pl);
		230	else
		231	((void ()(char , int, client ))pkt->cb)((char )data, datalen, this);
		232	}
149	else	233	else
		234	send_packet_printf ("drawinfo %d ERROR: you cannot execute '%s' now.", NDI_RED, pkt->name);
		235	}
		236
		237	bool
		238	client::handle_packet ()
		239	{
		240	int pkt_len = next_packet ();
		241
		242	if (!pkt_len)
		243	return false;
		244	else if (pkt_len < 0)
		245	{
		246	LOG (llevError, "read error on player %s\n",
		247	pl && pl->ob ? &pl->ob->name : "[anonymous]");
		248	destroy ();
		249	return false;
150	{	250	}
151	if (v.len)	251
		252	inbuf [pkt_len] = 0; /* Terminate buffer - useful for string data */
		253
		254	/* First, break out beginning word. There are at least
		255	* a few commands that do not have any paremeters. If
		256	* we get such a command, don't worry about trying
		257	* to break it up.
		258	*/
		259	int datalen;
		260	char data = strchr ((char )inbuf + 2, ' ');
		261
		262	if (data)
		263	{
		264	*data++ = 0;
		265	datalen = pkt_len - (data - (char *)inbuf);
		266	}
		267	else
		268	{
		269	data = (char *)inbuf + 2; // better read garbage than segfault
		270	datalen = 0;
		271	}
		272
		273	for (packet_type *pkt = packets; pkt < packets + (sizeof (packets) / sizeof (packets[0])); ++pkt)
		274	if (!strcmp ((char *)inbuf + 2, pkt->name))
152	{	275	{
153	memcpy (cur, v.ptr, v.len);	276	if (pkt->flags & PF_IMMEDIATE)
154	cur += v.len;	277	execute (pkt, data, datalen);
		278	else
		279	queue_command (pkt, data, datalen);
		280
		281	goto next_packet;
155	}	282	}
		283
		284	/* If we get here, we didn't find a valid command. Logging
		285	* this might be questionable, because a broken client/malicious
		286	* user could certainly send a whole bunch of invalid commands.
		287	*/
		288	send_packet_printf ("drawinfo %d ERROR: command '%s' not supported.", NDI_RED, (char *)inbuf + 2);
		289	next_packet:
		290	skip_packet (pkt_len);
		291
		292	// input buffer has space again
		293	socket_ev.poll (socket_ev.poll () \| PE_R);
		294
		295	return true;
		296	}
		297
		298	// callback called when socket is either readable or writable
		299	void
		300	client::socket_cb (iow &w, int got)
		301	{
		302	//TODO remove when we have better socket cleanup logic
		303	if (status == Ns_Dead)
156	}	304	{
157		305	socket_ev.poll (0);
158	return *this;	306	return;
159	}
160
161	packet &packet::operator <<(const data8 &v)
162	{
163	unsigned int len = min (v.len, 0x00FF);
164	return *this << uint8 (len) << data (v.ptr, len);
165	}
166
167	packet &packet::operator <<(const data16 &v)
168	{
169	unsigned int len = min (v.len, 0xFFFF);
170	return *this << uint16 (len) << data (v.ptr, len);
171	}
172
173	packet &packet::operator <<(const char *v)
174	{
175	return *this << data (v, strlen (v ? v : 0));
176	}
177
178	void
179	packet::printf (const char *format, ...)
180	{
181	int size = room ();
182
183	va_list ap;
184	va_start (ap, format);
185	int len = vsnprintf ((char *)cur, size, format, ap);
186	va_end (ap);
187
188	if (len >= size)
189	return reset ();
190
191	cur += len;
192	}
193
194	/******************************************************************************
195	*
196	* Start of read routines.
197	*
198	******************************************************************************/
199
200	int
201	client_socket::read_packet ()
202	{
203	for (;;)
204	{	307	}
205	if (inbuf_len >= 2)
206	{
207	unsigned int pkt_len = (inbuf [0] << 8) \| inbuf [1];
208		308
209	if (inbuf_len >= 2 + pkt_len)	309	if (got & PE_W)
210	return pkt_len + 2;	310	{
211	}	311	write_outputbuffer ();
		312
		313	if (!outputbuffer.len)
		314	socket_ev.poll (socket_ev.poll () & ~PE_W);
		315	}
		316
		317	if (got & PE_R)
		318	{
		319	//TODO: rate-limit tcp connection in better ways, important
212		320
213	int amount = sizeof (inbuf) - inbuf_len;	321	int amount = sizeof (inbuf) - inbuf_len;
214		322
215	if (amount <= 0)	323	if (!amount)
216	{	324	{
217	LOG (llevError, "packet too large");//TODO	325	// input buffer full
		326	socket_ev.poll (socket_ev.poll () & ~PE_R);
218	return -1;	327	return;
219	}	328	}
220		329
221	amount = read (fd, inbuf + inbuf_len, amount);	330	amount = read (fd, inbuf + inbuf_len, amount);
222		331
223	if (!amount)	332	if (!amount)
224	{	333	{
225	status = Ns_Dead;	334	destroy ();
226	return -1;	335	return;
227	}	336	}
228	else if (amount < 0)	337	else if (amount < 0)
229	{	338	{
230	if (errno != EAGAIN && errno != EINTR)	339	if (errno != EAGAIN && errno != EINTR)
231	{	340	{
232	LOG (llevError, "read error: %s\n", strerror (errno));	341	LOG (llevError, "read error: %s\n", strerror (errno));
		342	destroy ();
233	return -1;	343	return;
234	}	344	}
235		345
236	return 0;	346	// should not be here, normally
		347	}
		348	else
237	}	349	{
238
239	inbuf_len += amount;	350	inbuf_len += amount;
240		351
241	cst_tot.ibytes += amount;	352	cst_tot.ibytes += amount;
242	cst_lst.ibytes += amount;	353	cst_lst.ibytes += amount;
243	}
244	}
245		354
		355	cmd_ev.start ();
		356	}
		357	}
		358	}
		359
		360	// called whenever we have additional commands to process
246	void	361	void
247	client_socket::skip_packet (int len)	362	client::cmd_cb (iw &w)
248	{	363	{
249	inbuf_len -= len;	364	if (handle_packet () \|\| handle_command ())
250	memmove (inbuf, inbuf + len, inbuf_len);	365	w.start ();
		366	else
		367	flush ();
251	}	368	}
252		369
253	/*******************************************************************************	370	/*******************************************************************************
254	*	371	*
255	* Start of write related routines.	372	* Start of write related routines.
…		…
261	*	378	*
262	* ns is the socket we are adding the data to, buf is the start of the	379	* ns is the socket we are adding the data to, buf is the start of the
263	* data, and len is the number of bytes to add.	380	* data, and len is the number of bytes to add.
264	*/	381	*/
265	void	382	void
266	client_socket::send (void *buf_, int len)	383	client::send (void *buf_, int len)
267	{	384	{
268	char buf = (char )buf_;	385	char buf = (char )buf_;
269	char *pos = buf;	386	char *pos = buf;
270	int amt = 0;	387	int amt = 0;
271		388
272	if (status == Ns_Dead \|\| !buf)	389	if (status == Ns_Dead \|\| !buf)
273	{
274	LOG (llevDebug, "Write_To_Socket called with dead socket\n");
275	return;	390	return;
276	}
277		391
278	if ((len + outputbuffer.len) > SOCKETBUFSIZE)	392	if ((len + outputbuffer.len) > SOCKETBUFSIZE)
279	{	393	{
280	LOG (llevDebug, "Socket on fd %d has overrun internal buffer - marking as dead\n", fd);	394	LOG (llevDebug, "socket on fd %d has overrun internal buffer - marking as dead\n", fd);
281	status = Ns_Dead;	395	destroy ();
282	return;	396	return;
283	}	397	}
284		398
285	int avail, end;	399	int avail, end;
286		400
…		…
304	}	418	}
305		419
306	outputbuffer.len += len;	420	outputbuffer.len += len;
307	}	421	}
308		422
309	void
310	client_socket::socket_cb (iow &w, int got)
311	{
312	write_outputbuffer ();
313
314	if (!outputbuffer.len)
315	socket_ev.poll (socket_ev.poll () & ~PE_W);
316	}
317
318	/**	423	/**
319	* Takes a string of data, and writes it out to the socket. A very handy	424	* Takes a string of data, and writes it out to the socket. A very handy
320	* shortcut function.	425	* shortcut function.
321	*/	426	*/
322	void	427	void
323	client_socket::send_packet (packet &sl)	428	client::send_packet (packet &sl)
324	{	429	{
325	if (status == Ns_Dead)	430	if (status == Ns_Dead)
326	return;	431	return;
327		432
328	if (sl.length () >= MAXSOCKBUF)	433	if (sl.length () >= MAXSOCKBUF)
…		…
344		449
345	send (sl.buf_, sl.length () + sl.hdrlen);	450	send (sl.buf_, sl.length () + sl.hdrlen);
346	}	451	}
347		452
348	void	453	void
349	client_socket::send_packet (const char *buf, int len)	454	client::send_packet (const char *buf, int len)
350	{	455	{
351	packet sl;	456	packet sl;
352		457
353	sl << data (buf, len);	458	sl << data (buf, len);
354	send_packet (sl);	459	send_packet (sl);
355	}	460	}
356		461
357	void	462	void
358	client_socket::send_packet (const char *buf)	463	client::send_packet (const char *buf)
359	{	464	{
360	send_packet (buf, strlen (buf));	465	send_packet (buf, strlen (buf));
		466	}
		467
		468	void
		469	client::send_packet_printf (const char *format, ...)
		470	{
		471	packet sl;
		472
		473	va_list ap;
		474	va_start (ap, format);
		475	sl.vprintf (format, ap);
		476	va_end (ap);
		477
		478	send_packet (sl);
		479	}
		480
		481	/***********************************************************************
		482	*
		483	* packet functions/utilities
		484	*
		485	**********************************************************************/
		486
		487	packet &packet::operator <<(const data &v)
		488	{
		489	if (room () < v.len)
		490	reset ();
		491	else
		492	{
		493	if (v.len)
		494	{
		495	memcpy (cur, v.ptr, v.len);
		496	cur += v.len;
		497	}
		498	}
		499
		500	return *this;
		501	}
		502
		503	packet &packet::operator <<(const data8 &v)
		504	{
		505	unsigned int len = min (v.len, 0x00FF);
		506	return *this << uint8 (len) << data (v.ptr, len);
		507	}
		508
		509	packet &packet::operator <<(const data16 &v)
		510	{
		511	unsigned int len = min (v.len, 0xFFFF);
		512	return *this << uint16 (len) << data (v.ptr, len);
		513	}
		514
		515	packet &packet::operator <<(const char *v)
		516	{
		517	return *this << data (v, strlen (v ? v : 0));
		518	}
		519
		520	void
		521	packet::vprintf (const char *format, va_list ap)
		522	{
		523	int size = room ();
		524
		525	int len = vsnprintf ((char *)cur, size, format, ap);
		526
		527	if (len >= size)
		528	return reset ();
		529
		530	cur += len;
361	}	531	}
362		532
363	/******************************************************************************	533	/******************************************************************************
364	*	534	*
365	* statistics logging functions.	535	* statistics logging functions.

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing deliantra/server/socket/lowlevel.C (file contents): Revision 1.20 by root, Thu Dec 14 21:46:34 2006 UTC vs. Revision 1.25 by root, Sun Dec 17 19:14:00 2006 UTC

Diff Legend

Comparing deliantra/server/socket/lowlevel.C (file contents):
Revision 1.20 by root, Thu Dec 14 21:46:34 2006 UTC vs.
Revision 1.25 by root, Sun Dec 17 19:14:00 2006 UTC