tidy up, make less awful

1 files changed, 176 insertions, 0 deletions

diff --git a/net_rx.c b/net_rx.c
new file mode 100644
index 0000000..a3d95e4
--- /dev/null
+++ b/net_rx.c
@@ -0,0 +1,176 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <termios.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+
+#include "util.h"
+#include "sia.h"
+
+
+
+
+
+void msg (char *account, char *event, char *ascii)
+{
+  printf ("%s %64s %s\n", account, event, ascii);
+}
+
+
+static int new_block (int fd, SIA_Block *b)
+{
+  static int have_ascii_messages = 0; /*SIA level 3 doesn't have ascii, SIA level 4 does */
+
+  static char account[SIA_MAX_DATA_LENGTH + 1];
+  static char event[SIA_MAX_DATA_LENGTH + 1];
+  static char ascii[SIA_MAX_DATA_LENGTH + 1];
+
+  unsigned len = sia_data_length (b);
+
+  if (sia_ack_if_needed (fd, b))
+    return -1;
+
+  switch (b->function) {
+  case SIA_FN_ACCOUNT_ID:
+    memcpy (account, b->data, len);
+    account[len] = 0;
+    break;
+
+  case SIA_FN_NEW_EVENT:
+  case SIA_FN_OLD_EVENT:
+    memcpy (event, b->data, len);
+    event[len] = 0;
+
+    if (!have_ascii_messages) msg (account, event, "");
+
+    break;
+
+  case SIA_FN_ASCII:
+    have_ascii_messages = 1;
+    memcpy (ascii, b->data, len);
+    ascii[len] = 0;
+
+    msg (account, event, ascii);
+    break;
+  }
+
+  return 0;
+}
+
+static void periodic_task (void)
+{
+}
+
+static int usage (const char *name)
+{
+  fprintf (stderr, "Usage:\n");
+  fprintf (stderr, "%s [ -p listen_port ]\n\n", name);
+  fprintf (stderr, "listen_port defaults to 10002\n");
+
+  return 1;
+}
+
+
+#define TICK 5
+
+int main (int argc, char *argv[])
+{
+  int port = 10002;
+  unsigned opt;
+  int fd;
+  int afd = -1;
+  SIA_Block b;
+  struct timeval tv = {0};
+
+  unsigned char buf[SIA_MAX_BLOCK_LENGTH];
+  unsigned ptr = 0;
+
+
+  while ((opt = getopt (argc, argv, "p:")) != -1) {
+    switch (opt) {
+    case 'p':
+      port = atoi (optarg);
+      break;
+
+    default: /* '?' */
+      return usage (argv[0]);
+    }
+  }
+
+  fd = open_tcp_server (port);
+
+  if (fd < 0) {
+    perror ("open socket");
+    return -1;
+  }
+
+  set_blocking (fd);
+
+
+  // Epically budget TCP server
+
+  tv.tv_sec = TICK;
+
+  for (;;) {
+    fd_set rfds;
+    int rc;
+
+    FD_ZERO (&rfds);
+    FD_SET (fd, &rfds);
+
+    if (afd > 0)
+      FD_SET (afd, &rfds);
+
+    rc = select (afd > fd ? afd + 1 : fd + 1, &rfds, NULL, NULL, &tv);
+
+    if (!rc) periodic_task();
+
+    if (!tv.tv_sec && !tv.tv_usec)
+      tv.tv_sec = TICK;
+
+    if (FD_ISSET (fd, &rfds)) {
+      if (afd > 0) close (afd);
+
+      afd = accept (fd, NULL, NULL);
+      ptr = 0;
+    }
+
+    if ((afd > 0) && (FD_ISSET (afd, &rfds))) {
+
+      if (read (afd, &buf[ptr], 1) != 1) {
+        close (afd);
+        afd = -1;
+        ptr = 0;
+        continue;
+      }
+
+      ptr++;
+
+
+      switch (sia_parse (buf, ptr, &b)) {
+      case -1: /*Parse errror, form here is top drop connexion*/
+        close (afd);
+        afd = -1;
+        break;
+
+      case 0: /*More data needed*/
+        break;
+
+      default: /*Valid block */
+        new_block (afd, &b);
+        ptr = 0;
+        break;
+      }
+
+      if (ptr == SIA_MAX_BLOCK_LENGTH) {
+        close (afd);
+        afd = -1;
+      }
+
+    }
+
+  }
+}
+