OSDN Git Service

Initial revision
[pf3gnuchains/gcc-fork.git] / 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 }