mirror of
https://github.com/amadvance/snapraid.git
synced 2026-10-26 10:51:54 -04:00
123 lines
2.3 KiB
C
123 lines
2.3 KiB
C
/*
|
|
* Copyright (C) 2013 Andrea Mazzoleni
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*/
|
|
|
|
#include "internal.h"
|
|
#include "memory.h"
|
|
|
|
void *raid_malloc_align(size_t size, void **freeptr)
|
|
{
|
|
unsigned char *ptr;
|
|
uintptr_t offset;
|
|
|
|
ptr = malloc(size + RAID_MALLOC_ALIGN);
|
|
if (!ptr)
|
|
return 0;
|
|
|
|
*freeptr = ptr;
|
|
|
|
offset = ((uintptr_t)ptr) % RAID_MALLOC_ALIGN;
|
|
|
|
if (offset != 0)
|
|
ptr += RAID_MALLOC_ALIGN - offset;
|
|
|
|
return ptr;
|
|
}
|
|
|
|
void **raid_malloc_vector(int nd, int n, size_t size, void **freeptr)
|
|
{
|
|
void **v;
|
|
unsigned char *va;
|
|
int i;
|
|
|
|
v = malloc(n * sizeof(void *));
|
|
if (!v)
|
|
return 0;
|
|
|
|
va = raid_malloc_align(n * (size + RAID_MALLOC_DISPLACEMENT), freeptr);
|
|
if (!va) {
|
|
free(v);
|
|
return 0;
|
|
}
|
|
|
|
for (i = 0; i < n; ++i) {
|
|
v[i] = va;
|
|
va += size + RAID_MALLOC_DISPLACEMENT;
|
|
}
|
|
|
|
/* reverse order of the data blocks */
|
|
/* because they are usually accessed from the last one */
|
|
for (i = 0; i < nd/2; ++i) {
|
|
void *ptr = v[i];
|
|
v[i] = v[nd - 1 - i];
|
|
v[nd - 1 - i] = ptr;
|
|
}
|
|
|
|
return v;
|
|
}
|
|
|
|
void raid_mrand_vector(int n, size_t size, void **vv)
|
|
{
|
|
unsigned char **v = (unsigned char **)vv;
|
|
int i;
|
|
size_t j;
|
|
|
|
for (i = 0; i < n; ++i)
|
|
for (j = 0; j < size; ++j)
|
|
v[i][j] = rand();
|
|
}
|
|
|
|
int raid_mtest_vector(int n, size_t size, void **vv)
|
|
{
|
|
unsigned char **v = (unsigned char **)vv;
|
|
int i;
|
|
size_t j;
|
|
unsigned k;
|
|
unsigned char d;
|
|
unsigned char p;
|
|
|
|
/* fill with 0 */
|
|
d = 0;
|
|
for (i = 0; i < n; ++i)
|
|
for (j = 0; j < size; ++j)
|
|
v[i][j] = d;
|
|
|
|
/* test with all the byte patterns */
|
|
for (k = 1; k < 256; ++k) {
|
|
p = d;
|
|
d = k;
|
|
|
|
/* forward fill */
|
|
for (i = 0; i < n; ++i) {
|
|
for (j = 0; j < size; ++j) {
|
|
if (v[i][j] != p)
|
|
return -1;
|
|
v[i][j] = d;
|
|
}
|
|
}
|
|
|
|
p = d;
|
|
d = ~p;
|
|
/* backward fill with complement */
|
|
for (i = 0; i < n; ++i) {
|
|
for (j = size; j > 0; --j) {
|
|
if (v[i][j-1] != p)
|
|
return -1;
|
|
v[i][j-1] = d;
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|