[NaCl SDK] nacl_io: Add initial implementations of kill and signal

native_client_sdk/src/libraries/nacl_io/mount_node_tty.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "nacl_io/mount_node_tty.h"
 
 #include <assert.h>
 #include <errno.h>
+#include <signal.h>
 #include <stdio.h>
 #include <string.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
 
 #include <algorithm>
 
 #include "nacl_io/ioctl.h"
 #include "nacl_io/mount.h"
 #include "nacl_io/pepper_interface.h"
 #include "sdk_util/auto_lock.h"
 
 #define CHECK_LFLAG(TERMIOS, FLAG) (TERMIOS .c_lflag & FLAG)
 
 #define IS_ECHO CHECK_LFLAG(termios_, ECHO)
 #define IS_ECHOE CHECK_LFLAG(termios_, ECHOE)
 #define IS_ECHONL CHECK_LFLAG(termios_, ECHONL)
 #define IS_ECHOCTL CHECK_LFLAG(termios_, ECHOCTL)
 #define IS_ICANON CHECK_LFLAG(termios_, ICANON)
 
+#define DEFAULT_TTY_COLS 80
+#define DEFAULT_TTY_ROWS 30
+
 namespace nacl_io {
 
 MountNodeTty::MountNodeTty(Mount* mount) : MountNodeCharDevice(mount),
-                                           is_readable_(false) {
+                                           is_readable_(false),
+                                           did_resize_(false),
+                                           output_handler_(NULL),
+                                           rows_(DEFAULT_TTY_ROWS),
+                                           cols_(DEFAULT_TTY_COLS) {
   pthread_cond_init(&is_readable_cond_, NULL);
   InitTermios();
 }
 
 void MountNodeTty::InitTermios() {
   // Some sane values that produce good result.
   termios_.c_iflag = ICRNL | IXON | IXOFF | IUTF8;
   termios_.c_oflag = OPOST | ONLCR;
   termios_.c_cflag = CREAD | 077;
   termios_.c_lflag =
@@ skipping 20 matching lines @@
 }
 
 MountNodeTty::~MountNodeTty() {
   pthread_cond_destroy(&is_readable_cond_);
 }
 
 Error MountNodeTty::Write(size_t offs,
                      const void* buf,
                      size_t count,
                      int* out_bytes) {
-  return Write(offs, buf, count, out_bytes, false);
+  AUTO_LOCK(node_lock_);
+  return WriteWithLock(offs, buf, count, out_bytes);
 }
 
-Error MountNodeTty::Write(size_t offs,
+
+Error MountNodeTty::WriteWithLock(size_t offs,
                      const void* buf,
                      size_t count,
-                     int* out_bytes,
-                     bool locked) {
+                     int* out_bytes) {
   *out_bytes = 0;
 
-  if (!mount_->ppapi())
-    return ENOSYS;
+  // No handler registered.
+  if (output_handler_ == NULL)
+    return EIO;
 
-  MessagingInterface* msg_intr = mount_->ppapi()->GetMessagingInterface();
-  VarInterface* var_intr = mount_->ppapi()->GetVarInterface();
+  int rtn = output_handler_(reinterpret_cast<const char*>(buf), count);

[binji, 2013/08/22 18:09:36]

static_cast

[binji, 2013/08/22 18:09:36]

Not sure I like holding the node lock while calling a callback -- seems pretty
easy to accidentally deadlock. OTOH, we can't cache callback pointer, unlock,
and call through it, because the user may have changed the callback pointer (and
would not expect a call from the previous pointer).

I think we may need to add another lock, specifically for output_handler_.

[Sam Clegg, 2013/08/22 19:52:25]

Done.

[Sam Clegg, 2013/08/22 19:52:25]

Done.

+  if (rtn == -1)
+    return EIO;

[binji, 2013/08/22 18:09:36]

hm, we could delegate the errno return to the output_handler as well. What do
you think?

[Sam Clegg, 2013/08/22 19:52:25]

Done.

 
-  if (!(var_intr && msg_intr))
-    return ENOSYS;
-
-  // We append the prefix_ to the data in buf, then package it up
-  // and post it as a message.
-  const char* data = static_cast<const char*>(buf);
-  std::string message;
-  if (locked) {
-    message = prefix_;
-  } else {
-    AUTO_LOCK(node_lock_);
-    message = prefix_;
-  }
-
-  message.append(data, count);
-  uint32_t len = static_cast<uint32_t>(message.size());
-  struct PP_Var val = var_intr->VarFromUtf8(message.data(), len);
-  msg_intr->PostMessage(mount_->ppapi()->GetInstance(), val);
-  var_intr->Release(val);
-  *out_bytes = count;
+  *out_bytes = rtn;
   return 0;
 }
 
 Error MountNodeTty::Read(size_t offs, void* buf, size_t count, int* out_bytes) {
   AUTO_LOCK(node_lock_);
+  did_resize_ = false;
   while (!is_readable_) {
     pthread_cond_wait(&is_readable_cond_, node_lock_.mutex());
+    if (!is_readable_ && did_resize_) {
+      // If an async resize event occured then return the failure and
+      // set EINTR.
+      *out_bytes = 0;
+      return EINTR;
+    }
   }
 
   size_t bytes_to_copy = std::min(count, input_buffer_.size());
 
   if (IS_ICANON) {
     // Only read up to (and including) the first newline
     std::deque<char>::iterator nl = std::find(input_buffer_.begin(),
                                               input_buffer_.end(),
                                               '\n');
 
@@ skipping 20 matching lines @@
         std::find(input_buffer_.begin(),
                   input_buffer_.end(), '\n') != input_buffer_.end();
   else
     is_readable_ = input_buffer_.size() > 0;
 
   return 0;
 }
 
 Error MountNodeTty::Echo(const char* string, int count) {
   int wrote;
-  Error error = Write(0, string, count, &wrote, true);
+  Error error = WriteWithLock(0, string, count, &wrote);
   if (error != 0 || wrote != count) {
     // TOOD(sbc): Do something more useful in response to a
     // failure to echo.
     return error;
   }
 
   return 0;
 }
 
 Error MountNodeTty::ProcessInput(struct tioc_nacl_input_string* message) {
   AUTO_LOCK(node_lock_);
-  if (message->length < prefix_.size() ||
-      strncmp(message->buffer, prefix_.data(), prefix_.size()) != 0) {
-    return ENOTTY;
-  }
 
-  const char* buffer = message->buffer + prefix_.size();
-  int num_bytes = message->length - prefix_.size();
+  const char* buffer = message->buffer;
+  size_t num_bytes = message->length;
 
-  for (int i = 0; i < num_bytes; i++) {
+  for (size_t i = 0; i < num_bytes; i++) {
     char c = buffer[i];
     // Transform characters according to input flags.
     if (c == '\r') {
       if (termios_.c_iflag & IGNCR)
         continue;
       if (termios_.c_iflag & ICRNL)
         c = '\n';
     } else if (c == '\n') {
       if (termios_.c_iflag & INLCR)
         c = '\r';
@@ skipping 41 matching lines @@
       input_buffer_.push_back(c);
 
     if (c == '\n' || c == termios_.c_cc[VEOF] || !IS_ICANON)
       is_readable_ = true;
   }
 
   if (is_readable_) {
     RaiseEvent(POLLIN);
     pthread_cond_broadcast(&is_readable_cond_);
   }
+
   return 0;
 }
 
 Error MountNodeTty::Ioctl(int request, char* arg) {
-  if (request == TIOCNACLPREFIX) {
-    // This ioctl is used to change the prefix for this tty node.
-    // The prefix is used to distinguish messages intended for this
-    // tty node from all the other messages cluttering up the
-    // javascript postMessage() channel.
-    AUTO_LOCK(node_lock_);
-    prefix_ = arg;
-    return 0;
-  } else if (request == TIOCNACLINPUT) {
-    // This ioctl is used to deliver data from the user to this tty node's
-    // input buffer. We check if the prefix in the input data matches the
-    // prefix for this node, and only deliver the data if so.
-    struct tioc_nacl_input_string* message =
-      reinterpret_cast<struct tioc_nacl_input_string*>(arg);
-    return ProcessInput(message);
-  } else {
-    return EINVAL;
+  switch (request) {
+    case TIOCNACLOUTPUT: {
+      AUTO_LOCK(node_lock_);
+      if (arg == NULL) {
+        output_handler_ = NULL;
+        return 0;
+      }
+      if (output_handler_ != NULL)
+        return EALREADY;
+      output_handler_ = reinterpret_cast<nacl_io_tty_output_handler_t>(arg);
+      return 0;
+    }
+    case TIOCNACLINPUT: {
+      // This ioctl is used to deliver data from the user to this tty node's
+      // input buffer.
+      struct tioc_nacl_input_string* message =
+        reinterpret_cast<struct tioc_nacl_input_string*>(arg);
+      return ProcessInput(message);
+    }
+    case TIOCSWINSZ: {
+      struct winsize* size = reinterpret_cast<struct winsize*>(arg);
+      {
+        AUTO_LOCK(node_lock_);
+        rows_ = size->ws_row;
+        cols_ = size->ws_col;
+      }
+      kill(getpid(), SIGWINCH);
+
+      // Wake up any thread waiting on Read
+      {
+        AUTO_LOCK(node_lock_);
+        did_resize_ = true;
+        pthread_cond_broadcast(&is_readable_cond_);
+      }
+      return 0;
+    }
+    case TIOCGWINSZ: {
+      struct winsize* size = reinterpret_cast<struct winsize*>(arg);
+      size->ws_row = rows_;
+      size->ws_col = cols_;
+      return 0;
+    }
   }
+
+  return EINVAL;
 }
 
 Error MountNodeTty::Tcgetattr(struct termios* termios_p) {
   AUTO_LOCK(node_lock_);
   *termios_p = termios_;
   return 0;
 }
 
 Error MountNodeTty::Tcsetattr(int optional_actions,
                          const struct termios *termios_p) {
   AUTO_LOCK(node_lock_);
   termios_ = *termios_p;
   return 0;
 }
 
 }