lib/conn.c

   1 /* Copyright 2010, 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 <string.h>
   9 #include <assert.h>
  10 #include <event2/event.h>
  11 #include <event2/bufferevent.h>
  12 #include <radsec/radsec.h>
  13 #include <radsec/radsec-impl.h>
  14 #include "debug.h"
  15 #include "conn.h"
  16 #include "event.h"
  17 #include "packet.h"
  18 #include "tcp.h"
  19
  20 int
  21 conn_close (struct rs_connection **connp)
  22 {
  23   int r;
  24   assert (connp);
  25   assert (*connp);
  26   r = rs_conn_destroy (*connp);
  27   if (!r)
  28     *connp = NULL;
  29   return r;
  30 }
  31
  32 int
  33 conn_user_dispatch_p (const struct rs_connection *conn)
  34 {
  35   assert (conn);
  36
  37   return (conn->callbacks.connected_cb ||
  38           conn->callbacks.disconnected_cb ||
  39           conn->callbacks.received_cb ||
  40           conn->callbacks.sent_cb);
  41 }
  42
  43 int
  44 rs_conn_create (struct rs_context *ctx,
  45                 struct rs_connection **conn,
  46                 const char *config)
  47 {
  48   struct rs_connection *c;
  49
  50   c = (struct rs_connection *) malloc (sizeof(struct rs_connection));
  51   if (!c)
  52     return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__, NULL);
  53
  54   memset (c, 0, sizeof(struct rs_connection));
  55   c->ctx = ctx;
  56   c->fd = -1;
  57   if (config)
  58     {
  59       struct rs_realm *r = rs_conf_find_realm (ctx, config);
  60       if (r)
  61         {
  62           struct rs_peer *p;
  63
  64           c->realm = r;
  65           c->peers = r->peers;  /* FIXME: Copy instead?  */
  66           for (p = c->peers; p; p = p->next)
  67             p->conn = c;
  68           c->timeout.tv_sec = r->timeout;
  69           c->tryagain = r->retries;
  70         }
  71       else
  72         {
  73           c->realm = rs_malloc (ctx, sizeof (struct rs_realm));
  74           if (!c->realm)
  75             return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__,
  76                                        NULL);
  77           memset (c->realm, 0, sizeof (struct rs_realm));
  78         }
  79     }
  80
  81   if (conn)
  82     *conn = c;
  83   return RSE_OK;
  84 }
  85
  86 void
  87 rs_conn_set_type (struct rs_connection *conn, rs_conn_type_t type)
  88 {
  89   assert (conn);
  90   assert (conn->realm);
  91   conn->realm->type = type;
  92 }
  93
  94 int
  95 rs_conn_add_listener (struct rs_connection *conn,
  96                       rs_conn_type_t type,
  97                       const char *hostname,
  98                       int port)
  99 {
 100   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
 101 }
 102
 103
 104 int
 105 rs_conn_disconnect (struct rs_connection *conn)
 106 {
 107   int err = 0;
 108
 109   assert (conn);
 110
 111   err = evutil_closesocket (conn->fd);
 112   conn->fd = -1;
 113   return err;
 114 }
 115
 116 int
 117 rs_conn_destroy (struct rs_connection *conn)
 118 {
 119   int err = 0;
 120
 121   assert (conn);
 122
 123   /* NOTE: conn->realm is owned by context.  */
 124   /* NOTE: conn->peers is owned by context.  */
 125
 126   if (conn->is_connected)
 127     err = rs_conn_disconnect (conn);
 128   if (conn->tev)
 129     event_free (conn->tev);
 130   if (conn->bev)
 131     bufferevent_free (conn->bev);
 132   if (conn->evb)
 133     event_base_free (conn->evb);
 134
 135   /* TODO: free tls_ctx  */
 136   /* TODO: free tls_ssl  */
 137
 138   rs_free (conn->ctx, conn);
 139
 140   return err;
 141 }
 142
 143 int
 144 rs_conn_set_eventbase (struct rs_connection *conn, struct event_base *eb)
 145 {
 146   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
 147 }
 148
 149 void
 150 rs_conn_set_callbacks (struct rs_connection *conn, struct rs_conn_callbacks *cb)
 151 {
 152   assert (conn);
 153   memcpy (&conn->callbacks, cb, sizeof (conn->callbacks));
 154 }
 155
 156 void
 157 rs_conn_del_callbacks (struct rs_connection *conn)
 158 {
 159   assert (conn);
 160   memset (&conn->callbacks, 0, sizeof (conn->callbacks));
 161 }
 162
 163 struct rs_conn_callbacks *
 164 rs_conn_get_callbacks(struct rs_connection *conn)
 165 {
 166   assert (conn);
 167   return &conn->callbacks;
 168 }
 169
 170 int
 171 rs_conn_select_peer (struct rs_connection *conn, const char *name)
 172 {
 173   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
 174 }
 175
 176 int
 177 rs_conn_get_current_peer (struct rs_connection *conn,
 178                           const char *name,
 179                           size_t buflen)
 180 {
 181   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
 182 }
 183
 184 int rs_conn_fd (struct rs_connection *conn)
 185 {
 186   assert (conn);
 187   assert (conn->active_peer);
 188   return conn->fd;
 189 }
 190
 191 static void
 192 _rcb (struct rs_packet *packet, void *user_data)
 193 {
 194   struct rs_packet *pkt = (struct rs_packet *) user_data;
 195   assert (pkt);
 196   assert (pkt->conn);
 197
 198   pkt->flags |= rs_packet_received_flag;
 199   if (pkt->conn->bev)
 200     bufferevent_disable (pkt->conn->bev, EV_WRITE|EV_READ);
 201   else
 202     event_del (pkt->conn->rev);
 203 }
 204
 205 int
 206 rs_conn_receive_packet (struct rs_connection *conn,
 207                         struct rs_packet *req_msg,
 208                         struct rs_packet **pkt_out)
 209 {
 210   int err = 0;
 211   struct rs_packet *pkt = NULL;
 212
 213   assert (conn);
 214   assert (conn->realm);
 215   assert (!conn_user_dispatch_p (conn)); /* Dispatching mode only.  */
 216
 217   if (rs_packet_create (conn, &pkt))
 218     return -1;
 219
 220   assert (conn->evb);
 221   assert (conn->fd >= 0);
 222
 223   conn->callbacks.received_cb = _rcb;
 224   conn->user_data = pkt;
 225   pkt->flags &= ~rs_packet_received_flag;
 226
 227   if (conn->bev)                /* TCP.  */
 228     {
 229       bufferevent_setwatermark (conn->bev, EV_READ, RS_HEADER_LEN, 0);
 230       bufferevent_setcb (conn->bev, tcp_read_cb, NULL, tcp_event_cb, pkt);
 231       bufferevent_enable (conn->bev, EV_READ);
 232     }
 233   else                          /* UDP.  */
 234     {
 235       /* Put fresh packet in user_data for the callback and enable the
 236          read event.  */
 237       event_assign (conn->rev, conn->evb, event_get_fd (conn->rev),
 238                     EV_READ, event_get_callback (conn->rev), pkt);
 239       err = event_add (conn->rev, NULL);
 240       if (err < 0)
 241         return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
 242                                     "event_add: %s",
 243                                     evutil_gai_strerror (err));
 244
 245       /* Activae retransmission timer.  */
 246       conn_activate_timeout (pkt->conn);
 247     }
 248
 249   rs_debug (("%s: entering event loop\n", __func__));
 250   err = event_base_dispatch (conn->evb);
 251   conn->callbacks.received_cb = NULL;
 252   if (err < 0)
 253     return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
 254                                 "event_base_dispatch: %s",
 255                                 evutil_gai_strerror (err));
 256   rs_debug (("%s: event loop done\n", __func__));
 257
 258   if ((pkt->flags & rs_packet_received_flag) == 0
 259       || (req_msg
 260           && packet_verify_response (pkt->conn, pkt, req_msg) != RSE_OK))
 261     {
 262       assert (rs_err_conn_peek_code (pkt->conn));
 263       return rs_err_conn_peek_code (conn);
 264     }
 265
 266   if (pkt_out)
 267     *pkt_out = pkt;
 268   return RSE_OK;
 269 }
 270
 271 void
 272 rs_conn_set_timeout(struct rs_connection *conn, struct timeval *tv)
 273 {
 274   assert (conn);
 275   assert (tv);
 276   conn->timeout = *tv;
 277 }
 278
 279 int
 280 conn_activate_timeout (struct rs_connection *conn)
 281 {
 282   assert (conn);
 283   assert (conn->tev);
 284   assert (conn->evb);
 285   if (conn->timeout.tv_sec || conn->timeout.tv_usec)
 286     {
 287       rs_debug (("%s: activating timer: %d.%d\n", __func__,
 288                  conn->timeout.tv_sec, conn->timeout.tv_usec));
 289       if (evtimer_add (conn->tev, &conn->timeout))
 290         return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
 291                                     "evtimer_add: %d", errno);
 292     }
 293   return RSE_OK;
 294 }