#include "testutil.h"
#include "apr.h"
#include "apu.h"
#include "apr_general.h"
#include "apr_strings.h"
#include "apr_hash.h"
#include "apr_memcache.h"
#include "apr_network_io.h"
#if APR_HAVE_STDLIB_H
#include <stdlib.h>
#endif
#define HOST "localhost"
#define PORT 11211
#define TDATA_SIZE 3000
#define TDATA_SET 100
#define HASH_FUNC_RESULT 510
const char prefix[] = "testmemcache";
const char txt[] =
"Lorem ipsum dolor sit amet, consectetuer adipiscing elit. Duis at"
"lacus in ligula hendrerit consectetuer. Vestibulum tristique odio"
"iaculis leo. In massa arcu, ultricies a, laoreet nec, hendrerit non,"
"neque. Nulla sagittis sapien ac risus. Morbi ligula dolor, vestibulum"
"nec, viverra id, placerat dapibus, arcu. Curabitur egestas feugiat"
"tellus. Donec dignissim. Nunc ante. Curabitur id lorem. In mollis"
"tortor sit amet eros auctor dapibus. Proin nulla sem, tristique in,"
"convallis id, iaculis feugiat cras amet.";
typedef struct {
const char *someval;
apr_uint32_t which_server;
} my_hash_server_baton;
static apr_uint32_t my_hash_func(void *baton, const char *data,
apr_size_t data_len)
{
return HASH_FUNC_RESULT;
}
static apr_memcache_server_t *my_server_func(void *baton,
apr_memcache_t *mc,
const apr_uint32_t hash)
{
apr_memcache_server_t *ms = NULL;
my_hash_server_baton *mhsb = (my_hash_server_baton *)baton;
if(mc->ntotal == 0) {
return NULL;
}
if(mc->ntotal < mhsb->which_server) {
return NULL;
}
ms = mc->live_servers[mhsb->which_server - 1];
return ms;
}
apr_uint16_t firsttime = 0;
static int randval(apr_uint32_t high)
{
apr_uint32_t i = 0;
double d = 0;
if (firsttime == 0) {
srand((unsigned) (getpid()));
firsttime = 1;
}
d = (double) rand() / ((double) RAND_MAX + 1);
i = (int) (d * (high - 0 + 1));
return i > 0 ? i : 1;
}
static void test_memcache_create(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *server, *s;
apr_uint32_t max_servers = 10;
apr_uint32_t i;
apr_uint32_t hash;
rv = apr_memcache_create(pool, max_servers, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
for (i = 1; i <= max_servers; i++) {
apr_port_t port;
port = PORT + i;
rv =
apr_memcache_server_create(pool, HOST, PORT + i, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
s = apr_memcache_find_server(memcache, HOST, port);
ABTS_PTR_EQUAL(tc, server, s);
rv = apr_memcache_disable_server(memcache, s);
ABTS_ASSERT(tc, "server disable failed", rv == APR_SUCCESS);
rv = apr_memcache_enable_server(memcache, s);
ABTS_ASSERT(tc, "server enable failed", rv == APR_SUCCESS);
hash = apr_memcache_hash(memcache, prefix, strlen(prefix));
ABTS_ASSERT(tc, "hash failed", hash > 0);
s = apr_memcache_find_server_hash(memcache, hash);
ABTS_PTR_NOTNULL(tc, s);
}
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add should have failed", rv != APR_SUCCESS);
}
static int create_test_hash(apr_pool_t *p, apr_hash_t *h)
{
int i;
for (i = 0; i < TDATA_SIZE; i++) {
char *k, *v;
k = apr_pstrcat(p, prefix, apr_itoa(p, i), NULL);
v = apr_pstrndup(p, txt, randval((apr_uint32_t)strlen(txt)));
apr_hash_set(h, k, APR_HASH_KEY_STRING, v);
}
return i;
}
static void test_memcache_user_funcs(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *found;
apr_uint32_t max_servers = 10;
apr_uint32_t hres;
apr_uint32_t i;
my_hash_server_baton *baton =
apr_pcalloc(pool, sizeof(my_hash_server_baton));
rv = apr_memcache_create(pool, max_servers, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
memcache->hash_func = my_hash_func;
hres = apr_memcache_hash(memcache, "whatever", sizeof("whatever") - 1);
ABTS_INT_EQUAL(tc, HASH_FUNC_RESULT, hres);
for(i = 1; i <= 10; i++) {
apr_memcache_server_t *ms;
rv = apr_memcache_server_create(pool, HOST, i, 0, 1, 1, 60, &ms);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, ms);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
}
baton->which_server = 3;
memcache->server_func = my_server_func;
memcache->server_baton = baton;
found = apr_memcache_find_server_hash(memcache, 0);
ABTS_ASSERT(tc, "wrong server found", found->port == baton->which_server);
}
static void test_memcache_meta(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_memcache_t *memcache;
apr_memcache_server_t *server;
apr_memcache_stats_t *stats;
char *result;
apr_status_t rv;
rv = apr_memcache_create(pool, 1, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
rv = apr_memcache_version(server, pool, &result);
ABTS_PTR_NOTNULL(tc, result);
rv = apr_memcache_stats(server, p, &stats);
ABTS_PTR_NOTNULL(tc, stats);
ABTS_STR_NEQUAL(tc, stats->version, result, 5);
ABTS_ASSERT(tc, "pid", stats->pid >= 0);
ABTS_ASSERT(tc, "time", stats->time >= 0);
ABTS_ASSERT(tc, "rusage_user", stats->rusage_user >= 0);
ABTS_ASSERT(tc, "rusage_system", stats->rusage_system >= 0);
ABTS_ASSERT(tc, "curr_items", stats->curr_items >= 0);
ABTS_ASSERT(tc, "total_items", stats->total_items >= 0);
ABTS_ASSERT(tc, "bytes", stats->bytes >= 0);
ABTS_ASSERT(tc, "curr_connections", stats->curr_connections >= 0);
ABTS_ASSERT(tc, "total_connections", stats->total_connections >= 0);
ABTS_ASSERT(tc, "connection_structures",
stats->connection_structures >= 0);
ABTS_ASSERT(tc, "cmd_get", stats->cmd_get >= 0);
ABTS_ASSERT(tc, "cmd_set", stats->cmd_set >= 0);
ABTS_ASSERT(tc, "get_hits", stats->get_hits >= 0);
ABTS_ASSERT(tc, "get_misses", stats->get_misses >= 0);
ABTS_ASSERT(tc, "bytes_read", stats->bytes_read >= 0);
ABTS_ASSERT(tc, "bytes_written", stats->bytes_written >= 0);
ABTS_ASSERT(tc, "limit_maxbytes", stats->limit_maxbytes >= 0);
}
static void test_memcache_addreplace(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *server;
apr_hash_t *tdata;
apr_hash_index_t *hi;
char *result;
apr_size_t len;
rv = apr_memcache_create(pool, 1, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
tdata = apr_hash_make(p);
create_test_hash(pool, tdata);
for (hi = apr_hash_first(p, tdata); hi; hi = apr_hash_next(hi)) {
const void *k;
void *v;
const char *key;
apr_hash_this(hi, &k, NULL, &v);
key = k;
rv = apr_memcache_replace(memcache, key, v, strlen(v) - 1, 0, 27);
ABTS_ASSERT(tc, "replace should have failed", rv != APR_SUCCESS);
rv = apr_memcache_add(memcache, key, v, strlen(v), 0, 27);
ABTS_ASSERT(tc, "add failed", rv == APR_SUCCESS);
rv = apr_memcache_replace(memcache, key, "new", sizeof("new") - 1, 0, 27);
ABTS_ASSERT(tc, "replace failed", rv == APR_SUCCESS);
rv = apr_memcache_getp(memcache, pool, key, &result, &len, NULL);
ABTS_ASSERT(tc, "get failed", rv == APR_SUCCESS);
ABTS_STR_NEQUAL(tc, result, "new", 3);
rv = apr_memcache_add(memcache, key, v, strlen(v), 0, 27);
ABTS_ASSERT(tc, "add should have failed", rv != APR_SUCCESS);
rv = apr_memcache_delete(memcache, key, 0);
ABTS_ASSERT(tc, "delete failed", rv == APR_SUCCESS);
}
}
static void test_memcache_incrdecr(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *server;
apr_uint32_t new;
char *result;
apr_size_t len;
apr_uint32_t i;
rv = apr_memcache_create(pool, 1, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
rv = apr_memcache_set(memcache, prefix, "271", sizeof("271") - 1, 0, 27);
ABTS_ASSERT(tc, "set failed", rv == APR_SUCCESS);
for( i = 1; i <= TDATA_SIZE; i++) {
apr_uint32_t expect;
rv = apr_memcache_getp(memcache, pool, prefix, &result, &len, NULL);
ABTS_ASSERT(tc, "get failed", rv == APR_SUCCESS);
expect = i + atoi(result);
rv = apr_memcache_incr(memcache, prefix, i, &new);
ABTS_ASSERT(tc, "incr failed", rv == APR_SUCCESS);
ABTS_INT_EQUAL(tc, expect, new);
rv = apr_memcache_decr(memcache, prefix, i, &new);
ABTS_ASSERT(tc, "decr failed", rv == APR_SUCCESS);
ABTS_INT_EQUAL(tc, atoi(result), new);
}
rv = apr_memcache_getp(memcache, pool, prefix, &result, &len, NULL);
ABTS_ASSERT(tc, "get failed", rv == APR_SUCCESS);
ABTS_INT_EQUAL(tc, 271, atoi(result));
rv = apr_memcache_delete(memcache, prefix, 0);
ABTS_ASSERT(tc, "delete failed", rv == APR_SUCCESS);
}
static void test_memcache_multiget(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_pool_t *tmppool;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *server;
apr_hash_t *tdata, *values;
apr_hash_index_t *hi;
apr_uint32_t i;
rv = apr_memcache_create(pool, 1, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
tdata = apr_hash_make(p);
create_test_hash(pool, tdata);
for (hi = apr_hash_first(p, tdata); hi; hi = apr_hash_next(hi)) {
const void *k;
void *v;
const char *key;
apr_hash_this(hi, &k, NULL, &v);
key = k;
rv = apr_memcache_set(memcache, key, v, strlen(v), 0, 27);
ABTS_ASSERT(tc, "set failed", rv == APR_SUCCESS);
}
rv = apr_pool_create(&tmppool, pool);
for (i = 0; i < TDATA_SET; i++)
apr_memcache_add_multget_key(pool,
apr_pstrcat(pool, prefix,
apr_itoa(pool, i), NULL),
&values);
rv = apr_memcache_multgetp(memcache,
tmppool,
pool,
values);
ABTS_ASSERT(tc, "multgetp failed", rv == APR_SUCCESS);
ABTS_ASSERT(tc, "multgetp returned too few results",
apr_hash_count(values) == TDATA_SET);
for (hi = apr_hash_first(p, tdata); hi; hi = apr_hash_next(hi)) {
const void *k;
const char *key;
apr_hash_this(hi, &k, NULL, NULL);
key = k;
rv = apr_memcache_delete(memcache, key, 0);
ABTS_ASSERT(tc, "delete failed", rv == APR_SUCCESS);
}
}
static void test_memcache_setget(abts_case * tc, void *data)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_memcache_t *memcache;
apr_memcache_server_t *server;
apr_hash_t *tdata;
apr_hash_index_t *hi;
char *result;
apr_size_t len;
rv = apr_memcache_create(pool, 1, 0, &memcache);
ABTS_ASSERT(tc, "memcache create failed", rv == APR_SUCCESS);
rv = apr_memcache_server_create(pool, HOST, PORT, 0, 1, 1, 60, &server);
ABTS_ASSERT(tc, "server create failed", rv == APR_SUCCESS);
rv = apr_memcache_add_server(memcache, server);
ABTS_ASSERT(tc, "server add failed", rv == APR_SUCCESS);
tdata = apr_hash_make(pool);
create_test_hash(pool, tdata);
for (hi = apr_hash_first(p, tdata); hi; hi = apr_hash_next(hi)) {
const void *k;
void *v;
const char *key;
apr_hash_this(hi, &k, NULL, &v);
key = k;
rv = apr_memcache_set(memcache, key, v, strlen(v), 0, 27);
ABTS_ASSERT(tc, "set failed", rv == APR_SUCCESS);
rv = apr_memcache_getp(memcache, pool, key, &result, &len, NULL);
ABTS_ASSERT(tc, "get failed", rv == APR_SUCCESS);
}
rv = apr_memcache_getp(memcache, pool, "nothere3423", &result, &len, NULL);
ABTS_ASSERT(tc, "get should have failed", rv != APR_SUCCESS);
for (hi = apr_hash_first(p, tdata); hi; hi = apr_hash_next(hi)) {
const void *k;
const char *key;
apr_hash_this(hi, &k, NULL, NULL);
key = k;
rv = apr_memcache_delete(memcache, key, 0);
ABTS_ASSERT(tc, "delete failed", rv == APR_SUCCESS);
}
}
static apr_status_t check_mc(void)
{
apr_pool_t *pool = p;
apr_status_t rv;
apr_socket_t *sock = NULL;
apr_sockaddr_t *sa;
struct iovec vec[2];
apr_size_t written;
char buf[128];
apr_size_t len;
rv = apr_socket_create(&sock, APR_INET, SOCK_STREAM, 0, pool);
if(rv != APR_SUCCESS) {
return rv;
}
rv = apr_sockaddr_info_get(&sa, HOST, APR_INET, PORT, 0, pool);
if(rv != APR_SUCCESS) {
return rv;
}
rv = apr_socket_timeout_set(sock, 1 * APR_USEC_PER_SEC);
if (rv != APR_SUCCESS) {
return rv;
}
rv = apr_socket_connect(sock, sa);
if (rv != APR_SUCCESS) {
return rv;
}
rv = apr_socket_timeout_set(sock, -1);
if (rv != APR_SUCCESS) {
return rv;
}
vec[0].iov_base = "version";
vec[0].iov_len = sizeof("version") - 1;
vec[1].iov_base = "\r\n";
vec[1].iov_len = sizeof("\r\n") -1;
rv = apr_socket_sendv(sock, vec, 2, &written);
if (rv != APR_SUCCESS) {
return rv;
}
len = sizeof(buf);
rv = apr_socket_recv(sock, buf, &len);
if(rv != APR_SUCCESS) {
return rv;
}
if(strncmp(buf, "VERSION", sizeof("VERSION")-1) != 0) {
rv = APR_EGENERAL;
}
apr_socket_close(sock);
return rv;
}
abts_suite *testmemcache(abts_suite * suite)
{
apr_status_t rv;
suite = ADD_SUITE(suite);
rv = check_mc();
if(rv == APR_SUCCESS) {
abts_run_test(suite, test_memcache_create, NULL);
abts_run_test(suite, test_memcache_user_funcs, NULL);
abts_run_test(suite, test_memcache_meta, NULL);
abts_run_test(suite, test_memcache_setget, NULL);
abts_run_test(suite, test_memcache_multiget, NULL);
abts_run_test(suite, test_memcache_addreplace, NULL);
abts_run_test(suite, test_memcache_incrdecr, NULL);
}
return suite;
}