2020-05-24 14:15:49 -07:00
|
|
|
#include "configuration.h"
|
2021-06-27 10:56:28 -07:00
|
|
|
#include "CryptoEngine.h"
|
2020-05-24 14:15:49 -07:00
|
|
|
#include "ocrypto_aes_ctr.h"
|
2022-04-25 07:13:41 +02:00
|
|
|
// #include <Adafruit_nRFCrypto.h>
|
2020-05-09 21:02:56 -07:00
|
|
|
|
2020-05-24 14:15:49 -07:00
|
|
|
class NRF52CryptoEngine : public CryptoEngine
|
|
|
|
|
{
|
|
|
|
|
public:
|
|
|
|
|
NRF52CryptoEngine() {}
|
|
|
|
|
|
|
|
|
|
~NRF52CryptoEngine() {}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Encrypt a packet
|
|
|
|
|
*
|
|
|
|
|
* @param bytes is updated in place
|
|
|
|
|
*/
|
2022-03-20 11:40:13 +11:00
|
|
|
virtual void encrypt(uint32_t fromNode, uint64_t packetId, size_t numBytes, uint8_t *bytes) override
|
2020-05-24 14:15:49 -07:00
|
|
|
{
|
2022-04-25 07:13:41 +02:00
|
|
|
// DEBUG_MSG("NRF52 encrypt!\n");
|
2020-05-24 14:15:49 -07:00
|
|
|
|
2021-02-23 10:10:35 +08:00
|
|
|
if (key.length > 0) {
|
2020-05-24 14:15:49 -07:00
|
|
|
ocrypto_aes_ctr_ctx ctx;
|
|
|
|
|
|
2022-03-20 11:40:13 +11:00
|
|
|
initNonce(fromNode, packetId);
|
2021-02-23 10:10:35 +08:00
|
|
|
ocrypto_aes_ctr_init(&ctx, key.bytes, key.length, nonce);
|
2020-05-24 14:15:49 -07:00
|
|
|
|
|
|
|
|
ocrypto_aes_ctr_encrypt(&ctx, bytes, bytes, numBytes);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2022-03-20 11:40:13 +11:00
|
|
|
virtual void decrypt(uint32_t fromNode, uint64_t packetId, size_t numBytes, uint8_t *bytes) override
|
2020-05-24 14:15:49 -07:00
|
|
|
{
|
2022-04-25 07:13:41 +02:00
|
|
|
// DEBUG_MSG("NRF52 decrypt!\n");
|
2020-05-24 14:15:49 -07:00
|
|
|
|
2021-02-23 10:10:35 +08:00
|
|
|
if (key.length > 0) {
|
2020-05-24 14:15:49 -07:00
|
|
|
ocrypto_aes_ctr_ctx ctx;
|
|
|
|
|
|
2022-03-20 11:40:13 +11:00
|
|
|
initNonce(fromNode, packetId);
|
2021-02-23 10:10:35 +08:00
|
|
|
ocrypto_aes_ctr_init(&ctx, key.bytes, key.length, nonce);
|
2020-05-24 14:15:49 -07:00
|
|
|
|
|
|
|
|
ocrypto_aes_ctr_decrypt(&ctx, bytes, bytes, numBytes);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
};
|
|
|
|
|
|
2022-04-25 07:13:41 +02:00
|
|
|
// /**
|
|
|
|
|
// * Encrypt a packet
|
|
|
|
|
// *
|
|
|
|
|
// * @param bytes is updated in place
|
|
|
|
|
// */
|
|
|
|
|
// virtual void encrypt(uint32_t fromNode, uint64_t packetId, size_t numBytes, uint8_t *bytes) override
|
|
|
|
|
// {
|
|
|
|
|
// DEBUG_MSG("NRF52 encrypt!\n");
|
|
|
|
|
|
|
|
|
|
// if (key.length > 0) {
|
|
|
|
|
// nRFCrypto_AES ctx;
|
|
|
|
|
// uint8_t myLen = ctx.blockLen(numBytes);
|
|
|
|
|
// char encBuf[myLen] = {0};
|
|
|
|
|
// memcpy(encBuf, bytes, numBytes);
|
|
|
|
|
// initNonce(fromNode, packetId);
|
|
|
|
|
// nRFCrypto.begin();
|
|
|
|
|
// ctx.begin();
|
|
|
|
|
// ctx.Process(encBuf, numBytes, nonce, key.bytes, key.length, (char*)bytes, ctx.encryptFlag, ctx.ctrMode);
|
|
|
|
|
// ctx.end();
|
|
|
|
|
// nRFCrypto.end();
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
// virtual void decrypt(uint32_t fromNode, uint64_t packetId, size_t numBytes, uint8_t *bytes) override
|
|
|
|
|
// {
|
|
|
|
|
// DEBUG_MSG("NRF52 decrypt!\n");
|
|
|
|
|
|
|
|
|
|
// if (key.length > 0) {
|
|
|
|
|
// nRFCrypto_AES ctx;
|
|
|
|
|
// uint8_t myLen = ctx.blockLen(numBytes);
|
|
|
|
|
// char decBuf[myLen] = {0};
|
|
|
|
|
// memcpy(decBuf, bytes, numBytes);
|
|
|
|
|
// initNonce(fromNode, packetId);
|
|
|
|
|
// nRFCrypto.begin();
|
|
|
|
|
// ctx.begin();
|
|
|
|
|
// ctx.Process(decBuf, numBytes, nonce, key.bytes, key.length, (char*)bytes, ctx.decryptFlag, ctx.ctrMode);
|
|
|
|
|
// ctx.end();
|
|
|
|
|
// nRFCrypto.end();
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
// private:
|
|
|
|
|
// };
|
|
|
|
|
|
2020-05-24 14:15:49 -07:00
|
|
|
CryptoEngine *crypto = new NRF52CryptoEngine();
|