284 lines
8.2 KiB
C
284 lines
8.2 KiB
C
#pragma GCC diagnostic ignored "-Wunused-but-set-variable"
|
|
#pragma GCC diagnostic ignored "-Wunused-variable"
|
|
|
|
//#include "ks.h"
|
|
#include "../src/dht/ks_dht.h"
|
|
|
|
ks_dhtrt_routetable_t* rt;
|
|
ks_pool_t* pool;
|
|
|
|
|
|
int doquery(ks_dhtrt_routetable_t* rt, uint8_t* id, enum ks_dht_nodetype_t type, enum ks_afflags_t family)
|
|
{
|
|
ks_dhtrt_querynodes_t query;
|
|
memset(&query, 0, sizeof(query));
|
|
query.max = 30;
|
|
memcpy(&query.nodeid.id, id, KS_DHT_NODEID_SIZE);
|
|
query.family = family;
|
|
query.type = type;
|
|
|
|
return ks_dhtrt_findclosest_nodes(rt, &query);
|
|
}
|
|
|
|
void test01()
|
|
{
|
|
printf("testbuckets - test01 start\n"); fflush(stdout);
|
|
ks_dhtrt_routetable_t* rt;
|
|
ks_dhtrt_initroute(&rt, pool);
|
|
ks_dhtrt_deinitroute(&rt);
|
|
|
|
ks_dhtrt_initroute(&rt, pool);
|
|
ks_dht_nodeid_t nodeid, homeid;
|
|
memset(homeid.id, 0xdd, KS_DHT_NODEID_SIZE);
|
|
homeid.id[19] = 0;
|
|
|
|
char ip[] = "192.168.100.100";
|
|
unsigned short port = 7000;
|
|
ks_dht_node_t* peer;
|
|
ks_dht_node_t* peer1;
|
|
|
|
ks_status_t status;
|
|
status = ks_dhtrt_create_node(rt, homeid, KS_DHT_LOCAL, ip, port, &peer);
|
|
if (status == KS_STATUS_FAIL) {
|
|
printf("*** ks_dhtrt_create_node test01 failed\n");
|
|
exit(101);
|
|
}
|
|
|
|
peer = ks_dhtrt_find_node(rt, homeid);
|
|
if (peer == 0) {
|
|
printf("*** ks_dhtrt_find_node test01 failed \n"); fflush(stdout);
|
|
exit(102);
|
|
}
|
|
|
|
status = ks_dhtrt_create_node(rt, homeid, KS_DHT_LOCAL, ip, port, &peer1);
|
|
if (status == KS_STATUS_FAIL) {
|
|
printf("*** ks_dhtrt_create_node test01 did allow duplicate createnodes!!\n");
|
|
exit(103);
|
|
}
|
|
if (peer != peer1) {
|
|
printf("*** ks_dhtrt_create_node duplicate createnode did not return the same node!\n");
|
|
exit(104);
|
|
}
|
|
|
|
status = ks_dhtrt_delete_node(rt, peer);
|
|
if (status == KS_STATUS_FAIL) {
|
|
printf("*** ks_dhtrt_delete_node test01 failed\n");
|
|
exit(104);
|
|
}
|
|
|
|
printf("*** testbuckets - test01 complete\n"); fflush(stdout);
|
|
}
|
|
|
|
void test02()
|
|
{
|
|
ks_dht_node_t* peer;
|
|
ks_dht_nodeid_t nodeid;
|
|
memset(nodeid.id, 0xef, KS_DHT_NODEID_SIZE);
|
|
|
|
char ipv6[] = "1234:1234:1234:1234";
|
|
char ipv4[] = "123.123.123.123";
|
|
unsigned short port = 7000;
|
|
enum ks_afflags_t both = ifboth;
|
|
|
|
ks_status_t status;
|
|
|
|
nodeid.id[0] = 1;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_LOCAL, ipv6, port, &peer);
|
|
nodeid.id[0] = 2;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv6, port, &peer);
|
|
nodeid.id[0] = 3;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv6, port, &peer);
|
|
nodeid.id[0] = 4;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_LOCAL, ipv6, port, &peer);
|
|
nodeid.id[1] = 1;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv6, port, &peer);
|
|
|
|
|
|
nodeid.id[19] = 1;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv4, port, &peer);
|
|
nodeid.id[19] = 2;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv4, port, &peer);
|
|
nodeid.id[19] = 3;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv4, port, &peer);
|
|
nodeid.id[19] = 4;
|
|
status = ks_dhtrt_create_node(rt, nodeid, KS_DHT_LOCAL, ipv4, port, &peer);
|
|
|
|
int qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, both);
|
|
printf("\n*** local query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, both);
|
|
printf("\n*** remote query count expected 6, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, both);
|
|
printf("\n*** both query count expected 9, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, ifv4);
|
|
printf("\n*** local AF_INET query count expected 1, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, ifv6);
|
|
printf("\n*** local AF_INET6 query count expected 2, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, ifv6);
|
|
printf("\n*** AF_INET6 count expected 5, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, ifv4);
|
|
printf("\n*** remote AF_INET query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, ifv6);
|
|
printf("\n*** remote AF_INET6 query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, ifv4);
|
|
printf("\n*** AF_INET count expected 4, actual %d\n", qcount); fflush(stdout);
|
|
|
|
return;
|
|
}
|
|
|
|
/* this is similar to test2 but after mutiple table splits. */
|
|
|
|
void test03()
|
|
{
|
|
ks_dht_node_t* peer;
|
|
ks_dht_nodeid_t nodeid;
|
|
memset(nodeid.id, 0xef, KS_DHT_NODEID_SIZE);
|
|
|
|
char ipv6[] = "1234:1234:1234:1234";
|
|
char ipv4[] = "123.123.123.123";
|
|
unsigned short port = 7000;
|
|
enum ks_afflags_t both = ifboth;
|
|
|
|
ks_status_t status;
|
|
|
|
for (int i=0; i<200; ++i) {
|
|
if (i%10 == 0) {
|
|
++nodeid.id[0];
|
|
}
|
|
else {
|
|
++nodeid.id[1];
|
|
}
|
|
ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv4, port, &peer);
|
|
}
|
|
|
|
for (int i=0; i<2; ++i) {
|
|
if (i%10 == 0) {
|
|
++nodeid.id[0];
|
|
}
|
|
else {
|
|
++nodeid.id[1];
|
|
}
|
|
|
|
ks_dhtrt_create_node(rt, nodeid, KS_DHT_LOCAL, ipv4, port, &peer);
|
|
}
|
|
|
|
for (int i=0; i<201; ++i) {
|
|
if (i%10 == 0) {
|
|
++nodeid.id[0];
|
|
}
|
|
else {
|
|
++nodeid.id[1];
|
|
}
|
|
ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv6, port, &peer);
|
|
}
|
|
|
|
|
|
int qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, both);
|
|
printf("\n** local query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, both);
|
|
printf("\n*** remote query count expected 6, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, both);
|
|
printf("\n*** both query count expected 9, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, ifv4);
|
|
printf("\n*** local AF_INET query count expected 1, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_LOCAL, ifv6);
|
|
printf("\n*** local AF_INET6 query count expected 2, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, ifv6);
|
|
printf("\n*** AF_INET6 count expected 5, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, ifv4);
|
|
printf("\n** remote AF_INET query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_REMOTE, ifv6);
|
|
printf("\n*** remote AF_INET6 query count expected 3, actual %d\n", qcount); fflush(stdout);
|
|
|
|
qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, ifv4);
|
|
printf("\n*** AF_INET count expected 4, actual %d\n", qcount); fflush(stdout);
|
|
|
|
return;
|
|
}
|
|
|
|
void test04()
|
|
{
|
|
ks_dht_node_t* peer;
|
|
ks_dht_nodeid_t nodeid;
|
|
memset(nodeid.id, 0xef, KS_DHT_NODEID_SIZE);
|
|
|
|
char ipv6[] = "1234:1234:1234:1234";
|
|
char ipv4[] = "123.123.123.123";
|
|
unsigned short port = 7000;
|
|
enum ks_afflags_t both = ifboth;
|
|
|
|
ks_status_t status;
|
|
|
|
for (int i=0,i2=0; i<10000; ++i) {
|
|
if (i%40 == 0) {
|
|
++nodeid.id[0];
|
|
if(i2%40 == 0) {
|
|
++nodeid.id[1];
|
|
}
|
|
else {
|
|
++nodeid.id[2];
|
|
}
|
|
}
|
|
else {
|
|
++nodeid.id[1];
|
|
}
|
|
ks_dhtrt_create_node(rt, nodeid, KS_DHT_REMOTE, ipv4, port, &peer);
|
|
}
|
|
|
|
|
|
memset(nodeid.id, 0x2f, KS_DHT_NODEID_SIZE);
|
|
ks_time_t t0 = ks_time_now();
|
|
int qcount = doquery(rt, nodeid.id, KS_DHT_BOTH, ifv4);
|
|
ks_time_t t1 = ks_time_now();
|
|
|
|
int tx = t1 - t0;
|
|
t1 /= 1000;
|
|
|
|
printf("*** query on 10k nodes in %d ms\n", tx);
|
|
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
|
|
int main(int argx, char* argv[]) {
|
|
|
|
printf("testdhtbuckets - start\n");
|
|
|
|
ks_init();
|
|
|
|
ks_status_t status;
|
|
char *str = NULL;
|
|
int bytes = 1024;
|
|
ks_dht_nodeid_t homeid;
|
|
ks_dht_nodeid_t nodeid, nodeid1, nodeid2;
|
|
ks_dht_node_t *peer, *peer1, *peer2;
|
|
|
|
memset(homeid.id, 0xde, KS_DHT_NODEID_SIZE);
|
|
memset(nodeid.id, 0xef, KS_DHT_NODEID_SIZE);
|
|
|
|
ks_init();
|
|
status = ks_pool_open(&pool);
|
|
|
|
printf("init/deinit routeable\n"); fflush(stdout);
|
|
|
|
ks_dhtrt_initroute(&rt, pool);
|
|
ks_dhtrt_deinitroute(&rt);
|
|
|
|
ks_dhtrt_initroute(&rt, pool);
|
|
test01();
|
|
test02();
|
|
test03();
|
|
test04();
|
|
|
|
return 0;
|
|
|
|
}
|