added $SUDO_PROMPT support

diff --git a/sudo.c b/sudo.c
index 7ad53b1d556eeb817672894c9640768130ea3d5a..eab10c79a98685e953d41b5e000a554211d744b2 100644 (file)
--- a/sudo.c
+++ b/sudo.c
@@ -145,7 +145,7 @@ char *runas_user = "root";
 char *cmnd = NULL;
 char *cmnd_args = NULL;
 char *tty = NULL;
-char *prompt = PASSPROMPT;
+char *prompt;
 char host[MAXHOSTNAMELEN + 1];
 char *shost;
 char cwd[MAXPATHLEN + 1];
@@ -206,6 +206,12 @@ int main(argc, argv)
        exit(1);
     }
 
+    /*
+     * set the prompt based on $SUDO_PROMPT (can be overridden by `-p')
+     */
+    if ((prompt = getenv("SUDO_PROMPT")) == NULL)
+       prompt = PASSPROMPT;
+
     /*
      * parse our arguments
      */