1 : /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
2 : * ***** BEGIN LICENSE BLOCK *****
3 : * Version: MPL 1.1/GPL 2.0/LGPL 2.1
4 : *
5 : * The contents of this file are subject to the Mozilla Public License Version
6 : * 1.1 (the "License"); you may not use this file except in compliance with
7 : * the License. You may obtain a copy of the License at
8 : * http://www.mozilla.org/MPL/
9 : *
10 : * Software distributed under the License is distributed on an "AS IS" basis,
11 : * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
12 : * for the specific language governing rights and limitations under the
13 : * License.
14 : *
15 : * The Original Code is Mozilla code.
16 : *
17 : * The Initial Developer of the Original Code is
18 : * Mozilla Foundation.
19 : * Portions created by the Initial Developer are Copyright (C) 2010
20 : * the Initial Developer. All Rights Reserved.
21 : *
22 : * Contributor(s):
23 : * Benjamin Smedberg <benjamin@smedbergs.us>
24 : * Taras Glek <tglek@mozilla.com>
25 : *
26 : * Alternatively, the contents of this file may be used under the terms of
27 : * either the GNU General Public License Version 2 or later (the "GPL"), or
28 : * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
29 : * in which case the provisions of the GPL or the LGPL are applicable instead
30 : * of those above. If you wish to allow use of your version of this file only
31 : * under the terms of either the GPL or the LGPL, and not to allow others to
32 : * use your version of this file under the terms of the MPL, indicate your
33 : * decision by deleting the provisions above and replace them with the notice
34 : * and other provisions required by the GPL or the LGPL. If you do not delete
35 : * the provisions above, a recipient may use your version of this file under
36 : * the terms of any one of the MPL, the GPL or the LGPL.
37 : *
38 : * ***** END LICENSE BLOCK ***** */
39 :
40 : #ifndef mozilla_FileUtils_h
41 : #define mozilla_FileUtils_h
42 :
43 : #if defined(XP_UNIX) || defined(XP_OS2)
44 : # include <unistd.h>
45 : #elif defined(XP_WIN)
46 : # include <io.h>
47 : #endif
48 : #include "prio.h"
49 :
50 : namespace mozilla {
51 :
52 : /**
53 : * AutoFDClose is a RAII wrapper for PRFileDesc.
54 : **/
55 : class AutoFDClose
56 : {
57 : public:
58 322749 : AutoFDClose(PRFileDesc* fd = nsnull) : mFD(fd) { }
59 322749 : ~AutoFDClose() { if (mFD) PR_Close(mFD); }
60 :
61 320207 : PRFileDesc* operator= (PRFileDesc *fd) {
62 320207 : if (mFD) PR_Close(mFD);
63 320207 : mFD = fd;
64 320207 : return fd;
65 : }
66 :
67 1279768 : operator PRFileDesc* () { return mFD; }
68 320207 : PRFileDesc** operator &() { *this = nsnull; return &mFD; }
69 :
70 : private:
71 : PRFileDesc *mFD;
72 : };
73 :
74 : /**
75 : * Instances close() their fds when they go out of scope.
76 : */
77 : struct ScopedClose
78 : {
79 0 : ScopedClose(int aFd=-1) : mFd(aFd) {}
80 0 : ~ScopedClose() {
81 0 : if (0 <= mFd) {
82 0 : close(mFd);
83 : }
84 0 : }
85 : int mFd;
86 : };
87 :
88 : /**
89 : * Fallocate efficiently and continuously allocates files via fallocate-type APIs.
90 : * This is useful for avoiding fragmentation.
91 : * On sucess the file be padded with zeros to grow to aLength.
92 : *
93 : * @param aFD file descriptor.
94 : * @param aLength length of file to grow to.
95 : * @return true on success.
96 : */
97 : NS_COM_GLUE bool fallocate(PRFileDesc *aFD, PRInt64 aLength);
98 :
99 : } // namespace mozilla
100 : #endif
|