driver: added initial -o support, implemented the keyval vector.

1 files changed, 67 insertions, 0 deletions

diff --git a/src/driver/tpax_driver_ctx.c b/src/driver/tpax_driver_ctx.c
index 79518f0..dff6ef6 100644
--- a/src/driver/tpax_driver_ctx.c
+++ b/src/driver/tpax_driver_ctx.c
@@ -322,6 +322,12 @@ static void tpax_set_archive_block_size(struct tpax_common_ctx * cctx)
 		cctx->blksize = TPAX_USTAR_BLOCK_SIZE;
 }
 
+static int tpax_driver_is_valid_keyval(struct argv_keyval * keyval)
+{
+	(void)keyval;
+	return 0;
+}
+
 static struct tpax_driver_ctx_impl * tpax_driver_ctx_alloc(
 	struct argv_meta *		meta,
 	const struct tpax_fd_ctx *	fdctx,
@@ -333,6 +339,9 @@ static struct tpax_driver_ctx_impl * tpax_driver_ctx_alloc(
 	struct argv_entry *		entry;
 	const char **			units;
 	int				elements;
+	int                             nkeyval;
+	struct argv_keyval **           pkeyval;
+	struct argv_keyval *            keyval;
 
 	size =  sizeof(struct tpax_driver_ctx_alloc);
 	size += (nunits+1)*sizeof(const char *);
@@ -360,6 +369,22 @@ static struct tpax_driver_ctx_impl * tpax_driver_ctx_alloc(
 		if (!entry->fopt)
 			*units++ = entry->arg;
 
+	for (entry=meta->entries,nkeyval=0; entry->fopt || entry->arg; entry++)
+		if (entry->keyv)
+			for (keyval=entry->keyv; keyval->keyword; keyval++)
+				nkeyval++;
+
+	if (nkeyval && !(ictx->ctx.keyvalv = calloc(nkeyval+1,sizeof(*ictx->ctx.keyvalv)))) {
+		free(ictx);
+		return 0;
+	}
+
+	if ((pkeyval = ictx->ctx.keyvalv))
+		for (entry=meta->entries; entry->fopt || entry->arg; entry++)
+			if (entry->keyv)
+				for (keyval=entry->keyv; keyval->keyword; keyval++)
+					*pkeyval++ = keyval;
+
 	if (cctx->drvflags & TPAX_DRIVER_EXEC_MODE_WRITE_COPY) {
 		ictx->ctx.bufsize = TPAX_FILEIO_BUFLEN;
 		ictx->ctx.bufaddr = mmap(
@@ -369,6 +394,7 @@ static struct tpax_driver_ctx_impl * tpax_driver_ctx_alloc(
 			-1,0);
 
 		if (ictx->ctx.bufaddr == MAP_FAILED) {
+			free(ictx->ctx.keyvalv);
 			free(ictx);
 			return 0;
 		}
@@ -381,6 +407,7 @@ static struct tpax_driver_ctx_impl * tpax_driver_ctx_alloc(
 
 		if (ictx->ctx.dirbuff == MAP_FAILED) {
 			munmap(ictx->ctx.bufaddr,ictx->ctx.bufsize);
+			free(ictx->ctx.keyvalv);
 			free(ictx);
 			return 0;
 		}
@@ -399,6 +426,37 @@ static int tpax_get_driver_ctx_fail(struct argv_meta * meta)
 	return -1;
 }
 
+static int tpax_driver_keyval_error(
+	struct tpax_driver_ctx_impl *   ctx,
+	struct argv_keyval *            keyval,
+	const char *                    program)
+{
+	const char * equal;
+
+	switch (keyval->flags) {
+		case ARGV_KEYVAL_ASSIGN:
+			equal = "=";
+			break;
+
+		case ARGV_KEYVAL_OVERRIDE:
+			equal = ":=";
+			break;
+
+		default:
+			equal = "";
+	}
+
+	tpax_dprintf(
+		ctx->fdctx.fderr,
+		"%s: unsupported keyval argument (%s%s%s)\n",
+		program,keyval->keyword,equal,
+		keyval->value ? keyval->value : "");
+
+	tpax_lib_free_driver_ctx(&ctx->ctx);
+
+	return TPAX_ERROR;
+}
+
 int tpax_lib_get_driver_ctx(
 	char **				argv,
 	char **				envp,
@@ -413,6 +471,7 @@ int tpax_lib_get_driver_ctx(
 	struct argv_entry *		entry;
 	struct argv_entry *		archive;
 	struct argv_entry *		operand;
+	struct argv_keyval **           pkeyval;
 	struct tpax_fd_ctx              lfdctx;
 	size_t				nunits;
 	const char *			program;
@@ -699,6 +758,11 @@ int tpax_lib_get_driver_ctx(
 		return tpax_get_driver_ctx_fail(meta);
 	}
 
+	/* keyval validation */
+	for (pkeyval=ctx->keyvalv; pkeyval && *pkeyval; pkeyval++)
+		if (!tpax_driver_is_valid_keyval(*pkeyval))
+			return tpax_driver_keyval_error(ctx,*pkeyval,program);
+
 	if (archive) {
 		ctx->file     = archive->arg;
 		ctx->ctx.file = &ctx->file;
@@ -725,6 +789,9 @@ static void tpax_free_driver_ctx_impl(struct tpax_driver_ctx_alloc * ictx)
 		ictx->ctx.dirents = (struct tpax_dirent_buffer *)next;
 	}
 
+	if (ictx->ctx.keyvalv)
+		free(ictx->ctx.keyvalv);
+
 	if (ictx->ctx.bufaddr)
 		munmap(ictx->ctx.bufaddr,ictx->ctx.bufsize);