1 /*
2 * Copyright (c) 2014 Fujitsu Ltd.
3 * Author: Xing Gu <gux.fnst@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; if not, write the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
16 */
17 /*
18 * Description:
19 * Verify that,
20 * 1) tee() returns -1 and sets errno to EINVAL if fd_in does
21 * not refer to a pipe.
22 * 2) tee() returns -1 and sets errno to EINVAL if fd_out does
23 * not refer to a pipe.
24 * 3) tee() returns -1 and sets errno to EINVAL if fd_in and
25 * fd_out refer to the same pipe.
26 */
27
28 #define _GNU_SOURCE
29
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #include <fcntl.h>
33 #include <unistd.h>
34
35 #include "test.h"
36 #include "safe_macros.h"
37 #include "lapi/tee.h"
38
39 #define TEST_FILE "testfile"
40
41 #define STR "abcdefghigklmnopqrstuvwxyz"
42 #define TEE_TEST_LEN 10
43
44 static int fd;
45 static int pipes[2];
46
47 static struct test_case_t {
48 int *fdin;
49 int *fdout;
50 int exp_errno;
51 } test_cases[] = {
52 { &fd, &pipes[1], EINVAL },
53 { &pipes[0], &fd, EINVAL },
54 { &pipes[0], &pipes[1], EINVAL },
55 };
56
57 static void setup(void);
58 static void cleanup(void);
59 static void tee_verify(const struct test_case_t *);
60
61 char *TCID = "tee02";
62 int TST_TOTAL = ARRAY_SIZE(test_cases);
63
main(int ac,char ** av)64 int main(int ac, char **av)
65 {
66 int i, lc;
67
68 tst_parse_opts(ac, av, NULL, NULL);
69
70 setup();
71
72 for (lc = 0; TEST_LOOPING(lc); lc++) {
73 tst_count = 0;
74
75 for (i = 0; i < TST_TOTAL; i++)
76 tee_verify(&test_cases[i]);
77 }
78
79 cleanup();
80 tst_exit();
81 }
82
setup(void)83 static void setup(void)
84 {
85 if ((tst_kvercmp(2, 6, 17)) < 0) {
86 tst_brkm(TCONF, cleanup, "This test can only run on kernels "
87 "that are 2.6.17 or higher");
88 }
89
90 TEST_PAUSE;
91
92 tst_sig(FORK, DEF_HANDLER, cleanup);
93
94 tst_tmpdir();
95
96 fd = SAFE_OPEN(cleanup, TEST_FILE, O_RDWR | O_CREAT);
97
98 SAFE_PIPE(cleanup, pipes);
99 SAFE_WRITE(cleanup, 1, pipes[1], STR, sizeof(STR) - 1);
100 }
101
tee_verify(const struct test_case_t * tc)102 static void tee_verify(const struct test_case_t *tc)
103 {
104 TEST(tee(*(tc->fdin), *(tc->fdout), TEE_TEST_LEN, 0));
105
106 if (TEST_RETURN != -1) {
107 tst_resm(TFAIL, "tee() returned %ld, "
108 "expected -1, errno:%d", TEST_RETURN,
109 tc->exp_errno);
110 return;
111 }
112
113 if (TEST_ERRNO == tc->exp_errno) {
114 tst_resm(TPASS | TTERRNO, "tee() failed as expected");
115 } else {
116 tst_resm(TFAIL | TTERRNO,
117 "tee() failed unexpectedly; expected: %d - %s",
118 tc->exp_errno, strerror(tc->exp_errno));
119 }
120 }
121
cleanup(void)122 static void cleanup(void)
123 {
124 if (fd && close(fd) < 0)
125 tst_resm(TWARN | TERRNO, "close fd failed");
126
127 if (pipes[0] && close(pipes[0]) < 0)
128 tst_resm(TWARN | TERRNO, "close pipes[0] failed");
129
130 if (pipes[1] && close(pipes[1]) < 0)
131 tst_resm(TWARN | TERRNO, "close pipes[1] failed");
132
133 tst_rmdir();
134 }
135