regedit.c File Reference

#include "regedit.h"

Include dependency graph for regedit.c:

Go to the source code of this file.

Enumerations
enum	REGEDIT_ACTION {   ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE,   ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE,   ACTION_VIEW }

Functions
LPCWSTR	getAppName (void)
void	get_file_name (LPWSTR *command_line, LPWSTR file_name)
BOOL	PerformRegAction (REGEDIT_ACTION action, LPWSTR s, BOOL silent)
static void	error_unknown_switch (WCHAR chu, LPWSTR s)
BOOL	ProcessCmdLine (LPWSTR lpCmdLine)

Variables
static const LPCWSTR	usage

Enumeration Type Documentation

REGEDIT_ACTION

enum REGEDIT_ACTION

Enumerator
ACTION_UNDEF
ACTION_ADD
ACTION_EXPORT
ACTION_DELETE
ACTION_UNDEF
ACTION_ADD
ACTION_EXPORT
ACTION_DELETE
ACTION_VIEW

Definition at line 55 of file regedit.c.

{
    ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE
} REGEDIT_ACTION;

Function Documentation

error_unknown_switch()

static void error_unknown_switch ( WCHAR  chu, LPWSTR  s  )

static

Process unknown switch.

Params: chu - the switch character in upper-case. s - the command line string where s points to the switch character.

Definition at line 282 of file regedit.c.

{
    if (iswalpha(chu))
    {
        fwprintf(stderr, L"%s: Undefined switch /%c!\n", getAppName(), chu);
    }
    else
    {
        fwprintf(stderr, L"%s: Alphabetic character is expected after '%c' "
                          L"in switch specification\n", getAppName(), *(s - 1));
    }
    exit(1);
}

Referenced by ProcessCmdLine().

get_file_name()

void get_file_name	(	LPWSTR *	command_line,
		LPWSTR	file_name
	)

Definition at line 77 of file regedit.c.

{
    WCHAR *s = *command_line;
    size_t pos = 0; /* position of pointer "s" in *command_line */
    file_name[0] = 0;

    if (!s[0])
    {
        return;
    }

    if (s[0] == L'"')
    {
        s++;
        (*command_line)++;
        while(s[0] != L'"')
        {
            if (!s[0])
            {
                fwprintf(stderr, L"%s: Unexpected end of file name!\n", getAppName());
                exit(1);
            }
            s++;
            pos++;
        }
    }
    else
    {
        while(s[0] && !iswspace(s[0]))
        {
            s++;
            pos++;
        }
    }
    memcpy(file_name, *command_line, pos * sizeof(WCHAR));
    /* remove the last backslash */
    if (file_name[pos - 1] == L'\\')
    {
        file_name[pos - 1] = L'\0';
    }
    else
    {
        file_name[pos] = L'\0';
    }

    if (s[0])
    {
        s++;
        pos++;
    }
    while(s[0] && iswspace(s[0]))
    {
        s++;
        pos++;
    }
    (*command_line) += pos;
}

Referenced by PerformRegAction().

getAppName()

LPCWSTR getAppName

(

void

)

Definition at line 61 of file regedit.c.

{
    return L"regedit";
}

Referenced by _CmdWndProc(), convertHexCSVToHex(), convertHexToDWord(), delete_registry_key(), error_unknown_switch(), export_hkey(), export_registry_key(), FrameWndProc(), get_file_name(), PerformRegAction(), processRegEntry(), processRegLinesA(), processSetValue(), REGPROC_open_export_file(), REGPROC_print_error(), and setValue().

PerformRegAction()

BOOL PerformRegAction	(	REGEDIT_ACTION	action,
		LPWSTR	s,
		BOOL	silent
	)

Definition at line 135 of file regedit.c.

{
    switch (action)
    {
        case ACTION_ADD:
        {
            WCHAR szText[512];
            WCHAR filename[MAX_PATH];
            LPWSTR command_line = s;
            UINT count = 0, mbType = MB_YESNO;
            FILE *fp;

            get_file_name(&command_line, filename);
            while (filename[0])
            {
                count++;
                get_file_name(&command_line, filename);
            }

            if (count == 0)
            {
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No file name is specified.");
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
                exit(4);
            }

            if (count > 1)
            {
                /* Enable three buttons if we have more than one file */
                mbType = MB_YESNOCANCEL;
            }

            get_file_name(&s, filename);
            while (filename[0])
            {
                /* Request import confirmation */
                if (!silent)
                {
                    int choice;

                    LoadStringW(hInst, IDS_IMPORT_PROMPT, szText, COUNT_OF(szText));

                    choice = InfoMessageBox(NULL, mbType | MB_ICONQUESTION, szTitle, szText, filename);

                    switch (choice)
                    {
                        case IDNO:
                            goto cont;
                        case IDCANCEL:
                            /* The cancel case is useful if the user is importing more than one registry file
                            at a time, and wants to back out anytime during the import process. This way, the
                            user doesn't have to resort to ending the regedit process abruptly just to cancel
                            the operation. */
                            return TRUE;
                        default:
                            break;
                    }
                }

                /* Open the file */
                fp = _wfopen(filename, L"r");

                /* Import it */
                if (fp == NULL || !import_registry_file(fp))
                {
                    /* Error opening the file */
                    if (!silent)
                    {
                        LoadStringW(hInst, IDS_IMPORT_ERROR, szText, COUNT_OF(szText));
                        InfoMessageBox(NULL, MB_OK | MB_ICONERROR, szTitle, szText, filename);
                    }
                }
                else
                {
                    /* Show successful import */
                    if (!silent)
                    {
                        LoadStringW(hInst, IDS_IMPORT_OK, szText, COUNT_OF(szText));
                        InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, szText, filename);
                    }
                }

                /* Close the file */
                if (fp) fclose(fp);

cont:
                get_file_name(&s, filename);
            }
            break;
        }

        case ACTION_DELETE:
        {
            WCHAR reg_key_name[KEY_MAX_LEN];
            get_file_name(&s, reg_key_name);
            if (!reg_key_name[0])
            {
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No registry key is specified for removal.");
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
                exit(6);
            }
            delete_registry_key(reg_key_name);
            break;
        }

        case ACTION_EXPORT:
        {
            WCHAR filename[MAX_PATH];

            filename[0] = L'\0';
            get_file_name(&s, filename);
            if (!filename[0])
            {
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No file name is specified.");
                InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
                exit(7);
            }

            if (s[0])
            {
                WCHAR reg_key_name[KEY_MAX_LEN];
                get_file_name(&s, reg_key_name);
                export_registry_key(filename, reg_key_name, REG_FORMAT_4);
            }
            else
            {
                export_registry_key(filename, NULL, REG_FORMAT_4);
            }
            break;
        }

        default:
            fwprintf(stderr, L"%s: Unhandled action!\n", getAppName());
            exit(8);
            break;
    }

    return TRUE;
}

Referenced by ProcessCmdLine().

ProcessCmdLine()

BOOL ProcessCmdLine

(

LPWSTR

lpCmdLine

)

Definition at line 296 of file regedit.c.

{
    BOOL silent = FALSE;
    REGEDIT_ACTION action = ACTION_UNDEF;
    LPWSTR s = lpCmdLine;       /* command line pointer */
    WCHAR ch = *s;              /* current character */

    while (ch && ((ch == L'-') || (ch == L'/')))
    {
        WCHAR chu;
        WCHAR ch2;

        s++;
        ch = *s;
        ch2 = *(s + 1);
        chu = towupper(ch);
        if (!ch2 || iswspace(ch2))
        {
            if (chu == L'S')
            {
                /* Silence dialogs */
                silent = TRUE;
            }
            else if (chu == L'V')
            {
                /* Ignore this switch */
            }
            else
            {
                switch (chu)
                {
                    case L'D':
                        action = ACTION_DELETE;
                        break;
                    case L'E':
                        action = ACTION_EXPORT;
                        break;
                    case L'?':
                        InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
                        exit(3);
                        break;
                    default:
                        error_unknown_switch(chu, s);
                        break;
                }
            }
            s++;
        }
        else
        {
            if (ch2 == L':')
            {
                switch (chu)
                {
                    case L'L':
                        /* fall through */
                    case L'R':
                        s += 2;
                        while (*s && !iswspace(*s))
                        {
                            s++;
                        }
                        break;
                    default:
                        error_unknown_switch(chu, s);
                        break;
                }
            }
            else
            {
                /* this is a file name, starting from '/' */
                s--;
                break;
            }
        }
        /* skip spaces to the next parameter */
        ch = *s;
        while (ch && iswspace(ch))
        {
            s++;
            ch = *s;
        }
    }

    if (*s && action == ACTION_UNDEF)
        action = ACTION_ADD;

    if (action != ACTION_UNDEF)
        return PerformRegAction(action, s, silent);
    else
        return FALSE;
}

Referenced by wWinMain().

Variable Documentation

usage

const LPCWSTR usage

static

Definition at line 23 of file regedit.c.