haiku-beta6/src/kernel/boot/loader/stdio.cpp

/*
** Copyright 2003, Axel Dörfler, [email protected]. All rights reserved.
** Distributed under the terms of the OpenBeOS License.
*/


#include <boot/vfs.h>
#include <boot/stdio.h>
#include <util/kernel_cpp.h>

#include <stdarg.h>
#include <string.h>

//#undef stdout
//#undef stdin
//extern FILE *stdout;
//extern FILE *stdin;


// ToDo: does it make sense to have the extra "\r\n" handling in here?
// 		or should this be done by the platform dependent part?


int
vfprintf(FILE *file, const char *format, va_list list)
{
	ConsoleNode *node = (ConsoleNode *)file;
	char buffer[1024];
		// the buffer handling could (or should) be done better...
		// we should have a vsprintf() version that accepts a
		// maximum buffer size (for the kernel as well!)

	int length = vsprintf(buffer, format, list);
	if (length > 0) {
		int pos = 0;
		char *c;

		// be nice to our audience and replace single "\n" with "\r\n"

		for (c = buffer; c[0] && (size_t)length < sizeof(buffer) - 1; c++, pos++) {
			if (c[0] == '\r' && c[1] == '\n') {
				c++;
				pos++;
				continue;
			}
			if (c[0] == '\n') {
				// insert a '\r' here
				memmove(c + 1, c, length - pos);
				c[0] = '\r';
				length++;
			}
		}
		node->Write(buffer, length);
	}

	return length;
}


int
vprintf(const char *format, va_list args)
{
	return vfprintf(stdout, format, args);
}


int
printf(const char *format, ...)
{
	va_list args;

	va_start(args, format);
	int status = vfprintf(stdout, format, args);
	va_end(args);

	return status;
}


int
fprintf(FILE *file, const char *format, ...)
{
	va_list args;

	va_start(args, format);
	int status = vfprintf(file, format, args);
	va_end(args);

	return status;
}


int
fputc(int character, FILE *file)
{
	if (file == NULL)
		return B_FILE_ERROR;

    status_t status;

	// we only support direct console output right now...	
	if (character == '\n')
		status = ((ConsoleNode *)file)->Write("\r\n", 2);
	else
		status = ((ConsoleNode *)file)->Write(&character, 1);

	if (status > 0)
		return character;

	return status;
}


int
fputs(const char *string, FILE *file)
{
	if (file == NULL)
		return B_FILE_ERROR;

	// ToDo: for consistency, we should replace "\n" with "\r\n" here, too.

	status_t status = ((ConsoleNode *)file)->Write(string, strlen(string));
	fputc('\n', file);

	return status;
}


int
putc(int character)
{
	return fputc(character, stdout);
}


int
putchar(int character)
{
	return fputc(character, stdout);
}


int
puts(const char *string)
{
	return fputs(string, stdout);
}