linux - 基于tcp的阻塞socket规范读操作

引用:

int read_data(int sock_fd, char *buffer, int buffer_len)

{

char *ptemp = buffer;

int bytes_to_recv = buffer_len;

int bytes_recv = 0;



if (sock_fd < 0 || NULL == buffer || buffer_len <= 0)

{

return -1;

}



while (bytes_to_recv > 0)

{

bytes_recv = recv(sock_fd, ptemp, bytes_to_recv, 0);

if (0 == bytes_recv)

{

// the server has close the connection gracefully.

close(fd);

return -1;

}

else if (bytes_recv < 0)

{

if ((EINTR == errno) || (EAGAIN == errno))

{

continue;

}

else

{

close(fd);

return -1;

}

}

else

{

ptemp += bytes_recv;

bytes_to_recv -= bytes_recv;

}

}



return 0;

}