1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause 3 * 4 * Copyright (c) 2018 Alan Somers 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/cdefs.h> 29 #include <errno.h> 30 #include <fcntl.h> 31 #include <pthread.h> 32 #include <signal.h> 33 #include <sys/socket.h> 34 #include <sys/un.h> 35 36 #include <stdio.h> 37 38 #include <atf-c.h> 39 40 static void 41 do_socketpair(int *sv) 42 { 43 int s; 44 45 s = socketpair(PF_LOCAL, SOCK_STREAM, 0, sv); 46 ATF_REQUIRE_EQ(0, s); 47 ATF_REQUIRE(sv[0] >= 0); 48 ATF_REQUIRE(sv[1] >= 0); 49 ATF_REQUIRE(sv[0] != sv[1]); 50 } 51 52 /* getpeereid(3) should work with stream sockets created via socketpair(2) */ 53 ATF_TC_WITHOUT_HEAD(getpeereid); 54 ATF_TC_BODY(getpeereid, tc) 55 { 56 int sv[2]; 57 uid_t real_euid, euid; 58 gid_t real_egid, egid; 59 60 real_euid = geteuid(); 61 real_egid = getegid(); 62 63 do_socketpair(sv); 64 65 ATF_REQUIRE_EQ(0, getpeereid(sv[0], &euid, &egid)); 66 ATF_CHECK_EQ(real_euid, euid); 67 ATF_CHECK_EQ(real_egid, egid); 68 69 ATF_REQUIRE_EQ(0, getpeereid(sv[1], &euid, &egid)); 70 ATF_CHECK_EQ(real_euid, euid); 71 ATF_CHECK_EQ(real_egid, egid); 72 73 close(sv[0]); 74 close(sv[1]); 75 } 76 77 /* Sending zero bytes should succeed (once regressed in aba79b0f4a3f). */ 78 ATF_TC_WITHOUT_HEAD(send_0); 79 ATF_TC_BODY(send_0, tc) 80 { 81 int sv[2]; 82 83 do_socketpair(sv); 84 ATF_REQUIRE(send(sv[0], sv, 0, 0) == 0); 85 close(sv[0]); 86 close(sv[1]); 87 } 88 89 90 ATF_TP_ADD_TCS(tp) 91 { 92 ATF_TP_ADD_TC(tp, getpeereid); 93 ATF_TP_ADD_TC(tp, send_0); 94 95 return atf_no_error(); 96 } 97