| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/system/data_pipe_consumer_dispatcher.h" | 5 #include "mojo/system/data_pipe_consumer_dispatcher.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "mojo/system/data_pipe.h" | 8 #include "mojo/system/data_pipe.h" |
| 9 #include "mojo/system/memory.h" | 9 #include "mojo/system/memory.h" |
| 10 | 10 |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 49 data_pipe_ = NULL; | 49 data_pipe_ = NULL; |
| 50 return scoped_refptr<Dispatcher>(rv.get()); | 50 return scoped_refptr<Dispatcher>(rv.get()); |
| 51 } | 51 } |
| 52 | 52 |
| 53 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock( | 53 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock( |
| 54 void* elements, | 54 void* elements, |
| 55 uint32_t* num_bytes, | 55 uint32_t* num_bytes, |
| 56 MojoReadDataFlags flags) { | 56 MojoReadDataFlags flags) { |
| 57 lock().AssertAcquired(); | 57 lock().AssertAcquired(); |
| 58 | 58 |
| 59 if (!VerifyUserPointer<uint32_t>(num_bytes)) | 59 if (!VerifyUserPointer<uint32_t>(num_bytes, 1)) |
| 60 return MOJO_RESULT_INVALID_ARGUMENT; | 60 return MOJO_RESULT_INVALID_ARGUMENT; |
| 61 | 61 |
| 62 if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) { | 62 if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) { |
| 63 // These flags are mutally exclusive. | 63 // These flags are mutally exclusive. |
| 64 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) | 64 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) |
| 65 return MOJO_RESULT_INVALID_ARGUMENT; | 65 return MOJO_RESULT_INVALID_ARGUMENT; |
| 66 DVLOG_IF(2, elements) << "Discard mode: ignoring non-null |elements|"; | 66 DVLOG_IF(2, elements) << "Discard mode: ignoring non-null |elements|"; |
| 67 return data_pipe_->ConsumerDiscardData( | 67 return data_pipe_->ConsumerDiscardData( |
| 68 num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | 68 num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); |
| 69 } | 69 } |
| 70 | 70 |
| 71 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) { | 71 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) { |
| 72 DCHECK(!(flags & MOJO_READ_DATA_FLAG_DISCARD)); // Handled above. | 72 DCHECK(!(flags & MOJO_READ_DATA_FLAG_DISCARD)); // Handled above. |
| 73 DVLOG_IF(2, elements) << "Query mode: ignoring non-null |elements|"; | 73 DVLOG_IF(2, elements) << "Query mode: ignoring non-null |elements|"; |
| 74 return data_pipe_->ConsumerQueryData(num_bytes); | 74 return data_pipe_->ConsumerQueryData(num_bytes); |
| 75 } | 75 } |
| 76 | 76 |
| 77 // Only verify |elements| if we're neither discarding nor querying. | 77 // Only verify |elements| if we're neither discarding nor querying. |
| 78 if (!VerifyUserPointerWithSize<1>(elements, *num_bytes)) | 78 if (!VerifyUserPointer<void>(elements, *num_bytes)) |
| 79 return MOJO_RESULT_INVALID_ARGUMENT; | 79 return MOJO_RESULT_INVALID_ARGUMENT; |
| 80 | 80 |
| 81 return data_pipe_->ConsumerReadData( | 81 return data_pipe_->ConsumerReadData( |
| 82 elements, num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | 82 elements, num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); |
| 83 } | 83 } |
| 84 | 84 |
| 85 MojoResult DataPipeConsumerDispatcher::BeginReadDataImplNoLock( | 85 MojoResult DataPipeConsumerDispatcher::BeginReadDataImplNoLock( |
| 86 const void** buffer, | 86 const void** buffer, |
| 87 uint32_t* buffer_num_bytes, | 87 uint32_t* buffer_num_bytes, |
| 88 MojoReadDataFlags flags) { | 88 MojoReadDataFlags flags) { |
| 89 lock().AssertAcquired(); | 89 lock().AssertAcquired(); |
| 90 | 90 |
| 91 if (!VerifyUserPointerWithCount<const void*>(buffer, 1)) | 91 if (!VerifyUserPointer<const void*>(buffer, 1)) |
| 92 return MOJO_RESULT_INVALID_ARGUMENT; | 92 return MOJO_RESULT_INVALID_ARGUMENT; |
| 93 if (!VerifyUserPointer<uint32_t>(buffer_num_bytes)) | 93 if (!VerifyUserPointer<uint32_t>(buffer_num_bytes, 1)) |
| 94 return MOJO_RESULT_INVALID_ARGUMENT; | 94 return MOJO_RESULT_INVALID_ARGUMENT; |
| 95 // These flags may not be used in two-phase mode. | 95 // These flags may not be used in two-phase mode. |
| 96 if ((flags & MOJO_READ_DATA_FLAG_DISCARD) || | 96 if ((flags & MOJO_READ_DATA_FLAG_DISCARD) || |
| 97 (flags & MOJO_READ_DATA_FLAG_QUERY)) | 97 (flags & MOJO_READ_DATA_FLAG_QUERY)) |
| 98 return MOJO_RESULT_INVALID_ARGUMENT; | 98 return MOJO_RESULT_INVALID_ARGUMENT; |
| 99 | 99 |
| 100 return data_pipe_->ConsumerBeginReadData( | 100 return data_pipe_->ConsumerBeginReadData( |
| 101 buffer, buffer_num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | 101 buffer, buffer_num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); |
| 102 } | 102 } |
| 103 | 103 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 121 data_pipe_->ConsumerRemoveWaiter(waiter); | 121 data_pipe_->ConsumerRemoveWaiter(waiter); |
| 122 } | 122 } |
| 123 | 123 |
| 124 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { | 124 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { |
| 125 lock().AssertAcquired(); | 125 lock().AssertAcquired(); |
| 126 return data_pipe_->ConsumerIsBusy(); | 126 return data_pipe_->ConsumerIsBusy(); |
| 127 } | 127 } |
| 128 | 128 |
| 129 } // namespace system | 129 } // namespace system |
| 130 } // namespace mojo | 130 } // namespace mojo |
| OLD | NEW |