Detect unsigned time_t and adjust y2038 detection.
[gnupg.git] / util / w32reg.c
1 /* w32reg.c -  MS-Windows Registry access
2  *      Copyright (C) 1999, 2002, 2005 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 3 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, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21 #if defined (_WIN32) || defined (__CYGWIN32__)
22  /* This module is only used in this environment */
23
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <stdarg.h>
28 #include <windows.h>
29 #include "util.h"
30 #include "memory.h"
31
32 static HKEY
33 get_root_key(const char *root)
34 {
35     HKEY root_key;
36         
37     if( !root )
38         root_key = HKEY_CURRENT_USER;
39     else if( !strcmp( root, "HKEY_CLASSES_ROOT" ) )
40         root_key = HKEY_CLASSES_ROOT;
41     else if( !strcmp( root, "HKEY_CURRENT_USER" ) )
42         root_key = HKEY_CURRENT_USER;
43     else if( !strcmp( root, "HKEY_LOCAL_MACHINE" ) )
44         root_key = HKEY_LOCAL_MACHINE;
45     else if( !strcmp( root, "HKEY_USERS" ) )
46         root_key = HKEY_USERS;
47     else if( !strcmp( root, "HKEY_PERFORMANCE_DATA" ) )
48         root_key = HKEY_PERFORMANCE_DATA;
49     else if( !strcmp( root, "HKEY_CURRENT_CONFIG" ) )
50         root_key = HKEY_CURRENT_CONFIG;
51     else
52         return NULL;
53         
54     return root_key;
55 }
56
57
58 /****************
59  * Return a string from the Win32 Registry or NULL in case of
60  * error.  Caller must release the return value.   A NULL for root
61  * is an alias for HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE in turn.
62  * NOTE: The value is allocated with a plain malloc() - use free() and not
63  * the usual xfree()!!!
64  */
65 char *
66 read_w32_registry_string( const char *root, const char *dir, const char *name )
67 {
68     HKEY root_key, key_handle;
69     DWORD n1, nbytes, type;
70     char *result = NULL;
71
72     if ( !(root_key = get_root_key(root) ) )
73         return NULL;
74
75     if( RegOpenKeyEx( root_key, dir, 0, KEY_READ, &key_handle ) )
76       {
77         if (root)
78           return NULL; /* no need for a RegClose, so return direct */
79         /* It seems to be common practise to fall back to HKLM. */
80         if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, dir, 0, KEY_READ, &key_handle) )
81           return NULL; /* still no need for a RegClose, so return direct */
82       }
83
84     nbytes = 1;
85     if( RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes ) ) {
86         if (root)
87             goto leave;
88         /* Try to fallback to HKLM also vor a missing value.  */
89         RegCloseKey (key_handle);
90         if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, dir, 0, KEY_READ, &key_handle) )
91             return NULL; /* Nope.  */
92         if (RegQueryValueEx( key_handle, name, 0, NULL, NULL, &nbytes))
93             goto leave;
94     }
95     result = malloc( (n1=nbytes+1) );
96     if( !result )
97         goto leave;
98     if( RegQueryValueEx( key_handle, name, 0, &type, result, &n1 ) ) {
99         free(result); result = NULL;
100         goto leave;
101     }
102     result[nbytes] = 0; /* make sure it is really a string  */
103     if (type == REG_EXPAND_SZ && strchr (result, '%')) {
104         char *tmp;
105         
106         n1 += 1000;
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);
113             n1 = nbytes;
114             tmp = malloc (n1 + 1);
115             if (!tmp)
116                 goto leave;
117             nbytes = ExpandEnvironmentStrings (result, tmp, n1);
118             if (nbytes && nbytes > n1) {
119                 free (tmp); /* oops - truncated, better don't expand at all */
120                 goto leave;
121             }
122             tmp[nbytes] = 0;
123             free (result);
124             result = tmp;
125         }
126         else if (nbytes) { /* okay, reduce the length */
127             tmp[nbytes] = 0;
128             free (result);
129             result = malloc (strlen (tmp)+1);
130             if (!result)
131                 result = tmp;
132             else {
133                 strcpy (result, tmp);
134                 free (tmp);
135             }
136         }
137         else {  /* error - don't expand */
138             free (tmp);
139         }
140     }
141
142   leave:
143     RegCloseKey( key_handle );
144     return result;
145 }
146
147
148 int
149 write_w32_registry_string(const char *root, const char *dir, 
150                           const char *name, const char *value)
151 {
152     HKEY root_key, reg_key;
153         
154     if ( !(root_key = get_root_key(root) ) )
155         return -1;
156
157     if ( RegOpenKeyEx( root_key, dir, 0, KEY_WRITE, &reg_key ) 
158          != ERROR_SUCCESS )
159         return -1;
160         
161     if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value, 
162                         strlen( value ) ) != ERROR_SUCCESS ) {
163         if ( RegCreateKey( root_key, name, &reg_key ) != ERROR_SUCCESS ) {
164             RegCloseKey(reg_key);
165             return -1;
166         }
167         if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value,
168                             strlen( value ) ) != ERROR_SUCCESS ) {
169             RegCloseKey(reg_key);
170             return -1;
171         }
172     }
173
174     RegCloseKey( reg_key );
175         
176     return 0;
177 }
178
179 #endif /* __MINGW32__ || __CYGWIN32__ */