mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-24 18:32:28 -05:00
406 lines
9.3 KiB
C++
406 lines
9.3 KiB
C++
#include "VirtualConsole.h"
|
|
#include "VGA.h"
|
|
#include "kmalloc.h"
|
|
#include "i386.h"
|
|
#include "StdLib.h"
|
|
#include "Keyboard.h"
|
|
#include <AK/String.h>
|
|
|
|
static byte* s_vgaBuffer;
|
|
static VirtualConsole* s_consoles[6];
|
|
static int s_activeConsole;
|
|
|
|
void VirtualConsole::initialize()
|
|
{
|
|
s_vgaBuffer = (byte*)0xb8000;
|
|
memset(s_consoles, 0, sizeof(s_consoles));
|
|
s_activeConsole = -1;
|
|
}
|
|
|
|
VirtualConsole::VirtualConsole(unsigned index, InitialContents initialContents)
|
|
: TTY(4, index)
|
|
, m_index(index)
|
|
{
|
|
s_consoles[index] = this;
|
|
m_buffer = (byte*)kmalloc(80 * 25 * 2);
|
|
dbgprintf("VirtualConsole %u @ %p, m_buffer = %p\n", index, this, m_buffer);
|
|
if (initialContents == AdoptCurrentVGABuffer) {
|
|
memcpy(m_buffer, s_vgaBuffer, 80 * 25 * 2);
|
|
auto vgaCursor = vga_get_cursor();
|
|
m_cursorRow = vgaCursor / 80;
|
|
m_cursorColumn = vgaCursor % 80;
|
|
} else {
|
|
word* linemem = (word*)m_buffer;
|
|
for (word i = 0; i < 80 * 25; ++i)
|
|
linemem[i] = 0x0720;
|
|
}
|
|
}
|
|
|
|
VirtualConsole::~VirtualConsole()
|
|
{
|
|
}
|
|
|
|
void VirtualConsole::switchTo(unsigned index)
|
|
{
|
|
if ((int)index == s_activeConsole)
|
|
return;
|
|
dbgprintf("VC: Switch to %u (%p)\n", index, s_consoles[index]);
|
|
ASSERT(index < 6);
|
|
ASSERT(s_consoles[index]);
|
|
InterruptDisabler disabler;
|
|
if (s_activeConsole != -1)
|
|
s_consoles[s_activeConsole]->setActive(false);
|
|
s_activeConsole = index;
|
|
s_consoles[s_activeConsole]->setActive(true);
|
|
Console::the().setImplementation(s_consoles[s_activeConsole]);
|
|
}
|
|
|
|
void VirtualConsole::setActive(bool b)
|
|
{
|
|
if (b == m_active)
|
|
return;
|
|
|
|
InterruptDisabler disabler;
|
|
|
|
m_active = b;
|
|
if (!m_active) {
|
|
memcpy(m_buffer, s_vgaBuffer, 80 * 25 * 2);
|
|
return;
|
|
}
|
|
|
|
memcpy(s_vgaBuffer, m_buffer, 80 * 25 * 2);
|
|
vga_set_cursor(m_cursorRow, m_cursorColumn);
|
|
|
|
Keyboard::the().setClient(this);
|
|
}
|
|
|
|
inline bool isParameter(byte ch)
|
|
{
|
|
return ch >= 0x30 && ch <= 0x3f;
|
|
}
|
|
|
|
inline bool isIntermediate(byte ch)
|
|
{
|
|
return ch >= 0x20 && ch <= 0x2f;
|
|
}
|
|
|
|
inline bool isFinal(byte ch)
|
|
{
|
|
return ch >= 0x40 && ch <= 0x7e;
|
|
}
|
|
|
|
unsigned parseUInt(const String& str, bool& ok)
|
|
{
|
|
unsigned value = 0;
|
|
for (size_t i = 0; i < str.length(); ++i) {
|
|
if (str[i] < '0' || str[i] > '9') {
|
|
ok = false;
|
|
return 0;
|
|
}
|
|
value = value * 10;
|
|
value += str[i] - '0';
|
|
}
|
|
ok = true;
|
|
return value;
|
|
}
|
|
|
|
enum class VGAColor : byte {
|
|
Black = 0,
|
|
Blue,
|
|
Green,
|
|
Cyan,
|
|
Red,
|
|
Magenta,
|
|
Brown,
|
|
LightGray,
|
|
DarkGray,
|
|
BrightBlue,
|
|
BrightGreen,
|
|
BrightCyan,
|
|
BrightRed,
|
|
BrightMagenta,
|
|
Yellow,
|
|
White,
|
|
};
|
|
|
|
enum class ANSIColor : byte {
|
|
Black = 0,
|
|
Red,
|
|
Green,
|
|
Brown,
|
|
Blue,
|
|
Magenta,
|
|
Cyan,
|
|
LightGray,
|
|
DarkGray,
|
|
BrightRed,
|
|
BrightGreen,
|
|
Yellow,
|
|
BrightBlue,
|
|
BrightMagenta,
|
|
BrightCyan,
|
|
White,
|
|
};
|
|
|
|
static inline VGAColor ansiColorToVGA(ANSIColor color)
|
|
{
|
|
switch (color) {
|
|
case ANSIColor::Black: return VGAColor::Black;
|
|
case ANSIColor::Red: return VGAColor::Red;
|
|
case ANSIColor::Brown: return VGAColor::Brown;
|
|
case ANSIColor::Blue: return VGAColor::Blue;
|
|
case ANSIColor::Magenta: return VGAColor::Magenta;
|
|
case ANSIColor::Green: return VGAColor::Green;
|
|
case ANSIColor::Cyan: return VGAColor::Cyan;
|
|
case ANSIColor::LightGray: return VGAColor::LightGray;
|
|
case ANSIColor::DarkGray: return VGAColor::DarkGray;
|
|
case ANSIColor::BrightRed: return VGAColor::BrightRed;
|
|
case ANSIColor::BrightGreen: return VGAColor::BrightGreen;
|
|
case ANSIColor::Yellow: return VGAColor::Yellow;
|
|
case ANSIColor::BrightBlue: return VGAColor::BrightBlue;
|
|
case ANSIColor::BrightMagenta: return VGAColor::BrightMagenta;
|
|
case ANSIColor::BrightCyan: return VGAColor::BrightCyan;
|
|
case ANSIColor::White: return VGAColor::White;
|
|
}
|
|
ASSERT_NOT_REACHED();
|
|
return VGAColor::LightGray;
|
|
}
|
|
|
|
static inline byte ansiColorToVGA(byte color)
|
|
{
|
|
return (byte)ansiColorToVGA((ANSIColor)color);
|
|
}
|
|
|
|
void VirtualConsole::escape$m(const Vector<unsigned>& params)
|
|
{
|
|
for (auto param : params) {
|
|
switch (param) {
|
|
case 0:
|
|
// Reset
|
|
m_currentAttribute = 0x07;
|
|
break;
|
|
case 1:
|
|
// Bold
|
|
m_currentAttribute |= 8;
|
|
break;
|
|
case 30:
|
|
case 31:
|
|
case 32:
|
|
case 33:
|
|
case 34:
|
|
case 35:
|
|
case 36:
|
|
case 37:
|
|
// Foreground color
|
|
m_currentAttribute &= ~0x7;
|
|
m_currentAttribute |= ansiColorToVGA(param - 30);
|
|
break;
|
|
case 40:
|
|
case 41:
|
|
case 42:
|
|
case 43:
|
|
case 44:
|
|
case 45:
|
|
case 46:
|
|
case 47:
|
|
// Background color
|
|
m_currentAttribute &= ~0x70;
|
|
m_currentAttribute |= ansiColorToVGA(param - 30) << 8;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void VirtualConsole::escape$s(const Vector<unsigned>&)
|
|
{
|
|
m_savedCursorRow = m_cursorRow;
|
|
m_savedCursorColumn = m_cursorColumn;
|
|
}
|
|
|
|
void VirtualConsole::escape$u(const Vector<unsigned>&)
|
|
{
|
|
setCursor(m_savedCursorRow, m_savedCursorColumn);
|
|
}
|
|
|
|
void VirtualConsole::escape$H(const Vector<unsigned>& params)
|
|
{
|
|
unsigned row = 1;
|
|
unsigned col = 1;
|
|
if (params.size() >= 1)
|
|
row = params[0];
|
|
if (params.size() >= 2)
|
|
col = params[1];
|
|
setCursor(row - 1, col - 1);
|
|
}
|
|
|
|
void VirtualConsole::escape$J(const Vector<unsigned>& params)
|
|
{
|
|
int mode = 0;
|
|
if (params.size() >= 1)
|
|
mode = params[0];
|
|
switch (mode) {
|
|
case 0:
|
|
// FIXME: Clear from cursor to end of screen.
|
|
notImplemented();
|
|
break;
|
|
case 1:
|
|
// FIXME: Clear from cursor to beginning of screen.
|
|
notImplemented();
|
|
break;
|
|
case 2:
|
|
vga_clear();
|
|
break;
|
|
case 3:
|
|
// FIXME: <esc>[3J should also clear the scrollback buffer.
|
|
vga_clear();
|
|
break;
|
|
}
|
|
}
|
|
|
|
void VirtualConsole::executeEscapeSequence(byte final)
|
|
{
|
|
auto paramparts = String((const char*)m_parameters.data(), m_parameters.size()).split(';');
|
|
Vector<unsigned> params;
|
|
for (auto& parampart : paramparts) {
|
|
bool ok;
|
|
unsigned value = parseUInt(parampart, ok);
|
|
if (!ok) {
|
|
// FIXME: Should we do something else?
|
|
return;
|
|
}
|
|
params.append(value);
|
|
}
|
|
switch (final) {
|
|
case 'H': escape$H(params); break;
|
|
case 'J': escape$J(params); break;
|
|
case 'm': escape$m(params); break;
|
|
case 's': escape$s(params); break;
|
|
case 'u': escape$u(params); break;
|
|
default: break;
|
|
}
|
|
|
|
m_parameters.clear();
|
|
m_intermediates.clear();
|
|
}
|
|
|
|
void VirtualConsole::scrollUp()
|
|
{
|
|
if (m_cursorRow == (m_rows - 1)) {
|
|
memcpy(m_buffer, m_buffer + 160, 160 * 24);
|
|
word* linemem = (word*)&m_buffer[24 * 160];
|
|
for (word i = 0; i < 80; ++i)
|
|
linemem[i] = 0x0720;
|
|
if (m_active)
|
|
vga_scroll_up();
|
|
} else {
|
|
++m_cursorRow;
|
|
}
|
|
m_cursorColumn = 0;
|
|
}
|
|
|
|
void VirtualConsole::setCursor(unsigned row, unsigned column)
|
|
{
|
|
m_cursorRow = row;
|
|
m_cursorColumn = column;
|
|
if (m_active)
|
|
vga_set_cursor(m_cursorRow, m_cursorColumn);
|
|
}
|
|
|
|
void VirtualConsole::putCharacterAt(unsigned row, unsigned column, byte ch)
|
|
{
|
|
word cur = (row * 160) + (column * 2);
|
|
m_buffer[cur] = ch;
|
|
m_buffer[cur + 1] = m_currentAttribute;
|
|
if (m_active)
|
|
vga_putch_at(row, column, ch, m_currentAttribute);
|
|
}
|
|
|
|
void VirtualConsole::onChar(byte ch, bool shouldEmit)
|
|
{
|
|
if (shouldEmit)
|
|
emit(ch);
|
|
|
|
switch (m_escState) {
|
|
case ExpectBracket:
|
|
if (ch == '[')
|
|
m_escState = ExpectParameter;
|
|
else
|
|
m_escState = Normal;
|
|
return;
|
|
case ExpectParameter:
|
|
if (isParameter(ch)) {
|
|
m_parameters.append(ch);
|
|
return;
|
|
}
|
|
m_escState = ExpectIntermediate;
|
|
// fall through
|
|
case ExpectIntermediate:
|
|
if (isIntermediate(ch)) {
|
|
m_intermediates.append(ch);
|
|
return;
|
|
}
|
|
m_escState = ExpectFinal;
|
|
// fall through
|
|
case ExpectFinal:
|
|
if (isFinal(ch)) {
|
|
m_escState = Normal;
|
|
executeEscapeSequence(ch);
|
|
return;
|
|
}
|
|
m_escState = Normal;
|
|
return;
|
|
case Normal:
|
|
break;
|
|
}
|
|
|
|
switch (ch) {
|
|
case '\0':
|
|
return;
|
|
case '\033':
|
|
m_escState = ExpectBracket;
|
|
return;
|
|
case 8: // Backspace
|
|
if (m_cursorColumn) {
|
|
setCursor(m_cursorRow, m_cursorColumn - 1);
|
|
putCharacterAt(m_cursorRow, m_cursorColumn, ' ');
|
|
return;
|
|
}
|
|
break;
|
|
case '\n':
|
|
scrollUp();
|
|
setCursor(m_cursorRow, m_cursorColumn);
|
|
return;
|
|
}
|
|
|
|
putCharacterAt(m_cursorRow, m_cursorColumn, ch);
|
|
|
|
++m_cursorColumn;
|
|
if (m_cursorColumn >= m_columns)
|
|
scrollUp();
|
|
setCursor(m_cursorRow, m_cursorColumn);
|
|
}
|
|
|
|
void VirtualConsole::onKeyPress(byte ch)
|
|
{
|
|
emit(ch);
|
|
}
|
|
|
|
void VirtualConsole::onConsoleReceive(byte ch)
|
|
{
|
|
auto old_attribute = m_currentAttribute;
|
|
m_currentAttribute = 0x03;
|
|
onChar(ch, false);
|
|
m_currentAttribute = old_attribute;
|
|
}
|
|
|
|
void VirtualConsole::onTTYWrite(byte ch)
|
|
{
|
|
onChar(ch, false);
|
|
}
|
|
|
|
String VirtualConsole::ttyName() const
|
|
{
|
|
char buf[8];
|
|
ksprintf(buf, "tty%u", m_index);
|
|
return String(buf);
|
|
}
|