/*
 * Copyright (c) 2017 Red Hat, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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, see <http://www.gnu.org/licenses/>.
 *
 * 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 struct tst_option options[] = {
	{"l:", &str_len_data, "-l <num> Length of test data (in bytes)"},
	{NULL, NULL, NULL},
};

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);
	if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv))
		tst_brk(TBROK | TERRNO, "fail to create socket pair.");

	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 = options,
	.min_kver = "2.6.17"
};
