libgcc/libgcov-driver-system.c

   1 /* Routines required for instrumenting a program.  */
   2 /* Compile this one with gcc.  */
   3 /* Copyright (C) 1989-2015 Free Software Foundation, Inc.
   4
   5 This file is part of GCC.
   6
   7 GCC is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 3, or (at your option) any later
  10 version.
  11
  12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 Under Section 7 of GPL version 3, you are granted additional
  18 permissions described in the GCC Runtime Library Exception, version
  19 3.1, as published by the Free Software Foundation.
  20
  21 You should have received a copy of the GNU General Public License and
  22 a copy of the GCC Runtime Library Exception along with this program;
  23 see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
  24 <http://www.gnu.org/licenses/>.  */
  25
  26 /* A utility function for outputing errors.  */
  27
  28 static int __attribute__((format(printf, 1, 2)))
  29 gcov_error (const char *fmt, ...)
  30 {
  31   int ret;
  32   va_list argp;
  33   va_start (argp, fmt);
  34   ret = vfprintf (stderr, fmt, argp);
  35   va_end (argp);
  36   return ret;
  37 }
  38
  39 /* Make sure path component of the given FILENAME exists, create
  40    missing directories. FILENAME must be writable.
  41    Returns zero on success, or -1 if an error occurred.  */
  42
  43 static int
  44 create_file_directory (char *filename)
  45 {
  46 #if !defined(TARGET_POSIX_IO) && !defined(_WIN32)
  47   (void) filename;
  48   return -1;
  49 #else
  50   char *s;
  51
  52   s = filename;
  53
  54   if (HAS_DRIVE_SPEC(s))
  55     s += 2;
  56   if (IS_DIR_SEPARATOR(*s))
  57     ++s;
  58   for (; *s != '\0'; s++)
  59     if (IS_DIR_SEPARATOR(*s))
  60       {
  61         char sep = *s;
  62         *s  = '\0';
  63
  64         /* Try to make directory if it doesn't already exist.  */
  65         if (access (filename, F_OK) == -1
  66 #ifdef TARGET_POSIX_IO
  67             && mkdir (filename, 0755) == -1
  68 #else
  69 #ifdef mkdir
  70 #undef mkdir
  71 #endif
  72             && mkdir (filename) == -1
  73 #endif
  74             /* The directory might have been made by another process.  */
  75             && errno != EEXIST)
  76           {
  77             gcov_error ("profiling:%s:Cannot create directory\n", filename);
  78             *s = sep;
  79             return -1;
  80           };
  81
  82         *s = sep;
  83       };
  84   return 0;
  85 #endif
  86 }
  87
  88 static void
  89 allocate_filename_struct (struct gcov_filename *gf)
  90 {
  91   const char *gcov_prefix;
  92   size_t prefix_length;
  93   int strip = 0;
  94
  95   {
  96     /* Check if the level of dirs to strip off specified. */
  97     char *tmp = getenv("GCOV_PREFIX_STRIP");
  98     if (tmp)
  99       {
 100         strip = atoi (tmp);
 101         /* Do not consider negative values. */
 102         if (strip < 0)
 103           strip = 0;
 104       }
 105   }
 106   gf->strip = strip;
 107
 108   /* Get file name relocation prefix.  Non-absolute values are ignored. */
 109   gcov_prefix = getenv("GCOV_PREFIX");
 110   prefix_length = gcov_prefix ? strlen (gcov_prefix) : 0;
 111
 112   /* Remove an unnecessary trailing '/' */
 113   if (prefix_length && IS_DIR_SEPARATOR (gcov_prefix[prefix_length - 1]))
 114     prefix_length--;
 115
 116   /* If no prefix was specified and a prefix stip, then we assume
 117      relative.  */
 118   if (!prefix_length && gf->strip)
 119     {
 120       gcov_prefix = ".";
 121       prefix_length = 1;
 122     }
 123   gf->prefix = prefix_length;
 124
 125   /* Allocate and initialize the filename scratch space.  */
 126   gf->filename = (char *) xmalloc (gf->max_length + prefix_length + 2);
 127   if (prefix_length)
 128     memcpy (gf->filename, gcov_prefix, prefix_length);
 129 }
 130
 131 /* Open a gcda file specified by GI_FILENAME.
 132    Return -1 on error.  Return 0 on success.  */
 133
 134 static int
 135 gcov_exit_open_gcda_file (struct gcov_info *gi_ptr,
 136                           struct gcov_filename *gf)
 137 {
 138   const char *fname = gi_ptr->filename;
 139   char *dst = gf->filename + gf->prefix;
 140
 141   fname = gi_ptr->filename;
 142
 143   /* Build relocated filename, stripping off leading
 144      directories from the initial filename if requested. */
 145   if (gf->strip > 0)
 146     {
 147       const char *probe = fname;
 148       int level;
 149
 150       /* Remove a leading separator, without counting it.  */
 151       if (IS_DIR_SEPARATOR (*probe))
 152         probe++;
 153
 154       /* Skip selected directory levels.  If we fall off the end, we
 155          keep the final part.  */
 156       for (level = gf->strip; *probe && level; probe++)
 157         if (IS_DIR_SEPARATOR (*probe))
 158           {
 159             fname = probe;
 160             level--;
 161           }
 162     }
 163
 164   /* Update complete filename with stripped original. */
 165   if (gf->prefix)
 166     {
 167       /* Avoid to add multiple drive letters into combined path.  */
 168       if (HAS_DRIVE_SPEC(fname))
 169         fname += 2;
 170
 171       if (!IS_DIR_SEPARATOR (*fname))
 172         *dst++ = '/';
 173     }
 174   strcpy (dst, fname);
 175
 176   if (!gcov_open (gf->filename))
 177     {
 178       /* Open failed likely due to missed directory.
 179          Create directory and retry to open file. */
 180       if (create_file_directory (gf->filename))
 181         {
 182           fprintf (stderr, "profiling:%s:Skip\n", gf->filename);
 183           return -1;
 184         }
 185       if (!gcov_open (gf->filename))
 186         {
 187           fprintf (stderr, "profiling:%s:Cannot open\n", gf->filename);
 188           return -1;
 189         }
 190     }
 191
 192   return 0;
 193 }