v1.4| 增强unixScoket稳定性

src/components/linuxsocket/unix_socket.cpp

@@ -2,36 +2,71 @@
 using namespace iflytop;
 using namespace std;
 using namespace core;
-
-void UnixScoket::start() {
-  logger->info("Creating UNIX domain socket at path: {}", m_path);
-  logger->info("Connecting to opposite path: {}", m_opposite_path);
-
+int UnixScoket::initRxSocket() {
   sock_fd = socket(AF_UNIX, SOCK_DGRAM, 0);
   if (sock_fd == -1) {
     logger->error("socket creation failed: {}", strerror(errno));
-    exit(EXIT_FAILURE);
+    return -1;
   }
 
   // Remove socket file if it already exists
   unlink(m_path.c_str());
   if (bind(sock_fd, (struct sockaddr *)&rx_addr, sizeof(rx_addr)) == -1) {
-    perror("bind");
+    logger->error("bind failed: {}", strerror(errno));
     close(sock_fd);
-    exit(EXIT_FAILURE);
+    sock_fd = -1;
+    return -1;
   }
 
+  // 设置超时时间为5秒
+  struct timeval tv;
+  tv.tv_sec  = 1;  // 秒
+  tv.tv_usec = 0;  // 微秒
+
+  // 设置SO_RCVTIMEO选项
+  if (setsockopt(sock_fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) {
+    logger->error("setsockopt failed: {}", strerror(errno));
+    close(sock_fd);
+    sock_fd = -1;
+    return -1;
+  }
+  return 0;
+}
+void UnixScoket::start() {
+  logger->info("Creating UNIX domain socket at path: {}", m_path);
+  logger->info("Connecting to opposite path: {}", m_opposite_path);
+
+  initRxSocket();
+
   m_thread.reset(new Thread("UnixSocketThread", [this]() {
     uint8_t            rxbuffer[1024];
     struct sockaddr_un client_addr;
     ThisThread         thisThread;
     while (!thisThread.getExitFlag()) {
+      if (sock_fd < 0) {
+        usleep(100 * 1000);
+        logger->error("Socket is closed, reinitializing...");
+        initRxSocket();
+        continue;
+      }
+
       memset(rxbuffer, 0, sizeof(rxbuffer));
       socklen_t client_len = sizeof(client_addr);
       ssize_t   num_bytes  = recvfrom(sock_fd, rxbuffer, sizeof(rxbuffer) - 1, 0, (struct sockaddr *)&client_addr, &client_len);
       if (num_bytes > 0) {
         onPacket(rxbuffer, num_bytes);
       }
+      if (num_bytes < 0) {
+        if (errno == EAGAIN || errno == EWOULDBLOCK) {  // 超时错误
+          logger->info("recvfrom timed out, no data received");
+          continue;
+        } else {
+          logger->error("recvfrom failed: {}", strerror(errno));
+          close(sock_fd);
+          sock_fd = -1;
+          continue;
+        }
+      }
     }
     close(sock_fd);
     unlink(m_path.c_str());

src/components/linuxsocket/unix_socket.hpp

@@ -64,5 +64,6 @@ class UnixScoket {
 
   void start();
   void sendPacket(uint8_t* data, size_t len);
+  int initRxSocket();
 };
 }  // namespace iflytop

src/configs/version.hpp

@@ -1,2 +1,2 @@
 #pragma once
-#define VERSION "1.3"
+#define VERSION "1.4"

src/main.cpp

@@ -26,10 +26,29 @@ int main(int argc, char* argv[]) {
   mainLogger->info("#  company:{}", "ifytop");
   mainLogger->info("#  version:{}", VERSION);
   mainLogger->info("#");
+
     App app;
   app.initialize();
 
   while (true) {
     sleep(1);
   }
+
+  // unique_ptr<UnixScoket> server;
+  // server.reset(new UnixScoket(UnixScoket::Server, "zexcan"));
+  // server->start();
+  // server->onPacket.connect([mainLogger](uint8_t* data, size_t len) {
+  //   if (len > 0) {
+  //     mainLogger->info("Received packet: {}", string((char*)data, len));
+  //   }
+  // });
+
+  // unique_ptr<UnixScoket> client;
+  // client.reset(new UnixScoket(UnixScoket::Client, "zexcan"));
+  // client->start();
+
+  // while (true) {
+  //   client->sendPacket((uint8_t*)"hello", 5);
+  //   sleep(1);
+  // }
 }

test/auto.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+# 循环指令./sender 随即休眠0.1,0.2,0.3 后杀死 进程
+while true; do
+    ./sender &
+    sleep $(printf "0.%03d" $(( RANDOM % 100 )))
+    killall sender
+    echo "Killed sender process"
+done

test/receiver.c

@@ -1,9 +1,10 @@
+#include <errno.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <unistd.h>
 #include <sys/socket.h>
 #include <sys/un.h>
+#include <unistd.h>
 
 #define SOCKET_PATH "/tmp/unix_dgram_demo"
 
@@ -11,7 +12,7 @@ int main() {
   int                sock_fd;
   struct sockaddr_un server_addr, client_addr;
   socklen_t          client_len;
-    char buffer[256];
+  char               buffer[10240];
 
   // Create socket
   sock_fd = socket(AF_UNIX, SOCK_DGRAM, 0);
@@ -35,23 +36,34 @@ int main() {
     exit(EXIT_FAILURE);
   }
 
+  // 设置超时时间为5秒
+  struct timeval tv;
+  tv.tv_sec  = 1;  // 秒
+  tv.tv_usec = 0;  // 微秒
+
+  // 设置SO_RCVTIMEO选项
+  if (setsockopt(sock_fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) {
+    perror("setsockopt failed");
+    return -1;
+  }
+
   printf("Receiver is waiting for messages on %s...\n", SOCKET_PATH);
 
+  while (1) {
     // Receive message
     client_len        = sizeof(client_addr);
-    ssize_t num_bytes = recvfrom(sock_fd, buffer, sizeof(buffer) - 1, 0,
-                                (struct sockaddr *)&client_addr, &client_len);
+    ssize_t num_bytes = recvfrom(sock_fd, buffer, sizeof(buffer) - 1, 0, (struct sockaddr *)&client_addr, &client_len);
     if (num_bytes == -1) {
-        perror("recvfrom");
+      if (errno == EAGAIN || errno == EWOULDBLOCK) {  // 超时错误
+        printf("No message received within the timeout period.\n");
+        continue;
+      }
+
+      perror("!!!recvfrom");
+      exit(EXIT_FAILURE);
     } else {
       buffer[num_bytes] = '\0';
-        printf("Received message: %s\n", buffer);
-        
-        // Send response
-        const char *response = "Hello from receiver!";
-        if (sendto(sock_fd, response, strlen(response), 0,
-                  (struct sockaddr *)&client_addr, client_len) == -1) {
-            perror("sendto");
+      printf("Received message: %d\n", num_bytes);
     }
   }
 

test/sender.c

@@ -1,9 +1,9 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <unistd.h>
 #include <sys/socket.h>
 #include <sys/un.h>
+#include <unistd.h>
 
 #define SOCKET_PATH "/tmp/unix_dgram_demo"
 #define SENDER_PATH "/tmp/unix_dgram_sender"
@@ -41,20 +41,24 @@ int main() {
   strncpy(server_addr.sun_path, SOCKET_PATH, sizeof(server_addr.sun_path) - 1);
 
   // Send message to server
-    const char *message = "Hello from sender!";
-    if (sendto(sock_fd, message, strlen(message), 0,
-              (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
+  char txbuf[10240] = {0};
+  for (int i = 0; i < sizeof(txbuf) - 1; i++) {
+    txbuf[i] = 'a' + (i % 26);  // Fill with a-z characters
+  }
+
+  while (1) {
+    if (sendto(sock_fd, txbuf, 10240, 0, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
       perror("sendto");
       close(sock_fd);
       exit(EXIT_FAILURE);
     }
+  }
 
   printf("Message sent to receiver.\n");
 
   // Receive response
   socklen_t server_len = sizeof(server_addr);
-    ssize_t num_bytes = recvfrom(sock_fd, buffer, sizeof(buffer) - 1, 0,
-                                (struct sockaddr *)&server_addr, &server_len);
+  ssize_t   num_bytes  = recvfrom(sock_fd, buffer, sizeof(buffer) - 1, 0, (struct sockaddr *)&server_addr, &server_len);
   if (num_bytes == -1) {
     perror("recvfrom");
   } else {