/*
 * Copyright (C) 2012 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

/**
 * @file
 *
 * Standard Input/Output
 *
 */

#include <stdio.h>
#include <string.h>
#include "bootapp.h"
#include "wimboot.h"
#include "efi.h"

/**
 * Print character to console
 *
 * @v character		Character to print
 */
int putchar ( int character ) {
	EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *conout;
	struct bootapp_callback_params params;
	wchar_t wbuf[2];

	/* Convert LF to CR,LF */
	if ( character == '\n' )
		putchar ( '\r' );

	/* Print character to bochs debug port */
	__asm__ __volatile__ ( "outb %b0, $0xe9"
			       : : "a" ( character ) );

	/* Print character to EFI/BIOS console as applicable */
	if ( efi_systab ) {
		conout = efi_systab->ConOut;
		wbuf[0] = character;
		wbuf[1] = 0;
		conout->OutputString ( conout, wbuf );
	} else {
		memset ( &params, 0, sizeof ( params ) );
		params.vector.interrupt = 0x10;
		params.eax = ( 0x0e00 | character );
		params.ebx = 0x0007;
		call_interrupt ( &params );
	}

	return 0;
}

/**
 * Get character from console
 *
 * @ret character	Character
 */
int getchar ( void ) {
	EFI_BOOT_SERVICES *bs;
	EFI_SIMPLE_TEXT_INPUT_PROTOCOL *conin;
	EFI_INPUT_KEY key;
	UINTN index;
	struct bootapp_callback_params params;
	int character;

	/* Get character */
	if ( efi_systab ) {
		bs = efi_systab->BootServices;
		conin = efi_systab->ConIn;
		bs->WaitForEvent ( 1, &conin->WaitForKey, &index );
		conin->ReadKeyStroke ( conin, &key );
		character = key.UnicodeChar;
	} else {
		memset ( &params, 0, sizeof ( params ) );
		params.vector.interrupt = 0x16;
		call_interrupt ( &params );
		character = params.al;
	}

	return character;
}