[NaCl SDK] nacl_io: Add flow control the JavaScript pipes.

native_client_sdk/src/examples/demo/nacl_io/nacl_io_demo.c

 /* Copyright (c) 2012 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_demo.h"
 
 #include <assert.h>
+#include <errno.h>
+#include <fcntl.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/ioctl.h>
 #include <sys/mount.h>
+#include <sys/param.h>
+#include <sys/select.h>
+#include <sys/stat.h>
 #include <pthread.h>
+#include <unistd.h>
 
 #include "ppapi/c/pp_errors.h"
 #include "ppapi/c/pp_module.h"
 #include "ppapi/c/ppb.h"
 #include "ppapi/c/ppb_instance.h"
 #include "ppapi/c/ppb_messaging.h"
 #include "ppapi/c/ppb_var.h"
 #include "ppapi/c/ppb_var_array.h"
 #include "ppapi/c/ppb_var_dictionary.h"
 #include "ppapi/c/ppp.h"
 #include "ppapi/c/ppp_instance.h"
 #include "ppapi/c/ppp_messaging.h"
+#include "nacl_io/ioctl.h"
 #include "nacl_io/nacl_io.h"
 
 #include "handlers.h"
 #include "queue.h"
 
 #if defined(WIN32)
 #define va_copy(d, s) ((d) = (s))
 #endif
 
 typedef struct {
   const char* name;
   HandleFunc function;
 } FuncNameMapping;
 
-PP_Instance g_instance = 0;
-PPB_GetInterface g_get_browser_interface = NULL;
-PPB_Messaging* g_ppb_messaging = NULL;
+static PP_Instance g_instance = 0;
+static PPB_GetInterface g_get_browser_interface = NULL;
+static PPB_Messaging* g_ppb_messaging = NULL;
 PPB_Var* g_ppb_var = NULL;
 PPB_VarArray* g_ppb_var_array = NULL;
 PPB_VarDictionary* g_ppb_var_dictionary = NULL;
 
 static FuncNameMapping g_function_map[] = {
     {"fopen", HandleFopen},
     {"fwrite", HandleFwrite},
     {"fread", HandleFread},
     {"fseek", HandleFseek},
     {"fclose", HandleFclose},
@@ skipping 10 matching lines @@
     {"gethostbyname", HandleGethostbyname},
     {"connect", HandleConnect},
     {"send", HandleSend},
     {"recv", HandleRecv},
     {"close", HandleClose},
     {NULL, NULL},
 };
 
 /** A handle to the thread the handles messages. */
 static pthread_t g_handle_message_thread;
+static pthread_t g_echo_thread;
 
 /**
  * Create a new PP_Var from a C string.
  * @param[in] str The string to convert.
  * @return A new PP_Var with the contents of |str|.
  */
 struct PP_Var CStrToVar(const char* str) {
   return g_ppb_var->VarFromUtf8(str, strlen(str));
 }
 
@@ skipping 49 matching lines @@
   string = VprintfToNewString(format, args);
   va_end(args);
 
   var = g_ppb_var->VarFromUtf8(string, strlen(string));
   free(string);
 
   return var;
 }
 
 /**
- * Convert a PP_Var to a C string, given a buffer.
+ * Convert a PP_Var to a C string.
  * @param[in] var The PP_Var to convert.
- * @param[out] buffer The buffer to write to.
- * @param[in] length The length of |buffer|.
- * @return The number of characters written.
+ * @return A newly allocated, NULL-terminated string.
  */
-const char* VarToCStr(struct PP_Var var) {
+static const char* VarToCStr(struct PP_Var var) {
   uint32_t length;
   const char* str = g_ppb_var->VarToUtf8(var, &length);
   if (str == NULL) {
     return NULL;
   }
 
   /* str is NOT NULL-terminated. Copy using memcpy. */
   char* new_str = (char*)malloc(length + 1);
   memcpy(new_str, str, length);
   new_str[length] = 0;
@@ skipping 113 matching lines @@
     /* Post the error to JavaScript, so the user can see it. */
     PostMessageVar(var);
     return;
   }
 
   /* Function returned an output dictionary. Send it to JavaScript. */
   PostMessageVar(result_var);
   g_ppb_var->Release(result_var);
 }
 
+
+/**
+ * Helper function used by EchoThread which reads from a file descriptor
+ * and writes all the data that it reads back to the same descriptor.
+ */
+static void EchoInput(int fd) {
+  char buffer[512];
+  while (1) {
+    int rtn = read(fd, buffer, 512);
+    if (rtn > 0) {
+      int wrote = write(fd, buffer, rtn);
+      if (wrote < rtn)
+        fprintf(stderr, "only wrote %d/%d bytes\n", wrote, rtn);

[binji, 2014/05/01 20:22:31]

might be nice to send these errors to JavaScript, rather than the console.

[Sam Clegg, 2014/05/01 22:16:55]

Done.

+    } else {
+      if (rtn < 0 && errno != EAGAIN)
+        fprintf(stderr, "read failed: %d (%s)\n", errno, strerror(errno));
+      break;
+    }
+  }
+}
+
+/**
+ * Worker thread that listens for input on JS pipe nodes and echos all input
+ * back to the same pipe.
+ */
+static void* EchoThread(void* user_data) {
+  int fd1 = open("/dev/jspipe1", O_RDWR | O_NONBLOCK);
+  int fd2 = open("/dev/jspipe2", O_RDWR | O_NONBLOCK);
+  int fd3 = open("/dev/jspipe3", O_RDWR | O_NONBLOCK);
+  int nfds = MAX(fd1, fd2);
+  nfds = MAX(nfds, fd3);
+  while (1) {
+    fd_set readfds;
+    FD_ZERO(&readfds);
+    FD_SET(fd1, &readfds);
+    FD_SET(fd2, &readfds);
+    FD_SET(fd3, &readfds);
+    int rtn = select(nfds + 1, &readfds, NULL, NULL, NULL);
+    if (rtn < 0 && errno != EAGAIN) {
+      fprintf(stderr, "select failed: %s\n", strerror(errno));
+      break;
+    }
+    if (rtn > 0) {
+      if (FD_ISSET(fd1, &readfds))
+        EchoInput(fd1);
+      if (FD_ISSET(fd2, &readfds))
+        EchoInput(fd2);
+      if (FD_ISSET(fd3, &readfds))
+        EchoInput(fd3);
+    }
+
+  }
+  close(fd1);
+  close(fd2);
+  close(fd3);
+  return 0;
+}
+
 /**
  * A worker thread that handles messages from JavaScript.
  * @param[in] user_data Unused.
  * @return unused.
  */
 void* HandleMessageThread(void* user_data) {
   while (1) {
     struct PP_Var message = DequeueMessage();
     HandleMessage(message);
     g_ppb_var->Release(message);
@@ skipping 19 matching lines @@
         0,                                        /* mountflags */
         "type=PERSISTENT,expected_size=1048576"); /* data */
 
   mount("",       /* source. Use relative URL */
         "/http",  /* target */
         "httpfs", /* filesystemtype */
         0,        /* mountflags */
         "");      /* data */
 
   pthread_create(&g_handle_message_thread, NULL, &HandleMessageThread, NULL);
+  pthread_create(&g_echo_thread, NULL, &EchoThread, NULL);
   InitializeMessageQueue();
 
   return PP_TRUE;
 }
 
 static void Instance_DidDestroy(PP_Instance instance) {
 }
 
 static void Instance_DidChangeView(PP_Instance instance,
                                    PP_Resource view_resource) {
 }
 
 static void Instance_DidChangeFocus(PP_Instance instance, PP_Bool has_focus) {
 }
 
 static PP_Bool Instance_HandleDocumentLoad(PP_Instance instance,
                                            PP_Resource url_loader) {
   /* NaCl modules do not need to handle the document load function. */
   return PP_FALSE;
 }
 
 static void Messaging_HandleMessage(PP_Instance instance,
                                     struct PP_Var message) {
+  /* Special case for jspipe input handling */
+  if (message.type == PP_VARTYPE_DICTIONARY) {

[binji, 2014/05/01 20:22:31]

this won't work anymore because all messages are dictionaries.

[Sam Clegg, 2014/05/01 22:16:55]

Done.

+    struct PP_Var keys = g_ppb_var_dictionary->GetKeys(message);
+    struct PP_Var key = g_ppb_var_array->Get(keys, 0);
+    const char* key_string = VarToCStr(key);
+    char pipe_name[64];
+
+    sprintf(pipe_name, "/dev/%s", key_string);

[binji, 2014/05/01 20:22:31]

snprintf

+    int fd = open(pipe_name, O_RDONLY);
+    if (fd < 0) {
+      struct PP_Var var = PrintfToVar("Warning: opening %s failed.",
+                                      pipe_name);
+      g_ppb_messaging->PostMessage(g_instance, var);

[binji, 2014/05/01 20:22:31]

Use PostMessageVar to cleanup the newly created string variable.

[Sam Clegg, 2014/05/01 22:16:55]

Done.

+      return;

[binji, 2014/05/01 20:22:31]

release PP_Vars on failure

[Sam Clegg, 2014/05/01 22:16:55]

Done.

+    }
+    struct PP_Var payload = g_ppb_var_dictionary->Get(message, key);
+    if (ioctl(fd, NACL_IOC_HANDLEMESSAGE, (char*)&payload) != 0) {
+      struct PP_Var var = PrintfToVar("Error: ioctl on %s failed: %s",
+                                      pipe_name, strerror(errno));
+      g_ppb_messaging->PostMessage(g_instance, var);
+    }
+    g_ppb_var->Release(keys);
+    g_ppb_var->Release(key);
+    g_ppb_var->Release(payload);
+    close(fd);
+    return;
+  }
+
   g_ppb_var->AddRef(message);
   if (!EnqueueMessage(message)) {
     g_ppb_var->Release(message);
     PostMessageVar(
         PrintfToVar("Warning: dropped message because the queue was full."));
   }
 }
 
 #define GET_INTERFACE(var, type, name)            \
   var = (type*)(get_browser(name));               \
@@ skipping 27 matching lines @@
     static PPP_Messaging messaging_interface = {
         &Messaging_HandleMessage,
     };
     return &messaging_interface;
   }
   return NULL;
 }
 
 PP_EXPORT void PPP_ShutdownModule() {
 }