1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "gui.h"

     4

#include "command.h"

     5

#include "player.h"

     6

#include "console.h"

     7

     8

#if defined(WIN32)

     9

#	include <windows.h>

    10

#	include <winsock.h>

    11

    12

# pragma comment (lib, "ws2_32.lib")

    13

# define ENABLE_NETWORK

    14

# define GET_LAST_ERROR() WSAGetLastError()

    15

# define EWOULDBLOCK WSAEWOULDBLOCK

    16

#endif

    17

    18

#if defined(UNIX)

    19

// Make compatible with WIN32 names

    20

#	define SOCKET int

    21

#	define INVALID_SOCKET -1

    22

// we need different defines for MorphOS and AmigaOS

    23

#if !defined(__MORPHOS__) && !defined(__AMIGA__)

    24

#	define ioctlsocket ioctl

    25

# define closesocket close

    26

# define GET_LAST_ERROR() errno

    27

#endif

    28

// Need this for FIONREAD on solaris

    29

#	define BSD_COMP

    30

#	include <unistd.h>

    31

#	include <sys/ioctl.h>

    32

    33

// Socket stuff

    34

#	include <sys/socket.h>

    35

#	include <netinet/in.h>

    36

#	include <arpa/inet.h>

    37

# 	include <errno.h>

    38

# 	include <sys/time.h>

    39

// NetDB

    40

#   include <netdb.h>

    41

    42

# ifndef TCP_NODELAY

    43

#  define TCP_NODELAY 0x0001

    44

# endif

    45

    46

#endif

    47

    48

    49

#if defined(__MORPHOS__) || defined(__AMIGA__)

    50

#	include <exec/types.h>

    51

#	include <proto/exec.h> 		// required for Open/CloseLibrary()

    52

#	if defined(__MORPHOS__)

    53

#		include <sys/filio.h> 	// FION#? defines

    54

#	else // __AMIGA__

    55

#		include	<proto/socket.h>

    56

#	endif

    57

    58

// make source compatible with bsdsocket.library functions

    59

# define closesocket(s)     						CloseSocket(s)

    60

# define GET_LAST_ERROR() 							Errno()

    61

#	define ioctlsocket(s,request,status)  IoctlSocket((LONG)s,(ULONG)request,(char*)status)

    62

    63

struct Library *SocketBase = NULL;

    64

    65

#if !defined(__MORPHOS__)

    66

// usleep() implementation

    67

#include <devices/timer.h>

    68

#include <dos/dos.h>

    69

    70

struct Device       *TimerBase    = NULL;

    71

struct MsgPort      *TimerPort    = NULL;

    72

struct timerequest  *TimerRequest = NULL;

    73

#endif

    74

    75

#endif /* __MORPHOS__ || __AMIGA__ */

    76

    77

    78

#define SEND_MTU 1460

    79

    80

#if defined(ENABLE_NETWORK)

    81

    82

enum {

    83

	PACKET_TYPE_WELCOME = 0,

    84

	PACKET_TYPE_READY,

    85

	PACKET_TYPE_ACK,

    86

	PACKET_TYPE_SYNC,

    87

	PACKET_TYPE_FSYNC,

    88

	PACKET_TYPE_XMIT,

    89

	PACKET_TYPE_COMMAND,

    90

};

    91

    92

// sent from client -> server whenever the client wants to exec a command.

    93

// send from server -> client when another player execs a command.

    94

typedef struct CommandPacket {

    95

	byte packet_length;

    96

	byte packet_type;

    97

	uint16 cmd;

    98

	uint32 p1,p2;

    99

	TileIndex tile;

   100

	byte player;// player id, this is checked by the server.

   101

	byte when;  // offset from the current max_frame value minus 1. this is set by the server.

   102

	uint32 dp[8];

   103

} CommandPacket;

   104

   105

#define COMMAND_PACKET_BASE_SIZE (sizeof(CommandPacket) - 8 * sizeof(uint32))

   106

   107

// sent from server -> client periodically to tell the client about the current tick in the server

   108

// and how far the client may progress.

   109

typedef struct SyncPacket {

   110

	byte packet_length;

   111

	byte packet_type;

   112

	byte frames; // how many more frames may the client execute? this is relative to the old value of max.

   113

	byte server; // where is the server currently executing? this is negatively relative to the old value of max.

   114

	uint32 random_seed_1; // current random state at server. used to detect out of sync.

   115

	uint32 random_seed_2;

   116

} SyncPacket;

   117

   118

typedef struct FrameSyncPacket {

   119

	byte packet_length;

   120

	byte packet_type;

   121

	byte frames; // where is the server currently executing? this is negatively relative to the old value of max.

   122

} FrameSyncPacket;

   123

   124

// sent from server -> client as an acknowledgement that the server received the command.

   125

// the command will be executed at the current value of "max".

   126

typedef struct AckPacket {

   127

	byte packet_length;

   128

	byte packet_type;

   129

	int16 when;

   130

} AckPacket;

   131

   132

typedef struct ReadyPacket {

   133

	byte packet_length;

   134

	byte packet_type;

   135

} ReadyPacket;

   136

   137

typedef struct FilePacketHdr {

   138

	byte packet_length;

   139

	byte packet_type;

   140

} FilePacketHdr;

   141

   142

// sent from server to client when the client has joined.

   143

typedef struct WelcomePacket {

   144

	byte packet_length;

   145

	byte packet_type;

   146

	uint32 player_seeds[MAX_PLAYERS][2];

   147

	uint32 frames_max;

   148

	uint32 frames_srv;

   149

	uint32 frames_cnt;

   150

} WelcomePacket;

   151

   152

typedef struct Packet Packet;

   153

struct Packet {

   154

	Packet *next; // this one has to be the first element.

   155

	uint siz;

   156

	byte buf[SEND_MTU]; // packet payload

   157

};

   158

   159

typedef struct ClientState {

   160

	int socket;

   161

	bool inactive; // disable sending of commands/syncs to client

   162

	bool writable;

   163

	bool ready;

   164

	uint timeout;

   165

	uint xmitpos;

   166

   167

	uint eaten;

   168

	Packet *head, **last;

   169

   170

	uint buflen;											// receive buffer len

   171

	byte buf[1024];										// receive buffer

   172

} ClientState;

   173

   174

   175

typedef struct QueuedCommand QueuedCommand;

   176

struct QueuedCommand {

   177

	QueuedCommand *next;

   178

	CommandPacket cp;

   179

	CommandCallback *callback;

   180

	uint32 cmd;

   181

	uint32 frame;

   182

};

   183

   184

typedef struct CommandQueue CommandQueue;

   185

struct CommandQueue {

   186

	QueuedCommand *head, **last;

   187

};

   188

   189

#define MAX_CLIENTS (MAX_PLAYERS + 1)

   190

   191

// packets waiting to be executed, for each of the players.

   192

// this list is sorted in frame order, so the item on the front will be executed first.

   193

static CommandQueue _command_queue;

   194

   195

// in the client, this is the list of commands that have not yet been acked.

   196

// when it is acked, it will be moved to the appropriate position at the end of the player queue.

   197

static CommandQueue _ack_queue;

   198

   199

static ClientState _clients[MAX_CLIENTS];

   200

static int _num_clients;

   201

   202

// keep a history of the 16 most recent seeds to be able to capture out of sync errors.

   203

static uint32 _my_seed_list[16][2];

   204

static bool _network_ready_sent;

   205

static uint32 _frame_fsync_last;

   206

   207

typedef struct FutureSeeds {

   208

	uint32 frame;

   209

	uint32 seed[2];

   210

} FutureSeeds;

   211

   212

// remember some future seeds that the server sent to us.

   213

static FutureSeeds _future_seed[8];

   214

static uint _num_future_seed;

   215

   216

static SOCKET _listensocket; // tcp socket

   217

   218

static SOCKET _udp_client_socket; // udp server socket

   219

static SOCKET _udp_server_socket; // udp client socket

   220

   221

typedef struct UDPPacket {

   222

	byte command_code;

   223

	byte data_len;

   224

	byte command_check;

   225

	byte data[255];

   226

} UDPPacket;

   227

   228

enum {

   229

	NET_UDPCMD_SERVERSEARCH = 1,

   230

	NET_UDPCMD_SERVERACTIVE,

   231

	NET_UDPCMD_GETSERVERINFO,

   232

	NET_UDPCMD_SERVERINFO,

   233

};

   234

   235

void NetworkUDPSend(bool client, struct sockaddr_in recv,struct UDPPacket packet);

   236

static void HandleCommandPacket(ClientState *cs, CommandPacket *np);

   237

static void CloseClient(ClientState *cs);

   238

void NetworkSendWelcome(ClientState *cs, bool direct);

   239

   240

uint32 _network_ip_list[10]; // network ip list

   241

   242

// this is set to point to the savegame

   243

static byte *_transmit_file;

   244

static size_t _transmit_file_size;

   245

   246

static FILE *_recv_file;

   247

   248

/* multi os compatible sleep function */

   249

void CSleep(int milliseconds) {

   250

#if defined(WIN32)

   251

Sleep(milliseconds);

   252

#endif

   253

#if defined(UNIX)

   254

#if !defined(__BEOS__) && !defined(__MORPHOS__) && !defined(__AMIGAOS__)

   255

usleep(milliseconds*1000);

   256

#endif

   257

#ifdef __BEOS__

   258

snooze(milliseconds*1000);

   259

#endif

   260

#if defined(__MORPHOS__)

   261

usleep(milliseconds*1000);

   262

#endif

   263

#if defined(__AMIGAOS__) && !defined(__MORPHOS__)

   264

{

   265

	ULONG signals;

   266

	ULONG TimerSigBit = 1 << TimerPort->mp_SigBit;

   267

   268

	// send IORequest

   269

	TimerRequest->tr_node.io_Command = TR_ADDREQUEST;

   270

	TimerRequest->tr_time.tv_secs    = (milliseconds * 1000) / 1000000;

   271

	TimerRequest->tr_time.tv_micro   = (milliseconds * 1000) % 1000000;

   272

	SendIO((struct IORequest *)TimerRequest);

   273

   274

	if ( !((signals = Wait(TimerSigBit|SIGBREAKF_CTRL_C)) & TimerSigBit) ) {

   275

		AbortIO((struct IORequest *)TimerRequest);

   276

	}

   277

	WaitIO((struct IORequest *)TimerRequest);

   278

}

   279

#endif // __AMIGAOS__ && !__MORPHOS__

   280

#endif

   281

}

   282

   283

//////////////////////////////////////////////////////////////////////

   284

   285

// ****************************** //

   286

// * Network Error Handlers     * //

   287

// ****************************** //

   288

   289

static void NetworkHandleSaveGameError()

   290

{

   291

		_networking_sync = false;

   292

		_networking_queuing = true;

   293

		_switch_mode = SM_MENU;

   294

		_switch_mode_errorstr = STR_NETWORK_ERR_SAVEGAMEERROR;

   295

}

   296

   297

static void NetworkHandleConnectionLost()

   298

{

   299

		_networking_sync = false;

   300

		_networking_queuing = true;

   301

		_switch_mode = SM_MENU;

   302

		_switch_mode_errorstr = STR_NETWORK_ERR_LOSTCONNECTION;

   303

}

   304

   305

static void NetworkHandleDeSync()

   306

{

   307

	DEBUG(net, 0) ("[NET] Fatal ERROR: network sync error at frame %i", _frame_counter);

   308

	{

   309

		int i;

   310

		for (i=15; i>=0; i--) DEBUG(net,0) ("[NET] frame %i: [0]=%i, [1]=%i",_frame_counter-(i+1),_my_seed_list[i][0],_my_seed_list[i][1]);

   311

		for (i=0; i<8; i++) DEBUG(net,0) ("[NET] frame %i: [0]=%i, [1]=%i",_frame_counter+i,_future_seed[i].seed[0],_future_seed[i].seed[1]);

   312

	}

   313

	_networking_sync = false;

   314

	_networking_queuing = true;

   315

	_switch_mode = SM_MENU;

   316

	_switch_mode_errorstr = STR_NETWORK_ERR_DESYNC;

   317

}

   318

   319

// ****************************** //

   320

// * TCP Packets and Handlers   * //

   321

// ****************************** //

   322

   323

static QueuedCommand *AllocQueuedCommand(CommandQueue *nq)

   324

{

   325

	QueuedCommand *qp = (QueuedCommand*)calloc(sizeof(QueuedCommand), 1);

   326

	assert(qp);

   327

	*nq->last = qp;

   328

	nq->last = &qp->next;

   329

	return qp;

   330

}

   331

   332

static void QueueClear(CommandQueue *nq)

   333

{

   334

	QueuedCommand *qp;

   335

	while ((qp=nq->head)) {

   336

		// unlink it.

   337

		if (!(nq->head = qp->next)) nq->last = &nq->head;

   338

		free(qp);

   339

		}

   340

	nq->last = &nq->head;

   341

}

   342

   343

static int GetNextSyncFrame()

   344

{

   345

	uint32 newframe;

   346

	if (_frame_fsync_last == 0) return -1;

   347

	newframe = (_frame_fsync_last + 9);

   348

	if ( (newframe + 4) > _frame_counter_max) return -1;

   349

	return (_frame_counter_max - newframe);

   350

   351

}

   352

   353

// go through the player queues for each player and see if there are any pending commands

   354

// that should be executed this frame. if there are, execute them.

   355

void NetworkProcessCommands()

   356

{

   357

	CommandQueue *nq;

   358

	QueuedCommand *qp;

   359

   360

	// queue mode ?

   361

	if (_networking_queuing)

   362

		return;

   363

   364

	nq = &_command_queue;

   365

	while ( (qp=nq->head) && (!_networking_sync || qp->frame <= _frame_counter)) {

   366

		// unlink it.

   367

		if (!(nq->head = qp->next)) nq->last = &nq->head;

   368

   369

		if (qp->frame < _frame_counter && _networking_sync) {

   370

			DEBUG(net,0) ("error: !qp->cp.frame < _frame_counter, %d < %d\n", qp->frame, _frame_counter);

   371

		}

   372

   373

		// run the command

   374

		_current_player = qp->cp.player;

   375

		memcpy(_decode_parameters, qp->cp.dp, (qp->cp.packet_length - COMMAND_PACKET_BASE_SIZE));

   376

   377

		DoCommandP(qp->cp.tile, qp->cp.p1, qp->cp.p2, qp->callback, qp->cmd | CMD_DONT_NETWORK);

   378

		free(qp);

   379

	}

   380

   381

	if (!_networking_server) {

   382

		// remember the random seed so we can check if we're out of sync.

   383

		_my_seed_list[_frame_counter & 15][0] = _sync_seed_1;

   384

		_my_seed_list[_frame_counter & 15][1] = _sync_seed_2;

   385

   386

		while (_num_future_seed) {

   387

			assert(_future_seed[0].frame >= _frame_counter);

   388

			if (_future_seed[0].frame != _frame_counter) break;

   389

			if (_future_seed[0].seed[0] != _sync_seed_1 ||_future_seed[0].seed[1] != _sync_seed_2) NetworkHandleDeSync();

   390

			memcpy_overlapping(_future_seed, _future_seed + 1, --_num_future_seed * sizeof(FutureSeeds));

   391

		}

   392

	}

   393

}

   394

   395

// send a packet to a client

   396

static void SendBytes(ClientState *cs, void *bytes, uint len)

   397

{

   398

	byte *b = (byte*)bytes;

   399

	uint n;

   400

	Packet *p;

   401

   402

	assert(len != 0);

   403

   404

	// see if there's space in the last packet?

   405

	if (!cs->head || (p = (Packet*)cs->last, p->siz == sizeof(p->buf)))

   406

		p = NULL;

   407

   408

	do {

   409

		if (!p) {

   410

			// need to allocate a new packet buffer.

   411

			p = (Packet*)malloc(sizeof(Packet));

   412

   413

			// insert at the end of the linked list.

   414

			*cs->last = p;

   415

			cs->last = &p->next;

   416

			p->next = NULL;

   417

			p->siz = 0;

   418

		}

   419

   420

		// copy bytes to packet.

   421

		n = minu(sizeof(p->buf) - p->siz, len);

   422

		memcpy(p->buf + p->siz, b, n);

   423

		p->siz += n;

   424

		b += n;

   425

		p = NULL;

   426

	} while (len -= n);

   427

}

   428

   429

// send data direct to a client

   430

static void SendDirectBytes(ClientState *cs, void *bytes, uint len)

   431

{

   432

	char *buf = (char*)bytes;

   433

	uint n;

   434

   435

	n = send(cs->socket, buf, len, 0);

   436

	if (n == -1) {

   437

				int err = GET_LAST_ERROR();

   438

				DEBUG(net, 0) ("[NET] send() failed with error %d", err);

   439

				CloseClient(cs);

   440

			}

   441

}

   442

   443

// client:

   444

//   add it to the client's ack queue, and send the command to the server

   445

// server:

   446

//   add it to the server's player queue, and send it to all clients.

   447

void NetworkSendCommand(TileIndex tile, uint32 p1, uint32 p2, uint32 cmd, CommandCallback *callback)

   448

{

   449

	int nump;

   450

	QueuedCommand *qp;

   451

	ClientState *cs;

   452

   453

	if (!(cmd & CMD_NET_INSTANT)) {

   454

		qp = AllocQueuedCommand(_networking_server ? &_command_queue : &_ack_queue);

   455

	} else {

   456

		qp = (QueuedCommand*)calloc(sizeof(QueuedCommand), 1);

   457

		}

   458

	qp->cp.packet_type = PACKET_TYPE_COMMAND;

   459

	qp->cp.tile = tile;

   460

	qp->cp.p1 = p1;

   461

	qp->cp.p2 = p2;

   462

	qp->cp.cmd = (uint16)cmd;

   463

	qp->cp.player = _local_player;

   464

	qp->cp.when = 0;

   465

	qp->cmd = cmd;

   466

	qp->callback = callback;

   467

   468

	// so the server knows when to execute it.

   469

	qp->frame = _frame_counter_max - GetNextSyncFrame();

   470

   471

	// calculate the amount of extra bytes.

   472

	nump = 8;

   473

	while ( nump != 0 && ((uint32*)_decode_parameters)[nump-1] == 0) nump--;

   474

	qp->cp.packet_length = COMMAND_PACKET_BASE_SIZE + nump * sizeof(uint32);

   475

	if (nump != 0) memcpy(qp->cp.dp, _decode_parameters, nump * sizeof(uint32));

   476

   477

#if defined(TTD_BIG_ENDIAN)

   478

	// need to convert the command to little endian before sending it.

   479

	{

   480

		CommandPacket cp;

   481

		cp = qp->cp;

   482

		cp.cmd = TO_LE16(cp.cmd);

   483

		cp.tile = TO_LE16(cp.tile);

   484

		cp.p1 = TO_LE32(cp.p1);

   485

		cp.p2 = TO_LE32(cp.p2);

   486

		for(cs=_clients; cs->socket != INVALID_SOCKET; cs++) if (!cs->inactive) SendBytes(cs, &cp, cp.packet_length);

   487

	}

   488

#else

   489

	// send it to the peers

   490

	for(cs=_clients; cs->socket != INVALID_SOCKET; cs++) if (!cs->inactive) SendBytes(cs, &qp->cp, qp->cp.packet_length);

   491

   492

#endif

   493

	if (cmd & CMD_NET_INSTANT) {

   494

		free(qp);

   495

	}

   496

}

   497

   498

// client:

   499

//   server sends a command from another player that we should execute.

   500

//   put it in the command queue.

   501

//

   502

// server:

   503

//   client sends a command that it wants to execute.

   504

//   fill the when field so the client knows when to execute it.

   505

//   put it in the appropriate player queue.

   506

//   send it to all other clients.

   507

//   send an ack packet to the actual client.

   508

   509

static void HandleCommandPacket(ClientState *cs, CommandPacket *np)

   510

{

   511

	QueuedCommand *qp;