src: udp_sink add test of recv(2)

Suggested-by: David Laight <David.Laight@ACULAB.COM>
Signed-off-by: Jesper Dangaard Brouer <brouer@redhat.com>

Author: netoptimizer

src/udp_sink.c

@@ -44,7 +44,8 @@ static const char *__doc__=
 #define RUN_RECVMMSG  0x2
 #define RUN_RECVFROM  0x4
 #define RUN_READ      0x8
-#define RUN_ALL       (RUN_RECVMSG | RUN_RECVMMSG | RUN_RECVFROM | RUN_READ)
+#define RUN_RECV      0x10
+#define RUN_ALL (RUN_RECVMSG | RUN_RECVMMSG | RUN_RECVFROM | RUN_READ |RUN_RECV)
 
 struct sink_params {
 	struct params_common c;
@@ -79,6 +80,7 @@ static const struct option long_options[] = {
 	{"recvfrom",	no_argument,		NULL, 't' },
 	{"recvmsg",	no_argument,		NULL, 'u' },
 	{"recvmmsg",	no_argument,		NULL, 'U' },
+	{"recv",	no_argument,		NULL, 176 },
 	/* Other options */
 	{"help",	no_argument,		NULL, 'h' },
 	{"ipv4",	no_argument,		NULL, '4' },
@@ -321,6 +323,41 @@ static int sink_with_recvfrom(int sockfd, struct sink_params *p,
 	exit(EXIT_FAIL_SOCK);
 }
 
+static int sink_with_recv(int sockfd, struct sink_params *p,
+			  struct time_bench_record *r) {
+	int i, res;
+	uint64_t total = 0;
+	char *buffer = malloc_payload_buffer(p->buf_sz);
+	int flags = p->dontwait ? MSG_DONTWAIT : 0;
+
+	for (i = 0; i < p->count; i++) {
+		res = recv(sockfd, buffer, p->buf_sz, flags);
+		if (res < 0) {
+			if (errno == EAGAIN) {
+				r->try_again++;
+				continue;
+			}
+			goto error;
+		}
+		total += res;
+	}
+	r->bytes = total;
+	if (verbose > 0)
+		printf(" - read %lu bytes in %d packets = %lu bytes payload\n",
+		       total, i, total / i);
+
+	free(buffer);
+	return (i - r->try_again);
+
+ error: /* ugly construct to make sure the loop is small */
+	fprintf(stderr, "ERROR: %s() failed (%d) errno(%d) ",
+		__func__, res, errno);
+	perror("- recv");
+	free(buffer);
+	close(sockfd);
+	exit(EXIT_FAIL_SOCK);
+}
+
 static void setup_msg_name(struct msghdr *msg_hdr,
 			   struct sockaddr_storage *addr, int family)
 {
@@ -839,6 +876,7 @@ int main(int argc, char *argv[])
 		if (c == 'U') p.run_flag   |= RUN_RECVMMSG;
 		if (c == 't') p.run_flag   |= RUN_RECVFROM;
 		if (c == 'T') p.run_flag   |= RUN_READ;
+		if (c == 176) p.run_flag   |= RUN_RECV;
 		if (c == 'h' || c == '?') return usage(argv);
 	}
 
@@ -937,6 +975,11 @@ int main(int argc, char *argv[])
 		time_function(sockfd, &p, "recvfrom", sink_with_recvfrom);
 	}
 
+	if (p.run_flag       & RUN_RECV) {
+		init_stats(&p, RUN_RECV);
+		time_function(sockfd, &p, "recv", sink_with_recv);
+	}
+
 	close(sockfd);
 	return 0;
 }