1

/* $Id$ */

     2

     3

#ifdef ENABLE_NETWORK

     4

     5

#include "../../stdafx.h"

     6

#include "../../date.h"

     7

#include "../../debug.h"

     8

#include "../../macros.h"

     9

#include "../../newgrf_config.h"

    10

    11

#include "os_abstraction.h"

    12

#include "config.h"

    13

#include "game.h"

    14

#include "packet.h"

    15

#include "udp.h"

    16

    17

/**

    18

 * @file udp.c Basic functions to receive and send UDP packets.

    19

 */

    20

    21

/**

    22

 * Send a packet over UDP

    23

 * @param udp  the socket to send over

    24

 * @param p    the packet to send

    25

 * @param recv the receiver (target) of the packet

    26

 */

    27

void NetworkSendUDP_Packet(SOCKET udp, Packet *p, struct sockaddr_in *recv)

    28

{

    29

	int res;

    30

    31

	NetworkSend_FillPacketSize(p);

    32

    33

	/* Send the buffer */

    34

	res = sendto(udp, p->buffer, p->size, 0, (struct sockaddr *)recv, sizeof(*recv));

    35

    36

	/* Check for any errors, but ignore it otherwise */

    37

	if (res == -1) DEBUG(net, 1, "[udp] sendto failed with: %i", GET_LAST_ERROR());

    38

}

    39

    40

/**

    41

 * Start listening on the given host and port.

    42

 * @param udp       the place where the (references to the) UDP are stored

    43

 * @param host      the host (ip) to listen on

    44

 * @param port      the port to listen on

    45

 * @param broadcast whether to allow broadcast sending/receiving

    46

 * @return true if the listening succeeded

    47

 */

    48

bool NetworkUDPListen(SOCKET *udp, uint32 host, uint16 port, bool broadcast)

    49

{

    50

	struct sockaddr_in sin;

    51

    52

	/* Make sure socket is closed */

    53

	closesocket(*udp);

    54

    55

	*udp = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

    56

	if (*udp == INVALID_SOCKET) {

    57

		DEBUG(net, 0, "[udp] failed to start UDP listener");

    58

		return false;

    59

	}

    60

    61

	/* set nonblocking mode for socket */

    62

	{

    63

		unsigned long blocking = 1;

    64

#ifndef BEOS_NET_SERVER

    65

		ioctlsocket(*udp, FIONBIO, &blocking);

    66

#else

    67

		setsockopt(*udp, SOL_SOCKET, SO_NONBLOCK, &blocking, NULL);

    68

#endif

    69

	}

    70

    71

	sin.sin_family = AF_INET;

    72

	/* Listen on all IPs */

    73

	sin.sin_addr.s_addr = host;

    74

	sin.sin_port = htons(port);

    75

    76

	if (bind(*udp, (struct sockaddr*)&sin, sizeof(sin)) != 0) {

    77

		DEBUG(net, 0, "[udp] bind failed on %s:%i", inet_ntoa(*(struct in_addr *)&host), port);

    78

		return false;

    79

	}

    80

    81

	if (broadcast) {

    82

		/* Enable broadcast */

    83

		unsigned long val = 1;

    84

#ifndef BEOS_NET_SERVER // will work around this, some day; maybe.

    85

		setsockopt(*udp, SOL_SOCKET, SO_BROADCAST, (char *) &val , sizeof(val));

    86

#endif

    87

	}

    88

    89

	DEBUG(net, 1, "[udp] listening on port %s:%d", inet_ntoa(*(struct in_addr *)&host), port);

    90

    91

	return true;

    92

}

    93

    94

/**

    95

 * Receive a packet at UDP level

    96

 * @param udp the socket to receive the packet on

    97

 */

    98

void NetworkUDPReceive(SOCKET udp)

    99

{

   100

	struct sockaddr_in client_addr;

   101

	socklen_t client_len;

   102

	int nbytes;

   103

	Packet p;

   104

	int packet_len;

   105

   106

	packet_len = sizeof(p.buffer);

   107

	client_len = sizeof(client_addr);

   108

   109

	/* Try to receive anything */

   110

	nbytes = recvfrom(udp, p.buffer, packet_len, 0, (struct sockaddr *)&client_addr, &client_len);

   111

   112

	/* We got some bytes for the base header of the packet.

   113

	 * Assume we received the whole packet. */

   114

	if (nbytes > 2) {

   115

		NetworkRecv_ReadPacketSize(&p);

   116

   117

		/* Put the position on the right place */

   118

		p.pos = 2;

   119

		p.next = NULL;

   120

   121

		/* Handle the packet */

   122

		NetworkHandleUDPPacket(&p, &client_addr);

   123

	}

   124

}

   125

   126

   127

/**

   128

 * Serializes the GRFIdentifier (GRF ID and MD5 checksum) to the packet

   129

 * @param p the packet to write the data to

   130

 * @param c the configuration to write the GRF ID and MD5 checksum from

   131

 */

   132

void NetworkSend_GRFIdentifier(Packet *p, const GRFConfig *c)

   133

{

   134

	uint j;

   135

	NetworkSend_uint32(p, c->grfid);

   136

	for (j = 0; j < sizeof(c->md5sum); j++) {

   137

		NetworkSend_uint8 (p, c->md5sum[j]);

   138

	}

   139

}

   140

   141

/**

   142

 * Deserializes the GRFIdentifier (GRF ID and MD5 checksum) from the packet

   143

 * @param cs the client state (for closing connect on out-of-bounds reading etc)

   144

 * @param p  the packet to read the data from

   145

 * @param c  the configuration to write the GRF ID and MD5 checksum to

   146

 */

   147

void NetworkRecv_GRFIdentifier(NetworkClientState *cs, Packet *p, GRFConfig *c)

   148

{

   149

	uint j;

   150

	c->grfid = NetworkRecv_uint32(cs, p);

   151

	for (j = 0; j < sizeof(c->md5sum); j++) {

   152

		c->md5sum[j] = NetworkRecv_uint8(cs, p);

   153

	}

   154

}

   155

   156

   157

/**

   158

 * Serializes the NetworkGameInfo struct to the packet

   159

 * @param p    the packet to write the data to

   160

 * @param info the NetworkGameInfo struct to serialize

   161

 */

   162

void NetworkSend_NetworkGameInfo(Packet *p, const NetworkGameInfo *info)

   163

{

   164

	NetworkSend_uint8 (p, NETWORK_GAME_INFO_VERSION);

   165

   166

	/*

   167

	 *              Please observe the order.

   168

	 * The parts must be read in the same order as they are sent!

   169

	 */

   170

   171

   172

	/* NETWORK_GAME_INFO_VERSION = 4 */

   173

	{

   174

		/* Only send the GRF Identification (GRF_ID and MD5 checksum) of

   175

		 * the GRFs that are needed, i.e. the ones that the server has

   176

		 * selected in the NewGRF GUI and not the ones that are used due

   177

		 * to the fact that they are in [newgrf-static] in openttd.cfg */

   178

		const GRFConfig *c;

   179

		uint count = 0;

   180

   181

		/* Count number of GRFs to send information about */

   182

		for (c = info->grfconfig; c != NULL; c = c->next) {

   183

			if (!HASBIT(c->flags, GCF_STATIC)) count++;

   184

		}

   185

		NetworkSend_uint8 (p, count); // Send number of GRFs

   186

   187

		/* Send actual GRF Identifications */

   188

		for (c = info->grfconfig; c != NULL; c = c->next) {

   189

			if (!HASBIT(c->flags, GCF_STATIC)) NetworkSend_GRFIdentifier(p, c);

   190

		}

   191

	}

   192

   193

	/* NETWORK_GAME_INFO_VERSION = 3 */

   194

	NetworkSend_uint32(p, info->game_date);

   195

	NetworkSend_uint32(p, info->start_date);

   196

   197

	/* NETWORK_GAME_INFO_VERSION = 2 */

   198

	NetworkSend_uint8 (p, info->companies_max);

   199

	NetworkSend_uint8 (p, info->companies_on);

   200

	NetworkSend_uint8 (p, info->spectators_max);

   201

   202

	/* NETWORK_GAME_INFO_VERSION = 1 */

   203

	NetworkSend_string(p, info->server_name);

   204

	NetworkSend_string(p, info->server_revision);

   205

	NetworkSend_uint8 (p, info->server_lang);

   206

	NetworkSend_uint8 (p, info->use_password);

   207

	NetworkSend_uint8 (p, info->clients_max);

   208

	NetworkSend_uint8 (p, info->clients_on);

   209

	NetworkSend_uint8 (p, info->spectators_on);

   210

	NetworkSend_string(p, info->map_name);

   211

	NetworkSend_uint16(p, info->map_width);

   212

	NetworkSend_uint16(p, info->map_height);

   213

	NetworkSend_uint8 (p, info->map_set);

   214

	NetworkSend_uint8 (p, info->dedicated);

   215

}

   216

   217

/**

   218

 * Deserializes the NetworkGameInfo struct from the packet

   219

 * @param cs   the client state (for closing connect on out-of-bounds reading etc)

   220

 * @param p    the packet to read the data from

   221

 * @param info the NetworkGameInfo to deserialize into

   222

 */

   223

void NetworkRecv_NetworkGameInfo(NetworkClientState *cs, Packet *p, NetworkGameInfo *info)

   224

{

   225

	info->game_info_version = NetworkRecv_uint8(cs, p);

   226

   227

	/*

   228

	 *              Please observe the order.

   229

	 * The parts must be read in the same order as they are sent!

   230

	 */

   231

   232

	switch (info->game_info_version) {

   233

		case 4: {

   234

			GRFConfig *c, **dst = &info->grfconfig;

   235

			uint i;

   236

			uint num_grfs = NetworkRecv_uint8(cs, p);

   237

   238

			for (i = 0; i < num_grfs; i++) {

   239

				c = calloc(1, sizeof(*c));

   240

				NetworkRecv_GRFIdentifier(cs, p, c);

   241

				HandleIncomingNetworkGameInfoGRFConfig(c);

   242

   243

				/* Append GRFConfig to the list */

   244

				*dst = c;

   245

				dst = &c->next;

   246

			}

   247

		} /* Fallthrough */

   248

		case 3:

   249

			info->game_date      = NetworkRecv_uint32(cs, p);

   250

			info->start_date     = NetworkRecv_uint32(cs, p);

   251

			/* Fallthrough */

   252

		case 2:

   253

			info->companies_max  = NetworkRecv_uint8 (cs, p);

   254

			info->companies_on   = NetworkRecv_uint8 (cs, p);

   255

			info->spectators_max = NetworkRecv_uint8 (cs, p);

   256

			/* Fallthrough */

   257

		case 1:

   258

			NetworkRecv_string(cs, p, info->server_name,     sizeof(info->server_name));

   259

			NetworkRecv_string(cs, p, info->server_revision, sizeof(info->server_revision));

   260

			info->server_lang    = NetworkRecv_uint8 (cs, p);

   261

			info->use_password   = NetworkRecv_uint8 (cs, p);

   262

			info->clients_max    = NetworkRecv_uint8 (cs, p);

   263

			info->clients_on     = NetworkRecv_uint8 (cs, p);

   264

			info->spectators_on  = NetworkRecv_uint8 (cs, p);

   265

			if (info->game_info_version < 3) { // 16 bits dates got scrapped and are read earlier

   266

				info->game_date    = NetworkRecv_uint16(cs, p) + DAYS_TILL_ORIGINAL_BASE_YEAR;

   267

				info->start_date   = NetworkRecv_uint16(cs, p) + DAYS_TILL_ORIGINAL_BASE_YEAR;

   268

			}

   269

			NetworkRecv_string(cs, p, info->map_name, sizeof(info->map_name));

   270

			info->map_width      = NetworkRecv_uint16(cs, p);

   271

			info->map_height     = NetworkRecv_uint16(cs, p);

   272

			info->map_set        = NetworkRecv_uint8 (cs, p);

   273

			info->dedicated      = NetworkRecv_uint8 (cs, p);

   274

	}

   275

}

   276

   277

#endif /* ENABLE_NETWORK */