cs_ConnectionEncryption.h

Go to the documentation of this file.

/*
 * Author: Crownstone Team
 * Copyright: Crownstone (https://crownstone.rocks)
 * Date: Oct 21, 2020
 * License: LGPLv3+, Apache License 2.0, and/or MIT (triple-licensed)
 */
 
#pragma once
 
#include <events/cs_EventListener.h>
#include <protocol/cs_Packets.h>
#include <structs/cs_PacketsInternal.h>
 
#define DEFAULT_VALIDATION_KEY 0xCAFEBABE
 
enum class ConnectionEncryptionType { CTR, ECB };
 
class ConnectionEncryption : EventListener {
public:
    static ConnectionEncryption& getInstance() {
        static ConnectionEncryption instance;
        return instance;
    }
 
    void init();
 
    cs_ret_code_t encrypt(
            cs_data_t input,
            cs_data_t output,
            EncryptionAccessLevel accessLevel,
            ConnectionEncryptionType encryptionType);
 
    cs_ret_code_t decrypt(
            cs_data_t input,
            cs_data_t output,
            EncryptionAccessLevel& accessLevel,
            ConnectionEncryptionType encryptionType);
 
    static cs_buffer_size_t getEncryptedBufferSize(
            cs_buffer_size_t plaintextBufferSize, ConnectionEncryptionType encryptionType);
 
    static cs_buffer_size_t getPlaintextBufferSize(
            cs_buffer_size_t encryptedBufferSize, ConnectionEncryptionType encryptionType);
 
    cs_ret_code_t setSessionData(session_data_t& sessionData);
 
    void disconnect();
 
    void handleEvent(event_t& event) override;
 
private:
    // This class is singleton, make constructor private.
    ConnectionEncryption();
 
    // This class is singleton, deny implementation
    ConnectionEncryption(ConnectionEncryption const&);
 
    // This class is singleton, deny implementation
    void operator=(ConnectionEncryption const&);
 
    session_data_t _sessionData;
 
    encryption_nonce_t _nonce;
 
    void generateSessionData();
};