lib/packet.c

   1 /* See the file COPYING for licensing information.  */
   2
   3 #include <string.h>
   4 #include <assert.h>
   5 #include <freeradius/libradius.h>
   6 #include <event2/event.h>
   7 #include <event2/bufferevent.h>
   8 #include <radsec/radsec.h>
   9 #include <radsec/radsec-impl.h>
  10 #if defined DEBUG
  11 #include <netdb.h>
  12 #include <sys/socket.h>
  13 #include "debug.h"
  14 #endif
  15
  16 static int
  17 _packet_create (struct rs_connection *conn, struct rs_packet **pkt_out)
  18 {
  19   struct rs_packet *p;
  20   RADIUS_PACKET *rpkt;
  21
  22   *pkt_out = NULL;
  23
  24   rpkt = rad_alloc (1);
  25   if (!rpkt)
  26     return rs_err_conn_push (conn, RSE_NOMEM, __func__);
  27   rpkt->id = conn->nextid++;
  28
  29   p = (struct rs_packet *) malloc (sizeof (struct rs_packet));
  30   if (!p)
  31     {
  32       rad_free (&rpkt);
  33       return rs_err_conn_push (conn, RSE_NOMEM, __func__);
  34     }
  35   memset (p, 0, sizeof (struct rs_packet));
  36   p->conn = conn;
  37   p->rpkt = rpkt;
  38
  39   *pkt_out = p;
  40   return RSE_OK;
  41 }
  42
  43 static int
  44 _do_send (struct rs_packet *pkt)
  45 {
  46   int err;
  47
  48   if (rad_encode (pkt->rpkt, NULL, pkt->conn->active_peer->secret))
  49     return rs_err_conn_push_fl (pkt->conn, RSE_FR, __FILE__, __LINE__,
  50                                 "rad_encode: %s", fr_strerror ());
  51   if (rad_sign (pkt->rpkt, NULL, pkt->conn->active_peer->secret))
  52     return rs_err_conn_push_fl (pkt->conn, RSE_FR, __FILE__, __LINE__,
  53                                 "rad_sign: %s", fr_strerror ());
  54   assert (pkt->rpkt);
  55 #if defined (DEBUG)
  56   {
  57     char host[80], serv[80];
  58
  59     getnameinfo (pkt->conn->active_peer->addr->ai_addr,
  60                  pkt->conn->active_peer->addr->ai_addrlen,
  61                  host, sizeof(host), serv, sizeof(serv),
  62                  0 /* NI_NUMERICHOST|NI_NUMERICSERV*/);
  63     fprintf (stderr, "%s: about to send this to %s:%s:\n", __func__, host,
  64              serv);
  65     rs_dump_packet (pkt);
  66   }
  67 #endif
  68   err = bufferevent_write (pkt->conn->bev, pkt->rpkt->data,
  69                            pkt->rpkt->data_len);
  70   if (err < 0)
  71     return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
  72                                 "bufferevent_write: %s",
  73                                 evutil_gai_strerror(err));
  74   return RSE_OK;
  75 }
  76
  77 static void
  78 _event_cb (struct bufferevent *bev, short events, void *ctx)
  79 {
  80   struct rs_packet *pkt = (struct rs_packet *)ctx;
  81   struct rs_connection *conn;
  82   struct rs_peer *p;
  83   int err;
  84
  85   assert (pkt);
  86   assert (pkt->conn);
  87   assert (pkt->conn->active_peer);
  88   conn = pkt->conn;
  89   p = conn->active_peer;
  90
  91   p->is_connecting = 0;
  92   if (events & BEV_EVENT_CONNECTED)
  93     {
  94       p->is_connected = 1;
  95       if (conn->callbacks.connected_cb)
  96         conn->callbacks.connected_cb (conn->user_data);
  97
  98 #if defined (DEBUG)
  99       fprintf (stderr, "%s: connected\n", __func__);
 100 #endif
 101       if (_do_send (pkt))
 102         return;
 103       if (conn->callbacks.sent_cb)
 104         conn->callbacks.sent_cb (conn->user_data);
 105       /* Packet will be freed in write callback.  */
 106     }
 107   else if (events & BEV_EVENT_ERROR)
 108     rs_err_conn_push_fl (pkt->conn, RSE_CONNERR, __FILE__, __LINE__, NULL);
 109 }
 110
 111 static void
 112 _write_cb (struct bufferevent *bev, void *ctx)
 113 {
 114   struct rs_packet *pkt = (struct rs_packet *) ctx;
 115
 116   assert (pkt);
 117   assert (pkt->conn);
 118 #if defined (DEBUG)
 119   fprintf (stderr, "%s: packet written, breaking event loop\n", __func__);
 120 #endif
 121   if (event_base_loopbreak (pkt->conn->evb) < 0)
 122     abort ();                   /* FIXME */
 123   if (!pkt->conn->callbacks.sent_cb) /* Callback owns the packet now.  */
 124     rs_packet_destroy (pkt);
 125 }
 126
 127 static void
 128 _read_cb (struct bufferevent *bev, void *ctx)
 129 {
 130   struct rs_packet *pkt = (struct rs_packet *)ctx;
 131   size_t n;
 132
 133   assert (pkt);
 134   assert (pkt->conn);
 135
 136   pkt->rpkt->sockfd = pkt->conn->active_peer->fd; /* FIXME: Why?  */
 137   pkt->rpkt->vps = NULL;                          /* FIXME: Why?  */
 138
 139   if (!pkt->hdr_read_flag)
 140     {
 141       n = bufferevent_read (pkt->conn->bev, pkt->hdr, RS_HEADER_LEN);
 142       if (n == RS_HEADER_LEN)
 143         {
 144           pkt->hdr_read_flag = 1;
 145           pkt->rpkt->data_len = (pkt->hdr[2] << 8) + pkt->hdr[3];
 146           if (pkt->rpkt->data_len < 20 /* || len > 4096 */)
 147             abort ();           /* FIXME: Read and discard packet.  */
 148           pkt->rpkt->data = rs_malloc (pkt->conn->ctx, pkt->rpkt->data_len);
 149           if (!pkt->rpkt->data)
 150             {
 151               rs_err_conn_push_fl (pkt->conn, RSE_NOMEM, __FILE__, __LINE__,
 152                                    NULL);
 153               abort ();         /* FIXME: Read and discard packet.  */
 154             }
 155           memcpy (pkt->rpkt->data, pkt->hdr, RS_HEADER_LEN);
 156           bufferevent_setwatermark (pkt->conn->bev, EV_READ,
 157                                     pkt->rpkt->data_len - RS_HEADER_LEN, 0);
 158 #if defined (DEBUG)
 159           fprintf (stderr, "%s: packet header read, total pkt len=%d\n",
 160                    __func__, pkt->rpkt->data_len);
 161 #endif
 162         }
 163       else if (n < 0)
 164         return;                 /* Buffer frozen.  */
 165       else
 166         assert (!"short header");
 167     }
 168
 169 #if defined (DEBUG)
 170   printf ("%s: trying to read %d octets of packet data\n", __func__, pkt->rpkt->data_len - RS_HEADER_LEN);
 171 #endif
 172   n = bufferevent_read (pkt->conn->bev, pkt->rpkt->data + RS_HEADER_LEN, pkt->rpkt->data_len - RS_HEADER_LEN);
 173 #if defined (DEBUG)
 174   printf ("%s: read %d octets of packet data\n", __func__, n);
 175 #endif
 176   if (n == pkt->rpkt->data_len - RS_HEADER_LEN)
 177     {
 178       bufferevent_disable (pkt->conn->bev, EV_READ);
 179       pkt->hdr_read_flag = 0;
 180       memset (pkt->hdr, 0, sizeof(*pkt->hdr));
 181 #if defined (DEBUG)
 182       fprintf (stderr, "%s: complete packet read\n", __func__);
 183 #endif
 184       if (!rad_packet_ok (pkt->rpkt, 0) != 0)
 185         return;
 186       if (rad_decode (pkt->rpkt, NULL, pkt->conn->active_peer->secret) != 0)
 187         return;
 188
 189       if (pkt->conn->callbacks.received_cb)
 190         pkt->conn->callbacks.received_cb(pkt, pkt->conn->user_data);
 191
 192       if (event_base_loopbreak (pkt->conn->evb) < 0)
 193         abort ();               /* FIXME */
 194     }
 195   else if (n < 0)
 196     return;                     /* Buffer frozen.  */
 197   else
 198     assert (!"short packet");
 199 }
 200
 201 static void
 202 _evlog_cb (int severity, const char *msg)
 203 {
 204   const char *sevstr;
 205   switch (severity)
 206     {
 207     case _EVENT_LOG_DEBUG:
 208 #if !defined (DEBUG_LEVENT)
 209       return;
 210 #endif
 211       sevstr = "debug";
 212       break;
 213     case _EVENT_LOG_MSG:
 214       sevstr = "msg";
 215       break;
 216     case _EVENT_LOG_WARN:
 217       sevstr = "warn";
 218       break;
 219     case _EVENT_LOG_ERR:
 220       sevstr = "err";
 221       break;
 222     default:
 223       sevstr = "???";
 224       break;
 225     }
 226   fprintf (stderr, "libevent: [%s] %s\n", sevstr, msg);
 227 }
 228
 229 static int
 230 _init_evb (struct rs_connection *conn)
 231 {
 232   if (!conn->evb)
 233     {
 234 #if defined (DEBUG)
 235       event_enable_debug_mode ();
 236 #endif
 237       event_set_log_callback (_evlog_cb);
 238       conn->evb = event_base_new ();
 239       if (!conn->evb)
 240         return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
 241                                     "event_base_new");
 242     }
 243   return RSE_OK;
 244 }
 245
 246 static int
 247 _init_socket (struct rs_connection *conn, struct rs_peer *p)
 248 {
 249   if (p->fd != -1)
 250     return RSE_OK;
 251
 252   assert (p->addr);
 253   p->fd = socket (p->addr->ai_family, p->addr->ai_socktype,
 254                   p->addr->ai_protocol);
 255   if (p->fd < 0)
 256     return rs_err_conn_push_fl (conn, RSE_SOME_ERROR, __FILE__, __LINE__,
 257                                 strerror (errno));
 258   if (evutil_make_socket_nonblocking (p->fd) < 0)
 259     {
 260       evutil_closesocket (p->fd);
 261       return rs_err_conn_push_fl (conn, RSE_SOME_ERROR, __FILE__, __LINE__,
 262                                   strerror (errno));
 263     }
 264   return RSE_OK;
 265 }
 266
 267 static struct rs_peer *
 268 _pick_peer (struct rs_connection *conn)
 269 {
 270   if (!conn->active_peer)
 271     conn->active_peer = conn->peers;
 272   return conn->active_peer;
 273 }
 274
 275 static int
 276 _init_bev (struct rs_connection *conn, struct rs_peer *peer)
 277 {
 278   if (!conn->bev)
 279     {
 280       conn->bev = bufferevent_socket_new (conn->evb, peer->fd, 0);
 281       if (!conn->bev)
 282         return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
 283                                     "bufferevent_socket_new");
 284     }
 285   return RSE_OK;
 286 }
 287
 288 static void
 289 _do_connect (struct rs_peer *p)
 290 {
 291   int err;
 292
 293   err = bufferevent_socket_connect (p->conn->bev, p->addr->ai_addr,
 294                                     p->addr->ai_addrlen);
 295   if (err < 0)
 296     rs_err_conn_push_fl (p->conn, RSE_EVENT, __FILE__, __LINE__,
 297                          "bufferevent_socket_connect: %s",
 298                          evutil_gai_strerror(err));
 299   else
 300     p->is_connecting = 1;
 301 }
 302
 303 static int
 304 _conn_open(struct rs_connection *conn, struct rs_packet *pkt)
 305 {
 306   struct rs_peer *p;
 307
 308   if (_init_evb (conn))
 309     return -1;
 310
 311   p = _pick_peer (conn);
 312   if (!p)
 313     return rs_err_conn_push_fl (conn, RSE_NOPEER, __FILE__, __LINE__, NULL);
 314
 315   if (_init_socket (conn, p))
 316     return -1;
 317
 318   if (_init_bev (conn, p))
 319     return -1;
 320
 321   if (!p->is_connected)
 322     if (!p->is_connecting)
 323       _do_connect (p);
 324
 325   return RSE_OK;
 326 }
 327
 328 static int
 329 _conn_is_open_p (struct rs_connection *conn)
 330 {
 331   return conn->active_peer && conn->active_peer->is_connected;
 332 }
 333
 334 /* Public functions.  */
 335 int
 336 rs_packet_create_acc_request (struct rs_connection *conn,
 337                               struct rs_packet **pkt_out,
 338                               const char *user_name, const char *user_pw)
 339 {
 340   struct rs_packet *pkt;
 341   struct rs_attr *attr;
 342
 343   if (_packet_create (conn, pkt_out))
 344     return -1;
 345   pkt = *pkt_out;
 346   pkt->rpkt->code = PW_AUTHENTICATION_REQUEST;
 347
 348   if (user_name)
 349     {
 350       if (rs_attr_create (conn, &attr, "User-Name", user_name))
 351         return -1;
 352       rs_packet_add_attr (pkt, attr);
 353
 354       if (user_pw)
 355         {
 356           if (rs_attr_create (conn, &attr, "User-Password", user_pw))
 357             return -1;
 358           /* FIXME: need this too? rad_pwencode(user_pw, &pwlen, SECRET, reqauth) */
 359           rs_packet_add_attr (pkt, attr);
 360         }
 361     }
 362
 363   return RSE_OK;
 364 }
 365
 366 int
 367 rs_packet_send (struct rs_packet *pkt, void *user_data)
 368 {
 369   struct rs_connection *conn;
 370
 371   assert (pkt);
 372   conn = pkt->conn;
 373
 374   if (_conn_is_open_p (conn))
 375     _do_send (pkt);
 376   else
 377     if (_conn_open (conn, pkt))
 378       return -1;
 379
 380   assert (conn->evb);
 381   assert (conn->bev);
 382   assert (conn->active_peer);
 383   assert (conn->active_peer->fd >= 0);
 384
 385   conn->user_data = user_data;
 386   bufferevent_setcb (conn->bev, _read_cb, _write_cb, _event_cb, pkt);
 387   if (!conn->user_dispatch_flag)
 388     event_base_dispatch (conn->evb);
 389
 390 #if defined (DEBUG)
 391   fprintf (stderr, "%s: event loop done\n", __func__);
 392   assert (event_base_got_break(conn->evb));
 393 #endif
 394
 395   return RSE_OK;
 396 }
 397
 398 int
 399 rs_conn_receive_packet (struct rs_connection *conn, struct rs_packet **pkt_out)
 400 {
 401   struct rs_packet *pkt;
 402
 403   assert (conn);
 404
 405   if (_packet_create (conn, pkt_out))
 406     return -1;
 407   pkt = *pkt_out;
 408   pkt->conn = conn;
 409
 410   if (_conn_open (conn, pkt))
 411     return -1;
 412   assert (conn->evb);
 413   assert (conn->bev);
 414   assert (conn->active_peer);
 415   assert (conn->active_peer->fd >= 0);
 416
 417   bufferevent_setwatermark (conn->bev, EV_READ, RS_HEADER_LEN, 0);
 418   bufferevent_enable (conn->bev, EV_READ);
 419   bufferevent_setcb (conn->bev, _read_cb, _write_cb, _event_cb, pkt);
 420
 421   if (!conn->user_dispatch_flag)
 422     {
 423       event_base_dispatch (conn->evb);
 424 #if defined (DEBUG)
 425       fprintf (stderr, "%s: event loop done", __func__);
 426       if (event_base_got_break(conn->evb))
 427         {
 428           fprintf (stderr, ", got this:\n");
 429           rs_dump_packet (pkt);
 430         }
 431       else
 432         fprintf (stderr, ", no reply\n");
 433 #endif
 434     }
 435
 436   return RSE_OK;
 437 }
 438
 439 void
 440 rs_packet_add_attr(struct rs_packet *pkt, struct rs_attr *attr)
 441 {
 442   pairadd (&pkt->rpkt->vps, attr->vp);
 443   attr->pkt = pkt;
 444 }
 445
 446 struct radius_packet *
 447 rs_packet_frpkt(struct rs_packet *pkt)
 448 {
 449   assert (pkt);
 450   return pkt->rpkt;
 451 }
 452
 453 void
 454 rs_packet_destroy(struct rs_packet *pkt)
 455 {
 456   if (pkt)
 457     {
 458       rad_free (&pkt->rpkt);
 459       rs_free (pkt->conn->ctx, pkt);
 460     }
 461 }