| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |