[lttng-tools.git] / src / common / location.c

/*
 * Copyright (C) 2018 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * 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.
 *
 * 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 <lttng/location-internal.h>
#include <common/macros.h>
#include <stdlib.h>

static
struct lttng_trace_archive_location *lttng_trace_archive_location_create(
		enum lttng_trace_archive_location_type type)
{
	struct lttng_trace_archive_location *location;

	location = zmalloc(sizeof(*location));
	if (!location) {
		goto end;
	}

	location->type = type;
end:
	return location;
}

LTTNG_HIDDEN
void lttng_trace_archive_location_destroy(
		struct lttng_trace_archive_location *location)
{
	if (!location) {
		return;
	}

	switch (location->type) {
	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL:
		free(location->types.local.absolute_path);
		break;
	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY:
		free(location->types.relay.host);
		free(location->types.relay.relative_path);
		break;
	default:
		abort();
	}

	free(location);
}

LTTNG_HIDDEN
struct lttng_trace_archive_location *lttng_trace_archive_location_local_create(
		const char *absolute_path)
{
	struct lttng_trace_archive_location *location = NULL;

	if (!absolute_path) {
		goto end;
	}

	location = lttng_trace_archive_location_create(
			LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL);
	if (!location) {
		goto end;
	}

	location->types.local.absolute_path = strdup(absolute_path);
	if (!location->types.local.absolute_path) {
		goto error;
	}

end:
	return location;
error:
	lttng_trace_archive_location_destroy(location);
	return NULL;
}

LTTNG_HIDDEN
struct lttng_trace_archive_location *lttng_trace_archive_location_relay_create(
		const char *host,
		enum lttng_trace_archive_location_relay_protocol_type protocol,
		uint16_t control_port, uint16_t data_port,
		const char *relative_path)
{
	struct lttng_trace_archive_location *location = NULL;

	if (!host || !relative_path) {
		goto end;
	}

	location = lttng_trace_archive_location_create(
			LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY);
	if (!location) {
		goto end;
	}

	location->types.relay.host = strdup(host);
	if (!location->types.relay.host) {
		goto error;
	}
	location->types.relay.relative_path = strdup(relative_path);
	if (!location->types.relay.relative_path) {
		goto error;
	}

	location->types.relay.protocol = protocol;
	location->types.relay.ports.control = control_port;
	location->types.relay.ports.data = data_port;
end:
	return location;
error:
	lttng_trace_archive_location_destroy(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 (!location_comm_view.data) {
		goto error;
	}
	offset += location_comm_view.size;
	location_comm = (const struct lttng_trace_archive_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 (!absolute_path_view.data) {
			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 (!hostname_view.data || !relative_path_view.data) {
			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)
{
	enum lttng_trace_archive_location_type type;

	if (!location) {
		type = LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_UNKNOWN;
		goto end;
	}

	type = location->type;
end:
	return type;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_local_get_absolute_path(
		const struct lttng_trace_archive_location *location,
		const char **absolute_path)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !absolute_path ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*absolute_path = location->types.local.absolute_path;
end:
	return status;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_relay_get_host(
		const struct lttng_trace_archive_location *location,
		const char **relay_host)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !relay_host ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*relay_host = location->types.relay.host;
end:
	return status;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_relay_get_relative_path(
		const struct lttng_trace_archive_location *location,
		const char **relative_path)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !relative_path ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*relative_path = location->types.relay.relative_path;
end:
	return status;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_relay_get_control_port(
		const struct lttng_trace_archive_location *location,
		uint16_t *control_port)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !control_port ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*control_port = location->types.relay.ports.control;
end:
	return status;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_relay_get_data_port(
		const struct lttng_trace_archive_location *location,
		uint16_t *data_port)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !data_port ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*data_port = location->types.relay.ports.data;
end:
	return status;
}

enum lttng_trace_archive_location_status
lttng_trace_archive_location_relay_get_protocol_type(
		const struct lttng_trace_archive_location *location,
		enum lttng_trace_archive_location_relay_protocol_type *protocol)
{
	enum lttng_trace_archive_location_status status =
			LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;

	if (!location || !protocol ||
			location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
		status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
		goto end;
	}

	*protocol = location->types.relay.protocol;
end:
	return status;
}
Commit	Line	Data
434131e4 JG	1	/*
	2	* Copyright (C) 2018 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
	3	*
	4	* This library is free software; you can redistribute it and/or modify it
	5	* under the terms of the GNU Lesser General Public License, version 2.1 only,
	6	* as published by the Free Software Foundation.
	7	*
	8	* This library is distributed in the hope that it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
	11	* for more details.
	12	*
	13	* You should have received a copy of the GNU Lesser General Public License
	14	* along with this library; if not, write to the Free Software Foundation,
	15	* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	16	*/
	17
	18	#include <lttng/location-internal.h>
	19	#include <common/macros.h>
	20	#include <stdlib.h>
	21
	22	static
	23	struct lttng_trace_archive_location *lttng_trace_archive_location_create(
	24	enum lttng_trace_archive_location_type type)
	25	{
	26	struct lttng_trace_archive_location *location;
	27
	28	location = zmalloc(sizeof(*location));
	29	if (!location) {
	30	goto end;
	31	}
	32
	33	location->type = type;
	34	end:
	35	return location;
	36	}
	37
	38	LTTNG_HIDDEN
	39	void lttng_trace_archive_location_destroy(
	40	struct lttng_trace_archive_location *location)
	41	{
	42	if (!location) {
	43	return;
	44	}
	45
	46	switch (location->type) {
	47	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL:
	48	free(location->types.local.absolute_path);
	49	break;
	50	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY:
	51	free(location->types.relay.host);
	52	free(location->types.relay.relative_path);
	53	break;
	54	default:
	55	abort();
	56	}
	57
	58	free(location);
	59	}
	60
	61	LTTNG_HIDDEN
	62	struct lttng_trace_archive_location *lttng_trace_archive_location_local_create(
	63	const char *absolute_path)
	64	{
65	struct lttng_trace_archive_location *location = NULL;
66
67	if (!absolute_path) {
68	goto end;
69	}
70
71	location = lttng_trace_archive_location_create(
72	LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL);
73	if (!location) {
74	goto end;
75	}
76
77	location->types.local.absolute_path = strdup(absolute_path);
78	if (!location->types.local.absolute_path) {
79	goto error;
80	}
81
82	end:
83	return location;
84	error:
85	lttng_trace_archive_location_destroy(location);
86	return NULL;
87	}
88
89	LTTNG_HIDDEN
90	struct lttng_trace_archive_location *lttng_trace_archive_location_relay_create(
91	const char *host,
92	enum lttng_trace_archive_location_relay_protocol_type protocol,
93	uint16_t control_port, uint16_t data_port,
94	const char *relative_path)
95	{
96	struct lttng_trace_archive_location *location = NULL;
97
98	if (!host \|\| !relative_path) {
99	goto end;
100	}
101
102	location = lttng_trace_archive_location_create(
103	LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY);
104	if (!location) {
105	goto end;
106	}
107
108	location->types.relay.host = strdup(host);
109	if (!location->types.relay.host) {
110	goto error;
111	}
112	location->types.relay.relative_path = strdup(relative_path);
113	if (!location->types.relay.relative_path) {
114	goto error;
115	}
116
117	location->types.relay.protocol = protocol;
118	location->types.relay.ports.control = control_port;
119	location->types.relay.ports.data = data_port;
120	end:
121	return location;
122	error:
123	lttng_trace_archive_location_destroy(location);
124	return NULL;
125	}
126
1e9f1cf8 JG	127	LTTNG_HIDDEN
	128	ssize_t lttng_trace_archive_location_create_from_buffer(
	129	const struct lttng_buffer_view *view,
	130	struct lttng_trace_archive_location **location)
	131	{
	132	size_t offset = 0;
	133	const struct lttng_trace_archive_location_comm *location_comm;
	134	struct lttng_buffer_view location_comm_view;
	135
	136	location_comm_view = lttng_buffer_view_from_view(view, 0,
	137	sizeof(*location_comm));
	138	if (!location_comm_view.data) {
	139	goto error;
	140	}
	141	offset += location_comm_view.size;
	142	location_comm = (const struct lttng_trace_archive_location_comm *) view->data;
	143
	144	switch ((enum lttng_trace_archive_location_type) location_comm->type) {
	145	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL:
	146	{
	147	const struct lttng_buffer_view absolute_path_view =
	148	lttng_buffer_view_from_view(view, offset,
	149	location_comm->types.local.absolute_path_len);
	150
	151	if (!absolute_path_view.data) {
	152	goto error;
	153	}
	154	if (absolute_path_view.data[absolute_path_view.size - 1] != '\0') {
	155	goto error;
	156	}
	157	offset += absolute_path_view.size;
	158
	159	*location = lttng_trace_archive_location_local_create(
	160	absolute_path_view.data);
	161	if (!*location) {
	162	goto error;
	163	}
	164	break;
	165	}
	166	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY:
	167	{
	168	const struct lttng_buffer_view hostname_view =
	169	lttng_buffer_view_from_view(view, offset,
	170	location_comm->types.relay.hostname_len);
	171	const struct lttng_buffer_view relative_path_view =
	172	lttng_buffer_view_from_view(view,
	173	offset + hostname_view.size,
	174	location_comm->types.relay.relative_path_len);
	175
	176	if (!hostname_view.data \|\| !relative_path_view.data) {
	177	goto error;
	178	}
	179	if (hostname_view.data[hostname_view.size - 1] != '\0') {
	180	goto error;
	181	}
	182	if (relative_path_view.data[relative_path_view.size - 1] != '\0') {
	183	goto error;
	184	}
	185	offset += hostname_view.size + relative_path_view.size;
	186
	187	*location = lttng_trace_archive_location_relay_create(
	188	hostname_view.data,
	189	(enum lttng_trace_archive_location_relay_protocol_type) location_comm->types.relay.protocol,
	190	location_comm->types.relay.ports.control,
191	location_comm->types.relay.ports.data,
192	relative_path_view.data);
193	if (!*location) {
194	goto error;
195	}
196	break;
197	}
198	default:
199	goto error;
200	}
201
e4b6e50f	202	return offset;
1e9f1cf8 JG	203	error:
	204	return -1;
	205	}
	206
	207	LTTNG_HIDDEN
	208	ssize_t lttng_trace_archive_location_serialize(
	209	const struct lttng_trace_archive_location *location,
	210	struct lttng_dynamic_buffer *buffer)
	211	{
	212	int ret;
	213	struct lttng_trace_archive_location_comm location_comm;
1e9f1cf8 JG	214
	215	location_comm.type = (int8_t) location->type;
	216
	217	switch (location->type) {
	218	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL:
	219	location_comm.types.local.absolute_path_len =
	220	strlen(location->types.local.absolute_path) + 1;
	221	break;
	222	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY:
	223	location_comm.types.relay.hostname_len =
	224	strlen(location->types.relay.host) + 1;
	225	location_comm.types.relay.protocol =
	226	(int8_t) location->types.relay.protocol;
	227	location_comm.types.relay.ports.control =
	228	location->types.relay.ports.control;
	229	location_comm.types.relay.ports.data =
	230	location->types.relay.ports.data;
	231	location_comm.types.relay.relative_path_len =
	232	strlen(location->types.relay.relative_path) + 1;
	233	break;
	234	default:
	235	abort();
	236	}
	237
	238	ret = lttng_dynamic_buffer_append(buffer, &location_comm,
	239	sizeof(location_comm));
	240	if (ret) {
	241	goto error;
	242	}
	243
	244	switch (location->type) {
	245	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL:
	246	ret = lttng_dynamic_buffer_append(buffer,
	247	location->types.local.absolute_path,
	248	location_comm.types.local.absolute_path_len);
	249	if (ret) {
	250	goto error;
	251	}
	252	break;
	253	case LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY:
	254	ret = lttng_dynamic_buffer_append(buffer,
	255	location->types.relay.host,
	256	location_comm.types.relay.hostname_len);
	257	if (ret) {
	258	goto error;
	259	}
	260	ret = lttng_dynamic_buffer_append(buffer,
	261	location->types.relay.relative_path,
	262	location_comm.types.relay.relative_path_len);
	263	if (ret) {
	264	goto error;
	265	}
	266	break;
	267	default:
	268	abort();
	269	}
	270
f7daf93a	271	return 0;
1e9f1cf8 JG	272	error:
	273	return -1;
	274	}
	275
434131e4 JG	276	enum lttng_trace_archive_location_type lttng_trace_archive_location_get_type(
	277	const struct lttng_trace_archive_location *location)
	278	{
	279	enum lttng_trace_archive_location_type type;
	280
	281	if (!location) {
	282	type = LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_UNKNOWN;
	283	goto end;
	284	}
	285
	286	type = location->type;
	287	end:
	288	return type;
	289	}
	290
	291	enum lttng_trace_archive_location_status
	292	lttng_trace_archive_location_local_get_absolute_path(
	293	const struct lttng_trace_archive_location *location,
	294	const char **absolute_path)
	295	{
	296	enum lttng_trace_archive_location_status status =
	297	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
	298
	299	if (!location \|\| !absolute_path \|\|
	300	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_LOCAL) {
	301	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
	302	goto end;
	303	}
	304
	305	*absolute_path = location->types.local.absolute_path;
	306	end:
	307	return status;
	308	}
	309
	310	enum lttng_trace_archive_location_status
	311	lttng_trace_archive_location_relay_get_host(
	312	const struct lttng_trace_archive_location *location,
	313	const char **relay_host)
	314	{
	315	enum lttng_trace_archive_location_status status =
	316	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
	317
	318	if (!location \|\| !relay_host \|\|
	319	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
	320	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
	321	goto end;
	322	}
	323
	324	*relay_host = location->types.relay.host;
	325	end:
	326	return status;
	327	}
	328
	329	enum lttng_trace_archive_location_status
	330	lttng_trace_archive_location_relay_get_relative_path(
	331	const struct lttng_trace_archive_location *location,
	332	const char **relative_path)
	333	{
	334	enum lttng_trace_archive_location_status status =
	335	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
	336
	337	if (!location \|\| !relative_path \|\|
	338	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
	339	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
340	goto end;
341	}
342
343	*relative_path = location->types.relay.relative_path;
344	end:
345	return status;
346	}
347
348	enum lttng_trace_archive_location_status
349	lttng_trace_archive_location_relay_get_control_port(
350	const struct lttng_trace_archive_location *location,
351	uint16_t *control_port)
352	{
353	enum lttng_trace_archive_location_status status =
354	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
355
356	if (!location \|\| !control_port \|\|
357	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
358	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
359	goto end;
360	}
361
362	*control_port = location->types.relay.ports.control;
363	end:
364	return status;
365	}
366
367	enum lttng_trace_archive_location_status
368	lttng_trace_archive_location_relay_get_data_port(
369	const struct lttng_trace_archive_location *location,
370	uint16_t *data_port)
371	{
372	enum lttng_trace_archive_location_status status =
373	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
374
375	if (!location \|\| !data_port \|\|
376	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
377	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
378	goto end;
379	}
380
381	*data_port = location->types.relay.ports.data;
382	end:
383	return status;
384	}
385
386	enum lttng_trace_archive_location_status
387	lttng_trace_archive_location_relay_get_protocol_type(
388	const struct lttng_trace_archive_location *location,
389	enum lttng_trace_archive_location_relay_protocol_type *protocol)
390	{
391	enum lttng_trace_archive_location_status status =
392	LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_OK;
393
394	if (!location \|\| !protocol \|\|
395	location->type != LTTNG_TRACE_ARCHIVE_LOCATION_TYPE_RELAY) {
396	status = LTTNG_TRACE_ARCHIVE_LOCATION_STATUS_INVALID;
397	goto end;
398	}
399
400	*protocol = location->types.relay.protocol;
401	end:
402	return status;
403	}