/*
- * 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"
/*
struct buffer_reg_uid *reg;
const struct buffer_reg_uid *key;
- assert(node);
- assert(_key);
+ LTTNG_ASSERT(node);
+ LTTNG_ASSERT(_key);
reg = caa_container_of(node, struct buffer_reg_uid, node.node);
- assert(reg);
+ LTTNG_ASSERT(reg);
key = _key;
if (key->session_id != reg->session_id ||
* 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);
+ LTTNG_ASSERT(key);
xored_key = (uint64_t)(key->session_id ^ key->bits_per_long ^ key->uid);
return hash_key_u64(&xored_key, seed);
void buffer_reg_init_uid_registry(void)
{
/* Should be called once. */
- assert(!buffer_registry_uid);
+ LTTNG_ASSERT(!buffer_registry_uid);
buffer_registry_uid = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
- assert(buffer_registry_uid);
+ LTTNG_ASSERT(buffer_registry_uid);
buffer_registry_uid->match_fct = ht_match_reg_uid;
buffer_registry_uid->hash_fct = ht_hash_reg_uid;
* Return 0 on success else a negative value and regp is untouched.
*/
int buffer_reg_uid_create(uint64_t session_id, uint32_t bits_per_long, uid_t uid,
- enum lttng_domain_type domain, struct buffer_reg_uid **regp)
+ enum lttng_domain_type domain, struct buffer_reg_uid **regp,
+ const char *root_shm_path, const char *shm_path)
{
int ret = 0;
struct buffer_reg_uid *reg = NULL;
- assert(regp);
+ LTTNG_ASSERT(regp);
reg = zmalloc(sizeof(*reg));
if (!reg) {
reg->bits_per_long = bits_per_long;
reg->uid = uid;
reg->domain = domain;
-
+ if (shm_path[0]) {
+ strncpy(reg->root_shm_path, root_shm_path, sizeof(reg->root_shm_path));
+ reg->root_shm_path[sizeof(reg->root_shm_path) - 1] = '\0';
+ strncpy(reg->shm_path, shm_path, sizeof(reg->shm_path));
+ reg->shm_path[sizeof(reg->shm_path) - 1] = '\0';
+ DBG3("shm path '%s' is assigned to uid buffer registry for session id %" PRIu64,
+ reg->shm_path, session_id);
+ }
reg->registry->channels = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
if (!reg->registry->channels) {
ret = -ENOMEM;
struct cds_lfht_node *nodep;
struct lttng_ht *ht = buffer_registry_uid;
- assert(reg);
+ LTTNG_ASSERT(reg);
DBG3("Buffer registry per UID adding to global registry with id: %" PRIu64 ,
reg->session_id);
rcu_read_lock();
nodep = cds_lfht_add_unique(ht->ht, ht->hash_fct(reg, lttng_ht_seed),
ht->match_fct, reg, ®->node.node);
- assert(nodep == ®->node.node);
+ LTTNG_ASSERT(nodep == ®->node.node);
rcu_read_unlock();
}
void buffer_reg_init_pid_registry(void)
{
/* Should be called once. */
- assert(!buffer_registry_pid);
+ LTTNG_ASSERT(!buffer_registry_pid);
buffer_registry_pid = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
- assert(buffer_registry_pid);
+ LTTNG_ASSERT(buffer_registry_pid);
DBG3("Global buffer per PID registry initialized");
}
*
* Return 0 on success else a negative value and regp is untouched.
*/
-int buffer_reg_pid_create(uint64_t session_id, struct buffer_reg_pid **regp)
+int buffer_reg_pid_create(uint64_t session_id, struct buffer_reg_pid **regp,
+ const char *root_shm_path, const char *shm_path)
{
int ret = 0;
struct buffer_reg_pid *reg = NULL;
- assert(regp);
+ LTTNG_ASSERT(regp);
reg = zmalloc(sizeof(*reg));
if (!reg) {
/* A cast is done here so we can use the session ID as a u64 ht node. */
reg->session_id = session_id;
-
+ if (shm_path[0]) {
+ strncpy(reg->root_shm_path, root_shm_path, sizeof(reg->root_shm_path));
+ reg->root_shm_path[sizeof(reg->root_shm_path) - 1] = '\0';
+ strncpy(reg->shm_path, shm_path, sizeof(reg->shm_path));
+ reg->shm_path[sizeof(reg->shm_path) - 1] = '\0';
+ DBG3("shm path '%s' is assigned to pid buffer registry for session id %" PRIu64,
+ reg->shm_path, session_id);
+ }
reg->registry->channels = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
if (!reg->registry->channels) {
ret = -ENOMEM;
*/
void buffer_reg_pid_add(struct buffer_reg_pid *reg)
{
- assert(reg);
+ LTTNG_ASSERT(reg);
DBG3("Buffer registry per PID adding to global registry with id: %" PRIu64,
reg->session_id);
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.
{
struct buffer_reg_channel *reg;
- assert(regp);
+ LTTNG_ASSERT(regp);
DBG3("Buffer registry channel create with key: %" PRIu64, key);
{
struct buffer_reg_stream *reg;
- assert(regp);
+ LTTNG_ASSERT(regp);
DBG3("Buffer registry creating stream");
void buffer_reg_stream_add(struct buffer_reg_stream *stream,
struct buffer_reg_channel *channel)
{
- assert(stream);
- assert(channel);
+ LTTNG_ASSERT(stream);
+ LTTNG_ASSERT(channel);
pthread_mutex_lock(&channel->stream_list_lock);
cds_list_add_tail(&stream->lnode, &channel->streams);
void buffer_reg_channel_add(struct buffer_reg_session *session,
struct buffer_reg_channel *channel)
{
- assert(session);
- assert(channel);
+ LTTNG_ASSERT(session);
+ LTTNG_ASSERT(channel);
rcu_read_lock();
lttng_ht_add_unique_u64(session->channels, &channel->node);
struct buffer_reg_channel *chan = NULL;
struct lttng_ht *ht;
- assert(reg);
+ LTTNG_ASSERT(reg);
switch (reg->domain) {
case LTTNG_DOMAIN_UST:
ht = reg->registry->channels;
break;
default:
- assert(0);
+ abort();
goto end;
}
{
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);
break;
}
default:
- assert(0);
+ abort();
}
free(regp);
int ret;
struct lttng_ht_iter iter;
- assert(session);
- assert(regp);
+ LTTNG_ASSERT(session);
+ LTTNG_ASSERT(regp);
iter.iter.node = ®p->node.node;
ret = lttng_ht_del(session->channels, &iter);
- assert(!ret);
+ LTTNG_ASSERT(!ret);
}
/*
}
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);
break;
}
default:
- assert(0);
+ abort();
}
free(regp);
cds_lfht_for_each_entry(regp->channels->ht, &iter.iter, reg_chan,
node.node) {
ret = lttng_ht_del(regp->channels, &iter);
- assert(!ret);
+ LTTNG_ASSERT(!ret);
buffer_reg_channel_destroy(reg_chan, domain);
}
rcu_read_unlock();
free(regp->reg.ust);
break;
default:
- assert(0);
+ abort();
}
free(regp);
int ret;
struct lttng_ht_iter iter;
- assert(regp);
+ LTTNG_ASSERT(regp);
rcu_read_lock();
iter.iter.node = ®p->node.node;
ret = lttng_ht_del(buffer_registry_uid, &iter);
- assert(!ret);
+ LTTNG_ASSERT(!ret);
rcu_read_unlock();
}
}
break;
default:
- assert(0);
+ abort();
rcu_read_unlock();
return;
}
int ret;
struct lttng_ht_iter iter;
- assert(regp);
+ LTTNG_ASSERT(regp);
iter.iter.node = ®p->node.node;
ret = lttng_ht_del(buffer_registry_pid, &iter);
- assert(!ret);
+ LTTNG_ASSERT(!ret);
}
/*
projects / lttng-tools.git / blobdiff