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 (__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 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__ */