/*----------------function to store the pointer to the root of a function---------*/
store_func(funcptr head, char in_name[], nodeptr root)
{
funcptr p;
if (head==NULL) /* empty list */
{
p=new_func();
func_header=p;
strcpy(p->fname,in_name);
p->faddr=root;
p->nextf=NULL;
return;
}
p=head;
while (p->nextf!=NULL) p=p->nextf;
if (p->nextf==NULL) /* add the root of a new function at end*/
{
p->nextf=new_func();
strcpy(p->nextf->fname,in_name);
p->nextf->faddr=root;
p->nextf->nextf=NULL;
return;
}
}
void init_file_sys(Object *cxt) {
Object *file_object = context_get(cxt, "File");
set(cxt, file_object, "open:", new_func(cxt, native_file_open));
set(cxt, file_object, "open:mode:", new_func(cxt, native_file_open_mode));
set(cxt, file_object, "close", new_func(cxt, native_file_close));
set(cxt, file_object, "read:into:offset:length:", new_func(cxt, native_file_read_into_offset_length));
set(cxt, file_object, "eof", new_func(cxt, native_file_eof));
}
int main (int argc, char *argv[])
{
//"idle=halt nohz=off highres=on rcutree.blimit=3 nmi_watchdog=0 noirqdebug cgroups_disable=memory init=/sbin/react-init.sh isolcpus"
int c;
while ((c = getopt(argc, argv, "hedvsr:m:i:t:p:w:c:I:")) != -1) {
switch (c) {
case 'h':
printf("help\n");
exit(EXIT_SUCCESS);
break;
case 'w':
printf("optargs %s\n",optarg);
break;
default:
printf("default %c %s\n",c, optarg);
exit(EXIT_FAILURE);
break;
}
}
//react_elilo_adjust("./elilo.conf", 0 /*int add_to_file*/ );
new_func("./elilo.conf");
return 0;
}
int Resource::RegisterFunction(char *symbol, arg_list_t *(*func)(arg_list_t *))
{
arg_list_t *f, *sym_tab;
new_func(&f, func, symbol);
if (!func)
{
return -1;
}
if (!_sym_tab)
{
_sym_tab = f;
}
else
{
sym_tab = _sym_tab;
while (sym_tab->next)
{
sym_tab = sym_tab->next;
}
sym_tab->next = f;
}
return 0;
}
int main()
{
A* ap = new A();
B* bp = new B();
test_a(ap);
test_b(bp);
#ifdef _PROFILE_USE
new_func(10);
#endif
}
int do_add_pltmap_funcs( void * data , void * arg )
{
plt_map_t * map;
func_t * func;
if(!(map = data))
error_ret("no map",-1);
if(!(func = func_at_addr( map->vaddr )))
{
func = new_func();
OBJ_SET_ADDR(func->obj,map->vaddr);
add_func( func , 1 ) ;
}
OBJ_SET_NAME(func->obj,map->name);
return(0);
}
int get_objs_from_ehdr(Elf32_Ehdr * ehdr)
{
func_t * func;
if(!ehdr) error_ret("null arg",-1);
if(!(func = func_at_addr( ehdr->e_entry )))
{
func = new_func();
OBJ_SET_NAME(func->obj,"_start");
OBJ_SET_ADDR(func->obj,ehdr->e_entry);
if(add_func( func , 1 )<0)
error_ret("problem adding",-1);
}
else
{
if(strcmp(OBJ_GET_NAME(func->obj),"_start"))
OBJ_SET_NAME(func->obj,"_start");
}
return(0);
}
int get_objs_from_entry(elf_t * elf)
{
char * data;
func_t * func;
Elf32_Ehdr * ehdr;
x86_insn_t insn;
addr_t dst;
addr_t vma;
int nread,
tread = 0;
if(!elf) error_ret("null arg",-1);
if(!(ehdr = get_ehdr(elf)))
error_ret("can't get ehdr",-1);
if(get_objs_from_ehdr(ehdr)<0)
error_ret("can't get _start",-1);
if(!(data = data_at_addr( elf , ehdr->e_entry )))
error_ret("can't get data",-1);
/*
if(!(func = func_at_addr( ehdr->e_entry )))
error_ret("this is dumb",-1);
*/
do
{
nread =x86_disasm(data ,MAX_INSN_SIZE+tread,
ehdr->e_entry , tread , &insn);
if( nread <= 0 )
error_ret("can't get past first stage",-1);
tread += nread;
} while ( (tread < 512) && ( (insn.type != insn_push) ||
(insn.operands[op_dest].type != op_immediate ) ));
/* this should be libc_csu_fini */
dst = insn.operands[op_dest].data.dword;
if(dst <= 0 ) error_ret("bad operand",-1);
if(!(func = func_at_addr(dst))){
func = new_func();
OBJ_SET_NAME(func->obj,"__libc_csu_fini");
OBJ_SET_ADDR(func->obj,dst);
add_func(func,0);
}
nread =x86_disasm(data ,MAX_INSN_SIZE+tread ,
ehdr->e_entry , tread , &insn);
if( nread <= 0 )
error_ret("can't get past first stage",-1);
tread += nread;
if( (insn.type != insn_push) ||
(insn.operands[op_dest].type != op_immediate) )
error_ret("second push not ok",-1);
dst = insn.operands[op_dest].data.dword;
if(dst <= 0 )
error_ret("bad operand",-1);
if(!(func = func_at_addr(dst))){
func = new_func();
OBJ_SET_NAME(func->obj,"__libc_csu_init");
OBJ_SET_ADDR(func->obj,dst);
add_func( func,0 );
}
do
{
nread =x86_disasm(data ,MAX_INSN_SIZE+tread ,
ehdr->e_entry , tread , &insn);
if( nread <= 0 )
error_ret("problem getting main",-1);
tread += nread;
} while ( (tread < 512) && ((insn.type != insn_push) ||
(insn.operands[op_dest].type != op_immediate ) ));
if( (insn.type != insn_push) ||
(insn.operands[op_dest].type != op_immediate) )
error_ret("second push not ok",-1);
dst = insn.operands[op_dest].data.dword;
if(dst <= 0 ) error_ret("bad operand",-1);
if(!(func = func_at_addr(dst))){
func = new_func();
OBJ_SET_NAME(func->obj,"main");
OBJ_SET_ADDR(func->obj,dst);
add_func( func ,0);
}
return(tread);
}
void test_logdb(logdb_log_db* (*new_func)())
{
logdb_log_db *db;
enum logdb_error error = 0;
struct buffer key = {"key0", 4};
struct buffer value = {"val0", 4};
struct buffer key1;
struct buffer value1;
cstring *outtest;
cstring *value_test;
unsigned char testbin[4] = {0x00, 0x10, 0x20, 0x30};
struct buffer value0_new = {"dumb", 4};
struct buffer key2 = {"pkey", 4};
struct buffer value2;
struct buffer smp_value;
struct buffer smp_key;
uint8_t txbin[10240];
uint8_t txbin_rev[10240];
char hexrev[98];
int outlenrev;
long fsize;
char *buf;
char *wrk_buf;
FILE *f;
unsigned int i;
value2.p = testbin;
value2.len = 4;
key1.p = (char *)key1str;
key1.len = strlen(key1str);
value1.p = (char *)value1str;
value1.len = strlen(value1str);
unlink(dbtmpfile);
db = new_func();
u_assert_int_eq(logdb_load(db, "file_that_should_not_exists.dat", false, NULL), false);
u_assert_int_eq(logdb_load(db, dbtmpfile, true, NULL), true);
logdb_append(db, &key, &value);
logdb_append(db, &key1, &value1);
u_assert_int_eq(logdb_cache_size(db), 2);
outtest = logdb_find_cache(db, &key1);
u_assert_int_eq(strcmp(outtest->str, value1str),0);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
u_assert_int_eq(logdb_count_keys(db), 2);
value_test = logdb_find(db, &key1);
u_assert_int_eq(strcmp(value_test->str, value1str), 0);
value_test = logdb_find(db, &key);
u_assert_int_eq(memcmp(value_test->str, value.p, value.len), 0);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
logdb_append(db, &key2, &value2);
logdb_flush(db);
logdb_free(db);
/* check if private key is available */
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, &key2);
u_assert_int_eq(memcmp(value_test->str, value2.p, value2.len), 0);
value_test = logdb_find(db, &key);
u_assert_int_eq(memcmp(value_test->str, value.p, value.len), 0);
/* delete a record */
logdb_delete(db, &key2);
logdb_flush(db);
logdb_free(db);
/* find and check the deleted record */
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, &key);
u_assert_int_eq(memcmp(value_test->str, value.p, value.len), 0);
value_test = logdb_find(db, &key2);
u_assert_int_eq((int)value_test, 0); /* should be null */
/* overwrite a key */
logdb_append(db, &key, &value0_new);
value_test = logdb_find(db, &key);
u_assert_int_eq(memcmp(value_test->str, value0_new.p, value0_new.len), 0);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, &key);
u_assert_int_eq(memcmp(value_test->str, value0_new.p, value0_new.len), 0);
logdb_flush(db);
logdb_free(db);
/* simulate corruption */
f = fopen(dbtmpfile, "rb");
fseek(f, 0, SEEK_END);
fsize = ftell(f);
fseek(f, 0, SEEK_SET);
buf = malloc(fsize + 1);
fread(buf, fsize, 1, f);
fclose(f);
/* ---------------------------------------------------- */
wrk_buf = safe_malloc(fsize + 1);
memcpy(wrk_buf, buf, fsize);
wrk_buf[0] = 0x88; /* wrong header */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_WRONG_FILE_FORMAT);
logdb_free(db);
/* ---------------------------------------------------- */
memcpy(wrk_buf, buf, fsize);
wrk_buf[66] = 0x00; /* wrong checksum hash */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_CHECKSUM);
logdb_free(db);
/* ---------------------------------------------------- */
memcpy(wrk_buf, buf, fsize);
wrk_buf[42] = 0xFF; /* wrong value length */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_DATASTREAM_ERROR);
logdb_free(db);
free(buf);
free(wrk_buf);
/* --- large db test */
unlink(dbtmpfile);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, true, NULL), true);
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
smp_key.p = hashbin;
smp_key.len = outlen;
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
smp_value.p = txbin;
smp_value.len = outlen;
logdb_append(db, &smp_key, &smp_value);
}
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
smp_key.p = hashbin;
smp_key.len = outlen;
outtest = logdb_find(db, &smp_key);
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
u_assert_int_eq(outlen, outtest->len);
}
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
memcpy(hexrev, tx->txhash, sizeof(tx->txhash));
utils_reverse_hex(hexrev, strlen(tx->txhash));
outlenrev = sizeof(tx->txhash) / 2;
utils_hex_to_bin(hexrev, txbin_rev, strlen(hexrev), &outlenrev);
smp_key.p = hashbin;
smp_key.len = outlen;
outtest = logdb_find(db, &smp_key);
outlen = strlen(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
u_assert_int_eq(outlen, outtest->len);
/* hash transaction data and check hashes */
if (strlen(tx->hextx) > 2)
{
uint8_t tx_hash_check[SHA256_DIGEST_LENGTH];
sha256_Raw(txbin, outlen, tx_hash_check);
sha256_Raw(tx_hash_check, 32, tx_hash_check);
u_assert_int_eq(memcmp(tx_hash_check, txbin_rev, SHA256_DIGEST_LENGTH), 0);
}
}
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
smp_key.p = hashbin;
smp_key.len = outlen;
logdb_delete(db, &smp_key);
}
u_assert_int_eq(logdb_count_keys(db), 0);
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), 0);
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
smp_key.p = hashbin;
smp_key.len = outlen;
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
smp_value.p = txbin;
smp_value.len = outlen;
logdb_append(db, &smp_key, &smp_value);
}
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
smp_key.p = hashbin;
smp_key.len = outlen;
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
smp_value.p = txbin;
smp_value.len = outlen;
logdb_append(db, &smp_key, &smp_value);
}
logdb_flush(db);
logdb_free(db);
}
void test_logdb(logdb_log_db* (*new_func)())
{
logdb_log_db *db;
enum logdb_error error = 0;
cstring *key;// key= {"key0", 4};
cstring *value;// = {"val0", 4};
cstring *key1;
cstring *value1;
cstring *outtest;
cstring *value_test;
unsigned char testbin[4] = {0x00, 0x10, 0x20, 0x30};
cstring *value0;// = {"dumb", 4};
cstring *key2;// = {"pkey", 4};
cstring *value2;
cstring *smp_value;
cstring *smp_key;
uint8_t txbin[10240];
uint8_t txbin_rev[10240];
char hexrev[98];
int outlenrev;
long fsize;
char *buf;
char *wrk_buf;
FILE *f;
unsigned int i;
char bufs[300][65];
rb_red_blk_node *nodetest;
unsigned int cnt = 0;
logdb_record* rec;
key = cstr_new("key0");
value = cstr_new("val0");
value0 = cstr_new("dumb");
value1 = cstr_new_sz(10);
value2 = cstr_new_sz(10);
key1 = cstr_new_sz(10);
key2 = cstr_new("key2");
cstr_append_buf(value2, testbin, sizeof(testbin));
cstr_append_buf(value2, testbin, sizeof(testbin));
cstr_append_buf(key1, key1str, strlen(key1str));
cstr_append_buf(value1, value1str, strlen(value1str));
unlink(dbtmpfile);
db = new_func();
u_assert_int_eq(logdb_load(db, "file_that_should_not_exists.dat", false, NULL), false);
u_assert_int_eq(logdb_load(db, dbtmpfile, true, NULL), true);
logdb_append(db, NULL, key, value);
logdb_append(db, NULL, key1, value1);
u_assert_int_eq(logdb_cache_size(db), 2);
outtest = logdb_find_cache(db, key1);
u_assert_int_eq(strcmp(outtest->str, value1str),0);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
u_assert_int_eq(logdb_count_keys(db), 2);
value_test = logdb_find(db, key1);
u_assert_int_eq(strcmp(value_test->str, value1str), 0);
value_test = logdb_find(db, key);
u_assert_int_eq(memcmp(value_test->str, value->str, value->len), 0);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
logdb_append(db, NULL, key2, value2);
logdb_flush(db);
logdb_free(db);
/* check if private key is available */
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, key2);
u_assert_int_eq(memcmp(value_test->str, value2->str, value2->len), 0);
value_test = logdb_find(db, key);
u_assert_int_eq(memcmp(value_test->str, value->str, value->len), 0);
/* delete a record */
logdb_delete(db, NULL, key2);
logdb_flush(db);
logdb_free(db);
/* find and check the deleted record */
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, key);
u_assert_int_eq(memcmp(value_test->str, value->str, value->len), 0);
value_test = logdb_find(db, key2);
u_assert_int_eq((int)value_test, 0); /* should be null */
/* overwrite a key */
logdb_append(db, NULL, key, value0);
value_test = logdb_find(db, key);
u_assert_int_eq(memcmp(value_test->str, value0->str, value0->len), 0);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
value_test = logdb_find(db, key);
u_assert_int_eq(memcmp(value_test->str, value0->str, value0->len), 0);
logdb_flush(db);
logdb_free(db);
/* simulate corruption */
f = fopen(dbtmpfile, "rb");
fseek(f, 0, SEEK_END);
fsize = ftell(f);
fseek(f, 0, SEEK_SET);
buf = malloc(fsize + 1);
fread(buf, fsize, 1, f);
fclose(f);
/* ---------------------------------------------------- */
wrk_buf = safe_malloc(fsize + 1);
memcpy(wrk_buf, buf, fsize);
wrk_buf[0] = 0x88; /* wrong header */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_WRONG_FILE_FORMAT);
logdb_free(db);
/* ---------------------------------------------------- */
memcpy(wrk_buf, buf, fsize);
wrk_buf[66] = 0x00; /* wrong checksum hash */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_CHECKSUM);
logdb_free(db);
/* ---------------------------------------------------- */
memcpy(wrk_buf, buf, fsize);
wrk_buf[42] = 0xFF; /* wrong value length */
unlink(dbtmpfile);
f = fopen(dbtmpfile, "wb");
fwrite(wrk_buf, 1, fsize, f);
fclose(f);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), false);
u_assert_int_eq(error, LOGDB_ERROR_DATASTREAM_ERROR);
logdb_free(db);
free(buf);
free(wrk_buf);
/* --- large db test */
unlink(dbtmpfile);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, true, NULL), true);
smp_key = cstr_new_sz(100);
smp_value = cstr_new_sz(100);
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
cstr_erase(smp_value, 0, smp_value->len);
cstr_append_buf(smp_value, txbin, outlen);
logdb_append(db, NULL, smp_key, smp_value);
}
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
outtest = logdb_find(db, smp_key);
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
u_assert_int_eq(outlen, outtest->len);
}
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
memcpy(hexrev, tx->txhash, sizeof(tx->txhash));
utils_reverse_hex(hexrev, strlen(tx->txhash));
outlenrev = sizeof(tx->txhash) / 2;
utils_hex_to_bin(hexrev, txbin_rev, strlen(hexrev), &outlenrev);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
outtest = logdb_find(db, smp_key);
outlen = strlen(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
u_assert_int_eq(outlen, outtest->len);
/* hash transaction data and check hashes */
if (strlen(tx->hextx) > 2)
{
uint8_t tx_hash_check[SHA256_DIGEST_LENGTH];
sha256_Raw(txbin, outlen, tx_hash_check);
sha256_Raw(tx_hash_check, 32, tx_hash_check);
u_assert_int_eq(memcmp(tx_hash_check, txbin_rev, SHA256_DIGEST_LENGTH), 0);
}
}
/* check all records */
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
logdb_delete(db, NULL, smp_key);
}
u_assert_int_eq(logdb_count_keys(db), 0);
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), 0);
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
cstr_erase(smp_value, 0, smp_value->len);
cstr_append_buf(smp_value, txbin, outlen);
logdb_append(db, NULL, smp_key, smp_value);
}
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
logdb_flush(db);
logdb_free(db);
db = new_func();
error = 0;
u_assert_int_eq(logdb_load(db, dbtmpfile, false, &error), true);
u_assert_int_eq(error, LOGDB_SUCCESS);
u_assert_int_eq(logdb_count_keys(db), (sizeof(sampledata) / sizeof(sampledata[0])));
if(new_func == logdb_rbtree_new)
{
logdb_rbtree_db* handle = (logdb_rbtree_db *)db->cb_ctx;
size_t size = rbtree_count(handle->tree);
nodetest = NULL;
while ((nodetest = rbtree_enumerate_next(handle->tree)))
{
rec = (logdb_record *)nodetest->info;
utils_bin_to_hex((unsigned char *)rec->key->str, rec->key->len, bufs[cnt]);
for(i = 0; i < cnt; i++)
{
u_assert_int_eq(strcmp(bufs[i], bufs[cnt]) != 0, 1);
}
cnt++;
}
u_assert_int_eq(size, cnt);
}
for (i = 0; i < (sizeof(sampledata) / sizeof(sampledata[0])); i++) {
const struct txtest *tx = &sampledata[i];
uint8_t hashbin[sizeof(tx->txhash) / 2];
int outlen = sizeof(tx->txhash) / 2;
utils_hex_to_bin(tx->txhash, hashbin, strlen(tx->txhash), &outlen);
cstr_erase(smp_key, 0, smp_key->len);
cstr_append_buf(smp_key, hashbin, outlen);
outlen = sizeof(tx->hextx) / 2;
utils_hex_to_bin(tx->hextx, txbin, strlen(tx->hextx), &outlen);
cstr_erase(smp_value, 0, smp_value->len);
cstr_append_buf(smp_value, txbin, outlen);
logdb_append(db, NULL, smp_key, smp_value);
}
logdb_flush(db);
logdb_free(db);
/* test switch mem mapper after initialitaion. */
db = logdb_new();
logdb_set_memmapper(db, &logdb_rbtree_mapper, NULL);
logdb_flush(db);
logdb_free(db);
unlink(dbtmpfile);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, true, NULL), true);
// create transaction, don't store
logdb_txn* txn = logdb_txn_new();
logdb_append(db, txn, key, value);
logdb_append(db, txn, key1, value1);
u_assert_int_eq(logdb_cache_size(db), 0);
logdb_txn_free(txn);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
// db should still be empty
u_assert_int_eq(logdb_count_keys(db), 0);
// create transaction, store it this time
txn = logdb_txn_new();
logdb_append(db, txn, key, value);
logdb_append(db, txn, key1, value1);
logdb_txn_commit(db, txn);
u_assert_int_eq(logdb_cache_size(db), 2);
logdb_txn_free(txn);
logdb_flush(db);
logdb_free(db);
db = new_func();
u_assert_int_eq(logdb_load(db, dbtmpfile, false, NULL), true);
// now we should have the two persisted items from the txn
u_assert_int_eq(logdb_count_keys(db), 2);
logdb_flush(db);
logdb_free(db);
cstr_free(key, true);
cstr_free(value, true);
cstr_free(value0, true);
cstr_free(value1, true);
cstr_free(value2, true);
cstr_free(key1, true);
cstr_free(key2, true);
cstr_free(smp_key, true);
cstr_free(smp_value, true);
}
