网络编程（17）—— 使用recv检查输入缓存是否有数据

    之前一直用linux中的read/write函数进行socket描述符的读写，而在windows中用recv/send进行socket的读写，其实在linux中也有自己的recv/send函数，同样可以进行IO端口的读写操作。

ssize_t recv(int sockfd, void *buf, size_t len, int flags);

ssize_t send(int sockfd, const void *buf, size_t len, int flags);

       recv和send函数的原型和windows中基本类似，这里主要介绍下其最后一个参数flags，用于设置接收数据时指定的可选项信息。



    在recv函数中同时设置MSG_PEEK和MSG_DONTWAIT选项，以验证输入缓存上是否存在接收的数据。recv中设置了MSG_PEEK时，即使读取了输入缓冲的数据也不会被删除，一次该选项经常和MSG_DONTWAIT一起使用以非阻塞的方式验证输入缓冲数据是否存在。

peek_send.c：

#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>

void error_handling(char* message);

int main(int argc,char* argv[])
{
int sock;
struct sockaddr_in serv_addr;
if(argc!=3)
{
printf("Usage %s <address> <port>",argv[0]);
exit(1);
}
//创建socket
sock=socket(AF_INET,SOCK_STREAM,0);
//准备地址信息
memset(&serv_addr,0,sizeof(serv_addr));
serv_addr.sin_family=AF_INET;
serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
serv_addr.sin_port=htons(atoi(argv[2]));
//连接服务器
if(connect(sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr))==-1)
error_handling("connect error");
write(sock,"123",strlen("123"));
return 0;
}

void error_handling(char* message)
{
fputs(message,stderr);
fputc('\n',stderr);
exit(1);
}

  客户端实现功能向服务器发送一个字符串”123”。

peek_recv.c

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<string.h>

#define BUF_SIZE 30

void error_handling(char* message);

int main(int argc,char* argv[])
{
int serv_sock,clnt_sock;
struct sockaddr_in serv_addr,clnt_addr;
int clnt_addr_sz;
int str_len;
char buf[BUF_SIZE];
if(argc!=2)
{
printf("Usage %s <port>\n",argv[0]);
exit(1);
}
//创建socket
serv_sock=socket(AF_INET,SOCK_STREAM,0);
//设置地址
memset(&serv_addr,0,sizeof(serv_addr));
serv_addr.sin_family=AF_INET;
serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
serv_addr.sin_port=htons(atoi(argv[1]));
//绑定
if(bind(serv_sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr))==-1)
error_handling("bind error");
//开启监听
listen(serv_sock,5);
//接收链接
clnt_addr_sz=sizeof(clnt_addr);
clnt_sock=accept(serv_sock,(struct sockaddr*)&clnt_addr,&clnt_addr_sz);

while(1)
{
str_len=recv(clnt_sock,buf,BUF_SIZE-1,MSG_PEEK|MSG_DONTWAIT);
if(str_len>0)
break;
}
buf[str_len]=0;
puts(buf);
str_len=recv(clnt_sock,buf,BUF_SIZE-1,0);
buf[str_len]=0;
puts(buf);
return 0;
}
void error_handling(char* message)
{
fputs(message,stderr);
fputc('\n',stderr);
exit(1);
}

   服务器端对通过设置MSG_PEEK|MSG_DONTWAIT选项输入缓冲区进行了检查，如果发现存在数据就进行读取。

Github位置：

https://github.com/HymanLiuTS/NetDevelopment


克隆本项目：

git clone git@github.com:HymanLiuTS/NetDevelopment.git

获取本文源代码：

git checkout NL17