/* Copyright libuv project contributors. All rights reserved.

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to

 * deal in the Software without restriction, including without limitation the

 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or

 * sell copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice shall be included in

 * all copies or substantial portions of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS

 * IN THE SOFTWARE.

 */

#ifdef _WIN32

#include "uv.h"

#include "task.h"

#include <errno.h>

#include <io.h>

#include <string.h>

#include <windows.h>

#define ESC "\x1b"

#define EUR_UTF8 "\xe2\x82\xac"

#define EUR_UNICODE 0x20AC

const char* expect_str = NULL;

ssize_t expect_nread = 0;

static void dump_str(const char* str, ssize_t len) {

  ssize_t i;

  for (i = 0; i < len; i++) {

    fprintf(stderr, "%#02x ", *(str + i));

  }

}

static void print_err_msg(const char* expect, ssize_t expect_len,

                          const char* found, ssize_t found_len) {

  fprintf(stderr, "expect ");

  dump_str(expect, expect_len);

  fprintf(stderr, ", but found ");

  dump_str(found, found_len);

  fprintf(stderr, "\n");

}

static void tty_alloc(uv_handle_t* handle, size_t size, uv_buf_t* buf) {

  buf->base = malloc(size);

  ASSERT(buf->base != NULL);

  buf->len = size;

}

static void tty_read(uv_stream_t* tty_in, ssize_t nread, const uv_buf_t* buf) {

  if (nread > 0) {

    if (nread != expect_nread) {

      fprintf(stderr, "expected nread %ld, but found %ld\n",

              (long)expect_nread, (long)nread);

      print_err_msg(expect_str, expect_nread, buf->base, nread);

      ASSERT(FALSE);

    }

    if (strncmp(buf->base, expect_str, nread) != 0) {

      print_err_msg(expect_str, expect_nread, buf->base, nread);

      ASSERT(FALSE);

    }

    uv_close((uv_handle_t*) tty_in, NULL);

  } else {

    ASSERT(nread == 0);

  }

}

static void make_key_event_records(WORD virt_key, DWORD ctr_key_state,

                                   BOOL is_wsl, INPUT_RECORD* records) {

# define KEV(I) records[(I)].Event.KeyEvent

  BYTE kb_state[256] = {0};

  WCHAR buf[2];

  int ret;

  records[0].EventType = records[1].EventType = KEY_EVENT;

  KEV(0).bKeyDown = TRUE;

  KEV(1).bKeyDown = FALSE;

  KEV(0).wVirtualKeyCode = KEV(1).wVirtualKeyCode = virt_key;

  KEV(0).wRepeatCount = KEV(1).wRepeatCount = 1;

  KEV(0).wVirtualScanCode = KEV(1).wVirtualScanCode =

    MapVirtualKeyW(virt_key, MAPVK_VK_TO_VSC);

  KEV(0).dwControlKeyState = KEV(1).dwControlKeyState = ctr_key_state;

  if (ctr_key_state & LEFT_ALT_PRESSED) {

    kb_state[VK_LMENU] = 0x01;

  }

  if (ctr_key_state & RIGHT_ALT_PRESSED) {

    kb_state[VK_RMENU] = 0x01;

  }

  if (ctr_key_state & LEFT_CTRL_PRESSED) {

    kb_state[VK_LCONTROL] = 0x01;

  }

  if (ctr_key_state & RIGHT_CTRL_PRESSED) {

    kb_state[VK_RCONTROL] = 0x01;

  }

  if (ctr_key_state & SHIFT_PRESSED) {

    kb_state[VK_SHIFT] = 0x01;

  }

  ret = ToUnicode(virt_key, KEV(0).wVirtualScanCode, kb_state, buf, 2, 0);

  if (ret == 1) {

    if(!is_wsl &&

        ((ctr_key_state & LEFT_ALT_PRESSED) ||

         (ctr_key_state & RIGHT_ALT_PRESSED))) {

      /*

       * If ALT key is pressed, the UnicodeChar value of the keyup event is

       * set to 0 on nomal console. Emulate this behavior.

       * See https://github.com/Microsoft/console/issues/320

       */

      KEV(0).uChar.UnicodeChar = buf[0];

      KEV(1).uChar.UnicodeChar = 0;

    } else{

      /*

       * In WSL UnicodeChar is normally set. This behavior cause #2111.

       */

      KEV(0).uChar.UnicodeChar = KEV(1).uChar.UnicodeChar = buf[0];

    }

  } else {

    KEV(0).uChar.UnicodeChar = KEV(1).uChar.UnicodeChar = 0;

  }

# undef KEV

}

TEST_IMPL(tty_duplicate_vt100_fn_key) {

  int r;

  int ttyin_fd;

  uv_tty_t tty_in;

  uv_loop_t* loop;

  HANDLE handle;

  INPUT_RECORD records[2];

  DWORD written;

  loop = uv_default_loop();

  /* Make sure we have an FD that refers to a tty */

  handle = CreateFileA("conin$",

                       GENERIC_READ | GENERIC_WRITE,

                       FILE_SHARE_READ | FILE_SHARE_WRITE,

                       NULL,

                       OPEN_EXISTING,

                       FILE_ATTRIBUTE_NORMAL,

                       NULL);

  ASSERT(handle != INVALID_HANDLE_VALUE);

  ttyin_fd = _open_osfhandle((intptr_t) handle, 0);

  ASSERT(ttyin_fd >= 0);

  ASSERT(UV_TTY == uv_guess_handle(ttyin_fd));

  r = uv_tty_init(uv_default_loop(), &tty_in, ttyin_fd, 1);  /* Readable. */

  ASSERT(r == 0);

  ASSERT(uv_is_readable((uv_stream_t*) &tty_in));

  ASSERT(!uv_is_writable((uv_stream_t*) &tty_in));

  r = uv_read_start((uv_stream_t*)&tty_in, tty_alloc, tty_read);

  ASSERT(r == 0);

  expect_str = ESC"[[A";

  expect_nread = strlen(expect_str);

  /* Turn on raw mode. */

  r = uv_tty_set_mode(&tty_in, UV_TTY_MODE_RAW);

  ASSERT(r == 0);

  /*

   * Send F1 keystrokes. Test of issue cause by #2114 that vt100 fn key

   * duplicate.

   */

  make_key_event_records(VK_F1, 0, TRUE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == ARRAY_SIZE(records));

  uv_run(loop, UV_RUN_DEFAULT);

  MAKE_VALGRIND_HAPPY();

  return 0;

}

TEST_IMPL(tty_duplicate_alt_modifier_key) {

  int r;

  int ttyin_fd;

  uv_tty_t tty_in;

  uv_loop_t* loop;

  HANDLE handle;

  INPUT_RECORD records[2];

  INPUT_RECORD alt_records[2];

  DWORD written;

  loop = uv_default_loop();

  /* Make sure we have an FD that refers to a tty */

  handle = CreateFileA("conin$",

                       GENERIC_READ | GENERIC_WRITE,

                       FILE_SHARE_READ | FILE_SHARE_WRITE,

                       NULL,

                       OPEN_EXISTING,

                       FILE_ATTRIBUTE_NORMAL,

                       NULL);

  ASSERT(handle != INVALID_HANDLE_VALUE);

  ttyin_fd = _open_osfhandle((intptr_t) handle, 0);

  ASSERT(ttyin_fd >= 0);

  ASSERT(UV_TTY == uv_guess_handle(ttyin_fd));

  r = uv_tty_init(uv_default_loop(), &tty_in, ttyin_fd, 1);  /* Readable. */

  ASSERT(r == 0);

  ASSERT(uv_is_readable((uv_stream_t*) &tty_in));

  ASSERT(!uv_is_writable((uv_stream_t*) &tty_in));

  r = uv_read_start((uv_stream_t*)&tty_in, tty_alloc, tty_read);

  ASSERT(r == 0);

  expect_str = ESC"a"ESC"a";

  expect_nread = strlen(expect_str);

  /* Turn on raw mode. */

  r = uv_tty_set_mode(&tty_in, UV_TTY_MODE_RAW);

  ASSERT(r == 0);

  /* Emulate transmission of M-a at normal console */

  make_key_event_records(VK_MENU, 0, TRUE, alt_records);

  WriteConsoleInputW(handle, &alt_records[0], 1, &written);

  ASSERT(written == 1);

  make_key_event_records(L'A', LEFT_ALT_PRESSED, FALSE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == 2);

  WriteConsoleInputW(handle, &alt_records[1], 1, &written);

  ASSERT(written == 1);

  /* Emulate transmission of M-a at WSL(#2111) */

  make_key_event_records(VK_MENU, 0, TRUE, alt_records);

  WriteConsoleInputW(handle, &alt_records[0], 1, &written);

  ASSERT(written == 1);

  make_key_event_records(L'A', LEFT_ALT_PRESSED, TRUE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == 2);

  WriteConsoleInputW(handle, &alt_records[1], 1, &written);

  ASSERT(written == 1);

  uv_run(loop, UV_RUN_DEFAULT);

  MAKE_VALGRIND_HAPPY();

  return 0;

}

TEST_IMPL(tty_composing_character) {

  int r;

  int ttyin_fd;

  uv_tty_t tty_in;

  uv_loop_t* loop;

  HANDLE handle;

  INPUT_RECORD records[2];

  INPUT_RECORD alt_records[2];

  DWORD written;

  loop = uv_default_loop();

  /* Make sure we have an FD that refers to a tty */

  handle = CreateFileA("conin$",

                       GENERIC_READ | GENERIC_WRITE,

                       FILE_SHARE_READ | FILE_SHARE_WRITE,

                       NULL,

                       OPEN_EXISTING,

                       FILE_ATTRIBUTE_NORMAL,

                       NULL);

  ASSERT(handle != INVALID_HANDLE_VALUE);

  ttyin_fd = _open_osfhandle((intptr_t) handle, 0);

  ASSERT(ttyin_fd >= 0);

  ASSERT(UV_TTY == uv_guess_handle(ttyin_fd));

  r = uv_tty_init(uv_default_loop(), &tty_in, ttyin_fd, 1);  /* Readable. */

  ASSERT(r == 0);

  ASSERT(uv_is_readable((uv_stream_t*) &tty_in));

  ASSERT(!uv_is_writable((uv_stream_t*) &tty_in));

  r = uv_read_start((uv_stream_t*)&tty_in, tty_alloc, tty_read);

  ASSERT(r == 0);

  expect_str = EUR_UTF8;

  expect_nread = strlen(expect_str);

  /* Turn on raw mode. */

  r = uv_tty_set_mode(&tty_in, UV_TTY_MODE_RAW);

  ASSERT(r == 0);

  /* Emulate EUR inputs by LEFT ALT+NUMPAD ASCII KeyComos */

  make_key_event_records(VK_MENU, 0, FALSE, alt_records);

  alt_records[1].Event.KeyEvent.uChar.UnicodeChar = EUR_UNICODE;

  WriteConsoleInputW(handle, &alt_records[0], 1, &written);

  make_key_event_records(VK_NUMPAD0, LEFT_ALT_PRESSED, FALSE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == ARRAY_SIZE(records));

  make_key_event_records(VK_NUMPAD1, LEFT_ALT_PRESSED, FALSE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == ARRAY_SIZE(records));

  make_key_event_records(VK_NUMPAD2, LEFT_ALT_PRESSED, FALSE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == ARRAY_SIZE(records));

  make_key_event_records(VK_NUMPAD8, LEFT_ALT_PRESSED, FALSE, records);

  WriteConsoleInputW(handle, records, ARRAY_SIZE(records), &written);

  ASSERT(written == ARRAY_SIZE(records));

  WriteConsoleInputW(handle, &alt_records[1], 1, &written);

  uv_run(loop, UV_RUN_DEFAULT);

  MAKE_VALGRIND_HAPPY();

  return 0;

}

#else

typedef int file_has_no_tests;  /* ISO C forbids an empty translation unit. */

#endif  /* ifndef _WIN32 */