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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <sys/un.h>
#include <sys/stat.h>
#include <limits.h>
#include <fcntl.h>
#include "zdtmtst.h"
const char *test_doc = "Test external sockets\n";
const char *test_author = "Andrey Vagin <avagin@openvz.org";
#define SK_DATA "packet"
int main(int argc, char *argv[])
{
struct sockaddr_un addr;
unsigned int addrlen;
task_waiter_t lock;
char dir[] = "/tmp/zdtm.unix.sock.XXXXXX";
char *path;
pid_t pid;
int ret, sk;
char *val;
unsetenv("ZDTM_GROUPS");
val = getenv("ZDTM_GID");
if (val && (setgid(atoi(val)) == -1)) {
fprintf(stderr, "Can't set gid: %m");
exit(1);
}
val = getenv("ZDTM_UID");
if (val && (setuid(atoi(val)) == -1)) {
fprintf(stderr, "Can't set uid: %m");
exit(1);
}
if (mkdtemp(dir) < 0) {
pr_perror("mkdtemp(%s) failed", dir);
return 1;
}
chmod(dir, 0777);
addr.sun_family = AF_UNIX;
snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s", dir, "sock");
path = addr.sun_path;
addrlen = sizeof(addr.sun_family) + strlen(path);
task_waiter_init(&lock);
pid = fork();
if (pid < 0) {
pr_perror("fork() failed");
return 1;
} else if (pid == 0) {
char c;
test_ext_init(argc, argv);
sk = socket(AF_UNIX, SOCK_DGRAM, 0);
if (sk < 0) {
pr_perror("Can't create socket");
return 1;
}
ret = bind(sk, (struct sockaddr *)&addr, addrlen);
if (ret < 0) {
pr_perror("Can't bind socket to %s", path);
return 1;
}
chmod(dir, 0777);
chmod(path, 0777);
test_msg("The external socket %s\n", path);
task_waiter_complete(&lock, 1);
task_waiter_fini(&lock);
recv(sk, &c, sizeof(c), 0);
return 0;
}
task_waiter_wait4(&lock, 1);
task_waiter_fini(&lock);
test_init(argc, argv);
sk = socket(AF_UNIX, SOCK_DGRAM, 0);
if (sk < 0) {
pr_perror("Can't create socket");
return 1;
}
ret = connect(sk, (struct sockaddr *)&addr, addrlen);
if (ret < 0) {
pr_perror("Can't connect socket");
return 1;
}
test_daemon();
test_waitsig();
if (unlink(path)) {
pr_perror("Unable to remove %s", path);
return 1;
}
if (rmdir(dir)) {
pr_perror("Unable to remove %s", dir);
return 1;
}
ret = send(sk, "H", 1, 0);
if (ret != 1) {
pr_perror("Can't send a symbol");
fail();
return 1;
}
pass();
return 0;
}
|
