From 9a7acfd771e17bd7d2ca935fc17fa50ef579cce3 Mon Sep 17 00:00:00 2001
From: netblue30 <netblue30@yahoo.com>
Date: Sun, 11 Dec 2016 08:10:04 -0500
Subject: config support for firejail prompt in terminals

---
 src/firejail/checkcfg.c | 10 ++++++++++
 src/firejail/env.c      | 14 +++++++++++---
 src/firejail/firejail.h |  1 +
 3 files changed, 22 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/firejail/checkcfg.c b/src/firejail/checkcfg.c
index 6565f488a..098e8e967 100644
--- a/src/firejail/checkcfg.c
+++ b/src/firejail/checkcfg.c
@@ -43,6 +43,7 @@ int checkcfg(int val) {
 		cfg_val[CFG_RESTRICTED_NETWORK] = 0; // disabled by default
 		cfg_val[CFG_FORCE_NONEWPRIVS] = 0; // disabled by default
 		cfg_val[CFG_PRIVATE_BIN_NO_LOCAL] = 0; // disabled by default
+		cfg_val[CFG_FIREJAIL_PROMPT] = 0; // disabled by default
 		
 		// open configuration file
 		char *fname;
@@ -126,6 +127,15 @@ int checkcfg(int val) {
 				else
 					goto errout;
 			}
+			// prompt
+			else if (strncmp(ptr, "firejail-prompt ", 16) == 0) {
+				if (strcmp(ptr + 16, "yes") == 0)
+					cfg_val[CFG_FIREJAIL_PROMPT] = 1;
+				else if (strcmp(ptr + 16, "no") == 0)
+					cfg_val[CFG_FIREJAIL_PROMPT] = 0;
+				else
+					goto errout;
+			}
 			// nonewprivs
 			else if (strncmp(ptr, "force-nonewprivs ", 17) == 0) {
 				if (strcmp(ptr + 17, "yes") == 0)
diff --git a/src/firejail/env.c b/src/firejail/env.c
index a02c67ae1..783f019a6 100644
--- a/src/firejail/env.c
+++ b/src/firejail/env.c
@@ -129,13 +129,21 @@ void env_defaults(void) {
 		errExit("setenv");
 
 	// set prompt color to green
-	char *prompt = getenv("FIREJAIL_PROMPT");
-	if (prompt && strcmp(prompt, "yes") == 0) {
+	int set_prompt = 0;
+	if (checkcfg(CFG_FIREJAIL_PROMPT))
+		set_prompt = 1;
+	else { // check FIREJAIL_PROMPT="yes" environment variable
+		char *prompt = getenv("FIREJAIL_PROMPT");
+		if (prompt && strcmp(prompt, "yes") == 0)
+			set_prompt = 1;
+	}
+	
+	if (set_prompt) {
 		//export PS1='\[\e[1;32m\][\u@\h \W]\$\[\e[0m\] '
 		if (setenv("PROMPT_COMMAND", "export PS1=\"\\[\\e[1;32m\\][\\u@\\h \\W]\\$\\[\\e[0m\\] \"", 1) < 0)
 			errExit("setenv");
 	}
-
+	
 	// set the window title
 	if (!arg_quiet)
 		printf("\033]0;firejail %s\007", cfg.window_title);
diff --git a/src/firejail/firejail.h b/src/firejail/firejail.h
index 368e0d88d..8fede5a69 100644
--- a/src/firejail/firejail.h
+++ b/src/firejail/firejail.h
@@ -661,6 +661,7 @@ enum {
 	CFG_CHROOT_DESKTOP,
 	CFG_PRIVATE_HOME,
 	CFG_PRIVATE_BIN_NO_LOCAL,
+	CFG_FIREJAIL_PROMPT,
 	CFG_MAX // this should always be the last entry
 };
 extern char *xephyr_screen;
-- 
cgit v1.2.3-70-g09d2