(set_native_charset): Assume that ASCII,
[gnupg.git] / util / w32reg.c
1 /* w32reg.c -  MS-Windows Registry access
2  *      Copyright (C) 1999, 2002 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 (_WIN32) || 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 for HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE in turn.
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       {
78         if (root)
79           return NULL; /* no need for a RegClose, so return direct */
80         /* It seems to be common practise to fall back to HLM. */
81         if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, dir, 0, KEY_READ, &key_handle) )
82           return NULL; /* still no need for a RegClose, so return direct */
83       }
84
85     nbytes = 1;
86     if( RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes ) )
87         goto leave;
88     result = malloc( (n1=nbytes+1) );
89     if( !result )
90         goto leave;
91     if( RegQueryValueEx( key_handle, name, 0, &type, result, &n1 ) ) {
92         free(result); result = NULL;
93         goto leave;
94     }
95     result[nbytes] = 0; /* make sure it is really a string  */
96     if (type == REG_EXPAND_SZ && strchr (result, '%')) {
97         char *tmp;
98         
99         n1 += 1000;
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);
106             n1 = nbytes;
107             tmp = malloc (n1 + 1);
108             if (!tmp)
109                 goto leave;
110             nbytes = ExpandEnvironmentStrings (result, tmp, n1);
111             if (nbytes && nbytes > n1) {
112                 free (tmp); /* oops - truncated, better don't expand at all */
113                 goto leave;
114             }
115             tmp[nbytes] = 0;
116             free (result);
117             result = tmp;
118         }
119         else if (nbytes) { /* okay, reduce the length */
120             tmp[nbytes] = 0;
121             free (result);
122             result = malloc (strlen (tmp)+1);
123             if (!result)
124                 result = tmp;
125             else {
126                 strcpy (result, tmp);
127                 free (tmp);
128             }
129         }
130         else {  /* error - don't expand */
131             free (tmp);
132         }
133     }
134
135   leave:
136     RegCloseKey( key_handle );
137     return result;
138 }
139
140
141 int
142 write_w32_registry_string(const char *root, const char *dir, 
143                           const char *name, const char *value)
144 {
145     HKEY root_key, reg_key;
146         
147     if ( !(root_key = get_root_key(root) ) )
148         return -1;
149
150     if ( RegOpenKeyEx( root_key, dir, 0, KEY_WRITE, &reg_key ) 
151          != ERROR_SUCCESS )
152         return -1;
153         
154     if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value, 
155                         strlen( value ) ) != ERROR_SUCCESS ) {
156         if ( RegCreateKey( root_key, name, &reg_key ) != ERROR_SUCCESS ) {
157             RegCloseKey(reg_key);
158             return -1;
159         }
160         if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value,
161                             strlen( value ) ) != ERROR_SUCCESS ) {
162             RegCloseKey(reg_key);
163             return -1;
164         }
165     }
166
167     RegCloseKey( reg_key );
168         
169     return 0;
170 }
171
172 #endif /* __MINGW32__ || __CYGWIN32__ */