The Tor Browser: b2g/simulator/packages/subprocess/lib/subprocess_worker_win.js@b8a032363ba2 (annotated)

b2g/simulator/packages/subprocess/lib/subprocess_worker_win.js@b8a032363ba2 (annotated)

b2g/simulator/packages/subprocess/lib/subprocess_worker_win.js

Thu, 22 Jan 2015 13:21:57 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 22 Jan 2015 13:21:57 +0100
branch: TOR_BUG_9701
changeset 15: b8a032363ba2
permissions: -rw-r--r--

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
  */
 /*
  * ChromeWorker Object subprocess.jsm on Windows to process stdin/stdout/stderr
  * on separate threads.
  *
  */
 // Being a ChromeWorker object, implicitly uses the following:
 // Components.utils.import("resource://gre/modules/ctypes.jsm");
 'use strict';
 const BufferSize = 1024;
 const BOOL = ctypes.bool;
 const HANDLE = ctypes.size_t;
 const DWORD = ctypes.uint32_t;
 const LPDWORD = DWORD.ptr;
 const PVOID = ctypes.voidptr_t;
 const LPVOID = PVOID;
 /*
 typedef struct _OVERLAPPED {
   ULONG_PTR Internal;
   ULONG_PTR InternalHigh;
   union {
     struct {
       DWORD Offset;
       DWORD OffsetHigh;
     };
     PVOID  Pointer;
   };
   HANDLE    hEvent;
 } OVERLAPPED, *LPOVERLAPPED;
 */
 const OVERLAPPED = new ctypes.StructType("OVERLAPPED");
 var ReadFileBuffer = ctypes.char.array(BufferSize);
 var WriteFileBuffer = ctypes.uint8_t.array(BufferSize);
 var kernel32dll = null;
 var libFunc = {};
 function initLib(libName) {
     if (ctypes.size_t.size == 8) {
         var WinABI = ctypes.default_abi;
     } else {
         var WinABI = ctypes.winapi_abi;
     }
     kernel32dll = ctypes.open(libName);
     /*
     BOOL WINAPI WriteFile(
       __in         HANDLE hFile,
       __in         LPCVOID lpBuffer,
       __in         DWORD nNumberOfBytesToWrite,
       __out_opt    LPDWORD lpNumberOfBytesWritten,
       __inout_opt  LPOVERLAPPED lpOverlapped
     );
     NOTE: lpBuffer is declared as array of unsigned int8 instead of char to avoid
            implicit charset conversion
     */
     libFunc.WriteFile = kernel32dll.declare("WriteFile",
                                         WinABI,
                                         BOOL,
                                         HANDLE,
                                         WriteFileBuffer,
                                         DWORD,
                                         LPDWORD,
                                         OVERLAPPED.ptr
     );
     /*
     BOOL WINAPI ReadFile(
       __in         HANDLE hFile,
       __out        LPVOID ReadFileBuffer,
       __in         DWORD nNumberOfBytesToRead,
       __out_opt    LPDWORD lpNumberOfBytesRead,
       __inout_opt  LPOVERLAPPED lpOverlapped
     );
     */
     libFunc.ReadFile = kernel32dll.declare("ReadFile",
                                         WinABI,
                                         BOOL,
                                         HANDLE,
                                         ReadFileBuffer,
                                         DWORD,
                                         LPDWORD,
                                         OVERLAPPED.ptr
     );
     /*
     BOOL WINAPI CloseHandle(
       __in  HANDLE hObject
     );
     */
     libFunc.CloseHandle = kernel32dll.declare("CloseHandle",
                                             WinABI,
                                             BOOL,
                                             HANDLE
     );
 }
 function writePipe(pipe, data) {
     var bytesWritten = DWORD(0);
     var pData = new WriteFileBuffer();
     var numChunks = Math.floor(data.length / BufferSize);
     for (var chunk = 0; chunk <= numChunks; chunk ++) {
         var numBytes = chunk < numChunks ? BufferSize : data.length - chunk * BufferSize;
         for (var i=0; i < numBytes; i++) {
             pData[i] = data.charCodeAt(chunk * BufferSize + i) % 256;
         }
       var r = libFunc.WriteFile(pipe, pData, numBytes, bytesWritten.address(), null);
       if (bytesWritten.value != numBytes)
           throw("error: wrote "+bytesWritten.value+" instead of "+numBytes+" bytes");
     }
     postMessage("wrote "+data.length+" bytes of data");
 }
 function readString(data, length, charset) {
     var string = '', bytes = [];
     for(var i = 0;i < length; i++) {
         if(data[i] == 0 && charset != "null") // stop on NULL character for non-binary data
            break;
         bytes.push(data[i]);
     }
     return bytes;
 }
 function readPipe(pipe, charset) {
     while (true) {
         var bytesRead = DWORD(0);
         var line = new ReadFileBuffer();
         var r = libFunc.ReadFile(pipe, line, BufferSize, bytesRead.address(), null);
         if (!r) {
             // stop if we get an error (such as EOF reached)
             postMessage({msg: "info", data: "ReadFile failed"});
             break;
         }
         if (bytesRead.value > 0) {
             var c = readString(line, bytesRead.value, charset);
             postMessage({msg: "data", data: c, count: c.length});
         }
         else {
             break;
         }
     }
     libFunc.CloseHandle(pipe);
     postMessage({msg: "done"});
     kernel32dll.close();
     close();
 }
 onmessage = function (event) {
     let pipePtr;
     switch (event.data.msg) {
     case "init":
         initLib(event.data.libc);
         break;
     case "write":
         // data contents:
         //   msg: 'write'
         //   data: the data (string) to write
         //   pipe: ptr to pipe
         pipePtr = HANDLE.ptr(event.data.pipe);
         writePipe(pipePtr.contents, event.data.data);
         postMessage("WriteOK");
         break;
     case "read":
         initLib(event.data.libc);
         pipePtr = HANDLE.ptr(event.data.pipe);
         readPipe(pipePtr.contents, event.data.charset);
         break;
     case "close":
         pipePtr = HANDLE.ptr(event.data.pipe);
         postMessage("closing stdin\n");
         if (libFunc.CloseHandle(pipePtr.contents)) {
             postMessage("ClosedOK");
         }
         else
             postMessage("Could not close stdin handle");
         break;
     case "stop":
         kernel32dll.close();
         close();
         break;
     default:
         throw("error: Unknown command"+event.data.msg+"\n");
     }
     return;
 };