/* * 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 #include #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, sizeof(buffer)/sizeof(buffer[0])); 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_BACKEND); 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) { SERVER_START_REQ( read_console_output ) { req->handle = wine_server_obj_handle( data->hConOut ); req->x = 0; req->y = upd_tp; req->mode = CHAR_INFO_MODE_TEXTATTR; req->wrap = TRUE; wine_server_set_reply( req, &data->cells[upd_tp * data->curcfg.sb_width], (upd_bm-upd_tp+1) * data->curcfg.sb_width * sizeof(CHAR_INFO) ); wine_server_call( req ); } SERVER_END_REQ; 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) 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) { BOOL ret; SERVER_START_REQ(set_console_input_info) { req->handle = wine_server_obj_handle( hConIn ); req->mask = SET_CONSOLE_INPUT_INFO_HISTORY_SIZE; req->history_size = size; ret = !wine_server_call_err( req ); } SERVER_END_REQ; return ret; } /****************************************************************** * WINECON_SetHistoryMode * * */ static BOOL WINECON_SetHistoryMode(HANDLE hConIn, int mode) { BOOL ret; SERVER_START_REQ(set_console_input_info) { req->handle = wine_server_obj_handle( hConIn ); req->mask = SET_CONSOLE_INPUT_INFO_HISTORY_MODE; req->history_mode = mode; ret = !wine_server_call_err( req ); } SERVER_END_REQ; return ret; } /****************************************************************** * 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) { BOOL ret; if (len < sizeof(WCHAR)) return FALSE; SERVER_START_REQ( get_console_input_info ) { req->handle = wine_server_obj_handle( hConIn ); wine_server_set_reply( req, buffer, len - sizeof(WCHAR) ); if ((ret = !wine_server_call_err( req ))) { len = wine_server_reply_size( reply ); buffer[len / sizeof(WCHAR)] = 0; } } SERVER_END_REQ; return ret; } /****************************************************************** * WINECON_SetEditionMode * * */ static BOOL WINECON_SetEditionMode(HANDLE hConIn, int edition_mode) { BOOL ret; SERVER_START_REQ( set_console_input_info ) { req->handle = wine_server_obj_handle( hConIn ); req->mask = SET_CONSOLE_INPUT_INFO_EDITION_MODE; req->edition_mode = edition_mode; ret = !wine_server_call_err( req ); } SERVER_END_REQ; return ret; } /****************************************************************** * WINECON_GrabChanges * * A change occurs, try to figure out which */ void WINECON_GrabChanges(struct inner_data* data) { struct console_renderer_event evts[256]; int i, num, ev_found; HANDLE h; if (data->in_grab_changes) return; SERVER_START_REQ( get_console_renderer_events ) { wine_server_set_reply( req, evts, sizeof(evts) ); req->handle = wine_server_obj_handle( data->hSynchro ); if (!wine_server_call_err( req )) num = wine_server_reply_size(reply) / sizeof(evts[0]); else num = 0; } SERVER_END_REQ; if (!num) {WINE_WARN("hmm renderer signaled but no events available\n"); return;} 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_ACTIVE_SB_EVENT: SERVER_START_REQ( open_console ) { req->from = wine_server_obj_handle( data->hConIn ); req->access = GENERIC_READ | GENERIC_WRITE; req->attributes = 0; req->share = FILE_SHARE_READ | FILE_SHARE_WRITE; h = wine_server_call_err( req ) ? 0 : wine_server_ptr_handle(reply->handle); } SERVER_END_REQ; WINE_TRACE("%u/%u: active(%p)\n", i+1, num, h); if (h) { CloseHandle(data->hConOut); data->hConOut = h; } 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\n"); 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; } /****************************************************************** * 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; /* : 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->hConOut, &cinfo); /* */ cinfo.bVisible = cfg->cursor_visible; /* this shall update (through notif) curcfg */ SetConsoleCursorInfo(data->hConOut, &cinfo); } if (data->curcfg.history_size != cfg->history_size) { data->curcfg.history_size = cfg->history_size; WINECON_SetHistorySize(data->hConIn, cfg->history_size); } if (data->curcfg.history_nodup != cfg->history_nodup) { data->curcfg.history_nodup = cfg->history_nodup; WINECON_SetHistoryMode(data->hConIn, cfg->history_nodup); } if (data->curcfg.insert_mode != cfg->insert_mode) { data->curcfg.insert_mode = cfg->insert_mode; WINECON_SetInsertMode(data->hConIn, 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_weight != cfg->font_weight) { data->fnSetFont(data, cfg->face_name, cfg->cell_height, cfg->font_weight); SERVER_START_REQ(set_console_output_info) { req->handle = wine_server_obj_handle( data->hConOut ); req->mask = SET_CONSOLE_OUTPUT_INFO_FONT; req->font_width = cfg->cell_width; req->font_height = cfg->cell_height; wine_server_call( req ); } SERVER_END_REQ; } if (data->curcfg.def_attr != cfg->def_attr) { data->curcfg.def_attr = cfg->def_attr; SetConsoleTextAttribute(data->hConOut, cfg->def_attr); } /* 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 macros test if the (width/height) needs to be changed * for (window / ScreenBuffer) * The Change actually modify the dimension of . */ #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->hConOut, 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->hConOut, 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->hConOut, 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->hConOut, 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->hConIn, 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->hConIn) CloseHandle(data->hConIn); if (data->hConOut) CloseHandle(data->hConOut); if (data->hSynchro) CloseHandle(data->hSynchro); if (data->hProcess) CloseHandle(data->hProcess); 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) { BOOL ret; DWORD mode; SERVER_START_REQ(get_console_input_info) { req->handle = wine_server_obj_handle( data->hConIn ); ret = !wine_server_call_err( req ); data->curcfg.history_size = reply->history_size; data->curcfg.history_nodup = reply->history_mode; data->curcfg.edition_mode = reply->edition_mode; } SERVER_END_REQ; if (!ret) return FALSE; GetConsoleMode(data->hConIn, &mode); data->curcfg.insert_mode = (mode & (ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS)) == (ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS); SERVER_START_REQ(get_console_output_info) { req->handle = wine_server_obj_handle( data->hConOut ); ret = !wine_server_call_err( req ); data->curcfg.cursor_size = reply->cursor_size; data->curcfg.cursor_visible = reply->cursor_visible; data->curcfg.def_attr = reply->attr; data->curcfg.sb_width = reply->width; data->curcfg.sb_height = reply->height; data->curcfg.win_width = reply->win_right - reply->win_left + 1; data->curcfg.win_height = reply->win_bottom - reply->win_top + 1; } SERVER_END_REQ; WINECON_DumpConfig("first cfg: ", &data->curcfg); return ret; } /****************************************************************** * 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) { struct inner_data* data = NULL; DWORD ret; struct config_data cfg; STARTUPINFOW si; data = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*data)); if (!data) return 0; GetStartupInfoW(&si); if (pid == 0) { if (!si.lpTitle) WINECON_Fatal("Should have a title set"); 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 */ } /* 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; req->input_fd = -1; ret = !wine_server_call_err( req ); data->hConIn = wine_server_ptr_handle( reply->handle_in ); data->hSynchro = wine_server_ptr_handle( reply->event ); } SERVER_END_REQ; if (!ret) goto error; WINE_TRACE("using hConIn %p, hSynchro event %p\n", data->hConIn, data->hSynchro); SERVER_START_REQ(create_console_output) { req->handle_in = wine_server_obj_handle( data->hConIn ); req->access = GENERIC_WRITE|GENERIC_READ; req->attributes = 0; req->share = FILE_SHARE_READ|FILE_SHARE_WRITE; req->fd = -1; ret = !wine_server_call_err( req ); data->hConOut = wine_server_ptr_handle( reply->handle_out ); } SERVER_END_REQ; if (!ret) goto error; WINE_TRACE("using hConOut %p\n", data->hConOut); /* filling data->curcfg from cfg */ switch ((*backend)(data)) { case init_not_supported: if (backend == WCCURSES_InitBackend) { if (WCUSER_InitBackend( data ) != init_success) break; } else if (backend == WCUSER_InitBackend) { if (WCCURSES_InitBackend( data ) != init_success) break; } /* fall through */ case init_success: WINECON_GetServerConfig(data); data->cells = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, data->curcfg.sb_width * data->curcfg.sb_height * sizeof(CHAR_INFO)); if (!data->cells) WINECON_Fatal("OOM\n"); data->fnResizeScreenBuffer(data); data->fnComputePositions(data); WINECON_SetConfig(data, &cfg); data->curcfg.registry = cfg.registry; WINECON_DumpConfig("fint", &data->curcfg); SERVER_START_REQ( set_console_input_info ) { req->handle = wine_server_obj_handle( data->hConIn ); req->win = wine_server_user_handle( data->hWnd ); req->mask = SET_CONSOLE_INPUT_INFO_TITLE | SET_CONSOLE_INPUT_INFO_WIN; wine_server_add_data( req, appname, lstrlenW(appname) * sizeof(WCHAR) ); ret = !wine_server_call_err( req ); } SERVER_END_REQ; if (!ret) goto error; return data; case init_failed: break; } error: WINE_ERR("failed to init.\n"); WINECON_Delete(data); return NULL; } /****************************************************************** * WINECON_Spawn * * Spawn the child process when invoked with wineconsole foo bar */ static int WINECON_Spawn(struct inner_data* data, LPWSTR cmdLine) { PROCESS_INFORMATION info; STARTUPINFOW startup; BOOL done; /* we're in the case wineconsole ... 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(), data->hConIn, GetCurrentProcess(), &startup.hStdInput, GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE, TRUE, 0) || !DuplicateHandle(GetCurrentProcess(), data->hConOut, GetCurrentProcess(), &startup.hStdOutput, GENERIC_READ|GENERIC_WRITE, TRUE, 0) || !DuplicateHandle(GetCurrentProcess(), data->hConOut, 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 1; } done = CreateProcessW(NULL, cmdLine, NULL, NULL, TRUE, 0L, NULL, NULL, &startup, &info); if (done) { data->hProcess = info.hProcess; CloseHandle(info.hThread); } /* we no longer need the handles passed to the child for the console */ CloseHandle(startup.hStdInput); CloseHandle(startup.hStdOutput); CloseHandle(startup.hStdError); return !done; } 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, "--backend=", 10) == 0) { if (strncmp(wci->ptr + 10, "user", 4) == 0) { wci->backend = WCUSER_InitBackend; wci->ptr += 14; } else if (strncmp(wci->ptr + 10, "curses", 6) == 0) { wci->backend = WCCURSES_InitBackend; wci->ptr += 16; } else return IDS_CMD_INVALID_BACKEND; } 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= used when a new console is created (AllocConsole) * wineconsole used to start the program from the command line in * a freshly created console * --backend=(curses|user) can also be used to select the desired backend */ int PASCAL WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR lpCmdLine, INT nCmdShow) { struct inner_data* data; int ret = 1; struct wc_init wci; 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 , signal process that created us that we're up and running */ if (!(data = WINECON_Init(hInst, 0, NULL, wci.backend, nCmdShow))) 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); if (!(data = WINECON_Init(hInst, GetCurrentProcessId(), buffer, wci.backend, nCmdShow))) { HeapFree(GetProcessHeap(), 0, buffer); return 1; } ret = WINECON_Spawn(data, buffer); HeapFree(GetProcessHeap(), 0, buffer); if (ret != 0) { WINECON_Delete(data); printf_res(IDS_CMD_LAUNCH_FAILED, wine_dbgstr_a(wci.ptr)); return ret; } } 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; }