Don't write to the text console if we don't need to.

[netwatch.git] / watcher / watcher.c

diff --git a/watcher/watcher.c b/watcher/watcher.c
index 8b1eef7ad1c3f53b352bdcb4ce822a0c6469552d..65d79965908165180d577fa2d2b8e8c8e1ebdbae 100644 (file)
--- a/watcher/watcher.c
+++ b/watcher/watcher.c
@@ -1,4 +1,8 @@
 #include <pcap.h>
+#include <ctype.h>
+#include <termios.h>
+#include <unistd.h>
+#include <fcntl.h>
 #include <arpa/inet.h>
 #include <stdlib.h>
 #include <string.h>
@@ -17,21 +21,27 @@ typedef struct netwatch_frame {
        char data[];
 } netwatch_frame;
 
+pcap_t * pcap;
+
 void out(char *buf, unsigned short start)
 {
   int y,x;
   buf += start;
-  printf("\x1B[0;0H");
+  printf("\x1B[H");
   for (y = 0; y < 25; y++)
   {
+    printf("\x1B[%d;%df",y+1,1);
     for (x = 0; x < 80; x++)
     {
       unsigned char c, col;
+      static unsigned char last = 0xFF;
       
       c = *(buf++);
       col = *(buf++);
-      
-      printf("\x1B[0;");
+
+      if (col != last)
+      {
+      printf("\x1B[0m\x1B[");
       switch (col & 0x7)
       {
       case 0: printf("30;"); break;
@@ -44,9 +54,9 @@ void out(char *buf, unsigned short start)
       case 7: printf("37;"); break;
       }
       if (col & 0x8)
-        printf("1;");
+        printf("m\x1B[1;");
       if (col & 0x80)
-        printf("5;");
+        printf("m\x1B[5;");
       switch ((col >> 4) & 0x7)
       {
       case 0: printf("40"); break;
@@ -58,18 +68,77 @@ void out(char *buf, unsigned short start)
       case 6: printf("43"); break;
       case 7: printf("47"); break;
       }
-      
-      printf("m%c", c);
+      printf("m");
+      }
+      last = col;
+      if (c == 0)
+       c = ' ';
+      printf("%c", c);
+      fflush(stdout);
     }
-    printf("\n");
   }
 }
 
+
+
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netpacket/packet.h>
+#include <net/ethernet.h>
+
+typedef struct frame {
+       char header[14];
+       unsigned short datalen;
+       unsigned char cmd;
+       char data[];
+} frame;
+
+int sendsock;
+
+void send_text(char *d, int len)
+{
+       int datalen = sizeof(frame) + len;
+       frame * f = alloca(datalen + 1);
+       struct sockaddr_ll addr;
+
+       memcpy(f->header, "\x00\xb0\xd0\x97\xbc\xac\x00\x03\x93\x87\x84\x8C\x13\x38", 14);
+       f->datalen = len;
+       f->cmd = 0x42;
+       memcpy(f->data, d, len);
+       
+       addr.sll_family = AF_PACKET;
+       addr.sll_ifindex = 2;
+       addr.sll_halen = 0;
+
+       if (sendto(sendsock, f, datalen, 0, (struct sockaddr *)&addr, sizeof(addr)) < datalen)
+               perror("sendto");
+}
+
+void send_reboot() {
+       frame f;
+       struct sockaddr_ll addr;
+
+       memcpy(f.header, "\x00\xb0\xd0\x97\xbc\xac\x00\x03\x93\x87\x84\x8C\x13\x38", 14);
+       f.datalen = 0;
+       f.cmd = 0xFE;
+       addr.sll_family = AF_PACKET;
+       addr.sll_ifindex = 2;
+       addr.sll_halen = 0;
+
+       if (sendto(sendsock, &f, sizeof(frame), 0, (struct sockaddr *)&addr, sizeof(addr)) < sizeof(frame))
+               perror("sendto");
+}
+
 void handler (u_char * user, const struct pcap_pkthdr *h, const u_char *bytes)
 {
        netwatch_frame * f = (netwatch_frame *) bytes;
+       static int i = 0;
 
-       int data_len = h->caplen - sizeof(netwatch_frame);
+       int data_len = h->caplen - sizeof(netwatch_frame) - 4;
 
        if (data_len < 0) return;
 
@@ -86,12 +155,41 @@ void handler (u_char * user, const struct pcap_pkthdr *h, const u_char *bytes)
                memcpy(buf + f->buf_offset, f->data, data_len);
        }
 
-       out(buf, f->buf_window);
+       int c = getchar();
+       if (c > 0) {
+               if (c == 3)
+                       pcap_breakloop(pcap);
+               if (c == '\r') c = '\n';
+               if (c == 127) c = '\b';
+               if (c == 0x1B)
+               {
+                       /* Escape */
+                       c = getchar();
+                       if (c == '[')
+                       {
+                               c = getchar();
+                               if (c == 'A')
+                                       c = 0x82;
+                               if (c == 'B')
+                                       c = 0x83;
+                               if (c == 'D')
+                                       c = 0x84;
+                               if (c == 'C')
+                                       c = 0x85;
+                               if (c == '4')
+                                       send_reboot();
+                       }
+               }
+
+               char ch = c;
+               send_text(&ch, 1);
+       }
+
+       if (!(i++ % 2))
+               out(buf, f->buf_window);
 }
 
 int main() {
-       pcap_t * pcap;
-
        pcap = pcap_open_live(NULL, 65535, 1, 0, errbuf);
  
        if (!pcap)
@@ -99,8 +197,23 @@ int main() {
                printf("pcap_open_live: %s\n", errbuf);
                exit(1);
        }
+       sendsock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
+       if(sendsock < 0)
+       {
+               perror("socket");
+               exit(1);
+       }
+       fcntl(0, F_SETFL, O_NONBLOCK);
+       printf("\x1b[H\x1b[J\x1b[5C");
+
+       struct termios t, t_orig;
+       tcgetattr(0, &t);
+       tcgetattr(0, &t_orig);
+       cfmakeraw(&t);
+       tcsetattr(0, 0, &t);
 
        pcap_loop(pcap, -1, handler, NULL);
 
+       tcsetattr(0, 0, &t_orig);
        return 0;
 }