1 // -*- mode: cpp; mode: fold -*-
3 /* ######################################################################
7 CopyFile - Buffered copy of a single file
8 GetLock - dpkg compatible lock file manipulation (fcntl)
10 Most of this source is placed in the Public Domain, do with it what
12 It was originally written by Jason Gunthorpe <jgg@debian.org>.
13 FileFd gzip support added by Martin Pitt <martin.pitt@canonical.com>
15 The exception is RunScripts() it is under the GPLv2
17 ##################################################################### */
19 // Include Files /*{{{*/
22 #include <apt-pkg/fileutl.h>
23 #include <apt-pkg/strutl.h>
24 #include <apt-pkg/error.h>
25 #include <apt-pkg/sptr.h>
26 #include <apt-pkg/aptconfiguration.h>
27 #include <apt-pkg/configuration.h>
28 #include <apt-pkg/macros.h>
33 #include <sys/select.h>
71 // RunScripts - Run a set of scripts from a configuration subtree /*{{{*/
72 // ---------------------------------------------------------------------
74 bool RunScripts(const char *Cnf
)
76 Configuration::Item
const *Opts
= _config
->Tree(Cnf
);
77 if (Opts
== 0 || Opts
->Child
== 0)
81 // Fork for running the system calls
82 pid_t Child
= ExecFork();
87 if (_config
->FindDir("DPkg::Chroot-Directory","/") != "/")
89 std::cerr
<< "Chrooting into "
90 << _config
->FindDir("DPkg::Chroot-Directory")
92 if (chroot(_config
->FindDir("DPkg::Chroot-Directory","/").c_str()) != 0)
96 if (chdir("/tmp/") != 0)
99 unsigned int Count
= 1;
100 for (; Opts
!= 0; Opts
= Opts
->Next
, Count
++)
102 if (Opts
->Value
.empty() == true)
105 if(_config
->FindB("Debug::RunScripts", false) == true)
106 std::clog
<< "Running external script: '"
107 << Opts
->Value
<< "'" << std::endl
;
109 if (system(Opts
->Value
.c_str()) != 0)
115 // Wait for the child
117 while (waitpid(Child
,&Status
,0) != Child
)
121 return _error
->Errno("waitpid","Couldn't wait for subprocess");
124 // Restore sig int/quit
125 signal(SIGQUIT
,SIG_DFL
);
126 signal(SIGINT
,SIG_DFL
);
128 // Check for an error code.
129 if (WIFEXITED(Status
) == 0 || WEXITSTATUS(Status
) != 0)
131 unsigned int Count
= WEXITSTATUS(Status
);
135 for (; Opts
!= 0 && Count
!= 1; Opts
= Opts
->Next
, Count
--);
136 _error
->Error("Problem executing scripts %s '%s'",Cnf
,Opts
->Value
.c_str());
139 return _error
->Error("Sub-process returned an error code");
146 // CopyFile - Buffered copy of a file /*{{{*/
147 // ---------------------------------------------------------------------
148 /* The caller is expected to set things so that failure causes erasure */
149 bool CopyFile(FileFd
&From
,FileFd
&To
)
151 if (From
.IsOpen() == false || To
.IsOpen() == false ||
152 From
.Failed() == true || To
.Failed() == true)
155 // Buffered copy between fds
156 SPtrArray
<unsigned char> Buf
= new unsigned char[64000];
157 unsigned long long Size
= From
.Size();
160 unsigned long long ToRead
= Size
;
164 if (From
.Read(Buf
,ToRead
) == false ||
165 To
.Write(Buf
,ToRead
) == false)
174 // GetLock - Gets a lock file /*{{{*/
175 // ---------------------------------------------------------------------
176 /* This will create an empty file of the given name and lock it. Once this
177 is done all other calls to GetLock in any other process will fail with
178 -1. The return result is the fd of the file, the call should call
179 close at some time. */
180 int GetLock(string File
,bool Errors
)
182 // GetLock() is used in aptitude on directories with public-write access
183 // Use O_NOFOLLOW here to prevent symlink traversal attacks
184 int FD
= open(File
.c_str(),O_RDWR
| O_CREAT
| O_NOFOLLOW
,0640);
187 // Read only .. can't have locking problems there.
190 _error
->Warning(_("Not using locking for read only lock file %s"),File
.c_str());
191 return dup(0); // Need something for the caller to close
195 _error
->Errno("open",_("Could not open lock file %s"),File
.c_str());
197 // Feh.. We do this to distinguish the lock vs open case..
201 SetCloseExec(FD
,true);
203 // Acquire a write lock
206 fl
.l_whence
= SEEK_SET
;
209 if (fcntl(FD
,F_SETLK
,&fl
) == -1)
211 // always close to not leak resources
218 _error
->Warning(_("Not using locking for nfs mounted lock file %s"),File
.c_str());
219 return dup(0); // Need something for the caller to close
223 _error
->Errno("open",_("Could not get lock %s"),File
.c_str());
231 // FileExists - Check if a file exists /*{{{*/
232 // ---------------------------------------------------------------------
233 /* Beware: Directories are also files! */
234 bool FileExists(string File
)
237 if (stat(File
.c_str(),&Buf
) != 0)
242 // RealFileExists - Check if a file exists and if it is really a file /*{{{*/
243 // ---------------------------------------------------------------------
245 bool RealFileExists(string File
)
248 if (stat(File
.c_str(),&Buf
) != 0)
250 return ((Buf
.st_mode
& S_IFREG
) != 0);
253 // DirectoryExists - Check if a directory exists and is really one /*{{{*/
254 // ---------------------------------------------------------------------
256 bool DirectoryExists(string
const &Path
)
259 if (stat(Path
.c_str(),&Buf
) != 0)
261 return ((Buf
.st_mode
& S_IFDIR
) != 0);
264 // CreateDirectory - poor man's mkdir -p guarded by a parent directory /*{{{*/
265 // ---------------------------------------------------------------------
266 /* This method will create all directories needed for path in good old
267 mkdir -p style but refuses to do this if Parent is not a prefix of
268 this Path. Example: /var/cache/ and /var/cache/apt/archives are given,
269 so it will create apt/archives if /var/cache exists - on the other
270 hand if the parent is /var/lib the creation will fail as this path
271 is not a parent of the path to be generated. */
272 bool CreateDirectory(string
const &Parent
, string
const &Path
)
274 if (Parent
.empty() == true || Path
.empty() == true)
277 if (DirectoryExists(Path
) == true)
280 if (DirectoryExists(Parent
) == false)
283 // we are not going to create directories "into the blue"
284 if (Path
.compare(0, Parent
.length(), Parent
) != 0)
287 vector
<string
> const dirs
= VectorizeString(Path
.substr(Parent
.size()), '/');
288 string progress
= Parent
;
289 for (vector
<string
>::const_iterator d
= dirs
.begin(); d
!= dirs
.end(); ++d
)
291 if (d
->empty() == true)
294 progress
.append("/").append(*d
);
295 if (DirectoryExists(progress
) == true)
298 if (mkdir(progress
.c_str(), 0755) != 0)
304 // CreateAPTDirectoryIfNeeded - ensure that the given directory exists /*{{{*/
305 // ---------------------------------------------------------------------
306 /* a small wrapper around CreateDirectory to check if it exists and to
307 remove the trailing "/apt/" from the parent directory if needed */
308 bool CreateAPTDirectoryIfNeeded(string
const &Parent
, string
const &Path
)
310 if (DirectoryExists(Path
) == true)
313 size_t const len
= Parent
.size();
314 if (len
> 5 && Parent
.find("/apt/", len
- 6, 5) == len
- 5)
316 if (CreateDirectory(Parent
.substr(0,len
-5), Path
) == true)
319 else if (CreateDirectory(Parent
, Path
) == true)
325 // GetListOfFilesInDir - returns a vector of files in the given dir /*{{{*/
326 // ---------------------------------------------------------------------
327 /* If an extension is given only files with this extension are included
328 in the returned vector, otherwise every "normal" file is included. */
329 std::vector
<string
> GetListOfFilesInDir(string
const &Dir
, string
const &Ext
,
330 bool const &SortList
, bool const &AllowNoExt
)
332 std::vector
<string
> ext
;
334 if (Ext
.empty() == false)
336 if (AllowNoExt
== true && ext
.empty() == false)
338 return GetListOfFilesInDir(Dir
, ext
, SortList
);
340 std::vector
<string
> GetListOfFilesInDir(string
const &Dir
, std::vector
<string
> const &Ext
,
341 bool const &SortList
)
343 // Attention debuggers: need to be set with the environment config file!
344 bool const Debug
= _config
->FindB("Debug::GetListOfFilesInDir", false);
347 std::clog
<< "Accept in " << Dir
<< " only files with the following " << Ext
.size() << " extensions:" << std::endl
;
348 if (Ext
.empty() == true)
349 std::clog
<< "\tNO extension" << std::endl
;
351 for (std::vector
<string
>::const_iterator e
= Ext
.begin();
353 std::clog
<< '\t' << (e
->empty() == true ? "NO" : *e
) << " extension" << std::endl
;
356 std::vector
<string
> List
;
358 if (DirectoryExists(Dir
) == false)
360 _error
->Error(_("List of files can't be created as '%s' is not a directory"), Dir
.c_str());
364 Configuration::MatchAgainstConfig
SilentIgnore("Dir::Ignore-Files-Silently");
365 DIR *D
= opendir(Dir
.c_str());
368 _error
->Errno("opendir",_("Unable to read %s"),Dir
.c_str());
372 for (struct dirent
*Ent
= readdir(D
); Ent
!= 0; Ent
= readdir(D
))
374 // skip "hidden" files
375 if (Ent
->d_name
[0] == '.')
378 // Make sure it is a file and not something else
379 string
const File
= flCombine(Dir
,Ent
->d_name
);
380 #ifdef _DIRENT_HAVE_D_TYPE
381 if (Ent
->d_type
!= DT_REG
)
384 if (RealFileExists(File
) == false)
386 // do not show ignoration warnings for directories
388 #ifdef _DIRENT_HAVE_D_TYPE
389 Ent
->d_type
== DT_DIR
||
391 DirectoryExists(File
) == true)
393 if (SilentIgnore
.Match(Ent
->d_name
) == false)
394 _error
->Notice(_("Ignoring '%s' in directory '%s' as it is not a regular file"), Ent
->d_name
, Dir
.c_str());
399 // check for accepted extension:
400 // no extension given -> periods are bad as hell!
401 // extensions given -> "" extension allows no extension
402 if (Ext
.empty() == false)
404 string d_ext
= flExtension(Ent
->d_name
);
405 if (d_ext
== Ent
->d_name
) // no extension
407 if (std::find(Ext
.begin(), Ext
.end(), "") == Ext
.end())
410 std::clog
<< "Bad file: " << Ent
->d_name
<< " → no extension" << std::endl
;
411 if (SilentIgnore
.Match(Ent
->d_name
) == false)
412 _error
->Notice(_("Ignoring file '%s' in directory '%s' as it has no filename extension"), Ent
->d_name
, Dir
.c_str());
416 else if (std::find(Ext
.begin(), Ext
.end(), d_ext
) == Ext
.end())
419 std::clog
<< "Bad file: " << Ent
->d_name
<< " → bad extension »" << flExtension(Ent
->d_name
) << "«" << std::endl
;
420 if (SilentIgnore
.Match(Ent
->d_name
) == false)
421 _error
->Notice(_("Ignoring file '%s' in directory '%s' as it has an invalid filename extension"), Ent
->d_name
, Dir
.c_str());
426 // Skip bad filenames ala run-parts
427 const char *C
= Ent
->d_name
;
429 if (isalpha(*C
) == 0 && isdigit(*C
) == 0
430 && *C
!= '_' && *C
!= '-' && *C
!= ':') {
431 // no required extension -> dot is a bad character
432 if (*C
== '.' && Ext
.empty() == false)
437 // we don't reach the end of the name -> bad character included
441 std::clog
<< "Bad file: " << Ent
->d_name
<< " → bad character »"
442 << *C
<< "« in filename (period allowed: " << (Ext
.empty() ? "no" : "yes") << ")" << std::endl
;
446 // skip filenames which end with a period. These are never valid
450 std::clog
<< "Bad file: " << Ent
->d_name
<< " → Period as last character" << std::endl
;
455 std::clog
<< "Accept file: " << Ent
->d_name
<< " in " << Dir
<< std::endl
;
456 List
.push_back(File
);
460 if (SortList
== true)
461 std::sort(List
.begin(),List
.end());
464 std::vector
<string
> GetListOfFilesInDir(string
const &Dir
, bool SortList
)
466 bool const Debug
= _config
->FindB("Debug::GetListOfFilesInDir", false);
468 std::clog
<< "Accept in " << Dir
<< " all regular files" << std::endl
;
470 std::vector
<string
> List
;
472 if (DirectoryExists(Dir
) == false)
474 _error
->Error(_("List of files can't be created as '%s' is not a directory"), Dir
.c_str());
478 DIR *D
= opendir(Dir
.c_str());
481 _error
->Errno("opendir",_("Unable to read %s"),Dir
.c_str());
485 for (struct dirent
*Ent
= readdir(D
); Ent
!= 0; Ent
= readdir(D
))
487 // skip "hidden" files
488 if (Ent
->d_name
[0] == '.')
491 // Make sure it is a file and not something else
492 string
const File
= flCombine(Dir
,Ent
->d_name
);
493 #ifdef _DIRENT_HAVE_D_TYPE
494 if (Ent
->d_type
!= DT_REG
)
497 if (RealFileExists(File
) == false)
500 std::clog
<< "Bad file: " << Ent
->d_name
<< " → it is not a real file" << std::endl
;
505 // Skip bad filenames ala run-parts
506 const char *C
= Ent
->d_name
;
508 if (isalpha(*C
) == 0 && isdigit(*C
) == 0
509 && *C
!= '_' && *C
!= '-' && *C
!= '.')
512 // we don't reach the end of the name -> bad character included
516 std::clog
<< "Bad file: " << Ent
->d_name
<< " → bad character »" << *C
<< "« in filename" << std::endl
;
520 // skip filenames which end with a period. These are never valid
524 std::clog
<< "Bad file: " << Ent
->d_name
<< " → Period as last character" << std::endl
;
529 std::clog
<< "Accept file: " << Ent
->d_name
<< " in " << Dir
<< std::endl
;
530 List
.push_back(File
);
534 if (SortList
== true)
535 std::sort(List
.begin(),List
.end());
539 // SafeGetCWD - This is a safer getcwd that returns a dynamic string /*{{{*/
540 // ---------------------------------------------------------------------
541 /* We return / on failure. */
544 // Stash the current dir.
547 if (getcwd(S
,sizeof(S
)-2) == 0)
549 unsigned int Len
= strlen(S
);
555 // GetModificationTime - Get the mtime of the given file or -1 on error /*{{{*/
556 // ---------------------------------------------------------------------
557 /* We return / on failure. */
558 time_t GetModificationTime(string
const &Path
)
561 if (stat(Path
.c_str(), &St
) < 0)
566 // flNotDir - Strip the directory from the filename /*{{{*/
567 // ---------------------------------------------------------------------
569 string
flNotDir(string File
)
571 string::size_type Res
= File
.rfind('/');
572 if (Res
== string::npos
)
575 return string(File
,Res
,Res
- File
.length());
578 // flNotFile - Strip the file from the directory name /*{{{*/
579 // ---------------------------------------------------------------------
580 /* Result ends in a / */
581 string
flNotFile(string File
)
583 string::size_type Res
= File
.rfind('/');
584 if (Res
== string::npos
)
587 return string(File
,0,Res
);
590 // flExtension - Return the extension for the file /*{{{*/
591 // ---------------------------------------------------------------------
593 string
flExtension(string File
)
595 string::size_type Res
= File
.rfind('.');
596 if (Res
== string::npos
)
599 return string(File
,Res
,Res
- File
.length());
602 // flNoLink - If file is a symlink then deref it /*{{{*/
603 // ---------------------------------------------------------------------
604 /* If the name is not a link then the returned path is the input. */
605 string
flNoLink(string File
)
608 if (lstat(File
.c_str(),&St
) != 0 || S_ISLNK(St
.st_mode
) == 0)
610 if (stat(File
.c_str(),&St
) != 0)
613 /* Loop resolving the link. There is no need to limit the number of
614 loops because the stat call above ensures that the symlink is not
622 if ((Res
= readlink(NFile
.c_str(),Buffer
,sizeof(Buffer
))) <= 0 ||
623 (size_t)Res
>= sizeof(Buffer
))
626 // Append or replace the previous path
628 if (Buffer
[0] == '/')
631 NFile
= flNotFile(NFile
) + Buffer
;
633 // See if we are done
634 if (lstat(NFile
.c_str(),&St
) != 0)
636 if (S_ISLNK(St
.st_mode
) == 0)
641 // flCombine - Combine a file and a directory /*{{{*/
642 // ---------------------------------------------------------------------
643 /* If the file is an absolute path then it is just returned, otherwise
644 the directory is pre-pended to it. */
645 string
flCombine(string Dir
,string File
)
647 if (File
.empty() == true)
650 if (File
[0] == '/' || Dir
.empty() == true)
652 if (File
.length() >= 2 && File
[0] == '.' && File
[1] == '/')
654 if (Dir
[Dir
.length()-1] == '/')
656 return Dir
+ '/' + File
;
659 // SetCloseExec - Set the close on exec flag /*{{{*/
660 // ---------------------------------------------------------------------
662 void SetCloseExec(int Fd
,bool Close
)
664 if (fcntl(Fd
,F_SETFD
,(Close
== false)?0:FD_CLOEXEC
) != 0)
666 cerr
<< "FATAL -> Could not set close on exec " << strerror(errno
) << endl
;
671 // SetNonBlock - Set the nonblocking flag /*{{{*/
672 // ---------------------------------------------------------------------
674 void SetNonBlock(int Fd
,bool Block
)
676 int Flags
= fcntl(Fd
,F_GETFL
) & (~O_NONBLOCK
);
677 if (fcntl(Fd
,F_SETFL
,Flags
| ((Block
== false)?0:O_NONBLOCK
)) != 0)
679 cerr
<< "FATAL -> Could not set non-blocking flag " << strerror(errno
) << endl
;
684 // WaitFd - Wait for a FD to become readable /*{{{*/
685 // ---------------------------------------------------------------------
686 /* This waits for a FD to become readable using select. It is useful for
687 applications making use of non-blocking sockets. The timeout is
689 bool WaitFd(int Fd
,bool write
,unsigned long timeout
)
702 Res
= select(Fd
+1,0,&Set
,0,(timeout
!= 0?&tv
:0));
704 while (Res
< 0 && errno
== EINTR
);
714 Res
= select(Fd
+1,&Set
,0,0,(timeout
!= 0?&tv
:0));
716 while (Res
< 0 && errno
== EINTR
);
725 // MergeKeepFdsFromConfiguration - Merge APT::Keep-Fds configuration /*{{{*/
726 // ---------------------------------------------------------------------
727 /* This is used to merge the APT::Keep-Fds with the provided KeepFDs
730 void MergeKeepFdsFromConfiguration(std::set
<int> &KeepFDs
)
732 Configuration::Item
const *Opts
= _config
->Tree("APT::Keep-Fds");
733 if (Opts
!= 0 && Opts
->Child
!= 0)
736 for (; Opts
!= 0; Opts
= Opts
->Next
)
738 if (Opts
->Value
.empty() == true)
740 int fd
= atoi(Opts
->Value
.c_str());
746 // ExecFork - Magical fork that sanitizes the context before execing /*{{{*/
747 // ---------------------------------------------------------------------
748 /* This is used if you want to cleanse the environment for the forked
749 child, it fixes up the important signals and nukes all of the fds,
750 otherwise acts like normal fork. */
754 // we need to merge the Keep-Fds as external tools like
755 // debconf-apt-progress use it
756 MergeKeepFdsFromConfiguration(KeepFDs
);
757 return ExecFork(KeepFDs
);
760 pid_t
ExecFork(std::set
<int> KeepFDs
)
762 // Fork off the process
763 pid_t Process
= fork();
766 cerr
<< "FATAL -> Failed to fork." << endl
;
770 // Spawn the subprocess
774 signal(SIGPIPE
,SIG_DFL
);
775 signal(SIGQUIT
,SIG_DFL
);
776 signal(SIGINT
,SIG_DFL
);
777 signal(SIGWINCH
,SIG_DFL
);
778 signal(SIGCONT
,SIG_DFL
);
779 signal(SIGTSTP
,SIG_DFL
);
781 // Close all of our FDs - just in case
782 for (int K
= 3; K
!= sysconf(_SC_OPEN_MAX
); K
++)
784 if(KeepFDs
.find(K
) == KeepFDs
.end())
785 fcntl(K
,F_SETFD
,FD_CLOEXEC
);
792 // ExecWait - Fancy waitpid /*{{{*/
793 // ---------------------------------------------------------------------
794 /* Waits for the given sub process. If Reap is set then no errors are
795 generated. Otherwise a failed subprocess will generate a proper descriptive
797 bool ExecWait(pid_t Pid
,const char *Name
,bool Reap
)
802 // Wait and collect the error code
804 while (waitpid(Pid
,&Status
,0) != Pid
)
812 return _error
->Error(_("Waited for %s but it wasn't there"),Name
);
816 // Check for an error code.
817 if (WIFEXITED(Status
) == 0 || WEXITSTATUS(Status
) != 0)
821 if (WIFSIGNALED(Status
) != 0)
823 if( WTERMSIG(Status
) == SIGSEGV
)
824 return _error
->Error(_("Sub-process %s received a segmentation fault."),Name
);
826 return _error
->Error(_("Sub-process %s received signal %u."),Name
, WTERMSIG(Status
));
829 if (WIFEXITED(Status
) != 0)
830 return _error
->Error(_("Sub-process %s returned an error code (%u)"),Name
,WEXITSTATUS(Status
));
832 return _error
->Error(_("Sub-process %s exited unexpectedly"),Name
);
839 class FileFdPrivate
{ /*{{{*/
850 uint8_t buffer
[4096];
856 LZMAFILE() : file(NULL
), eof(false), compressing(false) {}
858 if (compressing
== true)
861 stream
.avail_out
= sizeof(buffer
)/sizeof(buffer
[0]);
862 stream
.next_out
= buffer
;
863 err
= lzma_code(&stream
, LZMA_FINISH
);
864 if (err
!= LZMA_OK
&& err
!= LZMA_STREAM_END
)
866 _error
->Error("~LZMAFILE: Compress finalisation failed");
869 size_t const n
= sizeof(buffer
)/sizeof(buffer
[0]) - stream
.avail_out
;
870 if (n
&& fwrite(buffer
, 1, n
, file
) != n
)
872 _error
->Errno("~LZMAFILE",_("Write error"));
875 if (err
== LZMA_STREAM_END
)
886 pid_t compressor_pid
;
888 APT::Configuration::Compressor compressor
;
889 unsigned int openmode
;
890 unsigned long long seekpos
;
901 compressed_fd(-1), compressor_pid(-1), pipe(false),
902 openmode(0), seekpos(0) {};
903 bool InternalClose(std::string
const &FileName
)
906 /* dummy so that the rest can be 'else if's */;
908 else if (gz
!= NULL
) {
909 int const e
= gzclose(gz
);
911 // gzdclose() on empty files always fails with "buffer error" here, ignore that
912 if (e
!= 0 && e
!= Z_BUF_ERROR
)
913 return _error
->Errno("close",_("Problem closing the gzip file %s"), FileName
.c_str());
917 else if (bz2
!= NULL
) {
923 else if (lzma
!= NULL
) {
930 bool CloseDown(std::string
const &FileName
)
932 bool const Res
= InternalClose(FileName
);
934 if (compressor_pid
> 0)
935 ExecWait(compressor_pid
, "FileFdCompressor", true);
940 bool InternalStream() const {
952 ~FileFdPrivate() { CloseDown(""); }
955 // FileFd::Open - Open a file /*{{{*/
956 // ---------------------------------------------------------------------
957 /* The most commonly used open mode combinations are given with Mode */
958 bool FileFd::Open(string FileName
,unsigned int const Mode
,CompressMode Compress
, unsigned long const AccessMode
)
960 if (Mode
== ReadOnlyGzip
)
961 return Open(FileName
, ReadOnly
, Gzip
, AccessMode
);
963 if (Compress
== Auto
&& (Mode
& WriteOnly
) == WriteOnly
)
964 return FileFdError("Autodetection on %s only works in ReadOnly openmode!", FileName
.c_str());
966 std::vector
<APT::Configuration::Compressor
> const compressors
= APT::Configuration::getCompressors();
967 std::vector
<APT::Configuration::Compressor
>::const_iterator compressor
= compressors
.begin();
968 if (Compress
== Auto
)
970 for (; compressor
!= compressors
.end(); ++compressor
)
972 std::string file
= FileName
+ compressor
->Extension
;
973 if (FileExists(file
) == false)
979 else if (Compress
== Extension
)
981 std::string::size_type
const found
= FileName
.find_last_of('.');
983 if (found
!= std::string::npos
)
985 ext
= FileName
.substr(found
);
986 if (ext
== ".new" || ext
== ".bak")
988 std::string::size_type
const found2
= FileName
.find_last_of('.', found
- 1);
989 if (found2
!= std::string::npos
)
990 ext
= FileName
.substr(found2
, found
- found2
);
995 for (; compressor
!= compressors
.end(); ++compressor
)
996 if (ext
== compressor
->Extension
)
998 // no matching extension - assume uncompressed (imagine files like 'example.org_Packages')
999 if (compressor
== compressors
.end())
1000 for (compressor
= compressors
.begin(); compressor
!= compressors
.end(); ++compressor
)
1001 if (compressor
->Name
== ".")
1009 case None
: name
= "."; break;
1010 case Gzip
: name
= "gzip"; break;
1011 case Bzip2
: name
= "bzip2"; break;
1012 case Lzma
: name
= "lzma"; break;
1013 case Xz
: name
= "xz"; break;
1017 return FileFdError("Opening File %s in None, Auto or Extension should be already handled?!?", FileName
.c_str());
1019 for (; compressor
!= compressors
.end(); ++compressor
)
1020 if (compressor
->Name
== name
)
1022 if (compressor
== compressors
.end())
1023 return FileFdError("Can't find a configured compressor %s for file %s", name
.c_str(), FileName
.c_str());
1026 if (compressor
== compressors
.end())
1027 return FileFdError("Can't find a match for specified compressor mode for file %s", FileName
.c_str());
1028 return Open(FileName
, Mode
, *compressor
, AccessMode
);
1030 bool FileFd::Open(string FileName
,unsigned int const Mode
,APT::Configuration::Compressor
const &compressor
, unsigned long const AccessMode
)
1035 if ((Mode
& WriteOnly
) != WriteOnly
&& (Mode
& (Atomic
| Create
| Empty
| Exclusive
)) != 0)
1036 return FileFdError("ReadOnly mode for %s doesn't accept additional flags!", FileName
.c_str());
1037 if ((Mode
& ReadWrite
) == 0)
1038 return FileFdError("No openmode provided in FileFd::Open for %s", FileName
.c_str());
1040 if ((Mode
& Atomic
) == Atomic
)
1044 else if ((Mode
& (Exclusive
| Create
)) == (Exclusive
| Create
))
1046 // for atomic, this will be done by rename in Close()
1047 unlink(FileName
.c_str());
1049 if ((Mode
& Empty
) == Empty
)
1052 if (lstat(FileName
.c_str(),&Buf
) == 0 && S_ISLNK(Buf
.st_mode
))
1053 unlink(FileName
.c_str());
1057 #define if_FLAGGED_SET(FLAG, MODE) if ((Mode & FLAG) == FLAG) fileflags |= MODE
1058 if_FLAGGED_SET(ReadWrite
, O_RDWR
);
1059 else if_FLAGGED_SET(ReadOnly
, O_RDONLY
);
1060 else if_FLAGGED_SET(WriteOnly
, O_WRONLY
);
1062 if_FLAGGED_SET(Create
, O_CREAT
);
1063 if_FLAGGED_SET(Empty
, O_TRUNC
);
1064 if_FLAGGED_SET(Exclusive
, O_EXCL
);
1065 #undef if_FLAGGED_SET
1067 if ((Mode
& Atomic
) == Atomic
)
1069 char *name
= strdup((FileName
+ ".XXXXXX").c_str());
1071 if((iFd
= mkstemp(name
)) == -1)
1074 return FileFdErrno("mkstemp", "Could not create temporary file for %s", FileName
.c_str());
1077 TemporaryFileName
= string(name
);
1080 // umask() will always set the umask and return the previous value, so
1081 // we first set the umask and then reset it to the old value
1082 mode_t
const CurrentUmask
= umask(0);
1083 umask(CurrentUmask
);
1084 // calculate the actual file permissions (just like open/creat)
1085 mode_t
const FilePermissions
= (AccessMode
& ~CurrentUmask
);
1087 if(fchmod(iFd
, FilePermissions
) == -1)
1088 return FileFdErrno("fchmod", "Could not change permissions for temporary file %s", TemporaryFileName
.c_str());
1091 iFd
= open(FileName
.c_str(), fileflags
, AccessMode
);
1093 this->FileName
= FileName
;
1094 if (iFd
== -1 || OpenInternDescriptor(Mode
, compressor
) == false)
1101 return FileFdErrno("open",_("Could not open file %s"), FileName
.c_str());
1104 SetCloseExec(iFd
,true);
1108 // FileFd::OpenDescriptor - Open a filedescriptor /*{{{*/
1109 // ---------------------------------------------------------------------
1111 bool FileFd::OpenDescriptor(int Fd
, unsigned int const Mode
, CompressMode Compress
, bool AutoClose
)
1113 std::vector
<APT::Configuration::Compressor
> const compressors
= APT::Configuration::getCompressors();
1114 std::vector
<APT::Configuration::Compressor
>::const_iterator compressor
= compressors
.begin();
1117 // compat with the old API
1118 if (Mode
== ReadOnlyGzip
&& Compress
== None
)
1123 case None
: name
= "."; break;
1124 case Gzip
: name
= "gzip"; break;
1125 case Bzip2
: name
= "bzip2"; break;
1126 case Lzma
: name
= "lzma"; break;
1127 case Xz
: name
= "xz"; break;
1130 if (AutoClose
== true && Fd
!= -1)
1132 return FileFdError("Opening Fd %d in Auto or Extension compression mode is not supported", Fd
);
1134 for (; compressor
!= compressors
.end(); ++compressor
)
1135 if (compressor
->Name
== name
)
1137 if (compressor
== compressors
.end())
1139 if (AutoClose
== true && Fd
!= -1)
1141 return FileFdError("Can't find a configured compressor %s for file %s", name
.c_str(), FileName
.c_str());
1143 return OpenDescriptor(Fd
, Mode
, *compressor
, AutoClose
);
1145 bool FileFd::OpenDescriptor(int Fd
, unsigned int const Mode
, APT::Configuration::Compressor
const &compressor
, bool AutoClose
)
1148 Flags
= (AutoClose
) ? FileFd::AutoClose
: 0;
1150 this->FileName
= "";
1151 if (OpenInternDescriptor(Mode
, compressor
) == false)
1154 (Flags
& Compressed
) == Compressed
||
1160 return FileFdError(_("Could not open file descriptor %d"), Fd
);
1164 bool FileFd::OpenInternDescriptor(unsigned int const Mode
, APT::Configuration::Compressor
const &compressor
)
1168 if (compressor
.Name
== "." || compressor
.Binary
.empty() == true)
1171 #if defined HAVE_ZLIB || defined HAVE_BZ2 || defined HAVE_LZMA
1172 // the API to open files is similar, so setup to avoid code duplicates later
1173 // and while at it ensure that we close before opening (if its a reopen)
1174 void* (*compress_open
)(int, const char *) = NULL
;
1176 /* dummy so that the rest can be 'else if's */;
1177 #define APT_COMPRESS_INIT(NAME,OPEN) \
1178 else if (compressor.Name == NAME) \
1180 compress_open = (void*(*)(int, const char *)) OPEN; \
1181 if (d != NULL) d->InternalClose(FileName); \
1184 APT_COMPRESS_INIT("gzip", gzdopen
)
1187 APT_COMPRESS_INIT("bzip2", BZ2_bzdopen
)
1190 APT_COMPRESS_INIT("xz", fdopen
)
1191 APT_COMPRESS_INIT("lzma", fdopen
)
1193 #undef APT_COMPRESS_INIT
1198 d
= new FileFdPrivate();
1200 d
->compressor
= compressor
;
1201 #if defined HAVE_ZLIB || defined HAVE_BZ2 || defined HAVE_LZMA
1202 if ((Flags
& AutoClose
) != AutoClose
&& compress_open
!= NULL
)
1204 // Need to duplicate fd here or gz/bz2 close for cleanup will close the fd as well
1205 int const internFd
= dup(iFd
);
1207 return FileFdErrno("OpenInternDescriptor", _("Could not open file descriptor %d"), iFd
);
1213 #if defined HAVE_ZLIB || defined HAVE_BZ2 || defined HAVE_LZMA
1214 if (compress_open
!= NULL
)
1216 void* compress_struct
= NULL
;
1217 if ((Mode
& ReadWrite
) == ReadWrite
)
1218 compress_struct
= compress_open(iFd
, "r+");
1219 else if ((Mode
& WriteOnly
) == WriteOnly
)
1220 compress_struct
= compress_open(iFd
, "w");
1222 compress_struct
= compress_open(iFd
, "r");
1223 if (compress_struct
== NULL
)
1227 /* dummy so that the rest can be 'else if's */;
1229 else if (compressor
.Name
== "gzip")
1230 d
->gz
= (gzFile
) compress_struct
;
1233 else if (compressor
.Name
== "bzip2")
1234 d
->bz2
= (BZFILE
*) compress_struct
;
1237 else if (compressor
.Name
== "xz" || compressor
.Name
== "lzma")
1239 uint32_t const xzlevel
= 6;
1240 uint64_t const memlimit
= UINT64_MAX
;
1241 if (d
->lzma
== NULL
)
1242 d
->lzma
= new FileFdPrivate::LZMAFILE
;
1243 d
->lzma
->file
= (FILE*) compress_struct
;
1244 d
->lzma
->stream
= LZMA_STREAM_INIT
;
1246 if ((Mode
& ReadWrite
) == ReadWrite
)
1247 return FileFdError("ReadWrite mode is not supported for file %s", FileName
.c_str());
1249 if ((Mode
& WriteOnly
) == WriteOnly
)
1251 if (compressor
.Name
== "xz")
1253 if (lzma_easy_encoder(&d
->lzma
->stream
, xzlevel
, LZMA_CHECK_CRC32
) != LZMA_OK
)
1258 lzma_options_lzma options
;
1259 lzma_lzma_preset(&options
, xzlevel
);
1260 if (lzma_alone_encoder(&d
->lzma
->stream
, &options
) != LZMA_OK
)
1263 d
->lzma
->compressing
= true;
1267 if (compressor
.Name
== "xz")
1269 if (lzma_auto_decoder(&d
->lzma
->stream
, memlimit
, 0) != LZMA_OK
)
1274 if (lzma_alone_decoder(&d
->lzma
->stream
, memlimit
) != LZMA_OK
)
1277 d
->lzma
->compressing
= false;
1281 Flags
|= Compressed
;
1286 // collect zombies here in case we reopen
1287 if (d
->compressor_pid
> 0)
1288 ExecWait(d
->compressor_pid
, "FileFdCompressor", true);
1290 if ((Mode
& ReadWrite
) == ReadWrite
)
1291 return FileFdError("ReadWrite mode is not supported for file %s", FileName
.c_str());
1293 bool const Comp
= (Mode
& WriteOnly
) == WriteOnly
;
1296 // Handle 'decompression' of empty files
1299 if (Buf
.st_size
== 0 && S_ISFIFO(Buf
.st_mode
) == false)
1302 // We don't need the file open - instead let the compressor open it
1303 // as he properly knows better how to efficiently read from 'his' file
1304 if (FileName
.empty() == false)
1311 // Create a data pipe
1312 int Pipe
[2] = {-1,-1};
1313 if (pipe(Pipe
) != 0)
1314 return FileFdErrno("pipe",_("Failed to create subprocess IPC"));
1315 for (int J
= 0; J
!= 2; J
++)
1316 SetCloseExec(Pipe
[J
],true);
1318 d
->compressed_fd
= iFd
;
1327 d
->compressor_pid
= ExecFork();
1328 if (d
->compressor_pid
== 0)
1332 dup2(d
->compressed_fd
,STDOUT_FILENO
);
1333 dup2(Pipe
[0],STDIN_FILENO
);
1337 if (d
->compressed_fd
!= -1)
1338 dup2(d
->compressed_fd
,STDIN_FILENO
);
1339 dup2(Pipe
[1],STDOUT_FILENO
);
1341 int const nullfd
= open("/dev/null", O_WRONLY
);
1344 dup2(nullfd
,STDERR_FILENO
);
1348 SetCloseExec(STDOUT_FILENO
,false);
1349 SetCloseExec(STDIN_FILENO
,false);
1351 std::vector
<char const*> Args
;
1352 Args
.push_back(compressor
.Binary
.c_str());
1353 std::vector
<std::string
> const * const addArgs
=
1354 (Comp
== true) ? &(compressor
.CompressArgs
) : &(compressor
.UncompressArgs
);
1355 for (std::vector
<std::string
>::const_iterator a
= addArgs
->begin();
1356 a
!= addArgs
->end(); ++a
)
1357 Args
.push_back(a
->c_str());
1358 if (Comp
== false && FileName
.empty() == false)
1360 // commands not needing arguments, do not need to be told about using standard output
1361 // in reality, only testcases with tools like cat, rev, rot13, … are able to trigger this
1362 if (compressor
.CompressArgs
.empty() == false && compressor
.UncompressArgs
.empty() == false)
1363 Args
.push_back("--stdout");
1364 if (TemporaryFileName
.empty() == false)
1365 Args
.push_back(TemporaryFileName
.c_str());
1367 Args
.push_back(FileName
.c_str());
1369 Args
.push_back(NULL
);
1371 execvp(Args
[0],(char **)&Args
[0]);
1372 cerr
<< _("Failed to exec compressor ") << Args
[0] << endl
;
1383 // FileFd::~File - Closes the file /*{{{*/
1384 // ---------------------------------------------------------------------
1385 /* If the proper modes are selected then we close the Fd and possibly
1386 unlink the file on error. */
1391 d
->CloseDown(FileName
);
1396 // FileFd::Read - Read a bit of the file /*{{{*/
1397 // ---------------------------------------------------------------------
1398 /* We are careful to handle interruption by a signal while reading
1400 bool FileFd::Read(void *To
,unsigned long long Size
,unsigned long long *Actual
)
1406 *((char *)To
) = '\0';
1410 /* dummy so that the rest can be 'else if's */;
1412 else if (d
!= NULL
&& d
->gz
!= NULL
)
1413 Res
= gzread(d
->gz
,To
,Size
);
1416 else if (d
!= NULL
&& d
->bz2
!= NULL
)
1417 Res
= BZ2_bzread(d
->bz2
,To
,Size
);
1420 else if (d
!= NULL
&& d
->lzma
!= NULL
)
1422 if (d
->lzma
->eof
== true)
1425 d
->lzma
->stream
.next_out
= (uint8_t *) To
;
1426 d
->lzma
->stream
.avail_out
= Size
;
1427 if (d
->lzma
->stream
.avail_in
== 0)
1429 d
->lzma
->stream
.next_in
= d
->lzma
->buffer
;
1430 d
->lzma
->stream
.avail_in
= fread(d
->lzma
->buffer
, 1, sizeof(d
->lzma
->buffer
)/sizeof(d
->lzma
->buffer
[0]), d
->lzma
->file
);
1432 d
->lzma
->err
= lzma_code(&d
->lzma
->stream
, LZMA_RUN
);
1433 if (d
->lzma
->err
== LZMA_STREAM_END
)
1435 d
->lzma
->eof
= true;
1436 Res
= Size
- d
->lzma
->stream
.avail_out
;
1438 else if (d
->lzma
->err
!= LZMA_OK
)
1445 Res
= Size
- d
->lzma
->stream
.avail_out
;
1448 // lzma run was okay, but produced no output…
1456 Res
= read(iFd
,To
,Size
);
1462 // trick the while-loop into running again
1468 /* dummy so that the rest can be 'else if's */;
1470 else if (d
!= NULL
&& d
->gz
!= NULL
)
1473 char const * const errmsg
= gzerror(d
->gz
, &err
);
1475 return FileFdError("gzread: %s (%d: %s)", _("Read error"), err
, errmsg
);
1479 else if (d
!= NULL
&& d
->bz2
!= NULL
)
1482 char const * const errmsg
= BZ2_bzerror(d
->bz2
, &err
);
1483 if (err
!= BZ_IO_ERROR
)
1484 return FileFdError("BZ2_bzread: %s (%d: %s)", _("Read error"), err
, errmsg
);
1488 else if (d
!= NULL
&& d
->lzma
!= NULL
)
1489 return FileFdError("lzma_read: %s (%d)", _("Read error"), d
->lzma
->err
);
1491 return FileFdErrno("read",_("Read error"));
1494 To
= (char *)To
+ Res
;
1501 while (Res
> 0 && Size
> 0);
1513 return FileFdError(_("read, still have %llu to read but none left"), Size
);
1516 // FileFd::ReadLine - Read a complete line from the file /*{{{*/
1517 // ---------------------------------------------------------------------
1518 /* Beware: This method can be quiet slow for big buffers on UNcompressed
1519 files because of the naive implementation! */
1520 char* FileFd::ReadLine(char *To
, unsigned long long const Size
)
1524 if (d
!= NULL
&& d
->gz
!= NULL
)
1525 return gzgets(d
->gz
, To
, Size
);
1528 unsigned long long read
= 0;
1529 while ((Size
- 1) != read
)
1531 unsigned long long done
= 0;
1532 if (Read(To
+ read
, 1, &done
) == false)
1536 if (To
[read
++] == '\n')
1545 // FileFd::Write - Write to the file /*{{{*/
1546 // ---------------------------------------------------------------------
1548 bool FileFd::Write(const void *From
,unsigned long long Size
)
1555 /* dummy so that the rest can be 'else if's */;
1557 else if (d
!= NULL
&& d
->gz
!= NULL
)
1558 Res
= gzwrite(d
->gz
,From
,Size
);
1561 else if (d
!= NULL
&& d
->bz2
!= NULL
)
1562 Res
= BZ2_bzwrite(d
->bz2
,(void*)From
,Size
);
1565 else if (d
!= NULL
&& d
->lzma
!= NULL
)
1567 d
->lzma
->stream
.next_in
= (uint8_t *)From
;
1568 d
->lzma
->stream
.avail_in
= Size
;
1569 d
->lzma
->stream
.next_out
= d
->lzma
->buffer
;
1570 d
->lzma
->stream
.avail_out
= sizeof(d
->lzma
->buffer
)/sizeof(d
->lzma
->buffer
[0]);
1571 d
->lzma
->err
= lzma_code(&d
->lzma
->stream
, LZMA_RUN
);
1572 if (d
->lzma
->err
!= LZMA_OK
)
1574 size_t const n
= sizeof(d
->lzma
->buffer
)/sizeof(d
->lzma
->buffer
[0]) - d
->lzma
->stream
.avail_out
;
1575 size_t const m
= (n
== 0) ? 0 : fwrite(d
->lzma
->buffer
, 1, n
, d
->lzma
->file
);
1579 Res
= Size
- d
->lzma
->stream
.avail_in
;
1583 Res
= write(iFd
,From
,Size
);
1585 if (Res
< 0 && errno
== EINTR
)
1590 /* dummy so that the rest can be 'else if's */;
1592 else if (d
!= NULL
&& d
->gz
!= NULL
)
1595 char const * const errmsg
= gzerror(d
->gz
, &err
);
1597 return FileFdError("gzwrite: %s (%d: %s)", _("Write error"), err
, errmsg
);
1601 else if (d
!= NULL
&& d
->bz2
!= NULL
)
1604 char const * const errmsg
= BZ2_bzerror(d
->bz2
, &err
);
1605 if (err
!= BZ_IO_ERROR
)
1606 return FileFdError("BZ2_bzwrite: %s (%d: %s)", _("Write error"), err
, errmsg
);
1610 else if (d
!= NULL
&& d
->lzma
!= NULL
)
1611 return FileFdErrno("lzma_fwrite", _("Write error"));
1613 return FileFdErrno("write",_("Write error"));
1616 From
= (char const *)From
+ Res
;
1621 while (Res
> 0 && Size
> 0);
1626 return FileFdError(_("write, still have %llu to write but couldn't"), Size
);
1628 bool FileFd::Write(int Fd
, const void *From
, unsigned long long Size
)
1634 Res
= write(Fd
,From
,Size
);
1635 if (Res
< 0 && errno
== EINTR
)
1638 return _error
->Errno("write",_("Write error"));
1640 From
= (char const *)From
+ Res
;
1643 while (Res
> 0 && Size
> 0);
1648 return _error
->Error(_("write, still have %llu to write but couldn't"), Size
);
1651 // FileFd::Seek - Seek in the file /*{{{*/
1652 // ---------------------------------------------------------------------
1654 bool FileFd::Seek(unsigned long long To
)
1658 if (d
!= NULL
&& (d
->pipe
== true || d
->InternalStream() == true))
1660 // Our poor man seeking in pipes is costly, so try to avoid it
1661 unsigned long long seekpos
= Tell();
1664 else if (seekpos
< To
)
1665 return Skip(To
- seekpos
);
1667 if ((d
->openmode
& ReadOnly
) != ReadOnly
)
1668 return FileFdError("Reopen is only implemented for read-only files!");
1669 d
->InternalClose(FileName
);
1673 if (TemporaryFileName
.empty() == false)
1674 iFd
= open(TemporaryFileName
.c_str(), O_RDONLY
);
1675 else if (FileName
.empty() == false)
1676 iFd
= open(FileName
.c_str(), O_RDONLY
);
1679 if (d
->compressed_fd
> 0)
1680 if (lseek(d
->compressed_fd
, 0, SEEK_SET
) != 0)
1681 iFd
= d
->compressed_fd
;
1683 return FileFdError("Reopen is not implemented for pipes opened with FileFd::OpenDescriptor()!");
1686 if (OpenInternDescriptor(d
->openmode
, d
->compressor
) == false)
1687 return FileFdError("Seek on file %s because it couldn't be reopened", FileName
.c_str());
1697 if (d
!= NULL
&& d
->gz
)
1698 res
= gzseek(d
->gz
,To
,SEEK_SET
);
1701 res
= lseek(iFd
,To
,SEEK_SET
);
1702 if (res
!= (off_t
)To
)
1703 return FileFdError("Unable to seek to %llu", To
);
1710 // FileFd::Skip - Seek in the file /*{{{*/
1711 // ---------------------------------------------------------------------
1713 bool FileFd::Skip(unsigned long long Over
)
1715 if (d
!= NULL
&& (d
->pipe
== true || d
->InternalStream() == true))
1720 unsigned long long toread
= std::min((unsigned long long) sizeof(buffer
), Over
);
1721 if (Read(buffer
, toread
) == false)
1722 return FileFdError("Unable to seek ahead %llu",Over
);
1730 if (d
!= NULL
&& d
->gz
!= NULL
)
1731 res
= gzseek(d
->gz
,Over
,SEEK_CUR
);
1734 res
= lseek(iFd
,Over
,SEEK_CUR
);
1736 return FileFdError("Unable to seek ahead %llu",Over
);
1743 // FileFd::Truncate - Truncate the file /*{{{*/
1744 // ---------------------------------------------------------------------
1746 bool FileFd::Truncate(unsigned long long To
)
1748 // truncating /dev/null is always successful - as we get an error otherwise
1749 if (To
== 0 && FileName
== "/dev/null")
1751 #if defined HAVE_ZLIB || defined HAVE_BZ2 || defined HAVE_LZMA
1752 if (d
!= NULL
&& (d
->InternalStream() == true
1757 return FileFdError("Truncating compressed files is not implemented (%s)", FileName
.c_str());
1759 if (ftruncate(iFd
,To
) != 0)
1760 return FileFdError("Unable to truncate to %llu",To
);
1765 // FileFd::Tell - Current seek position /*{{{*/
1766 // ---------------------------------------------------------------------
1768 unsigned long long FileFd::Tell()
1770 // In theory, we could just return seekpos here always instead of
1771 // seeking around, but not all users of FileFd use always Seek() and co
1772 // so d->seekpos isn't always true and we can just use it as a hint if
1773 // we have nothing else, but not always as an authority…
1774 if (d
!= NULL
&& (d
->pipe
== true || d
->InternalStream() == true))
1779 if (d
!= NULL
&& d
->gz
!= NULL
)
1780 Res
= gztell(d
->gz
);
1783 Res
= lseek(iFd
,0,SEEK_CUR
);
1784 if (Res
== (off_t
)-1)
1785 FileFdErrno("lseek","Failed to determine the current file position");
1791 static bool StatFileFd(char const * const msg
, int const iFd
, std::string
const &FileName
, struct stat
&Buf
, FileFdPrivate
* const d
) /*{{{*/
1793 bool ispipe
= (d
!= NULL
&& d
->pipe
== true);
1794 if (ispipe
== false)
1796 if (fstat(iFd
,&Buf
) != 0)
1797 // higher-level code will generate more meaningful messages,
1798 // even translated this would be meaningless for users
1799 return _error
->Errno("fstat", "Unable to determine %s for fd %i", msg
, iFd
);
1800 ispipe
= S_ISFIFO(Buf
.st_mode
);
1803 // for compressor pipes st_size is undefined and at 'best' zero
1806 // we set it here, too, as we get the info here for free
1807 // in theory the Open-methods should take care of it already
1810 if (stat(FileName
.c_str(), &Buf
) != 0)
1811 return _error
->Errno("fstat", "Unable to determine %s for file %s", msg
, FileName
.c_str());
1816 // FileFd::FileSize - Return the size of the file /*{{{*/
1817 unsigned long long FileFd::FileSize()
1820 if (StatFileFd("file size", iFd
, FileName
, Buf
, d
) == false)
1828 // FileFd::ModificationTime - Return the time of last touch /*{{{*/
1829 time_t FileFd::ModificationTime()
1832 if (StatFileFd("modification time", iFd
, FileName
, Buf
, d
) == false)
1837 return Buf
.st_mtime
;
1840 // FileFd::Size - Return the size of the content in the file /*{{{*/
1841 // ---------------------------------------------------------------------
1843 unsigned long long FileFd::Size()
1845 unsigned long long size
= FileSize();
1847 // for compressor pipes st_size is undefined and at 'best' zero,
1848 // so we 'read' the content and 'seek' back - see there
1849 if (d
!= NULL
&& (d
->pipe
== true || (d
->InternalStream() == true && size
> 0)))
1851 unsigned long long const oldSeek
= Tell();
1853 unsigned long long read
= 0;
1855 if (Read(ignore
, sizeof(ignore
), &read
) == false)
1865 // only check gzsize if we are actually a gzip file, just checking for
1866 // "gz" is not sufficient as uncompressed files could be opened with
1867 // gzopen in "direct" mode as well
1868 else if (d
!= NULL
&& d
->gz
&& !gzdirect(d
->gz
) && size
> 0)
1870 off_t
const oldPos
= lseek(iFd
,0,SEEK_CUR
);
1871 /* unfortunately zlib.h doesn't provide a gzsize(), so we have to do
1872 * this ourselves; the original (uncompressed) file size is the last 32
1873 * bits of the file */
1874 // FIXME: Size for gz-files is limited by 32bit… no largefile support
1875 if (lseek(iFd
, -4, SEEK_END
) < 0)
1877 FileFdErrno("lseek","Unable to seek to end of gzipped file");
1881 if (read(iFd
, &size
, 4) != 4)
1883 FileFdErrno("read","Unable to read original size of gzipped file");
1886 size
= le32toh(size
);
1888 if (lseek(iFd
, oldPos
, SEEK_SET
) < 0)
1890 FileFdErrno("lseek","Unable to seek in gzipped file");
1901 // FileFd::Close - Close the file if the close flag is set /*{{{*/
1902 // ---------------------------------------------------------------------
1904 bool FileFd::Close()
1910 if ((Flags
& AutoClose
) == AutoClose
)
1912 if ((Flags
& Compressed
) != Compressed
&& iFd
> 0 && close(iFd
) != 0)
1913 Res
&= _error
->Errno("close",_("Problem closing the file %s"), FileName
.c_str());
1917 Res
&= d
->CloseDown(FileName
);
1923 if ((Flags
& Replace
) == Replace
) {
1924 if (rename(TemporaryFileName
.c_str(), FileName
.c_str()) != 0)
1925 Res
&= _error
->Errno("rename",_("Problem renaming the file %s to %s"), TemporaryFileName
.c_str(), FileName
.c_str());
1927 FileName
= TemporaryFileName
; // for the unlink() below.
1928 TemporaryFileName
.clear();
1933 if ((Flags
& Fail
) == Fail
&& (Flags
& DelOnFail
) == DelOnFail
&&
1934 FileName
.empty() == false)
1935 if (unlink(FileName
.c_str()) != 0)
1936 Res
&= _error
->WarningE("unlnk",_("Problem unlinking the file %s"), FileName
.c_str());
1943 // FileFd::Sync - Sync the file /*{{{*/
1944 // ---------------------------------------------------------------------
1948 if (fsync(iFd
) != 0)
1949 return FileFdErrno("sync",_("Problem syncing the file"));
1953 // FileFd::FileFdErrno - set Fail and call _error->Errno *{{{*/
1954 bool FileFd::FileFdErrno(const char *Function
, const char *Description
,...)
1958 size_t msgSize
= 400;
1959 int const errsv
= errno
;
1962 va_start(args
,Description
);
1963 if (_error
->InsertErrno(GlobalError::ERROR
, Function
, Description
, args
, errsv
, msgSize
) == false)
1970 // FileFd::FileFdError - set Fail and call _error->Error *{{{*/
1971 bool FileFd::FileFdError(const char *Description
,...) {
1974 size_t msgSize
= 400;
1977 va_start(args
,Description
);
1978 if (_error
->Insert(GlobalError::ERROR
, Description
, args
, msgSize
) == false)
1986 APT_DEPRECATED gzFile
FileFd::gzFd() {
1995 // Glob - wrapper around "glob()" /*{{{*/
1996 // ---------------------------------------------------------------------
1998 std::vector
<std::string
> Glob(std::string
const &pattern
, int flags
)
2000 std::vector
<std::string
> result
;
2005 glob_res
= glob(pattern
.c_str(), flags
, NULL
, &globbuf
);
2009 if(glob_res
!= GLOB_NOMATCH
) {
2010 _error
->Errno("glob", "Problem with glob");
2016 for(i
=0;i
<globbuf
.gl_pathc
;i
++)
2017 result
.push_back(string(globbuf
.gl_pathv
[i
]));
2024 std::string
GetTempDir()
2026 const char *tmpdir
= getenv("TMPDIR");
2033 // check that tmpdir is set and exists
2035 if (!tmpdir
|| strlen(tmpdir
) == 0 || stat(tmpdir
, &st
) != 0)
2038 return string(tmpdir
);
2041 bool Rename(std::string From
, std::string To
)
2043 if (rename(From
.c_str(),To
.c_str()) != 0)
2045 _error
->Error(_("rename failed, %s (%s -> %s)."),strerror(errno
),
2046 From
.c_str(),To
.c_str());