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