Brian Silverman | 58899fd | 2019-03-24 11:03:11 -0700 | [diff] [blame] | 1 | #ifndef AOS_SCOPED_SCOPED_FD_H_ |
| 2 | #define AOS_SCOPED_SCOPED_FD_H_ |
Parker Schuh | 2cd173d | 2017-01-28 00:12:01 -0800 | [diff] [blame] | 3 | |
Brian Silverman | 8efe23e | 2013-07-07 23:31:37 -0700 | [diff] [blame] | 4 | #include <unistd.h> |
| 5 | |
John Park | 33858a3 | 2018-09-28 23:05:48 -0700 | [diff] [blame] | 6 | #include "aos/macros.h" |
brians | 343bc11 | 2013-02-10 01:53:46 +0000 | [diff] [blame] | 7 | |
| 8 | namespace aos { |
| 9 | |
| 10 | // Smart "pointer" (container) for a file descriptor. |
| 11 | class ScopedFD { |
| 12 | public: |
| 13 | explicit ScopedFD(int fd = -1) : fd_(fd) {} |
Parker Schuh | 2cd173d | 2017-01-28 00:12:01 -0800 | [diff] [blame] | 14 | ~ScopedFD() { Close(); } |
brians | 343bc11 | 2013-02-10 01:53:46 +0000 | [diff] [blame] | 15 | int get() const { return fd_; } |
| 16 | int release() { |
| 17 | const int r = fd_; |
| 18 | fd_ = -1; |
| 19 | return r; |
| 20 | } |
| 21 | void reset(int new_fd = -1) { |
| 22 | if (fd_ != new_fd) { |
| 23 | Close(); |
| 24 | fd_ = new_fd; |
| 25 | } |
| 26 | } |
| 27 | operator bool() const { return fd_ != -1; } |
Parker Schuh | 2cd173d | 2017-01-28 00:12:01 -0800 | [diff] [blame] | 28 | |
brians | 343bc11 | 2013-02-10 01:53:46 +0000 | [diff] [blame] | 29 | private: |
| 30 | int fd_; |
Brian Silverman | 58899fd | 2019-03-24 11:03:11 -0700 | [diff] [blame] | 31 | void Close(); |
brians | 343bc11 | 2013-02-10 01:53:46 +0000 | [diff] [blame] | 32 | DISALLOW_COPY_AND_ASSIGN(ScopedFD); |
| 33 | }; |
| 34 | |
| 35 | } // namespace aos |
Parker Schuh | 2cd173d | 2017-01-28 00:12:01 -0800 | [diff] [blame] | 36 | |
Brian Silverman | 58899fd | 2019-03-24 11:03:11 -0700 | [diff] [blame] | 37 | #endif // AOS_SCOPED_SCOPED_FD_H_ |