mirror of
https://github.com/dashr9230/SA-MP.git
synced 2024-12-22 14:37:29 +08:00
[raknet] Implement/match all TEABlockEncryptor functions
* Update `ReliabilityLayer.h`
This commit is contained in:
RD42
parent
8ca8b43b86
commit
13e2d26935
@ -18,6 +18,9 @@
|
||||
#ifndef __RELIABILITY_LAYER_H
|
||||
#define __RELIABILITY_LAYER_H
|
||||
|
||||
// SAMPSRV
|
||||
#define TEA_ENCRYPTOR
|
||||
|
||||
#include "SocketLayer.h"
|
||||
#include "MTUSize.h"
|
||||
#include "DS_LinkedList.h"
|
||||
@ -27,7 +30,13 @@
|
||||
#include "BitStream.h"
|
||||
#include "InternalPacket.h"
|
||||
#include "InternalPacketPool.h"
|
||||
|
||||
#ifdef TEA_ENCRYPTOR
|
||||
#include "TEABlockEncryptor.h"
|
||||
#else
|
||||
#include "DataBlockEncryptor.h"
|
||||
#endif
|
||||
|
||||
#include "RakNetStatistics.h"
|
||||
#include "SHA1.h"
|
||||
#include "DS_OrderedList.h"
|
||||
@ -328,7 +337,12 @@ private:
|
||||
double availableBandwidth;
|
||||
bool continuousSend;
|
||||
|
||||
#ifdef TEA_ENCRYPTOR
|
||||
TEABlockEncryptor encryptor;
|
||||
#else
|
||||
DataBlockEncryptor encryptor;
|
||||
#endif
|
||||
|
||||
unsigned sendPacketCount, receivePacketCount;
|
||||
RakNetTimeNS ackTimeIncrement;
|
||||
|
||||
|
||||
239
raknet/TEABlockEncryptor.cpp
Normal file
239
raknet/TEABlockEncryptor.cpp
Normal file
@ -0,0 +1,239 @@
|
||||
#include "TEABlockEncryptor.h"
|
||||
#include "CheckSum.h"
|
||||
#include "Rand.h"
|
||||
#include <assert.h>
|
||||
#include <string.h>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <windows.h>
|
||||
#else
|
||||
#define OutputDebugString NULL
|
||||
#endif
|
||||
|
||||
#define TEA_ROUNDS 32
|
||||
#define TEA_XOR_MASK 0x5E94A3CF
|
||||
|
||||
unsigned int TEABlockEncryptor::initObsDelta = (0x9E3779B9 ^ TEA_XOR_MASK);
|
||||
|
||||
#ifdef _DEBUG
|
||||
void DumpMemory(const char *preText, const unsigned char *data, unsigned int count)
|
||||
{
|
||||
return;
|
||||
|
||||
char buffer[2000];
|
||||
char temp[5];
|
||||
sprintf(buffer, "%s: ", preText);
|
||||
for(unsigned int i=0; i<count; i++)
|
||||
{
|
||||
sprintf(temp, "%02x ", data[i]);
|
||||
strcat(buffer, temp);
|
||||
}
|
||||
strcat(buffer, "\n");
|
||||
OutputDebugString(buffer);
|
||||
}
|
||||
#endif
|
||||
|
||||
TEABlockEncryptor::TEABlockEncryptor()
|
||||
{
|
||||
initDelta = initObsDelta ^ TEA_XOR_MASK;
|
||||
initSum = initDelta * TEA_ROUNDS;
|
||||
|
||||
keySet = false;
|
||||
}
|
||||
|
||||
TEABlockEncryptor::~TEABlockEncryptor()
|
||||
{
|
||||
// Note: This is not virtual
|
||||
// Make it cascaded virtual before adding any code here!
|
||||
}
|
||||
|
||||
bool TEABlockEncryptor::IsKeySet( void ) const
|
||||
{
|
||||
return keySet;
|
||||
}
|
||||
|
||||
void TEABlockEncryptor::SetKey( const unsigned char key[ 16 ] )
|
||||
{
|
||||
#ifdef _DEBUG
|
||||
#ifdef SAMPSRV
|
||||
DumpMemory("SvKey", &key[0], 16);
|
||||
#else
|
||||
DumpMemory("ClKey", &key[0], 16);
|
||||
#endif
|
||||
#endif
|
||||
|
||||
key = key;
|
||||
keySet = true;
|
||||
}
|
||||
|
||||
void TEABlockEncryptor::UnsetKey( void )
|
||||
{
|
||||
keySet = false;
|
||||
}
|
||||
|
||||
void TEABlockEncryptor::Encrypt( unsigned char *input, int inputLength, unsigned char *output, int *outputLength )
|
||||
{
|
||||
unsigned char checkSum;
|
||||
unsigned int checkSumInt;
|
||||
unsigned char paddingBytes;
|
||||
unsigned char encodedPad;
|
||||
unsigned int totalLength;
|
||||
CheckSum checkSumCalculator;
|
||||
|
||||
#ifdef _DEBUG
|
||||
assert( keySet );
|
||||
#endif
|
||||
|
||||
assert( input && inputLength );
|
||||
|
||||
// # of padding bytes
|
||||
totalLength = inputLength + sizeof( checkSum ) + sizeof( encodedPad );
|
||||
paddingBytes = 0;
|
||||
|
||||
/*if (totalLength < 16)
|
||||
{
|
||||
paddingBytes = 16 - totalLength;
|
||||
}
|
||||
else
|
||||
*/
|
||||
if ((totalLength % 8) != 0)
|
||||
{
|
||||
paddingBytes = 8 - (totalLength % 8);
|
||||
}
|
||||
|
||||
*outputLength = totalLength + paddingBytes;
|
||||
|
||||
// Randomize the pad size variable
|
||||
encodedPad = (unsigned char) randomMT();
|
||||
encodedPad <<= 4;
|
||||
encodedPad |= paddingBytes;
|
||||
|
||||
// Write the data first, in case we are overwriting ourselves
|
||||
if ( input == output )
|
||||
memmove( output + sizeof( checkSum ) + sizeof( encodedPad ) + paddingBytes, input, inputLength );
|
||||
else
|
||||
memcpy( output + sizeof( checkSum ) + sizeof( encodedPad ) + paddingBytes, input, inputLength );
|
||||
|
||||
// Write the pad size variable
|
||||
*(output + sizeof( checkSum )) = encodedPad;
|
||||
|
||||
// Write the padding
|
||||
for (unsigned int index = 0; index < paddingBytes; index++ )
|
||||
*( output + sizeof( checkSum ) + sizeof( encodedPad ) + index ) = (unsigned char) randomMT();
|
||||
|
||||
// Calculate the checksum on the data
|
||||
checkSumCalculator.Add( output + sizeof( checkSum ), inputLength + sizeof( encodedPad ) + paddingBytes );
|
||||
|
||||
checkSumInt = checkSumCalculator.Get();
|
||||
checkSum = (unsigned char) ( (checkSumInt << 4) ^ (checkSumInt) );
|
||||
|
||||
// Write checksum
|
||||
*(output) = checkSum;
|
||||
|
||||
#ifdef _DEBUG
|
||||
DumpMemory("Pre Encrypt", output, *outputLength);
|
||||
#endif
|
||||
|
||||
for(int i=0; i < (*outputLength); i+=8)
|
||||
{
|
||||
unsigned int &V0 = *(reinterpret_cast<unsigned int*>(output+i));
|
||||
unsigned int &V1 = *(reinterpret_cast<unsigned int*>(output+i+sizeof(unsigned int)));
|
||||
EncryptBlock(V0, V1);
|
||||
//*(reinterpret_cast<unsigned int*>(output+i)+0) = V0;
|
||||
//*(reinterpret_cast<unsigned int*>(output+i)+1) = V1;
|
||||
}
|
||||
|
||||
#ifdef _DEBUG
|
||||
DumpMemory("PostEncrypt", output, *outputLength);
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
bool TEABlockEncryptor::Decrypt( unsigned char *input, int inputLength, unsigned char *output, int *outputLength )
|
||||
{
|
||||
unsigned char checkSum;
|
||||
unsigned int checkSumInt;
|
||||
unsigned char paddingBytes;
|
||||
unsigned char encodedPad;
|
||||
CheckSum checkSumCalculator;
|
||||
|
||||
#ifdef _DEBUG
|
||||
assert( keySet );
|
||||
#endif
|
||||
|
||||
if ( input == 0 || inputLength < 8 || ( inputLength % 8 ) != 0 )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
#ifdef _DEBUG
|
||||
DumpMemory("Pre Decrypt", input, inputLength);
|
||||
#endif
|
||||
|
||||
for(int i=0; i<inputLength; i+=8)
|
||||
{
|
||||
unsigned int &V0 = *(reinterpret_cast<unsigned int*>(input+i));
|
||||
unsigned int &V1 = *(reinterpret_cast<unsigned int*>(input+i+sizeof(unsigned int)));
|
||||
DecryptBlock(V0, V1);
|
||||
//*(reinterpret_cast<unsigned int*>(input+i)+0) = V0;
|
||||
//*(reinterpret_cast<unsigned int*>(input+i)+1) = V1;
|
||||
}
|
||||
|
||||
#ifdef _DEBUG
|
||||
DumpMemory("PostDecrypt", input, inputLength);
|
||||
#endif
|
||||
|
||||
// Read checksum
|
||||
checkSum = *(output);
|
||||
|
||||
// Read the pad size variable
|
||||
encodedPad = *(output + sizeof( checkSum ));
|
||||
|
||||
// Only care about the last 4 bits
|
||||
paddingBytes = encodedPad & 0x0F;
|
||||
|
||||
// Get the data length
|
||||
*outputLength = inputLength - sizeof( checkSum ) - sizeof( encodedPad ) - paddingBytes;
|
||||
|
||||
// Calculate the checksum on the data.
|
||||
checkSumCalculator.Add( input + sizeof( checkSum ), *outputLength + sizeof( encodedPad ) + paddingBytes );
|
||||
|
||||
checkSumInt = checkSumCalculator.Get();
|
||||
checkSumInt = ( (checkSumInt << 4) ^ (checkSumInt) ) & 0xFF;
|
||||
|
||||
if ( (unsigned char)checkSumInt != checkSum )
|
||||
return false;
|
||||
|
||||
// Read the data
|
||||
if ( input == output )
|
||||
memmove( output, input + sizeof( checkSum ) + sizeof( encodedPad ) + paddingBytes, *outputLength );
|
||||
else
|
||||
memcpy( output, input + sizeof( checkSum ) + sizeof( encodedPad ) + paddingBytes, *outputLength );
|
||||
|
||||
return true;
|
||||
|
||||
}
|
||||
|
||||
void TEABlockEncryptor::EncryptBlock(unsigned int &V0, unsigned int &V1)
|
||||
{
|
||||
unsigned int sum = 0;
|
||||
|
||||
for(unsigned int i=0; i<TEA_ROUNDS; i++)
|
||||
{
|
||||
V0 += ((V1 << 4 ^ V1 >> 5) + V1) ^ (sum + key[sum & 3]);
|
||||
sum += initDelta;
|
||||
V1 += ((V0 << 4 ^ V0 >> 5) + V0) ^ (sum + key[sum>>11 & 3]);
|
||||
}
|
||||
}
|
||||
|
||||
void TEABlockEncryptor::DecryptBlock(unsigned int &V0, unsigned int &V1)
|
||||
{
|
||||
unsigned int sum = initSum;
|
||||
|
||||
for(unsigned int i=0; i<TEA_ROUNDS; i++)
|
||||
{
|
||||
V1 -= ((V0 << 4 ^ V0 >> 5) + V0) ^ (sum + key[sum>>11 & 3]);
|
||||
sum -= initDelta;
|
||||
V0 -= ((V1 << 4 ^ V1 >> 5) + V1) ^ (sum + key[sum & 3]);
|
||||
}
|
||||
}
|
||||
36
raknet/TEABlockEncryptor.h
Normal file
36
raknet/TEABlockEncryptor.h
Normal file
@ -0,0 +1,36 @@
|
||||
|
||||
// Implements a TEA block encryptor for RakNet
|
||||
|
||||
#ifndef __TEA_BLOCK_ENCRYPTOR_H
|
||||
#define __TEA_BLOCK_ENCRYPTOR_H
|
||||
|
||||
#include "DataBlockEncryptor.h"
|
||||
|
||||
class TEABlockEncryptor :
|
||||
public DataBlockEncryptor
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
TEABlockEncryptor();
|
||||
~TEABlockEncryptor();
|
||||
|
||||
bool IsKeySet( void ) const;
|
||||
void SetKey( const unsigned char key[ 16 ] );
|
||||
void UnsetKey( void );
|
||||
void Encrypt( unsigned char *input, int inputLength, unsigned char *output, int *outputLength );
|
||||
bool Decrypt( unsigned char *input, int inputLength, unsigned char *output, int *outputLength );
|
||||
|
||||
protected:
|
||||
unsigned char key[ 16 ];
|
||||
unsigned int initSum;
|
||||
unsigned int initDelta;
|
||||
|
||||
static unsigned int initObsDelta;
|
||||
|
||||
void EncryptBlock(unsigned int &V0, unsigned int &V1);
|
||||
void DecryptBlock(unsigned int &V0, unsigned int &V1);
|
||||
|
||||
};
|
||||
|
||||
#endif
|
||||
@ -862,6 +862,12 @@
|
||||
<File
|
||||
RelativePath="..\raknet\TCPInterface.h">
|
||||
</File>
|
||||
<File
|
||||
RelativePath="..\raknet\TEABlockEncryptor.cpp">
|
||||
</File>
|
||||
<File
|
||||
RelativePath="..\raknet\TEABlockEncryptor.h">
|
||||
</File>
|
||||
<File
|
||||
RelativePath="..\raknet\TelnetTransport.cpp">
|
||||
</File>
|
||||
|
||||
Loading…
Reference in New Issue
Block a user