folly/File.h

   1 /*
   2  * Copyright 2013 Facebook, Inc.
   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 FOLLY_FILE_H_
  18 #define FOLLY_FILE_H_
  19
  20 #include <sys/stat.h>
  21 #include <sys/types.h>
  22 #include <fcntl.h>
  23 #include <unistd.h>
  24
  25 namespace folly {
  26
  27 /**
  28  * A File represents an open file.
  29  */
  30 class File {
  31  public:
  32   /**
  33    * Creates an empty File object, for late initialization.
  34    */
  35   File();
  36
  37   /**
  38    * Create a File object from an existing file descriptor.
  39    * Takes ownership of the file descriptor if ownsFd is true.
  40    */
  41   /* implicit */ File(int fd,
  42                       bool ownsFd = false);
  43
  44   /**
  45    * Open and create a file object.  Throws on error.
  46    */
  47   /* implicit */ File(const char* name,
  48                       int flags = O_RDONLY,
  49                       mode_t mode = 0644);
  50
  51   ~File();
  52
  53   /**
  54    * Create and return a temporary, owned file (uses tmpfile()).
  55    */
  56   static File temporary();
  57
  58   /**
  59    * Return the file descriptor, or -1 if the file was closed.
  60    */
  61   int fd() const { return fd_; }
  62
  63   /**
  64    * Returns 'true' iff the file was successfully opened.
  65    */
  66   explicit operator bool() const {
  67     return fd_ != -1;
  68   }
  69
  70   /**
  71    * Duplicate file descriptor and return File that owns it.
  72    */
  73   File dup() const;
  74
  75   /**
  76    * If we own the file descriptor, close the file and throw on error.
  77    * Otherwise, do nothing.
  78    */
  79   void close();
  80
  81   /**
  82    * Closes the file (if owned).  Returns true on success, false (and sets
  83    * errno) on error.
  84    */
  85   bool closeNoThrow();
  86
  87   /**
  88    * Returns and releases the file descriptor; no longer owned by this File.
  89    * Returns -1 if the File object didn't wrap a file.
  90    */
  91   int release();
  92
  93   /**
  94    * Swap this File with another.
  95    */
  96   void swap(File& other);
  97
  98   // movable
  99   File(File&&);
 100   File& operator=(File&&);
 101
 102   // FLOCK (INTERPROCESS) LOCKS
 103   //
 104   // NOTE THAT THESE LOCKS ARE flock() LOCKS.  That is, they may only be used
 105   // for inter-process synchronization -- an attempt to acquire a second lock
 106   // on the same file descriptor from the same process may succeed.  Attempting
 107   // to acquire a second lock on a different file descriptor for the same file
 108   // should fail, but some systems might implement flock() using fcntl() locks,
 109   // in which case it will succeed.
 110   void lock();
 111   bool try_lock();
 112   void unlock();
 113
 114   void lock_shared();
 115   bool try_lock_shared();
 116   void unlock_shared();
 117
 118  private:
 119   void doLock(int op);
 120   bool doTryLock(int op);
 121
 122   // unique
 123   File(const File&) = delete;
 124   File& operator=(const File&) = delete;
 125
 126   int fd_;
 127   bool ownsFd_;
 128 };
 129
 130 void swap(File& a, File& b);
 131
 132
 133 }  // namespace folly
 134
 135 #endif /* FOLLY_FILE_H_ */