1

#include "stdafx.h"

     2

#include "network_data.h"

     3

     4

#ifdef ENABLE_NETWORK

     5

     6

#include "map.h"

     7

#include "network_gamelist.h"

     8

#include "network_udp.h"

     9

    10

extern void UpdateNetworkGameWindow(bool unselect);

    11

extern void NetworkPopulateCompanyInfo(void);

    12

    13

//

    14

// This file handles all the LAN-stuff

    15

// Stuff like:

    16

//   - UDP search over the network

    17

//

    18

    19

typedef enum {

    20

	PACKET_UDP_CLIENT_FIND_SERVER,

    21

	PACKET_UDP_SERVER_RESPONSE,

    22

	PACKET_UDP_CLIENT_DETAIL_INFO,

    23

	PACKET_UDP_SERVER_DETAIL_INFO, // Is not used in OpenTTD itself, only for external querying

    24

	PACKET_UDP_SERVER_REGISTER, // Packet to register itself to the master server

    25

	PACKET_UDP_MASTER_ACK_REGISTER, // Packet indicating registration has succedeed

    26

	PACKET_UDP_CLIENT_GET_LIST, // Request for serverlist from master server

    27

	PACKET_UDP_MASTER_RESPONSE_LIST, // Response from master server with server ip's + port's

    28

	PACKET_UDP_SERVER_UNREGISTER, // Request to be removed from the server-list

    29

	PACKET_UDP_END

    30

} PacketUDPType;

    31

    32

enum {

    33

	ADVERTISE_NORMAL_INTERVAL = 450,	// interval between advertising in days

    34

	ADVERTISE_RETRY_INTERVAL = 5,			// readvertise when no response after this amount of days

    35

	ADVERTISE_RETRY_TIMES = 3					// give up readvertising after this much failed retries

    36

};

    37

    38

#define DEF_UDP_RECEIVE_COMMAND(type) void NetworkPacketReceive_ ## type ## _command(Packet *p, struct sockaddr_in *client_addr)

    39

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

    40

    43

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)

    44

{

    45

	Packet *packet;

    46

	// Just a fail-safe.. should never happen

    47

	if (!_network_udp_server)

    48

		return;

    49

    50

	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);

    51

    52

	// Update some game_info

    53

	_network_game_info.game_date = _date;

    54

	_network_game_info.map_width = MapSizeX();

    55

	_network_game_info.map_height = MapSizeY();

    56

	_network_game_info.map_set = _opt.landscape;

    57

    58

	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);

    59

	NetworkSend_string(packet, _network_game_info.server_name);

    60

	NetworkSend_string(packet, _network_game_info.server_revision);

    61

	NetworkSend_uint8 (packet, _network_game_info.server_lang);

    62

	NetworkSend_uint8 (packet, _network_game_info.use_password);

    63

	NetworkSend_uint8 (packet, _network_game_info.clients_max);

    64

	NetworkSend_uint8 (packet, _network_game_info.clients_on);

    65

	NetworkSend_uint8 (packet, _network_game_info.spectators_on);

    66

	NetworkSend_uint16(packet, _network_game_info.game_date);

    67

	NetworkSend_uint16(packet, _network_game_info.start_date);

    68

	NetworkSend_string(packet, _network_game_info.map_name);

    69

	NetworkSend_uint16(packet, _network_game_info.map_width);

    70

	NetworkSend_uint16(packet, _network_game_info.map_height);

    71

	NetworkSend_uint8 (packet, _network_game_info.map_set);

    72

	NetworkSend_uint8 (packet, _network_game_info.dedicated);

    73

    74

	// Let the client know that we are here

    75

	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);

    76

    77

	free(packet);

    78

    79

	DEBUG(net, 2)("[NET][UDP] Queried from %s", inet_ntoa(client_addr->sin_addr));

    80

}

    81

    82

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)

    83

{

    84

	NetworkGameList *item;

    85

	byte game_info_version;

    86

    87

	// Just a fail-safe.. should never happen

    88

	if (_network_udp_server)

    89

		return;

    90

    92

    95

    96

	DEBUG(net, 6)("[NET][UDP] Server response from %s:%d", inet_ntoa(client_addr->sin_addr),ntohs(client_addr->sin_port));

    97

    98

	// Find next item

    99

	item = NetworkGameListAddItem(inet_addr(inet_ntoa(client_addr->sin_addr)), ntohs(client_addr->sin_port));

   100

   101

	if (game_info_version == 1) {

   116

   117

		if (item->info.hostname[0] == '\0')

   118

			snprintf(item->info.hostname, sizeof(item->info.hostname), "%s", inet_ntoa(client_addr->sin_addr));

   119

	}

   120

   121

	item->online = true;

   122

   123

	UpdateNetworkGameWindow(false);

   124

}

   125

   126

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)

   127

{

   128

	NetworkClientState *cs;

   129

	NetworkClientInfo *ci;

   130

	Packet *packet;

   131

	Player *player;

   132

	byte active = 0;

   133

	byte current = 0;

   134

	int i;

   135

   136

	// Just a fail-safe.. should never happen

   137

	if (!_network_udp_server)

   138

		return;

   139

   140

	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);

   141

   142

	FOR_ALL_PLAYERS(player) {

   143

		if (player->is_active)

   144

			active++;

   145

	}

   146

   147

	/* Send the amount of active companies */

   148

	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);

   149

	NetworkSend_uint8 (packet, active);

   150

   151

	/* Fetch the latest version of everything */

   152

	NetworkPopulateCompanyInfo();

   153

   154

	/* Go through all the players */

   155

	FOR_ALL_PLAYERS(player) {

   156

		/* Skip non-active players */

   157

		if (!player->is_active)

   158

			continue;

   159

   160

		current++;

   161

   162

		/* Send the information */

   163

		NetworkSend_uint8 (packet, current);

   164

   165

		NetworkSend_string(packet, _network_player_info[player->index].company_name);

   166

		NetworkSend_uint8 (packet, _network_player_info[player->index].inaugurated_year);

   167

		NetworkSend_uint64(packet, _network_player_info[player->index].company_value);

   168

		NetworkSend_uint64(packet, _network_player_info[player->index].money);

   169

		NetworkSend_uint64(packet, _network_player_info[player->index].income);

   170

		NetworkSend_uint16(packet, _network_player_info[player->index].performance);

   171

   172

		for (i = 0; i < NETWORK_VEHICLE_TYPES; i++)

   173

			NetworkSend_uint16(packet, _network_player_info[player->index].num_vehicle[i]);

   174

   175

		for (i = 0; i < NETWORK_STATION_TYPES; i++)

   176

			NetworkSend_uint16(packet, _network_player_info[player->index].num_station[i]);

   177

   178

		/* Find the clients that are connected to this player */

   179

		FOR_ALL_CLIENTS(cs) {

   180

			ci = DEREF_CLIENT_INFO(cs);

   181

			if ((ci->client_playas - 1) == player->index) {

   182

				/* The uint8 == 1 indicates that a client is following */

   183

				NetworkSend_uint8(packet, 1);

   184

				NetworkSend_string(packet, ci->client_name);

   185

				NetworkSend_string(packet, ci->unique_id);

   186

				NetworkSend_uint16(packet, ci->join_date);

   187

			}

   188

		}

   189

		/* Also check for the server itself */

   190

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   191

		if ((ci->client_playas - 1) == player->index) {

   192

			/* The uint8 == 1 indicates that a client is following */

   193

			NetworkSend_uint8(packet, 1);

   194

			NetworkSend_string(packet, ci->client_name);

   195

			NetworkSend_string(packet, ci->unique_id);

   196

			NetworkSend_uint16(packet, ci->join_date);

   197

		}

   198

   199

		/* Indicates end of client list */

   200

		NetworkSend_uint8(packet, 0);

   201

	}

   202

   203

	/* And check if we have any spectators */

   204

	FOR_ALL_CLIENTS(cs) {

   205

		ci = DEREF_CLIENT_INFO(cs);

   206

		if ((ci->client_playas - 1) > MAX_PLAYERS) {

   207

			/* The uint8 == 1 indicates that a client is following */

   208

			NetworkSend_uint8(packet, 1);

   209

			NetworkSend_string(packet, ci->client_name);

   210

			NetworkSend_string(packet, ci->unique_id);

   211

			NetworkSend_uint16(packet, ci->join_date);

   212

		}

   213

	}

   214

	/* Also check for the server itself */

   215

	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   216

	if ((ci->client_playas - 1) > MAX_PLAYERS) {

   217

		/* The uint8 == 1 indicates that a client is following */

   218

		NetworkSend_uint8(packet, 1);

   219

		NetworkSend_string(packet, ci->client_name);

   220

		NetworkSend_string(packet, ci->unique_id);

   221

		NetworkSend_uint16(packet, ci->join_date);

   222

	}

   223

   224

	/* Indicates end of client list */

   225

	NetworkSend_uint8(packet, 0);

   226

   227

	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);

   228

   229

	free(packet);

   230

}

   231

   232

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST) {

   233

	int i;

   234

	struct in_addr ip;

   235

	uint16 port;

   236

	uint8 ver;

   237

   238

	/* packet begins with the protocol version (uint8)

   239

	 * then an uint16 which indicates how many

   240

	 * ip:port pairs are in this packet, after that

   241

	 * an uint32 (ip) and an uint16 (port) for each pair

   242

	 */

   243

   248

   249

	if (ver == 1) {

   253

			NetworkUDPQueryServer(inet_ntoa(ip), port);

   254

		}

   255

	}

   256

}

   257

   258

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER) {

   259

	_network_advertise_retries = 0;

   260

	DEBUG(net, 2)("[NET][UDP] We are advertised on the master-server!");

   261

   262

	if (!_network_advertise)

   263

		/* We are advertised, but we don't want to! */

   264

		NetworkUDPRemoveAdvertise();

   265

}

   266

   267

   268

// The layout for the receive-functions by UDP

   269

typedef void NetworkUDPPacket(Packet *p, struct sockaddr_in *client_addr);

   270

   271

static NetworkUDPPacket* const _network_udp_packet[] = {

   272

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),

   273

	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),

   274

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),

   275

	NULL,

   276

	NULL,

   277

	RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER),

   278

	NULL,

   279

	RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST),

   280

	NULL

   281

};

   282

   283

   284

// If this fails, check the array above with network_data.h

   285

assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);

   286

   287

   288

void NetworkHandleUDPPacket(Packet *p, struct sockaddr_in *client_addr)

   289

{

   290

	byte type;

   291

   295

   299

		_network_udp_packet[type](p, client_addr);

   300

	}	else {

   301

		DEBUG(net, 0)("[NET][UDP] Received invalid packet type %d", type);

   302

	}

   303

}

   304

   305

   306

// Send a packet over UDP

   307

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

   308

{

   309

	int res;

   310

   311

	// Put the length in the buffer

   312

	p->buffer[0] = p->size & 0xFF;

   313

	p->buffer[1] = p->size >> 8;

   314

   315

	// Send the buffer

   316

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

   317

   318

	// Check for any errors, but ignore it for the rest

   319

	if (res == -1) {

   320

		DEBUG(net, 1)("[NET][UDP] Send error: %i", GET_LAST_ERROR());

   321

	}

   322

}

   323

   324

// Start UDP listener

   325

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

   326

{

   327

	struct sockaddr_in sin;

   328

   329

	// Make sure socket is closed

   330

	closesocket(*udp);

   331

   332

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

   333

	if (*udp == INVALID_SOCKET) {

   334

		DEBUG(net, 1)("[NET][UDP] Failed to start UDP support");

   335

		return false;

   336

	}

   337

   338

	// set nonblocking mode for socket

   339

	{

   340

		unsigned long blocking = 1;

   341

		ioctlsocket(*udp, FIONBIO, &blocking);

   342

	}

   343

   344

	sin.sin_family = AF_INET;

   345

	// Listen on all IPs

   346

	sin.sin_addr.s_addr = host;

   347

	sin.sin_port = htons(port);

   348

   349

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

   350

		DEBUG(net, 1) ("[NET][UDP] error: bind failed on %s:%i", inet_ntoa(*(struct in_addr *)&host), port);

   351

		return false;

   352

	}

   353

   354

	if (broadcast) {

   355

		/* Enable broadcast */

   356

		unsigned long val = 1;

   357

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

   358

	}

   359

   360

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

   361

   362

	return true;

   363

}

   364

   365

// Close UDP connection

   366

void NetworkUDPClose(void)

   367

{

   368

	DEBUG(net, 1) ("[NET][UDP] Closed listeners");

   369

   370

	if (_network_udp_server) {

   371

		closesocket(_udp_server_socket);

   372

		_udp_server_socket = INVALID_SOCKET;

   373

   374

		closesocket(_udp_master_socket);

   375

		_udp_master_socket = INVALID_SOCKET;

   376

   377

		_network_udp_server = false;

   378

		_network_udp_broadcast = 0;

   379

	} else {

   380

		closesocket(_udp_client_socket);

   381

		_udp_client_socket = INVALID_SOCKET;

   382

		_network_udp_broadcast = 0;

   383

	}

   384

}

   385

   386

// Receive something on UDP level

   387

void NetworkUDPReceive(SOCKET udp)

   388

{

   389

	struct sockaddr_in client_addr;

   390

#ifndef __MORPHOS__

   391

	int client_len;

   392

#else

   393

	LONG client_len; // for some reason we need a 'LONG' under MorphOS

   394

#endif

   395

	int nbytes;

   396

	static Packet *p = NULL;

   397

	int packet_len;

   398

   399

	// If p is NULL, malloc him.. this prevents unneeded mallocs

   400

	if (p == NULL)

   401

		p = malloc(sizeof(Packet));

   402

   403

	packet_len = sizeof(p->buffer);

   404

	client_len = sizeof(client_addr);

   405

   406

	// Try to receive anything

   407

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

   408

   409

	// We got some bytes.. just asume we receive the whole packet

   410

	if (nbytes > 0) {

   411

		// Get the size of the buffer

   412

		p->size = (uint16)p->buffer[0];

   413

		p->size += (uint16)p->buffer[1] << 8;

   414

		// Put the position on the right place

   415

		p->pos = 2;

   416

		p->next = NULL;

   417

   418

		// Handle the packet

   419

		NetworkHandleUDPPacket(p, &client_addr);

   420

   421

		// Free the packet

   422

		free(p);

   423

		p = NULL;

   424

	}

   425

}

   426

   427

// Broadcast to all ips

   428

void NetworkUDPBroadCast(SOCKET udp)

   429

{

   430

	int i;

   431

	struct sockaddr_in out_addr;

   432

	byte *bcptr;

   433

	uint32 bcaddr;

   434

	Packet *p;

   435

   436

	// Init the packet

   437

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);

   438

   439

	// Go through all the ips on this pc

   440

	i = 0;

   441

	while (_network_ip_list[i] != 0) {

   442

		bcaddr = _network_ip_list[i];

   443

		bcptr = (byte *)&bcaddr;

   444

		// Make the address a broadcast address

   445

		bcptr[3] = 255;

   446

   447

		DEBUG(net, 6)("[NET][UDP] Broadcasting to %s", inet_ntoa(*(struct in_addr *)&bcaddr));

   448

   449

		out_addr.sin_family = AF_INET;

   450

		out_addr.sin_port = htons(_network_server_port);

   451

		out_addr.sin_addr.s_addr = bcaddr;

   452

   453

		NetworkSendUDP_Packet(udp, p, &out_addr);

   454

   455

		i++;

   456

	}

   457

}

   458

   459

   460

// Request the the server-list from the master server

   461

void NetworkUDPQueryMasterServer(void)

   462

{

   463

	struct sockaddr_in out_addr;

   464

	Packet *p;

   465

   466

	if (_udp_client_socket == INVALID_SOCKET)

   467

		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))

   468

			return;

   469

   470

	p = NetworkSend_Init(PACKET_UDP_CLIENT_GET_LIST);

   471

   472

	out_addr.sin_family = AF_INET;

   473

	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);

   474

	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);

   475

   476

	// packet only contains protocol version

   477

	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);

   478

   479

	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);

   480

   481

	DEBUG(net, 2)("[NET][UDP] Queried Master Server at %s:%d", inet_ntoa(out_addr.sin_addr),ntohs(out_addr.sin_port));

   482

   483

	free(p);

   484

}

   485

   486

// Find all servers

   487

void NetworkUDPSearchGame(void)

   488

{

   489

	// We are still searching..

   490

	if (_network_udp_broadcast > 0)

   491

		return;

   492

   493

	// No UDP-socket yet..

   494

	if (_udp_client_socket == INVALID_SOCKET)

   495

		if (!NetworkUDPListen(&_udp_client_socket, 0, 0, true))

   496

			return;

   497

   498

	DEBUG(net, 0)("[NET][UDP] Searching server");

   499

   500

	NetworkUDPBroadCast(_udp_client_socket);

   501

	_network_udp_broadcast = 300; // Stay searching for 300 ticks

   502

}

   503

   504

NetworkGameList *NetworkUDPQueryServer(const byte* host, unsigned short port)

   505

{

   506

	struct sockaddr_in out_addr;