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