lib/udp.c

   1 /* Copyright 2011 NORDUnet A/S. All rights reserved.
   2    See the file COPYING for licensing information.  */
   3
   4 #if defined HAVE_CONFIG_H
   5 #include <config.h>
   6 #endif
   7
   8 #include <assert.h>
   9 #include <event2/event.h>
  10 #include <radsec/radsec.h>
  11 #include <radsec/radsec-impl.h>
  12 #include "debug.h"
  13 #include "event.h"
  14 #include "compat.h"
  15 #include "udp.h"
  16
  17 /* Callback for conn->wev and conn->rev.  FIXME: Rename.  */
  18 static void
  19 _evcb (evutil_socket_t fd, short what, void *user_data)
  20 {
  21   //rs_debug (("%s: fd=%d what=0x%x\n", __func__, fd, what));
  22   if (what & EV_TIMEOUT)
  23     {
  24       struct rs_connection *conn = (struct rs_connection *) user_data;
  25       assert (conn);
  26       conn->is_connecting = 0;
  27       rs_debug (("%s: UDP timeout NYI", __func__));
  28     }
  29   else if (what & EV_READ)
  30     {
  31       struct rs_connection *conn = (struct rs_connection *) user_data;
  32       assert (conn);
  33       /* read a single UDP packet and stick it in a new struct
  34          rs_packet */
  35
  36       rs_debug (("%s: UDP read NYI", __func__));
  37     }
  38   else if (what & EV_WRITE)
  39     {
  40       struct rs_packet *pkt = (struct rs_packet *) user_data;
  41       assert (pkt);
  42       /* Socket ready for writing, possibly as a result of a
  43          successful connect.  */
  44       if (!pkt->conn->is_connected)
  45         event_on_connect (pkt->conn, pkt);
  46       if (pkt->conn->out_queue)
  47         {
  48           /* Send one packet, the first.  */
  49           ssize_t r = 0;
  50           struct rs_packet *p = pkt->conn->out_queue;
  51
  52           assert (p->rpkt);
  53           assert (p->rpkt->data);
  54           r = compat_send (fd, p->rpkt->data, p->rpkt->data_len, 0);
  55           if (r == -1)
  56             {
  57               int sockerr = evutil_socket_geterror (p->conn->fd);
  58               if (sockerr != EAGAIN)
  59                 rs_err_conn_push_fl (p->conn, RSE_SOCKERR, __FILE__, __LINE__,
  60                                      "%d: send: %d (%s)", fd, sockerr,
  61                                      evutil_socket_error_to_string (sockerr));
  62               return;           /* Don't unlink packet. */
  63             }
  64           pkt->conn->out_queue = p->next;
  65         }
  66     }
  67 }
  68
  69 int
  70 udp_init (struct rs_connection *conn, struct rs_packet *pkt)
  71 {
  72   assert (!conn->bev);
  73
  74   conn->rev = event_new (conn->evb, conn->fd, EV_READ|EV_PERSIST, _evcb, conn);
  75   conn->wev = event_new (conn->evb, conn->fd, EV_WRITE|EV_PERSIST, _evcb, pkt);
  76   if (!conn->rev || !conn->wev)
  77     {
  78       if (conn->rev)
  79         event_free (conn->rev);
  80       /* ENOMEM _or_ EINVAL but EINVAL only if we use EV_SIGNAL, at
  81          least for now (libevent-2.0.5).  */
  82       return rs_err_conn_push_fl (conn, RSE_NOMEM, __FILE__, __LINE__, NULL);
  83     }
  84   return RSE_OK;
  85 }