mirror of
https://git.code.sf.net/p/quake/quakeforge
synced 2024-11-23 21:02:50 +00:00
9fb337fac4
This unifies all the checks and plugs a set of potential buffer overflows.
271 lines
5.5 KiB
C
271 lines
5.5 KiB
C
/*
|
|
net_loop.c
|
|
|
|
Loopback network driver.
|
|
|
|
Copyright (C) 1996-1997 Id Software, Inc.
|
|
|
|
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:
|
|
|
|
Free Software Foundation, Inc.
|
|
59 Temple Place - Suite 330
|
|
Boston, MA 02111-1307, USA
|
|
|
|
*/
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include "QF/cvar.h"
|
|
#include "QF/msg.h"
|
|
#include "QF/sys.h"
|
|
|
|
#include "netmain.h"
|
|
#include "net_loop.h"
|
|
|
|
#include "../nq/include/client.h"
|
|
#include "../nq/include/server.h"
|
|
|
|
qboolean localconnectpending = false;
|
|
qsocket_t *loop_client = NULL;
|
|
qsocket_t *loop_server = NULL;
|
|
|
|
__attribute__((pure)) int
|
|
Loop_Init (void)
|
|
{
|
|
if (cls.state == ca_dedicated)
|
|
return -1;
|
|
return 0;
|
|
}
|
|
|
|
|
|
void
|
|
Loop_Shutdown (void)
|
|
{
|
|
}
|
|
|
|
|
|
void
|
|
Loop_Listen (qboolean state)
|
|
{
|
|
}
|
|
|
|
|
|
void
|
|
Loop_SearchForHosts (qboolean xmit)
|
|
{
|
|
if (!sv.active)
|
|
return;
|
|
|
|
const char *name = "local";
|
|
if (strcmp (hostname->string, "UNNAMED") != 0) {
|
|
name = hostname->string;
|
|
}
|
|
const char *map = sv.name;
|
|
int users = net_activeconnections;
|
|
int maxusers = svs.maxclients;
|
|
const char *cname = "local";
|
|
NET_AddCachedHost (name, map, cname, users, maxusers, net_driverlevel,
|
|
0, 0);
|
|
}
|
|
|
|
|
|
qsocket_t *
|
|
Loop_Connect (const char *host)
|
|
{
|
|
if (strcmp (host, "local") != 0)
|
|
return NULL;
|
|
|
|
localconnectpending = true;
|
|
|
|
if (!loop_client) {
|
|
if ((loop_client = NET_NewQSocket ()) == NULL) {
|
|
Sys_Printf ("Loop_Connect: no qsocket available\n");
|
|
return NULL;
|
|
}
|
|
strcpy (loop_client->address, "localhost");
|
|
}
|
|
loop_client->receiveMessageLength = 0;
|
|
loop_client->sendMessageLength = 0;
|
|
loop_client->canSend = true;
|
|
|
|
if (!loop_server) {
|
|
if ((loop_server = NET_NewQSocket ()) == NULL) {
|
|
Sys_Printf ("Loop_Connect: no qsocket available\n");
|
|
return NULL;
|
|
}
|
|
strcpy (loop_server->address, "LOCAL");
|
|
}
|
|
loop_server->receiveMessageLength = 0;
|
|
loop_server->sendMessageLength = 0;
|
|
loop_server->canSend = true;
|
|
|
|
loop_client->driverdata = (void *) loop_server;
|
|
loop_server->driverdata = (void *) loop_client;
|
|
|
|
return loop_client;
|
|
}
|
|
|
|
|
|
qsocket_t *
|
|
Loop_CheckNewConnections (void)
|
|
{
|
|
if (!localconnectpending)
|
|
return NULL;
|
|
|
|
localconnectpending = false;
|
|
loop_server->sendMessageLength = 0;
|
|
loop_server->receiveMessageLength = 0;
|
|
loop_server->canSend = true;
|
|
loop_client->sendMessageLength = 0;
|
|
loop_client->receiveMessageLength = 0;
|
|
loop_client->canSend = true;
|
|
return loop_server;
|
|
}
|
|
|
|
|
|
static int
|
|
IntAlign (int value)
|
|
{
|
|
return (value + (sizeof (int) - 1)) & (~(sizeof (int) - 1));
|
|
}
|
|
|
|
|
|
int
|
|
Loop_GetMessage (qsocket_t * sock)
|
|
{
|
|
int ret;
|
|
int length;
|
|
|
|
if (sock->receiveMessageLength == 0)
|
|
return 0;
|
|
|
|
ret = sock->receiveMessage[0];
|
|
length = sock->receiveMessage[1] + (sock->receiveMessage[2] << 8);
|
|
// alignment byte skipped here
|
|
SZ_Clear (net_message->message);
|
|
SZ_Write (net_message->message, &sock->receiveMessage[4], length);
|
|
|
|
length = IntAlign (length + 4);
|
|
sock->receiveMessageLength -= length;
|
|
|
|
if (sock->receiveMessageLength)
|
|
memmove (sock->receiveMessage, &sock->receiveMessage[length],
|
|
sock->receiveMessageLength);
|
|
|
|
if (sock->driverdata && ret == 1)
|
|
((qsocket_t *) sock->driverdata)->canSend = true;
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
int
|
|
Loop_SendMessage (qsocket_t * sock, sizebuf_t *data)
|
|
{
|
|
byte *buffer;
|
|
int *bufferLength;
|
|
|
|
if (!sock->driverdata)
|
|
return -1;
|
|
|
|
bufferLength = &((qsocket_t *) sock->driverdata)->receiveMessageLength;
|
|
|
|
if ((*bufferLength + data->cursize + 4) > NET_MAXMESSAGE)
|
|
Sys_Error ("Loop_SendMessage: overflow");
|
|
|
|
buffer = ((qsocket_t *) sock->driverdata)->receiveMessage + *bufferLength;
|
|
|
|
// message type
|
|
*buffer++ = 1;
|
|
|
|
// length
|
|
*buffer++ = data->cursize & 0xff;
|
|
*buffer++ = data->cursize >> 8;
|
|
|
|
// align
|
|
buffer++;
|
|
|
|
// message
|
|
memcpy (buffer, data->data, data->cursize);
|
|
*bufferLength = IntAlign (*bufferLength + data->cursize + 4);
|
|
|
|
sock->canSend = false;
|
|
return 1;
|
|
}
|
|
|
|
|
|
int
|
|
Loop_SendUnreliableMessage (qsocket_t * sock, sizebuf_t *data)
|
|
{
|
|
byte *buffer;
|
|
int *bufferLength;
|
|
|
|
if (!sock->driverdata)
|
|
return -1;
|
|
|
|
bufferLength = &((qsocket_t *) sock->driverdata)->receiveMessageLength;
|
|
|
|
if ((*bufferLength + data->cursize + sizeof (byte) + sizeof (short)) >
|
|
NET_MAXMESSAGE) return 0;
|
|
|
|
buffer = ((qsocket_t *) sock->driverdata)->receiveMessage + *bufferLength;
|
|
|
|
// message type
|
|
*buffer++ = 2;
|
|
|
|
// length
|
|
*buffer++ = data->cursize & 0xff;
|
|
*buffer++ = data->cursize >> 8;
|
|
|
|
// align
|
|
buffer++;
|
|
|
|
// message
|
|
memcpy (buffer, data->data, data->cursize);
|
|
*bufferLength = IntAlign (*bufferLength + data->cursize + 4);
|
|
return 1;
|
|
}
|
|
|
|
|
|
__attribute__((pure)) qboolean
|
|
Loop_CanSendMessage (qsocket_t * sock)
|
|
{
|
|
if (!sock->driverdata)
|
|
return false;
|
|
return sock->canSend;
|
|
}
|
|
|
|
|
|
__attribute__((const)) qboolean
|
|
Loop_CanSendUnreliableMessage (qsocket_t * sock)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
|
|
void
|
|
Loop_Close (qsocket_t * sock)
|
|
{
|
|
if (sock->driverdata)
|
|
((qsocket_t *) sock->driverdata)->driverdata = NULL;
|
|
sock->receiveMessageLength = 0;
|
|
sock->sendMessageLength = 0;
|
|
sock->canSend = true;
|
|
if (sock == loop_client)
|
|
loop_client = NULL;
|
|
else
|
|
loop_server = NULL;
|
|
}
|