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/common/data_pipe_utils.h" | 5 #include "mojo/common/data_pipe_utils.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "mojo/public/cpp/system/wait.h" |
10 | 11 |
11 namespace mojo { | 12 namespace mojo { |
12 namespace common { | 13 namespace common { |
13 namespace { | 14 namespace { |
14 | 15 |
15 bool BlockingCopyHelper(ScopedDataPipeConsumerHandle source, | 16 bool BlockingCopyHelper(ScopedDataPipeConsumerHandle source, |
16 const base::Callback<size_t(const void*, uint32_t)>& write_bytes) { | 17 const base::Callback<size_t(const void*, uint32_t)>& write_bytes) { |
17 for (;;) { | 18 for (;;) { |
18 const void* buffer; | 19 const void* buffer; |
19 uint32_t num_bytes; | 20 uint32_t num_bytes; |
20 MojoResult result = BeginReadDataRaw( | 21 MojoResult result = BeginReadDataRaw( |
21 source.get(), &buffer, &num_bytes, MOJO_READ_DATA_FLAG_NONE); | 22 source.get(), &buffer, &num_bytes, MOJO_READ_DATA_FLAG_NONE); |
22 if (result == MOJO_RESULT_OK) { | 23 if (result == MOJO_RESULT_OK) { |
23 size_t bytes_written = write_bytes.Run(buffer, num_bytes); | 24 size_t bytes_written = write_bytes.Run(buffer, num_bytes); |
24 result = EndReadDataRaw(source.get(), num_bytes); | 25 result = EndReadDataRaw(source.get(), num_bytes); |
25 if (bytes_written < num_bytes || result != MOJO_RESULT_OK) | 26 if (bytes_written < num_bytes || result != MOJO_RESULT_OK) |
26 return false; | 27 return false; |
27 } else if (result == MOJO_RESULT_SHOULD_WAIT) { | 28 } else if (result == MOJO_RESULT_SHOULD_WAIT) { |
28 result = Wait(source.get(), | 29 result = Wait(source.get(), MOJO_HANDLE_SIGNAL_READABLE); |
29 MOJO_HANDLE_SIGNAL_READABLE, | |
30 MOJO_DEADLINE_INDEFINITE, | |
31 nullptr); | |
32 if (result != MOJO_RESULT_OK) { | 30 if (result != MOJO_RESULT_OK) { |
33 // If the producer handle was closed, then treat as EOF. | 31 // If the producer handle was closed, then treat as EOF. |
34 return result == MOJO_RESULT_FAILED_PRECONDITION; | 32 return result == MOJO_RESULT_FAILED_PRECONDITION; |
35 } | 33 } |
36 } else if (result == MOJO_RESULT_FAILED_PRECONDITION) { | 34 } else if (result == MOJO_RESULT_FAILED_PRECONDITION) { |
37 // If the producer handle was closed, then treat as EOF. | 35 // If the producer handle was closed, then treat as EOF. |
38 return true; | 36 return true; |
39 } else { | 37 } else { |
40 // Some other error occurred. | 38 // Some other error occurred. |
41 break; | 39 break; |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
75 if (result == MOJO_RESULT_OK) { | 73 if (result == MOJO_RESULT_OK) { |
76 char* char_buffer = static_cast<char*>(buffer); | 74 char* char_buffer = static_cast<char*>(buffer); |
77 uint32_t byte_index = 0; | 75 uint32_t byte_index = 0; |
78 while (it != source.end() && byte_index < buffer_num_bytes) { | 76 while (it != source.end() && byte_index < buffer_num_bytes) { |
79 char_buffer[byte_index++] = *it++; | 77 char_buffer[byte_index++] = *it++; |
80 } | 78 } |
81 EndWriteDataRaw(destination.get(), byte_index); | 79 EndWriteDataRaw(destination.get(), byte_index); |
82 if (it == source.end()) | 80 if (it == source.end()) |
83 return true; | 81 return true; |
84 } else if (result == MOJO_RESULT_SHOULD_WAIT) { | 82 } else if (result == MOJO_RESULT_SHOULD_WAIT) { |
85 result = Wait(destination.get(), MOJO_HANDLE_SIGNAL_WRITABLE, | 83 result = Wait(destination.get(), MOJO_HANDLE_SIGNAL_WRITABLE); |
86 MOJO_DEADLINE_INDEFINITE, nullptr); | |
87 if (result != MOJO_RESULT_OK) { | 84 if (result != MOJO_RESULT_OK) { |
88 // If the consumer handle was closed, then treat as EOF. | 85 // If the consumer handle was closed, then treat as EOF. |
89 return result == MOJO_RESULT_FAILED_PRECONDITION; | 86 return result == MOJO_RESULT_FAILED_PRECONDITION; |
90 } | 87 } |
91 } else { | 88 } else { |
92 // If the consumer handle was closed, then treat as EOF. | 89 // If the consumer handle was closed, then treat as EOF. |
93 return result == MOJO_RESULT_FAILED_PRECONDITION; | 90 return result == MOJO_RESULT_FAILED_PRECONDITION; |
94 } | 91 } |
95 } | 92 } |
96 } | 93 } |
97 | 94 |
98 } // namespace common | 95 } // namespace common |
99 } // namespace mojo | 96 } // namespace mojo |
OLD | NEW |