/* Copyright (C) 2002-2018 Free Software Foundation, Inc.

   This file is part of the GNU C Library.

   Contributed by Ulrich Drepper <drepper@redhat.com>, 2002.

   The GNU C Library is free software; you can redistribute it and/or

   modify it under the terms of the GNU Lesser General Public

   License as published by the Free Software Foundation; either

   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public

   License along with the GNU C Library; if not, see

   <http://www.gnu.org/licenses/>.  */

/* NOTE: this tests functionality beyond POSIX.  POSIX does not allow

   exit to be called more than once.  */

#include <stddef.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <sys/un.h>

#include <sys/ipc.h>

#include <sys/msg.h>

#include <unistd.h>

#include <errno.h>

#include <limits.h>

#include <pthread.h>

#include <fcntl.h>

#include <termios.h>

#include <sys/mman.h>

#include <sys/poll.h>

#include <sys/wait.h>

#include <sys/stat.h>

#include <sys/uio.h>

/* Since STREAMS are not supported in the standard Linux kernel and

   there we don't advertise STREAMS as supported is no need to test

   the STREAMS related functions.  This affects

     getmsg()              getpmsg()          putmsg()

     putpmsg()

   lockf() and fcntl() are tested in tst-cancel16.

   pthread_join() is tested in tst-join5.

   pthread_testcancel()'s only purpose is to allow cancellation.  This

   is tested in several places.

   sem_wait() and sem_timedwait() are checked in tst-cancel1[2345] tests.

   mq_send(), mq_timedsend(), mq_receive() and mq_timedreceive() are checked

   in tst-mqueue8{,x} tests.

   aio_suspend() is tested in tst-cancel17.

   clock_nanosleep() is tested in tst-cancel18.

   Linux sendmmsg and recvmmsg are checked in tst-cancel4_1.c and

   tst-cancel4_2.c respectively.

*/

#include "tst-cancel4-common.h"

#ifndef IPC_ADDVAL

# define IPC_ADDVAL 0

#endif

static void *

tf_read  (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  ssize_t s;

  pthread_cleanup_push (cl, NULL);

  char buf[100];

  s = read (fd, buf, sizeof (buf));

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("read returns with %zd", s);

}

static void *

tf_readv  (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  ssize_t s;

  pthread_cleanup_push (cl, NULL);

  char buf[100];

  struct iovec iov[1] = { [0] = { .iov_base = buf, .iov_len = sizeof (buf) } };

  s = readv (fd, iov, 1);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("readv returns with %zd", s);

}

static void *

tf_write  (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[1];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  ssize_t s;

  pthread_cleanup_push (cl, NULL);

  char buf[WRITE_BUFFER_SIZE];

  memset (buf, '\0', sizeof (buf));

  s = write (fd, buf, sizeof (buf));

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("write returns with %zd", s);

}

static void *

tf_writev  (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[1];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  ssize_t s;

  pthread_cleanup_push (cl, NULL);

  char buf[WRITE_BUFFER_SIZE];

  memset (buf, '\0', sizeof (buf));

  struct iovec iov[1] = { [0] = { .iov_base = buf, .iov_len = sizeof (buf) } };

  s = writev (fd, iov, 1);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("writev returns with %zd", s);

}

static void *

tf_sleep (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  sleep (arg == NULL ? 1000000 : 0);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sleep returns");

}

static void *

tf_usleep (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  usleep (arg == NULL ? (useconds_t) ULONG_MAX : 0);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("usleep returns");

}

static void *

tf_nanosleep (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  struct timespec ts = { .tv_sec = arg == NULL ? 10000000 : 0, .tv_nsec = 0 };

  TEMP_FAILURE_RETRY (nanosleep (&ts, &ts);;

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("nanosleep returns");

}

static void *

tf_select (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  fd_set rfs;

  FD_ZERO (&rfs;;

  FD_SET (fd, &rfs;;

  int s;

  pthread_cleanup_push (cl, NULL);

  s = select (fd + 1, &rfs, NULL, NULL, NULL);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("select returns with %d: %m", s);

}

static void *

tf_pselect (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  fd_set rfs;

  FD_ZERO (&rfs;;

  FD_SET (fd, &rfs;;

  int s;

  pthread_cleanup_push (cl, NULL);

  s = pselect (fd + 1, &rfs, NULL, NULL, NULL, NULL);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("pselect returns with %d: %m", s);

}

static void *

tf_poll (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  struct pollfd rfs[1] = { [0] = { .fd = fd, .events = POLLIN } };

  int s;

  pthread_cleanup_push (cl, NULL);

  s = poll (rfs, 1, -1);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("poll returns with %d: %m", s);

}

static void *

tf_ppoll (void *arg)

{

  int fd;

  if (arg == NULL)

    fd = fds[0];

  else

    {

      char fname[] = "/tmp/tst-cancel4-fd-XXXXXX";

      tempfd = fd = mkstemp (fname);

      if (fd == -1)

	FAIL_EXIT1 ("mkstemp failed: %m");

      unlink (fname);

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  struct pollfd rfs[1] = { [0] = { .fd = fd, .events = POLLIN } };

  int s;

  pthread_cleanup_push (cl, NULL);

  s = ppoll (rfs, 1, NULL, NULL);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("ppoll returns with %d: %m", s);

}

static void *

tf_wait (void *arg)

{

  pid_t pid = fork ();

  if (pid == -1)

    FAIL_EXIT1 ("fork: %m");

  if (pid == 0)

    {

      /* Make the program disappear after a while.  */

      if (arg == NULL)

	sleep (10);

      exit (0);

    }

  if (arg != NULL)

    {

      struct timespec  ts = { .tv_sec = 0, .tv_nsec = 100000000 };

      while (nanosleep (&ts, &ts) != 0)

	continue;

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  int s;

  pthread_cleanup_push (cl, NULL);

  s = wait (NULL);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("wait returns with %d: %m", s);

}

static void *

tf_waitpid (void *arg)

{

  pid_t pid = fork ();

  if (pid == -1)

    FAIL_EXIT1 ("fork: %m");

  if (pid == 0)

    {

      /* Make the program disappear after a while.  */

      if (arg == NULL)

	sleep (10);

      exit (0);

    }

  if (arg != NULL)

    {

      struct timespec  ts = { .tv_sec = 0, .tv_nsec = 100000000 };

      while (nanosleep (&ts, &ts) != 0)

	continue;

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  int s;

  pthread_cleanup_push (cl, NULL);

  s = waitpid (-1, NULL, 0);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("waitpid returns with %d: %m", s);

}

static void *

tf_waitid (void *arg)

{

  pid_t pid = fork ();

  if (pid == -1)

    FAIL_EXIT1 ("fork: %m");

  if (pid == 0)

    {

      /* Make the program disappear after a while.  */

      if (arg == NULL)

	sleep (10);

      exit (0);

    }

  if (arg != NULL)

    {

      struct timespec  ts = { .tv_sec = 0, .tv_nsec = 100000000 };

      while (nanosleep (&ts, &ts) != 0)

	continue;

      xpthread_barrier_wait (&b2;;

    }

  xpthread_barrier_wait (&b2;;

  int s;

  pthread_cleanup_push (cl, NULL);

#ifndef WEXITED

# define WEXITED 0

#endif

  siginfo_t si;

  s = waitid (P_PID, pid, &si, WEXITED);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("waitid returns with %d: %m", s);

}

static void *

tf_sigpause (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  sigpause (sigmask (SIGINT));

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sigpause returned");

}

static void *

tf_sigsuspend (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  /* Just for fun block all signals.  */

  sigset_t mask;

  sigfillset (&mask);

  sigsuspend (&mask);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sigsuspend returned");

}

static void *

tf_sigwait (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  /* Block SIGUSR1.  */

  sigset_t mask;

  sigemptyset (&mask);

  sigaddset (&mask, SIGUSR1);

  TEST_VERIFY_EXIT (pthread_sigmask (SIG_BLOCK, &mask, NULL) == 0);

  int sig;

  pthread_cleanup_push (cl, NULL);

  /* Wait for SIGUSR1.  */

  sigwait (&mask, &sig);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sigwait returned with signal %d", sig);

}

static void *

tf_sigwaitinfo (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  /* Block SIGUSR1.  */

  sigset_t mask;

  sigemptyset (&mask);

  sigaddset (&mask, SIGUSR1);

  TEST_VERIFY_EXIT (pthread_sigmask (SIG_BLOCK, &mask, NULL) == 0);

  siginfo_t info;

  pthread_cleanup_push (cl, NULL);

  /* Wait for SIGUSR1.  */

  sigwaitinfo (&mask, &info;;

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sigwaitinfo returned with signal %d", info.si_signo);

}

static void *

tf_sigtimedwait (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  /* Block SIGUSR1.  */

  sigset_t mask;

  sigemptyset (&mask);

  sigaddset (&mask, SIGUSR1);

  TEST_VERIFY_EXIT (pthread_sigmask (SIG_BLOCK, &mask, NULL) == 0);

  /* Wait for SIGUSR1.  */

  siginfo_t info;

  struct timespec ts = { .tv_sec = 60, .tv_nsec = 0 };

  pthread_cleanup_push (cl, NULL);

  sigtimedwait (&mask, &info, &ts);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("sigtimedwait returned with signal %d", info.si_signo);

}

static void *

tf_pause (void *arg)

{

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  pause ();

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("pause returned");

}

static void *

tf_accept (void *arg)

{

  struct sockaddr_un sun;

  /* To test a non-blocking accept call we make the call file by using

     a datagrame socket.  */

  int pf = arg == NULL ? SOCK_STREAM : SOCK_DGRAM;

  tempfd = socket (AF_UNIX, pf, 0);

  if (tempfd == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, %s, 0): %m", arg == NULL ? "SOCK_STREAM"

					       : "SOCK_DGRAM");

  int tries = 0;

  do

    {

      TEST_VERIFY_EXIT (tries++ < 10);

      strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-1-XXXXXX");

      TEST_VERIFY_EXIT (mktemp (sun.sun_path) != NULL);

      sun.sun_family = AF_UNIX;

    }

  while (bind (tempfd, (struct sockaddr *) &sun,

	       offsetof (struct sockaddr_un, sun_path)

	       + strlen (sun.sun_path) + 1) != 0);

  unlink (sun.sun_path);

  listen (tempfd, 5);

  socklen_t len = sizeof (sun);

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  accept (tempfd, (struct sockaddr *) &sun, &len;;

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("accept returned");

}

static void *

tf_send (void *arg)

{

  struct sockaddr_un sun;

  tempfd = socket (AF_UNIX, SOCK_STREAM, 0);

  if (tempfd == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_STREAM, 0): %m");

  int tries = 0;

  do

    {

      TEST_VERIFY_EXIT (tries++ < 10);

      strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-2-XXXXXX");

      TEST_VERIFY_EXIT (mktemp (sun.sun_path) != NULL);

      sun.sun_family = AF_UNIX;

    }

  while (bind (tempfd, (struct sockaddr *) &sun,

	       offsetof (struct sockaddr_un, sun_path)

	       + strlen (sun.sun_path) + 1) != 0);

  listen (tempfd, 5);

  tempfd2 = socket (AF_UNIX, SOCK_STREAM, 0);

  if (tempfd2 == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_STREAM, 0): %m");

  set_socket_buffer (tempfd2);

  if (connect (tempfd2, (struct sockaddr *) &sun, sizeof (sun)) != 0)

    FAIL_EXIT1 ("connect: %m");

  unlink (sun.sun_path);

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  char mem[WRITE_BUFFER_SIZE];

  send (tempfd2, mem, arg == NULL ? sizeof (mem) : 1, 0);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("send returned");

}

static void *

tf_recv (void *arg)

{

  struct sockaddr_un sun;

  tempfd = socket (AF_UNIX, SOCK_STREAM, 0);

  if (tempfd == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_STREAM, 0): %m");

  int tries = 0;

  do

    {

      TEST_VERIFY_EXIT (tries++ < 10);

      strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-3-XXXXXX");

      TEST_VERIFY_EXIT (mktemp (sun.sun_path) != NULL);

      sun.sun_family = AF_UNIX;

    }

  while (bind (tempfd, (struct sockaddr *) &sun,

	       offsetof (struct sockaddr_un, sun_path)

	       + strlen (sun.sun_path) + 1) != 0);

  listen (tempfd, 5);

  tempfd2 = socket (AF_UNIX, SOCK_STREAM, 0);

  if (tempfd2 == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_STREAM, 0): %m");

  if (connect (tempfd2, (struct sockaddr *) &sun, sizeof (sun)) != 0)

    FAIL_EXIT1 ("connect: %m");

  unlink (sun.sun_path);

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  char mem[70];

  recv (tempfd2, mem, arg == NULL ? sizeof (mem) : 0, 0);

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("recv returned");

}

static void *

tf_recvfrom (void *arg)

{

  struct sockaddr_un sun;

  tempfd = socket (AF_UNIX, SOCK_DGRAM, 0);

  if (tempfd == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");

  int tries = 0;

  do

    {

      TEST_VERIFY_EXIT (tries++ < 10);

      strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-4-XXXXXX");

      TEST_VERIFY_EXIT (mktemp (sun.sun_path) != NULL);

      sun.sun_family = AF_UNIX;

    }

  while (bind (tempfd, (struct sockaddr *) &sun,

	       offsetof (struct sockaddr_un, sun_path)

	       + strlen (sun.sun_path) + 1) != 0);

  tempfname = strdup (sun.sun_path);

  tempfd2 = socket (AF_UNIX, SOCK_DGRAM, 0);

  if (tempfd2 == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");

  xpthread_barrier_wait (&b2;;

  if (arg != NULL)

    xpthread_barrier_wait (&b2;;

  pthread_cleanup_push (cl, NULL);

  char mem[70];

  socklen_t len = sizeof (sun);

  recvfrom (tempfd2, mem, arg == NULL ? sizeof (mem) : 0, 0,

	    (struct sockaddr *) &sun, &len;;

  pthread_cleanup_pop (0);

  FAIL_EXIT1 ("recvfrom returned");

}

static void *

tf_recvmsg (void *arg)

{

  struct sockaddr_un sun;

  tempfd = socket (AF_UNIX, SOCK_DGRAM, 0);

  if (tempfd == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");

  int tries = 0;

  do

    {

      TEST_VERIFY_EXIT (tries++ < 10);

      strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-5-XXXXXX");

      TEST_VERIFY_EXIT (mktemp (sun.sun_path) != NULL);

      sun.sun_family = AF_UNIX;

    }

  while (bind (tempfd, (struct sockaddr *) &sun,

	       offsetof (struct sockaddr_un, sun_path)

	       + strlen (sun.sun_path) + 1) != 0);

  tempfname = strdup (sun.sun_path);

  tempfd2 = socket (AF_UNIX, SOCK_DGRAM, 0);

  if (tempfd2 == -1)

    FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");