Socket recv()之前进行select代码
2019獨角獸企業重金招聘Python工程師標準>>>
================ 收藏備用。 ================ 參數解釋: sck - socket buf - 接收緩沖區 size-緩沖區大小 time_out-等待時間(按秒計)如果超時則返回 返回值:收到字節數,0表示超時等錯誤 ?int rcv(int sck, void * buf, int size, int time_out)
{
if (sck < 1 || !buf || size < 1) return 0;
timeval tv = { 0, 0}; timeval * ptv = 0;
if (time_out > 0) { tv.tv_sec = time_out; ptv = &tv; }
memset(buf, 0, size);
int r = 0; char * b = (char*) buf; int sz = size;
fd_set rd, er; int total = 0; time_t t0 = time(0); time_t t1 = 0;
do {
FD_ZERO(&rd); FD_SET(sck, &rd);
FD_ZERO(&er); FD_SET(sck, &er);
r = select(sck + 1, &rd, 0, &er, ptv);
if (r == -1) { nperror("select()"); return -1; }
if (FD_ISSET(sck, &er)) {
nperror("socket(shutdown)"); return -1;
}//end if
if (FD_ISSET(sck, &rd)) {
r = recv(sck, b, sz, 0);
if (r == -1) { nperror("recv()"); return -1; }
total += r; sz -= r; b+= r;
}//end if
if (time_out > 0)
t1 = time(0) - t0;
else
t1 = time_out - 1;
//end if
}while(sz && t1 < time_out);
return total;
}//end if
轉載于:https://my.oschina.net/u/4000302/blog/3044777
總結
以上是生活随笔為你收集整理的Socket recv()之前进行select代码的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 一卡通变“一脸通” 人脸识别更新校园考勤
- 下一篇: MySql入门知识(一)