mirror of
https://github.com/fergalmoran/ladybird.git
synced 2025-12-22 09:19:03 +00:00
107 lines
2.9 KiB
C++
107 lines
2.9 KiB
C++
/*
|
|
* Copyright (c) 2020, Ali Mohammad Pur <mpfard@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/ByteBuffer.h>
|
|
#include <AK/ByteString.h>
|
|
#include <LibCrypto/ASN1/DER.h>
|
|
|
|
namespace Crypto::PK {
|
|
|
|
template<typename ExportableKey>
|
|
ErrorOr<ByteBuffer> wrap_in_private_key_info(ExportableKey key, Span<int> algorithm_identifier)
|
|
requires requires(ExportableKey k) {
|
|
k.export_as_der();
|
|
}
|
|
{
|
|
ASN1::Encoder encoder;
|
|
TRY(encoder.write_constructed(ASN1::Class::Universal, ASN1::Kind::Sequence, [&]() -> ErrorOr<void> {
|
|
TRY(encoder.write(0x00u)); // version
|
|
|
|
// AlgorithmIdentifier
|
|
TRY(encoder.write_constructed(ASN1::Class::Universal, ASN1::Kind::Sequence, [&]() -> ErrorOr<void> {
|
|
TRY(encoder.write(algorithm_identifier)); // algorithm
|
|
|
|
// FIXME: This assumes we have a NULL parameter, this is not always the case
|
|
TRY(encoder.write(nullptr)); // parameters
|
|
|
|
return {};
|
|
}));
|
|
|
|
// PrivateKey
|
|
auto data = TRY(key.export_as_der());
|
|
TRY(encoder.write(data));
|
|
|
|
return {};
|
|
}));
|
|
|
|
return encoder.finish();
|
|
}
|
|
|
|
template<typename ExportableKey>
|
|
ErrorOr<ByteBuffer> wrap_in_subject_public_key_info(ExportableKey key, Span<int> algorithm_identifier)
|
|
requires requires(ExportableKey k) {
|
|
k.export_as_der();
|
|
}
|
|
{
|
|
ASN1::Encoder encoder;
|
|
TRY(encoder.write_constructed(ASN1::Class::Universal, ASN1::Kind::Sequence, [&]() -> ErrorOr<void> {
|
|
// AlgorithmIdentifier
|
|
TRY(encoder.write_constructed(ASN1::Class::Universal, ASN1::Kind::Sequence, [&]() -> ErrorOr<void> {
|
|
TRY(encoder.write(algorithm_identifier)); // algorithm
|
|
|
|
// FIXME: This assumes we have a NULL parameter, this is not always the case
|
|
TRY(encoder.write(nullptr)); // parameters
|
|
|
|
return {};
|
|
}));
|
|
|
|
// subjectPublicKey
|
|
auto data = TRY(key.export_as_der());
|
|
auto bitstring = ::Crypto::ASN1::BitStringView(data, 0);
|
|
TRY(encoder.write(bitstring));
|
|
|
|
return {};
|
|
}));
|
|
|
|
return encoder.finish();
|
|
}
|
|
|
|
// FIXME: Fixing name up for grabs
|
|
template<typename PrivKeyT, typename PubKeyT>
|
|
class PKSystem {
|
|
public:
|
|
using PublicKeyType = PubKeyT;
|
|
using PrivateKeyType = PrivKeyT;
|
|
|
|
PKSystem(PublicKeyType& pubkey, PrivateKeyType& privkey)
|
|
: m_public_key(pubkey)
|
|
, m_private_key(privkey)
|
|
{
|
|
}
|
|
|
|
PKSystem() = default;
|
|
|
|
virtual void encrypt(ReadonlyBytes in, Bytes& out) = 0;
|
|
virtual void decrypt(ReadonlyBytes in, Bytes& out) = 0;
|
|
|
|
virtual void sign(ReadonlyBytes in, Bytes& out) = 0;
|
|
virtual void verify(ReadonlyBytes in, Bytes& out) = 0;
|
|
|
|
virtual ByteString class_name() const = 0;
|
|
|
|
virtual size_t output_size() const = 0;
|
|
|
|
protected:
|
|
virtual ~PKSystem() = default;
|
|
|
|
PublicKeyType m_public_key;
|
|
PrivateKeyType m_private_key;
|
|
};
|
|
|
|
}
|