Mirror/aicodix___code
1
0
Fork 0
mirror of https://github.com/aicodix/code.git synced 2026-04-28 09:43:11 +00:00
Code Issues Projects Releases Packages Wiki Activity

added more asserts for block length

Browse source
This commit is contained in:
Ahmet Inan 2024-03-26 20:21:48 +01:00
commit c17ca2f674
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
cauchy_prime_field_erasure_coding.hh
View file

@ -11,6 +11,7 @@ namespace CODE {
template <typename PF, typename IO, int MAX_LEN> template <typename PF, typename IO, int MAX_LEN>
struct CauchyPrimeFieldErasureCoding struct CauchyPrimeFieldErasureCoding
{ {
static_assert(MAX_LEN < int(PF::P-1), "Block length must be smaller than largest field value");
PF temp[MAX_LEN]; PF temp[MAX_LEN];
bool used[PF::P]; bool used[PF::P];
PF row_num, row_den; PF row_num, row_den;
@ -114,7 +115,7 @@ struct CauchyPrimeFieldErasureCoding
int encode(const IO *data, IO *block, int block_id, int block_len, int block_cnt) int encode(const IO *data, IO *block, int block_id, int block_len, int block_cnt)
{ {
assert(block_id >= block_cnt && block_id < int(PF::P) / 2); assert(block_id >= block_cnt && block_id < int(PF::P) / 2);
assert(block_len < int(PF::P-1) && block_len <= MAX_LEN); assert(block_len <= MAX_LEN);
for (int k = 0; k < block_cnt; k++) { for (int k = 0; k < block_cnt; k++) {
PF a_ik = cauchy_matrix(block_id, k); PF a_ik = cauchy_matrix(block_id, k);
mac(data + block_len * k, a_ik, block_len, !k, k == block_cnt - 1); mac(data + block_len * k, a_ik, block_len, !k, k == block_cnt - 1);
@ -126,6 +127,7 @@ struct CauchyPrimeFieldErasureCoding
} }
void decode(IO *data, const IO *blocks, const IO *block_subs, const IO *block_ids, int block_idx, int block_len, int block_cnt) void decode(IO *data, const IO *blocks, const IO *block_subs, const IO *block_ids, int block_idx, int block_len, int block_cnt)
{ {
assert(block_len <= MAX_LEN);
for (int k = 0; k < block_cnt; k++) { for (int k = 0; k < block_cnt; k++) {
PF b_ik = inverse_cauchy_matrix(block_ids, block_idx, k, block_cnt); PF b_ik = inverse_cauchy_matrix(block_ids, block_idx, k, block_cnt);
mac_sub(data, blocks + block_len * k, b_ik, block_subs[k], block_len, !k, k == block_cnt - 1); mac_sub(data, blocks + block_len * k, b_ik, block_subs[k], block_len, !k, k == block_cnt - 1);