linux中send函数MSG_NOSIGNAL

linux下当连接断开,还发数据的时候,不仅send()的返回值会有反映,而且还会向系统发送一个异常消息,如果不作处理,系统会出BrokePipe,程序会退出,这对于服务器提供稳定的服务将造成巨大的灾难。为此,send()函数的最后一个参数可以设MSG_NOSIGNAL,禁止send()函数向系统发送异常消息

//中间的select就是为了侦测断开后再次连接并二次send

int socket_send(async_socket_t* async_sock, uint8_t* buf, int len){
  uint8_t* ptr = buf;
  int left = len;
  int rc;
  while (left > 0) {
#ifdef __IOS__
    rc = send(async_sock->fd, ptr, left, 0);
#else
    rc = send(async_sock->fd, ptr, left, MSG_NOSIGNAL);
#endif


    if(rc>0){
      left -= rc;
      ptr += rc;
    }else if (rc<0) {
      if (errno == EAGAIN) {
        break;
      } else if (errno == EINTR) {
        continue;
      } else {
        printf("send step 1 failed\n");
        return -1;
      }
    }
  }

  if(left>0){
    printf("wait writeable");
    fd_set writeset;
    FD_ZERO(&writeset);
    int max_fd = async_sock->fd + 1;
    struct timeval timeout;
    while(!async_sock->bstop && left>0){      
      timeout.tv_sec = 0;
      timeout.tv_usec = 100*1000;
      FD_SET(async_sock->fd, &writeset);
      int ret = select(max_fd, NULL, &writeset, NULL, &timeout);
      if( ret == 0){//TIMEOUT
        continue;
      }else if( ret < 0){
        printf("select failed\n");
        if(errno == EINTR){
          continue;
        }
        return -1;
      }
      //writeable
#ifdef __IOS__
      rc = send(async_sock->fd, ptr, left, 0);
#else
      rc = send(async_sock->fd, ptr, left, MSG_NOSIGNAL);
#endif

      if(rc>0){
        left -= rc;
        ptr += rc;
      }else if (rc<0) {
        printf("send step 2 failed\n");
        return -1;
      }
    }

  }
  return len-left;
}

客户端代码 #define _WINSOCK_DEPRECATED_NO_WARNINGS #include <stdio.h> #include <Winsock2.h> #ifndef MSG_NOSIGNAL #define MSG_NOSIGNAL 0 #endif #pragma comment(lib,"ws2_32.lib") int main() { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD(1, 1); err = WSAStartup(wVersionRequested, &wsaData); if (err != 0) { return -1; } if (LOBYTE(wsaData.wVersion) != 1 || HIBYTE(wsaData.wVersion) != 1) { WSACleanup(); return -1; } SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0); SOCKADDR_IN addrSrv; addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); addrSrv.sin_family = AF_INET; addrSrv.sin_port = htons(6000); connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)); send(sockClient, "hello\0", strlen("hello") + 1, 0); char recvBuf[50]; recv(sockClient, recvBuf, 50, 0); printf("%s\n", recvBuf); closesocket(sockClient); WSACleanup(); return 0;} 服务器端:#define _WINSOCK_DEPRECATED_NO_WARNINGS #include <stdio.h> #include <Winsock2.h> #ifndef MSG_NOSIGNAL #define MSG_NOSIGNAL 0 #endif #pragma comment(lib,"ws2_32.lib") int main() { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD(2, 2); err = WSAStartup(wVersionRequested, &wsaData); if (err != 0) { return 1; } if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) { WSACleanup(); return 1; } SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0); int optval = 1; setsockopt(sockSrv, SOL_SOCKET, SO_REUSEADDR, (const char*)&optval, sizeof(optval)); SOCKADDR_IN addrSrv; addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY); addrSrv.sin_family = AF_INET; addrSrv.sin_port = htons(6000); bind(sockSrv, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)); listen(sockSrv, 5); SOCKADDR_IN addrClient; int len = sizeof(SOCKADDR); while (1) { SOCKET sockConn = accept(sockSrv, (SOCKADDR*)&addrClient, &len); char sendBuf[50]; printf(sendBuf, "Welcome %s to here!", inet_ntoa(addrClient.sin_addr)); send(sockConn, sendBuf, strlen(sendBuf) + 1, MSG_NOSIGNAL); char recvBuf[50]; recv(sockConn, recvBuf, 50, 0); printf("%s\n", recvBuf); closesocket(sockConn); } WSACleanup(); return 0;} 如何修改代码改成可以一直聊天的 不要预输入进去的 要我自己在客户端进行打字操作
06-06
你刚刚给我修改的服务器端和客户端的代码后为什么服务器端不显示任何消息 服务器端代码 #define _WINSOCK_DEPRECATED_NO_WARNINGS#include <stdio.h>#include <Winsock2.h>#ifndef MSG_NOSIGNAL#define MSG_NOSIGNAL 0#endif#pragma comment(lib,"ws2_32.lib")int main() { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD(2, 2); err = WSAStartup(wVersionRequested, &wsaData); if (err != 0) { return 1; } if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) { WSACleanup(); return 1; } // 创建套接字并绑定到本地地址和端口 SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0); int optval = 1; setsockopt(sockSrv, SOL_SOCKET, SO_REUSEADDR, (const char*)&optval, sizeof(optval)); SOCKADDR_IN addrSrv; addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY); addrSrv.sin_family = AF_INET; addrSrv.sin_port = htons(6000); bind(sockSrv, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)); // 监听连接请求并接受客户端连接 listen(sockSrv, 5); SOCKADDR_IN addrClient; int len = sizeof(SOCKADDR); SOCKET sockConn; fd_set fdReads; while (1) { // 使用 select() 函数实现非阻塞接收客户端连接 FD_ZERO(&fdReads); FD_SET(sockSrv, &fdReads); int ret = select(sockSrv + 1, &fdReads, NULL, NULL, NULL); if (ret < 0) { break; } if (FD_ISSET(sockSrv, &fdReads)) { sockConn = accept(sockSrv, (SOCKADDR*)&addrClient, &len); char sendBuf[50]; printf(sendBuf, "Welcome %s to here!\n", inet_ntoa(addrClient.sin_addr)); send(sockConn, sendBuf, strlen(sendBuf) + 1, MSG_NOSIGNAL); } // 使用 select() 函数实现非阻塞接收客户端消息 FD_ZERO(&fdReads); FD_SET(sockConn, &fdReads); ret = select(sockConn + 1, &fdReads, NULL, NULL, NULL); if (ret < 0) { break; } if (FD_ISSET(sockConn, &fdReads)) { char recvBuf[50]; recv(sockConn, recvBuf, 50, 0); printf("Received from client: %s\n", recvBuf); // 将客户端发送的消息广播给所有连接的客户端 for (SOCKET i = 0; i < FD_SETSIZE; i++) { if (FD_ISSET(i, &fdReads)) { send(i, recvBuf, strlen(recvBuf) + 1, MSG_NOSIGNAL); } } } } closesocket(sockSrv); WSACleanup(); return 0;}
06-06
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值