Home | History | Annotate | Download | only in src 
      1 /***************************************************************************
      2  *                                  _   _ ____  _
      3  *  Project                     ___| | | |  _ \| |
      4  *                             / __| | | | |_) | |
      5  *                            | (__| |_| |  _ <| |___
      6  *                             \___|\___/|_| \_\_____|
      7  *
      8  * Copyright (C) 1998 - 2016, Daniel Stenberg, <daniel (at) 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 = GetEnvironmentVariableA(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 = ExpandEnvironmentStringsA(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 }
     96