Блог

Я прошу прощения за длину, но я подозреваю, что ошибка может быть связана с обработкой ошибок, и есть немного необходимого шаблона для создания сети.

Я запускаю следующую реализацию tcp-сервера с разветвлением одного файла.

 #include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <netdb.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <errno.h>
#include <pty.h>

#define PORT 9600
#define BUF_SIZE 128
#define SERVER_BACKLOG 10

void handle_conn(int conn_fd) {
  // Create a TTY / PTY pair and hook it up to the connection.
  int master_fd;
  char slave_name[100];
  pid_t pid = forkpty(amp;master_fd, slave_name, NULL, NULL);
  if (pid == -1) {
    fprintf(stderr, "Error on first fork: %d %s",  errno, strerror(errno));
    return;
  } else if (pid == 0) { // Child process attached to slave pty
    execl("/bin/bash", "bash", (char*) NULL);
  }

  // Set both fds to be non-blocking
  int flags = fcntl(master_fd, F_GETFL, 0);
  fcntl(master_fd, F_SETFL, flags | O_NONBLOCK);
  flags = fcntl(conn_fd, F_GETFL, 0);
  fcntl(conn_fd, F_SETFL, flags | O_NONBLOCK);

  // Parent continues here, shuffling data back and forth from the socket to
  // the master fd.
  int n;
  char c;
  while(1) {
    fd_set fds;
    FD_ZERO(amp;fds);
    FD_SET(conn_fd, amp;fds);
    FD_SET(master_fd, amp;fds);
    int numFds = select((conn_fd > master_fd ? conn_fd : master_fd)   1, amp;fds, NULL, NULL, NULL);

    if (numFds == -1) {
      fprintf(stderr, "Error on select: %d %s",  errno, strerror(errno));
      return;
    }
    if (numFds == 0) {
      continue;
    }
    if (FD_ISSET(conn_fd, amp;fds)) {
      // Read from socket and write to tty.
      while ((n = read(conn_fd, amp;c, 1)) > 0) {
        write(master_fd, amp;c, 1);
      }

      if (n == 0) {
        // Reached EOF
        break;
      }
      // Presumed n is -1
      if (errno != EAGAIN || errno != EWOULDBLOCK) {
        fprintf(stderr, "Error reading socket %d %sn", errno, strerror(errno));
        break;
      }
    }

    if (FD_ISSET(master_fd, amp;fds)) {
      // Read from tty and write to socket as long as things are still
      // readable.
      while ((n = read(master_fd, amp;c, 1)) > 0) {
        write(conn_fd, amp;c, 1);
      }

      if (n == 0) {
        // Reached EOF
        break;
      }

      // Presumed n is -1
      if (errno != EAGAIN amp;amp; errno != EWOULDBLOCK) {
        fprintf(stderr, "Error reading master FD %d %sn", errno, strerror(errno));
        break;
      }
    }
  }
  printf("Closing connection [%d]n", conn_fd);
  int err = close(conn_fd);
  if (err != 0) {
    fprintf(stderr, "Error while closing connection [%d]n", conn_fd);
  }

  err = close(master_fd);
  if (err != 0) {
    fprintf(stderr, "Error while closing master fd [%d]n", conn_fd);
  }
}

int main(int argc, char *argv[]) {
  int server_fd = socket(PF_INET, SOCK_STREAM, 0);
  if (server_fd < 0) {
    fprintf(stderr, "Cannot create socketn");
    return 1;
  }

  struct sockaddr_in server_addr;
  server_addr.sin_family = AF_INET;
  server_addr.sin_port = htons(PORT);
  int ok = inet_pton(AF_INET, "127.0.0.1", amp;server_addr.sin_addr);
  if (!ok) {
    fprintf(stderr, "Cannot parse IP addressn");
    return 1;
  }

  // Allow reuse of port.
  int optval = 1;
  setsockopt(server_fd, SOL_SOCKET, SO_REUSEPORT, amp;optval, sizeof(optval));

  int err = bind(server_fd, (struct sockaddr *) amp;server_addr, sizeof(server_addr));
  if (err != 0) {
    fprintf(stderr, "Cannot bind server: Error %d %sn", errno, strerror(errno));
    return 1;
  }

  err = listen(server_fd, SERVER_BACKLOG);
  if (err != 0) {
    fprintf(stderr, "Cannot listenn");
    return 1;
  }

  printf("Server listening on port %dn", PORT);

  int client_fd;
  struct sockaddr_in client_addr;
  socklen_t client_addr_len;
  while (1) {
    client_fd = accept(server_fd, (struct sockaddr *) amp;client_addr, amp;client_addr_len);
    if (client_fd < 0) {
      fprintf(stderr, "Cannot accept: Error %d %sn", errno, strerror(errno));
      return 1;
    }

    printf("New client [%d]n", client_fd);

    // Using fork instead of threads here to avoid pitfalls involved with
    // mixing forking and threads, since this use case case of a tty host is
    // not expected to involve a great many requests.
    int pid = fork();
    if (pid == -1) {
      fprintf(stderr, "Error on forking to handle new connection: %d %s",  errno, strerror(errno));
      // Server does not need to die here.
      continue;
    } else if (pid == 0) { // Child will handle the connection and then return.
      handle_conn(client_fd);
      return 0;
    }
    // Parent continues to accept
  }
}
  

Затем на другом терминале я запускаю следующий клиент netcat.

 stty -icanon -echo; nc localhost 9600
  

На этом этапе я могу вводить команды типа ls или cat на терминале с nc помощью .

Однако, когда я печатаю exit , сервер печатает Error reading master FD 5 Input/output errornClosing connection [4] , и больше никаких ошибок, поэтому я предполагаю, что сервер успешно закрыл соединение, но клиент продолжает зависать, пока я не нажму Control-C на клиента.

Как я могу изменить код сервера, чтобы заставить клиента отключиться?

Я обнаружил ошибку! Оказывается, я забыл закрыть сокет в родительском процессе. Как только я это сделал, close() сработал, как и ожидалось, в дочернем элементе.

Простая модификация приведенного выше кода, как показано ниже, устраняет проблему.

     ...
    // Parent continues to accept
    close(client_fd);