recv send处理

读数据的时候需要考虑的是当recv()返回的大小如果等于请求的大小,那么很有可能是缓冲区还有数据未读完,也意味着该次事件还没有处理完,所以还需要再次读取
while(rs)
{
buflen = recv(activeevents[i].data.fd, buf, sizeof(buf), 0);
if(buflen < 0)
{
    // 
由于是非阻塞的模式,所以当errnoEAGAIN,表示当前缓冲区已无数据可读
    // 
在这里就当作是该次事件已处理处 .
    if(errno == EAGAIN)
     break;
    else
     return;
   }
   else if(buflen == 0)
   {
     // 
这里表示对端的socket已正常关闭
.
   }
   if(buflen == sizeof(buf)
     rs = 1;   // 
需要再次读取
   else
     rs = 0;

}



假如发送端流量大于接收端的流量(意思是epoll所在的程序读比转发的socket要快),由于是非阻塞的socket,那么send()函数虽然返回,但实际缓冲区的数据并未真正发给接收端,这样不断的读和发,当缓冲区满后会产生EAGAIN错误(参考man send),同时,不理会这次请求发送的数据.所以,需要封装socket_send()的函数用来处理这种情况,该函数会尽量将数据写完再返回,返回-1表示出错。在socket_send()内部,当写缓冲已满(send()返回-1,errnoEAGAIN),那么会等待后再重试.这种方式并不很完美,在理论上可能会长时间的阻塞在socket_send()内部,但暂没有更好的办法.

ssize_t socket_send(int sockfd, const char* buffer, size_t buflen)
{
ssize_t tmp;
size_t total = buflen;
const char *p = buffer;

while(1)
{
    tmp = send(sockfd, p, total, 0);
    if(tmp < 0)
    {
      // 
send收到信号时,可以继续写,但这里返回-1.
      if(errno == EINTR)
        return -1;

      // 
socket是非阻塞时,如返回此错误,表示写缓冲队列已满,
      // 
在这里做延时后再重试.
      if(errno == EAGAIN)
      {
        usleep(1000);
        continue;
      }

      return -1;
    }

    if((size_t)tmp == total)
      return buflen;

    total -= tmp;
    p += tmp;
}

return tmp;
}


转载自http://blog.csdn.net/xiaofei0859/article/details/6037814/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是一个使用 epoll 实现的简单的 TCP 服务器和客户端的例子。服务器和客户端都是非阻塞的,使用 epoll 监听和处理文件描述符事件。其中,服务器使用了多线程技术,每个线程负责处理一个客户端连接。 服务器代码: ``` #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <errno.h> #include <signal.h> #include <netinet/in.h> #include <sys/epoll.h> #include <pthread.h> #define MAX_EVENTS 1024 #define BUF_SIZE 1024 #define THREAD_NUM 4 int listen_fd; int epoll_fd; pthread_t thread_pool[THREAD_NUM]; void setnonblocking(int fd) { int flags = fcntl(fd, F_GETFL, 0); fcntl(fd, F_SETFL, flags | O_NONBLOCK); } void add_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd, &ev); } void del_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_DEL, fd, &ev); } void mod_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &ev); } void *worker(void *arg) { int client_fd = *(int*)arg; char buf[BUF_SIZE]; int nread; while (1) { nread = recv(client_fd, buf, BUF_SIZE, 0); if (nread == -1) { if (errno == EAGAIN) { printf("recv EAGAIN\n"); break; } else { perror("recv"); break; } } else if (nread == 0) { printf("client close\n"); break; } else { printf("recv: %s\n", buf); if (send(client_fd, buf, nread, 0) == -1) { perror("send"); break; } } } close(client_fd); pthread_exit(NULL); } void *accepter(void *arg) { struct sockaddr_in client_addr; socklen_t client_len = sizeof(client_addr); int client_fd; int i; while (1) { client_fd = accept(listen_fd, (struct sockaddr*)&client_addr, &client_len); if (client_fd == -1) { if (errno == EAGAIN) { printf("accept EAGAIN\n"); continue; } else { perror("accept"); break; } } else { setnonblocking(client_fd); add_event(client_fd, EPOLLIN|EPOLLET); printf("new client: %d\n", client_fd); } } close(listen_fd); pthread_exit(NULL); } int main(int argc, char *argv[]) { struct sockaddr_in server_addr; int port = 8888; struct epoll_event events[MAX_EVENTS]; int nready; int i, j; if (argc > 1) { port = atoi(argv[1]); } listen_fd = socket(AF_INET, SOCK_STREAM, 0); if (listen_fd == -1) { perror("socket"); exit(1); } setnonblocking(listen_fd); memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = htonl(INADDR_ANY); server_addr.sin_port = htons(port); if (bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) { perror("bind"); exit(1); } if (listen(listen_fd, 1024) == -1) { perror("listen"); exit(1); } epoll_fd = epoll_create(MAX_EVENTS); if (epoll_fd == -1) { perror("epoll_create"); exit(1); } add_event(listen_fd, EPOLLIN|EPOLLET); for (i = 0; i < THREAD_NUM; i++) { if (pthread_create(&thread_pool[i], NULL, accepter, NULL) == -1) { perror("pthread_create"); exit(1); } } while (1) { nready = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); if (nready == -1) { if (errno == EINTR) { continue; } else { perror("epoll_wait"); break; } } for (i = 0; i < nready; i++) { if (events[i].data.fd == listen_fd) { continue; } if (events[i].events & EPOLLIN) { pthread_t thread_id; int client_fd = events[i].data.fd; del_event(client_fd, EPOLLIN|EPOLLET); if (pthread_create(&thread_id, NULL, worker, (void*)&client_fd) == -1) { perror("pthread_create"); } } } } for (i = 0; i < THREAD_NUM; i++) { if (pthread_join(thread_pool[i], NULL) == -1) { perror("pthread_join"); } } close(epoll_fd); return 0; } ``` 客户端代码: ``` #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <errno.h> #include <netinet/in.h> #include <arpa/inet.h> #include <sys/epoll.h> #define MAX_EVENTS 1024 #define BUF_SIZE 1024 void setnonblocking(int fd) { int flags = fcntl(fd, F_GETFL, 0); fcntl(fd, F_SETFL, flags | O_NONBLOCK); } void add_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd, &ev); } void del_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_DEL, fd, &ev); } void mod_event(int fd, int events) { struct epoll_event ev; ev.events = events; ev.data.fd = fd; epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &ev); } int main(int argc, char *argv[]) { struct sockaddr_in server_addr; char *ip = "127.0.0.1"; int port = 8888; int sockfd; struct epoll_event events[MAX_EVENTS]; int nready; char buf[BUF_SIZE]; int nread, nwrite; int i; if (argc > 1) { ip = argv[1]; } if (argc > 2) { port = atoi(argv[2]); } sockfd = socket(AF_INET, SOCK_STREAM, 0); if (sockfd == -1) { perror("socket"); exit(1); } setnonblocking(sockfd); memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = inet_addr(ip); server_addr.sin_port = htons(port); if (connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) { if (errno != EINPROGRESS) { perror("connect"); exit(1); } } epoll_fd = epoll_create(MAX_EVENTS); if (epoll_fd == -1) { perror("epoll_create"); exit(1); } add_event(sockfd, EPOLLOUT); while (1) { nready = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); if (nready == -1) { if (errno == EINTR) { continue; } else { perror("epoll_wait"); break; } } for (i = 0; i < nready; i++) { if (events[i].events & EPOLLOUT) { del_event(sockfd, EPOLLOUT); printf("connected\n"); break; } } } while (1) { scanf("%s", buf); if (strcmp(buf, "quit") == 0) { break; } nwrite = send(sockfd, buf, strlen(buf), 0); if (nwrite == -1) { perror("send"); break; } nread = recv(sockfd, buf, BUF_SIZE, 0); if (nread == -1) { if (errno == EAGAIN) { printf("recv EAGAIN\n"); continue; } else { perror("recv"); break; } } else if (nread == 0) { printf("server close\n"); break; } else { printf("recv: %s\n", buf); } } close(sockfd); close(epoll_fd); return 0; } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值