| 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 <stdio.h> | 7 #include <stdio.h> |
| 8 | 8 |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 base::ScopedFILE fp(base::OpenFile(destination, "wb")); | 21 base::ScopedFILE fp(base::OpenFile(destination, "wb")); |
| 22 if (!fp) | 22 if (!fp) |
| 23 return false; | 23 return false; |
| 24 | 24 |
| 25 for (;;) { | 25 for (;;) { |
| 26 const void* buffer; | 26 const void* buffer; |
| 27 uint32_t num_bytes; | 27 uint32_t num_bytes; |
| 28 MojoResult result = BeginReadDataRaw(source.get(), &buffer, &num_bytes, | 28 MojoResult result = BeginReadDataRaw(source.get(), &buffer, &num_bytes, |
| 29 MOJO_READ_DATA_FLAG_NONE); | 29 MOJO_READ_DATA_FLAG_NONE); |
| 30 if (result == MOJO_RESULT_OK) { | 30 if (result == MOJO_RESULT_OK) { |
| 31 fwrite(buffer, 1, num_bytes, fp.get()); | 31 size_t bytes_written = fwrite(buffer, 1, num_bytes, fp.get()); |
| 32 result = EndReadDataRaw(source.get(), num_bytes); | 32 result = EndReadDataRaw(source.get(), num_bytes); |
| 33 if (result != MOJO_RESULT_OK) | 33 if (bytes_written < num_bytes || result != MOJO_RESULT_OK) |
| 34 return false; | 34 return false; |
| 35 } else if (result == MOJO_RESULT_SHOULD_WAIT) { | 35 } else if (result == MOJO_RESULT_SHOULD_WAIT) { |
| 36 result = Wait(source.get(), | 36 result = Wait(source.get(), |
| 37 MOJO_WAIT_FLAG_READABLE, | 37 MOJO_WAIT_FLAG_READABLE, |
| 38 MOJO_DEADLINE_INDEFINITE); | 38 MOJO_DEADLINE_INDEFINITE); |
| 39 if (result != MOJO_RESULT_OK) { | 39 if (result != MOJO_RESULT_OK) { |
| 40 // If the producer handle was closed, then treat as EOF. | 40 // If the producer handle was closed, then treat as EOF. |
| 41 return result == MOJO_RESULT_FAILED_PRECONDITION; | 41 return result == MOJO_RESULT_FAILED_PRECONDITION; |
| 42 } | 42 } |
| 43 } else if (result == MOJO_RESULT_FAILED_PRECONDITION) { | 43 } else if (result == MOJO_RESULT_FAILED_PRECONDITION) { |
| (...skipping 19 matching lines...) Expand all Loading... |
| 63 const base::Callback<void(bool)>& callback) { | 63 const base::Callback<void(bool)>& callback) { |
| 64 base::PostTaskAndReplyWithResult( | 64 base::PostTaskAndReplyWithResult( |
| 65 task_runner, | 65 task_runner, |
| 66 FROM_HERE, | 66 FROM_HERE, |
| 67 base::Bind(&BlockingCopyToFile, base::Passed(&source), destination), | 67 base::Bind(&BlockingCopyToFile, base::Passed(&source), destination), |
| 68 callback); | 68 callback); |
| 69 } | 69 } |
| 70 | 70 |
| 71 } // namespace common | 71 } // namespace common |
| 72 } // namespace mojo | 72 } // namespace mojo |
| OLD | NEW |