freebsd-src/tools/regression/sockets/accept_fd_leak/accept_fd_leak.c
Nik Clayton 00e13b1d67 Switch over to a different, more flexible test output protocol that's
understood by Perl's Test::Harness module and prove(1) commands.

Update README to describe the new protocol.  The work's broken down into
two main sets of changes.

First, update the existing test programs (shell scripts and C programs)
to produce output in the ok/not ok format, and to, where possible, also
produce a header describing the number of tests that are expected to be
run.

Second, provide the .t files that actually run the tests.  In some cases
these are copies of, or very similar too, scripts that already existed.
I've kept the old scripts around so that it's possible to verify that
behaviour under this new system (in terms of whether or not a test fails)
is identical to the behaviour under the old system.

Add a TODO file.
2004-11-11 19:47:55 +00:00

116 lines
3.5 KiB
C

/*-
* Copyright (c) 2004 Robert N. M. Watson
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
* $FreeBSD$
*/
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#define LOOPS 500
/*
* This test is intended to detect a leak of a file descriptor in the process
* following a failed non-blocking accept. It measures an available fd
* baseline, then performs 1000 failing accepts, then checks to see what the
* next fd is. It relies on sequential fd allocation, and will test for it
* briefly before beginning (not 100% reliable, but a good start).
*/
int
main(int argc, char *argv[])
{
struct sockaddr_in sin;
socklen_t size;
int fd1, fd2, fd3, i, s;
printf("1..1\n");
/*
* Check for sequential fd allocation, and give up early if not.
*/
fd1 = dup(STDIN_FILENO);
fd2 = dup(STDIN_FILENO);
if (fd2 != fd1 + 1)
errx(-1, "Non-sequential fd allocation\n");
s = socket(PF_INET, SOCK_STREAM, 0);
if (s == -1)
errx(-1, "socket: %s", strerror(errno));
bzero(&sin, sizeof(sin));
sin.sin_len = sizeof(sin);
sin.sin_family = AF_INET;
sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
sin.sin_port = htons(8080);
if (bind(s, (struct sockaddr *) &sin, sizeof(sin)) != 0)
errx(-1, "bind: %s", strerror(errno));
if (listen(s, -1) != 0)
errx(-1, "listen: %s", strerror(errno));
i = fcntl(s, F_GETFL);
if (i == -1)
errx(-1, "ioctl(F_GETFL): %s", strerror(errno));
i |= O_NONBLOCK;
if (fcntl(s, F_SETFL, i) != 0)
errx(-1, "ioctl(F_SETFL): %s", strerror(errno));
i = fcntl(s, F_GETFL);
if (i == -1)
errx(-1, "ioctl(F_GETFL): %s", strerror(errno));
if ((i & O_NONBLOCK) != O_NONBLOCK)
errx(-1, "Failed to set O_NONBLOCK (i=0x%x)\n", i);
for (i = 0; i < LOOPS; i++) {
size = sizeof(sin);
if (accept(s, (struct sockaddr *)&sin, &size) != -1)
errx(-1, "accept succeeded\n");
if (errno != EAGAIN)
errx(-1, "accept: %s", strerror(errno));
}
/*
* Allocate a file descriptor and make sure it's fd2+2. 2 because
* we allocate an fd for the socket.
*/
fd3 = dup(STDIN_FILENO);
if (fd3 != fd2 + 2)
printf("not ok 1 - (%d, %d, %d)\n", fd1, fd2, fd3);
else
printf("ok 1\n");
return (0);
}