pull/63/head
Bryan Biedenkapp
2 years ago
parent
cb5329b2ca
commit
8ba9e5c846
@ -0,0 +1,100 @@
|
|||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
/*
|
||||||
|
* Digital Voice Modem - Common Library
|
||||||
|
* MIT Open Source. Use is subject to license terms.
|
||||||
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2024 Bryan Biedenkapp, N2PLL
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
#include "Defines.h"
|
||||||
|
#include "RC4Crypto.h"
|
||||||
|
#include "Log.h"
|
||||||
|
#include "Utils.h"
|
||||||
|
|
||||||
|
using namespace crypto;
|
||||||
|
|
||||||
|
#include <cassert>
|
||||||
|
#include <cstring>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Public Class Members
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
/* Initializes a new instance of the AES class. */
|
||||||
|
|
||||||
|
RC4::RC4() = default;
|
||||||
|
|
||||||
|
/* Encrypt/Decrypt input buffer with given key. */
|
||||||
|
|
||||||
|
uint8_t* RC4::crypt(const uint8_t in[], uint32_t inLen, const uint8_t key[], uint32_t keyLen)
|
||||||
|
{
|
||||||
|
uint8_t permutation[RC4_PERMUTATION_CNT];
|
||||||
|
::memset(permutation, 0x00U, RC4_PERMUTATION_CNT);
|
||||||
|
|
||||||
|
init(key, keyLen, permutation);
|
||||||
|
|
||||||
|
uint8_t* out = new uint8_t[inLen];
|
||||||
|
::memset(out, 0x00U, inLen);
|
||||||
|
|
||||||
|
transform(in, inLen, permutation, out);
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Private Class Members
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
/* */
|
||||||
|
|
||||||
|
void RC4::swap(uint8_t* a, uint8_t i1, uint8_t i2)
|
||||||
|
{
|
||||||
|
uint8_t temp = a[i1];
|
||||||
|
a[i1] = a[i2];
|
||||||
|
a[i2] = temp;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* */
|
||||||
|
|
||||||
|
void RC4::init(const uint8_t key[], uint8_t keyLen, uint8_t* permutation)
|
||||||
|
{
|
||||||
|
assert(permutation != nullptr);
|
||||||
|
|
||||||
|
// init state variable
|
||||||
|
for (uint32_t i = 0U; i < RC4_PERMUTATION_CNT; i++)
|
||||||
|
permutation[i] = (uint8_t)i;
|
||||||
|
|
||||||
|
m_i1 = 0U;
|
||||||
|
m_i2 = 0U;
|
||||||
|
|
||||||
|
// randomize, using key
|
||||||
|
for (int j = 0, i = 0; i < 256; i++) {
|
||||||
|
j = (j + permutation[i] + key[i % keyLen]) % RC4_PERMUTATION_CNT;
|
||||||
|
|
||||||
|
// swap permutation[i] and permutation[j]
|
||||||
|
swap(permutation, i, j);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* */
|
||||||
|
|
||||||
|
void RC4::transform(const uint8_t* input, uint32_t length, uint8_t* permutation, uint8_t* output)
|
||||||
|
{
|
||||||
|
assert(input != nullptr);
|
||||||
|
assert(output != nullptr);
|
||||||
|
|
||||||
|
uint32_t i = 0U, j = 0U;
|
||||||
|
for (; i < length; i++, j++) {
|
||||||
|
// update indices
|
||||||
|
m_i1 = (uint8_t)((m_i1 + 1) % RC4_PERMUTATION_CNT);
|
||||||
|
m_i2 = (uint8_t)((m_i2 + permutation[m_i1]) % RC4_PERMUTATION_CNT);
|
||||||
|
|
||||||
|
// swap permutation[m_i1] and permutation[m_i2]
|
||||||
|
swap(permutation, m_i1, m_i2);
|
||||||
|
|
||||||
|
// transform byte
|
||||||
|
uint8_t b = (uint8_t)((permutation[m_i1] + permutation[m_i2]) % RC4_PERMUTATION_CNT);
|
||||||
|
output[j] = (uint8_t)(input[i] ^ permutation[b]);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,68 @@
|
|||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
/*
|
||||||
|
* Digital Voice Modem - Common Library
|
||||||
|
* MIT Open Source. Use is subject to license terms.
|
||||||
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2024 Bryan Biedenkapp, N2PLL
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
/**
|
||||||
|
* @defgroup crypto Cryptography
|
||||||
|
* @brief Defines and implements cryptography routines.
|
||||||
|
* @ingroup common
|
||||||
|
*
|
||||||
|
* @file RC4Crypto.h
|
||||||
|
* @ingroup crypto
|
||||||
|
* @file RC4Crypto.cpp
|
||||||
|
* @ingroup crypto
|
||||||
|
*/
|
||||||
|
#if !defined(__RC4_CRYPTO_H__)
|
||||||
|
#define __RC4_CRYPTO_H__
|
||||||
|
|
||||||
|
#include "common/Defines.h"
|
||||||
|
|
||||||
|
namespace crypto
|
||||||
|
{
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Constants
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
const uint32_t RC4_PERMUTATION_CNT = 256;
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// Class Declaration
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Rivest Cipher 4 Algorithm.
|
||||||
|
* @ingroup crypto
|
||||||
|
*/
|
||||||
|
class HOST_SW_API RC4 {
|
||||||
|
public:
|
||||||
|
/**
|
||||||
|
* @brief Initializes a new instance of the RC4 class.
|
||||||
|
*/
|
||||||
|
explicit RC4();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Encrypt/Decrypt input buffer with given key.
|
||||||
|
* @param in Input buffer (if encrypted, will decrypt, if decrypted, will encrypt)
|
||||||
|
* @param inLen Input buffer length.
|
||||||
|
* @param key Encryption key.
|
||||||
|
* @param keyLen Encryption key length.
|
||||||
|
* @returns uint8_t* Encrypted input buffer.
|
||||||
|
*/
|
||||||
|
uint8_t* crypt(const uint8_t in[], uint32_t inLen, const uint8_t key[], uint32_t keyLen);
|
||||||
|
|
||||||
|
private:
|
||||||
|
uint32_t m_i1;
|
||||||
|
uint32_t m_i2;
|
||||||
|
|
||||||
|
void swap(uint8_t* a, uint8_t i1, uint8_t i2);
|
||||||
|
void init(const uint8_t key[], uint8_t keyLen, uint8_t* permutation);
|
||||||
|
void transform(const uint8_t* input, uint32_t length, uint8_t* permutation, uint8_t* output);
|
||||||
|
};
|
||||||
|
} // namespace crypto
|
||||||
|
|
||||||
|
#endif // __RC4_CRYPTO_H__
|
||||||
@ -0,0 +1,67 @@
|
|||||||
|
// SPDX-License-Identifier: GPL-2.0-only
|
||||||
|
/**
|
||||||
|
* Digital Voice Modem - Test Suite
|
||||||
|
* GPLv2 Open Source. Use is subject to license terms.
|
||||||
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||||
|
*
|
||||||
|
* @package DVM / Test Suite
|
||||||
|
* @license GPLv2 License (https://opensource.org/licenses/GPL-2.0)
|
||||||
|
*
|
||||||
|
* Copyright (C) 2024 Bryan Biedenkapp, N2PLL
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
#include "host/Defines.h"
|
||||||
|
#include "common/RC4Crypto.h"
|
||||||
|
#include "common/Log.h"
|
||||||
|
#include "common/Utils.h"
|
||||||
|
|
||||||
|
using namespace crypto;
|
||||||
|
|
||||||
|
#include <catch2/catch_test_macros.hpp>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <time.h>
|
||||||
|
|
||||||
|
TEST_CASE("RC4", "[Crypto Test]") {
|
||||||
|
SECTION("RC4_Crypto_Test") {
|
||||||
|
bool failed = false;
|
||||||
|
|
||||||
|
INFO("RC4 Crypto Test");
|
||||||
|
|
||||||
|
srand((unsigned int)time(NULL));
|
||||||
|
|
||||||
|
// key (K)
|
||||||
|
uint8_t K[8] =
|
||||||
|
{
|
||||||
|
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07
|
||||||
|
};
|
||||||
|
|
||||||
|
// message
|
||||||
|
uint8_t message[48] =
|
||||||
|
{
|
||||||
|
0x90, 0x56, 0x00, 0x00, 0x2D, 0x75, 0xE6, 0x8D, 0x00, 0x89, 0x69, 0xCF, 0x00, 0xFE, 0x00, 0x04,
|
||||||
|
0x4F, 0xC7, 0x60, 0xFF, 0x30, 0x3E, 0x2B, 0xAD, 0x00, 0x89, 0x69, 0xCF, 0x00, 0x00, 0x00, 0x08,
|
||||||
|
0x52, 0x50, 0x54, 0x4C, 0x00, 0x89, 0x69, 0xCF, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
|
||||||
|
};
|
||||||
|
|
||||||
|
// perform crypto
|
||||||
|
RC4* rc4 = new RC4();
|
||||||
|
|
||||||
|
Utils::dump(2U, "RC4_Crypto_Test, Message", message, 48);
|
||||||
|
|
||||||
|
uint8_t* crypted = rc4->crypt(message, 48 * sizeof(uint8_t), K, 8);
|
||||||
|
Utils::dump(2U, "RC4_Crypto_Test, Encrypted", crypted, 48);
|
||||||
|
|
||||||
|
uint8_t* decrypted = rc4->crypt(crypted, 48 * sizeof(uint8_t), K, 8);
|
||||||
|
Utils::dump(2U, "RC4_Crypto_Test, Decrypted", decrypted, 48);
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < 48U; i++) {
|
||||||
|
if (decrypted[i] != message[i]) {
|
||||||
|
::LogDebug("T", "RC4_Crypto_Test, INVALID AT IDX %d\n", i);
|
||||||
|
failed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
delete rc4;
|
||||||
|
REQUIRE(failed==false);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue