cryptModule.h

Go to the documentation of this file.

 
#ifndef _CRYPTMODULE_h
#define _CRYPTMODULE_h
 
#if defined(ARDUINO) && ARDUINO >= 100
#include "Arduino.h"
#else
#include "WProgram.h"
#endif
#include "EnigmaIoTconfig.h"
 
#define CRYPTMODULE_DEBUG_TAG "CryptModule"
 
#ifdef ESP8266
#define RANDOM_32 0x3FF20E44
#endif
 
const uint8_t RANDOM_LENGTH = sizeof (uint32_t); 
const uint8_t CRC_LENGTH = sizeof (uint32_t); 
 
class CryptModule {
public:
    static uint32_t random ();
 
    static uint32_t random (uint32_t max, uint32_t min = 0) {
        uint32_t _max, _min;
 
        if (max > min) {
            _max = max;
            _min = min;
        } else {
            _max = min;
            _min = max;
        }
        if (_max != _min) {
            return _min + (random () % (_max - _min));
        } else {
            return _min;
        }
    }
 
    static uint8_t* random (const uint8_t* buf, size_t len);
 
    static bool decryptBuffer (const uint8_t* data, size_t length,
                               const uint8_t* iv, uint8_t ivlen, const uint8_t* key, uint8_t keylen,
                               const uint8_t* aad, uint8_t aadLen, const uint8_t* tag, uint8_t tagLen);
 
    static uint8_t* getSHA256 (uint8_t* buffer, uint8_t length);
 
    static bool encryptBuffer (const uint8_t* data, size_t length,
                               const uint8_t* iv, uint8_t ivlen, const uint8_t* key, uint8_t keylen,
                               const uint8_t* aad, uint8_t aadLen, const uint8_t* tag, uint8_t tagLen);
 
    void getDH1 ();
 
    bool getDH2 (const uint8_t* remotePubKey);
 
    uint8_t* getPrivDHKey () {
        return privateDHKey;
    }
 
    uint8_t* getPubDHKey () {
        return publicDHKey;
    }
 
protected:
    uint8_t privateDHKey[KEY_LENGTH]; 
    uint8_t publicDHKey[KEY_LENGTH];  
};
 
extern CryptModule Crypto; 
 
#endif