1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include "zdtmtst.h"
const char *test_doc = "Check that data in dgram socket are restored correctly";
const char *test_author = "Andrew Vagin <avagin@openvz.org";
#define SK_SRV "\0socket_dgram_srv"
#define MSG "hello"
int main(int argc, char **argv)
{
struct sockaddr_un addr;
unsigned int addrlen;
int srv, clnt1, clnt2, ret;
char buf[1024];
test_init(argc, argv);
srv = socket(PF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0);
if (srv < 0) {
pr_perror("socket");
return 1;
}
clnt1 = socket(PF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0);
if (clnt1 < 0) {
pr_perror("socket");
return 1;
}
clnt2 = socket(PF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0);
if (clnt2 < 0) {
pr_perror("socket");
return 1;
}
addr.sun_family = AF_UNIX;
memcpy(addr.sun_path, SK_SRV, sizeof(SK_SRV));
addrlen = sizeof(addr.sun_family) + sizeof(SK_SRV);
if (bind(srv, (struct sockaddr *)&addr, addrlen)) {
fail("bind");
exit(1);
}
if (connect(clnt1, (struct sockaddr *)&addr, addrlen)) {
fail("connect");
exit(1);
}
if (connect(clnt2, (struct sockaddr *)&addr, addrlen)) {
fail("connect");
exit(1);
}
if (write(clnt1, MSG, sizeof(MSG)) != sizeof(MSG)) {
pr_perror("write");
return 1;
}
test_daemon();
test_waitsig();
ret = read(srv, buf, sizeof(buf));
buf[ret > 0 ? ret : 0] = 0;
if (ret != sizeof(MSG)) {
fail("%d: %s", ret, buf);
return 1;
}
ret = read(srv, buf, sizeof(buf));
if (ret != -1 || errno != EAGAIN) {
fail("unexpected data: %d", ret);
return 1;
}
pass();
return 0;
}
|