gcc/prefix.c

   1 /* Utility to update paths from internal to external forms.
   2    Copyright (C) 1997 Free Software Foundation, Inc.
   3
   4 This file is part of GNU CC.
   5
   6 GNU CC is free software; you can redistribute it and/or
   7 modify it under the terms of the GNU Library General Public
   8 License as published by the Free Software Foundation; either
   9 version 2 of the License, or (at your option) any later version.
  10
  11 GCC 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 Library General Public License for more details.
  15
  16 You should have received a copy of the GNU Library General Public
  17 License along with GCC; see the file COPYING.  If not, write to the Free
  18 Software Foundation, Inc., 59 Temple Place - Suite 330,
  19 Boston, MA 02111-1307, USA.  */
  20
  21 /* This file contains routines to update a path, both to canonicalize
  22    the directory format and to handle any prefix translation.
  23
  24    This file must be compiled with -DPREFIX= to specify the "prefix"
  25    value used by configure.  If a filename does not begin with this
  26    prefix, it will not be affected other than by directory canonicalization.
  27
  28    Each caller of 'update_path' may specify both a filename and
  29    a translation prefix and consist of the name of the package that contains
  30    the file ("@GCC", "@BINUTIL", "@GNU", etc).
  31
  32    If the prefix is not specified, the filename will only undergo
  33    directory canonicalization.
  34
  35    If it is specified, the string given by PREFIX will be replaced
  36    by the specified prefix (with a '@' in front unless the prefix begins
  37    with a '$') and further translation will be done as follows
  38    until none of the two conditions below are met:
  39
  40    1) If the filename begins with '@', the string between the '@' and
  41    the end of the name or the first '/' or directory separator will
  42    be considered a "key" and looked up as follows:
  43
  44    -- If this is a Win32 OS, then the Registry will be examined for
  45       an entry of "key" in
  46
  47       HKEY_LOCAL_MACHINE\SOFTWARE\Free Software Foundation\
  48
  49       if found, that value will be used.
  50
  51    -- If not found (or not a Win32 OS), the environment variable
  52       key_ROOT (the value of "key" concatenated with the constant "_ROOT")
  53       is tried.  If that fails, then PREFIX (see above) is used.
  54
  55    2) If the filename begins with a '$', the rest of the string up
  56    to the end or the first '/' or directory separator will be used
  57    as an environment variable, whose value will be returned.
  58
  59    Once all this is done, any '/' will be converted to DIR_SEPARATOR,
  60    if they are different.
  61
  62    NOTE:  using resolve_keyed_path under Win32 requires linking with
  63    advapi32.dll.  */
  64
  65
  66 #include "config.h"
  67 #include "gansidecl.h"
  68 #include "stdarg.h"
  69
  70 #ifdef _WIN32
  71 #include <windows.h>
  72 #endif
  73
  74 static char *get_key_value      PROTO((char *));
  75 static char *translate_name     PROTO((char *));
  76 static char *concat             PVPROTO((char *, ...));
  77 static char *save_string        PROTO((char *, int));
  78
  79 #ifdef _WIN32
  80 static char *lookup_key         PROTO((char *));
  81 static HKEY reg_key = (HKEY) INVALID_HANDLE_VALUE;
  82 #endif
  83
  84 extern char *getenv ();
  85
  86 /* Given KEY, as above, return its value.  */
  87
  88 static char *
  89 get_key_value (key)
  90      char *key;
  91 {
  92   char *prefix = 0;
  93
  94 #ifdef _WIN32
  95   prefix = lookup_key (key);
  96 #endif
  97
  98   if (prefix == 0)
  99     prefix = getenv (concat (key, "_ROOT", NULL_PTR));
 100
 101   if (prefix == 0)
 102     prefix = PREFIX;
 103
 104   return prefix;
 105 }
 106
 107 /* Concatenate a sequence of strings, returning the result.
 108
 109    This function is based on the one in libiberty.  */
 110
 111 static char *
 112 concat VPROTO((char *first, ...))
 113 {
 114   register int length;
 115   register char *newstr;
 116   register char *end;
 117   register char *arg;
 118   va_list args;
 119 #ifndef __STDC__
 120   char *first;
 121 #endif
 122
 123   /* First compute the size of the result and get sufficient memory.  */
 124
 125   VA_START (args, first);
 126 #ifndef __STDC__
 127   first = va_arg (args, char *);
 128 #endif
 129
 130   arg = first;
 131   length = 0;
 132
 133   while (arg != 0)
 134     {
 135       length += strlen (arg);
 136       arg = va_arg (args, char *);
 137     }
 138
 139   newstr = (char *) malloc (length + 1);
 140   va_end (args);
 141
 142   /* Now copy the individual pieces to the result string.  */
 143
 144   VA_START (args, first);
 145 #ifndef __STDC__
 146   first = va_arg (args, char *);
 147 #endif
 148
 149   end = newstr;
 150   arg = first;
 151   while (arg != 0)
 152     {
 153       while (*arg)
 154         *end++ = *arg++;
 155       arg = va_arg (args, char *);
 156     }
 157   *end = '\000';
 158   va_end (args);
 159
 160   return (newstr);
 161 }
 162
 163 /* Return a copy of a string that has been placed in the heap.  */
 164
 165 static char *
 166 save_string (s, len)
 167      char *s;
 168      int len;
 169 {
 170   register char *result = (char *) malloc (len + 1);
 171
 172   bcopy (s, result, len);
 173   result[len] = 0;
 174   return result;
 175 }
 176
 177 #ifdef _WIN32
 178
 179 /* Look up "key" in the registry, as above.  */
 180
 181 static char *
 182 lookup_key (key)
 183      char *key;
 184 {
 185   char *dst;
 186   DWORD size;
 187   DWORD type;
 188   LONG res;
 189
 190   if (reg_key == (HKEY) INVALID_HANDLE_VALUE)
 191     {
 192       res = RegOpenKeyExA (HKEY_LOCAL_MACHINE, "SOFTWARE", 0,
 193                            KEY_READ, &reg_key);
 194
 195       if (res == ERROR_SUCCESS)
 196         res = RegOpenKeyExA (reg_key, "Free Software Foundation", 0,
 197                              KEY_READ, &reg_key);
 198
 199       if (res != ERROR_SUCCESS)
 200         {
 201           reg_key = (HKEY) INVALID_HANDLE_VALUE;
 202           return 0;
 203         }
 204     }
 205
 206   size = 32;
 207   dst = (char *) malloc (size);
 208
 209   res = RegQueryValueExA (reg_key, key, 0, &type, dst, &size);
 210   if (res == ERROR_MORE_DATA && type == REG_SZ)
 211     {
 212       dst = (char *) realloc (dst, size);
 213       res = RegQueryValueExA (reg_key, key, 0, &type, dst, &size);
 214     }
 215
 216   if (type != REG_SZ || res != ERROR_SUCCESS)
 217     {
 218       free (dst);
 219       dst = 0;
 220     }
 221
 222   return dst;
 223 }
 224 #endif
 225
 226 /* If NAME starts with a '@' or '$', apply the translation rules above
 227    and return a new name.  Otherwise, return the given name.  */
 228
 229 static char *
 230 translate_name (name)
 231      char *name;
 232 {
 233   char code = name[0];
 234   char *key, *prefix;
 235   int keylen;
 236
 237   if (code != '@' && code != '$')
 238     return name;
 239
 240   for (keylen = 0;
 241        (name[keylen + 1] != 0 && name[keylen + 1] != '/'
 242 #ifdef DIR_SEPARATOR
 243         && name[keylen + 1] != DIR_SEPARATOR
 244 #endif
 245         );
 246        keylen++)
 247     ;
 248
 249   key = alloca (keylen + 1);
 250   strncpy (key, &name[1], keylen);
 251   key[keylen] = 0;
 252
 253   name = &name[keylen + 1];
 254
 255   if (code == '@')
 256     {
 257       prefix = get_key_value (key);
 258       if (prefix == 0)
 259         prefix = PREFIX;
 260     }
 261   else
 262     {
 263       prefix = getenv (key);
 264       if (prefix == 0)
 265         prefix = concat ("$", key, NULL_PTR);
 266     }
 267
 268   /* Remove any trailing directory separator from what we got.  */
 269   if (prefix[strlen (prefix) - 1] == '/'
 270 #ifdef DIR_SEPARATOR
 271       || prefix[strlen (prefix) - 1] == DIR_SEPARATOR
 272 #endif
 273       )
 274     {
 275       prefix = save_string (prefix, strlen (prefix));
 276       prefix[strlen (prefix) - 1] = 0;
 277     }
 278
 279   return concat (prefix, name, NULL_PTR);
 280 }
 281
 282 /* Update PATH using KEY if PATH starts with PREFIX.  */
 283
 284 char *
 285 update_path (path, key)
 286      char *path;
 287      char *key;
 288 {
 289   if (! strncmp (path, PREFIX, strlen (PREFIX)) && key != 0)
 290     {
 291       if (key[0] != '$')
 292         key = concat ("@", key, NULL_PTR);
 293
 294       path = concat (key, &path[strlen (PREFIX)], NULL_PTR);
 295
 296       while (path[0] == '@' || path[0] == '$')
 297         path = translate_name (path);
 298     }
 299
 300 #ifdef DIR_SEPARATOR
 301   if (DIR_SEPARATOR != '/')
 302     {
 303       int i;
 304       int len = strlen (path);
 305
 306       path = save_string (path, len);
 307       for (i = 0; i < len; i++)
 308         if (path[i] == '/')
 309           path[i] = DIR_SEPARATOR;
 310     }
 311 #endif
 312
 313   return path;
 314 }