tools/debugserver/source/TTYState.cpp

   1 //===-- TTYState.cpp --------------------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 //  Created by Greg Clayton on 3/26/07.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "TTYState.h"
  15 #include <fcntl.h>
  16 #include <unistd.h>
  17 #include <sys/signal.h>
  18
  19 TTYState::TTYState() :
  20     m_fd(-1),
  21     m_tflags(-1),
  22     m_ttystateErr(-1),
  23     m_processGroup(-1)
  24 {
  25 }
  26
  27 TTYState::~TTYState()
  28 {
  29 }
  30
  31 bool
  32 TTYState::GetTTYState (int fd, bool saveProcessGroup)
  33 {
  34     if (fd >= 0 && ::isatty (fd))
  35     {
  36         m_fd = fd;
  37         m_tflags = fcntl (fd, F_GETFL, 0);
  38         m_ttystateErr = tcgetattr (fd, &m_ttystate);
  39         if (saveProcessGroup)
  40             m_processGroup = tcgetpgrp (0);
  41         else
  42             m_processGroup = -1;
  43     }
  44     else
  45     {
  46         m_fd = -1;
  47         m_tflags = -1;
  48         m_ttystateErr = -1;
  49         m_processGroup = -1;
  50     }
  51     return m_ttystateErr == 0;
  52 }
  53
  54 bool
  55 TTYState::SetTTYState () const
  56 {
  57     int result = 0;
  58     if (IsValid())
  59     {
  60         if (TFlagsValid())
  61             result = fcntl (m_fd, F_SETFL, m_tflags);
  62
  63         if (TTYStateValid())
  64             result = tcsetattr (m_fd, TCSANOW, &m_ttystate);
  65
  66         if (ProcessGroupValid())
  67         {
  68             // Save the original signal handler.
  69             void (*saved_sigttou_callback) (int) = NULL;
  70             saved_sigttou_callback = (void (*)(int)) signal (SIGTTOU, SIG_IGN);
  71             // Set the process group
  72             result = tcsetpgrp (m_fd, m_processGroup);
  73             // Restore the original signal handler.
  74             signal (SIGTTOU, saved_sigttou_callback);
  75         }
  76         return true;
  77     }
  78     return false;
  79 }
  80
  81
  82
  83 TTYStateSwitcher::TTYStateSwitcher() :
  84     m_currentState(~0)
  85 {
  86 }
  87
  88 TTYStateSwitcher::~TTYStateSwitcher()
  89 {
  90 }
  91
  92 bool
  93 TTYStateSwitcher::GetState(uint32_t idx, int fd, bool saveProcessGroup)
  94 {
  95     if (ValidStateIndex(idx))
  96         return m_ttystates[idx].GetTTYState(fd, saveProcessGroup);
  97     return false;
  98 }
  99
 100 bool
 101 TTYStateSwitcher::SetState(uint32_t idx) const
 102 {
 103     if (!ValidStateIndex(idx))
 104         return false;
 105
 106     // See if we already are in this state?
 107     if (ValidStateIndex(m_currentState) && (idx == m_currentState) && m_ttystates[idx].IsValid())
 108         return true;
 109
 110     // Set the state to match the index passed in and only update the
 111     // current state if there are no errors.
 112     if (m_ttystates[idx].SetTTYState())
 113     {
 114         m_currentState = idx;
 115         return true;
 116     }
 117
 118     // We failed to set the state. The tty state was invalid or not
 119     // initialized.
 120     return false;
 121 }
 122