C++程序  |  221行  |  4.84 KB

/*
 *
 *   Copyright (c) International Business Machines  Corp., 2002
 *
 *   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
 */

/* 12/20/2002   Port to LTP     robbiew@us.ibm.com */
/* 06/30/2001   Port to Linux   nsharoff@us.ibm.com */

/*
 * NAME
 *	shmt06
 *
 * CALLS
 *	shmctl(2) shmget(2) shmat(2)
 *
 * ALGORITHM
 * Parent process forks a child. Child pauses until parent has created
 * a shared memory segment, attached to it and written to it too. At that
 * time child gets the shared memory segment id, attaches to it at two
 * different addresses than the parents and verifies that their contents
 * are the same as the contents of the parent attached segment.
 *
 */

#include <stdio.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/utsname.h>
#include <signal.h>
#include <errno.h>
#include <stdlib.h>
#include <unistd.h>

#define		SIZE		16*1024

/** LTP Port **/
#include "test.h"

char *TCID = "shmt06";		/* Test program identifier.    */
int TST_TOTAL = 2;		/* Total number of test cases. */
/**************/

key_t key;
sigset_t set;

int child();
static int rm_shm(int);

int main(void)
{
	char *cp = NULL;
	int pid, pid1, shmid;
	int status;

	key = (key_t) getpid();

	sigemptyset(&set);
	sigaddset(&set, SIGUSR1);
	sigprocmask(SIG_BLOCK, &set, NULL);

	pid = fork();
	switch (pid) {
	case -1:
		tst_brkm(TBROK, NULL, "fork failed");
	case 0:
		child();
	}

/*------------------------------------------------------*/

	if ((shmid = shmget(key, SIZE, IPC_CREAT | 0666)) < 0) {
		perror("shmget");
		tst_resm(TFAIL, "Error: shmget: shmid = %d, errno = %d\n",
			 shmid, errno);
		/*
		 * kill the child if parent failed to do the attach
		 */
		(void)kill(pid, SIGINT);
	} else {
		cp = shmat(shmid, NULL, 0);

		if (cp == (char *)-1) {
			perror("shmat");
			tst_resm(TFAIL,
				 "Error: shmat: shmid = %d, errno = %d\n",
				 shmid, errno);

			/* kill the child if parent failed to do the attch */

			kill(pid, SIGINT);

			/* remove shared memory segment */

			rm_shm(shmid);

			tst_exit();
		}
		*cp = 'A';
		*(cp + 1) = 'B';
		*(cp + 2) = 'C';

		kill(pid, SIGUSR1);
		while ((pid1 = wait(&status)) < 0 && (errno == EINTR)) ;
		if (pid1 != pid) {
			tst_resm(TFAIL, "Waited on the wrong child");
			tst_resm(TFAIL,
				 "Error: wait_status = %d, pid1= %d\n", status,
				 pid1);
		}
	}

	tst_resm(TPASS, "shmget,shmat");

/*---------------------------------------------------------------*/

	if (shmdt(cp) < 0) {
		tst_resm(TFAIL, "shmdt");
	}

	tst_resm(TPASS, "shmdt");

/*-------------------------------------------------------------*/

	rm_shm(shmid);
	tst_exit();
}

int child(void)
{
	int shmid, chld_pid;
	char *cp;
	int sig;

	sigwait(&set, &sig);
	chld_pid = getpid();

	if ((shmid = shmget(key, SIZE, 0)) < 0) {
		perror("shmget:child process");
		tst_resm(TFAIL,
			 "Error: shmget: errno=%d, shmid=%d, child_pid=%d\n",
			 errno, shmid, chld_pid);
	} else {
		cp = shmat(shmid, NULL, 0);

		if (cp == (char *)-1) {
			perror("shmat:child process");
			tst_resm(TFAIL,
				 "Error: shmat: errno=%d, shmid=%d, child_pid=%d\n",
				 errno, shmid, chld_pid);
		} else {
			if (*cp != 'A') {
				tst_resm(TFAIL, "child: not A\n");
			}
			if (*(cp + 1) != 'B') {
				tst_resm(TFAIL, "child: not B\n");
			}
			if (*(cp + 2) != 'C') {
				tst_resm(TFAIL, "child: not C\n");
			}
			if (*(cp + 8192) != 0) {
				tst_resm(TFAIL, "child: not 0\n");
			}
		}

		/*
		 * Attach the segment to a different addresse
		 * and verify it's contents again.
		 */
		cp = shmat(shmid, NULL, 0);

		if (cp == (char *)-1) {
			perror("shmat:child process");
			tst_resm(TFAIL,
				 "Error: shmat: errno=%d, shmid=%d, child_pid=%d\n",
				 errno, shmid, chld_pid);
		} else {
			if (*cp != 'A') {
				tst_resm(TFAIL, "child: not A\n");
			}
			if (*(cp + 1) != 'B') {
				tst_resm(TFAIL, "child: not B\n");
			}
			if (*(cp + 2) != 'C') {
				tst_resm(TFAIL, "child: not C\n");
			}
			if (*(cp + 8192) != 0) {
				tst_resm(TFAIL, "child: not 0\n");
			}
		}
	}
	tst_exit();
}

static int rm_shm(int shmid)
{
	if (shmctl(shmid, IPC_RMID, NULL) == -1) {
		perror("shmctl");
		tst_brkm(TFAIL,
			 NULL,
			 "shmctl Failed to remove: shmid = %d, errno = %d\n",
			 shmid, errno);
	}
	return (0);
}