/*
- * Copyright (C) 2013 - David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2013 David Goulet <dgoulet@efficios.com>
*
- * This program is free software; you can redistribute it and/or modify it
- * under the terms of the GNU General Public License, version 2 only, as
- * published by the Free Software Foundation.
+ * SPDX-License-Identifier: GPL-2.0-only
*
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
- * more details.
- *
- * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 51
- * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-#define _GNU_SOURCE
#define _LGPL_SOURCE
#include <inttypes.h>
#include "buffer-registry.h"
#include "fd-limit.h"
#include "ust-consumer.h"
-#include "ust-ctl.h"
+#include "lttng-ust-ctl.h"
+#include "lttng-ust-error.h"
#include "utils.h"
/*
* Hash function for the per UID registry hash table. This XOR the triplet
* together.
*/
-static unsigned long ht_hash_reg_uid(void *_key, unsigned long seed)
+static unsigned long ht_hash_reg_uid(const void *_key, unsigned long seed)
{
uint64_t xored_key;
- struct buffer_reg_uid *key = _key;
+ const struct buffer_reg_uid *key = _key;
assert(key);
goto error_session;
}
+ reg->registry->maps = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
+ if (!reg->registry->maps) {
+ lttng_ht_destroy(reg->registry->channels);
+ ret = -ENOMEM;
+ goto error_session;
+ }
+
+
cds_lfht_node_init(®->node.node);
*regp = reg;
goto error_session;
}
+ reg->registry->maps = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
+ if (!reg->registry->maps) {
+ lttng_ht_destroy(reg->registry->channels);
+ ret = -ENOMEM;
+ goto error_session;
+ }
+
lttng_ht_node_init_u64(®->node, reg->session_id);
*regp = reg;
return reg;
}
+/*
+ * Find the consumer channel key from a UST session per-uid channel key.
+ *
+ * Return the matching key or -1 if not found.
+ */
+int buffer_reg_uid_consumer_channel_key(
+ struct cds_list_head *buffer_reg_uid_list,
+ uint64_t chan_key, uint64_t *consumer_chan_key)
+{
+ struct lttng_ht_iter iter;
+ struct buffer_reg_uid *uid_reg = NULL;
+ struct buffer_reg_session *session_reg = NULL;
+ struct buffer_reg_channel *reg_chan;
+ int ret = -1;
+
+ rcu_read_lock();
+ /*
+ * For the per-uid registry, we have to iterate since we don't have the
+ * uid and bitness key.
+ */
+ cds_list_for_each_entry(uid_reg, buffer_reg_uid_list, lnode) {
+ session_reg = uid_reg->registry;
+ cds_lfht_for_each_entry(session_reg->channels->ht,
+ &iter.iter, reg_chan, node.node) {
+ if (reg_chan->key == chan_key) {
+ *consumer_chan_key = reg_chan->consumer_key;
+ ret = 0;
+ goto end;
+ }
+ }
+ }
+
+end:
+ rcu_read_unlock();
+ return ret;
+}
+
/*
* Allocate and initialize a buffer registry channel with the given key. Set
* regp with the object pointer.
return 0;
}
+/*
+ * Allocate and initialize a buffer registry map with the given key. Set
+ * regp with the object pointer.
+ *
+ * Return 0 on success or else a negative value keeping regp untouched.
+ */
+int buffer_reg_map_create(uint64_t key, struct buffer_reg_map **regp)
+{
+ struct buffer_reg_map *reg;
+
+ assert(regp);
+
+ DBG3("Buffer registry map create with key: %" PRIu64, key);
+
+ reg = zmalloc(sizeof(*reg));
+ if (!reg) {
+ PERROR("zmalloc buffer registry map");
+ return -ENOMEM;
+ }
+
+ reg->key = key;
+ CDS_INIT_LIST_HEAD(®->counters);
+ pthread_mutex_init(®->counter_list_lock, NULL);
+
+ lttng_ht_node_init_u64(®->node, key);
+ *regp = reg;
+
+ return 0;
+}
+
/*
* Allocate and initialize a buffer registry stream. Set regp with the object
* pointer.
return 0;
}
+/*
+ * Allocate and initialize a buffer registry map_counter. Set regp with the object
+ * pointer.
+ *
+ * Return 0 on success or else a negative value keeping regp untouched.
+ */
+int buffer_reg_map_counter_create(struct buffer_reg_map_counter **regp)
+{
+ struct buffer_reg_map_counter *reg;
+
+ assert(regp);
+
+ DBG3("Buffer registry creating map_counter");
+
+ reg = zmalloc(sizeof(*reg));
+ if (!reg) {
+ PERROR("zmalloc buffer registry map_counter");
+ return -ENOMEM;
+ }
+
+ *regp = reg;
+
+ return 0;
+}
+
/*
* Add stream to the list in the channel.
*/
pthread_mutex_unlock(&channel->stream_list_lock);
}
+/*
+ * Add map_counter to the list in the map.
+ */
+void buffer_reg_map_counter_add(struct buffer_reg_map_counter *map_counter,
+ struct buffer_reg_map *map)
+{
+ assert(map_counter);
+ assert(map);
+
+ pthread_mutex_lock(&map->counter_list_lock);
+ cds_list_add_tail(&map_counter->lnode, &map->counters);
+ map->counter_count++;
+ pthread_mutex_unlock(&map->counter_list_lock);
+}
+
/*
* Add a buffer registry channel object to the given session.
*/
rcu_read_unlock();
}
+/*
+ * Add a buffer registry map object to the given session.
+ */
+void buffer_reg_map_add(struct buffer_reg_session *session,
+ struct buffer_reg_map *map)
+{
+ assert(session);
+ assert(map);
+
+ rcu_read_lock();
+ lttng_ht_add_unique_u64(session->maps, &map->node);
+ rcu_read_unlock();
+}
+
/*
* Find a buffer registry channel object with the given key. RCU read side lock
* MUST be acquired and hold on until the object reference is not needed
return chan;
}
+/*
+ * Find a buffer registry map object with the given key. RCU read side lock
+ * MUST be acquired and hold on until the object reference is not needed
+ * anymore.
+ *
+ * Return the object pointer or NULL on error.
+ */
+struct buffer_reg_map *buffer_reg_map_find(uint64_t key,
+ struct buffer_reg_uid *reg)
+{
+ struct lttng_ht_node_u64 *node;
+ struct lttng_ht_iter iter;
+ struct buffer_reg_map *map = NULL;
+ struct lttng_ht *ht;
+
+ assert(reg);
+
+ switch (reg->domain) {
+ case LTTNG_DOMAIN_UST:
+ ht = reg->registry->maps;
+ break;
+ default:
+ assert(0);
+ goto end;
+ }
+
+ lttng_ht_lookup(ht, &key, &iter);
+ node = lttng_ht_iter_get_node_u64(&iter);
+ if (!node) {
+ goto end;
+ }
+ map = caa_container_of(node, struct buffer_reg_map, node);
+
+end:
+ return map;
+}
+
/*
* Destroy a buffer registry stream with the given domain.
*/
{
int ret;
- ret = ust_ctl_release_object(-1, regp->obj.ust);
+ ret = ust_app_release_object(NULL, regp->obj.ust);
if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
ERR("Buffer reg stream release obj handle %d failed with ret %d",
regp->obj.ust->handle, ret);
return;
}
+/*
+ * Destroy a buffer registry map_counter with the given domain.
+ */
+void buffer_reg_map_counter_destroy(struct buffer_reg_map_counter *regp,
+ enum lttng_domain_type domain)
+{
+ if (!regp) {
+ return;
+ }
+
+ DBG3("Buffer registry map counter destroy with handle %d",
+ regp->obj.ust->handle);
+
+ switch (domain) {
+ case LTTNG_DOMAIN_UST:
+ {
+ int ret;
+
+ ret = ust_app_release_object(NULL, regp->obj.ust);
+ if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
+ ERR("Buffer reg map counter release obj handle %d failed with ret %d",
+ regp->obj.ust->handle, ret);
+ }
+ free(regp->obj.ust);
+ lttng_fd_put(LTTNG_FD_APPS, 2);
+ break;
+ }
+ default:
+ assert(0);
+ }
+
+ free(regp);
+ return;
+}
+
/*
* Remove buffer registry channel object from the session hash table. RCU read
* side lock MUST be acquired before calling this.
assert(!ret);
}
+/*
+ * Remove buffer registry map object from the session hash table. RCU read
+ * side lock MUST be acquired before calling this.
+ */
+void buffer_reg_map_remove(struct buffer_reg_session *session,
+ struct buffer_reg_map *regp)
+{
+ int ret;
+ struct lttng_ht_iter iter;
+
+ assert(session);
+ assert(regp);
+
+ iter.iter.node = ®p->node.node;
+ ret = lttng_ht_del(session->maps, &iter);
+ assert(!ret);
+}
+
/*
* Destroy a buffer registry channel with the given domain.
*/
}
if (regp->obj.ust) {
- ret = ust_ctl_release_object(-1, regp->obj.ust);
+ ret = ust_app_release_object(NULL, regp->obj.ust);
if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
ERR("Buffer reg channel release obj handle %d failed with ret %d",
regp->obj.ust->handle, ret);
return;
}
+/*
+ * Destroy a buffer registry map with the given domain.
+ */
+void buffer_reg_map_destroy(struct buffer_reg_map *regp,
+ enum lttng_domain_type domain)
+{
+ if (!regp) {
+ return;
+ }
+
+ DBG3("Buffer registry map destroy with key %" PRIu32, regp->key);
+
+ switch (domain) {
+ case LTTNG_DOMAIN_UST:
+ {
+ int ret;
+ struct buffer_reg_map_counter *map_counter_reg, *tmp;
+ /* Wipe counter */
+ cds_list_for_each_entry_safe(map_counter_reg, tmp, ®p->counters, lnode) {
+ cds_list_del(&map_counter_reg->lnode);
+ regp->counter_count--;
+ buffer_reg_map_counter_destroy(map_counter_reg, domain);
+ }
+
+ if (regp->obj.ust) {
+ ret = ust_app_release_object(NULL, regp->obj.ust);
+ if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
+ ERR("Buffer reg map release obj handle %d failed with ret %d",
+ regp->obj.ust->handle, ret);
+ }
+ free(regp->obj.ust);
+ }
+ lttng_fd_put(LTTNG_FD_APPS, 1);
+ break;
+ }
+ default:
+ assert(0);
+ }
+
+ free(regp);
+ return;
+}
+
/*
* Destroy a buffer registry session with the given domain.
*
int ret;
struct lttng_ht_iter iter;
struct buffer_reg_channel *reg_chan;
+ struct buffer_reg_map *reg_map;
DBG3("Buffer registry session destroy");
assert(!ret);
buffer_reg_channel_destroy(reg_chan, domain);
}
+ cds_lfht_for_each_entry(regp->maps->ht, &iter.iter, reg_map,
+ node.node) {
+ ret = lttng_ht_del(regp->maps, &iter);
+ assert(!ret);
+ buffer_reg_map_destroy(reg_map, domain);
+ }
rcu_read_unlock();
ht_cleanup_push(regp->channels);
+ ht_cleanup_push(regp->maps);
switch (domain) {
case LTTNG_DOMAIN_UST:
projects / lttng-tools.git / blobdiff