C++程序  |  145行  |  3.26 KB

/*
* Copyright (c) International Business Machines Corp., 2007
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
* the GNU General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*
* Author: Serge Hallyn <serue@us.ibm.com>
*
* Create shm with key 0xEAEAEA
* clone, clone(CLONE_NEWIPC), or unshare(CLONE_NEWIPC)
* In cloned process, try to get the created shm

***************************************************************************/

#define _GNU_SOURCE 1
#include <sys/wait.h>
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include "ipcns_helper.h"
#include "test.h"

char *TCID = "sysvipc_namespace";
int TST_TOTAL = 1;
#define TESTKEY 0xEAEAEA

int p1[2];
int p2[2];

int check_shmid(void *vtest)
{
	char buf[3];
	int id;

	(void) vtest;

	close(p1[1]);
	close(p2[0]);

	read(p1[0], buf, 3);
	id = shmget(TESTKEY, 100, 0);
	if (id == -1) {
		write(p2[1], "notfnd", 7);
	} else {
		write(p2[1], "exists", 7);
		shmctl(id, IPC_RMID, NULL);
	}

	tst_exit();
}

static void setup(void)
{
	tst_require_root();
	check_newipc();
}

#define UNSHARESTR "unshare"
#define CLONESTR "clone"
#define NONESTR "none"
int main(int argc, char *argv[])
{
	int r, use_clone = T_NONE;
	int id;
	char *tsttype = NONESTR;
	char buf[7];

	setup();

	if (argc != 2) {
		tst_resm(TFAIL, "Usage: %s <clone|unshare|none>", argv[0]);
		tst_brkm(TFAIL,
			 NULL,
			 " where clone, unshare, or fork specifies unshare method.");
	}
	if (pipe(p1) == -1) {
		perror("pipe");
		exit(EXIT_FAILURE);
	}
	if (pipe(p2) == -1) {
		perror("pipe");
		exit(EXIT_FAILURE);
	}
	tsttype = NONESTR;
	if (strcmp(argv[1], "clone") == 0) {
		use_clone = T_CLONE;
		tsttype = CLONESTR;
	} else if (strcmp(argv[1], "unshare") == 0) {
		use_clone = T_UNSHARE;
		tsttype = UNSHARESTR;
	}

	/* first create the key */
	id = shmget(TESTKEY, 100, IPC_CREAT);
	if (id == -1) {
		perror("shmget");
		tst_brkm(TFAIL, NULL, "shmget failed");
	}

	tst_resm(TINFO, "shmid namespaces test : %s", tsttype);
	/* fire off the test */
	r = do_clone_unshare_test(use_clone, CLONE_NEWIPC, check_shmid, NULL);
	if (r < 0) {
		tst_brkm(TFAIL, NULL, "%s failed", tsttype);
	}

	close(p1[0]);
	close(p2[1]);
	write(p1[1], "go", 3);
	read(p2[0], buf, 7);
	if (strcmp(buf, "exists") == 0) {
		if (use_clone == T_NONE)
			tst_resm(TPASS, "plain cloned process found shmid");
		else
			tst_resm(TFAIL, "%s: child process found shmid",
				 tsttype);
	} else {
		if (use_clone == T_NONE)
			tst_resm(TFAIL,
				 "plain cloned process didn't find shmid");
		else
			tst_resm(TPASS, "%s: child process didn't find shmid",
				 tsttype);
	}

	/* destroy the key */
	shmctl(id, IPC_RMID, NULL);

	tst_exit();
}