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

    10

//

    11

// This file handles all the LAN-stuff

    12

// Stuff like:

    13

//   - UDP search over the network

    14

//

    15

    16

typedef enum {

    17

	PACKET_UDP_FIND_SERVER,

    18

	PACKET_UDP_SERVER_RESPONSE,

    19

	PACKET_UDP_END

    20

} PacketUDPType;

    21

    22

static SOCKET _udp_server_socket; // udp server socket

    23

    24

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

    25

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

    26

    27

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_FIND_SERVER)

    28

{

    29

	Packet *packet;

    30

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

    31

	if (!_network_udp_server)

    32

		return;

    33

    34

	packet = NetworkSend_Init(PACKET_UDP_SERVER_RESPONSE);

    35

    36

	// Update some game_info

    37

	_network_game_info.game_date = _date;

    38

	_network_game_info.map_set = _opt.landscape;

    39

    40

	NetworkSend_uint8 (packet, NETWORK_GAME_INFO_VERSION);

    41

	NetworkSend_string(packet, _network_game_info.server_name);

    42

	NetworkSend_string(packet, _network_game_info.server_revision);

    43

	NetworkSend_uint8 (packet, _network_game_info.server_lang);

    44

	NetworkSend_uint8 (packet, _network_game_info.use_password);

    45

	NetworkSend_uint8 (packet, _network_game_info.clients_max);

    46

	NetworkSend_uint8 (packet, _network_game_info.clients_on);

    47

	NetworkSend_uint8 (packet, _network_game_info.spectators_on);

    48

	NetworkSend_uint16(packet, _network_game_info.game_date);

    49

	NetworkSend_uint16(packet, _network_game_info.start_date);

    50

	NetworkSend_string(packet, _network_game_info.map_name);

    51

	NetworkSend_uint16(packet, _network_game_info.map_width);

    52

	NetworkSend_uint16(packet, _network_game_info.map_height);

    53

	NetworkSend_uint8 (packet, _network_game_info.map_set);

    54

	NetworkSend_uint8 (packet, _network_game_info.dedicated);

    55

    56

	// Let the client know that we are here

    57

	NetworkSendUDP_Packet(packet, client_addr);

    58

    59

	free(packet);

    60

    61

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

    62

}

    63

    64

DEF_UDP_RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE)

    65

{

    66

	NetworkGameList *item;

    67

	byte game_info_version;

    68

    69

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

    70

	if (_network_udp_server)

    71

		return;

    72

    73

	game_info_version = NetworkRecv_uint8(p);

    74

    75

	// Find next item

    76

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

    77

    78

	if (game_info_version == 1) {

    79

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

    80

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

    81

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

    82

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

    83

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

    84

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

    85

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

    86

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

    87

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

    88

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

    89

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

    90

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

    91

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

    92

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

    93

    94

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

    95

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

    96

	}

    97

    98

	item->online = true;

    99

   100

	UpdateNetworkGameWindow(false);

   101

}

   102

   103

   104

// The layout for the receive-functions by UDP

   105

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

   106

   107

static NetworkUDPPacket* const _network_udp_packet[] = {

   108

	RECEIVE_COMMAND(PACKET_UDP_FIND_SERVER),

   109

	RECEIVE_COMMAND(PACKET_UDP_SERVER_RESPONSE),

   110

};

   111

   112

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

   113

assert_compile(lengthof(_network_udp_packet) == PACKET_UDP_END);

   114

   115

   116

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

   117

{

   118

	byte type;

   119

   120

	type = NetworkRecv_uint8(p);

   121

   122

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

   123

		_network_udp_packet[type](p, client_addr);

   124

	}	else {

   125

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

   126

	}

   127

}

   128

   129

   130

// Send a packet over UDP

   131

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

   132

{

   133

	SOCKET udp;

   134

	int res;

   135

   136

	// Find the correct socket

   137

	if (_network_udp_server)

   138

		udp = _udp_server_socket;

   139

	else

   140

		udp = _udp_client_socket;

   141

   142

	// Put the length in the buffer

   143

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

   144

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

   145

   146

	// Send the buffer

   147

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

   148

   149

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

   150

	if (res == -1) {

   151

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

   152

	}

   153

}

   154

   155

// Start UDP listener

   156

bool NetworkUDPListen(uint32 host, uint16 port)

   157

{

   158

	struct sockaddr_in sin;

   159

	SOCKET udp;

   160

   161

	// Make sure sockets are closed

   162

	if (_network_udp_server)

   163

		closesocket(_udp_server_socket);

   164

	else

   165

		closesocket(_udp_client_socket);

   166

   167

	udp = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

   168

	if (udp == INVALID_SOCKET) {

   169

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

   170

		return false;

   171

	}

   172

   173

	// set nonblocking mode for socket

   174

	{

   175

		unsigned long blocking = 1;

   176

		ioctlsocket(udp, FIONBIO, &blocking);

   177

	}

   178

   179

	sin.sin_family = AF_INET;

   180

	// Listen on all IPs

   181

	sin.sin_addr.s_addr = host;

   182

	sin.sin_port = htons(port);

   183

   184

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

   185

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

   186

		return false;

   187

	}

   188

   189

	// enable broadcasting

   190

	// allow reusing

   191

	{

   192

		unsigned long val = 1;

   193

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

   194

		val = 1;

   195

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

   196

	}

   197

   198

	if (_network_udp_server)

   199

		_udp_server_socket = udp;

   200

	else

   201

		_udp_client_socket = udp;

   202

   203

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

   204

   205

	return true;

   206

}

   207

   208

// Close UDP connection

   209

void NetworkUDPClose(void)

   210

{

   211

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

   212

   213

	if (_network_udp_server) {

   214

		closesocket(_udp_server_socket);

   215

		_udp_server_socket = INVALID_SOCKET;

   216

		_network_udp_server = false;

   217

		_network_udp_broadcast = 0;

   218

	} else {

   219

		closesocket(_udp_client_socket);

   220

		_udp_client_socket = INVALID_SOCKET;

   221

		_network_udp_broadcast = 0;

   222

	}

   223

}

   224

   225

// Receive something on UDP level

   226

void NetworkUDPReceive(void)

   227

{

   228

	struct sockaddr_in client_addr;

   229

#ifndef __MORPHOS__

   230

	int client_len;

   231

#else

   232

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

   233

#endif

   234

	int nbytes;

   235

	static Packet *p = NULL;

   236

	int packet_len;

   237

	SOCKET udp;

   238

   239

	if (_network_udp_server)

   240

		udp = _udp_server_socket;

   241

	else

   242

		udp = _udp_client_socket;

   243

   244

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

   245

	if (p == NULL)

   246

		p = malloc(sizeof(Packet));

   247

   248

	packet_len = sizeof(p->buffer);

   249

	client_len = sizeof(client_addr);

   250

   251

	// Try to receive anything

   252

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

   253

   254

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

   255

	if (nbytes > 0) {

   256

		// Get the size of the buffer

   257

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

   258

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

   259

		// Put the position on the right place

   260

		p->pos = 2;

   261

		p->next = NULL;

   262

   263

		// Handle the packet

   264

		NetworkHandleUDPPacket(p, &client_addr);

   265

   266

		// Free the packet

   267

		free(p);

   268

		p = NULL;

   269

	}

   270

}

   271

   272

// Broadcast to all ips

   273

void NetworkUDPBroadCast(void)

   274

{

   275

	int i;

   276

	struct sockaddr_in out_addr;

   277

	byte *bcptr;

   278

	uint32 bcaddr;

   279

	Packet *p;

   280

   281

	// Init the packet

   282

	p = NetworkSend_Init(PACKET_UDP_FIND_SERVER);

   283

   284

	// Go through all the ips on this pc

   285

	i = 0;

   286

	while (_network_ip_list[i] != 0) {

   287

		bcaddr = _network_ip_list[i];

   288

		bcptr = (byte *)&bcaddr;

   289

		// Make the address a broadcast address

   290

		bcptr[3] = 255;

   291

   292

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

   293

   294

		out_addr.sin_family = AF_INET;

   295

		out_addr.sin_port = htons(_network_server_port);

   296

		out_addr.sin_addr.s_addr = bcaddr;

   297

   298

		NetworkSendUDP_Packet(p, &out_addr);

   299

   300

		i++;

   301

	}

   302

   303

	free(p);

   304

}

   305

   306

// Find all servers

   307

void NetworkUDPSearchGame(void)

   308

{

   309

	// We are still searching..

   310

	if (_network_udp_broadcast > 0)

   311

		return;

   312

   313

	// No UDP-socket yet..

   314

	if (_udp_client_socket == INVALID_SOCKET)

   315

		if (!NetworkUDPListen(0, 0))

   316

			return;

   317

   318

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

   319

   320

	NetworkUDPBroadCast();

   321

	_network_udp_broadcast = 300; // Stay searching for 300 ticks

   322

}

   323

   324

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

   325

{

   326

	struct sockaddr_in out_addr;

   327

	Packet *p;

   328

	NetworkGameList *item;

   329

	char hostname[NETWORK_HOSTNAME_LENGTH];

   330

   331

	// No UDP-socket yet..

   332

	if (_udp_client_socket == INVALID_SOCKET)

   333

		if (!NetworkUDPListen(0, 0))

   334

			return;

   335

   336

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

   337

   338

	out_addr.sin_family = AF_INET;

   339

	out_addr.sin_port = htons(port);

   340

	out_addr.sin_addr.s_addr = NetworkResolveHost(host);

   341

   342

	// Clear item in gamelist

   343

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

   344

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

   345

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

   346

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

   347

	item->online = false;

   348

   349

	// Init the packet

   350

	p = NetworkSend_Init(PACKET_UDP_FIND_SERVER);

   351

   352

	NetworkSendUDP_Packet(p, &out_addr);

   353

   354

	free(p);

   355

   356

	UpdateNetworkGameWindow(false);

   357

}

   358

   359

void NetworkUDPInitialize(void)

   360

{

   361

	_udp_client_socket = INVALID_SOCKET;

   362

	_udp_server_socket = INVALID_SOCKET;

   363

   364

	_network_udp_server = false;

   365

	_network_udp_broadcast = 0;

   366

}

   367

   368

#endif /* ENABLE_NETWORK */