1 //
2 // Copyright (C) 2012 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 //      http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16 
17 #ifndef UPDATE_ENGINE_PAYLOAD_CONSUMER_FILE_DESCRIPTOR_H_
18 #define UPDATE_ENGINE_PAYLOAD_CONSUMER_FILE_DESCRIPTOR_H_
19 
20 #include <errno.h>
21 #include <sys/types.h>
22 #include <memory>
23 
24 #include <base/macros.h>
25 
26 // Abstraction for managing opening, reading, writing and closing of file
27 // descriptors. This includes an abstract class and one standard implementation
28 // based on POSIX system calls.
29 //
30 // TODO(garnold) this class is modeled after (and augments the functionality of)
31 // the FileWriter class; ultimately, the latter should be replaced by the former
32 // throughout the codebase.  A few deviations from the original FileWriter:
33 //
34 // * Providing two flavors of Open()
35 //
36 // * A FileDescriptor is reusable and can be used to read/write multiple files
37 //   as long as open/close preconditions are respected.
38 //
39 // * Write() returns the number of bytes written: this appears to be more useful
40 //   for clients, who may wish to retry or otherwise do something useful with
41 //   the remaining data that was not written.
42 
43 namespace chromeos_update_engine {
44 
45 class FileDescriptor;
46 using FileDescriptorPtr = std::shared_ptr<FileDescriptor>;
47 
48 // An abstract class defining the file descriptor API.
49 class FileDescriptor {
50  public:
FileDescriptor()51   FileDescriptor() {}
~FileDescriptor()52   virtual ~FileDescriptor() {}
53 
54   // Opens a file descriptor. The descriptor must be in the closed state prior
55   // to this call. Returns true on success, false otherwise. Specific
56   // implementations may set errno accordingly.
57   virtual bool Open(const char* path, int flags, mode_t mode) = 0;
58   virtual bool Open(const char* path, int flags) = 0;
59 
60   // Reads from a file descriptor up to a given count. The descriptor must be
61   // open prior to this call. Returns the number of bytes read, or -1 on error.
62   // Specific implementations may set errno accordingly.
63   virtual ssize_t Read(void* buf, size_t count) = 0;
64 
65   // Writes to a file descriptor. The descriptor must be open prior to this
66   // call. Returns the number of bytes written, or -1 if an error occurred and
67   // no bytes were written. Specific implementations may set errno accordingly.
68   virtual ssize_t Write(const void* buf, size_t count) = 0;
69 
70   // Seeks to an offset. Returns the resulting offset location as measured in
71   // bytes from the beginning. On error, return -1. Specific implementations
72   // may set errno accordingly.
73   virtual off64_t Seek(off64_t offset, int whence) = 0;
74 
75   // Return the size of the block device in bytes, or 0 if the device is not a
76   // block device or an error occurred.
77   virtual uint64_t BlockDevSize() = 0;
78 
79   // Runs a ioctl() on the file descriptor if supported. Returns whether
80   // the operation is supported. The |request| can be one of BLKDISCARD,
81   // BLKZEROOUT and BLKSECDISCARD to discard, write zeros or securely discard
82   // the blocks. These ioctls accept a range of bytes (|start| and |length|)
83   // over which they perform the operation. The return value from the ioctl is
84   // stored in |result|.
85   virtual bool BlkIoctl(int request,
86                         uint64_t start,
87                         uint64_t length,
88                         int* result) = 0;
89 
90   // Flushes any cached data. The descriptor must be opened prior to this
91   // call. Returns false if it fails to write data. Implementations may set
92   // errno accrodingly.
93   virtual bool Flush() = 0;
94 
95   // Closes a file descriptor. The descriptor must be open prior to this call.
96   // Returns true on success, false otherwise. Specific implementations may set
97   // errno accordingly.
98   virtual bool Close() = 0;
99 
100   // Indicates whether or not an implementation sets meaningful errno.
101   virtual bool IsSettingErrno() = 0;
102 
103   // Indicates whether the descriptor is currently open.
104   virtual bool IsOpen() = 0;
105 
106  private:
107   DISALLOW_COPY_AND_ASSIGN(FileDescriptor);
108 };
109 
110 // A simple EINTR-immune wrapper implementation around standard system calls.
111 class EintrSafeFileDescriptor : public FileDescriptor {
112  public:
EintrSafeFileDescriptor()113   EintrSafeFileDescriptor() : fd_(-1) {}
114   ~EintrSafeFileDescriptor();
115 
116   // Interface methods.
117   bool Open(const char* path, int flags, mode_t mode) override;
118   bool Open(const char* path, int flags) override;
119   ssize_t Read(void* buf, size_t count) override;
120   ssize_t Write(const void* buf, size_t count) override;
121   off64_t Seek(off64_t offset, int whence) override;
122   uint64_t BlockDevSize() override;
123   bool BlkIoctl(int request,
124                 uint64_t start,
125                 uint64_t length,
126                 int* result) override;
127   bool Flush() override;
128   bool Close() override;
IsSettingErrno()129   bool IsSettingErrno() override { return true; }
IsOpen()130   bool IsOpen() override { return (fd_ >= 0); }
131 
132  protected:
133   int fd_;
134 };
135 
136 }  // namespace chromeos_update_engine
137 
138 #endif  // UPDATE_ENGINE_PAYLOAD_CONSUMER_FILE_DESCRIPTOR_H_
139