sysdeps/unix/sysv/linux/dl-origin.c

   1 /* Find path of executable.
   2    Copyright (C) 1998-2000, 2002, 2004, 2008 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, write to the Free
  18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  19    02111-1307 USA.  */
  20
  21 #include <assert.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25 #include <sys/param.h>
  26 #include <ldsodefs.h>
  27 #include <sysdep.h>
  28
  29 #include <dl-dst.h>
  30
  31 /* On Linux >= 2.1 systems which have the dcache implementation we can get
  32    the path of the application from the /proc/self/exe symlink.  Try this
  33    first and fall back on the generic method if necessary.  */
  34
  35 const char *
  36 _dl_get_origin (void)
  37 {
  38   char linkval[PATH_MAX];
  39   const char *str;
  40   char *result = (char *) -1l;
  41   int len;
  42
  43   str = GLRO(dl_execfn);
  44   if (str == NULL || str[0] != '/')
  45     {
  46       INTERNAL_SYSCALL_DECL (err);
  47
  48       len = INTERNAL_SYSCALL (readlink, err, 3, "/proc/self/exe", linkval,
  49                               sizeof (linkval));
  50       if (! INTERNAL_SYSCALL_ERROR_P (len, err)
  51           && len > 0 && linkval[0] != '[')
  52         str = linkval;
  53       else
  54         str = NULL;
  55     }
  56   else
  57     len = strlen (str);
  58
  59   if (str == NULL)
  60     {
  61       /* We use the environment variable LD_ORIGIN_PATH.  If it is set make
  62          a copy and strip out trailing slashes.  */
  63       if (GLRO(dl_origin_path) != NULL)
  64         {
  65           size_t len = strlen (GLRO(dl_origin_path));
  66           result = (char *) malloc (len + 1);
  67           if (result == NULL)
  68             result = (char *) -1;
  69           else
  70             {
  71               char *cp = __mempcpy (result, GLRO(dl_origin_path), len);
  72               while (cp > result + 1 && cp[-1] == '/')
  73                 --cp;
  74               *cp = '\0';
  75             }
  76         }
  77     }
  78   else
  79     {
  80       /* We can use this value.  */
  81       assert (str[0] == '/');
  82       while (len > 1 && str[len - 1] != '/')
  83         --len;
  84       result = (char *) malloc (len + 1);
  85       if (result == NULL)
  86         result = (char *) -1;
  87       else if (len == 1)
  88         memcpy (result, "/", 2);
  89       else
  90         *((char *) __mempcpy (result, str, len - 1)) = '\0';
  91     }
  92
  93   return result;
  94 }