Home | History | Annotate | Download | only in test 
      1 // Copyright 2014 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "mojo/common/test/test_utils.h"
      6 
      7 #include <fcntl.h>
      8 #include <io.h>
      9 #include <string.h>
     10 #include <windows.h>
     11 
     12 #include "base/base_paths.h"
     13 #include "base/path_service.h"
     14 #include "base/strings/string_util.h"
     15 
     16 namespace mojo {
     17 namespace test {
     18 
     19 bool BlockingWrite(const embedder::PlatformHandle& handle,
     20                    const void* buffer,
     21                    size_t bytes_to_write,
     22                    size_t* bytes_written) {
     23   OVERLAPPED overlapped = { 0 };
     24   DWORD bytes_written_dword = 0;
     25 
     26   if (!WriteFile(handle.handle, buffer, static_cast<DWORD>(bytes_to_write),
     27                  &bytes_written_dword, &overlapped)) {
     28     if (GetLastError() != ERROR_IO_PENDING ||
     29         !GetOverlappedResult(handle.handle, &overlapped, &bytes_written_dword,
     30                              TRUE)) {
     31       return false;
     32     }
     33   }
     34 
     35   *bytes_written = bytes_written_dword;
     36   return true;
     37 }
     38 
     39 bool BlockingRead(const embedder::PlatformHandle& handle,
     40                   void* buffer,
     41                   size_t buffer_size,
     42                   size_t* bytes_read) {
     43   OVERLAPPED overlapped = { 0 };
     44   DWORD bytes_read_dword = 0;
     45 
     46   if (!ReadFile(handle.handle, buffer, static_cast<DWORD>(buffer_size),
     47                 &bytes_read_dword, &overlapped)) {
     48     if (GetLastError() != ERROR_IO_PENDING ||
     49         !GetOverlappedResult(handle.handle, &overlapped, &bytes_read_dword,
     50                              TRUE)) {
     51       return false;
     52     }
     53   }
     54 
     55   *bytes_read = bytes_read_dword;
     56   return true;
     57 }
     58 
     59 bool NonBlockingRead(const embedder::PlatformHandle& handle,
     60                      void* buffer,
     61                      size_t buffer_size,
     62                      size_t* bytes_read) {
     63   OVERLAPPED overlapped = { 0 };
     64   DWORD bytes_read_dword = 0;
     65 
     66   if (!ReadFile(handle.handle, buffer, static_cast<DWORD>(buffer_size),
     67                 &bytes_read_dword, &overlapped)) {
     68     if (GetLastError() != ERROR_IO_PENDING)
     69       return false;
     70 
     71     CancelIo(handle.handle);
     72 
     73     if (!GetOverlappedResult(handle.handle, &overlapped, &bytes_read_dword,
     74                              TRUE)) {
     75       *bytes_read = 0;
     76       return true;
     77     }
     78   }
     79 
     80   *bytes_read = bytes_read_dword;
     81   return true;
     82 }
     83 
     84 embedder::ScopedPlatformHandle PlatformHandleFromFILE(base::ScopedFILE fp) {
     85   CHECK(fp);
     86 
     87   HANDLE rv = INVALID_HANDLE_VALUE;
     88   PCHECK(DuplicateHandle(
     89       GetCurrentProcess(),
     90       reinterpret_cast<HANDLE>(_get_osfhandle(_fileno(fp.get()))),
     91       GetCurrentProcess(),
     92       &rv,
     93       0,
     94       TRUE,
     95       DUPLICATE_SAME_ACCESS)) << "DuplicateHandle";
     96   return embedder::ScopedPlatformHandle(embedder::PlatformHandle(rv));
     97 }
     98 
     99 base::ScopedFILE FILEFromPlatformHandle(embedder::ScopedPlatformHandle h,
    100                                         const char* mode) {
    101   CHECK(h.is_valid());
    102   // Microsoft's documentation for |_open_osfhandle()| only discusses these
    103   // flags (and |_O_WTEXT|). Hmmm.
    104   int flags = 0;
    105   if (strchr(mode, 'a'))
    106     flags |= _O_APPEND;
    107   if (strchr(mode, 'r'))
    108     flags |= _O_RDONLY;
    109   if (strchr(mode, 't'))
    110     flags |= _O_TEXT;
    111   base::ScopedFILE rv(
    112       _fdopen(_open_osfhandle(reinterpret_cast<intptr_t>(h.release().handle),
    113                               flags),
    114               mode));
    115   PCHECK(rv) << "_fdopen";
    116   return rv.Pass();
    117 }
    118 
    119 base::FilePath GetFilePathForJSResource(const std::string& path) {
    120   std::string binding_path = "gen/" + path + ".js";
    121   base::ReplaceChars(binding_path, "//", "\\", &binding_path);
    122   base::FilePath exe_dir;
    123   PathService::Get(base::DIR_EXE, &exe_dir);
    124   return exe_dir.AppendASCII(binding_path);
    125 }
    126 
    127 }  // namespace test
    128 }  // namespace mojo
    129