Send triggered updates (not really tested).
[trust_router.git] / tr / tr_main.c
index 5b2559f..f022fa7 100644 (file)
 #include <tr_cfgwatch.h>
 #include <tr_debug.h>
 
+#define TALLOC_DEBUG_ENABLE 1
+#define DEBUG_HARDCODED_PEER_TABLE 1
+#define DEBUG_PING_SELF 0
+
 /***** command-line option handling / setup *****/
 
 /* Strip trailing / from a path name.*/
@@ -106,15 +110,87 @@ static error_t parse_option(int key, char *arg, struct argp_state *state)
 static struct argp argp = {cmdline_options, parse_option, arg_doc, doc};
 
 
-int main (int argc, char *argv[])
+/***** talloc error handling *****/
+/* called when talloc tries to abort */
+static void tr_abort(const char *reason)
+{
+  tr_crit("tr_abort: Critical error, talloc aborted. Reason: %s", reason);
+  abort();
+}
+
+#if TALLOC_DEBUG_ENABLE
+static void tr_talloc_log(const char *msg)
 {
-  TALLOC_CTX *main_ctx=talloc_new(NULL);
+  tr_debug("talloc: %s", msg);
+}
+#endif /* TALLOC_DEBUG_ENABLE */
+
+
+#if DEBUG_PING_SELF
+struct thingy {
+  TRPS_INSTANCE *trps;
+  struct event *ev;
+};
+
+static void debug_ping(evutil_socket_t fd, short what, void *arg)
+{
+  TALLOC_CTX *tmp_ctx=talloc_new(NULL);
+  struct thingy *thingy=(struct thingy *)arg;
+  TRPS_INSTANCE *trps=thingy->trps;
+  TRP_REQ *req=NULL;
+  TR_MSG msg;
+  char *encoded=NULL;
+  struct timeval interval={1, 0};
+  static int count=10;
+  TR_NAME *name=NULL;
+
+  tr_debug("debug_ping entered");
+  if (trps->trpc==NULL)
+    tr_trpc_initiate(trps, trps->hostname, trps->port);
+
+  /* create a TRP route request msg */
+  req=trp_req_new(tmp_ctx);
+  name=tr_new_name("community");
+  trp_req_set_comm(req, name);
+  name=tr_new_name("realm");
+  trp_req_set_realm(req, name);
+  tr_msg_set_trp_req(&msg, req);
+  encoded=tr_msg_encode(&msg);
+  if (encoded==NULL)
+    tr_err("debug_ping: error encoding TRP message.");
+  else {
+    tr_debug("debug_ping: sending message");
+    trps_send_msg(trps, NULL, encoded);
+    tr_msg_free_encoded(encoded);
+  }
+  if (count-- > 0)
+    evtimer_add(thingy->ev, &interval);
+}
+#endif /* DEBUG_PING_SELF */
+
+int main(int argc, char *argv[])
+{
+  TALLOC_CTX *main_ctx=NULL;
 
   TR_INSTANCE *tr = NULL;
   struct cmdline_args opts;
   struct event_base *ev_base;
-  struct tr_socket_event tids_ev, trps_ev;
+  struct tr_socket_event tids_ev;
+  TR_TRPS_EVENTS *trps_ev;
   struct event *cfgwatch_ev;
+#if DEBUG_PING_SELF
+  struct event *debug_ping_ev;
+  struct timeval notime={0, 0};
+  struct thingy thingy={NULL};
+#endif /* DEBUG_PING_SELF */
+
+  /* we're going to be multithreaded, so disable null context tracking */
+  talloc_set_abort_fn(tr_abort);
+  talloc_disable_null_tracking();
+#if TALLOC_DEBUG_ENABLE
+  talloc_set_log_fn(tr_talloc_log);
+#endif /* TALLOC_DEBUG_ENABLE */
+  main_ctx=talloc_new(NULL);
 
   /* Use standalone logging */
   tr_log_open();
@@ -135,6 +211,18 @@ int main (int argc, char *argv[])
     return 1;
   }
 
+  /***** initialize the trust path query server instance *****/
+  if (NULL == (tr->tids = tids_create (tr))) {
+    tr_crit("Error initializing Trust Path Query Server instance.");
+    return 1;
+  }
+
+  /***** initialize the trust router protocol server instance *****/
+  if (NULL == (tr->trps = trps_new(tr))) {
+    tr_crit("Error initializing Trust Router Protocol Server instance.");
+    return 1;
+  }
+
   /***** process configuration *****/
   tr->cfgwatch=tr_cfgwatch_create(tr);
   if (tr->cfgwatch == NULL) {
@@ -143,23 +231,13 @@ int main (int argc, char *argv[])
   }
   tr->cfgwatch->config_dir=opts.config_dir;
   tr->cfgwatch->cfg_mgr=tr->cfg_mgr;
+  tr->cfgwatch->update_cb=tr_config_changed; /* handle configuration changes */
+  tr->cfgwatch->update_cookie=(void *)(tr->trps);
   if (0 != tr_read_and_apply_config(tr->cfgwatch)) {
     tr_crit("Error reading configuration, exiting.");
     return 1;
   }
 
-  /***** initialize the trust path query server instance *****/
-  if (0 == (tr->tids = tids_create (tr))) {
-    tr_crit("Error initializing Trust Path Query Server instance.");
-    return 1;
-  }
-
-  /***** initialize the trust router protocol server instance *****/
-  if (0 == (tr->trps = trps_create (tr))) {
-    tr_crit("Error initializing Trust Router Protocol Server instance.");
-    return 1;
-  }
-
   /***** Set up the event loop *****/
   ev_base=tr_event_loop_init(); /* Set up the event loop */
   if (ev_base==NULL) {
@@ -190,14 +268,57 @@ int main (int argc, char *argv[])
   }
 
   /* install TRP handler events */
+  trps_ev=tr_trps_events_new(main_ctx);
   if (0 != tr_trps_event_init(ev_base,
                               tr->trps,
                               tr->cfg_mgr,
-                             &trps_ev)) {
+                              trps_ev)) {
     tr_crit("Error initializing Trust Path Query Server instance.");
     return 1;
   }
 
+#if DEBUG_HARDCODED_PEER_TABLE
+  {
+    TRP_PEER *hc_peer=NULL;
+    char *s=NULL;
+
+    hc_peer=trp_peer_new(main_ctx); /* will later be stolen by ptable context */
+    if (hc_peer==NULL) {
+      tr_crit("Unable to allocate new peer. Aborting.");
+      return 1;
+    }
+    trp_peer_set_server(hc_peer, "epsilon.vmnet");
+    trp_peer_set_gssname(hc_peer, tr_new_name("tr-epsilon-vmnet@apc.painless-security.com"));
+    switch (tr->trps->port) {
+    case 10000:
+      trp_peer_set_port(hc_peer, 10001);
+      break;
+    case 10001:
+      trp_peer_set_port(hc_peer, 10000);
+      break;
+    default:
+      tr_crit("Cannot use hardcoded peer table with port other than 10000 or 10001.");
+      return 1;
+    }
+    if (TRP_SUCCESS != trps_add_peer(tr->trps, hc_peer)) {
+      tr_crit("Unable to add peer.");
+      return 1;
+    }
+
+    s=trp_ptable_to_str(main_ctx, tr->trps->ptable, NULL, NULL);
+    tr_debug("Peer Table:\n%s\n", s);
+    talloc_free(s);
+  }
+#endif /* DEBUG_HARDCODED_PEER_TABLE */
+
+#if DEBUG_PING_SELF
+  /* for debugging, send a message to peers on a timer */
+  debug_ping_ev=evtimer_new(ev_base, debug_ping, (void *)&thingy);
+  thingy.trps=tr->trps;
+  thingy.ev=debug_ping_ev;
+  evtimer_add(debug_ping_ev, &notime);
+#endif /* DEBUG_PING_SELF */
+
   tr_event_loop_run(ev_base); /* does not return until we are done */
 
   /* TODO: ensure talloc is properly used so this actually works */