]> git.saurik.com Git - apt.git/blob - apt-pkg/contrib/fileutl.cc
d5e7192e3d4a854ed3319848602d5218736434ad
[apt.git] / apt-pkg / contrib / fileutl.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description /*{{{*/
3 // $Id: fileutl.cc,v 1.42 2002/09/14 05:29:22 jgg Exp $
4 /* ######################################################################
5
6 File Utilities
7
8 CopyFile - Buffered copy of a single file
9 GetLock - dpkg compatible lock file manipulation (fcntl)
10
11 Most of this source is placed in the Public Domain, do with it what
12 you will
13 It was originally written by Jason Gunthorpe <jgg@debian.org>.
14 FileFd gzip support added by Martin Pitt <martin.pitt@canonical.com>
15
16 The exception is RunScripts() it is under the GPLv2
17
18 ##################################################################### */
19 /*}}}*/
20 // Include Files /*{{{*/
21 #include <config.h>
22
23 #include <apt-pkg/fileutl.h>
24 #include <apt-pkg/strutl.h>
25 #include <apt-pkg/error.h>
26 #include <apt-pkg/sptr.h>
27 #include <apt-pkg/configuration.h>
28
29 #include <cstdlib>
30 #include <cstring>
31 #include <cstdio>
32
33 #include <iostream>
34 #include <unistd.h>
35 #include <fcntl.h>
36 #include <sys/stat.h>
37 #include <sys/types.h>
38 #include <sys/time.h>
39 #include <sys/wait.h>
40 #include <dirent.h>
41 #include <signal.h>
42 #include <errno.h>
43 #include <set>
44 #include <algorithm>
45
46 #ifdef WORDS_BIGENDIAN
47 #include <inttypes.h>
48 #endif
49
50 #include <apti18n.h>
51 /*}}}*/
52
53 using namespace std;
54
55 // RunScripts - Run a set of scripts from a configuration subtree /*{{{*/
56 // ---------------------------------------------------------------------
57 /* */
58 bool RunScripts(const char *Cnf)
59 {
60 Configuration::Item const *Opts = _config->Tree(Cnf);
61 if (Opts == 0 || Opts->Child == 0)
62 return true;
63 Opts = Opts->Child;
64
65 // Fork for running the system calls
66 pid_t Child = ExecFork();
67
68 // This is the child
69 if (Child == 0)
70 {
71 if (_config->FindDir("DPkg::Chroot-Directory","/") != "/")
72 {
73 std::cerr << "Chrooting into "
74 << _config->FindDir("DPkg::Chroot-Directory")
75 << std::endl;
76 if (chroot(_config->FindDir("DPkg::Chroot-Directory","/").c_str()) != 0)
77 _exit(100);
78 }
79
80 if (chdir("/tmp/") != 0)
81 _exit(100);
82
83 unsigned int Count = 1;
84 for (; Opts != 0; Opts = Opts->Next, Count++)
85 {
86 if (Opts->Value.empty() == true)
87 continue;
88
89 if (system(Opts->Value.c_str()) != 0)
90 _exit(100+Count);
91 }
92 _exit(0);
93 }
94
95 // Wait for the child
96 int Status = 0;
97 while (waitpid(Child,&Status,0) != Child)
98 {
99 if (errno == EINTR)
100 continue;
101 return _error->Errno("waitpid","Couldn't wait for subprocess");
102 }
103
104 // Restore sig int/quit
105 signal(SIGQUIT,SIG_DFL);
106 signal(SIGINT,SIG_DFL);
107
108 // Check for an error code.
109 if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
110 {
111 unsigned int Count = WEXITSTATUS(Status);
112 if (Count > 100)
113 {
114 Count -= 100;
115 for (; Opts != 0 && Count != 1; Opts = Opts->Next, Count--);
116 _error->Error("Problem executing scripts %s '%s'",Cnf,Opts->Value.c_str());
117 }
118
119 return _error->Error("Sub-process returned an error code");
120 }
121
122 return true;
123 }
124 /*}}}*/
125
126 // CopyFile - Buffered copy of a file /*{{{*/
127 // ---------------------------------------------------------------------
128 /* The caller is expected to set things so that failure causes erasure */
129 bool CopyFile(FileFd &From,FileFd &To)
130 {
131 if (From.IsOpen() == false || To.IsOpen() == false)
132 return false;
133
134 // Buffered copy between fds
135 SPtrArray<unsigned char> Buf = new unsigned char[64000];
136 unsigned long long Size = From.Size();
137 while (Size != 0)
138 {
139 unsigned long long ToRead = Size;
140 if (Size > 64000)
141 ToRead = 64000;
142
143 if (From.Read(Buf,ToRead) == false ||
144 To.Write(Buf,ToRead) == false)
145 return false;
146
147 Size -= ToRead;
148 }
149
150 return true;
151 }
152 /*}}}*/
153 // GetLock - Gets a lock file /*{{{*/
154 // ---------------------------------------------------------------------
155 /* This will create an empty file of the given name and lock it. Once this
156 is done all other calls to GetLock in any other process will fail with
157 -1. The return result is the fd of the file, the call should call
158 close at some time. */
159 int GetLock(string File,bool Errors)
160 {
161 // GetLock() is used in aptitude on directories with public-write access
162 // Use O_NOFOLLOW here to prevent symlink traversal attacks
163 int FD = open(File.c_str(),O_RDWR | O_CREAT | O_NOFOLLOW,0640);
164 if (FD < 0)
165 {
166 // Read only .. cant have locking problems there.
167 if (errno == EROFS)
168 {
169 _error->Warning(_("Not using locking for read only lock file %s"),File.c_str());
170 return dup(0); // Need something for the caller to close
171 }
172
173 if (Errors == true)
174 _error->Errno("open",_("Could not open lock file %s"),File.c_str());
175
176 // Feh.. We do this to distinguish the lock vs open case..
177 errno = EPERM;
178 return -1;
179 }
180 SetCloseExec(FD,true);
181
182 // Aquire a write lock
183 struct flock fl;
184 fl.l_type = F_WRLCK;
185 fl.l_whence = SEEK_SET;
186 fl.l_start = 0;
187 fl.l_len = 0;
188 if (fcntl(FD,F_SETLK,&fl) == -1)
189 {
190 if (errno == ENOLCK)
191 {
192 _error->Warning(_("Not using locking for nfs mounted lock file %s"),File.c_str());
193 return dup(0); // Need something for the caller to close
194 }
195 if (Errors == true)
196 _error->Errno("open",_("Could not get lock %s"),File.c_str());
197
198 int Tmp = errno;
199 close(FD);
200 errno = Tmp;
201 return -1;
202 }
203
204 return FD;
205 }
206 /*}}}*/
207 // FileExists - Check if a file exists /*{{{*/
208 // ---------------------------------------------------------------------
209 /* Beware: Directories are also files! */
210 bool FileExists(string File)
211 {
212 struct stat Buf;
213 if (stat(File.c_str(),&Buf) != 0)
214 return false;
215 return true;
216 }
217 /*}}}*/
218 // RealFileExists - Check if a file exists and if it is really a file /*{{{*/
219 // ---------------------------------------------------------------------
220 /* */
221 bool RealFileExists(string File)
222 {
223 struct stat Buf;
224 if (stat(File.c_str(),&Buf) != 0)
225 return false;
226 return ((Buf.st_mode & S_IFREG) != 0);
227 }
228 /*}}}*/
229 // DirectoryExists - Check if a directory exists and is really one /*{{{*/
230 // ---------------------------------------------------------------------
231 /* */
232 bool DirectoryExists(string const &Path)
233 {
234 struct stat Buf;
235 if (stat(Path.c_str(),&Buf) != 0)
236 return false;
237 return ((Buf.st_mode & S_IFDIR) != 0);
238 }
239 /*}}}*/
240 // CreateDirectory - poor man's mkdir -p guarded by a parent directory /*{{{*/
241 // ---------------------------------------------------------------------
242 /* This method will create all directories needed for path in good old
243 mkdir -p style but refuses to do this if Parent is not a prefix of
244 this Path. Example: /var/cache/ and /var/cache/apt/archives are given,
245 so it will create apt/archives if /var/cache exists - on the other
246 hand if the parent is /var/lib the creation will fail as this path
247 is not a parent of the path to be generated. */
248 bool CreateDirectory(string const &Parent, string const &Path)
249 {
250 if (Parent.empty() == true || Path.empty() == true)
251 return false;
252
253 if (DirectoryExists(Path) == true)
254 return true;
255
256 if (DirectoryExists(Parent) == false)
257 return false;
258
259 // we are not going to create directories "into the blue"
260 if (Path.find(Parent, 0) != 0)
261 return false;
262
263 vector<string> const dirs = VectorizeString(Path.substr(Parent.size()), '/');
264 string progress = Parent;
265 for (vector<string>::const_iterator d = dirs.begin(); d != dirs.end(); ++d)
266 {
267 if (d->empty() == true)
268 continue;
269
270 progress.append("/").append(*d);
271 if (DirectoryExists(progress) == true)
272 continue;
273
274 if (mkdir(progress.c_str(), 0755) != 0)
275 return false;
276 }
277 return true;
278 }
279 /*}}}*/
280 // CreateAPTDirectoryIfNeeded - ensure that the given directory exists /*{{{*/
281 // ---------------------------------------------------------------------
282 /* a small wrapper around CreateDirectory to check if it exists and to
283 remove the trailing "/apt/" from the parent directory if needed */
284 bool CreateAPTDirectoryIfNeeded(string const &Parent, string const &Path)
285 {
286 if (DirectoryExists(Path) == true)
287 return true;
288
289 size_t const len = Parent.size();
290 if (len > 5 && Parent.find("/apt/", len - 6, 5) == len - 5)
291 {
292 if (CreateDirectory(Parent.substr(0,len-5), Path) == true)
293 return true;
294 }
295 else if (CreateDirectory(Parent, Path) == true)
296 return true;
297
298 return false;
299 }
300 /*}}}*/
301 // GetListOfFilesInDir - returns a vector of files in the given dir /*{{{*/
302 // ---------------------------------------------------------------------
303 /* If an extension is given only files with this extension are included
304 in the returned vector, otherwise every "normal" file is included. */
305 std::vector<string> GetListOfFilesInDir(string const &Dir, string const &Ext,
306 bool const &SortList, bool const &AllowNoExt)
307 {
308 std::vector<string> ext;
309 ext.reserve(2);
310 if (Ext.empty() == false)
311 ext.push_back(Ext);
312 if (AllowNoExt == true && ext.empty() == false)
313 ext.push_back("");
314 return GetListOfFilesInDir(Dir, ext, SortList);
315 }
316 std::vector<string> GetListOfFilesInDir(string const &Dir, std::vector<string> const &Ext,
317 bool const &SortList)
318 {
319 // Attention debuggers: need to be set with the environment config file!
320 bool const Debug = _config->FindB("Debug::GetListOfFilesInDir", false);
321 if (Debug == true)
322 {
323 std::clog << "Accept in " << Dir << " only files with the following " << Ext.size() << " extensions:" << std::endl;
324 if (Ext.empty() == true)
325 std::clog << "\tNO extension" << std::endl;
326 else
327 for (std::vector<string>::const_iterator e = Ext.begin();
328 e != Ext.end(); ++e)
329 std::clog << '\t' << (e->empty() == true ? "NO" : *e) << " extension" << std::endl;
330 }
331
332 std::vector<string> List;
333
334 if (DirectoryExists(Dir.c_str()) == false)
335 {
336 _error->Error(_("List of files can't be created as '%s' is not a directory"), Dir.c_str());
337 return List;
338 }
339
340 Configuration::MatchAgainstConfig SilentIgnore("Dir::Ignore-Files-Silently");
341 DIR *D = opendir(Dir.c_str());
342 if (D == 0)
343 {
344 _error->Errno("opendir",_("Unable to read %s"),Dir.c_str());
345 return List;
346 }
347
348 for (struct dirent *Ent = readdir(D); Ent != 0; Ent = readdir(D))
349 {
350 // skip "hidden" files
351 if (Ent->d_name[0] == '.')
352 continue;
353
354 // Make sure it is a file and not something else
355 string const File = flCombine(Dir,Ent->d_name);
356 #ifdef _DIRENT_HAVE_D_TYPE
357 if (Ent->d_type != DT_REG)
358 #endif
359 {
360 if (RealFileExists(File.c_str()) == false)
361 {
362 if (SilentIgnore.Match(Ent->d_name) == false)
363 _error->Notice(_("Ignoring '%s' in directory '%s' as it is not a regular file"), Ent->d_name, Dir.c_str());
364 continue;
365 }
366 }
367
368 // check for accepted extension:
369 // no extension given -> periods are bad as hell!
370 // extensions given -> "" extension allows no extension
371 if (Ext.empty() == false)
372 {
373 string d_ext = flExtension(Ent->d_name);
374 if (d_ext == Ent->d_name) // no extension
375 {
376 if (std::find(Ext.begin(), Ext.end(), "") == Ext.end())
377 {
378 if (Debug == true)
379 std::clog << "Bad file: " << Ent->d_name << " → no extension" << std::endl;
380 if (SilentIgnore.Match(Ent->d_name) == false)
381 _error->Notice(_("Ignoring file '%s' in directory '%s' as it has no filename extension"), Ent->d_name, Dir.c_str());
382 continue;
383 }
384 }
385 else if (std::find(Ext.begin(), Ext.end(), d_ext) == Ext.end())
386 {
387 if (Debug == true)
388 std::clog << "Bad file: " << Ent->d_name << " → bad extension »" << flExtension(Ent->d_name) << "«" << std::endl;
389 if (SilentIgnore.Match(Ent->d_name) == false)
390 _error->Notice(_("Ignoring file '%s' in directory '%s' as it has an invalid filename extension"), Ent->d_name, Dir.c_str());
391 continue;
392 }
393 }
394
395 // Skip bad filenames ala run-parts
396 const char *C = Ent->d_name;
397 for (; *C != 0; ++C)
398 if (isalpha(*C) == 0 && isdigit(*C) == 0
399 && *C != '_' && *C != '-') {
400 // no required extension -> dot is a bad character
401 if (*C == '.' && Ext.empty() == false)
402 continue;
403 break;
404 }
405
406 // we don't reach the end of the name -> bad character included
407 if (*C != 0)
408 {
409 if (Debug == true)
410 std::clog << "Bad file: " << Ent->d_name << " → bad character »"
411 << *C << "« in filename (period allowed: " << (Ext.empty() ? "no" : "yes") << ")" << std::endl;
412 continue;
413 }
414
415 // skip filenames which end with a period. These are never valid
416 if (*(C - 1) == '.')
417 {
418 if (Debug == true)
419 std::clog << "Bad file: " << Ent->d_name << " → Period as last character" << std::endl;
420 continue;
421 }
422
423 if (Debug == true)
424 std::clog << "Accept file: " << Ent->d_name << " in " << Dir << std::endl;
425 List.push_back(File);
426 }
427 closedir(D);
428
429 if (SortList == true)
430 std::sort(List.begin(),List.end());
431 return List;
432 }
433 /*}}}*/
434 // SafeGetCWD - This is a safer getcwd that returns a dynamic string /*{{{*/
435 // ---------------------------------------------------------------------
436 /* We return / on failure. */
437 string SafeGetCWD()
438 {
439 // Stash the current dir.
440 char S[300];
441 S[0] = 0;
442 if (getcwd(S,sizeof(S)-2) == 0)
443 return "/";
444 unsigned int Len = strlen(S);
445 S[Len] = '/';
446 S[Len+1] = 0;
447 return S;
448 }
449 /*}}}*/
450 // GetModificationTime - Get the mtime of the given file or -1 on error /*{{{*/
451 // ---------------------------------------------------------------------
452 /* We return / on failure. */
453 time_t GetModificationTime(string const &Path)
454 {
455 struct stat St;
456 if (stat(Path.c_str(), &St) < 0)
457 return -1;
458 return St.st_mtime;
459 }
460 /*}}}*/
461 // flNotDir - Strip the directory from the filename /*{{{*/
462 // ---------------------------------------------------------------------
463 /* */
464 string flNotDir(string File)
465 {
466 string::size_type Res = File.rfind('/');
467 if (Res == string::npos)
468 return File;
469 Res++;
470 return string(File,Res,Res - File.length());
471 }
472 /*}}}*/
473 // flNotFile - Strip the file from the directory name /*{{{*/
474 // ---------------------------------------------------------------------
475 /* Result ends in a / */
476 string flNotFile(string File)
477 {
478 string::size_type Res = File.rfind('/');
479 if (Res == string::npos)
480 return "./";
481 Res++;
482 return string(File,0,Res);
483 }
484 /*}}}*/
485 // flExtension - Return the extension for the file /*{{{*/
486 // ---------------------------------------------------------------------
487 /* */
488 string flExtension(string File)
489 {
490 string::size_type Res = File.rfind('.');
491 if (Res == string::npos)
492 return File;
493 Res++;
494 return string(File,Res,Res - File.length());
495 }
496 /*}}}*/
497 // flNoLink - If file is a symlink then deref it /*{{{*/
498 // ---------------------------------------------------------------------
499 /* If the name is not a link then the returned path is the input. */
500 string flNoLink(string File)
501 {
502 struct stat St;
503 if (lstat(File.c_str(),&St) != 0 || S_ISLNK(St.st_mode) == 0)
504 return File;
505 if (stat(File.c_str(),&St) != 0)
506 return File;
507
508 /* Loop resolving the link. There is no need to limit the number of
509 loops because the stat call above ensures that the symlink is not
510 circular */
511 char Buffer[1024];
512 string NFile = File;
513 while (1)
514 {
515 // Read the link
516 int Res;
517 if ((Res = readlink(NFile.c_str(),Buffer,sizeof(Buffer))) <= 0 ||
518 (unsigned)Res >= sizeof(Buffer))
519 return File;
520
521 // Append or replace the previous path
522 Buffer[Res] = 0;
523 if (Buffer[0] == '/')
524 NFile = Buffer;
525 else
526 NFile = flNotFile(NFile) + Buffer;
527
528 // See if we are done
529 if (lstat(NFile.c_str(),&St) != 0)
530 return File;
531 if (S_ISLNK(St.st_mode) == 0)
532 return NFile;
533 }
534 }
535 /*}}}*/
536 // flCombine - Combine a file and a directory /*{{{*/
537 // ---------------------------------------------------------------------
538 /* If the file is an absolute path then it is just returned, otherwise
539 the directory is pre-pended to it. */
540 string flCombine(string Dir,string File)
541 {
542 if (File.empty() == true)
543 return string();
544
545 if (File[0] == '/' || Dir.empty() == true)
546 return File;
547 if (File.length() >= 2 && File[0] == '.' && File[1] == '/')
548 return File;
549 if (Dir[Dir.length()-1] == '/')
550 return Dir + File;
551 return Dir + '/' + File;
552 }
553 /*}}}*/
554 // SetCloseExec - Set the close on exec flag /*{{{*/
555 // ---------------------------------------------------------------------
556 /* */
557 void SetCloseExec(int Fd,bool Close)
558 {
559 if (fcntl(Fd,F_SETFD,(Close == false)?0:FD_CLOEXEC) != 0)
560 {
561 cerr << "FATAL -> Could not set close on exec " << strerror(errno) << endl;
562 exit(100);
563 }
564 }
565 /*}}}*/
566 // SetNonBlock - Set the nonblocking flag /*{{{*/
567 // ---------------------------------------------------------------------
568 /* */
569 void SetNonBlock(int Fd,bool Block)
570 {
571 int Flags = fcntl(Fd,F_GETFL) & (~O_NONBLOCK);
572 if (fcntl(Fd,F_SETFL,Flags | ((Block == false)?0:O_NONBLOCK)) != 0)
573 {
574 cerr << "FATAL -> Could not set non-blocking flag " << strerror(errno) << endl;
575 exit(100);
576 }
577 }
578 /*}}}*/
579 // WaitFd - Wait for a FD to become readable /*{{{*/
580 // ---------------------------------------------------------------------
581 /* This waits for a FD to become readable using select. It is useful for
582 applications making use of non-blocking sockets. The timeout is
583 in seconds. */
584 bool WaitFd(int Fd,bool write,unsigned long timeout)
585 {
586 fd_set Set;
587 struct timeval tv;
588 FD_ZERO(&Set);
589 FD_SET(Fd,&Set);
590 tv.tv_sec = timeout;
591 tv.tv_usec = 0;
592 if (write == true)
593 {
594 int Res;
595 do
596 {
597 Res = select(Fd+1,0,&Set,0,(timeout != 0?&tv:0));
598 }
599 while (Res < 0 && errno == EINTR);
600
601 if (Res <= 0)
602 return false;
603 }
604 else
605 {
606 int Res;
607 do
608 {
609 Res = select(Fd+1,&Set,0,0,(timeout != 0?&tv:0));
610 }
611 while (Res < 0 && errno == EINTR);
612
613 if (Res <= 0)
614 return false;
615 }
616
617 return true;
618 }
619 /*}}}*/
620 // ExecFork - Magical fork that sanitizes the context before execing /*{{{*/
621 // ---------------------------------------------------------------------
622 /* This is used if you want to cleanse the environment for the forked
623 child, it fixes up the important signals and nukes all of the fds,
624 otherwise acts like normal fork. */
625 pid_t ExecFork()
626 {
627 // Fork off the process
628 pid_t Process = fork();
629 if (Process < 0)
630 {
631 cerr << "FATAL -> Failed to fork." << endl;
632 exit(100);
633 }
634
635 // Spawn the subprocess
636 if (Process == 0)
637 {
638 // Setup the signals
639 signal(SIGPIPE,SIG_DFL);
640 signal(SIGQUIT,SIG_DFL);
641 signal(SIGINT,SIG_DFL);
642 signal(SIGWINCH,SIG_DFL);
643 signal(SIGCONT,SIG_DFL);
644 signal(SIGTSTP,SIG_DFL);
645
646 set<int> KeepFDs;
647 Configuration::Item const *Opts = _config->Tree("APT::Keep-Fds");
648 if (Opts != 0 && Opts->Child != 0)
649 {
650 Opts = Opts->Child;
651 for (; Opts != 0; Opts = Opts->Next)
652 {
653 if (Opts->Value.empty() == true)
654 continue;
655 int fd = atoi(Opts->Value.c_str());
656 KeepFDs.insert(fd);
657 }
658 }
659
660 // Close all of our FDs - just in case
661 for (int K = 3; K != 40; K++)
662 {
663 if(KeepFDs.find(K) == KeepFDs.end())
664 fcntl(K,F_SETFD,FD_CLOEXEC);
665 }
666 }
667
668 return Process;
669 }
670 /*}}}*/
671 // ExecWait - Fancy waitpid /*{{{*/
672 // ---------------------------------------------------------------------
673 /* Waits for the given sub process. If Reap is set then no errors are
674 generated. Otherwise a failed subprocess will generate a proper descriptive
675 message */
676 bool ExecWait(pid_t Pid,const char *Name,bool Reap)
677 {
678 if (Pid <= 1)
679 return true;
680
681 // Wait and collect the error code
682 int Status;
683 while (waitpid(Pid,&Status,0) != Pid)
684 {
685 if (errno == EINTR)
686 continue;
687
688 if (Reap == true)
689 return false;
690
691 return _error->Error(_("Waited for %s but it wasn't there"),Name);
692 }
693
694
695 // Check for an error code.
696 if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
697 {
698 if (Reap == true)
699 return false;
700 if (WIFSIGNALED(Status) != 0)
701 {
702 if( WTERMSIG(Status) == SIGSEGV)
703 return _error->Error(_("Sub-process %s received a segmentation fault."),Name);
704 else
705 return _error->Error(_("Sub-process %s received signal %u."),Name, WTERMSIG(Status));
706 }
707
708 if (WIFEXITED(Status) != 0)
709 return _error->Error(_("Sub-process %s returned an error code (%u)"),Name,WEXITSTATUS(Status));
710
711 return _error->Error(_("Sub-process %s exited unexpectedly"),Name);
712 }
713
714 return true;
715 }
716 /*}}}*/
717
718 // FileFd::Open - Open a file /*{{{*/
719 // ---------------------------------------------------------------------
720 /* The most commonly used open mode combinations are given with Mode */
721 bool FileFd::Open(string FileName,OpenMode Mode,CompressMode Compress, unsigned long Perms)
722 {
723 if (Mode == ReadOnlyGzip)
724 return Open(FileName, ReadOnly, Gzip, Perms);
725 Close();
726 Flags = AutoClose;
727
728 int fileflags = 0;
729 #define if_FLAGGED_SET(FLAG, MODE) if ((Mode & FLAG) == FLAG) fileflags |= MODE
730 if_FLAGGED_SET(ReadWrite, O_RDWR);
731 else if_FLAGGED_SET(ReadOnly, O_RDONLY);
732 else if_FLAGGED_SET(WriteOnly, O_WRONLY);
733 else return _error->Error("No openmode provided in FileFd::Open for %s", FileName.c_str());
734
735 if_FLAGGED_SET(Create, O_CREAT);
736 if_FLAGGED_SET(Exclusive, O_EXCL);
737 else if_FLAGGED_SET(Atomic, O_EXCL);
738 if_FLAGGED_SET(Empty, O_TRUNC);
739 #undef if_FLAGGED_SET
740
741 if ((Mode & Atomic) == Atomic)
742 {
743 Flags |= Replace;
744 char *name = strdup((FileName + ".XXXXXX").c_str());
745 TemporaryFileName = string(mktemp(name));
746 free(name);
747 }
748 else if ((Mode & (Exclusive | Create)) == (Exclusive | Create))
749 {
750 // for atomic, this will be done by rename in Close()
751 unlink(FileName.c_str());
752 }
753 if ((Mode & Empty) == Empty)
754 {
755 struct stat Buf;
756 if (lstat(FileName.c_str(),&Buf) == 0 && S_ISLNK(Buf.st_mode))
757 unlink(FileName.c_str());
758 }
759
760 if (TemporaryFileName.empty() == false)
761 iFd = open(TemporaryFileName.c_str(), fileflags, Perms);
762 else
763 iFd = open(FileName.c_str(), fileflags, Perms);
764
765 if (iFd != -1 && Compress == Gzip)
766 {
767 gz = gzdopen (iFd, "r");
768 if (gz == NULL)
769 {
770 close (iFd);
771 iFd = -1;
772 }
773 }
774
775 if (iFd == -1)
776 return _error->Errno("open",_("Could not open file %s"),FileName.c_str());
777
778 this->FileName = FileName;
779 SetCloseExec(iFd,true);
780 return true;
781 }
782 /*}}}*/
783 // FileFd::OpenDescriptor - Open a filedescriptor /*{{{*/
784 // ---------------------------------------------------------------------
785 /* */
786 bool FileFd::OpenDescriptor(int Fd, OpenMode Mode, CompressMode Compress, bool AutoClose)
787 {
788 Close();
789 Flags = (AutoClose) ? FileFd::AutoClose : 0;
790 iFd = Fd;
791 if (Mode == ReadOnlyGzip) {
792 gz = gzdopen (iFd, "r");
793 if (gz == NULL) {
794 if (AutoClose)
795 close (iFd);
796 return _error->Errno("gzdopen",_("Could not open file descriptor %d"),
797 Fd);
798 }
799 }
800 this->FileName = "";
801 return true;
802 }
803 /*}}}*/
804 // FileFd::~File - Closes the file /*{{{*/
805 // ---------------------------------------------------------------------
806 /* If the proper modes are selected then we close the Fd and possibly
807 unlink the file on error. */
808 FileFd::~FileFd()
809 {
810 Close();
811 }
812 /*}}}*/
813 // FileFd::Read - Read a bit of the file /*{{{*/
814 // ---------------------------------------------------------------------
815 /* We are carefull to handle interruption by a signal while reading
816 gracefully. */
817 bool FileFd::Read(void *To,unsigned long long Size,unsigned long long *Actual)
818 {
819 int Res;
820 errno = 0;
821 if (Actual != 0)
822 *Actual = 0;
823
824 do
825 {
826 if (gz != NULL)
827 Res = gzread(gz,To,Size);
828 else
829 Res = read(iFd,To,Size);
830 if (Res < 0 && errno == EINTR)
831 continue;
832 if (Res < 0)
833 {
834 Flags |= Fail;
835 return _error->Errno("read",_("Read error"));
836 }
837
838 To = (char *)To + Res;
839 Size -= Res;
840 if (Actual != 0)
841 *Actual += Res;
842 }
843 while (Res > 0 && Size > 0);
844
845 if (Size == 0)
846 return true;
847
848 // Eof handling
849 if (Actual != 0)
850 {
851 Flags |= HitEof;
852 return true;
853 }
854
855 Flags |= Fail;
856 return _error->Error(_("read, still have %llu to read but none left"), Size);
857 }
858 /*}}}*/
859 // FileFd::Write - Write to the file /*{{{*/
860 // ---------------------------------------------------------------------
861 /* */
862 bool FileFd::Write(const void *From,unsigned long long Size)
863 {
864 int Res;
865 errno = 0;
866 do
867 {
868 if (gz != NULL)
869 Res = gzwrite(gz,From,Size);
870 else
871 Res = write(iFd,From,Size);
872 if (Res < 0 && errno == EINTR)
873 continue;
874 if (Res < 0)
875 {
876 Flags |= Fail;
877 return _error->Errno("write",_("Write error"));
878 }
879
880 From = (char *)From + Res;
881 Size -= Res;
882 }
883 while (Res > 0 && Size > 0);
884
885 if (Size == 0)
886 return true;
887
888 Flags |= Fail;
889 return _error->Error(_("write, still have %llu to write but couldn't"), Size);
890 }
891 /*}}}*/
892 // FileFd::Seek - Seek in the file /*{{{*/
893 // ---------------------------------------------------------------------
894 /* */
895 bool FileFd::Seek(unsigned long long To)
896 {
897 int res;
898 if (gz)
899 res = gzseek(gz,To,SEEK_SET);
900 else
901 res = lseek(iFd,To,SEEK_SET);
902 if (res != (signed)To)
903 {
904 Flags |= Fail;
905 return _error->Error("Unable to seek to %llu", To);
906 }
907
908 return true;
909 }
910 /*}}}*/
911 // FileFd::Skip - Seek in the file /*{{{*/
912 // ---------------------------------------------------------------------
913 /* */
914 bool FileFd::Skip(unsigned long long Over)
915 {
916 int res;
917 if (gz)
918 res = gzseek(gz,Over,SEEK_CUR);
919 else
920 res = lseek(iFd,Over,SEEK_CUR);
921 if (res < 0)
922 {
923 Flags |= Fail;
924 return _error->Error("Unable to seek ahead %llu",Over);
925 }
926
927 return true;
928 }
929 /*}}}*/
930 // FileFd::Truncate - Truncate the file /*{{{*/
931 // ---------------------------------------------------------------------
932 /* */
933 bool FileFd::Truncate(unsigned long long To)
934 {
935 if (gz)
936 {
937 Flags |= Fail;
938 return _error->Error("Truncating gzipped files is not implemented (%s)", FileName.c_str());
939 }
940 if (ftruncate(iFd,To) != 0)
941 {
942 Flags |= Fail;
943 return _error->Error("Unable to truncate to %llu",To);
944 }
945
946 return true;
947 }
948 /*}}}*/
949 // FileFd::Tell - Current seek position /*{{{*/
950 // ---------------------------------------------------------------------
951 /* */
952 unsigned long long FileFd::Tell()
953 {
954 off_t Res;
955 if (gz)
956 Res = gztell(gz);
957 else
958 Res = lseek(iFd,0,SEEK_CUR);
959 if (Res == (off_t)-1)
960 _error->Errno("lseek","Failed to determine the current file position");
961 return Res;
962 }
963 /*}}}*/
964 // FileFd::FileSize - Return the size of the file /*{{{*/
965 // ---------------------------------------------------------------------
966 /* */
967 unsigned long long FileFd::FileSize()
968 {
969 struct stat Buf;
970
971 if (fstat(iFd,&Buf) != 0)
972 return _error->Errno("fstat","Unable to determine the file size");
973 return Buf.st_size;
974 }
975 /*}}}*/
976 // FileFd::Size - Return the size of the content in the file /*{{{*/
977 // ---------------------------------------------------------------------
978 /* */
979 unsigned long long FileFd::Size()
980 {
981 unsigned long long size = FileSize();
982
983 // only check gzsize if we are actually a gzip file, just checking for
984 // "gz" is not sufficient as uncompressed files will be opened with
985 // gzopen in "direct" mode as well
986 if (gz && !gzdirect(gz) && size > 0)
987 {
988 /* unfortunately zlib.h doesn't provide a gzsize(), so we have to do
989 * this ourselves; the original (uncompressed) file size is the last 32
990 * bits of the file */
991 // FIXME: Size for gz-files is limited by 32bit… no largefile support
992 off_t orig_pos = lseek(iFd, 0, SEEK_CUR);
993 if (lseek(iFd, -4, SEEK_END) < 0)
994 return _error->Errno("lseek","Unable to seek to end of gzipped file");
995 size = 0L;
996 if (read(iFd, &size, 4) != 4)
997 return _error->Errno("read","Unable to read original size of gzipped file");
998
999 #ifdef WORDS_BIGENDIAN
1000 uint32_t tmp_size = size;
1001 uint8_t const * const p = (uint8_t const * const) &tmp_size;
1002 tmp_size = (p[3] << 24) | (p[2] << 16) | (p[1] << 8) | p[0];
1003 size = tmp_size;
1004 #endif
1005
1006 if (lseek(iFd, orig_pos, SEEK_SET) < 0)
1007 return _error->Errno("lseek","Unable to seek in gzipped file");
1008 return size;
1009 }
1010
1011 return size;
1012 }
1013 /*}}}*/
1014 // FileFd::Close - Close the file if the close flag is set /*{{{*/
1015 // ---------------------------------------------------------------------
1016 /* */
1017 bool FileFd::Close()
1018 {
1019 bool Res = true;
1020 if ((Flags & AutoClose) == AutoClose)
1021 {
1022 if (gz != NULL) {
1023 int const e = gzclose(gz);
1024 // gzdopen() on empty files always fails with "buffer error" here, ignore that
1025 if (e != 0 && e != Z_BUF_ERROR)
1026 Res &= _error->Errno("close",_("Problem closing the gzip file %s"), FileName.c_str());
1027 } else
1028 if (iFd > 0 && close(iFd) != 0)
1029 Res &= _error->Errno("close",_("Problem closing the file %s"), FileName.c_str());
1030 }
1031
1032 if ((Flags & Replace) == Replace && iFd >= 0) {
1033 if (rename(TemporaryFileName.c_str(), FileName.c_str()) != 0)
1034 Res &= _error->Errno("rename",_("Problem renaming the file %s to %s"), TemporaryFileName.c_str(), FileName.c_str());
1035
1036 FileName = TemporaryFileName; // for the unlink() below.
1037 TemporaryFileName.clear();
1038 }
1039
1040 iFd = -1;
1041 gz = NULL;
1042
1043 if ((Flags & Fail) == Fail && (Flags & DelOnFail) == DelOnFail &&
1044 FileName.empty() == false)
1045 if (unlink(FileName.c_str()) != 0)
1046 Res &= _error->WarningE("unlnk",_("Problem unlinking the file %s"), FileName.c_str());
1047
1048
1049 return Res;
1050 }
1051 /*}}}*/
1052 // FileFd::Sync - Sync the file /*{{{*/
1053 // ---------------------------------------------------------------------
1054 /* */
1055 bool FileFd::Sync()
1056 {
1057 #ifdef _POSIX_SYNCHRONIZED_IO
1058 if (fsync(iFd) != 0)
1059 return _error->Errno("sync",_("Problem syncing the file"));
1060 #endif
1061 return true;
1062 }
1063 /*}}}*/