1 /*
2 *
3 * Copyright (c) International Business Machines Corp., 2004
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
13 * the GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 */
19
20 /*
21 * NAME
22 * hugeshmat02.c
23 *
24 * DESCRIPTION
25 * hugeshmat02 - check for EINVAL and EACCES errors with hugetlb
26 *
27 * ALGORITHM
28 * loop if that option was specified
29 * call shmat() using three invalid test cases
30 * check the errno value
31 * issue a PASS message if we get EINVAL or EACCES
32 * otherwise, the tests fails
33 * issue a FAIL message
34 * call cleanup
35 *
36 * USAGE: <for command-line>
37 * hugeshmat02 [-c n] [-e] [-i n] [-I x] [-P x] [-t]
38 * where, -c n : Run n copies concurrently.
39 * -e : Turn on errno logging.
40 * -i n : Execute test n times.
41 * -I x : Execute test for x seconds.
42 * -P x : Pause for x seconds between iterations.
43 * -t : Turn on syscall timing.
44 *
45 * HISTORY
46 * 03/2001 - Written by Wayne Boyer
47 * 04/2004 - Updated By Robbie Williamson
48 *
49 * RESTRICTIONS
50 * Must be ran as root
51 */
52
53 #include <pwd.h>
54 #include "hugetlb.h"
55 #include "safe_macros.h"
56 #include "mem.h"
57
58 char *TCID = "hugeshmat02";
59 int TST_TOTAL = 2;
60
61 #if __WORDSIZE == 64
62 #define NADDR 0x10000000eef /* a 64bit non alligned address value */
63 #else
64 #define NADDR 0x60000eef /* a non alligned address value */
65 #endif
66
67 static size_t shm_size;
68 static int shm_id_1 = -1;
69 static int shm_id_2 = -1;
70 static void *addr;
71
72 static long hugepages = 128;
73 static option_t options[] = {
74 {"s:", &sflag, &nr_opt},
75 {NULL, NULL, NULL}
76 };
77
78 struct test_case_t {
79 int *shmid;
80 void *addr;
81 int error;
82 } TC[] = {
83 /* EINVAL - the shared memory ID is not valid */
84 {
85 &shm_id_1, NULL, EINVAL},
86 /* EINVAL - the address is not page aligned and SHM_RND is not given */
87 {
88 &shm_id_2, (void *)NADDR, EINVAL},};
89
main(int ac,char ** av)90 int main(int ac, char **av)
91 {
92 int lc, i;
93
94 tst_parse_opts(ac, av, options, NULL);
95
96 if (sflag)
97 hugepages = SAFE_STRTOL(NULL, nr_opt, 0, LONG_MAX);
98
99 setup();
100
101 for (lc = 0; TEST_LOOPING(lc); lc++) {
102 tst_count = 0;
103
104 for (i = 0; i < TST_TOTAL; i++) {
105 addr = shmat(*(TC[i].shmid), TC[i].addr, 0);
106 if (addr != (void *)-1) {
107 tst_resm(TFAIL, "shmat suceeded unexpectedly");
108 continue;
109 }
110 if (errno == TC[i].error)
111 tst_resm(TPASS | TERRNO, "shmat failed as "
112 "expected");
113 else
114 tst_resm(TFAIL | TERRNO, "shmat failed "
115 "unexpectedly - expect errno=%d, "
116 "got", TC[i].error);
117 }
118 }
119 cleanup();
120 tst_exit();
121 }
122
setup(void)123 void setup(void)
124 {
125 long hpage_size;
126
127 tst_require_root();
128 check_hugepage();
129 tst_sig(NOFORK, DEF_HANDLER, cleanup);
130 tst_tmpdir();
131
132 orig_hugepages = get_sys_tune("nr_hugepages");
133 set_sys_tune("nr_hugepages", hugepages, 1);
134 hpage_size = read_meminfo("Hugepagesize:") * 1024;
135
136 shm_size = hpage_size * hugepages / 2;
137 update_shm_size(&shm_size);
138 shmkey = getipckey(cleanup);
139
140 /* create a shared memory resource with read and write permissions */
141 /* also post increment the shmkey for the next shmget call */
142 shm_id_2 = shmget(shmkey++, shm_size,
143 SHM_HUGETLB | SHM_RW | IPC_CREAT | IPC_EXCL);
144 if (shm_id_2 == -1)
145 tst_brkm(TBROK | TERRNO, cleanup, "shmget");
146
147 TEST_PAUSE;
148 }
149
cleanup(void)150 void cleanup(void)
151 {
152 rm_shm(shm_id_2);
153
154 set_sys_tune("nr_hugepages", orig_hugepages, 0);
155
156 tst_rmdir();
157 }
158