Files
2025-09-18 17:55:52 -04:00

74 lines
2.2 KiB
C++

// Roblox: Moved here from RakNet 3.x
/// \file DataBlockEncryptor.h
/// \internal
/// \brief Encrypts and decrypts data blocks. Used as part of secure connections.
///
/// This file is part of RakNet Copyright 2003 Jenkins Software LLC
///
/// Usage of RakNet is subject to the appropriate license agreement.
#ifndef __DATA_BLOCK_ENCRYPTOR_H
#define __DATA_BLOCK_ENCRYPTOR_H
#include "Rijndael.h"
#include "RakMemoryOverride.h"
namespace RakNet
{
class RakNetRandom;
}
/// Encrypts and decrypts data blocks.
class DataBlockEncryptor
{
public:
// Constructor
DataBlockEncryptor();
// Destructor
~DataBlockEncryptor();
/// \return true if SetKey has been called previously
bool IsKeySet( void ) const;
/// \brief Set the encryption key
/// \param[in] key The new encryption key
void SetKey( const unsigned char key[ 16 ] );
/// \brief Unset the encryption key
void UnsetKey( void );
/// \brief Encryption adds 6 data bytes and then pads the number of bytes to be a multiple of 16.
/// \details 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, RakNet::RakNetRandom *rnr );
/// \brief Decryption removes bytes, as few as 6.
/// \details 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