+++ /dev/null
-/*
- * Copyright (C) 2018 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
- * Copyright (C) 2020 Jérémie Galarneau <jeremie.galarneau@efficios.com>
- *
- * SPDX-License-Identifier: GPL-2.0-only
- *
- */
-
-#include "lttng/tracker.h"
-#include "common/dynamic-array.h"
-#include "common/macros.h"
-#define _LGPL_SOURCE
-#include <grp.h>
-#include <pwd.h>
-#include <sys/types.h>
-#include <unistd.h>
-#include <urcu.h>
-#include <urcu/list.h>
-#include <urcu/rculfhash.h>
-
-#include "tracker.h"
-#include <common/defaults.h>
-#include <common/error.h>
-#include <common/hashtable/hashtable.h>
-#include <common/hashtable/utils.h>
-#include <common/tracker.h>
-#include <lttng/lttng-error.h>
-
-struct process_attr_tracker_value_node {
- struct process_attr_value *value;
- struct cds_lfht_node inclusion_set_ht_node;
- struct rcu_head rcu_head;
-};
-
-struct process_attr_tracker {
- enum lttng_tracking_policy policy;
- struct cds_lfht *inclusion_set_ht;
-};
-
-static void process_attr_tracker_value_node_rcu_free(struct rcu_head *rcu_head)
-{
- struct process_attr_tracker_value_node *node =
- container_of(rcu_head, typeof(*node), rcu_head);
-
- free(node);
-}
-
-struct process_attr_tracker *process_attr_tracker_create(void)
-{
- struct process_attr_tracker *tracker;
-
- tracker = zmalloc(sizeof(*tracker));
- if (!tracker) {
- return NULL;
- }
-
- (void) process_attr_tracker_set_tracking_policy(
- tracker, LTTNG_TRACKING_POLICY_INCLUDE_ALL);
-
- tracker->inclusion_set_ht = cds_lfht_new(DEFAULT_HT_SIZE, 1, 0,
- CDS_LFHT_AUTO_RESIZE | CDS_LFHT_ACCOUNTING, NULL);
- if (!tracker->inclusion_set_ht) {
- goto error;
- }
-
- return tracker;
-error:
- process_attr_tracker_destroy(tracker);
- return NULL;
-}
-
-static void process_attr_tracker_remove_value_node(
- struct process_attr_tracker *tracker,
- struct process_attr_tracker_value_node *value_node)
-{
- cds_lfht_del(tracker->inclusion_set_ht,
- &value_node->inclusion_set_ht_node);
- process_attr_value_destroy(value_node->value);
- call_rcu(&value_node->rcu_head,
- process_attr_tracker_value_node_rcu_free);
-}
-
-static void process_attr_tracker_clear_inclusion_set(
- struct process_attr_tracker *tracker)
-{
- int ret;
- struct lttng_ht_iter iter;
- struct process_attr_tracker_value_node *value_node;
-
- if (!tracker->inclusion_set_ht) {
- return;
- }
-
- rcu_read_lock();
- cds_lfht_for_each_entry (tracker->inclusion_set_ht, &iter.iter,
- value_node, inclusion_set_ht_node) {
- process_attr_tracker_remove_value_node(tracker, value_node);
- }
- rcu_read_unlock();
- ret = cds_lfht_destroy(tracker->inclusion_set_ht, NULL);
- LTTNG_ASSERT(ret == 0);
- tracker->inclusion_set_ht = NULL;
-}
-
-static int process_attr_tracker_create_inclusion_set(
- struct process_attr_tracker *tracker)
-{
- LTTNG_ASSERT(!tracker->inclusion_set_ht);
- tracker->inclusion_set_ht = cds_lfht_new(DEFAULT_HT_SIZE, 1, 0,
- CDS_LFHT_AUTO_RESIZE | CDS_LFHT_ACCOUNTING, NULL);
- return tracker->inclusion_set_ht ? 0 : -1;
-}
-
-void process_attr_tracker_destroy(struct process_attr_tracker *tracker)
-{
- if (!tracker) {
- return;
- }
-
- process_attr_tracker_clear_inclusion_set(tracker);
- free(tracker);
-}
-
-enum lttng_tracking_policy process_attr_tracker_get_tracking_policy(
- const struct process_attr_tracker *tracker)
-{
- return tracker->policy;
-}
-
-int process_attr_tracker_set_tracking_policy(
- struct process_attr_tracker *tracker,
- enum lttng_tracking_policy tracking_policy)
-{
- int ret = 0;
-
- if (tracker->policy == tracking_policy) {
- goto end;
- }
-
- process_attr_tracker_clear_inclusion_set(tracker);
- ret = process_attr_tracker_create_inclusion_set(tracker);
- if (ret) {
- goto end;
- }
- tracker->policy = tracking_policy;
-end:
- return ret;
-}
-
-static int match_inclusion_set_value(
- struct cds_lfht_node *node, const void *key)
-{
- const struct process_attr_value *value_key = key;
- const struct process_attr_tracker_value_node *value_node =
- caa_container_of(node,
- struct process_attr_tracker_value_node,
- inclusion_set_ht_node);
-
- return process_attr_tracker_value_equal(value_node->value, value_key);
-}
-
-static struct process_attr_tracker_value_node *process_attr_tracker_lookup(
- const struct process_attr_tracker *tracker,
- const struct process_attr_value *value)
-{
- struct cds_lfht_iter iter;
- struct cds_lfht_node *node;
-
- LTTNG_ASSERT(tracker->policy == LTTNG_TRACKING_POLICY_INCLUDE_SET);
-
- rcu_read_lock();
- cds_lfht_lookup(tracker->inclusion_set_ht,
- process_attr_value_hash(value),
- match_inclusion_set_value, value, &iter);
- node = cds_lfht_iter_get_node(&iter);
- rcu_read_unlock();
-
- return node ? container_of(node, struct process_attr_tracker_value_node,
- inclusion_set_ht_node) :
- NULL;
-}
-
-/* Protected by session mutex held by caller. */
-enum process_attr_tracker_status process_attr_tracker_inclusion_set_add_value(
- struct process_attr_tracker *tracker,
- const struct process_attr_value *value)
-{
- enum process_attr_tracker_status status =
- PROCESS_ATTR_TRACKER_STATUS_OK;
- struct process_attr_value *value_copy = NULL;
- struct process_attr_tracker_value_node *value_node = NULL;
-
- rcu_read_lock();
- if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) {
- status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY;
- goto end;
- }
-
- if (process_attr_tracker_lookup(tracker, value)) {
- status = PROCESS_ATTR_TRACKER_STATUS_EXISTS;
- goto end;
- }
-
- value_node = zmalloc(sizeof(*value_node));
- if (!value_node) {
- status = PROCESS_ATTR_TRACKER_STATUS_ERROR;
- goto end;
- }
-
- value_copy = process_attr_value_copy(value);
- if (!value_copy) {
- status = PROCESS_ATTR_TRACKER_STATUS_ERROR;
- goto end;
- }
-
- value_node->value = value_copy;
- cds_lfht_add(tracker->inclusion_set_ht,
- process_attr_value_hash(value_copy),
- &value_node->inclusion_set_ht_node);
- value_copy = NULL;
- value_node = NULL;
-end:
- if (value_copy) {
- process_attr_value_destroy(value_copy);
- }
- if (value_node) {
- free(value_node);
- }
- rcu_read_unlock();
- return status;
-}
-
-/* Protected by session mutex held by caller. */
-enum process_attr_tracker_status
-process_attr_tracker_inclusion_set_remove_value(
- struct process_attr_tracker *tracker,
- const struct process_attr_value *value)
-{
- struct process_attr_tracker_value_node *value_node;
- enum process_attr_tracker_status status =
- PROCESS_ATTR_TRACKER_STATUS_OK;
-
- rcu_read_lock();
- if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) {
- status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY;
- goto end;
- }
-
- value_node = process_attr_tracker_lookup(tracker, value);
- if (!value_node) {
- status = PROCESS_ATTR_TRACKER_STATUS_MISSING;
- goto end;
- }
-
- process_attr_tracker_remove_value_node(tracker, value_node);
-end:
- rcu_read_unlock();
- return status;
-}
-
-enum process_attr_tracker_status process_attr_tracker_get_inclusion_set(
- const struct process_attr_tracker *tracker,
- struct lttng_process_attr_values **_values)
-{
- struct lttng_ht_iter iter;
- struct process_attr_tracker_value_node *value_node;
- enum process_attr_tracker_status status =
- PROCESS_ATTR_TRACKER_STATUS_OK;
- struct lttng_process_attr_values *values;
- struct process_attr_value *new_value = NULL;
-
- values = lttng_process_attr_values_create();
- if (!values) {
- status = PROCESS_ATTR_TRACKER_STATUS_ERROR;
- goto error;
- }
-
- if (tracker->policy != LTTNG_TRACKING_POLICY_INCLUDE_SET) {
- status = PROCESS_ATTR_TRACKER_STATUS_INVALID_TRACKING_POLICY;
- goto error;
- }
-
- rcu_read_lock();
- cds_lfht_for_each_entry (tracker->inclusion_set_ht, &iter.iter,
- value_node, inclusion_set_ht_node) {
- int ret;
-
- new_value = process_attr_value_copy(value_node->value);
- if (!new_value) {
- status = PROCESS_ATTR_TRACKER_STATUS_ERROR;
- goto error_unlock;
- }
-
- ret = lttng_dynamic_pointer_array_add_pointer(
- &values->array, new_value);
- if (ret) {
- status = PROCESS_ATTR_TRACKER_STATUS_ERROR;
- goto error_unlock;
- }
-
- new_value = NULL;
- }
- rcu_read_unlock();
- *_values = values;
- return status;
-error_unlock:
- rcu_read_unlock();
-error:
- lttng_process_attr_values_destroy(values);
- process_attr_value_destroy(new_value);
- return status;
-}