diff --git a/ChangeLog b/ChangeLog index 774595f28a..3277c98963 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,10 @@ +Sun Nov 28 04:40:00 2010 Luis Lavena + + * io.c (io_fwrite): use rb_w32_write_console under Windows. + + * win32/win32.c (rb_w32_write_console): added to write to write + Unicode using WriteConsoleW for stdout/stderr. [ruby-core:33166] + Sun Nov 28 03:58:47 2010 NARUSE, Yui * lib/net/http.rb: improve rdoc. diff --git a/io.c b/io.c index afd4f3110e..8e1f1b826e 100644 --- a/io.c +++ b/io.c @@ -961,6 +961,10 @@ do_writeconv(VALUE str, rb_io_t *fptr) static long io_fwrite(VALUE str, rb_io_t *fptr, int nosync) { +#ifdef _WIN32 + long len = rb_w32_write_console(str, fptr->fd); + if (len > 0) return len; +#endif str = do_writeconv(str, fptr); return io_binwrite(str, RSTRING_PTR(str), RSTRING_LEN(str), fptr, nosync); diff --git a/win32/win32.c b/win32/win32.c index aa83b5ec40..6637dcb9e3 100644 --- a/win32/win32.c +++ b/win32/win32.c @@ -5286,6 +5286,29 @@ rb_w32_write(int fd, const void *buf, size_t size) return ret; } +long +rb_w32_write_console(VALUE str, int fd) +{ + static int disable; + HANDLE handle; + DWORD dwMode, reslen; + + if (disable) return -1L; + handle = (HANDLE)_osfhnd(fd); + if (!GetConsoleMode(handle, &dwMode) || + !rb_econv_has_convpath_p(rb_enc_name(rb_enc_get(str)), "UTF-16LE")) + return -1L; + + str = rb_str_encode(str, rb_enc_from_encoding(rb_enc_find("UTF-16LE")), 0, + Qnil); + if (!WriteConsoleW(handle, (LPWSTR)RSTRING_PTR(str), RSTRING_LEN(str)/2, &reslen, NULL)) { + if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) + disable = TRUE; + return -1L; + } + return (long)reslen; +} + static int unixtime_to_filetime(time_t time, FILETIME *ft) {