像TAB和KEY_UP这样的键无法按预期工作(例如,KEY_UP成为" ^ [[A")在PTY程序

时间:2018-04-08 11:09:41

标签: c sh pty

我有一个pty代码示例,(可能是最狡猾的pty示例)我试图用它来启动sh [sehll terminal]在奴隶但是热键如cd或key up [last command] not working [什么也没做]     我用gcc -o pty.o pty.c来填充它     并运行./pty.o "sh"     但是当我尝试修改密钥时,它只是打印^[[A并且标签执行选项卡缩进而不是建议目录选项' s。     代码位于http://rachid.koucha.free.fr/tech_corner/pty_pdip.html

在mypty3下:

  

我们可以使mypty2更通用,以便能够执行pty(slave端)后面的任何程序。在mypty3中,父进程将所有数据从其标准输入写入pty的主端,并将所有数据从pty的主端写入其标准输出。子进程的行为与mypty2中的行为相同,但执行交互式程序及其作为参数传递给程序的参数。我们可以注意到setsid()ioctl(TIOCSCTTY)的调用,使pty成为执行程序的控制终端。我们还可以注意到fds文件描述符的关闭在调用dup()后变得无用。

 #define _XOPEN_SOURCE 600
#include <stdlib.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
#include <stdio.h>
#define __USE_BSD
#include <termios.h>
#include <sys/select.h>
#include <sys/ioctl.h>
#include <string.h>


int main(int ac, char *av[])
{
int fdm, fds;
int rc;
char input[150];

// Check arguments
if (ac <= 1)
{
fprintf(stderr, "Usage: %s program_name [parameters]\n", av[0]);
exit(1);
}

fdm = posix_openpt(O_RDWR);
if (fdm < 0)
{
fprintf(stderr, "Error %d on posix_openpt()\n", errno);
return 1;
}

rc = grantpt(fdm);
if (rc != 0)
{
fprintf(stderr, "Error %d on grantpt()\n", errno);
return 1;
}

rc = unlockpt(fdm);
if (rc != 0)
{
fprintf(stderr, "Error %d on unlockpt()\n", errno);
return 1;
}

// Open the slave side ot the PTY
fds = open(ptsname(fdm), O_RDWR);

// Create the child process
if (fork())
{
fd_set fd_in;

  // FATHER

  // Close the slave side of the PTY
  close(fds);

  while (1)
  {
    // Wait for data from standard input and master side of PTY
    FD_ZERO(&fd_in);
    FD_SET(0, &fd_in);
    FD_SET(fdm, &fd_in);

    rc = select(fdm + 1, &fd_in, NULL, NULL, NULL);
    switch(rc)
    {
      case -1 : fprintf(stderr, "Error %d on select()\n", errno);
                exit(1);

      default :
      {
        // If data on standard input
        if (FD_ISSET(0, &fd_in))
        {
          rc = read(0, input, sizeof(input));
          if (rc > 0)
          {
            // Send data on the master side of PTY
            write(fdm, input, rc);
          }
          else
          {
            if (rc < 0)
            {
              fprintf(stderr, "Error %d on read standard input\n", errno);
              exit(1);
            }
          }
        }

        // If data on master side of PTY
        if (FD_ISSET(fdm, &fd_in))
        {
          rc = read(fdm, input, sizeof(input));
          if (rc > 0)
          {
            // Send data on standard output
            write(1, input, rc);
          }
          else
          {
            if (rc < 0)
            {
              fprintf(stderr, "Error %d on read master PTY\n", errno);
              exit(1);
            }
          }
        }
      }
    } // End switch
  } // End while
}
else
{
struct termios slave_orig_term_settings; // Saved terminal settings
struct termios new_term_settings; // Current terminal settings

  // CHILD

  // Close the master side of the PTY
  close(fdm);

  // Save the defaults parameters of the slave side of the PTY
  rc = tcgetattr(fds, &slave_orig_term_settings);

  // Set RAW mode on slave side of PTY
  new_term_settings = slave_orig_term_settings;
  cfmakeraw (&new_term_settings);
  tcsetattr (fds, TCSANOW, &new_term_settings);

  // The slave side of the PTY becomes the standard input and outputs of the child process
  close(0); // Close standard input (current terminal)
  close(1); // Close standard output (current terminal)
  close(2); // Close standard error (current terminal)

  dup(fds); // PTY becomes standard input (0)
  dup(fds); // PTY becomes standard output (1)
  dup(fds); // PTY becomes standard error (2)

  // Now the original file descriptor is useless
  close(fds);

  // Make the current process a new session leader
  setsid();

  // As the child is a session leader, set the controlling terminal to be the slave side of the PTY
  // (Mandatory for programs like the shell to make them manage correctly their outputs)
  ioctl(0, TIOCSCTTY, 1);

  // Execution of the program
  {
  char **child_av;
  int i;

    // Build the command line
    child_av = (char **)malloc(ac * sizeof(char *));
    for (i = 1; i < ac; i ++)
    {
      child_av[i - 1] = strdup(av[i]);
    }
    child_av[i - 1] = NULL;
    rc = execvp(child_av[0], child_av);
  }

  // if Error...
  return 1;
}

return 0;
} // main

1 个答案:

答案 0 :(得分:0)

cfmakeraw()应在父进程中调用,而不是在子进程中调用。然后,所有用户输入( TAB ,...)将由子进程处理。并且父母需要在退出之前恢复其原始终端设置。

更新了代码(1.将cfmakeraw()从子级移到了父级; 2.在exit()之前恢复终端设置:

#define _GNU_SOURCE

#include <stdlib.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
#include <stdio.h>
#include <termios.h>
#include <sys/select.h>
#include <sys/ioctl.h>
#include <string.h>

struct termios save_termios;

void restore_term(void)
{
    tcsetattr(STDIN_FILENO, TCSANOW, &save_termios);
}

int main(int ac, char *av[])
{
    int fdm, fds;
    int rc;
    char input[150];

    tcgetattr(STDIN_FILENO, & save_termios);

    // Check arguments
    if (ac <= 1)
    {
        fprintf(stderr, "Usage: %s program_name [parameters]\n", av[0]);
        exit(1);
    }

    fdm = posix_openpt(O_RDWR);
    if (fdm < 0)
    {
        fprintf(stderr, "Error %d on posix_openpt()\n", errno);
        return 1;
    }

    rc = grantpt(fdm);
    if (rc != 0)
    {
        fprintf(stderr, "Error %d on grantpt()\n", errno);
        return 1;
    }

    rc = unlockpt(fdm);
    if (rc != 0)
    {
        fprintf(stderr, "Error %d on unlockpt()\n", errno);
        return 1;
    }

    // Open the slave side ot the PTY
    fds = open(ptsname(fdm), O_RDWR);

    // Create the child process
    if (fork())
    {
        fd_set fd_in;
        struct termios new_setting;

        // FATHER
        atexit(restore_term);

        new_setting = save_termios;
        cfmakeraw ( & new_setting);
        tcsetattr(STDIN_FILENO, TCSANOW, & new_setting);

        // Close the slave side of the PTY
        close(fds);

        while (1)
        {
            // Wait for data from standard input and master side of PTY
            FD_ZERO(&fd_in);
            FD_SET(0, &fd_in);
            FD_SET(fdm, &fd_in);

            rc = select(fdm + 1, &fd_in, NULL, NULL, NULL);
            switch(rc)
            {
            case -1 : fprintf(stderr, "Error %d on select()\n", errno);
                      exit(1);

            default :
                      {
                          // If data on standard input
                          if (FD_ISSET(0, &fd_in))
                          {
                              rc = read(0, input, sizeof(input));
                              if (rc > 0)
                              {
                                  // Send data on the master side of PTY
                                  write(fdm, input, rc);
                              }
                              else
                              {
                                  if (rc < 0)
                                  {
                                      fprintf(stderr, "Error %d on read standard input\n", errno);
                                      exit(1);
                                  }
                              }
                          }

                          // If data on master side of PTY
                          if (FD_ISSET(fdm, &fd_in))
                          {
                              rc = read(fdm, input, sizeof(input));
                              if (rc > 0)
                              {
                                  // Send data on standard output
                                  write(1, input, rc);
                              }
                              else
                              {
                                  if (rc < 0)
                                  {
                                      fprintf(stderr, "Error %d on read master PTY\n", errno);
                                      exit(1);
                                  }
                              }
                          }
                      }
            } // End switch
        } // End while
    }
    else
    {
        // CHILD

        // Close the master side of the PTY
        close(fdm);

        // The slave side of the PTY becomes the standard input and outputs of the child process
        close(0); // Close standard input (current terminal)
        close(1); // Close standard output (current terminal)
        close(2); // Close standard error (current terminal)

        dup(fds); // PTY becomes standard input (0)
        dup(fds); // PTY becomes standard output (1)
        dup(fds); // PTY becomes standard error (2)

        // Now the original file descriptor is useless
        close(fds);

        // Make the current process a new session leader
        setsid();

        // As the child is a session leader, set the controlling terminal to be the slave side of the PTY
        // (Mandatory for programs like the shell to make them manage correctly their outputs)
        ioctl(0, TIOCSCTTY, 1);

        // Execution of the program
        {
            char **child_av;
            int i;

            // Build the command line
            child_av = (char **)malloc(ac * sizeof(char *));
            for (i = 1; i < ac; i ++)
            {
                child_av[i - 1] = strdup(av[i]);
            }
            child_av[i - 1] = NULL;
            rc = execvp(child_av[0], child_av);
        }

        // if Error...
        return 1;
    }

    return 0;
} // main