path: root/lib/libalpm/be_sync.c

/*
 *  be_sync.c
 *
 *  Copyright (c) 2006-2010 Pacman Development Team <pacman-dev@archlinux.org>
 *  Copyright (c) 2002-2006 by Judd Vinet <jvinet@zeroflux.org>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "config.h"

#include <errno.h>
#include <dirent.h>
#include <ctype.h>
#include <locale.h>

/* libarchive */
#include <archive.h>
#include <archive_entry.h>

/* libalpm */
#include "util.h"
#include "log.h"
#include "alpm.h"
#include "alpm_list.h"
#include "package.h"
#include "handle.h"
#include "delta.h"
#include "deps.h"
#include "dload.h"

/* create list of directories in db */
static int dirlist_from_tar(const char *archive, alpm_list_t **dirlist)
{
	struct archive *_archive;
	struct archive_entry *entry;

	if((_archive = archive_read_new()) == NULL)
		RET_ERR(PM_ERR_LIBARCHIVE, -1);

	archive_read_support_compression_all(_archive);
	archive_read_support_format_all(_archive);

	if(archive_read_open_filename(_archive, archive,
				ARCHIVE_DEFAULT_BYTES_PER_BLOCK) != ARCHIVE_OK) {
		_alpm_log(PM_LOG_ERROR, _("could not open %s: %s\n"), archive,
				archive_error_string(_archive));
		RET_ERR(PM_ERR_PKG_OPEN, -1);
	}

	while(archive_read_next_header(_archive, &entry) == ARCHIVE_OK) {
		const struct stat *st;
		const char *entryname; /* the name of the file in the archive */

		st = archive_entry_stat(entry);
		entryname = archive_entry_pathname(entry);

		if(S_ISDIR(st->st_mode)) {
			char *name = strdup(entryname);
			*dirlist = alpm_list_add(*dirlist, name);
		}
	}
	archive_read_finish(_archive);

	*dirlist = alpm_list_msort(*dirlist, alpm_list_count(*dirlist), _alpm_str_cmp);
	return(0);
}

static int is_dir(const char *path, struct dirent *entry)
{
#ifdef DT_DIR
	return(entry->d_type == DT_DIR);
#else
	char buffer[PATH_MAX];
	snprintf(buffer, PATH_MAX, "%s/%s", path, entry->d_name);

	struct stat sbuf;
	if (!stat(buffer, &sbuf)) {
		return(S_ISDIR(sbuf.st_mode));
	}

	return(0);
#endif
}

/* create list of directories in db */
static int dirlist_from_fs(const char *syncdbpath, alpm_list_t **dirlist)
{
	DIR *dbdir;
	struct dirent *ent = NULL;

	dbdir = opendir(syncdbpath);
	if (dbdir != NULL) {
		while((ent = readdir(dbdir)) != NULL) {
			char *name = ent->d_name;
			size_t len;
			char *entry;

			if(strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
				continue;
			}

			if(!is_dir(syncdbpath, ent)) {
				continue;
			}

			len = strlen(name);
			MALLOC(entry, len + 2, RET_ERR(PM_ERR_MEMORY, -1));
			strcpy(entry, name);
			entry[len] = '/';
			entry[len+1] = '\0';
			*dirlist = alpm_list_add(*dirlist, entry);
		}
		closedir(dbdir);
	}

	*dirlist = alpm_list_msort(*dirlist, alpm_list_count(*dirlist), _alpm_str_cmp);
	return(0);
}

/* remove old directories from dbdir */
static int remove_olddir(const char *syncdbpath, alpm_list_t *dirlist)
{
	alpm_list_t *i;
	for (i = dirlist; i; i = i->next) {
		const char *name = i->data;
		char *dbdir;
		size_t len = strlen(syncdbpath) + strlen(name) + 2;
		MALLOC(dbdir, len, RET_ERR(PM_ERR_MEMORY, -1));
		snprintf(dbdir, len, "%s%s", syncdbpath, name);
		_alpm_log(PM_LOG_DEBUG, "removing: %s\n", dbdir);
		if(_alpm_rmrf(dbdir) != 0) {
			_alpm_log(PM_LOG_ERROR, _("could not remove database directory %s\n"), dbdir);
			free(dbdir);
			RET_ERR(PM_ERR_DB_REMOVE, -1);
		}
		free(dbdir);
	}
	return(0);
}

/** Update a package database
 *
 * An update of the package database \a db will be attempted. Unless
 * \a force is true, the update will only be performed if the remote
 * database was modified since the last update.
 *
 * A transaction is necessary for this operation, in order to obtain a
 * database lock. During this transaction the front-end will be informed
 * of the download progress of the database via the download callback.
 *
 * Example:
 * @code
 * pmdb_t *db;
 * int result;
 * db = alpm_list_getdata(alpm_option_get_syncdbs());
 * if(alpm_trans_init(0, NULL, NULL, NULL) == 0) {
 *     result = alpm_db_update(0, db);
 *     alpm_trans_release();
 *
 *     if(result > 0) {
 *	       printf("Unable to update database: %s\n", alpm_strerrorlast());
 *     } else if(result < 0) {
 *         printf("Database already up to date\n");
 *     } else {
 *         printf("Database updated\n");
 *     }
 * }
 * @endcode
 *
 * @ingroup alpm_databases
 * @note After a successful update, the \link alpm_db_get_pkgcache()
 * package cache \endlink will be invalidated
 * @param force if true, then forces the update, otherwise update only in case
 * the database isn't up to date
 * @param db pointer to the package database to update
 * @return 0 on success, > 0 on error (pm_errno is set accordingly), < 0 if up
 * to date
 */
int SYMEXPORT alpm_db_update(int force, pmdb_t *db)
{
	char *dbfile, *dbfilepath, *syncpath;
	const char *dbpath, *syncdbpath;
	alpm_list_t *newdirlist = NULL, *olddirlist = NULL;
	alpm_list_t *onlynew = NULL, *onlyold = NULL;
	size_t len;
	int ret;

	ALPM_LOG_FUNC;

	/* Sanity checks */
	ASSERT(handle != NULL, RET_ERR(PM_ERR_HANDLE_NULL, -1));
	ASSERT(db != NULL && db != handle->db_local, RET_ERR(PM_ERR_WRONG_ARGS, -1));
	/* Verify we are in a transaction.  This is done _mainly_ because we need a DB
	 * lock - if we update without a db lock, we may kludge some other pacman
	 * process that _has_ a lock.
	 */
	ASSERT(handle->trans != NULL, RET_ERR(PM_ERR_TRANS_NULL, -1));
	ASSERT(handle->trans->state == STATE_INITIALIZED, RET_ERR(PM_ERR_TRANS_NOT_INITIALIZED, -1));

	if(!alpm_list_find_ptr(handle->dbs_sync, db)) {
		RET_ERR(PM_ERR_DB_NOT_FOUND, -1);
	}

	len = strlen(db->treename) + 4;
	MALLOC(dbfile, len, RET_ERR(PM_ERR_MEMORY, -1));
	sprintf(dbfile, "%s.db", db->treename);

	dbpath = alpm_option_get_dbpath();
	len = strlen(dbpath) + 6;
	MALLOC(syncpath, len, RET_ERR(PM_ERR_MEMORY, -1));
	sprintf(syncpath, "%s%s", dbpath, "sync/");

	ret = _alpm_download_single_file(dbfile, db->servers, syncpath, force);
	free(dbfile);
	free(syncpath);

	if(ret == 1) {
		/* files match, do nothing */
		pm_errno = 0;
		return(1);
	} else if(ret == -1) {
		/* pm_errno was set by the download code */
		_alpm_log(PM_LOG_DEBUG, "failed to sync db: %s\n", alpm_strerrorlast());
		return(-1);
	}

	syncdbpath = _alpm_db_path(db);

	/* form the path to the db location */
	len = strlen(dbpath) + strlen(db->treename) + 9;
	MALLOC(dbfilepath, len, RET_ERR(PM_ERR_MEMORY, -1));
	sprintf(dbfilepath, "%ssync/%s.db", dbpath, db->treename);

	if(force) {
		/* if forcing update, remove the old dir and extract the db */
		if(_alpm_rmrf(syncdbpath) != 0) {
			_alpm_log(PM_LOG_ERROR, _("could not remove database %s\n"), db->treename);
			RET_ERR(PM_ERR_DB_REMOVE, -1);
		} else {
			_alpm_log(PM_LOG_DEBUG, "database dir %s removed\n", _alpm_db_path(db));
		}
	} else {
		/* if not forcing, only remove and extract what is necessary */
		ret = dirlist_from_tar(dbfilepath, &newdirlist);
		if(ret) {
			goto cleanup;
		}
		ret = dirlist_from_fs(syncdbpath, &olddirlist);
		if(ret) {
			goto cleanup;
		}

		alpm_list_diff_sorted(olddirlist, newdirlist, _alpm_str_cmp, &onlyold, &onlynew);

		ret = remove_olddir(syncdbpath, onlyold);
		if(ret) {
			goto cleanup;
		}
	}

	/* Cache needs to be rebuilt */
	_alpm_db_free_pkgcache(db);

	checkdbdir(db);
	ret = _alpm_unpack(dbfilepath, syncdbpath, onlynew, 0);

cleanup:
	FREELIST(newdirlist);
	FREELIST(olddirlist);
	alpm_list_free(onlynew);
	alpm_list_free(onlyold);

	free(dbfilepath);

	if(ret) {
		RET_ERR(PM_ERR_SYSTEM, -1);
	}

	return(0);
}

int _alpm_sync_db_populate(pmdb_t *db)
{
	int count = 0;
	struct archive *archive;
	struct archive_entry *entry;
	const char * archive_path;

	ALPM_LOG_FUNC;

	ASSERT(db != NULL, RET_ERR(PM_ERR_DB_NULL, -1));

	if((archive = archive_read_new()) == NULL)
		RET_ERR(PM_ERR_LIBARCHIVE, 1);

	archive_read_support_compression_all(archive);
	archive_read_support_format_all(archive);

	if(archive_read_open_filename(archive, _alpm_db_path(db),
				ARCHIVE_DEFAULT_BYTES_PER_BLOCK) != ARCHIVE_OK) {
		_alpm_log(PM_LOG_ERROR, _("could not open %s: %s\n"), _alpm_db_path(db),
				archive_error_string(archive));
		RET_ERR(PM_ERR_PKG_OPEN, 1);
	}

	while(archive_read_next_header(archive, &entry) == ARCHIVE_OK) {
		const struct stat *st;
		const char *name;
		pmpkg_t *pkg;

		st = archive_entry_stat(entry);

		if(S_ISDIR(st->st_mode)) {
			archive_path = archive_entry_pathname(entry);

			pkg = _alpm_pkg_new();
			if(pkg == NULL) {
				archive_read_finish(archive);
				return(-1);
			}

			name = archive_entry_pathname(entry);

			if(splitname(name, pkg) != 0) {
				_alpm_log(PM_LOG_ERROR, _("invalid name for database entry '%s'\n"),
						name);
				_alpm_pkg_free(pkg);
				continue;
			}

			/* duplicated database entries are not allowed */
			if(_alpm_pkg_find(db->pkgcache, pkg->name)) {
				_alpm_log(PM_LOG_ERROR, _("duplicated database entry '%s'\n"), pkg->name);
				_alpm_pkg_free(pkg);
				continue;
			}

			pkg->origin = PKG_FROM_SYNCDB;
			pkg->ops = &default_pkg_ops;
			pkg->origin_data.db = db;

			/* add to the collection */
			_alpm_log(PM_LOG_FUNCTION, "adding '%s' to package cache for db '%s'\n",
					pkg->name, db->treename);
			db->pkgcache = alpm_list_add(db->pkgcache, pkg);
			count++;
		} else {
			/* we have desc, depends or deltas - parse it */
			_alpm_sync_db_read(db, archive, entry);
		}
	}

	db->pkgcache = alpm_list_msort(db->pkgcache, count, _alpm_pkg_cmp);
	archive_read_finish(archive);

	return(count);
}

int _alpm_sync_db_read(pmdb_t *db, struct archive *archive, struct archive_entry *entry)
{
	char line[1024];
	const char *entryname;
	char *filename, *pkgname, *p, *q;
	pmpkg_t *pkg;

	ALPM_LOG_FUNC;

	if(db == NULL) {
		RET_ERR(PM_ERR_DB_NULL, -1);
	}

	if(entry == NULL) {
		_alpm_log(PM_LOG_DEBUG, "invalid archive entry provided to _alpm_sync_db_read, skipping\n");
		return(-1);
	}

	entryname = archive_entry_pathname(entry);

	_alpm_log(PM_LOG_FUNCTION, "loading package data from archive entry %s\n",
			entryname);

	/* get package and db file names */
	STRDUP(pkgname, entryname, RET_ERR(PM_ERR_MEMORY, -1));
	p = pkgname + strlen(pkgname);
	for(q = --p; *q && *q != '/'; q--);
	STRDUP(filename, q+1, RET_ERR(PM_ERR_MEMORY, -1));
	for(p = --q; *p && *p != '-'; p--);
	for(q = --p; *q && *q != '-'; q--);
	*q = '\0';

	/* package is already in db due to parsing of directory name */
	pkg = _alpm_pkg_find(db->pkgcache, pkgname);
	if(pkg == NULL) {
		_alpm_log(PM_LOG_DEBUG, "package %s not found in %s sync database",
					pkgname, db->treename);
		return(-1);
	}

	if(strcmp(filename, "desc") == 0) {
		while(_alpm_archive_fgets(line, sizeof(line), archive) != NULL) {
			_alpm_strtrim(line);
			if(strcmp(line, "%NAME%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				if(strcmp(_alpm_strtrim(line), pkg->name) != 0) {
					_alpm_log(PM_LOG_ERROR, _("%s database is inconsistent: name "
								"mismatch on package %s\n"), db->treename, pkg->name);
				}
			} else if(strcmp(line, "%VERSION%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				if(strcmp(_alpm_strtrim(line), pkg->version) != 0) {
					_alpm_log(PM_LOG_ERROR, _("%s database is inconsistent: version "
								"mismatch on package %s\n"), db->treename, pkg->name);
				}
			} else if(strcmp(line, "%FILENAME%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->filename, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%DESC%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->desc, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%GROUPS%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) && strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->groups = alpm_list_add(pkg->groups, linedup);
				}
			} else if(strcmp(line, "%URL%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->url, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%LICENSE%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->licenses = alpm_list_add(pkg->licenses, linedup);
				}
			} else if(strcmp(line, "%ARCH%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->arch, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%BUILDDATE%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				_alpm_strtrim(line);

				char first = tolower((unsigned char)line[0]);
				if(first > 'a' && first < 'z') {
					struct tm tmp_tm = {0}; /* initialize to null in case of failure */
					setlocale(LC_TIME, "C");
					strptime(line, "%a %b %e %H:%M:%S %Y", &tmp_tm);
					pkg->builddate = mktime(&tmp_tm);
					setlocale(LC_TIME, "");
				} else {
					pkg->builddate = atol(line);
				}
			} else if(strcmp(line, "%INSTALLDATE%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				_alpm_strtrim(line);

				char first = tolower((unsigned char)line[0]);
				if(first > 'a' && first < 'z') {
					struct tm tmp_tm = {0}; /* initialize to null in case of failure */
					setlocale(LC_TIME, "C");
					strptime(line, "%a %b %e %H:%M:%S %Y", &tmp_tm);
					pkg->installdate = mktime(&tmp_tm);
					setlocale(LC_TIME, "");
				} else {
					pkg->installdate = atol(line);
				}
			} else if(strcmp(line, "%PACKAGER%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->packager, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%REASON%") == 0) {
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				pkg->reason = (pmpkgreason_t)atol(_alpm_strtrim(line));
			} else if(strcmp(line, "%SIZE%") == 0 || strcmp(line, "%CSIZE%") == 0) {
				/* NOTE: the CSIZE and SIZE fields both share the "size" field
				 *       in the pkginfo_t struct.  This can be done b/c CSIZE
				 *       is currently only used in sync databases, and SIZE is
				 *       only used in local databases.
				 */
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				pkg->size = atol(_alpm_strtrim(line));
				/* also store this value to isize if isize is unset */
				if(pkg->isize == 0) {
					pkg->isize = pkg->size;
				}
			} else if(strcmp(line, "%ISIZE%") == 0) {
				/* ISIZE (installed size) tag only appears in sync repositories,
				 * not the local one. */
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				pkg->isize = atol(_alpm_strtrim(line));
			} else if(strcmp(line, "%MD5SUM%") == 0) {
				/* MD5SUM tag only appears in sync repositories,
				 * not the local one. */
				if(_alpm_archive_fgets(line, sizeof(line), archive) == NULL) {
					goto error;
				}
				STRDUP(pkg->md5sum, _alpm_strtrim(line), goto error);
			} else if(strcmp(line, "%REPLACES%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->replaces = alpm_list_add(pkg->replaces, linedup);
				}
			} else if(strcmp(line, "%FORCE%") == 0) {
				pkg->force = 1;
			}
		}
	} else if(strcmp(filename, "depends") == 0) {
		while(_alpm_archive_fgets(line, sizeof(line), archive) != NULL) {
			_alpm_strtrim(line);
			if(strcmp(line, "%DEPENDS%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					pmdepend_t *dep = _alpm_splitdep(_alpm_strtrim(line));
					pkg->depends = alpm_list_add(pkg->depends, dep);
				}
			} else if(strcmp(line, "%OPTDEPENDS%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->optdepends = alpm_list_add(pkg->optdepends, linedup);
				}
			} else if(strcmp(line, "%CONFLICTS%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->conflicts = alpm_list_add(pkg->conflicts, linedup);
				}
			} else if(strcmp(line, "%PROVIDES%") == 0) {
				while(_alpm_archive_fgets(line, sizeof(line), archive) &&
							strlen(_alpm_strtrim(line))) {
					char *linedup;
					STRDUP(linedup, _alpm_strtrim(line), goto error);
					pkg->provides = alpm_list_add(pkg->provides, linedup);
				}
			}
		}
	} else if(strcmp(filename, "deltas") == 0) {
		while(_alpm_archive_fgets(line, sizeof(line), archive) != NULL) {
			_alpm_strtrim(line);
				if(strcmp(line, "%DELTAS%") == 0) {
					while(_alpm_archive_fgets(line, sizeof(line), archive) && strlen(_alpm_strtrim(line))) {
						pmdelta_t *delta = _alpm_delta_parse(line);
						if(delta) {
							pkg->deltas = alpm_list_add(pkg->deltas, delta);
						}
					}
				}
		}
	} else {
		 /* unknown database file */
		_alpm_log(PM_LOG_DEBUG, "unknown database file: %s", filename);
	}

error:
	FREE(pkgname);
	FREE(filename);
	return(0);
}

struct db_operations sync_db_ops = {
	.populate         = _alpm_sync_db_populate,
	.unregister       = _alpm_db_unregister,
};

pmdb_t *_alpm_db_register_sync(const char *treename)
{
	pmdb_t *db;
	alpm_list_t *i;

	ALPM_LOG_FUNC;

	for(i = handle->dbs_sync; i; i = i->next) {
		pmdb_t *sdb = i->data;
		if(strcmp(treename, sdb->treename) == 0) {
			_alpm_log(PM_LOG_DEBUG, "attempt to re-register the '%s' database, using existing\n", sdb->treename);
			return sdb;
		}
	}

	_alpm_log(PM_LOG_DEBUG, "registering sync database '%s'\n", treename);

	db = _alpm_db_new(treename, 0);
	db->ops = &sync_db_ops;
	if(db == NULL) {
		RET_ERR(PM_ERR_DB_CREATE, NULL);
	}

	handle->dbs_sync = alpm_list_add(handle->dbs_sync, db);
	return(db);
}


/* vim: set ts=2 sw=2 noet: */