+++ /dev/null
-// Copyright (c) 2015 Ryan Prichard
-//
-// Permission is hereby granted, free of charge, to any person obtaining a copy
-// of this software and associated documentation files (the "Software"), to
-// deal in the Software without restriction, including without limitation the
-// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
-// sell copies of the Software, and to permit persons to whom the Software is
-// furnished to do so, subject to the following conditions:
-//
-// The above copyright notice and this permission notice shall be included in
-// all copies or substantial portions of the Software.
-//
-// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
-// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-// IN THE SOFTWARE.
-
-#include <windows.h>
-
-#include <cassert>
-#include <cctype>
-#include <cstdio>
-#include <cstdlib>
-#include <cwchar>
-#include <vector>
-
-#include "../include/winpty.h"
-#include "../shared/DebugClient.h"
-
-static std::vector<unsigned char> filterContent(
- const std::vector<unsigned char> &content) {
- std::vector<unsigned char> result;
- auto it = content.begin();
- const auto itEnd = content.end();
- while (it < itEnd) {
- if (*it == '\r') {
- // Filter out carriage returns. Sometimes the output starts with
- // a single CR; other times, it has multiple CRs.
- it++;
- } else if (*it == '\x1b' && (it + 1) < itEnd && *(it + 1) == '[') {
- // Filter out escape sequences. They have no interior letters and
- // end with a single letter.
- it += 2;
- while (it < itEnd && !isalpha(*it)) {
- it++;
- }
- it++;
- } else {
- // Let everything else through.
- result.push_back(*it);
- it++;
- }
- }
- return result;
-}
-
-// Read bytes from the non-overlapped file handle until the file is closed or
-// until an I/O error occurs.
-static std::vector<unsigned char> readAll(HANDLE handle) {
- unsigned char buf[1024];
- std::vector<unsigned char> result;
- while (true) {
- DWORD amount = 0;
- BOOL ret = ReadFile(handle, buf, sizeof(buf), &amount, nullptr);
- if (!ret || amount == 0) {
- break;
- }
- result.insert(result.end(), buf, buf + amount);
- }
- return result;
-}
-
-static void parentTest() {
- wchar_t program[1024];
- wchar_t cmdline[1024];
- GetModuleFileNameW(nullptr, program, 1024);
-
- {
- // XXX: We'd like to use swprintf, which is part of C99 and takes a
- // size_t maxlen argument. MinGW-w64 has this function, as does MSVC.
- // The old MinGW doesn't, though -- instead, it apparently provides an
- // swprintf taking no maxlen argument. This *might* be a regression?
- // (There is also no swnprintf, but that function is obsolescent with a
- // correct swprintf, and it isn't in POSIX or ISO C.)
- //
- // Visual C++ 6 also provided this non-conformant swprintf, and I'm
- // guessing MSVCRT.DLL does too. (My impression is that the old MinGW
- // prefers to rely on MSVCRT.DLL for convenience?)
- //
- // I could compile differently for old MinGW, but what if it fixes its
- // function later? Instead, use a workaround. It's starting to make
- // sense to drop MinGW support in favor of MinGW-w64. This is too
- // annoying.
- //
- // grepbait: OLD-MINGW / WINPTY_TARGET_MSYS1
- cmdline[0] = L'\0';
- wcscat(cmdline, L"\"");
- wcscat(cmdline, program);
- wcscat(cmdline, L"\" CHILD");
- }
- // swnprintf(cmdline, sizeof(cmdline) / sizeof(cmdline[0]),
- // L"\"%ls\" CHILD", program);
-
- auto agentCfg = winpty_config_new(0, nullptr);
- assert(agentCfg != nullptr);
- auto pty = winpty_open(agentCfg, nullptr);
- assert(pty != nullptr);
- winpty_config_free(agentCfg);
-
- HANDLE conin = CreateFileW(
- winpty_conin_name(pty),
- GENERIC_WRITE, 0, nullptr, OPEN_EXISTING, 0, nullptr);
- HANDLE conout = CreateFileW(
- winpty_conout_name(pty),
- GENERIC_READ, 0, nullptr, OPEN_EXISTING, 0, nullptr);
- assert(conin != INVALID_HANDLE_VALUE);
- assert(conout != INVALID_HANDLE_VALUE);
-
- auto spawnCfg = winpty_spawn_config_new(
- WINPTY_SPAWN_FLAG_AUTO_SHUTDOWN, program, cmdline,
- nullptr, nullptr, nullptr);
- assert(spawnCfg != nullptr);
- HANDLE process = nullptr;
- BOOL spawnSuccess = winpty_spawn(
- pty, spawnCfg, &process, nullptr, nullptr, nullptr);
- assert(spawnSuccess && process != nullptr);
-
- auto content = readAll(conout);
- content = filterContent(content);
-
- std::vector<unsigned char> expectedContent = {
- 'H', 'I', '\n', 'X', 'Y', '\n'
- };
- DWORD exitCode = 0;
- assert(GetExitCodeProcess(process, &exitCode) && exitCode == 42);
- CloseHandle(process);
- CloseHandle(conin);
- CloseHandle(conout);
- assert(content == expectedContent);
- winpty_free(pty);
-}
-
-static void childTest() {
- printf("HI\nXY\n");
- exit(42);
-}
-
-int main(int argc, char *argv[]) {
- if (argc == 1) {
- parentTest();
- } else {
- childTest();
- }
- return 0;
-}
projects / VSoRC / .git / blobdiff