Sweden-Number/programs/wineconsole/wineconsole.c

883 lines
31 KiB
C

/*
* an application for displaying Win32 console
*
* Copyright 2001, 2002 Eric Pouech
*
* 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 "config.h"
#include "wine/port.h"
#include <stdio.h>
#include <stdarg.h>
#include "wine/server.h"
#include "winecon_private.h"
#include "winnls.h"
#include "winuser.h"
#include "wine/unicode.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(wineconsole);
void WINECON_Fatal(const char* msg)
{
WINE_ERR("%s\n", msg);
ExitProcess(0);
}
static void printf_res(UINT uResId, ...)
{
WCHAR buffer[1024];
CHAR ansi[1024];
va_list args;
va_start(args, uResId);
LoadStringW(GetModuleHandleW(NULL), uResId, buffer, ARRAY_SIZE(buffer));
WideCharToMultiByte(CP_UNIXCP, 0, buffer, -1, ansi, sizeof(ansi), NULL, NULL);
vprintf(ansi, args);
va_end(args);
}
static void WINECON_Usage(void)
{
printf_res(IDS_USAGE_HEADER);
printf_res(IDS_USAGE_COMMAND);
printf_res(IDS_USAGE_FOOTER);
}
/******************************************************************
* WINECON_FetchCells
*
* updates the local copy of cells (band to update)
*/
static void WINECON_FetchCells(struct inner_data* data, int upd_tp, int upd_bm)
{
SMALL_RECT region = { 0, upd_tp, data->curcfg.sb_width - 1, upd_bm };
COORD size = { data->curcfg.sb_width, data->curcfg.sb_height };
COORD coord = { 0, upd_tp };
if (ReadConsoleOutputW(data->console, data->cells, size, coord, &region))
data->fnRefresh(data, upd_tp, upd_bm);
}
/******************************************************************
* WINECON_ResizeWithContainer
*
* For console embedded in a container (e.g. user in a win32 window, or (n)curses
* in a TERM, perform resize of console (screen buffer and window) to fit in
* (new) container size.
*/
void WINECON_ResizeWithContainer(struct inner_data* data, int width, int height)
{
struct config_data cfg;
if (data->in_set_config || data->in_grab_changes) return;
cfg = data->curcfg;
cfg.win_width = width;
cfg.win_height = height;
/* auto size screen-buffer if it's now smaller than window */
if (cfg.sb_width < cfg.win_width) cfg.sb_width = cfg.win_width;
if (cfg.sb_height < cfg.win_height) cfg.sb_height = cfg.win_height;
/* and reset window pos so that we don't display outside of the screen-buffer */
if (cfg.win_pos.X + cfg.win_width > cfg.sb_width) cfg.win_pos.X = cfg.sb_width - cfg.win_width;
if (cfg.win_pos.Y + cfg.win_height > cfg.sb_height) cfg.win_pos.Y = cfg.sb_height - cfg.win_height;
WINECON_SetConfig(data, &cfg);
}
/******************************************************************
* WINECON_SetHistorySize
*
*
*/
static BOOL WINECON_SetHistorySize(HANDLE hConIn, int size)
{
struct condrv_input_info_params params = { SET_CONSOLE_INPUT_INFO_HISTORY_SIZE };
params.info.history_size = size;
return DeviceIoControl(hConIn, IOCTL_CONDRV_SET_INPUT_INFO, &params, sizeof(params), NULL, 0, NULL, NULL);
}
/******************************************************************
* WINECON_SetHistoryMode
*
*
*/
static BOOL WINECON_SetHistoryMode(HANDLE hConIn, int mode)
{
struct condrv_input_info_params params = { SET_CONSOLE_INPUT_INFO_HISTORY_MODE };
params.info.history_mode = mode;
return DeviceIoControl(hConIn, IOCTL_CONDRV_SET_INPUT_INFO, &params, sizeof(params), NULL, 0, NULL, NULL);
}
/******************************************************************
* WINECON_SetInsertMode
*
*
*/
static void WINECON_SetInsertMode(HANDLE hConIn, unsigned int enable)
{
DWORD mode;
GetConsoleMode(hConIn, &mode);
if (enable)
mode |= ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS;
else
mode &= ~ENABLE_INSERT_MODE;
SetConsoleMode(hConIn, mode);
}
/******************************************************************
* WINECON_GetConsoleTitle
*
*
*/
BOOL WINECON_GetConsoleTitle(HANDLE hConIn, WCHAR* buffer, size_t len)
{
DWORD size;
if (!DeviceIoControl(hConIn, IOCTL_CONDRV_GET_TITLE, NULL, 0, buffer, len - sizeof(WCHAR), &size, NULL))
return FALSE;
buffer[size / sizeof(WCHAR)] = 0;
return TRUE;
}
/******************************************************************
* WINECON_SetEditionMode
*
*
*/
static BOOL WINECON_SetEditionMode(HANDLE hConIn, int edition_mode)
{
struct condrv_input_info_params params = { SET_CONSOLE_INPUT_INFO_EDITION_MODE };
params.info.edition_mode = edition_mode;
return DeviceIoControl(hConIn, IOCTL_CONDRV_SET_INPUT_INFO, &params, sizeof(params), NULL, 0, NULL, NULL);
}
/******************************************************************
* WINECON_SetColors
*
* Sets ColorTable and Pop-up menu colors
*/
static void WINECON_SetColors(struct inner_data *data, const struct config_data* cfg)
{
struct condrv_output_info_params params =
{ SET_CONSOLE_OUTPUT_INFO_COLORTABLE | SET_CONSOLE_OUTPUT_INFO_POPUP_ATTR };
memcpy(data->curcfg.color_map, cfg->color_map, sizeof(data->curcfg.color_map));
data->curcfg.popup_attr = cfg->popup_attr;
params.info.popup_attr = cfg->popup_attr;
memcpy(params.info.color_map, cfg->color_map, sizeof(cfg->color_map));
DeviceIoControl(data->console, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL, NULL);
}
/******************************************************************
* WINECON_GrabChanges
*
* A change occurs, try to figure out which
*/
void WINECON_GrabChanges(struct inner_data* data)
{
struct condrv_renderer_event *evts = data->events;
int i, ev_found;
DWORD num;
if (data->in_grab_changes) return;
if (!GetOverlappedResult(data->console, &data->overlapped, &num, FALSE))
{
if (GetLastError() == ERROR_IO_INCOMPLETE) return;
ERR( "failed to get renderer events: %u\n", GetLastError() );
data->dying = TRUE;
return;
}
num /= sizeof(data->events[0]);
WINE_TRACE( "got %u events\n", num );
/* FIXME: should do some event compression here (cursor pos, update) */
/* step 1: keep only last cursor pos event */
ev_found = -1;
for (i = num - 1; i >= 0; i--)
{
if (evts[i].event == CONSOLE_RENDERER_CURSOR_POS_EVENT)
{
if (ev_found != -1)
{
WINE_TRACE("%u/%u: curs-pos(%d,%d) ignoring\n", i+1, num, evts[i].u.cursor_pos.x, evts[i].u.cursor_pos.y);
evts[i].event = CONSOLE_RENDERER_NONE_EVENT;
}
ev_found = i;
}
}
/* step 2: manage update events */
ev_found = -1;
for (i = 0; i < num; i++)
{
if (evts[i].event == CONSOLE_RENDERER_NONE_EVENT ||
evts[i].event == CONSOLE_RENDERER_CURSOR_POS_EVENT ||
evts[i].event == CONSOLE_RENDERER_CURSOR_GEOM_EVENT) continue;
if (evts[i].event != CONSOLE_RENDERER_UPDATE_EVENT)
{
ev_found = -1;
continue;
}
if (ev_found != -1 && /* Only 2 cases where they CANNOT merge */
!(evts[i ].u.update.bottom + 1 < evts[ev_found].u.update.top ||
evts[ev_found].u.update.bottom + 1 < evts[i ].u.update.top))
{
WINE_TRACE("%u/%u: update(%d,%d) merging with %u\n", ev_found+1, num, evts[i].u.update.top, evts[i].u.update.bottom, i+1);
evts[i].u.update.top = min(evts[i ].u.update.top,
evts[ev_found].u.update.top);
evts[i].u.update.bottom = max(evts[i ].u.update.bottom,
evts[ev_found].u.update.bottom);
evts[ev_found].event = CONSOLE_RENDERER_NONE_EVENT;
}
ev_found = i;
}
data->in_grab_changes = TRUE;
for (i = 0; i < num; i++)
{
switch (evts[i].event)
{
case CONSOLE_RENDERER_NONE_EVENT:
WINE_TRACE("%u/%u: NOP\n", i+1, num);
break;
case CONSOLE_RENDERER_TITLE_EVENT:
WINE_TRACE("%u/%u: title()\n", i+1, num);
data->fnSetTitle(data);
break;
case CONSOLE_RENDERER_SB_RESIZE_EVENT:
if (data->curcfg.sb_width != evts[i].u.resize.width ||
data->curcfg.sb_height != evts[i].u.resize.height)
{
WINE_TRACE("%u/%u: resize(%d,%d)\n", i+1, num, evts[i].u.resize.width, evts[i].u.resize.height);
data->curcfg.sb_width = evts[i].u.resize.width;
data->curcfg.sb_height = evts[i].u.resize.height;
data->cells = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, data->cells,
data->curcfg.sb_width * data->curcfg.sb_height * sizeof(CHAR_INFO));
if (!data->cells) WINECON_Fatal("OOM");
data->fnResizeScreenBuffer(data);
data->fnComputePositions(data);
}
break;
case CONSOLE_RENDERER_UPDATE_EVENT:
WINE_TRACE("%u/%u: update(%d,%d)\n", i+1, num, evts[i].u.update.top, evts[i].u.update.bottom);
WINECON_FetchCells(data, evts[i].u.update.top, evts[i].u.update.bottom);
break;
case CONSOLE_RENDERER_CURSOR_POS_EVENT:
if (evts[i].u.cursor_pos.x != data->cursor.X || evts[i].u.cursor_pos.y != data->cursor.Y)
{
WINE_TRACE("%u/%u: curs-pos(%d,%d)\n", i+1, num, evts[i].u.cursor_pos.x, evts[i].u.cursor_pos.y);
data->cursor.X = evts[i].u.cursor_pos.x;
data->cursor.Y = evts[i].u.cursor_pos.y;
data->fnPosCursor(data);
}
break;
case CONSOLE_RENDERER_CURSOR_GEOM_EVENT:
if (evts[i].u.cursor_geom.size != data->curcfg.cursor_size ||
evts[i].u.cursor_geom.visible != data->curcfg.cursor_visible)
{
WINE_TRACE("%u/%u: curs-geom(%d,%d)\n", i+1, num,
evts[i].u.cursor_geom.size, evts[i].u.cursor_geom.visible);
data->fnShapeCursor(data, evts[i].u.cursor_geom.size,
evts[i].u.cursor_geom.visible, FALSE);
}
break;
case CONSOLE_RENDERER_DISPLAY_EVENT:
if (evts[i].u.display.left != data->curcfg.win_pos.X)
{
WINE_TRACE("%u/%u: h-scroll(%d)\n", i+1, num, evts[i].u.display.left);
data->fnScroll(data, evts[i].u.display.left, TRUE);
data->fnPosCursor(data);
}
if (evts[i].u.display.top != data->curcfg.win_pos.Y)
{
WINE_TRACE("%u/%u: v-scroll(%d)\n", i+1, num, evts[i].u.display.top);
data->fnScroll(data, evts[i].u.display.top, FALSE);
data->fnPosCursor(data);
}
if (evts[i].u.display.width != data->curcfg.win_width ||
evts[i].u.display.height != data->curcfg.win_height)
{
WINE_TRACE("%u/%u: win-size(%d,%d)\n", i+1, num, evts[i].u.display.width, evts[i].u.display.height);
data->curcfg.win_width = evts[i].u.display.width;
data->curcfg.win_height = evts[i].u.display.height;
data->fnComputePositions(data);
}
break;
case CONSOLE_RENDERER_EXIT_EVENT:
data->dying = TRUE;
WINE_TRACE("%u/%u: Exit!!\n", i+1, num);
return;
default:
WINE_FIXME("Unknown event type (%d)\n", evts[i].event);
}
}
data->in_grab_changes = FALSE;
if (!DeviceIoControl(data->console, IOCTL_CONDRV_GET_RENDERER_EVENTS, NULL, 0, data->events,
sizeof(data->events), NULL, &data->overlapped) && GetLastError() != ERROR_IO_PENDING)
{
ERR("failed to get renderer events: %u\n", GetLastError());
data->dying = TRUE;
}
}
/******************************************************************
* WINECON_SetConfig
*
* Apply to data all the configuration elements from cfg. This includes modification
* of server side equivalent and visual parts.
* If force is FALSE, only the changed items are modified.
*/
void WINECON_SetConfig(struct inner_data* data, const struct config_data* cfg)
{
if (data->in_set_config) return;
data->in_set_config = TRUE;
if (data->curcfg.cursor_size != cfg->cursor_size ||
data->curcfg.cursor_visible != cfg->cursor_visible)
{
CONSOLE_CURSOR_INFO cinfo;
cinfo.dwSize = cfg->cursor_size;
/* <FIXME>: this hack is needed to pass through the invariant test operation on the server side
* (no notification is sent when invariant operation is requested)
*/
cinfo.bVisible = !cfg->cursor_visible;
SetConsoleCursorInfo(data->console, &cinfo);
/* </FIXME> */
cinfo.bVisible = cfg->cursor_visible;
/* this shall update (through notif) curcfg */
SetConsoleCursorInfo(data->console, &cinfo);
}
if (data->curcfg.history_size != cfg->history_size)
{
data->curcfg.history_size = cfg->history_size;
WINECON_SetHistorySize(data->console, cfg->history_size);
}
if (data->curcfg.history_nodup != cfg->history_nodup)
{
data->curcfg.history_nodup = cfg->history_nodup;
WINECON_SetHistoryMode(data->console, cfg->history_nodup);
}
if (data->curcfg.insert_mode != cfg->insert_mode)
{
data->curcfg.insert_mode = cfg->insert_mode;
WINECON_SetInsertMode(data->console, cfg->insert_mode);
}
data->curcfg.menu_mask = cfg->menu_mask;
data->curcfg.quick_edit = cfg->quick_edit;
if (strcmpiW(data->curcfg.face_name, cfg->face_name) || data->curcfg.cell_width != cfg->cell_width ||
data->curcfg.cell_height != cfg->cell_height || data->curcfg.font_pitch_family != cfg->font_pitch_family ||
data->curcfg.font_weight != cfg->font_weight)
{
struct condrv_output_info_params *params;
size_t len = lstrlenW(cfg->face_name);
RECT r;
data->fnSetFont(data, cfg->face_name, cfg->cell_height, cfg->font_weight);
SystemParametersInfoW(SPI_GETWORKAREA, 0, &r, 0);
if ((params = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*params) + len * sizeof(WCHAR))))
{
params->mask = SET_CONSOLE_OUTPUT_INFO_MAX_SIZE | SET_CONSOLE_OUTPUT_INFO_FONT;
params->info.max_width = (r.right - r.left) / cfg->cell_width;
params->info.max_height = (r.bottom - r.top - GetSystemMetrics(SM_CYCAPTION)) / cfg->cell_height;
params->info.font_width = cfg->cell_width;
params->info.font_height = cfg->cell_height;
params->info.font_weight = cfg->font_weight;
params->info.font_pitch_family = cfg->font_pitch_family;
memcpy(params + 1, cfg->face_name, len * sizeof(WCHAR));
DeviceIoControl(data->console, IOCTL_CONDRV_SET_OUTPUT_INFO, params, sizeof(*params) + len * sizeof(WCHAR),
NULL, 0, NULL, NULL);
}
}
if (data->curcfg.def_attr != cfg->def_attr)
{
DWORD screen_size, written;
COORD top_left = {0,0};
data->curcfg.def_attr = cfg->def_attr;
screen_size = cfg->win_width * (cfg->win_height + 1);
FillConsoleOutputAttribute(data->console, cfg->def_attr, screen_size, top_left, &written);
SetConsoleTextAttribute(data->console, cfg->def_attr);
}
WINECON_SetColors(data, cfg);
/* now let's look at the window / sb size changes...
* since the server checks that sb is always bigger than window,
* we have to take care of doing the operations in the right order
*/
/* a set of macros to make things easier to read
* The Test<A><B> macros test if the <A> (width/height) needs to be changed
* for <B> (window / ScreenBuffer)
* The Change<A><B> actually modify the <B> dimension of <A>.
*/
#define TstSBfWidth() (data->curcfg.sb_width != cfg->sb_width)
#define TstWinHPos() (data->curcfg.win_width != cfg->win_width || data->curcfg.win_pos.X != cfg->win_pos.X)
#define ChgSBfWidth() do {c.X = cfg->sb_width; \
c.Y = data->curcfg.sb_height;\
SetConsoleScreenBufferSize(data->console, c);\
} while (0)
#define ChgWinHPos() do {pos.Left = cfg->win_pos.X - data->curcfg.win_pos.X; \
pos.Top = 0; \
pos.Right = pos.Left + cfg->win_width - data->curcfg.win_width; \
pos.Bottom = 0; \
SetConsoleWindowInfo(data->console, FALSE, &pos);\
} while (0)
#define TstSBfHeight() (data->curcfg.sb_height != cfg->sb_height)
#define TstWinVPos() (data->curcfg.win_height != cfg->win_height || data->curcfg.win_pos.Y != cfg->win_pos.Y)
/* since we're going to apply height after width is done, we use width as defined
* in cfg, and not in data->curcfg because if won't be updated yet */
#define ChgSBfHeight() do {c.X = cfg->sb_width; c.Y = cfg->sb_height; \
SetConsoleScreenBufferSize(data->console, c); \
} while (0)
#define ChgWinVPos() do {pos.Left = 0; \
pos.Top = cfg->win_pos.Y - data->curcfg.win_pos.Y; \
pos.Right = 0; \
pos.Bottom = pos.Top + cfg->win_height - data->curcfg.win_height; \
SetConsoleWindowInfo(data->console, FALSE, &pos);\
} while (0)
do
{
COORD c;
SMALL_RECT pos;
if (TstSBfWidth())
{
if (TstWinHPos())
{
/* we're changing both at the same time, do it in the right order */
if (cfg->sb_width >= data->curcfg.win_width)
{
ChgSBfWidth(); ChgWinHPos();
}
else
{
ChgWinHPos(); ChgSBfWidth();
}
}
else ChgSBfWidth();
}
else if (TstWinHPos()) ChgWinHPos();
if (TstSBfHeight())
{
if (TstWinVPos())
{
if (cfg->sb_height >= data->curcfg.win_height)
{
ChgSBfHeight(); ChgWinVPos();
}
else
{
ChgWinVPos(); ChgSBfHeight();
}
}
else ChgSBfHeight();
}
else if (TstWinVPos()) ChgWinVPos();
} while (0);
#undef TstSBfWidth
#undef TstWinHPos
#undef ChgSBfWidth
#undef ChgWinHPos
#undef TstSBfHeight
#undef TstWinVPos
#undef ChgSBfHeight
#undef ChgWinVPos
data->curcfg.exit_on_die = cfg->exit_on_die;
if (data->curcfg.edition_mode != cfg->edition_mode)
{
data->curcfg.edition_mode = cfg->edition_mode;
WINECON_SetEditionMode(data->console, cfg->edition_mode);
}
/* we now need to gather all events we got from the operations above,
* in order to get data correctly updated
*/
WINECON_GrabChanges(data);
data->in_set_config = FALSE;
}
/******************************************************************
* WINECON_Delete
*
* Destroy wineconsole internal data
*/
static void WINECON_Delete(struct inner_data* data)
{
if (!data) return;
if (data->fnDeleteBackend) data->fnDeleteBackend(data);
if (data->console) CloseHandle(data->console);
if (data->hProcess) CloseHandle(data->hProcess);
if (data->overlapped.hEvent) CloseHandle(data->overlapped.hEvent);
HeapFree(GetProcessHeap(), 0, data->curcfg.registry);
HeapFree(GetProcessHeap(), 0, data->cells);
HeapFree(GetProcessHeap(), 0, data);
}
/******************************************************************
* WINECON_GetServerConfig
*
* Fills data->curcfg with the actual configuration running in the server
* (getting real information on the server, and not relying on cached
* information in data)
*/
static BOOL WINECON_GetServerConfig(struct inner_data* data)
{
struct condrv_input_info input_info;
struct condrv_output_info output_info;
DWORD mode;
if (!DeviceIoControl(data->console, IOCTL_CONDRV_GET_INPUT_INFO, NULL, 0,
&input_info, sizeof(input_info), NULL, NULL))
return FALSE;
data->curcfg.history_size = input_info.history_size;
data->curcfg.history_nodup = input_info.history_mode;
data->curcfg.edition_mode = input_info.edition_mode;
GetConsoleMode(data->console, &mode);
data->curcfg.insert_mode = (mode & (ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS)) ==
(ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS);
if (!DeviceIoControl(data->console, IOCTL_CONDRV_GET_OUTPUT_INFO, NULL, 0,
&output_info, sizeof(output_info), NULL, NULL))
return FALSE;
data->curcfg.cursor_size = output_info.cursor_size;
data->curcfg.cursor_visible = output_info.cursor_visible;
data->curcfg.def_attr = output_info.attr;
data->curcfg.sb_width = output_info.width;
data->curcfg.sb_height = output_info.height;
data->curcfg.win_width = output_info.win_right - output_info.win_left + 1;
data->curcfg.win_height = output_info.win_bottom - output_info.win_top + 1;
WINECON_DumpConfig("first cfg: ", &data->curcfg);
return TRUE;
}
/******************************************************************
* WINECON_Spawn
*
* Spawn the child process when invoked with wineconsole foo bar
*/
static BOOL WINECON_Spawn(struct inner_data* data, LPWSTR cmdLine, HANDLE con_in, HANDLE con_out)
{
PROCESS_INFORMATION info;
STARTUPINFOW startup;
BOOL done;
/* we're in the case wineconsole <exe> <options>... spawn the new process */
memset(&startup, 0, sizeof(startup));
startup.cb = sizeof(startup);
startup.dwFlags = STARTF_USESTDHANDLES;
/* the attributes of wineconsole's handles are not adequate for inheritance, so
* get them with the correct attributes before process creation
*/
if (!DuplicateHandle(GetCurrentProcess(), con_in, GetCurrentProcess(),
&startup.hStdInput, GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE, TRUE, 0) ||
!DuplicateHandle(GetCurrentProcess(), con_out, GetCurrentProcess(),
&startup.hStdOutput, GENERIC_READ|GENERIC_WRITE, TRUE, 0) ||
!DuplicateHandle(GetCurrentProcess(), con_out, GetCurrentProcess(),
&startup.hStdError, GENERIC_READ|GENERIC_WRITE, TRUE, 0))
{
WINE_ERR("Can't dup handles\n");
/* no need to delete handles, we're exiting the program anyway */
return FALSE;
}
done = CreateProcessW(NULL, cmdLine, NULL, NULL, TRUE, 0L, NULL, NULL, &startup, &info);
if (done)
{
data->hProcess = info.hProcess;
CloseHandle(info.hThread);
}
else printf_res(IDS_CMD_LAUNCH_FAILED, wine_dbgstr_w(cmdLine));
/* we no longer need the handles passed to the child for the console */
CloseHandle(startup.hStdInput);
CloseHandle(startup.hStdOutput);
CloseHandle(startup.hStdError);
return done;
}
/******************************************************************
* WINECON_Init
*
* Initialisation part I. Creation of server object (console input and
* active screen buffer)
*/
static struct inner_data* WINECON_Init(HINSTANCE hInst, DWORD pid, LPCWSTR appname,
enum init_return (*backend)(struct inner_data*),
INT nCmdShow, WCHAR *cmdline)
{
struct condrv_input_info_params input_params;
OBJECT_ATTRIBUTES attr = {sizeof(attr)};
struct inner_data* data = NULL;
DWORD ret;
struct config_data cfg;
STARTUPINFOW si;
UNICODE_STRING string;
IO_STATUS_BLOCK io;
condrv_handle_t h;
HANDLE con_in;
HANDLE con_out;
NTSTATUS status;
static const WCHAR renderer_pathW[] = {'\\','D','e','v','i','c','e','\\','C','o','n','D','r','v',
'\\','R','e','n','d','e','r','e','r',0};
data = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*data));
if (!data) return 0;
GetStartupInfoW(&si);
if (pid == 0) appname = si.lpTitle;
data->nCmdShow = nCmdShow;
/* load settings */
WINECON_RegLoad(appname, &cfg);
/* some overrides */
if (pid == 0)
{
if (si.dwFlags & STARTF_USECOUNTCHARS)
{
cfg.sb_width = si.dwXCountChars;
cfg.sb_height = si.dwYCountChars;
}
if (si.dwFlags & STARTF_USEFILLATTRIBUTE)
cfg.def_attr = si.dwFillAttribute;
/* should always be defined */
}
if (!(data->overlapped.hEvent = CreateEventW(NULL, TRUE, TRUE, NULL))) goto error;
/* the handles here are created without the whistles and bells required by console
* (mainly because wineconsole doesn't need it)
* - they are not inheritable
* - hConIn is not synchronizable
*/
SERVER_START_REQ(alloc_console)
{
req->access = GENERIC_READ | GENERIC_WRITE;
req->attributes = 0;
req->pid = pid;
ret = !wine_server_call_err( req );
con_in = wine_server_ptr_handle( reply->handle_in );
}
SERVER_END_REQ;
if (!ret) goto error;
WINE_TRACE("using con_in %p, renderer %p\n", con_in, data->console);
RtlInitUnicodeString(&string, renderer_pathW);
attr.ObjectName = &string;
status = NtCreateFile(&data->console, FILE_READ_DATA | FILE_WRITE_DATA | FILE_WRITE_PROPERTIES
| FILE_READ_PROPERTIES | SYNCHRONIZE, &attr, &io, NULL, FILE_ATTRIBUTE_NORMAL,
0, FILE_OPEN, FILE_NON_DIRECTORY_FILE, NULL, 0);
if (status) goto error;
h = condrv_handle(con_in);
if (!DeviceIoControl(data->console, IOCTL_CONDRV_ATTACH_RENDERER, &h, sizeof(h), NULL, 0, NULL, NULL))
goto error;
SERVER_START_REQ(create_console_output)
{
req->handle_in = wine_server_obj_handle( con_in );
req->access = GENERIC_WRITE|GENERIC_READ;
req->attributes = 0;
req->share = FILE_SHARE_READ|FILE_SHARE_WRITE;
ret = !wine_server_call_err( req );
con_out = wine_server_ptr_handle( reply->handle_out );
}
SERVER_END_REQ;
if (!ret) goto error;
WINE_TRACE("using con_out %p\n", con_out);
/* filling data->curcfg from cfg */
if ((*backend)(data) != init_success) goto error;
WINECON_GetServerConfig(data);
data->cells = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
data->curcfg.sb_width * data->curcfg.sb_height * sizeof(CHAR_INFO));
if (!data->cells) goto error;
data->fnResizeScreenBuffer(data);
data->fnComputePositions(data);
WINECON_SetConfig(data, &cfg);
data->curcfg.registry = cfg.registry;
WINECON_DumpConfig("fint", &data->curcfg);
memset(&input_params, 0, sizeof(input_params));
input_params.mask = SET_CONSOLE_INPUT_INFO_WIN;
input_params.info.win = condrv_handle(data->hWnd);
ret = DeviceIoControl(data->console, IOCTL_CONDRV_SET_INPUT_INFO, &input_params, sizeof(input_params),
NULL, 0, NULL, NULL);
if (!ret) goto error;
ret = DeviceIoControl(data->console, IOCTL_CONDRV_SET_TITLE, (void *)appname,
lstrlenW(appname) * sizeof(WCHAR), NULL, 0, NULL, NULL);
if (!ret) goto error;
if (cmdline && !WINECON_Spawn(data, cmdline, con_in, con_out)) goto error;
CloseHandle(con_in);
CloseHandle(con_out);
return data;
error:
WINE_ERR("failed to init.\n");
WINECON_Delete(data);
return NULL;
}
struct wc_init {
LPCSTR ptr;
enum {from_event, from_process_name} mode;
enum init_return (*backend)(struct inner_data*);
HANDLE event;
};
#define WINECON_CMD_SHOW_USAGE 0x10000
/******************************************************************
* WINECON_ParseOptions
*
* RETURNS
* On success: 0
* On error: error string id optionally with the CMD_SHOW_USAGE flag
*/
static UINT WINECON_ParseOptions(const char* lpCmdLine, struct wc_init* wci)
{
memset(wci, 0, sizeof(*wci));
wci->ptr = lpCmdLine;
wci->mode = from_process_name;
wci->backend = WCUSER_InitBackend;
for (;;)
{
while (*wci->ptr == ' ' || *wci->ptr == '\t') wci->ptr++;
if (wci->ptr[0] != '-') break;
if (strncmp(wci->ptr, "--use-event=", 12) == 0)
{
char* end;
wci->event = ULongToHandle( strtoul(wci->ptr + 12, &end, 10) );
if (end == wci->ptr + 12) return IDS_CMD_INVALID_EVENT_ID;
wci->mode = from_event;
wci->ptr = end;
}
else if (!strncmp(wci->ptr, "--help", 6) &&
(!wci->ptr[6] || wci->ptr[6] == ' ' || wci->ptr[6] == '\t'))
return IDS_CMD_ABOUT|WINECON_CMD_SHOW_USAGE;
else
return IDS_CMD_INVALID_OPTION|WINECON_CMD_SHOW_USAGE;
}
if (wci->mode == from_event)
return 0;
while (*wci->ptr == ' ' || *wci->ptr == '\t') wci->ptr++;
if (*wci->ptr == 0) wci->ptr = "cmd";
return 0;
}
/******************************************************************
* WinMain
*
* wineconsole can either be started as:
* wineconsole --use-event=<int> used when a new console is created (AllocConsole)
* wineconsole <pgm> <arguments> used to start the program <pgm> from the command line in
* a freshly created console
*/
int PASCAL WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR lpCmdLine, INT nCmdShow)
{
struct inner_data* data;
int ret = 1;
struct wc_init wci;
/* make sure we're not connected to inherited console */
FreeConsole();
if ((ret = WINECON_ParseOptions(lpCmdLine, &wci)) != 0)
{
printf_res(ret & 0xffff);
if (ret & WINECON_CMD_SHOW_USAGE)
WINECON_Usage();
return 0;
}
switch (wci.mode)
{
case from_event:
/* case of wineconsole <evt>, signal process that created us that we're up and running */
if (!(data = WINECON_Init(hInst, 0, NULL, wci.backend, nCmdShow, NULL))) return 1;
ret = !SetEvent(wci.event);
if (ret != 0) WINE_ERR("SetEvent failed.\n");
break;
case from_process_name:
{
int len;
WCHAR *buffer;
len = MultiByteToWideChar(CP_ACP, 0, wci.ptr, -1, NULL, 0);
buffer = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
if (!buffer)
return 1;
MultiByteToWideChar(CP_ACP, 0, wci.ptr, -1, buffer, len);
data = WINECON_Init(hInst, GetCurrentProcessId(), buffer, wci.backend, nCmdShow, buffer);
HeapFree(GetProcessHeap(), 0, buffer);
if (!data) return 1;
}
break;
default:
return 1;
}
if (!ret)
{
DWORD exitcode;
WINE_TRACE("calling MainLoop.\n");
ret = data->fnMainLoop(data);
if (!ret && data->hProcess &&
WaitForSingleObject(data->hProcess, INFINITE) == WAIT_OBJECT_0 &&
GetExitCodeProcess(data->hProcess, &exitcode))
{
WINE_TRACE("forwarding exitcode %u from child process\n", exitcode);
ret = exitcode;
}
}
WINECON_Delete(data);
return ret;
}