113 lines
2.4 KiB
C
113 lines
2.4 KiB
C
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
/*
|
|
* Copyright (c) 2017 Red Hat, Inc.
|
|
* Author: Boyang Xue <bxue@redhat.com>
|
|
*/
|
|
|
|
/*
|
|
* Functional test for splice(2): pipe <-> socket
|
|
*
|
|
* This test case tests splice(2) from a pipe to a socket and vice versa
|
|
*/
|
|
|
|
#define _GNU_SOURCE
|
|
#include <errno.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/wait.h>
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include "tst_test.h"
|
|
#include "lapi/splice.h"
|
|
#include "splice.h"
|
|
|
|
#define PIPE_MAX (64*1024)
|
|
|
|
static char *str_len_data;
|
|
static int num_len_data = PIPE_MAX;
|
|
static char *arr_in, *arr_out;
|
|
|
|
static void setup(void)
|
|
{
|
|
int i, pipe_limit;
|
|
|
|
pipe_limit = get_max_limit(num_len_data);
|
|
num_len_data = pipe_limit;
|
|
|
|
if (tst_parse_int(str_len_data, &num_len_data, 1, pipe_limit)) {
|
|
tst_brk(TBROK, "Invalid length of data: '%s', "
|
|
"valid value: [1, %d]", str_len_data, pipe_limit);
|
|
}
|
|
tst_res(TINFO, "splice size = %d", num_len_data);
|
|
arr_in = SAFE_MALLOC(num_len_data);
|
|
arr_out = SAFE_MALLOC(num_len_data);
|
|
for (i = 0; i < num_len_data; i++)
|
|
arr_in[i] = i & 0xff;
|
|
|
|
}
|
|
|
|
static void cleanup(void)
|
|
{
|
|
free(arr_in);
|
|
free(arr_out);
|
|
}
|
|
|
|
static void pipe_socket(void)
|
|
{
|
|
int pp1[2], pp2[2], sv[2], i, ret;
|
|
|
|
SAFE_PIPE(pp1);
|
|
SAFE_PIPE(pp2);
|
|
SAFE_SOCKETPAIR(AF_UNIX, SOCK_STREAM, 0, sv);
|
|
|
|
SAFE_WRITE(1, pp1[1], arr_in, num_len_data);
|
|
for (i = num_len_data; i > 0; i = i - ret) {
|
|
ret = splice(pp1[0], NULL, sv[0], 0, i, 0);
|
|
if (ret == -1) {
|
|
tst_res(TFAIL | TERRNO, "splice error");
|
|
goto exit;
|
|
}
|
|
}
|
|
for (i = num_len_data; i > 0; i = i - ret) {
|
|
ret = splice(sv[1], 0, pp2[1], NULL, i, 0);
|
|
if (ret == -1) {
|
|
if (errno == EINVAL) {
|
|
tst_res(TCONF, "splice does not support "
|
|
"af_unix sockets");
|
|
} else {
|
|
tst_res(TFAIL | TERRNO, "splice error");
|
|
}
|
|
goto exit;
|
|
}
|
|
SAFE_READ(1, pp2[0], arr_out + num_len_data - i, ret);
|
|
}
|
|
|
|
for (i = 0; i < num_len_data; i++) {
|
|
if (arr_in[i] != arr_out[i]) {
|
|
tst_res(TFAIL, "wrong data at %d: expected: %d, "
|
|
"actual: %d", i, arr_in[i], arr_out[i]);
|
|
goto exit;
|
|
}
|
|
}
|
|
tst_res(TPASS, "splice(2): pipe <-> socket run pass.");
|
|
exit:
|
|
for (i = 0; i < 2; i++) {
|
|
SAFE_CLOSE(pp1[i]);
|
|
SAFE_CLOSE(pp2[i]);
|
|
SAFE_CLOSE(sv[i]);
|
|
}
|
|
}
|
|
|
|
static struct tst_test test = {
|
|
.test_all = pipe_socket,
|
|
.setup = setup,
|
|
.cleanup = cleanup,
|
|
.options = (struct tst_option[]) {
|
|
{"l:", &str_len_data, "-l <num> Length of test data (in bytes)"},
|
|
{}
|
|
},
|
|
.min_kver = "2.6.17"
|
|
};
|