solana/programs/sbf/c/src/poseidon/poseidon.c

360 lines
12 KiB
C

/**
* @brief Poseidon syscall test
*/
#include <sol/assert.h>
#include <sol/poseidon.h>
#include <sol/string.h>
extern uint64_t entrypoint(const uint8_t *input) {
// Two inputs: ones and twos (big-endian).
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {13, 84, 225, 147, 143, 138, 140, 28, 125, 235, 94,
3, 85, 242, 99, 25, 32, 123, 132, 254, 156, 162,
206, 27, 38, 231, 53, 200, 41, 130, 25, 144};
uint8_t input1[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
uint8_t input2[] = {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input2, SOL_ARRAY_SIZE(input2)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// Two inputs: ones and twos (little-endian).
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {144, 25, 130, 41, 200, 53, 231, 38, 27, 206, 162,
156, 254, 132, 123, 32, 25, 99, 242, 85, 3, 94,
235, 125, 28, 140, 138, 143, 147, 225, 84, 13};
uint8_t input1[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
uint8_t input2[] = {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input2, SOL_ARRAY_SIZE(input2)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_LITTLE_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
uint8_t input1[] = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
};
// 1 input.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
41, 23, 97, 0, 234, 169, 98, 189, 193, 254, 108,
101, 77, 106, 60, 19, 14, 150, 164, 209, 22, 139,
51, 132, 139, 137, 125, 197, 2, 130, 1, 51,
};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 2 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
0, 122, 243, 70, 226, 211, 4, 39, 158, 121, 224,
169, 243, 2, 63, 119, 18, 148, 167, 138, 203, 112,
231, 63, 144, 175, 226, 124, 173, 64, 30, 129,
};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 3 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
2, 192, 6, 110, 16, 167, 42, 189, 43, 51, 195,
178, 20, 203, 62, 129, 188, 177, 182, 227, 9, 97,
205, 35, 194, 2, 177, 134, 115, 191, 37, 67,
};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 4 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
8, 44, 156, 55, 10, 13, 36, 244, 65, 111, 188,
65, 74, 55, 104, 31, 120, 68, 45, 39, 216, 99,
133, 153, 28, 23, 214, 252, 12, 75, 125, 113,
};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 5 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
16, 56, 150, 5, 174, 104, 141, 79, 20, 219, 133,
49, 34, 196, 125, 102, 168, 3, 199, 43, 65, 88,
156, 177, 191, 134, 135, 65, 178, 6, 185, 187,
};
const SolBytes inputs[] = {{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 6 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
42, 115, 246, 121, 50, 140, 62, 171, 114, 74, 163,
229, 189, 191, 80, 179, 144, 53, 215, 114, 159, 19,
91, 151, 9, 137, 15, 133, 197, 220, 94, 118,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 7 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
34, 118, 49, 10, 167, 243, 52, 58, 40, 66, 20,
19, 157, 157, 169, 89, 190, 42, 49, 178, 199, 8,
165, 248, 25, 84, 178, 101, 229, 58, 48, 184,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 8 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
23, 126, 20, 83, 196, 70, 225, 176, 125, 43, 66,
51, 66, 81, 71, 9, 92, 79, 202, 187, 35, 61,
35, 11, 109, 70, 162, 20, 217, 91, 40, 132,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 9 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
14, 143, 238, 47, 228, 157, 163, 15, 222, 235, 72,
196, 46, 187, 68, 204, 110, 231, 5, 95, 97, 251,
202, 94, 49, 59, 138, 95, 202, 131, 76, 71,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 10 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
46, 196, 198, 94, 99, 120, 171, 140, 115, 48, 133,
79, 74, 112, 119, 193, 255, 146, 96, 228, 72, 133,
196, 184, 29, 209, 49, 173, 58, 134, 205, 150,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 11 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
0, 113, 61, 65, 236, 166, 53, 241, 23, 212, 236,
188, 235, 95, 58, 102, 220, 65, 66, 235, 112, 181,
103, 101, 188, 53, 143, 27, 236, 64, 187, 155,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
// 12 inputs.
{
uint8_t result[POSEIDON_RESULT_LENGTH];
uint8_t expected[] = {
20, 57, 11, 224, 186, 239, 36, 155, 212, 124, 101,
221, 172, 101, 194, 229, 46, 133, 19, 192, 129, 193,
205, 114, 201, 128, 6, 9, 142, 154, 143, 190,
};
const SolBytes inputs[] = {
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)},
{input1, SOL_ARRAY_SIZE(input1)}, {input1, SOL_ARRAY_SIZE(input1)}};
sol_poseidon(
POSEIDON_PARAMETERS_BN254_X5,
POSEIDON_ENDIANNESS_BIG_ENDIAN,
inputs,
SOL_ARRAY_SIZE(inputs),
result
);
sol_assert(0 == sol_memcmp(result, expected, POSEIDON_RESULT_LENGTH));
}
return SUCCESS;
}