From 9af1c6db0f56854348ef0d3d11f36da16d41f5d0 Mon Sep 17 00:00:00 2001
From: Wilco Baan Hofman <wilco@baanhofman.nl>
Date: Sat, 26 Sep 2009 17:05:34 +0200
Subject: regshell: Add support for 'cd ..' and cd relative to the root.

Signed-off-by: Jelmer Vernooij <jelmer@samba.org>
---
 source4/lib/registry/tools/regshell.c | 81 +++++++++++++++++++++++++++++++----
 1 file changed, 73 insertions(+), 8 deletions(-)

(limited to 'source4')

diff --git a/source4/lib/registry/tools/regshell.c b/source4/lib/registry/tools/regshell.c
index 0fc06d1219..314f063891 100644
--- a/source4/lib/registry/tools/regshell.c
+++ b/source4/lib/registry/tools/regshell.c
@@ -3,6 +3,7 @@
    simple registry frontend
 
    Copyright (C) Jelmer Vernooij 2004-2007
+   Copyright (C) Wilco Baan Hofman 2009
 
    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
@@ -30,10 +31,59 @@
 
 struct regshell_context {
 	struct registry_context *registry;
-	const char *path;
+	char *path;
+	char *predef;
 	struct registry_key *current;
+	struct registry_key *root;
 };
 
+static WERROR get_full_path(struct regshell_context *ctx, char *path, char **ret_path) 
+{
+	char *dir;
+	char *tmp;
+	char *new_path;
+
+	if (path[0] == '\\') {
+		new_path = talloc_strdup(ctx, "");
+	} else {
+ 		new_path = talloc_strdup(ctx, ctx->path);
+	}		
+
+	dir = strtok(path, "\\");
+	if (dir == NULL) {
+		*ret_path = new_path;
+		return WERR_OK;
+	}
+	do {
+		if (strcmp(dir, "..") == 0) {
+			if (strchr(new_path, '\\')) {
+				new_path[strrchr(new_path, '\\') - new_path] = '\0';
+			} else {
+				tmp = new_path;
+				new_path = talloc_strdup(ctx, "");
+				talloc_free(tmp);
+			}
+			continue;
+		} 
+		if (strcmp(dir, ".") == 0) {
+			continue;
+		}
+
+		tmp = new_path;
+		/* No prepending a backslash */
+		if (strcmp(new_path, "") == 0) {
+			new_path = talloc_strdup(ctx, dir);
+		} else {
+			new_path = talloc_asprintf(ctx, "%s\\%s", new_path, dir);
+		}
+		talloc_free(tmp);
+		
+	} while ((dir = strtok(NULL, "\\")));
+
+	*ret_path = new_path;
+	return WERR_OK;
+}
+
 /* *
  * ck/cd - change key
  * ls - list values/keys
@@ -117,8 +167,9 @@ static WERROR cmd_predef(struct regshell_context *ctx, int argc, char **argv)
 			return error;
 		}
 
-		ctx->path = strupper_talloc(ctx, argv[1]);
+		ctx->predef = strupper_talloc(ctx, argv[1]);
 		ctx->current = ret;
+		ctx->root = ret;
 	}
 
 	return WERR_OK;
@@ -127,6 +178,9 @@ static WERROR cmd_predef(struct regshell_context *ctx, int argc, char **argv)
 static WERROR cmd_pwd(struct regshell_context *ctx,
 		      int argc, char **argv)
 {
+	if (ctx->predef) {
+		printf("%s\\", ctx->predef);
+	}
 	printf("%s\n", ctx->path);
 	return WERR_OK;
 }
@@ -160,10 +214,15 @@ static WERROR cmd_set(struct regshell_context *ctx, int argc, char **argv)
 static WERROR cmd_ck(struct regshell_context *ctx, int argc, char **argv)
 {
 	struct registry_key *nkey = NULL;
+	char *full_path;
 	WERROR error;
 
 	if(argc == 2) {
-		error = reg_open_key(ctx->registry, ctx->current, argv[1],
+		if (!W_ERROR_IS_OK(get_full_path(ctx, argv[1], &full_path))) {
+			fprintf(stderr, "Unable to parse the supplied path\n");
+			return WERR_INVALID_PARAM;
+		}
+		error = reg_open_key(ctx->registry, ctx->root, full_path,
 				     &nkey);
 		if(!W_ERROR_IS_OK(error)) {
 			DEBUG(0, ("Error opening specified key: %s\n",
@@ -171,10 +230,12 @@ static WERROR cmd_ck(struct regshell_context *ctx, int argc, char **argv)
 			return error;
 		}
 
-		ctx->path = talloc_asprintf(ctx, "%s\\%s", ctx->path, argv[1]);
+		talloc_free(ctx->path);
+		ctx->path = full_path;
+
 		ctx->current = nkey;
 	}
-	printf("New path is: %s\n", ctx->path);
+	printf("New path is: %s\\%s\n", ctx->predef?ctx->predef:"", ctx->path);
 
 	return WERR_OK;
 }
@@ -299,7 +360,6 @@ _NORETURN_ static WERROR cmd_exit(struct regshell_context *ctx,
 				  int argc, char **argv)
 {
 	exit(0);
-	return WERR_OK;
 }
 
 static WERROR cmd_help(struct regshell_context *ctx, int, char **);
@@ -523,6 +583,8 @@ int main(int argc, char **argv)
 			return 1;
 		ctx->registry = ctx->current->context;
 		ctx->path = talloc_strdup(ctx, "");
+		ctx->predef = NULL;
+		ctx->root = ctx->current;
 	} else {
 		ctx->registry = reg_common_open_local(cmdline_credentials, ev_ctx, cmdline_lp_ctx);
 	}
@@ -540,11 +602,14 @@ int main(int argc, char **argv)
 						     reg_predefined_keys[i].handle,
 						     &ctx->current);
 			if (W_ERROR_IS_OK(err)) {
-				ctx->path = talloc_strdup(ctx,
+				ctx->predef = talloc_strdup(ctx,
 							  reg_predefined_keys[i].name);
+				ctx->path = talloc_strdup(ctx, "");
+				ctx->root = ctx->current;
 				break;
 			} else {
 				ctx->current = NULL;
+				ctx->root = NULL;
 			}
 		}
 	}
@@ -559,7 +624,7 @@ int main(int argc, char **argv)
 	while (true) {
 		char *line, *prompt;
 
-		asprintf(&prompt, "%s> ", ctx->path);
+		asprintf(&prompt, "%s\\%s> ", ctx->predef?ctx->predef:"", ctx->path);
 
 		current_key = ctx->current; 		/* No way to pass a void * pointer
 							   via readline :-( */
-- 
cgit