request.c

   1 /* Copyright 2010-2011 NORDUnet A/S. All rights reserved.
   2    See LICENSE for licensing information. */
   3
   4 #if defined HAVE_CONFIG_H
   5 #include <config.h>
   6 #endif
   7
   8 #include <stdint.h>
   9 #include <stdlib.h>
  10 #include <assert.h>
  11 #include <sys/time.h>
  12 #include <event2/event.h>
  13 #include <radsec/radsec.h>
  14 #include <radsec/radsec-impl.h>
  15 #include <radsec/request.h>
  16 #include <radsec/request-impl.h>
  17 #include <radius/client.h>
  18 #include "debug.h"
  19 #include "conn.h"
  20 #include "tcp.h"
  21 #include "udp.h"
  22
  23 /* RFC 5080 2.2.1.  Retransmission Behavior.  */
  24 #define IRT 2
  25 #define MRC 5
  26 #define MRT 16
  27 #define MRD 30
  28 #define RAND 100                /* Rand factor, milliseconds. */
  29
  30 int
  31 rs_request_create (struct rs_connection *conn, struct rs_request **req_out)
  32 {
  33   struct rs_request *req = rs_malloc (conn->ctx, sizeof(*req));
  34   assert (req_out);
  35   if (!req)
  36     return rs_err_conn_push_fl (conn, RSE_NOMEM, __FILE__, __LINE__, NULL);
  37   memset (req, 0, sizeof(*req));
  38   req->conn = conn;
  39   *req_out = req;
  40   return RSE_OK;
  41 }
  42
  43 void
  44 rs_request_add_reqpkt (struct rs_request *req, struct rs_packet *req_msg)
  45 {
  46   assert (req);
  47   req->req_msg = req_msg;
  48 }
  49
  50 int
  51 rs_request_create_authn (struct rs_connection *conn,
  52                          struct rs_request **req_out,
  53                          const char *user_name,
  54                          const char *user_pw)
  55 {
  56   struct rs_request *req = NULL;
  57   assert (req_out);
  58
  59   if (rs_request_create (conn, &req))
  60     return -1;
  61
  62   if (rs_packet_create_authn_request (conn, &req->req_msg, user_name, user_pw))
  63     return -1;
  64
  65   if (req_out)
  66     *req_out = req;
  67   return RSE_OK;
  68 }
  69
  70 void
  71 rs_request_destroy (struct rs_request *request)
  72 {
  73   assert (request);
  74   assert (request->conn);
  75   assert (request->conn->ctx);
  76
  77   if (request->req_msg)
  78     rs_packet_destroy (request->req_msg);
  79   rs_free (request->conn->ctx, request);
  80 }
  81
  82 static void
  83 _rand_rt (struct timeval *res, uint32_t rtprev, uint32_t factor)
  84 {
  85   uint32_t ms = rtprev * (nr_rand () % factor);
  86   res->tv_sec = rtprev + ms / 1000;
  87   res->tv_usec = (ms % 1000) * 1000;
  88 }
  89
  90 int
  91 rs_request_send (struct rs_request *request, struct rs_packet **resp_msg)
  92 {
  93   int r = 0;
  94   struct rs_connection *conn = NULL;
  95   int count = 0;
  96   struct timeval rt = {0,0};
  97   struct timeval end = {0,0};
  98   struct timeval now = {0,0};
  99   struct timeval tmp_tv = {0,0};
 100   const struct timeval mrt_tv = {MRT,0};
 101
 102   if (!request || !request->conn || !request->req_msg || !resp_msg)
 103     return rs_err_conn_push_fl (conn, RSE_INVAL, __FILE__, __LINE__, NULL);
 104   conn = request->conn;
 105   assert (!conn_user_dispatch_p (conn)); /* This function is high level.  */
 106
 107   gettimeofday (&end, NULL);
 108   end.tv_sec += MRD;
 109   _rand_rt (&rt, IRT, RAND);
 110   while (1)
 111     {
 112       rs_conn_set_timeout (conn, &rt);
 113
 114       r = rs_packet_send (request->req_msg, NULL);
 115       if (r == RSE_OK)
 116         {
 117           r = rs_conn_receive_packet (request->conn,
 118                                       request->req_msg,
 119                                       resp_msg);
 120           if (r == RSE_OK)
 121             break;              /* Success.  */
 122         }
 123       if (r != RSE_TIMEOUT_CONN && r != RSE_TIMEOUT_IO)
 124         break;                  /* Error.  */
 125
 126       /* Timing out reading or writing. Pop the timeout error from the
 127          stack and continue the loop. */
 128       rs_err_conn_pop (request->conn);
 129
 130       gettimeofday (&now, NULL);
 131       if (++count > MRC || timercmp (&now, &end, >))
 132         {
 133           r = rs_err_conn_push_fl (request->conn, RSE_TIMEOUT,
 134                                    __FILE__, __LINE__, NULL);
 135           break;                /* Timeout.  */
 136         }
 137
 138       /* rt = 2 * rt + rand_rt (rt, RAND); */
 139       timeradd (&rt, &rt, &rt);
 140       _rand_rt (&tmp_tv, IRT, RAND);
 141       timeradd (&rt, &tmp_tv, &rt);
 142       if (timercmp (&rt, &mrt_tv, >))
 143         _rand_rt (&rt, MRT, RAND);
 144     }
 145
 146   timerclear (&rt);
 147   rs_conn_set_timeout (conn, &rt);
 148
 149   rs_debug (("%s: returning %d\n", __func__, r));
 150   return r;
 151 }
 152
 153 struct rs_packet *
 154 rs_request_get_reqmsg (const struct rs_request *request)
 155 {
 156   assert (request);
 157   return request->req_msg;
 158 }