1

/* $Id$ */

     2

     3

/**

     4

 * @file packet.cpp Basic functions to create, fill and read packets.

     5

 */

     6

     7

#ifdef ENABLE_NETWORK

     8

     9

#include "../../stdafx.h"

    10

#include "../../string.h"

    11

    12

#include "packet.h"

    13

    14

    15

/* Do not want to include functions.h and all required headers */

    16

extern void NORETURN CDECL error(const char *str, ...);

    17

    18

/**

    19

 * Create a packet that is used to read from a network socket

    20

 * @param cs the socket handler associated with the socket we are reading from

    21

 */

    22

Packet::Packet(NetworkSocketHandler *cs)

    23

{

    24

	assert(cs != NULL);

    25

    26

	this->cs   = cs;

    27

	this->next = NULL;

    28

	this->pos  = 0; // We start reading from here

    29

	this->size = 0;

    30

}

    31

    32

/**

    33

 * Creates a packet to send

    34

 * @param type of the packet to send

    35

 */

    36

Packet::Packet(PacketType type)

    37

{

    38

	this->cs                   = NULL;

    39

	this->next                 = NULL;

    40

    41

	/* Skip the size so we can write that in before sending the packet */

    42

	this->pos                  = 0;

    43

	this->size                 = sizeof(PacketSize);

    44

	this->buffer[this->size++] = type;

    45

}

    46

    47

/**

    48

 * Create a packet for sending

    49

 * @param type the of packet

    50

 * @return the newly created packet

    51

 */

    52

Packet *NetworkSend_Init(PacketType type)

    53

{

    54

	Packet *packet = new Packet(type);

    55

	/* An error is inplace here, because it simply means we ran out of memory. */

    56

	if (packet == NULL) error("Failed to allocate Packet");

    57

    58

	return packet;

    59

}

    60

    61

/**

    62

 * Writes the packet size from the raw packet from packet->size

    63

 */

    64

void Packet::PrepareToSend()

    65

{

    66

	assert(this->cs == NULL && this->next == NULL);

    67

    68

	this->buffer[0] = GB(this->size, 0, 8);

    69

	this->buffer[1] = GB(this->size, 8, 8);

    70

    71

	this->pos  = 0; // We start reading from here

    72

}

    73

    74

/**

    75

 * The next couple of functions make sure we can send

    76

 *  uint8, uint16, uint32 and uint64 endian-safe

    77

 *  over the network. The least significant bytes are

    78

 *  sent first.

    79

 *

    80

 *  So 0x01234567 would be sent as 67 45 23 01.

    81

 *

    82

 * A bool is sent as a uint8 where zero means false

    83

 *  and non-zero means true.

    84

 */

    85

    86

void Packet::Send_bool(bool data)

    87

{

    88

	this->Send_uint8(data ? 1 : 0);

    89

}

    90

    91

void Packet::Send_uint8(uint8 data)

    92

{

    93

	assert(this->size < sizeof(this->buffer) - sizeof(data));

    94

	this->buffer[this->size++] = data;

    95

}

    96

    97

void Packet::Send_uint16(uint16 data)

    98

{

    99

	assert(this->size < sizeof(this->buffer) - sizeof(data));

   100

	this->buffer[this->size++] = GB(data, 0, 8);

   101

	this->buffer[this->size++] = GB(data, 8, 8);

   102

}

   103

   104

void Packet::Send_uint32(uint32 data)

   105

{

   106

	assert(this->size < sizeof(this->buffer) - sizeof(data));

   107

	this->buffer[this->size++] = GB(data,  0, 8);

   108

	this->buffer[this->size++] = GB(data,  8, 8);

   109

	this->buffer[this->size++] = GB(data, 16, 8);

   110

	this->buffer[this->size++] = GB(data, 24, 8);

   111

}

   112

   113

void Packet::Send_uint64(uint64 data)

   114

{

   115

	assert(this->size < sizeof(this->buffer) - sizeof(data));

   116

	this->buffer[this->size++] = GB(data,  0, 8);

   117

	this->buffer[this->size++] = GB(data,  8, 8);

   118

	this->buffer[this->size++] = GB(data, 16, 8);

   119

	this->buffer[this->size++] = GB(data, 24, 8);

   120

	this->buffer[this->size++] = GB(data, 32, 8);

   121

	this->buffer[this->size++] = GB(data, 40, 8);

   122

	this->buffer[this->size++] = GB(data, 48, 8);

   123

	this->buffer[this->size++] = GB(data, 56, 8);

   124

}

   125

   126

/**

   127

 *  Sends a string over the network. It sends out

   128

 *  the string + '\0'. No size-byte or something.

   129

 * @param data   the string to send

   130

 */

   131

void Packet::Send_string(const char* data)

   132

{

   133

	assert(data != NULL);

   134

	/* The <= *is* valid due to the fact that we are comparing sizes and not the index. */

   135

	assert(this->size + strlen(data) + 1 <= sizeof(this->buffer));

   136

	while ((this->buffer[this->size++] = *data++) != '\0') {}

   137

}

   138

   139

   140

/**

   141

 * Receiving commands

   142

 * Again, the next couple of functions are endian-safe

   143

 *  see the comment before Send_bool for more info.

   144

 */

   145

   146

   147

/** Is it safe to read from the packet, i.e. didn't we run over the buffer ? */

   148

bool Packet::CanReadFromPacket(uint bytes_to_read)

   149

{

   150

	/* Don't allow reading from a quit client/client who send bad data */

   151

	if (this->cs->HasClientQuit()) return false;

   152

   153

	/* Check if variable is within packet-size */

   154

	if (this->pos + bytes_to_read > this->size) {

   155

		this->cs->CloseConnection();

   156

		return false;

   157

	}

   158

   159

	return true;

   160

}

   161

   162

/**

   163

 * Reads the packet size from the raw packet and stores it in the packet->size

   164

 */

   165

void Packet::ReadRawPacketSize()

   166

{

   167

	assert(this->cs != NULL && this->next == NULL);

   168

	this->size  = (PacketSize)this->buffer[0];

   169

	this->size += (PacketSize)this->buffer[1] << 8;

   170

}

   171

   172

/**

   173

 * Prepares the packet so it can be read

   174

 */

   175

void Packet::PrepareToRead()

   176

{

   177

	this->ReadRawPacketSize();

   178

   179

	/* Put the position on the right place */

   180

	this->pos = sizeof(PacketSize);

   181

}

   182

   183

bool Packet::Recv_bool()

   184

{

   185

	return this->Recv_uint8() != 0;

   186

}

   187

   188

uint8 Packet::Recv_uint8()

   189

{

   190

	uint8 n;

   191

   192

	if (!this->CanReadFromPacket(sizeof(n))) return 0;

   193

   194

	n = this->buffer[this->pos++];

   195

	return n;

   196

}

   197

   198

uint16 Packet::Recv_uint16()

   199

{

   200

	uint16 n;

   201

   202

	if (!this->CanReadFromPacket(sizeof(n))) return 0;

   203

   204

	n  = (uint16)this->buffer[this->pos++];

   205

	n += (uint16)this->buffer[this->pos++] << 8;

   206

	return n;

   207

}

   208

   209

uint32 Packet::Recv_uint32()

   210

{

   211

	uint32 n;

   212

   213

	if (!this->CanReadFromPacket(sizeof(n))) return 0;

   214

   215

	n  = (uint32)this->buffer[this->pos++];

   216

	n += (uint32)this->buffer[this->pos++] << 8;

   217

	n += (uint32)this->buffer[this->pos++] << 16;

   218

	n += (uint32)this->buffer[this->pos++] << 24;

   219

	return n;

   220

}

   221

   222

uint64 Packet::Recv_uint64()

   223

{

   224

	uint64 n;

   225

   226

	if (!this->CanReadFromPacket(sizeof(n))) return 0;

   227

   228

	n  = (uint64)this->buffer[this->pos++];

   229

	n += (uint64)this->buffer[this->pos++] << 8;

   230

	n += (uint64)this->buffer[this->pos++] << 16;

   231

	n += (uint64)this->buffer[this->pos++] << 24;

   232

	n += (uint64)this->buffer[this->pos++] << 32;

   233

	n += (uint64)this->buffer[this->pos++] << 40;

   234

	n += (uint64)this->buffer[this->pos++] << 48;

   235

	n += (uint64)this->buffer[this->pos++] << 56;

   236

	return n;

   237

}

   238

   239

/** Reads a string till it finds a '\0' in the stream */

   240

void Packet::Recv_string(char *buffer, size_t size)

   241

{

   242

	PacketSize pos;

   243

	char *bufp = buffer;

   244

   245

	/* Don't allow reading from a closed socket */

   246

	if (cs->HasClientQuit()) return;

   247

   248

	pos = this->pos;

   249

	while (--size > 0 && pos < this->size && (*buffer++ = this->buffer[pos++]) != '\0') {}

   250

   251

	if (size == 0 || pos == this->size) {

   252

		*buffer = '\0';

   253

		/* If size was sooner to zero then the string in the stream

   254

		 *  skip till the \0, so than packet can be read out correctly for the rest */

   255

		while (pos < this->size && this->buffer[pos] != '\0') pos++;

   256

		pos++;

   257

	}

   258

	this->pos = pos;

   259

   260

	str_validate(bufp);

   261

}

   262

   263

#endif /* ENABLE_NETWORK */