Avoid NULL pointer access

2012-02-13 16:40:54 +01:00
parent e1ebfebf1b
commit 002affcfbe
1 changed files with 5 additions and 3 deletions
--- a/cpukit/libmisc/shell/shell_getprompt.c
+++ b/cpukit/libmisc/shell/shell_getprompt.c
@@ -37,13 +37,15 @@ void rtems_shell_get_prompt(
  size_t             size
 )
 {
-  char curdir[256];
+  char buf[256];
+  char *cwd;

  /* XXX: show_prompt user adjustable */
-  getcwd(curdir,sizeof(curdir));
+  cwd = getcwd(buf,sizeof(buf));
+  cwd = cwd != NULL ? cwd : "?";
  snprintf(prompt, size - 1, "%s%s[%s] %c ",
          ((shell_env->taskname) ? shell_env->taskname : ""),
          ((shell_env->taskname) ? " " : ""),
-          curdir,
+          cwd,
          geteuid()?'$':'#');
 }