| /* |
| * |
| * 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" |
| #include "usctest.h" |
| |
| char *TCID = "shmt06"; /* Test program identifier. */ |
| int TST_TOTAL = 2; /* Total number of test cases. */ |
| /**************/ |
| |
| key_t key; |
| sigset_t sigset; |
| |
| int child(); |
| int rm_shm(int); |
| |
| int main() |
| { |
| char *cp = NULL; |
| int pid, pid1, shmid; |
| int status; |
| |
| key = (key_t) getpid(); |
| |
| sigemptyset(&sigset); |
| sigaddset(&sigset, SIGUSR1); |
| sigprocmask(SIG_BLOCK, &sigset, 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() |
| { |
| int shmid, chld_pid; |
| char *cp; |
| int sig; |
| |
| sigwait(&sigset, &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(); |
| } |
| |
| int rm_shm(shmid) |
| 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); |
| } |