diff options
Diffstat (limited to 'svm/svmdb.c')
| -rw-r--r-- | svm/svmdb.c | 520 |
1 files changed, 520 insertions, 0 deletions
diff --git a/svm/svmdb.c b/svm/svmdb.c new file mode 100644 index 00000000000..515bc7d276b --- /dev/null +++ b/svm/svmdb.c @@ -0,0 +1,520 @@ +/* + *------------------------------------------------------------------ + * svmdb.c -- simple shared memory database + * + * Copyright (c) 2009 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *------------------------------------------------------------------ + */ + +#include <stdio.h> +#include <stdlib.h> +#include <sys/types.h> +#include <sys/mman.h> +#include <sys/stat.h> +#include <netinet/in.h> +#include <signal.h> +#include <pthread.h> +#include <unistd.h> +#include <time.h> +#include <fcntl.h> +#include <string.h> +#include <vppinfra/clib.h> +#include <vppinfra/vec.h> +#include <vppinfra/hash.h> +#include <vppinfra/bitmap.h> +#include <vppinfra/fifo.h> +#include <vppinfra/time.h> +#include <vppinfra/mheap.h> +#include <vppinfra/heap.h> +#include <vppinfra/pool.h> +#include <vppinfra/format.h> + +#include "svmdb.h" + +static void local_set_variable_nolock (svmdb_client_t *client, + svmdb_namespace_t namespace, + u8 * var, u8 * val, u32 elsize); + +always_inline void region_lock(svm_region_t *rp, int tag) +{ + pthread_mutex_lock(&rp->mutex); +#ifdef MUTEX_DEBUG + rp->mutex_owner_pid = getpid(); + rp->mutex_owner_tag = tag; +#endif +} + +always_inline void region_unlock(svm_region_t *rp) +{ +#ifdef MUTEX_DEBUG + rp->mutex_owner_pid = 0; + rp->mutex_owner_tag = 0; +#endif + pthread_mutex_unlock(&rp->mutex); +} + +static svmdb_client_t *svmdb_map_internal (char *root_path, uword size) +{ + svmdb_client_t *client = 0; + svm_map_region_args_t *a = 0; + svm_region_t *db_rp; + void *oldheap; + svmdb_shm_hdr_t *hp = 0; + + vec_validate (client, 0); + vec_validate (a, 0); + + svm_region_init_chroot(root_path); + + a->root_path = root_path; + a->name = "/db"; + a->size = size ? size : SVMDB_DEFAULT_SIZE; + a->flags = SVM_FLAGS_MHEAP; + + db_rp = client->db_rp = svm_region_find_or_create (a); + + ASSERT(db_rp); + + vec_free (a); + + region_lock (client->db_rp, 10); + /* Has someone else set up the shared-memory variable table? */ + if (db_rp->user_ctx) { + client->shm = (void *) db_rp->user_ctx; + client->pid = getpid(); + region_unlock (client->db_rp); + ASSERT (client->shm->version == SVMDB_SHM_VERSION); + return (client); + } + /* Nope, it's our problem... */ + + /* Add a bogus client (pid=0) so the svm won't be deallocated */ + oldheap = svm_push_pvt_heap (db_rp); + vec_add1(client->db_rp->client_pids, 0); + svm_pop_heap (oldheap); + + oldheap = svm_push_data_heap (db_rp); + + vec_validate(hp, 0); + hp->version = SVMDB_SHM_VERSION; + hp->namespaces[SVMDB_NAMESPACE_STRING] + = hash_create_string(0, sizeof(uword)); + hp->namespaces[SVMDB_NAMESPACE_VEC] + = hash_create_string(0, sizeof(uword)); + + db_rp->user_ctx = hp; + client->shm = hp; + + svm_pop_heap (oldheap); + region_unlock (client->db_rp); + client->pid = getpid(); + + return (client); +} +svmdb_client_t *svmdb_map (void) +{ + return svmdb_map_internal (0, 0); +} + +svmdb_client_t *svmdb_map_size (uword size) +{ + return svmdb_map_internal (0, size); +} + +svmdb_client_t *svmdb_map_chroot (char *root_path) +{ + return svmdb_map_internal (root_path, 0); +} + +svmdb_client_t *svmdb_map_chroot_size (char *root_path, uword size) +{ + return svmdb_map_internal (root_path, size); +} + +void svmdb_unmap (svmdb_client_t *client) +{ + ASSERT(client); + + if (! svm_get_root_rp()) + return; + + svm_region_unmap ((void *) client->db_rp); + svm_region_exit (); + vec_free(client); +} + +static void notify_value (svmdb_value_t * v, svmdb_action_t a) +{ + int i; + int rv; + union sigval sv; + u32 value; + u32 *dead_registrations = 0; + + svmdb_notify_t *np; + + for (i = 0; i < vec_len (v->notifications); i++) { + np = vec_elt_at_index (v->notifications, i); + if (np->action == a) { + value = (np->action<<28) | (np->opaque); + sv.sival_ptr = (void *)(uword)value; + do { + rv = 0; + if (sigqueue (np->pid, np->signum, sv) == 0) + break; + rv = errno; + } while (rv == EAGAIN); + if (rv == 0) + continue; + vec_add1 (dead_registrations, i); + } + } + + for (i = 0; i < vec_len (dead_registrations); i++) { + np = vec_elt_at_index (v->notifications, dead_registrations[i]); + clib_warning ("dead reg pid %d sig %d action %d opaque %x", + np->pid, np->signum, np->action, np->opaque); + vec_delete (v->notifications, 1, dead_registrations[i]); + } + vec_free (dead_registrations); +} + +int svmdb_local_add_del_notification (svmdb_client_t *client, + svmdb_notification_args_t *a) +{ + uword *h; + void *oldheap; + hash_pair_t *hp; + svmdb_shm_hdr_t * shm; + u8 *dummy_value = 0; + svmdb_value_t *value; + svmdb_notify_t *np; + int i; + int rv = 0; + + ASSERT (a->elsize); + + region_lock (client->db_rp, 18); + shm = client->shm; + oldheap = svm_push_data_heap (client->db_rp); + + h = shm->namespaces[a->nspace]; + + hp = hash_get_pair_mem (h, a->var); + if (hp == 0) { + local_set_variable_nolock (client, a->nspace, (u8 *)a->var, + dummy_value, a->elsize); + /* might have moved */ + h = shm->namespaces[a->nspace]; + hp = hash_get_pair_mem (h, a->var); + ASSERT(hp); + } + + value = pool_elt_at_index (shm->values, hp->value[0]); + + for (i = 0; i < vec_len (value->notifications); i++) { + np = vec_elt_at_index (value->notifications, i); + if ((np->pid == client->pid) + && (np->signum == a->signum) + && (np->action == a->action) + && (np->opaque == a->opaque)) { + if (a->add_del == 0 /* delete */) { + vec_delete (value->notifications, 1, i); + goto out; + } else { /* add */ + clib_warning ( + "%s: ignore dup reg pid %d signum %d action %d opaque %x", + a->var, client->pid, a->signum, a->action, a->opaque); + rv = -2; + goto out; + } + } + } + if (a->add_del == 0) { + rv = -3; + goto out; + } + + vec_add2 (value->notifications, np, 1); + np->pid = client->pid; + np->signum = a->signum; + np->action = a->action; + np->opaque = a->opaque; + +out: + svm_pop_heap(oldheap); + region_unlock (client->db_rp); + return rv; +} + + +static void local_unset_variable_nolock (svmdb_client_t *client, + svmdb_namespace_t namespace, + char * var) +{ + uword *h; + svmdb_value_t *oldvalue; + hash_pair_t *hp; + + h = client->shm->namespaces[namespace]; + hp = hash_get_pair_mem (h, var); + if (hp) { + oldvalue = pool_elt_at_index (client->shm->values, hp->value[0]); + if (vec_len (oldvalue->notifications)) + notify_value (oldvalue, SVMDB_ACTION_UNSET); + /* zero length value means unset */ + _vec_len (oldvalue->value) = 0; + } + client->shm->namespaces[namespace] = h; +} + +void svmdb_local_unset_string_variable (svmdb_client_t *client, char *var) +{ + void *oldheap; + + region_lock (client->db_rp, 11); + oldheap = svm_push_data_heap (client->db_rp); + local_unset_variable_nolock (client, SVMDB_NAMESPACE_STRING, var); + svm_pop_heap(oldheap); + region_unlock (client->db_rp); +} + +static void local_set_variable_nolock (svmdb_client_t *client, + svmdb_namespace_t namespace, + u8 * var, u8 * val, u32 elsize) +{ + uword *h; + hash_pair_t *hp; + u8 *name; + svmdb_shm_hdr_t * shm; + + shm = client->shm; + h = shm->namespaces[namespace]; + hp = hash_get_pair_mem (h, var); + if (hp) { + svmdb_value_t * oldvalue; + oldvalue = pool_elt_at_index (client->shm->values, hp->value[0]); + vec_alloc (oldvalue->value, vec_len(val)*elsize); + memcpy (oldvalue->value, val, vec_len(val)*elsize); + _vec_len (oldvalue->value) = vec_len(val); + notify_value (oldvalue, SVMDB_ACTION_SET); + } else { + svmdb_value_t * newvalue; + pool_get (shm->values, newvalue); + memset (newvalue, 0, sizeof (*newvalue)); + newvalue->elsize = elsize; + vec_alloc (newvalue->value, vec_len(val)*elsize); + memcpy (newvalue->value, val, vec_len(val)*elsize); + _vec_len (newvalue->value) = vec_len(val); + name = format (0, "%s%c", var, 0); + hash_set_mem (h, name, newvalue - shm->values); + } + shm->namespaces[namespace] = h; +} + +void svmdb_local_set_string_variable (svmdb_client_t *client, + char *var, char *val) +{ + void *oldheap; + + region_lock (client->db_rp, 12); + oldheap = svm_push_data_heap (client->db_rp); + + local_unset_variable_nolock (client, SVMDB_NAMESPACE_STRING, var); + + local_set_variable_nolock (client, SVMDB_NAMESPACE_STRING, + (u8 *) var, (u8 *) val, 1 /* elsize */); + svm_pop_heap(oldheap); + region_unlock (client->db_rp); +} + +static u8 * local_get_variable_nolock (svmdb_client_t *client, + svmdb_namespace_t namespace, + u8 * var) +{ + uword *h; + uword *p; + svmdb_shm_hdr_t * shm; + svmdb_value_t *oldvalue; + + shm = client->shm; + h = shm->namespaces[namespace]; + p = hash_get_mem (h, var); + if (p) { + oldvalue = pool_elt_at_index (shm->values, p[0]); + notify_value (oldvalue, SVMDB_ACTION_GET); + return (oldvalue->value); + } + return 0; +} + +void *svmdb_local_get_variable_reference (svmdb_client_t *client, + svmdb_namespace_t namespace, + char *var) +{ + u8 *rv; + + region_lock (client->db_rp, 19); + rv = local_get_variable_nolock (client, namespace, (u8 *)var); + region_unlock (client->db_rp); + return (void *)rv; +} + +char *svmdb_local_get_string_variable (svmdb_client_t *client, char *var) +{ + u8 *rv = 0; + + region_lock (client->db_rp, 13); + rv = local_get_variable_nolock (client, SVMDB_NAMESPACE_STRING, (u8 *) var); + + if (rv && vec_len (rv)) { + rv = format (0, "%s", rv); + vec_add1(rv, 0); + } + region_unlock (client->db_rp); + return ((char *) rv); +} + +void svmdb_local_dump_strings (svmdb_client_t *client) +{ + uword *h; + u8 *key; + u32 value; + svmdb_shm_hdr_t *shm = client->shm; + + region_lock (client->db_rp, 14); + + h = client->shm->namespaces [SVMDB_NAMESPACE_STRING]; + + hash_foreach_mem(key, value, h, + ({ + svmdb_value_t *v = pool_elt_at_index (shm->values, value); + + fformat(stdout, "%s: %s\n", key, + vec_len(v->value) ? v->value : (u8 *)"(nil)"); + })); + region_unlock (client->db_rp); +} + +void svmdb_local_unset_vec_variable (svmdb_client_t *client, char *var) +{ + void *oldheap; + + region_lock (client->db_rp, 15); + oldheap = svm_push_data_heap (client->db_rp); + local_unset_variable_nolock (client, SVMDB_NAMESPACE_VEC, var); + svm_pop_heap(oldheap); + region_unlock (client->db_rp); +} + +void svmdb_local_set_vec_variable (svmdb_client_t *client, + char *var, void *val_arg, u32 elsize) +{ + u8 *val = (u8 *)val_arg; + void *oldheap; + + region_lock (client->db_rp, 16); + oldheap = svm_push_data_heap (client->db_rp); + + local_unset_variable_nolock (client, SVMDB_NAMESPACE_VEC, var); + local_set_variable_nolock (client, SVMDB_NAMESPACE_VEC, (u8 *) var, + val, elsize); + + svm_pop_heap(oldheap); + region_unlock (client->db_rp); +} + +void *svmdb_local_get_vec_variable (svmdb_client_t *client, char *var, + u32 elsize) +{ + u8 *rv = 0; + u8 *copy = 0; + + region_lock (client->db_rp, 17); + + rv = local_get_variable_nolock (client, SVMDB_NAMESPACE_VEC, (u8 *) var); + + if (rv && vec_len(rv)) { + /* Make a copy in process-local memory */ + vec_alloc (copy, vec_len(rv)*elsize); + memcpy (copy, rv, vec_len(rv)*elsize); + _vec_len(copy) = vec_len(rv); + region_unlock (client->db_rp); + return (copy); + } + region_unlock (client->db_rp); + return (0); +} + +void svmdb_local_dump_vecs (svmdb_client_t *client) +{ + uword *h; + u8 *key; + u32 value; + svmdb_shm_hdr_t *shm; + + region_lock (client->db_rp, 17); + shm = client->shm; + + h = client->shm->namespaces [SVMDB_NAMESPACE_VEC]; + + hash_foreach_mem(key, value, h, + ({ + svmdb_value_t *v = pool_elt_at_index (shm->values, value); + (void) fformat(stdout, "%s:\n %U\n", key, + format_hex_bytes, v->value, + vec_len(v->value)*v->elsize); + })); + + region_unlock (client->db_rp); +} + +void *svmdb_local_find_or_add_vec_variable (svmdb_client_t *client, + char *var, u32 nbytes) +{ + void *oldheap; + u8 *rv = 0; + + region_lock (client->db_rp, 18); + oldheap = svm_push_data_heap (client->db_rp); + + rv = local_get_variable_nolock (client, SVMDB_NAMESPACE_VEC, (u8 *)var); + + if (rv) { + goto out; + } else { + uword *h; + u8 *name; + svmdb_shm_hdr_t * shm; + svmdb_value_t * newvalue; + + shm = client->shm; + h = shm->namespaces[SVMDB_NAMESPACE_VEC]; + + pool_get (shm->values, newvalue); + memset (newvalue, 0, sizeof (*newvalue)); + newvalue->elsize = 1; + vec_alloc (newvalue->value, nbytes); + _vec_len (newvalue->value) = nbytes; + name = format (0, "%s%c", var, 0); + hash_set_mem (h, name, newvalue - shm->values); + shm->namespaces[SVMDB_NAMESPACE_VEC] = h; + rv = newvalue->value; + } + +out: + svm_pop_heap(oldheap); + region_unlock (client->db_rp); + return (rv); +} |