上篇提到网络io无缓冲造成的问题,这里就详细讲解一下并给出相关解决方法
现有服务端代码如下,将套接字设为例无缓冲
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include <sys/fcntl.h>
const int SERV_PORT = 8000;
const int SIZE = 100;
const int MAXLINE = 9000;
void setnonblocking(int fd)
{
int flags = fcntl(fd, F_GETFL, 0);
flags |= O_NONBLOCK;
if(fcntl(fd, F_SETFL, flags) < 0)
perror("fcntl error");
}
int main()
{
struct sockaddr_in servaddr,cliaddr;
socklen_t cliaddr_len;
int listenfd,connfd;
char *buf;
int i,n,flag = 0;
buf = (char *)malloc(sizeof(char) * MAXLINE);
listenfd = socket(AF_INET,SOCK_STREAM,0);
bzero(&servaddr,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); //因为本机地址可以有很多种表示,htonl只是为了保险起见,可以去掉
servaddr.sin_port = htons(SERV_PORT);
bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr)); //端口绑定
listen(listenfd,20);//监听端口,20表示最大阻塞
// printf("Accepting connections..\n");
while(1){
cliaddr_len = sizeof(cliaddr);
connfd = accept(listenfd,(struct sockaddr *)&cliaddr,&cliaddr_len); //如果得不到客户端发来的消息,将会被阻塞,一直等到消息到来
setnonblocking(connfd);
memset(buf, 0, sizeof(char) * MAXLINE);
n = read(connfd,buf,MAXLINE); //如果n<=0,表示客户端已断开
while (1){
if (n != 0){
printf("%s",buf); //输出客户端发来的信息
}
else {
// printf("Client say close the connection..\n");
break;
}
memset(buf, 0, sizeof(char) * MAXLINE);
n = read(connfd,buf,MAXLINE);
}
close(connfd);
}
}
客户端代码如下
#include <unistd.h>
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <error.h>
#include <stdio.h>
#include <sys/fcntl.h>
#include <sys/epoll.h>
#include <arpa/inet.h>
using namespace std;
void setnonblocking(int fd)
{
int flags = fcntl(fd, F_GETFL, 0);
flags |= O_NONBLOCK;
if(fcntl(fd, F_SETFL, flags) < 0)
perror("fcntl error");
}
int main(int argc, char *argv[])
{
if (argc < 2)
{
perror("input ip and port");
}
const char *ip = argv[1];
const int port = atoi(argv[2]);
int clnt_sock;
socklen_t clnt_len;
struct sockaddr_in clnt_addr;
clnt_sock = socket(AF_INET, SOCK_STREAM, 0);
clnt_addr.sin_family = AF_INET;
clnt_addr.sin_addr.s_addr = inet_addr(ip);
clnt_addr.sin_port = htons(port);
clnt_len = sizeof(clnt_addr);
int ret = connect(clnt_sock, (sockaddr*)&clnt_addr, clnt_len);
setnonblocking(clnt_sock);
for (int i = 0; i < 100; ++i)
send(clnt_sock, "123", 3, 0);
return 0;
}
现执行./listen > data
./connect 192.168.16.162 8000
可得到data文件大小经常为0,原因是因为他是无缓冲的,所以数据还未输出就直接执行到了memset语句,然后就覆盖了原数据,但如果你注释了memset却又会得到比既定大小大很多的字符,因为第一波没接收到的字符后面一起接收到了,就会造成未重置为0导致输出变多,所以无缓冲时必须知道既定字符大小,这种情况一般就需要传送头表示接收大小了,如可以像tcp三次握手般建立相关连接,然后进行发送接收保证数据的正确性,一般没必要,不如直接使用缓冲io,这里就不细说了
现我们看第二种情况,我们将接收不设置无缓冲,这时就可以接收的返回值n来判断接收的字符并进行相关输出了
for (int i = 0; i < n; ++i)
printf("%c", buf[i]);
但又会出现问题,你会发现接收不到消息,这是因为发送端是非阻塞,很快走完了流程并关闭了套接字,所以服务端并没有接收的,这时需要在发送端sleep才能正常接收,但这时通过输出你又会发现问题还是没有输出,这就是上一篇提到的行缓冲问题了,因为其没到缓冲限制也没有换行符,这时就需要fflush来刷新缓冲区了,现在输出你会发现正常了
但我们把发送数据增多,加到亿,如下代码所示
setnonblocking(clnt_sock);
char buf[9000];
for (int i = 0; i < 4000; i++)
buf[i] = '1';
for (int i = 0; i < 1000000; ++i)
send(clnt_sock, buf, 4000, 0);
sleep(10);
send(clnt_sock, "2", 1, 0);
sleep(2);
通过接收你会发现2收到了但少了很多数据,这就是典型的缓冲区溢出,发送的数据已经超过了内核发送缓冲区的大小,因为是无缓冲不会进行相关控制,导致了数据 被直接丢弃,解决方法可以通过增大缓冲区,发送时连续发送一批sleep一会儿来缓解缓冲区溢出,这里就不介绍了
本文到此结束,有问题欢迎指出