blob: 79e43e7e9f746e26158b4f520782807b6ab7daf8 [file] [log] [blame]
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <time.h>
#include <osmocom/core/utils.h>
#include <osmocom/core/application.h>
#include <osmocom/core/logging.h>
#include "../../lib/in46_addr.h"
#include "../../lib/ippool.h"
#include "../../lib/syserr.h"
static struct ippool_t *create_pool(const char *prefix_str, unsigned int flags)
{
struct ippool_t *pool;
struct in46_prefix pfx;
size_t t;
int rc;
/* dynamic-only v4 */
rc = ippool_aton(&pfx.addr, &t, prefix_str, flags);
OSMO_ASSERT(rc == 0);
pfx.prefixlen = t;
rc = ippool_new(&pool, &pfx, NULL, flags);
OSMO_ASSERT(rc == 0);
//ippool_printaddr(pool);
return pool;
}
static void test_pool_size(const char *pfx, unsigned int flags, unsigned int expected_size)
{
struct ippool_t *pool;
struct ippoolm_t *member;
struct in46_addr addr;
int i, rc, n;
printf("testing pool for prefix %s, flags=0x%x, expected_size=%u\n", pfx, flags, expected_size);
pool = create_pool(pfx, flags);
OSMO_ASSERT(pool->listsize == expected_size);
memset(&addr, 0, sizeof(addr));
addr.len = pool->member[0].addr.len;
/* allocate all addresses */
for (i = 0; i < expected_size; i++) {
member = NULL;
rc = ippool_newip(pool, &member, &addr, 0);
OSMO_ASSERT(rc == 0);
OSMO_ASSERT(member);
printf("allocated address %s\n", in46a_ntoa(&member->addr));
}
/* allocate one more, expect that to fail */
rc = ippool_newip(pool, &member, &addr, 0);
OSMO_ASSERT(rc < 0);
/* release a (random) number N of (random) member address */
n = rand() % pool->listsize;
for (i = 0; i < n; i++) {
int r;
/* chose a random index that is in use */
do {
r = rand() % pool->listsize;
} while (!pool->member[r].inuse);
/* and free it... */
rc = ippool_freeip(pool, &pool->member[r]);
OSMO_ASSERT(rc == 0);
}
/* allocate all N previously released addresses */
for (i = 0; i < n; i++) {
member = NULL;
rc = ippool_newip(pool, &member, &addr, 0);
OSMO_ASSERT(rc == 0);
OSMO_ASSERT(member);
}
/* allocate one more, expect that to fail */
rc = ippool_newip(pool, &member, &addr, 0);
OSMO_ASSERT(rc < 0);
ippool_free(pool);
}
static void test_pool_sizes(void)
{
/* 255 addresses [0..254] */
test_pool_size("192.168.23.0/24", 0, 255);
/* 254 addresses [1..254] */
test_pool_size("192.168.23.0/24", IPPOOL_NONETWORK, 254);
/* 253 addresses [2..254] */
test_pool_size("192.168.23.0/24", IPPOOL_NOGATEWAY, 253);
/* 65533 addresses [3..255.254] */
test_pool_size("192.168.0.0/16", IPPOOL_NOGATEWAY, 65533);
/* 255 prefixes of /64 each */
test_pool_size("2001:DB8::/56", 0, 255);
}
int main(int argc, char **argv)
{
osmo_init_logging(&log_info);
log_set_use_color(osmo_stderr_target, 0);
log_set_print_filename(osmo_stderr_target, 0);
srand(time(NULL));
test_pool_sizes();
}