TCP读取定长数据接口的编写：
int readn(int fd,char*bp,size_t len)
{
int cnt;
int rc;
cnt=len;
while(cnt>0)
{
rc=recv(fd,bp,cnt,0);
if(rc<0)//read error
{
if(errno==EINTR)
continue;
return -1;//return error
}
if(0==rc) //over?
{
return len-cnt;
}
bp+=rc;
cnt-=rc;
}
return len;
}
TCP读取变长数据接口的编写（增加消息丢弃处理）：
需要增加消息头，在消息头中说明数据的长度：
这里数据包的结构定义如下：
typedef struct
{
u_int32_t recLen;
char buf[128];
}packet;
int readvRec(int fd,char* bp,size_t len)//len:接收缓冲区的长度
{
u_int32_t recLen;
int rc;
//retrieve the length of the record
rc=readn(fd,&recLen,sizeof(u_int32_t));
if(rc!=sizeof(u_int32_t))
{
return rc<0?-1:0;
}
recLen=ntohl(recLen);
if(recLen>len)
{
//discard the record
while(recLen>0)
{
rc=readn(fd,bp,len);
if(rc!=len)
{
return rc<0?-1:0;
}
recLen-=len;
if(len>recLen)
{
len=recLen;
}
}
set_errno(EMSGSIZE);
return -1;
}
//retrieve the record
rc=readn(fd,bp,recLen);
if(rc!=len)
{
return rc<0?-1:0;
}
return rc;
}