numam-dpdk/app/test/test_telemetry_data.c
Ciara Power 89ce9f09e2 test/telemetry: fix socket resource leak
If an error occurred when reading from the socket, the function
returned without closing the socket. This is now fixed to avoid the
resource leak of the sock variable going out of scope.

Coverity issue: 363043
Fixes: bd78cf693e ("test/telemetry: add unit tests for data to JSON")

Signed-off-by: Ciara Power <ciara.power@intel.com>
Acked-by: Kevin Laatz <kevin.laatz@intel.com>
2020-10-19 17:01:58 +02:00

374 lines
10 KiB
C

/* SPDX-License-Identifier: BSD-3-Clause
* Copyright 2020 Intel Corporation
*/
#include <string.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <limits.h>
#include <rte_eal.h>
#include <rte_common.h>
#include <rte_telemetry.h>
#include <rte_string_fns.h>
#include "test.h"
#include "telemetry_data.h"
#define TELEMETRY_VERSION "v2"
#define REQUEST_CMD "/test"
#define BUF_SIZE 1024
#define TEST_OUTPUT(exp) test_output(__func__, exp)
static struct rte_tel_data response_data;
static int sock;
/*
* This function is the callback registered with Telemetry to be used when
* the /test command is requested. This callback returns the global data built
* up by the individual test cases.
*/
static int
test_cb(const char *cmd __rte_unused, const char *params __rte_unused,
struct rte_tel_data *d)
{
*d = response_data;
return 0;
}
/*
* This function is called by each test case function. It communicates with
* the telemetry socket by requesting the /test command, and reading the
* response. The expected response is passed in by the test case function,
* and is compared to the actual response received from Telemetry.
*/
static int
test_output(const char *func_name, const char *expected)
{
int bytes;
char buf[BUF_SIZE * 16];
if (write(sock, REQUEST_CMD, strlen(REQUEST_CMD)) < 0) {
printf("%s: Error with socket write - %s\n", __func__,
strerror(errno));
return -1;
}
bytes = read(sock, buf, sizeof(buf) - 1);
if (bytes < 0) {
printf("%s: Error with socket read - %s\n", __func__,
strerror(errno));
return -1;
}
buf[bytes] = '\0';
printf("%s: buf = '%s', expected = '%s'\n", func_name, buf, expected);
return strncmp(expected, buf, sizeof(buf));
}
static int
test_dict_with_array_int_values(void)
{
int i;
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_INT_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_INT_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
for (i = 0; i < 5; i++) {
rte_tel_data_add_array_int(child_data, i);
rte_tel_data_add_array_int(child_data2, i);
}
rte_tel_data_add_dict_container(&response_data, "dict_0",
child_data, 0);
rte_tel_data_add_dict_container(&response_data, "dict_1",
child_data2, 0);
return TEST_OUTPUT("{\"/test\":{\"dict_0\":[0,1,2,3,4],"
"\"dict_1\":[0,1,2,3,4]}}");
}
static int
test_array_with_array_int_values(void)
{
int i;
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_INT_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_INT_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_CONTAINER);
for (i = 0; i < 5; i++) {
rte_tel_data_add_array_int(child_data, i);
rte_tel_data_add_array_int(child_data2, i);
}
rte_tel_data_add_array_container(&response_data, child_data, 0);
rte_tel_data_add_array_container(&response_data, child_data2, 0);
return TEST_OUTPUT("{\"/test\":[[0,1,2,3,4],[0,1,2,3,4]]}");
}
static int
test_case_array_int(void)
{
int i;
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_INT_VAL);
for (i = 0; i < 5; i++)
rte_tel_data_add_array_int(&response_data, i);
return TEST_OUTPUT("{\"/test\":[0,1,2,3,4]}");
}
static int
test_case_add_dict_int(void)
{
int i = 0;
char name_of_value[8];
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
for (i = 0; i < 5; i++) {
sprintf(name_of_value, "dict_%d", i);
rte_tel_data_add_dict_int(&response_data, name_of_value, i);
}
return TEST_OUTPUT("{\"/test\":{\"dict_0\":0,\"dict_1\":1,\"dict_2\":2,"
"\"dict_3\":3,\"dict_4\":4}}");
}
static int
test_case_array_string(void)
{
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_STRING_VAL);
rte_tel_data_add_array_string(&response_data, "aaaa");
rte_tel_data_add_array_string(&response_data, "bbbb");
rte_tel_data_add_array_string(&response_data, "cccc");
rte_tel_data_add_array_string(&response_data, "dddd");
rte_tel_data_add_array_string(&response_data, "eeee");
return TEST_OUTPUT("{\"/test\":[\"aaaa\",\"bbbb\",\"cccc\",\"dddd\","
"\"eeee\"]}");
}
static int
test_case_add_dict_string(void)
{
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
rte_tel_data_add_dict_string(&response_data, "dict_0", "aaaa");
rte_tel_data_add_dict_string(&response_data, "dict_1", "bbbb");
rte_tel_data_add_dict_string(&response_data, "dict_2", "cccc");
rte_tel_data_add_dict_string(&response_data, "dict_3", "dddd");
return TEST_OUTPUT("{\"/test\":{\"dict_0\":\"aaaa\",\"dict_1\":"
"\"bbbb\",\"dict_2\":\"cccc\",\"dict_3\":\"dddd\"}}");
}
static int
test_dict_with_array_string_values(void)
{
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_STRING_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_STRING_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
rte_tel_data_add_array_string(child_data, "aaaa");
rte_tel_data_add_array_string(child_data2, "bbbb");
rte_tel_data_add_dict_container(&response_data, "dict_0",
child_data, 0);
rte_tel_data_add_dict_container(&response_data, "dict_1",
child_data2, 0);
return TEST_OUTPUT("{\"/test\":{\"dict_0\":[\"aaaa\"],\"dict_1\":"
"[\"bbbb\"]}}");
}
static int
test_array_with_array_string_values(void)
{
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_STRING_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_STRING_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_CONTAINER);
rte_tel_data_add_array_string(child_data, "aaaa");
rte_tel_data_add_array_string(child_data2, "bbbb");
rte_tel_data_add_array_container(&response_data, child_data, 0);
rte_tel_data_add_array_container(&response_data, child_data2, 0);
return TEST_OUTPUT("{\"/test\":[[\"aaaa\"],[\"bbbb\"]]}");
}
static int
test_case_array_u64(void)
{
int i;
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_U64_VAL);
for (i = 0; i < 5; i++)
rte_tel_data_add_array_u64(&response_data, i);
return TEST_OUTPUT("{\"/test\":[0,1,2,3,4]}");
}
static int
test_case_add_dict_u64(void)
{
int i = 0;
char name_of_value[8];
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
for (i = 0; i < 5; i++) {
sprintf(name_of_value, "dict_%d", i);
rte_tel_data_add_dict_u64(&response_data, name_of_value, i);
}
return TEST_OUTPUT("{\"/test\":{\"dict_0\":0,\"dict_1\":1,\"dict_2\":2,"
"\"dict_3\":3,\"dict_4\":4}}");
}
static int
test_dict_with_array_u64_values(void)
{
int i;
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_U64_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_U64_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_dict(&response_data);
for (i = 0; i < 10; i++) {
rte_tel_data_add_array_u64(child_data, i);
rte_tel_data_add_array_u64(child_data2, i);
}
rte_tel_data_add_dict_container(&response_data, "dict_0",
child_data, 0);
rte_tel_data_add_dict_container(&response_data, "dict_1",
child_data2, 0);
return TEST_OUTPUT("{\"/test\":{\"dict_0\":[0,1,2,3,4,5,6,7,8,9],"
"\"dict_1\":[0,1,2,3,4,5,6,7,8,9]}}");
}
static int
test_array_with_array_u64_values(void)
{
int i;
struct rte_tel_data *child_data = rte_tel_data_alloc();
rte_tel_data_start_array(child_data, RTE_TEL_U64_VAL);
struct rte_tel_data *child_data2 = rte_tel_data_alloc();
rte_tel_data_start_array(child_data2, RTE_TEL_U64_VAL);
memset(&response_data, 0, sizeof(response_data));
rte_tel_data_start_array(&response_data, RTE_TEL_CONTAINER);
for (i = 0; i < 5; i++) {
rte_tel_data_add_array_u64(child_data, i);
rte_tel_data_add_array_u64(child_data2, i);
}
rte_tel_data_add_array_container(&response_data, child_data, 0);
rte_tel_data_add_array_container(&response_data, child_data2, 0);
return TEST_OUTPUT("{\"/test\":[[0,1,2,3,4],[0,1,2,3,4]]}");
}
static int
connect_to_socket(void)
{
char buf[BUF_SIZE];
int sock, bytes;
struct sockaddr_un telem_addr;
sock = socket(AF_UNIX, SOCK_SEQPACKET, 0);
if (sock < 0) {
printf("\n%s: Error creating socket: %s\n", __func__,
strerror(errno));
return -1;
}
telem_addr.sun_family = AF_UNIX;
snprintf(telem_addr.sun_path, sizeof(telem_addr.sun_path),
"%s/dpdk_telemetry.%s", rte_eal_get_runtime_dir(),
TELEMETRY_VERSION);
if (connect(sock, (struct sockaddr *) &telem_addr,
sizeof(telem_addr)) < 0) {
printf("\n%s: Error connecting to socket: %s\n", __func__,
strerror(errno));
close(sock);
return -1;
}
bytes = read(sock, buf, sizeof(buf) - 1);
if (bytes < 0) {
printf("%s: Error with socket read - %s\n", __func__,
strerror(errno));
close(sock);
return -1;
}
buf[bytes] = '\0';
printf("\n%s: %s\n", __func__, buf);
return sock;
}
static int
test_telemetry_data(void)
{
typedef int (*test_case)(void);
unsigned int i = 0;
sock = connect_to_socket();
if (sock <= 0)
return -1;
test_case test_cases[] = {test_case_array_string,
test_case_array_int, test_case_array_u64,
test_case_add_dict_int, test_case_add_dict_u64,
test_case_add_dict_string,
test_dict_with_array_int_values,
test_dict_with_array_u64_values,
test_dict_with_array_string_values,
test_array_with_array_int_values,
test_array_with_array_u64_values,
test_array_with_array_string_values };
rte_telemetry_register_cmd(REQUEST_CMD, test_cb, "Test");
for (i = 0; i < RTE_DIM(test_cases); i++) {
if (test_cases[i]() != 0) {
close(sock);
return -1;
}
}
close(sock);
return 0;
}
REGISTER_TEST_COMMAND(telemetry_data_autotest, test_telemetry_data);