1*ac011361SKuniyuki Iwashima // SPDX-License-Identifier: GPL-2.0
2*ac011361SKuniyuki Iwashima /* Copyright Amazon.com Inc. or its affiliates. */
3*ac011361SKuniyuki Iwashima
4*ac011361SKuniyuki Iwashima #define _GNU_SOURCE
5*ac011361SKuniyuki Iwashima #include <sched.h>
6*ac011361SKuniyuki Iwashima
7*ac011361SKuniyuki Iwashima #include <unistd.h>
8*ac011361SKuniyuki Iwashima #include <linux/netlink.h>
9*ac011361SKuniyuki Iwashima #include <linux/rtnetlink.h>
10*ac011361SKuniyuki Iwashima #include <linux/sock_diag.h>
11*ac011361SKuniyuki Iwashima #include <linux/unix_diag.h>
12*ac011361SKuniyuki Iwashima #include <sys/socket.h>
13*ac011361SKuniyuki Iwashima #include <sys/stat.h>
14*ac011361SKuniyuki Iwashima #include <sys/types.h>
15*ac011361SKuniyuki Iwashima #include <sys/un.h>
16*ac011361SKuniyuki Iwashima
17*ac011361SKuniyuki Iwashima #include "../../kselftest_harness.h"
18*ac011361SKuniyuki Iwashima
FIXTURE(diag_uid)19*ac011361SKuniyuki Iwashima FIXTURE(diag_uid)
20*ac011361SKuniyuki Iwashima {
21*ac011361SKuniyuki Iwashima int netlink_fd;
22*ac011361SKuniyuki Iwashima int unix_fd;
23*ac011361SKuniyuki Iwashima __u32 inode;
24*ac011361SKuniyuki Iwashima __u64 cookie;
25*ac011361SKuniyuki Iwashima };
26*ac011361SKuniyuki Iwashima
FIXTURE_VARIANT(diag_uid)27*ac011361SKuniyuki Iwashima FIXTURE_VARIANT(diag_uid)
28*ac011361SKuniyuki Iwashima {
29*ac011361SKuniyuki Iwashima int unshare;
30*ac011361SKuniyuki Iwashima int udiag_show;
31*ac011361SKuniyuki Iwashima };
32*ac011361SKuniyuki Iwashima
FIXTURE_VARIANT_ADD(diag_uid,uid)33*ac011361SKuniyuki Iwashima FIXTURE_VARIANT_ADD(diag_uid, uid)
34*ac011361SKuniyuki Iwashima {
35*ac011361SKuniyuki Iwashima .unshare = 0,
36*ac011361SKuniyuki Iwashima .udiag_show = UDIAG_SHOW_UID
37*ac011361SKuniyuki Iwashima };
38*ac011361SKuniyuki Iwashima
FIXTURE_VARIANT_ADD(diag_uid,uid_unshare)39*ac011361SKuniyuki Iwashima FIXTURE_VARIANT_ADD(diag_uid, uid_unshare)
40*ac011361SKuniyuki Iwashima {
41*ac011361SKuniyuki Iwashima .unshare = CLONE_NEWUSER,
42*ac011361SKuniyuki Iwashima .udiag_show = UDIAG_SHOW_UID
43*ac011361SKuniyuki Iwashima };
44*ac011361SKuniyuki Iwashima
FIXTURE_SETUP(diag_uid)45*ac011361SKuniyuki Iwashima FIXTURE_SETUP(diag_uid)
46*ac011361SKuniyuki Iwashima {
47*ac011361SKuniyuki Iwashima struct stat file_stat;
48*ac011361SKuniyuki Iwashima socklen_t optlen;
49*ac011361SKuniyuki Iwashima int ret;
50*ac011361SKuniyuki Iwashima
51*ac011361SKuniyuki Iwashima if (variant->unshare)
52*ac011361SKuniyuki Iwashima ASSERT_EQ(unshare(variant->unshare), 0);
53*ac011361SKuniyuki Iwashima
54*ac011361SKuniyuki Iwashima self->netlink_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG);
55*ac011361SKuniyuki Iwashima ASSERT_NE(self->netlink_fd, -1);
56*ac011361SKuniyuki Iwashima
57*ac011361SKuniyuki Iwashima self->unix_fd = socket(AF_UNIX, SOCK_STREAM, 0);
58*ac011361SKuniyuki Iwashima ASSERT_NE(self->unix_fd, -1);
59*ac011361SKuniyuki Iwashima
60*ac011361SKuniyuki Iwashima ret = fstat(self->unix_fd, &file_stat);
61*ac011361SKuniyuki Iwashima ASSERT_EQ(ret, 0);
62*ac011361SKuniyuki Iwashima
63*ac011361SKuniyuki Iwashima self->inode = file_stat.st_ino;
64*ac011361SKuniyuki Iwashima
65*ac011361SKuniyuki Iwashima optlen = sizeof(self->cookie);
66*ac011361SKuniyuki Iwashima ret = getsockopt(self->unix_fd, SOL_SOCKET, SO_COOKIE, &self->cookie, &optlen);
67*ac011361SKuniyuki Iwashima ASSERT_EQ(ret, 0);
68*ac011361SKuniyuki Iwashima }
69*ac011361SKuniyuki Iwashima
FIXTURE_TEARDOWN(diag_uid)70*ac011361SKuniyuki Iwashima FIXTURE_TEARDOWN(diag_uid)
71*ac011361SKuniyuki Iwashima {
72*ac011361SKuniyuki Iwashima close(self->netlink_fd);
73*ac011361SKuniyuki Iwashima close(self->unix_fd);
74*ac011361SKuniyuki Iwashima }
75*ac011361SKuniyuki Iwashima
send_request(struct __test_metadata * _metadata,FIXTURE_DATA (diag_uid)* self,const FIXTURE_VARIANT (diag_uid)* variant)76*ac011361SKuniyuki Iwashima int send_request(struct __test_metadata *_metadata,
77*ac011361SKuniyuki Iwashima FIXTURE_DATA(diag_uid) *self,
78*ac011361SKuniyuki Iwashima const FIXTURE_VARIANT(diag_uid) *variant)
79*ac011361SKuniyuki Iwashima {
80*ac011361SKuniyuki Iwashima struct {
81*ac011361SKuniyuki Iwashima struct nlmsghdr nlh;
82*ac011361SKuniyuki Iwashima struct unix_diag_req udr;
83*ac011361SKuniyuki Iwashima } req = {
84*ac011361SKuniyuki Iwashima .nlh = {
85*ac011361SKuniyuki Iwashima .nlmsg_len = sizeof(req),
86*ac011361SKuniyuki Iwashima .nlmsg_type = SOCK_DIAG_BY_FAMILY,
87*ac011361SKuniyuki Iwashima .nlmsg_flags = NLM_F_REQUEST
88*ac011361SKuniyuki Iwashima },
89*ac011361SKuniyuki Iwashima .udr = {
90*ac011361SKuniyuki Iwashima .sdiag_family = AF_UNIX,
91*ac011361SKuniyuki Iwashima .udiag_ino = self->inode,
92*ac011361SKuniyuki Iwashima .udiag_cookie = {
93*ac011361SKuniyuki Iwashima (__u32)self->cookie,
94*ac011361SKuniyuki Iwashima (__u32)(self->cookie >> 32)
95*ac011361SKuniyuki Iwashima },
96*ac011361SKuniyuki Iwashima .udiag_show = variant->udiag_show
97*ac011361SKuniyuki Iwashima }
98*ac011361SKuniyuki Iwashima };
99*ac011361SKuniyuki Iwashima struct sockaddr_nl nladdr = {
100*ac011361SKuniyuki Iwashima .nl_family = AF_NETLINK
101*ac011361SKuniyuki Iwashima };
102*ac011361SKuniyuki Iwashima struct iovec iov = {
103*ac011361SKuniyuki Iwashima .iov_base = &req,
104*ac011361SKuniyuki Iwashima .iov_len = sizeof(req)
105*ac011361SKuniyuki Iwashima };
106*ac011361SKuniyuki Iwashima struct msghdr msg = {
107*ac011361SKuniyuki Iwashima .msg_name = &nladdr,
108*ac011361SKuniyuki Iwashima .msg_namelen = sizeof(nladdr),
109*ac011361SKuniyuki Iwashima .msg_iov = &iov,
110*ac011361SKuniyuki Iwashima .msg_iovlen = 1
111*ac011361SKuniyuki Iwashima };
112*ac011361SKuniyuki Iwashima
113*ac011361SKuniyuki Iwashima return sendmsg(self->netlink_fd, &msg, 0);
114*ac011361SKuniyuki Iwashima }
115*ac011361SKuniyuki Iwashima
render_response(struct __test_metadata * _metadata,struct unix_diag_req * udr,__u32 len)116*ac011361SKuniyuki Iwashima void render_response(struct __test_metadata *_metadata,
117*ac011361SKuniyuki Iwashima struct unix_diag_req *udr, __u32 len)
118*ac011361SKuniyuki Iwashima {
119*ac011361SKuniyuki Iwashima unsigned int rta_len = len - NLMSG_LENGTH(sizeof(*udr));
120*ac011361SKuniyuki Iwashima struct rtattr *attr;
121*ac011361SKuniyuki Iwashima uid_t uid;
122*ac011361SKuniyuki Iwashima
123*ac011361SKuniyuki Iwashima ASSERT_GT(len, sizeof(*udr));
124*ac011361SKuniyuki Iwashima ASSERT_EQ(udr->sdiag_family, AF_UNIX);
125*ac011361SKuniyuki Iwashima
126*ac011361SKuniyuki Iwashima attr = (struct rtattr *)(udr + 1);
127*ac011361SKuniyuki Iwashima ASSERT_NE(RTA_OK(attr, rta_len), 0);
128*ac011361SKuniyuki Iwashima ASSERT_EQ(attr->rta_type, UNIX_DIAG_UID);
129*ac011361SKuniyuki Iwashima
130*ac011361SKuniyuki Iwashima uid = *(uid_t *)RTA_DATA(attr);
131*ac011361SKuniyuki Iwashima ASSERT_EQ(uid, getuid());
132*ac011361SKuniyuki Iwashima }
133*ac011361SKuniyuki Iwashima
receive_response(struct __test_metadata * _metadata,FIXTURE_DATA (diag_uid)* self)134*ac011361SKuniyuki Iwashima void receive_response(struct __test_metadata *_metadata,
135*ac011361SKuniyuki Iwashima FIXTURE_DATA(diag_uid) *self)
136*ac011361SKuniyuki Iwashima {
137*ac011361SKuniyuki Iwashima long buf[8192 / sizeof(long)];
138*ac011361SKuniyuki Iwashima struct sockaddr_nl nladdr = {
139*ac011361SKuniyuki Iwashima .nl_family = AF_NETLINK
140*ac011361SKuniyuki Iwashima };
141*ac011361SKuniyuki Iwashima struct iovec iov = {
142*ac011361SKuniyuki Iwashima .iov_base = buf,
143*ac011361SKuniyuki Iwashima .iov_len = sizeof(buf)
144*ac011361SKuniyuki Iwashima };
145*ac011361SKuniyuki Iwashima struct msghdr msg = {
146*ac011361SKuniyuki Iwashima .msg_name = &nladdr,
147*ac011361SKuniyuki Iwashima .msg_namelen = sizeof(nladdr),
148*ac011361SKuniyuki Iwashima .msg_iov = &iov,
149*ac011361SKuniyuki Iwashima .msg_iovlen = 1
150*ac011361SKuniyuki Iwashima };
151*ac011361SKuniyuki Iwashima struct nlmsghdr *nlh;
152*ac011361SKuniyuki Iwashima int ret;
153*ac011361SKuniyuki Iwashima
154*ac011361SKuniyuki Iwashima ret = recvmsg(self->netlink_fd, &msg, 0);
155*ac011361SKuniyuki Iwashima ASSERT_GT(ret, 0);
156*ac011361SKuniyuki Iwashima
157*ac011361SKuniyuki Iwashima nlh = (struct nlmsghdr *)buf;
158*ac011361SKuniyuki Iwashima ASSERT_NE(NLMSG_OK(nlh, ret), 0);
159*ac011361SKuniyuki Iwashima ASSERT_EQ(nlh->nlmsg_type, SOCK_DIAG_BY_FAMILY);
160*ac011361SKuniyuki Iwashima
161*ac011361SKuniyuki Iwashima render_response(_metadata, NLMSG_DATA(nlh), nlh->nlmsg_len);
162*ac011361SKuniyuki Iwashima
163*ac011361SKuniyuki Iwashima nlh = NLMSG_NEXT(nlh, ret);
164*ac011361SKuniyuki Iwashima ASSERT_EQ(NLMSG_OK(nlh, ret), 0);
165*ac011361SKuniyuki Iwashima }
166*ac011361SKuniyuki Iwashima
167*ac011361SKuniyuki Iwashima TEST_F(diag_uid, 1)
168*ac011361SKuniyuki Iwashima {
169*ac011361SKuniyuki Iwashima int ret;
170*ac011361SKuniyuki Iwashima
171*ac011361SKuniyuki Iwashima ret = send_request(_metadata, self, variant);
172*ac011361SKuniyuki Iwashima ASSERT_GT(ret, 0);
173*ac011361SKuniyuki Iwashima
174*ac011361SKuniyuki Iwashima receive_response(_metadata, self);
175*ac011361SKuniyuki Iwashima }
176*ac011361SKuniyuki Iwashima
177*ac011361SKuniyuki Iwashima TEST_HARNESS_MAIN
178