libgcc/libgcov-driver-system.c

   1 /* Routines required for instrumenting a program.  */
   2 /* Compile this one with gcc.  */
   3 /* Copyright (C) 1989-2014 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             && mkdir (filename) == -1
  70 #endif
  71             /* The directory might have been made by another process.  */
  72             && errno != EEXIST)
  73           {
  74             gcov_error ("profiling:%s:Cannot create directory\n", filename);
  75             *s = sep;
  76             return -1;
  77           };
  78
  79         *s = sep;
  80       };
  81   return 0;
  82 #endif
  83 }
  84
  85 static void
  86 allocate_filename_struct (struct gcov_filename *gf)
  87 {
  88   const char *gcov_prefix;
  89   size_t prefix_length;
  90   int strip = 0;
  91
  92   {
  93     /* Check if the level of dirs to strip off specified. */
  94     char *tmp = getenv("GCOV_PREFIX_STRIP");
  95     if (tmp)
  96       {
  97         strip = atoi (tmp);
  98         /* Do not consider negative values. */
  99         if (strip < 0)
 100           strip = 0;
 101       }
 102   }
 103   gf->strip = strip;
 104
 105   /* Get file name relocation prefix.  Non-absolute values are ignored. */
 106   gcov_prefix = getenv("GCOV_PREFIX");
 107   prefix_length = gcov_prefix ? strlen (gcov_prefix) : 0;
 108
 109   /* Remove an unnecessary trailing '/' */
 110   if (prefix_length && IS_DIR_SEPARATOR (gcov_prefix[prefix_length - 1]))
 111     prefix_length--;
 112
 113   /* If no prefix was specified and a prefix stip, then we assume
 114      relative.  */
 115   if (!prefix_length && gf->strip)
 116     {
 117       gcov_prefix = ".";
 118       prefix_length = 1;
 119     }
 120   gf->prefix = prefix_length;
 121
 122   /* Allocate and initialize the filename scratch space.  */
 123   gf->filename = (char *) xmalloc (gf->max_length + prefix_length + 2);
 124   if (prefix_length)
 125     memcpy (gf->filename, gcov_prefix, prefix_length);
 126 }
 127
 128 /* Open a gcda file specified by GI_FILENAME.
 129    Return -1 on error.  Return 0 on success.  */
 130
 131 static int
 132 gcov_exit_open_gcda_file (struct gcov_info *gi_ptr,
 133                           struct gcov_filename *gf)
 134 {
 135   const char *fname = gi_ptr->filename;
 136   char *dst = gf->filename + gf->prefix;
 137
 138   fname = gi_ptr->filename;
 139
 140   /* Build relocated filename, stripping off leading
 141      directories from the initial filename if requested. */
 142   if (gf->strip > 0)
 143     {
 144       const char *probe = fname;
 145       int level;
 146
 147       /* Remove a leading separator, without counting it.  */
 148       if (IS_DIR_SEPARATOR (*probe))
 149         probe++;
 150
 151       /* Skip selected directory levels.  If we fall off the end, we
 152          keep the final part.  */
 153       for (level = gf->strip; *probe && level; probe++)
 154         if (IS_DIR_SEPARATOR (*probe))
 155           {
 156             fname = probe;
 157             level--;
 158           }
 159     }
 160
 161   /* Update complete filename with stripped original. */
 162   if (gf->prefix)
 163     {
 164       /* Avoid to add multiple drive letters into combined path.  */
 165       if (HAS_DRIVE_SPEC(fname))
 166         fname += 2;
 167
 168       if (!IS_DIR_SEPARATOR (*fname))
 169         *dst++ = '/';
 170     }
 171   strcpy (dst, fname);
 172
 173   if (!gcov_open (gf->filename))
 174     {
 175       /* Open failed likely due to missed directory.
 176          Create directory and retry to open file. */
 177       if (create_file_directory (gf->filename))
 178         {
 179           fprintf (stderr, "profiling:%s:Skip\n", gf->filename);
 180           return -1;
 181         }
 182       if (!gcov_open (gf->filename))
 183         {
 184           fprintf (stderr, "profiling:%s:Cannot open\n", gf->filename);
 185           return -1;
 186         }
 187     }
 188
 189   return 0;
 190 }