lib/tcp.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 <event2/bufferevent.h>
  11 #if defined (RS_ENABLE_TLS)
  12 #include <event2/bufferevent_ssl.h>
  13 #include <openssl/err.h>
  14 #endif
  15 #include <radsec/radsec.h>
  16 #include <radsec/radsec-impl.h>
  17 #include "tcp.h"
  18 #include "packet.h"
  19 #include "conn.h"
  20 #include "debug.h"
  21 #include "event.h"
  22
  23 #if defined (DEBUG)
  24 #include <event2/buffer.h>
  25 #endif
  26
  27 /* Read one RADIUS packet header.  Return !0 on error.  A return value
  28    of 0 means that we need more data.  */
  29 static int
  30 _read_header (struct rs_packet *pkt)
  31 {
  32   size_t n = 0;
  33
  34   n = bufferevent_read (pkt->conn->bev, pkt->hdr, RS_HEADER_LEN);
  35   if (n == RS_HEADER_LEN)
  36     {
  37       pkt->flags |= rs_packet_hdr_read_flag;
  38       pkt->rpkt->data_len = (pkt->hdr[2] << 8) + pkt->hdr[3];
  39       if (pkt->rpkt->data_len < 20 || pkt->rpkt->data_len > 4096)
  40         {
  41           conn_close (&pkt->conn);
  42           return rs_err_conn_push (pkt->conn, RSE_INVALID_PKT,
  43                                    "invalid packet length: %d",
  44                                    pkt->rpkt->data_len);
  45         }
  46       pkt->rpkt->data = rs_malloc (pkt->conn->ctx, pkt->rpkt->data_len);
  47       if (!pkt->rpkt->data)
  48         {
  49           conn_close (&pkt->conn);
  50           return rs_err_conn_push_fl (pkt->conn, RSE_NOMEM, __FILE__, __LINE__,
  51                                       NULL);
  52         }
  53       memcpy (pkt->rpkt->data, pkt->hdr, RS_HEADER_LEN);
  54       bufferevent_setwatermark (pkt->conn->bev, EV_READ,
  55                                 pkt->rpkt->data_len - RS_HEADER_LEN, 0);
  56       rs_debug (("%s: packet header read, total pkt len=%d\n",
  57                  __func__, pkt->rpkt->data_len));
  58     }
  59   else if (n < 0)
  60     {
  61       rs_debug (("%s: buffer frozen while reading header\n", __func__));
  62     }
  63   else      /* Error: libevent gave us less than the low watermark. */
  64     {
  65       conn_close (&pkt->conn);
  66       return rs_err_conn_push_fl (pkt->conn, RSE_INTERNAL, __FILE__, __LINE__,
  67                                   "got %d octets reading header", n);
  68     }
  69
  70   return 0;
  71 }
  72
  73 static int
  74 _read_packet (struct rs_packet *pkt)
  75 {
  76   size_t n = 0;
  77
  78   rs_debug (("%s: trying to read %d octets of packet data\n", __func__,
  79              pkt->rpkt->data_len - RS_HEADER_LEN));
  80
  81   n = bufferevent_read (pkt->conn->bev,
  82                         pkt->rpkt->data + RS_HEADER_LEN,
  83                         pkt->rpkt->data_len - RS_HEADER_LEN);
  84
  85   rs_debug (("%s: read %ld octets of packet data\n", __func__, n));
  86
  87   if (n == pkt->rpkt->data_len - RS_HEADER_LEN)
  88     {
  89       bufferevent_disable (pkt->conn->bev, EV_READ);
  90       rs_debug (("%s: complete packet read\n", __func__));
  91       pkt->flags &= ~rs_packet_hdr_read_flag;
  92       memset (pkt->hdr, 0, sizeof(*pkt->hdr));
  93
  94       /* Checks done by rad_packet_ok:
  95          - lenghts (FIXME: checks really ok for tcp?)
  96          - invalid code field
  97          - attribute lengths >= 2
  98          - attribute sizes adding up correctly  */
  99       if (!rad_packet_ok (pkt->rpkt, 0))
 100         {
 101           conn_close (&pkt->conn);
 102           return rs_err_conn_push_fl (pkt->conn, RSE_FR, __FILE__, __LINE__,
 103                                       "invalid packet: %s", fr_strerror ());
 104         }
 105
 106 #if defined (DEBUG)
 107       /* Find out what happens if there's data left in the buffer.  */
 108       {
 109         size_t rest = 0;
 110         rest = evbuffer_get_length (bufferevent_get_input (pkt->conn->bev));
 111         if (rest)
 112           rs_debug (("%s: returning with %d octets left in buffer\n", __func__,
 113                      rest));
 114       }
 115 #endif
 116
 117       /* Hand over message to user.  This changes ownership of pkt.
 118          Don't touch it afterwards -- it might have been freed.  */
 119       if (pkt->conn->callbacks.received_cb)
 120         pkt->conn->callbacks.received_cb (pkt, pkt->conn->user_data);
 121     }
 122   else if (n < 0)               /* Buffer frozen.  */
 123     rs_debug (("%s: buffer frozen when reading packet\n", __func__));
 124   else                          /* Short packet.  */
 125     rs_debug (("%s: waiting for another %d octets\n", __func__,
 126                pkt->rpkt->data_len - RS_HEADER_LEN - n));
 127
 128   return 0;
 129 }
 130
 131 /* The read callback for TCP.
 132
 133    Read exactly one RADIUS message from BEV and store it in struct
 134    rs_packet passed in USER_DATA.
 135
 136    Inform upper layer about successful reception of received RADIUS
 137    message by invoking conn->callbacks.recevied_cb(), if !NULL.  */
 138 void
 139 tcp_read_cb (struct bufferevent *bev, void *user_data)
 140 {
 141   struct rs_packet *pkt = (struct rs_packet *) user_data;
 142
 143   assert (pkt);
 144   assert (pkt->conn);
 145   assert (pkt->rpkt);
 146
 147   pkt->rpkt->sockfd = pkt->conn->fd;
 148   pkt->rpkt->vps = NULL;
 149
 150   if ((pkt->flags & rs_packet_hdr_read_flag) == 0)
 151     if (_read_header (pkt))
 152       return;                   /* Error.  */
 153   _read_packet (pkt);
 154 }
 155
 156 void
 157 tcp_event_cb (struct bufferevent *bev, short events, void *user_data)
 158 {
 159   struct rs_packet *pkt = (struct rs_packet *) user_data;
 160   struct rs_connection *conn = NULL;
 161   int sockerr = 0;
 162 #if defined (RS_ENABLE_TLS)
 163   unsigned long tlserr = 0;
 164 #endif
 165 #if defined (DEBUG)
 166   struct rs_peer *p = NULL;
 167 #endif
 168
 169   assert (pkt);
 170   assert (pkt->conn);
 171   conn = pkt->conn;
 172 #if defined (DEBUG)
 173   assert (pkt->conn->active_peer);
 174   p = conn->active_peer;
 175 #endif
 176
 177   conn->is_connecting = 0;
 178   if (events & BEV_EVENT_CONNECTED)
 179     {
 180       if (conn->tev)
 181         evtimer_del (conn->tev); /* Cancel connect timer.  */
 182       event_on_connect (conn, pkt);
 183     }
 184   else if (events & BEV_EVENT_EOF)
 185     {
 186       event_on_disconnect (conn);
 187     }
 188   else if (events & BEV_EVENT_TIMEOUT)
 189     {
 190       rs_debug (("%s: %p times out on %s\n", __func__, p,
 191                  (events & BEV_EVENT_READING) ? "read" : "write"));
 192       rs_err_conn_push_fl (conn, RSE_TIMEOUT_IO, __FILE__, __LINE__, NULL);
 193     }
 194   else if (events & BEV_EVENT_ERROR)
 195     {
 196       sockerr = evutil_socket_geterror (conn->active_peer->fd);
 197       if (sockerr == 0) /* FIXME: True that errno == 0 means closed? */
 198         {
 199           event_on_disconnect (conn);
 200           rs_err_conn_push_fl (conn, RSE_DISCO, __FILE__, __LINE__, NULL);
 201         }
 202       else
 203         {
 204           rs_debug (("%s: %d: %d (%s)\n", __func__, conn->fd, sockerr,
 205                      evutil_socket_error_to_string (sockerr)));
 206           rs_err_conn_push_fl (conn, RSE_SOCKERR, __FILE__, __LINE__,
 207                                "%d: %d (%s)", conn->fd, sockerr,
 208                                evutil_socket_error_to_string (sockerr));
 209         }
 210 #if defined (RS_ENABLE_TLS)
 211       if (conn->tls_ssl)        /* FIXME: correct check?  */
 212         {
 213           for (tlserr = bufferevent_get_openssl_error (conn->bev);
 214                tlserr;
 215                tlserr = bufferevent_get_openssl_error (conn->bev))
 216             {
 217               rs_debug (("%s: openssl error: %s\n", __func__,
 218                          ERR_error_string (tlserr, NULL)));
 219               rs_err_conn_push_fl (conn, RSE_SSLERR, __FILE__, __LINE__,
 220                                    ERR_error_string (tlserr, NULL));
 221             }
 222         }
 223 #endif  /* RS_ENABLE_TLS */
 224       event_loopbreak (conn);
 225     }
 226
 227 #if defined (DEBUG)
 228   if (events & BEV_EVENT_ERROR && events != BEV_EVENT_ERROR)
 229     rs_debug (("%s: BEV_EVENT_ERROR and more: 0x%x\n", __func__, events));
 230 #endif
 231 }
 232
 233 void
 234 tcp_write_cb (struct bufferevent *bev, void *ctx)
 235 {
 236   struct rs_packet *pkt = (struct rs_packet *) ctx;
 237
 238   assert (pkt);
 239   assert (pkt->conn);
 240
 241   if (pkt->conn->callbacks.sent_cb)
 242     pkt->conn->callbacks.sent_cb (pkt->conn->user_data);
 243 }
 244
 245 int
 246 tcp_init_connect_timer (struct rs_connection *conn)
 247 {
 248   assert (conn);
 249
 250   if (conn->tev)
 251     event_free (conn->tev);
 252   conn->tev = evtimer_new (conn->evb, event_conn_timeout_cb, conn);
 253   if (!conn->tev)
 254     return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
 255                                 "evtimer_new");
 256
 257   return RSE_OK;
 258 }