From 7764466998e490d7b2e365b74b87330ea913be2e Mon Sep 17 00:00:00 2001
From: Max Lv <max.c.lv@gmail.com>
Date: Sun, 12 Jul 2015 17:09:56 +0800
Subject: [PATCH] fix the timer

---
 src/udprelay.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/udprelay.c b/src/udprelay.c
index 6d3a0475..8d1172b6 100644
--- a/src/udprelay.c
+++ b/src/udprelay.c
@@ -518,6 +518,12 @@ static void query_resolve_cb(struct sockaddr *addr, void *data)
         struct remote_ctx *remote_ctx = query_ctx->remote_ctx;
         int cache_hit = 0;
 
+        // Lookup in the conn cache
+        if (remote_ctx == NULL) {
+            char *key = hash_key(0, &query_ctx->src_addr);
+            cache_lookup(conn_cache, key, (void *)&remote_ctx);
+        }
+
         if (remote_ctx == NULL) {
             int remotefd = create_remote_socket(addr->sa_family == AF_INET6);
             if (remotefd != -1) {
@@ -567,8 +573,8 @@ static void query_resolve_cb(struct sockaddr *addr, void *data)
                     close_and_free_remote(EV_A_ remote_ctx);
                 }
             } else {
-                // Add to conn cache
                 if (!cache_hit) {
+                    // Add to conn cache
                     char *key = hash_key(0, &remote_ctx->src_addr);
                     cache_insert(query_ctx->server_ctx->conn_cache, key,
                             (void *)remote_ctx);
@@ -945,6 +951,11 @@ static void server_recv_cb(EV_P_ ev_io *w, int revents)
         }
     }
 
+    // reset the timer
+    if (remote_ctx != NULL) {
+        ev_timer_again(EV_A_ & remote_ctx->watcher);
+    }
+
     if (remote_ctx == NULL) {
         if (verbose) {
 #ifdef UDPRELAY_REDIR