1

#include "stdafx.h"

     2

#include "network_data.h"

     3

     4

#ifdef ENABLE_NETWORK

     5

     6

#include "network_gamelist.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

    27

	PACKET_UDP_END

    28

} PacketUDPType;

    29

    35

    36

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

    38

    39

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)

    40

{

    41

	Packet *packet;

    42

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

    43

	if (!_network_udp_server)

    44

		return;

    45

    46

	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);

    47

    48

	// Update some game_info

    49

	_network_game_info.game_date = _date;

    50

	_network_game_info.map_set = _opt.landscape;

    51

    52

	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);

    53

	NetworkSend_string(packet, _network_game_info.server_name);

    54

	NetworkSend_string(packet, _network_game_info.server_revision);

    55

	NetworkSend_uint8 (packet, _network_game_info.server_lang);

    56

	NetworkSend_uint8 (packet, _network_game_info.use_password);

    57

	NetworkSend_uint8 (packet, _network_game_info.clients_max);

    58

	NetworkSend_uint8 (packet, _network_game_info.clients_on);

    59

	NetworkSend_uint8 (packet, _network_game_info.spectators_on);

    60

	NetworkSend_uint16(packet, _network_game_info.game_date);

    61

	NetworkSend_uint16(packet, _network_game_info.start_date);

    62

	NetworkSend_string(packet, _network_game_info.map_name);

    63

	NetworkSend_uint16(packet, _network_game_info.map_width);

    64

	NetworkSend_uint16(packet, _network_game_info.map_height);

    65

	NetworkSend_uint8 (packet, _network_game_info.map_set);

    66

	NetworkSend_uint8 (packet, _network_game_info.dedicated);

    67

    68

	// Let the client know that we are here

    70

    71

	free(packet);

    72

    73

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

    74

}

    75

    76

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)

    77

{

    78

	NetworkGameList *item;

    79

	byte game_info_version;

    80

    81

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

    82

	if (_network_udp_server)

    83

		return;

    84

    85

	game_info_version = NetworkRecv_uint8(p);

    86

    90

	// Find next item

    91

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

    92

    93

	if (game_info_version == 1) {

    94

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

    95

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

    96

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

    97

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

    98

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

    99

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

   100

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

   101

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

   102

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

   103

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

   104

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

   105

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

   106

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

   107

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

   108

   109

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

   110

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

   111

	}

   112

   113

	item->online = true;

   114

   115

	UpdateNetworkGameWindow(false);

   116

}

   117

   118

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)

   119

{

   120

	NetworkClientState *cs;

   121

	NetworkClientInfo *ci;

   122

	Packet *packet;

   123

	Player *player;

   124

	byte active = 0;

   125

	byte current = 0;

   126

	int i;

   127

   128

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

   129

	if (!_network_udp_server)

   130

		return;

   131

   132

	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);

   133

   134

	FOR_ALL_PLAYERS(player) {

   135

		if (player->is_active)

   136

			active++;

   137

	}

   138

   139

	/* Send the amount of active companies */

   140

	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);

   141

	NetworkSend_uint8 (packet, active);

   142

   143

	/* Fetch the latest version of everything */

   144

	NetworkPopulateCompanyInfo();

   145

   146

	/* Go through all the players */

   147

	FOR_ALL_PLAYERS(player) {

   148

		/* Skip non-active players */

   149

		if (!player->is_active)

   150

			continue;

   151

   152

		current++;

   153

   154

		/* Send the information */

   155

		NetworkSend_uint8 (packet, current);

   156

   157

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

   158

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

   159

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

   160

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

   161

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

   162

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

   163

   164

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

   165

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

   166

   167

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

   168

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

   169

   170

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

   171

		FOR_ALL_CLIENTS(cs) {

   172

			ci = DEREF_CLIENT_INFO(cs);

   173

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

   174

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

   175

				NetworkSend_uint8(packet, 1);

   176

				NetworkSend_string(packet, ci->client_name);

   177

				NetworkSend_string(packet, ci->unique_id);

   178

				NetworkSend_uint16(packet, ci->join_date);

   179

			}

   180

		}

   181

		/* Also check for the server itself */

   182

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   183

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

   184

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

   185

			NetworkSend_uint8(packet, 1);

   186

			NetworkSend_string(packet, ci->client_name);

   187

			NetworkSend_string(packet, ci->unique_id);

   188

			NetworkSend_uint16(packet, ci->join_date);

   189

		}

   190

   191

		/* Indicates end of client list */

   192

		NetworkSend_uint8(packet, 0);

   193

	}

   194

   195

	/* And check if we have any spectators */

   196

	FOR_ALL_CLIENTS(cs) {

   197

		ci = DEREF_CLIENT_INFO(cs);

   198

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

   199

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

   200

			NetworkSend_uint8(packet, 1);

   201

			NetworkSend_string(packet, ci->client_name);

   202

			NetworkSend_string(packet, ci->unique_id);

   203

			NetworkSend_uint16(packet, ci->join_date);

   204

		}

   205

	}

   206

	/* Also check for the server itself */

   207

	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   208

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

   209

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

   210

		NetworkSend_uint8(packet, 1);

   211

		NetworkSend_string(packet, ci->client_name);

   212

		NetworkSend_string(packet, ci->unique_id);

   213

		NetworkSend_uint16(packet, ci->join_date);

   214

	}

   215

   216

	/* Indicates end of client list */

   217

	NetworkSend_uint8(packet, 0);

   218

   220

   221

	free(packet);

   222

}

   223

   252

   253

// The layout for the receive-functions by UDP

   254

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

   255

   256

static NetworkUDPPacket* const _network_udp_packet[] = {

   257

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),

   258

	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),

   259

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),

   260

	NULL,

   261

	NULL,

   265

};

   266

   268

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

   269

assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);

   270

   271

   272

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

   273

{

   274

	byte type;

   275

   276

	type = NetworkRecv_uint8(p);

   277

   278

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

   279

		_network_udp_packet[type](p, client_addr);

   280

	}	else {

   281

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

   282

	}

   283

}

   284

   285

   286

// Send a packet over UDP

   288

{

   289

	int res;

   290

   291

	// Put the length in the buffer

   292

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

   293

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

   294

   295

	// Send the buffer

   296

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

   297

   298

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

   299

	if (res == -1) {

   300

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

   301

	}

   302

}

   303

   304

// Start UDP listener

   306

{

   307

	struct sockaddr_in sin;

   308

   311

   314

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

   315

		return false;

   316

	}

   317

   318

	// set nonblocking mode for socket

   319

	{

   320

		unsigned long blocking = 1;

   322

	}

   323

   324

	sin.sin_family = AF_INET;

   325

	// Listen on all IPs

   326

	sin.sin_addr.s_addr = host;

   327

	sin.sin_port = htons(port);

   328

   330

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

   331

		return false;

   332

	}

   333

   336

		unsigned long val = 1;

   338

	}

   339

   340

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

   341

   342

	return true;

   343

}

   344

   345

// Close UDP connection

   346

void NetworkUDPClose(void)

   347

{

   349

   350

	if (_network_udp_server) {

   351

		closesocket(_udp_server_socket);

   352

		_udp_server_socket = INVALID_SOCKET;

   357

		_network_udp_server = false;

   358

		_network_udp_broadcast = 0;

   359

	} else {

   360

		closesocket(_udp_client_socket);

   361

		_udp_client_socket = INVALID_SOCKET;

   362

		_network_udp_broadcast = 0;

   363

	}

   364

}

   365

   366

// Receive something on UDP level

   368

{

   369

	struct sockaddr_in client_addr;

   370

#ifndef __MORPHOS__

   371

	int client_len;

   372

#else

   373

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

   374

#endif

   375

	int nbytes;

   376

	static Packet *p = NULL;

   377

	int packet_len;

   378

   379

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

   380

	if (p == NULL)

   381

		p = malloc(sizeof(Packet));

   382

   383

	packet_len = sizeof(p->buffer);

   384

	client_len = sizeof(client_addr);

   385

   386

	// Try to receive anything

   387

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

   388

   389

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

   390

	if (nbytes > 0) {

   391

		// Get the size of the buffer

   392

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

   393

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

   394

		// Put the position on the right place

   395

		p->pos = 2;

   396

		p->next = NULL;

   397

   398

		// Handle the packet

   399

		NetworkHandleUDPPacket(p, &client_addr);

   400

   401

		// Free the packet

   402

		free(p);

   403

		p = NULL;

   404

	}

   405

}

   406

   407

// Broadcast to all ips

   409

{

   410

	int i;

   411

	struct sockaddr_in out_addr;

   412

	byte *bcptr;

   413

	uint32 bcaddr;

   414

	Packet *p;

   415

   416

	// Init the packet

   417

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);

   418

   419

	// Go through all the ips on this pc

   420

	i = 0;

   421

	while (_network_ip_list[i] != 0) {

   422

		bcaddr = _network_ip_list[i];

   423

		bcptr = (byte *)&bcaddr;

   424

		// Make the address a broadcast address

   425

		bcptr[3] = 255;

   426

   427

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

   428

   429

		out_addr.sin_family = AF_INET;

   430

		out_addr.sin_port = htons(_network_server_port);

   431

		out_addr.sin_addr.s_addr = bcaddr;

   432

   434

   435

		i++;

   436

	}

   462

   463

	free(p);

   464

}

   465

   466

// Find all servers

   467

void NetworkUDPSearchGame(void)

   468

{

   469

	// We are still searching..

   470

	if (_network_udp_broadcast > 0)

   471

		return;

   472

   473

	// No UDP-socket yet..

   474

	if (_udp_client_socket == INVALID_SOCKET)

   476

			return;

   477

   478

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

   479

   481

	_network_udp_broadcast = 300; // Stay searching for 300 ticks

   482

}

   483

   484

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

   485

{

   486

	struct sockaddr_in out_addr;

   487

	Packet *p;

   488

	NetworkGameList *item;

   489

	char hostname[NETWORK_HOSTNAME_LENGTH];

   490

   491

	// No UDP-socket yet..

   492

	if (_udp_client_socket == INVALID_SOCKET)

   494

			return NULL;

   495

   496

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

   497

   498

	out_addr.sin_family = AF_INET;

   499

	out_addr.sin_port = htons(port);

   500

	out_addr.sin_addr.s_addr = NetworkResolveHost(host);

   501

   502

	// Clear item in gamelist

   503

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

   504

	memset(&item->info, 0, sizeof(item->info));

   505

	snprintf(item->info.server_name, sizeof(item->info.server_name), "%s", hostname);

   506

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

   507

	item->online = false;

   508

   509

	// Init the packet

   510

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);