Home | History | Annotate | Line # | Download | only in kqueue 
      1 /* $NetBSD: t_proc4.c,v 1.1 2021/10/10 17:43:15 thorpej Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2021 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     26  * POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 #include <sys/cdefs.h>
     30 __RCSID("$NetBSD: t_proc4.c,v 1.1 2021/10/10 17:43:15 thorpej Exp $");
     31 
     32 #include <sys/event.h>
     33 #include <sys/time.h>
     34 #include <sys/types.h>
     35 #include <sys/wait.h>
     36 
     37 #include <err.h>
     38 #include <limits.h>
     39 #include <pwd.h>
     40 #include <stdio.h>
     41 #include <stdlib.h>
     42 #include <unistd.h>
     43 
     44 #include <atf-c.h>
     45 
     46 /* 10 children each create 10 grandchildren. */
     47 #define	TOTAL_CHILDREN		10
     48 #define	TOTAL_GRANDCHILDREN	(TOTAL_CHILDREN * TOTAL_CHILDREN)
     49 #define	TOTAL_DESCENDENTS	(TOTAL_CHILDREN + TOTAL_GRANDCHILDREN)
     50 
     51 #define	EXIT_CHILD		66
     52 #define	EXIT_GRANDCHILD		76
     53 
     54 static void
     55 child(void)
     56 {
     57 	int i, status;
     58 	pid_t pid;
     59 
     60 	for (i = 0; i < TOTAL_CHILDREN; i++) {
     61 		ATF_REQUIRE((pid = fork()) != -1);
     62 		if (pid == 0) {
     63 			/* grandchild */
     64 			_exit(EXIT_GRANDCHILD);
     65 		}
     66 	}
     67 
     68 	for (i = 0; i < TOTAL_CHILDREN; i++) {
     69 		ATF_REQUIRE((pid = wait(&status)) != -1);
     70 	}
     71 	_exit(EXIT_CHILD);
     72 }
     73 
     74 ATF_TC(proc4);
     75 ATF_TC_HEAD(proc4, tc)
     76 {
     77 	atf_tc_set_md_var(tc, "descr", "Exercises EVFILT_PROC + NOTE_TRACK");
     78 }
     79 
     80 ATF_TC_BODY(proc4, tc)
     81 {
     82 	int total_tracks = 0;
     83 	int child_exits = 0;
     84 	int grandchild_exits = 0;
     85 	pid_t pid;
     86 	int kq, status;
     87 	struct kevent event;
     88 	struct timespec ts = { 0, 0 };
     89 	int i, rv;
     90 
     91 	ATF_REQUIRE((kq = kqueue()) >= 0);
     92 
     93 	EV_SET(&event, getpid(), EVFILT_PROC, EV_ADD,
     94 	    NOTE_TRACK | NOTE_EXIT, 0, 0);
     95 
     96 	ATF_REQUIRE(kevent(kq, &event, 1, NULL, 0, NULL) == 0);
     97 
     98 	for (i = 0; i < TOTAL_CHILDREN; i++) {
     99 		ATF_REQUIRE((pid = fork()) != -1);
    100 		if (pid == 0) {
    101 			child();
    102 			/* NOTREACHED */
    103 		}
    104 	}
    105 
    106 	for (;;) {
    107 		if (total_tracks == TOTAL_DESCENDENTS &&
    108 		    child_exits == TOTAL_CHILDREN &&
    109 		    grandchild_exits == TOTAL_GRANDCHILDREN) {
    110 			break;
    111 		}
    112 		memset(&event, 0, sizeof(event));
    113 		rv = kevent(kq, NULL, 0, &event, 1, NULL);
    114 		ATF_REQUIRE(rv != -1);
    115 		ATF_REQUIRE(rv == 1);
    116 		if (event.fflags & NOTE_CHILD) {
    117 			total_tracks++;
    118 			ATF_REQUIRE(total_tracks <= TOTAL_DESCENDENTS);
    119 		} else if (event.fflags & NOTE_EXIT) {
    120 			ATF_REQUIRE(event.flags & EV_EOF);
    121 			ATF_REQUIRE(event.data >= 0);
    122 			ATF_REQUIRE(event.data <= UINT_MAX);
    123 			status = (int)event.data;
    124 			ATF_REQUIRE(WIFEXITED(status));
    125 			ATF_REQUIRE(WEXITSTATUS(status) == EXIT_CHILD ||
    126 				    WEXITSTATUS(status) == EXIT_GRANDCHILD);
    127 			if (WEXITSTATUS(status) == EXIT_CHILD) {
    128 				child_exits++;
    129 				ATF_REQUIRE(child_exits <= TOTAL_CHILDREN);
    130 				ATF_REQUIRE(wait4((pid_t)event.ident,
    131 						  &status,
    132 						  WNOHANG,
    133 						  NULL) != -1);
    134 				ATF_REQUIRE(status == (int)event.data);
    135 			} else {
    136 				grandchild_exits++;
    137 				ATF_REQUIRE(grandchild_exits <=
    138 					    TOTAL_GRANDCHILDREN);
    139 			}
    140 		} else {
    141 			/*
    142 			 * We didn't ask for NOTE_FORK, so we don't
    143 			 * expect to ever see it, even though we are
    144 			 * getting NOTE_CHILD as the result of the
    145 			 * NOTE_TRACK.
    146 			 */
    147 			ATF_REQUIRE((event.fflags & NOTE_FORK) == 0);
    148 
    149 			/*
    150 			 * Bomb out of we are getting a TRACKERR.
    151 			 * XXX This could legitimately happen if
    152 			 * the kernel is low on memory because the
    153 			 * code path involved specifically chooses
    154 			 * not to block when allocating memory.
    155 			 */
    156 			ATF_REQUIRE((event.fflags & NOTE_TRACKERR) == 0);
    157 		}
    158 	}
    159 
    160 	ATF_REQUIRE(kevent(kq, NULL, 0, &event, 1, &ts) == 0);
    161 
    162 	(void)close(kq);
    163 }
    164 
    165 ATF_TP_ADD_TCS(tp)
    166 {
    167 	ATF_TP_ADD_TC(tp, proc4);
    168 
    169 	return atf_no_error();
    170 }
    171