audit: new command to check changed files

Currently only implement --backup to get list of (config) files in
protected directories to backup.

This also fixes a database corruption bug in database.c.

1 files changed, 146 insertions, 0 deletions

diff --git a/src/audit.c b/src/audit.c
new file mode 100644
index 0000000..f04b14a
--- /dev/null
+++ b/src/audit.c
@@ -0,0 +1,146 @@
+/* audit.c - Alpine Package Keeper (APK)
+ *
+ * Copyright (C) 2005-2008 Natanael Copa <n@tanael.org>
+ * Copyright (C) 2008 Timo Teräs <timo.teras@iki.fi>
+ * All rights reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify it 
+ * under the terms of the GNU General Public License version 2 as published
+ * by the Free Software Foundation. See http://www.gnu.org/ for details.
+ */
+
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <dirent.h>
+#include <sys/stat.h>
+#include "apk_applet.h"
+#include "apk_database.h"
+
+struct audit_ctx {
+	int (*action)(struct apk_database *db);
+};
+
+static int audit_directory(apk_hash_item item, void *ctx)
+{
+	struct apk_db_dir *dbd = (struct apk_db_dir *) item;
+	struct apk_db_file *dbf;
+	struct apk_database *db = (struct apk_database *) ctx;
+	struct dirent *de;
+	struct stat st;
+	struct apk_bstream *bs;
+	char tmp[512], reason;
+	DIR *dir;
+	apk_blob_t bdir = APK_BLOB_STR(dbd->dirname);
+	csum_t csum;
+
+	if (!(dbd->flags & APK_DBDIRF_PROTECTED))
+		return 0;
+
+	dir = opendir(dbd->dirname);
+	if (dir == NULL)
+		return 0;
+
+	while ((de = readdir(dir)) != NULL) {
+		if (strcmp(de->d_name, ".") == 0 ||
+		    strcmp(de->d_name, "..") == 0)
+			continue;
+
+		snprintf(tmp, sizeof(tmp), "%s/%s",
+			 dbd->dirname, de->d_name);
+
+		if (stat(tmp, &st) < 0)
+			continue;
+
+		if (S_ISDIR(st.st_mode)) {
+			if (apk_db_dir_query(db, APK_BLOB_STR(tmp)) != NULL)
+				continue;
+
+			reason = 'D';
+		} else {
+			dbf = apk_db_file_query(db, bdir, APK_BLOB_STR(de->d_name));
+			if (dbf != NULL) {
+				bs = apk_bstream_from_file(tmp);
+				if (bs == NULL)
+					continue;
+
+				bs->close(bs, csum, NULL);
+
+				if (apk_blob_compare(APK_BLOB_BUF(csum),
+						     APK_BLOB_BUF(dbf->csum)) == 0)
+					continue;
+
+				reason = 'M';
+			} else {
+				reason = 'F';
+			}
+		}
+		if (apk_quiet)
+			printf("%s\n", tmp);
+		else
+			printf("%c %s\n", reason, tmp);
+	}
+
+	closedir(dir);
+	return 0;
+}
+
+static int audit_backup(struct apk_database *db)
+{
+	fchdir(db->root_fd);
+	return apk_hash_foreach(&db->installed.dirs, audit_directory, db);
+}
+
+static int audit_parse(void *ctx, int optch, int optindex, const char *optarg)
+{
+	struct audit_ctx *actx = (struct audit_ctx *) ctx;
+
+	switch (optch) {
+	case 0x10000:
+		actx->action = audit_backup;
+		break;
+	default:
+		return -1;
+	}
+	return 0;
+}
+
+static int audit_main(void *ctx, int argc, char **argv)
+{
+	struct audit_ctx *actx = (struct audit_ctx *) ctx;
+	struct apk_database db;
+	int r;
+
+	if (actx->action == NULL) {
+		apk_error("No audit action requested");
+		return 2;
+	}
+
+	r = apk_db_open(&db, apk_root);
+	if (r != 0) {
+		apk_error("APK database not present");
+		return 1;
+	}
+
+	r = actx->action(&db);
+
+	apk_db_close(&db);
+	return r;
+}
+
+static struct option audit_options[] = {
+	{ "backup",	no_argument,		NULL, 0x10000 },
+};
+
+static struct apk_applet apk_audit = {
+	.name = "audit",
+	.usage = "--backup",
+	.context_size = sizeof(struct audit_ctx),
+	.num_options = ARRAY_SIZE(audit_options),
+	.options = audit_options,
+	.parse = audit_parse,
+	.main = audit_main,
+};
+
+APK_DEFINE_APPLET(apk_audit);
+