#include <set>
#include <string>
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <List.h>
#include "BPlusTree.h"
#include "Inode.h"
#include "Volume.h"
#define DEFAULT_ITERATIONS 10
#define DEFAULT_NUM_KEYS 100
#define DEFAULT_KEY_TYPE S_STR_INDEX
#define MIN_STRING 3
#define MAX_STRING 256
struct key {
void* data;
uint32 length;
int32 in;
int32 count;
off_t value;
};
key* gKeys;
int32 gNum = DEFAULT_NUM_KEYS;
int32 gType = DEFAULT_KEY_TYPE;
int32 gTreeCount = 0;
bool gVerbose, gExcessive;
int32 gIterations = DEFAULT_ITERATIONS;
int32 gHard = 1;
Volume* gVolume;
int32 gSeed = 42;
extern BList gBlocks;
void bailOut();
void bailOutWithKey(void* key, uint16 length);
void dumpTree();
void dumpKey(void* key, int32 length);
void dumpKeys();
void
dumpTree()
{
puts("\n*** Tree-Dump:\n");
bplustree_header* header = (bplustree_header*)gBlocks.ItemAt(0);
dump_bplustree_header(header);
for (int32 i = 1; i < gBlocks.CountItems(); i++) {
bplustree_node* node = (bplustree_node*)gBlocks.ItemAt(i);
printf("\n--- %s node at %ld --------------------------------------\n",
node->overflow_link == BPLUSTREE_NULL ? "leaf" : "index",
i * BPLUSTREE_NODE_SIZE);
dump_bplustree_node(node, header, gVolume);
}
}
void
bailOut()
{
if (gVerbose) {
dumpTree();
}
shutdown_cache(gVolume->Device(), gVolume->BlockSize());
exit(-1);
}
void
bailOutWithKey(void* key, uint16 length)
{
dumpKey(key, length);
putchar('\n');
bailOut();
}
void
dumpKey(void* key, int32 length)
{
switch (gType) {
case S_STR_INDEX:
printf("\"%s\" (%ld bytes)", (char*)key, length);
break;
case S_INT_INDEX:
printf("%ld", *(int32*)key);
break;
case S_UINT_INDEX:
printf("%lu", *(uint32*)key);
break;
case S_LONG_LONG_INDEX:
printf("%lld", *(int64*)key);
break;
case S_ULONG_LONG_INDEX:
printf("%Lu", *(uint64*)key);
break;
case S_FLOAT_INDEX:
printf("%g", *(float*)key);
break;
case S_DOUBLE_INDEX:
printf("%g", *(double*)key);
break;
}
if ((gType == S_INT_INDEX || gType == S_UINT_INDEX
|| gType == S_FLOAT_INDEX) && length != 4)
printf(" (wrong length %ld)", length);
else if ((gType == S_LONG_LONG_INDEX || gType == S_ULONG_LONG_INDEX
|| gType == S_DOUBLE_INDEX) && length != 8)
printf(" (wrong length %ld)", length);
}
void
dumpKeys()
{
const char* type;
switch (gType) {
case S_STR_INDEX:
type = "string";
break;
case S_INT_INDEX:
type = "int32";
break;
case S_UINT_INDEX:
type = "uint32";
break;
case S_LONG_LONG_INDEX:
type = "int64";
break;
case S_ULONG_LONG_INDEX:
type = "uint64";
break;
case S_FLOAT_INDEX:
type = "float";
break;
case S_DOUBLE_INDEX:
type = "double";
break;
default:
debugger("unknown type in gType");
return;
}
printf("Dumping %ld keys of type %s\n", gNum, type);
for (int32 i = 0; i < gNum; i++) {
printf("% 8ld. (%3ld) key = ", i, gKeys[i].in);
dumpKey(gKeys[i].data, gKeys[i].length);
putchar('\n');
}
}
void
generateName(int32 i, char* name, int32* _length)
{
int32 length = rand() % (MAX_STRING - MIN_STRING) + MIN_STRING;
for (int32 i = 0; i < length; i++) {
int32 c = int32(52.0 * rand() / RAND_MAX);
if (c >= 26)
name[i] = 'A' + c - 26;
else
name[i] = 'a' + c;
}
name[length] = 0;
*_length = length;
}
void
fillBuffer(void* buffer, int32 start)
{
for (int32 i = 0; i < gNum; i++) {
switch (gType) {
case S_INT_INDEX:
{
int32* array = (int32*)buffer;
array[i] = start + i;
break;
}
case S_UINT_INDEX:
{
uint32* array = (uint32*)buffer;
array[i] = start + i;
break;
}
case S_LONG_LONG_INDEX:
{
int64* array = (int64*)buffer;
array[i] = start + i;
break;
}
case S_ULONG_LONG_INDEX:
{
uint64* array = (uint64*)buffer;
array[i] = start + i;
break;
}
case S_FLOAT_INDEX:
{
float* array = (float*)buffer;
array[i] = start + i * 1.0001;
break;
}
case S_DOUBLE_INDEX:
{
double* array = (double*)buffer;
array[i] = start + i * 1.0001;
break;
}
}
gKeys[i].value = i;
}
}
status_t
createKeys()
{
gKeys = (key*)malloc(gNum * sizeof(key));
if (gKeys == NULL)
return B_NO_MEMORY;
if (gType == S_STR_INDEX) {
std::set<std::string> set;
for (int32 i = 0; i < gNum; i++) {
char name[B_FILE_NAME_LENGTH];
int32 length;
int32 tries = 0;
std::set<std::string>::const_iterator found;
while (tries++ < 100) {
generateName(i, name, &length);
found = set.find(string(name));
if (found == set.end())
break;
}
if (found != set.end()) {
printf("Couldn't create unique key list!\n");
dumpKeys();
bailOut();
}
gKeys[i].data = malloc(length + 1);
memcpy(gKeys[i].data, name, length + 1);
gKeys[i].length = length;
gKeys[i].in = 0;
gKeys[i].count = 0;
gKeys[i].value = i;
set.insert(name);
}
} else {
int32 length;
int32 start = 0;
switch (gType) {
case S_FLOAT_INDEX:
case S_INT_INDEX:
start = -gNum / 2;
case S_UINT_INDEX:
length = 4;
break;
case S_DOUBLE_INDEX:
case S_LONG_LONG_INDEX:
start = -gNum / 2;
case S_ULONG_LONG_INDEX:
length = 8;
break;
default:
return B_BAD_VALUE;
}
uint8* buffer = (uint8*)malloc(length * gNum);
if (buffer == NULL)
return B_NO_MEMORY;
for (int32 i = 0; i < gNum; i++) {
gKeys[i].data = (void*)(buffer + i * length);
gKeys[i].length = length;
gKeys[i].in = 0;
gKeys[i].count = 0;
}
fillBuffer(buffer, start);
}
return B_OK;
}
void
checkTreeContents(BPlusTree* tree)
{
for (int32 i = 0; i < gNum; i++)
gKeys[i].count = 0;
TreeIterator iterator(tree);
char key[B_FILE_NAME_LENGTH];
uint16 length;
uint16 duplicate;
off_t value;
status_t status;
while ((status = iterator.GetNextEntry(key, &length, B_FILE_NAME_LENGTH,
&value, &duplicate)) == B_OK) {
if (value < 0 || value >= gNum) {
iterator.Dump();
printf("\ninvalid value %lld in tree: ", value);
bailOutWithKey(key, length);
}
if (gKeys[value].value != value) {
iterator.Dump();
printf("\nkey pointing to the wrong value %lld (should be %lld)\n",
value, gKeys[value].value);
bailOutWithKey(key, length);
}
if (length != gKeys[value].length
|| memcmp(key, gKeys[value].data, length)) {
iterator.Dump();
printf("\nkeys don't match (key index = %lld, %ld times in tree, "
"%ld. occassion):\n\tfound: ", value, gKeys[value].in,
gKeys[value].count + 1);
dumpKey(key, length);
printf("\n\texpected: ");
dumpKey(gKeys[value].data, gKeys[value].length);
putchar('\n');
bailOut();
}
gKeys[value].count++;
}
if (status != B_ENTRY_NOT_FOUND) {
printf("TreeIterator::GetNext() returned: %s\n", strerror(status));
iterator.Dump();
bailOut();
}
for (int32 i = 0; i < gNum; i++) {
if (gKeys[i].in != gKeys[i].count) {
printf("Key ");
dumpKey(gKeys[i].data, gKeys[i].length);
printf(" found only %ld from %ld\n", gKeys[i].count, gKeys[i].in);
bailOut();
}
}
}
void
checkTreeIntegrity(BPlusTree* tree)
{
TreeIterator iterator(tree);
for (int32 i = 0; i < gNum; i++) {
status_t status = iterator.Find((uint8*)gKeys[i].data, gKeys[i].length);
if (gKeys[i].in == 0) {
if (status == B_OK) {
printf("found key %" B_PRId32 " even though it's not in!\n", i);
bailOutWithKey(gKeys[i].data, gKeys[i].length);
}
} else if (status != B_OK) {
printf("TreeIterator::Find() returned: %s\n", strerror(status));
bailOutWithKey(gKeys[i].data, gKeys[i].length);
}
}
}
void
checkTree(BPlusTree* tree)
{
if (!gExcessive)
printf("* Check tree...\n");
checkTreeContents(tree);
checkTreeIntegrity(tree);
bool errorsFound = false;
tree->Validate(false, errorsFound);
if (errorsFound) {
printf("BPlusTree::Validate() found errors\n");
bailOut();
}
}
void
addAllKeys(Transaction& transaction, BPlusTree* tree)
{
printf("*** Adding all keys to the tree...\n");
for (int32 i = 0; i < gNum; i++) {
status_t status = tree->Insert(transaction, (uint8*)gKeys[i].data,
gKeys[i].length, gKeys[i].value);
if (status != B_OK) {
printf("BPlusTree::Insert() returned: %s\n", strerror(status));
printf("key: ");
bailOutWithKey(gKeys[i].data, gKeys[i].length);
} else {
gKeys[i].in++;
gTreeCount++;
}
}
checkTree(tree);
}
void
removeAllKeys(Transaction& transaction, BPlusTree* tree)
{
printf("*** Removing all keys from the tree...\n");
for (int32 i = 0; i < gNum; i++) {
while (gKeys[i].in > 0) {
status_t status = tree->Remove(transaction, (uint8*)gKeys[i].data,
gKeys[i].length, gKeys[i].value);
if (status != B_OK) {
printf("BPlusTree::Remove() returned: %s\n", strerror(status));
printf("key: ");
dumpKey(gKeys[i].data, gKeys[i].length);
putchar('\n');
} else {
gKeys[i].in--;
gTreeCount--;
}
}
}
checkTree(tree);
}
void
duplicateTest(Transaction& transaction, BPlusTree* tree)
{
int32 index = int32(1.0 * gNum * rand() / RAND_MAX);
if (index == gNum)
index = gNum - 1;
printf("*** Duplicate test with key ");
dumpKey(gKeys[index].data, gKeys[index].length);
puts("...");
status_t status;
int32 insertTotal = 0;
for (int32 i = 0; i < 8; i++) {
int32 insertCount = int32(1000.0 * rand() / RAND_MAX);
if (gVerbose) {
printf("* insert %ld to %ld old entries...\n", insertCount,
insertTotal + gKeys[index].in);
}
for (int32 j = 0; j < insertCount; j++) {
status = tree->Insert(transaction, (uint8*)gKeys[index].data,
gKeys[index].length, gKeys[index].value);
if (status != B_OK) {
printf("BPlusTree::Insert() returned: %s\n", strerror(status));
bailOutWithKey(gKeys[index].data, gKeys[index].length);
}
insertTotal++;
gTreeCount++;
gKeys[index].in++;
if (gExcessive)
checkTree(tree);
}
int32 count;
if (i < 7) {
count = int32(1000.0 * rand() / RAND_MAX);
if (count > insertTotal)
count = insertTotal;
} else
count = insertTotal;
if (gVerbose) {
printf("* remove %ld from %ld entries...\n", count,
insertTotal + gKeys[index].in);
}
for (int32 j = 0; j < count; j++) {
status_t status = tree->Remove(transaction,
(uint8*)gKeys[index].data, gKeys[index].length,
gKeys[index].value);
if (status != B_OK) {
printf("BPlusTree::Remove() returned: %s\n", strerror(status));
bailOutWithKey(gKeys[index].data, gKeys[index].length);
}
insertTotal--;
gTreeCount--;
gKeys[index].in--;
if (gExcessive)
checkTree(tree);
}
}
if (gExcessive)
checkTree(tree);
}
void
addRandomSet(Transaction& transaction, BPlusTree* tree, int32 num)
{
printf("*** Add random set to tree (%ld to %ld old entries)...\n",
num, gTreeCount);
for (int32 i = 0; i < num; i++) {
int32 index = int32(1.0 * gNum * rand() / RAND_MAX);
if (index == gNum)
index = gNum - 1;
if (gVerbose) {
printf("adding key %ld (%ld times in the tree)\n", index,
gKeys[index].in);
}
status_t status = tree->Insert(transaction, (uint8*)gKeys[index].data,
gKeys[index].length, gKeys[index].value);
if (status != B_OK) {
printf("BPlusTree::Insert() returned: %s\n", strerror(status));
bailOutWithKey(gKeys[index].data, gKeys[index].length);
}
gKeys[index].in++;
gTreeCount++;
if (gExcessive)
checkTree(tree);
}
if (!gExcessive)
checkTree(tree);
}
void
removeRandomSet(Transaction& transaction, BPlusTree* tree, int32 num)
{
printf("*** Remove random set from tree (%ld from %ld entries)...\n",
num, gTreeCount);
int32 tries = 500;
for (int32 i = 0; i < num; i++) {
if (gTreeCount < 1)
break;
int32 index = int32(1.0 * gNum * rand() / RAND_MAX);
if (index == gNum)
index = gNum - 1;
if (gKeys[index].in == 0) {
i--;
if (tries-- < 0)
break;
continue;
}
if (gVerbose) {
printf("removing key %ld (%ld times in the tree)\n", index,
gKeys[index].in);
}
status_t status = tree->Remove(transaction, (uint8*)gKeys[index].data,
gKeys[index].length, gKeys[index].value);
if (status != B_OK) {
printf("BPlusTree::Remove() returned: %s\n", strerror(status));
bailOutWithKey(gKeys[index].data, gKeys[index].length);
}
gKeys[index].in--;
gTreeCount--;
if (gExcessive)
checkTree(tree);
}
if (!gExcessive)
checkTree(tree);
}
void
usage(char* program)
{
if (strrchr(program, '/'))
program = strrchr(program, '/') + 1;
fprintf(stderr, "usage: %s [-veh] [-t type] [-n keys] [-i iterations] "
"[-h times] [-r seed]\n"
"BFS B+Tree torture test\n"
"\t-t\ttype is one of string, int32, uint32, int64, uint64, float,\n"
"\t\tor double; defaults to string.\n"
"\t-n\tkeys is the number of keys to be used,\n"
"\t\tminimum is 1, defaults to %d.\n"
"\t-i\titerations is the number of the test cycles, defaults to %d.\n"
"\t-r\tthe seed for the random function, defaults to %ld.\n"
"\t-h\tremoves the keys and start over again for x times.\n"
"\t-e\texcessive validity tests: tree contents will be tested after "
"every operation\n"
"\t-v\tfor verbose output.\n",
program, DEFAULT_NUM_KEYS, DEFAULT_ITERATIONS, gSeed);
exit(0);
}
int
main(int argc, char** argv)
{
char* program = argv[0];
while (*++argv) {
char* arg = *argv;
if (*arg == '-') {
if (arg[1] == '-')
usage(program);
while (*++arg && isalpha(*arg)) {
switch (*arg) {
case 'v':
gVerbose = true;
break;
case 'e':
gExcessive = true;
break;
case 't':
if (*++argv == NULL)
usage(program);
if (!strcmp(*argv, "string"))
gType = S_STR_INDEX;
else if (!strcmp(*argv, "int32")
|| !strcmp(*argv, "int"))
gType = S_INT_INDEX;
else if (!strcmp(*argv, "uint32")
|| !strcmp(*argv, "uint"))
gType = S_UINT_INDEX;
else if (!strcmp(*argv, "int64")
|| !strcmp(*argv, "llong"))
gType = S_LONG_LONG_INDEX;
else if (!strcmp(*argv, "uint64")
|| !strcmp(*argv, "ullong"))
gType = S_ULONG_LONG_INDEX;
else if (!strcmp(*argv, "float"))
gType = S_FLOAT_INDEX;
else if (!strcmp(*argv, "double"))
gType = S_DOUBLE_INDEX;
else
usage(program);
break;
case 'n':
if (*++argv == NULL || !isdigit(**argv))
usage(program);
gNum = atoi(*argv);
if (gNum < 1)
gNum = 1;
break;
case 'h':
if (*++argv == NULL || !isdigit(**argv))
usage(program);
gHard = atoi(*argv);
if (gHard < 1)
gHard = 1;
break;
case 'i':
if (*++argv == NULL || !isdigit(**argv))
usage(program);
gIterations = atoi(*argv);
if (gIterations < 1)
gIterations = 1;
break;
case 'r':
if (*++argv == NULL || !isdigit(**argv))
usage(program);
gSeed = atoi(*argv);
break;
}
}
} else
break;
}
if (gVerbose)
printf("Set seed to %ld\n", gSeed);
srand(gSeed);
Inode inode("tree.data", gType | S_ALLOW_DUPS);
rw_lock_write_lock(&inode.Lock());
gVolume = inode.GetVolume();
Transaction transaction(gVolume, 0);
init_cache(gVolume->Device(), gVolume->BlockSize());
BPlusTree tree(transaction, &inode);
status_t status = tree.InitCheck();
if (status != B_OK) {
fprintf(stderr, "creating tree failed: %s\n", strerror(status));
bailOut();
}
printf("*** Creating %ld keys...\n", gNum);
status = createKeys();
if (status != B_OK) {
fprintf(stderr, "creating keys failed: %s\n", strerror(status));
bailOut();
}
if (gVerbose)
dumpKeys();
for (int32 j = 0; j < gHard; j++) {
addAllKeys(transaction, &tree);
for (int32 i = 0; i < gIterations; i++) {
printf("---------- Test iteration %ld --------------------------\n",
i + 1);
addRandomSet(transaction, &tree,
int32(1.0 * gNum * rand() / RAND_MAX));
removeRandomSet(transaction, &tree,
int32(1.0 * gNum * rand() / RAND_MAX));
duplicateTest(transaction, &tree);
}
removeAllKeys(transaction, &tree);
}
transaction.Done();
shutdown_cache(gVolume->Device(), gVolume->BlockSize());
return 0;
}
