Commit 12329bd5 authored by Stefano Garzarella's avatar Stefano Garzarella Committed by David S. Miller
Browse files

vsock/test: add send_buf() utility function 



Move the code of send_byte() out in a new utility function that
can be used to send a generic buffer.

This new function can be used when we need to send a custom
buffer and not just a single 'A' byte.

Signed-off-by: default avatarStefano Garzarella <sgarzare@redhat.com>
Reviewed-by: default avatarArseniy Krasnov <avkrasnov@salutedevices.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent a0bcb835

tools/testing/vsock/util.c

+54 −36
Original line number Diff line number Diff line
@@ -211,105 +211,123 @@ int vsock_seqpacket_accept(unsigned int cid, unsigned int port, 
	return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET);

}



/* Receive bytes in a buffer and check the return value.

/* Transmit bytes from a buffer and check the return value.

 *

 * expected_ret:

 *  <0 Negative errno (for testing errors)

 *   0 End-of-file

 *  >0 Success (bytes successfully read)

 *  >0 Success (bytes successfully written)

 */

void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret)

void send_buf(int fd, const void *buf, size_t len, int flags,

	      ssize_t expected_ret)

{

	ssize_t nread = 0;

	ssize_t nwritten = 0;

	ssize_t ret;



	timeout_begin(TIMEOUT);

	do {

		ret = recv(fd, buf + nread, len - nread, flags);

		timeout_check("recv");

		ret = send(fd, buf + nwritten, len - nwritten, flags);

		timeout_check("send");



		if (ret == 0 || (ret < 0 && errno != EINTR))

			break;



		nread += ret;

	} while (nread < len);

		nwritten += ret;

	} while (nwritten < len);

	timeout_end();



	if (expected_ret < 0) {

		if (ret != -1) {

			fprintf(stderr, "bogus recv(2) return value %zd (expected %zd)\n",

			fprintf(stderr, "bogus send(2) return value %zd (expected %zd)\n",

				ret, expected_ret);

			exit(EXIT_FAILURE);

		}

		if (errno != -expected_ret) {

			perror("recv");

			perror("send");

			exit(EXIT_FAILURE);

		}

		return;

	}



	if (ret < 0) {

		perror("recv");

		perror("send");

		exit(EXIT_FAILURE);

	}



	if (nread != expected_ret) {

	if (nwritten != expected_ret) {

		if (ret == 0)

			fprintf(stderr, "unexpected EOF while receiving bytes\n");

			fprintf(stderr, "unexpected EOF while sending bytes\n");



		fprintf(stderr, "bogus recv(2) bytes read %zd (expected %zd)\n",

			nread, expected_ret);

		fprintf(stderr, "bogus send(2) bytes written %zd (expected %zd)\n",

			nwritten, expected_ret);

		exit(EXIT_FAILURE);

	}

}



/* Transmit one byte and check the return value.

/* Receive bytes in a buffer and check the return value.

 *

 * expected_ret:

 *  <0 Negative errno (for testing errors)

 *   0 End-of-file

 *   1 Success

 *  >0 Success (bytes successfully read)

 */

void send_byte(int fd, int expected_ret, int flags)

void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret)

{

	const uint8_t byte = 'A';

	ssize_t nwritten;

	ssize_t nread = 0;

	ssize_t ret;



	timeout_begin(TIMEOUT);

	do {

		nwritten = send(fd, &byte, sizeof(byte), flags);

		timeout_check("write");

	} while (nwritten < 0 && errno == EINTR);

		ret = recv(fd, buf + nread, len - nread, flags);

		timeout_check("recv");



		if (ret == 0 || (ret < 0 && errno != EINTR))

			break;



		nread += ret;

	} while (nread < len);

	timeout_end();



	if (expected_ret < 0) {

		if (nwritten != -1) {

			fprintf(stderr, "bogus send(2) return value %zd\n",

				nwritten);

		if (ret != -1) {

			fprintf(stderr, "bogus recv(2) return value %zd (expected %zd)\n",

				ret, expected_ret);

			exit(EXIT_FAILURE);

		}

		if (errno != -expected_ret) {

			perror("write");

			perror("recv");

			exit(EXIT_FAILURE);

		}

		return;

	}



	if (nwritten < 0) {

		perror("write");

	if (ret < 0) {

		perror("recv");

		exit(EXIT_FAILURE);

	}

	if (nwritten == 0) {

		if (expected_ret == 0)

			return;



		fprintf(stderr, "unexpected EOF while sending byte\n");

	if (nread != expected_ret) {

		if (ret == 0)

			fprintf(stderr, "unexpected EOF while receiving bytes\n");



		fprintf(stderr, "bogus recv(2) bytes read %zd (expected %zd)\n",

			nread, expected_ret);

		exit(EXIT_FAILURE);

	}

	if (nwritten != sizeof(byte)) {

		fprintf(stderr, "bogus send(2) return value %zd\n", nwritten);

		exit(EXIT_FAILURE);

}



/* Transmit one byte and check the return value.

 *

 * expected_ret:

 *  <0 Negative errno (for testing errors)

 *   0 End-of-file

 *   1 Success

 */

void send_byte(int fd, int expected_ret, int flags)

{

	const uint8_t byte = 'A';



	send_buf(fd, &byte, sizeof(byte), flags, expected_ret);

}



/* Receive one byte and check the return value.

tools/testing/vsock/util.h

+2 −0
Original line number Diff line number Diff line
@@ -42,6 +42,8 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, 
int vsock_seqpacket_accept(unsigned int cid, unsigned int port,

			   struct sockaddr_vm *clientaddrp);

void vsock_wait_remote_close(int fd);

void send_buf(int fd, const void *buf, size_t len, int flags,

	      ssize_t expected_ret);

void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret);

void send_byte(int fd, int expected_ret, int flags);

void recv_byte(int fd, int expected_ret, int flags);