programs/wineconsole/wineconsole.c

   1 /*
   2  * an application for displaying Win32 console
   3  *
   4  * Copyright 2001, 2002 Eric Pouech
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <stdio.h>
  25 #include <stdarg.h>
  26 #include "wine/server.h"
  27 #include "winecon_private.h"
  28 #include "winnls.h"
  29 #include "winuser.h"
  30 #include "wine/unicode.h"
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(wineconsole);
  34
  35 void WINECON_Fatal(const char* msg)
  36 {
  37     WINE_ERR("%s\n", msg);
  38     ExitProcess(0);
  39 }
  40
  41 static void printf_res(UINT uResId, ...)
  42 {
  43     WCHAR buffer[1024];
  44     CHAR ansi[1024];
  45     va_list args;
  46
  47     va_start(args, uResId);
  48     LoadStringW(GetModuleHandleW(NULL), uResId, buffer, sizeof(buffer)/sizeof(buffer[0]));
  49     WideCharToMultiByte(CP_UNIXCP, 0, buffer, -1, ansi, sizeof(ansi), NULL, NULL);
  50     vprintf(ansi, args);
  51     va_end(args);
  52 }
  53
  54 static void WINECON_Usage(void)
  55 {
  56     printf_res(IDS_USAGE_HEADER);
  57     printf_res(IDS_USAGE_BACKEND);
  58     printf_res(IDS_USAGE_COMMAND);
  59     printf_res(IDS_USAGE_FOOTER);
  60 }
  61
  62 /******************************************************************
  63  *              WINECON_FetchCells
  64  *
  65  * updates the local copy of cells (band to update)
  66  */
  67 static void WINECON_FetchCells(struct inner_data* data, int upd_tp, int upd_bm)
  68 {
  69     SERVER_START_REQ( read_console_output )
  70     {
  71         req->handle = wine_server_obj_handle( data->hConOut );
  72         req->x      = 0;
  73         req->y      = upd_tp;
  74         req->mode   = CHAR_INFO_MODE_TEXTATTR;
  75         req->wrap   = TRUE;
  76         wine_server_set_reply( req, &data->cells[upd_tp * data->curcfg.sb_width],
  77                                (upd_bm-upd_tp+1) * data->curcfg.sb_width * sizeof(CHAR_INFO) );
  78         wine_server_call( req );
  79     }
  80     SERVER_END_REQ;
  81     data->fnRefresh(data, upd_tp, upd_bm);
  82 }
  83
  84 /******************************************************************
  85  *              WINECON_ResizeWithContainer
  86  *
  87  * For console embedded in a container (e.g. user in a win32 window, or (n)curses
  88  * in a TERM, perform resize of console (screen buffer and window) to fit in
  89  * (new) container size.
  90  */
  91 void WINECON_ResizeWithContainer(struct inner_data* data, int width, int height)
  92 {
  93     struct config_data  cfg;
  94
  95     if (data->in_set_config) return;
  96
  97     cfg = data->curcfg;
  98     cfg.win_width  = width;
  99     cfg.win_height = height;
 100
 101     /* auto size screen-buffer if it's now smaller than window */
 102     if (cfg.sb_width  < cfg.win_width)  cfg.sb_width = cfg.win_width;
 103     if (cfg.sb_height < cfg.win_height) cfg.sb_height = cfg.win_height;
 104
 105     /* and reset window pos so that we don't display outside of the screen-buffer */
 106     if (cfg.win_pos.X + cfg.win_width  > cfg.sb_width)  cfg.win_pos.X = cfg.sb_width  - cfg.win_width;
 107     if (cfg.win_pos.Y + cfg.win_height > cfg.sb_height) cfg.win_pos.Y = cfg.sb_height - cfg.win_height;
 108
 109     WINECON_SetConfig(data, &cfg);
 110 }
 111
 112 /******************************************************************
 113  *              WINECON_SetHistorySize
 114  *
 115  *
 116  */
 117 static BOOL WINECON_SetHistorySize(HANDLE hConIn, int size)
 118 {
 119     BOOL        ret;
 120
 121     SERVER_START_REQ(set_console_input_info)
 122     {
 123         req->handle = wine_server_obj_handle( hConIn );
 124         req->mask = SET_CONSOLE_INPUT_INFO_HISTORY_SIZE;
 125         req->history_size = size;
 126         ret = !wine_server_call_err( req );
 127     }
 128     SERVER_END_REQ;
 129     return ret;
 130 }
 131
 132 /******************************************************************
 133  *              WINECON_SetHistoryMode
 134  *
 135  *
 136  */
 137 static BOOL WINECON_SetHistoryMode(HANDLE hConIn, int mode)
 138 {
 139     BOOL        ret;
 140
 141     SERVER_START_REQ(set_console_input_info)
 142     {
 143         req->handle = wine_server_obj_handle( hConIn );
 144         req->mask = SET_CONSOLE_INPUT_INFO_HISTORY_MODE;
 145         req->history_mode = mode;
 146         ret = !wine_server_call_err( req );
 147     }
 148     SERVER_END_REQ;
 149     return ret;
 150 }
 151
 152 /******************************************************************
 153  *              WINECON_SetInsertMode
 154  *
 155  *
 156  */
 157 static void WINECON_SetInsertMode(HANDLE hConIn, unsigned int enable)
 158 {
 159     DWORD mode;
 160
 161     GetConsoleMode(hConIn, &mode);
 162     if (enable)
 163         mode |= ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS;
 164     else
 165         mode &= ~ENABLE_INSERT_MODE;
 166     SetConsoleMode(hConIn, mode);
 167 }
 168
 169 /******************************************************************
 170  *              WINECON_GetConsoleTitle
 171  *
 172  *
 173  */
 174 BOOL WINECON_GetConsoleTitle(HANDLE hConIn, WCHAR* buffer, size_t len)
 175 {
 176     BOOL ret;
 177
 178     if (len < sizeof(WCHAR)) return FALSE;
 179
 180     SERVER_START_REQ( get_console_input_info )
 181     {
 182         req->handle = wine_server_obj_handle( hConIn );
 183         wine_server_set_reply( req, buffer, len - sizeof(WCHAR) );
 184         if ((ret = !wine_server_call_err( req )))
 185         {
 186             len = wine_server_reply_size( reply );
 187             buffer[len / sizeof(WCHAR)] = 0;
 188         }
 189     }
 190     SERVER_END_REQ;
 191     return ret;
 192 }
 193
 194 /******************************************************************
 195  *              WINECON_SetEditionMode
 196  *
 197  *
 198  */
 199 static BOOL WINECON_SetEditionMode(HANDLE hConIn, int edition_mode)
 200 {
 201     BOOL ret;
 202
 203     SERVER_START_REQ( set_console_input_info )
 204     {
 205         req->handle = wine_server_obj_handle( hConIn );
 206         req->mask = SET_CONSOLE_INPUT_INFO_EDITION_MODE;
 207         req->edition_mode = edition_mode;
 208         ret = !wine_server_call_err( req );
 209     }
 210     SERVER_END_REQ;
 211     return ret;
 212 }
 213
 214 /******************************************************************
 215  *              WINECON_SetColors
 216  *
 217  * Sets ColorTable and Pop-up menu colors
 218  */
 219 static void WINECON_SetColors(struct inner_data *data, const struct config_data* cfg)
 220 {
 221     size_t color_map_size = sizeof(data->curcfg.color_map);
 222
 223     memcpy(data->curcfg.color_map, cfg->color_map, color_map_size);
 224     data->curcfg.popup_attr = cfg->popup_attr;
 225
 226     SERVER_START_REQ( set_console_output_info )
 227     {
 228         req->handle = wine_server_obj_handle( data->hConOut );
 229         req->mask = SET_CONSOLE_OUTPUT_INFO_COLORTABLE | SET_CONSOLE_OUTPUT_INFO_POPUP_ATTR;
 230         req->popup_attr = cfg->popup_attr;
 231         wine_server_add_data( req, cfg->color_map, color_map_size );
 232         wine_server_call( req );
 233     }
 234     SERVER_END_REQ;
 235 }
 236
 237 /******************************************************************
 238  *              WINECON_GrabChanges
 239  *
 240  * A change occurs, try to figure out which
 241  */
 242 void    WINECON_GrabChanges(struct inner_data* data)
 243 {
 244     struct console_renderer_event       evts[256];
 245     int i, num, ev_found;
 246     HANDLE h;
 247
 248     if (data->in_grab_changes) return;
 249
 250     SERVER_START_REQ( get_console_renderer_events )
 251     {
 252         wine_server_set_reply( req, evts, sizeof(evts) );
 253         req->handle = wine_server_obj_handle( data->hSynchro );
 254         if (!wine_server_call_err( req )) num = wine_server_reply_size(reply) / sizeof(evts[0]);
 255         else num = 0;
 256     }
 257     SERVER_END_REQ;
 258     if (!num) {WINE_WARN("hmm renderer signaled but no events available\n"); return;}
 259     WINE_TRACE( "got %u events\n", num );
 260
 261     /* FIXME: should do some event compression here (cursor pos, update) */
 262     /* step 1: keep only last cursor pos event */
 263     ev_found = -1;
 264     for (i = num - 1; i >= 0; i--)
 265     {
 266         if (evts[i].event == CONSOLE_RENDERER_CURSOR_POS_EVENT)
 267         {
 268             if (ev_found != -1)
 269             {
 270                 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);
 271                 evts[i].event = CONSOLE_RENDERER_NONE_EVENT;
 272             }
 273             ev_found = i;
 274         }
 275     }
 276     /* step 2: manage update events */
 277     ev_found = -1;
 278     for (i = 0; i < num; i++)
 279     {
 280         if (evts[i].event == CONSOLE_RENDERER_NONE_EVENT ||
 281             evts[i].event == CONSOLE_RENDERER_CURSOR_POS_EVENT ||
 282             evts[i].event == CONSOLE_RENDERER_CURSOR_GEOM_EVENT) continue;
 283         if (evts[i].event != CONSOLE_RENDERER_UPDATE_EVENT)
 284         {
 285             ev_found = -1;
 286             continue;
 287         }
 288
 289         if (ev_found != -1 &&  /* Only 2 cases where they CANNOT merge */
 290             !(evts[i       ].u.update.bottom + 1 < evts[ev_found].u.update.top ||
 291               evts[ev_found].u.update.bottom + 1 < evts[i       ].u.update.top))
 292         {
 293             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);
 294             evts[i].u.update.top    = min(evts[i       ].u.update.top,
 295                                           evts[ev_found].u.update.top);
 296             evts[i].u.update.bottom = max(evts[i       ].u.update.bottom,
 297                                           evts[ev_found].u.update.bottom);
 298             evts[ev_found].event = CONSOLE_RENDERER_NONE_EVENT;
 299         }
 300         ev_found = i;
 301     }
 302
 303     data->in_grab_changes = TRUE;
 304     for (i = 0; i < num; i++)
 305     {
 306         switch (evts[i].event)
 307         {
 308         case CONSOLE_RENDERER_NONE_EVENT:
 309             WINE_TRACE("%u/%u: NOP\n", i+1, num);
 310             break;
 311         case CONSOLE_RENDERER_TITLE_EVENT:
 312             WINE_TRACE("%u/%u: title()\n", i+1, num);
 313             data->fnSetTitle(data);
 314             break;
 315         case CONSOLE_RENDERER_ACTIVE_SB_EVENT:
 316             SERVER_START_REQ( open_console )
 317             {
 318                 req->from       = wine_server_obj_handle( data->hConIn );
 319                 req->access     = GENERIC_READ | GENERIC_WRITE;
 320                 req->attributes = 0;
 321                 req->share      = FILE_SHARE_READ | FILE_SHARE_WRITE;
 322                 h = wine_server_call_err( req ) ? 0 : wine_server_ptr_handle(reply->handle);
 323             }
 324             SERVER_END_REQ;
 325             WINE_TRACE("%u/%u: active(%p)\n", i+1, num, h);
 326             if (h)
 327             {
 328                 CloseHandle(data->hConOut);
 329                 data->hConOut = h;
 330             }
 331             break;
 332         case CONSOLE_RENDERER_SB_RESIZE_EVENT:
 333             if (data->curcfg.sb_width != evts[i].u.resize.width ||
 334                 data->curcfg.sb_height != evts[i].u.resize.height)
 335             {
 336                 WINE_TRACE("%u/%u: resize(%d,%d)\n", i+1, num, evts[i].u.resize.width, evts[i].u.resize.height);
 337                 data->curcfg.sb_width  = evts[i].u.resize.width;
 338                 data->curcfg.sb_height = evts[i].u.resize.height;
 339
 340                 data->cells = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, data->cells,
 341                                           data->curcfg.sb_width * data->curcfg.sb_height * sizeof(CHAR_INFO));
 342
 343                 if (!data->cells) WINECON_Fatal("OOM");
 344                 data->fnResizeScreenBuffer(data);
 345                 data->fnComputePositions(data);
 346             }
 347             break;
 348         case CONSOLE_RENDERER_UPDATE_EVENT:
 349             WINE_TRACE("%u/%u: update(%d,%d)\n", i+1, num, evts[i].u.update.top, evts[i].u.update.bottom);
 350             WINECON_FetchCells(data, evts[i].u.update.top, evts[i].u.update.bottom);
 351             break;
 352         case CONSOLE_RENDERER_CURSOR_POS_EVENT:
 353             if (evts[i].u.cursor_pos.x != data->cursor.X || evts[i].u.cursor_pos.y != data->cursor.Y)
 354             {
 355                 WINE_TRACE("%u/%u: curs-pos(%d,%d)\n", i+1, num, evts[i].u.cursor_pos.x, evts[i].u.cursor_pos.y);
 356                 data->cursor.X = evts[i].u.cursor_pos.x;
 357                 data->cursor.Y = evts[i].u.cursor_pos.y;
 358                 data->fnPosCursor(data);
 359             }
 360             break;
 361         case CONSOLE_RENDERER_CURSOR_GEOM_EVENT:
 362             if (evts[i].u.cursor_geom.size != data->curcfg.cursor_size ||
 363                 evts[i].u.cursor_geom.visible != data->curcfg.cursor_visible)
 364             {
 365                 WINE_TRACE("%u/%u: curs-geom(%d,%d)\n", i+1, num,
 366                            evts[i].u.cursor_geom.size, evts[i].u.cursor_geom.visible);
 367                 data->fnShapeCursor(data, evts[i].u.cursor_geom.size,
 368                                     evts[i].u.cursor_geom.visible, FALSE);
 369             }
 370             break;
 371         case CONSOLE_RENDERER_DISPLAY_EVENT:
 372             if (evts[i].u.display.left != data->curcfg.win_pos.X)
 373             {
 374                 WINE_TRACE("%u/%u: h-scroll(%d)\n", i+1, num, evts[i].u.display.left);
 375                 data->fnScroll(data, evts[i].u.display.left, TRUE);
 376                 data->fnPosCursor(data);
 377             }
 378             if (evts[i].u.display.top != data->curcfg.win_pos.Y)
 379             {
 380                 WINE_TRACE("%u/%u: v-scroll(%d)\n", i+1, num, evts[i].u.display.top);
 381                 data->fnScroll(data, evts[i].u.display.top, FALSE);
 382                 data->fnPosCursor(data);
 383             }
 384             if (evts[i].u.display.width != data->curcfg.win_width ||
 385                 evts[i].u.display.height != data->curcfg.win_height)
 386             {
 387                 WINE_TRACE("%u/%u: win-size(%d,%d)\n", i+1, num, evts[i].u.display.width, evts[i].u.display.height);
 388                 data->curcfg.win_width = evts[i].u.display.width;
 389                 data->curcfg.win_height = evts[i].u.display.height;
 390                 data->fnComputePositions(data);
 391             }
 392             break;
 393         case CONSOLE_RENDERER_EXIT_EVENT:
 394             data->dying = TRUE;
 395             WINE_TRACE("%u/%u: Exit!!\n", i+1, num);
 396             return;
 397         default:
 398             WINE_FIXME("Unknown event type (%d)\n", evts[i].event);
 399         }
 400     }
 401     data->in_grab_changes = FALSE;
 402 }
 403
 404 /******************************************************************
 405  *              WINECON_SetConfig
 406  *
 407  * Apply to data all the configuration elements from cfg. This includes modification
 408  * of server side equivalent and visual parts.
 409  * If force is FALSE, only the changed items are modified.
 410  */
 411 void     WINECON_SetConfig(struct inner_data* data, const struct config_data* cfg)
 412 {
 413     if (data->in_set_config) return;
 414     data->in_set_config = TRUE;
 415     if (data->curcfg.cursor_size != cfg->cursor_size ||
 416         data->curcfg.cursor_visible != cfg->cursor_visible)
 417     {
 418         CONSOLE_CURSOR_INFO cinfo;
 419         cinfo.dwSize = cfg->cursor_size;
 420         /* <FIXME>: this hack is needed to pass through the invariant test operation on the server side
 421          * (no notification is sent when invariant operation is requested)
 422          */
 423         cinfo.bVisible = !cfg->cursor_visible;
 424         SetConsoleCursorInfo(data->hConOut, &cinfo);
 425         /* </FIXME> */
 426         cinfo.bVisible = cfg->cursor_visible;
 427         /* this shall update (through notif) curcfg */
 428         SetConsoleCursorInfo(data->hConOut, &cinfo);
 429     }
 430     if (data->curcfg.history_size != cfg->history_size)
 431     {
 432         data->curcfg.history_size = cfg->history_size;
 433         WINECON_SetHistorySize(data->hConIn, cfg->history_size);
 434     }
 435     if (data->curcfg.history_nodup != cfg->history_nodup)
 436     {
 437         data->curcfg.history_nodup = cfg->history_nodup;
 438         WINECON_SetHistoryMode(data->hConIn, cfg->history_nodup);
 439     }
 440     if (data->curcfg.insert_mode != cfg->insert_mode)
 441     {
 442         data->curcfg.insert_mode = cfg->insert_mode;
 443         WINECON_SetInsertMode(data->hConIn, cfg->insert_mode);
 444     }
 445     data->curcfg.menu_mask = cfg->menu_mask;
 446     data->curcfg.quick_edit = cfg->quick_edit;
 447     if (strcmpiW(data->curcfg.face_name, cfg->face_name) || data->curcfg.cell_width != cfg->cell_width ||
 448         data->curcfg.cell_height != cfg->cell_height || data->curcfg.font_weight != cfg->font_weight)
 449     {
 450         RECT r;
 451         data->fnSetFont(data, cfg->face_name, cfg->cell_height, cfg->font_weight);
 452         SystemParametersInfoW(SPI_GETWORKAREA, 0, &r, 0);
 453         SERVER_START_REQ(set_console_output_info)
 454         {
 455             req->handle = wine_server_obj_handle( data->hConOut );
 456             req->mask = SET_CONSOLE_OUTPUT_INFO_MAX_SIZE | SET_CONSOLE_OUTPUT_INFO_FONT;
 457             req->max_width  = (r.right - r.left) / cfg->cell_width;
 458             req->max_height = (r.bottom - r.top - GetSystemMetrics(SM_CYCAPTION)) / cfg->cell_height;
 459             req->font_width = cfg->cell_width;
 460             req->font_height = cfg->cell_height;
 461             wine_server_call( req );
 462         }
 463         SERVER_END_REQ;
 464     }
 465     if (data->curcfg.def_attr != cfg->def_attr)
 466     {
 467         DWORD screen_size, written;
 468         COORD top_left = {0,0};
 469
 470         data->curcfg.def_attr = cfg->def_attr;
 471         screen_size = cfg->win_width * (cfg->win_height + 1);
 472         FillConsoleOutputAttribute(data->hConOut, cfg->def_attr, screen_size, top_left, &written);
 473         SetConsoleTextAttribute(data->hConOut, cfg->def_attr);
 474     }
 475     WINECON_SetColors(data, cfg);
 476     /* now let's look at the window / sb size changes...
 477      * since the server checks that sb is always bigger than window,
 478      * we have to take care of doing the operations in the right order
 479      */
 480     /* a set of macros to make things easier to read
 481      * The Test<A><B> macros test if the <A> (width/height) needs to be changed
 482      * for <B> (window / ScreenBuffer)
 483      * The Change<A><B> actually modify the <B> dimension of <A>.
 484      */
 485 #define TstSBfWidth()   (data->curcfg.sb_width != cfg->sb_width)
 486 #define TstWinHPos()    (data->curcfg.win_width != cfg->win_width || data->curcfg.win_pos.X != cfg->win_pos.X)
 487
 488 #define ChgSBfWidth()   do {c.X = cfg->sb_width; \
 489                             c.Y = data->curcfg.sb_height;\
 490                             SetConsoleScreenBufferSize(data->hConOut, c);\
 491                         } while (0)
 492 #define ChgWinHPos()    do {pos.Left = cfg->win_pos.X - data->curcfg.win_pos.X; \
 493                             pos.Top = 0; \
 494                             pos.Right = pos.Left + cfg->win_width - data->curcfg.win_width; \
 495                             pos.Bottom = 0; \
 496                             SetConsoleWindowInfo(data->hConOut, FALSE, &pos);\
 497                         } while (0)
 498 #define TstSBfHeight()  (data->curcfg.sb_height != cfg->sb_height)
 499 #define TstWinVPos()    (data->curcfg.win_height != cfg->win_height || data->curcfg.win_pos.Y != cfg->win_pos.Y)
 500
 501 /* since we're going to apply height after width is done, we use width as defined
 502  * in cfg, and not in data->curcfg because if won't be updated yet */
 503 #define ChgSBfHeight()  do {c.X = cfg->sb_width; c.Y = cfg->sb_height; \
 504                             SetConsoleScreenBufferSize(data->hConOut, c); \
 505                         } while (0)
 506 #define ChgWinVPos()    do {pos.Left = 0; \
 507                             pos.Top = cfg->win_pos.Y - data->curcfg.win_pos.Y; \
 508                             pos.Right = 0; \
 509                             pos.Bottom = pos.Top + cfg->win_height - data->curcfg.win_height; \
 510                             SetConsoleWindowInfo(data->hConOut, FALSE, &pos);\
 511                         } while (0)
 512
 513     do
 514     {
 515         COORD       c;
 516         SMALL_RECT  pos;
 517
 518         if (TstSBfWidth())
 519         {
 520             if (TstWinHPos())
 521             {
 522                 /* we're changing both at the same time, do it in the right order */
 523                 if (cfg->sb_width >= data->curcfg.win_width)
 524                 {
 525                     ChgSBfWidth(); ChgWinHPos();
 526                 }
 527                 else
 528                 {
 529                     ChgWinHPos(); ChgSBfWidth();
 530                 }
 531             }
 532             else ChgSBfWidth();
 533         }
 534         else if (TstWinHPos()) ChgWinHPos();
 535         if (TstSBfHeight())
 536         {
 537             if (TstWinVPos())
 538             {
 539                 if (cfg->sb_height >= data->curcfg.win_height)
 540                 {
 541                     ChgSBfHeight(); ChgWinVPos();
 542                 }
 543                 else
 544                 {
 545                     ChgWinVPos(); ChgSBfHeight();
 546                 }
 547             }
 548             else ChgSBfHeight();
 549         }
 550         else if (TstWinVPos()) ChgWinVPos();
 551     } while (0);
 552 #undef TstSBfWidth
 553 #undef TstWinHPos
 554 #undef ChgSBfWidth
 555 #undef ChgWinHPos
 556 #undef TstSBfHeight
 557 #undef TstWinVPos
 558 #undef ChgSBfHeight
 559 #undef ChgWinVPos
 560
 561     data->curcfg.exit_on_die = cfg->exit_on_die;
 562     if (data->curcfg.edition_mode != cfg->edition_mode)
 563     {
 564         data->curcfg.edition_mode = cfg->edition_mode;
 565         WINECON_SetEditionMode(data->hConIn, cfg->edition_mode);
 566     }
 567     /* we now need to gather all events we got from the operations above,
 568      * in order to get data correctly updated
 569      */
 570     WINECON_GrabChanges(data);
 571     data->in_set_config = FALSE;
 572 }
 573
 574 /******************************************************************
 575  *              WINECON_Delete
 576  *
 577  * Destroy wineconsole internal data
 578  */
 579 static void WINECON_Delete(struct inner_data* data)
 580 {
 581     if (!data) return;
 582
 583     if (data->fnDeleteBackend)  data->fnDeleteBackend(data);
 584     if (data->hConIn)           CloseHandle(data->hConIn);
 585     if (data->hConOut)          CloseHandle(data->hConOut);
 586     if (data->hSynchro)         CloseHandle(data->hSynchro);
 587     if (data->hProcess)         CloseHandle(data->hProcess);
 588     HeapFree(GetProcessHeap(), 0, data->curcfg.registry);
 589     HeapFree(GetProcessHeap(), 0, data->cells);
 590     HeapFree(GetProcessHeap(), 0, data);
 591 }
 592
 593 /******************************************************************
 594  *              WINECON_GetServerConfig
 595  *
 596  * Fills data->curcfg with the actual configuration running in the server
 597  * (getting real information on the server, and not relying on cached
 598  * information in data)
 599  */
 600 static BOOL WINECON_GetServerConfig(struct inner_data* data)
 601 {
 602     BOOL  ret;
 603     DWORD mode;
 604
 605     SERVER_START_REQ(get_console_input_info)
 606     {
 607         req->handle = wine_server_obj_handle( data->hConIn );
 608         ret = !wine_server_call_err( req );
 609         data->curcfg.history_size = reply->history_size;
 610         data->curcfg.history_nodup = reply->history_mode;
 611         data->curcfg.edition_mode = reply->edition_mode;
 612     }
 613     SERVER_END_REQ;
 614     if (!ret) return FALSE;
 615
 616     GetConsoleMode(data->hConIn, &mode);
 617     data->curcfg.insert_mode = (mode & (ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS)) ==
 618                                        (ENABLE_INSERT_MODE|ENABLE_EXTENDED_FLAGS);
 619
 620     SERVER_START_REQ(get_console_output_info)
 621     {
 622         req->handle = wine_server_obj_handle( data->hConOut );
 623         ret = !wine_server_call_err( req );
 624         data->curcfg.cursor_size = reply->cursor_size;
 625         data->curcfg.cursor_visible = reply->cursor_visible;
 626         data->curcfg.def_attr = reply->attr;
 627         data->curcfg.sb_width = reply->width;
 628         data->curcfg.sb_height = reply->height;
 629         data->curcfg.win_width = reply->win_right - reply->win_left + 1;
 630         data->curcfg.win_height = reply->win_bottom - reply->win_top + 1;
 631     }
 632     SERVER_END_REQ;
 633     WINECON_DumpConfig("first cfg: ", &data->curcfg);
 634
 635     return ret;
 636 }
 637
 638 /******************************************************************
 639  *              WINECON_Init
 640  *
 641  * Initialisation part I. Creation of server object (console input and
 642  * active screen buffer)
 643  */
 644 static struct inner_data* WINECON_Init(HINSTANCE hInst, DWORD pid, LPCWSTR appname,
 645                                        enum init_return (*backend)(struct inner_data*),
 646                                        INT nCmdShow)
 647 {
 648     struct inner_data*  data = NULL;
 649     DWORD               ret;
 650     struct config_data  cfg;
 651     STARTUPINFOW        si;
 652
 653     data = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*data));
 654     if (!data) return 0;
 655
 656     GetStartupInfoW(&si);
 657
 658     if (pid == 0) appname = si.lpTitle;
 659
 660     data->nCmdShow = nCmdShow;
 661     /* load settings */
 662     WINECON_RegLoad(appname, &cfg);
 663
 664     /* some overrides */
 665     if (pid == 0)
 666     {
 667         if (si.dwFlags & STARTF_USECOUNTCHARS)
 668         {
 669             cfg.sb_width  = si.dwXCountChars;
 670             cfg.sb_height = si.dwYCountChars;
 671         }
 672         if (si.dwFlags & STARTF_USEFILLATTRIBUTE)
 673             cfg.def_attr = si.dwFillAttribute;
 674         /* should always be defined */
 675     }
 676
 677     /* the handles here are created without the whistles and bells required by console
 678      * (mainly because wineconsole doesn't need it)
 679      * - they are not inheritable
 680      * - hConIn is not synchronizable
 681      */
 682     SERVER_START_REQ(alloc_console)
 683     {
 684         req->access     = GENERIC_READ | GENERIC_WRITE;
 685         req->attributes = 0;
 686         req->pid        = pid;
 687         req->input_fd   = -1;
 688
 689         ret = !wine_server_call_err( req );
 690         data->hConIn = wine_server_ptr_handle( reply->handle_in );
 691         data->hSynchro = wine_server_ptr_handle( reply->event );
 692     }
 693     SERVER_END_REQ;
 694     if (!ret) goto error;
 695     WINE_TRACE("using hConIn %p, hSynchro event %p\n", data->hConIn, data->hSynchro);
 696
 697     SERVER_START_REQ(create_console_output)
 698     {
 699         req->handle_in  = wine_server_obj_handle( data->hConIn );
 700         req->access     = GENERIC_WRITE|GENERIC_READ;
 701         req->attributes = 0;
 702         req->share      = FILE_SHARE_READ|FILE_SHARE_WRITE;
 703         req->fd         = -1;
 704         ret = !wine_server_call_err( req );
 705         data->hConOut   = wine_server_ptr_handle( reply->handle_out );
 706     }
 707     SERVER_END_REQ;
 708     if (!ret) goto error;
 709     WINE_TRACE("using hConOut %p\n", data->hConOut);
 710
 711     /* filling data->curcfg from cfg */
 712     switch ((*backend)(data))
 713     {
 714     case init_not_supported:
 715         if (backend == WCCURSES_InitBackend)
 716         {
 717             if (WCUSER_InitBackend( data ) != init_success) break;
 718         }
 719         else if (backend == WCUSER_InitBackend)
 720         {
 721             if (WCCURSES_InitBackend( data ) != init_success) break;
 722         }
 723         /* fall through */
 724     case init_success:
 725         WINECON_GetServerConfig(data);
 726         data->cells = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
 727                                 data->curcfg.sb_width * data->curcfg.sb_height * sizeof(CHAR_INFO));
 728         if (!data->cells) goto error;
 729         data->fnResizeScreenBuffer(data);
 730         data->fnComputePositions(data);
 731         WINECON_SetConfig(data, &cfg);
 732         data->curcfg.registry = cfg.registry;
 733         WINECON_DumpConfig("fint", &data->curcfg);
 734         SERVER_START_REQ( set_console_input_info )
 735         {
 736             req->handle = wine_server_obj_handle( data->hConIn );
 737             req->win = wine_server_user_handle( data->hWnd );
 738             req->mask = SET_CONSOLE_INPUT_INFO_TITLE |
 739                         SET_CONSOLE_INPUT_INFO_WIN;
 740             wine_server_add_data( req, appname, lstrlenW(appname) * sizeof(WCHAR) );
 741             ret = !wine_server_call_err( req );
 742         }
 743         SERVER_END_REQ;
 744         if (!ret) goto error;
 745
 746         return data;
 747     case init_failed:
 748         break;
 749     }
 750
 751  error:
 752     WINE_ERR("failed to init.\n");
 753
 754     WINECON_Delete(data);
 755     return NULL;
 756 }
 757
 758 /******************************************************************
 759  *              WINECON_Spawn
 760  *
 761  * Spawn the child process when invoked with wineconsole foo bar
 762  */
 763 static int WINECON_Spawn(struct inner_data* data, LPWSTR cmdLine)
 764 {
 765     PROCESS_INFORMATION info;
 766     STARTUPINFOW        startup;
 767     BOOL                done;
 768
 769     /* we're in the case wineconsole <exe> <options>... spawn the new process */
 770     memset(&startup, 0, sizeof(startup));
 771     startup.cb          = sizeof(startup);
 772     startup.dwFlags     = STARTF_USESTDHANDLES;
 773
 774     /* the attributes of wineconsole's handles are not adequate for inheritance, so
 775      * get them with the correct attributes before process creation
 776      */
 777     if (!DuplicateHandle(GetCurrentProcess(), data->hConIn,  GetCurrentProcess(),
 778                          &startup.hStdInput, GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE, TRUE, 0) ||
 779         !DuplicateHandle(GetCurrentProcess(), data->hConOut, GetCurrentProcess(),
 780                          &startup.hStdOutput, GENERIC_READ|GENERIC_WRITE, TRUE, 0) ||
 781         !DuplicateHandle(GetCurrentProcess(), data->hConOut, GetCurrentProcess(),
 782                          &startup.hStdError, GENERIC_READ|GENERIC_WRITE, TRUE, 0))
 783     {
 784         WINE_ERR("Can't dup handles\n");
 785         /* no need to delete handles, we're exiting the program anyway */
 786         return 1;
 787     }
 788
 789     done = CreateProcessW(NULL, cmdLine, NULL, NULL, TRUE, 0L, NULL, NULL, &startup, &info);
 790     if (done)
 791     {
 792         data->hProcess = info.hProcess;
 793         CloseHandle(info.hThread);
 794     }
 795
 796     /* we no longer need the handles passed to the child for the console */
 797     CloseHandle(startup.hStdInput);
 798     CloseHandle(startup.hStdOutput);
 799     CloseHandle(startup.hStdError);
 800
 801     return !done;
 802 }
 803
 804 struct wc_init {
 805     LPCSTR              ptr;
 806     enum {from_event, from_process_name} mode;
 807     enum init_return    (*backend)(struct inner_data*);
 808     HANDLE              event;
 809 };
 810
 811 #define WINECON_CMD_SHOW_USAGE 0x10000
 812
 813 /******************************************************************
 814  *               WINECON_ParseOptions
 815  *
 816  * RETURNS
 817  *   On success: 0
 818  *   On error:   error string id optionally with the CMD_SHOW_USAGE flag
 819  */
 820 static UINT WINECON_ParseOptions(const char* lpCmdLine, struct wc_init* wci)
 821 {
 822     memset(wci, 0, sizeof(*wci));
 823     wci->ptr = lpCmdLine;
 824     wci->mode = from_process_name;
 825     wci->backend = WCUSER_InitBackend;
 826
 827     for (;;)
 828     {
 829         while (*wci->ptr == ' ' || *wci->ptr == '\t') wci->ptr++;
 830         if (wci->ptr[0] != '-') break;
 831         if (strncmp(wci->ptr, "--use-event=", 12) == 0)
 832         {
 833             char*           end;
 834             wci->event = ULongToHandle( strtoul(wci->ptr + 12, &end, 10) );
 835             if (end == wci->ptr + 12) return IDS_CMD_INVALID_EVENT_ID;
 836             wci->mode = from_event;
 837             wci->ptr = end;
 838         }
 839         else if (strncmp(wci->ptr, "--backend=", 10) == 0)
 840         {
 841             if (strncmp(wci->ptr + 10, "user", 4) == 0)
 842             {
 843                 wci->backend = WCUSER_InitBackend;
 844                 wci->ptr += 14;
 845             }
 846             else if (strncmp(wci->ptr + 10, "curses", 6) == 0)
 847             {
 848                 wci->backend = WCCURSES_InitBackend;
 849                 wci->ptr += 16;
 850             }
 851             else
 852                 return IDS_CMD_INVALID_BACKEND;
 853         }
 854         else if (!strncmp(wci->ptr, "--help", 6) &&
 855                  (!wci->ptr[6] || wci->ptr[6] == ' ' || wci->ptr[6] == '\t'))
 856             return IDS_CMD_ABOUT|WINECON_CMD_SHOW_USAGE;
 857         else
 858             return IDS_CMD_INVALID_OPTION|WINECON_CMD_SHOW_USAGE;
 859     }
 860
 861     if (wci->mode == from_event)
 862         return 0;
 863
 864     while (*wci->ptr == ' ' || *wci->ptr == '\t') wci->ptr++;
 865     if (*wci->ptr == 0) wci->ptr = "cmd";
 866
 867     return 0;
 868 }
 869
 870 /******************************************************************
 871  *              WinMain
 872  *
 873  * wineconsole can either be started as:
 874  *      wineconsole --use-event=<int>   used when a new console is created (AllocConsole)
 875  *      wineconsole <pgm> <arguments>   used to start the program <pgm> from the command line in
 876  *                                      a freshly created console
 877  * --backend=(curses|user) can also be used to select the desired backend
 878  */
 879 int PASCAL WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR lpCmdLine, INT nCmdShow)
 880 {
 881     struct inner_data*  data;
 882     int                 ret = 1;
 883     struct wc_init      wci;
 884
 885     if ((ret = WINECON_ParseOptions(lpCmdLine, &wci)) != 0)
 886     {
 887         printf_res(ret & 0xffff);
 888         if (ret & WINECON_CMD_SHOW_USAGE)
 889             WINECON_Usage();
 890         return 0;
 891     }
 892
 893     switch (wci.mode)
 894     {
 895     case from_event:
 896         /* case of wineconsole <evt>, signal process that created us that we're up and running */
 897         if (!(data = WINECON_Init(hInst, 0, NULL, wci.backend, nCmdShow))) return 1;
 898         ret = !SetEvent(wci.event);
 899         if (ret != 0) WINE_ERR("SetEvent failed.\n");
 900         break;
 901     case from_process_name:
 902         {
 903             int len;
 904             WCHAR *buffer;
 905
 906             len = MultiByteToWideChar(CP_ACP, 0, wci.ptr, -1, NULL, 0);
 907
 908             buffer = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 909             if (!buffer)
 910                 return 1;
 911
 912             MultiByteToWideChar(CP_ACP, 0, wci.ptr, -1, buffer, len);
 913
 914             if (!(data = WINECON_Init(hInst, GetCurrentProcessId(), buffer, wci.backend, nCmdShow)))
 915             {
 916                 HeapFree(GetProcessHeap(), 0, buffer);
 917                 return 1;
 918             }
 919             ret = WINECON_Spawn(data, buffer);
 920             HeapFree(GetProcessHeap(), 0, buffer);
 921             if (ret != 0)
 922             {
 923                 WINECON_Delete(data);
 924                 printf_res(IDS_CMD_LAUNCH_FAILED, wine_dbgstr_a(wci.ptr));
 925                 return ret;
 926             }
 927         }
 928         break;
 929     default:
 930         return 1;
 931     }
 932
 933     if (!ret)
 934     {
 935         DWORD exitcode;
 936
 937         WINE_TRACE("calling MainLoop.\n");
 938         ret = data->fnMainLoop(data);
 939
 940         if (!ret && data->hProcess &&
 941             WaitForSingleObject(data->hProcess, INFINITE) == WAIT_OBJECT_0 &&
 942             GetExitCodeProcess(data->hProcess, &exitcode))
 943         {
 944             WINE_TRACE("forwarding exitcode %u from child process\n", exitcode);
 945             ret = exitcode;
 946         }
 947     }
 948
 949     WINECON_Delete(data);
 950
 951     return ret;
 952 }