test/stress/stress2/testcases/socket/socket.c

   1 /*-
   2  * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  */
  27
  28 #include <sys/param.h>
  29 #include <sys/socket.h>
  30 #include <netinet/in.h>
  31 #include <netdb.h>
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <unistd.h>
  35 #include <signal.h>
  36 #include <errno.h>
  37 #include <string.h>
  38 #include <err.h>
  39
  40 #include "stress.h"
  41
  42 #define NB (400 * 1024 * 1024)
  43
  44 int port;
  45 int bufsize;
  46 int sv[2];
  47
  48 static void
  49 reader(void) {
  50         int n, t, *buf;
  51
  52         t = 0;
  53         if ((buf = malloc(bufsize)) == NULL)
  54                         err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
  55         while (done_testing == 0) {
  56                 if ((n = read(sv[0], buf, bufsize)) < 0)
  57                         err(1, "read(), %s:%d", __FILE__, __LINE__);
  58                 t += n;
  59                 if (n == 0) break;
  60         }
  61         close(sv[0]);
  62         return;
  63 }
  64
  65 static void
  66 writer(void) {
  67         int i, *buf, r;
  68
  69         if (r < 0)
  70                 err(1, "connect(), %s:%d", __FILE__, __LINE__);
  71
  72         if ((buf = malloc(bufsize)) == NULL)
  73                         err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
  74         for (i = 0; i < bufsize / sizeof(int); i++)
  75                 buf[i] = i;
  76
  77         for (;;) {
  78                 for (i = 0; i < NB; i+= bufsize) {
  79                         if (write(sv[1], buf, bufsize) < 0) {
  80                                 if (errno == EPIPE)
  81                                         return;
  82                                 err(1, "write(%d), %s:%d", sv[1],
  83                                                 __FILE__, __LINE__);
  84                         }
  85                 }
  86         }
  87         return;
  88 }
  89
  90 int
  91 setup(int nb)
  92 {
  93         port = 12340 + nb;
  94         bufsize = 2 << random_int(2, 12);
  95         return (0);
  96 }
  97
  98 void
  99 cleanup(void)
 100 {
 101 }
 102
 103 int
 104 test(void)
 105 {
 106         pid_t pid;
 107
 108         if (socketpair(PF_UNIX, SOCK_STREAM, 0, sv) != 0)
 109                 err(1, "socketpair()");
 110         if ((pid = fork()) == 0) {
 111                 writer();
 112                 exit(EXIT_SUCCESS);
 113
 114         } else if (pid > 0) {
 115                 reader();
 116                 kill(pid, SIGINT);
 117         } else
 118                 err(1, "fork(), %s:%d",  __FILE__, __LINE__);
 119
 120         return (0);
 121 }