xref: /aosp_15_r20/external/ltp/testcases/kernel/containers/sysvipc/shmnstest.c (revision 49cdfc7efb34551c7342be41a7384b9c40d7cab7)
1*49cdfc7eSAndroid Build Coastguard Worker // SPDX-License-Identifier: GPL-2.0-or-later
2*49cdfc7eSAndroid Build Coastguard Worker /*
3*49cdfc7eSAndroid Build Coastguard Worker  * Copyright (c) International Business Machines Corp., 2007
4*49cdfc7eSAndroid Build Coastguard Worker  *				Serge Hallyn <[email protected]>
5*49cdfc7eSAndroid Build Coastguard Worker  * Copyright (C) 2022 SUSE LLC Andrea Cervesato <[email protected]>
6*49cdfc7eSAndroid Build Coastguard Worker  */
7*49cdfc7eSAndroid Build Coastguard Worker 
8*49cdfc7eSAndroid Build Coastguard Worker /*\
9*49cdfc7eSAndroid Build Coastguard Worker  * [Description]
10*49cdfc7eSAndroid Build Coastguard Worker  *
11*49cdfc7eSAndroid Build Coastguard Worker  * Test if SysV IPC shared memory with a specific key is shared between
12*49cdfc7eSAndroid Build Coastguard Worker  * processes and namespaces.
13*49cdfc7eSAndroid Build Coastguard Worker  */
14*49cdfc7eSAndroid Build Coastguard Worker 
15*49cdfc7eSAndroid Build Coastguard Worker #define _GNU_SOURCE
16*49cdfc7eSAndroid Build Coastguard Worker 
17*49cdfc7eSAndroid Build Coastguard Worker #include <sys/wait.h>
18*49cdfc7eSAndroid Build Coastguard Worker #include <sys/msg.h>
19*49cdfc7eSAndroid Build Coastguard Worker #include <sys/types.h>
20*49cdfc7eSAndroid Build Coastguard Worker #include "tst_safe_sysv_ipc.h"
21*49cdfc7eSAndroid Build Coastguard Worker #include "tst_test.h"
22*49cdfc7eSAndroid Build Coastguard Worker #include "common.h"
23*49cdfc7eSAndroid Build Coastguard Worker 
24*49cdfc7eSAndroid Build Coastguard Worker #define TESTKEY 0xEAEAEA
25*49cdfc7eSAndroid Build Coastguard Worker 
26*49cdfc7eSAndroid Build Coastguard Worker static char *str_op;
27*49cdfc7eSAndroid Build Coastguard Worker static int use_clone;
28*49cdfc7eSAndroid Build Coastguard Worker static int ipc_id = -1;
29*49cdfc7eSAndroid Build Coastguard Worker 
check_shmid(void)30*49cdfc7eSAndroid Build Coastguard Worker static void check_shmid(void)
31*49cdfc7eSAndroid Build Coastguard Worker {
32*49cdfc7eSAndroid Build Coastguard Worker 	TEST(shmget(TESTKEY, 100, 0));
33*49cdfc7eSAndroid Build Coastguard Worker 	if (TST_RET < 0) {
34*49cdfc7eSAndroid Build Coastguard Worker 		if (use_clone == T_NONE)
35*49cdfc7eSAndroid Build Coastguard Worker 			tst_res(TFAIL, "plain cloned process didn't find shmid");
36*49cdfc7eSAndroid Build Coastguard Worker 		else
37*49cdfc7eSAndroid Build Coastguard Worker 			tst_res(TPASS, "%s: child process didn't find shmid", str_op);
38*49cdfc7eSAndroid Build Coastguard Worker 	} else {
39*49cdfc7eSAndroid Build Coastguard Worker 		if (use_clone == T_NONE)
40*49cdfc7eSAndroid Build Coastguard Worker 			tst_res(TPASS, "plain cloned process found shmid");
41*49cdfc7eSAndroid Build Coastguard Worker 		else
42*49cdfc7eSAndroid Build Coastguard Worker 			tst_res(TFAIL, "%s: child process found shmid", str_op);
43*49cdfc7eSAndroid Build Coastguard Worker 	}
44*49cdfc7eSAndroid Build Coastguard Worker }
45*49cdfc7eSAndroid Build Coastguard Worker 
run(void)46*49cdfc7eSAndroid Build Coastguard Worker static void run(void)
47*49cdfc7eSAndroid Build Coastguard Worker {
48*49cdfc7eSAndroid Build Coastguard Worker 	clone_unshare_test(use_clone, CLONE_NEWIPC, check_shmid);
49*49cdfc7eSAndroid Build Coastguard Worker }
50*49cdfc7eSAndroid Build Coastguard Worker 
setup(void)51*49cdfc7eSAndroid Build Coastguard Worker static void setup(void)
52*49cdfc7eSAndroid Build Coastguard Worker {
53*49cdfc7eSAndroid Build Coastguard Worker 	use_clone = get_clone_unshare_enum(str_op);
54*49cdfc7eSAndroid Build Coastguard Worker 	ipc_id = shmget(TESTKEY, 100, IPC_CREAT);
55*49cdfc7eSAndroid Build Coastguard Worker }
56*49cdfc7eSAndroid Build Coastguard Worker 
cleanup(void)57*49cdfc7eSAndroid Build Coastguard Worker static void cleanup(void)
58*49cdfc7eSAndroid Build Coastguard Worker {
59*49cdfc7eSAndroid Build Coastguard Worker 	if (ipc_id != -1) {
60*49cdfc7eSAndroid Build Coastguard Worker 		tst_res(TINFO, "Destroying shared memory");
61*49cdfc7eSAndroid Build Coastguard Worker 		SAFE_SHMCTL(ipc_id, IPC_RMID, NULL);
62*49cdfc7eSAndroid Build Coastguard Worker 	}
63*49cdfc7eSAndroid Build Coastguard Worker }
64*49cdfc7eSAndroid Build Coastguard Worker 
65*49cdfc7eSAndroid Build Coastguard Worker static struct tst_test test = {
66*49cdfc7eSAndroid Build Coastguard Worker 	.test_all = run,
67*49cdfc7eSAndroid Build Coastguard Worker 	.setup = setup,
68*49cdfc7eSAndroid Build Coastguard Worker 	.cleanup = cleanup,
69*49cdfc7eSAndroid Build Coastguard Worker 	.forks_child = 1,
70*49cdfc7eSAndroid Build Coastguard Worker 	.needs_root = 1,
71*49cdfc7eSAndroid Build Coastguard Worker 	.needs_checkpoints = 1,
72*49cdfc7eSAndroid Build Coastguard Worker 	.options = (struct tst_option[]) {
73*49cdfc7eSAndroid Build Coastguard Worker 		{ "m:", &str_op, "Test execution mode <clone|unshare|none>" },
74*49cdfc7eSAndroid Build Coastguard Worker 		{},
75*49cdfc7eSAndroid Build Coastguard Worker 	},
76*49cdfc7eSAndroid Build Coastguard Worker };
77