329 lines
11 KiB
C
329 lines
11 KiB
C
/*
|
|
* Copyright 2014 Kevin M Greenan
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* Redistributions in binary form must reproduce the above copyright notice, this
|
|
* list of conditions and the following disclaimer in the documentation and/or
|
|
* other materials provided with the distribution. THIS SOFTWARE IS PROVIDED BY
|
|
* THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
|
|
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
|
|
* EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
|
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
|
|
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
|
* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
* jerasure_rs_vand backend implementation
|
|
*
|
|
* vi: set noai tw=79 ts=4 sw=4:
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "erasurecode.h"
|
|
#include "erasurecode_backend.h"
|
|
#include "erasurecode_helpers.h"
|
|
|
|
/* Forward declarations */
|
|
struct ec_backend_op_stubs jerasure_rs_vand_ops;
|
|
struct ec_backend jerasure_rs_vand;
|
|
|
|
typedef int* (*reed_sol_vandermonde_coding_matrix_func)(int, int, int);
|
|
typedef void (*jerasure_matrix_encode_func)(int, int, int, int*, char **, char **, int);
|
|
typedef int (*jerasure_matrix_decode_func)(int, int, int, int *, int, int*, char **, char **, int);
|
|
typedef int (*jerasure_make_decoding_matrix_func)(int, int, int, int *, int *, int *, int *);
|
|
typedef int * (*jerasure_erasures_to_erased_func)(int, int, int *);
|
|
typedef void (*jerasure_matrix_dotprod_func)(int, int, int *,int *, int,char **, char **, int);
|
|
|
|
struct jerasure_rs_vand_descriptor {
|
|
/* calls required for init */
|
|
reed_sol_vandermonde_coding_matrix_func reed_sol_vandermonde_coding_matrix;
|
|
|
|
/* calls required for encode */
|
|
jerasure_matrix_encode_func jerasure_matrix_encode;
|
|
|
|
/* calls required for decode */
|
|
jerasure_matrix_decode_func jerasure_matrix_decode;
|
|
|
|
/* calls required for reconstruct */
|
|
jerasure_make_decoding_matrix_func jerasure_make_decoding_matrix;
|
|
jerasure_erasures_to_erased_func jerasure_erasures_to_erased;
|
|
jerasure_matrix_dotprod_func jerasure_matrix_dotprod;
|
|
|
|
/* fields needed to hold state */
|
|
int *matrix;
|
|
int k;
|
|
int m;
|
|
int w;
|
|
};
|
|
|
|
static int jerasure_rs_vand_encode(void *desc, char **data, char **parity,
|
|
int blocksize)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc =
|
|
(struct jerasure_rs_vand_descriptor*) desc;
|
|
|
|
/* FIXME - make jerasure_matrix_encode return a value */
|
|
jerasure_desc->jerasure_matrix_encode(jerasure_desc->k, jerasure_desc->m,
|
|
jerasure_desc->w, jerasure_desc->matrix, data, parity, blocksize);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int jerasure_rs_vand_decode(void *desc, char **data, char **parity,
|
|
int *missing_idxs, int blocksize)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc =
|
|
(struct jerasure_rs_vand_descriptor*)desc;
|
|
|
|
/* FIXME - make jerasure_matrix_decode return a value */
|
|
jerasure_desc->jerasure_matrix_decode(jerasure_desc->k,
|
|
jerasure_desc->m, jerasure_desc->w,
|
|
jerasure_desc->matrix, 1, missing_idxs, data, parity, blocksize);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int jerasure_rs_vand_reconstruct(void *desc, char **data, char **parity,
|
|
int *missing_idxs, int destination_idx, int blocksize)
|
|
{
|
|
int ret = 1; /* return code */
|
|
int *decoding_row; /* decoding matrix row for decode */
|
|
int *erased = NULL; /* k+m length list of erased frag ids */
|
|
int *dm_ids = NULL; /* k length list of frag ids */
|
|
int *decoding_matrix = NULL; /* matrix for decoding */
|
|
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc =
|
|
(struct jerasure_rs_vand_descriptor*) desc;
|
|
|
|
dm_ids = (int *) alloc_zeroed_buffer(sizeof(int) * jerasure_desc->k);
|
|
decoding_matrix = (int *)
|
|
alloc_zeroed_buffer(sizeof(int*) * jerasure_desc->k * jerasure_desc->k);
|
|
if (NULL == decoding_matrix || NULL == dm_ids) {
|
|
goto out;
|
|
}
|
|
|
|
erased = jerasure_desc->jerasure_erasures_to_erased(jerasure_desc->k,
|
|
jerasure_desc->m, missing_idxs);
|
|
if (NULL == erased) {
|
|
goto out;
|
|
}
|
|
|
|
ret = jerasure_desc->jerasure_make_decoding_matrix(jerasure_desc->k,
|
|
jerasure_desc->m, jerasure_desc->w, jerasure_desc->matrix,
|
|
erased, decoding_matrix, dm_ids);
|
|
if (destination_idx < jerasure_desc->k) {
|
|
decoding_row = decoding_matrix + (destination_idx * jerasure_desc->k);
|
|
} else {
|
|
decoding_row = jerasure_desc->matrix +
|
|
((destination_idx - jerasure_desc->k) * jerasure_desc->k);
|
|
}
|
|
|
|
if (ret == 0) {
|
|
jerasure_desc->jerasure_matrix_dotprod(jerasure_desc->k,
|
|
jerasure_desc->w, decoding_row, dm_ids, destination_idx,
|
|
data, parity, blocksize);
|
|
} else {
|
|
goto out;
|
|
}
|
|
|
|
out:
|
|
free(erased);
|
|
free(decoding_matrix);
|
|
free(dm_ids);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static int jerasure_rs_vand_min_fragments(void *desc, int *missing_idxs,
|
|
int *fragments_to_exclude, int *fragments_needed)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc =
|
|
(struct jerasure_rs_vand_descriptor*)desc;
|
|
|
|
uint64_t exclude_bm = convert_list_to_bitmap(fragments_to_exclude);
|
|
uint64_t missing_bm = convert_list_to_bitmap(missing_idxs) | exclude_bm;
|
|
int i;
|
|
int j = 0;
|
|
int ret = -1;
|
|
|
|
for (i = 0; i < (jerasure_desc->k + jerasure_desc->m); i++) {
|
|
if (!(missing_bm & (1 << i))) {
|
|
fragments_needed[j] = i;
|
|
j++;
|
|
}
|
|
if (j == jerasure_desc->k) {
|
|
ret = 0;
|
|
fragments_needed[j] = -1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
#define DEFAULT_W 16
|
|
static void * jerasure_rs_vand_init(struct ec_backend_args *args,
|
|
void *backend_sohandle)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *desc = NULL;
|
|
|
|
desc = (struct jerasure_rs_vand_descriptor *)
|
|
malloc(sizeof(struct jerasure_rs_vand_descriptor));
|
|
if (NULL == desc) {
|
|
return NULL;
|
|
}
|
|
|
|
desc->k = args->uargs.k;
|
|
desc->m = args->uargs.m;
|
|
|
|
if (args->uargs.w <= 0)
|
|
args->uargs.w = DEFAULT_W;
|
|
|
|
/* store w back in args so upper layer can get to it */
|
|
desc->w = args->uargs.w;
|
|
|
|
/* validate EC arguments */
|
|
{
|
|
long long max_symbols;
|
|
if (desc->w != 8 && desc->w != 16 && desc->w != 32) {
|
|
goto error;
|
|
}
|
|
max_symbols = 1LL << desc->w;
|
|
if ((desc->k + desc->m) > max_symbols) {
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* ISO C forbids casting a void* to a function pointer.
|
|
* Since dlsym return returns a void*, we use this union to
|
|
* "transform" the void* to a function pointer.
|
|
*/
|
|
union {
|
|
reed_sol_vandermonde_coding_matrix_func initp;
|
|
jerasure_matrix_encode_func encodep;
|
|
jerasure_matrix_decode_func decodep;
|
|
jerasure_make_decoding_matrix_func decodematrixp;
|
|
jerasure_erasures_to_erased_func erasep;
|
|
jerasure_matrix_dotprod_func dotprodp;
|
|
void *vptr;
|
|
} func_handle = {.vptr = NULL};
|
|
|
|
|
|
/* fill in function addresses */
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "jerasure_matrix_encode");
|
|
desc->jerasure_matrix_encode = func_handle.encodep;
|
|
if (NULL == desc->jerasure_matrix_encode) {
|
|
goto error;
|
|
}
|
|
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "jerasure_matrix_decode");
|
|
desc->jerasure_matrix_decode = func_handle.decodep;
|
|
if (NULL == desc->jerasure_matrix_decode) {
|
|
goto error;
|
|
}
|
|
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "jerasure_make_decoding_matrix");
|
|
desc->jerasure_make_decoding_matrix = func_handle.decodematrixp;
|
|
if (NULL == desc->jerasure_make_decoding_matrix) {
|
|
goto error;
|
|
}
|
|
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "jerasure_matrix_dotprod");
|
|
desc->jerasure_matrix_dotprod = func_handle.dotprodp;
|
|
if (NULL == desc->jerasure_matrix_dotprod) {
|
|
goto error;
|
|
}
|
|
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "jerasure_erasures_to_erased");
|
|
desc->jerasure_erasures_to_erased = func_handle.erasep;
|
|
if (NULL == desc->jerasure_erasures_to_erased) {
|
|
goto error;
|
|
}
|
|
|
|
func_handle.vptr = NULL;
|
|
func_handle.vptr = dlsym(backend_sohandle, "reed_sol_vandermonde_coding_matrix");
|
|
desc->reed_sol_vandermonde_coding_matrix = func_handle.initp;
|
|
if (NULL == desc->reed_sol_vandermonde_coding_matrix) {
|
|
goto error;
|
|
}
|
|
|
|
desc->matrix = desc->reed_sol_vandermonde_coding_matrix(
|
|
desc->k, desc->m, desc->w);
|
|
if (NULL == desc->matrix) {
|
|
goto error;
|
|
}
|
|
|
|
return desc;
|
|
|
|
error:
|
|
free(desc);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/**
|
|
* Return the element-size, which is the number of bits stored
|
|
* on a given device, per codeword. For Vandermonde, this is
|
|
* 'w'. For somthing like cauchy, this is packetsize * w.
|
|
*
|
|
* Returns the size in bits!
|
|
*/
|
|
static int
|
|
jerasure_rs_vand_element_size(void* desc)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc =
|
|
(struct jerasure_rs_vand_descriptor*)desc;
|
|
|
|
/* Note that cauchy will return pyeclib_handle->w * PYECC_CAUCHY_PACKETSIZE * 8 */
|
|
return jerasure_desc->w;
|
|
}
|
|
|
|
static int jerasure_rs_vand_exit(void *desc)
|
|
{
|
|
struct jerasure_rs_vand_descriptor *jerasure_desc = NULL;
|
|
|
|
jerasure_desc = (struct jerasure_rs_vand_descriptor*) desc;
|
|
|
|
free(jerasure_desc);
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct ec_backend_op_stubs jerasure_rs_vand_op_stubs = {
|
|
.INIT = jerasure_rs_vand_init,
|
|
.EXIT = jerasure_rs_vand_exit,
|
|
.ENCODE = jerasure_rs_vand_encode,
|
|
.DECODE = jerasure_rs_vand_decode,
|
|
.FRAGSNEEDED = jerasure_rs_vand_min_fragments,
|
|
.RECONSTRUCT = jerasure_rs_vand_reconstruct,
|
|
.ELEMENTSIZE = jerasure_rs_vand_element_size,
|
|
};
|
|
|
|
struct ec_backend_common backend_jerasure_rs_vand = {
|
|
.id = EC_BACKEND_JERASURE_RS_VAND,
|
|
.name = "jerasure_rs_vand",
|
|
#if defined(__MACOS__) || defined(__MACOSX__) || defined(__OSX__) || defined(__APPLE__)
|
|
.soname = "libJerasure.dylib",
|
|
#else
|
|
.soname = "libJerasure.so",
|
|
#endif
|
|
.soversion = "2.0",
|
|
.ops = &jerasure_rs_vand_op_stubs,
|
|
};
|