mirror of
https://github.com/jakcron/nstool.git
synced 2024-12-23 03:05:27 +00:00
92 lines
3.2 KiB
C++
92 lines
3.2 KiB
C++
#pragma once
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <crypto/sha.h>
|
|
|
|
namespace crypto
|
|
{
|
|
namespace rsa
|
|
{
|
|
const size_t kRsa1024Size = 0x80;
|
|
const size_t kRsa2048Size = 0x100;
|
|
const size_t kRsa4096Size = 0x200;
|
|
const size_t kRsaPublicExponentSize = 4;
|
|
|
|
enum RsaType
|
|
{
|
|
RSA_1024,
|
|
RSA_2048,
|
|
RSA_4096,
|
|
};
|
|
|
|
#pragma pack (push, 1)
|
|
struct sRsa1024Key
|
|
{
|
|
uint8_t modulus[kRsa1024Size];
|
|
uint8_t priv_exponent[kRsa1024Size];
|
|
uint8_t public_exponent[kRsaPublicExponentSize];
|
|
|
|
void operator=(const sRsa1024Key& other)
|
|
{
|
|
memcpy(this->modulus, modulus, kRsa1024Size);
|
|
memcpy(this->priv_exponent, priv_exponent, kRsa1024Size);
|
|
memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize);
|
|
}
|
|
|
|
bool operator==(const sRsa1024Key& other)
|
|
{
|
|
return memcmp(this->modulus, other.modulus, kRsa1024Size) == 0 && memcmp(this->priv_exponent, other.priv_exponent, kRsa1024Size) == 0 && memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize) == 0;
|
|
}
|
|
};
|
|
|
|
struct sRsa2048Key
|
|
{
|
|
uint8_t modulus[kRsa2048Size];
|
|
uint8_t priv_exponent[kRsa2048Size];
|
|
uint8_t public_exponent[kRsaPublicExponentSize];
|
|
|
|
void operator=(const sRsa2048Key& other)
|
|
{
|
|
memcpy(this->modulus, other.modulus, kRsa2048Size);
|
|
memcpy(this->priv_exponent, other.priv_exponent, kRsa2048Size);
|
|
memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize);
|
|
}
|
|
|
|
bool operator==(const sRsa2048Key& other)
|
|
{
|
|
return memcmp(this->modulus, other.modulus, kRsa2048Size) == 0 && memcmp(this->priv_exponent, other.priv_exponent, kRsa2048Size) == 0 && memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize) == 0;
|
|
}
|
|
};
|
|
|
|
struct sRsa4096Key
|
|
{
|
|
uint8_t modulus[kRsa4096Size];
|
|
uint8_t priv_exponent[kRsa4096Size];
|
|
uint8_t public_exponent[kRsaPublicExponentSize];
|
|
|
|
void operator=(const sRsa4096Key& other)
|
|
{
|
|
memcpy(this->modulus, other.modulus, kRsa4096Size);
|
|
memcpy(this->priv_exponent, other.priv_exponent, kRsa4096Size);
|
|
memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize);
|
|
}
|
|
|
|
bool operator==(const sRsa4096Key& other)
|
|
{
|
|
return memcmp(this->modulus, other.modulus, kRsa4096Size) == 0 && memcmp(this->priv_exponent, other.priv_exponent, kRsa4096Size) == 0 && memcpy(this->public_exponent, other.public_exponent, kRsaPublicExponentSize) == 0;
|
|
}
|
|
};
|
|
#pragma pack (pop)
|
|
|
|
// rsa1024
|
|
int RsaSign(const sRsa1024Key& key, sha::HashType hash_type, const uint8_t* hash, uint8_t signature[kRsa1024Size]);
|
|
int RsaVerify(const sRsa1024Key& key, sha::HashType hash_type, const uint8_t* hash, const uint8_t signature[kRsa1024Size]);
|
|
// rsa2048
|
|
int RsaSign(const sRsa2048Key& key, sha::HashType hash_type, const uint8_t* hash, uint8_t signature[kRsa2048Size]);
|
|
int RsaVerify(const sRsa2048Key& key, sha::HashType hash_type, const uint8_t* hash, const uint8_t signature[kRsa2048Size]);
|
|
// rsa4096
|
|
int RsaSign(const sRsa4096Key& key, sha::HashType hash_type, const uint8_t* hash, uint8_t signature[kRsa4096Size]);
|
|
int RsaVerify(const sRsa4096Key& key, sha::HashType hash_type, const uint8_t* hash, const uint8_t signature[kRsa4096Size]);
|
|
}
|
|
}
|