Work in progress global dedupe changes.

This commit is contained in:
Moinak Ghosh 2013-03-20 22:47:03 +05:30
parent f2806d4ffa
commit b7fdeb08bc
9 changed files with 126 additions and 64 deletions

View file

@ -30,6 +30,8 @@
#include <sys/types.h> #include <sys/types.h>
#include <stdint.h> #include <stdint.h>
#include <utils.h>
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif #endif
@ -58,27 +60,6 @@ extern "C" {
#define MAX_NONCE 32 #define MAX_NONCE 32
#define KECCAK_MAX_SEG (2305843009213693950ULL) #define KECCAK_MAX_SEG (2305843009213693950ULL)
/*
* Public checksum properties. CKSUM_MAX_BYTES must be updated if a
* newer larger checksum is added to the list.
*/
typedef enum {
CKSUM_CRC64 = 0x100,
CKSUM_BLAKE256 = 0x200,
CKSUM_BLAKE512 = 0x300,
CKSUM_SHA256 = 0x400,
CKSUM_SHA512 = 0x500,
CKSUM_KECCAK256 = 0x600,
CKSUM_KECCAK512 = 0x700,
/*
* Backwards compatibility options. SKEIN in release 1.2 was replaced with
* Blake2 from 1.3 onwards (for sheer speed of Blake2). We want to be able
* to decode archives created with 1.2. New archives do not use SKEIN.
*/
CKSUM_SKEIN256 = 0x800,
CKSUM_SKEIN512 = 0x900,
CKSUM_INVALID = 0
} cksum_t;
typedef struct { typedef struct {
void *crypto_ctx; void *crypto_ctx;

28
main.c
View file

@ -87,6 +87,7 @@ static int nthreads = 0;
static int hide_mem_stats = 1; static int hide_mem_stats = 1;
static int hide_cmp_stats = 1; static int hide_cmp_stats = 1;
static int enable_rabin_scan = 0; static int enable_rabin_scan = 0;
static int enable_rabin_global = 0;
static int enable_delta_encode = 0; static int enable_delta_encode = 0;
static int enable_delta2_encode = 0; static int enable_delta2_encode = 0;
static int enable_rabin_split = 1; static int enable_rabin_split = 1;
@ -770,6 +771,15 @@ start_decompress(const char *filename, const char *to_filename)
if (flags & FLAG_DEDUP) { if (flags & FLAG_DEDUP) {
enable_rabin_scan = 1; enable_rabin_scan = 1;
if (flags & FLAG_DEDUP_FIXED) {
if (version > 7) {
enable_rabin_global = 1;
} else {
fprintf(stderr, "Invalid file deduplication flags.\n");
err = 1;
goto uncomp_done;
}
}
} else if (flags & FLAG_DEDUP_FIXED) { } else if (flags & FLAG_DEDUP_FIXED) {
enable_fixed_scan = 1; enable_fixed_scan = 1;
} }
@ -1577,8 +1587,8 @@ start_compress(const char *filename, uint64_t chunksize, int level)
short version, flags; short version, flags;
struct stat sbuf; struct stat sbuf;
int compfd = -1, uncompfd = -1, err; int compfd = -1, uncompfd = -1, err;
int i, thread, bail, single_chunk; int thread, bail, single_chunk;
int nprocs, np, p; uint32_t i, nprocs, np, p;
struct cmp_data **dary = NULL, *tdat; struct cmp_data **dary = NULL, *tdat;
pthread_t writer_thr; pthread_t writer_thr;
uchar_t *cread_buf, *pos; uchar_t *cread_buf, *pos;
@ -1599,6 +1609,7 @@ start_compress(const char *filename, uint64_t chunksize, int level)
*/ */
compressed_chunksize = chunksize + CHUNK_HDR_SZ + zlib_buf_extra(chunksize); compressed_chunksize = chunksize + CHUNK_HDR_SZ + zlib_buf_extra(chunksize);
init_algo_props(&props); init_algo_props(&props);
props.cksum = cksum;
cread_buf = NULL; cread_buf = NULL;
if (_props_func) { if (_props_func) {
@ -1610,8 +1621,10 @@ start_compress(const char *filename, uint64_t chunksize, int level)
} }
flags = 0; flags = 0;
if (enable_rabin_scan || enable_fixed_scan) { if (enable_rabin_scan || enable_fixed_scan || enable_rabin_global) {
if (enable_rabin_scan) if (enable_rabin_global)
flags |= (FLAG_DEDUP | FLAG_DEDUP_FIXED);
else if (enable_rabin_scan)
flags |= FLAG_DEDUP; flags |= FLAG_DEDUP;
else else
flags |= FLAG_DEDUP_FIXED; flags |= FLAG_DEDUP_FIXED;
@ -1801,6 +1814,10 @@ start_compress(const char *filename, uint64_t chunksize, int level)
sem_init(&(tdat->start_sem), 0, 0); sem_init(&(tdat->start_sem), 0, 0);
sem_init(&(tdat->cmp_done_sem), 0, 0); sem_init(&(tdat->cmp_done_sem), 0, 0);
sem_init(&(tdat->write_done_sem), 0, 1); sem_init(&(tdat->write_done_sem), 0, 1);
sem_init(&(tdat->index_sem), 0, 0);
// i is unsigned so this wraps around backwards for i == 0
tdat->index_sem_other = &(dary[(i - 1) % nprocs]->index_sem);
if (_init_func) { if (_init_func) {
if (_init_func(&(tdat->data), &(tdat->level), props.nthreads, chunksize, VERSION, COMPRESS) != 0) { if (_init_func(&(tdat->data), &(tdat->level), props.nthreads, chunksize, VERSION, COMPRESS) != 0) {
COMP_BAIL; COMP_BAIL;
@ -1830,6 +1847,9 @@ start_compress(const char *filename, uint64_t chunksize, int level)
} }
thread = 1; thread = 1;
// When doing global dedupe first thread does not wait to access the index.
sem_post(&(dary[0]->index_sem));
w.dary = dary; w.dary = dary;
w.wfd = compfd; w.wfd = compfd;
w.nprocs = nprocs; w.nprocs = nprocs;

View file

@ -194,6 +194,8 @@ struct cmp_data {
sem_t start_sem; sem_t start_sem;
sem_t cmp_done_sem; sem_t cmp_done_sem;
sem_t write_done_sem; sem_t write_done_sem;
sem_t index_sem;
sem_t *index_sem_other;
void *data; void *data;
pthread_t thr; pthread_t thr;
mac_ctx_t chunk_hmac; mac_ctx_t chunk_hmac;

View file

@ -42,7 +42,8 @@
* Hashtable structures for in-memory index. * Hashtable structures for in-memory index.
*/ */
typedef struct _hash_entry { typedef struct _hash_entry {
uint64_t seg_offset; uint64_t item_offset;
uint32_t item_size;
struct _hash_entry *next; struct _hash_entry *next;
uchar_t cksum[1]; uchar_t cksum[1];
} hash_entry_t; } hash_entry_t;
@ -53,7 +54,6 @@ typedef struct {
typedef struct { typedef struct {
htab_t *list; htab_t *list;
pthread_mutex_t *mlist;
uint64_t memlimit; uint64_t memlimit;
uint64_t memused; uint64_t memused;
int hash_entry_size, intervals, hash_slots; int hash_entry_size, intervals, hash_slots;
@ -91,8 +91,6 @@ static cleanup_indx(index_t *indx)
} }
free(indx->list); free(indx->list);
} }
if (indx->mlist)
free(indx->mlist);
free(indx); free(indx);
} }
} }
@ -153,7 +151,6 @@ init_global_db_s(char *path, char *tmppath, uint32_t chunksize, uint64_t user_ch
indx->memlimit = memlimit; indx->memlimit = memlimit;
indx->list = (htab_t *)calloc(intervals, sizeof (htab_t)); indx->list = (htab_t *)calloc(intervals, sizeof (htab_t));
indx->mlist = (pthread_mutex_t *)malloc(intervals * sizeof (pthread_mutex_t));
indx->hash_entry_size = hash_entry_size; indx->hash_entry_size = hash_entry_size;
indx->intervals = intervals; indx->intervals = intervals;
indx->hash_slots = hash_slots / intervals; indx->hash_slots = hash_slots / intervals;
@ -167,9 +164,9 @@ init_global_db_s(char *path, char *tmppath, uint32_t chunksize, uint64_t user_ch
return (NULL); return (NULL);
} }
indx->memused += ((hash_slots / intervals) * (sizeof (hash_entry_t *))); indx->memused += ((hash_slots / intervals) * (sizeof (hash_entry_t *)));
pthread_mutex_init(&(indx->mlist[i]), NULL);
} }
if (pct_interval > 0) {
strcpy(cfg->rootdir, tmppath); strcpy(cfg->rootdir, tmppath);
strcat(cfg->rootdir, "/.segXXXXXX"); strcat(cfg->rootdir, "/.segXXXXXX");
cfg->seg_fd_w = mkstemp(cfg->rootdir); cfg->seg_fd_w = mkstemp(cfg->rootdir);
@ -185,6 +182,7 @@ init_global_db_s(char *path, char *tmppath, uint32_t chunksize, uint64_t user_ch
for (i = 0; i < nthreads; i++) { for (i = 0; i < nthreads; i++) {
cfg->seg_fd_r[i] = open(cfg->rootdir, O_RDONLY); cfg->seg_fd_r[i] = open(cfg->rootdir, O_RDONLY);
} }
}
cfg->dbdata = indx; cfg->dbdata = indx;
} }
return (cfg); return (cfg);
@ -213,9 +211,12 @@ mycmp(uchar_t *a, uchar_t *b, int sz)
return (0); return (0);
} }
/*
* Lookup and insert item if indicated. Not thread-safe by design.
*/
uint64_t uint64_t
db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval, db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval,
uint64_t seg_offset, int do_insert) uint64_t item_offset, uint32_t item_size, int do_insert)
{ {
uint32_t htab_entry; uint32_t htab_entry;
index_t *indx = (index_t *)(cfg->dbdata); index_t *indx = (index_t *)(cfg->dbdata);
@ -228,18 +229,25 @@ db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval,
htab = indx->list[interval].tab; htab = indx->list[interval].tab;
pent = &(htab[htab_entry]); pent = &(htab[htab_entry]);
pthread_mutex_lock(&(indx->mlist[interval]));
ent = htab[htab_entry]; ent = htab[htab_entry];
if (cfg->pct_interval == 0) { // Single file global dedupe case
while (ent) { while (ent) {
if (mycmp(sim_cksum, ent->cksum, cfg->similarity_cksum_sz) == 0) { if (mycmp(sim_cksum, ent->cksum, cfg->similarity_cksum_sz) == 0 &&
uint64_t off; ent->item_size == item_size) {
off = ent->seg_offset; return (ent->item_offset);
pthread_mutex_unlock(&(indx->mlist[interval]));
return (off+1);
} }
pent = &(ent->next); pent = &(ent->next);
ent = ent->next; ent = ent->next;
} }
} else {
while (ent) {
if (mycmp(sim_cksum, ent->cksum, cfg->similarity_cksum_sz) == 0) {
return (ent->item_offset);
}
pent = &(ent->next);
ent = ent->next;
}
}
if (do_insert) { if (do_insert) {
if (indx->memused + indx->hash_entry_size >= indx->memlimit - (indx->hash_entry_size << 2)) { if (indx->memused + indx->hash_entry_size >= indx->memlimit - (indx->hash_entry_size << 2)) {
ent = htab[htab_entry]; ent = htab[htab_entry];
@ -249,11 +257,11 @@ db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval,
ent = (hash_entry_t *)malloc(indx->hash_entry_size); ent = (hash_entry_t *)malloc(indx->hash_entry_size);
indx->memused += indx->hash_entry_size; indx->memused += indx->hash_entry_size;
} }
ent->seg_offset = seg_offset; ent->item_offset = item_offset;
ent->item_size = item_size;
ent->next = 0; ent->next = 0;
memcpy(ent->cksum, sim_cksum, cfg->similarity_cksum_sz); memcpy(ent->cksum, sim_cksum, cfg->similarity_cksum_sz);
*pent = ent; *pent = ent;
} }
pthread_mutex_unlock(&(indx->mlist[interval]));
return (0); return (0);
} }

View file

@ -37,7 +37,7 @@ archive_config_t *init_global_db_s(char *path, char *tmppath, uint32_t chunksize
cksum_t ck, cksum_t ck_sim, size_t file_sz, size_t memlimit, cksum_t ck, cksum_t ck_sim, size_t file_sz, size_t memlimit,
int nthreads); int nthreads);
uint64_t db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval, uint64_t db_lookup_insert_s(archive_config_t *cfg, uchar_t *sim_cksum, int interval,
uint64_t seg_offset, int do_insert); uint64_t item_offset, uint32_t item_size, int do_insert);
#ifdef __cplusplus #ifdef __cplusplus
} }

View file

@ -128,7 +128,7 @@ dedupe_buf_extra(uint64_t chunksize, int rab_blk_sz, const char *algo, int delta
*/ */
dedupe_context_t * dedupe_context_t *
create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, int rab_blk_sz, create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, int rab_blk_sz,
const char *algo, const algo_props_t *props, int delta_flag, int fixed_flag, const char *algo, const algo_props_t *props, int delta_flag, int dedupe_flag,
int file_version, compress_op_t op) { int file_version, compress_op_t op) {
dedupe_context_t *ctx; dedupe_context_t *ctx;
uint32_t i; uint32_t i;
@ -136,7 +136,7 @@ create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, int rab_blk_s
if (rab_blk_sz < 1 || rab_blk_sz > 5) if (rab_blk_sz < 1 || rab_blk_sz > 5)
rab_blk_sz = RAB_BLK_DEFAULT; rab_blk_sz = RAB_BLK_DEFAULT;
if (fixed_flag) { if (dedupe_flag == RABIN_DEDUPE_FIXED || dedupe_flag == RABIN_DEDUPE_FILE_GLOBAL) {
delta_flag = 0; delta_flag = 0;
inited = 1; inited = 1;
} }
@ -196,7 +196,7 @@ create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, int rab_blk_s
ctx->rabin_poly_max_block_size = RAB_POLYNOMIAL_MAX_BLOCK_SIZE; ctx->rabin_poly_max_block_size = RAB_POLYNOMIAL_MAX_BLOCK_SIZE;
ctx->current_window_data = NULL; ctx->current_window_data = NULL;
ctx->fixed_flag = fixed_flag; ctx->dedupe_flag = dedupe_flag;
ctx->rabin_break_patt = 0; ctx->rabin_break_patt = 0;
ctx->rabin_poly_avg_block_size = RAB_BLK_AVG_SZ(rab_blk_sz); ctx->rabin_poly_avg_block_size = RAB_BLK_AVG_SZ(rab_blk_sz);
ctx->rabin_avg_block_mask = RAB_BLK_MASK; ctx->rabin_avg_block_mask = RAB_BLK_MASK;
@ -220,7 +220,7 @@ create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, int rab_blk_s
ctx->delta_flag = 2; ctx->delta_flag = 2;
} }
if (!fixed_flag) if (dedupe_flag != RABIN_DEDUPE_FIXED)
ctx->blknum = chunksize / ctx->rabin_poly_min_block_size; ctx->blknum = chunksize / ctx->rabin_poly_min_block_size;
else else
ctx->blknum = chunksize / ctx->rabin_poly_avg_block_size; ctx->blknum = chunksize / ctx->rabin_poly_avg_block_size;
@ -330,7 +330,7 @@ dedupe_compress(dedupe_context_t *ctx, uchar_t *buf, uint64_t *size, uint64_t of
if (*size < ctx->rabin_poly_avg_block_size) return (0); if (*size < ctx->rabin_poly_avg_block_size) return (0);
DEBUG_STAT_EN(strt = get_wtime_millis()); DEBUG_STAT_EN(strt = get_wtime_millis());
if (ctx->fixed_flag) { if (ctx->dedupe_flag == RABIN_DEDUPE_FIXED) {
blknum = *size / ctx->rabin_poly_avg_block_size; blknum = *size / ctx->rabin_poly_avg_block_size;
j = *size % ctx->rabin_poly_avg_block_size; j = *size % ctx->rabin_poly_avg_block_size;
if (j) if (j)

View file

@ -118,6 +118,10 @@
#define GET_SIMILARITY_FLAG SET_SIMILARITY_FLAG #define GET_SIMILARITY_FLAG SET_SIMILARITY_FLAG
#define CLEAR_SIMILARITY_FLAG (0xBFFFFFFFUL) #define CLEAR_SIMILARITY_FLAG (0xBFFFFFFFUL)
#define RABIN_DEDUPE_SEGMENTED 0
#define RABIN_DEDUPE_FIXED 1
#define RABIN_DEDUPE_FILE_GLOBAL 2
// Mask to extract value from a rabin index entry // Mask to extract value from a rabin index entry
#define RABIN_INDEX_VALUE (0x3FFFFFFFUL) #define RABIN_INDEX_VALUE (0x3FFFFFFFUL)
@ -171,7 +175,7 @@ typedef struct {
uint64_t real_chunksize; uint64_t real_chunksize;
short valid; short valid;
void *lzma_data; void *lzma_data;
int level, delta_flag, fixed_flag, deltac_min_distance; int level, delta_flag, dedupe_flag, deltac_min_distance;
} dedupe_context_t; } dedupe_context_t;
extern dedupe_context_t *create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize, extern dedupe_context_t *create_dedupe_context(uint64_t chunksize, uint64_t real_chunksize,

View file

@ -41,6 +41,8 @@
#include <cpuid.h> #include <cpuid.h>
#include <xxhash.h> #include <xxhash.h>
#include <sys/sysinfo.h>
#define _IN_UTILS_ #define _IN_UTILS_
#include "utils.h" #include "utils.h"
@ -367,3 +369,16 @@ get_mb_s(uint64_t bytes, double strt, double en)
bytes_sec = ((double)bytes / (en - strt)) * 1000; bytes_sec = ((double)bytes / (en - strt)) * 1000;
return (BYTES_TO_MB(bytes_sec)); return (BYTES_TO_MB(bytes_sec));
} }
void
get_sysinfo(my_sysinfo *msys_info)
{
struct sysinfo sys_info;
sysinfo(&sys_info);
msys_info->totalram = sys_info.totalram;
msys_info->freeram = sys_info.freeram;
msys_info->totalswap = sys_info.totalswap;
msys_info->freeswap = sys_info.freeswap;
msys_info->mem_unit = sys_info.mem_unit;
}

View file

@ -134,6 +134,28 @@ typedef int32_t bsize_t;
#define BYTES_TO_MB(x) ((x) / (1024 * 1024)) #define BYTES_TO_MB(x) ((x) / (1024 * 1024))
/*
* Public checksum properties. CKSUM_MAX_BYTES must be updated if a
* newer larger checksum is added to the list.
*/
typedef enum {
CKSUM_CRC64 = 0x100,
CKSUM_BLAKE256 = 0x200,
CKSUM_BLAKE512 = 0x300,
CKSUM_SHA256 = 0x400,
CKSUM_SHA512 = 0x500,
CKSUM_KECCAK256 = 0x600,
CKSUM_KECCAK512 = 0x700,
/*
* Backwards compatibility options. SKEIN in release 1.2 was replaced with
* Blake2 from 1.3 onwards (for sheer speed of Blake2). We want to be able
* to decode archives created with 1.2. New archives do not use SKEIN.
*/
CKSUM_SKEIN256 = 0x800,
CKSUM_SKEIN512 = 0x900,
CKSUM_INVALID = 0
} cksum_t;
typedef enum { typedef enum {
COMPRESS_NONE = 0, COMPRESS_NONE = 0,
COMPRESS_LZFX, COMPRESS_LZFX,
@ -155,6 +177,7 @@ typedef struct {
int d_max_threads; int d_max_threads;
int delta2_span; int delta2_span;
int deltac_min_distance; int deltac_min_distance;
cksum_t cksum;
} algo_props_t; } algo_props_t;
typedef enum { typedef enum {
@ -162,6 +185,14 @@ typedef enum {
DECOMPRESS_THREADS DECOMPRESS_THREADS
} algo_threads_type_t; } algo_threads_type_t;
typedef struct{
int64_t totalram;
int64_t freeram;
int64_t totalswap;
int64_t freeswap;
int64_t mem_unit;
} my_sysinfo;
#ifndef _IN_UTILS_ #ifndef _IN_UTILS_
extern processor_info_t proc_info; extern processor_info_t proc_info;
#endif #endif
@ -179,6 +210,7 @@ extern void set_threadcounts(algo_props_t *props, int *nthreads, int nprocs,
extern uint64_t get_total_ram(); extern uint64_t get_total_ram();
extern double get_wtime_millis(void); extern double get_wtime_millis(void);
extern double get_mb_s(uint64_t bytes, double strt, double en); extern double get_mb_s(uint64_t bytes, double strt, double en);
extern void get_sysinfo(my_sysinfo *msys_info);
extern void init_algo_props(algo_props_t *props); extern void init_algo_props(algo_props_t *props);
extern void init_pcompress(); extern void init_pcompress();