X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=src%2Fcommon%2Flocation.c;h=97fed09b1ff380e87712c2565e96344894a8a6aa;hb=d37406195ec30adfb1bd903e3ee8dfdda79700ef;hp=d1fd538f2933994e0ba91be7e6ecfa28d8bd4687;hpb=434131e45c3b55d74c3feec8b21b0979eecc8671;p=lttng-tools.git diff --git a/src/common/location.c b/src/common/location.c index d1fd538f2..97fed09b1 100644 --- a/src/common/location.c +++ b/src/common/location.c @@ -1,23 +1,14 @@ /* - * Copyright (C) 2018 - Jérémie Galarneau + * Copyright (C) 2018 Jérémie Galarneau * - * This library is free software; you can redistribute it and/or modify it - * under the terms of the GNU Lesser General Public License, version 2.1 only, - * as published by the Free Software Foundation. + * SPDX-License-Identifier: LGPL-2.1-only * - * This library 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 Lesser General Public License - * for more details. - * - * You should have received a copy of the GNU Lesser General Public License - * along with this library; if not, write to the Free Software Foundation, - * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include #include #include +#include static struct lttng_trace_archive_location *lttng_trace_archive_location_create( @@ -30,18 +21,17 @@ struct lttng_trace_archive_location *lttng_trace_archive_location_create( goto end; } + urcu_ref_init(&location->ref); location->type = type; end: return location; } -LTTNG_HIDDEN -void lttng_trace_archive_location_destroy( - struct lttng_trace_archive_location *location) +static +void trace_archive_location_destroy_ref(struct urcu_ref *ref) { - if (!location) { - return; - } + struct lttng_trace_archive_location *location = + container_of(ref, struct lttng_trace_archive_location, ref); switch (location->type) { case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL: @@ -58,6 +48,22 @@ void lttng_trace_archive_location_destroy( free(location); } +LTTNG_HIDDEN +void lttng_trace_archive_location_get(struct lttng_trace_archive_location *location) +{ + urcu_ref_get(&location->ref); +} + +LTTNG_HIDDEN +void lttng_trace_archive_location_put(struct lttng_trace_archive_location *location) +{ + if (!location) { + return; + } + + urcu_ref_put(&location->ref, trace_archive_location_destroy_ref); +} + LTTNG_HIDDEN struct lttng_trace_archive_location *lttng_trace_archive_location_local_create( const char *absolute_path) @@ -82,7 +88,7 @@ struct lttng_trace_archive_location *lttng_trace_archive_location_local_create( end: return location; error: - lttng_trace_archive_location_destroy(location); + lttng_trace_archive_location_put(location); return NULL; } @@ -120,10 +126,164 @@ struct lttng_trace_archive_location *lttng_trace_archive_location_relay_create( end: return location; error: - lttng_trace_archive_location_destroy(location); + lttng_trace_archive_location_put(location); return NULL; } +LTTNG_HIDDEN +ssize_t lttng_trace_archive_location_create_from_buffer( + const struct lttng_buffer_view *view, + struct lttng_trace_archive_location **location) +{ + size_t offset = 0; + const struct lttng_trace_archive_location_comm *location_comm; + struct lttng_buffer_view location_comm_view; + + location_comm_view = lttng_buffer_view_from_view(view, 0, + sizeof(*location_comm)); + if (!lttng_buffer_view_is_valid(&location_comm_view)) { + goto error; + } + + offset += location_comm_view.size; + location_comm = (const struct lttng_trace_archive_location_comm *) location_comm_view.data; + + switch ((enum lttng_trace_archive_location_type) location_comm->type) { + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL: + { + const struct lttng_buffer_view absolute_path_view = + lttng_buffer_view_from_view(view, offset, + location_comm->types.local.absolute_path_len); + + if (!lttng_buffer_view_is_valid(&absolute_path_view)) { + goto error; + } + + if (absolute_path_view.data[absolute_path_view.size - 1] != '\0') { + goto error; + } + offset += absolute_path_view.size; + + *location = lttng_trace_archive_location_local_create( + absolute_path_view.data); + if (!*location) { + goto error; + } + break; + } + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY: + { + const struct lttng_buffer_view hostname_view = + lttng_buffer_view_from_view(view, offset, + location_comm->types.relay.hostname_len); + const struct lttng_buffer_view relative_path_view = + lttng_buffer_view_from_view(view, + offset + hostname_view.size, + location_comm->types.relay.relative_path_len); + + if (!lttng_buffer_view_is_valid(&hostname_view) || + !lttng_buffer_view_is_valid( + &relative_path_view)) { + goto error; + } + + if (hostname_view.data[hostname_view.size - 1] != '\0') { + goto error; + } + if (relative_path_view.data[relative_path_view.size - 1] != '\0') { + goto error; + } + offset += hostname_view.size + relative_path_view.size; + + *location = lttng_trace_archive_location_relay_create( + hostname_view.data, + (enum lttng_trace_archive_location_relay_protocol_type) location_comm->types.relay.protocol, + location_comm->types.relay.ports.control, + location_comm->types.relay.ports.data, + relative_path_view.data); + if (!*location) { + goto error; + } + break; + } + default: + goto error; + } + + return offset; +error: + return -1; +} + +LTTNG_HIDDEN +ssize_t lttng_trace_archive_location_serialize( + const struct lttng_trace_archive_location *location, + struct lttng_dynamic_buffer *buffer) +{ + int ret; + struct lttng_trace_archive_location_comm location_comm; + + location_comm.type = (int8_t) location->type; + + switch (location->type) { + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL: + location_comm.types.local.absolute_path_len = + strlen(location->types.local.absolute_path) + 1; + break; + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY: + location_comm.types.relay.hostname_len = + strlen(location->types.relay.host) + 1; + location_comm.types.relay.protocol = + (int8_t) location->types.relay.protocol; + location_comm.types.relay.ports.control = + location->types.relay.ports.control; + location_comm.types.relay.ports.data = + location->types.relay.ports.data; + location_comm.types.relay.relative_path_len = + strlen(location->types.relay.relative_path) + 1; + break; + default: + abort(); + } + + ret = lttng_dynamic_buffer_append(buffer, &location_comm, + sizeof(location_comm)); + if (ret) { + goto error; + } + + switch (location->type) { + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL: + ret = lttng_dynamic_buffer_append(buffer, + location->types.local.absolute_path, + location_comm.types.local.absolute_path_len); + if (ret) { + goto error; + } + break; + case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY: + ret = lttng_dynamic_buffer_append(buffer, + location->types.relay.host, + location_comm.types.relay.hostname_len); + if (ret) { + goto error; + } + ret = lttng_dynamic_buffer_append(buffer, + location->types.relay.relative_path, + location_comm.types.relay.relative_path_len); + if (ret) { + goto error; + } + break; + default: + abort(); + } + + return 0; +error: + return -1; +} + enum lttng_trace_archive_location_type lttng_trace_archive_location_get_type( const struct lttng_trace_archive_location *location) {