1 /*
2 * Copyright (c) 2014 Fujitsu Ltd.
3 * Author: Zeng Linggang <zenglg.jy@cn.fujitsu.com>
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of version 2 of the GNU General Public License as
7 * published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope that it would be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12 *
13 * You should have received a copy of the GNU General Public License along
14 * with this program.
15 */
16 /*
17 * Test Description:
18 * Verify that,
19 * The flag of fchownat() is AT_SYMLINK_NOFOLLOW and the pathname would
20 * not be dereferenced if the pathname is a symbolic link.
21 */
22
23 #define _GNU_SOURCE
24
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <fcntl.h>
28 #include <unistd.h>
29 #include <stdlib.h>
30 #include <errno.h>
31 #include <string.h>
32 #include <signal.h>
33 #include "test.h"
34 #include "safe_macros.h"
35 #include "fchownat.h"
36 #include "lapi/fcntl.h"
37
38 #define TESTFILE "testfile"
39 #define TESTFILE_LINK "testfile_link"
40
41 char *TCID = "fchownat02";
42 int TST_TOTAL = 1;
43
44 static int dir_fd;
45 static uid_t set_uid = 1000;
46 static gid_t set_gid = 1000;
47 static void setup(void);
48 static void cleanup(void);
49 static void test_verify(void);
50 static void fchownat_verify(void);
51
main(int ac,char ** av)52 int main(int ac, char **av)
53 {
54 int lc;
55 int i;
56
57 tst_parse_opts(ac, av, NULL, NULL);
58
59 setup();
60
61 for (lc = 0; TEST_LOOPING(lc); lc++) {
62 tst_count = 0;
63 for (i = 0; i < TST_TOTAL; i++)
64 fchownat_verify();
65 }
66
67 cleanup();
68 tst_exit();
69 }
70
setup(void)71 static void setup(void)
72 {
73 struct stat c_buf, l_buf;
74
75 if ((tst_kvercmp(2, 6, 16)) < 0)
76 tst_brkm(TCONF, NULL, "This test needs kernel 2.6.16 or newer");
77
78 tst_require_root();
79
80 tst_sig(NOFORK, DEF_HANDLER, cleanup);
81
82 TEST_PAUSE;
83
84 tst_tmpdir();
85
86 dir_fd = SAFE_OPEN(cleanup, "./", O_DIRECTORY);
87
88 SAFE_TOUCH(cleanup, TESTFILE, 0600, NULL);
89
90 SAFE_SYMLINK(cleanup, TESTFILE, TESTFILE_LINK);
91
92 SAFE_STAT(cleanup, TESTFILE_LINK, &c_buf);
93
94 SAFE_LSTAT(cleanup, TESTFILE_LINK, &l_buf);
95
96 if (l_buf.st_uid == set_uid || l_buf.st_gid == set_gid) {
97 tst_brkm(TBROK | TERRNO, cleanup,
98 "link_uid(%d) == set_uid(%d) or link_gid(%d) == "
99 "set_gid(%d)", l_buf.st_uid, set_uid, l_buf.st_gid,
100 set_gid);
101 }
102 }
103
fchownat_verify(void)104 static void fchownat_verify(void)
105 {
106 TEST(fchownat(dir_fd, TESTFILE_LINK, set_uid, set_gid,
107 AT_SYMLINK_NOFOLLOW));
108
109 if (TEST_RETURN != 0) {
110 tst_resm(TFAIL | TTERRNO, "fchownat() failed, errno=%d : %s",
111 TEST_ERRNO, strerror(TEST_ERRNO));
112 } else {
113 test_verify();
114 }
115 }
116
test_verify(void)117 static void test_verify(void)
118 {
119 struct stat c_buf, l_buf;
120
121 SAFE_STAT(cleanup, TESTFILE_LINK, &c_buf);
122
123 SAFE_LSTAT(cleanup, TESTFILE_LINK, &l_buf);
124
125 if (c_buf.st_uid != set_uid && l_buf.st_uid == set_uid &&
126 c_buf.st_gid != set_gid && l_buf.st_gid == set_gid) {
127 tst_resm(TPASS, "fchownat() test AT_SYMLINK_NOFOLLOW success");
128 } else {
129 tst_resm(TFAIL,
130 "fchownat() test AT_SYMLINK_NOFOLLOW fail with uid=%d "
131 "link_uid=%d set_uid=%d | gid=%d link_gid=%d "
132 "set_gid=%d", c_buf.st_uid, l_buf.st_uid, set_uid,
133 c_buf.st_gid, l_buf.st_gid, set_gid);
134 }
135 }
136
cleanup(void)137 static void cleanup(void)
138 {
139 tst_rmdir();
140 }
141