export.c File Reference

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <wine/heap.h>
#include "reg.h"

Include dependency graph for export.c:

Go to the source code of this file.

Macros
#define	MAX_HEX_CHARS   77

Functions
static void	write_file (HANDLE hFile, const WCHAR *str)
static WCHAR *	escape_string (WCHAR *str, size_t str_len, size_t *line_len)
static size_t	export_value_name (HANDLE hFile, WCHAR *name, size_t len)
static void	export_string_data (WCHAR **buf, WCHAR *data, size_t size)
static void	export_dword_data (WCHAR **buf, DWORD *data)
static size_t	export_hex_data_type (HANDLE hFile, DWORD type)
static void	export_hex_data (HANDLE hFile, WCHAR **buf, DWORD type, DWORD line_len, void *data, DWORD size)
static void	export_newline (HANDLE hFile)
static void	export_data (HANDLE hFile, WCHAR *value_name, DWORD value_len, DWORD type, void *data, size_t size)
static void	export_key_name (HANDLE hFile, WCHAR *name)
static int	export_registry_data (HANDLE hFile, HKEY key, WCHAR *path)
static void	export_file_header (HANDLE hFile)
static HANDLE	create_file (const WCHAR *filename, DWORD action)
static HANDLE	get_file_handle (WCHAR *filename, BOOL overwrite_file)
static BOOL	is_overwrite_switch (const WCHAR *s)
int	reg_export (int argc, WCHAR *argv[])

Macro Definition Documentation

MAX_HEX_CHARS

#define MAX_HEX_CHARS   77

Definition at line 153 of file export.c.

Function Documentation

create_file()

static HANDLE create_file ( const WCHAR *  filename, DWORD  action  )

static

Definition at line 324 of file export.c.

{
    return CreateFileW(filename, GENERIC_WRITE, 0, NULL, action, FILE_ATTRIBUTE_NORMAL, NULL);
}

Referenced by get_file_handle().

escape_string()

static WCHAR* escape_string ( WCHAR *  str, size_t  str_len, size_t *  line_len  )

static

Definition at line 34 of file export.c.

{
    size_t i, escape_count, pos;
    WCHAR *buf;

    for (i = 0, escape_count = 0; i < str_len; i++)
    {
        WCHAR c = str[i];

        if (!c) break;

        if (c == '\r' || c == '\n' || c == '\\' || c == '"')
            escape_count++;
    }

    buf = heap_xalloc((str_len + escape_count + 1) * sizeof(WCHAR));

    for (i = 0, pos = 0; i < str_len; i++, pos++)
    {
        WCHAR c = str[i];

        if (!c) break;

        switch (c)
        {
        case '\r':
            buf[pos++] = '\\';
            buf[pos] = 'r';
            break;
        case '\n':
            buf[pos++] = '\\';
            buf[pos] = 'n';
            break;
        case '\\':
            buf[pos++] = '\\';
            buf[pos] = '\\';
            break;
        case '"':
            buf[pos++] = '\\';
            buf[pos] = '"';
            break;
        default:
            buf[pos] = c;
        }
    }

    buf[pos] = 0;
    *line_len = pos;
    return buf;
}

Referenced by export_string_data(), and export_value_name().

export_data()

static void export_data ( HANDLE  hFile, WCHAR *  value_name, DWORD  value_len, DWORD  type, void *  data, size_t  size  )

static

Definition at line 194 of file export.c.

{
    WCHAR *buf = NULL;
    size_t line_len = export_value_name(hFile, value_name, value_len);

    switch (type)
    {
    case REG_SZ:
        export_string_data(&buf, data, size);
        break;
    case REG_DWORD:
        if (size)
        {
            export_dword_data(&buf, data);
            break;
        }
        /* fall through */
    case REG_NONE:
    case REG_EXPAND_SZ:
    case REG_BINARY:
    case REG_MULTI_SZ:
    default:
        export_hex_data(hFile, &buf, type, line_len, data, size);
        break;
    }

    if (size || type == REG_SZ)
    {
        write_file(hFile, buf);
        heap_free(buf);
    }

    export_newline(hFile);
}

Referenced by export_registry_data().

export_dword_data()

static void export_dword_data ( WCHAR **  buf, DWORD *  data  )

static

Definition at line 123 of file export.c.

{
    static const WCHAR fmt[] = {'d','w','o','r','d',':','%','0','8','x',0};

    *buf = heap_xalloc(15 * sizeof(WCHAR));
    swprintf(*buf, fmt, *data);
}

Referenced by export_data().

export_file_header()

static void export_file_header ( HANDLE  hFile )

static

Definition at line 315 of file export.c.

{
    static const WCHAR header[] = { 0xfeff,'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',0};

    write_file(hFile, header);
}

Referenced by reg_export().

export_hex_data()

static void export_hex_data ( HANDLE  hFile, WCHAR **  buf, DWORD  type, DWORD  line_len, void *  data, DWORD  size  )

static

Definition at line 155 of file export.c.

{
    static const WCHAR fmt[] = {'%','0','2','x',0};
    static const WCHAR hex_concat[] = {'\\','\r','\n',' ',' ',0};
    size_t num_commas, i, pos;

    line_len += export_hex_data_type(hFile, type);

    if (!size) return;

    num_commas = size - 1;
    *buf = heap_xalloc(size * 3 * sizeof(WCHAR));

    for (i = 0, pos = 0; i < size; i++)
    {
        pos += swprintf(*buf + pos, fmt, ((BYTE *)data)[i]);
        if (i == num_commas) break;
        (*buf)[pos++] = ',';
        (*buf)[pos] = 0;
        line_len += 3;

        if (line_len >= MAX_HEX_CHARS)
        {
            write_file(hFile, *buf);
            write_file(hFile, hex_concat);
            line_len = 2;
            pos = 0;
        }
    }
}

Referenced by export_data().

export_hex_data_type()

static size_t export_hex_data_type ( HANDLE  hFile, DWORD  type  )

static

Definition at line 131 of file export.c.

{
    static const WCHAR hex[] = {'h','e','x',':',0};
    static const WCHAR hexp_fmt[] = {'h','e','x','(','%','x',')',':',0};
    size_t line_len;

    if (type == REG_BINARY)
    {
        line_len = lstrlenW(hex);
        write_file(hFile, hex);
    }
    else
    {
        WCHAR *buf = heap_xalloc(15 * sizeof(WCHAR));
        line_len = swprintf(buf, hexp_fmt, type);
        write_file(hFile, buf);
        heap_free(buf);
    }

    return line_len;
}

Referenced by export_hex_data().

export_key_name()

static void export_key_name ( HANDLE  hFile, WCHAR *  name  )

static

Definition at line 230 of file export.c.

{
    static const WCHAR fmt[] = {'\r','\n','[','%','s',']','\r','\n',0};
    WCHAR *buf;

    buf = heap_xalloc((lstrlenW(name) + 7) * sizeof(WCHAR));
    swprintf(buf, fmt, name);
    write_file(hFile, buf);
    heap_free(buf);
}

Referenced by export_registry_data().

export_newline()

static void export_newline ( HANDLE  hFile )

static

Definition at line 187 of file export.c.

{
    static const WCHAR newline[] = {'\r','\n',0};

    write_file(hFile, newline);
}

Referenced by export_data(), and reg_export().

export_registry_data()

static int export_registry_data ( HANDLE  hFile, HKEY  key, WCHAR *  path  )

static

Definition at line 241 of file export.c.

{
    LONG rc;
    DWORD max_value_len = 256, value_len;
    DWORD max_data_bytes = 2048, data_size;
    DWORD subkey_len;
    DWORD i, type, path_len;
    WCHAR *value_name, *subkey_name, *subkey_path;
    BYTE *data;
    HKEY subkey;

    export_key_name(hFile, path);

    value_name = heap_xalloc(max_value_len * sizeof(WCHAR));
    data = heap_xalloc(max_data_bytes);

    i = 0;
    for (;;)
    {
        value_len = max_value_len;
        data_size = max_data_bytes;
        rc = RegEnumValueW(key, i, value_name, &value_len, NULL, &type, data, &data_size);

        if (rc == ERROR_SUCCESS)
        {
            export_data(hFile, value_name, value_len, type, data, data_size);
            i++;
        }
        else if (rc == ERROR_MORE_DATA)
        {
            if (data_size > max_data_bytes)
            {
                max_data_bytes = data_size;
                data = heap_xrealloc(data, max_data_bytes);
            }
            else
            {
                max_value_len *= 2;
                value_name = heap_xrealloc(value_name, max_value_len * sizeof(WCHAR));
            }
        }
        else break;
    }

    heap_free(data);
    heap_free(value_name);

    subkey_name = heap_xalloc(MAX_SUBKEY_LEN * sizeof(WCHAR));

    path_len = lstrlenW(path);

    i = 0;
    for (;;)
    {
        subkey_len = MAX_SUBKEY_LEN;
        rc = RegEnumKeyExW(key, i, subkey_name, &subkey_len, NULL, NULL, NULL, NULL);
        if (rc == ERROR_SUCCESS)
        {
            subkey_path = build_subkey_path(path, path_len, subkey_name, subkey_len);
            if (!RegOpenKeyExW(key, subkey_name, 0, KEY_READ, &subkey))
            {
                export_registry_data(hFile, subkey, subkey_path);
                RegCloseKey(subkey);
            }
            heap_free(subkey_path);
            i++;
        }
        else break;
    }

    heap_free(subkey_name);
    return 0;
}

Referenced by reg_export().

export_string_data()

static void export_string_data ( WCHAR **  buf, WCHAR *  data, size_t  size  )

static

Definition at line 109 of file export.c.

{
    size_t len = 0, line_len;
    WCHAR *str;
    static const WCHAR fmt[] = {'"','%','s','"',0};

    if (size)
        len = size / sizeof(WCHAR) - 1;
    str = escape_string(data, len, &line_len);
    *buf = heap_xalloc((line_len + 3) * sizeof(WCHAR));
    swprintf(*buf, fmt, str);
    heap_free(str);
}

Referenced by export_data().

export_value_name()

static size_t export_value_name ( HANDLE  hFile, WCHAR *  name, size_t  len  )

static

Definition at line 85 of file export.c.

{
    static const WCHAR quoted_fmt[] = {'"','%','s','"','=',0};
    static const WCHAR default_name[] = {'@','=',0};
    size_t line_len;

    if (name && *name)
    {
        WCHAR *str = escape_string(name, len, &line_len);
        WCHAR *buf = heap_xalloc((line_len + 4) * sizeof(WCHAR));
        line_len = swprintf(buf, quoted_fmt, str);
        write_file(hFile, buf);
        heap_free(buf);
        heap_free(str);
    }
    else
    {
        line_len = lstrlenW(default_name);
        write_file(hFile, default_name);
    }

    return line_len;
}

Referenced by export_data().

get_file_handle()

static HANDLE get_file_handle ( WCHAR *  filename, BOOL  overwrite_file  )

static

Definition at line 329 of file export.c.

{
    HANDLE hFile = create_file(filename, overwrite_file ? CREATE_ALWAYS : CREATE_NEW);

    if (hFile == INVALID_HANDLE_VALUE)
    {
        DWORD error = GetLastError();

        if (error == ERROR_FILE_EXISTS)
        {
            if (!ask_confirm(STRING_OVERWRITE_FILE, filename))
            {
                output_message(STRING_CANCELLED);
                exit(0);
            }

            hFile = create_file(filename, CREATE_ALWAYS);
        }
        else
        {
            WCHAR *str;

            FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
                           FORMAT_MESSAGE_IGNORE_INSERTS, NULL, error, 0, (WCHAR *)&str, 0, NULL);
            output_writeconsole(str, lstrlenW(str));
            LocalFree(str);
            exit(1);
        }
    }

    return hFile;
}

Referenced by reg_export().

is_overwrite_switch()

static BOOL is_overwrite_switch ( const WCHAR *  s )

static

Definition at line 362 of file export.c.

{
    if (lstrlenW(s) > 2)
        return FALSE;

    if ((s[0] == '/' || s[0] == '-') && (s[1] == 'y' || s[1] == 'Y'))
        return TRUE;

    return FALSE;
}

Referenced by reg_export().

reg_export()

int reg_export	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 373 of file export.c.

{
    HKEY root, hkey;
    WCHAR *path, *long_key;
    BOOL overwrite_file = FALSE;
    HANDLE hFile;
    int ret;

    if (argc == 3 || argc > 5)
        goto error;

    if (!parse_registry_key(argv[2], &root, &path, &long_key))
        return 1;

    if (argc == 5 && !(overwrite_file = is_overwrite_switch(argv[4])))
        goto error;

    if (RegOpenKeyExW(root, path, 0, KEY_READ, &hkey))
    {
        output_message(STRING_INVALID_KEY);
        return 1;
    }

    hFile = get_file_handle(argv[3], overwrite_file);
    export_file_header(hFile);
    ret = export_registry_data(hFile, hkey, long_key);
    export_newline(hFile);
    CloseHandle(hFile);

    RegCloseKey(hkey);

    return ret;

error:
    output_message(STRING_INVALID_SYNTAX);
    output_message(STRING_FUNC_HELP, _wcsupr(argv[1]));
    return 1;
}

Referenced by wmain().

write_file()

static void write_file ( HANDLE  hFile, const WCHAR *  str  )

static

Definition at line 27 of file export.c.

{
    DWORD written;

    WriteFile(hFile, str, lstrlenW(str) * sizeof(WCHAR), &written, NULL);
}

Referenced by do_write_job(), export_data(), export_file_header(), export_hex_data(), export_hex_data_type(), export_key_name(), export_newline(), and export_value_name().