/*
 * :ts=4
 *
 * Portable ISO 'C' (1994) runtime library for the Amiga computer
 * Copyright (c) 2002-2019 by Olaf Barthel <obarthel (at) gmx.net>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *   - Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *
 *   - Neither the name of Olaf Barthel nor the names of contributors
 *     may be used to endorse or promote products derived from this
 *     software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef _STDLIB_NULL_POINTER_CHECK_H
#include "stdlib_null_pointer_check.h"
#endif /* _STDLIB_NULL_POINTER_CHECK_H */

/****************************************************************************/

#ifndef _STDIO_HEADERS_H
#include "stdio_headers.h"
#endif /* _STDIO_HEADERS_H */

/****************************************************************************/

/* Both fputs() and puts() share this function. */
int
__fputs(const char *s, int line_feed, FILE *stream)
{
	struct iob * file = (struct iob *)stream;
	size_t total_size;
	int result = EOF;
	int buffer_mode;
	int c;

	ENTER();

	SHOWSTRING(s);
	SHOWPOINTER(stream);

	assert( s != NULL && stream != NULL );

	if (__check_abort_enabled)
		__check_abort();

	#if defined(CHECK_FOR_NULL_POINTERS)
	{
		if (s == NULL || stream == NULL)
		{
			__set_errno(EFAULT);
			goto out;
		}
	}
	#endif /* CHECK_FOR_NULL_POINTERS */

	flockfile(stream);

	assert( __is_valid_iob(file) );
	assert( FLAG_IS_SET(file->iob_Flags,IOBF_IN_USE) );
	assert( file->iob_BufferSize > 0 );

	if (__fputc_check(stream) < 0)
		goto out;

	assert( 0 <= file->iob_BufferWriteBytes );
	assert( file->iob_BufferWriteBytes <= file->iob_BufferSize );
	assert( file->iob_BufferPosition <= file->iob_BufferSize );

	/* We perform line buffering to improve readability of the
	 * buffered text if buffering was disabled and the output
	 * goes to an interactive stream.
	 */
	buffer_mode = (file->iob_Flags & IOBF_BUFFER_MODE);
	if (buffer_mode == IOBF_BUFFER_MODE_NONE)
	{
		struct fd * fd = __fd[file->iob_Descriptor];

		__fd_lock(fd);

		if (FLAG_IS_SET(fd->fd_Flags, FDF_IS_INTERACTIVE))
			buffer_mode = IOBF_BUFFER_MODE_LINE;

		__fd_unlock(fd);
	}

	total_size = strlen(s);
	if (total_size > 0)
	{
		if (buffer_mode == IOBF_BUFFER_MODE_LINE)
		{
			while (total_size > 0)
			{
				/* Is there still room in the write buffer to store
				 * more of the string?
				 */
				if (file->iob_BufferWriteBytes < file->iob_BufferSize)
				{
					unsigned char * buffer = &file->iob_Buffer[file->iob_BufferWriteBytes];
					size_t num_buffer_bytes;
					const char * lf;

					/* Give the user a chance to abort what could otherwise
					 * become an uninterrupted series of copying operations.
					 */
					if (__check_abort_enabled)
						__check_abort();

					/* Store only as many characters as will fit into the write buffer. */
					assert( file->iob_BufferSize >= file->iob_BufferWriteBytes );

					num_buffer_bytes = file->iob_BufferSize - file->iob_BufferWriteBytes;
					if (total_size < num_buffer_bytes)
						num_buffer_bytes = total_size;

					/* Try to find a line feed in the string. If there is one,
					 * reduce the number of characters to write to the sequence
					 * which ends with the line feed character.
					 */
					lf = memchr(s, '\n', num_buffer_bytes);
					if (lf != NULL)
					{
						assert( (size_t)(lf + 1 - s) <= num_buffer_bytes );

						num_buffer_bytes = lf + 1 - s;
					}

					assert( num_buffer_bytes > 0 );

					memmove(buffer, s, num_buffer_bytes);
					s += num_buffer_bytes;

					assert( file->iob_BufferWriteBytes + num_buffer_bytes <= file->iob_BufferSize );

					file->iob_BufferWriteBytes += num_buffer_bytes;

					/* Write the buffer to disk if it's full or contains a line feed. */
					if ((lf != NULL || __iob_write_buffer_is_full(file)) && __flush_iob_write_buffer(file) < 0)
					{
						/* Abort with error. */
						goto out;
					}

					/* Stop as soon as no further data needs to be written. */
					assert( total_size >= num_buffer_bytes );

					total_size -= num_buffer_bytes;
					if (total_size == 0)
						break;

					/* If there is again room in the output buffer,
					 * repeat this optimization.
					 */
					if (file->iob_BufferWriteBytes < file->iob_BufferSize)
						continue;
				}

				c = (*s++);

				if (__putc_line_buffered(c, (FILE *)file) == EOF)
					goto out;

				total_size--;
			}
		}
		else if (buffer_mode == IOBF_BUFFER_MODE_NONE)
		{
			ssize_t num_bytes_written;

			/* We bypass the buffer entirely. */
			num_bytes_written = write(file->iob_Descriptor, s, total_size);
			if (num_bytes_written == -1)
			{
				SET_FLAG(file->iob_Flags, IOBF_ERROR);
				goto out;
			}
		}
		else
		{
			while (total_size > 0)
			{
				/* If there is more data to be written than the write buffer will hold
				 * and the write buffer is empty anyway, then we'll bypass the write
				 * buffer entirely.
				 */
				if (file->iob_BufferWriteBytes == 0 && total_size >= (size_t)file->iob_BufferSize)
				{
					ssize_t num_bytes_written;

					/* We bypass the buffer entirely. */
					num_bytes_written = write(file->iob_Descriptor, s, total_size);
					if (num_bytes_written == -1)
					{
						SET_FLAG(file->iob_Flags, IOBF_ERROR);
						goto out;
					}

					break;
				}

				/* Is there still room in the write buffer to store
				 * more of the string?
				 */
				if (file->iob_BufferWriteBytes < file->iob_BufferSize)
				{
					unsigned char * buffer = &file->iob_Buffer[file->iob_BufferWriteBytes];
					size_t num_buffer_bytes;

					/* Give the user a chance to abort what could otherwise
					 * become an uninterrupted series of copying operations.
					 */
					if (__check_abort_enabled)
						__check_abort();

					/* Store only as many characters as will fit into the write buffer. */
					assert( file->iob_BufferSize >= file->iob_BufferWriteBytes );

					num_buffer_bytes = file->iob_BufferSize - file->iob_BufferWriteBytes;
					if (total_size < num_buffer_bytes)
						num_buffer_bytes = total_size;

					assert( num_buffer_bytes > 0 );

					memmove(buffer, s, num_buffer_bytes);
					s += num_buffer_bytes;

					assert( file->iob_BufferWriteBytes + num_buffer_bytes <= file->iob_BufferSize );

					file->iob_BufferWriteBytes += num_buffer_bytes;

					/* Write a full buffer to disk. */
					if (__iob_write_buffer_is_full(file) && __flush_iob_write_buffer(file) < 0)
					{
						/* Abort with error. */
						goto out;
					}

					/* Stop as soon as no further data needs to be written. */
					assert( total_size >= num_buffer_bytes );

					total_size -= num_buffer_bytes;
					if (total_size == 0)
						break;

					/* If there is again room in the output buffer,
					 * try this optimization again.
					 */
					if (file->iob_BufferWriteBytes < file->iob_BufferSize)
						continue;
				}

				c = (*s++);

				if (__putc_fully_buffered(c, (FILE *)file) == EOF)
					goto out;

				total_size--;
			}
		}
	}

	if (line_feed != 0 && __putc(line_feed, stream, buffer_mode) == EOF)
		goto out;

	result = OK;

 out:

	/* Note: if buffering is disabled for this stream, then we still
	 * may have buffered data around, queued to be printed right now.
	 * This is intended to improve performance as it takes more effort
	 * to write a single character to a file than to write a bunch.
	 */
	if (result == 0 && (file->iob_Flags & IOBF_BUFFER_MODE) == IOBF_BUFFER_MODE_NONE)
	{
		if (__iob_write_buffer_is_valid(file) && __flush_iob_write_buffer(file) < 0)
		{
			SHOWMSG("couldn't flush the write buffer");
			result = EOF;
		}
	}

	if (stream != NULL)
		funlockfile(stream);

	RETURN(result);
	return(result);
}

/****************************************************************************/

int
fputs(const char *s, FILE *stream)
{
	int result;

	ENTER();

	SHOWSTRING(s);
	SHOWPOINTER(stream);

	assert( s != NULL && stream != NULL );

	if (__check_abort_enabled)
		__check_abort();

	#if defined(CHECK_FOR_NULL_POINTERS)
	{
		if (s == NULL || stream == NULL)
		{
			__set_errno(EFAULT);
			goto out;
		}
	}
	#endif /* CHECK_FOR_NULL_POINTERS */

	result = __fputs(s, 0, stream);

	RETURN(result);
	return(result);
}