4 // Copyright (c) 2000-2005 Per Liden
6 // This program is free software; you can redistribute it and/or modify
7 // it under the terms of the GNU General Public License as published by
8 // the Free Software Foundation; either version 2 of the License, or
9 // (at your option) any later version.
11 // This program is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
16 // You should have received a copy of the GNU General Public License
17 // along with this program; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
31 #include <ext/stdio_filebuf.h>
34 #include <sys/types.h>
38 #include <sys/param.h>
43 #include <archive_entry.h>
45 using __gnu_cxx::stdio_filebuf
;
47 pkgutil::pkgutil(const string
& name
)
52 memset(&sa
, 0, sizeof(sa
));
53 sa
.sa_handler
= SIG_IGN
;
54 sigaction(SIGHUP
, &sa
, 0);
55 sigaction(SIGINT
, &sa
, 0);
56 sigaction(SIGQUIT
, &sa
, 0);
57 sigaction(SIGTERM
, &sa
, 0);
60 void pkgutil::db_open(const string
& path
)
63 root
= trim_filename(path
+ "/");
64 const string filename
= root
+ PKG_DB
;
66 int fd
= open(filename
.c_str(), O_RDONLY
);
68 throw runtime_error_with_errno("could not open " + filename
);
70 stdio_filebuf
<char> filebuf(fd
, ios::in
, getpagesize());
73 throw runtime_error_with_errno("could not read " + filename
);
80 getline(in
, info
.version
);
86 break; // End of record
88 info
.files
.insert(info
.files
.end(), file
);
90 if (!info
.files
.empty())
91 packages
[name
] = info
;
95 cerr
<< packages
.size() << " packages found in database" << endl
;
99 void pkgutil::db_commit()
101 const string dbfilename
= root
+ PKG_DB
;
102 const string dbfilename_new
= dbfilename
+ ".incomplete_transaction";
103 const string dbfilename_bak
= dbfilename
+ ".backup";
105 // Remove failed transaction (if it exists)
106 if (unlink(dbfilename_new
.c_str()) == -1 && errno
!= ENOENT
)
107 throw runtime_error_with_errno("could not remove " + dbfilename_new
);
109 // Write new database
110 int fd_new
= creat(dbfilename_new
.c_str(), 0444);
112 throw runtime_error_with_errno("could not create " + dbfilename_new
);
114 stdio_filebuf
<char> filebuf_new(fd_new
, ios::out
, getpagesize());
115 ostream
db_new(&filebuf_new
);
116 for (packages_t::const_iterator i
= packages
.begin(); i
!= packages
.end(); ++i
) {
117 if (!i
->second
.files
.empty()) {
118 db_new
<< i
->first
<< "\n";
119 db_new
<< i
->second
.version
<< "\n";
120 copy(i
->second
.files
.begin(), i
->second
.files
.end(), ostream_iterator
<string
>(db_new
, "\n"));
127 // Make sure the new database was successfully written
129 throw runtime_error("could not write " + dbfilename_new
);
131 // Synchronize file to disk
132 if (fsync(fd_new
) == -1)
133 throw runtime_error_with_errno("could not synchronize " + dbfilename_new
);
135 // Relink database backup
136 if (unlink(dbfilename_bak
.c_str()) == -1 && errno
!= ENOENT
)
137 throw runtime_error_with_errno("could not remove " + dbfilename_bak
);
138 if (link(dbfilename
.c_str(), dbfilename_bak
.c_str()) == -1)
139 throw runtime_error_with_errno("could not create " + dbfilename_bak
);
141 // Move new database into place
142 if (rename(dbfilename_new
.c_str(), dbfilename
.c_str()) == -1)
143 throw runtime_error_with_errno("could not rename " + dbfilename_new
+ " to " + dbfilename
);
146 cerr
<< packages
.size() << " packages written to database" << endl
;
150 void pkgutil::db_add_pkg(const string
& name
, const pkginfo_t
& info
)
152 packages
[name
] = info
;
155 bool pkgutil::db_find_pkg(const string
& name
)
157 return (packages
.find(name
) != packages
.end());
160 void pkgutil::db_rm_pkg(const string
& name
)
162 set
<string
> files
= packages
[name
].files
;
163 packages
.erase(name
);
166 cerr
<< "Removing package phase 1 (all files in package):" << endl
;
167 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
171 // Don't delete files that still have references
172 for (packages_t::const_iterator i
= packages
.begin(); i
!= packages
.end(); ++i
)
173 for (set
<string
>::const_iterator j
= i
->second
.files
.begin(); j
!= i
->second
.files
.end(); ++j
)
177 cerr
<< "Removing package phase 2 (files that still have references excluded):" << endl
;
178 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
183 for (set
<string
>::const_reverse_iterator i
= files
.rbegin(); i
!= files
.rend(); ++i
) {
184 const string filename
= root
+ *i
;
185 if (file_exists(filename
) && remove(filename
.c_str()) == -1) {
186 const char* msg
= strerror(errno
);
187 cerr
<< utilname
<< ": could not remove " << filename
<< ": " << msg
<< endl
;
192 void pkgutil::db_rm_pkg(const string
& name
, const set
<string
>& keep_list
)
194 set
<string
> files
= packages
[name
].files
;
195 packages
.erase(name
);
198 cerr
<< "Removing package phase 1 (all files in package):" << endl
;
199 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
203 // Don't delete files found in the keep list
204 for (set
<string
>::const_iterator i
= keep_list
.begin(); i
!= keep_list
.end(); ++i
)
208 cerr
<< "Removing package phase 2 (files that is in the keep list excluded):" << endl
;
209 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
213 // Don't delete files that still have references
214 for (packages_t::const_iterator i
= packages
.begin(); i
!= packages
.end(); ++i
)
215 for (set
<string
>::const_iterator j
= i
->second
.files
.begin(); j
!= i
->second
.files
.end(); ++j
)
219 cerr
<< "Removing package phase 3 (files that still have references excluded):" << endl
;
220 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
225 for (set
<string
>::const_reverse_iterator i
= files
.rbegin(); i
!= files
.rend(); ++i
) {
226 const string filename
= root
+ *i
;
227 if (file_exists(filename
) && remove(filename
.c_str()) == -1) {
228 if (errno
== ENOTEMPTY
)
230 const char* msg
= strerror(errno
);
231 cerr
<< utilname
<< ": could not remove " << filename
<< ": " << msg
<< endl
;
236 void pkgutil::db_rm_files(set
<string
> files
, const set
<string
>& keep_list
)
238 // Remove all references
239 for (packages_t::iterator i
= packages
.begin(); i
!= packages
.end(); ++i
)
240 for (set
<string
>::const_iterator j
= files
.begin(); j
!= files
.end(); ++j
)
241 i
->second
.files
.erase(*j
);
244 cerr
<< "Removing files:" << endl
;
245 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
249 // Don't delete files found in the keep list
250 for (set
<string
>::const_iterator i
= keep_list
.begin(); i
!= keep_list
.end(); ++i
)
254 for (set
<string
>::const_reverse_iterator i
= files
.rbegin(); i
!= files
.rend(); ++i
) {
255 const string filename
= root
+ *i
;
256 if (file_exists(filename
) && remove(filename
.c_str()) == -1) {
257 if (errno
== ENOTEMPTY
)
259 const char* msg
= strerror(errno
);
260 cerr
<< utilname
<< ": could not remove " << filename
<< ": " << msg
<< endl
;
265 set
<string
> pkgutil::db_find_conflicts(const string
& name
, const pkginfo_t
& info
)
269 // Find conflicting files in database
270 for (packages_t::const_iterator i
= packages
.begin(); i
!= packages
.end(); ++i
) {
271 if (i
->first
!= name
) {
272 set_intersection(info
.files
.begin(), info
.files
.end(),
273 i
->second
.files
.begin(), i
->second
.files
.end(),
274 inserter(files
, files
.end()));
279 cerr
<< "Conflicts phase 1 (conflicts in database):" << endl
;
280 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
284 // Find conflicting files in filesystem
285 for (set
<string
>::iterator i
= info
.files
.begin(); i
!= info
.files
.end(); ++i
) {
286 const string filename
= root
+ *i
;
287 if (file_exists(filename
) && files
.find(*i
) == files
.end())
288 files
.insert(files
.end(), *i
);
292 cerr
<< "Conflicts phase 2 (conflicts in filesystem added):" << endl
;
293 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
297 // Exclude directories
298 set
<string
> tmp
= files
;
299 for (set
<string
>::const_iterator i
= tmp
.begin(); i
!= tmp
.end(); ++i
) {
300 if ((*i
)[i
->length() - 1] == '/')
305 cerr
<< "Conflicts phase 3 (directories excluded):" << endl
;
306 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
310 // If this is an upgrade, remove files already owned by this package
311 if (packages
.find(name
) != packages
.end()) {
312 for (set
<string
>::const_iterator i
= packages
[name
].files
.begin(); i
!= packages
[name
].files
.end(); ++i
)
316 cerr
<< "Conflicts phase 4 (files already owned by this package excluded):" << endl
;
317 copy(files
.begin(), files
.end(), ostream_iterator
<string
>(cerr
, "\n"));
325 pair
<string
, pkgutil::pkginfo_t
> pkgutil::pkg_open(const string
& filename
) const
327 pair
<string
, pkginfo_t
> result
;
329 struct archive
* archive
;
330 struct archive_entry
* entry
;
332 // Extract name and version from filename
333 string
basename(filename
, filename
.rfind('/') + 1);
334 string
name(basename
, 0, basename
.find(VERSION_DELIM
));
335 string
version(basename
, 0, basename
.rfind(PKG_EXT
));
336 version
.erase(0, version
.find(VERSION_DELIM
) == string::npos
? string::npos
: version
.find(VERSION_DELIM
) + 1);
338 if (name
.empty() || version
.empty())
339 throw runtime_error("could not determine name and/or version of " + basename
+ ": Invalid package name");
342 result
.second
.version
= version
;
344 archive
= archive_read_new();
345 archive_read_support_compression_all(archive
);
346 archive_read_support_format_all(archive
);
348 if (archive_read_open_filename(archive
,
349 const_cast<char*>(filename
.c_str()),
350 ARCHIVE_DEFAULT_BYTES_PER_BLOCK
) != ARCHIVE_OK
)
351 throw runtime_error_with_errno("could not open " + filename
, archive_errno(archive
));
353 for (i
= 0; archive_read_next_header(archive
, &entry
) ==
356 result
.second
.files
.insert(result
.second
.files
.end(),
357 archive_entry_pathname(entry
));
359 mode_t mode
= archive_entry_mode(entry
);
362 archive_read_data_skip(archive
) != ARCHIVE_OK
)
363 throw runtime_error_with_errno("could not read " + filename
, archive_errno(archive
));
367 if (archive_errno(archive
) == 0)
368 throw runtime_error("empty package");
370 throw runtime_error("could not read " + filename
);
373 archive_read_finish(archive
);
378 void pkgutil::pkg_install(const string
& filename
, const set
<string
>& keep_list
, const set
<string
>& non_install_list
) const
380 struct archive
* archive
;
381 struct archive_entry
* entry
;
384 archive
= archive_read_new();
385 archive_read_support_compression_all(archive
);
386 archive_read_support_format_all(archive
);
388 if (archive_read_open_filename(archive
,
389 const_cast<char*>(filename
.c_str()),
390 ARCHIVE_DEFAULT_BYTES_PER_BLOCK
) != ARCHIVE_OK
)
391 throw runtime_error_with_errno("could not open " + filename
, archive_errno(archive
));
395 for (i
= 0; archive_read_next_header(archive
, &entry
) ==
397 string archive_filename
= archive_entry_pathname(entry
);
398 string reject_dir
= trim_filename(root
+ string("/") + string(PKG_REJECTED
));
399 string original_filename
= trim_filename(root
+ string("/") + archive_filename
);
400 string real_filename
= original_filename
;
402 // Check if file is filtered out via INSTALL
403 if (non_install_list
.find(archive_filename
) != non_install_list
.end()) {
406 cout
<< utilname
<< ": ignoring " << archive_filename
<< endl
;
408 mode
= archive_entry_mode(entry
);
411 archive_read_data_skip(archive
);
416 // Check if file should be rejected
417 if (file_exists(real_filename
) && keep_list
.find(archive_filename
) != keep_list
.end())
418 real_filename
= trim_filename(reject_dir
+ string("/") + archive_filename
);
420 archive_entry_set_pathname(entry
, const_cast<char*>
421 (real_filename
.c_str()));
424 unsigned int flags
= ARCHIVE_EXTRACT_OWNER
| ARCHIVE_EXTRACT_PERM
| ARCHIVE_EXTRACT_UNLINK
;
426 if (archive_read_extract(archive
, entry
, flags
) != ARCHIVE_OK
) {
427 // If a file fails to install we just print an error message and
428 // continue trying to install the rest of the package.
429 const char* msg
= archive_error_string(archive
);
430 cerr
<< utilname
<< ": could not install " + archive_filename
<< ": " << msg
<< endl
;
434 // Check rejected file
435 if (real_filename
!= original_filename
) {
436 bool remove_file
= false;
437 mode_t mode
= archive_entry_mode(entry
);
441 remove_file
= permissions_equal(real_filename
, original_filename
);
444 remove_file
= permissions_equal(real_filename
, original_filename
) &&
445 (file_empty(real_filename
) || file_equal(real_filename
, original_filename
));
447 // Remove rejected file or signal about its existence
449 file_remove(reject_dir
, real_filename
);
451 cout
<< utilname
<< ": rejecting " << archive_filename
<< ", keeping existing version" << endl
;
456 if (archive_errno(archive
) == 0)
457 throw runtime_error("empty package");
459 throw runtime_error("could not read " + filename
);
462 archive_read_finish(archive
);
465 void pkgutil::ldconfig() const
467 // Only execute ldconfig if /etc/ld.so.conf exists
468 if (file_exists(root
+ LDCONFIG_CONF
)) {
472 throw runtime_error_with_errno("fork() failed");
475 execl(LDCONFIG
, LDCONFIG
, "-r", root
.c_str(), (char *) 0);
476 const char* msg
= strerror(errno
);
477 cerr
<< utilname
<< ": could not execute " << LDCONFIG
<< ": " << msg
<< endl
;
480 if (waitpid(pid
, 0, 0) == -1)
481 throw runtime_error_with_errno("waitpid() failed");
486 void pkgutil::pkg_footprint(string
& filename
) const
489 struct archive
* archive
;
490 struct archive_entry
* entry
;
492 map
<string
, mode_t
> hardlink_target_modes
;
494 // We first do a run over the archive and remember the modes
496 // In the second run, we print the footprint - using the stored
497 // modes for hardlinks.
499 // FIXME the code duplication here is butt ugly
500 archive
= archive_read_new();
501 archive_read_support_compression_all(archive
);
502 archive_read_support_format_all(archive
);
504 if (archive_read_open_filename(archive
,
505 const_cast<char*>(filename
.c_str()),
506 ARCHIVE_DEFAULT_BYTES_PER_BLOCK
) != ARCHIVE_OK
)
507 throw runtime_error_with_errno("could not open " + filename
, archive_errno(archive
));
509 for (i
= 0; archive_read_next_header(archive
, &entry
) ==
512 mode_t mode
= archive_entry_mode(entry
);
514 if (!archive_entry_hardlink(entry
)) {
515 const char *s
= archive_entry_pathname(entry
);
517 hardlink_target_modes
[s
] = mode
;
520 if (S_ISREG(mode
) && archive_read_data_skip(archive
))
521 throw runtime_error_with_errno("could not read " + filename
, archive_errno(archive
));
524 archive_read_finish(archive
);
526 // Too bad, there doesn't seem to be a way to reuse our archive
528 archive
= archive_read_new();
529 archive_read_support_compression_all(archive
);
530 archive_read_support_format_all(archive
);
532 if (archive_read_open_filename(archive
,
533 const_cast<char*>(filename
.c_str()),
534 ARCHIVE_DEFAULT_BYTES_PER_BLOCK
) != ARCHIVE_OK
)
535 throw runtime_error_with_errno("could not open " + filename
, archive_errno(archive
));
537 for (i
= 0; archive_read_next_header(archive
, &entry
) ==
539 mode_t mode
= archive_entry_mode(entry
);
541 // Access permissions
543 // Access permissions on symlinks differ among filesystems, e.g. XFS and ext2 have different.
544 // To avoid getting different footprints we always use "lrwxrwxrwx".
545 cout
<< "lrwxrwxrwx";
547 const char *h
= archive_entry_hardlink(entry
);
550 cout
<< mtos(hardlink_target_modes
[h
]);
558 uid_t uid
= archive_entry_uid(entry
);
559 struct passwd
* pw
= getpwuid(uid
);
568 gid_t gid
= archive_entry_gid(entry
);
569 struct group
* gr
= getgrgid(gid
);
576 cout
<< '\t' << archive_entry_pathname(entry
);
581 cout
<< " -> " << archive_entry_symlink(entry
);
582 } else if (S_ISCHR(mode
) ||
585 cout
<< " (" << archive_entry_rdevmajor(entry
)
586 << ", " << archive_entry_rdevminor(entry
)
588 } else if (S_ISREG(mode
) &&
589 archive_entry_size(entry
) == 0) {
590 // Empty regular file
596 if (S_ISREG(mode
) && archive_read_data_skip(archive
))
597 throw runtime_error_with_errno("could not read " + filename
, archive_errno(archive
));
601 if (archive_errno(archive
) == 0)
602 throw runtime_error("empty package");
604 throw runtime_error("could not read " + filename
);
607 archive_read_finish(archive
);
610 void pkgutil::print_version() const
612 cout
<< utilname
<< " (pkgutils) " << VERSION
<< endl
;
615 db_lock::db_lock(const string
& root
, bool exclusive
)
618 const string dirname
= trim_filename(root
+ string("/") + PKG_DIR
);
620 if (!(dir
= opendir(dirname
.c_str())))
621 throw runtime_error_with_errno("could not read directory " + dirname
);
623 if (flock(dirfd(dir
), (exclusive
? LOCK_EX
: LOCK_SH
) | LOCK_NB
) == -1) {
624 if (errno
== EWOULDBLOCK
)
625 throw runtime_error("package database is currently locked by another process");
627 throw runtime_error_with_errno("could not lock directory " + dirname
);
634 flock(dirfd(dir
), LOCK_UN
);
639 void assert_argument(char** argv
, int argc
, int index
)
641 if (argc
- 1 < index
+ 1)
642 throw runtime_error("option " + string(argv
[index
]) + " requires an argument");
645 string
itos(unsigned int value
)
648 sprintf(buf
, "%u", value
);
652 string
mtos(mode_t mode
)
657 switch (mode
& S_IFMT
) {
658 case S_IFREG
: s
+= '-'; break; // Regular
659 case S_IFDIR
: s
+= 'd'; break; // Directory
660 case S_IFLNK
: s
+= 'l'; break; // Symbolic link
661 case S_IFCHR
: s
+= 'c'; break; // Character special
662 case S_IFBLK
: s
+= 'b'; break; // Block special
663 case S_IFSOCK
: s
+= 's'; break; // Socket
664 case S_IFIFO
: s
+= 'p'; break; // Fifo
665 default: s
+= '?'; break; // Unknown
669 s
+= (mode
& S_IRUSR
) ? 'r' : '-';
670 s
+= (mode
& S_IWUSR
) ? 'w' : '-';
671 switch (mode
& (S_IXUSR
| S_ISUID
)) {
672 case S_IXUSR
: s
+= 'x'; break;
673 case S_ISUID
: s
+= 'S'; break;
674 case S_IXUSR
| S_ISUID
: s
+= 's'; break;
675 default: s
+= '-'; break;
679 s
+= (mode
& S_IRGRP
) ? 'r' : '-';
680 s
+= (mode
& S_IWGRP
) ? 'w' : '-';
681 switch (mode
& (S_IXGRP
| S_ISGID
)) {
682 case S_IXGRP
: s
+= 'x'; break;
683 case S_ISGID
: s
+= 'S'; break;
684 case S_IXGRP
| S_ISGID
: s
+= 's'; break;
685 default: s
+= '-'; break;
689 s
+= (mode
& S_IROTH
) ? 'r' : '-';
690 s
+= (mode
& S_IWOTH
) ? 'w' : '-';
691 switch (mode
& (S_IXOTH
| S_ISVTX
)) {
692 case S_IXOTH
: s
+= 'x'; break;
693 case S_ISVTX
: s
+= 'T'; break;
694 case S_IXOTH
| S_ISVTX
: s
+= 't'; break;
695 default: s
+= '-'; break;
701 string
trim_filename(const string
& filename
)
704 string result
= filename
;
706 for (string::size_type pos
= result
.find(search
); pos
!= string::npos
; pos
= result
.find(search
))
707 result
.replace(pos
, search
.size(), "/");
712 bool file_exists(const string
& filename
)
715 return !lstat(filename
.c_str(), &buf
);
718 bool file_empty(const string
& filename
)
722 if (lstat(filename
.c_str(), &buf
) == -1)
725 return (S_ISREG(buf
.st_mode
) && buf
.st_size
== 0);
728 bool file_equal(const string
& file1
, const string
& file2
)
730 struct stat buf1
, buf2
;
732 if (lstat(file1
.c_str(), &buf1
) == -1)
735 if (lstat(file2
.c_str(), &buf2
) == -1)
739 if (S_ISREG(buf1
.st_mode
) && S_ISREG(buf2
.st_mode
)) {
740 ifstream
f1(file1
.c_str());
741 ifstream
f2(file2
.c_str());
749 f1
.read(buffer1
, 4096);
750 f2
.read(buffer2
, 4096);
751 if (f1
.gcount() != f2
.gcount() ||
752 memcmp(buffer1
, buffer2
, f1
.gcount()) ||
753 f1
.eof() != f2
.eof())
760 else if (S_ISLNK(buf1
.st_mode
) && S_ISLNK(buf2
.st_mode
)) {
761 char symlink1
[MAXPATHLEN
];
762 char symlink2
[MAXPATHLEN
];
764 memset(symlink1
, 0, MAXPATHLEN
);
765 memset(symlink2
, 0, MAXPATHLEN
);
767 if (readlink(file1
.c_str(), symlink1
, MAXPATHLEN
- 1) == -1)
770 if (readlink(file2
.c_str(), symlink2
, MAXPATHLEN
- 1) == -1)
773 return !strncmp(symlink1
, symlink2
, MAXPATHLEN
);
776 else if (S_ISCHR(buf1
.st_mode
) && S_ISCHR(buf2
.st_mode
)) {
777 return buf1
.st_dev
== buf2
.st_dev
;
780 else if (S_ISBLK(buf1
.st_mode
) && S_ISBLK(buf2
.st_mode
)) {
781 return buf1
.st_dev
== buf2
.st_dev
;
787 bool permissions_equal(const string
& file1
, const string
& file2
)
792 if (lstat(file1
.c_str(), &buf1
) == -1)
795 if (lstat(file2
.c_str(), &buf2
) == -1)
798 return(buf1
.st_mode
== buf2
.st_mode
) &&
799 (buf1
.st_uid
== buf2
.st_uid
) &&
800 (buf1
.st_gid
== buf2
.st_gid
);
803 void file_remove(const string
& basedir
, const string
& filename
)
805 if (filename
!= basedir
&& !remove(filename
.c_str())) {
806 char* path
= strdup(filename
.c_str());
807 file_remove(basedir
, dirname(path
));