root/lang/objective-cplusplus/i3/trunk/src/os-windows/CygwinPty.cpp @ 34664

#include <PrecompiledHeaders.h>
#include "CygwinPty.h"

namespace i3 {
template <size_t Size>
bool query_registory(HKEY root, const TCHAR* SubKey, const TCHAR* value, TCHAR (&retval)[Size]) {
    HKEY hkResult = NULL;
    LONG result = RegOpenKeyEx(
        root,
        SubKey,
        0,
        KEY_EXECUTE,
        &hkResult);
    
    if (result != ERROR_SUCCESS) {
        return false;
    }

    DWORD type = 0;
    DWORD cbCd = sizeof(retval);
    result = RegQueryValueEx(
        hkResult,
        value,
        NULL,
        &type,
        (BYTE*)retval,
        &cbCd);

    if (result != ERROR_SUCCESS) {
        return false;
    }
    return true;
}


CygwinPty::CygwinPty() {
}

bool CygwinPty::interrupt() {
    BOOL result = GenerateConsoleCtrlEvent(CTRL_C_EVENT, pi.dwProcessId);
    return (result != 0);
    //return write("\003", 1);
}

bool CygwinPty::connect() {
    //XXX clean handles ???
    //XXX MAX_PATH safe???

    TCHAR fakecygpty[2000] = {};
    {
        DWORD result = GetModuleFileName(
                           GetModuleHandle(NULL),
                           fakecygpty,
                           _countof(fakecygpty));

        if (result == 0 || result == _countof(fakecygpty)) { // error
            return false;
        }
        if (PathRemoveFileSpec(fakecygpty) == FALSE) {
            return false;
        }
        if (_tcscat_s(fakecygpty, _T("\\misc\\fakecygpty.exe")) != 0) {
            return false;
        }
    }

    TCHAR cd[2000] = {};
    if (!query_registory(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Cygwin\\setup"), TEXT("rootdir"), retval) &&
        !query_registory(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Wow6432Node\\Cygwin\\setup"), TEXT("rootdir"), retval) &&
        !query_registory(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Cygnus Solutions\\Cygwin\\mounts v2\\/"), TEXT("native"), retval) &&
        !query_registory(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Wow6432Node\\Cygnus Solutions\\Cygwin\\mounts v2\\/"), TEXT("native"), retval)) {
        return false;
    }

    SECURITY_ATTRIBUTES sa = {};
    sa.bInheritHandle = TRUE;

    STARTUPINFO si = {};
    memset((void*)&pi, 0, sizeof(pi));

    CreatePipe(&hStdInput    , &hMyStdInput , &sa, 0);
    CreatePipe(&hMyStdOutput , &hStdOutput  , &sa, 0);
    CreatePipe(&hMyStdError  , &hStdError   , &sa, 0);

    if (!DuplicateHandle(GetCurrentProcess(),hMyStdInput ,GetCurrentProcess(),NULL, 0, FALSE, DUPLICATE_SAME_ACCESS) ||
        !DuplicateHandle(GetCurrentProcess(),hMyStdOutput,GetCurrentProcess(),NULL, 0, FALSE, DUPLICATE_SAME_ACCESS) ||
        !DuplicateHandle(GetCurrentProcess(),hMyStdError ,GetCurrentProcess(),NULL, 0, FALSE, DUPLICATE_SAME_ACCESS)) {

        return false;
    }

    si.cb = sizeof(STARTUPINFO);
    si.dwFlags = STARTF_USESTDHANDLES;

    si.hStdInput = hStdInput;
    // si.hStdOutput = hStdOutput;
    // si.hStdError  = hStdError;
    // si.hStdInput  = GetStdHandle(STD_INPUT_HANDLE);
    si.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
    si.hStdError  = GetStdHandle(STD_ERROR_HANDLE);
    {
        TCHAR arg[] = _T("fakecygpty.exe /bin/zsh --login -i");
        BOOL result = CreateProcess(fakecygpty, arg, NULL, NULL, TRUE, CREATE_NEW_PROCESS_GROUP, NULL, cd, &si, &pi);
        if (result == 0) {
            return false;
        }
    }
    return true;
}
}