From 48aad66df0ba07f153b88587aac00830f3913d40 Mon Sep 17 00:00:00 2001
From: Artem Prilutskiy <performer@aesyle>
Date: Wed, 7 Jun 2017 16:39:10 +0300
Subject: [PATCH] Fixes

---
 RewindClient.c | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/RewindClient.c b/RewindClient.c
index 22c27b0..6b25e8c 100644
--- a/RewindClient.c
+++ b/RewindClient.c
@@ -208,7 +208,8 @@ int ConnectRewindClient(struct RewindContext* context, const char* location, con
   ssize_t length;
 
   size_t attempt = 0;
-  time_t threshold = time(NULL) + CONNECT_TIMEOUT;
+  struct timeval now;
+  struct timeval threshold;
 
   uint8_t* digest = (uint8_t*)alloca(SHA256_DIGEST_LENGTH);
 
@@ -238,11 +239,17 @@ int ConnectRewindClient(struct RewindContext* context, const char* location, con
 
   // Do login procedure
 
-  while (threshold > time(NULL))
+  gettimeofday(&now, NULL);
+  threshold.tv_sec  = now.tv_sec + CONNECT_TIMEOUT;
+  threshold.tv_usec = now.tv_usec; 
+
+  while (timercmp(&now, &threshold, <))
   {
     TransmitRewindData(context, REWIND_TYPE_KEEP_ALIVE, REWIND_FLAG_NONE, context->data, context->length);
     length = ReceiveRewindData(context, buffer, BUFFER_SIZE);
 
+    gettimeofday(&now, NULL);
+
     if ((length == CLIENT_ERROR_WRONG_ADDRESS) ||
         (length == CLIENT_ERROR_SOCKET_IO) &&
         ((errno == EWOULDBLOCK) ||
@@ -292,16 +299,19 @@ int WaitForRewindSessionEnd(struct RewindContext* context, struct RewindSessionP
   struct timeval now;
   struct timeval threshold;
 
-  gettimeofday(&threshold, NULL);
-  threshold.tv_sec += interval;
+  gettimeofday(&now, NULL);
+  threshold.tv_sec  = now.tv_sec + CONNECT_TIMEOUT;
+  threshold.tv_usec = now.tv_usec; 
 
-  do
+  while (timercmp(&now, &threshold, <))
   {
     TransmitRewindData(context, REWIND_TYPE_KEEP_ALIVE, REWIND_FLAG_NONE, context->data, context->length);
     TransmitRewindData(context, REWIND_TYPE_SESSION_POLL, REWIND_FLAG_NONE, request, sizeof(struct RewindSessionPollData));
 
     length = ReceiveRewindData(context, buffer, BUFFER_SIZE);
 
+    gettimeofday(&now, NULL);
+
     if ((length == CLIENT_ERROR_WRONG_ADDRESS) ||
         (length == CLIENT_ERROR_SOCKET_IO) &&
         ((errno == EWOULDBLOCK) ||
@@ -334,10 +344,7 @@ int WaitForRewindSessionEnd(struct RewindContext* context, struct RewindSessionP
       sleep(RECEIVE_TIMEOUT);
       state = 0b00;
     }
-
-    gettimeofday(&now, NULL);
   }
-  while (timercmp(&now, &threshold, <));
 
   return CLIENT_ERROR_RESPONSE_TIMEOUT;
 }