src/testzzuf/socat.c

/*
     This file is part of libmicrohttpd
     Copyright (C) 2008,2016 Christian Grothoff

     libmicrohttpd is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published
     by the Free Software Foundation; either version 2, or (at your
     option) any later version.

     libmicrohttpd 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
     General Public License for more details.

     You should have received a copy of the GNU General Public License
     along with libmicrohttpd; see the file COPYING.  If not, write to the
     Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
     Boston, MA 02110-1301, USA.
*/

/**
 * @file socat.c
 * @brief  Code to fork-exec zzuf and start the socat process
 * @author Christian Grothoff
 */

#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>

#ifdef _WIN32
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN 1
#endif /* !WIN32_LEAN_AND_MEAN */
#include <windows.h>
#endif


/**
 * A larger loop count will run more random tests --
 * which would be good, except that it may take too
 * long for most user's patience.  So this small
 * value is the default.
 */
#ifndef _MHD_VHEAVY_TESTS
#define LOOP_COUNT 10
#else  /* ! _MHD_VHEAVY_TESTS */
#define LOOP_COUNT 200
#endif /* ! _MHD_VHEAVY_TESTS */

#define CURL_TIMEOUT 50L

static pid_t zzuf_pid;

static void
zzuf_socat_start ()
{
  int status;
  char *const args[] = {
    "zzuf",
    "--ratio=0.0:0.75",
    "-n",
    "-A",
    "socat",
    "-lf",
    "/dev/null",
    "TCP4-LISTEN:11081,reuseaddr,fork",
    "TCP4:127.0.0.1:11080",
    NULL,
  };
  zzuf_pid = fork ();
  if (zzuf_pid == -1)
  {
    fprintf (stderr, "fork failed: %s\n", strerror (errno));
    exit (1);
  }
  if (zzuf_pid != 0)
  {
    (void) sleep (1);                 /* allow zzuf and socat to start */
    status = 0;
    if (0 < waitpid (zzuf_pid, &status, WNOHANG))
    {
      if (WIFEXITED (status))
        fprintf (stderr,
                 "zzuf died with status code %d!\n",
                 WEXITSTATUS (status));
      if (WIFSIGNALED (status))
        fprintf (stderr,
                 "zzuf died from signal %d!\n", WTERMSIG (status));
      exit (1);
    }
    return;
  }
  setpgid (0, 0);
  execvp ("zzuf", args);
  fprintf (stderr, "execution of `zzuf' failed: %s\n", strerror (errno));
  exit (1);
}


static void
zzuf_socat_stop ()
{
  int status;
  if (zzuf_pid != 0)
  {
    if (0 != killpg (zzuf_pid, SIGINT))
      fprintf (stderr, "Failed to killpg: %s\n", strerror (errno));
    kill (zzuf_pid, SIGINT);
    waitpid (zzuf_pid, &status, 0);
    (void) sleep (1);                 /* allow socat to also die in peace */
  }
}


/* end of socat.c */