cgetws.cpp File Reference

#include <conio.h>
#include <corecrt_internal_lowio.h>
#include <corecrt_internal_securecrt.h>
#include <stdlib.h>

Include dependency graph for cgetws.cpp:

Go to the source code of this file.

Functions
errno_t __cdecl	_cgetws_s (wchar_t *const string_buffer, size_t const size_in_words, size_t *const size_read)
wchar_t *__cdecl	_cgetws (_Inout_z_ wchar_t *const string)

Variables
wchar_t	__console_wchar_buffer = 0
int	__console_wchar_buffer_used = 0

Function Documentation

_cgetws()

wchar_t *__cdecl _cgetws

(

_Inout_z_ wchar_t *const

string

)

Definition at line 138 of file cgetws.cpp.

{
    _VALIDATE_CLEAR_OSSERR_RETURN(string != nullptr, EINVAL, nullptr);
    _VALIDATE_CLEAR_OSSERR_RETURN(string[0] > 0,     EINVAL, nullptr);
 
    size_t const size_in_words = static_cast<size_t>(string[0]);
 
    size_t size_read = 0;
    // warning 26018: Potential overflow of null terminated buffer using expression string+2
    // Suppressing warning since _cgetws is purposefully unsafe.
#pragma warning(suppress:__WARNING_POTENTIAL_BUFFER_OVERFLOW_NULLTERMINATED)
    errno_t const result = _cgetws_s(string + 2, size_in_words, &size_read);
 
    // warning 26018: Potential overflow of null terminated buffer using expression string[1]
    // Suppressing warning since _cgetws is purposefully unsafe.
#pragma warning(suppress:__WARNING_POTENTIAL_BUFFER_OVERFLOW_NULLTERMINATED)
    string[1] = static_cast<wchar_t>(size_read);
 
    return result == 0 ? string + 2 : nullptr;
}

_cgetws_s()

errno_t __cdecl _cgetws_s	(	wchar_t *const	string_buffer,
		size_t const	size_in_words,
		size_t *const	size_read
	)

Definition at line 25 of file cgetws.cpp.

{
    _VALIDATE_CLEAR_OSSERR_RETURN_ERRCODE(string_buffer != nullptr, EINVAL);
    _VALIDATE_CLEAR_OSSERR_RETURN_ERRCODE(size_in_words > 0, EINVAL);
    _RESET_STRING(string_buffer, size_in_words);
 
    _VALIDATE_CLEAR_OSSERR_RETURN_ERRCODE(size_read != nullptr, EINVAL);
 
    __acrt_lock(__acrt_conio_lock);
    errno_t retval = 0;
    __try
    {
        wchar_t* string = string_buffer;
 
        // We need to decrement size_in_words because ReadConsole reads as many
        // characters as the parameter passed.  It doesn't null-terminate:
        size_t size_remaining = size_in_words - 1;
        *size_read = 0;
 
        // If size_in_words was 1, then there's only room for the null terminator:
        if (size_remaining == 0)
            __leave;
 
        // If there is a buffered character, first fill with the buffered
        // character then proceed with reading from the console:
        if (__console_wchar_buffer_used != 0)
        {
            *string++ = __console_wchar_buffer;
            --size_remaining;
            (*size_read)++;
 
            if (__console_wchar_buffer == L'\0')
                size_remaining = 0;
 
            __console_wchar_buffer = 0;
        }
 
        if (size_remaining == 0)
            __leave;
 
        /*
        * __dcrt_lowio_console_input_handle, the handle to the console input, is created the first
        * time that either _getch() or _cgets() or _kbhit() is called.
        */
 
        if (__dcrt_lowio_ensure_console_input_initialized() == FALSE)
        {
            __acrt_errno_map_os_error(GetLastError());
            retval = errno;
            __leave;
        }
 
        ULONG old_state;
        __dcrt_get_input_console_mode(&old_state);
        __dcrt_set_input_console_mode(ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT | ENABLE_ECHO_INPUT);
 
        ULONG wchars_read;
        BOOL read_console_result = __dcrt_read_console(
            string,
            static_cast<DWORD>(size_remaining),
            &wchars_read);
 
        if (!read_console_result)
        {
            __acrt_errno_map_os_error(GetLastError());
            retval = errno;
            __leave;
        }
 
        // Set the length of the string and null terminate it:
        if (wchars_read >= 2 && string[wchars_read - 2] == L'\r')
        {
            *size_read += wchars_read - 2;
            string[wchars_read - 2] = L'\0';
        }
        else if (wchars_read == size_remaining && string[wchars_read - 1] == L'\r')
        {
            // Special case 1:  \r\n straddles the boundary:
            string[wchars_read - 1] = L'\0';
            *size_read += wchars_read - 1;
        }
        else if (wchars_read == 1 && string[0] == L'\n')
        {
            // Special case 2:  Read a single \n:
            string[0] = L'\0';
            *size_read += 0;
        }
        else
        {
            *size_read += wchars_read;
            string[wchars_read] = L'\0';
        }
 
        __dcrt_set_input_console_mode(old_state);
    }
    __finally
    {
        __acrt_unlock(__acrt_conio_lock);
    }
    __endtry
 
    return retval;
}

Referenced by _cgets_s(), and _cgetws().

Variable Documentation

__console_wchar_buffer

wchar_t __console_wchar_buffer = 0

Definition at line 18 of file cgetws.cpp.

Referenced by _cgets_s(), and _cgetws_s().

__console_wchar_buffer_used

int __console_wchar_buffer_used = 0

Definition at line 19 of file cgetws.cpp.

Referenced by _cgets_s(), and _cgetws_s().