#include "regedit.h"
#include <assert.h>
#include <fcntl.h>
#include <io.h>
#include <wine/unicode.h>

Include dependency graph for regproc.c:

Macros
#define	REG_VAL_BUF_SIZE 4096

#define	REG_FILE_HEX_LINE_LEN (2 + 25 * 3)

#define	REG_CLASS_NUMBER (sizeof(reg_class_keys) / sizeof(reg_class_keys[0]))

#define	NOT_ENOUGH_MEMORY 1

#define	IO_ERROR 2

#define	CHECK_ENOUGH_MEMORY(p)

Functions
WCHAR *	GetWideString (const char *strA)

static WCHAR *	GetWideStringN (const char strA, int chars, DWORD len)

char *	GetMultiByteString (const WCHAR *strW)

static char *	GetMultiByteStringN (const WCHAR strW, int chars, DWORD len)

static BOOL	convertHexToDWord (WCHAR str, DWORD dw)

static BYTE *	convertHexCSVToHex (WCHAR str, DWORD size)

static DWORD	getDataType (LPWSTR lpValue, DWORD parse_type)

static void	REGPROC_unescape_string (WCHAR *str)

static BOOL	parseKeyName (LPWSTR lpKeyName, HKEY hKey, LPWSTR lpKeyPath)

static LONG	setValue (WCHAR val_name, WCHAR val_data, BOOL is_unicode)

static LONG	openKeyW (WCHAR *stdInput)

static void	closeKey (void)

static void	processSetValue (WCHAR *line, BOOL is_unicode)

static void	processRegEntry (WCHAR *stdInput, BOOL isUnicode)

static void	processRegLinesA (FILE in, char first_chars)

static void	processRegLinesW (FILE *in)

static void	REGPROC_print_error (void)

static void	REGPROC_resize_char_buffer (WCHAR *buffer, DWORD len, DWORD required_len)

static void	REGPROC_resize_binary_buffer (BYTE *buffer, DWORD size, DWORD required_size)

static void	REGPROC_export_string (WCHAR *line_buf, DWORD line_buf_size, DWORD line_len, WCHAR str, DWORD str_len)

static void	REGPROC_export_binary (WCHAR *line_buf, DWORD line_buf_size, DWORD line_len, DWORD type, BYTE value, DWORD value_size, BOOL unicode)

static void	REGPROC_write_line (FILE file, const WCHAR str, BOOL unicode)

static void	export_hkey (FILE file, HKEY key, WCHAR reg_key_name_buf, DWORD reg_key_name_size, WCHAR *val_name_buf, DWORD val_name_size, BYTE *val_buf, DWORD val_size, WCHAR *line_buf, DWORD line_buf_size, BOOL unicode)

static FILE *	REGPROC_open_export_file (WCHAR *file_name, BOOL unicode)

BOOL	export_registry_key (WCHAR file_name, WCHAR reg_key_name, DWORD format)

BOOL	import_registry_file (FILE *reg_file)

void	delete_registry_key (WCHAR *reg_key_name)

Variables
const WCHAR *	reg_class_namesW []

static HKEY	reg_class_keys []

static LPSTR	currentKeyName

static HKEY	currentKeyHandle = NULL

Macro Definition Documentation

◆ CHECK_ENOUGH_MEMORY

#define CHECK_ENOUGH_MEMORY ( p )

Value:

if (!(p)) \
{ \
    fprintf(stderr,"%S: file %s, line %d: Not enough memory\n", \
            getAppName(), __FILE__, __LINE__); \
    exit(NOT_ENOUGH_MEMORY); \
}

Definition at line 58 of file regproc.c.

Referenced by convertHexCSVToHex(), export_registry_key(), GetMultiByteString(), GetMultiByteStringN(), GetWideString(), GetWideStringN(), processRegLinesA(), processRegLinesW(), REGPROC_resize_binary_buffer(), and REGPROC_resize_char_buffer().

◆ IO_ERROR

#define IO_ERROR 2

Definition at line 53 of file regproc.c.

Referenced by processRegLinesA(), and processRegLinesW().

◆ NOT_ENOUGH_MEMORY

#define NOT_ENOUGH_MEMORY 1

Definition at line 52 of file regproc.c.

◆ REG_CLASS_NUMBER

#define REG_CLASS_NUMBER (sizeof(reg_class_keys) / sizeof(reg_class_keys[0]))

Definition at line 49 of file regproc.c.

Referenced by export_registry_key(), and parseKeyName().

◆ REG_FILE_HEX_LINE_LEN

#define REG_FILE_HEX_LINE_LEN (2 + 25 * 3)

Definition at line 36 of file regproc.c.

Referenced by REGPROC_export_binary().

◆ REG_VAL_BUF_SIZE

#define REG_VAL_BUF_SIZE 4096

Definition at line 31 of file regproc.c.

Referenced by export_registry_key(), processRegLinesA(), and processRegLinesW().

Function Documentation

◆ closeKey()

static void closeKey ( void )

static

Definition at line 465 of file regproc.c.

Referenced by processRegEntry().

 {
     if (currentKeyHandle)
     {
         HeapFree(GetProcessHeap(), 0, currentKeyName);
         RegCloseKey(currentKeyHandle);
         currentKeyHandle = NULL;
     }
 }

◆ convertHexCSVToHex()

static BYTE* convertHexCSVToHex	(	WCHAR *	str,
		DWORD *	size
	)

static

Definition at line 163 of file regproc.c.

Referenced by setValue().

 {
     WCHAR *s;
     BYTE *d, *data;
 
     /* The worst case is 1 digit + 1 comma per byte */
     *size=(lstrlenW(str)+1)/2;
     data=HeapAlloc(GetProcessHeap(), 0, *size);
     CHECK_ENOUGH_MEMORY(data);
 
     s = str;
     d = data;
     *size=0;
     while (*s != '\0') {
         UINT wc;
         WCHAR *end;
 
         wc = strtoulW(s,&end,16);
         if (end == s || wc > 0xff || (*end && *end != ',')) {
             char* strA = GetMultiByteString(s);
             fprintf(stderr,"%S: ERROR converting CSV hex stream. Invalid value at '%s'\n",
                     getAppName(), strA);
             HeapFree(GetProcessHeap(), 0, data);
             HeapFree(GetProcessHeap(), 0, strA);
             return NULL;
         }
         *d++ =(BYTE)wc;
         (*size)++;
         if (*end) end++;
         s = end;
     }
 
     return data;
 }

◆ convertHexToDWord()

static BOOL convertHexToDWord	(	WCHAR *	str,
		DWORD *	dw
	)

static

Definition at line 147 of file regproc.c.

Referenced by setValue().

 {
     char buf[9];
     char dummy;
 
     WideCharToMultiByte(CP_ACP, 0, str, -1, buf, 9, NULL, NULL);
     if (lstrlenW(str) > 8 || sscanf(buf, "%lx%c", dw, &dummy) != 1) {
         fprintf(stderr,"%S: ERROR, invalid hex value\n", getAppName());
         return FALSE;
     }
     return TRUE;
 }

◆ delete_registry_key()

void delete_registry_key ( WCHAR * reg_key_name )

Definition at line 1432 of file regproc.c.

Referenced by PerformRegAction(), and processRegEntry().

 {
     WCHAR *key_name = NULL;
     HKEY key_class;
 
     if (!reg_key_name || !reg_key_name[0])
         return;
 
     if (!parseKeyName(reg_key_name, &key_class, &key_name)) {
         char* reg_key_nameA = GetMultiByteString(reg_key_name);
         fprintf(stderr,"%S: Incorrect registry class specification in '%s'\n",
                 getAppName(), reg_key_nameA);
         HeapFree(GetProcessHeap(), 0, reg_key_nameA);
         exit(1);
     }
     if (!*key_name) {
         char* reg_key_nameA = GetMultiByteString(reg_key_name);
         fprintf(stderr,"%S: Can't delete registry class '%s'\n",
                 getAppName(), reg_key_nameA);
         HeapFree(GetProcessHeap(), 0, reg_key_nameA);
         exit(1);
     }
 
     SHDeleteKey(key_class, key_name);
 }

◆ export_hkey()

static void export_hkey	(	FILE *	file,
		HKEY	key,
		WCHAR **	reg_key_name_buf,
		DWORD *	reg_key_name_size,
		WCHAR **	val_name_buf,
		DWORD *	val_name_size,
		BYTE **	val_buf,
		DWORD *	val_size,
		WCHAR **	line_buf,
		DWORD *	line_buf_size,
		BOOL	unicode
	)

static

Definition at line 1096 of file regproc.c.

Referenced by export_registry_key().

 {
     DWORD max_sub_key_len;
     DWORD max_val_name_len;
     DWORD max_val_size;
     DWORD curr_len;
     DWORD i;
     BOOL more_data;
     LONG ret;
     WCHAR key_format[] = {'\r','\n','[','%','s',']','\r','\n',0};
 
     /* get size information and resize the buffers if necessary */
     if (RegQueryInfoKeyW(key, NULL, NULL, NULL, NULL,
                         &max_sub_key_len, NULL,
                         NULL, &max_val_name_len, &max_val_size, NULL, NULL
                        ) != ERROR_SUCCESS) {
         REGPROC_print_error();
     }
     curr_len = strlenW(*reg_key_name_buf);
     REGPROC_resize_char_buffer(reg_key_name_buf, reg_key_name_size,
                                max_sub_key_len + curr_len + 1);
     REGPROC_resize_char_buffer(val_name_buf, val_name_size,
                                max_val_name_len);
     REGPROC_resize_binary_buffer(val_buf, val_size, max_val_size);
     REGPROC_resize_char_buffer(line_buf, line_buf_size, lstrlenW(*reg_key_name_buf) + 4);
     /* output data for the current key */
     sprintfW(*line_buf, key_format, *reg_key_name_buf);
     REGPROC_write_line(file, *line_buf, unicode);
 
     /* print all the values */
     i = 0;
     more_data = TRUE;
     while(more_data) {
         DWORD value_type;
         DWORD val_name_size1 = *val_name_size;
         DWORD val_size1 = *val_size;
         ret = RegEnumValueW(key, i, *val_name_buf, &val_name_size1, NULL,
                            &value_type, *val_buf, &val_size1);
         if (ret == ERROR_MORE_DATA) {
             /* Increase the size of the buffers and retry */
             REGPROC_resize_char_buffer(val_name_buf, val_name_size, val_name_size1);
             REGPROC_resize_binary_buffer(val_buf, val_size, val_size1);
         } else if (ret != ERROR_SUCCESS) {
             more_data = FALSE;
             if (ret != ERROR_NO_MORE_ITEMS) {
                 REGPROC_print_error();
             }
         } else {
             DWORD line_len;
             i++;
 
             if ((*val_name_buf)[0]) {
                 const WCHAR val_start[] = {'"','%','s','"','=',0};
 
                 line_len = 0;
                 REGPROC_export_string(line_buf, line_buf_size, &line_len, *val_name_buf, lstrlenW(*val_name_buf));
                 REGPROC_resize_char_buffer(val_name_buf, val_name_size, lstrlenW(*line_buf) + 1);
                 lstrcpyW(*val_name_buf, *line_buf);
 
                 line_len = 3 + lstrlenW(*val_name_buf);
                 REGPROC_resize_char_buffer(line_buf, line_buf_size, line_len);
                 sprintfW(*line_buf, val_start, *val_name_buf);
             } else {
                 const WCHAR std_val[] = {'@','=',0};
                 line_len = 2;
                 REGPROC_resize_char_buffer(line_buf, line_buf_size, line_len);
                 lstrcpyW(*line_buf, std_val);
             }
 
             switch (value_type) {
             case REG_SZ:
             {
                 WCHAR* wstr = (WCHAR*)*val_buf;
 
                 if (val_size1 < sizeof(WCHAR) || val_size1 % sizeof(WCHAR) ||
                     wstr[val_size1 / sizeof(WCHAR) - 1]) {
                     REGPROC_export_binary(line_buf, line_buf_size, &line_len, value_type, *val_buf, val_size1, unicode);
                 } else {
                     const WCHAR start[] = {'"',0};
                     const WCHAR end[] = {'"','\r','\n',0};
                     DWORD len;
 
                     len = lstrlenW(start);
                     REGPROC_resize_char_buffer(line_buf, line_buf_size, line_len + len);
                     lstrcpyW(*line_buf + line_len, start);
                     line_len += len;
 
                     /* At this point we know wstr is '\0'-terminated
                      * so we can subtract 1 from the size
                      */
                     REGPROC_export_string(line_buf, line_buf_size, &line_len, wstr, val_size1 / sizeof(WCHAR) - 1);
 
                     REGPROC_resize_char_buffer(line_buf, line_buf_size, line_len + lstrlenW(end));
                     lstrcpyW(*line_buf + line_len, end);
                 }
                 break;
             }
 
             case REG_DWORD:
             {
                 WCHAR format[] = {'d','w','o','r','d',':','%','0','8','x','\r','\n',0};
 
                 REGPROC_resize_char_buffer(line_buf, line_buf_size, line_len + 15);
                 sprintfW(*line_buf + line_len, format, *((DWORD *)*val_buf));
                 break;
             }
 
             default:
             {
                 char* key_nameA = GetMultiByteString(*reg_key_name_buf);
                 char* value_nameA = GetMultiByteString(*val_name_buf);
                 fprintf(stderr,"%S: warning - unsupported registry format '%ld', "
                         "treat as binary\n",
                         getAppName(), value_type);
                 fprintf(stderr,"key name: \"%s\"\n", key_nameA);
                 fprintf(stderr,"value name:\"%s\"\n\n", value_nameA);
                 HeapFree(GetProcessHeap(), 0, key_nameA);
                 HeapFree(GetProcessHeap(), 0, value_nameA);
             }
                 /* falls through */
             case REG_EXPAND_SZ:
             case REG_MULTI_SZ:
                 /* falls through */
             case REG_BINARY:
                 REGPROC_export_binary(line_buf, line_buf_size, &line_len, value_type, *val_buf, val_size1, unicode);
             }
             REGPROC_write_line(file, *line_buf, unicode);
         }
     }
 
     i = 0;
     more_data = TRUE;
     (*reg_key_name_buf)[curr_len] = '\\';
     while(more_data) {
         DWORD buf_size = *reg_key_name_size - curr_len - 1;
 
         ret = RegEnumKeyExW(key, i, *reg_key_name_buf + curr_len + 1, &buf_size,
                            NULL, NULL, NULL, NULL);
         if (ret == ERROR_MORE_DATA) {
             /* Increase the size of the buffer and retry */
             REGPROC_resize_char_buffer(reg_key_name_buf, reg_key_name_size, curr_len + 1 + buf_size);
         } else if (ret != ERROR_SUCCESS) {
             more_data = FALSE;
             if (ret != ERROR_NO_MORE_ITEMS) {
                 REGPROC_print_error();
             }
         } else {
             HKEY subkey;
 
             i++;
             if (RegOpenKeyW(key, *reg_key_name_buf + curr_len + 1,
                            &subkey) == ERROR_SUCCESS) {
                 export_hkey(file, subkey, reg_key_name_buf, reg_key_name_size,
                             val_name_buf, val_name_size, val_buf, val_size,
                             line_buf, line_buf_size, unicode);
                 RegCloseKey(subkey);
             } else {
                 REGPROC_print_error();
             }
         }
     }
     (*reg_key_name_buf)[curr_len] = '\0';
 }

◆ export_registry_key()

BOOL export_registry_key	(	WCHAR *	file_name,
		WCHAR *	reg_key_name,
		DWORD	format
	)

Definition at line 1310 of file regproc.c.

Referenced by ExportRegistryFile(), and PerformRegAction().

 {
     WCHAR *reg_key_name_buf;
     WCHAR *val_name_buf;
     BYTE *val_buf;
     WCHAR *line_buf;
     DWORD reg_key_name_size = KEY_MAX_LEN;
     DWORD val_name_size = KEY_MAX_LEN;
     DWORD val_size = REG_VAL_BUF_SIZE;
     DWORD line_buf_size = KEY_MAX_LEN + REG_VAL_BUF_SIZE;
     FILE *file = NULL;
     BOOL unicode = (format == REG_FORMAT_5);
 
     reg_key_name_buf = HeapAlloc(GetProcessHeap(), 0,
                                  reg_key_name_size  * sizeof(*reg_key_name_buf));
     val_name_buf = HeapAlloc(GetProcessHeap(), 0,
                              val_name_size * sizeof(*val_name_buf));
     val_buf = HeapAlloc(GetProcessHeap(), 0, val_size);
     line_buf = HeapAlloc(GetProcessHeap(), 0, line_buf_size * sizeof(*line_buf));
     CHECK_ENOUGH_MEMORY(reg_key_name_buf && val_name_buf && val_buf && line_buf);
 
     if (reg_key_name && reg_key_name[0]) {
         HKEY reg_key_class;
         WCHAR *branch_name = NULL;
         HKEY key;
 
         REGPROC_resize_char_buffer(&reg_key_name_buf, &reg_key_name_size,
                                    lstrlenW(reg_key_name));
         lstrcpyW(reg_key_name_buf, reg_key_name);
 
         /* open the specified key */
         if (!parseKeyName(reg_key_name, &reg_key_class, &branch_name)) {
             CHAR* key_nameA = GetMultiByteString(reg_key_name);
             fprintf(stderr,"%S: Incorrect registry class specification in '%s'\n",
                     getAppName(), key_nameA);
             HeapFree(GetProcessHeap(), 0, key_nameA);
             exit(1);
         }
         if (!branch_name[0]) {
             /* no branch - registry class is specified */
             file = REGPROC_open_export_file(file_name, unicode);
             export_hkey(file, reg_key_class,
                         &reg_key_name_buf, &reg_key_name_size,
                         &val_name_buf, &val_name_size,
                         &val_buf, &val_size, &line_buf,
                         &line_buf_size, unicode);
         } else if (RegOpenKeyW(reg_key_class, branch_name, &key) == ERROR_SUCCESS) {
             file = REGPROC_open_export_file(file_name, unicode);
             export_hkey(file, key,
                         &reg_key_name_buf, &reg_key_name_size,
                         &val_name_buf, &val_name_size,
                         &val_buf, &val_size, &line_buf,
                         &line_buf_size, unicode);
             RegCloseKey(key);
         } else {
             CHAR* key_nameA = GetMultiByteString(reg_key_name);
             fprintf(stderr,"%S: Can't export. Registry key '%s' does not exist!\n",
                     getAppName(), key_nameA);
             HeapFree(GetProcessHeap(), 0, key_nameA);
             REGPROC_print_error();
         }
     } else {
         unsigned int i;
 
         /* export all registry classes */
         file = REGPROC_open_export_file(file_name, unicode);
         for (i = 0; i < REG_CLASS_NUMBER; i++) {
             /* do not export HKEY_CLASSES_ROOT */
             if (reg_class_keys[i] != HKEY_CLASSES_ROOT &&
                     reg_class_keys[i] != HKEY_CURRENT_USER &&
                     reg_class_keys[i] != HKEY_CURRENT_CONFIG &&
                     reg_class_keys[i] != HKEY_DYN_DATA) {
                 lstrcpyW(reg_key_name_buf, reg_class_namesW[i]);
                 export_hkey(file, reg_class_keys[i],
                             &reg_key_name_buf, &reg_key_name_size,
                             &val_name_buf, &val_name_size,
                             &val_buf, &val_size, &line_buf,
                             &line_buf_size, unicode);
             }
         }
     }
 
     if (file) {
         fclose(file);
     }
     HeapFree(GetProcessHeap(), 0, reg_key_name);
     HeapFree(GetProcessHeap(), 0, val_name_buf);
     HeapFree(GetProcessHeap(), 0, val_buf);
     HeapFree(GetProcessHeap(), 0, line_buf);
     return TRUE;
 }

◆ getDataType()

static DWORD getDataType	(	LPWSTR *	lpValue,
		DWORD *	parse_type
	)

static

Definition at line 205 of file regproc.c.

Referenced by setValue().

 {
     struct data_type { const WCHAR *tag; int len; int type; int parse_type; };
 
     static const WCHAR quote[] = {'"'};
     static const WCHAR str[] = {'s','t','r',':','"'};
     static const WCHAR str2[] = {'s','t','r','(','2',')',':','"'};
     static const WCHAR hex[] = {'h','e','x',':'};
     static const WCHAR dword[] = {'d','w','o','r','d',':'};
     static const WCHAR hexp[] = {'h','e','x','('};
 
     static const struct data_type data_types[] = {                   /* actual type */  /* type to assume for parsing */
                 { quote,       1,   REG_SZ,              REG_SZ },
                 { str,         5,   REG_SZ,              REG_SZ },
                 { str2,        8,   REG_EXPAND_SZ,       REG_SZ },
                 { hex,         4,   REG_BINARY,          REG_BINARY },
                 { dword,       6,   REG_DWORD,           REG_DWORD },
                 { hexp,        4,   -1,                  REG_BINARY },
                 { NULL,        0,    0,                  0 }
             };
 
     const struct data_type *ptr;
     int type;
 
     for (ptr = data_types; ptr->tag; ptr++) {
         if (strncmpW( ptr->tag, *lpValue, ptr->len ))
             continue;
 
         /* Found! */
         *parse_type = ptr->parse_type;
         type=ptr->type;
         *lpValue+=ptr->len;
         if (type == -1) {
             WCHAR* end;
 
             /* "hex(xx):" is special */
             type = (int)strtoulW( *lpValue , &end, 16 );
             if (**lpValue=='\0' || *end!=')' || *(end+1)!=':') {
                 type=REG_NONE;
             } else {
                 *lpValue = end + 2;
             }
         }
         return type;
     }
     *parse_type=REG_NONE;
     return REG_NONE;
 }

◆ GetMultiByteString()

char* GetMultiByteString ( const WCHAR * strW )

Definition at line 109 of file regproc.c.

Referenced by convertHexCSVToHex(), delete_registry_key(), export_hkey(), export_registry_key(), openKeyW(), processRegEntry(), processSetValue(), REGPROC_open_export_file(), and REGPROC_write_line().

 {
     if(strW)
     {
         char* strA;
         int len = WideCharToMultiByte(CP_ACP, 0, strW, -1, NULL, 0, NULL, NULL);
 
         strA = HeapAlloc(GetProcessHeap(), 0, len);
         CHECK_ENOUGH_MEMORY(strA);
         WideCharToMultiByte(CP_ACP, 0, strW, -1, strA, len, NULL, NULL);
         return strA;
     }
     return NULL;
 }

◆ GetMultiByteStringN()

static char* GetMultiByteStringN	(	const WCHAR *	strW,
		int	chars,
		DWORD *	len
	)

static

Definition at line 128 of file regproc.c.

Referenced by REGPROC_export_binary().

 {
     if(strW)
     {
         char* strA;
         *len = WideCharToMultiByte(CP_ACP, 0, strW, chars, NULL, 0, NULL, NULL);
 
         strA = HeapAlloc(GetProcessHeap(), 0, *len);
         CHECK_ENOUGH_MEMORY(strA);
         WideCharToMultiByte(CP_ACP, 0, strW, chars, strA, *len, NULL, NULL);
         return strA;
     }
     *len = 0;
     return NULL;
 }

◆ GetWideString()

WCHAR* GetWideString ( const char * strA )

Definition at line 70 of file regproc.c.

Referenced by processRegLinesA().

 {
     if(strA)
     {
         WCHAR* strW;
         int len = MultiByteToWideChar(CP_ACP, 0, strA, -1, NULL, 0);
 
         strW = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
         CHECK_ENOUGH_MEMORY(strW);
         MultiByteToWideChar(CP_ACP, 0, strA, -1, strW, len);
         return strW;
     }
     return NULL;
 }

◆ GetWideStringN()

static WCHAR* GetWideStringN	(	const char *	strA,
		int	chars,
		DWORD *	len
	)

static

Definition at line 89 of file regproc.c.

Referenced by setValue().

 {
     if(strA)
     {
         WCHAR* strW;
         *len = MultiByteToWideChar(CP_ACP, 0, strA, chars, NULL, 0);
 
         strW = HeapAlloc(GetProcessHeap(), 0, *len * sizeof(WCHAR));
         CHECK_ENOUGH_MEMORY(strW);
         MultiByteToWideChar(CP_ACP, 0, strA, chars, strW, *len);
         return strW;
     }
     *len = 0;
     return NULL;
 }

◆ import_registry_file()

BOOL import_registry_file ( FILE * reg_file )

Definition at line 1405 of file regproc.c.

Referenced by ImportRegistryFile(), and PerformRegAction().

 {
     if (reg_file)
     {
         BYTE s[2];
         if (fread( s, 2, 1, reg_file) == 1)
         {
             if (s[0] == 0xff && s[1] == 0xfe)
             {
                 processRegLinesW(reg_file);
             } else
             {
                 processRegLinesA(reg_file, (char*)s);
             }
         }
         return TRUE;
     }
     return FALSE;
 }

◆ openKeyW()

static LONG openKeyW ( WCHAR * stdInput )

static

Definition at line 426 of file regproc.c.

Referenced by processRegEntry().

 {
     HKEY keyClass;
     WCHAR* keyPath;
     DWORD dwDisp;
     LONG res;
 
     /* Sanity checks */
     if (stdInput == NULL)
         return ERROR_INVALID_PARAMETER;
 
     /* Get the registry class */
     if (!parseKeyName(stdInput, &keyClass, &keyPath))
         return ERROR_INVALID_PARAMETER;
 
     res = RegCreateKeyExW(
                keyClass,                 /* Class     */
                keyPath,                  /* Sub Key   */
                0,                        /* MUST BE 0 */
                NULL,                     /* object type */
                REG_OPTION_NON_VOLATILE,  /* option, REG_OPTION_NON_VOLATILE ... */
                KEY_ALL_ACCESS,           /* access mask, KEY_ALL_ACCESS */
                NULL,                     /* security attribute */
                &currentKeyHandle,        /* result */
                &dwDisp);                 /* disposition, REG_CREATED_NEW_KEY or
                                                         REG_OPENED_EXISTING_KEY */
 
     if (res == ERROR_SUCCESS)
         currentKeyName = GetMultiByteString(stdInput);
     else
         currentKeyHandle = NULL;
 
     return res;
 
 }

◆ parseKeyName()

static BOOL parseKeyName	(	LPWSTR	lpKeyName,
		HKEY *	hKey,
		LPWSTR *	lpKeyPath
	)

static

Definition at line 286 of file regproc.c.

Referenced by delete_registry_key(), export_registry_key(), and openKeyW().

 {
     WCHAR* lpSlash = NULL;
     unsigned int i, len;
 
     if (lpKeyName == NULL)
         return FALSE;
 
     for(i = 0; *(lpKeyName+i) != 0; i++)
     {
         if(*(lpKeyName+i) == '\\')
         {
             lpSlash = lpKeyName+i;
             break;
         }
     }
 
     if (lpSlash)
     {
         len = lpSlash-lpKeyName;
     }
     else
     {
         len = lstrlenW(lpKeyName);
         lpSlash = lpKeyName+len;
     }
     *hKey = NULL;
 
     for (i = 0; i < REG_CLASS_NUMBER; i++) {
         if (CompareStringW(LOCALE_USER_DEFAULT, 0, lpKeyName, len, reg_class_namesW[i], len) == CSTR_EQUAL &&
             len == lstrlenW(reg_class_namesW[i])) {
             *hKey = reg_class_keys[i];
             break;
         }
     }
 
     if (*hKey == NULL)
         return FALSE;
 
 
     if (*lpSlash != '\0')
         lpSlash++;
     *lpKeyPath = lpSlash;
     return TRUE;
 }

◆ processRegEntry()

static void processRegEntry	(	WCHAR *	stdInput,
		BOOL	isUnicode
	)

static

Definition at line 563 of file regproc.c.

Referenced by processRegLinesA(), and processRegLinesW().

 {
     /*
      * We encountered the end of the file, make sure we
      * close the opened key and exit
      */
     if (stdInput == NULL) {
         closeKey();
         return;
     }
 
     if      ( stdInput[0] == '[')      /* We are reading a new key */
     {
         WCHAR* keyEnd;
         closeKey();                    /* Close the previous key */
 
         /* Get rid of the square brackets */
         stdInput++;
         keyEnd = strrchrW(stdInput, ']');
         if (keyEnd)
             *keyEnd='\0';
 
         /* delete the key if we encounter '-' at the start of reg key */
         if ( stdInput[0] == '-')
         {
             delete_registry_key(stdInput + 1);
         } else if ( openKeyW(stdInput) != ERROR_SUCCESS )
         {
             char* stdInputA = GetMultiByteString(stdInput);
             fprintf(stderr,"%S: setValue failed to open key %s\n",
                     getAppName(), stdInputA);
             HeapFree(GetProcessHeap(), 0, stdInputA);
         }
     } else if( currentKeyHandle &&
                (( stdInput[0] == '@') || /* reading a default @=data pair */
                 ( stdInput[0] == '\"'))) /* reading a new value=data pair */
     {
         processSetValue(stdInput, isUnicode);
     } else
     {
         /* Since we are assuming that the file format is valid we must be
          * reading a blank line which indicates the end of this key processing
          */
         closeKey();
     }
 }

◆ processRegLinesA()

static void processRegLinesA	(	FILE *	in,
		char *	first_chars
	)

static

Definition at line 618 of file regproc.c.

Referenced by import_registry_file().

 {
     LPSTR line           = NULL;  /* line read from input stream */
     ULONG lineSize       = REG_VAL_BUF_SIZE;
 
     line = HeapAlloc(GetProcessHeap(), 0, lineSize);
     CHECK_ENOUGH_MEMORY(line);
     memcpy(line, first_chars, 2);
 
     while (!feof(in)) {
         LPSTR s; /* The pointer into line for where the current fgets should read */
         WCHAR* lineW;
         s = line;
 
         if(first_chars)
         {
             s += 2;
             first_chars = NULL;
         }
 
         for (;;) {
             size_t size_remaining;
             int size_to_get, i;
             char *s_eol; /* various local uses */
 
             /* Do we need to expand the buffer ? */
             assert (s >= line && s <= line + lineSize);
             size_remaining = lineSize - (s-line);
             if (size_remaining < 2) /* room for 1 character and the \0 */
             {
                 char *new_buffer;
                 size_t new_size = lineSize + REG_VAL_BUF_SIZE;
                 if (new_size > lineSize) /* no arithmetic overflow */
                     new_buffer = HeapReAlloc (GetProcessHeap(), 0, line, new_size);
                 else
                     new_buffer = NULL;
                 CHECK_ENOUGH_MEMORY(new_buffer);
                 line = new_buffer;
                 s = line + lineSize - size_remaining;
                 lineSize = new_size;
                 size_remaining = lineSize - (s-line);
             }
 
             /* Get as much as possible into the buffer, terminated either by
              * eof, error, eol or getting the maximum amount.  Abort on error.
              */
             size_to_get = (size_remaining > INT_MAX ? INT_MAX : size_remaining);
 
             /* get a single line. note that `i' must be one past the last
              * meaningful character in `s' when this loop exits */
             for(i = 0; i < size_to_get-1; ++i){
                 int xchar;
 
                 xchar = fgetc(in);
                 s[i] = xchar;
                 if(xchar == EOF){
                     if(ferror(in)){
                         perror("While reading input");
                         exit(IO_ERROR);
                     }else
                         assert(feof(in));
                     break;
                 }
                 if(s[i] == '\r'){
                     /* read the next character iff it's \n */
                     if(i+2 >= size_to_get){
                         /* buffer too short, so put back the EOL char to
                          * read next cycle */
                         ungetc('\r', in);
                         break;
                     }
                     s[i+1] = fgetc(in);
                     if(s[i+1] != '\n'){
                         ungetc(s[i+1], in);
                         i = i+1;
                     }else
                         i = i+2;
                     break;
                 }
                 if(s[i] == '\n'){
                     i = i+1;
                     break;
                 }
             }
             s[i] = '\0';
 
             /* If we didn't read the eol nor the eof go around for the rest */
             s_eol = strpbrk (s, "\r\n");
             if (!feof (in) && !s_eol) {
                 s = strchr (s, '\0');
                 continue;
             }
 
             /* If it is a comment line then discard it and go around again */
             if (line [0] == '#' || line [0] == ';') {
                 s = line;
                 continue;
             }
 
             /* Remove any line feed.  Leave s_eol on the first \0 */
             if (s_eol) {
                if (*s_eol == '\r' && *(s_eol+1) == '\n')
                    *(s_eol+1) = '\0';
                *s_eol = '\0';
             } else
                 s_eol = strchr (s, '\0');
 
             /* If there is a concatenating \\ then go around again */
             if (s_eol > line && *(s_eol-1) == '\\') {
                 int c;
                 s = s_eol-1;
 
                 do
                 {
                     c = fgetc(in);
                 } while(c == ' ' || c == '\t');
 
                 if(c == EOF)
                 {
                     fprintf(stderr,"%S: ERROR - invalid continuation.\n",
                             getAppName());
                 }
                 else
                 {
                     *s = c;
                     s++;
                 }
                 continue;
             }
 
             lineW = GetWideString(line);
 
             break; /* That is the full virtual line */
         }
 
         processRegEntry(lineW, FALSE);
         HeapFree(GetProcessHeap(), 0, lineW);
     }
     processRegEntry(NULL, FALSE);
 
     HeapFree(GetProcessHeap(), 0, line);
 }

◆ processRegLinesW()

static void processRegLinesW ( FILE * in )

static

Definition at line 761 of file regproc.c.

Referenced by import_registry_file().

 {
     WCHAR* buf           = NULL;  /* line read from input stream */
     ULONG lineSize       = REG_VAL_BUF_SIZE;
     size_t CharsInBuf = -1;
 
     WCHAR* s; /* The pointer into buf for where the current fgets should read */
     WCHAR* line; /* The start of the current line */
 
     buf = HeapAlloc(GetProcessHeap(), 0, lineSize * sizeof(WCHAR));
     CHECK_ENOUGH_MEMORY(buf);
 
     s = buf;
     line = buf;
 
     while(!feof(in)) {
         size_t size_remaining;
         int size_to_get;
         WCHAR *s_eol = NULL; /* various local uses */
 
         /* Do we need to expand the buffer ? */
         assert (s >= buf && s <= buf + lineSize);
         size_remaining = lineSize - (s-buf);
         if (size_remaining < 2) /* room for 1 character and the \0 */
         {
             WCHAR *new_buffer;
             size_t new_size = lineSize + (REG_VAL_BUF_SIZE / sizeof(WCHAR));
             if (new_size > lineSize) /* no arithmetic overflow */
                 new_buffer = HeapReAlloc (GetProcessHeap(), 0, buf, new_size * sizeof(WCHAR));
             else
                 new_buffer = NULL;
             CHECK_ENOUGH_MEMORY(new_buffer);
             buf = new_buffer;
             line = buf;
             s = buf + lineSize - size_remaining;
             lineSize = new_size;
             size_remaining = lineSize - (s-buf);
         }
 
         /* Get as much as possible into the buffer, terminated either by
         * eof, error or getting the maximum amount.  Abort on error.
         */
         size_to_get = (size_remaining > INT_MAX ? INT_MAX : size_remaining);
 
         CharsInBuf = fread(s, sizeof(WCHAR), size_to_get - 1, in);
         s[CharsInBuf] = 0;
 
         if (CharsInBuf == 0) {
             if (ferror(in)) {
                 perror ("While reading input");
                 exit (IO_ERROR);
             } else {
                 assert (feof(in));
                 *s = '\0';
                 /* It is not clear to me from the definition that the
                 * contents of the buffer are well defined on detecting
                 * an eof without managing to read anything.
                 */
             }
         }
 
         /* If we didn't read the eol nor the eof go around for the rest */
         while(1)
         {
             const WCHAR line_endings[] = {'\r','\n',0};
             s_eol = strpbrkW(line, line_endings);
 
             if(!s_eol) {
                 /* Move the stub of the line to the start of the buffer so
                  * we get the maximum space to read into, and so we don't
                  * have to recalculate 'line' if the buffer expands */
                 MoveMemory(buf, line, (strlenW(line)+1) * sizeof(WCHAR));
                 line = buf;
                 s = strchrW(line, '\0');
                 break;
             }
 
             /* If it is a comment line then discard it and go around again */
             if (*line == '#' || *line == ';') {
                 if (*s_eol == '\r' && *(s_eol+1) == '\n')
                     line = s_eol + 2;
                 else
                     line = s_eol + 1;
                 continue;
             }
 
             /* If there is a concatenating \\ then go around again */
             if (*(s_eol-1) == '\\') {
                 WCHAR* NextLine = s_eol + 1;
 
                 if(*s_eol == '\r' && *(s_eol+1) == '\n')
                     NextLine++;
 
                 while(isspaceW(*NextLine))
                     NextLine++;
 
                 if (!*NextLine)
                 {
                     s = NextLine;
                     break;
                 }
 
                 MoveMemory(s_eol - 1, NextLine, (CharsInBuf - (NextLine - s) + 1)*sizeof(WCHAR));
                 CharsInBuf -= NextLine - s_eol + 1;
                 s_eol = 0;
                 continue;
             }
 
             /* Remove any line feed.  Leave s_eol on the last \0 */
             if (*s_eol == '\r' && *(s_eol + 1) == '\n')
                 *s_eol++ = '\0';
             *s_eol = '\0';
 
             processRegEntry(line, TRUE);
             line = s_eol + 1;
             s_eol = 0;
             continue; /* That is the full virtual line */
         }
     }
 
     processRegEntry(NULL, TRUE);
 
     HeapFree(GetProcessHeap(), 0, buf);
 }

◆ processSetValue()

static void processSetValue	(	WCHAR *	line,
		BOOL	is_unicode
	)

static

Definition at line 483 of file regproc.c.

Referenced by processRegEntry().

 {
     WCHAR* val_name;                   /* registry value name   */
     WCHAR* val_data;                   /* registry value data   */
     int line_idx = 0;                 /* current character under analysis */
     LONG res;
 
     /* get value name */
     while ( isspaceW(line[line_idx]) ) line_idx++;
     if (line[line_idx] == '@' && line[line_idx + 1] == '=') {
         line[line_idx] = '\0';
         val_name = line;
         line_idx++;
     } else if (line[line_idx] == '\"') {
         line_idx++;
         val_name = line + line_idx;
         while (line[line_idx]) {
             if (line[line_idx] == '\\')   /* skip escaped character */
             {
                 line_idx += 2;
             } else {
                 if (line[line_idx] == '\"') {
                     line[line_idx] = '\0';
                     line_idx++;
                     break;
                 } else {
                     line_idx++;
                 }
             }
         }
         while ( isspaceW(line[line_idx]) ) line_idx++;
         if (!line[line_idx]) {
             fprintf(stderr, "%S: warning: unexpected EOL\n", getAppName());
             return;
         }
         if (line[line_idx] != '=') {
             char* lineA;
             line[line_idx] = '\"';
             lineA = GetMultiByteString(line);
             fprintf(stderr,"%S: warning: unrecognized line: '%s'\n", getAppName(), lineA);
             HeapFree(GetProcessHeap(), 0, lineA);
             return;
         }
 
     } else {
         char* lineA = GetMultiByteString(line);
         fprintf(stderr,"%S: warning: unrecognized line: '%s'\n", getAppName(), lineA);
         HeapFree(GetProcessHeap(), 0, lineA);
         return;
     }
     line_idx++;                   /* skip the '=' character */
 
     while ( isspaceW(line[line_idx]) ) line_idx++;
     val_data = line + line_idx;
     /* trim trailing blanks */
     line_idx = strlenW(val_data);
     while (line_idx > 0 && isspaceW(val_data[line_idx-1])) line_idx--;
     val_data[line_idx] = '\0';
 
     REGPROC_unescape_string(val_name);
     res = setValue(val_name, val_data, is_unicode);
     if ( res != ERROR_SUCCESS )
     {
         char* val_nameA = GetMultiByteString(val_name);
         char* val_dataA = GetMultiByteString(val_data);
         fprintf(stderr,"%S: ERROR Key %s not created. Value: %s, Data: %s\n",
                 getAppName(),
                 currentKeyName,
                 val_nameA,
                 val_dataA);
         HeapFree(GetProcessHeap(), 0, val_nameA);
         HeapFree(GetProcessHeap(), 0, val_dataA);
     }
 }

◆ REGPROC_export_binary()

static void REGPROC_export_binary	(	WCHAR **	line_buf,
		DWORD *	line_buf_size,
		DWORD *	line_len,
		DWORD	type,
		BYTE *	value,
		DWORD	value_size,
		BOOL	unicode
	)

static

Definition at line 992 of file regproc.c.

Referenced by export_hkey().

 {
     DWORD hex_pos, data_pos;
     const WCHAR *hex_prefix;
     const WCHAR hex[] = {'h','e','x',':',0};
     WCHAR hex_buf[17];
     const WCHAR concat[] = {'\\','\r','\n',' ',' ',0};
     DWORD concat_prefix, concat_len;
     const WCHAR newline[] = {'\r','\n',0};
     CHAR* value_multibyte = NULL;
 
     if (type == REG_BINARY) {
         hex_prefix = hex;
     } else {
         const WCHAR hex_format[] = {'h','e','x','(','%','x',')',':',0};
         hex_prefix = hex_buf;
         sprintfW(hex_buf, hex_format, type);
         if ((type == REG_SZ || type == REG_EXPAND_SZ || type == REG_MULTI_SZ) && !unicode)
         {
             value_multibyte = GetMultiByteStringN((WCHAR*)value, value_size / sizeof(WCHAR), &value_size);
             value = (BYTE*)value_multibyte;
         }
     }
 
     concat_len = lstrlenW(concat);
     concat_prefix = 2;
 
     hex_pos = *line_len;
     *line_len += lstrlenW(hex_prefix);
     data_pos = *line_len;
     *line_len += value_size * 3;
     /* - The 2 spaces that concat places at the start of the
      *   line effectively reduce the space available for data.
      * - If the value name and hex prefix are very long
      *   ( > REG_FILE_HEX_LINE_LEN) or *line_len divides
      *   without a remainder then we may overestimate
      *   the needed number of lines by one. But that's ok.
      * - The trailing '\r' takes the place of a comma so
      *   we only need to add 1 for the trailing '\n'
      */
     *line_len += *line_len / (REG_FILE_HEX_LINE_LEN - concat_prefix) * concat_len + 1;
     REGPROC_resize_char_buffer(line_buf, line_buf_size, *line_len);
     lstrcpyW(*line_buf + hex_pos, hex_prefix);
     if (value_size)
     {
         const WCHAR format[] = {'%','0','2','x',0};
         DWORD i, column;
 
         column = data_pos; /* no line wrap yet */
         i = 0;
         while (1)
         {
             sprintfW(*line_buf + data_pos, format, (unsigned int)value[i]);
             data_pos += 2;
             if (++i == value_size)
                 break;
 
             (*line_buf)[data_pos++] = ',';
             column += 3;
 
             /* wrap the line */
             if (column >= REG_FILE_HEX_LINE_LEN) {
                 lstrcpyW(*line_buf + data_pos, concat);
                 data_pos += concat_len;
                 column = concat_prefix;
             }
         }
     }
     lstrcpyW(*line_buf + data_pos, newline);
     HeapFree(GetProcessHeap(), 0, value_multibyte);
 }

◆ REGPROC_export_string()

static void REGPROC_export_string	(	WCHAR **	line_buf,
		DWORD *	line_buf_size,
		DWORD *	line_len,
		WCHAR *	str,
		DWORD	str_len
	)

static

Definition at line 957 of file regproc.c.

Referenced by export_hkey().

 {
     DWORD i, pos;
     DWORD extra = 0;
 
     REGPROC_resize_char_buffer(line_buf, line_buf_size, *line_len + str_len + 10);
 
     /* escaping characters */
     pos = *line_len;
     for (i = 0; i < str_len; i++) {
         WCHAR c = str[i];
         switch (c) {
         case '\n':
             extra++;
             REGPROC_resize_char_buffer(line_buf, line_buf_size, *line_len + str_len + extra);
             (*line_buf)[pos++] = '\\';
             (*line_buf)[pos++] = 'n';
             break;
 
         case '\\':
         case '"':
             extra++;
             REGPROC_resize_char_buffer(line_buf, line_buf_size, *line_len + str_len + extra);
             (*line_buf)[pos++] = '\\';
             /* Fall through */
 
         default:
             (*line_buf)[pos++] = c;
             break;
         }
     }
     (*line_buf)[pos] = '\0';
     *line_len = pos;
 }

◆ REGPROC_open_export_file()

static FILE* REGPROC_open_export_file	(	WCHAR *	file_name,
		BOOL	unicode
	)

static

Definition at line 1268 of file regproc.c.

Referenced by export_registry_key().

 {
     FILE *file;
     WCHAR dash = '-';
 
     if (strncmpW(file_name,&dash,1)==0) {
         file=stdout;
         _setmode(_fileno(file), _O_BINARY);
     } else
     {
         CHAR* file_nameA = GetMultiByteString(file_name);
         file = fopen(file_nameA, "wb");
         if (!file) {
             perror("");
             fprintf(stderr,"%S: Can't open file \"%s\"\n", getAppName(), file_nameA);
             HeapFree(GetProcessHeap(), 0, file_nameA);
             exit(1);
         }
         HeapFree(GetProcessHeap(), 0, file_nameA);
     }
     if(unicode)
     {
         const BYTE unicode_seq[] = {0xff,0xfe};
         const WCHAR header[] = {'W','i','n','d','o','w','s',' ','R','e','g','i','s','t','r','y',' ','E','d','i','t','o','r',' ','V','e','r','s','i','o','n',' ','5','.','0','0','\r','\n'};
         fwrite(unicode_seq, sizeof(BYTE), sizeof(unicode_seq)/sizeof(unicode_seq[0]), file);
         fwrite(header, sizeof(WCHAR), sizeof(header)/sizeof(header[0]), file);
     } else
     {
         fputs("REGEDIT4\r\n", file);
     }
 
     return file;
 }

◆ REGPROC_print_error()

static void REGPROC_print_error ( void )

static

Definition at line 892 of file regproc.c.

Referenced by export_hkey(), and export_registry_key().

 {
     LPVOID lpMsgBuf;
     DWORD error_code;
     int status;
 
     error_code = GetLastError ();
     status = FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
                             NULL, error_code, 0, (LPSTR) &lpMsgBuf, 0, NULL);
     if (!status) {
         fprintf(stderr,"%S: Cannot display message for error %lu, status %lu\n",
                 getAppName(), error_code, GetLastError());
         exit(1);
     }
     puts(lpMsgBuf);
     LocalFree(lpMsgBuf);
     exit(1);
 }

◆ REGPROC_resize_binary_buffer()

static void REGPROC_resize_binary_buffer	(	BYTE **	buffer,
		DWORD *	size,
		DWORD	required_size
	)

static

Definition at line 942 of file regproc.c.

Referenced by export_hkey().

 {
     if (required_size > *size) {
         *size = required_size;
         if (!*buffer)
             *buffer = HeapAlloc(GetProcessHeap(), 0, *size);
         else
             *buffer = HeapReAlloc(GetProcessHeap(), 0, *buffer, *size);
         CHECK_ENOUGH_MEMORY(*buffer);
     }
 }

◆ REGPROC_resize_char_buffer()

static void REGPROC_resize_char_buffer	(	WCHAR **	buffer,
		DWORD *	len,
		DWORD	required_len
	)

static

Definition at line 921 of file regproc.c.

Referenced by export_hkey(), export_registry_key(), REGPROC_export_binary(), and REGPROC_export_string().

 {
     required_len++;
     if (required_len > *len) {
         *len = required_len;
         if (!*buffer)
             *buffer = HeapAlloc(GetProcessHeap(), 0, *len * sizeof(**buffer));
         else
             *buffer = HeapReAlloc(GetProcessHeap(), 0, *buffer, *len * sizeof(**buffer));
         CHECK_ENOUGH_MEMORY(*buffer);
     }
 }

◆ REGPROC_unescape_string()

static void REGPROC_unescape_string ( WCHAR * str )

static

Definition at line 257 of file regproc.c.

Referenced by processSetValue(), and setValue().

 {
     int str_idx = 0;            /* current character under analysis */
     int val_idx = 0;            /* the last character of the unescaped string */
     int len = lstrlenW(str);
     for (str_idx = 0; str_idx < len; str_idx++, val_idx++) {
         if (str[str_idx] == '\\') {
             str_idx++;
             switch (str[str_idx]) {
             case 'n':
                 str[val_idx] = '\n';
                 break;
             case '\\':
             case '"':
                 str[val_idx] = str[str_idx];
                 break;
             default:
                 fprintf(stderr,"Warning! Unrecognized escape sequence: \\%c'\n",
                         str[str_idx]);
                 str[val_idx] = str[str_idx];
                 break;
             }
         } else {
             str[val_idx] = str[str_idx];
         }
     }
     str[val_idx] = '\0';
 }

◆ REGPROC_write_line()

static void REGPROC_write_line	(	FILE *	file,
		const WCHAR *	str,
		BOOL	unicode
	)

static

Definition at line 1067 of file regproc.c.

Referenced by export_hkey().

 {
     if(unicode)
     {
         fwrite(str, sizeof(WCHAR), lstrlenW(str), file);
     } else
     {
         char* strA = GetMultiByteString(str);
         fputs(strA, file);
         HeapFree(GetProcessHeap(), 0, strA);
     }
 }

◆ setValue()

static LONG setValue	(	WCHAR *	val_name,
		WCHAR *	val_data,
		BOOL	is_unicode
	)

static

Definition at line 344 of file regproc.c.

Referenced by processSetValue().

 {
     LONG res;
     DWORD  dwDataType, dwParseType;
     LPBYTE lpbData;
     DWORD  dwData, dwLen;
     WCHAR del[] = {'-',0};
 
     if ( (val_name == NULL) || (val_data == NULL) )
         return ERROR_INVALID_PARAMETER;
 
     if (lstrcmpW(val_data, del) == 0)
     {
         res=RegDeleteValueW(currentKeyHandle,val_name);
         return (res == ERROR_FILE_NOT_FOUND ? ERROR_SUCCESS : res);
     }
 
     /* Get the data type stored into the value field */
     dwDataType = getDataType(&val_data, &dwParseType);
 
     if (dwParseType == REG_SZ)          /* no conversion for string */
     {
         REGPROC_unescape_string(val_data);
         /* Compute dwLen after REGPROC_unescape_string because it may
          * have changed the string length and we don't want to store
          * the extra garbage in the registry.
          */
         dwLen = lstrlenW(val_data);
         if(val_data[dwLen-1] != '"')
             return ERROR_INVALID_DATA;
         if (dwLen>0 && val_data[dwLen-1]=='"')
         {
             dwLen--;
             val_data[dwLen]='\0';
         }
         lpbData = (BYTE*) val_data;
         dwLen++;  /* include terminating null */
         dwLen = dwLen * sizeof(WCHAR); /* size is in bytes */
     }
     else if (dwParseType == REG_DWORD)  /* Convert the dword types */
     {
         if (!convertHexToDWord(val_data, &dwData))
             return ERROR_INVALID_DATA;
         lpbData = (BYTE*)&dwData;
         dwLen = sizeof(dwData);
     }
     else if (dwParseType == REG_BINARY) /* Convert the binary data */
     {
         lpbData = convertHexCSVToHex(val_data, &dwLen);
         if (!lpbData)
             return ERROR_INVALID_DATA;
 
         if((dwDataType == REG_MULTI_SZ || dwDataType == REG_EXPAND_SZ) && !is_unicode)
         {
             LPBYTE tmp = lpbData;
             lpbData = (LPBYTE)GetWideStringN((char*)lpbData, dwLen, &dwLen);
             dwLen *= sizeof(WCHAR);
             HeapFree(GetProcessHeap(), 0, tmp);
         }
     }
     else                                /* unknown format */
     {
         fprintf(stderr,"%S: ERROR, unknown data format\n", getAppName());
         return ERROR_INVALID_DATA;
     }
 
     res = RegSetValueExW(
                currentKeyHandle,
                val_name,
                0,                  /* Reserved */
                dwDataType,
                lpbData,
                dwLen);
     if (dwParseType == REG_BINARY)
         HeapFree(GetProcessHeap(), 0, lpbData);
     return res;
 }

Variable Documentation

◆ currentKeyHandle

HKEY currentKeyHandle = NULL

static

Definition at line 334 of file regproc.c.

◆ currentKeyName

LPSTR currentKeyName

static

Definition at line 333 of file regproc.c.

Referenced by closeKey(), openKeyW(), and processSetValue().

◆ reg_class_keys

HKEY reg_class_keys[]

static

Initial value:

= {
            HKEY_LOCAL_MACHINE, HKEY_USERS, HKEY_CLASSES_ROOT,
            HKEY_CURRENT_CONFIG, HKEY_CURRENT_USER, HKEY_DYN_DATA
        }

Definition at line 44 of file regproc.c.

◆ reg_class_namesW

const WCHAR* reg_class_namesW[]

Initial value:

=
{
    L"HKEY_LOCAL_MACHINE", L"HKEY_USERS", L"HKEY_CLASSES_ROOT",
    L"HKEY_CURRENT_CONFIG", L"HKEY_CURRENT_USER", L"HKEY_DYN_DATA"
}

Definition at line 38 of file regproc.c.

Referenced by export_registry_key(), and parseKeyName().

Macros

Functions

Variables

Macro Definition Documentation

◆ CHECK_ENOUGH_MEMORY

◆ IO_ERROR

◆ NOT_ENOUGH_MEMORY

◆ REG_CLASS_NUMBER

◆ REG_FILE_HEX_LINE_LEN

◆ REG_VAL_BUF_SIZE

Function Documentation

◆ closeKey()

◆ convertHexCSVToHex()

◆ convertHexToDWord()

◆ delete_registry_key()

◆ export_hkey()

◆ export_registry_key()

◆ getDataType()

◆ GetMultiByteString()

◆ GetMultiByteStringN()

◆ GetWideString()

◆ GetWideStringN()

◆ import_registry_file()

◆ openKeyW()

◆ parseKeyName()

◆ processRegEntry()

◆ processRegLinesA()

◆ processRegLinesW()

◆ processSetValue()

◆ REGPROC_export_binary()

◆ REGPROC_export_string()

◆ REGPROC_open_export_file()

◆ REGPROC_print_error()

◆ REGPROC_resize_binary_buffer()

◆ REGPROC_resize_char_buffer()

◆ REGPROC_unescape_string()

◆ REGPROC_write_line()

◆ setValue()

Variable Documentation

◆ currentKeyHandle

◆ currentKeyName

◆ reg_class_keys

◆ reg_class_namesW