1

#include "stdafx.h"

     2

#include "network_data.h"

     3

     4

#ifdef ENABLE_NETWORK

     5

     6

#include "network_gamelist.h"

     7

#include "network_udp.h"

     8

     9

extern void UpdateNetworkGameWindow(bool unselect);

    10

extern void NetworkPopulateCompanyInfo(void);

    11

    12

//

    13

// This file handles all the LAN-stuff

    14

// Stuff like:

    15

//   - UDP search over the network

    16

//

    17

    18

typedef enum {

    19

	PACKET_UDP_CLIENT_FIND_SERVER,

    20

	PACKET_UDP_SERVER_RESPONSE,

    21

	PACKET_UDP_CLIENT_DETAIL_INFO,

    22

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

    23

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

    24

	PACKET_UDP_MASTER_ACK_REGISTER, // Packet indicating registration has succedeed

    25

	PACKET_UDP_CLIENT_GET_LIST, // Request for serverlist from master server

    26

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

    27

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

    28

	PACKET_UDP_END

    29

} PacketUDPType;

    30

    31

enum {

    32

	ADVERTISE_NORMAL_INTERVAL = 450,	// interval between advertising in days

    33

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

    34

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

    35

};

    36

    37

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

    38

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

    39

    40

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)

    41

{

    42

	Packet *packet;

    43

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

    44

	if (!_network_udp_server)

    45

		return;

    46

    47

	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);

    48

    49

	// Update some game_info

    50

	_network_game_info.game_date = _date;

    51

	_network_game_info.map_set = _opt.landscape;

    52

    53

	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);

    54

	NetworkSend_string(packet, _network_game_info.server_name);

    55

	NetworkSend_string(packet, _network_game_info.server_revision);

    56

	NetworkSend_uint8 (packet, _network_game_info.server_lang);

    57

	NetworkSend_uint8 (packet, _network_game_info.use_password);

    58

	NetworkSend_uint8 (packet, _network_game_info.clients_max);

    59

	NetworkSend_uint8 (packet, _network_game_info.clients_on);

    60

	NetworkSend_uint8 (packet, _network_game_info.spectators_on);

    61

	NetworkSend_uint16(packet, _network_game_info.game_date);

    62

	NetworkSend_uint16(packet, _network_game_info.start_date);

    63

	NetworkSend_string(packet, _network_game_info.map_name);

    64

	NetworkSend_uint16(packet, _network_game_info.map_width);

    65

	NetworkSend_uint16(packet, _network_game_info.map_height);

    66

	NetworkSend_uint8 (packet, _network_game_info.map_set);

    67

	NetworkSend_uint8 (packet, _network_game_info.dedicated);

    68

    69

	// Let the client know that we are here

    70

	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);

    71

    72

	free(packet);

    73

    74

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

    75

}

    76

    77

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)

    78

{

    79

	NetworkGameList *item;

    80

	byte game_info_version;

    81

    82

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

    83

	if (_network_udp_server)

    84

		return;

    85

    86

	game_info_version = NetworkRecv_uint8(p);

    87

    88

    89

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

    90

    91

	// Find next item

    92

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

    93

    94

	if (game_info_version == 1) {

    95

		NetworkRecv_string(p, item->info.server_name, sizeof(item->info.server_name));

    96

		NetworkRecv_string(p, item->info.server_revision, sizeof(item->info.server_revision));

    97

		item->info.server_lang = NetworkRecv_uint8(p);

    98

		item->info.use_password = NetworkRecv_uint8(p);

    99

		item->info.clients_max = NetworkRecv_uint8(p);

   100

		item->info.clients_on = NetworkRecv_uint8(p);

   101

		item->info.spectators_on = NetworkRecv_uint8(p);

   102

		item->info.game_date = NetworkRecv_uint16(p);

   103

		item->info.start_date = NetworkRecv_uint16(p);

   104

		NetworkRecv_string(p, item->info.map_name, sizeof(item->info.map_name));

   105

		item->info.map_width = NetworkRecv_uint16(p);

   106

		item->info.map_height = NetworkRecv_uint16(p);

   107

		item->info.map_set = NetworkRecv_uint8(p);

   108

		item->info.dedicated = NetworkRecv_uint8(p);

   109

   110

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

   111

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

   112

	}

   113

   114

	item->online = true;

   115

   116

	UpdateNetworkGameWindow(false);

   117

}

   118

   119

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)

   120

{

   121

	NetworkClientState *cs;

   122

	NetworkClientInfo *ci;

   123

	Packet *packet;

   124

	Player *player;

   125

	byte active = 0;

   126

	byte current = 0;

   127

	int i;

   128

   129

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

   130

	if (!_network_udp_server)

   131

		return;

   132

   133

	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);

   134

   135

	FOR_ALL_PLAYERS(player) {

   136

		if (player->is_active)

   137

			active++;

   138

	}

   139

   140

	/* Send the amount of active companies */

   141

	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);

   142

	NetworkSend_uint8 (packet, active);

   143

   144

	/* Fetch the latest version of everything */

   145

	NetworkPopulateCompanyInfo();

   146

   147

	/* Go through all the players */

   148

	FOR_ALL_PLAYERS(player) {

   149

		/* Skip non-active players */

   150

		if (!player->is_active)

   151

			continue;

   152

   153

		current++;

   154

   155

		/* Send the information */

   156

		NetworkSend_uint8 (packet, current);

   157

   158

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

   159

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

   160

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

   161

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

   162

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

   163

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

   164

   165

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

   166

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

   167

   168

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

   169

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

   170

   171

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

   172

		FOR_ALL_CLIENTS(cs) {

   173

			ci = DEREF_CLIENT_INFO(cs);

   174

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

   175

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

   176

				NetworkSend_uint8(packet, 1);

   177

				NetworkSend_string(packet, ci->client_name);

   178

				NetworkSend_string(packet, ci->unique_id);

   179

				NetworkSend_uint16(packet, ci->join_date);

   180

			}

   181

		}

   182

		/* Also check for the server itself */

   183

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   184

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

   185

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

   186

			NetworkSend_uint8(packet, 1);

   187

			NetworkSend_string(packet, ci->client_name);

   188

			NetworkSend_string(packet, ci->unique_id);

   189

			NetworkSend_uint16(packet, ci->join_date);

   190

		}

   191

   192

		/* Indicates end of client list */

   193

		NetworkSend_uint8(packet, 0);

   194

	}

   195

   196

	/* And check if we have any spectators */

   197

	FOR_ALL_CLIENTS(cs) {

   198

		ci = DEREF_CLIENT_INFO(cs);

   199

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

   200

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

   201

			NetworkSend_uint8(packet, 1);

   202

			NetworkSend_string(packet, ci->client_name);

   203

			NetworkSend_string(packet, ci->unique_id);

   204

			NetworkSend_uint16(packet, ci->join_date);

   205

		}

   206

	}

   207

	/* Also check for the server itself */

   208

	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   209

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

   210

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

   211

		NetworkSend_uint8(packet, 1);

   212

		NetworkSend_string(packet, ci->client_name);

   213

		NetworkSend_string(packet, ci->unique_id);

   214

		NetworkSend_uint16(packet, ci->join_date);

   215

	}

   216

   217

	/* Indicates end of client list */

   218

	NetworkSend_uint8(packet, 0);

   219

   220

	NetworkSendUDP_Packet(_udp_server_socket, packet, client_addr);

   221

   222

	free(packet);

   223

}

   224

   225

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST) {

   226

	int i;

   227

	struct in_addr ip;

   228

	uint16 port;

   229

	uint8 ver;

   230

   231

	/* packet begins with the protocol version (uint8)

   232

	 * then an uint16 which indicates how many

   233

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

   234

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

   235

	 */

   236

   237

	ver = NetworkRecv_uint8(p);

   238

   239

	if (ver == 1) {

   240

		for (i = NetworkRecv_uint16(p); i != 0 ; i--) {

   241

			ip.s_addr = TO_LE32(NetworkRecv_uint32(p));

   242

			port = NetworkRecv_uint16(p);

   243

			NetworkUDPQueryServer(inet_ntoa(ip), port);

   244

		}

   245

	}

   246

}

   247

   248

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER) {

   249

	_network_advertise_retries = 0;

   250

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

   251

   252

	if (!_network_advertise)

   253

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

   254

		NetworkUDPRemoveAdvertise();

   255

}

   256

   257

   258

// The layout for the receive-functions by UDP

   259

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

   260

   261

static NetworkUDPPacket* const _network_udp_packet[] = {

   262

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),

   263

	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),

   264

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),

   265

	NULL,

   266

	NULL,

   267

	RECEIVE_COMMAND(PACKET_UDP_MASTER_ACK_REGISTER),

   268

	NULL,

   269

	RECEIVE_COMMAND(PACKET_UDP_MASTER_RESPONSE_LIST),

   270

	NULL

   271

};

   272

   273

   274

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

   275

assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);

   276

   277

   278

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

   279

{

   280

	byte type;

   281

   282

	type = NetworkRecv_uint8(p);

   283

   284

	if (type < PACKET_UDP_END && _network_udp_packet[type] != NULL) {

   285

		_network_udp_packet[type](p, client_addr);

   286

	}	else {

   287

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

   288

	}

   289

}

   290

   291

   292

// Send a packet over UDP

   293

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

   294

{

   295

	int res;

   296

   297

	// Put the length in the buffer

   298

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

   299

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

   300

   301

	// Send the buffer

   302

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

   303

   304

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

   305

	if (res == -1) {

   306

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

   307

	}

   308

}

   309

   310

// Start UDP listener

   311

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

   312

{

   313

	struct sockaddr_in sin;

   314

   315

	// Make sure socket is closed

   316

	closesocket(*udp);

   317

   318

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

   319

	if (*udp == INVALID_SOCKET) {

   320

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

   321

		return false;

   322

	}

   323

   324

	// set nonblocking mode for socket

   325

	{

   326

		unsigned long blocking = 1;

   327

		ioctlsocket(*udp, FIONBIO, &blocking);

   328

	}

   329

   330

	sin.sin_family = AF_INET;

   331

	// Listen on all IPs

   332

	sin.sin_addr.s_addr = host;

   333

	sin.sin_port = htons(port);

   334

   335

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

   336

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

   337

		return false;

   338

	}

   339

   340

	if (broadcast) {

   341

		/* Enable broadcast */

   342

		unsigned long val = 1;

   343

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

   344

	}

   345

   346

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

   347

   348

	return true;

   349

}

   350

   351

// Close UDP connection

   352

void NetworkUDPClose(void)

   353

{

   354

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

   355

   356

	if (_network_udp_server) {

   357

		closesocket(_udp_server_socket);

   358

		_udp_server_socket = INVALID_SOCKET;

   359

   360

		closesocket(_udp_master_socket);

   361

		_udp_master_socket = INVALID_SOCKET;

   362

   363

		_network_udp_server = false;

   364

		_network_udp_broadcast = 0;

   365

	} else {

   366

		closesocket(_udp_client_socket);

   367

		_udp_client_socket = INVALID_SOCKET;

   368

		_network_udp_broadcast = 0;

   369

	}

   370

}

   371

   372

// Receive something on UDP level

   373

void NetworkUDPReceive(SOCKET udp)

   374

{

   375

	struct sockaddr_in client_addr;

   376

#ifndef __MORPHOS__

   377

	int client_len;

   378

#else

   379

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

   380

#endif

   381

	int nbytes;

   382

	static Packet *p = NULL;

   383

	int packet_len;

   384

   385

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

   386

	if (p == NULL)

   387

		p = malloc(sizeof(Packet));

   388

   389

	packet_len = sizeof(p->buffer);

   390

	client_len = sizeof(client_addr);

   391

   392

	// Try to receive anything

   393

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

   394

   395

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

   396

	if (nbytes > 0) {

   397

		// Get the size of the buffer

   398

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

   399

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

   400

		// Put the position on the right place

   401

		p->pos = 2;

   402

		p->next = NULL;

   403

   404

		// Handle the packet

   405

		NetworkHandleUDPPacket(p, &client_addr);

   406

   407

		// Free the packet

   408

		free(p);

   409

		p = NULL;

   410

	}

   411

}

   412

   413

// Broadcast to all ips

   414

void NetworkUDPBroadCast(SOCKET udp)

   415

{

   416

	int i;

   417

	struct sockaddr_in out_addr;

   418

	byte *bcptr;

   419

	uint32 bcaddr;

   420

	Packet *p;

   421

   422

	// Init the packet

   423

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);

   424

   425

	// Go through all the ips on this pc

   426

	i = 0;

   427

	while (_network_ip_list[i] != 0) {

   428

		bcaddr = _network_ip_list[i];

   429

		bcptr = (byte *)&bcaddr;

   430

		// Make the address a broadcast address

   431

		bcptr[3] = 255;

   432

   433

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

   434

   435

		out_addr.sin_family = AF_INET;

   436

		out_addr.sin_port = htons(_network_server_port);

   437

		out_addr.sin_addr.s_addr = bcaddr;

   438

   439

		NetworkSendUDP_Packet(udp, p, &out_addr);

   440

   441

		i++;

   442

	}

   443

}

   444

   445

   446

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

   447

void NetworkUDPQueryMasterServer(void)

   448

{

   449

	struct sockaddr_in out_addr;

   450

	Packet *p;

   451

   452

	if (_udp_client_socket == INVALID_SOCKET)

   453

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

   454

			return;

   455

   456

	p = NetworkSend_Init(PACKET_UDP_CLIENT_GET_LIST);

   457

   458

	out_addr.sin_family = AF_INET;

   459

	out_addr.sin_port = htons(NETWORK_MASTER_SERVER_PORT);

   460

	out_addr.sin_addr.s_addr = NetworkResolveHost(NETWORK_MASTER_SERVER_HOST);

   461

   462

	// packet only contains protocol version

   463

	NetworkSend_uint8(p, NETWORK_MASTER_SERVER_VERSION);

   464

   465

	NetworkSendUDP_Packet(_udp_client_socket, p, &out_addr);

   466

   467

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

   468

   469

	free(p);

   470

}

   471

   472

// Find all servers

   473

void NetworkUDPSearchGame(void)

   474

{

   475

	// We are still searching..

   476

	if (_network_udp_broadcast > 0)

   477

		return;

   478

   479

	// No UDP-socket yet..

   480

	if (_udp_client_socket == INVALID_SOCKET)

   481

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

   482

			return;

   483

   484

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

   485

   486

	NetworkUDPBroadCast(_udp_client_socket);

   487

	_network_udp_broadcast = 300; // Stay searching for 300 ticks

   488

}

   489

   490

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

   491

{

   492

	struct sockaddr_in out_addr;

   493

	Packet *p;

   494

	NetworkGameList *item;

   495

	char hostname[NETWORK_HOSTNAME_LENGTH];

   496

   497

	// No UDP-socket yet..

   498

	if (_udp_client_socket == INVALID_SOCKET)

   499

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

   500

			return NULL;

   501

   502

	ttd_strlcpy(hostname, host, sizeof(hostname));

   503

   504

	out_addr.sin_family = AF_INET;

   505

	out_addr.sin_port = htons(port);

   506

	out_addr.sin_addr.s_addr = NetworkResolveHost(host);

   507

   508

	// Clear item in gamelist

   509

	item = NetworkGameListAddItem(inet_addr(inet_ntoa(out_addr.sin_addr)), ntohs(out_addr.sin_port));