Update head to match stable 1.0
[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 NUKK 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;
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, NULL, result, &n1 ) ) {
86         free(result); result = NULL;
87         goto leave;
88     }
89     result[nbytes] = 0; /* make sure it is really a string  */
90
91   leave:
92     RegCloseKey( key_handle );
93     return result;
94 }
95
96
97 int
98 write_w32_registry_string(const char *root, const char *dir, 
99                           const char *name, const char *value)
100 {
101     HKEY root_key, reg_key;
102         
103     if ( !(root_key = get_root_key(root) ) )
104         return -1;
105
106     if ( RegOpenKeyEx( root_key, dir, 0, KEY_WRITE, &reg_key ) 
107          != ERROR_SUCCESS )
108         return -1;
109         
110     if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value, 
111                         strlen( value ) ) != ERROR_SUCCESS ) {
112         if ( RegCreateKey( root_key, name, &reg_key ) != ERROR_SUCCESS ) {
113             RegCloseKey(reg_key);
114             return -1;
115         }
116         if ( RegSetValueEx( reg_key, name, 0, REG_SZ, (BYTE *)value,
117                             strlen( value ) ) != ERROR_SUCCESS ) {
118             RegCloseKey(reg_key);
119             return -1;
120         }
121     }
122
123     RegCloseKey( reg_key );
124         
125     return 0;
126 }
127
128 #endif /* __MINGW32__ || __CYGWIN32__ */