1

/* $Id$ */

     2

     3

#ifdef ENABLE_NETWORK

     4

     5

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

     6

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

     7

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

     8

#include "../../helpers.hpp"

     9

#include "packet.h"

    10

#include "udp.h"

    11

    12

/**

    13

 * @file core/udp.cpp Basic functions to receive and send UDP packets.

    14

 */

    15

    16

/**

    17

 * Start listening on the given host and port.

    18

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

    19

 * @param port      the port to listen on

    20

 * @param broadcast whether to allow broadcast sending/receiving

    21

 * @return true if the listening succeeded

    22

 */

    23

bool NetworkUDPSocketHandler::Listen(const uint32 host, const uint16 port, const bool broadcast)

    24

{

    25

	struct sockaddr_in sin;

    26

    27

	/* Make sure socket is closed */

    28

	this->Close();

    29

    30

	this->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

    31

	if (!this->IsConnected()) {

    32

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

    33

		return false;

    34

	}

    35

    36

	/* set nonblocking mode for socket */

    37

	{

    38

		unsigned long blocking = 1;

    39

#ifndef BEOS_NET_SERVER

    40

		ioctlsocket(this->sock, FIONBIO, &blocking);

    41

#else

    42

		setsockopt(this->sock, SOL_SOCKET, SO_NONBLOCK, &blocking, NULL);

    43

#endif

    44

	}

    45

    46

	sin.sin_family = AF_INET;

    47

	/* Listen on all IPs */

    48

	sin.sin_addr.s_addr = host;

    49

	sin.sin_port = htons(port);

    50

    51

	if (bind(this->sock, (struct sockaddr*)&sin, sizeof(sin)) != 0) {

    52

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

    53

		return false;

    54

	}

    55

    56

	if (broadcast) {

    57

		/* Enable broadcast */

    58

		unsigned long val = 1;

    59

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

    60

		setsockopt(this->sock, SOL_SOCKET, SO_BROADCAST, (char *) &val , sizeof(val));

    61

#endif

    62

	}

    63

    64

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

    65

    66

	return true;

    67

}

    68

    69

/**

    70

 * Close the given UDP socket

    71

 */

    72

void NetworkUDPSocketHandler::Close()

    73

{

    74

	if (!this->IsConnected()) return;

    75

    76

	closesocket(this->sock);

    77

	this->sock = INVALID_SOCKET;

    78

}

    79

    80

NetworkRecvStatus NetworkUDPSocketHandler::CloseConnection()

    81

{

    82

	this->has_quit = true;

    83

	return NETWORK_RECV_STATUS_OKAY;

    84

}

    85

    86

/**

    87

 * Send a packet over UDP

    88

 * @param p    the packet to send

    89

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

    90

 */

    91

void NetworkUDPSocketHandler::SendPacket(Packet *p, const struct sockaddr_in *recv)

    92

{

    93

	int res;

    94

    95

	NetworkSend_FillPacketSize(p);

    96

    97

	/* Send the buffer */

    98

	res = sendto(this->sock, (const char*)p->buffer, p->size, 0, (struct sockaddr *)recv, sizeof(*recv));

    99

   100

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

   101

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

   102

}

   103

   104

/**

   105

 * Receive a packet at UDP level

   106

 */

   107

void NetworkUDPSocketHandler::ReceivePackets()

   108

{

   109

	struct sockaddr_in client_addr;

   110

	socklen_t client_len;

   111

	int nbytes;

   112

	Packet p;

   113

	int packet_len;

   114

   115

	if (!this->IsConnected()) return;

   116

   117

	packet_len = sizeof(p.buffer);

   118

	client_len = sizeof(client_addr);

   119

   120

	/* Try to receive anything */

   121

	nbytes = recvfrom(this->sock, (char*)p.buffer, packet_len, 0, (struct sockaddr *)&client_addr, &client_len);

   122

   123

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

   124

	if (nbytes > 2) {

   125

		NetworkRecv_ReadPacketSize(&p);

   126

   127

		/* If the size does not match the packet must be corrupted.

   128

		 * Otherwise it will be marked as corrupted later on. */

   129

		if (nbytes != p.size) {

   130

			DEBUG(net, 1, "received a packet with mismatching size from %s:%d",

   131

					inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));

   132

   133

			return;

   134

		}

   135

   136

		/* Put the position on the right place */

   137

		p.pos = 2;

   138

		p.next = NULL;

   139

   140

		/* Handle the packet */

   141

		this->HandleUDPPacket(&p, &client_addr);

   142

	}

   143

}

   144

   145

   146

/**

   147

 * Serializes the NetworkGameInfo struct to the packet

   148

 * @param p    the packet to write the data to

   149

 * @param info the NetworkGameInfo struct to serialize

   150

 */

   151

void NetworkUDPSocketHandler::Send_NetworkGameInfo(Packet *p, const NetworkGameInfo *info)

   152

{

   153

	NetworkSend_uint8 (p, NETWORK_GAME_INFO_VERSION);

   154

   155

	/*

   156

	 *              Please observe the order.

   157

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

   158

	 */

   159

   160

	/* Update the documentation in udp.h on changes

   161

	 * to the NetworkGameInfo wire-protocol! */

   162

   163

	/* NETWORK_GAME_INFO_VERSION = 4 */

   164

	{

   165

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

   166

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

   167

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

   168

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

   169

		const GRFConfig *c;

   170

		uint count = 0;

   171

   172

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

   173

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

   174

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

   175

		}

   176

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

   177

   178

		/* Send actual GRF Identifications */

   179

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

   180

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

   181

		}

   182

	}

   183

   184

	/* NETWORK_GAME_INFO_VERSION = 3 */

   185

	NetworkSend_uint32(p, info->game_date);

   186

	NetworkSend_uint32(p, info->start_date);

   187

   188

	/* NETWORK_GAME_INFO_VERSION = 2 */

   189

	NetworkSend_uint8 (p, info->companies_max);

   190

	NetworkSend_uint8 (p, info->companies_on);

   191

	NetworkSend_uint8 (p, info->spectators_max);

   192

   193

	/* NETWORK_GAME_INFO_VERSION = 1 */

   194

	NetworkSend_string(p, info->server_name);

   195

	NetworkSend_string(p, info->server_revision);

   196

	NetworkSend_uint8 (p, info->server_lang);

   197

	NetworkSend_uint8 (p, info->use_password);

   198

	NetworkSend_uint8 (p, info->clients_max);

   199

	NetworkSend_uint8 (p, info->clients_on);

   200

	NetworkSend_uint8 (p, info->spectators_on);

   201

	NetworkSend_string(p, info->map_name);

   202

	NetworkSend_uint16(p, info->map_width);

   203

	NetworkSend_uint16(p, info->map_height);

   204

	NetworkSend_uint8 (p, info->map_set);

   205

	NetworkSend_uint8 (p, info->dedicated);

   206

}

   207

   208

/**

   209

 * Deserializes the NetworkGameInfo struct from the packet

   210

 * @param p    the packet to read the data from

   211

 * @param info the NetworkGameInfo to deserialize into

   212

 */

   213

void NetworkUDPSocketHandler::Recv_NetworkGameInfo(Packet *p, NetworkGameInfo *info)

   214

{

   215

	static const Date MAX_DATE = ConvertYMDToDate(MAX_YEAR, 11, 31); // December is month 11

   216

   217

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

   218

   219

	/*

   220

	 *              Please observe the order.

   221

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

   222

	 */

   223

   224

	/* Update the documentation in udp.h on changes

   225

	 * to the NetworkGameInfo wire-protocol! */

   226

   227

	switch (info->game_info_version) {

   228

		case 4: {

   229

			GRFConfig **dst = &info->grfconfig;

   230

			uint i;

   231

			uint num_grfs = NetworkRecv_uint8(this, p);

   232

   233

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

   234

				GRFConfig *c = CallocT<GRFConfig>(1);

   235

				this->Recv_GRFIdentifier(p, c);

   236

				this->HandleIncomingNetworkGameInfoGRFConfig(c);

   237

   238

				/* Append GRFConfig to the list */

   239

				*dst = c;

   240

				dst = &c->next;

   241

			}

   242

		} /* Fallthrough */

   243

		case 3:

   244

			info->game_date      = clamp(NetworkRecv_uint32(this, p), 0, MAX_DATE);

   245

			info->start_date     = clamp(NetworkRecv_uint32(this, p), 0, MAX_DATE);

   246

			/* Fallthrough */

   247

		case 2:

   248

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

   249

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

   250

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

   251

			/* Fallthrough */

   252

		case 1:

   253

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

   254

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

   255

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

   256

			info->use_password   = (NetworkRecv_uint8 (this, p) != 0);

   257

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

   258

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

   259

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

   260

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

   261

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

   262

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

   263

			}

   264

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

   265

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

   266

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

   267

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

   268

			info->dedicated      = (NetworkRecv_uint8(this, p) != 0);

   269

   270

			if (info->server_lang >= NETWORK_NUM_LANGUAGES)  info->server_lang = 0;

   271

			if (info->map_set     >= NETWORK_NUM_LANDSCAPES) info->map_set     = 0;

   272

	}

   273

}

   274

   275

/**

   276

 * Defines a simple (switch) case for each network packet

   277

 * @param type the packet type to create the case for

   278

 */

   279

#define UDP_COMMAND(type) case type: this->NetworkPacketReceive_ ## type ## _command(p, client_addr); break;

   280

   281

/**

   282

 * Handle an incoming packets by sending it to the correct function.

   283

 * @param p the received packet

   284

 * @param client_addr the sender of the packet

   285

 */

   286

void NetworkUDPSocketHandler::HandleUDPPacket(Packet *p, const struct sockaddr_in *client_addr)

   287

{

   288

	PacketUDPType type;

   289

   290

	/* New packet == new client, which has not quit yet */

   291

	this->has_quit = false;

   292

   293

	type = (PacketUDPType)NetworkRecv_uint8(this, p);

   294

   295

	switch (this->HasClientQuit() ? PACKET_UDP_END : type) {

   296

		UDP_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER);

   297

		UDP_COMMAND(PACKET_UDP_SERVER_RESPONSE);

   298

		UDP_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO);

   299

		UDP_COMMAND(PACKET_UDP_SERVER_DETAIL_INFO);

   300

		UDP_COMMAND(PACKET_UDP_SERVER_REGISTER);

   301

		UDP_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER);

   302

		UDP_COMMAND(PACKET_UDP_CLIENT_GET_LIST);

   303

		UDP_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST);

   304

		UDP_COMMAND(PACKET_UDP_SERVER_UNREGISTER);

   305

		UDP_COMMAND(PACKET_UDP_CLIENT_GET_NEWGRFS);

   306

		UDP_COMMAND(PACKET_UDP_SERVER_NEWGRFS);

   307

   308

		default:

   309

			if (this->HasClientQuit()) {

   310

				DEBUG(net, 0, "[udp] received invalid packet type %d from %s:%d", type,  inet_ntoa(client_addr->sin_addr), ntohs(client_addr->sin_port));

   311

			} else {

   312

				DEBUG(net, 0, "[udp] received illegal packet from %s:%d", inet_ntoa(client_addr->sin_addr), ntohs(client_addr->sin_port));

   313

			}

   314

			break;

   315

	}

   316

}

   317

   318

/**

   319

 * Create stub implementations for all receive commands that only

   320

 * show a warning that the given command is not available for the

   321

 * socket where the packet came from.

   322

 * @param type the packet type to create the stub for

   323

 */

   324

#define DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(type) \

   325

void NetworkUDPSocketHandler::NetworkPacketReceive_## type ##_command(\

   326

		Packet *p, const struct sockaddr_in *client_addr) { \

   327

	DEBUG(net, 0, "[udp] received packet type %d on wrong port from %s:%d", \

   328

			type, inet_ntoa(client_addr->sin_addr), ntohs(client_addr->sin_port)); \

   329

}

   330

   331

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER);

   332

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE);

   333

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO);

   334

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_DETAIL_INFO);

   335

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_REGISTER);

   336

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER);

   337

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_GET_LIST);

   338

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST);

   339

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_UNREGISTER);

   340

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_GET_NEWGRFS);

   341

DEFINE_UNAVAILABLE_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_NEWGRFS);

   342

   343

#endif /* ENABLE_NETWORK */