You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
c532d4e15b
#Jira: UENET-911, UENET-888, UENET-887 #rb: john.barrett, ryan.gerleve [CL 4875396 by Jake Leonard in Dev-Networking branch]
142 lines
4.1 KiB
C++
142 lines
4.1 KiB
C++
// Copyright 1998-2019 Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "PacketHandler.h"
|
|
|
|
THIRD_PARTY_INCLUDES_START
|
|
#include "CryptoPP/5.6.5/include/rsa.h"
|
|
#include "CryptoPP/5.6.5/include/osrng.h"
|
|
THIRD_PARTY_INCLUDES_END
|
|
|
|
/* RSA Encryptor Module Interface */
|
|
class FRSAEncryptorHandlerComponentModuleInterface : public FPacketHandlerComponentModuleInterface
|
|
{
|
|
public:
|
|
/* Creates an instance of this component */
|
|
virtual TSharedPtr<HandlerComponent> CreateComponentInstance(FString& Options) override;
|
|
};
|
|
|
|
|
|
namespace ERSAEncryptionHandler
|
|
{
|
|
enum State
|
|
{
|
|
UnInitialized,
|
|
InitializedLocalKeysSentLocal,
|
|
Initialized
|
|
};
|
|
}
|
|
|
|
/*
|
|
* Asymmetric block cipher handler component using RSA as the cipher
|
|
*/
|
|
class RSAENCRYPTIONHANDLERCOMPONENT_API RSAEncryptionHandlerComponent : public HandlerComponent
|
|
{
|
|
public:
|
|
/*
|
|
* Initializes default data, can specify key size in bits, increasing the key size
|
|
* will increase the fixed cipher text and max plain text sizes
|
|
*/
|
|
RSAEncryptionHandlerComponent(int32 InKeySizeInBits = 1024);
|
|
|
|
virtual void CountBytes(FArchive& Ar) const override;
|
|
|
|
/* Initializes the handler component */
|
|
virtual void Initialize() override;
|
|
|
|
virtual void NotifyHandshakeBegin() override;
|
|
|
|
/* Whether the handler component is valid */
|
|
virtual bool IsValid() const override;
|
|
|
|
/* Handles any incoming packets */
|
|
virtual void Incoming(FBitReader& Packet) override;
|
|
|
|
/* Handles any outgoing packets */
|
|
virtual void Outgoing(FBitWriter& Packet, FOutPacketTraits& Traits) override;
|
|
|
|
virtual void IncomingConnectionless(const TSharedPtr<const FInternetAddr>& Address, FBitReader& Packet) override
|
|
{
|
|
}
|
|
|
|
virtual void OutgoingConnectionless(const TSharedPtr<const FInternetAddr>& Address, FBitWriter& Packet, FOutPacketTraits& Traits) override
|
|
{
|
|
}
|
|
|
|
virtual int32 GetReservedPacketBits() const override
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
protected:
|
|
/* Set the state of the handler */
|
|
void SetState(ERSAEncryptionHandler::State State);
|
|
|
|
/* Encrypt outgoing packets */
|
|
virtual void Encrypt(FBitWriter& Packet);
|
|
|
|
/* Decrypt incoming packets */
|
|
virtual void Decrypt(FBitReader& Packet);
|
|
|
|
/* Pack the local key into a packet */
|
|
virtual void PackLocalKey(FBitWriter& Packet);
|
|
|
|
/* Unpack the remote key from a packet */
|
|
virtual void UnPackRemoteKey(FBitReader& Packet);
|
|
|
|
/* Encrypt outgoing packet with remote connection's public key */
|
|
virtual void EncryptWithRemotePublic(const TArray<uint8>& InPlainText, TArray<uint8>& OutCipherText);
|
|
|
|
/* Decrypt incoming packet with local private key */
|
|
virtual void DecryptWithPrivate(const TArray<uint8>& InCipherText, TArray<uint8>& OutPlainText);
|
|
|
|
/* Save the public key's modulus in the provided array */
|
|
void SavePublicKeyModulus(TArray<uint8>& OutModulus);
|
|
|
|
/* Save the public key's exponent in the provided array */
|
|
void SavePublicKeyExponent(TArray<uint8>& OutExponent);
|
|
|
|
/** Maximum plain text length that can be encrypted with private key */
|
|
uint32 PrivateKeyMaxPlaintextLength;
|
|
|
|
/** Fixed cipher text length that will result of private key encryption */
|
|
uint32 PrivateKeyFixedCiphertextLength;
|
|
|
|
/** Maximum plain text length that can be encrypted with remote public key */
|
|
uint32 RemotePublicKeyMaxPlaintextLength;
|
|
|
|
/** Fixed cipher text length that will be result of RemotePublicKey encryption */
|
|
uint32 RemotePublicKeyFixedCiphertextLength;
|
|
|
|
/** Size of Key in Bits */
|
|
uint32 KeySizeInBits;
|
|
|
|
/* State of the handler */
|
|
ERSAEncryptionHandler::State State;
|
|
|
|
/* Random number generator used to generate key */
|
|
CryptoPP::AutoSeededRandomPool Rng;
|
|
|
|
/* RSA parameters for generating key */
|
|
CryptoPP::InvertibleRSAFunction Params;
|
|
|
|
/* Encryptor for encrypting using remote's public key */
|
|
CryptoPP::RSAES_OAEP_SHA_Encryptor RemotePublicEncryptor;
|
|
|
|
/* Encryptor for encrypting with local private key */
|
|
CryptoPP::RSAES_OAEP_SHA_Encryptor PrivateEncryptor;
|
|
|
|
/* Decryptor for decrypting with local private key */
|
|
CryptoPP::RSAES_OAEP_SHA_Decryptor PrivateDecryptor;
|
|
|
|
/* Local Public Key */
|
|
CryptoPP::RSA::PublicKey PublicKey;
|
|
|
|
/* Local Private Key */
|
|
CryptoPP::RSA::PrivateKey PrivateKey;
|
|
|
|
/* Remote Public Key */
|
|
CryptoPP::RSA::PublicKey RemotePublicKey;
|
|
};
|