dlls/msvcrt/console.c

   1 /*
   2  * msvcrt.dll console functions
   3  *
   4  * Copyright 2000 Jon Griffiths
   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  * Note: init and free don't need MT locking since they are called at DLL
  21  * (de)attachment time, which is syncronised for us
  22  */
  23
  24 #include "msvcrt.h"
  25 #include "winnls.h"
  26 #include "wincon.h"
  27 #include "mtdll.h"
  28 #include "wine/debug.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
  31
  32
  33
  34 /* MT */
  35 #define LOCK_CONSOLE   _mlock(_CONIO_LOCK)
  36 #define UNLOCK_CONSOLE _munlock(_CONIO_LOCK)
  37
  38 static HANDLE MSVCRT_console_in = INVALID_HANDLE_VALUE;
  39 static HANDLE MSVCRT_console_out= INVALID_HANDLE_VALUE;
  40 static int __MSVCRT_console_buffer = MSVCRT_EOF;
  41
  42 /* INTERNAL: Initialise console handles */
  43 void msvcrt_init_console(void)
  44 {
  45   TRACE(":Opening console handles\n");
  46
  47   MSVCRT_console_in = CreateFileA("CONIN$", GENERIC_READ, FILE_SHARE_READ,
  48                                   NULL, OPEN_EXISTING, 0, NULL);
  49   MSVCRT_console_out= CreateFileA("CONOUT$", GENERIC_WRITE, FILE_SHARE_WRITE,
  50                                     NULL, OPEN_EXISTING, 0, NULL);
  51
  52   if ((MSVCRT_console_in == INVALID_HANDLE_VALUE) ||
  53       (MSVCRT_console_out== INVALID_HANDLE_VALUE))
  54     WARN(":Console handle Initialisation FAILED!\n");
  55 }
  56
  57 /* INTERNAL: Free console handles */
  58 void msvcrt_free_console(void)
  59 {
  60   TRACE(":Closing console handles\n");
  61   CloseHandle(MSVCRT_console_in);
  62   CloseHandle(MSVCRT_console_out);
  63 }
  64
  65 /*********************************************************************
  66  *              _cputs (MSVCRT.@)
  67  */
  68 int CDECL _cputs(const char* str)
  69 {
  70   DWORD count;
  71   int retval = MSVCRT_EOF;
  72
  73   LOCK_CONSOLE;
  74   if (WriteConsoleA(MSVCRT_console_out, str, strlen(str), &count, NULL)
  75       && count == 1)
  76     retval = 0;
  77   UNLOCK_CONSOLE;
  78   return retval;
  79 }
  80
  81 /*********************************************************************
  82  *              _cputws (MSVCRT.@)
  83  */
  84 int CDECL _cputws(const MSVCRT_wchar_t* str)
  85 {
  86   DWORD count;
  87   int retval = MSVCRT_EOF;
  88
  89   LOCK_CONSOLE;
  90   if (WriteConsoleW(MSVCRT_console_out, str, lstrlenW(str), &count, NULL)
  91       && count == 1)
  92     retval = 0;
  93   UNLOCK_CONSOLE;
  94   return retval;
  95 }
  96
  97 #define NORMAL_CHAR     0
  98 #define ALT_CHAR        1
  99 #define CTRL_CHAR       2
 100 #define SHIFT_CHAR      3
 101
 102 static const struct {unsigned vk; unsigned ch[4][2];} enh_map[] = {
 103     {0x47, {{0xE0, 0x47}, {0x00, 0x97}, {0xE0, 0x77}, {0xE0, 0x47}}},
 104     {0x48, {{0xE0, 0x48}, {0x00, 0x98}, {0xE0, 0x8D}, {0xE0, 0x48}}},
 105     {0x49, {{0xE0, 0x49}, {0x00, 0x99}, {0xE0, 0x86}, {0xE0, 0x49}}},
 106     {0x4B, {{0xE0, 0x4B}, {0x00, 0x9B}, {0xE0, 0x73}, {0xE0, 0x4B}}},
 107     {0x4D, {{0xE0, 0x4D}, {0x00, 0x9D}, {0xE0, 0x74}, {0xE0, 0x4D}}},
 108     {0x4F, {{0xE0, 0x4F}, {0x00, 0x9F}, {0xE0, 0x75}, {0xE0, 0x4F}}},
 109     {0x50, {{0xE0, 0x50}, {0x00, 0xA0}, {0xE0, 0x91}, {0xE0, 0x50}}},
 110     {0x51, {{0xE0, 0x51}, {0x00, 0xA1}, {0xE0, 0x76}, {0xE0, 0x51}}},
 111     {0x52, {{0xE0, 0x52}, {0x00, 0xA2}, {0xE0, 0x92}, {0xE0, 0x52}}},
 112     {0x53, {{0xE0, 0x53}, {0x00, 0xA3}, {0xE0, 0x93}, {0xE0, 0x53}}},
 113 };
 114
 115 /*********************************************************************
 116  *              _getch (MSVCRT.@)
 117  */
 118 int CDECL _getch(void)
 119 {
 120   int retval = MSVCRT_EOF;
 121
 122   LOCK_CONSOLE;
 123   if (__MSVCRT_console_buffer != MSVCRT_EOF)
 124   {
 125     retval = __MSVCRT_console_buffer;
 126     __MSVCRT_console_buffer = MSVCRT_EOF;
 127   }
 128   else
 129   {
 130     INPUT_RECORD ir;
 131     DWORD count;
 132     DWORD mode = 0;
 133
 134     GetConsoleMode(MSVCRT_console_in, &mode);
 135     if(mode)
 136       SetConsoleMode(MSVCRT_console_in, 0);
 137
 138     do {
 139       if (ReadConsoleInputA(MSVCRT_console_in, &ir, 1, &count))
 140       {
 141           unsigned int i;
 142         /* Only interested in ASCII chars */
 143         if (ir.EventType == KEY_EVENT &&
 144             ir.Event.KeyEvent.bKeyDown)
 145         {
 146             if (ir.Event.KeyEvent.uChar.AsciiChar)
 147             {
 148                 retval = ir.Event.KeyEvent.uChar.AsciiChar;
 149                 break;
 150             }
 151             for (i = 0; i < sizeof(enh_map) / sizeof(enh_map[0]); i++)
 152             {
 153                 if (ir.Event.KeyEvent.wVirtualScanCode == enh_map[i].vk) break;
 154             }
 155             if (i < sizeof(enh_map) / sizeof(enh_map[0]))
 156             {
 157                 unsigned idx;
 158
 159                 if (ir.Event.KeyEvent.dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
 160                     idx = ALT_CHAR;
 161                 else if (ir.Event.KeyEvent.dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED) )
 162                     idx = CTRL_CHAR;
 163                 else if (ir.Event.KeyEvent.dwControlKeyState & SHIFT_PRESSED)
 164                     idx = SHIFT_CHAR;
 165                 else
 166                     idx = NORMAL_CHAR;
 167
 168                 retval = enh_map[i].ch[idx][0];
 169                 __MSVCRT_console_buffer = enh_map[i].ch[idx][1];
 170                 break;
 171             }
 172             WARN("Unmapped char keyState=%x vk=%x\n",
 173                  ir.Event.KeyEvent.dwControlKeyState, ir.Event.KeyEvent.wVirtualScanCode);
 174         }
 175       }
 176       else
 177         break;
 178     } while(1);
 179     if (mode)
 180       SetConsoleMode(MSVCRT_console_in, mode);
 181   }
 182   UNLOCK_CONSOLE;
 183   return retval;
 184 }
 185
 186 /*********************************************************************
 187  *              _putch (MSVCRT.@)
 188  */
 189 int CDECL _putch(int c)
 190 {
 191   int retval = MSVCRT_EOF;
 192   DWORD count;
 193   LOCK_CONSOLE;
 194   if (WriteConsoleA(MSVCRT_console_out, &c, 1, &count, NULL) && count == 1)
 195     retval = c;
 196   UNLOCK_CONSOLE;
 197   return retval;
 198 }
 199
 200 /*********************************************************************
 201  *              _getche (MSVCRT.@)
 202  */
 203 int CDECL _getche(void)
 204 {
 205   int retval;
 206   LOCK_CONSOLE;
 207   retval = _getch();
 208   if (retval != MSVCRT_EOF)
 209     retval = _putch(retval);
 210   UNLOCK_CONSOLE;
 211   return retval;
 212 }
 213
 214 /*********************************************************************
 215  *              _cgets (MSVCRT.@)
 216  */
 217 char* CDECL _cgets(char* str)
 218 {
 219   char *buf = str + 2;
 220   DWORD got;
 221   DWORD conmode = 0;
 222
 223   TRACE("(%p)\n", str);
 224   str[1] = 0; /* Length */
 225   LOCK_CONSOLE;
 226   GetConsoleMode(MSVCRT_console_in, &conmode);
 227   SetConsoleMode(MSVCRT_console_in, ENABLE_LINE_INPUT|ENABLE_ECHO_INPUT|ENABLE_PROCESSED_INPUT);
 228
 229   if(ReadConsoleA(MSVCRT_console_in, buf, str[0], &got, NULL)) {
 230     if(buf[got-2] == '\r') {
 231       buf[got-2] = 0;
 232       str[1] = got-2;
 233     }
 234     else if(got == 1 && buf[got-1] == '\n') {
 235       buf[0] = 0;
 236       str[1] = 0;
 237     }
 238     else if(got == str[0] && buf[got-1] == '\r') {
 239       buf[got-1] = 0;
 240       str[1] = got-1;
 241     }
 242     else
 243       str[1] = got;
 244   }
 245   else
 246     buf = NULL;
 247   SetConsoleMode(MSVCRT_console_in, conmode);
 248   UNLOCK_CONSOLE;
 249   return buf;
 250 }
 251
 252 /*********************************************************************
 253  *              _ungetch (MSVCRT.@)
 254  */
 255 int CDECL _ungetch(int c)
 256 {
 257   int retval = MSVCRT_EOF;
 258   LOCK_CONSOLE;
 259   if (c != MSVCRT_EOF && __MSVCRT_console_buffer == MSVCRT_EOF)
 260     retval = __MSVCRT_console_buffer = c;
 261   UNLOCK_CONSOLE;
 262   return retval;
 263 }
 264
 265 /*********************************************************************
 266  *              _kbhit (MSVCRT.@)
 267  */
 268 int CDECL _kbhit(void)
 269 {
 270   int retval = 0;
 271
 272   LOCK_CONSOLE;
 273   if (__MSVCRT_console_buffer != MSVCRT_EOF)
 274     retval = 1;
 275   else
 276   {
 277     /* FIXME: There has to be a faster way than this in Win32.. */
 278     INPUT_RECORD *ir = NULL;
 279     DWORD count = 0, i;
 280
 281     GetNumberOfConsoleInputEvents(MSVCRT_console_in, &count);
 282
 283     if (count && (ir = MSVCRT_malloc(count * sizeof(INPUT_RECORD))) &&
 284         PeekConsoleInputA(MSVCRT_console_in, ir, count, &count))
 285       for(i = 0; i < count - 1; i++)
 286       {
 287         if (ir[i].EventType == KEY_EVENT &&
 288             ir[i].Event.KeyEvent.bKeyDown &&
 289             ir[i].Event.KeyEvent.uChar.AsciiChar)
 290         {
 291           retval = 1;
 292           break;
 293         }
 294       }
 295     MSVCRT_free(ir);
 296   }
 297   UNLOCK_CONSOLE;
 298   return retval;
 299 }
 300
 301
 302 /*********************************************************************
 303  *              _vcprintf (MSVCRT.@)
 304  */
 305 int CDECL _vcprintf(const char* format, __ms_va_list valist)
 306 {
 307   char buf[2048];
 308   LPWSTR formatW = NULL;
 309   DWORD sz;
 310   pf_output out;
 311   int retval;
 312
 313   out.unicode = FALSE;
 314   out.buf.A = out.grow.A = buf;
 315   out.used = 0;
 316   out.len = sizeof(buf);
 317
 318   sz = MultiByteToWideChar( CP_ACP, 0, format, -1, NULL, 0 );
 319   formatW = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
 320   MultiByteToWideChar( CP_ACP, 0, format, -1, formatW, sz );
 321
 322   if ((retval = pf_vsnprintf( &out, formatW, NULL, FALSE, valist )) > 0)
 323   {
 324       LOCK_CONSOLE;
 325       retval = _cputs( out.buf.A );
 326       UNLOCK_CONSOLE;
 327   }
 328   HeapFree( GetProcessHeap(), 0, formatW );
 329   if (out.buf.A != buf)
 330     MSVCRT_free (out.buf.A);
 331   return retval;
 332 }
 333
 334 /*********************************************************************
 335  *              _cprintf (MSVCRT.@)
 336  */
 337 int CDECL _cprintf(const char* format, ...)
 338 {
 339   int retval;
 340   __ms_va_list valist;
 341
 342   __ms_va_start( valist, format );
 343   retval = _vcprintf(format, valist);
 344   __ms_va_end(valist);
 345
 346   return retval;
 347 }
 348
 349
 350 /*********************************************************************
 351  *              _vcwprintf (MSVCRT.@)
 352  */
 353 int CDECL _vcwprintf(const MSVCRT_wchar_t* format, __ms_va_list valist)
 354 {
 355   MSVCRT_wchar_t buf[2048];
 356   pf_output out;
 357   int retval;
 358
 359   out.unicode = TRUE;
 360   out.buf.W = out.grow.W = buf;
 361   out.used = 0;
 362   out.len = sizeof(buf) / sizeof(buf[0]);
 363
 364   if ((retval = pf_vsnprintf( &out, format, NULL, FALSE, valist )) >= 0)
 365   {
 366       LOCK_CONSOLE;
 367       retval = _cputws( out.buf.W );
 368       UNLOCK_CONSOLE;
 369   }
 370   if (out.buf.W != buf)
 371     MSVCRT_free (out.buf.W);
 372   return retval;
 373 }
 374
 375 /*********************************************************************
 376  *              _cwprintf (MSVCRT.@)
 377  */
 378 int CDECL _cwprintf(const MSVCRT_wchar_t* format, ...)
 379 {
 380   int retval;
 381   __ms_va_list valist;
 382
 383   __ms_va_start( valist, format );
 384   retval = _vcwprintf(format, valist);
 385   __ms_va_end(valist);
 386
 387   return retval;
 388 }