1 /*
2 * Copyright (c) Wipro Technologies Ltd, 2002. All Rights Reserved.
3 * Copyright (c) 2012 Wanlong Gao <gaowanlong@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
19 /*
20 * Test to verify inheritance of environment variables by child.
21 */
22
23 #if defined UCLINUX && !__THROW
24 /* workaround for libc bug */
25 #define __THROW
26 #endif
27
28 #include <errno.h>
29 #include <sched.h>
30 #include <sys/wait.h>
31 #include <fcntl.h>
32 #include "test.h"
33 #include "clone_platform.h"
34
35 #define MAX_LINE_LENGTH 256
36
37 static void setup(void);
38 static void cleanup(void);
39 static int child_environ();
40
41 static int pfd[2];
42
43 char *TCID = "clone06";
44 int TST_TOTAL = 1;
45
main(int ac,char ** av)46 int main(int ac, char **av)
47 {
48
49 int lc, status;
50 void *child_stack;
51 char *parent_env;
52 char buff[MAX_LINE_LENGTH];
53
54 tst_parse_opts(ac, av, NULL, NULL);
55
56 setup();
57
58 child_stack = malloc(CHILD_STACK_SIZE);
59 if (child_stack == NULL)
60 tst_brkm(TBROK, cleanup, "Cannot allocate stack for child");
61
62 for (lc = 0; TEST_LOOPING(lc); lc++) {
63 tst_count = 0;
64
65 if ((pipe(pfd)) == -1)
66 tst_brkm(TBROK | TERRNO, cleanup, "pipe() failed");
67
68 TEST(ltp_clone(SIGCHLD, child_environ, NULL, CHILD_STACK_SIZE,
69 child_stack));
70
71 if (TEST_RETURN == -1)
72 tst_brkm(TFAIL | TTERRNO, cleanup, "clone failed");
73
74 /* close write end from parent */
75 if ((close(pfd[1])) == -1)
76 tst_resm(TWARN | TERRNO, "close(pfd[1]) failed");
77
78 /* Read env var from read end */
79 if ((read(pfd[0], buff, sizeof(buff))) == -1)
80 tst_brkm(TBROK | TERRNO, cleanup,
81 "read from pipe failed");
82
83 /* Close read end from parent */
84 if ((close(pfd[0])) == -1)
85 tst_resm(TWARN | TERRNO, "close(pfd[0]) failed");
86
87 parent_env = getenv("TERM") ? : "";
88
89 if ((strcmp(buff, parent_env)) == 0)
90 tst_resm(TPASS, "Test Passed");
91 else
92 tst_resm(TFAIL, "Test Failed");
93
94 if ((wait(&status)) == -1)
95 tst_brkm(TBROK | TERRNO, cleanup,
96 "wait failed, status: %d", status);
97 }
98
99 free(child_stack);
100
101 cleanup();
102 tst_exit();
103 }
104
setup(void)105 static void setup(void)
106 {
107 tst_sig(FORK, DEF_HANDLER, cleanup);
108 TEST_PAUSE;
109 }
110
cleanup(void)111 static void cleanup(void)
112 {
113 }
114
115 /*
116 * Function executed by child.
117 * Gets the value for environment variable,TERM &
118 * writes it to a pipe.
119 */
child_environ(void)120 static int child_environ(void)
121 {
122
123 char var[MAX_LINE_LENGTH];
124
125 /* Close read end from child */
126 if ((close(pfd[0])) == -1)
127 tst_brkm(TBROK | TERRNO, cleanup, "close(pfd[0]) failed");
128
129 if ((sprintf(var, "%s", getenv("TERM") ? : "")) < 0)
130 tst_resm(TWARN | TERRNO, "sprintf() failed");
131
132 if ((write(pfd[1], var, MAX_LINE_LENGTH)) == -1)
133 tst_resm(TWARN | TERRNO, "write to pipe failed");
134
135 /* Close write end of pipe from child */
136 if ((close(pfd[1])) == -1)
137 tst_resm(TWARN | TERRNO, "close(pfd[1]) failed");
138
139 exit(0);
140 }
141