C 语言

关注公众号 jb51net

关闭
首页 > 软件编程 > C 语言 > C++ socket通信

C++ socket通信遇到的问题及解决方法

作者:BoBo玩ROS

这篇文章主要介绍了C++ socket通信遇到的问题,通过代码修改来解决这个问题,本文结合实例代码给大家介绍的非常详细,需要的朋友可以参考下

为了解决这两个问题,你可以对代码进行以下修改:

1.处理socket断开的情况:

在recv函数返回0的判断分支中,你可以在断开连接时进行一些处理,例如打印提示信息或进行重连操作。

if (ret == 0) {
  printf("Remote socket close!!!\n");
  // 处理socket断开,例如进行重连操作
  Reconnect();
  break;
}

你可以在Reconnect函数中实现重新连接的逻辑。

2.处理socket连接没有发送消息的情况:

在接收数据失败的判断分支中,你可以添加一个超时机制,如果在一定时间内没有接收到数据,就认为连接没有发送消息。

if (ret < 0) {
  if ((errno == EAGAIN) || (errno == EWOULDBLOCK) || (errno == EINTR)) {
    std::this_thread::sleep_for(std::chrono::microseconds(10));
    continue;
  }
  if (errno == EWOULDBLOCK || errno == EAGAIN) {
    // 没有数据可读,可以继续执行其他任务
    printf("Failed to receive data!!!\n");
    // 处理连接没有发送消息的情况,例如重新发送消息或断开连接
    HandleNoMessage();
    continue;
  } else {
    printf("Remote socket error!!!\n");
    break;
  }
}

你可以在HandleNoMessage函数中实现处理连接没有发送消息的逻辑。

请注意,以上只是一种修改代码的思路,具体的实现方式可能需要根据你的需求和业务逻辑来进行调整。

要将这段C++代码改为非阻塞执行,需要使用非阻塞的套接字操作和事件循环机制。下面是修改后的代码:

void NtripServer::ThreadHandler(void) {
  int ret;
  char recv_buffer[1024] = {};
  thread_is_running_ = true;
  // 将socket_fd_设置为非阻塞模式
  if (SetNonBlocking(socket_fd_) == -1) {
    printf("Failed to set socket to non-blocking mode\n");
    return;
  }
  while (thread_is_running_) {
    ret = recv(socket_fd_, recv_buffer, sizeof(recv_buffer), 0);
    if (ret == 0) {
      printf("Remote socket close!!!\n");
      break;
    } else if (ret < 0) {
      if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) {
        printf("Remote socket error!!!\n");
        break;
      }
    } else {
      // 处理接收到的数据
    }
    std::this_thread::sleep_for(std::chrono::milliseconds(10));
  }
  close(socket_fd_);
  socket_fd_ = -1;
  thread_is_running_ = false;
  service_is_running_ = false;
}

在修改后的代码中,我们首先调用SetNonBlocking函数将socket_fd_设置为非阻塞模式。然后,在while循环中,我们使用recv函数接收数据,并根据返回值进行处理。

如果recv返回值为0,表示远程socket关闭,我们打印相应的信息并跳出循环。

如果recv返回值小于0,我们首先判断errno是否为EAGAIN、EWOULDBLOCK或EINTR。如果不是这些错误码,表示发生了其他错误,我们打印相应的信息并跳出循环。

如果recv返回值大于0,表示成功接收到数据,我们可以在代码中添加处理接收到的数据的逻辑。

在每次循环结束后,我们使用std::this_thread::sleep_for函数暂停10毫秒,以避免忙等待。

最后,在循环结束后,我们关闭socket_fd_并将其置为-1,同时将thread_is_running_和service_is_running_标志置为false,表示线程已停止。

是的,SetNonBlocking函数需要自定义。下面是一个示例的SetNonBlocking函数的实现:

int SetNonBlocking(int socket_fd) {
  int flags = fcntl(socket_fd, F_GETFL, 0);
  if (flags == -1) {
    return -1;
  }
  if (fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK) == -1) {
    return -1;
  }
  return 0;
}

这个函数的实现与之前提到的设置socket为非阻塞模式的代码相同。它首先使用fcntl函数获取socket_fd的文件描述符标志,并将返回值保存在变量flags中。如果获取失败,函数返回-1。

然后,它使用fcntl函数的F_SETFL命令来修改socket_fd的文件描述符标志。修改的方式是将O_NONBLOCK标志添加到原有标志中,使用位运算符|进行操作。如果修改失败,函数返回-1。

如果设置成功,函数返回0。

下面是修改后的完整代码示例:

#include <iostream>
#include <thread>
#include <chrono>
#include <cstring>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
class NtripServer {
public:
  NtripServer() : socket_fd_(-1), thread_is_running_(false), service_is_running_(false) {}
  ~NtripServer() {}
  void Start() {
    if (service_is_running_) {
      std::cout << "Service is already running!" << std::endl;
      return;
    }
    // 创建socket
    socket_fd_ = socket(AF_INET, SOCK_STREAM, 0);
    if (socket_fd_ == -1) {
      std::cout << "Failed to create socket!" << std::endl;
      return;
    }
    // 设置socket选项
    int optval = 1;
    setsockopt(socket_fd_, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
    // 绑定地址和端口
    sockaddr_in server_addr{};
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(1234);
    server_addr.sin_addr.s_addr = INADDR_ANY;
    if (bind(socket_fd_, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
      std::cout << "Failed to bind address!" << std::endl;
      close(socket_fd_);
      socket_fd_ = -1;
      return;
    }
    // 监听连接
    if (listen(socket_fd_, 5) == -1) {
      std::cout << "Failed to listen!" << std::endl;
      close(socket_fd_);
      socket_fd_ = -1;
      return;
    }
    // 启动线程处理连接
    service_is_running_ = true;
    std::thread thread(&NtripServer::ThreadHandler, this);
    thread.detach();
  }
  void Stop() {
    if (!service_is_running_) {
      std::cout << "Service is not running!" << std::endl;
      return;
    }
    service_is_running_ = false;
  }
private:
  int socket_fd_;
  bool thread_is_running_;
  bool service_is_running_;
  void ThreadHandler() {
    int ret;
    char recv_buffer[1024] = {};
    thread_is_running_ = true;
    // 将socket_fd_设置为非阻塞模式
    if (SetNonBlocking(socket_fd_) == -1) {
      std::cout << "Failed to set socket to non-blocking mode" << std::endl;
      return;
    }
    while (thread_is_running_) {
      ret = recv(socket_fd_, recv_buffer, sizeof(recv_buffer), 0);
      if (ret == 0) {
        std::cout << "Remote socket close!!!" << std::endl;
        break;
      } else if (ret < 0) {
        if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) {
          std::cout << "Remote socket error!!!" << std::endl;
          break;
        }
      } else {
        // 处理接收到的数据
        std::cout << "Received data: " << recv_buffer << std::endl;
      }
      std::this_thread::sleep_for(std::chrono::milliseconds(10));
    }
    close(socket_fd_);
    socket_fd_ = -1;
    thread_is_running_ = false;
    service_is_running_ = false;
  }
  int SetNonBlocking(int socket_fd) {
    int flags = fcntl(socket_fd, F_GETFL, 0);
    if (flags == -1) {
      return -1;
    }
    if (fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK) == -1) {
      return -1;
    }
    return 0;
  }
};
int main() {
  NtripServer server;
  server.Start();
  // 服务运行一段时间后停止
  std::this_thread::sleep_for(std::chrono::seconds(10));
  server.Stop();
  return 0;
}

在这个示例代码中,我们创建了一个NtripServer类来管理Ntrip服务。在Start函数中,我们创建了一个socket,并设置选项、绑定地址和端口,然后开始监听连接。接着,我们启动一个线程来处理连接,线程的入口函数是ThreadHandler。

在ThreadHandler函数中,我们将socket_fd_设置为非阻塞模式,并使用recv函数接收数据。根据返回值进行处理,如果返回值为0,表示远程socket关闭;如果返回值小于0,并且errno不是EAGAIN、EWOULDBLOCK或EINTR,表示发生了其他错误;如果返回值大于0,表示成功接收到数据,我们可以在代码中添加处理接收到的数据的逻辑。

在主函数中,我们创建了一个NtripServer对象,并调用Start函数启动服务。然后,让服务运行一段时间后调用Stop函数停止服务。

请注意,这只是一个简单的示例代码,实际应用中可能需要进行更多的错误处理和逻辑处理。对于Linux系统,可以使用fcntl函数来设置socket为非阻塞模式。在上述代码中,SetNonBlocking函数就是用来设置socket为非阻塞的。具体来说,它通过调用fcntl函数,使用F_GETFL命令获取socket的当前标志位,然后将O_NONBLOCK标志位加入到标志位中,再次调用fcntl函数,使用F_SETFL命令将新的标志位设置回socket。

这样设置之后,socket的recv函数将变为非阻塞模式,即不会一直等待数据到达,而是立即返回。如果没有数据到达,recv函数将返回-1,并且errno会被设置为EAGAIN或EWOULDBLOCK。这样,我们可以在代码中根据返回值和errno来判断是否有数据到达,并进行相应的处理。如果在SetNonBlocking函数中设置socket为非阻塞模式失败,可以在调用该函数的地方进行错误处理。可以根据具体情况选择合适的错误处理方式,例如打印错误信息、记录日志、关闭socket等。以下是一个示例的错误处理代码:

if (SetNonBlocking(socket_fd) == -1) {
  std::cout << "Failed to set socket to non-blocking mode" << std::endl;
  // 处理错误,例如关闭socket
  close(socket_fd);
  return;
}

在上述代码中,如果SetNonBlocking函数返回-1,表示设置socket为非阻塞模式失败。我们可以打印错误信息,然后进行相应的错误处理,例如关闭socket。这样可以确保在设置失败的情况下,不会继续使用这个socket进行后续操作。

您可能感兴趣的文章:
阅读全文