FdUtils.cxx

Go to the documentation of this file.

 
#include "FdUtils.hxx"
 
#ifdef __linux__
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/stat.h>
#include <termios.h> /* tc* functions */
#endif
 
#include "nmranet_config.h"
 
#define PCALL_LOGERR(where, callfn, fd, args...)                               \
    do                                                                         \
    {                                                                          \
        int ret = callfn(fd, args);                                            \
        if (ret < 0)                                                           \
        {                                                                      \
            char buf[256];                                                     \
            strerror_r(errno, buf, sizeof(buf));                               \
            LOG_ERROR("fd %d %s: %s", fd, where, buf);                         \
        }                                                                      \
    } while (0)
 
void FdUtils::optimize_socket_fd(int fd)
{
#ifdef __linux__
    const int rcvbuf = config_gridconnect_tcp_rcv_buffer_size();
    if (rcvbuf > 1)
    {
        PCALL_LOGERR("setsockopt SO_RCVBUF", ::setsockopt, fd, SOL_SOCKET,
            SO_RCVBUF, &rcvbuf, sizeof(rcvbuf));
    }
    const int sndbuf = config_gridconnect_tcp_snd_buffer_size();
    if (sndbuf > 1)
    {
        PCALL_LOGERR("setsockopt SO_SNDBUF", ::setsockopt, fd, SOL_SOCKET,
            SO_SNDBUF, &sndbuf, sizeof(sndbuf));
        int ret = 0;
        socklen_t retsize = sizeof(ret);
        ::getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &ret, &retsize);
        LOG(ALWAYS, "fd %d sndbuf %d", fd, ret);
    }
    const int lowat = config_gridconnect_tcp_notsent_lowat_buffer_size();
    if (lowat > 1)
    {
        PCALL_LOGERR("setsockopt tcp_notsent_lowat", ::setsockopt, fd,
            IPPROTO_TCP, TCP_NOTSENT_LOWAT, &lowat, sizeof(lowat));
        int ret = 0;
        socklen_t retsize = sizeof(ret);
        ::getsockopt(fd, IPPROTO_TCP, TCP_NOTSENT_LOWAT, &ret, &retsize);
        LOG(ALWAYS, "fd %d lowat %d", fd, ret);
    }
#endif
}
 
void FdUtils::optimize_tty_fd(int fd)
{
#ifdef __linux__
    // Sets up the terminal in raw mode. Otherwise linux might echo
    // characters coming in from the device and that will make
    // packets go back to where they came from.
    HASSERT(!tcflush(fd, TCIOFLUSH));
    struct termios settings;
    HASSERT(!tcgetattr(fd, &settings));
    cfmakeraw(&settings);
    cfsetspeed(&settings, B115200);
    HASSERT(!tcsetattr(fd, TCSANOW, &settings));
#endif
}
 
void FdUtils::optimize_fd(int fd)
{
#ifdef __linux__
    struct stat statbuf;
    fstat(fd, &statbuf);
 
    if (S_ISSOCK(statbuf.st_mode))
    {
        optimize_socket_fd(fd);
    }
    else if (isatty(fd))
    {
        optimize_tty_fd(fd);
    }
#endif
}