mirror of
https://github.com/henrydcase/pqc.git
synced 2024-12-04 21:34:01 +00:00
173 lines
12 KiB
C
173 lines
12 KiB
C
|
#include <string.h>
|
||
|
|
|
||
|
|
#include "address.h"
|
||
|
|
#include "params.h"
|
||
|
|
#include "thashx8.h"
|
||
|
|
#include "utils.h"
|
||
|
|
|
||
|
|
#include "utilsx8.h"
|
||
|
|
|
||
|
|
/**
|
||
|
|
* For a given leaf index, computes the authentication path and the resulting
|
||
|
|
* root node using Merkle's TreeHash algorithm.
|
||
|
|
* Expects the layer and tree parts of the tree_addr to be set, as well as the
|
||
|
|
* tree type (i.e. PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_ADDR_TYPE_HASHTREE or PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_ADDR_TYPE_FORSTREE).
|
||
|
|
* Applies the offset idx_offset to indices before building addresses, so that
|
||
|
|
* it is possible to continue counting indices across trees.
|
||
|
|
*/
|
||
|
|
static void treehashx8(unsigned char *rootx8, unsigned char *auth_pathx8,
|
||
|
|
unsigned char *stackx8, unsigned int *heights,
|
||
|
|
const unsigned char *sk_seed, const unsigned char *pub_seed,
|
||
|
|
const uint32_t leaf_idx[8], uint32_t idx_offset[8],
|
||
|
|
uint32_t tree_height,
|
||
|
|
void (*gen_leafx8)(
|
||
|
|
unsigned char * /* leaf0 */,
|
||
|
|
unsigned char * /* leaf1 */,
|
||
|
|
unsigned char * /* leaf2 */,
|
||
|
|
unsigned char * /* leaf3 */,
|
||
|
|
unsigned char * /* leaf4 */,
|
||
|
|
unsigned char * /* leaf5 */,
|
||
|
|
unsigned char * /* leaf6 */,
|
||
|
|
unsigned char * /* leaf7 */,
|
||
|
|
const unsigned char * /* sk_seed */,
|
||
|
|
const unsigned char * /* pub_seed */,
|
||
|
|
uint32_t /* addr_idx0 */,
|
||
|
|
uint32_t /* addr_idx1 */,
|
||
|
|
uint32_t /* addr_idx2 */,
|
||
|
|
uint32_t /* addr_idx3 */,
|
||
|
|
uint32_t /* addr_idx4 */,
|
||
|
|
uint32_t /* addr_idx5 */,
|
||
|
|
uint32_t /* addr_idx6 */,
|
||
|
|
uint32_t /* addr_idx7 */,
|
||
|
|
const uint32_t[8] /* tree_addr */,
|
||
|
|
const hash_state * /* state_seeded */),
|
||
|
|
uint32_t tree_addrx8[8 * 8],
|
||
|
|
const hash_state *state_seeded) {
|
||
|
|
unsigned int offset = 0;
|
||
|
|
uint32_t idx;
|
||
|
|
uint32_t tree_idx;
|
||
|
|
unsigned int j;
|
||
|
|
|
||
|
|
for (idx = 0; idx < (uint32_t)(1 << tree_height); idx++) {
|
||
|
|
/* Add the next leaf node to the stack. */
|
||
|
|
gen_leafx8(stackx8 + 0 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 1 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 2 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 3 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 4 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 5 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 6 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 7 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + offset * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
sk_seed, pub_seed,
|
||
|
|
idx + idx_offset[0],
|
||
|
|
idx + idx_offset[1],
|
||
|
|
idx + idx_offset[2],
|
||
|
|
idx + idx_offset[3],
|
||
|
|
idx + idx_offset[4],
|
||
|
|
idx + idx_offset[5],
|
||
|
|
idx + idx_offset[6],
|
||
|
|
idx + idx_offset[7],
|
||
|
|
tree_addrx8,
|
||
|
|
state_seeded);
|
||
|
|
offset++;
|
||
|
|
heights[offset - 1] = 0;
|
||
|
|
|
||
|
|
/* If this is a node we need for the auth path.. */
|
||
|
|
for (j = 0; j < 8; j++) {
|
||
|
|
if ((leaf_idx[j] ^ 0x1) == idx) {
|
||
|
|
memcpy(auth_pathx8 + j * tree_height * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + j * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N, PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* While the top-most nodes are of equal height.. */
|
||
|
|
while (offset >= 2 && heights[offset - 1] == heights[offset - 2]) {
|
||
|
|
/* Compute index of the new node, in the next layer. */
|
||
|
|
tree_idx = (idx >> (heights[offset - 1] + 1));
|
||
|
|
|
||
|
|
/* Set the address of the node we're creating. */
|
||
|
|
for (j = 0; j < 8; j++) {
|
||
|
|
PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_set_tree_height(tree_addrx8 + j * 8, heights[offset - 1] + 1);
|
||
|
|
PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_set_tree_index(tree_addrx8 + j * 8,
|
||
|
|
tree_idx + (idx_offset[j] >> (heights[offset - 1] + 1)));
|
||
|
|
}
|
||
|
|
/* Hash the top-most nodes from the stack together. */
|
||
|
|
PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_thashx8_2(stackx8 + 0 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 1 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 2 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 3 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 4 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 5 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 6 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 7 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 0 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 1 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 2 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 3 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 4 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 5 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 6 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + 7 * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 2)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
pub_seed, tree_addrx8, state_seeded);
|
||
|
|
offset--;
|
||
|
|
/* Note that the top-most node is now one layer higher. */
|
||
|
|
heights[offset - 1]++;
|
||
|
|
|
||
|
|
/* If this is a node we need for the auth path.. */
|
||
|
|
for (j = 0; j < 8; j++) {
|
||
|
|
if (((leaf_idx[j] >> heights[offset - 1]) ^ 0x1) == tree_idx) {
|
||
|
|
memcpy(auth_pathx8 + j * tree_height * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + heights[offset - 1]*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N,
|
||
|
|
stackx8 + j * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N + (offset - 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N, PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for (j = 0; j < 8; j++) {
|
||
|
|
memcpy(rootx8 + j * PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N, stackx8 + j * (tree_height + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N, PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* The wrappers below ensure we used fixed-size buffers on the stack (no VLAs) */
|
||
|
|
|
||
|
|
|
||
|
|
#define treehashx8_variant(name, size) \
|
||
|
|
void PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_treehashx8_##name( \
|
||
|
|
unsigned char *rootx8, unsigned char *auth_pathx8, \
|
||
|
|
const unsigned char *sk_seed, const unsigned char *pub_seed, \
|
||
|
|
const uint32_t leaf_idx[8], uint32_t idx_offset[8], \
|
||
|
|
void (*gen_leafx8)( \
|
||
|
|
unsigned char* /* leaf0 */, \
|
||
|
|
unsigned char* /* leaf1 */, \
|
||
|
|
unsigned char* /* leaf2 */, \
|
||
|
|
unsigned char* /* leaf3 */, \
|
||
|
|
unsigned char* /* leaf4 */, \
|
||
|
|
unsigned char* /* leaf5 */, \
|
||
|
|
unsigned char* /* leaf6 */, \
|
||
|
|
unsigned char* /* leaf7 */, \
|
||
|
|
const unsigned char* /* sk_seed */, \
|
||
|
|
const unsigned char* /* pub_seed */, \
|
||
|
|
uint32_t /* addr_idx0 */, \
|
||
|
|
uint32_t /* addr_idx1 */, \
|
||
|
|
uint32_t /* addr_idx2 */, \
|
||
|
|
uint32_t /* addr_idx3 */, \
|
||
|
|
uint32_t /* addr_idx4 */, \
|
||
|
|
uint32_t /* addr_idx5 */, \
|
||
|
|
uint32_t /* addr_idx6 */, \
|
||
|
|
uint32_t /* addr_idx7 */, \
|
||
|
|
const uint32_t[8] /* tree_addr */, \
|
||
|
|
const hash_state* /* state_seeded */), \
|
||
|
|
uint32_t tree_addrx8[8*8], \
|
||
|
|
const hash_state *state_seeded) \
|
||
|
|
{ \
|
||
|
|
const uint32_t tree_height = (size); \
|
||
|
|
unsigned char stackx8[8*((size) + 1)*PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_N]; \
|
||
|
|
unsigned int heights[(size) + 1]; \
|
||
|
|
treehashx8(rootx8, auth_pathx8, stackx8, heights, sk_seed, pub_seed, \
|
||
|
|
leaf_idx, idx_offset, tree_height, gen_leafx8, tree_addrx8, state_seeded); \
|
||
|
|
}
|
||
|
|
|
||
|
|
treehashx8_variant(FORS_HEIGHT, PQCLEAN_SPHINCSSHA256192FSIMPLE_AVX2_FORS_HEIGHT)
|
||
|
|
|
||
|
|
#undef treehashx8_variant
|