Home | History | Annotate | Line # | Download | only in sys
t_kevent.c revision 1.4 
      1 /*	$NetBSD: t_kevent.c,v 1.4 2012/06/02 16:52:18 martin Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundatiom
      8  * by Christos Zoulas.
      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 #include <sys/cdefs.h>
     32 __RCSID("$NetBSD: t_kevent.c,v 1.4 2012/06/02 16:52:18 martin Exp $");
     33 
     34 #include <sys/types.h>
     35 #include <sys/event.h>
     36 
     37 #include <atf-c.h>
     38 #include <errno.h>
     39 #include <time.h>
     40 #include <stdio.h>
     41 #include <stdlib.h>
     42 #include <string.h>
     43 #include <unistd.h>
     44 #include <err.h>
     45 #include <sys/event.h>
     46 #include <sys/time.h>
     47 #include <sys/socket.h>
     48 #include <sys/wait.h>
     49 
     50 ATF_TC(kevent_zerotimer);
     51 ATF_TC_HEAD(kevent_zerotimer, tc)
     52 {
     53 	atf_tc_set_md_var(tc, "descr", "Checks that kevent with a 0 timer "
     54 	    "does not crash the system (PR lib/45618)");
     55 }
     56 
     57 ATF_TC_BODY(kevent_zerotimer, tc)
     58 {
     59 	struct kevent ev;
     60 	int kq;
     61 
     62 	ATF_REQUIRE((kq = kqueue()) != -1);
     63 	EV_SET(&ev, 1, EVFILT_TIMER, EV_ADD|EV_ENABLE, 0, 1, 0);
     64 	ATF_REQUIRE(kevent(kq, &ev, 1, NULL, 0, NULL) != -1);
     65 	ATF_REQUIRE(kevent(kq, NULL, 0, &ev, 1, NULL) == 1);
     66 }
     67 
     68 ATF_TC(kqueue_desc_passing);
     69 ATF_TC_HEAD(kqueue_desc_passing, tc)
     70 {
     71 	atf_tc_set_md_var(tc, "descr", "Checks that passing a kqueue to "
     72 		"another process does not crash the kernel (PR 46463)");
     73 }
     74 
     75 ATF_TC_BODY(kqueue_desc_passing, tc)
     76 {
     77 	pid_t child;
     78 	int s[2], storage, status, kq;
     79 	struct cmsghdr *msg;
     80 	struct iovec iov;
     81 	struct msghdr m;
     82 	struct kevent ev;
     83 
     84 	ATF_REQUIRE((kq = kqueue()) != -1);
     85 
     86 	// atf_tc_skip("crashes kernel (PR 46463)");
     87 
     88 	ATF_REQUIRE(socketpair(AF_LOCAL, SOCK_STREAM, 0, s) != -1);
     89 	msg = malloc(CMSG_SPACE(sizeof(int)));
     90 	m.msg_iov = &iov;
     91 	m.msg_iovlen = 1;
     92 	m.msg_name = NULL;
     93 	m.msg_namelen = 0;
     94 	m.msg_control = msg;
     95 	m.msg_controllen = CMSG_SPACE(sizeof(int));
     96 
     97 	child = fork();
     98 	if (child == 0) {
     99 		close(s[0]);
    100 
    101 		iov.iov_base = &storage;
    102 		iov.iov_len = sizeof(int);
    103 		m.msg_iov = &iov;
    104 		m.msg_iovlen = 1;
    105 
    106 		if (recvmsg(s[1], &m, 0) == -1)
    107 			err(1, "child: could not recvmsg");
    108 
    109 		kq = *(int *)CMSG_DATA(msg);
    110 		printf("child (pid %d): received kq fd %d\n", getpid(), kq);
    111 		exit(0);
    112 	}
    113 
    114 	close(s[1]);
    115 
    116 	iov.iov_base = &storage;
    117 	iov.iov_len = sizeof(int);
    118 
    119 	msg->cmsg_level = SOL_SOCKET;
    120 	msg->cmsg_type = SCM_RIGHTS;
    121 	msg->cmsg_len = CMSG_LEN(sizeof(int));
    122 
    123 	*(int *)CMSG_DATA(msg) = kq;
    124 
    125 	EV_SET(&ev, 1, EVFILT_TIMER, EV_ADD|EV_ENABLE, 0, 1, 0);
    126 	ATF_CHECK(kevent(kq, &ev, 1, NULL, 0, NULL) != -1);
    127 
    128 	printf("parent (pid %d): sending kq fd %d\n", getpid(), kq);
    129 	if (sendmsg(s[0], &m, 0) == -1) {
    130 		ATF_REQUIRE_EQ_MSG(errno, EBADF, "errno is %d", errno);
    131 		atf_tc_skip("PR kern/46523");
    132 	}
    133 
    134 	close(kq);
    135 
    136 	waitpid(child, &status, 0);
    137 	ATF_CHECK(WIFEXITED(status) && WEXITSTATUS(status)==0);
    138 }
    139 
    140 ATF_TP_ADD_TCS(tp)
    141 {
    142 
    143 	ATF_TP_ADD_TC(tp, kevent_zerotimer);
    144 	ATF_TP_ADD_TC(tp, kqueue_desc_passing);
    145 
    146 	return atf_no_error();
    147 }
    148