blob: 496111bf230dbfdefc76685430f5278f5bd53586 [file] [log] [blame]
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <sys/socket.h>
#include <unistd.h>
#include <fcntl.h>
#include <assert.h>
int
main(void)
{
#if defined(HAVE_SENDMMSG) && defined(HAVE_STRUCT_MMSGHDR)
const int R = 0, W = 1;
int fd;
int sv[2];
char one[] = "one";
char two[] = "two";
char three[] = "three";
struct iovec iov[] = {
{
.iov_base = one,
.iov_len = sizeof(one) - 1
}, {
.iov_base = two,
.iov_len = sizeof(two) - 1
}, {
.iov_base = three,
.iov_len = sizeof(three) - 1
}
};
struct mmsghdr mmh[] = {
{
.msg_hdr = {
.msg_iov = iov + 0,
.msg_iovlen = 2,
}
}, {
.msg_hdr = {
.msg_iov = iov + 2,
.msg_iovlen = 1,
}
}
};
#define n_mmh (sizeof(mmh)/sizeof(mmh[0]))
/*
* Following open/dup2/close calls make the output of strace
* more predictable, so we can just compare the output and
* expected output (mmsg.expected) for testing purposes.
*/
while ((fd = open("/dev/null", O_RDWR)) < 3)
assert(fd >= 0);
(void) close(3);
assert(socketpair(AF_UNIX, SOCK_DGRAM, 0, sv) == 0);
assert(dup2(sv[W], W) == W);
assert(close(sv[W]) == 0);
assert(dup2(sv[R], R) == R);
assert(close(sv[R]) == 0);
assert(sendmmsg(W, mmh, n_mmh, 0) == n_mmh);
assert(close(W) == 0);
assert(recvmmsg(R, mmh, n_mmh, 0, NULL) == n_mmh);
assert(close(R) == 0);
return 0;
#else
return 77;
#endif
}