more reliable keyboard injection

[netwatch.git] / net / net.c

diff --git a/net/net.c b/net/net.c
index 7312fdec89426630cff34b898f5b1a764a915d05..e1c74446162042bd5aa788f6782e43b4ba6ac045 100644 (file)
--- a/net/net.c
+++ b/net/net.c
@@ -4,7 +4,9 @@
 #include <smram.h>
 #include <pci-bother.h>
 #include <output.h>
+#include <minilib.h>
 #include "net.h"
+#include "../aseg/keyboard.h"
 
 static struct nic *_nic = 0x0;
 
@@ -18,6 +20,8 @@ typedef struct packet_t {
        char from[6];
        char to[6];
        unsigned short ethertype;
+       unsigned short datalen;
+       unsigned char command;
        char data[];
 } packet_t;
 
@@ -33,6 +37,20 @@ static unsigned int vga_base()
             + (((unsigned int) vga_read(CRTC_START_ADDR_LSB_IDX)) << 1);
 }
 
+void handle_command(packet_t * p)
+{
+       uint16_t dl = htons(p->datalen);
+       int i;
+
+       outputf("NIC: Command: 0x%x, %d bytes", p->command, dl);
+
+       if (p->command == 0x42)
+       {
+               for (i = 0; i < dl; i++)
+                       kbd_inject_key(p->data[i]);
+       }
+}
+
 void eth_poll()
 {
        int i;
@@ -51,9 +69,13 @@ void eth_poll()
 
                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));
+               outputf("NIC: Packet: %d 0x%x", _nic->packetlen, htons(p->ethertype));
+               if (htons(p->ethertype) == 0x1338) {
+                       if (htons(p->datalen) + sizeof(packet_t) > _nic->packetlen) {
+                               outputf("NIC: Malformed packet");
+                       } else {
+                               handle_command(p);
+                       }
                }
        }
        smram_tseg_set_state(SMRAM_TSEG_OPEN);