From 122b6830b4fe9d015875f88b3eb06fda2fe1975f Mon Sep 17 00:00:00 2001
From: Gustavo Sverzut Barbieri <barbieri@profusion.mobi>
Date: Mon, 12 Sep 2016 14:48:03 -0300
Subject: [PATCH] efl_net_dialer_tcp: add connect timeout.

---
 src/lib/ecore_con/efl_net_dialer_tcp.c  | 79 ++++++++++++++++++++++++-
 src/lib/ecore_con/efl_net_dialer_tcp.eo |  1 +
 2 files changed, 79 insertions(+), 1 deletion(-)

diff --git a/src/lib/ecore_con/efl_net_dialer_tcp.c b/src/lib/ecore_con/efl_net_dialer_tcp.c
index c5ce629362..beb078c7ff 100644
--- a/src/lib/ecore_con/efl_net_dialer_tcp.c
+++ b/src/lib/ecore_con/efl_net_dialer_tcp.c
@@ -39,6 +39,7 @@ typedef struct _Efl_Net_Dialer_Tcp_Data
    } resolve;
    struct {
       Ecore_Thread *thread;
+      Eo *timer;
    } connect;
    Eina_Stringshare *address_dial;
    Eina_Stringshare *proxy;
@@ -47,6 +48,16 @@ typedef struct _Efl_Net_Dialer_Tcp_Data
    double timeout_dial;
 } Efl_Net_Dialer_Tcp_Data;
 
+EOLIAN static Eo*
+_efl_net_dialer_tcp_efl_object_constructor(Eo *o, Efl_Net_Dialer_Tcp_Data *pd EINA_UNUSED)
+{
+   o = efl_constructor(efl_super(o, MY_CLASS));
+   if (!o) return NULL;
+
+   efl_net_dialer_timeout_dial_set(o, 30.0);
+   return o;
+}
+
 EOLIAN static void
 _efl_net_dialer_tcp_efl_object_destructor(Eo *o, Efl_Net_Dialer_Tcp_Data *pd)
 {
@@ -54,6 +65,12 @@ _efl_net_dialer_tcp_efl_object_destructor(Eo *o, Efl_Net_Dialer_Tcp_Data *pd)
        (!efl_io_closer_closed_get(o)))
      efl_io_closer_close(o);
 
+   if (pd->connect.timer)
+     {
+        efl_del(pd->connect.timer);
+        pd->connect.timer = NULL;
+     }
+
    if (pd->connect.thread)
      {
         ecore_thread_cancel(pd->connect.thread);
@@ -185,6 +202,39 @@ _efl_net_dialer_tcp_resolved(void *data, const char *host EINA_UNUSED, const cha
    _efl_net_dialer_tcp_connect(o, pd);
 }
 
+static void
+_efl_net_dialer_tcp_connect_timeout(void *data, const Efl_Event *event EINA_UNUSED)
+{
+   Eo *o = data;
+   Efl_Net_Dialer_Tcp_Data *pd = efl_data_scope_get(o, MY_CLASS);
+   Eina_Error err = ETIMEDOUT;
+
+   pd->connect.timer = NULL;
+
+   if (pd->resolve.thread)
+     {
+        ecore_thread_cancel(pd->resolve.thread);
+        pd->resolve.thread = NULL;
+     }
+   if (pd->resolve.names)
+     {
+        freeaddrinfo(pd->resolve.names);
+        pd->resolve.names = NULL;
+     }
+   pd->resolve.current = NULL;
+
+   if (pd->connect.thread)
+     {
+        ecore_thread_cancel(pd->connect.thread);
+        pd->connect.thread = NULL;
+     }
+
+   efl_ref(o);
+   efl_io_reader_eos_set(o, EINA_TRUE);
+   efl_event_callback_call(o, EFL_NET_DIALER_EVENT_ERROR, &err);
+   efl_unref(o);
+}
+
 EOLIAN static Eina_Error
 _efl_net_dialer_tcp_efl_net_dialer_dial(Eo *o, Efl_Net_Dialer_Tcp_Data *pd EINA_UNUSED, const char *address)
 {
@@ -234,6 +284,18 @@ _efl_net_dialer_tcp_efl_net_dialer_dial(Eo *o, Efl_Net_Dialer_Tcp_Data *pd EINA_
 
    efl_net_dialer_address_dial_set(o, address);
 
+   if (pd->connect.timer)
+     {
+        efl_del(pd->connect.timer);
+        pd->connect.timer = NULL;
+     }
+   if (pd->timeout_dial > 0.0)
+     {
+        pd->connect.timer = efl_add(EFL_LOOP_TIMER_CLASS, efl_loop_user_loop_get(o),
+                                    efl_loop_timer_interval_set(efl_added, pd->timeout_dial),
+                                    efl_event_callback_add(efl_added, EFL_LOOP_TIMER_EVENT_TICK, _efl_net_dialer_tcp_connect_timeout, o));
+     }
+
    return 0;
 }
 
@@ -265,8 +327,18 @@ _efl_net_dialer_tcp_efl_net_dialer_proxy_get(Eo *o EINA_UNUSED, Efl_Net_Dialer_T
 EOLIAN static void
 _efl_net_dialer_tcp_efl_net_dialer_timeout_dial_set(Eo *o EINA_UNUSED, Efl_Net_Dialer_Tcp_Data *pd, double seconds)
 {
-   // TODO: when using ecore_con_info/threads, set timeout
    pd->timeout_dial = seconds;
+   if (pd->connect.timer)
+     {
+        efl_del(pd->connect.timer);
+        pd->connect.timer = NULL;
+     }
+   if (pd->timeout_dial > 0.0)
+     {
+        pd->connect.timer = efl_add(EFL_LOOP_TIMER_CLASS, efl_loop_user_loop_get(o),
+                                    efl_loop_timer_interval_set(efl_added, pd->timeout_dial),
+                                    efl_event_callback_add(efl_added, EFL_LOOP_TIMER_EVENT_TICK, _efl_net_dialer_tcp_connect_timeout, o));
+     }
 }
 
 EOLIAN static double
@@ -278,6 +350,11 @@ _efl_net_dialer_tcp_efl_net_dialer_timeout_dial_get(Eo *o EINA_UNUSED, Efl_Net_D
 EOLIAN static void
 _efl_net_dialer_tcp_efl_net_dialer_connected_set(Eo *o, Efl_Net_Dialer_Tcp_Data *pd, Eina_Bool connected)
 {
+   if (pd->connect.timer)
+     {
+        efl_del(pd->connect.timer);
+        pd->connect.timer = NULL;
+     }
    if (pd->connected == connected) return;
    pd->connected = connected;
    if (connected) efl_event_callback_call(o, EFL_NET_DIALER_EVENT_CONNECTED, NULL);
diff --git a/src/lib/ecore_con/efl_net_dialer_tcp.eo b/src/lib/ecore_con/efl_net_dialer_tcp.eo
index ba2497d10f..e3f7ec6df2 100644
--- a/src/lib/ecore_con/efl_net_dialer_tcp.eo
+++ b/src/lib/ecore_con/efl_net_dialer_tcp.eo
@@ -5,6 +5,7 @@ class Efl.Net.Dialer.Tcp (Efl.Net.Socket.Tcp, Efl.Net.Dialer) {
     ]]
 
     implements {
+        Efl.Object.constructor;
         Efl.Object.destructor;
         Efl.Net.Dialer.dial;
         Efl.Net.Dialer.address_dial;