From 50d89a314a394e5427d0791a08ffc3a3b79b6670 Mon Sep 17 00:00:00 2001 From: Jacob Potter Date: Mon, 6 Oct 2008 03:20:56 -0400 Subject: [PATCH] packet receive code; keyboard injection --- aseg/Makefile | 2 +- aseg/counter.c | 20 +++++- aseg/keyboard.c | 182 +++++++++++++++++++++++++++++++++++++++++++++++ aseg/keyboard.h | 10 +++ net/net.c | 21 +++++- tools/Makefile | 5 +- tools/frob-rls.c | 48 +++++++++++++ 7 files changed, 282 insertions(+), 6 deletions(-) create mode 100644 aseg/keyboard.c create mode 100644 aseg/keyboard.h create mode 100644 tools/frob-rls.c diff --git a/aseg/Makefile b/aseg/Makefile index 64b3c5a..c976b1e 100644 --- a/aseg/Makefile +++ b/aseg/Makefile @@ -4,7 +4,7 @@ CFLAGS=-I../include -I../include/raw -nostdlib -nostdinc -fno-builtin -D__RAW__ OBJS=counter.o firstrun.o ../pci/pci-raw.o ../lib/minilib.o ../lib/console.o \ ../ich2/smram-ich2.o ../ich2/smi.o vga-overlay.o packet.o \ ../lib/sprintf.o ../lib/doprnt.o ../pci/pci.o ../net/net.o \ - ../ich2/ich2-timer.o ../pci/pci-bother.o ../net/3c90x.o + ../ich2/ich2-timer.o ../pci/pci-bother.o ../net/3c90x.o keyboard.o all: aseg.elf diff --git a/aseg/counter.c b/aseg/counter.c index f5a2602..73003da 100644 --- a/aseg/counter.c +++ b/aseg/counter.c @@ -7,6 +7,7 @@ #include "../net/net.h" #include "vga-overlay.h" #include "packet.h" +#include "keyboard.h" unsigned int counter = 0; unsigned long pcisave; @@ -29,8 +30,21 @@ void pci_dump() { //dologf("READ: %08x (%02x)", cts, b); if ((cts & 0xFFFF) == 0x64) curdev = (b & 0x20) ? 1 : 0; - if ((curdev == 0) && ((cts & 0xFFFF) == 0x60) && (b == 0x01)) - outb(0xCF9, 0x4); + + if ((curdev == 0) && ((cts & 0xFFFF) == 0x60)) + { + /* This is a keyboard read. */ + if (b == 0x01) { + /* Reset. */ + outb(0xCF9, 0x4); + return; + } + + if (kbd_get_injected_scancode()) { + b = kbd_get_injected_scancode(); + } + } + *(unsigned char*)0xAFFD0 /* EAX */ = b; break; } @@ -86,6 +100,8 @@ void gbl_rls_handler(smi_event_t ev) if (packet->type == 42) { dump_log((char *)packet->data); *(unsigned long*)0xAFFD4 = 42; + } else if (packet->type == 0xAA) { + kbd_inject_key('A'); } else { *(unsigned long*)0xAFFD4 = 0x2BADD00D; } diff --git a/aseg/keyboard.c b/aseg/keyboard.c new file mode 100644 index 0000000..878fa33 --- /dev/null +++ b/aseg/keyboard.c @@ -0,0 +1,182 @@ +#include "keyboard.h" +#include + +static unsigned char kbd_inj_buffer[128]; +static int kbd_inj_start = 0; +static int kbd_inj_end = 0; +int kbd_mode = 2; + +static const char scancodes2[][8] = { + ['a'] = "\x1e\xf0\x1e", + ['b'] = "\x30\xf0\x30", + ['c'] = "\x2e\xf0\x2e", + ['d'] = "\x20\xf0\x20", + ['e'] = "\x12\xf0\x12", + ['f'] = "\x21\xf0\x21", + ['g'] = "\x22\xf0\x22", + ['h'] = "\x23\xf0\x23", + ['i'] = "\x17\xf0\x17", + ['j'] = "\x24\xf0\x24", + ['k'] = "\x25\xf0\x25", + ['l'] = "\x26\xf0\x26", + ['m'] = "\x32\xf0\x32", + ['n'] = "\x31\xf0\x31", + ['o'] = "\x18\xf0\x18", + ['p'] = "\x19\xf0\x19", + ['q'] = "\x10\xf0\x10", + ['r'] = "\x13\xf0\x13", + ['s'] = "\x1f\xf0\x1f", + ['t'] = "\x14\xf0\x14", + ['u'] = "\x16\xf0\x16", + ['v'] = "\x2f\xf0\x2f", + ['w'] = "\x11\xf0\x11", + ['x'] = "\x2d\xf0\x2d", + ['y'] = "\x15\xf0\x15", + ['z'] = "\x2c\xf0\x2c", + ['A'] = "\x12\x1e\xf0\x1e\xf0\x12", + ['B'] = "\x12\x30\xf0\x30\xf0\x12", + ['C'] = "\x12\x2e\xf0\x2e\xf0\x12", + ['D'] = "\x12\x20\xf0\x20\xf0\x12", + ['E'] = "\x12\x12\xf0\x12\xf0\x12", + ['F'] = "\x12\x21\xf0\x21\xf0\x12", + ['G'] = "\x12\x22\xf0\x22\xf0\x12", + ['H'] = "\x12\x23\xf0\x23\xf0\x12", + ['I'] = "\x12\x17\xf0\x17\xf0\x12", + ['J'] = "\x12\x24\xf0\x24\xf0\x12", + ['K'] = "\x12\x25\xf0\x25\xf0\x12", + ['L'] = "\x12\x26\xf0\x26\xf0\x12", + ['M'] = "\x12\x32\xf0\x32\xf0\x12", + ['N'] = "\x12\x31\xf0\x31\xf0\x12", + ['O'] = "\x12\x18\xf0\x18\xf0\x12", + ['P'] = "\x12\x19\xf0\x19\xf0\x12", + ['Q'] = "\x12\x10\xf0\x10\xf0\x12", + ['R'] = "\x12\x13\xf0\x13\xf0\x12", + ['S'] = "\x12\x1f\xf0\x1f\xf0\x12", + ['T'] = "\x12\x14\xf0\x14\xf0\x12", + ['U'] = "\x12\x16\xf0\x16\xf0\x12", + ['V'] = "\x12\x2f\xf0\x2f\xf0\x12", + ['W'] = "\x12\x11\xf0\x11\xf0\x12", + ['X'] = "\x12\x2d\xf0\x2d\xf0\x12", + ['Y'] = "\x12\x15\xf0\x15\xf0\x12", + ['Z'] = "\x12\x2c\xf0\x2c\xf0\x12", + ['`'] = "\x0e\xf0\x0e", + ['~'] = "\x12\x0e\xf0\x0e\xf0\x12", + ['1'] = "\x16\xf0\x16", + ['!'] = "\x12\x16\xf0\x16\xf0\x12", + ['2'] = "\x1e\xf0\x1e", + ['@'] = "\x12\x1e\xf0\x1e\xf0\x12", + ['3'] = "\x26\xf0\x26", + ['#'] = "\x12\x26\xf0\x26\xf0\x12", + ['4'] = "\x25\xf0\x25", + ['$'] = "\x12\x25\xf0\x25\xf0\x12", + ['5'] = "\x2e\xf0\x2e", + ['%'] = "\x12\x2e\xf0\x2e\xf0\x12", + ['6'] = "\x36\xf0\x36", + ['^'] = "\x12\x36\xf0\x36\xf0\x12", + ['7'] = "\x3d\xf0\x3d", + ['&'] = "\x12\x3d\xf0\x3d\xf0\x12", + ['8'] = "\x3e\xf0\x3e", + ['*'] = "\x12\x3e\xf0\x3e\xf0\x12", + ['9'] = "\x46\xf0\x46", + ['('] = "\x12\x46\xf0\x46\xf0\x12", + ['0'] = "\x45\xf0\x45", + [')'] = "\x12\x45\xf0\x45\xf0\x12", + ['-'] = "\x4e\xf0\x4e", + ['_'] = "\x12\x4e\xf0\x4e\xf0\x12", + ['='] = "\x55\xf0\x55", + ['+'] = "\x12\x55\xf0\x55\xf0\x12", + ['['] = "\x54\xf0\x54", + ['{'] = "\x12\x54\xf0\x54\xf0\x12", + [']'] = "\x5b\xf0\x5b", + ['}'] = "\x12\x5b\xf0\x5b\xf0\x12", + ['\\'] = "\x5d\xf0\x5d", + ['|'] = "\x12\x5d\xf0\x5d\xf0\x12", + [';'] = "\x4c\xf0\x4c", + [':'] = "\x12\x4c\xf0\x4c\xf0\x12", + ['\''] = "\x52\xf0\x52", + ['"'] = "\x12\x52\xf0\x52\xf0\x12", + [','] = "\x41\xf0\x41", + ['<'] = "\x12\x41\xf0\x41\xf0\x12", + ['.'] = "\x49\xf0\x49", + ['>'] = "\x12\x49\xf0\x49\xf0\x12", + ['/'] = "\x4a\xf0\x4a", + ['?'] = "\x12\x4a\xf0\x4a\xf0\x12", +}; + +const unsigned char convert_table[] = { + 0xff, 0x43, 0x41, 0x3f, 0x3d, 0x3b, 0x3c, 0x58, 0x64, 0x44, 0x42, 0x40, 0x3e, 0x0f, 0x29, 0x59, + 0x65, 0x38, 0x2a, 0x70, 0x1d, 0x10, 0x02, 0x5a, 0x66, 0x71, 0x2c, 0x1f, 0x1e, 0x11, 0x03, 0x5b, + 0x67, 0x2e, 0x2d, 0x20, 0x12, 0x05, 0x04, 0x5c, 0x68, 0x39, 0x2f, 0x21, 0x14, 0x13, 0x06, 0x5d, + 0x69, 0x31, 0x30, 0x23, 0x22, 0x15, 0x07, 0x5e, 0x6a, 0x72, 0x32, 0x24, 0x16, 0x08, 0x09, 0x5f, + 0x6b, 0x33, 0x25, 0x17, 0x18, 0x0b, 0x0a, 0x60, 0x6c, 0x34, 0x35, 0x26, 0x27, 0x19, 0x0c, 0x61, + 0x6d, 0x73, 0x28, 0x74, 0x1a, 0x0d, 0x62, 0x6e, 0x3a, 0x36, 0x1c, 0x1b, 0x75, 0x2b, 0x63, 0x76, + 0x55, 0x56, 0x77, 0x78, 0x79, 0x7a, 0x0e, 0x7b, 0x7c, 0x4f, 0x7d, 0x4b, 0x47, 0x7e, 0x7f, 0x6f, + 0x52, 0x53, 0x50, 0x4c, 0x4d, 0x48, 0x01, 0x45, 0x57, 0x4e, 0x51, 0x4a, 0x37, 0x49, 0x46, 0x54, + 0x80, 0x81, 0x82, 0x41, 0x54, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f, + 0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f, + 0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf, + 0xb0, 0xb1, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf, + 0xc0, 0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf, + 0xd0, 0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf, + 0xe0, 0xe1, 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef, + 0x00, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff +}; + +unsigned char sc_convert_1(unsigned char in) { + static int shifted = 0; + + if (shifted) { + shifted = 0; + return convert_table[in] | 0x80; + } + + if (in == 0xF0) { + shifted = 1; + return 0; + } else { + return convert_table[in]; + } +} + +void kbd_inject_scancode (unsigned char sc) { + outputf("Buffering %02x", sc); + kbd_inj_buffer[kbd_inj_end] = sc; + kbd_inj_end += 1; + kbd_inj_end %= sizeof(kbd_inj_buffer); +} + +void kbd_inject_key(unsigned char k) { + + if (kbd_mode == 1) { + const char * c = scancodes2[k]; + if (!c) return; + while (*c) { + char cconv = sc_convert_1(*c); + if (cconv) kbd_inject_scancode(cconv); + c++; + } + } else { + const char * c = scancodes2[k]; + if (!c) return; + while (*c) { + kbd_inject_scancode(*c); + c++; + } + } +} + +unsigned char kbd_get_injected_scancode() +{ + unsigned char b; + + if (kbd_inj_end != kbd_inj_start) { + b = kbd_inj_buffer[kbd_inj_start]; + kbd_inj_start += 1; + kbd_inj_start %= sizeof(kbd_inj_buffer); + outputf("Injecting %02x", b); + return b; + } else { + outputf("Not injecting"); + return 0; + } +} diff --git a/aseg/keyboard.h b/aseg/keyboard.h new file mode 100644 index 0000000..684ce45 --- /dev/null +++ b/aseg/keyboard.h @@ -0,0 +1,10 @@ +#ifndef KEYBOARD_H +#define KEYBOARD_H + +void kbd_inject_key(unsigned char k); + +unsigned char kbd_get_injected_scancode(); + +extern int kbd_mode; + +#endif diff --git a/net/net.c b/net/net.c index 98571b0..7312fde 100644 --- a/net/net.c +++ b/net/net.c @@ -12,6 +12,15 @@ extern struct pci_driver a3c90x_driver; static char test[1024] = {0}; +static char packet[4096] = {0}; + +typedef struct packet_t { + char from[6]; + char to[6]; + unsigned short ethertype; + char data[]; +} packet_t; + static unsigned char vga_read(unsigned char idx) { outb(CRTC_IDX_REG, idx); @@ -36,7 +45,17 @@ void eth_poll() if (!_nic) return; - _nic->poll(_nic, 0); + if (_nic->poll(_nic, 0)) { + _nic->packet = packet; + _nic->poll(_nic, 1); + + packet_t * p = (packet_t *) packet; + + outputf("NIC: Packet: %d 0x%x", _nic->packetlen, p->ethertype); + if (p->ethertype == 0x3813) { + outputf("NIC: Command: 0x%x", *((uint16_t *)p->data)); + } + } smram_tseg_set_state(SMRAM_TSEG_OPEN); old_state = smram_save_state(); diff --git a/tools/Makefile b/tools/Makefile index 3bd46a5..4b0978b 100644 --- a/tools/Makefile +++ b/tools/Makefile @@ -3,6 +3,7 @@ CC=gcc SMRAM_ICH2_OBJS=smram-linux-tool.o ../pci/pci-linux.o ../ich2/smram-ich2.noraw.o PCI_OBJS=pci.o ../pci/pci-linux.o POKE_RLS_OBJS=poke-rls.o poke-rls-asm.o ../pci/pci-linux.o +FROB_RLS_OBJS=frob-rls.o poke-rls-asm.o ../pci/pci-linux.o all: smram-ich2 port pci poke-rls @@ -18,8 +19,8 @@ port: port.o pci: $(PCI_OBJS) gcc $(CFLAGS) -o pci $(PCI_OBJS) -poke-rls: $(POKE_RLS_OBJS) - gcc $(CFLAGS) -o poke-rls $(POKE_RLS_OBJS) +frob-rls: $(FROB_RLS_OBJS) + gcc $(CFLAGS) -o frob-rls $(FROB_RLS_OBJS) clean: rm -f $(SMRAM_ICH2_OBJS) smram-ich2 diff --git a/tools/frob-rls.c b/tools/frob-rls.c new file mode 100644 index 0000000..3ff4e53 --- /dev/null +++ b/tools/frob-rls.c @@ -0,0 +1,48 @@ +#include +#include +#include +#include +#include + +static uint16_t _get_PMBASE() +{ + static long pmbase = -1; + + if (pmbase == -1) + pmbase = pci_read32(ICH2_LPC_BUS, ICH2_LPC_DEV, ICH2_LPC_FN, ICH2_LPC_PCI_PMBASE) & ICH2_PMBASE_MASK; + + return pmbase; +} + +typedef struct { + uint32_t signature; + uint32_t type; + uint8_t data[]; +} packet_t; + +extern unsigned int poke(unsigned long addr, unsigned long * value); + +int main(int argc, char **argv) +{ + unsigned int res; + packet_t * packet = (packet_t *)memalign(4096, sizeof(packet_t)); + + packet->signature = 0x1BADD00D; + packet->type = 0xAA; + strcpy(packet->data, "hello, world!"); + + if (iopl(3) < 0) + { + perror("iopl"); + return 1; + } + + res = poke(_get_PMBASE() + 0x04, (void *)packet); + printf("returned %p\n", res); + + if (res == 42) { + int i; + for (i = 0; i < 96; i++) + printf("%s\n", packet->data + i * 41); + } +} -- 2.39.2