EncryptionKeyManager.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Async/Mutex.h"
#include "Async/UniqueLock.h"
#include "Containers/Map.h"
#include "Delegates/Delegate.h"
#include "Misc/AES.h"
#include "Misc/Guid.h"
 
#define UE_API CORE_API
 
namespace UE
{
 
class FEncryptionKeyManager
{
public:
    FEncryptionKeyManager(const FEncryptionKeyManager&) = delete;
    FEncryptionKeyManager(FEncryptionKeyManager&&) = delete;
    FEncryptionKeyManager& operator=(const FEncryptionKeyManager&) = delete;
    FEncryptionKeyManager& operator=(FEncryptionKeyManager&&) = delete;
    UE_API ~FEncryptionKeyManager();
 
    UE_API bool ContainsKey(const FGuid& Id) const;
    UE_API void AddKey(const FGuid& Id, const FAES::FAESKey& Key);
    UE_API bool TryGetKey(const FGuid& Id, FAES::FAESKey& OutKey) const;
    UE_API TMap<FGuid, FAES::FAESKey> GetAllKeys() const;
 
    DECLARE_MULTICAST_DELEGATE_TwoParams(FEncryptionKeyAddedDelegate, const FGuid&, const FAES::FAESKey&);
    FEncryptionKeyAddedDelegate& OnKeyAdded() { return KeyAdded; }
    static UE_API FEncryptionKeyManager& Get();
 
private:
    UE_API FEncryptionKeyManager();
 
    mutable FMutex Mutex;
    TMap<FGuid, FAES::FAESKey> Keys;
    FEncryptionKeyAddedDelegate KeyAdded;
};
 
} // namespace UE
 
#undef UE_API