使用select从套接字和标准输入流中读取数据

Question

我正在编写一个基于ncurses的聊天程序。起初，我只编写了网络相关的内容（没有使用ncurses），一切都正常工作，但是在添加图形后，我无法使客户端应用程序正常工作。

主要问题是同时从stdin和socket读取。在没有ncurses版本中，我使用了pthread，它能够完美工作。不幸的是，pthread和ncurses似乎不太搭配，因此我不得不寻找另一种解决方案。我认为select()会有用处，但它仅从stdin读取，并完全忽略了socket。

这是整个代码：code 有趣的部分是：

char message[1024];
fd_set master;
fd_set read_fds;

FD_ZERO(&master);
FD_ZERO(&read_fds);

FD_SET(0,&master);
FD_SET(s,&master); // s is a socket descriptor
while(true){
read_fds = master;
if (select(2,&read_fds,NULL,NULL,NULL) == -1){
  perror("select:");
  exit(1);
}
// if there are any data ready to read from the socket
if (FD_ISSET(s, &read_fds)){
  n = read(s,buf,max);
  buf[n]=0;
  if(n<0)
  {
    printf("Blad odczytu z gniazdka");
    exit(1);
  } 
  mvwprintw(output_window,1,1,"%s\n",buf);
}
// if there is something in stdin
if (FD_ISSET(0, &read_fds)){
  getstr(message);
  move(CURS_Y++,CURS_X);
  if (CURS_Y == LINES-2){
    CURS_Y = 1;
  }
  n = write(s,message,strlen(message));
  if (n < 0){
    perror("writeThread:");
    exit(1);
  }
}
}

我可能不太了解select()函数的工作原理，或者我不应该调用connect()函数连接套接字。我感到很迷茫，请给予帮助！谢谢。

- michauwilliam

---

你知道当通过套接字传入某些东西时，第一件事情就是退出程序吗？ - Mario The Spoon

请勿使用FD_ZERO(0)，而是使用FD_SET(fileno(stdin)，... - Mario The Spoon

抱歉，exit 只是为了测试程序是否会到达那个点。不过感谢你指出来了 :P - michauwilliam

我认为你误解了ncurses和线程不兼容的想法。如果你遇到问题，那么你肯定做错了什么（比如试图同时从多个线程操纵curses数据结构）。 - R.. GitHub STOP HELPING ICE

Answer 1

你的问题出在select()函数上。
第一个参数不是你在read_fds中传递的文件描述符的数量，而是最高套接字ID + 1。

来自man手册的说明：

每个集合中检查前nfds个描述符；即，在描述符集中从0到nfds-1的描述符将被检查。（例如：如果你设置了两个文件描述符“4”和“17”，那么nfds不应该是“2”，而应该是“17 + 1”或者“18”。）

所以在你的代码中，尝试传递's+1'而不是'2'。

- fabrizi0

---

谢谢，现在已经通过FD_ISSET（s，&read_fds）了，但仍然不能在屏幕上打印缓冲区。我猜现在只是ncurses的问题，我明天会尝试解决它。谢谢 :) 另外，如果您有任何想法为什么无法打印，请告诉我！ - michauwilliam

我在打印后忘记了使用wrefresh。问题已解决，谢谢。 - michauwilliam

Answer 2

你需要指定最高的文件描述符来进行选择：

if (select(s + 1,&read_fds,NULL,NULL,NULL) == -1){

select()需要知道它要监视的文件描述符数量。

- Richard Pennington

---

它只应该监视两个描述符 - stdin 和 s。 - michauwilliam

不，这个数字不是文件描述符的数量，而是最高文件描述符加一。而且，“s + 1”肯定比2大。 - Richard Pennington