/*++
/* NAME
/* peekfd 3
/* SUMMARY
/* determine amount of data ready to read
/* SYNOPSIS
/* #include <iostuff.h>
/*
/* ssize_t peekfd(fd)
/* int fd;
/* DESCRIPTION
/* peekfd() attempts to find out how many bytes are available to
/* be read from the named file descriptor. The result value is
/* the number of available bytes.
/* DIAGNOSTICS
/* peekfd() returns -1 in case of trouble. The global \fIerrno\fR
/* variable reflects the nature of the problem.
/* BUGS
/* On some systems, non-blocking read() may fail even after a
/* positive return from peekfd(). The smtp-sink program works
/* around this by using the readable() function instead.
/* LICENSE
/* .ad
/* .fi
/* The Secure Mailer license must be distributed with this software.
/* AUTHOR(S)
/* Wietse Venema
/* IBM T.J. Watson Research
/* P.O. Box 704
/* Yorktown Heights, NY 10598, USA
/*
/* Wietse Venema
/* Google, Inc.
/* 111 8th Avenue
/* New York, NY 10011, USA
/*--*/

/* System library. */

#include <sys_defs.h>
#include <sys/ioctl.h>
#ifdef SUNOS5
#include <sys/socket.h>   /* shutdown(2) */
#endif
#ifdef FIONREAD_IN_SYS_FILIO_H
#include <sys/filio.h>
#endif
#ifdef FIONREAD_IN_TERMIOS_H
#include <termios.h>
#endif
#include <unistd.h>

#ifndef SHUT_RDWR
#define SHUT_RDWR  2
#endif

/* Utility library. */

#include "iostuff.h"

/* peekfd - return amount of data ready to read */

ssize_t peekfd(int fd)
{

    /*
     * Anticipate a series of system-dependent code fragments.
     */
#ifdef FIONREAD
    int     count;

#ifdef SUNOS5

    /*
     * With Solaris10, write_wait() hangs in poll() until timeout, when
     * invoked after peekfd() has received an ECONNRESET error indication.
     * This happens when a client sends QUIT and closes the connection
     * immediately.
     */
    if (ioctl(fd, FIONREAD, (char *) &count) < 0) {
 (void) shutdown(fd, SHUT_RDWR);
 return (-1);
    } else {
 return (count);
    }
#else      /* SUNOS5 */
    return (ioctl(fd, FIONREAD, (char *) &count) < 0 ? -1 : count);
#endif      /* SUNOS5 */
#else
#error "don't know how to look ahead"
#endif
}