Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PQC: Classic McEliece #3883

Merged
merged 3 commits into from
Nov 21, 2024
Merged

Conversation

FAlbertDev
Copy link
Collaborator

@FAlbertDev FAlbertDev commented Jan 11, 2024

Pull Request Dependencies

Description

This PR relates to the Classic McEliece KEM as specified in this ISO draft. It also contains the instances defined in the NIST Round 4 submission. The test cases were generated using the NIST submissions reference implementation. Note that Classic McEliece (module cmce) is not the original McEliece Algorithm that is implemented in Botan's mce module. See the Classic McEliece homepage, for a brief comparison.

TODO Tracker

@coveralls
Copy link

coveralls commented Jan 11, 2024

Coverage Status

coverage: 91.243% (+0.2%) from 91.063%
when pulling c256e1c on Rohde-Schwarz:pqc/classic_mceliece
into e5ec408 on randombit:master.

@reneme reneme added this to the Botan 3.4.0 milestone Jan 11, 2024
@FAlbertDev FAlbertDev force-pushed the pqc/classic_mceliece branch 4 times, most recently from f73829f to e505390 Compare January 19, 2024 14:53
@atreiber94 atreiber94 force-pushed the pqc/classic_mceliece branch 2 times, most recently from 04b9314 to 190261d Compare January 19, 2024 15:59
@reneme reneme self-requested a review January 22, 2024 09:33
@FAlbertDev

This comment was marked as resolved.

@reneme reneme force-pushed the pqc/classic_mceliece branch 6 times, most recently from 7662592 to 2a3e60f Compare January 23, 2024 15:55
@reneme

This comment was marked as resolved.

Copy link
Collaborator

@reneme reneme left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This mostly looks at cmce_decaps.cpp, cmce_encaps.cpp and cmce.cpp, noting many minor code style things and a few suggestions for alternative code structuring. Not looking into the Classic McEliece specifics at all here.

src/build-data/oids.txt Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_types.h Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_encaps.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_encaps.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_encaps.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_encaps.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_encaps.cpp Outdated Show resolved Hide resolved
src/lib/utils/strong_type.h Outdated Show resolved Hide resolved
Copy link
Collaborator

@reneme reneme left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

First pass on cmce_field_orderings.cpp. I'm somewhat concerned that this isn't very efficient. But it may well be "good enough". Should we profile that?

src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_field_ordering.cpp Outdated Show resolved Hide resolved
Copy link
Collaborator

@reneme reneme left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

More comments. I didn't look at cmce_parameter_set.*, cmce_parameters_* and cmce_poly.*, yet.

src/lib/pubkey/classic_mceliece/cmce_keys_internal.cpp Outdated Show resolved Hide resolved
src/lib/utils/strong_type.h Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce.cpp Outdated Show resolved Hide resolved
Comment on lines 41 to 49
// TODO: Only for test instances. Remove on final PR
size_t m = Classic_McEliece_GF::log_q_from_mod(mod);

for(int i = static_cast<int>(m) - 2; i >= 0; --i) {
x ^= CT::Mask<uint32_t>::expand((uint32_t(1) << (i + m)) & x)
.if_set_return(static_cast<uint32_t>(mod.get()) << i);
}

return GF_Elem(static_cast<uint16_t>(x));
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Don't forget to remove (and probably replace by some exception?)

src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_matrix.cpp Outdated Show resolved Hide resolved
Comment on lines 247 to 295
Code_Word Classic_McEliece_Matrix::mul(const Classic_McEliece_Parameters& params, const Error_Vector& e) const {
auto s = e.subvector(0, params.pk_no_rows());
auto e_T = e.subvector(params.pk_no_rows());
auto pk_slicer = BufferSlicer(m_mat_bytes);

for(size_t i = 0; i < params.pk_no_rows(); ++i) {
auto pk_current_bytes = pk_slicer.take(params.pk_row_size_bytes());
auto row = secure_bitvector(pk_current_bytes, params.n() - params.pk_no_rows());
row &= e_T;
s.at(i) = s.at(i) ^ row.has_odd_hamming_weight();
}

BOTAN_ASSERT_NOMSG(pk_slicer.empty());
return s.as<Code_Word>();
}
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This produces quite a few copies and allocations. If that's in the implementation's hot path, we should probably want to have another look. Here's which (I believe cause allocations/copies):

  1. .subvector() always creates a new bitvector and copies the content into a newly allocated buffer,
  2. the c'tor of bitvector copies the passed-in buffer (particularly pk_current_bytes
  3. .as<> produces a copy of the bitvector with a new type

For (3): This could already be the right type using e.subvector<Code_Word>(), no?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some context: This function is called once per encapsulation. Also, e and s are pretty small, while m_mat_bytes is gigantic. Therefore, 1) and 3) are not critical; I'll apply your suggestion for 3) anyway. I don't know how we can prevent 2) without having something like a bitvector view or dropping the bitvector altogether.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think I'll have a timeboxed look into a bitvector that doesn't own its underlying storage. Perhaps that isn't too hard to achieve, especially when we can limit it to byte-aligned subvectors.

Copy link
Collaborator

@reneme reneme left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done with a first pass on the implementation. Don't get put off by the number of comments. Most are just C++ style nits and smaller programming suggestions.

That's really good work! 😃

Comment on lines 48 to 57
/// Reduced instances for side channel analysis (Self-created test instance with
/// m=8, n=128, t=8, f(z)=z^8+z^7+z^2+z+1, F(y)=y^8+y^4+y^3+y^2+1)
/// Minimal instance without semi-systematic matrix creation and no plaintext confirmation
test,
/// Minimal instance with semi-systematic matrix creation and no plaintext confirmation
testf,
/// Minimal instance without semi-systematic matrix creation and with plaintext confirmation
testpc,
/// Minimal instance with semi-systematic matrix creation and with plaintext confirmation
testpcf
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TODO: remove before merging

src/lib/pubkey/classic_mceliece/cmce_parameter_set.h Outdated Show resolved Hide resolved
Comment on lines 27 to 32
/**
* @returns ceil(n/d)
* TODO: Remove once LMS is merged
*/
constexpr size_t ceil_div(size_t n, size_t d) {
return (n + d - 1) / d;
}
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Re-evaluate before merging this.

src/lib/pubkey/classic_mceliece/cmce_parameters.h Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_parameters.h Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_poly.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_poly.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_poly.cpp Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/info.txt Outdated Show resolved Hide resolved
src/lib/utils/bit_ops.h Outdated Show resolved Hide resolved
@FAlbertDev
Copy link
Collaborator Author

Thanks a lot for your extensive review, @reneme! I addressed your suggestions and am optimistic that this PR is ready to drop its Draft status 🎉. Note that some suggestions that depend on other PRs are still open, which are not critical, though. Also, note that an extensive side-channel analysis is still in progress.

@FAlbertDev FAlbertDev marked this pull request as ready for review February 5, 2024 14:49
constexpr bitref& operator^=(bool other) noexcept { return assign(this->is_set() ^ other); }

private:
constexpr bitref& assign(bool bit) noexcept { return (bit) ? set() : unset(); }
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As discussed, DATA identified this line as a leakage during our SCA review.

Due to the ? operator, a control-flow difference is observed based on the boolean input bit variable.

The assign() routine is used within the push_back() routine, which in turn is used within the decode() routine. This may allow an adversary to observe the error vector e and, hence recover the shared secret.

We suggest to perform both the set() and unset() functions with the input as a mask, for example:

         private:
            constexpr bitref& assign(bool bit) noexcept {
                const block_type assign_mask = 0 - static_cast<block_type>(bit);
                this->m_block |=  (this->m_mask &  assign_mask);
                this->m_block &= ~(this->m_mask & ~assign_mask);
                return *this;
            }

In our case, this results in the following instructions - without any conditional branch based on the input:

[ ... ] 
                                         const block_type assign_mask = 0 - static_cast<block_type>(bit);
41d397: 41 f7 dc                   neg %r12d
[ ... ]
                                         this->m_block |= (this->m_mask & assign_mask);
41d3ab: 44 89 e1                   mov %r12d,%ecx
41d3ae: 21 c1                      and %eax,%ecx
                                         this->m_block &= ~(this->m_mask & ~assign_mask);
41d3b0: f7 d0                      not %eax
                                         this->m_block |= (this->m_mask & assign_mask);
41d3b2: 0a 0a                      or (%rdx),%cl
                                         this->m_block &= ~(this->m_mask & ~assign_mask);
41d3b4: 44 09 e0                   or %r12d,%eax
41d3b7: 21 c8                      and %ecx,%eax
[ ... ]

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for your analysis and your report! I applied your fix using Botan's constant-time helper class (7a0fe59)

@@ -1,6 +1,8 @@
/*
* An abstraction for an arbitrarily large bitvector that can
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@reneme the code still contains some artefacts for varying the underlying data type that we'd ideally want to get rid of.

@reneme
Copy link
Collaborator

reneme commented Oct 18, 2024

Rebased and resolved several conflicts after all other PQC algorithms got merged.

Copy link
Owner

@randombit randombit left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some review comments. Also I left some comments in July that never got responded too

@@ -419,6 +429,12 @@ std::unique_ptr<Private_Key> load_private_key(const AlgorithmIdentifier& alg_id,
}
#endif

#if defined(BOTAN_HAS_CLASSICMCELIECE) || defined(BOTAN_HAS_CLASSICMCELIECE)
if(alg_name.starts_with("mceliece")) {
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Uses ClassicMcEliece elsewhere we should be consistent

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We named the algorithm identifiers as defined in the specifications:

The OIDs were taken from the IETF Hackathon GitHub. IMO, we should stay with the same naming. But I agree that it is annoying that these are not named classic...

src/lib/pubkey/pk_algs.cpp Show resolved Hide resolved
src/build-data/oids.txt Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_gf.h Outdated Show resolved Hide resolved
src/lib/pubkey/classic_mceliece/cmce_decaps.cpp Outdated Show resolved Hide resolved

} // anonymous namespace

std::optional<Classic_McEliece_Field_Ordering> Classic_McEliece_Field_Ordering::create_field_ordering(
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does the non-constant time nature of std::optional bring up issues here? If so we should consider using CT::Option, if not a comment that it's fine would be helpful.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A std::optional should suffice. The field ordering creation is performed in a rejection sampling loop. I.e., if the result is a nullopt, all secrets are dismissed anyway and a new rejection sampling loop starts from the beginning.

/**
* Renders this bitvector into a sequence of "0"s and "1"s.
*/
std::string to_string() const {
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fine to leave it here

Copy link
Owner

@randombit randombit left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Finally made it through a complete review. Left one minor comment re side channels. Otherwise LGTM / ok to merge.

/// Swaps bit i with bit j in val
void swap_bits(uint64_t& val, size_t i, size_t j) {
uint64_t bit_i = (val >> i) & 1;
uint64_t bit_j = (val >> j) & 1;
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The single bits here are very vulnerable to Clang doing something smart/unpleasant - maybe add a CT::value_barrier

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good catch, this seems very sensible. I've also found a & 1 in cmce_gf, where I also applied a preventive value barrier.

Copy link
Collaborator Author

@FAlbertDev FAlbertDev Nov 20, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Interesting... If I apply the value barrier like:

uint64_t bit_i = CT::value_barrier( (val >> i) & 1 );
uint64_t bit_j = CT::value_barrier( (val >> j) & 1 );

valgrind reports a new side-channel. Instead, if I apply the value barrier like (or none at all):

uint64_t bit_i = (val >> i) & CT::value_barrier( 1 );
uint64_t bit_j = (val >> j) & CT::value_barrier( 1 );

it seems to work. It seems we should prevent the compiler from noticing something is bit-ANDed with 1 (in which case only two branches are possible). Side-channels are scary...

Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

😱

Copy link
Collaborator

@reneme reneme left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I didn't do an additional review, just reverting my "request changes"

@FAlbertDev
Copy link
Collaborator Author

I've applied the final suggestion and cleaned-up the history. I'll merge as soon as the CI is done. Thanks @randombit for your review!

- constant time conditional swap with mask
- floor_log2

Co-Authored-By: Amos Treiber <[email protected]>
@FAlbertDev FAlbertDev force-pushed the pqc/classic_mceliece branch 3 times, most recently from 2ddd2e6 to ed61330 Compare November 20, 2024 15:56
reneme and others added 2 commits November 21, 2024 10:33
This is an implementation of the Classic McEliece KEM according to the
NIST Round 4 submission and the ISO draft 20230419.

Co-Authored-By: Amos Treiber <[email protected]>
@FAlbertDev FAlbertDev merged commit c5ca299 into randombit:master Nov 21, 2024
38 checks passed
@FAlbertDev FAlbertDev deleted the pqc/classic_mceliece branch November 21, 2024 10:24
This was referenced Nov 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants