1272343Sngie/* $NetBSD: t_getpid.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2011 The NetBSD Foundation, Inc. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * This code is derived from software contributed to The NetBSD Foundation 8272343Sngie * by Jukka Ruohonen. 9272343Sngie * 10272343Sngie * Redistribution and use in source and binary forms, with or without 11272343Sngie * modification, are permitted provided that the following conditions 12272343Sngie * are met: 13272343Sngie * 1. Redistributions of source code must retain the above copyright 14272343Sngie * notice, this list of conditions and the following disclaimer. 15272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 16272343Sngie * notice, this list of conditions and the following disclaimer in the 17272343Sngie * documentation and/or other materials provided with the distribution. 18272343Sngie * 19272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22272343Sngie * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29272343Sngie * POSSIBILITY OF SUCH DAMAGE. 30272343Sngie */ 31272343Sngie#include <sys/cdefs.h> 32272343Sngie__RCSID("$NetBSD: t_getpid.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $"); 33272343Sngie 34272343Sngie#include <sys/wait.h> 35272343Sngie 36272343Sngie#include <stdlib.h> 37272343Sngie#include <pthread.h> 38272343Sngie#include <unistd.h> 39272343Sngie 40272343Sngie#include <atf-c.h> 41272343Sngie 42272343Sngiestatic int maxiter = 10; 43272343Sngiestatic void *threadfunc(void *); 44272343Sngie 45272343Sngiestatic void * 46272343Sngiethreadfunc(void *arg) 47272343Sngie{ 48272343Sngie *(pid_t *)arg = getpid(); 49272343Sngie 50272343Sngie return NULL; 51272343Sngie} 52272343Sngie 53272343SngieATF_TC(getpid_process); 54272343SngieATF_TC_HEAD(getpid_process, tc) 55272343Sngie{ 56272343Sngie atf_tc_set_md_var(tc, "descr", "Test getpid(2) with processes"); 57272343Sngie} 58272343Sngie 59272343SngieATF_TC_BODY(getpid_process, tc) 60272343Sngie{ 61272343Sngie pid_t ppid, fpid, cpid, tpid, wpid; 62272343Sngie int i, sta; 63272343Sngie 64272343Sngie for (i = 0; i < maxiter; i++) { 65272343Sngie 66272343Sngie tpid = getpid(); 67272343Sngie fpid = fork(); 68272343Sngie 69272343Sngie ATF_REQUIRE(fpid >= 0); 70272343Sngie 71272343Sngie if (fpid == 0) { 72272343Sngie 73272343Sngie cpid = getpid(); 74272343Sngie ppid = getppid(); 75272343Sngie 76272343Sngie if (tpid != ppid) 77272343Sngie _exit(EXIT_FAILURE); 78272343Sngie 79272343Sngie if (cpid == ppid) 80272343Sngie _exit(EXIT_FAILURE); 81272343Sngie 82272343Sngie if (tpid == fpid) 83272343Sngie _exit(EXIT_FAILURE); 84272343Sngie 85272343Sngie _exit(EXIT_SUCCESS); 86272343Sngie } 87272343Sngie 88272343Sngie wpid = wait(&sta); 89272343Sngie 90272343Sngie if (wpid != fpid) 91272343Sngie atf_tc_fail("PID mismatch"); 92272343Sngie 93272343Sngie ATF_REQUIRE(WIFEXITED(sta) != 0); 94272343Sngie 95272343Sngie if (WEXITSTATUS(sta) != EXIT_SUCCESS) 96272343Sngie atf_tc_fail("PID mismatch"); 97272343Sngie } 98272343Sngie} 99272343Sngie 100272343SngieATF_TC(getpid_thread); 101272343SngieATF_TC_HEAD(getpid_thread, tc) 102272343Sngie{ 103272343Sngie atf_tc_set_md_var(tc, "descr", "Test getpid(2) with threads"); 104272343Sngie} 105272343Sngie 106272343SngieATF_TC_BODY(getpid_thread, tc) 107272343Sngie{ 108272343Sngie pid_t pid, tpid; 109272343Sngie pthread_t tid; 110272343Sngie int i, rv; 111272343Sngie 112272343Sngie for (i = 0; i < maxiter; i++) { 113272343Sngie 114272343Sngie pid = getpid(); 115272343Sngie 116272343Sngie rv = pthread_create(&tid, NULL, threadfunc, &tpid); 117272343Sngie ATF_REQUIRE(rv == 0); 118272343Sngie 119272343Sngie rv = pthread_join(tid, NULL); 120272343Sngie ATF_REQUIRE(rv == 0); 121272343Sngie 122272343Sngie if (pid != tpid) 123272343Sngie atf_tc_fail("Unequal PIDs"); 124272343Sngie } 125272343Sngie} 126272343Sngie 127272343SngieATF_TP_ADD_TCS(tp) 128272343Sngie{ 129272343Sngie 130272343Sngie ATF_TP_ADD_TC(tp, getpid_process); 131272343Sngie ATF_TP_ADD_TC(tp, getpid_thread); 132272343Sngie 133272343Sngie return atf_no_error(); 134272343Sngie} 135