hop-2012/server/main.c

/* Copyright (C) 2010, 2011 Tony Garnock-Jones. All rights reserved. */

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <signal.h>

#include <netinet/in.h>
#include <ucontext.h>

#include <assert.h>

typedef unsigned char u_char;
#include <event.h>

#include "cmsg_private.h"
#include "harness.h"
#include "net.h"
#include "ref.h"
#include "sexp.h"
#include "hashtable.h"
#include "node.h"
#include "queue.h"
#include "direct.h"
#include "fanout.h"
#include "meta.h"

#define WANT_CONSOLE_LISTENER 1

static void factory_handle_message(node_t *n, sexp_t *m) {
  size_t msglen = sexp_length(m);
  sexp_t *args;
  cmsg_bytes_t selector;

  if (msglen == 0 || !sexp_stringp(sexp_head(m))) {
    warn("Invalid message in factory\n");
    return;
  }

  selector = sexp_data(sexp_head(m));
  args = sexp_tail(m);

  if ((msglen == 5) && !cmsg_bytes_cmp(selector, cmsg_cstring_bytes("create"))) {
    sexp_t *classname = sexp_listref(args, 0);
    sexp_t *ctor_arg = sexp_listref(args, 1);
    sexp_t *reply_sink = sexp_listref(args, 2);
    sexp_t *reply_name = sexp_listref(args, 3);
    if (sexp_stringp(classname) && sexp_stringp(reply_sink) && sexp_stringp(reply_name)) {
      cmsg_bytes_t classname_bytes = sexp_data(classname);
      node_class_t *nc = lookup_node_class(classname_bytes);
      if (nc == NULL) {
	warn("Node class not found <<%.*s>>\n", classname_bytes.len, classname_bytes.bytes);
      } else {
	sexp_t *error = NULL;
	sexp_t *reply;
	if (new_node(nc, ctor_arg, &error) != NULL) {
	  reply = sexp_cons(sexp_cstring("create-ok"), NULL);
	} else {
	  reply = sexp_cons(sexp_cstring("create-failed"), sexp_cons(error, NULL));
	}
	post_node(sexp_data(reply_sink), sexp_data(reply_name), reply, sexp_empty_bytes);
      }
    }
    return;
  }

  warn("Message not understood in factory; selector <<%.*s>>, length %u\n",
       selector.len, selector.bytes,
       msglen);
}

static node_class_t factory_class = {
  .name = "factory",
  .extend = NULL,
  .destroy = NULL,
  .handle_message = factory_handle_message
};

static void init_factory(void) {
  bind_node(cmsg_cstring_bytes("factory"), new_node(&factory_class, NULL, NULL));
}

#if WANT_CONSOLE_LISTENER
static void console_listener(void *arg) {
  IOHandle *in_handle = new_iohandle(0);
  while (1) {
    cmsg_bytes_t buf = iohandle_readwait(in_handle, 1);
    if (buf.len == 0) break;
    iohandle_drain(in_handle, buf.len);
  }
  delete_iohandle(in_handle);
  interrupt_harness();
}
#endif

int main(int argc, char *argv[]) {
  info("cmsg ALPHA, Copyright (C) 2010, 2011 Tony Garnock-Jones. All rights reserved.\n");
  event_init();
  signal(SIGPIPE, SIG_IGN); /* avoid EPIPE when connections drop unexpectedly */
  info("Using libevent version %s\n", event_get_version());
  init_sexp();
  init_node(cmsg_cstring_bytes("server"));
  init_factory();
  init_queue();
  init_direct();
  init_fanout();
  init_meta();
#if WANT_CONSOLE_LISTENER
  spawn(console_listener, NULL);
#endif
  start_net(5671);
  boot_harness();
  done_sexp();
  return 0;
}