/*
 *
 *   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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

/* 03/21/2003   enable ia64     Jacky.Malcles */
/* 12/20/2002   Port to LTP     robbiew@us.ibm.com */
/* 06/30/2001   Port to Linux   nsharoff@us.ibm.com */

/*
 * NAME
 *		 shmt04
 *
 * 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 and 
 * verifies that its contents are the same as the contents of the
 * parent attached segment.
 *
 */


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

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


char *TCID="shmt04";            /* Test program identifier.    */
int TST_TOTAL=2;                /* Total number of test cases. */
extern int Tst_count;           /* Test Case counter for tst_* routines */
/**************/


key_t key;
sigset_t mysigset;

#define  ADDR1  (void *)0x40000000
#define  ADDR  (void *)0x80000
#define  SIZE  16*1024


int child(void);
int rm_shm(int);

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

 key = (key_t) getpid() ;

 signal(SIGUSR1, SIG_DFL);

 sigemptyset(&mysigset);
 sigaddset(&mysigset,SIGUSR1);
 sigprocmask(SIG_BLOCK,&mysigset,NULL);
 
 pid = fork();
 switch (pid) {
 case -1:
  tst_resm(TBROK,"fork failed");
  tst_exit() ;
 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 {
#ifdef __ia64__
  cp = (char *) shmat(shmid, ADDR1, 0);
#elif defined(__ARM_ARCH_4T__)
  cp = (char *) shmat(shmid, NULL, 0);
#else
  cp = (char *) shmat(shmid, ADDR, 0);
#endif

 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() ;

/*----------------------------------------------------------*/
return(0);
}

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

sigemptyset(&mysigset);
sigsuspend(&mysigset);
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 
{
#ifdef __ia64__
  cp = (char *) shmat(shmid, ADDR1, 0);
#elif defined(__ARM_ARCH_4T__)
  cp = (char *) shmat(shmid, NULL, 0);
#else
  cp = (char *) shmat(shmid, ADDR, 0);
#endif
 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() ;
return(0);
}

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