126 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			126 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
| // SPDX-License-Identifier: GPL-2.0-or-later
 | |
| /*
 | |
|  * Copyright (c) International Business Machines  Corp., 2001
 | |
|  *   07/2001 Ported by Wayne Boyer
 | |
|  * Copyright (c) 2013 Cyril Hrubis <chrubis@suse.cz>
 | |
|  */
 | |
| 
 | |
| /*
 | |
|  * DESCRIPTION
 | |
|  *	Testcase to check the basic functionality of the readv(2) system call.
 | |
|  *
 | |
|  * ALGORITHM
 | |
|  *	Create a IO vector, and attempt to readv() various components of it.
 | |
|  */
 | |
| #include <stdlib.h>
 | |
| #include <sys/types.h>
 | |
| #include <sys/uio.h>
 | |
| #include <fcntl.h>
 | |
| #include <memory.h>
 | |
| 
 | |
| #include "tst_test.h"
 | |
| 
 | |
| /* Note: multi_iovec test assumes CHUNK is divisible by 4 */
 | |
| #define	CHUNK		64
 | |
| 
 | |
| static char buf[CHUNK];
 | |
| static struct iovec *rd_iovec, *big_iovec, *multi_iovec, *lockup_iovec;
 | |
| static int fd;
 | |
| 
 | |
| static struct testcase {
 | |
| 	struct iovec **iov;
 | |
| 	int iov_count, exp_ret;
 | |
| 	const char *name;
 | |
| } testcase_list[] = {
 | |
| 	{&rd_iovec, 0, 0, "readv() with 0 I/O vectors"},
 | |
| 	{&rd_iovec, 3, CHUNK, "readv() with NULL I/O vectors"},
 | |
| 	{&big_iovec, 2, CHUNK, "readv() with too big I/O vectors"},
 | |
| 	{&multi_iovec, 2, 3*CHUNK/4, "readv() with multiple I/O vectors"},
 | |
| 	{&lockup_iovec, 2, CHUNK, "readv() with zero-len buffer"},
 | |
| };
 | |
| 
 | |
| static void test_readv(unsigned int n)
 | |
| {
 | |
| 	int i, fpos, fail = 0;
 | |
| 	size_t j;
 | |
| 	char *ptr;
 | |
| 	const struct testcase *tc = testcase_list + n;
 | |
| 	struct iovec *vec;
 | |
| 
 | |
| 	SAFE_LSEEK(fd, 0, SEEK_SET);
 | |
| 	vec = *tc->iov;
 | |
| 
 | |
| 	for (i = 0; i < tc->iov_count; i++) {
 | |
| 		if (vec[i].iov_base && vec[i].iov_len)
 | |
| 			memset(vec[i].iov_base, 0, vec[i].iov_len);
 | |
| 	}
 | |
| 
 | |
| 	TEST(readv(fd, vec, tc->iov_count));
 | |
| 
 | |
| 	if (TST_RET == -1)
 | |
| 		tst_res(TFAIL | TTERRNO, "readv() failed unexpectedly");
 | |
| 	else if (TST_RET < 0)
 | |
| 		tst_res(TFAIL | TTERRNO, "readv() returned invalid value");
 | |
| 	else if (TST_RET != tc->exp_ret)
 | |
| 		tst_res(TFAIL, "readv() returned unexpected value %ld",
 | |
| 			TST_RET);
 | |
| 
 | |
| 	if (TST_RET != tc->exp_ret)
 | |
| 		return;
 | |
| 
 | |
| 	tst_res(TPASS, "%s", tc->name);
 | |
| 
 | |
| 	for (i = 0, fpos = 0; i < tc->iov_count; i++) {
 | |
| 		ptr = vec[i].iov_base;
 | |
| 
 | |
| 		for (j = 0; j < vec[i].iov_len; j++, fpos++) {
 | |
| 			if (ptr[j] != (fpos < tc->exp_ret ? 0x42 : 0))
 | |
| 				fail++;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (fail)
 | |
| 		tst_res(TFAIL, "Wrong buffer content");
 | |
| 	else
 | |
| 		tst_res(TPASS, "readv() correctly read %d bytes ", tc->exp_ret);
 | |
| }
 | |
| 
 | |
| static void setup(void)
 | |
| {
 | |
| 	/* replace the default NULL pointer with invalid address */
 | |
| 	lockup_iovec[0].iov_base = tst_get_bad_addr(NULL);
 | |
| 
 | |
| 	memset(buf, 0x42, sizeof(buf));
 | |
| 
 | |
| 	fd = SAFE_OPEN("data_file", O_WRONLY | O_CREAT | O_TRUNC, 0666);
 | |
| 	SAFE_WRITE(1, fd, buf, sizeof(buf));
 | |
| 	SAFE_CLOSE(fd);
 | |
| 	fd = SAFE_OPEN("data_file", O_RDONLY);
 | |
| }
 | |
| 
 | |
| static void cleanup(void)
 | |
| {
 | |
| 	if (fd >= 0)
 | |
| 		SAFE_CLOSE(fd);
 | |
| }
 | |
| 
 | |
| static struct tst_test test = {
 | |
| 	.setup = setup,
 | |
| 	.cleanup = cleanup,
 | |
| 	.test = test_readv,
 | |
| 	.tcnt = ARRAY_SIZE(testcase_list),
 | |
| 	.needs_tmpdir = 1,
 | |
| 	.timeout = 15,
 | |
| 	.tags = (const struct tst_tag[]) {
 | |
| 		{"linux-git", "19f18459330f"},
 | |
| 		{}
 | |
| 	},
 | |
| 	.bufs = (struct tst_buffers[]) {
 | |
| 		{&rd_iovec, .iov_sizes = (int[]){CHUNK, 0, 0, -1}},
 | |
| 		{&big_iovec, .iov_sizes = (int[]){2*CHUNK, CHUNK, -1}},
 | |
| 		{&multi_iovec, .iov_sizes = (int[]){CHUNK/4, CHUNK/2, -1}},
 | |
| 		{&lockup_iovec, .iov_sizes = (int[]){0, CHUNK, -1}},
 | |
| 		{}
 | |
| 	}
 | |
| };
 |