* w32reg.c (read_w32_registry_string): Handle REG_EXPAND_SZ.
[gnupg.git] / util / w32reg.c
1 /* w32reg.c -  MS-Windows Registry access
2  *      Copyright (C) 1999 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG 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
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #if defined (__MINGW32__) || defined (__CYGWIN32__)
23  /* This module is only used in this environment */
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdarg.h>
29 #include <windows.h>
30 #include "util.h"
31 #include "memory.h"
32
33 static HKEY
34 get_root_key(const char *root)
35 {
36     HKEY root_key;
37         
38     if( !root )
39         root_key = HKEY_CURRENT_USER;
40     else if( !strcmp( root, "HKEY_CLASSES_ROOT" ) )
41         root_key = HKEY_CLASSES_ROOT;
42     else if( !strcmp( root, "HKEY_CURRENT_USER" ) )
43         root_key = HKEY_CURRENT_USER;
44     else if( !strcmp( root, "HKEY_LOCAL_MACHINE" ) )
45         root_key = HKEY_LOCAL_MACHINE;
46     else if( !strcmp( root, "HKEY_USERS" ) )
47         root_key = HKEY_USERS;
48     else if( !strcmp( root, "HKEY_PERFORMANCE_DATA" ) )
49         root_key = HKEY_PERFORMANCE_DATA;
50     else if( !strcmp( root, "HKEY_CURRENT_CONFIG" ) )
51         root_key = HKEY_CURRENT_CONFIG;
52     else
53         return NULL;
54         
55     return root_key;
56 }
57
58
59 /****************
60  * Return a string from the Win32 Registry or NULL in case of
61  * error.  Caller must release the return value.   A NULL for root
62  * is an alias fro HKEY_CURRENT_USER
63  * NOTE: The value is allocated with a plain malloc() - use free() and not
64  * the usual m_free()!!!
65  */
66 char *
67 read_w32_registry_string( const char *root, const char *dir, const char *name )
68 {
69     HKEY root_key, key_handle;
70     DWORD n1, nbytes, type;
71     char *result = NULL;
72
73     if ( !(root_key = get_root_key(root) ) )
74         return NULL;
75
76     if( RegOpenKeyEx( root_key, dir, 0, KEY_READ, &key_handle ) )
77         return NULL; /* no need for a RegClose, so return direct */
78
79     nbytes = 1;
80     if( RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes ) )
81         goto leave;
82     result = malloc( (n1=nbytes+1) );
83     if( !result )
84         goto leave;
85     if( RegQueryValueEx( key_handle, name, 0, &type, result, &n1 ) ) {
86         free(result); result = NULL;
87         goto leave;
88     }
89     result[nbytes] = 0; /* make sure it is really a string  */
90     if (type == REG_EXPAND_SZ && strchr (result, '%')) {
91         char *tmp;
92         
93         n1 += 1000;
94         tmp = malloc (n1+1);
95         if (!tmp)
96             goto leave;
97         nbytes = ExpandEnvironmentStrings (result, tmp, n1);
98         if (nbytes && nbytes > n1) {
99             free (tmp);
100             tmp = malloc (n1 + 1);
101             if (!tmp)
102                 goto leave;
103             nbytes = ExpandEnvironmentStrings (result, tmp, n1);
104             if (nbytes && nbytes > n1) {
105                 free (tmp); /* oops - truncated, better don't expand at all */
106                 goto leave;
107             }
108             tmp[nbytes] = 0;
109             free (result);
110             result = tmp;
111         }
112         else if (nbytes) { /* okay, reduce the length */
113             tmp[nbytes] = 0;
114             free (result);
115             result = malloc (strlen (tmp)+1);
116             if (!result)
117                 result = tmp;
118             else {
119                 strcpy (result, tmp);
120                 free (tmp);
121             }
122         }
123         else {  /* error - don't expand */
124             free (tmp);
125         }
126     }
127
128   leave:
129     RegCloseKey( key_handle );
130     return result;
131 }
132
133
134 int
135 write_w32_registry_string(const char *root, const char *dir, 
136                           const char *name, const char *value)
137 {
138     HKEY root_key, reg_key;
139         
140     if ( !(root_key = get_root_key(root) ) )
141         return -1;
142
143     if ( RegOpenKeyEx( root_key, dir, 0, KEY_WRITE, &reg_key ) 
144          != ERROR_SUCCESS )
145         return -1;
146         
147     if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value, 
148                         strlen( value ) ) != ERROR_SUCCESS ) {
149         if ( RegCreateKey( root_key, name, &reg_key ) != ERROR_SUCCESS ) {
150             RegCloseKey(reg_key);
151             return -1;
152         }
153         if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value,
154                             strlen( value ) ) != ERROR_SUCCESS ) {
155             RegCloseKey(reg_key);
156             return -1;
157         }
158     }
159
160     RegCloseKey( reg_key );
161         
162     return 0;
163 }
164
165 #endif /* __MINGW32__ || __CYGWIN32__ */