doskey.c File Reference

#include <stdio.h>
#include <wchar.h>
#include <locale.h>
#include <windef.h>
#include <winbase.h>
#include <winuser.h>
#include <wincon.h>
#include "doskey.h"

Include dependency graph for doskey.c:

Go to the source code of this file.

Macros
#define	EXENAME_LENGTH   (255 + 1)
#define	MAX_STRING   2000

Functions
VOID WINAPI	ExpungeConsoleCommandHistoryW (LPCWSTR lpExeName)
DWORD WINAPI	GetConsoleCommandHistoryW (LPWSTR lpHistory, DWORD cbHistory, LPCWSTR lpExeName)
DWORD WINAPI	GetConsoleCommandHistoryLengthW (LPCWSTR lpExeName)
BOOL WINAPI	SetConsoleNumberOfCommandsW (DWORD dwNumCommands, LPCWSTR lpExeName)
static VOID	SetInsert (DWORD dwFlag)
static VOID	PrintHistory (VOID)
static INT	SetMacro (LPWSTR definition)
static VOID	PrintMacros (LPWSTR pszExeName, LPWSTR Indent)
static VOID	PrintAllMacros (VOID)
static LPWSTR	RemoveQuotes (LPWSTR str)
static VOID	ReadFromFile (LPWSTR FileName)
static BOOL	GetArg (WCHAR **pStart, WCHAR **pEnd)
int	wmain (VOID)

Variables
WCHAR	szStringBuf [MAX_STRING]
LPWSTR	pszExeName = L"cmd.exe"

Macro Definition Documentation

EXENAME_LENGTH

#define EXENAME_LENGTH   (255 + 1)

Definition at line 20 of file doskey.c.

MAX_STRING

#define MAX_STRING   2000

Definition at line 43 of file doskey.c.

Function Documentation

ExpungeConsoleCommandHistoryW()

VOID WINAPI ExpungeConsoleCommandHistoryW

(

LPCWSTR

lpExeName

)

Referenced by wmain().

GetArg()

static BOOL GetArg ( WCHAR **  pStart, WCHAR **  pEnd  )

static

Definition at line 257 of file doskey.c.

{
    BOOL bInQuotes = FALSE;
    WCHAR *p = *pEnd;
    p += wcsspn(p, L" \t");
    if (!*p)
        return FALSE;
    *pStart = p;
    do
    {
        if (!bInQuotes && (*p == L' ' || *p == L'\t'))
            break;
        bInQuotes ^= (*p++ == L'"');
    } while (*p);
    *pEnd = p;
    return TRUE;
}

Referenced by wmain().

GetConsoleCommandHistoryLengthW()

DWORD WINAPI GetConsoleCommandHistoryLengthW

(

LPCWSTR

lpExeName

)

Referenced by PrintHistory().

GetConsoleCommandHistoryW()

DWORD WINAPI GetConsoleCommandHistoryW	(	LPWSTR	lpHistory,
		DWORD	cbHistory,
		LPCWSTR	lpExeName
	)

Referenced by PrintHistory().

PrintAllMacros()

static VOID PrintAllMacros ( VOID  )

static

Definition at line 145 of file doskey.c.

{
    DWORD Length = GetConsoleAliasExesLength();
    PBYTE ExeNameBuf;
    WCHAR *ExeName;
    WCHAR *ExeNameEnd;
 
    ExeNameBuf = HeapAlloc(GetProcessHeap(),
                           HEAP_ZERO_MEMORY,
                           Length * sizeof(BYTE));
    if (!ExeNameBuf) return;
    ExeName = (WCHAR *)ExeNameBuf;
    ExeNameEnd = (WCHAR *)&ExeNameBuf[Length];
 
    if (GetConsoleAliasExesW(ExeName, Length * sizeof(BYTE)))
    {
        for (; ExeName < ExeNameEnd; ExeName += wcslen(ExeName) + 1)
        {
            wprintf(L"[%s]\n", ExeName);
            PrintMacros(ExeName, L"    ");
            wprintf(L"\n");
        }
    }
 
    HeapFree(GetProcessHeap(), 0, ExeNameBuf);
}

Referenced by wmain().

PrintHistory()

static VOID PrintHistory ( VOID  )

static

Definition at line 60 of file doskey.c.

{
    DWORD Length = GetConsoleCommandHistoryLengthW(pszExeName);
    PBYTE HistBuf;
    WCHAR *Hist;
    WCHAR *HistEnd;
 
    HistBuf = HeapAlloc(GetProcessHeap(),
                        HEAP_ZERO_MEMORY,
                        Length);
    if (!HistBuf) return;
    Hist = (WCHAR *)HistBuf;
    HistEnd = (WCHAR *)&HistBuf[Length];
 
    if (GetConsoleCommandHistoryW(Hist, Length, pszExeName))
    {
        for (; Hist < HistEnd; Hist += wcslen(Hist) + 1)
        {
            wprintf(L"%s\n", Hist);
        }
    }
 
    HeapFree(GetProcessHeap(), 0, HistBuf);
}

Referenced by wmain().

PrintMacros()

static VOID PrintMacros ( LPWSTR  pszExeName, LPWSTR  Indent  )

static

Definition at line 120 of file doskey.c.

{
    DWORD Length = GetConsoleAliasesLengthW(pszExeName);
    PBYTE AliasBuf;
    WCHAR *Alias;
    WCHAR *AliasEnd;
 
    AliasBuf = HeapAlloc(GetProcessHeap(),
                         HEAP_ZERO_MEMORY,
                         Length * sizeof(BYTE));
    if (!AliasBuf) return;
    Alias = (WCHAR *)AliasBuf;
    AliasEnd = (WCHAR *)&AliasBuf[Length];
 
    if (GetConsoleAliasesW(Alias, Length * sizeof(BYTE), pszExeName))
    {
        for (; Alias < AliasEnd; Alias += wcslen(Alias) + 1)
        {
            wprintf(L"%s%s\n", Indent, Alias);
        }
    }
 
    HeapFree(GetProcessHeap(), 0, AliasBuf);
}

Referenced by PrintAllMacros(), and wmain().

ReadFromFile()

static VOID ReadFromFile ( LPWSTR  FileName )

static

Definition at line 184 of file doskey.c.

{
    FILE* fp;
    WCHAR line[MAX_PATH];
    WCHAR ExeNameBuffer[EXENAME_LENGTH];
    LPWSTR pszOrgExeName = pszExeName;
 
    /* Open the file */
    fp = _wfopen(FileName, L"rt");
    if (!fp)
    {
        _wperror(FileName);
        return;
    }
 
    while (fgetws(line, ARRAYSIZE(line), fp) != NULL)
    {
        PWCHAR end;
 
        if (!*line)
            continue;
 
        /* Remove trailing newline character */
        end = &line[wcslen(line) - 1];
        if (*end == L'\n')
            *end = L'\0';
 
        if (!*line)
            continue;
 
        /* Check for any section redefining the current executable name */
        end = NULL;
        if (*line == L'[')
            end = wcschr(line, L']');
 
        if (end != NULL)
        {
            /* New section: change the current executable name */
 
            *end = L'\0'; // NULL-terminate it
            pszExeName = RemoveQuotes(line + 1);
            if (*pszExeName)
            {
                /* Capture the new executable name and truncate it if needed */
                end = &pszExeName[wcslen(pszExeName)];
                if (end - pszExeName >= EXENAME_LENGTH)
                    end = &pszExeName[EXENAME_LENGTH - 1];
                *end = L'\0'; // Truncate it
                wcscpy(ExeNameBuffer, pszExeName);
                pszExeName = ExeNameBuffer;
            }
            else
            {
                /* Restore the original current executable name */
                pszExeName = pszOrgExeName;
            }
        }
        else
        {
            /* Set the new macro for the current executable */
            SetMacro(line);
        }
    }
 
    /* Restore the original current executable name if it has changed */
    pszExeName = pszOrgExeName;
 
    /* Close the file and return */
    fclose(fp);
    return;
}

Referenced by wmain().

RemoveQuotes()

static LPWSTR RemoveQuotes ( LPWSTR  str )

static

Definition at line 173 of file doskey.c.

{
    WCHAR *end;
    if (*str == L'"' && *(end = str + wcslen(str) - 1) == L'"')
    {
        str++;
        *end = L'\0';
    }
    return str;
}

Referenced by ReadFromFile(), and wmain().

SetConsoleNumberOfCommandsW()

BOOL WINAPI SetConsoleNumberOfCommandsW	(	DWORD	dwNumCommands,
		LPCWSTR	lpExeName
	)

Referenced by wmain().

SetInsert()

static VOID SetInsert ( DWORD  dwFlag )

static

Definition at line 47 of file doskey.c.

{
    /*
     * NOTE: Enabling the ENABLE_INSERT_MODE mode can also be done by calling
     * kernel32:SetConsoleCommandHistoryMode(CONSOLE_OVERSTRIKE) (deprecated).
     */
    DWORD dwMode;
    HANDLE hConsole = GetStdHandle(STD_INPUT_HANDLE);
    GetConsoleMode(hConsole, &dwMode);
    dwMode |= ENABLE_EXTENDED_FLAGS;
    SetConsoleMode(hConsole, (dwMode & ~ENABLE_INSERT_MODE) | dwFlag);
}

Referenced by wmain().

SetMacro()

static INT SetMacro ( LPWSTR  definition )

static

Definition at line 85 of file doskey.c.

{
    WCHAR *name, *nameend, *text, temp;
 
    name = definition;
    while (*name == L' ')
        name++;
 
    /* error if no '=' found */
    if ((nameend = wcschr(name, L'=')) != NULL)
    {
        text = nameend + 1;
        while (*text == L' ')
            text++;
 
        while (nameend > name && nameend[-1] == L' ')
            nameend--;
 
        /* Split rest into name and substitute */
        temp = *nameend;
        *nameend = L'\0';
        /* Don't allow spaces in the name, since such a macro would be unusable */
        if (!wcschr(name, L' ') && AddConsoleAliasW(name, text, pszExeName))
            return 0;
        *nameend = temp;
    }
 
    LoadStringW(GetModuleHandle(NULL),
                IDS_INVALID_MACRO_DEF,
                szStringBuf,
                ARRAYSIZE(szStringBuf));
    wprintf(szStringBuf, definition);
    return 1;
}

Referenced by ReadFromFile(), and wmain().

wmain()

int wmain

(

VOID

)

Definition at line 276 of file doskey.c.

{
    LPWSTR pArgStart, pArgEnd;
 
    setlocale(LC_ALL, "");
 
    /* Get the full command line using GetCommandLine(). We can't just use argv,
     * because then a parameter like "gotoroot=cd \" wouldn't be passed completely. */
    pArgEnd = GetCommandLineW();
 
    /* Skip the application name */
    GetArg(&pArgStart, &pArgEnd);
 
    while (GetArg(&pArgStart, &pArgEnd))
    {
        /* NULL-terminate this argument to make processing easier */
        WCHAR tmp = *pArgEnd;
        *pArgEnd = L'\0';
 
        if (!wcscmp(pArgStart, L"/?"))
        {
            LoadStringW(GetModuleHandle(NULL),
                        IDS_HELP,
                        szStringBuf,
                        ARRAYSIZE(szStringBuf));
            wprintf(szStringBuf);
            break;
        }
        else if (!_wcsnicmp(pArgStart, L"/EXENAME=", 9))
        {
            pszExeName = RemoveQuotes(pArgStart + 9);
        }
        else if (!wcsicmp(pArgStart, L"/H") ||
                 !wcsicmp(pArgStart, L"/HISTORY"))
        {
            PrintHistory();
        }
        else if (!_wcsnicmp(pArgStart, L"/LISTSIZE=", 10))
        {
            SetConsoleNumberOfCommandsW(_wtoi(pArgStart + 10), pszExeName);
        }
        else if (!wcsicmp(pArgStart, L"/REINSTALL"))
        {
            ExpungeConsoleCommandHistoryW(pszExeName);
        }
        else if (!wcsicmp(pArgStart, L"/INSERT"))
        {
            SetInsert(ENABLE_INSERT_MODE);
        }
        else if (!wcsicmp(pArgStart, L"/OVERSTRIKE"))
        {
            SetInsert(0);
        }
        else if (!wcsicmp(pArgStart, L"/M") ||
                 !wcsicmp(pArgStart, L"/MACROS"))
        {
            PrintMacros(pszExeName, L"");
        }
        else if (!_wcsnicmp(pArgStart, L"/M:",      3) ||
                 !_wcsnicmp(pArgStart, L"/MACROS:", 8))
        {
            LPWSTR exe = RemoveQuotes(wcschr(pArgStart, L':') + 1);
            if (!wcsicmp(exe, L"ALL"))
                PrintAllMacros();
            else
                PrintMacros(exe, L"");
        }
        else if (!_wcsnicmp(pArgStart, L"/MACROFILE=", 11))
        {
            ReadFromFile(RemoveQuotes(pArgStart + 11));
        }
        else
        {
            /* This is the beginning of a macro definition. It includes
             * the entire remainder of the line, so first put back the
             * character that we replaced with NULL. */
            *pArgEnd = tmp;
            return SetMacro(pArgStart);
        }
 
        if (!tmp) break;
        pArgEnd++;
    }
 
    return 0;
}

Variable Documentation

pszExeName

LPWSTR pszExeName = L"cmd.exe"

Definition at line 45 of file doskey.c.

Referenced by PrintHistory(), PrintMacros(), ReadFromFile(), SetMacro(), and wmain().

szStringBuf

WCHAR szStringBuf[MAX_STRING]

Definition at line 44 of file doskey.c.

Referenced by SetMacro(), and wmain().