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