tests/pipe_drain.c

   1 /*
   2  * Copyright (c) 2019 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28
  29 #include <darwintest.h>
  30 #include <darwintest_multiprocess.h>
  31 #include <stdio.h>
  32 #include <unistd.h>
  33 #include <sys/wait.h>
  34 #include <pthread.h>
  35 #include <stdlib.h>
  36 #include <signal.h>
  37
  38 static void
  39 signal_handler(int sig, siginfo_t *sip __unused, void *ucontext __unused)
  40 {
  41         if (sig == SIGPIPE) {
  42                 T_FAIL("Received SIGPIPE");
  43         }
  44
  45         exit(141);
  46 }
  47
  48 static void *
  49 thread_read(void *arg)
  50 {
  51         int fd = (int) (uintptr_t)arg;
  52         char buf[10];
  53
  54         read(fd, buf, 10);
  55         T_LOG("thread returned from read");
  56         return 0;
  57 }
  58
  59 T_DECL(pipe_drain,
  60     "test a pipe with multiple read descriptor could close one descriptor and drain that descriptor")
  61 {
  62         int pipe_fd[2];
  63         int dup_fd;
  64         int ret;
  65         char buf[10] = "Hello";
  66         pthread_t thread;
  67
  68         /* Install the signal handler for SIGPIPE */
  69
  70         struct sigaction sa = {
  71                 .sa_sigaction = signal_handler,
  72                 .sa_flags = SA_SIGINFO
  73         };
  74         sigfillset(&sa.sa_mask);
  75
  76         T_QUIET; T_ASSERT_POSIX_ZERO(sigaction(SIGPIPE, &sa, NULL), NULL);
  77
  78         ret = pipe(pipe_fd);
  79         T_EXPECT_EQ(ret, 0, NULL);
  80
  81         dup_fd = dup(pipe_fd[0]);
  82         T_EXPECT_GE(dup_fd, 0, NULL);
  83
  84         pthread_create(&thread, NULL, thread_read, (void *) (uintptr_t) pipe_fd[0]);
  85
  86         sleep(5);
  87
  88         close(pipe_fd[0]);
  89         ret = (int)write(pipe_fd[1], buf, strlen(buf) + 1);
  90         T_EXPECT_EQ(ret, (int)strlen(buf) + 1, NULL);
  91 }