Repository navigation
Expand file tree
/
Copy pathfecpp.h
More file actions
85 lines (68 loc) · 1.79 KB
/
Copy pathfecpp.h
File metadata and controls
85 lines (68 loc) · 1.79 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
8677
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
/*
* Forward error correction based on Vandermonde matrices
*
* (C) 1997-1998 Luigi Rizzo (luigi@iet.unipi.it)
* (C) 2009 Jack Lloyd (jack@randombit.net)
*
* Distributed under the terms given in license.txt
*/
#ifndef FECPP_H_
#define FECPP_H_
#include <map>
#include <vector>
#include <functional>
#include <cstdint>
namespace fecpp {
using std::uint8_t;
using std::size_t;
using byte = std::uint8_t;
#if defined(__i386__) || defined(__x86_64__)
#define FECPP_IS_X86
#endif
/**
* Forward error correction code
*/
class fec_code
{
public:
/**
* fec_code constructor
* @param K the number of shares needed for recovery
* @param N the number of shares generated
*/
fec_code(size_t K, size_t n);
size_t get_K() const { return K; }
size_t get_N() const { return N; }
/**
* @param input the data to FEC
* @param size the length in bytes of input
* @param out the output callback
*/
void encode(
const uint8_t input[], size_t size,
std::function<void (size_t, size_t, const uint8_t[], size_t)> out)
const;
/**
* @param shares map of share id to share contents
* @param share_size size in bytes of each share
* @param out the output callback
*/
void decode(
const std::map<size_t, const uint8_t*>& shares, size_t share_size,
std::function<void (size_t, size_t, const uint8_t[], size_t)> out)
const;
private:
size_t K, N;
std::vector<uint8_t> enc_matrix;
};
#if defined(FECPP_IS_X86)
/**
* CPU runtime detection
*/
bool has_sse2();
bool has_ssse3();
size_t addmul_sse2(uint8_t z[], const uint8_t x[], uint8_t y, size_t size);
size_t addmul_ssse3(uint8_t z[], const uint8_t x[], uint8_t y, size_t size);
#endif
}
#endif