]> git.saurik.com Git - apt.git/blame - apt-pkg/contrib/fileutl.cc
releasing package apt version 0.9.13~exp1
[apt.git] / apt-pkg / contrib / fileutl.cc
CommitLineData
578bfd0a
AL
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
7da2b375 3// $Id: fileutl.cc,v 1.42 2002/09/14 05:29:22 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
614adaa0
MV
11 Most of this source is placed in the Public Domain, do with it what
12 you will
7da2b375 13 It was originally written by Jason Gunthorpe <jgg@debian.org>.
a3a03f5d 14 FileFd gzip support added by Martin Pitt <martin.pitt@canonical.com>
578bfd0a 15
614adaa0
MV
16 The exception is RunScripts() it is under the GPLv2
17
578bfd0a
AL
18 ##################################################################### */
19 /*}}}*/
20// Include Files /*{{{*/
ea542140
DK
21#include <config.h>
22
094a497d 23#include <apt-pkg/fileutl.h>
1cd1c398 24#include <apt-pkg/strutl.h>
094a497d 25#include <apt-pkg/error.h>
b2e465d6 26#include <apt-pkg/sptr.h>
468720c5 27#include <apt-pkg/aptconfiguration.h>
75ef8f14 28#include <apt-pkg/configuration.h>
b2e465d6 29
152ab79e 30#include <cstdlib>
4f333a8b 31#include <cstring>
3010fb0e 32#include <cstdio>
4f333a8b 33
4d055c05 34#include <iostream>
578bfd0a 35#include <unistd.h>
2c206aa4 36#include <fcntl.h>
578bfd0a 37#include <sys/stat.h>
578bfd0a 38#include <sys/types.h>
cc2313b7 39#include <sys/time.h>
1ae93c94 40#include <sys/wait.h>
46e39c8e 41#include <dirent.h>
54676e1a 42#include <signal.h>
65a1e968 43#include <errno.h>
488011fa
MV
44#include <glob.h>
45
75ef8f14 46#include <set>
46e39c8e 47#include <algorithm>
2cae0ccb 48
7efb8c8e
DK
49#ifdef HAVE_ZLIB
50 #include <zlib.h>
699b209e 51#endif
c4997486
DK
52#ifdef HAVE_BZ2
53 #include <bzlib.h>
54#endif
032bd56f 55
2a79d5b5 56#ifdef WORDS_BIGENDIAN
2cae0ccb
DK
57#include <inttypes.h>
58#endif
ea542140
DK
59
60#include <apti18n.h>
578bfd0a
AL
61 /*}}}*/
62
4d055c05
AL
63using namespace std;
64
032bd56f
DK
65class FileFdPrivate {
66 public:
7efb8c8e 67#ifdef HAVE_ZLIB
032bd56f 68 gzFile gz;
699b209e
DK
69#else
70 void* gz;
c4997486
DK
71#endif
72#ifdef HAVE_BZ2
73 BZFILE* bz2;
74#else
75 void* bz2;
699b209e 76#endif
561f860a 77 int compressed_fd;
699b209e
DK
78 pid_t compressor_pid;
79 bool pipe;
80 APT::Configuration::Compressor compressor;
52b47296 81 unsigned int openmode;
1abbc47c 82 unsigned long long seekpos;
c4997486
DK
83 FileFdPrivate() : gz(NULL), bz2(NULL),
84 compressed_fd(-1), compressor_pid(-1), pipe(false),
1abbc47c 85 openmode(0), seekpos(0) {};
500400fe
DK
86 bool CloseDown(std::string const &FileName)
87 {
88 bool Res = true;
89#ifdef HAVE_ZLIB
90 if (gz != NULL) {
91 int const e = gzclose(gz);
92 gz = NULL;
93 // gzdclose() on empty files always fails with "buffer error" here, ignore that
94 if (e != 0 && e != Z_BUF_ERROR)
95 Res &= _error->Errno("close",_("Problem closing the gzip file %s"), FileName.c_str());
96 }
97#endif
98#ifdef HAVE_BZ2
99 if (bz2 != NULL) {
100 BZ2_bzclose(bz2);
101 bz2 = NULL;
102 }
103#endif
104 if (compressor_pid > 0)
105 ExecWait(compressor_pid, "FileFdCompressor", true);
106 compressor_pid = -1;
107
108 return Res;
109 }
110 ~FileFdPrivate() { CloseDown(""); }
032bd56f
DK
111};
112
614adaa0
MV
113// RunScripts - Run a set of scripts from a configuration subtree /*{{{*/
114// ---------------------------------------------------------------------
115/* */
116bool RunScripts(const char *Cnf)
117{
118 Configuration::Item const *Opts = _config->Tree(Cnf);
119 if (Opts == 0 || Opts->Child == 0)
120 return true;
121 Opts = Opts->Child;
122
123 // Fork for running the system calls
124 pid_t Child = ExecFork();
125
126 // This is the child
127 if (Child == 0)
128 {
cfba4f69
MV
129 if (_config->FindDir("DPkg::Chroot-Directory","/") != "/")
130 {
131 std::cerr << "Chrooting into "
132 << _config->FindDir("DPkg::Chroot-Directory")
133 << std::endl;
134 if (chroot(_config->FindDir("DPkg::Chroot-Directory","/").c_str()) != 0)
135 _exit(100);
136 }
137
614adaa0
MV
138 if (chdir("/tmp/") != 0)
139 _exit(100);
140
141 unsigned int Count = 1;
142 for (; Opts != 0; Opts = Opts->Next, Count++)
143 {
144 if (Opts->Value.empty() == true)
145 continue;
146
147 if (system(Opts->Value.c_str()) != 0)
148 _exit(100+Count);
149 }
150 _exit(0);
151 }
152
153 // Wait for the child
154 int Status = 0;
155 while (waitpid(Child,&Status,0) != Child)
156 {
157 if (errno == EINTR)
158 continue;
159 return _error->Errno("waitpid","Couldn't wait for subprocess");
160 }
161
162 // Restore sig int/quit
163 signal(SIGQUIT,SIG_DFL);
164 signal(SIGINT,SIG_DFL);
165
166 // Check for an error code.
167 if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
168 {
169 unsigned int Count = WEXITSTATUS(Status);
170 if (Count > 100)
171 {
172 Count -= 100;
173 for (; Opts != 0 && Count != 1; Opts = Opts->Next, Count--);
174 _error->Error("Problem executing scripts %s '%s'",Cnf,Opts->Value.c_str());
175 }
176
177 return _error->Error("Sub-process returned an error code");
178 }
179
180 return true;
181}
182 /*}}}*/
183
578bfd0a
AL
184// CopyFile - Buffered copy of a file /*{{{*/
185// ---------------------------------------------------------------------
186/* The caller is expected to set things so that failure causes erasure */
8b89e57f 187bool CopyFile(FileFd &From,FileFd &To)
578bfd0a 188{
2128d3fc
DK
189 if (From.IsOpen() == false || To.IsOpen() == false ||
190 From.Failed() == true || To.Failed() == true)
578bfd0a
AL
191 return false;
192
193 // Buffered copy between fds
b2e465d6 194 SPtrArray<unsigned char> Buf = new unsigned char[64000];
650faab0 195 unsigned long long Size = From.Size();
b0db36b1 196 while (Size != 0)
578bfd0a 197 {
650faab0 198 unsigned long long ToRead = Size;
b0db36b1
AL
199 if (Size > 64000)
200 ToRead = 64000;
201
4a6d5862 202 if (From.Read(Buf,ToRead) == false ||
b0db36b1 203 To.Write(Buf,ToRead) == false)
578bfd0a 204 return false;
b0db36b1
AL
205
206 Size -= ToRead;
578bfd0a
AL
207 }
208
578bfd0a
AL
209 return true;
210}
211 /*}}}*/
212// GetLock - Gets a lock file /*{{{*/
213// ---------------------------------------------------------------------
214/* This will create an empty file of the given name and lock it. Once this
215 is done all other calls to GetLock in any other process will fail with
216 -1. The return result is the fd of the file, the call should call
217 close at some time. */
218int GetLock(string File,bool Errors)
219{
f659b39a
OS
220 // GetLock() is used in aptitude on directories with public-write access
221 // Use O_NOFOLLOW here to prevent symlink traversal attacks
222 int FD = open(File.c_str(),O_RDWR | O_CREAT | O_NOFOLLOW,0640);
578bfd0a
AL
223 if (FD < 0)
224 {
b2e465d6
AL
225 // Read only .. cant have locking problems there.
226 if (errno == EROFS)
227 {
228 _error->Warning(_("Not using locking for read only lock file %s"),File.c_str());
229 return dup(0); // Need something for the caller to close
230 }
231
578bfd0a 232 if (Errors == true)
b2e465d6
AL
233 _error->Errno("open",_("Could not open lock file %s"),File.c_str());
234
235 // Feh.. We do this to distinguish the lock vs open case..
236 errno = EPERM;
578bfd0a
AL
237 return -1;
238 }
b2e465d6
AL
239 SetCloseExec(FD,true);
240
578bfd0a
AL
241 // Aquire a write lock
242 struct flock fl;
c71bc556
AL
243 fl.l_type = F_WRLCK;
244 fl.l_whence = SEEK_SET;
245 fl.l_start = 0;
246 fl.l_len = 0;
578bfd0a
AL
247 if (fcntl(FD,F_SETLK,&fl) == -1)
248 {
3d165906
MV
249 // always close to not leak resources
250 int Tmp = errno;
251 close(FD);
252 errno = Tmp;
253
d89df07a
AL
254 if (errno == ENOLCK)
255 {
b2e465d6
AL
256 _error->Warning(_("Not using locking for nfs mounted lock file %s"),File.c_str());
257 return dup(0); // Need something for the caller to close
3d165906
MV
258 }
259
578bfd0a 260 if (Errors == true)
b2e465d6
AL
261 _error->Errno("open",_("Could not get lock %s"),File.c_str());
262
578bfd0a
AL
263 return -1;
264 }
265
266 return FD;
267}
268 /*}}}*/
269// FileExists - Check if a file exists /*{{{*/
270// ---------------------------------------------------------------------
36f1098a 271/* Beware: Directories are also files! */
578bfd0a
AL
272bool FileExists(string File)
273{
274 struct stat Buf;
275 if (stat(File.c_str(),&Buf) != 0)
276 return false;
277 return true;
278}
279 /*}}}*/
36f1098a
DK
280// RealFileExists - Check if a file exists and if it is really a file /*{{{*/
281// ---------------------------------------------------------------------
282/* */
283bool RealFileExists(string File)
284{
285 struct stat Buf;
286 if (stat(File.c_str(),&Buf) != 0)
287 return false;
288 return ((Buf.st_mode & S_IFREG) != 0);
289}
290 /*}}}*/
1cd1c398
DK
291// DirectoryExists - Check if a directory exists and is really one /*{{{*/
292// ---------------------------------------------------------------------
293/* */
294bool DirectoryExists(string const &Path)
295{
296 struct stat Buf;
297 if (stat(Path.c_str(),&Buf) != 0)
298 return false;
299 return ((Buf.st_mode & S_IFDIR) != 0);
300}
301 /*}}}*/
302// CreateDirectory - poor man's mkdir -p guarded by a parent directory /*{{{*/
303// ---------------------------------------------------------------------
304/* This method will create all directories needed for path in good old
305 mkdir -p style but refuses to do this if Parent is not a prefix of
306 this Path. Example: /var/cache/ and /var/cache/apt/archives are given,
307 so it will create apt/archives if /var/cache exists - on the other
308 hand if the parent is /var/lib the creation will fail as this path
309 is not a parent of the path to be generated. */
310bool CreateDirectory(string const &Parent, string const &Path)
311{
312 if (Parent.empty() == true || Path.empty() == true)
313 return false;
314
315 if (DirectoryExists(Path) == true)
316 return true;
317
318 if (DirectoryExists(Parent) == false)
319 return false;
320
321 // we are not going to create directories "into the blue"
322 if (Path.find(Parent, 0) != 0)
323 return false;
324
325 vector<string> const dirs = VectorizeString(Path.substr(Parent.size()), '/');
326 string progress = Parent;
327 for (vector<string>::const_iterator d = dirs.begin(); d != dirs.end(); ++d)
328 {
329 if (d->empty() == true)
330 continue;
331
332 progress.append("/").append(*d);
333 if (DirectoryExists(progress) == true)
334 continue;
335
336 if (mkdir(progress.c_str(), 0755) != 0)
337 return false;
338 }
339 return true;
340}
341 /*}}}*/
7753e468 342// CreateAPTDirectoryIfNeeded - ensure that the given directory exists /*{{{*/
b29c3712
DK
343// ---------------------------------------------------------------------
344/* a small wrapper around CreateDirectory to check if it exists and to
345 remove the trailing "/apt/" from the parent directory if needed */
7753e468 346bool CreateAPTDirectoryIfNeeded(string const &Parent, string const &Path)
b29c3712
DK
347{
348 if (DirectoryExists(Path) == true)
349 return true;
350
351 size_t const len = Parent.size();
352 if (len > 5 && Parent.find("/apt/", len - 6, 5) == len - 5)
353 {
354 if (CreateDirectory(Parent.substr(0,len-5), Path) == true)
355 return true;
356 }
357 else if (CreateDirectory(Parent, Path) == true)
358 return true;
359
360 return false;
361}
362 /*}}}*/
46e39c8e
MV
363// GetListOfFilesInDir - returns a vector of files in the given dir /*{{{*/
364// ---------------------------------------------------------------------
365/* If an extension is given only files with this extension are included
366 in the returned vector, otherwise every "normal" file is included. */
b39c1859
MV
367std::vector<string> GetListOfFilesInDir(string const &Dir, string const &Ext,
368 bool const &SortList, bool const &AllowNoExt)
369{
370 std::vector<string> ext;
371 ext.reserve(2);
372 if (Ext.empty() == false)
373 ext.push_back(Ext);
374 if (AllowNoExt == true && ext.empty() == false)
375 ext.push_back("");
376 return GetListOfFilesInDir(Dir, ext, SortList);
377}
378std::vector<string> GetListOfFilesInDir(string const &Dir, std::vector<string> const &Ext,
379 bool const &SortList)
380{
381 // Attention debuggers: need to be set with the environment config file!
382 bool const Debug = _config->FindB("Debug::GetListOfFilesInDir", false);
383 if (Debug == true)
384 {
385 std::clog << "Accept in " << Dir << " only files with the following " << Ext.size() << " extensions:" << std::endl;
386 if (Ext.empty() == true)
387 std::clog << "\tNO extension" << std::endl;
388 else
389 for (std::vector<string>::const_iterator e = Ext.begin();
390 e != Ext.end(); ++e)
391 std::clog << '\t' << (e->empty() == true ? "NO" : *e) << " extension" << std::endl;
392 }
393
46e39c8e 394 std::vector<string> List;
36f1098a 395
69c2ecbd 396 if (DirectoryExists(Dir) == false)
36f1098a
DK
397 {
398 _error->Error(_("List of files can't be created as '%s' is not a directory"), Dir.c_str());
399 return List;
400 }
401
1408e219 402 Configuration::MatchAgainstConfig SilentIgnore("Dir::Ignore-Files-Silently");
46e39c8e
MV
403 DIR *D = opendir(Dir.c_str());
404 if (D == 0)
405 {
406 _error->Errno("opendir",_("Unable to read %s"),Dir.c_str());
407 return List;
408 }
409
410 for (struct dirent *Ent = readdir(D); Ent != 0; Ent = readdir(D))
411 {
b39c1859 412 // skip "hidden" files
46e39c8e
MV
413 if (Ent->d_name[0] == '.')
414 continue;
415
491058e3
DK
416 // Make sure it is a file and not something else
417 string const File = flCombine(Dir,Ent->d_name);
418#ifdef _DIRENT_HAVE_D_TYPE
419 if (Ent->d_type != DT_REG)
420#endif
421 {
69c2ecbd 422 if (RealFileExists(File) == false)
491058e3 423 {
84e254d6
DK
424 // do not show ignoration warnings for directories
425 if (
426#ifdef _DIRENT_HAVE_D_TYPE
427 Ent->d_type == DT_DIR ||
428#endif
69c2ecbd 429 DirectoryExists(File) == true)
84e254d6 430 continue;
491058e3
DK
431 if (SilentIgnore.Match(Ent->d_name) == false)
432 _error->Notice(_("Ignoring '%s' in directory '%s' as it is not a regular file"), Ent->d_name, Dir.c_str());
433 continue;
434 }
435 }
436
b39c1859
MV
437 // check for accepted extension:
438 // no extension given -> periods are bad as hell!
439 // extensions given -> "" extension allows no extension
440 if (Ext.empty() == false)
441 {
442 string d_ext = flExtension(Ent->d_name);
443 if (d_ext == Ent->d_name) // no extension
444 {
445 if (std::find(Ext.begin(), Ext.end(), "") == Ext.end())
446 {
447 if (Debug == true)
448 std::clog << "Bad file: " << Ent->d_name << " → no extension" << std::endl;
5edc3966 449 if (SilentIgnore.Match(Ent->d_name) == false)
491058e3 450 _error->Notice(_("Ignoring file '%s' in directory '%s' as it has no filename extension"), Ent->d_name, Dir.c_str());
b39c1859
MV
451 continue;
452 }
453 }
454 else if (std::find(Ext.begin(), Ext.end(), d_ext) == Ext.end())
455 {
456 if (Debug == true)
457 std::clog << "Bad file: " << Ent->d_name << " → bad extension »" << flExtension(Ent->d_name) << "«" << std::endl;
1408e219 458 if (SilentIgnore.Match(Ent->d_name) == false)
491058e3 459 _error->Notice(_("Ignoring file '%s' in directory '%s' as it has an invalid filename extension"), Ent->d_name, Dir.c_str());
b39c1859
MV
460 continue;
461 }
462 }
46e39c8e 463
b39c1859 464 // Skip bad filenames ala run-parts
46e39c8e
MV
465 const char *C = Ent->d_name;
466 for (; *C != 0; ++C)
467 if (isalpha(*C) == 0 && isdigit(*C) == 0
b39c1859
MV
468 && *C != '_' && *C != '-') {
469 // no required extension -> dot is a bad character
470 if (*C == '.' && Ext.empty() == false)
471 continue;
46e39c8e 472 break;
b39c1859 473 }
46e39c8e 474
b39c1859 475 // we don't reach the end of the name -> bad character included
46e39c8e 476 if (*C != 0)
b39c1859
MV
477 {
478 if (Debug == true)
479 std::clog << "Bad file: " << Ent->d_name << " → bad character »"
480 << *C << "« in filename (period allowed: " << (Ext.empty() ? "no" : "yes") << ")" << std::endl;
481 continue;
482 }
483
fbb2c7e0
DK
484 // skip filenames which end with a period. These are never valid
485 if (*(C - 1) == '.')
486 {
487 if (Debug == true)
488 std::clog << "Bad file: " << Ent->d_name << " → Period as last character" << std::endl;
489 continue;
490 }
491
492 if (Debug == true)
493 std::clog << "Accept file: " << Ent->d_name << " in " << Dir << std::endl;
494 List.push_back(File);
495 }
496 closedir(D);
497
498 if (SortList == true)
499 std::sort(List.begin(),List.end());
500 return List;
501}
502std::vector<string> GetListOfFilesInDir(string const &Dir, bool SortList)
503{
504 bool const Debug = _config->FindB("Debug::GetListOfFilesInDir", false);
505 if (Debug == true)
506 std::clog << "Accept in " << Dir << " all regular files" << std::endl;
507
508 std::vector<string> List;
509
69c2ecbd 510 if (DirectoryExists(Dir) == false)
fbb2c7e0
DK
511 {
512 _error->Error(_("List of files can't be created as '%s' is not a directory"), Dir.c_str());
513 return List;
514 }
515
516 DIR *D = opendir(Dir.c_str());
517 if (D == 0)
518 {
519 _error->Errno("opendir",_("Unable to read %s"),Dir.c_str());
520 return List;
521 }
522
523 for (struct dirent *Ent = readdir(D); Ent != 0; Ent = readdir(D))
524 {
525 // skip "hidden" files
526 if (Ent->d_name[0] == '.')
527 continue;
528
529 // Make sure it is a file and not something else
530 string const File = flCombine(Dir,Ent->d_name);
531#ifdef _DIRENT_HAVE_D_TYPE
532 if (Ent->d_type != DT_REG)
533#endif
534 {
69c2ecbd 535 if (RealFileExists(File) == false)
fbb2c7e0
DK
536 {
537 if (Debug == true)
538 std::clog << "Bad file: " << Ent->d_name << " → it is not a real file" << std::endl;
539 continue;
540 }
541 }
542
543 // Skip bad filenames ala run-parts
544 const char *C = Ent->d_name;
545 for (; *C != 0; ++C)
546 if (isalpha(*C) == 0 && isdigit(*C) == 0
547 && *C != '_' && *C != '-' && *C != '.')
548 break;
549
550 // we don't reach the end of the name -> bad character included
551 if (*C != 0)
552 {
553 if (Debug == true)
554 std::clog << "Bad file: " << Ent->d_name << " → bad character »" << *C << "« in filename" << std::endl;
555 continue;
556 }
557
b39c1859
MV
558 // skip filenames which end with a period. These are never valid
559 if (*(C - 1) == '.')
560 {
561 if (Debug == true)
562 std::clog << "Bad file: " << Ent->d_name << " → Period as last character" << std::endl;
46e39c8e 563 continue;
b39c1859 564 }
46e39c8e 565
b39c1859
MV
566 if (Debug == true)
567 std::clog << "Accept file: " << Ent->d_name << " in " << Dir << std::endl;
46e39c8e
MV
568 List.push_back(File);
569 }
570 closedir(D);
571
572 if (SortList == true)
573 std::sort(List.begin(),List.end());
574 return List;
575}
576 /*}}}*/
578bfd0a
AL
577// SafeGetCWD - This is a safer getcwd that returns a dynamic string /*{{{*/
578// ---------------------------------------------------------------------
579/* We return / on failure. */
580string SafeGetCWD()
581{
582 // Stash the current dir.
583 char S[300];
584 S[0] = 0;
7f25bdff 585 if (getcwd(S,sizeof(S)-2) == 0)
578bfd0a 586 return "/";
7f25bdff
AL
587 unsigned int Len = strlen(S);
588 S[Len] = '/';
589 S[Len+1] = 0;
578bfd0a
AL
590 return S;
591}
592 /*}}}*/
2ec858bc
MV
593// GetModificationTime - Get the mtime of the given file or -1 on error /*{{{*/
594// ---------------------------------------------------------------------
595/* We return / on failure. */
596time_t GetModificationTime(string const &Path)
597{
598 struct stat St;
599 if (stat(Path.c_str(), &St) < 0)
600 return -1;
601 return St.st_mtime;
602}
603 /*}}}*/
8ce4327b
AL
604// flNotDir - Strip the directory from the filename /*{{{*/
605// ---------------------------------------------------------------------
606/* */
607string flNotDir(string File)
608{
609 string::size_type Res = File.rfind('/');
610 if (Res == string::npos)
611 return File;
612 Res++;
613 return string(File,Res,Res - File.length());
614}
615 /*}}}*/
d38b7b3d
AL
616// flNotFile - Strip the file from the directory name /*{{{*/
617// ---------------------------------------------------------------------
171c45bc 618/* Result ends in a / */
d38b7b3d
AL
619string flNotFile(string File)
620{
621 string::size_type Res = File.rfind('/');
622 if (Res == string::npos)
171c45bc 623 return "./";
d38b7b3d
AL
624 Res++;
625 return string(File,0,Res);
626}
627 /*}}}*/
b2e465d6
AL
628// flExtension - Return the extension for the file /*{{{*/
629// ---------------------------------------------------------------------
630/* */
631string flExtension(string File)
632{
633 string::size_type Res = File.rfind('.');
634 if (Res == string::npos)
635 return File;
636 Res++;
637 return string(File,Res,Res - File.length());
638}
639 /*}}}*/
421c8d10
AL
640// flNoLink - If file is a symlink then deref it /*{{{*/
641// ---------------------------------------------------------------------
642/* If the name is not a link then the returned path is the input. */
643string flNoLink(string File)
644{
645 struct stat St;
646 if (lstat(File.c_str(),&St) != 0 || S_ISLNK(St.st_mode) == 0)
647 return File;
648 if (stat(File.c_str(),&St) != 0)
649 return File;
650
651 /* Loop resolving the link. There is no need to limit the number of
652 loops because the stat call above ensures that the symlink is not
653 circular */
654 char Buffer[1024];
655 string NFile = File;
656 while (1)
657 {
658 // Read the link
3286ad13 659 ssize_t Res;
421c8d10 660 if ((Res = readlink(NFile.c_str(),Buffer,sizeof(Buffer))) <= 0 ||
3286ad13 661 (size_t)Res >= sizeof(Buffer))
421c8d10
AL
662 return File;
663
664 // Append or replace the previous path
665 Buffer[Res] = 0;
666 if (Buffer[0] == '/')
667 NFile = Buffer;
668 else
669 NFile = flNotFile(NFile) + Buffer;
670
671 // See if we are done
672 if (lstat(NFile.c_str(),&St) != 0)
673 return File;
674 if (S_ISLNK(St.st_mode) == 0)
675 return NFile;
676 }
677}
678 /*}}}*/
b2e465d6
AL
679// flCombine - Combine a file and a directory /*{{{*/
680// ---------------------------------------------------------------------
681/* If the file is an absolute path then it is just returned, otherwise
682 the directory is pre-pended to it. */
683string flCombine(string Dir,string File)
684{
685 if (File.empty() == true)
686 return string();
687
688 if (File[0] == '/' || Dir.empty() == true)
689 return File;
690 if (File.length() >= 2 && File[0] == '.' && File[1] == '/')
691 return File;
692 if (Dir[Dir.length()-1] == '/')
693 return Dir + File;
694 return Dir + '/' + File;
695}
696 /*}}}*/
3b5421b4
AL
697// SetCloseExec - Set the close on exec flag /*{{{*/
698// ---------------------------------------------------------------------
699/* */
700void SetCloseExec(int Fd,bool Close)
701{
702 if (fcntl(Fd,F_SETFD,(Close == false)?0:FD_CLOEXEC) != 0)
703 {
704 cerr << "FATAL -> Could not set close on exec " << strerror(errno) << endl;
705 exit(100);
706 }
707}
708 /*}}}*/
709// SetNonBlock - Set the nonblocking flag /*{{{*/
710// ---------------------------------------------------------------------
711/* */
712void SetNonBlock(int Fd,bool Block)
713{
0a8a80e5
AL
714 int Flags = fcntl(Fd,F_GETFL) & (~O_NONBLOCK);
715 if (fcntl(Fd,F_SETFL,Flags | ((Block == false)?0:O_NONBLOCK)) != 0)
3b5421b4
AL
716 {
717 cerr << "FATAL -> Could not set non-blocking flag " << strerror(errno) << endl;
718 exit(100);
719 }
720}
721 /*}}}*/
722// WaitFd - Wait for a FD to become readable /*{{{*/
723// ---------------------------------------------------------------------
b2e465d6 724/* This waits for a FD to become readable using select. It is useful for
6d5dd02a
AL
725 applications making use of non-blocking sockets. The timeout is
726 in seconds. */
1084d58a 727bool WaitFd(int Fd,bool write,unsigned long timeout)
3b5421b4
AL
728{
729 fd_set Set;
cc2313b7 730 struct timeval tv;
3b5421b4
AL
731 FD_ZERO(&Set);
732 FD_SET(Fd,&Set);
6d5dd02a
AL
733 tv.tv_sec = timeout;
734 tv.tv_usec = 0;
1084d58a 735 if (write == true)
b0db36b1
AL
736 {
737 int Res;
738 do
739 {
740 Res = select(Fd+1,0,&Set,0,(timeout != 0?&tv:0));
741 }
742 while (Res < 0 && errno == EINTR);
743
744 if (Res <= 0)
745 return false;
1084d58a
AL
746 }
747 else
748 {
b0db36b1
AL
749 int Res;
750 do
751 {
752 Res = select(Fd+1,&Set,0,0,(timeout != 0?&tv:0));
753 }
754 while (Res < 0 && errno == EINTR);
755
756 if (Res <= 0)
757 return false;
cc2313b7 758 }
1084d58a 759
3b5421b4
AL
760 return true;
761}
762 /*}}}*/
54676e1a
AL
763// ExecFork - Magical fork that sanitizes the context before execing /*{{{*/
764// ---------------------------------------------------------------------
765/* This is used if you want to cleanse the environment for the forked
766 child, it fixes up the important signals and nukes all of the fds,
767 otherwise acts like normal fork. */
75ef8f14 768pid_t ExecFork()
e45c4617
MV
769{
770 set<int> KeepFDs;
771
772 Configuration::Item const *Opts = _config->Tree("APT::Keep-Fds");
773 if (Opts != 0 && Opts->Child != 0)
774 {
775 Opts = Opts->Child;
776 for (; Opts != 0; Opts = Opts->Next)
777 {
778 if (Opts->Value.empty() == true)
779 continue;
780 int fd = atoi(Opts->Value.c_str());
781 KeepFDs.insert(fd);
782 }
783 }
784 return ExecFork(KeepFDs);
785}
786
787pid_t ExecFork(std::set<int> KeepFDs)
54676e1a
AL
788{
789 // Fork off the process
790 pid_t Process = fork();
791 if (Process < 0)
792 {
793 cerr << "FATAL -> Failed to fork." << endl;
794 exit(100);
795 }
796
797 // Spawn the subprocess
798 if (Process == 0)
799 {
800 // Setup the signals
801 signal(SIGPIPE,SIG_DFL);
802 signal(SIGQUIT,SIG_DFL);
803 signal(SIGINT,SIG_DFL);
804 signal(SIGWINCH,SIG_DFL);
805 signal(SIGCONT,SIG_DFL);
806 signal(SIGTSTP,SIG_DFL);
75ef8f14 807
54676e1a
AL
808 // Close all of our FDs - just in case
809 for (int K = 3; K != 40; K++)
75ef8f14
MV
810 {
811 if(KeepFDs.find(K) == KeepFDs.end())
007dc9e0 812 fcntl(K,F_SETFD,FD_CLOEXEC);
75ef8f14 813 }
54676e1a
AL
814 }
815
816 return Process;
817}
818 /*}}}*/
ddc1d8d0
AL
819// ExecWait - Fancy waitpid /*{{{*/
820// ---------------------------------------------------------------------
2c9a72d1 821/* Waits for the given sub process. If Reap is set then no errors are
ddc1d8d0
AL
822 generated. Otherwise a failed subprocess will generate a proper descriptive
823 message */
3826564e 824bool ExecWait(pid_t Pid,const char *Name,bool Reap)
ddc1d8d0
AL
825{
826 if (Pid <= 1)
827 return true;
828
829 // Wait and collect the error code
830 int Status;
831 while (waitpid(Pid,&Status,0) != Pid)
832 {
833 if (errno == EINTR)
834 continue;
835
836 if (Reap == true)
837 return false;
838
db0db9fe 839 return _error->Error(_("Waited for %s but it wasn't there"),Name);
ddc1d8d0
AL
840 }
841
842
843 // Check for an error code.
844 if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
845 {
846 if (Reap == true)
847 return false;
ab7f4d7c 848 if (WIFSIGNALED(Status) != 0)
40e7fe0e 849 {
ab7f4d7c
MV
850 if( WTERMSIG(Status) == SIGSEGV)
851 return _error->Error(_("Sub-process %s received a segmentation fault."),Name);
852 else
853 return _error->Error(_("Sub-process %s received signal %u."),Name, WTERMSIG(Status));
40e7fe0e 854 }
ddc1d8d0
AL
855
856 if (WIFEXITED(Status) != 0)
b2e465d6 857 return _error->Error(_("Sub-process %s returned an error code (%u)"),Name,WEXITSTATUS(Status));
ddc1d8d0 858
b2e465d6 859 return _error->Error(_("Sub-process %s exited unexpectedly"),Name);
ddc1d8d0
AL
860 }
861
862 return true;
863}
864 /*}}}*/
578bfd0a 865
13d87e2e 866// FileFd::Open - Open a file /*{{{*/
578bfd0a
AL
867// ---------------------------------------------------------------------
868/* The most commonly used open mode combinations are given with Mode */
52b47296 869bool FileFd::Open(string FileName,unsigned int const Mode,CompressMode Compress, unsigned long const Perms)
578bfd0a 870{
257e8d66
DK
871 if (Mode == ReadOnlyGzip)
872 return Open(FileName, ReadOnly, Gzip, Perms);
257e8d66 873
468720c5 874 if (Compress == Auto && (Mode & WriteOnly) == WriteOnly)
ae635e3c 875 return FileFdError("Autodetection on %s only works in ReadOnly openmode!", FileName.c_str());
257e8d66 876
468720c5
DK
877 std::vector<APT::Configuration::Compressor> const compressors = APT::Configuration::getCompressors();
878 std::vector<APT::Configuration::Compressor>::const_iterator compressor = compressors.begin();
879 if (Compress == Auto)
880 {
468720c5
DK
881 for (; compressor != compressors.end(); ++compressor)
882 {
883 std::string file = std::string(FileName).append(compressor->Extension);
884 if (FileExists(file) == false)
885 continue;
886 FileName = file;
468720c5
DK
887 break;
888 }
889 }
890 else if (Compress == Extension)
891 {
52b47296
DK
892 std::string::size_type const found = FileName.find_last_of('.');
893 std::string ext;
894 if (found != std::string::npos)
895 {
896 ext = FileName.substr(found);
897 if (ext == ".new" || ext == ".bak")
898 {
899 std::string::size_type const found2 = FileName.find_last_of('.', found - 1);
900 if (found2 != std::string::npos)
901 ext = FileName.substr(found2, found - found2);
902 else
903 ext.clear();
904 }
905 }
aee1aac6
DK
906 for (; compressor != compressors.end(); ++compressor)
907 if (ext == compressor->Extension)
908 break;
909 // no matching extension - assume uncompressed (imagine files like 'example.org_Packages')
910 if (compressor == compressors.end())
911 for (compressor = compressors.begin(); compressor != compressors.end(); ++compressor)
912 if (compressor->Name == ".")
468720c5 913 break;
468720c5 914 }
aee1aac6 915 else
468720c5
DK
916 {
917 std::string name;
918 switch (Compress)
919 {
aee1aac6 920 case None: name = "."; break;
468720c5
DK
921 case Gzip: name = "gzip"; break;
922 case Bzip2: name = "bzip2"; break;
923 case Lzma: name = "lzma"; break;
924 case Xz: name = "xz"; break;
aee1aac6
DK
925 case Auto:
926 case Extension:
52b47296 927 // Unreachable
ae635e3c 928 return FileFdError("Opening File %s in None, Auto or Extension should be already handled?!?", FileName.c_str());
468720c5
DK
929 }
930 for (; compressor != compressors.end(); ++compressor)
931 if (compressor->Name == name)
932 break;
aee1aac6 933 if (compressor == compressors.end())
ae635e3c 934 return FileFdError("Can't find a configured compressor %s for file %s", name.c_str(), FileName.c_str());
468720c5
DK
935 }
936
aee1aac6 937 if (compressor == compressors.end())
ae635e3c 938 return FileFdError("Can't find a match for specified compressor mode for file %s", FileName.c_str());
aee1aac6
DK
939 return Open(FileName, Mode, *compressor, Perms);
940}
52b47296 941bool FileFd::Open(string FileName,unsigned int const Mode,APT::Configuration::Compressor const &compressor, unsigned long const Perms)
aee1aac6
DK
942{
943 Close();
aee1aac6
DK
944 Flags = AutoClose;
945
946 if ((Mode & WriteOnly) != WriteOnly && (Mode & (Atomic | Create | Empty | Exclusive)) != 0)
ae635e3c 947 return FileFdError("ReadOnly mode for %s doesn't accept additional flags!", FileName.c_str());
aee1aac6 948 if ((Mode & ReadWrite) == 0)
ae635e3c 949 return FileFdError("No openmode provided in FileFd::Open for %s", FileName.c_str());
468720c5 950
257e8d66
DK
951 if ((Mode & Atomic) == Atomic)
952 {
953 Flags |= Replace;
257e8d66
DK
954 }
955 else if ((Mode & (Exclusive | Create)) == (Exclusive | Create))
956 {
957 // for atomic, this will be done by rename in Close()
958 unlink(FileName.c_str());
959 }
960 if ((Mode & Empty) == Empty)
578bfd0a 961 {
257e8d66
DK
962 struct stat Buf;
963 if (lstat(FileName.c_str(),&Buf) == 0 && S_ISLNK(Buf.st_mode))
964 unlink(FileName.c_str());
965 }
c4fc2fd7 966
561f860a
DK
967 int fileflags = 0;
968 #define if_FLAGGED_SET(FLAG, MODE) if ((Mode & FLAG) == FLAG) fileflags |= MODE
969 if_FLAGGED_SET(ReadWrite, O_RDWR);
970 else if_FLAGGED_SET(ReadOnly, O_RDONLY);
971 else if_FLAGGED_SET(WriteOnly, O_WRONLY);
4a9db827 972
561f860a
DK
973 if_FLAGGED_SET(Create, O_CREAT);
974 if_FLAGGED_SET(Empty, O_TRUNC);
975 if_FLAGGED_SET(Exclusive, O_EXCL);
561f860a 976 #undef if_FLAGGED_SET
52b47296 977
7335eebe
AGM
978 if ((Mode & Atomic) == Atomic)
979 {
980 char *name = strdup((FileName + ".XXXXXX").c_str());
981
dc545c0b 982 if((iFd = mkstemp(name)) == -1)
7335eebe
AGM
983 {
984 free(name);
98b69f9d 985 return FileFdErrno("mkstemp", "Could not create temporary file for %s", FileName.c_str());
7335eebe
AGM
986 }
987
988 TemporaryFileName = string(name);
7335eebe 989 free(name);
dc545c0b
DK
990
991 if(Perms != 600 && fchmod(iFd, Perms) == -1)
992 return FileFdErrno("fchmod", "Could not change permissions for temporary file %s", TemporaryFileName.c_str());
7335eebe 993 }
468720c5 994 else
561f860a 995 iFd = open(FileName.c_str(), fileflags, Perms);
468720c5 996
b711c01e 997 this->FileName = FileName;
561f860a
DK
998 if (iFd == -1 || OpenInternDescriptor(Mode, compressor) == false)
999 {
468720c5 1000 if (iFd != -1)
fc81e8f2 1001 {
561f860a
DK
1002 close (iFd);
1003 iFd = -1;
fc81e8f2 1004 }
ae635e3c 1005 return FileFdErrno("open",_("Could not open file %s"), FileName.c_str());
257e8d66 1006 }
578bfd0a 1007
13d87e2e
AL
1008 SetCloseExec(iFd,true);
1009 return true;
578bfd0a 1010}
257e8d66
DK
1011 /*}}}*/
1012// FileFd::OpenDescriptor - Open a filedescriptor /*{{{*/
1013// ---------------------------------------------------------------------
1014/* */
52b47296 1015bool FileFd::OpenDescriptor(int Fd, unsigned int const Mode, CompressMode Compress, bool AutoClose)
aee1aac6
DK
1016{
1017 std::vector<APT::Configuration::Compressor> const compressors = APT::Configuration::getCompressors();
1018 std::vector<APT::Configuration::Compressor>::const_iterator compressor = compressors.begin();
1019 std::string name;
bce778a3
MV
1020
1021 // compat with the old API
1022 if (Mode == ReadOnlyGzip && Compress == None)
1023 Compress = Gzip;
1024
aee1aac6
DK
1025 switch (Compress)
1026 {
1027 case None: name = "."; break;
1028 case Gzip: name = "gzip"; break;
1029 case Bzip2: name = "bzip2"; break;
1030 case Lzma: name = "lzma"; break;
1031 case Xz: name = "xz"; break;
1032 case Auto:
1033 case Extension:
f97bb523
DK
1034 if (AutoClose == true && Fd != -1)
1035 close(Fd);
ae635e3c 1036 return FileFdError("Opening Fd %d in Auto or Extension compression mode is not supported", Fd);
aee1aac6
DK
1037 }
1038 for (; compressor != compressors.end(); ++compressor)
1039 if (compressor->Name == name)
1040 break;
1041 if (compressor == compressors.end())
f97bb523
DK
1042 {
1043 if (AutoClose == true && Fd != -1)
1044 close(Fd);
ae635e3c 1045 return FileFdError("Can't find a configured compressor %s for file %s", name.c_str(), FileName.c_str());
f97bb523 1046 }
aee1aac6
DK
1047 return OpenDescriptor(Fd, Mode, *compressor, AutoClose);
1048}
52b47296 1049bool FileFd::OpenDescriptor(int Fd, unsigned int const Mode, APT::Configuration::Compressor const &compressor, bool AutoClose)
144c0969
JAK
1050{
1051 Close();
1052 Flags = (AutoClose) ? FileFd::AutoClose : 0;
d3aac32e
DK
1053 if (AutoClose == false && (
1054#ifdef HAVE_ZLIB
1055 compressor.Name == "gzip" ||
1056#endif
1057#ifdef HAVE_BZ2
1058 compressor.Name == "bzip2" ||
1059#endif
1060 false))
1061 {
1062 // Need to duplicate fd here or gzclose for cleanup will close the fd as well
1063 iFd = dup(Fd);
1064 }
1065 else
1066 iFd = Fd;
b711c01e 1067 this->FileName = "";
f97bb523 1068 if (Fd == -1 || OpenInternDescriptor(Mode, compressor) == false)
468720c5 1069 {
f97bb523
DK
1070 if (iFd != -1 && (
1071#ifdef HAVE_ZLIB
1072 compressor.Name == "gzip" ||
1073#endif
1074#ifdef HAVE_BZ2
1075 compressor.Name == "bzip2" ||
1076#endif
1077 AutoClose == true))
1078 {
468720c5 1079 close (iFd);
f97bb523
DK
1080 iFd = -1;
1081 }
1082 return FileFdError(_("Could not open file descriptor %d"), Fd);
144c0969 1083 }
144c0969 1084 return true;
468720c5 1085}
52b47296 1086bool FileFd::OpenInternDescriptor(unsigned int const Mode, APT::Configuration::Compressor const &compressor)
468720c5 1087{
ff477ee1
DK
1088 if (compressor.Name == "." || compressor.Binary.empty() == true)
1089 return true;
1090
ba667cf7
DK
1091 if (d == NULL)
1092 {
1093 d = new FileFdPrivate();
1094 d->openmode = Mode;
1095 d->compressor = compressor;
1096 }
ff477ee1 1097
7efb8c8e 1098#ifdef HAVE_ZLIB
ff477ee1 1099 if (compressor.Name == "gzip")
468720c5 1100 {
adbd7eb4
DK
1101 if (d->gz != NULL)
1102 {
1103 gzclose(d->gz);
1104 d->gz = NULL;
1105 }
468720c5 1106 if ((Mode & ReadWrite) == ReadWrite)
032bd56f 1107 d->gz = gzdopen(iFd, "r+");
468720c5 1108 else if ((Mode & WriteOnly) == WriteOnly)
032bd56f 1109 d->gz = gzdopen(iFd, "w");
468720c5 1110 else
c4997486 1111 d->gz = gzdopen(iFd, "r");
032bd56f 1112 if (d->gz == NULL)
468720c5 1113 return false;
032bd56f 1114 Flags |= Compressed;
561f860a 1115 return true;
468720c5 1116 }
699b209e 1117#endif
c4997486 1118#ifdef HAVE_BZ2
ff477ee1 1119 if (compressor.Name == "bzip2")
c4997486 1120 {
adbd7eb4
DK
1121 if (d->bz2 != NULL)
1122 {
1123 BZ2_bzclose(d->bz2);
1124 d->bz2 = NULL;
1125 }
c4997486
DK
1126 if ((Mode & ReadWrite) == ReadWrite)
1127 d->bz2 = BZ2_bzdopen(iFd, "r+");
1128 else if ((Mode & WriteOnly) == WriteOnly)
1129 d->bz2 = BZ2_bzdopen(iFd, "w");
1130 else
1131 d->bz2 = BZ2_bzdopen(iFd, "r");
1132 if (d->bz2 == NULL)
1133 return false;
1134 Flags |= Compressed;
1135 return true;
1136 }
1137#endif
1138
adbd7eb4
DK
1139 // collect zombies here in case we reopen
1140 if (d->compressor_pid > 0)
1141 ExecWait(d->compressor_pid, "FileFdCompressor", true);
561f860a
DK
1142
1143 if ((Mode & ReadWrite) == ReadWrite)
ae635e3c 1144 return FileFdError("ReadWrite mode is not supported for file %s", FileName.c_str());
561f860a
DK
1145
1146 bool const Comp = (Mode & WriteOnly) == WriteOnly;
561f860a
DK
1147 if (Comp == false)
1148 {
adbd7eb4 1149 // Handle 'decompression' of empty files
561f860a
DK
1150 struct stat Buf;
1151 fstat(iFd, &Buf);
1152 if (Buf.st_size == 0 && S_ISFIFO(Buf.st_mode) == false)
1153 return true;
1154
1155 // We don't need the file open - instead let the compressor open it
1156 // as he properly knows better how to efficiently read from 'his' file
1157 if (FileName.empty() == false)
afb093cd 1158 {
561f860a 1159 close(iFd);
afb093cd
DK
1160 iFd = -1;
1161 }
561f860a
DK
1162 }
1163
1164 // Create a data pipe
1165 int Pipe[2] = {-1,-1};
1166 if (pipe(Pipe) != 0)
ae635e3c 1167 return FileFdErrno("pipe",_("Failed to create subprocess IPC"));
561f860a
DK
1168 for (int J = 0; J != 2; J++)
1169 SetCloseExec(Pipe[J],true);
1170
1171 d->compressed_fd = iFd;
1172 d->pipe = true;
1173
1174 if (Comp == true)
1175 iFd = Pipe[1];
1176 else
1177 iFd = Pipe[0];
1178
1179 // The child..
1180 d->compressor_pid = ExecFork();
1181 if (d->compressor_pid == 0)
1182 {
1183 if (Comp == true)
1184 {
1185 dup2(d->compressed_fd,STDOUT_FILENO);
1186 dup2(Pipe[0],STDIN_FILENO);
1187 }
1188 else
1189 {
1190 if (FileName.empty() == true)
1191 dup2(d->compressed_fd,STDIN_FILENO);
1192 dup2(Pipe[1],STDOUT_FILENO);
1193 }
0b4895d3
DK
1194 int const nullfd = open("/dev/null", O_WRONLY);
1195 if (nullfd != -1)
1196 {
1197 dup2(nullfd,STDERR_FILENO);
1198 close(nullfd);
1199 }
561f860a
DK
1200
1201 SetCloseExec(STDOUT_FILENO,false);
1202 SetCloseExec(STDIN_FILENO,false);
1203
1204 std::vector<char const*> Args;
1205 Args.push_back(compressor.Binary.c_str());
1206 std::vector<std::string> const * const addArgs =
1207 (Comp == true) ? &(compressor.CompressArgs) : &(compressor.UncompressArgs);
1208 for (std::vector<std::string>::const_iterator a = addArgs->begin();
1209 a != addArgs->end(); ++a)
1210 Args.push_back(a->c_str());
1211 if (Comp == false && FileName.empty() == false)
1212 {
1213 Args.push_back("--stdout");
1214 if (TemporaryFileName.empty() == false)
1215 Args.push_back(TemporaryFileName.c_str());
1216 else
1217 Args.push_back(FileName.c_str());
1218 }
1219 Args.push_back(NULL);
1220
1221 execvp(Args[0],(char **)&Args[0]);
1222 cerr << _("Failed to exec compressor ") << Args[0] << endl;
1223 _exit(100);
1224 }
1225 if (Comp == true)
1226 close(Pipe[0]);
468720c5 1227 else
561f860a 1228 close(Pipe[1]);
561f860a 1229
468720c5 1230 return true;
144c0969 1231}
578bfd0a 1232 /*}}}*/
8e06abb2 1233// FileFd::~File - Closes the file /*{{{*/
578bfd0a
AL
1234// ---------------------------------------------------------------------
1235/* If the proper modes are selected then we close the Fd and possibly
1236 unlink the file on error. */
8e06abb2 1237FileFd::~FileFd()
578bfd0a
AL
1238{
1239 Close();
500400fe 1240 if (d != NULL)
500400fe 1241 d->CloseDown(FileName);
96ab3c6f
MV
1242 delete d;
1243 d = NULL;
578bfd0a
AL
1244}
1245 /*}}}*/
8e06abb2 1246// FileFd::Read - Read a bit of the file /*{{{*/
578bfd0a 1247// ---------------------------------------------------------------------
b0db36b1
AL
1248/* We are carefull to handle interruption by a signal while reading
1249 gracefully. */
650faab0 1250bool FileFd::Read(void *To,unsigned long long Size,unsigned long long *Actual)
578bfd0a 1251{
3286ad13 1252 ssize_t Res;
b0db36b1 1253 errno = 0;
f604cf55
AL
1254 if (Actual != 0)
1255 *Actual = 0;
699b209e 1256 *((char *)To) = '\0';
b0db36b1 1257 do
578bfd0a 1258 {
7efb8c8e 1259#ifdef HAVE_ZLIB
ff477ee1 1260 if (d != NULL && d->gz != NULL)
c4997486
DK
1261 Res = gzread(d->gz,To,Size);
1262 else
1263#endif
1264#ifdef HAVE_BZ2
ff477ee1 1265 if (d != NULL && d->bz2 != NULL)
c4997486 1266 Res = BZ2_bzread(d->bz2,To,Size);
a3a03f5d 1267 else
699b209e 1268#endif
a3a03f5d 1269 Res = read(iFd,To,Size);
b711c01e 1270
b0db36b1
AL
1271 if (Res < 0)
1272 {
b711c01e
DK
1273 if (errno == EINTR)
1274 continue;
7efb8c8e 1275#ifdef HAVE_ZLIB
ff477ee1 1276 if (d != NULL && d->gz != NULL)
b711c01e
DK
1277 {
1278 int err;
1279 char const * const errmsg = gzerror(d->gz, &err);
1280 if (err != Z_ERRNO)
ae635e3c 1281 return FileFdError("gzread: %s (%d: %s)", _("Read error"), err, errmsg);
b711c01e 1282 }
c4997486
DK
1283#endif
1284#ifdef HAVE_BZ2
ff477ee1 1285 if (d != NULL && d->bz2 != NULL)
c4997486
DK
1286 {
1287 int err;
1288 char const * const errmsg = BZ2_bzerror(d->bz2, &err);
1289 if (err != BZ_IO_ERROR)
ae635e3c 1290 return FileFdError("BZ2_bzread: %s (%d: %s)", _("Read error"), err, errmsg);
c4997486 1291 }
b711c01e 1292#endif
ae635e3c 1293 return FileFdErrno("read",_("Read error"));
b0db36b1 1294 }
578bfd0a 1295
b0db36b1
AL
1296 To = (char *)To + Res;
1297 Size -= Res;
ff477ee1
DK
1298 if (d != NULL)
1299 d->seekpos += Res;
f604cf55
AL
1300 if (Actual != 0)
1301 *Actual += Res;
b0db36b1
AL
1302 }
1303 while (Res > 0 && Size > 0);
1304
1305 if (Size == 0)
1306 return true;
1307
ddc1d8d0 1308 // Eof handling
f604cf55 1309 if (Actual != 0)
ddc1d8d0
AL
1310 {
1311 Flags |= HitEof;
1312 return true;
1313 }
ae635e3c
DK
1314
1315 return FileFdError(_("read, still have %llu to read but none left"), Size);
578bfd0a
AL
1316}
1317 /*}}}*/
032bd56f
DK
1318// FileFd::ReadLine - Read a complete line from the file /*{{{*/
1319// ---------------------------------------------------------------------
1320/* Beware: This method can be quiet slow for big buffers on UNcompressed
1321 files because of the naive implementation! */
1322char* FileFd::ReadLine(char *To, unsigned long long const Size)
1323{
699b209e 1324 *To = '\0';
7efb8c8e 1325#ifdef HAVE_ZLIB
ff477ee1 1326 if (d != NULL && d->gz != NULL)
032bd56f 1327 return gzgets(d->gz, To, Size);
699b209e 1328#endif
032bd56f
DK
1329
1330 unsigned long long read = 0;
40468850
DK
1331 while ((Size - 1) != read)
1332 {
1333 unsigned long long done = 0;
1334 if (Read(To + read, 1, &done) == false)
1335 return NULL;
1336 if (done == 0)
1337 break;
1338 if (To[read++] == '\n')
1339 break;
1340 }
1341 if (read == 0)
032bd56f 1342 return NULL;
40468850 1343 To[read] = '\0';
032bd56f
DK
1344 return To;
1345}
1346 /*}}}*/
8e06abb2 1347// FileFd::Write - Write to the file /*{{{*/
578bfd0a
AL
1348// ---------------------------------------------------------------------
1349/* */
650faab0 1350bool FileFd::Write(const void *From,unsigned long long Size)
578bfd0a 1351{
3286ad13 1352 ssize_t Res;
b0db36b1
AL
1353 errno = 0;
1354 do
578bfd0a 1355 {
7efb8c8e 1356#ifdef HAVE_ZLIB
ff477ee1 1357 if (d != NULL && d->gz != NULL)
032bd56f 1358 Res = gzwrite(d->gz,From,Size);
a3a03f5d 1359 else
c4997486
DK
1360#endif
1361#ifdef HAVE_BZ2
ff477ee1 1362 if (d != NULL && d->bz2 != NULL)
c4997486
DK
1363 Res = BZ2_bzwrite(d->bz2,(void*)From,Size);
1364 else
699b209e 1365#endif
a3a03f5d 1366 Res = write(iFd,From,Size);
b0db36b1
AL
1367 if (Res < 0 && errno == EINTR)
1368 continue;
1369 if (Res < 0)
1370 {
c4997486 1371#ifdef HAVE_ZLIB
ff477ee1 1372 if (d != NULL && d->gz != NULL)
c4997486
DK
1373 {
1374 int err;
1375 char const * const errmsg = gzerror(d->gz, &err);
1376 if (err != Z_ERRNO)
ae635e3c 1377 return FileFdError("gzwrite: %s (%d: %s)", _("Write error"), err, errmsg);
c4997486
DK
1378 }
1379#endif
1380#ifdef HAVE_BZ2
ff477ee1 1381 if (d != NULL && d->bz2 != NULL)
c4997486
DK
1382 {
1383 int err;
1384 char const * const errmsg = BZ2_bzerror(d->bz2, &err);
1385 if (err != BZ_IO_ERROR)
ae635e3c 1386 return FileFdError("BZ2_bzwrite: %s (%d: %s)", _("Write error"), err, errmsg);
c4997486
DK
1387 }
1388#endif
ae635e3c 1389 return FileFdErrno("write",_("Write error"));
b0db36b1
AL
1390 }
1391
1392 From = (char *)From + Res;
1393 Size -= Res;
ff477ee1
DK
1394 if (d != NULL)
1395 d->seekpos += Res;
578bfd0a 1396 }
b0db36b1 1397 while (Res > 0 && Size > 0);
578bfd0a 1398
b0db36b1
AL
1399 if (Size == 0)
1400 return true;
ae635e3c
DK
1401
1402 return FileFdError(_("write, still have %llu to write but couldn't"), Size);
d68d65ad
DK
1403}
1404bool FileFd::Write(int Fd, const void *From, unsigned long long Size)
1405{
3286ad13 1406 ssize_t Res;
d68d65ad
DK
1407 errno = 0;
1408 do
1409 {
1410 Res = write(Fd,From,Size);
1411 if (Res < 0 && errno == EINTR)
1412 continue;
1413 if (Res < 0)
1414 return _error->Errno("write",_("Write error"));
1415
1416 From = (char *)From + Res;
1417 Size -= Res;
1418 }
1419 while (Res > 0 && Size > 0);
1420
1421 if (Size == 0)
1422 return true;
1423
1424 return _error->Error(_("write, still have %llu to write but couldn't"), Size);
578bfd0a
AL
1425}
1426 /*}}}*/
8e06abb2 1427// FileFd::Seek - Seek in the file /*{{{*/
578bfd0a
AL
1428// ---------------------------------------------------------------------
1429/* */
650faab0 1430bool FileFd::Seek(unsigned long long To)
578bfd0a 1431{
ff477ee1 1432 if (d != NULL && (d->pipe == true
c4997486 1433#ifdef HAVE_BZ2
ff477ee1 1434 || d->bz2 != NULL
c4997486 1435#endif
ff477ee1 1436 ))
699b209e 1437 {
1abbc47c
DK
1438 // Our poor man seeking in pipes is costly, so try to avoid it
1439 unsigned long long seekpos = Tell();
1440 if (seekpos == To)
1441 return true;
1442 else if (seekpos < To)
1443 return Skip(To - seekpos);
1444
561f860a 1445 if ((d->openmode & ReadOnly) != ReadOnly)
ae635e3c 1446 return FileFdError("Reopen is only implemented for read-only files!");
c4997486 1447#ifdef HAVE_BZ2
e3b402f4
MV
1448 if (d->bz2 != NULL)
1449 {
1450 BZ2_bzclose(d->bz2);
1451 d->bz2 = NULL;
1452 }
c4997486 1453#endif
afb093cd
DK
1454 if (iFd != -1)
1455 close(iFd);
1456 iFd = -1;
561f860a
DK
1457 if (TemporaryFileName.empty() == false)
1458 iFd = open(TemporaryFileName.c_str(), O_RDONLY);
1459 else if (FileName.empty() == false)
1460 iFd = open(FileName.c_str(), O_RDONLY);
1461 else
6fd947bd
DK
1462 {
1463 if (d->compressed_fd > 0)
1464 if (lseek(d->compressed_fd, 0, SEEK_SET) != 0)
1465 iFd = d->compressed_fd;
500400fe 1466 if (iFd < 0)
ae635e3c 1467 return FileFdError("Reopen is not implemented for pipes opened with FileFd::OpenDescriptor()!");
6fd947bd 1468 }
561f860a
DK
1469
1470 if (OpenInternDescriptor(d->openmode, d->compressor) == false)
ae635e3c 1471 return FileFdError("Seek on file %s because it couldn't be reopened", FileName.c_str());
561f860a
DK
1472
1473 if (To != 0)
1474 return Skip(To);
1abbc47c
DK
1475
1476 d->seekpos = To;
561f860a 1477 return true;
699b209e 1478 }
3286ad13 1479 off_t res;
7efb8c8e 1480#ifdef HAVE_ZLIB
ff477ee1 1481 if (d != NULL && d->gz)
032bd56f 1482 res = gzseek(d->gz,To,SEEK_SET);
a3a03f5d 1483 else
699b209e 1484#endif
a3a03f5d 1485 res = lseek(iFd,To,SEEK_SET);
3286ad13 1486 if (res != (off_t)To)
ae635e3c 1487 return FileFdError("Unable to seek to %llu", To);
1abbc47c 1488
ff477ee1
DK
1489 if (d != NULL)
1490 d->seekpos = To;
727f18af
AL
1491 return true;
1492}
1493 /*}}}*/
1494// FileFd::Skip - Seek in the file /*{{{*/
1495// ---------------------------------------------------------------------
1496/* */
650faab0 1497bool FileFd::Skip(unsigned long long Over)
727f18af 1498{
ff477ee1 1499 if (d != NULL && (d->pipe == true
c4997486 1500#ifdef HAVE_BZ2
ff477ee1 1501 || d->bz2 != NULL
c4997486 1502#endif
ff477ee1 1503 ))
40468850
DK
1504 {
1505 d->seekpos += Over;
1506 char buffer[1024];
1507 while (Over != 0)
1508 {
1509 unsigned long long toread = std::min((unsigned long long) sizeof(buffer), Over);
1510 if (Read(buffer, toread) == false)
ae635e3c 1511 return FileFdError("Unable to seek ahead %llu",Over);
40468850
DK
1512 Over -= toread;
1513 }
1514 return true;
1515 }
1516
3286ad13 1517 off_t res;
7efb8c8e 1518#ifdef HAVE_ZLIB
ff477ee1 1519 if (d != NULL && d->gz != NULL)
032bd56f 1520 res = gzseek(d->gz,Over,SEEK_CUR);
a3a03f5d 1521 else
699b209e 1522#endif
a3a03f5d 1523 res = lseek(iFd,Over,SEEK_CUR);
1524 if (res < 0)
ae635e3c 1525 return FileFdError("Unable to seek ahead %llu",Over);
ff477ee1
DK
1526 if (d != NULL)
1527 d->seekpos = res;
1abbc47c 1528
6d5dd02a
AL
1529 return true;
1530}
1531 /*}}}*/
1532// FileFd::Truncate - Truncate the file /*{{{*/
1533// ---------------------------------------------------------------------
1534/* */
650faab0 1535bool FileFd::Truncate(unsigned long long To)
6d5dd02a 1536{
c4997486 1537#if defined HAVE_ZLIB || defined HAVE_BZ2
ff477ee1 1538 if (d != NULL && (d->gz != NULL || d->bz2 != NULL))
ae635e3c 1539 return FileFdError("Truncating compressed files is not implemented (%s)", FileName.c_str());
c4997486 1540#endif
6d5dd02a 1541 if (ftruncate(iFd,To) != 0)
ae635e3c
DK
1542 return FileFdError("Unable to truncate to %llu",To);
1543
578bfd0a
AL
1544 return true;
1545}
1546 /*}}}*/
7f25bdff
AL
1547// FileFd::Tell - Current seek position /*{{{*/
1548// ---------------------------------------------------------------------
1549/* */
650faab0 1550unsigned long long FileFd::Tell()
7f25bdff 1551{
1abbc47c
DK
1552 // In theory, we could just return seekpos here always instead of
1553 // seeking around, but not all users of FileFd use always Seek() and co
1554 // so d->seekpos isn't always true and we can just use it as a hint if
1555 // we have nothing else, but not always as an authority…
ff477ee1 1556 if (d != NULL && (d->pipe == true
c4997486 1557#ifdef HAVE_BZ2
ff477ee1 1558 || d->bz2 != NULL
c4997486 1559#endif
ff477ee1 1560 ))
1abbc47c
DK
1561 return d->seekpos;
1562
a3a03f5d 1563 off_t Res;
7efb8c8e 1564#ifdef HAVE_ZLIB
ff477ee1 1565 if (d != NULL && d->gz != NULL)
032bd56f 1566 Res = gztell(d->gz);
a3a03f5d 1567 else
699b209e 1568#endif
a3a03f5d 1569 Res = lseek(iFd,0,SEEK_CUR);
7f25bdff 1570 if (Res == (off_t)-1)
ae635e3c 1571 FileFdErrno("lseek","Failed to determine the current file position");
ff477ee1
DK
1572 if (d != NULL)
1573 d->seekpos = Res;
7f25bdff
AL
1574 return Res;
1575}
1576 /*}}}*/
4260fd39 1577// FileFd::FileSize - Return the size of the file /*{{{*/
578bfd0a
AL
1578// ---------------------------------------------------------------------
1579/* */
650faab0 1580unsigned long long FileFd::FileSize()
578bfd0a
AL
1581{
1582 struct stat Buf;
ff477ee1 1583 if ((d == NULL || d->pipe == false) && fstat(iFd,&Buf) != 0)
ae635e3c 1584 return FileFdErrno("fstat","Unable to determine the file size");
699b209e
DK
1585
1586 // for compressor pipes st_size is undefined and at 'best' zero
ff477ee1 1587 if ((d != NULL && d->pipe == true) || S_ISFIFO(Buf.st_mode))
699b209e
DK
1588 {
1589 // we set it here, too, as we get the info here for free
1590 // in theory the Open-methods should take care of it already
ff477ee1
DK
1591 if (d != NULL)
1592 d->pipe = true;
699b209e 1593 if (stat(FileName.c_str(), &Buf) != 0)
ae635e3c 1594 return FileFdErrno("stat","Unable to determine the file size");
699b209e
DK
1595 }
1596
4260fd39
DK
1597 return Buf.st_size;
1598}
1599 /*}}}*/
1600// FileFd::Size - Return the size of the content in the file /*{{{*/
1601// ---------------------------------------------------------------------
1602/* */
650faab0 1603unsigned long long FileFd::Size()
4260fd39 1604{
650faab0 1605 unsigned long long size = FileSize();
44dc669e 1606
699b209e
DK
1607 // for compressor pipes st_size is undefined and at 'best' zero,
1608 // so we 'read' the content and 'seek' back - see there
ff477ee1 1609 if (d != NULL && (d->pipe == true
c4997486 1610#ifdef HAVE_BZ2
ff477ee1 1611 || (d->bz2 && size > 0)
c4997486 1612#endif
ff477ee1 1613 ))
699b209e 1614 {
1abbc47c 1615 unsigned long long const oldSeek = Tell();
699b209e
DK
1616 char ignore[1000];
1617 unsigned long long read = 0;
1618 do {
638593bd
DK
1619 if (Read(ignore, sizeof(ignore), &read) == false)
1620 {
1621 Seek(oldSeek);
1622 return 0;
1623 }
699b209e 1624 } while(read != 0);
1abbc47c
DK
1625 size = Tell();
1626 Seek(oldSeek);
699b209e 1627 }
7efb8c8e 1628#ifdef HAVE_ZLIB
44dc669e 1629 // only check gzsize if we are actually a gzip file, just checking for
032bd56f 1630 // "gz" is not sufficient as uncompressed files could be opened with
44dc669e 1631 // gzopen in "direct" mode as well
ff477ee1 1632 else if (d != NULL && d->gz && !gzdirect(d->gz) && size > 0)
9c182afa 1633 {
65c72a4b 1634 off_t const oldPos = lseek(iFd,0,SEEK_CUR);
9c182afa
MP
1635 /* unfortunately zlib.h doesn't provide a gzsize(), so we have to do
1636 * this ourselves; the original (uncompressed) file size is the last 32
1637 * bits of the file */
650faab0 1638 // FIXME: Size for gz-files is limited by 32bit… no largefile support
9c182afa 1639 if (lseek(iFd, -4, SEEK_END) < 0)
638593bd
DK
1640 {
1641 FileFdErrno("lseek","Unable to seek to end of gzipped file");
1642 return 0;
1643 }
1644 size = 0;
9c182afa 1645 if (read(iFd, &size, 4) != 4)
638593bd
DK
1646 {
1647 FileFdErrno("read","Unable to read original size of gzipped file");
1648 return 0;
1649 }
f330c0f3
DK
1650
1651#ifdef WORDS_BIGENDIAN
2cae0ccb
DK
1652 uint32_t tmp_size = size;
1653 uint8_t const * const p = (uint8_t const * const) &tmp_size;
1654 tmp_size = (p[3] << 24) | (p[2] << 16) | (p[1] << 8) | p[0];
1655 size = tmp_size;
f330c0f3 1656#endif
9c182afa 1657
65c72a4b 1658 if (lseek(iFd, oldPos, SEEK_SET) < 0)
638593bd
DK
1659 {
1660 FileFdErrno("lseek","Unable to seek in gzipped file");
1661 return 0;
1662 }
65c72a4b 1663
9c182afa
MP
1664 return size;
1665 }
699b209e 1666#endif
9c182afa 1667
44dc669e 1668 return size;
578bfd0a
AL
1669}
1670 /*}}}*/
76a763e1
DK
1671// FileFd::ModificationTime - Return the time of last touch /*{{{*/
1672// ---------------------------------------------------------------------
1673/* */
1674time_t FileFd::ModificationTime()
1675{
1676 struct stat Buf;
ff477ee1 1677 if ((d == NULL || d->pipe == false) && fstat(iFd,&Buf) != 0)
76a763e1 1678 {
ae635e3c 1679 FileFdErrno("fstat","Unable to determine the modification time of file %s", FileName.c_str());
76a763e1
DK
1680 return 0;
1681 }
699b209e
DK
1682
1683 // for compressor pipes st_size is undefined and at 'best' zero
ff477ee1 1684 if ((d != NULL && d->pipe == true) || S_ISFIFO(Buf.st_mode))
699b209e
DK
1685 {
1686 // we set it here, too, as we get the info here for free
1687 // in theory the Open-methods should take care of it already
ff477ee1
DK
1688 if (d != NULL)
1689 d->pipe = true;
699b209e
DK
1690 if (stat(FileName.c_str(), &Buf) != 0)
1691 {
ae635e3c 1692 FileFdErrno("fstat","Unable to determine the modification time of file %s", FileName.c_str());
699b209e
DK
1693 return 0;
1694 }
1695 }
1696
76a763e1
DK
1697 return Buf.st_mtime;
1698}
1699 /*}}}*/
8e06abb2 1700// FileFd::Close - Close the file if the close flag is set /*{{{*/
578bfd0a
AL
1701// ---------------------------------------------------------------------
1702/* */
8e06abb2 1703bool FileFd::Close()
578bfd0a 1704{
032bd56f
DK
1705 if (iFd == -1)
1706 return true;
1707
578bfd0a
AL
1708 bool Res = true;
1709 if ((Flags & AutoClose) == AutoClose)
d13c2d3f 1710 {
500400fe
DK
1711 if ((Flags & Compressed) != Compressed && iFd > 0 && close(iFd) != 0)
1712 Res &= _error->Errno("close",_("Problem closing the file %s"), FileName.c_str());
1713
1714 if (d != NULL)
1715 {
1716 Res &= d->CloseDown(FileName);
1717 delete d;
1718 d = NULL;
1719 }
d13c2d3f 1720 }
3010fb0e 1721
d3aac32e 1722 if ((Flags & Replace) == Replace) {
3010fb0e 1723 if (rename(TemporaryFileName.c_str(), FileName.c_str()) != 0)
62d073d9
DK
1724 Res &= _error->Errno("rename",_("Problem renaming the file %s to %s"), TemporaryFileName.c_str(), FileName.c_str());
1725
fd3b761e 1726 FileName = TemporaryFileName; // for the unlink() below.
257e8d66 1727 TemporaryFileName.clear();
3010fb0e 1728 }
62d073d9
DK
1729
1730 iFd = -1;
1731
578bfd0a
AL
1732 if ((Flags & Fail) == Fail && (Flags & DelOnFail) == DelOnFail &&
1733 FileName.empty() == false)
1734 if (unlink(FileName.c_str()) != 0)
62d073d9 1735 Res &= _error->WarningE("unlnk",_("Problem unlinking the file %s"), FileName.c_str());
3010fb0e 1736
fbb89d94
DK
1737 if (Res == false)
1738 Flags |= Fail;
578bfd0a
AL
1739 return Res;
1740}
1741 /*}}}*/
b2e465d6
AL
1742// FileFd::Sync - Sync the file /*{{{*/
1743// ---------------------------------------------------------------------
1744/* */
1745bool FileFd::Sync()
1746{
b2e465d6 1747 if (fsync(iFd) != 0)
ae635e3c
DK
1748 return FileFdErrno("sync",_("Problem syncing the file"));
1749 return true;
1750}
1751 /*}}}*/
1752// FileFd::FileFdErrno - set Fail and call _error->Errno *{{{*/
1753bool FileFd::FileFdErrno(const char *Function, const char *Description,...)
1754{
1755 Flags |= Fail;
1756 va_list args;
1757 size_t msgSize = 400;
1758 int const errsv = errno;
1759 while (true)
fbb89d94 1760 {
ae635e3c
DK
1761 va_start(args,Description);
1762 if (_error->InsertErrno(GlobalError::ERROR, Function, Description, args, errsv, msgSize) == false)
1763 break;
1764 va_end(args);
fbb89d94 1765 }
ae635e3c
DK
1766 return false;
1767}
1768 /*}}}*/
1769// FileFd::FileFdError - set Fail and call _error->Error *{{{*/
1770bool FileFd::FileFdError(const char *Description,...) {
1771 Flags |= Fail;
1772 va_list args;
1773 size_t msgSize = 400;
1774 while (true)
1775 {
1776 va_start(args,Description);
1777 if (_error->Insert(GlobalError::ERROR, Description, args, msgSize) == false)
1778 break;
1779 va_end(args);
1780 }
1781 return false;
b2e465d6
AL
1782}
1783 /*}}}*/
699b209e
DK
1784
1785gzFile FileFd::gzFd() { return (gzFile) d->gz; }
488011fa
MV
1786
1787
1788// Glob - wrapper around "glob()" /*{{{*/
1789// ---------------------------------------------------------------------
1790/* */
1791std::vector<std::string> Glob(std::string const &pattern, int flags)
1792{
1793 std::vector<std::string> result;
1794 glob_t globbuf;
ec4835a1
ÁGM
1795 int glob_res;
1796 unsigned int i;
488011fa
MV
1797
1798 glob_res = glob(pattern.c_str(), flags, NULL, &globbuf);
1799
1800 if (glob_res != 0)
1801 {
1802 if(glob_res != GLOB_NOMATCH) {
1803 _error->Errno("glob", "Problem with glob");
1804 return result;
1805 }
1806 }
1807
1808 // append results
1809 for(i=0;i<globbuf.gl_pathc;i++)
1810 result.push_back(string(globbuf.gl_pathv[i]));
1811
1812 globfree(&globbuf);
1813 return result;
1814}
1815 /*}}}*/