fix: features and enhancements — issues #70, #36, #34, #33, #32, #57, #40, #37

src/shared/protocol.c

@@ -34,11 +34,17 @@ static void bw_throttle(size_t bytes_written) {
   struct timespec now;
   clock_gettime(CLOCK_MONOTONIC, &now);
 
-  long long elapsed_ns =
-      (now.tv_sec - bw_last_refill.tv_sec) * 1000000000LL + (now.tv_nsec - bw_last_refill.tv_nsec);
+  /* Use unsigned long long for intermediate computation to avoid overflow.
+   * sec_diff * 1000000000ULL could overflow signed 64-bit for large deltas;
+   * clamp to a safe maximum. */
+  unsigned long long sec_diff = (unsigned long long)(now.tv_sec - bw_last_refill.tv_sec);
+  if (sec_diff > 9223372036ULL)
+    sec_diff = 9223372036ULL;
+  unsigned long long elapsed_ns =
+      sec_diff * 1000000000ULL + (unsigned long long)(now.tv_nsec - bw_last_refill.tv_nsec);
   bw_last_refill = now;
 
-  long long tokens_to_add = (long long)((double)io_bwlimit * elapsed_ns / 1000000000.0);
+  long long tokens_to_add = (long long)((double)io_bwlimit * (double)elapsed_ns / 1000000000.0);
   bw_tokens += tokens_to_add;
   if (bw_tokens > (long long)io_bwlimit)
     bw_tokens = (long long)io_bwlimit;
@@ -74,13 +80,23 @@ bool send_n_data(int file_descriptor, const void* data, size_t data_size) {
     if (io_bwlimit > 0 && chunk > 65536)
       chunk = 65536;
     ssize_t bytes_send;
-    if (io_ssl)
+    if (io_ssl) {
       bytes_send = SSL_write(io_ssl, (const char*)data + total_bytes_send, chunk);
-    else
+      if (bytes_send <= 0) {
+        int err = SSL_get_error(io_ssl, (int)bytes_send);
+        if (err == SSL_ERROR_WANT_WRITE || err == SSL_ERROR_WANT_READ) {
+          /* Non-fatal: retry without counting progress */
+          continue;
+        }
+        log_message(LOG_LEVEL_ERROR, "Could not send data (SSL error: %d)", err);
+        return false;
+      }
+    } else {
       bytes_send = write(fd, (const char*)data + total_bytes_send, chunk);
-    if (bytes_send <= 0) {
-      log_message(LOG_LEVEL_ERROR, "Could not send data");
-      return false;
+      if (bytes_send <= 0) {
+        log_message(LOG_LEVEL_ERROR, "Could not send data");
+        return false;
+      }
     }
     bw_throttle((size_t)bytes_send);
     total_bytes_send += bytes_send;
@@ -95,18 +111,31 @@ bool receive_n_data(int file_descriptor, void* data, size_t data_size) {
   size_t total_bytes_received = 0;
   while (total_bytes_received < data_size) {
     ssize_t bytes_received;
-    if (io_ssl)
+    if (io_ssl) {
       bytes_received =
           SSL_read(io_ssl, (char*)data + total_bytes_received, data_size - total_bytes_received);
-    else
+      if (bytes_received <= 0) {
+        int err = SSL_get_error(io_ssl, (int)bytes_received);
+        if (err == SSL_ERROR_WANT_READ || err == SSL_ERROR_WANT_WRITE) {
+          /* Non-fatal: retry without counting progress */
+          continue;
+        }
+        if (bytes_received == 0)
+          log_message(LOG_LEVEL_ERROR, "Connection closed while receiving data");
+        else
+          log_message(LOG_LEVEL_ERROR, "Could not receive bytes (SSL error: %d)", err);
+        return false;
+      }
+    } else {
       bytes_received =
           read(fd, (char*)data + total_bytes_received, data_size - total_bytes_received);
-    if (bytes_received <= 0) {
-      if (bytes_received == 0)
-        log_message(LOG_LEVEL_ERROR, "Connection closed while receiving data");
-      else
-        log_message(LOG_LEVEL_ERROR, "Could not receive bytes");
-      return false;
+      if (bytes_received <= 0) {
+        if (bytes_received == 0)
+          log_message(LOG_LEVEL_ERROR, "Connection closed while receiving data");
+        else
+          log_message(LOG_LEVEL_ERROR, "Could not receive bytes");
+        return false;
+      }
     }
     total_bytes_received += bytes_received;
   }
@@ -151,6 +180,10 @@ char* receive_str(int file_descriptor) {
   size_t size;
   if (!receive_n_data(file_descriptor, &size, sizeof(size_t)))
     return NULL;
+  if (size > MAX_STRING_SIZE) {
+    log_message(LOG_LEVEL_ERROR, "String size %zu exceeds maximum %d", size, MAX_STRING_SIZE);
+    return NULL;
+  }
   char* data = (char*)malloc(size + 1);
   if (data == NULL)
     return NULL;