cmd/tests: Add tests for console InsertMode

[Hugh McMaster]

---
 programs/cmd/tests/Makefile.in  |   4 +-
 programs/cmd/tests/insertmode.c | 136 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 139 insertions(+), 1 deletion(-)
 create mode 100644 programs/cmd/tests/insertmode.c

diff --git a/programs/cmd/tests/Makefile.in b/programs/cmd/tests/Makefile.in
index 87942fe..4777ac0 100644
--- a/programs/cmd/tests/Makefile.in
+++ b/programs/cmd/tests/Makefile.in
@@ -1,6 +1,8 @@
 TESTDLL   = cmd.exe
+IMPORTS   = advapi32
 
 C_SRCS = \
-	batch.c
+	batch.c \
+	insertmode.c
 
 RC_SRCS   = rsrc.rc
diff --git a/programs/cmd/tests/insertmode.c b/programs/cmd/tests/insertmode.c
new file mode 100644
index 0000000..89fb6ed
--- /dev/null
+++ b/programs/cmd/tests/insertmode.c
@@ -0,0 +1,136 @@
+/*
+ * Copyright 2015 Hugh McMaster
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <windows.h>
+#include <string.h>
+
+#include "wine/test.h"
+
+/* AttachConsole is not present on Windows 2000, although the PSDK says it should be */
+static BOOL (__cdecl *p_attachconsole)(DWORD process_id);
+
+static BOOL have_attachconsole(void)
+{
+    HMODULE hmod = GetModuleHandleA("kernel32.dll");
+
+    p_attachconsole = (void *)GetProcAddress(hmod, "AttachConsole");
+    if (!p_attachconsole)
+        return FALSE;
+    return TRUE;
+}
+
+static LONG reg_get_insertmode(DWORD *value)
+{
+    HKEY hkey;
+    DWORD size = sizeof(DWORD);
+    LONG ret;
+
+    if (!(ret = RegOpenKeyExA(HKEY_CURRENT_USER, "Console", 0, KEY_READ, &hkey)))
+    {
+        ret = RegQueryValueExA(hkey, "InsertMode", NULL, NULL, (BYTE *)value, &size);
+        RegCloseKey(hkey);
+    }
+    return ret;
+}
+
+static LONG reg_set_insertmode(DWORD value)
+{
+    HKEY hkey;
+    LONG ret;
+
+    if (!(ret = RegOpenKeyExA(HKEY_CURRENT_USER, "Console", 0, KEY_SET_VALUE, &hkey)))
+    {
+        ret = RegSetValueExA(hkey, "InsertMode", 0, REG_DWORD, (const BYTE *)&value, sizeof(value));
+        RegCloseKey(hkey);
+    }
+    return ret;
+}
+
+#define INSERT_MODE_FLAGS (ENABLE_INSERT_MODE | ENABLE_EXTENDED_FLAGS)
+static DWORD is_insertmode(DWORD mode)
+{
+    if ((mode & INSERT_MODE_FLAGS) == INSERT_MODE_FLAGS)
+        return 1;
+    else
+        return 0;
+}
+
+static BOOL get_child_console_mode(DWORD *mode)
+{
+    STARTUPINFOA si;
+    PROCESS_INFORMATION pi;
+    char cmd[32];
+    DWORD res;
+
+    ZeroMemory(&si, sizeof(si));
+    si.cb = sizeof(si);
+    ZeroMemory(&pi, sizeof(pi));
+
+    strcpy(cmd, "cmd.exe /c ping 127.0.0.1 -n 4");
+    if (!CreateProcessA(NULL, cmd, NULL, NULL, FALSE, CREATE_NO_WINDOW, NULL, NULL, &si, &pi))
+        return FALSE;
+
+    Sleep(1000); /* Wait for console to fully initialise */
+
+    FreeConsole();
+    p_attachconsole(pi.dwProcessId);
+
+    GetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), mode);
+
+    FreeConsole();
+    p_attachconsole(ATTACH_PARENT_PROCESS);
+
+    res = WaitForSingleObject(pi.hProcess, 5000);
+    if (res == WAIT_TIMEOUT)
+        TerminateProcess(pi.hProcess, 1);
+
+    CloseHandle(pi.hThread);
+    CloseHandle(pi.hProcess);
+    return TRUE;
+}
+
+START_TEST(insertmode)
+{
+    DWORD orig_mode, orig_reg_value, status, child_mode;
+
+    GetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), &orig_mode);
+    status = is_insertmode(orig_mode);
+    reg_get_insertmode(&orig_reg_value);
+    ok(status == orig_reg_value, "Console InsertMode (%x) does not match registry value (%x)\n",
+       status, orig_reg_value);
+
+    if (!have_attachconsole()) {
+        win_skip("AttachConsole not present in kernel32.dll, skipping InsertMode tests\n");
+        return;
+    }
+
+    reg_set_insertmode(0);
+    if (!get_child_console_mode(&child_mode)) {
+        win_skip("Error creating cmd process (%d), skipping InsertMode tests\n", GetLastError());
+        return;
+    }
+    status = is_insertmode(child_mode);
+    ok(status == 0, "Incorrect console status, got %x, expected 0\n", status);
+
+    reg_set_insertmode(1);
+    get_child_console_mode(&child_mode);
+    status = is_insertmode(child_mode);
+    ok(status == 1, "Incorrect console status, got %x, expected 1\n", status);
+
+    reg_set_insertmode(orig_reg_value); /* Restore original InsertMode value */
+}
-- 
1.9.1