我对一个协议进行了封装,希望能在loadrunner下调用。结果调用收包函数的时候老是压不上去。
函数大致内容如下
extern "C" LRDLLTEST_API int receive_version_data(int sockfd) { char* receiveBuf = (char *)malloc(1024); memset(receiveBuf,0,1024); int n = recv(sockfd,(char *)receiveBuf,1024,0); free(receiveBuf); return n; } |
经过反复的验证,网上查资料终于发现
int n = recv(sockfd,(char *)receiveBuf,1024,0);
语句有问题;
经修改如下:
int n = recv(sockfd,(char *)receiveBuf,1024,MSG_PEEK);
参阅资料如下:
旦,我们建立好了tcp连接之后,我们就可以把得到的fd当作文件描述符来使用。
由此网络程序里最基本的函数就是read和write函数了。
ssize_t write(int fd, const void*buf,size_t nbytes);
write函数将buf中的nbytes字节内容写入文件描述符fd.成功时返回写的字节数.失败时返回-1. 并设置errno变量. 在网络程序中,当我们向套接字文件描述符写时有两可能.
1)write的返回值大于0,表示写了部分或者是全部的数据. 这样我们用一个while循环来不停的写入,但是循环过程中的buf参数和nbyte参数得由我们来更新。也就是说,网络写函数是不负责将全部数据写完之后在返回的。
2)返回的值小于0,此时出现了错误.我们要根据错误类型来处理.
如果错误为EINTR表示在写的时候出现了中断错误.
如果为EPIPE表示网络连接出现了问题(对方已经关闭了连接).
为了处理以上的情况,我们自己编写一个写函数来处理这几种情况.
int my_write(int fd,void *buffer,int length) { int bytes_left; int written_bytes; char *ptr; ptr=buffer; bytes_left=length; while(bytes_left>0) { /* 开始写*/ written_bytes=write(fd,ptr,bytes_left); if(written_bytes<=0) /* 出错了*/ { if(errno==EINTR) /* 中断错误 我们继续写*/ written_bytes=0; else /* 其他错误 没有办法,只好撤退了*/ return(-1); } bytes_left-=written_bytes; ptr+=written_bytes; /* 从剩下的地方继续写 */ } return(0); } |