newio/src/s_bsd.c

/*
 *  ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd).
 *
 *  Copyright (C) 1997-2013 ircd-hybrid development team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 *  USA
 */

/*! \file s_bsd.c
 * \brief Network functions.
 * \version $Id$
 */

#include "stdinc.h"
#include <netinet/in_systm.h>
#include <netinet/ip.h>
#include <netinet/tcp.h>
#include <sys/uio.h>
#include "ircd_defs.h"
#include "list.h"
#include "s_bsd.h"
#include "parse.h"
#include "client.h"
#include "dbuf.h"
#include "irc_string.h"
#include "ircd.h"
#include "listener.h"
#include "numeric.h"
#include "packet.h"
#include "irc_res.h"
#include "restart.h"
#include "s_auth.h"
#include "conf.h"
#include "log.h"
#include "s_serv.h"
#include "send.h"
#include "memory.h"
#include "s_user.h"
#include "msgq.h"
#include "ioengine.h"
#include "dbuf.h"
#include "hash.h"
#include "ssl.h"


#ifndef IOV_MAX
#define IOV_MAX 16  /**< Minimum required length of an iovec array */
#endif


dlink_list connection_list;

static void client_sock_callback(struct Event *);
static void client_timer_callback(struct Event *);

/** Temporary buffer for reading data from a peer. */
static char readbuf[SERVER_TCP_WINDOW];

/*
 * report_error text constants
 */
const char *const ACCEPT_ERROR_MSG    = "error accepting connection for %s: %s";
const char *const BIND_ERROR_MSG      = "bind error for %s: %s";
const char *const CONNECT_ERROR_MSG   = "connect to host %s failed: %s";
const char *const CONNLIMIT_ERROR_MSG = "connect limit exceeded for %s: %s";
const char *const LISTEN_ERROR_MSG    = "listen error for %s: %s";
const char *const NONB_ERROR_MSG      = "error setting non-blocking for %s: %s";
const char *const PEERNAME_ERROR_MSG  = "getpeername failed for %s: %s";
const char *const POLL_ERROR_MSG      = "poll error for %s: %s";
const char *const REGISTER_ERROR_MSG  = "registering %s: %s";
const char *const REUSEADDR_ERROR_MSG = "error setting SO_REUSEADDR for %s: %s";
const char *const SELECT_ERROR_MSG    = "select error for %s: %s";
const char *const SETBUFS_ERROR_MSG   = "error setting buffer size for %s: %s";
const char *const SOCKET_ERROR_MSG    = "error creating socket for %s: %s";
const char *const TOS_ERROR_MSG       = "error setting TOS for %s: %s";


static int
is_blocked(int error)
{
  return EWOULDBLOCK == error
#ifdef ENOMEM
    || ENOMEM == error
#endif
#ifdef ENOBUFS
    || ENOBUFS == error
#endif
    || EAGAIN == error;
}

/* get_sockerr - get the error value from the socket or the current errno
 *
 * Get the *real* error from the socket (well try to anyway..).
 * This may only work when SO_DEBUG is enabled but its worth the
 * gamble anyway.
 */
int
os_get_sockerr(int fd)
{
  int errtmp = errno;
#ifdef SO_ERROR
  int err = 0;
  socklen_t len = sizeof(err);

  if (-1 < fd && !getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &len))
  {
    if (err)
      errtmp = err;
  }
  errno = errtmp;
#endif
  return errtmp;
}

/** Set a file descriptor to non-blocking mode.
 * @param[in] fd %Socket file descriptor.
 * @return Non-zero on success, or zero on failure.
 */
int
os_set_nonblocking(int fd)
{
  int res = 0;

  if ((res = fcntl(fd, F_GETFL, 0)) == -1)
    return 0;
  if (fcntl(fd, F_SETFL, res | O_NONBLOCK) == -1)
    return 0;
  return 1;
}

/*
 * setup_socket()
 *
 * Set the socket non-blocking, and other wonderful bits.
 */
static void
setup_socket(int fd)
{
  int opt = 1;

  setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt));

#ifdef IPTOS_LOWDELAY
  opt = IPTOS_LOWDELAY;
  setsockopt(fd, IPPROTO_IP, IP_TOS, &opt, sizeof(opt));
#endif
}

/** Mark a socket's address as reusable.
 * @param[in] fd %Socket file descriptor to manipulate.
 * @return Non-zero on success, or zero on failure.
 */
int
os_set_reuseaddr(int fd)
{
  unsigned int opt = 1;

  return !setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
}

/** Set a socket's send and receive buffer sizes.
 * @param[in] fd %Socket file descriptor to manipulate.
 * @param[in] ssize New send buffer size.
 * @param[in] rsize New receive buffer size.
 * @return Non-zero on success, or zero on failure.
 */
int
os_set_sockbufs(int fd, unsigned int ssize, unsigned int rsize)
{
  unsigned int sopt = ssize;
  unsigned int ropt = rsize;

  return !setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &ropt, sizeof(ropt)) &&
         !setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &sopt, sizeof(sopt));
}

/** Set a socket's "type of service" value.
 * @param[in] fd %Socket file descriptor to manipulate.
 * @param[in] tos New type of service value to use.
 * @return Non-zero on success, or zero on failure.
 */
int
os_set_tos(int fd, int tos)
{
#if defined(IP_TOS) && defined(IPPROTO_IP)
  unsigned int opt = tos;

  return !setsockopt(fd, IPPROTO_IP, IP_TOS, &opt, sizeof(opt));
#else
  return 1;
#endif
}

/** Disable IP options on a socket.
 * @param[in] fd %Socket file descriptor to manipulate.
 * @return Non-zero on success, or zero on failure.
 */
int
os_disable_options(int fd)
{
#if defined(IP_OPTIONS) && defined(IPPROTO_IP)
  return !setsockopt(fd, IPPROTO_IP, IP_OPTIONS, NULL, 0);
#else
  return 1;
#endif
}

/** Start listening on a socket.
 * @param[in] fd Disconnected file descriptor.
 * @param[in] backlog Maximum number of un-accept()ed connections to keep.
 * @return Non-zero on success; zero on error.
 */
int
os_set_listen(int fd, int backlog)
{
  return !listen(fd, backlog);
}

/** Attempt to read from a non-blocking socket.
 * @param[in] fd File descriptor to read from.
 * @param[out] buf Output buffer to read into.
 * @param[in] length Number of bytes to read.
 * @param[out] count_out Receives number of bytes actually read.
 * @return An IOResult value indicating status.
 */
IOResult
os_recv_nonb(int fd, char *buf, unsigned int length, unsigned int *count_out)
{
  int res = 0;

  assert(buf);
  assert(count_out);

  if ((res = recv(fd, buf, length, 0)) > 0)
  {
    *count_out = (unsigned int)res;
    return IO_SUCCESS;
  }

  if (res == 0)
  {
    *count_out = 0;
    errno = 0;  /* Or ECONNRESET? */
    return IO_FAILURE;
  }

  *count_out = 0;
  return is_blocked(errno) ? IO_BLOCKED : IO_FAILURE;
}

/** Attempt to read from a non-blocking UDP socket.
 * @param[in] fd File descriptor to read from.
 * @param[out] buf Output buffer to read into.
 * @param[in] length Number of bytes to read.
 * @param[out] length_out Receives number of bytes actually read.
 * @param[out] addr_out Peer address that sent the message.
 * @return An IOResult value indicating status.
 */
IOResult
os_recvfrom_nonb(int fd, char *buf, unsigned int length,
                 unsigned int *length_out,
                 struct irc_ssaddr *addr)
{
  unsigned int len = sizeof(struct irc_ssaddr);
  int res = 0;

  assert(buf);
  assert(length_out);
  assert(addr);

  res = recvfrom(fd, buf, length, 0, (struct sockaddr *)addr, &len);

  if (res > -1)
  {
    *length_out = (unsigned int)res;
    return IO_SUCCESS;
  }

  *length_out = 0;
  return is_blocked(errno) ? IO_BLOCKED : IO_FAILURE;
}

/** Attempt to write on a connected socket.
 * @param[in] fd File descriptor to write to.
 * @param[in] buf Output buffer to send from.
 * @param[in] length Number of bytes to write.
 * @param[out] count_out Receives number of bytes actually written.
 * @return An IOResult value indicating status.
 */
IOResult
os_send_nonb(int fd, const char *buf, unsigned int length,
             unsigned int *count_out)
{
  int res = 0;

  assert(buf);
  assert(count_out);

  if ((res = send(fd, buf, length, 0)) > -1)
  {
    *count_out = (unsigned int)res;
    return IO_SUCCESS;
  }

  *count_out = 0;
  return is_blocked(errno) ? IO_BLOCKED : IO_FAILURE;
}

/** Attempt a vectored write on a connected socket.
 * @param[in] fd File descriptor to write to.
 * @param[in] buf Message queue to send from.
 * @param[out] count_in Number of bytes mapped from \a buf.
 * @param[out] count_out Receives number of bytes actually written.
 * @return An IOResult value indicating status.
 */
IOResult
os_sendv_nonb(int fd, struct MsgQ *buf, unsigned int *count_in,
              unsigned int *count_out)
{
  int res = 0, count = 0;
  struct iovec iov[IOV_MAX];

  assert(buf);
  assert(count_in);
  assert(count_out);

  *count_in = 0;
  count = msgq_mapiov(buf, iov, IOV_MAX, count_in);

  if ((res = writev(fd, iov, count)) > -1)
  {
    *count_out = (unsigned int)res;
    return IO_SUCCESS;
  }

  *count_out = 0;
  return is_blocked(errno) ? IO_BLOCKED : IO_FAILURE;
}

/** Attempt to write on a non-blocking UDP socket.
 * @param[in] fd File descriptor to write to.
 * @param[in] buf Output buffer to send from.
 * @param[in] length Number of bytes to write.
 * @param[out] count_out Receives number of bytes actually written.
 * @param[in] flags Flags for call to sendto().
 * @param[in] peer Destination address of the message.
 * @return An IOResult value indicating status.
 */
IOResult os_sendto_nonb(int fd, const char *buf, unsigned int length,
                        unsigned int *count_out, unsigned int flags,
                        struct irc_ssaddr *peer)
{
  int res = 0;

  assert(buf);

  if ((res = sendto(fd, buf, length, flags, (struct sockaddr *)peer, peer->ss_len)) > -1)
  {
    if (count_out)
      *count_out = (unsigned int)res;
    return IO_SUCCESS;
  }

  if (count_out)
    *count_out = 0;
  return is_blocked(errno) ? IO_BLOCKED : IO_FAILURE;
}

/** Start a non-blocking connection.
 * @param[in] fd Disconnected file descriptor.
 * @param[in] sin Target address for connection.
 * @return IOResult code indicating status.
 */
IOResult os_connect_nonb(int fd, const struct irc_ssaddr *addr)
{
  if (connect(fd, (const struct sockaddr *)addr, addr->ss_len))
    return (errno == EINPROGRESS) ? IO_BLOCKED : IO_FAILURE;
  return IO_SUCCESS;
}

int
os_accept(int fd, struct irc_ssaddr *addr)
{
  int new_fd = 0;
  socklen_t addrlen = sizeof(struct irc_ssaddr);

  new_fd = accept(fd, (struct sockaddr *)addr, &addrlen);
  if (new_fd < 0)
    return -1;

  remove_ipv6_mapping(addr);
  setup_socket(new_fd);
  os_set_nonblocking(new_fd);
  return new_fd;
}

int
os_socket(int family, int sock_type, const char *port_name)
{
  int fd;

  if ((fd = socket(family, sock_type, 0)) < 0)
  {
    report_error(L_ALL, SOCKET_ERROR_MSG, port_name, errno);
    return -1;
  }

  if (fd > MAXCLIENTS - 1)
  {
    report_error(L_ALL, CONNLIMIT_ERROR_MSG, port_name, 0);
    close(fd);
    return -1;
  }

  if (!os_set_reuseaddr(fd))
  {
    report_error(L_ALL, REUSEADDR_ERROR_MSG, port_name, errno);
    close(fd);
    return -1;
  }

  if (!os_set_nonblocking(fd))
  {
    report_error(L_ALL, NONB_ERROR_MSG, port_name, errno);
    close(fd);
    return -1;
  }

  setup_socket(fd);
  return fd;
}

int
os_bind(int fd, struct irc_ssaddr *local)
{
  return bind(fd, (struct sockaddr *)local, local->ss_len);
}

/** Closes all file descriptors.
 * @param close_stderr If non-zero, also close stderr.
 */
void
close_connections(const int close_stderr)
{
  int i = 0;

  if (close_stderr)
  {
    close(0);
    close(1);
    close(2);
  }

  for (i = 3; i < MAXCONNECTIONS; ++i)
    close(i);
}

#ifdef HAVE_LIBCRYPTO
static IOResult
client_recv(struct Client *client_p, char *buf, unsigned int length, unsigned int *count_out)
{
  if (client_p->localClient->socket.ssl)
    return ssl_recv(&client_p->localClient->socket, buf, length, count_out);

  return os_recv_nonb(s_fd(&client_p->localClient->socket), buf, length, count_out);
}

static IOResult
client_sendv(struct Client *client_p, struct MsgQ *buf, unsigned int *count_in, unsigned int *count_out)
{
  if (client_p->localClient->socket.ssl)
    return ssl_sendv(&client_p->localClient->socket, buf, count_in, count_out);
  return os_sendv_nonb(s_fd(&client_p->localClient->socket), buf, count_in, count_out);
}
#endif

/*
 * report_error - report an error from an errno. 
 * Record error to log and also send a copy to all *LOCAL* opers online.
 *
 *        text        is a *format* string for outputing error. It must
 *                contain only two '%s', the first will be replaced
 *                by the sockhost from the client_p, and the latter will
 *                be taken from sys_errlist[errno].
 *
 *        client_p        if not NULL, is the *LOCAL* client associated with
 *                the error.
 *
 * Cannot use perror() within daemon. stderr is closed in
 * ircd and cannot be used. And, worse yet, it might have
 * been reassigned to a normal connection...
 * 
 * Actually stderr is still there IFF ircd was run with -s --Rodder
 */

void
report_error(int level, const char *text, const char *who, int error)
{
  int errtmp = errno; /* debug may change 'errno' */
  const char *errmsg = (error) ? strerror(error) : "";

  if (!errmsg)
    errmsg = "Unknown error";

  if (EmptyString(who))
    who = "unknown";

  sendto_realops_flags(UMODE_DEBUG, level, SEND_NOTICE,
                       text, who, errmsg);
  ilog(LOG_TYPE_IRCD, text, who, errmsg);
  errno = errtmp;
}

/** Complete non-blocking connect()-sequence. Check access and
 * terminate connection, if trouble detected.
 * @param client_p Client to which we have connected, with all ConfItem structs attached.
 * @return Zero on failure (caller should exit_client()), non-zero on success.
 */
static int
completed_connection(struct Client *client_p)
{
  struct MaskItem *conf = NULL;

  assert(client_p);

  /*
   * Get the socket status from the fd first to check if
   * connection actually succeeded
   */
  if ((client_p->localClient->error = os_get_sockerr(s_fd(&client_p->localClient->socket))))
  {
    const char *msg = strerror(client_p->localClient->error);

    if (!msg)
      msg = "Unknown error";

    sendto_realops_flags(UMODE_ALL, L_ADMIN, SEND_NOTICE,
                         "Connection failed to %s: %s",
                         get_client_name(client_p, HIDE_IP), msg);
    sendto_realops_flags(UMODE_ALL, L_OPER, SEND_NOTICE,
                         "Connection failed to %s: %s",
                         get_client_name(client_p, MASK_IP), msg);
    return 0;
  }

  conf = find_conf_name(&client_p->localClient->confs,
                        client_p->name, CONF_SERVER);
  if (!conf)
  {
    sendto_realops_flags(UMODE_ALL, L_ADMIN, SEND_NOTICE,
                         "Lost connect{} block for %s", get_client_name(client_p, HIDE_IP));
    sendto_realops_flags(UMODE_ALL, L_OPER, SEND_NOTICE,
                         "Lost connect{} block for %s", get_client_name(client_p, MASK_IP));
    return 0;
  }

  if (s_state(&client_p->localClient->socket) == SS_CONNECTING)
    socket_state(&client_p->localClient->socket, SS_CONNECTED);

  if (!EmptyString(conf->spasswd))
    sendto_one(client_p, "PASS %s TS %d %s", conf->spasswd, TS_CURRENT, me.id);

  send_capabilities(client_p, 0);

  assert(client_p->serv);
  SetHandshake(client_p);

  /*
   * Make us timeout after twice the timeout for DNS look ups
   */
  client_p->localClient->lasttime = CurrentTime;
  AddFlag(client_p, FLAGS_PINGSENT);

  sendto_one(client_p, "SERVER %s 1 :%s%s", me.name,
             ConfigServerHide.hidden ? "(H) " : "", me.info);

  return !IsDead(client_p);
}

/*
 * close_connection
 *        Close the physical connection. This function must make
 *        MyConnect(client_p) == FALSE, and set client_p->from == NULL.
 */
void
close_connection(struct Client *client_p)
{
  dlink_node *ptr = NULL;

  assert(client_p);

  if (IsClient(client_p))
  {
    ++ServerStats.is_cl;
    ServerStats.is_cbs += client_p->localClient->send.bytes;
    ServerStats.is_cbr += client_p->localClient->recv.bytes;
    ServerStats.is_cti += CurrentTime - client_p->localClient->firsttime;
  }
  else if (IsServer(client_p))
  {
    ++ServerStats.is_sv;
    ServerStats.is_sbs += client_p->localClient->send.bytes;
    ServerStats.is_sbr += client_p->localClient->recv.bytes;
    ServerStats.is_sti += CurrentTime - client_p->localClient->firsttime;

    DLINK_FOREACH(ptr, server_items.head)
    {
      struct MaskItem *conf = ptr->data;

      if (irccmp(conf->name, client_p->name))
        continue;

      /*
       * Reset next-connect cycle of all connect{} blocks that match
       * this servername.
       */
      conf->until = CurrentTime + conf->class->con_freq;
    }
  }
  else
    ++ServerStats.is_ni;

  if (s_fd(&client_p->localClient->socket)> -1)
  {
    flush_connections(client_p);
    dlinkDelete(&client_p->localClient->node, &connection_list);
    close(s_fd(&client_p->localClient->socket));
    socket_del(&client_p->localClient->socket);  /* Queue a socket delete */
    s_fd(&client_p->localClient->socket) = -1;
  }

  AddFlag(client_p, FLAGS_DEADSOCKET);

  MsgQClear(&client_p->localClient->sendQ);
  client_drop_sendq(client_p->localClient);
  DBufClear(&client_p->localClient->recvQ);

  MyFree(client_p->localClient->passwd);
  client_p->localClient->passwd = NULL;

  detach_conf(client_p, CONF_CLIENT|CONF_OPER|CONF_SERVER);
  client_p->from = NULL; /* ...this should catch them! >:) --msa */

  if (client_p->localClient->listener)
  {
    assert(0 < client_p->localClient->listener->ref_count);
    listener_release(client_p->localClient->listener);
    client_p->localClient->listener = NULL;
  }
}

/*
 * add_connection - creates a client which has just connected to us on 
 * the given fd. The sockhost field is initialized with the ip# of the host.
 * An unique id is calculated now, in case it is needed for auth.
 * The client is sent to the auth module for verification, and not put in
 * any client list yet.
 */
void
add_connection(struct Listener *listener, int fd, struct irc_ssaddr *irn)
{
  struct Client *new_client = make_client(NULL);

  /*
   * copy address to 'sockhost' as a string, copy it to host too
   * so we have something valid to put into error messages...
   */
  memcpy(&new_client->localClient->ip, irn, sizeof(struct irc_ssaddr));

  getnameinfo((struct sockaddr *)&new_client->localClient->ip,
              new_client->localClient->ip.ss_len, new_client->sockhost, 
              sizeof(new_client->sockhost), NULL, 0, NI_NUMERICHOST);
  new_client->localClient->aftype = new_client->localClient->ip.ss.ss_family;

#ifdef HAVE_LIBGEOIP
  /* XXX IPV6 SUPPORT XXX */
  if (irn->ss.ss_family == AF_INET && geoip_ctx)
  {
    const struct sockaddr_in *v4 = (const struct sockaddr_in *)&new_client->localClient->ip;
    new_client->localClient->country_id = GeoIP_id_by_ipnum(geoip_ctx, (unsigned long)ntohl(v4->sin_addr.s_addr));
  }
#endif

  if (new_client->sockhost[0] == ':' && new_client->sockhost[1] == ':')
  {
    strlcpy(new_client->host, "0", sizeof(new_client->host));
    strlcpy(new_client->host+1, new_client->sockhost, sizeof(new_client->host)-1);
    memmove(new_client->sockhost+1, new_client->sockhost, sizeof(new_client->sockhost)-1);
    new_client->sockhost[0] = '0';
  }
  else
    strlcpy(new_client->host, new_client->sockhost, sizeof(new_client->host));

  s_fd(&new_client->localClient->socket) = fd;
  if (!socket_add(&new_client->localClient->socket, client_sock_callback,
                  new_client->localClient, SS_CONNECTED, 0, fd))
  {
    ++ServerStats.is_ref;
//    write(fd, register_message, strlen(register_message));
    close(fd);
    s_fd(&new_client->localClient->socket) = -1;
    return;
  }

#ifdef HAVE_LIBCRYPTO
  if ((listener->flags & LISTENER_SSL))
  {
    if (!ssl_new_accept(new_client, fd))
    {
      ++ServerStats.is_ref;

      close(fd);
      s_fd(&new_client->localClient->socket) = -1;
      return;
    }
  }
#endif

  new_client->localClient->freeflag |= FREEFLAG_SOCKET;
  new_client->localClient->listener = listener;
  ++listener->ref_count;

  /* If we've made it this far we can put the client on the auth query pile */
  start_auth(new_client);
}

/** Determines whether to tell the events engine we're interested in
 * writable events.
 * \param client_p Client for which to decide this.
 */
void
update_write(struct Client *client_p)
{
  /*
   * If there are messages that need to be sent along, or if the client
   * is in the middle of a /list, then we need to tell the engine that
   * we're interested in writable events--otherwise, we need to drop
   * that interest.
   */
  socket_events(&client_p->localClient->socket,
                ((MsgQLength(&client_p->localClient->sendQ) || client_p->localClient->list_task) ?
                 SOCK_ACTION_ADD : SOCK_ACTION_DEL) | SOCK_EVENT_WRITABLE);
}

/** Read a 'packet' of data from a connection and process it.  Read in
 * 8k chunks to give a better performance rating (for server
 * connections).  Do some tricky stuff for client connections to make
 * sure they don't do any flooding >:-) -avalon
 * @param client_p Client from which to read data.
 * @param socket_ready If non-zero, more data can be read from the client's socket.
 * @return Positive number on success, zero on connection-fatal failure, negative
 *   if user is killed.
 */
static int
read_packet(struct Client *client_p, int socket_ready)
{
  unsigned int dolen = 0, length = 0;

  if (socket_ready && !(IsClient(client_p) &&
                        DBufLength(&client_p->localClient->recvQ) > get_recvq(&client_p->localClient->confs)))
  {
#ifdef HAVE_LIBCRYPTO
    switch (client_recv(client_p, readbuf, sizeof(readbuf), &length))
#else
    switch (os_recv_nonb(s_fd(&client_p->localClient->socket), readbuf, sizeof(readbuf), &length))
#endif
    {
      case IO_SUCCESS:
        if (length)
        {
          client_p->localClient->lasttime = CurrentTime;

          if (client_p->localClient->lasttime > client_p->localClient->since)
            client_p->localClient->since = CurrentTime;

          ClearPingSent(client_p);
          DelFlag(client_p, FLAGS_NONEWLINE);
        }
        break;

      case IO_BLOCKED:
        break;
      case IO_FAILURE:
        client_p->localClient->error = errno;
        /* AddFlag(client_p, FLAGS_DEADSOCKET); */
        return 0;
    }
  }

  /*
   * For server connections, we process as many as we can without
   * worrying about the time of day or anything :)
   */
  if (length > 0 && IsServer(client_p))
    return server_dopacket(client_p, readbuf, length);
  else if (length > 0 && (IsHandshake(client_p) || IsConnecting(client_p)))
    return connect_dopacket(client_p, readbuf, length);
  else
  {
    /*
     * Before we even think of parsing what we just read, stick
     * it on the end of the receive queue and do it when its
     * turn comes around.
     */
    if (length > 0 && !dbuf_put(&client_p->localClient->recvQ, readbuf, length))
      return exit_client(client_p, client_p, &me, "dbuf_put fail");

    if (DBufLength(&client_p->localClient->recvQ) > get_recvq(&client_p->localClient->confs))
      return exit_client(client_p, client_p, &me, "Excess Flood");

    while (DBufLength(&client_p->localClient->recvQ) && !HasFlag(client_p, FLAGS_NONEWLINE) &&
           (IsTrusted(client_p) || client_p->localClient->since - CurrentTime < 10))
    {
      dolen = dbuf_getmsg(&client_p->localClient->recvQ,
                          client_p->localClient->buffer, IRCD_BUFSIZE);

      /*
       * Devious looking...whats it do ? well..if a client
       * sends a *long* message without any CR or LF, then
       * dbuf_getmsg fails and we pull it out using this
       * loop which just gets the next 512 bytes and then
       * deletes the rest of the buffer contents.
       * -avalon
       */
      if (dolen == 0)
      {
        if (DBufLength(&client_p->localClient->recvQ) < 510)
          AddFlag(client_p, FLAGS_NONEWLINE);
        else
        {
          /*
           * More than 512 bytes in the line - drop the input and yell
           * at the client.
           */
          DBufClear(&client_p->localClient->recvQ);
          sendto_one(client_p, form_str(ERR_INPUTTOOLONG), me.name,
                     client_p->name[0] ? client_p->name : "*");
        }
      }
      else if (client_dopacket(client_p, dolen) == CPTR_KILLED)
        return CPTR_KILLED;

      /*
       * If it has become registered as a Server
       * then skip the per-message parsing below.
       */
      if (IsHandshake(client_p) || IsServer(client_p))
      {
        while (-1)
        {
          dolen = dbuf_get(&client_p->localClient->recvQ, readbuf, sizeof(readbuf));

          if (dolen <= 0)
            return 1;
          if (dolen == 0)
          {
            if (DBufLength(&client_p->localClient->recvQ) < 510)
              AddFlag(client_p, FLAGS_NONEWLINE);
            else
              DBufClear(&client_p->localClient->recvQ);
          }
          else if ((IsServer(client_p) &&
                    server_dopacket(client_p, readbuf, dolen) == CPTR_KILLED) ||
                   (!IsServer(client_p) &&
                    connect_dopacket(client_p, readbuf, dolen) == CPTR_KILLED))
            return CPTR_KILLED;
        }
      }
    }

    /* If there's still data to process, wait 2 seconds first */
    if (DBufLength(&client_p->localClient->recvQ) && !HasFlag(client_p, FLAGS_NONEWLINE) &&
        !t_onqueue(&client_p->localClient->proc))
    {
      client_p->localClient->freeflag |= FREEFLAG_TIMER;
      timer_add(&client_p->localClient->proc, client_timer_callback, client_p->localClient,
                TT_RELATIVE, 2);
    }
  }
  return 1;
}

/** Attempt to send a sequence of bytes to the connection.
 * As a side effect, updates \a client_p's FLAGS_BLOCKED setting
 * and sendB/sendK fields.
 * @param client_p Client that should receive data.
 * @param buf Message buffer to send to client.
 * @return Negative on connection-fatal error; otherwise
 *  number of bytes sent.
 */
unsigned int
deliver_it(struct Client *client_p, struct MsgQ *buf)
{
  unsigned int bytes_written = 0, bytes_count = 0;

  assert(client_p);

#ifdef HAVE_LIBCRYPTO
  switch (client_sendv(client_p, buf, &bytes_count, &bytes_written))
#else
  switch (os_sendv_nonb(s_fd(&client_p->localClient->socket), buf, &bytes_count, &bytes_written))
#endif
  {
    case IO_SUCCESS:
      DelFlag(client_p, FLAGS_BLOCKED);

      client_p->localClient->send.bytes += bytes_written;
      me.localClient->send.bytes        += bytes_written;

      /* A partial write implies that future writes will block. */
      if (bytes_written < bytes_count)
        AddFlag(client_p, FLAGS_BLOCKED);
      break;
    case IO_BLOCKED:
      AddFlag(client_p, FLAGS_BLOCKED);
      break;
    case IO_FAILURE:
      client_p->localClient->error = errno;
      AddFlag(client_p, FLAGS_DEADSOCKET);
      break;
  }

  return bytes_written;
}

/** Set up address and port and make a connection.
 * @param conf Provides the connection information.
 * @param client_p Client structure for the peer.
 * @return Non-zero on success; zero on failure.
 */
static int
connect_inet(struct MaskItem *conf, struct Client *client_p)
{
  struct addrinfo hints, *res;
  char portname[PORTNAMELEN + 1];
  /* conversion structs */
  struct sockaddr_in *v4;
  struct sockaddr_in6 *v6;
  IOResult result;

  assert(conf);
  assert(client_p);

  s_fd(&client_p->localClient->socket) = os_socket(conf->aftype, SOCK_STREAM, client_p->name);
  if (s_fd(&client_p->localClient->socket) < 0)
    return 0;

  switch (conf->aftype)
  {
    case AF_INET:
      v4 = (struct sockaddr_in *)&conf->bind;
      if (v4->sin_addr.s_addr != 0)
      {
        struct irc_ssaddr ipn;
        memset(&ipn, 0, sizeof(struct irc_ssaddr));
        ipn.ss.ss_family = AF_INET;
        ipn.ss_port = 0;
        memcpy(&ipn, &conf->bind, sizeof(struct irc_ssaddr));
        /* XXX ipn.ss_len */
        os_bind(s_fd(&client_p->localClient->socket), &ipn);
      }
      else if (ServerInfo.specific_ipv4_vhost)
      {
        struct irc_ssaddr ipn;
        memset(&ipn, 0, sizeof(struct irc_ssaddr));
        ipn.ss.ss_family = AF_INET;
        ipn.ss_port = 0;
        memcpy(&ipn, &ServerInfo.ip, sizeof(struct irc_ssaddr));

        os_bind(s_fd(&client_p->localClient->socket), &ipn);
      }

      break;

    case AF_INET6:
    {
      struct irc_ssaddr ipn;
      struct sockaddr_in6 *v6conf;

      memset(&ipn, 0, sizeof(struct irc_ssaddr));
      v6conf = (struct sockaddr_in6 *)&conf->bind;
      v6 = (struct sockaddr_in6 *)&ipn;

      if (memcmp(&v6conf->sin6_addr, &v6->sin6_addr, sizeof(struct in6_addr)) != 0)
      {
        memcpy(&ipn, &conf->bind, sizeof(struct irc_ssaddr));
        ipn.ss.ss_family = AF_INET6;
        ipn.ss_port = 0;
        os_bind(s_fd(&client_p->localClient->socket), &ipn);
      }
      else if (ServerInfo.specific_ipv6_vhost)
      {
        memcpy(&ipn, &ServerInfo.ip6, sizeof(struct irc_ssaddr));
        ipn.ss.ss_family = AF_INET6;
        ipn.ss_port = 0;
        os_bind(s_fd(&client_p->localClient->socket), &ipn);
      }
    }
  }


  /*
   * Save connection info in client
   */
#ifdef XXX___
  memcpy(&cli_ip(cptr), &conf->addr, sizeof(cli_ip(cptr)));
  ircd_ntoa_r(cli_sock_ip(cptr), &cli_ip(cptr));
#endif
  /*
   * We want a big buffer for server connections
   */
  if (!os_set_sockbufs(s_fd(&client_p->localClient->socket), /* XXX */ CLIENT_TCP_WINDOW, CLIENT_TCP_WINDOW))
  {
    client_p->localClient->error = errno;
    report_error(L_ALL, SETBUFS_ERROR_MSG, client_p->name, errno);
    close(s_fd(&client_p->localClient->socket));
    s_fd(&client_p->localClient->socket) = -1;
    return 0;
  }

  /*
   * Set the TOS bits - this is nonfatal if it doesn't stick.
   */
#ifdef XXX___
  if (!os_set_tos(cli_fd(cptr), FEAT_TOS_SERVER))
    report_error(L_ALL, TOS_ERROR_MSG, cli_name(cptr), errno);
#endif

  memset(&hints, 0, sizeof(hints));
  hints.ai_family = AF_UNSPEC;
  hints.ai_socktype = SOCK_STREAM;
  hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST;

  snprintf(portname, sizeof(portname), "%d", conf->port);

#if 0
  if (getaddrinfo(host, portname, &hints, &res))
  {
#if 0
    /* Send the DNS request, for the next level */
    if (aftype == AF_INET6)
      gethost_byname_type(comm_connect_dns_callback, fd, host, T_AAAA);
    else
      gethost_byname_type(comm_connect_dns_callback, fd, host, T_A);
  }
#endif
  else
  {
    /* We have a valid IP, so we just call tryconnect */
    /* Make sure we actually set the timeout here .. */
    assert(res != NULL);
    memcpy(&fd->connect.hostaddr, res->ai_addr, res->ai_addrlen);
    fd->connect.hostaddr.ss_len = res->ai_addrlen;
    fd->connect.hostaddr.ss.ss_family = res->ai_family;
    freeaddrinfo(res);
  }
#endif
  if ((result = os_connect_nonb(s_fd(&client_p->localClient->socket), &conf->addr)) == IO_FAILURE)
  {
    client_p->localClient->error = errno;
    report_error(L_ALL, CONNECT_ERROR_MSG, client_p->name, errno);
    close(s_fd(&client_p->localClient->socket));
    s_fd(&client_p->localClient->socket) = -1;
    return 0;
  }

  if (!socket_add(&client_p->localClient->socket, client_sock_callback, client_p->localClient,
                  (result == IO_SUCCESS) ? SS_CONNECTED : SS_CONNECTING,
                  SOCK_EVENT_READABLE, s_fd(&client_p->localClient->socket)))
  {
    client_p->localClient->error = ENFILE;
    report_error(L_ALL, REGISTER_ERROR_MSG, client_p->name, ENFILE);
    close(s_fd(&client_p->localClient->socket));
    s_fd(&client_p->localClient->socket) = -1;
    return 0;
  }

  client_p->localClient->freeflag |= FREEFLAG_SOCKET;
  return 1;
}

/** Start a connection to another server.
 * @param aconf Connect block data for target server.
 * @param by Client who requested the connection (if any).
 * @return Non-zero on success; zero on failure.
 */
int
connect_server(struct MaskItem *conf, struct Client *by)
{
  struct Client *client_p = NULL;

  assert(conf);

  if (conf->dns_pending)
  {
    sendto_realops_flags(UMODE_ALL, L_ALL, SEND_NOTICE,
                         "Server %s connect DNS pending",
                         conf->name);
    return 0;
  }

  if ((client_p = hash_find_client(conf->name)))
  {
    if (IsServer(client_p) || IsMe(client_p))
    {
      sendto_realops_flags(UMODE_ALL, L_ADMIN, SEND_NOTICE,
                           "Server %s already present from %s",
                           conf->name, get_client_name(client_p, SHOW_IP));
      sendto_realops_flags(UMODE_ALL, L_OPER, SEND_NOTICE,
                           "Server %s already present from %s",
                           conf->name, get_client_name(client_p, MASK_IP));
      if (by && IsClient(by) && !MyClient(by))
        sendto_one(by, ":%s NOTICE %s :Server %s already present from %s",
                 me.name, by->name, conf->name,
                 get_client_name(client_p, MASK_IP));
      return 0;
    }
    else if (IsHandshake(client_p) || IsConnecting(client_p))
    {
      if (by && IsClient(by))
        sendto_one(by, ":%s NOTICE %s :Connection to %s already in progress",
                 me.name, by->name, client_p->name,
                 get_client_name(client_p, MASK_IP));
      return 0;
    }
  }
#ifdef XXX___
  /*
   * If we don't know the IP# for this host and it is a hostname and
   * not a ip# string, then try and find the appropriate host record.
   */
  if (!irc_in_addr_valid(&aconf->address.addr)
      && !ircd_aton(&aconf->address.addr, aconf->host)) {
    gethost_byname(connect_dns_callback, conf, conf->host);
    aconf->dns_pending = 1;
    return 0;
  }
#endif
  /* Create a local client */
  client_p = make_client(NULL);

  /*
   * Copy these in so we have something for error detection.
   */
  strlcpy(client_p->name, conf->name, sizeof(client_p->name));
  strlcpy(client_p->host, conf->host, sizeof(client_p->host));

  /* We already converted the ip once, so lets use it - stu */ /* XXX */
//  strlcpy(client_p->sockhost, buf, sizeof(client_p->sockhost));
  client_p->localClient->aftype = conf->aftype;

  /*
   * Attach config entries to client here rather than in
   * completed_connection. This to avoid null pointer references
   */
  if (!attach_connect_block(client_p, conf->name, conf->host))
  {
    sendto_realops_flags(UMODE_ALL, L_ALL, SEND_NOTICE,
                         "Host %s is not enabled for connecting: no connect{} block",
                         conf->name);
    if (by && IsClient(by) && !MyClient(by))
      sendto_one(by, ":%s NOTICE %s :Connect to host %s failed: no connect{} block",
                 me.name, by->name, client_p->name);

    detach_conf(client_p, CONF_CLIENT|CONF_OPER|CONF_SERVER);
    free_client(client_p);
    return 0;
  }

  /*
   * Attempt to connect to the server in the conf line
   */
  if (!connect_inet(conf, client_p))
  {
    if (by && IsClient(by) && !MyClient(by))
      sendto_one(by, ":%s NOTICE %s :Couldn't connect to %s",
                 me.name, by->name, client_p->name);

    detach_conf(client_p, CONF_CLIENT|CONF_OPER|CONF_SERVER);
    free_client(client_p);
    return 0;
  }

  /*
   * NOTE: if we're here we have a valid C:Line and the client should
   * have started the connection and stored the remote address/port and
   * ip address name in itself
   *
   * The socket has been connected or connect is in progress.
   */
  make_server(client_p);

  if (by && IsClient(by))
    strlcpy(client_p->serv->by, by->name, sizeof(client_p->serv->by));
  else
    strlcpy(client_p->serv->by, "AutoConn.", sizeof(client_p->serv->by));

  SetConnecting(client_p);
  dlinkAdd(client_p, &client_p->node, &global_client_list);

  /* from def_fam */
  client_p->localClient->aftype = conf->aftype;

  dlinkAdd(client_p, &client_p->localClient->node, &connection_list);

  /*
   * Actually we lie, the connect hasn't succeeded yet, but we have a valid
   * client_p, so we register it now.
   * Maybe these two calls should be merged.
   */
  add_client_to_list(client_p);
  hash_add_client(client_p);
/*    nextping = CurrentTime; */

  return (s_state(&client_p->localClient->socket) == SS_CONNECTED) ?
    completed_connection(client_p) : 1;
}

/* 
 * remove_ipv6_mapping() - Removes IPv4-In-IPv6 mapping from an address
 * OSes with IPv6 mapping listening on both
 * AF_INET and AF_INET6 map AF_INET connections inside AF_INET6 structures
 * 
 */
void
remove_ipv6_mapping(struct irc_ssaddr *addr)
{
  if (addr->ss.ss_family == AF_INET6)
  {
    if (IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)addr)->sin6_addr))
    {
      struct sockaddr_in6 v6; 
      struct sockaddr_in *v4 = (struct sockaddr_in *)addr;

      memcpy(&v6, addr, sizeof(v6));
      memset(v4, 0, sizeof(struct sockaddr_in));
      memcpy(&v4->sin_addr, &v6.sin6_addr.s6_addr[12], sizeof(v4->sin_addr));

      addr->ss.ss_family = AF_INET;
      addr->ss_len = sizeof(struct sockaddr_in);
    }
    else 
      addr->ss_len = sizeof(struct sockaddr_in6);
  }
  else
    addr->ss_len = sizeof(struct sockaddr_in);
} 

/** Process events on a client socket.
 * @param ev Socket event structure that has a struct Connection as
 *   its associated data.
 */
static void
client_sock_callback(struct Event *ev)
{
  struct Client *client_p = NULL;
  struct Connection *con = NULL;
  const char *fmt = "%s";
  const char *fallback = NULL;

  assert(ev_socket(ev));
  assert(s_data(ev_socket(ev)));

  con = s_data(ev_socket(ev));

  assert(con->client_p || ev_type(ev) == ET_DESTROY);

  client_p = con->client_p;

  assert(!client_p || con == client_p->localClient);

  switch (ev_type(ev))
  {
    case ET_DESTROY:
      con->freeflag &= ~FREEFLAG_SOCKET;

      if (!con->freeflag && !client_p)
        free_connection(con);
      break;

    case ET_CONNECT:  /* Socket connection completed */
      if (!completed_connection(client_p) || IsDead(client_p))
        fallback = client_p->info;
      break;

    case ET_ERROR:  /* An error occurred */
      fallback = client_p->info;
      client_p->localClient->error = ev_data(ev);

      /*
       * If the OS told us we have a bad file descriptor, we should
       * record that for future reference.
       */
      if (client_p->localClient->error == EBADF)
        s_fd(&client_p->localClient->socket) = -1;

      if (s_state(&con->socket) == SS_CONNECTING)
      {
        completed_connection(client_p);

        /*
         * For some reason, the os_get_sockerr() in completed_connect()
         * can return 0 even when ev_data(ev) indicates a real error, so
         * re-assign the client error here.
         */
        client_p->localClient->error = ev_data(ev);
        break;
      }

    /*FALLTHROUGH*/
    case ET_EOF:  /* End of file on socket */
      AddFlag(client_p, FLAGS_DEADSOCKET);

      if ((IsServer(client_p) || IsHandshake(client_p)) && client_p->localClient->error == 0)
      {
        exit_client_msg(client_p, client_p, &me, "Server %s closed the connection (%s)",
                        client_p->name, client_p->serv->last_error_msg);
        return;
      }
      else
      {
        fmt = "Read error: %s";
        fallback = "EOF from client";
      }

      break;

    case ET_WRITE:  /* Socket is writable */
      DelFlag(client_p, FLAGS_BLOCKED);

      if (client_p->localClient->list_task && MsgQLength(&client_p->localClient->sendQ) < 2048)
        safe_list_channels(client_p, 0);
      send_queued(client_p);
      break;

    case ET_READ:  /* Socket is readable */
      if (!IsDead(client_p))
      {
        if (read_packet(client_p, 1) == 0)  /* Error while reading packet */
          fallback = "EOF from client";
      }

      break;

    default:
      assert(0 && "Unrecognized socket event in client_sock_callback()");
      break;
  }

  assert(!client_p || !client_p->localClient || con == client_p->localClient);

  if (fallback)
  {
    const char *msg = (client_p->localClient->error) ? strerror(client_p->localClient->error) : fallback;

    if (!msg)
      msg = "Unknown error";
    exit_client_msg(client_p, client_p, &me, fmt, msg);
  }
}

/** Process a timer on client socket.
 * @param ev Timer event that has a struct Connection as its
 * associated data.
 */
static void
client_timer_callback(struct Event *ev)
{
  struct Client *client_p = NULL;
  struct Connection *con = NULL;

  assert(ev_timer(ev));
  assert(t_data(ev_timer(ev)));
  assert(ET_DESTROY == ev_type(ev) || ET_EXPIRE == ev_type(ev));

  con = t_data(ev_timer(ev));

  assert(con->client_p || ev_type(ev) == ET_DESTROY);

  client_p = con->client_p;

  assert(!client_p || con == client_p->localClient);

  if (ev_type(ev) == ET_DESTROY)
  {
    con->freeflag &= ~FREEFLAG_TIMER;  /* Timer has expired... */

    if (!con->freeflag && !client_p)
      free_connection(con);  /* Client is being destroyed */
  }
  else
    read_packet(client_p, 0);  /* read_packet will re-add timer if needed */

  assert(!client_p || !client_p->localClient || con == client_p->localClient);
}
Revision:	2456
Committed:	Wed Aug 14 17:46:30 2013 UTC (10 years, 8 months ago) by michael
Content type:	text/x-csrc
File size:	41257 byte(s)
Log Message:	- ioengine changes as of 14AUG13