OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 |
| 11 #include "webrtc/base/file.h" |
| 12 |
| 13 #include <io.h> |
| 14 #include "webrtc/base/win32.h" |
| 15 |
| 16 #include <limits> // NOLINT: win32.h should be considered a system header |
| 17 |
| 18 #include "webrtc/base/checks.h" |
| 19 |
| 20 namespace rtc { |
| 21 |
| 22 File File::Open(const std::string& path) { |
| 23 HANDLE handle = |
| 24 ::CreateFile(ToUtf16(path).c_str(), GENERIC_READ | GENERIC_WRITE, 0, |
| 25 nullptr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr); |
| 26 return File(handle); |
| 27 } |
| 28 |
| 29 size_t File::Write(const uint8_t* data, size_t length) { |
| 30 RTC_DCHECK_LT(length, std::numeric_limits<DWORD>::max()); |
| 31 size_t total_written = 0; |
| 32 do { |
| 33 DWORD written; |
| 34 if (!::WriteFile(file_, data + total_written, |
| 35 static_cast<DWORD>(length - total_written), &written, |
| 36 nullptr)) { |
| 37 break; |
| 38 } |
| 39 total_written += written; |
| 40 } while (total_written < length); |
| 41 return total_written; |
| 42 } |
| 43 |
| 44 size_t File::Read(uint8_t* buffer, size_t length) { |
| 45 RTC_DCHECK_LT(length, std::numeric_limits<DWORD>::max()); |
| 46 size_t total_read = 0; |
| 47 do { |
| 48 DWORD read; |
| 49 if (!::ReadFile(file_, buffer + total_read, |
| 50 static_cast<DWORD>(length - total_read), &read, nullptr)) { |
| 51 break; |
| 52 } |
| 53 total_read += read; |
| 54 } while (total_read < length); |
| 55 return total_read; |
| 56 } |
| 57 |
| 58 size_t File::WriteAt(const uint8_t* data, size_t length, size_t offset) { |
| 59 RTC_DCHECK_LT(length, std::numeric_limits<DWORD>::max()); |
| 60 size_t total_written = 0; |
| 61 do { |
| 62 DWORD written; |
| 63 |
| 64 LARGE_INTEGER offset_li; |
| 65 offset_li.QuadPart = offset + total_written; |
| 66 |
| 67 OVERLAPPED overlapped = {0}; |
| 68 overlapped.Offset = offset_li.LowPart; |
| 69 overlapped.OffsetHigh = offset_li.HighPart; |
| 70 |
| 71 if (!::WriteFile(file_, data + total_written, |
| 72 static_cast<DWORD>(length - total_written), &written, |
| 73 &overlapped)) { |
| 74 break; |
| 75 } |
| 76 |
| 77 total_written += written; |
| 78 } while (total_written < length); |
| 79 return total_written; |
| 80 } |
| 81 |
| 82 size_t File::ReadAt(uint8_t* buffer, size_t length, size_t offset) { |
| 83 RTC_DCHECK_LT(length, std::numeric_limits<DWORD>::max()); |
| 84 size_t total_read = 0; |
| 85 do { |
| 86 DWORD read; |
| 87 |
| 88 LARGE_INTEGER offset_li; |
| 89 offset_li.QuadPart = offset + total_read; |
| 90 |
| 91 OVERLAPPED overlapped = {0}; |
| 92 overlapped.Offset = offset_li.LowPart; |
| 93 overlapped.OffsetHigh = offset_li.HighPart; |
| 94 |
| 95 if (!::ReadFile(file_, buffer + total_read, |
| 96 static_cast<DWORD>(length - total_read), &read, |
| 97 &overlapped)) { |
| 98 break; |
| 99 } |
| 100 |
| 101 total_read += read; |
| 102 } while (total_read < length); |
| 103 return total_read; |
| 104 } |
| 105 |
| 106 bool File::Seek(size_t offset) { |
| 107 LARGE_INTEGER distance; |
| 108 distance.QuadPart = offset; |
| 109 return SetFilePointerEx(file_, distance, nullptr, FILE_BEGIN) != 0; |
| 110 } |
| 111 |
| 112 bool File::Close() { |
| 113 if (file_ == kInvalidPlatformFileValue) |
| 114 return false; |
| 115 bool ret = CloseHandle(file_) != 0; |
| 116 file_ = kInvalidPlatformFileValue; |
| 117 return ret; |
| 118 } |
| 119 |
| 120 } // namespace rtc |
OLD | NEW |