* Create a branch for USB experiments.

svn path=/branches/usb-experiments/; revision=72629

base/applications/cmdutils/doskey/doskey.c

@@ -0,0 +1,300 @@
+#include <stdio.h>
+#include <wchar.h>
+#include <locale.h>
+
+#include <windef.h>
+#include <winbase.h>
+#include <winuser.h>
+#include <wincon.h>
+
+/* Console API functions which are absent from wincon.h */
+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);
+
+#include "doskey.h"
+
+#define MAX_STRING 2000
+WCHAR szStringBuf[MAX_STRING];
+LPWSTR pszExeName = L"cmd.exe";
+
+static VOID SetInsert(DWORD dwFlag)
+{
+    /*
+     * NOTE: Enabling the ENABLE_INSERT_MODE mode can also be done by calling
+     * kernel32:SetConsoleCommandHistoryMode(CONSOLE_OVERSTRIKE) .
+     */
+    DWORD dwMode;
+    HANDLE hConsole = GetStdHandle(STD_INPUT_HANDLE);
+    GetConsoleMode(hConsole, &dwMode);
+    dwMode |= ENABLE_EXTENDED_FLAGS;
+    SetConsoleMode(hConsole, (dwMode & ~ENABLE_INSERT_MODE) | dwFlag);
+}
+
+static VOID PrintHistory(VOID)
+{
+    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);
+}
+
+static INT SetMacro(LPWSTR definition)
+{
+    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, MAX_STRING);
+    wprintf(szStringBuf, definition);
+    return 1;
+}
+
+static VOID PrintMacros(LPWSTR pszExeName, LPWSTR Indent)
+{
+    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);
+}
+
+static VOID PrintAllMacros(VOID)
+{
+    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);
+}
+
+static VOID ReadFromFile(LPWSTR param)
+{
+    FILE* fp;
+    WCHAR line[MAX_PATH];
+
+    fp = _wfopen(param, L"r");
+    if (!fp)
+    {
+        _wperror(param);
+        return;
+    }
+
+    while ( fgetws(line, MAX_PATH, fp) != NULL)
+    {
+        /* Remove newline character */
+        WCHAR *end = &line[wcslen(line) - 1];
+        if (*end == L'\n')
+            *end = L'\0';
+
+        if (*line)
+            SetMacro(line);
+    }
+
+    fclose(fp);
+    return;
+}
+
+/* Get the start and end of the next command-line argument. */
+static BOOL GetArg(WCHAR **pStart, WCHAR **pEnd)
+{
+    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;
+}
+
+/* Remove starting and ending quotes from a string, if present */
+static LPWSTR RemoveQuotes(LPWSTR str)
+{
+    WCHAR *end;
+    if (*str == L'"' && *(end = str + wcslen(str) - 1) == L'"')
+    {
+        str++;
+        *end = L'\0';
+    }
+    return str;
+}
+
+int
+wmain(VOID)
+{
+    WCHAR *pArgStart;
+    WCHAR *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))
+    {
+        /* NUL-terminate this argument to make processing easier */
+        WCHAR tmp = *pArgEnd;
+        *pArgEnd = L'\0';
+
+        if (!wcscmp(pArgStart, L"/?"))
+        {
+            LoadStringW(GetModuleHandle(NULL), IDS_HELP, szStringBuf, MAX_STRING);
+            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 NUL. */
+            *pArgEnd = tmp;
+            return SetMacro(pArgStart);
+        }
+
+        if (!tmp) break;
+        pArgEnd++;
+    }
+
+    return 0;
+}