c语言 readline实现



要实现类似java的readline功能，需要用到recv函数的MSG_PEEK标示，它表示recv在读缓冲区的时候不对缓冲区进行清空，

只是一次窥视，以此来找到第一个出现\n的位置，并调用recv进行读取清空

int readline(int sockfd, char *buf, int len)
{
 int count;

 while (1)
 {
  // 设置标志位，读取后不清除缓冲区
  count = recv(sockfd, buf, len, MSG_PEEK);

  // 小于0表示失败，等于0表示对方关闭连接了
  if (count < 0 || count == 0)
   return count;

  // 找到第一次出现\n的位置，并截取到此位置的字符串
  for (int i = 0; i < count; ++i)
   if (buf[i] == '\n')
   {
    memset(buf, 0, len);

    len = i + 1;

    if (recv(sockfd, buf, len, 0) != len)
     return -1;

    return len;
   }
 }

 return -1;
}