task-manager: Add helper function to calculate the total retransmit timeout

diff --git a/src/libcharon/sa/task_manager.c b/src/libcharon/sa/task_manager.c
index c42008ba9144a318f824386504a706610b2b1642..bd1191406c735252ab24e8954c27dce885d0ce1f 100644 (file)
--- a/src/libcharon/sa/task_manager.c
+++ b/src/libcharon/sa/task_manager.c
@@ -15,10 +15,40 @@
 
 #include "task_manager.h"
 
+#include <math.h>
 #include <sa/ikev1/task_manager_v1.h>
 #include <sa/ikev2/task_manager_v2.h>
 
-/**
+/*
+ * See header
+ */
+u_int task_manager_total_retransmit_timeout()
+{
+       double timeout, base, limit = 0, total = 0;
+       int tries, i;
+
+       tries = lib->settings->get_int(lib->settings, "%s.retransmit_tries",
+                                                                  RETRANSMIT_TRIES, lib->ns);
+       base = lib->settings->get_double(lib->settings, "%s.retransmit_base",
+                                                                        RETRANSMIT_BASE, lib->ns);
+       timeout = lib->settings->get_double(lib->settings, "%s.retransmit_timeout",
+                                                                               RETRANSMIT_TIMEOUT, lib->ns);
+       limit = lib->settings->get_double(lib->settings, "%s.retransmit_limit",
+                                                                         0, lib->ns);
+
+       for (i = 0; i <= tries; i++)
+       {
+               double interval = timeout * pow(base, i);
+               if (limit)
+               {
+                       interval = min(interval, limit);
+               }
+               total += interval;
+       }
+       return (u_int)total;
+}
+
+/*
  * See header
  */
 task_manager_t *task_manager_create(ike_sa_t *ike_sa)

diff --git a/src/libcharon/sa/task_manager.h b/src/libcharon/sa/task_manager.h
index 4d3f9bef552c2b4728213e6cacc33dc9e761b0b8..e3fddf39bdf92ad290eada20e857984135baedd1 100644 (file)
--- a/src/libcharon/sa/task_manager.h
+++ b/src/libcharon/sa/task_manager.h
@@ -302,6 +302,17 @@ struct task_manager_t {
        void (*destroy) (task_manager_t *this);
 };
 
+/**
+ * Calculate total timeout of the retransmission mechanism.
+ *
+ * This is affected by modifications of retransmit_base, retransmit_timeout,
+ * retransmit_limit or retransmit_tries. The resulting value can then be used
+ * e.g. in kernel plugins to set the system's acquire timeout properly.
+ *
+ * @return                                     calculated total retransmission timeout in seconds
+ */
+u_int task_manager_total_retransmit_timeout();
+
 /**
  * Create a task manager instance for the correct IKE version.
  *