winpty/agent/Win32Console.cc
2012-04-01 02:13:21 -07:00

300 lines
9.3 KiB
C++

// Copyright (c) 2011-2012 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 "Win32Console.h"
#include "AgentAssert.h"
#include "../shared/DebugClient.h"
#include <windows.h>
Win32Console::Win32Console()
{
m_conin = GetStdHandle(STD_INPUT_HANDLE);
m_conout = GetStdHandle(STD_OUTPUT_HANDLE);
}
Win32Console::~Win32Console()
{
CloseHandle(m_conin);
CloseHandle(m_conout);
}
HANDLE Win32Console::conin()
{
return m_conin;
}
HANDLE Win32Console::conout()
{
return m_conout;
}
HWND Win32Console::hwnd()
{
return GetConsoleWindow();
}
void Win32Console::postCloseMessage()
{
HWND h = hwnd();
if (h != NULL)
PostMessage(h, WM_CLOSE, 0, 0);
}
// A Windows console window can never be larger than the desktop window. To
// maximize the possible size of the console in rows*cols, try to configure
// the console with a small font.
void Win32Console::setSmallFont()
{
// Some of these types and functions are missing from the MinGW headers.
// Others are undocumented.
struct AGENT_CONSOLE_FONT_INFO {
DWORD nFont;
COORD dwFontSize;
};
struct AGENT_CONSOLE_FONT_INFOEX {
ULONG cbSize;
DWORD nFont;
COORD dwFontSize;
UINT FontFamily;
UINT FontWeight;
WCHAR FaceName[LF_FACESIZE];
};
typedef BOOL WINAPI SetConsoleFontType(
HANDLE hOutput,
DWORD dwFontIndex);
typedef BOOL WINAPI GetCurrentConsoleFontType(
HANDLE hOutput,
BOOL bMaximize,
AGENT_CONSOLE_FONT_INFO *pFontInfo);
typedef BOOL WINAPI SetCurrentConsoleFontExType(
HANDLE hConsoleOutput,
BOOL bMaximumWindow,
AGENT_CONSOLE_FONT_INFOEX *lpConsoleCurrentFontEx);
typedef COORD WINAPI GetConsoleFontSizeType(
HANDLE hConsoleOutput,
DWORD nFont);
HINSTANCE dll = LoadLibrary(L"kernel32.dll");
ASSERT(dll != NULL);
SetConsoleFontType *pSetConsoleFont =
(SetConsoleFontType*)GetProcAddress(dll, "SetConsoleFont");
GetCurrentConsoleFontType *pGetCurrentConsoleFont =
(GetCurrentConsoleFontType*)GetProcAddress(dll, "GetCurrentConsoleFont");
SetCurrentConsoleFontExType *pSetCurrentConsoleFontEx =
(SetCurrentConsoleFontExType*)GetProcAddress(dll, "SetCurrentConsoleFontEx");
GetConsoleFontSizeType *pGetConsoleFontSize =
(GetConsoleFontSizeType*)GetProcAddress(dll, "GetConsoleFontSize");
BOOL success;
// The undocumented GetNumberOfConsoleFonts API reports that my Windows 7
// system has 12 fonts on it. Each font is really just a differently-sized
// raster/Terminal font. Font index 0 is the smallest font, so we want to
// choose it.
if (pGetConsoleFontSize == NULL) {
// This API should exist even on Windows XP.
trace("error: GetConsoleFontSize API is missing");
return;
}
if (pGetCurrentConsoleFont == NULL) {
// This API should exist even on Windows XP.
trace("error: GetCurrentConsoleFont API is missing");
return;
}
AGENT_CONSOLE_FONT_INFO fi;
success = pGetCurrentConsoleFont(m_conout, FALSE, &fi);
if (!success) {
trace("error: GetCurrentConsoleFont failed");
return;
}
COORD smallest = pGetConsoleFontSize(m_conout, 0);
if (smallest.X == 0 || smallest.Y == 0) {
trace("error: GetConsoleFontSize failed");
return;
}
trace("font #0: X=%d Y=%d", smallest.X, smallest.Y);
trace("current font: idx=%d X=%d Y=%d",
(int)fi.nFont, fi.dwFontSize.X, fi.dwFontSize.Y);
if (fi.dwFontSize.X <= smallest.X && fi.dwFontSize.Y <= smallest.Y)
return;
// First try to call the documented Vista API.
if (pSetCurrentConsoleFontEx != NULL) {
AGENT_CONSOLE_FONT_INFOEX fix = {0};
fix.cbSize = sizeof(fix);
fix.nFont = 0;
fix.dwFontSize = smallest;
success = pSetCurrentConsoleFontEx(m_conout, FALSE, &fix);
trace("SetCurrentConsoleFontEx call %s",
success ? "succeeded" : "failed");
return;
}
// Then try to call the undocumented Windows XP API.
//
// Somewhat described here:
// http://blogs.microsoft.co.il/blogs/pavely/archive/2009/07/23/changing-console-fonts.aspx
//
if (pSetConsoleFont != NULL) {
success = pSetConsoleFont(m_conout, 0);
trace("SetConsoleFont call %s", success ? "succeeded" : "failed");
return;
}
trace("Not setting console font size -- "
"neither SetConsoleFont nor SetCurrentConsoleFontEx API exists");
}
Coord Win32Console::bufferSize()
{
// TODO: error handling
CONSOLE_SCREEN_BUFFER_INFO info;
memset(&info, 0, sizeof(info));
if (!GetConsoleScreenBufferInfo(m_conout, &info)) {
trace("GetConsoleScreenBufferInfo failed");
}
return info.dwSize;
}
SmallRect Win32Console::windowRect()
{
// TODO: error handling
CONSOLE_SCREEN_BUFFER_INFO info;
memset(&info, 0, sizeof(info));
if (!GetConsoleScreenBufferInfo(m_conout, &info)) {
trace("GetConsoleScreenBufferInfo failed");
}
return info.srWindow;
}
void Win32Console::resizeBuffer(const Coord &size)
{
// TODO: error handling
if (!SetConsoleScreenBufferSize(m_conout, size)) {
trace("SetConsoleScreenBufferSize failed");
}
}
void Win32Console::moveWindow(const SmallRect &rect)
{
// TODO: error handling
if (!SetConsoleWindowInfo(m_conout, TRUE, &rect)) {
trace("SetConsoleWindowInfo failed");
}
}
void Win32Console::reposition(const Coord &newBufferSize,
const SmallRect &newWindowRect)
{
// Windows has one API for resizing the screen buffer and a different one
// for resizing the window. It seems that either API can fail if the
// window does not fit on the screen buffer.
const SmallRect origWindowRect(windowRect());
const SmallRect origBufferRect(Coord(), bufferSize());
ASSERT(!newBufferSize.isEmpty());
SmallRect bufferRect(Coord(), newBufferSize);
ASSERT(bufferRect.contains(newWindowRect));
SmallRect tempWindowRect = origWindowRect.intersected(bufferRect);
if (tempWindowRect.width() <= 0) {
tempWindowRect.setLeft(newBufferSize.X - 1);
tempWindowRect.setWidth(1);
}
if (tempWindowRect.height() <= 0) {
tempWindowRect.setTop(newBufferSize.Y - 1);
tempWindowRect.setHeight(1);
}
// Alternatively, if we can immediately use the new window size,
// do that instead.
if (origBufferRect.contains(newWindowRect))
tempWindowRect = newWindowRect;
if (tempWindowRect != origWindowRect)
moveWindow(tempWindowRect);
resizeBuffer(newBufferSize);
if (newWindowRect != tempWindowRect)
moveWindow(newWindowRect);
}
Coord Win32Console::cursorPosition()
{
// TODO: error handling
CONSOLE_SCREEN_BUFFER_INFO info;
memset(&info, 0, sizeof(info));
if (!GetConsoleScreenBufferInfo(m_conout, &info)) {
trace("GetConsoleScreenBufferInfo failed");
}
return info.dwCursorPosition;
}
void Win32Console::setCursorPosition(const Coord &coord)
{
// TODO: error handling
if (!SetConsoleCursorPosition(m_conout, coord)) {
trace("SetConsoleCursorPosition failed");
}
}
void Win32Console::writeInput(const INPUT_RECORD *ir, int count)
{
// TODO: error handling
DWORD dummy = 0;
if (!WriteConsoleInput(m_conin, ir, count, &dummy)) {
trace("WriteConsoleInput failed");
}
}
bool Win32Console::processedInputMode()
{
// TODO: error handling
DWORD mode = 0;
if (!GetConsoleMode(m_conin, &mode)) {
trace("GetConsoleMode failed");
}
return (mode & ENABLE_PROCESSED_INPUT) == ENABLE_PROCESSED_INPUT;
}
void Win32Console::read(const SmallRect &rect, CHAR_INFO *data)
{
// TODO: error handling
SmallRect tmp(rect);
if (!ReadConsoleOutput(m_conout, data, rect.size(), Coord(), &tmp)) {
trace("ReadConsoleOutput failed [x:%d,y:%d,w:%d,h:%d]",
rect.Left, rect.Top, rect.width(), rect.height());
}
}
void Win32Console::write(const SmallRect &rect, const CHAR_INFO *data)
{
// TODO: error handling
SmallRect tmp(rect);
if (!WriteConsoleOutput(m_conout, data, rect.size(), Coord(), &tmp)) {
trace("WriteConsoleOutput failed");
}
}