From 913f780e1077efada6674c7a14b687ba5ac1b6d0 Mon Sep 17 00:00:00 2001
From: Ben Walker <benjamin.walker@intel.com>
Date: Thu, 17 Oct 2019 13:58:46 -0700
Subject: [PATCH] test: Add mocks for sendmsg and recvmsg

These will be used from the posix sock layer.

Change-Id: I6427d9e9aee889e11ba7e36876112a5aba449e31
Signed-off-by: Ben Walker <benjamin.walker@intel.com>
Reviewed-on: https://review.gerrithub.io/c/spdk/spdk/+/471751
Tested-by: SPDK CI Jenkins <sys_sgci@intel.com>
Reviewed-by: Paul Luse <paul.e.luse@intel.com>
Reviewed-by: Alexey Marchuk <alexeymar@mellanox.com>
Reviewed-by: Shuhei Matsumoto <shuhei.matsumoto.xt@hitachi.com>
Reviewed-by: Jim Harris <james.r.harris@intel.com>
---
 include/spdk_internal/mock.h | 4 ++++
 lib/ut_mock/mock.c           | 5 +++++
 mk/spdk.mock.unittest.mk     | 2 ++
 3 files changed, 11 insertions(+)

diff --git a/include/spdk_internal/mock.h b/include/spdk_internal/mock.h
index 397ad5f3c..705378fac 100644
--- a/include/spdk_internal/mock.h
+++ b/include/spdk_internal/mock.h
@@ -116,6 +116,10 @@ DECLARE_WRAPPER(pthread_mutex_init, int,
 DECLARE_WRAPPER(pthread_mutexattr_init, int,
 		(pthread_mutexattr_t *attr));
 
+DECLARE_WRAPPER(recvmsg, ssize_t, (int sockfd, struct msghdr *msg, int flags));
+
+DECLARE_WRAPPER(sendmsg, ssize_t, (int sockfd, const struct msghdr *msg, int flags));
+
 DECLARE_WRAPPER(writev, ssize_t, (int fd, const struct iovec *iov, int iovcnt));
 
 #endif /* SPDK_INTERNAL_MOCK_H */
diff --git a/lib/ut_mock/mock.c b/lib/ut_mock/mock.c
index 4ecb2b27c..a7b2a23d3 100644
--- a/lib/ut_mock/mock.c
+++ b/lib/ut_mock/mock.c
@@ -42,4 +42,9 @@ DEFINE_WRAPPER(pthread_mutex_init, int,
 DEFINE_WRAPPER(pthread_mutexattr_init, int,
 	       (pthread_mutexattr_t *attr), (attr))
 
+DEFINE_WRAPPER(recvmsg, ssize_t, (int sockfd, struct msghdr *msg, int flags), (sockfd, msg, flags))
+
+DEFINE_WRAPPER(sendmsg, ssize_t, (int sockfd, const struct msghdr *msg, int flags), (sockfd, msg,
+		flags))
+
 DEFINE_WRAPPER(writev, ssize_t, (int fd, const struct iovec *iov, int iovcnt), (fd, iov, iovcnt))
diff --git a/mk/spdk.mock.unittest.mk b/mk/spdk.mock.unittest.mk
index ee2a41906..90b28cf4d 100644
--- a/mk/spdk.mock.unittest.mk
+++ b/mk/spdk.mock.unittest.mk
@@ -34,4 +34,6 @@ LDFLAGS += \
 	-Wl,--wrap,calloc \
 	-Wl,--wrap,pthread_mutexattr_init \
 	-Wl,--wrap,pthread_mutex_init \
+	-Wl,--wrap,recvmsg \
+	-Wl,--wrap,sendmsg \
 	-Wl,--wrap,writev