src/tool_homedir.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2016, Daniel Stenberg, <daniel@haxx.se>, et al.
   9  *
  10  * This software is licensed as described in the file COPYING, which
  11  * you should have received as part of this distribution. The terms
  12  * are also available at https://curl.haxx.se/docs/copyright.html.
  13  *
  14  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  15  * copies of the Software, and permit persons to whom the Software is
  16  * furnished to do so, under the terms of the COPYING file.
  17  *
  18  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  19  * KIND, either express or implied.
  20  *
  21  ***************************************************************************/
  22 #include "tool_setup.h"
  23
  24 #ifdef HAVE_PWD_H
  25 #  include <pwd.h>
  26 #endif
  27
  28 #include "tool_homedir.h"
  29
  30 #include "memdebug.h" /* keep this as LAST include */
  31
  32 static char *GetEnv(const char *variable, char do_expand)
  33 {
  34   char *env = NULL;
  35 #ifdef WIN32
  36   char  buf1[1024], buf2[1024];
  37   DWORD rc;
  38
  39   /* Don't use getenv(); it doesn't find variable added after program was
  40    * started. Don't accept truncated results (i.e. rc >= sizeof(buf1)).  */
  41
  42   rc = GetEnvironmentVariable(variable, buf1, sizeof(buf1));
  43   if(rc > 0 && rc < sizeof(buf1)) {
  44     env = buf1;
  45     variable = buf1;
  46   }
  47   if(do_expand && strchr(variable, '%')) {
  48     /* buf2 == variable if not expanded */
  49     rc = ExpandEnvironmentStrings(variable, buf2, sizeof(buf2));
  50     if(rc > 0 && rc < sizeof(buf2) &&
  51        !strchr(buf2, '%'))    /* no vars still unexpanded */
  52       env = buf2;
  53   }
  54 #else
  55   (void)do_expand;
  56   /* no length control */
  57   env = getenv(variable);
  58 #endif
  59   return (env && env[0]) ? strdup(env) : NULL;
  60 }
  61
  62 /* return the home directory of the current user as an allocated string */
  63 char *homedir(void)
  64 {
  65   char *home;
  66
  67   home = GetEnv("CURL_HOME", FALSE);
  68   if(home)
  69     return home;
  70
  71   home = GetEnv("HOME", FALSE);
  72   if(home)
  73     return home;
  74
  75 #if defined(HAVE_GETPWUID) && defined(HAVE_GETEUID)
  76  {
  77    struct passwd *pw = getpwuid(geteuid());
  78
  79    if(pw) {
  80      home = pw->pw_dir;
  81      if(home && home[0])
  82        home = strdup(home);
  83      else
  84        home = NULL;
  85    }
  86  }
  87 #endif /* PWD-stuff */
  88 #ifdef WIN32
  89   home = GetEnv("APPDATA", TRUE);
  90   if(!home)
  91     home = GetEnv("%USERPROFILE%\\Application Data", TRUE); /* Normally only
  92                                                                on Win-2K/XP */
  93 #endif /* WIN32 */
  94   return home;
  95 }