xref: /aosp_15_r20/art/libartbase/base/unix_file/fd_file.cc (revision 795d594fd825385562da6b089ea9b2033f3abf5a)
1*795d594fSAndroid Build Coastguard Worker /*
2*795d594fSAndroid Build Coastguard Worker  * Copyright (C) 2009 The Android Open Source Project
3*795d594fSAndroid Build Coastguard Worker  *
4*795d594fSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*795d594fSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*795d594fSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*795d594fSAndroid Build Coastguard Worker  *
8*795d594fSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*795d594fSAndroid Build Coastguard Worker  *
10*795d594fSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*795d594fSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*795d594fSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*795d594fSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*795d594fSAndroid Build Coastguard Worker  * limitations under the License.
15*795d594fSAndroid Build Coastguard Worker  */
16*795d594fSAndroid Build Coastguard Worker 
17*795d594fSAndroid Build Coastguard Worker #include "fd_file.h"
18*795d594fSAndroid Build Coastguard Worker 
19*795d594fSAndroid Build Coastguard Worker #include <errno.h>
20*795d594fSAndroid Build Coastguard Worker #include <stdio.h>
21*795d594fSAndroid Build Coastguard Worker #include <sys/stat.h>
22*795d594fSAndroid Build Coastguard Worker #include <sys/types.h>
23*795d594fSAndroid Build Coastguard Worker #include <unistd.h>
24*795d594fSAndroid Build Coastguard Worker 
25*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
26*795d594fSAndroid Build Coastguard Worker #include <android/fdsan.h>
27*795d594fSAndroid Build Coastguard Worker #endif
28*795d594fSAndroid Build Coastguard Worker 
29*795d594fSAndroid Build Coastguard Worker #if defined(_WIN32)
30*795d594fSAndroid Build Coastguard Worker #include <windows.h>
31*795d594fSAndroid Build Coastguard Worker #endif
32*795d594fSAndroid Build Coastguard Worker 
33*795d594fSAndroid Build Coastguard Worker #include <limits>
34*795d594fSAndroid Build Coastguard Worker #include <vector>
35*795d594fSAndroid Build Coastguard Worker 
36*795d594fSAndroid Build Coastguard Worker #include <android-base/file.h>
37*795d594fSAndroid Build Coastguard Worker #include <android-base/logging.h>
38*795d594fSAndroid Build Coastguard Worker 
39*795d594fSAndroid Build Coastguard Worker // Includes needed for FdFile::Copy().
40*795d594fSAndroid Build Coastguard Worker #include "base/globals.h"
41*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
42*795d594fSAndroid Build Coastguard Worker #include "base/bit_utils.h"
43*795d594fSAndroid Build Coastguard Worker #include "base/mem_map.h"
44*795d594fSAndroid Build Coastguard Worker #include "sys/mman.h"
45*795d594fSAndroid Build Coastguard Worker #else
46*795d594fSAndroid Build Coastguard Worker #include <algorithm>
47*795d594fSAndroid Build Coastguard Worker #include "base/stl_util.h"
48*795d594fSAndroid Build Coastguard Worker #endif
49*795d594fSAndroid Build Coastguard Worker 
50*795d594fSAndroid Build Coastguard Worker namespace unix_file {
51*795d594fSAndroid Build Coastguard Worker 
52*795d594fSAndroid Build Coastguard Worker #if defined(_WIN32)
53*795d594fSAndroid Build Coastguard Worker // RAII wrapper for an event object to allow asynchronous I/O to correctly signal completion.
54*795d594fSAndroid Build Coastguard Worker class ScopedEvent {
55*795d594fSAndroid Build Coastguard Worker  public:
ScopedEvent()56*795d594fSAndroid Build Coastguard Worker   ScopedEvent() {
57*795d594fSAndroid Build Coastguard Worker     handle_ = CreateEventA(/*lpEventAttributes*/ nullptr,
58*795d594fSAndroid Build Coastguard Worker                            /*bManualReset*/ true,
59*795d594fSAndroid Build Coastguard Worker                            /*bInitialState*/ false,
60*795d594fSAndroid Build Coastguard Worker                            /*lpName*/ nullptr);
61*795d594fSAndroid Build Coastguard Worker   }
62*795d594fSAndroid Build Coastguard Worker 
~ScopedEvent()63*795d594fSAndroid Build Coastguard Worker   ~ScopedEvent() { CloseHandle(handle_); }
64*795d594fSAndroid Build Coastguard Worker 
handle()65*795d594fSAndroid Build Coastguard Worker   HANDLE handle() { return handle_; }
66*795d594fSAndroid Build Coastguard Worker 
67*795d594fSAndroid Build Coastguard Worker  private:
68*795d594fSAndroid Build Coastguard Worker   HANDLE handle_;
69*795d594fSAndroid Build Coastguard Worker   DISALLOW_COPY_AND_ASSIGN(ScopedEvent);
70*795d594fSAndroid Build Coastguard Worker };
71*795d594fSAndroid Build Coastguard Worker 
72*795d594fSAndroid Build Coastguard Worker // Windows implementation of pread/pwrite. Note that these DO move the file descriptor's read/write
73*795d594fSAndroid Build Coastguard Worker // position, but do so atomically.
pread(int fd,void * data,size_t byte_count,off64_t offset)74*795d594fSAndroid Build Coastguard Worker static ssize_t pread(int fd, void* data, size_t byte_count, off64_t offset) {
75*795d594fSAndroid Build Coastguard Worker   ScopedEvent event;
76*795d594fSAndroid Build Coastguard Worker   if (event.handle() == INVALID_HANDLE_VALUE) {
77*795d594fSAndroid Build Coastguard Worker     PLOG(ERROR) << "Could not create event handle.";
78*795d594fSAndroid Build Coastguard Worker     errno = EIO;
79*795d594fSAndroid Build Coastguard Worker     return static_cast<ssize_t>(-1);
80*795d594fSAndroid Build Coastguard Worker   }
81*795d594fSAndroid Build Coastguard Worker 
82*795d594fSAndroid Build Coastguard Worker   auto handle = reinterpret_cast<HANDLE>(_get_osfhandle(fd));
83*795d594fSAndroid Build Coastguard Worker   DWORD bytes_read = 0;
84*795d594fSAndroid Build Coastguard Worker   OVERLAPPED overlapped = {};
85*795d594fSAndroid Build Coastguard Worker   overlapped.Offset = static_cast<DWORD>(offset);
86*795d594fSAndroid Build Coastguard Worker   overlapped.OffsetHigh = static_cast<DWORD>(offset >> 32);
87*795d594fSAndroid Build Coastguard Worker   overlapped.hEvent = event.handle();
88*795d594fSAndroid Build Coastguard Worker   if (!ReadFile(handle, data, static_cast<DWORD>(byte_count), &bytes_read, &overlapped)) {
89*795d594fSAndroid Build Coastguard Worker     // If the read failed with other than ERROR_IO_PENDING, return an error.
90*795d594fSAndroid Build Coastguard Worker     // ERROR_IO_PENDING signals the write was begun asynchronously.
91*795d594fSAndroid Build Coastguard Worker     // Block until the asynchronous operation has finished or fails, and return
92*795d594fSAndroid Build Coastguard Worker     // result accordingly.
93*795d594fSAndroid Build Coastguard Worker     if (::GetLastError() != ERROR_IO_PENDING ||
94*795d594fSAndroid Build Coastguard Worker         !::GetOverlappedResult(handle, &overlapped, &bytes_read, TRUE)) {
95*795d594fSAndroid Build Coastguard Worker       // In case someone tries to read errno (since this is masquerading as a POSIX call).
96*795d594fSAndroid Build Coastguard Worker       errno = EIO;
97*795d594fSAndroid Build Coastguard Worker       return static_cast<ssize_t>(-1);
98*795d594fSAndroid Build Coastguard Worker     }
99*795d594fSAndroid Build Coastguard Worker   }
100*795d594fSAndroid Build Coastguard Worker   return static_cast<ssize_t>(bytes_read);
101*795d594fSAndroid Build Coastguard Worker }
102*795d594fSAndroid Build Coastguard Worker 
pwrite(int fd,const void * buf,size_t count,off64_t offset)103*795d594fSAndroid Build Coastguard Worker static ssize_t pwrite(int fd, const void* buf, size_t count, off64_t offset) {
104*795d594fSAndroid Build Coastguard Worker   ScopedEvent event;
105*795d594fSAndroid Build Coastguard Worker   if (event.handle() == INVALID_HANDLE_VALUE) {
106*795d594fSAndroid Build Coastguard Worker     PLOG(ERROR) << "Could not create event handle.";
107*795d594fSAndroid Build Coastguard Worker     errno = EIO;
108*795d594fSAndroid Build Coastguard Worker     return static_cast<ssize_t>(-1);
109*795d594fSAndroid Build Coastguard Worker   }
110*795d594fSAndroid Build Coastguard Worker 
111*795d594fSAndroid Build Coastguard Worker   auto handle = reinterpret_cast<HANDLE>(_get_osfhandle(fd));
112*795d594fSAndroid Build Coastguard Worker   DWORD bytes_written = 0;
113*795d594fSAndroid Build Coastguard Worker   OVERLAPPED overlapped = {};
114*795d594fSAndroid Build Coastguard Worker   overlapped.Offset = static_cast<DWORD>(offset);
115*795d594fSAndroid Build Coastguard Worker   overlapped.OffsetHigh = static_cast<DWORD>(offset >> 32);
116*795d594fSAndroid Build Coastguard Worker   overlapped.hEvent = event.handle();
117*795d594fSAndroid Build Coastguard Worker   if (!::WriteFile(handle, buf, count, &bytes_written, &overlapped)) {
118*795d594fSAndroid Build Coastguard Worker     // If the write failed with other than ERROR_IO_PENDING, return an error.
119*795d594fSAndroid Build Coastguard Worker     // ERROR_IO_PENDING signals the write was begun asynchronously.
120*795d594fSAndroid Build Coastguard Worker     // Block until the asynchronous operation has finished or fails, and return
121*795d594fSAndroid Build Coastguard Worker     // result accordingly.
122*795d594fSAndroid Build Coastguard Worker     if (::GetLastError() != ERROR_IO_PENDING ||
123*795d594fSAndroid Build Coastguard Worker         !::GetOverlappedResult(handle, &overlapped, &bytes_written, TRUE)) {
124*795d594fSAndroid Build Coastguard Worker       // In case someone tries to read errno (since this is masquerading as a POSIX call).
125*795d594fSAndroid Build Coastguard Worker       errno = EIO;
126*795d594fSAndroid Build Coastguard Worker       return static_cast<ssize_t>(-1);
127*795d594fSAndroid Build Coastguard Worker     }
128*795d594fSAndroid Build Coastguard Worker   }
129*795d594fSAndroid Build Coastguard Worker   return static_cast<ssize_t>(bytes_written);
130*795d594fSAndroid Build Coastguard Worker }
131*795d594fSAndroid Build Coastguard Worker 
fsync(int fd)132*795d594fSAndroid Build Coastguard Worker static int fsync(int fd) {
133*795d594fSAndroid Build Coastguard Worker   auto handle = reinterpret_cast<HANDLE>(_get_osfhandle(fd));
134*795d594fSAndroid Build Coastguard Worker   if (handle != INVALID_HANDLE_VALUE && ::FlushFileBuffers(handle)) {
135*795d594fSAndroid Build Coastguard Worker     return 0;
136*795d594fSAndroid Build Coastguard Worker   }
137*795d594fSAndroid Build Coastguard Worker   errno = EINVAL;
138*795d594fSAndroid Build Coastguard Worker   return -1;
139*795d594fSAndroid Build Coastguard Worker }
140*795d594fSAndroid Build Coastguard Worker #endif
141*795d594fSAndroid Build Coastguard Worker 
142*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
GetFdFileOwnerTag(FdFile * fd_file)143*795d594fSAndroid Build Coastguard Worker static uint64_t GetFdFileOwnerTag(FdFile* fd_file) {
144*795d594fSAndroid Build Coastguard Worker   return android_fdsan_create_owner_tag(ANDROID_FDSAN_OWNER_TYPE_ART_FDFILE,
145*795d594fSAndroid Build Coastguard Worker                                         reinterpret_cast<uint64_t>(fd_file));
146*795d594fSAndroid Build Coastguard Worker }
147*795d594fSAndroid Build Coastguard Worker #endif
148*795d594fSAndroid Build Coastguard Worker 
FdFile(int fd,bool check_usage)149*795d594fSAndroid Build Coastguard Worker FdFile::FdFile(int fd, bool check_usage)
150*795d594fSAndroid Build Coastguard Worker     : FdFile(fd, std::string(), check_usage) {}
151*795d594fSAndroid Build Coastguard Worker 
FdFile(int fd,const std::string & path,bool check_usage)152*795d594fSAndroid Build Coastguard Worker FdFile::FdFile(int fd, const std::string& path, bool check_usage)
153*795d594fSAndroid Build Coastguard Worker     : FdFile(fd, path, check_usage, false) {}
154*795d594fSAndroid Build Coastguard Worker 
FdFile(int fd,const std::string & path,bool check_usage,bool read_only_mode)155*795d594fSAndroid Build Coastguard Worker FdFile::FdFile(int fd, const std::string& path, bool check_usage,
156*795d594fSAndroid Build Coastguard Worker                bool read_only_mode)
157*795d594fSAndroid Build Coastguard Worker     : guard_state_(check_usage ? GuardState::kBase : GuardState::kNoCheck),
158*795d594fSAndroid Build Coastguard Worker       fd_(fd),
159*795d594fSAndroid Build Coastguard Worker       file_path_(path),
160*795d594fSAndroid Build Coastguard Worker       read_only_mode_(read_only_mode) {
161*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
162*795d594fSAndroid Build Coastguard Worker   if (fd >= 0) {
163*795d594fSAndroid Build Coastguard Worker     android_fdsan_exchange_owner_tag(fd, 0, GetFdFileOwnerTag(this));
164*795d594fSAndroid Build Coastguard Worker   }
165*795d594fSAndroid Build Coastguard Worker #endif
166*795d594fSAndroid Build Coastguard Worker }
167*795d594fSAndroid Build Coastguard Worker 
FdFile(const std::string & path,int flags,mode_t mode,bool check_usage)168*795d594fSAndroid Build Coastguard Worker FdFile::FdFile(const std::string& path, int flags, mode_t mode,
169*795d594fSAndroid Build Coastguard Worker                bool check_usage) {
170*795d594fSAndroid Build Coastguard Worker   Open(path, flags, mode);
171*795d594fSAndroid Build Coastguard Worker   if (!check_usage || !IsOpened()) {
172*795d594fSAndroid Build Coastguard Worker     guard_state_ = GuardState::kNoCheck;
173*795d594fSAndroid Build Coastguard Worker   }
174*795d594fSAndroid Build Coastguard Worker }
175*795d594fSAndroid Build Coastguard Worker 
Destroy()176*795d594fSAndroid Build Coastguard Worker void FdFile::Destroy() {
177*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage && (guard_state_ < GuardState::kNoCheck)) {
178*795d594fSAndroid Build Coastguard Worker     if (guard_state_ < GuardState::kFlushed) {
179*795d594fSAndroid Build Coastguard Worker       LOG(ERROR) << "File " << file_path_ << " wasn't explicitly flushed before destruction.";
180*795d594fSAndroid Build Coastguard Worker     }
181*795d594fSAndroid Build Coastguard Worker     if (guard_state_ < GuardState::kClosed) {
182*795d594fSAndroid Build Coastguard Worker       LOG(ERROR) << "File " << file_path_ << " wasn't explicitly closed before destruction.";
183*795d594fSAndroid Build Coastguard Worker     }
184*795d594fSAndroid Build Coastguard Worker     DCHECK_GE(guard_state_, GuardState::kClosed);
185*795d594fSAndroid Build Coastguard Worker   }
186*795d594fSAndroid Build Coastguard Worker   if (fd_ != kInvalidFd) {
187*795d594fSAndroid Build Coastguard Worker     if (Close() != 0) {
188*795d594fSAndroid Build Coastguard Worker       PLOG(WARNING) << "Failed to close file with fd=" << fd_ << " path=" << file_path_;
189*795d594fSAndroid Build Coastguard Worker     }
190*795d594fSAndroid Build Coastguard Worker   }
191*795d594fSAndroid Build Coastguard Worker }
192*795d594fSAndroid Build Coastguard Worker 
FdFile(FdFile && other)193*795d594fSAndroid Build Coastguard Worker FdFile::FdFile(FdFile&& other) noexcept
194*795d594fSAndroid Build Coastguard Worker     : guard_state_(other.guard_state_),
195*795d594fSAndroid Build Coastguard Worker       fd_(other.fd_),
196*795d594fSAndroid Build Coastguard Worker       file_path_(std::move(other.file_path_)),
197*795d594fSAndroid Build Coastguard Worker       read_only_mode_(other.read_only_mode_) {
198*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
199*795d594fSAndroid Build Coastguard Worker   if (fd_ >= 0) {
200*795d594fSAndroid Build Coastguard Worker     android_fdsan_exchange_owner_tag(fd_, GetFdFileOwnerTag(&other), GetFdFileOwnerTag(this));
201*795d594fSAndroid Build Coastguard Worker   }
202*795d594fSAndroid Build Coastguard Worker #endif
203*795d594fSAndroid Build Coastguard Worker   other.guard_state_ = GuardState::kClosed;
204*795d594fSAndroid Build Coastguard Worker   other.fd_ = kInvalidFd;
205*795d594fSAndroid Build Coastguard Worker }
206*795d594fSAndroid Build Coastguard Worker 
operator =(FdFile && other)207*795d594fSAndroid Build Coastguard Worker FdFile& FdFile::operator=(FdFile&& other) noexcept {
208*795d594fSAndroid Build Coastguard Worker   if (this == &other) {
209*795d594fSAndroid Build Coastguard Worker     return *this;
210*795d594fSAndroid Build Coastguard Worker   }
211*795d594fSAndroid Build Coastguard Worker 
212*795d594fSAndroid Build Coastguard Worker   if (this->fd_ != other.fd_) {
213*795d594fSAndroid Build Coastguard Worker     Destroy();  // Free old state.
214*795d594fSAndroid Build Coastguard Worker   }
215*795d594fSAndroid Build Coastguard Worker 
216*795d594fSAndroid Build Coastguard Worker   guard_state_ = other.guard_state_;
217*795d594fSAndroid Build Coastguard Worker   fd_ = other.fd_;
218*795d594fSAndroid Build Coastguard Worker   file_path_ = std::move(other.file_path_);
219*795d594fSAndroid Build Coastguard Worker   read_only_mode_ = other.read_only_mode_;
220*795d594fSAndroid Build Coastguard Worker 
221*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
222*795d594fSAndroid Build Coastguard Worker   if (fd_ >= 0) {
223*795d594fSAndroid Build Coastguard Worker     android_fdsan_exchange_owner_tag(fd_, GetFdFileOwnerTag(&other), GetFdFileOwnerTag(this));
224*795d594fSAndroid Build Coastguard Worker   }
225*795d594fSAndroid Build Coastguard Worker #endif
226*795d594fSAndroid Build Coastguard Worker   other.guard_state_ = GuardState::kClosed;
227*795d594fSAndroid Build Coastguard Worker   other.fd_ = kInvalidFd;
228*795d594fSAndroid Build Coastguard Worker   return *this;
229*795d594fSAndroid Build Coastguard Worker }
230*795d594fSAndroid Build Coastguard Worker 
~FdFile()231*795d594fSAndroid Build Coastguard Worker FdFile::~FdFile() {
232*795d594fSAndroid Build Coastguard Worker   Destroy();
233*795d594fSAndroid Build Coastguard Worker }
234*795d594fSAndroid Build Coastguard Worker 
Release()235*795d594fSAndroid Build Coastguard Worker int FdFile::Release() {
236*795d594fSAndroid Build Coastguard Worker   int tmp_fd = fd_;
237*795d594fSAndroid Build Coastguard Worker   fd_ = kInvalidFd;
238*795d594fSAndroid Build Coastguard Worker   guard_state_ = GuardState::kNoCheck;
239*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
240*795d594fSAndroid Build Coastguard Worker   if (tmp_fd >= 0) {
241*795d594fSAndroid Build Coastguard Worker     android_fdsan_exchange_owner_tag(tmp_fd, GetFdFileOwnerTag(this), 0);
242*795d594fSAndroid Build Coastguard Worker   }
243*795d594fSAndroid Build Coastguard Worker #endif
244*795d594fSAndroid Build Coastguard Worker   return tmp_fd;
245*795d594fSAndroid Build Coastguard Worker }
246*795d594fSAndroid Build Coastguard Worker 
Reset(int fd,bool check_usage)247*795d594fSAndroid Build Coastguard Worker void FdFile::Reset(int fd, bool check_usage) {
248*795d594fSAndroid Build Coastguard Worker   CHECK_NE(fd, fd_);
249*795d594fSAndroid Build Coastguard Worker 
250*795d594fSAndroid Build Coastguard Worker   if (fd_ != kInvalidFd) {
251*795d594fSAndroid Build Coastguard Worker     Destroy();
252*795d594fSAndroid Build Coastguard Worker   }
253*795d594fSAndroid Build Coastguard Worker   fd_ = fd;
254*795d594fSAndroid Build Coastguard Worker 
255*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
256*795d594fSAndroid Build Coastguard Worker   if (fd_ >= 0) {
257*795d594fSAndroid Build Coastguard Worker     android_fdsan_exchange_owner_tag(fd_, 0, GetFdFileOwnerTag(this));
258*795d594fSAndroid Build Coastguard Worker   }
259*795d594fSAndroid Build Coastguard Worker #endif
260*795d594fSAndroid Build Coastguard Worker 
261*795d594fSAndroid Build Coastguard Worker   if (check_usage) {
262*795d594fSAndroid Build Coastguard Worker     guard_state_ = fd == kInvalidFd ? GuardState::kNoCheck : GuardState::kBase;
263*795d594fSAndroid Build Coastguard Worker   } else {
264*795d594fSAndroid Build Coastguard Worker     guard_state_ = GuardState::kNoCheck;
265*795d594fSAndroid Build Coastguard Worker   }
266*795d594fSAndroid Build Coastguard Worker }
267*795d594fSAndroid Build Coastguard Worker 
moveTo(GuardState target,GuardState warn_threshold,const char * warning)268*795d594fSAndroid Build Coastguard Worker void FdFile::moveTo(GuardState target, GuardState warn_threshold, const char* warning) {
269*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage) {
270*795d594fSAndroid Build Coastguard Worker     if (guard_state_ < GuardState::kNoCheck) {
271*795d594fSAndroid Build Coastguard Worker       if (warn_threshold < GuardState::kNoCheck && guard_state_ >= warn_threshold) {
272*795d594fSAndroid Build Coastguard Worker         LOG(ERROR) << warning;
273*795d594fSAndroid Build Coastguard Worker       }
274*795d594fSAndroid Build Coastguard Worker       guard_state_ = target;
275*795d594fSAndroid Build Coastguard Worker     }
276*795d594fSAndroid Build Coastguard Worker   }
277*795d594fSAndroid Build Coastguard Worker }
278*795d594fSAndroid Build Coastguard Worker 
moveUp(GuardState target,const char * warning)279*795d594fSAndroid Build Coastguard Worker void FdFile::moveUp(GuardState target, const char* warning) {
280*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage) {
281*795d594fSAndroid Build Coastguard Worker     if (guard_state_ < GuardState::kNoCheck) {
282*795d594fSAndroid Build Coastguard Worker       if (guard_state_ < target) {
283*795d594fSAndroid Build Coastguard Worker         guard_state_ = target;
284*795d594fSAndroid Build Coastguard Worker       } else if (target < guard_state_) {
285*795d594fSAndroid Build Coastguard Worker         LOG(ERROR) << warning;
286*795d594fSAndroid Build Coastguard Worker       }
287*795d594fSAndroid Build Coastguard Worker     }
288*795d594fSAndroid Build Coastguard Worker   }
289*795d594fSAndroid Build Coastguard Worker }
290*795d594fSAndroid Build Coastguard Worker 
Open(const std::string & path,int flags)291*795d594fSAndroid Build Coastguard Worker bool FdFile::Open(const std::string& path, int flags) {
292*795d594fSAndroid Build Coastguard Worker   return Open(path, flags, 0640);
293*795d594fSAndroid Build Coastguard Worker }
294*795d594fSAndroid Build Coastguard Worker 
Open(const std::string & path,int flags,mode_t mode)295*795d594fSAndroid Build Coastguard Worker bool FdFile::Open(const std::string& path, int flags, mode_t mode) {
296*795d594fSAndroid Build Coastguard Worker   static_assert(O_RDONLY == 0, "Readonly flag has unexpected value.");
297*795d594fSAndroid Build Coastguard Worker   DCHECK_EQ(fd_, kInvalidFd) << path;
298*795d594fSAndroid Build Coastguard Worker   read_only_mode_ = ((flags & O_ACCMODE) == O_RDONLY);
299*795d594fSAndroid Build Coastguard Worker   fd_ = TEMP_FAILURE_RETRY(open(path.c_str(), flags, mode));
300*795d594fSAndroid Build Coastguard Worker   if (fd_ == kInvalidFd) {
301*795d594fSAndroid Build Coastguard Worker     return false;
302*795d594fSAndroid Build Coastguard Worker   }
303*795d594fSAndroid Build Coastguard Worker 
304*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
305*795d594fSAndroid Build Coastguard Worker   android_fdsan_exchange_owner_tag(fd_, 0, GetFdFileOwnerTag(this));
306*795d594fSAndroid Build Coastguard Worker #endif
307*795d594fSAndroid Build Coastguard Worker 
308*795d594fSAndroid Build Coastguard Worker   file_path_ = path;
309*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage && (flags & (O_RDWR | O_CREAT | O_WRONLY)) != 0) {
310*795d594fSAndroid Build Coastguard Worker     // Start in the base state (not flushed, not closed).
311*795d594fSAndroid Build Coastguard Worker     guard_state_ = GuardState::kBase;
312*795d594fSAndroid Build Coastguard Worker   } else {
313*795d594fSAndroid Build Coastguard Worker     // We are not concerned with read-only files. In that case, proper flushing and closing is
314*795d594fSAndroid Build Coastguard Worker     // not important.
315*795d594fSAndroid Build Coastguard Worker     guard_state_ = GuardState::kNoCheck;
316*795d594fSAndroid Build Coastguard Worker   }
317*795d594fSAndroid Build Coastguard Worker   return true;
318*795d594fSAndroid Build Coastguard Worker }
319*795d594fSAndroid Build Coastguard Worker 
Close()320*795d594fSAndroid Build Coastguard Worker int FdFile::Close() {
321*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
322*795d594fSAndroid Build Coastguard Worker   int result = android_fdsan_close_with_tag(fd_, GetFdFileOwnerTag(this));
323*795d594fSAndroid Build Coastguard Worker #else
324*795d594fSAndroid Build Coastguard Worker   int result = close(fd_);
325*795d594fSAndroid Build Coastguard Worker #endif
326*795d594fSAndroid Build Coastguard Worker 
327*795d594fSAndroid Build Coastguard Worker   // Test here, so the file is closed and not leaked.
328*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage) {
329*795d594fSAndroid Build Coastguard Worker     DCHECK_GE(guard_state_, GuardState::kFlushed) << "File " << file_path_
330*795d594fSAndroid Build Coastguard Worker         << " has not been flushed before closing.";
331*795d594fSAndroid Build Coastguard Worker     moveUp(GuardState::kClosed, nullptr);
332*795d594fSAndroid Build Coastguard Worker   }
333*795d594fSAndroid Build Coastguard Worker 
334*795d594fSAndroid Build Coastguard Worker #if defined(__linux__)
335*795d594fSAndroid Build Coastguard Worker   // close always succeeds on linux, even if failure is reported.
336*795d594fSAndroid Build Coastguard Worker   UNUSED(result);
337*795d594fSAndroid Build Coastguard Worker #else
338*795d594fSAndroid Build Coastguard Worker   if (result == -1) {
339*795d594fSAndroid Build Coastguard Worker     return -errno;
340*795d594fSAndroid Build Coastguard Worker   }
341*795d594fSAndroid Build Coastguard Worker #endif
342*795d594fSAndroid Build Coastguard Worker 
343*795d594fSAndroid Build Coastguard Worker   fd_ = kInvalidFd;
344*795d594fSAndroid Build Coastguard Worker   file_path_ = "";
345*795d594fSAndroid Build Coastguard Worker   return 0;
346*795d594fSAndroid Build Coastguard Worker }
347*795d594fSAndroid Build Coastguard Worker 
Flush(bool flush_metadata)348*795d594fSAndroid Build Coastguard Worker int FdFile::Flush(bool flush_metadata) {
349*795d594fSAndroid Build Coastguard Worker   DCHECK(flush_metadata || !read_only_mode_);
350*795d594fSAndroid Build Coastguard Worker 
351*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
352*795d594fSAndroid Build Coastguard Worker   int rc;
353*795d594fSAndroid Build Coastguard Worker   if (flush_metadata) {
354*795d594fSAndroid Build Coastguard Worker     rc = TEMP_FAILURE_RETRY(fsync(fd_));
355*795d594fSAndroid Build Coastguard Worker   } else {
356*795d594fSAndroid Build Coastguard Worker     rc = TEMP_FAILURE_RETRY(fdatasync(fd_));
357*795d594fSAndroid Build Coastguard Worker   }
358*795d594fSAndroid Build Coastguard Worker #else
359*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(fsync(fd_));
360*795d594fSAndroid Build Coastguard Worker #endif
361*795d594fSAndroid Build Coastguard Worker 
362*795d594fSAndroid Build Coastguard Worker   moveUp(GuardState::kFlushed, "Flushing closed file.");
363*795d594fSAndroid Build Coastguard Worker   if (rc == 0) {
364*795d594fSAndroid Build Coastguard Worker     return 0;
365*795d594fSAndroid Build Coastguard Worker   }
366*795d594fSAndroid Build Coastguard Worker 
367*795d594fSAndroid Build Coastguard Worker   // Don't report failure if we just tried to flush a pipe or socket.
368*795d594fSAndroid Build Coastguard Worker   return errno == EINVAL ? 0 : -errno;
369*795d594fSAndroid Build Coastguard Worker }
370*795d594fSAndroid Build Coastguard Worker 
Read(char * buf,int64_t byte_count,int64_t offset) const371*795d594fSAndroid Build Coastguard Worker int64_t FdFile::Read(char* buf, int64_t byte_count, int64_t offset) const {
372*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
373*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(pread64(fd_, buf, byte_count, offset));
374*795d594fSAndroid Build Coastguard Worker #else
375*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(pread(fd_, buf, byte_count, offset));
376*795d594fSAndroid Build Coastguard Worker #endif
377*795d594fSAndroid Build Coastguard Worker   return (rc == -1) ? -errno : rc;
378*795d594fSAndroid Build Coastguard Worker }
379*795d594fSAndroid Build Coastguard Worker 
SetLength(int64_t new_length)380*795d594fSAndroid Build Coastguard Worker int FdFile::SetLength(int64_t new_length) {
381*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
382*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
383*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(ftruncate64(fd_, new_length));
384*795d594fSAndroid Build Coastguard Worker #else
385*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(ftruncate(fd_, new_length));
386*795d594fSAndroid Build Coastguard Worker #endif
387*795d594fSAndroid Build Coastguard Worker   moveTo(GuardState::kBase, GuardState::kClosed, "Truncating closed file.");
388*795d594fSAndroid Build Coastguard Worker   return (rc == -1) ? -errno : rc;
389*795d594fSAndroid Build Coastguard Worker }
390*795d594fSAndroid Build Coastguard Worker 
GetLength() const391*795d594fSAndroid Build Coastguard Worker int64_t FdFile::GetLength() const {
392*795d594fSAndroid Build Coastguard Worker   struct stat s;
393*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(fstat(fd_, &s));
394*795d594fSAndroid Build Coastguard Worker   return (rc == -1) ? -errno : s.st_size;
395*795d594fSAndroid Build Coastguard Worker }
396*795d594fSAndroid Build Coastguard Worker 
Write(const char * buf,int64_t byte_count,int64_t offset)397*795d594fSAndroid Build Coastguard Worker int64_t FdFile::Write(const char* buf, int64_t byte_count, int64_t offset) {
398*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
399*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
400*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(pwrite64(fd_, buf, byte_count, offset));
401*795d594fSAndroid Build Coastguard Worker #else
402*795d594fSAndroid Build Coastguard Worker   int rc = TEMP_FAILURE_RETRY(pwrite(fd_, buf, byte_count, offset));
403*795d594fSAndroid Build Coastguard Worker #endif
404*795d594fSAndroid Build Coastguard Worker   moveTo(GuardState::kBase, GuardState::kClosed, "Writing into closed file.");
405*795d594fSAndroid Build Coastguard Worker   return (rc == -1) ? -errno : rc;
406*795d594fSAndroid Build Coastguard Worker }
407*795d594fSAndroid Build Coastguard Worker 
Fd() const408*795d594fSAndroid Build Coastguard Worker int FdFile::Fd() const {
409*795d594fSAndroid Build Coastguard Worker   return fd_;
410*795d594fSAndroid Build Coastguard Worker }
411*795d594fSAndroid Build Coastguard Worker 
ReadOnlyMode() const412*795d594fSAndroid Build Coastguard Worker bool FdFile::ReadOnlyMode() const {
413*795d594fSAndroid Build Coastguard Worker   return read_only_mode_;
414*795d594fSAndroid Build Coastguard Worker }
415*795d594fSAndroid Build Coastguard Worker 
CheckUsage() const416*795d594fSAndroid Build Coastguard Worker bool FdFile::CheckUsage() const {
417*795d594fSAndroid Build Coastguard Worker   return guard_state_ != GuardState::kNoCheck;
418*795d594fSAndroid Build Coastguard Worker }
419*795d594fSAndroid Build Coastguard Worker 
IsOpened() const420*795d594fSAndroid Build Coastguard Worker bool FdFile::IsOpened() const {
421*795d594fSAndroid Build Coastguard Worker   return FdFile::IsOpenFd(fd_);
422*795d594fSAndroid Build Coastguard Worker }
423*795d594fSAndroid Build Coastguard Worker 
ReadIgnoreOffset(int fd,void * buf,size_t count,off_t offset)424*795d594fSAndroid Build Coastguard Worker static ssize_t ReadIgnoreOffset(int fd, void *buf, size_t count, off_t offset) {
425*795d594fSAndroid Build Coastguard Worker   DCHECK_EQ(offset, 0);
426*795d594fSAndroid Build Coastguard Worker   return read(fd, buf, count);
427*795d594fSAndroid Build Coastguard Worker }
428*795d594fSAndroid Build Coastguard Worker 
429*795d594fSAndroid Build Coastguard Worker template <ssize_t (*read_func)(int, void*, size_t, off_t)>
ReadFullyGeneric(int fd,void * buffer,size_t byte_count,size_t offset)430*795d594fSAndroid Build Coastguard Worker static bool ReadFullyGeneric(int fd, void* buffer, size_t byte_count, size_t offset) {
431*795d594fSAndroid Build Coastguard Worker   char* ptr = static_cast<char*>(buffer);
432*795d594fSAndroid Build Coastguard Worker   while (byte_count > 0) {
433*795d594fSAndroid Build Coastguard Worker     ssize_t bytes_read = TEMP_FAILURE_RETRY(read_func(fd, ptr, byte_count, offset));
434*795d594fSAndroid Build Coastguard Worker     if (bytes_read <= 0) {
435*795d594fSAndroid Build Coastguard Worker       // 0: end of file
436*795d594fSAndroid Build Coastguard Worker       // -1: error
437*795d594fSAndroid Build Coastguard Worker       return false;
438*795d594fSAndroid Build Coastguard Worker     }
439*795d594fSAndroid Build Coastguard Worker     byte_count -= bytes_read;  // Reduce the number of remaining bytes.
440*795d594fSAndroid Build Coastguard Worker     ptr += bytes_read;  // Move the buffer forward.
441*795d594fSAndroid Build Coastguard Worker     offset += static_cast<size_t>(bytes_read);  // Move the offset forward.
442*795d594fSAndroid Build Coastguard Worker   }
443*795d594fSAndroid Build Coastguard Worker   return true;
444*795d594fSAndroid Build Coastguard Worker }
445*795d594fSAndroid Build Coastguard Worker 
ReadFully(void * buffer,size_t byte_count)446*795d594fSAndroid Build Coastguard Worker bool FdFile::ReadFully(void* buffer, size_t byte_count) {
447*795d594fSAndroid Build Coastguard Worker   return ReadFullyGeneric<ReadIgnoreOffset>(fd_, buffer, byte_count, 0);
448*795d594fSAndroid Build Coastguard Worker }
449*795d594fSAndroid Build Coastguard Worker 
PreadFully(void * buffer,size_t byte_count,size_t offset)450*795d594fSAndroid Build Coastguard Worker bool FdFile::PreadFully(void* buffer, size_t byte_count, size_t offset) {
451*795d594fSAndroid Build Coastguard Worker   return ReadFullyGeneric<pread>(fd_, buffer, byte_count, offset);
452*795d594fSAndroid Build Coastguard Worker }
453*795d594fSAndroid Build Coastguard Worker 
454*795d594fSAndroid Build Coastguard Worker template <bool kUseOffset>
WriteFullyGeneric(const void * buffer,size_t byte_count,size_t offset)455*795d594fSAndroid Build Coastguard Worker bool FdFile::WriteFullyGeneric(const void* buffer, size_t byte_count, size_t offset) {
456*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
457*795d594fSAndroid Build Coastguard Worker   moveTo(GuardState::kBase, GuardState::kClosed, "Writing into closed file.");
458*795d594fSAndroid Build Coastguard Worker   DCHECK(kUseOffset || offset == 0u);
459*795d594fSAndroid Build Coastguard Worker   const char* ptr = static_cast<const char*>(buffer);
460*795d594fSAndroid Build Coastguard Worker   while (byte_count > 0) {
461*795d594fSAndroid Build Coastguard Worker     ssize_t bytes_written = kUseOffset
462*795d594fSAndroid Build Coastguard Worker         ? TEMP_FAILURE_RETRY(pwrite(fd_, ptr, byte_count, offset))
463*795d594fSAndroid Build Coastguard Worker         : TEMP_FAILURE_RETRY(write(fd_, ptr, byte_count));
464*795d594fSAndroid Build Coastguard Worker     if (bytes_written == -1) {
465*795d594fSAndroid Build Coastguard Worker       return false;
466*795d594fSAndroid Build Coastguard Worker     }
467*795d594fSAndroid Build Coastguard Worker     byte_count -= bytes_written;  // Reduce the number of remaining bytes.
468*795d594fSAndroid Build Coastguard Worker     ptr += bytes_written;  // Move the buffer forward.
469*795d594fSAndroid Build Coastguard Worker     offset += static_cast<size_t>(bytes_written);
470*795d594fSAndroid Build Coastguard Worker   }
471*795d594fSAndroid Build Coastguard Worker   return true;
472*795d594fSAndroid Build Coastguard Worker }
473*795d594fSAndroid Build Coastguard Worker 
PwriteFully(const void * buffer,size_t byte_count,size_t offset)474*795d594fSAndroid Build Coastguard Worker bool FdFile::PwriteFully(const void* buffer, size_t byte_count, size_t offset) {
475*795d594fSAndroid Build Coastguard Worker   return WriteFullyGeneric<true>(buffer, byte_count, offset);
476*795d594fSAndroid Build Coastguard Worker }
477*795d594fSAndroid Build Coastguard Worker 
WriteFully(const void * buffer,size_t byte_count)478*795d594fSAndroid Build Coastguard Worker bool FdFile::WriteFully(const void* buffer, size_t byte_count) {
479*795d594fSAndroid Build Coastguard Worker   return WriteFullyGeneric<false>(buffer, byte_count, 0u);
480*795d594fSAndroid Build Coastguard Worker }
481*795d594fSAndroid Build Coastguard Worker 
Rename(const std::string & new_path)482*795d594fSAndroid Build Coastguard Worker bool FdFile::Rename(const std::string& new_path) {
483*795d594fSAndroid Build Coastguard Worker   if (kCheckSafeUsage) {
484*795d594fSAndroid Build Coastguard Worker     // Filesystems that use delayed allocation (e.g., ext4) may journal a rename before a data
485*795d594fSAndroid Build Coastguard Worker     // update is written to disk. Therefore on system crash, the data update may not persist.
486*795d594fSAndroid Build Coastguard Worker     // Guard against this by ensuring the file has been flushed prior to rename.
487*795d594fSAndroid Build Coastguard Worker     if (guard_state_ < GuardState::kFlushed) {
488*795d594fSAndroid Build Coastguard Worker       LOG(ERROR) << "File " << file_path_ << " has not been flushed before renaming.";
489*795d594fSAndroid Build Coastguard Worker     }
490*795d594fSAndroid Build Coastguard Worker     DCHECK_GE(guard_state_, GuardState::kFlushed);
491*795d594fSAndroid Build Coastguard Worker   }
492*795d594fSAndroid Build Coastguard Worker 
493*795d594fSAndroid Build Coastguard Worker   if (!FilePathMatchesFd()) {
494*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "Failed rename because the file descriptor is not backed by the expected file "
495*795d594fSAndroid Build Coastguard Worker                << "path: " << file_path_;
496*795d594fSAndroid Build Coastguard Worker     return false;
497*795d594fSAndroid Build Coastguard Worker   }
498*795d594fSAndroid Build Coastguard Worker 
499*795d594fSAndroid Build Coastguard Worker   std::string old_path = file_path_;
500*795d594fSAndroid Build Coastguard Worker   int rc = std::rename(old_path.c_str(), new_path.c_str());
501*795d594fSAndroid Build Coastguard Worker   if (rc != 0) {
502*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "Rename from '" << old_path << "' to '" << new_path << "' failed.";
503*795d594fSAndroid Build Coastguard Worker     return false;
504*795d594fSAndroid Build Coastguard Worker   }
505*795d594fSAndroid Build Coastguard Worker   file_path_ = new_path;
506*795d594fSAndroid Build Coastguard Worker 
507*795d594fSAndroid Build Coastguard Worker   // Rename modifies the directory entries mapped within the parent directory file descriptor(s),
508*795d594fSAndroid Build Coastguard Worker   // rather than the file, so flushing the file will not persist the change to disk. Therefore, we
509*795d594fSAndroid Build Coastguard Worker   // flush the parent directory file descriptor(s).
510*795d594fSAndroid Build Coastguard Worker   std::string old_dir = android::base::Dirname(old_path);
511*795d594fSAndroid Build Coastguard Worker   std::string new_dir = android::base::Dirname(new_path);
512*795d594fSAndroid Build Coastguard Worker   std::vector<std::string> sync_dirs = {new_dir};
513*795d594fSAndroid Build Coastguard Worker   if (new_dir != old_dir) {
514*795d594fSAndroid Build Coastguard Worker     sync_dirs.emplace_back(old_dir);
515*795d594fSAndroid Build Coastguard Worker   }
516*795d594fSAndroid Build Coastguard Worker   for (auto& dirname : sync_dirs) {
517*795d594fSAndroid Build Coastguard Worker     FdFile dir = FdFile(dirname, O_RDONLY, /*check_usage=*/false);
518*795d594fSAndroid Build Coastguard Worker     rc = dir.Flush(/*flush_metadata=*/true);
519*795d594fSAndroid Build Coastguard Worker     if (rc != 0) {
520*795d594fSAndroid Build Coastguard Worker       LOG(ERROR) << "Flushing directory '" << dirname << "' during rename failed.";
521*795d594fSAndroid Build Coastguard Worker       return false;
522*795d594fSAndroid Build Coastguard Worker     }
523*795d594fSAndroid Build Coastguard Worker     rc = dir.Close();
524*795d594fSAndroid Build Coastguard Worker     if (rc != 0) {
525*795d594fSAndroid Build Coastguard Worker       LOG(ERROR) << "Closing directory '" << dirname << "' during rename failed.";
526*795d594fSAndroid Build Coastguard Worker       return false;
527*795d594fSAndroid Build Coastguard Worker     }
528*795d594fSAndroid Build Coastguard Worker   }
529*795d594fSAndroid Build Coastguard Worker   return true;
530*795d594fSAndroid Build Coastguard Worker }
531*795d594fSAndroid Build Coastguard Worker 
532*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
SparseWrite(const uint8_t * data,size_t size,const std::vector<uint8_t> & zeroes)533*795d594fSAndroid Build Coastguard Worker bool FdFile::SparseWrite(const uint8_t* data,
534*795d594fSAndroid Build Coastguard Worker                          size_t size,
535*795d594fSAndroid Build Coastguard Worker                          const std::vector<uint8_t>& zeroes) {
536*795d594fSAndroid Build Coastguard Worker   DCHECK_GE(zeroes.size(), size);
537*795d594fSAndroid Build Coastguard Worker   if (memcmp(zeroes.data(), data, size) == 0) {
538*795d594fSAndroid Build Coastguard Worker     // These bytes are all zeroes, skip them by moving the file offset via lseek SEEK_CUR (available
539*795d594fSAndroid Build Coastguard Worker     // since linux kernel 3.1).
540*795d594fSAndroid Build Coastguard Worker     if (TEMP_FAILURE_RETRY(lseek(Fd(), size, SEEK_CUR)) < 0) {
541*795d594fSAndroid Build Coastguard Worker       return false;
542*795d594fSAndroid Build Coastguard Worker     }
543*795d594fSAndroid Build Coastguard Worker   } else {
544*795d594fSAndroid Build Coastguard Worker     if (!WriteFully(data, size)) {
545*795d594fSAndroid Build Coastguard Worker       return false;
546*795d594fSAndroid Build Coastguard Worker     }
547*795d594fSAndroid Build Coastguard Worker   }
548*795d594fSAndroid Build Coastguard Worker   return true;
549*795d594fSAndroid Build Coastguard Worker }
550*795d594fSAndroid Build Coastguard Worker 
UserspaceSparseCopy(const FdFile * input_file,off_t off,size_t size,size_t fs_blocksize)551*795d594fSAndroid Build Coastguard Worker bool FdFile::UserspaceSparseCopy(const FdFile* input_file,
552*795d594fSAndroid Build Coastguard Worker                                  off_t off,
553*795d594fSAndroid Build Coastguard Worker                                  size_t size,
554*795d594fSAndroid Build Coastguard Worker                                  size_t fs_blocksize) {
555*795d594fSAndroid Build Coastguard Worker   // Map the input file. We will begin the copy 'off' bytes into the map.
556*795d594fSAndroid Build Coastguard Worker   art::MemMap::Init();
557*795d594fSAndroid Build Coastguard Worker   std::string error_msg;
558*795d594fSAndroid Build Coastguard Worker   art::MemMap mmap = art::MemMap::MapFile(off + size,
559*795d594fSAndroid Build Coastguard Worker                                           PROT_READ,
560*795d594fSAndroid Build Coastguard Worker                                           MAP_PRIVATE,
561*795d594fSAndroid Build Coastguard Worker                                           input_file->Fd(),
562*795d594fSAndroid Build Coastguard Worker                                           /*start=*/0,
563*795d594fSAndroid Build Coastguard Worker                                           /*low_4gb=*/false,
564*795d594fSAndroid Build Coastguard Worker                                           input_file->GetPath().c_str(),
565*795d594fSAndroid Build Coastguard Worker                                           &error_msg);
566*795d594fSAndroid Build Coastguard Worker   if (!mmap.IsValid()) {
567*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "Failed to mmap " << input_file->GetPath() << " for copying: " << error_msg;
568*795d594fSAndroid Build Coastguard Worker     return false;
569*795d594fSAndroid Build Coastguard Worker   }
570*795d594fSAndroid Build Coastguard Worker 
571*795d594fSAndroid Build Coastguard Worker   std::vector<uint8_t> zeroes(/*n=*/fs_blocksize, /*val=*/0);
572*795d594fSAndroid Build Coastguard Worker 
573*795d594fSAndroid Build Coastguard Worker   // Iterate through each fs_blocksize of the copy region.
574*795d594fSAndroid Build Coastguard Worker   uint8_t* input_ptr = mmap.Begin() + off;
575*795d594fSAndroid Build Coastguard Worker   for (; (input_ptr + fs_blocksize) <= mmap.End(); input_ptr += fs_blocksize) {
576*795d594fSAndroid Build Coastguard Worker     if (!SparseWrite(input_ptr, fs_blocksize, zeroes)) {
577*795d594fSAndroid Build Coastguard Worker       return false;
578*795d594fSAndroid Build Coastguard Worker     }
579*795d594fSAndroid Build Coastguard Worker   }
580*795d594fSAndroid Build Coastguard Worker   // Finish copying any remaining bytes.
581*795d594fSAndroid Build Coastguard Worker   const size_t remaining_bytes = size % fs_blocksize;
582*795d594fSAndroid Build Coastguard Worker   if (remaining_bytes > 0) {
583*795d594fSAndroid Build Coastguard Worker     if (!SparseWrite(input_ptr, remaining_bytes, zeroes)) {
584*795d594fSAndroid Build Coastguard Worker       return false;
585*795d594fSAndroid Build Coastguard Worker     }
586*795d594fSAndroid Build Coastguard Worker   }
587*795d594fSAndroid Build Coastguard Worker   // Update the input file FD offset to the end of the copy region.
588*795d594fSAndroid Build Coastguard Worker   off_t input_offset = TEMP_FAILURE_RETRY(lseek(input_file->Fd(), off + size, SEEK_SET));
589*795d594fSAndroid Build Coastguard Worker   if (input_offset != (off + static_cast<off_t>(size))) {
590*795d594fSAndroid Build Coastguard Worker     return false;
591*795d594fSAndroid Build Coastguard Worker   }
592*795d594fSAndroid Build Coastguard Worker   return true;
593*795d594fSAndroid Build Coastguard Worker }
594*795d594fSAndroid Build Coastguard Worker #endif
595*795d594fSAndroid Build Coastguard Worker 
Copy(FdFile * input_file,int64_t offset,int64_t size)596*795d594fSAndroid Build Coastguard Worker bool FdFile::Copy(FdFile* input_file, int64_t offset, int64_t size) {
597*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
598*795d594fSAndroid Build Coastguard Worker   off_t off = static_cast<off_t>(offset);
599*795d594fSAndroid Build Coastguard Worker   off_t sz = static_cast<off_t>(size);
600*795d594fSAndroid Build Coastguard Worker   if (offset < 0 || static_cast<int64_t>(off) != offset ||
601*795d594fSAndroid Build Coastguard Worker       size < 0 || static_cast<int64_t>(sz) != size ||
602*795d594fSAndroid Build Coastguard Worker       sz > std::numeric_limits<off_t>::max() - off) {
603*795d594fSAndroid Build Coastguard Worker     errno = EINVAL;
604*795d594fSAndroid Build Coastguard Worker     return false;
605*795d594fSAndroid Build Coastguard Worker   }
606*795d594fSAndroid Build Coastguard Worker   if (size == 0) {
607*795d594fSAndroid Build Coastguard Worker     return true;
608*795d594fSAndroid Build Coastguard Worker   }
609*795d594fSAndroid Build Coastguard Worker 
610*795d594fSAndroid Build Coastguard Worker #ifdef __linux__
611*795d594fSAndroid Build Coastguard Worker   off_t current_offset = TEMP_FAILURE_RETRY(lseek(Fd(), 0, SEEK_CUR));
612*795d594fSAndroid Build Coastguard Worker   if (GetLength() > current_offset) {
613*795d594fSAndroid Build Coastguard Worker     // Copying to an existing region of the destination file is not supported. The current
614*795d594fSAndroid Build Coastguard Worker     // implementation would incorrectly preserve all existing data regions within the output file
615*795d594fSAndroid Build Coastguard Worker     // which match the locations of holes within the input file.
616*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "Cannot copy into an existing region of the destination file.";
617*795d594fSAndroid Build Coastguard Worker     errno = EINVAL;
618*795d594fSAndroid Build Coastguard Worker     return false;
619*795d594fSAndroid Build Coastguard Worker   }
620*795d594fSAndroid Build Coastguard Worker   struct stat output_stat;
621*795d594fSAndroid Build Coastguard Worker   if (TEMP_FAILURE_RETRY(fstat(Fd(), &output_stat)) < 0) {
622*795d594fSAndroid Build Coastguard Worker     return false;
623*795d594fSAndroid Build Coastguard Worker   }
624*795d594fSAndroid Build Coastguard Worker   const off_t fs_blocksize = output_stat.st_blksize;
625*795d594fSAndroid Build Coastguard Worker   if (!art::IsAlignedParam(current_offset, fs_blocksize)) {
626*795d594fSAndroid Build Coastguard Worker     // The input region is copied (skipped or written) in chunks of the output file's blocksize. For
627*795d594fSAndroid Build Coastguard Worker     // those chunks to be represented as holes or data, they should land as aligned blocks in the
628*795d594fSAndroid Build Coastguard Worker     // output file. Therefore, here we enforce that the current output offset is aligned.
629*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "Copy destination FD offset (" << current_offset << ") must be aligned with"
630*795d594fSAndroid Build Coastguard Worker                << " blocksize (" << fs_blocksize << ").";
631*795d594fSAndroid Build Coastguard Worker     errno = EINVAL;
632*795d594fSAndroid Build Coastguard Worker     return false;
633*795d594fSAndroid Build Coastguard Worker   }
634*795d594fSAndroid Build Coastguard Worker   const size_t end_length = GetLength() + sz;
635*795d594fSAndroid Build Coastguard Worker   if (!UserspaceSparseCopy(input_file, off, sz, fs_blocksize)) {
636*795d594fSAndroid Build Coastguard Worker     return false;
637*795d594fSAndroid Build Coastguard Worker   }
638*795d594fSAndroid Build Coastguard Worker   // In case the last blocks of the input file were a hole, fix the length to what would have been
639*795d594fSAndroid Build Coastguard Worker   // set if they had been data.
640*795d594fSAndroid Build Coastguard Worker   if (SetLength(end_length) != 0) {
641*795d594fSAndroid Build Coastguard Worker     return false;
642*795d594fSAndroid Build Coastguard Worker   }
643*795d594fSAndroid Build Coastguard Worker #else
644*795d594fSAndroid Build Coastguard Worker   if (lseek(input_file->Fd(), off, SEEK_SET) != off) {
645*795d594fSAndroid Build Coastguard Worker     return false;
646*795d594fSAndroid Build Coastguard Worker   }
647*795d594fSAndroid Build Coastguard Worker   constexpr size_t kMaxBufferSize = 16 * ::art::KB;
648*795d594fSAndroid Build Coastguard Worker   const size_t buffer_size = std::min<uint64_t>(size, kMaxBufferSize);
649*795d594fSAndroid Build Coastguard Worker   art::UniqueCPtr<void> buffer(malloc(buffer_size));
650*795d594fSAndroid Build Coastguard Worker   if (buffer == nullptr) {
651*795d594fSAndroid Build Coastguard Worker     errno = ENOMEM;
652*795d594fSAndroid Build Coastguard Worker     return false;
653*795d594fSAndroid Build Coastguard Worker   }
654*795d594fSAndroid Build Coastguard Worker   while (size != 0) {
655*795d594fSAndroid Build Coastguard Worker     size_t chunk_size = std::min<uint64_t>(buffer_size, size);
656*795d594fSAndroid Build Coastguard Worker     if (!input_file->ReadFully(buffer.get(), chunk_size) ||
657*795d594fSAndroid Build Coastguard Worker         !WriteFully(buffer.get(), chunk_size)) {
658*795d594fSAndroid Build Coastguard Worker       return false;
659*795d594fSAndroid Build Coastguard Worker     }
660*795d594fSAndroid Build Coastguard Worker     size -= chunk_size;
661*795d594fSAndroid Build Coastguard Worker   }
662*795d594fSAndroid Build Coastguard Worker #endif
663*795d594fSAndroid Build Coastguard Worker   return true;
664*795d594fSAndroid Build Coastguard Worker }
665*795d594fSAndroid Build Coastguard Worker 
FilePathMatchesFd()666*795d594fSAndroid Build Coastguard Worker bool FdFile::FilePathMatchesFd() {
667*795d594fSAndroid Build Coastguard Worker   if (file_path_.empty()) {
668*795d594fSAndroid Build Coastguard Worker     return false;
669*795d594fSAndroid Build Coastguard Worker   }
670*795d594fSAndroid Build Coastguard Worker   // Try to figure out whether file_path_ is still referring to the one on disk.
671*795d594fSAndroid Build Coastguard Worker   bool is_current = false;
672*795d594fSAndroid Build Coastguard Worker   struct stat this_stat, current_stat;
673*795d594fSAndroid Build Coastguard Worker   int cur_fd = TEMP_FAILURE_RETRY(open(file_path_.c_str(), O_RDONLY | O_CLOEXEC));
674*795d594fSAndroid Build Coastguard Worker   if (cur_fd > 0) {
675*795d594fSAndroid Build Coastguard Worker     // File still exists.
676*795d594fSAndroid Build Coastguard Worker     if (fstat(fd_, &this_stat) == 0 && fstat(cur_fd, &current_stat) == 0) {
677*795d594fSAndroid Build Coastguard Worker       is_current = (this_stat.st_dev == current_stat.st_dev) &&
678*795d594fSAndroid Build Coastguard Worker                    (this_stat.st_ino == current_stat.st_ino);
679*795d594fSAndroid Build Coastguard Worker     }
680*795d594fSAndroid Build Coastguard Worker     close(cur_fd);
681*795d594fSAndroid Build Coastguard Worker   }
682*795d594fSAndroid Build Coastguard Worker   return is_current;
683*795d594fSAndroid Build Coastguard Worker }
684*795d594fSAndroid Build Coastguard Worker 
Unlink()685*795d594fSAndroid Build Coastguard Worker bool FdFile::Unlink() {
686*795d594fSAndroid Build Coastguard Worker   bool is_current = FilePathMatchesFd();
687*795d594fSAndroid Build Coastguard Worker   if (is_current) {
688*795d594fSAndroid Build Coastguard Worker     unlink(file_path_.c_str());
689*795d594fSAndroid Build Coastguard Worker   }
690*795d594fSAndroid Build Coastguard Worker 
691*795d594fSAndroid Build Coastguard Worker   return is_current;
692*795d594fSAndroid Build Coastguard Worker }
693*795d594fSAndroid Build Coastguard Worker 
Erase(bool unlink)694*795d594fSAndroid Build Coastguard Worker bool FdFile::Erase(bool unlink) {
695*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
696*795d594fSAndroid Build Coastguard Worker 
697*795d594fSAndroid Build Coastguard Worker   bool ret_result = true;
698*795d594fSAndroid Build Coastguard Worker   if (unlink) {
699*795d594fSAndroid Build Coastguard Worker     ret_result = Unlink();
700*795d594fSAndroid Build Coastguard Worker   }
701*795d594fSAndroid Build Coastguard Worker 
702*795d594fSAndroid Build Coastguard Worker   int result;
703*795d594fSAndroid Build Coastguard Worker   result = SetLength(0);
704*795d594fSAndroid Build Coastguard Worker   result = Flush();
705*795d594fSAndroid Build Coastguard Worker   result = Close();
706*795d594fSAndroid Build Coastguard Worker   // Ignore the errors.
707*795d594fSAndroid Build Coastguard Worker   (void) result;
708*795d594fSAndroid Build Coastguard Worker 
709*795d594fSAndroid Build Coastguard Worker   return ret_result;
710*795d594fSAndroid Build Coastguard Worker }
711*795d594fSAndroid Build Coastguard Worker 
FlushCloseOrErase()712*795d594fSAndroid Build Coastguard Worker int FdFile::FlushCloseOrErase() {
713*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
714*795d594fSAndroid Build Coastguard Worker   int flush_result = Flush();
715*795d594fSAndroid Build Coastguard Worker   if (flush_result != 0) {
716*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "CloseOrErase failed while flushing a file.";
717*795d594fSAndroid Build Coastguard Worker     Erase();
718*795d594fSAndroid Build Coastguard Worker     return flush_result;
719*795d594fSAndroid Build Coastguard Worker   }
720*795d594fSAndroid Build Coastguard Worker   int close_result = Close();
721*795d594fSAndroid Build Coastguard Worker   if (close_result != 0) {
722*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "CloseOrErase failed while closing a file.";
723*795d594fSAndroid Build Coastguard Worker     Erase();
724*795d594fSAndroid Build Coastguard Worker     return close_result;
725*795d594fSAndroid Build Coastguard Worker   }
726*795d594fSAndroid Build Coastguard Worker   return 0;
727*795d594fSAndroid Build Coastguard Worker }
728*795d594fSAndroid Build Coastguard Worker 
FlushClose()729*795d594fSAndroid Build Coastguard Worker int FdFile::FlushClose() {
730*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
731*795d594fSAndroid Build Coastguard Worker   int flush_result = Flush();
732*795d594fSAndroid Build Coastguard Worker   if (flush_result != 0) {
733*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "FlushClose failed while flushing a file.";
734*795d594fSAndroid Build Coastguard Worker   }
735*795d594fSAndroid Build Coastguard Worker   int close_result = Close();
736*795d594fSAndroid Build Coastguard Worker   if (close_result != 0) {
737*795d594fSAndroid Build Coastguard Worker     LOG(ERROR) << "FlushClose failed while closing a file.";
738*795d594fSAndroid Build Coastguard Worker   }
739*795d594fSAndroid Build Coastguard Worker   return (flush_result != 0) ? flush_result : close_result;
740*795d594fSAndroid Build Coastguard Worker }
741*795d594fSAndroid Build Coastguard Worker 
MarkUnchecked()742*795d594fSAndroid Build Coastguard Worker void FdFile::MarkUnchecked() {
743*795d594fSAndroid Build Coastguard Worker   guard_state_ = GuardState::kNoCheck;
744*795d594fSAndroid Build Coastguard Worker }
745*795d594fSAndroid Build Coastguard Worker 
ClearContent()746*795d594fSAndroid Build Coastguard Worker bool FdFile::ClearContent() {
747*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
748*795d594fSAndroid Build Coastguard Worker   if (SetLength(0) < 0) {
749*795d594fSAndroid Build Coastguard Worker     PLOG(ERROR) << "Failed to reset the length";
750*795d594fSAndroid Build Coastguard Worker     return false;
751*795d594fSAndroid Build Coastguard Worker   }
752*795d594fSAndroid Build Coastguard Worker   return ResetOffset();
753*795d594fSAndroid Build Coastguard Worker }
754*795d594fSAndroid Build Coastguard Worker 
ResetOffset()755*795d594fSAndroid Build Coastguard Worker bool FdFile::ResetOffset() {
756*795d594fSAndroid Build Coastguard Worker   DCHECK(!read_only_mode_);
757*795d594fSAndroid Build Coastguard Worker   off_t rc =  TEMP_FAILURE_RETRY(lseek(fd_, 0, SEEK_SET));
758*795d594fSAndroid Build Coastguard Worker   if (rc == static_cast<off_t>(-1)) {
759*795d594fSAndroid Build Coastguard Worker     PLOG(ERROR) << "Failed to reset the offset";
760*795d594fSAndroid Build Coastguard Worker     return false;
761*795d594fSAndroid Build Coastguard Worker   }
762*795d594fSAndroid Build Coastguard Worker   return true;
763*795d594fSAndroid Build Coastguard Worker }
764*795d594fSAndroid Build Coastguard Worker 
Compare(FdFile * other)765*795d594fSAndroid Build Coastguard Worker int FdFile::Compare(FdFile* other) {
766*795d594fSAndroid Build Coastguard Worker   int64_t length = GetLength();
767*795d594fSAndroid Build Coastguard Worker   int64_t length2 = other->GetLength();
768*795d594fSAndroid Build Coastguard Worker   if (length != length2) {
769*795d594fSAndroid Build Coastguard Worker     return length < length2 ? -1 : 1;
770*795d594fSAndroid Build Coastguard Worker   }
771*795d594fSAndroid Build Coastguard Worker   static const size_t kBufferSize = 4096;
772*795d594fSAndroid Build Coastguard Worker   std::unique_ptr<uint8_t[]> buffer1(new uint8_t[kBufferSize]);
773*795d594fSAndroid Build Coastguard Worker   std::unique_ptr<uint8_t[]> buffer2(new uint8_t[kBufferSize]);
774*795d594fSAndroid Build Coastguard Worker   size_t offset = 0;
775*795d594fSAndroid Build Coastguard Worker   while (length > 0) {
776*795d594fSAndroid Build Coastguard Worker     size_t len = std::min(kBufferSize, static_cast<size_t>(length));
777*795d594fSAndroid Build Coastguard Worker     if (!PreadFully(&buffer1[0], len, offset)) {
778*795d594fSAndroid Build Coastguard Worker       return -1;
779*795d594fSAndroid Build Coastguard Worker     }
780*795d594fSAndroid Build Coastguard Worker     if (!other->PreadFully(&buffer2[0], len, offset)) {
781*795d594fSAndroid Build Coastguard Worker       return 1;
782*795d594fSAndroid Build Coastguard Worker     }
783*795d594fSAndroid Build Coastguard Worker     int result = memcmp(&buffer1[0], &buffer2[0], len);
784*795d594fSAndroid Build Coastguard Worker     if (result != 0) {
785*795d594fSAndroid Build Coastguard Worker       return result;
786*795d594fSAndroid Build Coastguard Worker     }
787*795d594fSAndroid Build Coastguard Worker     length -= len;
788*795d594fSAndroid Build Coastguard Worker     offset += len;
789*795d594fSAndroid Build Coastguard Worker   }
790*795d594fSAndroid Build Coastguard Worker   return 0;
791*795d594fSAndroid Build Coastguard Worker }
792*795d594fSAndroid Build Coastguard Worker 
IsOpenFd(int fd)793*795d594fSAndroid Build Coastguard Worker bool FdFile::IsOpenFd(int fd) {
794*795d594fSAndroid Build Coastguard Worker   if (fd == kInvalidFd) {
795*795d594fSAndroid Build Coastguard Worker     return false;
796*795d594fSAndroid Build Coastguard Worker   }
797*795d594fSAndroid Build Coastguard Worker   #ifdef _WIN32  // Windows toolchain does not support F_GETFD.
798*795d594fSAndroid Build Coastguard Worker     return true;
799*795d594fSAndroid Build Coastguard Worker   #else
800*795d594fSAndroid Build Coastguard Worker     int saved_errno = errno;
801*795d594fSAndroid Build Coastguard Worker     bool is_open = (fcntl(fd, F_GETFD) != -1);
802*795d594fSAndroid Build Coastguard Worker     errno = saved_errno;
803*795d594fSAndroid Build Coastguard Worker     return is_open;
804*795d594fSAndroid Build Coastguard Worker   #endif
805*795d594fSAndroid Build Coastguard Worker }
806*795d594fSAndroid Build Coastguard Worker 
807*795d594fSAndroid Build Coastguard Worker }  // namespace unix_file
808