Class ShellWebSocketClient

py/ovl.py:677–719

class ShellWebSocketClient(SSLEnabledWebSocketBaseClient):

  def __init__(self, state, output, *args, **kwargs):
    """Constructor.

    Args:
      output: output file object.
    """
    super().__init__(state, *args, **kwargs)
    self._output = output
    self._input_thread = threading.Thread(target=self._FeedInput)
    self._stop = threading.Event()

  def handshake_ok(self):
    pass

  def _FeedInput(self):
    try:
      while True:
        rd, unused_w, unused_x = select.select([sys.stdin], [], [], 0.5)
        if self._stop.is_set():
          break

        if sys.stdin in rd:
          data = sys.stdin.buffer.read()
          if not data:
            self.send(json.dumps({'type': 'stdin_close'}))
            break
          self.send(data, binary=True)
    except (KeyboardInterrupt, RuntimeError):
      pass

  def opened(self):
    self._input_thread.start()

  def closed(self, code, reason=None):
    self._stop.set()
    self._input_thread.join()

  def received_message(self, message):
    if message.is_binary:
      self._output.write(message.data)
      self._output.flush()


class PersistentShellWebSocketClient(SSLEnabledWebSocketBaseClient):