1

#include "stdafx.h"

     2

#include "network_data.h"

     3

     4

#ifdef ENABLE_NETWORK

     5

     6

#include "network_gamelist.h"

     7

     8

extern void UpdateNetworkGameWindow(bool unselect);

     9

extern void NetworkPopulateCompanyInfo(void);

    10

    11

//

    12

// This file handles all the LAN-stuff

    13

// Stuff like:

    14

//   - UDP search over the network

    15

//

    16

    17

typedef enum {

    18

	PACKET_UDP_CLIENT_FIND_SERVER,

    19

	PACKET_UDP_SERVER_RESPONSE,

    20

	PACKET_UDP_CLIENT_DETAIL_INFO,

    21

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

    23

	PACKET_UDP_END

    24

} PacketUDPType;

    25

    26

static SOCKET _udp_server_socket; // udp server socket

    27

    28

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

    29

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

    30

    31

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER)

    32

{

    33

	Packet *packet;

    34

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

    35

	if (!_network_udp_server)

    36

		return;

    37

    38

	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);

    39

    40

	// Update some game_info

    41

	_network_game_info.game_date = _date;

    42

	_network_game_info.map_set = _opt.landscape;

    43

    44

	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);

    45

	NetworkSend_string(packet, _network_game_info.server_name);

    46

	NetworkSend_string(packet, _network_game_info.server_revision);

    47

	NetworkSend_uint8 (packet, _network_game_info.server_lang);

    48

	NetworkSend_uint8 (packet, _network_game_info.use_password);

    49

	NetworkSend_uint8 (packet, _network_game_info.clients_max);

    50

	NetworkSend_uint8 (packet, _network_game_info.clients_on);

    51

	NetworkSend_uint8 (packet, _network_game_info.spectators_on);

    52

	NetworkSend_uint16(packet, _network_game_info.game_date);

    53

	NetworkSend_uint16(packet, _network_game_info.start_date);

    54

	NetworkSend_string(packet, _network_game_info.map_name);

    55

	NetworkSend_uint16(packet, _network_game_info.map_width);

    56

	NetworkSend_uint16(packet, _network_game_info.map_height);

    57

	NetworkSend_uint8 (packet, _network_game_info.map_set);

    58

	NetworkSend_uint8 (packet, _network_game_info.dedicated);

    59

    60

	// Let the client know that we are here

    61

	NetworkSendUDP_Packet(packet, client_addr);

    62

    63

	free(packet);

    64

    65

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

    66

}

    67

    68

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)

    69

{

    70

	NetworkGameList *item;

    71

	byte game_info_version;

    72

    73

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

    74

	if (_network_udp_server)

    75

		return;

    76

    77

	game_info_version = NetworkRecv_uint8(p);

    78

    79

	// Find next item

    80

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

    81

    82

	if (game_info_version == 1) {

    83

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

    84

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

    85

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

    86

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

    87

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

    88

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

    89

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

    90

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

    91

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

    92

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

    93

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

    94

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

    95

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

    96

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

    97

    98

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

    99

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

   100

	}

   101

   102

	item->online = true;

   103

   104

	UpdateNetworkGameWindow(false);

   105

}

   106

   107

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO)

   108

{

   109

	ClientState *cs;

   110

	NetworkClientInfo *ci;

   111

	Packet *packet;

   112

	Player *player;

   113

	byte active = 0;

   114

	byte current = 0;

   115

	int i;

   116

   117

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

   118

	if (!_network_udp_server)

   119

		return;

   120

   121

	packet = NetworkSend_Init(PACKET_UDP_SERVER_DETAIL_INFO);

   122

   123

	FOR_ALL_PLAYERS(player) {

   124

		if (player->is_active)

   125

			active++;

   126

	}

   127

   128

	/* Send the amount of active companies */

   129

	NetworkSend_uint8 (packet, NETWORK_COMPANY_INFO_VERSION);

   130

	NetworkSend_uint8 (packet, active);

   131

   132

	/* Fetch the latest version of everything */

   133

	NetworkPopulateCompanyInfo();

   134

   135

	/* Go through all the players */

   136

	FOR_ALL_PLAYERS(player) {

   137

		/* Skip non-active players */

   138

		if (!player->is_active)

   139

			continue;

   140

   141

		current++;

   142

   143

		/* Send the information */

   144

		NetworkSend_uint8 (packet, current);

   145

   146

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

   147

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

   148

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

   149

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

   150

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

   151

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

   152

   153

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

   154

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

   155

   156

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

   157

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

   158

   159

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

   160

		FOR_ALL_CLIENTS(cs) {

   161

			ci = DEREF_CLIENT_INFO(cs);

   162

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

   163

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

   164

				NetworkSend_uint8(packet, 1);

   165

				NetworkSend_string(packet, ci->client_name);

   166

				NetworkSend_string(packet, ci->unique_id);

   167

				NetworkSend_uint16(packet, ci->join_date);

   168

			}

   169

		}

   170

		/* Also check for the server itself */

   171

		ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   172

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

   173

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

   174

			NetworkSend_uint8(packet, 1);

   175

			NetworkSend_string(packet, ci->client_name);

   176

			NetworkSend_string(packet, ci->unique_id);

   177

			NetworkSend_uint16(packet, ci->join_date);

   178

		}

   179

   180

		/* Indicates end of client list */

   181

		NetworkSend_uint8(packet, 0);

   182

	}

   183

   184

	/* And check if we have any spectators */

   185

	FOR_ALL_CLIENTS(cs) {

   186

		ci = DEREF_CLIENT_INFO(cs);

   187

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

   188

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

   189

			NetworkSend_uint8(packet, 1);

   190

			NetworkSend_string(packet, ci->client_name);

   191

			NetworkSend_string(packet, ci->unique_id);

   192

			NetworkSend_uint16(packet, ci->join_date);

   193

		}

   194

	}

   195

	/* Also check for the server itself */

   196

	ci = NetworkFindClientInfoFromIndex(NETWORK_SERVER_INDEX);

   197

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

   198

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

   199

		NetworkSend_uint8(packet, 1);

   200

		NetworkSend_string(packet, ci->client_name);

   201

		NetworkSend_string(packet, ci->unique_id);

   202

		NetworkSend_uint16(packet, ci->join_date);

   203

	}

   204

   205

	/* Indicates end of client list */

   206

	NetworkSend_uint8(packet, 0);

   207

   208

	NetworkSendUDP_Packet(packet, client_addr);

   209

   210

	free(packet);

   211

}

   212

   213

   214

// The layout for the receive-functions by UDP

   215

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

   216

   217

static NetworkUDPPacket* const _network_udp_packet[] = {

   218

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_FIND_SERVER),

   219

	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),

   220

	RECEIVE_COMMAND(PACKET_UDP_CLIENT_DETAIL_INFO),

   221

	NULL,

   223

};

   224

   225

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

   226

assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);

   227

   228

   229

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

   230

{

   231

	byte type;

   232

   233

	type = NetworkRecv_uint8(p);

   234

   235

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

   236

		_network_udp_packet[type](p, client_addr);

   237

	}	else {

   238

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

   239

	}

   240

}

   241

   242

   243

// Send a packet over UDP

   244

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

   245

{

   246

	SOCKET udp;

   247

	int res;

   248

   249

	// Find the correct socket

   250

	if (_network_udp_server)

   251

		udp = _udp_server_socket;

   252

	else

   253

		udp = _udp_client_socket;

   254

   255

	// Put the length in the buffer

   256

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

   257

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

   258

   259

	// Send the buffer

   260

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

   261

   262

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

   263

	if (res == -1) {

   264

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

   265

	}

   266

}

   267

   268

// Start UDP listener

   269

bool NetworkUDPListen(uint32 host, uint16 port)

   270

{

   271

	struct sockaddr_in sin;

   272

	SOCKET udp;

   273

   274

	// Make sure sockets are closed

   275

	if (_network_udp_server)

   276

		closesocket(_udp_server_socket);

   277

	else

   278

		closesocket(_udp_client_socket);

   279

   280

	udp = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

   281

	if (udp == INVALID_SOCKET) {

   282

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

   283

		return false;

   284

	}

   285

   286

	// set nonblocking mode for socket

   287

	{

   288

		unsigned long blocking = 1;

   289

		ioctlsocket(udp, FIONBIO, &blocking);

   290

	}

   291

   292

	sin.sin_family = AF_INET;

   293

	// Listen on all IPs

   294

	sin.sin_addr.s_addr = host;

   295

	sin.sin_port = htons(port);

   296

   297

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

   298

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

   299

		return false;

   300

	}

   301

   302

	// enable broadcasting

   303

	// allow reusing

   304

	{

   305

		unsigned long val = 1;

   306

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

   307

		val = 1;

   308

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

   309

	}

   310

   311

	if (_network_udp_server)

   312

		_udp_server_socket = udp;

   313

	else

   314

		_udp_client_socket = udp;

   315

   316

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

   317

   318

	return true;

   319

}

   320

   321

// Close UDP connection

   322

void NetworkUDPClose(void)

   323

{

   324

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

   325

   326

	if (_network_udp_server) {

   327

		closesocket(_udp_server_socket);

   328

		_udp_server_socket = INVALID_SOCKET;

   329

		_network_udp_server = false;

   330

		_network_udp_broadcast = 0;

   331

	} else {

   332

		closesocket(_udp_client_socket);

   333

		_udp_client_socket = INVALID_SOCKET;

   334

		_network_udp_broadcast = 0;

   335

	}

   336

}

   337

   338

// Receive something on UDP level

   339

void NetworkUDPReceive(void)

   340

{

   341

	struct sockaddr_in client_addr;

   342

#ifndef __MORPHOS__

   343

	int client_len;

   344

#else

   345

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

   346

#endif

   347

	int nbytes;

   348

	static Packet *p = NULL;

   349

	int packet_len;

   350

	SOCKET udp;

   351

   352

	if (_network_udp_server)

   353

		udp = _udp_server_socket;

   354

	else

   355

		udp = _udp_client_socket;

   356

   357

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

   358

	if (p == NULL)

   359

		p = malloc(sizeof(Packet));

   360

   361

	packet_len = sizeof(p->buffer);

   362

	client_len = sizeof(client_addr);

   363

   364

	// Try to receive anything

   365

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

   366

   367

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

   368

	if (nbytes > 0) {

   369

		// Get the size of the buffer

   370

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

   371

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

   372

		// Put the position on the right place

   373

		p->pos = 2;

   374

		p->next = NULL;

   375

   376

		// Handle the packet

   377

		NetworkHandleUDPPacket(p, &client_addr);

   378

   379

		// Free the packet

   380

		free(p);

   381

		p = NULL;

   382

	}

   383

}

   384

   385

// Broadcast to all ips

   386

void NetworkUDPBroadCast(void)

   387

{

   388

	int i;

   389

	struct sockaddr_in out_addr;

   390

	byte *bcptr;

   391

	uint32 bcaddr;

   392

	Packet *p;

   393

   394

	// Init the packet

   395

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);

   396

   397

	// Go through all the ips on this pc

   398

	i = 0;

   399

	while (_network_ip_list[i] != 0) {

   400

		bcaddr = _network_ip_list[i];

   401

		bcptr = (byte *)&bcaddr;

   402

		// Make the address a broadcast address

   403

		bcptr[3] = 255;

   404

   405

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

   406

   407

		out_addr.sin_family = AF_INET;

   408

		out_addr.sin_port = htons(_network_server_port);

   409

		out_addr.sin_addr.s_addr = bcaddr;

   410

   411

		NetworkSendUDP_Packet(p, &out_addr);

   412

   413

		i++;

   414

	}

   415

   416

	free(p);

   417

}

   418

   419

// Find all servers

   420

void NetworkUDPSearchGame(void)

   421

{

   422

	// We are still searching..

   423

	if (_network_udp_broadcast > 0)

   424

		return;

   425

   426

	// No UDP-socket yet..

   427

	if (_udp_client_socket == INVALID_SOCKET)

   428

		if (!NetworkUDPListen(0, 0))

   429

			return;

   430

   431

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

   432

   433

	NetworkUDPBroadCast();

   434

	_network_udp_broadcast = 300; // Stay searching for 300 ticks

   435

}

   436

   437

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

   438

{

   439

	struct sockaddr_in out_addr;

   440

	Packet *p;

   441

	NetworkGameList *item;

   442

	char hostname[NETWORK_HOSTNAME_LENGTH];

   443

   444

	// No UDP-socket yet..

   445

	if (_udp_client_socket == INVALID_SOCKET)

   446

		if (!NetworkUDPListen(0, 0))

   447

			return;

   448

   449

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

   450

   451

	out_addr.sin_family = AF_INET;

   452

	out_addr.sin_port = htons(port);

   453

	out_addr.sin_addr.s_addr = NetworkResolveHost(host);

   454

   455

	// Clear item in gamelist

   456

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

   457

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

   458

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

   459

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

   460

	item->online = false;

   461

   462

	// Init the packet

   463

	p = NetworkSend_Init(PACKET_UDP_CLIENT_FIND_SERVER);

   464

   465

	NetworkSendUDP_Packet(p, &out_addr);

   466

   467

	free(p);

   468

   469

	UpdateNetworkGameWindow(false);

   470

}

   471

   505

void NetworkUDPInitialize(void)

   506

{

   507

	_udp_client_socket = INVALID_SOCKET;

   508

	_udp_server_socket = INVALID_SOCKET;

   509

   510

	_network_udp_server = false;

   511

	_network_udp_broadcast = 0;

   512

}

   513

   514

#endif /* ENABLE_NETWORK */