Signed-off-by: Andrew Gregory
---
lib/libalpm/Makefile.am | 2 +
lib/libalpm/alpm.h | 1 +
lib/libalpm/error.c | 2 +
lib/libalpm/hook.c | 279 ++++++++++++++++++++++++++++++++++++++++++++++++
lib/libalpm/hook.h | 2 +
lib/libalpm/ini.c | 1 +
lib/libalpm/ini.h | 1 +
lib/libalpm/trans.c | 6 ++
8 files changed, 294 insertions(+)
create mode 120000 lib/libalpm/ini.c
create mode 120000 lib/libalpm/ini.h
diff --git a/lib/libalpm/Makefile.am b/lib/libalpm/Makefile.am
index 1668da1..4272ae7 100644
--- a/lib/libalpm/Makefile.am
+++ b/lib/libalpm/Makefile.am
@@ -43,6 +43,8 @@ libalpm_la_SOURCES = \
graph.h graph.c \
group.h group.c \
handle.h handle.c \
+ hook.h hook.c \
+ ini.h ini.c \
libarchive-compat.h \
log.h log.c \
package.h package.c \
diff --git a/lib/libalpm/alpm.h b/lib/libalpm/alpm.h
index f224927..3049f2f 100644
--- a/lib/libalpm/alpm.h
+++ b/lib/libalpm/alpm.h
@@ -87,6 +87,7 @@ typedef enum _alpm_errno_t {
ALPM_ERR_TRANS_ABORT,
ALPM_ERR_TRANS_TYPE,
ALPM_ERR_TRANS_NOT_LOCKED,
+ ALPM_ERR_TRANS_HOOK_FAILED,
/* Packages */
ALPM_ERR_PKG_NOT_FOUND,
ALPM_ERR_PKG_IGNORED,
diff --git a/lib/libalpm/error.c b/lib/libalpm/error.c
index 76b710c..0a12cde 100644
--- a/lib/libalpm/error.c
+++ b/lib/libalpm/error.c
@@ -100,6 +100,8 @@ const char SYMEXPORT *alpm_strerror(alpm_errno_t err)
return _("operation not compatible with the transaction type");
case ALPM_ERR_TRANS_NOT_LOCKED:
return _("transaction commit attempt when database is not locked");
+ case ALPM_ERR_TRANS_HOOK_FAILED:
+ return _("failed to run transaction hooks");
/* Packages */
case ALPM_ERR_PKG_NOT_FOUND:
return _("could not find or read package");
diff --git a/lib/libalpm/hook.c b/lib/libalpm/hook.c
index c4f232b..7c50787 100644
--- a/lib/libalpm/hook.c
+++ b/lib/libalpm/hook.c
@@ -17,11 +17,15 @@
* along with this program. If not, see http://www.gnu.org/licenses/.
*/
+#include
#include
+#include
#include "handle.h"
#include "hook.h"
+#include "ini.h"
#include "log.h"
+#include "trans.h"
#include "util.h"
enum _alpm_hook_op_t {
@@ -153,4 +157,279 @@ static int _alpm_hook_parse_cb(const char *file, int line,
return 0;
}
+static int _alpm_hook_trigger_match_file(alpm_handle_t *handle, struct _alpm_trigger_t *t)
+{
+ alpm_list_t *i, *j, *install = NULL, *upgrade = NULL, *remove = NULL;
+ size_t isize = 0, rsize = 0;
+ int ret = 0;
+
+ /* check if file will be installed */
+ for(i = handle->trans->add; i; i = i->next) {
+ alpm_pkg_t *pkg = i->data;
+ alpm_filelist_t filelist = pkg->files;
+ size_t f;
+ for(f = 0; f < filelist.count; f++) {
+ if(alpm_option_match_noextract(handle, filelist.files[f].name) == 0) {
+ continue;
+ }
+ if(_alpm_fnmatch_patterns(t->targets, filelist.files[f].name) == 0) {
+ install = alpm_list_add(install, filelist.files[f].name);
+ isize++;
+ }
+ }
+ }
+
+ /* check if file will be removed due to package upgrade */
+ for(i = handle->trans->add; i; i = i->next) {
+ alpm_pkg_t *spkg = i->data;
+ alpm_pkg_t *pkg = alpm_db_get_pkg(handle->db_local, spkg->name);
+ if(pkg) {
+ alpm_filelist_t filelist = pkg->files;
+ size_t f;
+ for(f = 0; f < filelist.count; f++) {
+ if(_alpm_fnmatch_patterns(t->targets, filelist.files[f].name) == 0) {
+ remove = alpm_list_add(remove, filelist.files[f].name);
+ rsize++;
+ }
+ }
+ }
+ }
+
+ /* check if file will be removed due to package removal */
+ for(i = handle->trans->remove; i; i = i->next) {
+ alpm_pkg_t *pkg = i->data;
+ alpm_filelist_t filelist = pkg->files;
+ size_t f;
+ for(f = 0; f < filelist.count; f++) {
+ if(_alpm_fnmatch_patterns(t->targets, filelist.files[f].name) == 0) {
+ remove = alpm_list_add(remove, filelist.files[f].name);
+ rsize++;
+ }
+ }
+ }
+
+ i = install = alpm_list_msort(install, isize, (alpm_list_fn_cmp)strcmp);
+ j = remove = alpm_list_msort(remove, rsize, (alpm_list_fn_cmp)strcmp);
+ while(i) {
+ while(j && strcmp(i->data, j->data) > 0) {
+ j = j->next;
+ }
+ if(j == NULL) {
+ break;
+ }
+ if(strcmp(i->data, j->data) == 0) {
+ char *path = i->data;
+ upgrade = alpm_list_add(upgrade, path);
+ while(i && strcmp(i->data, path) == 0) {
+ alpm_list_t *next = i->next;
+ install = alpm_list_remove_item(install, i);
+ free(i);
+ i = next;
+ }
+ while(j && strcmp(j->data, path) == 0) {
+ alpm_list_t *next = j->next;
+ remove = alpm_list_remove_item(remove, j);
+ free(j);
+ j = next;
+ }
+ } else {
+ i = i->next;
+ }
+ }
+
+ ret = (t->op & ALPM_HOOK_OP_INSTALL && install)
+ || (t->op & ALPM_HOOK_OP_UPGRADE && upgrade)
+ || (t->op & ALPM_HOOK_OP_REMOVE && remove);
+
+ alpm_list_free(install);
+ alpm_list_free(upgrade);
+ alpm_list_free(remove);
+
+ return ret;
+}
+
+static int _alpm_hook_trigger_match_pkg(alpm_handle_t *handle, struct _alpm_trigger_t *t)
+{
+ if(t->op & ALPM_HOOK_OP_INSTALL || t->op & ALPM_HOOK_OP_UPGRADE) {
+ alpm_list_t *i;
+ for(i = handle->trans->add; i; i = i->next) {
+ alpm_pkg_t *pkg = i->data;
+ if(_alpm_fnmatch_patterns(t->targets, pkg->name) == 0) {
+ if(alpm_db_get_pkg(handle->db_local, pkg->name)) {
+ if(t->op & ALPM_HOOK_OP_UPGRADE) {
+ return 1;
+ }
+ } else {
+ if(t->op & ALPM_HOOK_OP_INSTALL) {
+ return 1;
+ }
+ }
+ }
+ }
+ }
+
+ if(t->op & ALPM_HOOK_OP_REMOVE) {
+ alpm_list_t *i;
+ for(i = handle->trans->remove; i; i = i->next) {
+ alpm_pkg_t *pkg = i->data;
+ if(pkg && _alpm_fnmatch_patterns(t->targets, pkg->name) == 0) {
+ if(!alpm_list_find(handle->trans->add, pkg, _alpm_pkg_cmp)) {
+ return 1;
+ }
+ }
+ }
+ }
+
+ return 0;
+}
+
+static int _alpm_hook_trigger_match(alpm_handle_t *handle, struct _alpm_trigger_t *t)
+{
+ return t->type == ALPM_HOOK_TYPE_PACKAGE
+ ? _alpm_hook_trigger_match_pkg(handle, t)
+ : _alpm_hook_trigger_match_file(handle, t);
+}
+
+static int _alpm_hook_triggered(alpm_handle_t *handle, struct _alpm_hook_t *hook)
+{
+ alpm_list_t *i;
+ for(i = hook->triggers; i; i = i->next) {
+ if(_alpm_hook_trigger_match(handle, i->data)) {
+ return 1;
+ }
+ }
+ return 0;
+}
+
+static alpm_list_t *find_hook(alpm_list_t *haystack, const void *needle)
+{
+ while(haystack) {
+ struct _alpm_hook_t *h = haystack->data;
+ if(h && strcmp(h->name, needle) == 0) {
+ return haystack;
+ }
+ haystack = haystack->next;
+ }
+ return NULL;
+}
+
+static int _alpm_hook_run_hook(alpm_handle_t *handle, struct _alpm_hook_t *hook)
+{
+ alpm_list_t *i, *pkgs = _alpm_db_get_pkgcache(handle->db_local);
+ char *const argv[] = { hook->cmd, NULL };
+
+ for(i = hook->depends; i; i = i->next) {
+ if(!alpm_find_satisfier(pkgs, i->data)) {
+ _alpm_log(handle, ALPM_LOG_ERROR, _("unable to run hook %s: %s\n"),
+ hook->name, _("could not satisfy dependencies"));
+ return -1;
+ }
+ }
+
+ return _alpm_run_chroot(handle, hook->cmd, argv);
+}
+
+int _alpm_hook_run(alpm_handle_t *handle, enum _alpm_hook_when_t when)
+{
+ alpm_list_t *i, *hooks = NULL;
+ const char *suffix = ".hook";
+ size_t suflen = strlen(suffix);
+ int ret = 0;
+
+ for(i = alpm_list_last(handle->hookdirs); i; i = alpm_list_previous(i)) {
+ int err;
+ char path[PATH_MAX];
+ size_t dirlen;
+ struct dirent entry, *result;
+ DIR *d;
+
+ if(!(d = opendir(i->data))) {
+ if(errno == ENOENT) {
+ continue;
+ } else {
+ _alpm_log(handle, ALPM_LOG_ERROR, _("could not open directory: %s: %s\n"),
+ (char *)i->data, strerror(errno));
+ ret = -1;
+ continue;
+ }
+ }
+
+ strncpy(path, i->data, PATH_MAX);
+ dirlen = strlen(i->data);
+
+ while((err = readdir_r(d, &entry, &result)) == 0 && result) {
+ struct _alpm_hook_cb_ctx ctx = { handle, NULL };
+ struct stat buf;
+ size_t name_len = strlen(entry.d_name);
+
+ if(strcmp(entry.d_name, ".") == 0 || strcmp(entry.d_name, "..") == 0) {
+ continue;
+ }
+
+ strncpy(path + dirlen, entry.d_name, PATH_MAX - dirlen);
+
+ if(name_len < suflen
+ || strcmp(entry.d_name + name_len - suflen, suffix) != 0) {
+ _alpm_log(handle, ALPM_LOG_DEBUG, "skipping non-hook file %s\n", path);
+ continue;
+ }
+
+ if(find_hook(hooks, entry.d_name)) {
+ _alpm_log(handle, ALPM_LOG_DEBUG, "skipping overridden hook %s\n", path);
+ continue;
+ }
+
+ if(fstatat(dirfd(d), entry.d_name, &buf, 0) != 0) {
+ _alpm_log(handle, ALPM_LOG_ERROR,
+ _("could not stat file %s: %s\n"), path, strerror(errno));
+ ret = -1;
+ continue;
+ }
+
+ if(S_ISDIR(buf.st_mode)) {
+ _alpm_log(handle, ALPM_LOG_DEBUG, "skipping directory %s\n", path);
+ continue;
+ }
+
+ CALLOC(ctx.hook, sizeof(struct _alpm_hook_t), 1,
+ ret = -1; closedir(d); goto cleanup);
+
+ _alpm_log(handle, ALPM_LOG_DEBUG, "parsing hook file %s\n", path);
+ if(parse_ini(path, _alpm_hook_parse_cb, &ctx) != 0) {
+ _alpm_log(handle, ALPM_LOG_DEBUG, "parsing hook file %s failed\n", path);
+ _alpm_hook_free(ctx.hook);
+ ret = -1;
+ continue;
+ }
+
+ STRDUP(ctx.hook->name, entry.d_name, ret = -1; closedir(d); goto cleanup);
+ hooks = alpm_list_add(hooks, ctx.hook);
+ }
+
+ if(err != 0) {
+ _alpm_log(handle, ALPM_LOG_ERROR, _("could not read directory: %s: %s\n"),
+ (char *) i->data, strerror(errno));
+ ret = -1;
+ }
+
+ closedir(d);
+ }
+
+ for(i = hooks; i; i = i->next) {
+ struct _alpm_hook_t *hook = i->data;
+ if(hook && hook->when == when && _alpm_hook_triggered(handle, hook)) {
+ _alpm_log(handle, ALPM_LOG_DEBUG, "running hook %s\n", hook->name);
+ if(_alpm_hook_run_hook(handle, hook) != 0 && hook->abort_on_fail) {
+ ret = -1;
+ }
+ }
+ }
+
+cleanup:
+ alpm_list_free_inner(hooks, (alpm_list_fn_free) _alpm_hook_free);
+ alpm_list_free(hooks);
+
+ return ret;
+}
+
/* vim: set noet: */
diff --git a/lib/libalpm/hook.h b/lib/libalpm/hook.h
index 0692176..4894a19 100644
--- a/lib/libalpm/hook.h
+++ b/lib/libalpm/hook.h
@@ -27,6 +27,8 @@ enum _alpm_hook_when_t {
ALPM_HOOK_POST_TRANSACTION
};
+int _alpm_hook_run(alpm_handle_t *handle, enum _alpm_hook_when_t when);
+
#endif /* _ALPM_HOOK_H */
/* vim: set noet: */
diff --git a/lib/libalpm/ini.c b/lib/libalpm/ini.c
new file mode 120000
index 0000000..b16d2f2
--- /dev/null
+++ b/lib/libalpm/ini.c
@@ -0,0 +1 @@
+../../src/common/ini.c
\ No newline at end of file
diff --git a/lib/libalpm/ini.h b/lib/libalpm/ini.h
new file mode 120000
index 0000000..e395fa0
--- /dev/null
+++ b/lib/libalpm/ini.h
@@ -0,0 +1 @@
+../../src/common/ini.h
\ No newline at end of file
diff --git a/lib/libalpm/trans.c b/lib/libalpm/trans.c
index ed073c0..a6b1aef 100644
--- a/lib/libalpm/trans.c
+++ b/lib/libalpm/trans.c
@@ -40,6 +40,7 @@
#include "sync.h"
#include "alpm.h"
#include "deps.h"
+#include "hook.h"
/** \addtogroup alpm_trans Transaction Functions
* @brief Functions to manipulate libalpm transactions
@@ -189,6 +190,10 @@ int SYMEXPORT alpm_trans_commit(alpm_handle_t *handle, alpm_list_t **data)
}
}
+ if(_alpm_hook_run(handle, ALPM_HOOK_PRE_TRANSACTION) != 0) {
+ RET_ERR(handle, ALPM_ERR_TRANS_HOOK_FAILED, -1);
+ }
+
trans->state = STATE_COMMITING;
alpm_logaction(handle, ALPM_CALLER_PREFIX, "transaction started\n");
@@ -215,6 +220,7 @@ int SYMEXPORT alpm_trans_commit(alpm_handle_t *handle, alpm_list_t **data)
alpm_logaction(handle, ALPM_CALLER_PREFIX, "transaction interrupted\n");
} else {
alpm_logaction(handle, ALPM_CALLER_PREFIX, "transaction completed\n");
+ _alpm_hook_run(handle, ALPM_HOOK_POST_TRANSACTION);
}
trans->state = STATE_COMMITED;
--
2.6.1