Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Crashpad Authors. All rights reserved. | 1 // Copyright 2014 The Crashpad Authors. All rights reserved. |
| 2 // | 2 // |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | 3 // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 // you may not use this file except in compliance with the License. | 4 // you may not use this file except in compliance with the License. |
| 5 // You may obtain a copy of the License at | 5 // You may obtain a copy of the License at |
| 6 // | 6 // |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | 7 // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 // | 8 // |
| 9 // Unless required by applicable law or agreed to in writing, software | 9 // Unless required by applicable law or agreed to in writing, software |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | 10 // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 // See the License for the specific language governing permissions and | 12 // See the License for the specific language governing permissions and |
| 13 // limitations under the License. | 13 // limitations under the License. |
| 14 | 14 |
| 15 #ifndef CRASHPAD_UTIL_NUMERIC_IN_RANGE_CAST_H_ | 15 #include "util/file/file_io.h" |
| 16 #define CRASHPAD_UTIL_NUMERIC_IN_RANGE_CAST_H_ | |
| 17 | 16 |
| 18 #include "base/logging.h" | 17 #include "base/logging.h" |
| 19 #include "base/numerics/safe_conversions.h" | 18 #include "base/numerics/safe_conversions.h" |
| 20 | 19 |
| 21 namespace crashpad { | 20 namespace crashpad { |
| 22 | 21 |
| 23 //! \brief Casts to a different type if it can be done without data loss, | 22 // TODO(scottmg): Handle > DWORD sized writes if necessary. |
| 24 //! logging a warning message and returing a default value otherwise. | |
| 25 //! | |
| 26 //! \param[in] source The value to convert and return. | |
| 27 //! \param[in] default_value The default value to return, in the event that \a | |
| 28 //! source cannot be represented in the destination type. | |
| 29 //! | |
| 30 //! \return \a source if it can be represented in the destination type, | |
| 31 //! otherwise \a default_value. | |
| 32 template <typename Destination, typename Source> | |
| 33 Destination InRangeCast(Source source, Destination default_value) { | |
| 34 if (base::IsValueInRangeForNumericType<Destination>(source)) { | |
| 35 return static_cast<Destination>(source); | |
| 36 } | |
| 37 | 23 |
| 38 LOG(WARNING) << "value " << source << " out of range"; | 24 ssize_t ReadFile(FileHandle file, void* buffer, size_t size) { |
| 39 return static_cast<Destination>(default_value); | 25 DWORD size_dword = base::checked_cast<DWORD>(size); |
| 26 DWORD total_read = 0; | |
| 27 BOOL success; | |
| 28 do { | |
| 29 DWORD bytes_read; | |
| 30 success = ::ReadFile(file, buffer, size_dword, &bytes_read, nullptr); | |
|
Mark Mentovai
2014/12/17 01:05:24
On the second pass, buffer needs to point beyond i
scottmg
2014/12/17 01:50:27
Oh boy! Hopefully tests would have saved me eventu
| |
| 31 if (!success && GetLastError() != ERROR_MORE_DATA) | |
|
Mark Mentovai
2014/12/17 01:05:24
ReadFile()’s docs say that you’ll get ERROR_MORE_D
scottmg
2014/12/17 01:50:28
Raymond gets upset when Windows programmers poke a
Mark Mentovai
2014/12/17 05:39:16
scottmg wrote:
| |
| 32 return -1; | |
| 33 total_read += bytes_read; | |
| 34 } while (!success); | |
| 35 return total_read; | |
| 36 } | |
| 37 | |
| 38 ssize_t WriteFile(FileHandle file, const void* buffer, size_t size) { | |
| 39 // TODO(scottmg): This might need to handle the limit for pipes across a | |
| 40 // network in the future. | |
| 41 DWORD size_dword = base::checked_cast<DWORD>(size); | |
| 42 DWORD bytes_written; | |
| 43 BOOL rv = ::WriteFile(file, buffer, size_dword, &bytes_written, nullptr); | |
|
Mark Mentovai
2014/12/17 01:05:24
WriteFile()’s docs aren’t 100% convincing about sh
scottmg
2014/12/17 01:50:28
SG. The link above is also clear that it cannot su
| |
| 44 if (!rv) | |
| 45 return -1; | |
| 46 return bytes_written; | |
|
Mark Mentovai
2014/12/17 01:05:24
…so let’s make it explicit and [D]CHECK_EQ(bytes_w
scottmg
2014/12/17 01:50:28
Done.
| |
| 47 } | |
| 48 | |
| 49 bool LoggingCloseFile(FileHandle file) { | |
| 50 BOOL rv = CloseHandle(file); | |
| 51 PLOG_IF(ERROR, !rv) << "CloseHandle"; | |
| 52 return rv; | |
| 40 } | 53 } |
| 41 | 54 |
| 42 } // namespace crashpad | 55 } // namespace crashpad |
| 43 | |
| 44 #endif // CRASHPAD_UTIL_NUMERIC_IN_RANGE_CAST_H_ | |
| OLD | NEW |