mirror of
https://github.com/DarkflameUniverse/DarkflameServer
synced 2024-08-30 18:43:58 +00:00
72 lines
2.5 KiB
C
72 lines
2.5 KiB
C
|
/// \file
|
||
|
|
/// \brief \b [Internal] Encrypts and decrypts data blocks. Used as part of secure connections.
|
||
|
|
///
|
||
|
|
/// This file is part of RakNet Copyright 2003 Kevin Jenkins.
|
||
|
|
///
|
||
|
|
/// Usage of RakNet is subject to the appropriate license agreement.
|
||
|
|
/// Creative Commons Licensees are subject to the
|
||
|
|
/// license found at
|
||
|
|
/// http://creativecommons.org/licenses/by-nc/2.5/
|
||
|
|
/// Single application licensees are subject to the license found at
|
||
|
|
/// http://www.jenkinssoftware.com/SingleApplicationLicense.html
|
||
|
|
/// Custom license users are subject to the terms therein.
|
||
|
|
/// GPL license users are subject to the GNU General Public
|
||
|
|
/// License as published by the Free
|
||
|
|
/// Software Foundation; either version 2 of the License, or (at your
|
||
|
|
/// option) any later version.
|
||
|
|
|
||
|
|
#ifndef __DATA_BLOCK_ENCRYPTOR_H
|
||
|
|
#define __DATA_BLOCK_ENCRYPTOR_H
|
||
|
|
|
||
|
|
#include "Rijndael.h"
|
||
|
|
#include "RakMemoryOverride.h"
|
||
|
|
|
||
|
|
/// Encrypts and decrypts data blocks.
|
||
|
|
class DataBlockEncryptor : public RakNet::RakMemoryOverride
|
||
|
|
{
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
/// Constructor
|
||
|
|
DataBlockEncryptor();
|
||
|
|
|
||
|
|
/// Destructor
|
||
|
|
~DataBlockEncryptor();
|
||
|
|
|
||
|
|
/// \return true if SetKey has been called previously
|
||
|
|
bool IsKeySet( void ) const;
|
||
|
|
|
||
|
|
/// Set the encryption key
|
||
|
|
/// \param[in] key The new encryption key
|
||
|
|
void SetKey( const unsigned char key[ 16 ] );
|
||
|
|
|
||
|
|
/// Unset the encryption key
|
||
|
|
void UnsetKey( void );
|
||
|
|
|
||
|
|
/// Encryption adds 6 data bytes and then pads the number of bytes to be a multiple of 16. Output should be large enough to hold this.
|
||
|
|
/// Output can be the same memory block as input
|
||
|
|
/// \param[in] input the input buffer to encrypt
|
||
|
|
/// \param[in] inputLength the size of the @em input buffer
|
||
|
|
/// \param[in] output the output buffer to store encrypted data
|
||
|
|
/// \param[in] outputLength the size of the output buffer
|
||
|
|
void Encrypt( unsigned char *input, unsigned int inputLength, unsigned char *output, unsigned int *outputLength );
|
||
|
|
|
||
|
|
/// Decryption removes bytes, as few as 6. Output should be large enough to hold this.
|
||
|
|
/// Output can be the same memory block as input
|
||
|
|
/// \param[in] input the input buffer to decrypt
|
||
|
|
/// \param[in] inputLength the size of the @em input buffer
|
||
|
|
/// \param[in] output the output buffer to store decrypted data
|
||
|
|
/// \param[in] outputLength the size of the @em output buffer
|
||
|
|
/// \return False on bad checksum or input, true on success
|
||
|
|
bool Decrypt( unsigned char *input, unsigned int inputLength, unsigned char *output, unsigned int *outputLength );
|
||
|
|
|
||
|
|
protected:
|
||
|
|
|
||
|
|
keyInstance keyEncrypt;
|
||
|
|
keyInstance keyDecrypt;
|
||
|
|
cipherInstance cipherInst;
|
||
|
|
bool keySet;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|