windows-nt-4.0/private/sdktools/vctools/crt/crtw32/stdio/ungetwc.c

/***
*ungetwc.c - unget a wide character from a stream
*
*	Copyright (c) 1993-1995, Microsoft Corporation. All rights reserved.
*
*Purpose:
*	defines ungetwc() - pushes a wide character back onto an input stream
*
*Revision History:
*	04-26-93  CFW	Module created.
*	04-30-93  CFW	Bring wide char support from ungetc.c.
*	05-10-93  CFW	Optimize, fix error handling.
*	06-02-93  CFW	Wide get/put use wint_t.
*	07-16-93  SRW	ALPHA Merge
*       09-15-93  CFW   Use ANSI conformant "__" names.
*       10-01-93  CFW   Test only for TEXT, update comments.
*	10-28-93  CFW	Test for both IOSTRG and TEXT.
*	11-10-93  GJF	Merged in NT SDK version (picked up fix to a cast
*			expression). Also replaced MTHREAD with _MT.
*       02-07-94  CFW   POSIXify.
*       08-31-94  CFW   Fix for "C" locale, call wctomb().
*	02-06-94  CFW	assert -> _ASSERTE.
*	03-07-95  GJF	_[un]lock_str macros now take FILE * arg.
*	06-12-95  GJF	Replaced _osfile[] with _osfile() (macro referencing
*			field in ioinfo struct).
*       07-28-95  GJF   Replaced _osfile() with _osfile_safe().
*
*******************************************************************************/

#ifndef _POSIX_

#include <cruntime.h>
#include <stdio.h>
#include <stdlib.h>
#include <file2.h>
#include <dbgint.h>
#include <internal.h>
#include <mtdll.h>
#include <msdos.h>
#include <errno.h>
#include <wchar.h>
#include <setlocal.h>

#ifdef	_MT	/* multi-thread; define both ungetwc and _lk_ungetwc */

/***
*wint_t ungetwc(ch, stream) - put a wide character back onto a stream
*
*Purpose:
*	Guaranteed one char pushback on a stream as long as open for reading.
*	More than one char pushback in a row is not guaranteed, and will fail
*	if it follows an ungetc which pushed the first char in buffer. Failure
*	causes return of WEOF.
*
*Entry:
*	wint_t ch - wide character to push back
*	FILE *stream - stream to push character onto
*
*Exit:
*	returns ch
*	returns WEOF if tried to push WEOF, stream not opened for reading or
*	or if we have already ungetc'd back to beginning of buffer.
*
*Exceptions:
*
*******************************************************************************/

wint_t __cdecl ungetwc (
	REG2 wint_t ch,
	REG1 FILE *stream
	)
{
	wint_t retval;

	_ASSERTE(stream != NULL);

	_lock_str(stream);

	retval = _ungetwc_lk (ch, stream);

	_unlock_str(stream);

	return(retval);
}

/***
*_ungetwc_lk() -  Ungetwc() core routine (locked version)
*
*Purpose:
*	Core ungetwc() routine; assumes stream is already locked.
*
*	[See ungetwc() above for more info.]
*
*Entry: [See ungetwc()]
*
*Exit:	[See ungetwc()]
*
*Exceptions:
*
*******************************************************************************/

wint_t __cdecl _ungetwc_lk (
	wint_t ch,
	FILE *str
	)
{

#else	/* non multi-thread; just define ungetc */

wint_t __cdecl ungetwc (
	wint_t ch,
	FILE *str
	)
{

#endif	/* rejoin common code */

	_ASSERTE(str != NULL);

	/*
	 * Requirements for success:
	 *
	 * 1. Character to be pushed back on the stream must not be WEOF.
	 *
	 * 2. The stream must currently be in read mode, or must be open for
	 *    update (i.e., read/write) and must NOT currently be in write
	 *    mode.
	 */
	if ( (ch != WEOF) &&
	     ( (str->_flag & _IOREAD) || ((str->_flag & _IORW) &&
		!(str->_flag & _IOWRT))
	     )
	   )
	{
		/* If stream is unbuffered, get one. */
		if (str->_base == NULL)
			_getbuf(str);

#ifndef _NTSUBSET_
	    if (!(str->_flag & _IOSTRG) && (_osfile_safe(_fileno(str)) & 
                  FTEXT))
		{
			/*
			 * Text mode, sigh... Convert the wc to a mbc.
			 */
			int size;
			char mbc[4];

                if ((size = wctomb(mbc, ch)) == -1)
			{
				/*
				 * Conversion failed! Set errno and return
				 * failure.
				 */
				errno = EILSEQ;
				return WEOF;
			}

			/* we know _base != NULL; since file is buffered */
			if (str->_ptr == str->_base)
			{
				if (str->_cnt)
				/* my back is against the wall; i've already done
				 * ungetwc, and there's no room for this one
				 */
					return WEOF;
				str->_ptr += size;
			}

			if ( size == 1 )
			{
		                *--str->_ptr = mbc[0];
			}
			else /* size == 2 */
			{
		                *--str->_ptr = mbc[1];
		                *--str->_ptr = mbc[0];
			}

			str->_cnt += size;

			str->_flag &= ~_IOEOF;
			str->_flag |= _IOREAD;	/* may already be set */
			return (wint_t) (0x0ffff & ch);
		}
#endif
	      	/*
		 * Binary mode - push back the wide char.
		 */
		/* we know _base != NULL; since file is buffered */
		if (str->_ptr == str->_base)
		{
			if (str->_cnt)
				/* my back is against the wall; i've already done
				 * ungetc, and there's no room for this one
				 */
				return WEOF;
			str->_ptr += sizeof(wchar_t);
		}

		str->_cnt += sizeof(wchar_t);

		str->_flag &= ~_IOEOF;
		str->_flag |= _IOREAD;	/* may already be set */

		return (wint_t) (*--((wchar_t *)(str->_ptr)) = (wchar_t)(ch & 0xffff));

	}
	return WEOF;
}

#endif /* _POSIX_ */