follow method attribute suggestions by gcc
[ntk/apt.git] / apt-pkg / contrib / fileutl.h
CommitLineData
578bfd0a
AL
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
233b185f 3// $Id: fileutl.h,v 1.26 2001/05/07 05:06:52 jgg Exp $
578bfd0a
AL
4/* ######################################################################
5
6 File Utilities
7
8 CopyFile - Buffered copy of a single file
9 GetLock - dpkg compatible lock file manipulation (fcntl)
10 FileExists - Returns true if the file exists
11 SafeGetCWD - Returns the CWD in a string with overrun protection
12
13 The file class is a handy abstraction for various functions+classes
14 that need to accept filenames.
15
16 This source is placed in the Public Domain, do with it what you will
17 It was originally written by Jason Gunthorpe.
18
19 ##################################################################### */
20 /*}}}*/
578bfd0a
AL
21#ifndef PKGLIB_FILEUTL_H
22#define PKGLIB_FILEUTL_H
23
1cd1c398 24#include <apt-pkg/macros.h>
699b209e 25#include <apt-pkg/aptconfiguration.h>
6c139d6e 26
578bfd0a 27#include <string>
46e39c8e 28#include <vector>
e45c4617 29#include <set>
453b82a3 30#include <time.h>
578bfd0a 31
a3a03f5d 32#include <zlib.h>
33
a4f6bdc8
DK
34#ifndef APT_8_CLEANER_HEADERS
35using std::string;
36#endif
37
144c0969
JAK
38/* Define this for python-apt */
39#define APT_HAS_GZIP 1
40
032bd56f 41class FileFdPrivate;
8e06abb2 42class FileFd
578bfd0a
AL
43{
44 protected:
45 int iFd;
46
ddc1d8d0 47 enum LocalFlags {AutoClose = (1<<0),Fail = (1<<1),DelOnFail = (1<<2),
032bd56f 48 HitEof = (1<<3), Replace = (1<<4), Compressed = (1<<5) };
578bfd0a 49 unsigned long Flags;
8f3ba4e8
DK
50 std::string FileName;
51 std::string TemporaryFileName;
3184b4cf 52
578bfd0a 53 public:
257e8d66
DK
54 enum OpenMode {
55 ReadOnly = (1 << 0),
56 WriteOnly = (1 << 1),
57 ReadWrite = ReadOnly | WriteOnly,
58
59 Create = (1 << 2),
60 Exclusive = (1 << 3),
61 Atomic = Exclusive | (1 << 4),
62 Empty = (1 << 5),
63
64 WriteEmpty = ReadWrite | Create | Empty,
65 WriteExists = ReadWrite,
66 WriteAny = ReadWrite | Create,
67 WriteTemp = ReadWrite | Create | Exclusive,
68 ReadOnlyGzip,
69 WriteAtomic = ReadWrite | Create | Atomic
70 };
468720c5 71 enum CompressMode { Auto = 'A', None = 'N', Extension = 'E', Gzip = 'G', Bzip2 = 'B', Lzma = 'L', Xz = 'X' };
578bfd0a 72
650faab0 73 inline bool Read(void *To,unsigned long long Size,bool AllowEof)
f604cf55 74 {
650faab0 75 unsigned long long Jnk;
f604cf55
AL
76 if (AllowEof)
77 return Read(To,Size,&Jnk);
78 return Read(To,Size);
79 }
650faab0 80 bool Read(void *To,unsigned long long Size,unsigned long long *Actual = 0);
032bd56f 81 char* ReadLine(char *To, unsigned long long const Size);
650faab0 82 bool Write(const void *From,unsigned long long Size);
d68d65ad 83 bool static Write(int Fd, const void *From, unsigned long long Size);
650faab0
DK
84 bool Seek(unsigned long long To);
85 bool Skip(unsigned long long To);
86 bool Truncate(unsigned long long To);
87 unsigned long long Tell();
88 unsigned long long Size();
89 unsigned long long FileSize();
76a763e1 90 time_t ModificationTime();
650faab0
DK
91
92 /* You want to use 'unsigned long long' if you are talking about a file
93 to be able to support large files (>2 or >4 GB) properly.
94 This shouldn't happen all to often for the indexes, but deb's might be…
95 And as the auto-conversation converts a 'unsigned long *' to a 'bool'
96 instead of 'unsigned long long *' we need to provide this explicitely -
97 otherwise applications magically start to fail… */
453b82a3 98 bool Read(void *To,unsigned long long Size,unsigned long *Actual) APT_DEPRECATED
650faab0
DK
99 {
100 unsigned long long R;
101 bool const T = Read(To, Size, &R);
102 *Actual = R;
103 return T;
104 }
105
52b47296
DK
106 bool Open(std::string FileName,unsigned int const Mode,CompressMode Compress,unsigned long const Perms = 0666);
107 bool Open(std::string FileName,unsigned int const Mode,APT::Configuration::Compressor const &compressor,unsigned long const Perms = 0666);
108 inline bool Open(std::string const &FileName,unsigned int const Mode, unsigned long const Perms = 0666) {
257e8d66
DK
109 return Open(FileName, Mode, None, Perms);
110 };
52b47296
DK
111 bool OpenDescriptor(int Fd, unsigned int const Mode, CompressMode Compress, bool AutoClose=false);
112 bool OpenDescriptor(int Fd, unsigned int const Mode, APT::Configuration::Compressor const &compressor, bool AutoClose=false);
113 inline bool OpenDescriptor(int Fd, unsigned int const Mode, bool AutoClose=false) {
bce778a3 114 return OpenDescriptor(Fd, Mode, None, AutoClose);
257e8d66 115 };
578bfd0a 116 bool Close();
b2e465d6
AL
117 bool Sync();
118
578bfd0a
AL
119 // Simple manipulators
120 inline int Fd() {return iFd;};
d0254ba7 121 inline void Fd(int fd) { OpenDescriptor(fd, ReadWrite);};
a02db58f 122 gzFile gzFd() APT_DEPRECATED APT_PURE;
699b209e 123
578bfd0a
AL
124 inline bool IsOpen() {return iFd >= 0;};
125 inline bool Failed() {return (Flags & Fail) == Fail;};
126 inline void EraseOnFailure() {Flags |= DelOnFail;};
127 inline void OpFail() {Flags |= Fail;};
ddc1d8d0 128 inline bool Eof() {return (Flags & HitEof) == HitEof;};
032bd56f 129 inline bool IsCompressed() {return (Flags & Compressed) == Compressed;};
8f3ba4e8 130 inline std::string &Name() {return FileName;};
c7b5ce1c 131
52b47296 132 FileFd(std::string FileName,unsigned int const Mode,unsigned long Perms = 0666) : iFd(-1), Flags(0), d(NULL)
13d87e2e 133 {
468720c5
DK
134 Open(FileName,Mode, None, Perms);
135 };
52b47296 136 FileFd(std::string FileName,unsigned int const Mode, CompressMode Compress, unsigned long Perms = 0666) : iFd(-1), Flags(0), d(NULL)
468720c5
DK
137 {
138 Open(FileName,Mode, Compress, Perms);
13d87e2e 139 };
032bd56f 140 FileFd() : iFd(-1), Flags(AutoClose), d(NULL) {};
52b47296 141 FileFd(int const Fd, unsigned int const Mode = ReadWrite, CompressMode Compress = None) : iFd(-1), Flags(0), d(NULL)
032bd56f
DK
142 {
143 OpenDescriptor(Fd, Mode, Compress);
144 };
145 FileFd(int const Fd, bool const AutoClose) : iFd(-1), Flags(0), d(NULL)
146 {
147 OpenDescriptor(Fd, ReadWrite, None, AutoClose);
148 };
8e06abb2 149 virtual ~FileFd();
468720c5
DK
150
151 private:
032bd56f 152 FileFdPrivate* d;
52b47296 153 bool OpenInternDescriptor(unsigned int const Mode, APT::Configuration::Compressor const &compressor);
ae635e3c
DK
154
155 // private helpers to set Fail flag and call _error->Error
453b82a3
DK
156 bool FileFdErrno(const char* Function, const char* Description,...) APT_PRINTF(3) APT_COLD;
157 bool FileFdError(const char* Description,...) APT_PRINTF(2) APT_COLD;
578bfd0a
AL
158};
159
614adaa0 160bool RunScripts(const char *Cnf);
8b89e57f 161bool CopyFile(FileFd &From,FileFd &To);
8f3ba4e8
DK
162int GetLock(std::string File,bool Errors = true);
163bool FileExists(std::string File);
164bool RealFileExists(std::string File);
453b82a3 165bool DirectoryExists(std::string const &Path) APT_CONST;
8f3ba4e8
DK
166bool CreateDirectory(std::string const &Parent, std::string const &Path);
167time_t GetModificationTime(std::string const &Path);
b29c3712 168
68e01721
MV
169std::string GetTempDir();
170
b29c3712
DK
171/** \brief Ensure the existence of the given Path
172 *
173 * \param Parent directory of the Path directory - a trailing
174 * /apt/ will be removed before CreateDirectory call.
175 * \param Path which should exist after (successful) call
176 */
8f3ba4e8 177bool CreateAPTDirectoryIfNeeded(std::string const &Parent, std::string const &Path);
b29c3712 178
8f3ba4e8 179std::vector<std::string> GetListOfFilesInDir(std::string const &Dir, std::string const &Ext,
b3793d41 180 bool const &SortList, bool const &AllowNoExt=false);
8f3ba4e8 181std::vector<std::string> GetListOfFilesInDir(std::string const &Dir, std::vector<std::string> const &Ext,
b39c1859 182 bool const &SortList);
fbb2c7e0 183std::vector<std::string> GetListOfFilesInDir(std::string const &Dir, bool SortList);
8f3ba4e8 184std::string SafeGetCWD();
3b5421b4
AL
185void SetCloseExec(int Fd,bool Close);
186void SetNonBlock(int Fd,bool Block);
1084d58a 187bool WaitFd(int Fd,bool write = false,unsigned long timeout = 0);
75ef8f14 188pid_t ExecFork();
e45c4617 189pid_t ExecFork(std::set<int> keep_fds);
96ae6de5 190void MergeKeepFdsFromConfiguration(std::set<int> &keep_fds);
3826564e 191bool ExecWait(pid_t Pid,const char *Name,bool Reap = false);
578bfd0a 192
8ce4327b 193// File string manipulators
8f3ba4e8
DK
194std::string flNotDir(std::string File);
195std::string flNotFile(std::string File);
196std::string flNoLink(std::string File);
197std::string flExtension(std::string File);
198std::string flCombine(std::string Dir,std::string File);
8ce4327b 199
488011fa
MV
200// simple c++ glob
201std::vector<std::string> Glob(std::string const &pattern, int flags=0);
202
578bfd0a 203#endif