tools/regression/netipx/spxabort/spxabort.c

   1 /*-
   2  * Copyright (c) 2006 Robert N. M. Watson
   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  * $FreeBSD$
  27  */
  28
  29 /*
  30  * Exercise the pru_abort() code for SPX by opening an SPX connection to a
  31  * listen socket, then closing the listen socket before accepting.
  32  *
  33  * We would also like to be able to test the other two abort cases, in which
  34  * incomplete connections are aborted due to overflow, and due to close of
  35  * the listen socket, but that requires a packet level test rather than using
  36  * the socket API.
  37  */
  38
  39 #include <sys/types.h>
  40 #include <sys/socket.h>
  41
  42 #include <netipx/ipx.h>
  43
  44 #include <err.h>
  45 #include <errno.h>
  46 #include <fcntl.h>
  47 #include <stdio.h>
  48 #include <string.h>
  49 #include <unistd.h>
  50
  51 #define IPX_ENDPOINT    "0xbebe.1.0x8a13"
  52
  53 int
  54 main(int argc, char *argv[])
  55 {
  56         struct sockaddr_ipx sipx;
  57         int sock_listen, sock;
  58
  59         sock_listen = socket(PF_IPX, SOCK_STREAM, 0);
  60         if (sock_listen < 0)
  61                 err(-1, "sock_listen = socket(PF_IPX, SOCK_STREAM, 0)");
  62
  63         bzero(&sipx, sizeof(sipx));
  64         sipx.sipx_len = sizeof(sipx);
  65         sipx.sipx_family = AF_IPX;
  66         sipx.sipx_addr = ipx_addr(IPX_ENDPOINT);
  67
  68         if (bind(sock_listen, (struct sockaddr *)&sipx, sizeof(sipx)) < 0)
  69                 err(-1, "bind(sock_listen)");
  70
  71         if (listen(sock_listen, -1) < 0)
  72                 err(-1, "listen(sock_listen)");
  73
  74         sock = socket(PF_IPX, SOCK_STREAM, 0);
  75         if (sock < 0)
  76                 err(-1, "sock = socket(PF_IPX, SOCK_STREAM, 0)");
  77
  78         bzero(&sipx, sizeof(sipx));
  79         sipx.sipx_len = sizeof(sipx);
  80         sipx.sipx_family = AF_IPX;
  81         sipx.sipx_addr = ipx_addr(IPX_ENDPOINT);
  82
  83         if (fcntl(sock, F_SETFL, O_NONBLOCK) < 0)
  84                 err(-1, "fcntl(sock, F_SETFL, O_NONBLOCKING)");
  85
  86         if (connect(sock, (struct sockaddr *)&sipx, sizeof(sipx)) < 0) {
  87                 if (errno != EINPROGRESS)
  88                         err(-1, "sock = socket(PF_IPX, SOCK_STREAM, 0)");
  89         }
  90
  91         sleep(1);       /* Arbitrary. */
  92
  93         close(sock_listen);
  94
  95         return (0);
  96 };