X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fcmd-2-4.c;h=02edb7713e54245b2c55beb6ec6ffb94ed4499e6;hp=b269bc68a2e84473e9061afcc32697105e41a81e;hb=640b9481e1805d314256361ed49d55dcc35d6172;hpb=246777db6078ff9dd8046d5701fc12cb34547402 diff --git a/src/bin/lttng-relayd/cmd-2-4.c b/src/bin/lttng-relayd/cmd-2-4.c index b269bc68a..02edb7713 100644 --- a/src/bin/lttng-relayd/cmd-2-4.c +++ b/src/bin/lttng-relayd/cmd-2-4.c @@ -1,20 +1,10 @@ /* - * Copyright (C) 2013 - Julien Desfossez - * David Goulet - * 2015 - Mathieu Desnoyers + * Copyright (C) 2013 Julien Desfossez + * Copyright (C) 2013 David Goulet + * Copyright (C) 2015 Mathieu Desnoyers * - * 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 _LGPL_SOURCE @@ -27,10 +17,10 @@ #include #include -#include "cmd-generic.h" +#include "cmd-2-4.h" #include "lttng-relayd.h" -int cmd_create_session_2_4(struct relay_connection *conn, +int cmd_create_session_2_4(const struct lttng_buffer_view *payload, char *session_name, char *hostname, uint32_t *live_timer, bool *snapshot) { @@ -38,20 +28,26 @@ int cmd_create_session_2_4(struct relay_connection *conn, struct lttcomm_relayd_create_session_2_4 session_info; size_t len; - ret = cmd_recv(conn->sock, &session_info, sizeof(session_info)); - if (ret < 0) { - ERR("Unable to recv session info version 2.4"); + if (payload->size < sizeof(session_info)) { + ERR("Unexpected payload size in \"cmd_create_session_2_4\": expected >= %zu bytes, got %zu bytes", + sizeof(session_info), payload->size); + ret = -1; goto error; } + memcpy(&session_info, payload->data, sizeof(session_info)); + len = lttng_strnlen(session_info.session_name, sizeof(session_info.session_name)); /* Ensure that NULL-terminated and fits in local filename length. */ if (len == sizeof(session_info.session_name) || len >= LTTNG_NAME_MAX) { ret = -ENAMETOOLONG; ERR("Session name too long"); goto error; + } else if (len == 0) { + ret = -EINVAL; + ERR("Session name can't be of length 0"); + goto error; } - strncpy(session_name, session_info.session_name, - sizeof(session_info.session_name)); + strncpy(session_name, session_info.session_name, LTTNG_NAME_MAX); len = lttng_strnlen(session_info.hostname, sizeof(session_info.hostname)); if (len == sizeof(session_info.hostname) || len >= LTTNG_HOST_NAME_MAX) { @@ -59,8 +55,8 @@ int cmd_create_session_2_4(struct relay_connection *conn, ERR("Session name too long"); goto error; } - strncpy(hostname, session_info.hostname, - sizeof(session_info.hostname)); + strncpy(hostname, session_info.hostname, LTTNG_HOST_NAME_MAX); + *live_timer = be32toh(session_info.live_timer); *snapshot = be32toh(session_info.snapshot);