tests/kernel/kqueue/t_proc1.c

   1 /* $NetBSD: t_proc1.c,v 1.2 2015/01/14 22:22:32 christos Exp $ */
   2
   3 /*-
   4  * Copyright (c) 2002, 2008 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Luke Mewburn and Jaromir Dolecek.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include <sys/cdefs.h>
  33 __COPYRIGHT("@(#) Copyright (c) 2008\
  34  The NetBSD Foundation, inc. All rights reserved.");
  35 __RCSID("$NetBSD: t_proc1.c,v 1.2 2015/01/14 22:22:32 christos Exp $");
  36
  37 /*
  38  * this also used to trigger problem fixed in
  39  * rev. 1.1.1.1.2.13 of sys/kern/kern_event.c
  40  */
  41
  42 #include <sys/param.h>
  43 #include <sys/event.h>
  44 #include <sys/wait.h>
  45
  46 #include <err.h>
  47 #include <stdio.h>
  48 #include <stdlib.h>
  49 #include <unistd.h>
  50 #include <inttypes.h>
  51
  52 #include <atf-c.h>
  53
  54 #include "../../h_macros.h"
  55
  56 static int
  57 child(void)
  58 {
  59         pid_t ch;
  60         int status;
  61         char *argv[] = { NULL, NULL };
  62         char *envp[] = { NULL, NULL };
  63
  64         if ((argv[0] = strdup("true")) == NULL)
  65                 err(EXIT_FAILURE, "strdup(\"true\")");
  66
  67         if ((envp[0] = strdup("FOO=BAZ")) == NULL)
  68                 err(EXIT_FAILURE, "strdup(\"FOO=BAZ\")");
  69
  70         /* Ensure parent is ready */
  71         (void)sleep(2);
  72
  73         /* Do fork */
  74         switch (ch = fork()) {
  75         case -1:
  76                 return EXIT_FAILURE;
  77                 /* NOTREACHED */
  78         case 0:
  79                 return EXIT_SUCCESS;
  80                 /* NOTREACHED */
  81         default:
  82                 wait(&status);
  83                 break;
  84         }
  85
  86         /* Exec */
  87         execve("/usr/bin/true", argv, envp);
  88
  89         /* NOTREACHED */
  90         return EXIT_FAILURE;
  91 }
  92
  93 ATF_TC(proc1);
  94 ATF_TC_HEAD(proc1, tc)
  95 {
  96         atf_tc_set_md_var(tc, "descr", "Checks EVFILT_PROC");
  97 }
  98 ATF_TC_BODY(proc1, tc)
  99 {
 100         struct kevent event[1];
 101         pid_t pid;
 102         int kq, status;
 103         u_int want;
 104
 105         RL(kq = kqueue());
 106
 107         /* fork a child for doing the events */
 108         RL(pid = fork());
 109         if (pid == 0) {
 110                 _exit(child());
 111                 /* NOTREACHED */
 112         }
 113
 114         (void)sleep(1); /* give child some time to come up */
 115
 116         event[0].ident = (uintptr_t)pid;
 117         event[0].filter = EVFILT_PROC;
 118         event[0].flags = EV_ADD | EV_ENABLE;
 119         event[0].fflags = NOTE_EXIT | NOTE_FORK | NOTE_EXEC; /* | NOTE_TRACK;*/
 120         want = NOTE_EXIT | NOTE_FORK | NOTE_EXEC;
 121
 122         RL(kevent(kq, event, 1, NULL, 0, NULL));
 123
 124         /* wait until we get all events we want */
 125         while (want) {
 126                 RL(kevent(kq, NULL, 0, event, 1, NULL));
 127                 printf("%ld:", (long)event[0].ident);
 128
 129                 if (event[0].fflags & NOTE_EXIT) {
 130                         want &= ~NOTE_EXIT;
 131                         printf(" NOTE_EXIT");
 132                 }
 133                 if (event[0].fflags & NOTE_EXEC) {
 134                         want &= ~NOTE_EXEC;
 135                         printf(" NOTE_EXEC");
 136                 }
 137                 if (event[0].fflags & NOTE_FORK) {
 138                         want &= ~NOTE_FORK;
 139                         printf(" NOTE_FORK");
 140                 }
 141                 if (event[0].fflags & NOTE_CHILD)
 142                         printf(" NOTE_CHILD, parent = %" PRId64, event[0].data);
 143
 144                 printf("\n");
 145         }
 146
 147         (void)waitpid(pid, &status, 0);
 148 }
 149
 150 ATF_TP_ADD_TCS(tp)
 151 {
 152         ATF_TP_ADD_TC(tp, proc1);
 153
 154         return atf_no_error();
 155 }