include/trp_internal.h

   1 #ifndef TRP_INTERNAL_H
   2 #define TRP_INTERNAL_H
   3
   4 #include <pthread.h>
   5 #include <talloc.h>
   6
   7 #include <gsscon.h>
   8 #include <trust_router/tr_dh.h>
   9 #include <tr_mq.h>
  10 #include <tr_msg.h>
  11 #include <trp_ptable.h>
  12 #include <trp_rtable.h>
  13 #include <trust_router/trp.h>
  14
  15 /* info records */
  16 /* TRP update record types */
  17 typedef struct trp_inforec_route {
  18   TR_NAME *comm;
  19   TR_NAME *realm;
  20   TR_NAME *trust_router;
  21   TR_NAME *next_hop;
  22   unsigned int metric;
  23   unsigned int interval;
  24 } TRP_INFOREC_ROUTE;
  25
  26 /* TODO: define struct trp_msg_info_community */
  27
  28 typedef union trp_inforec_data {
  29   TRP_INFOREC_ROUTE *route;
  30   /* TRP_INFOREC_COMM *comm; */
  31 } TRP_INFOREC_DATA;
  32
  33 struct trp_inforec {
  34   TRP_INFOREC *next;
  35   TRP_INFOREC_TYPE type;
  36   TRP_INFOREC_DATA data; /* contains pointer to one of the record types */
  37 };
  38
  39 struct trp_update {
  40   TRP_INFOREC *records;
  41   TR_NAME *peer; /* who did this update come from? */
  42 };
  43
  44 struct trp_req {
  45   TR_NAME *comm;
  46   TR_NAME *realm;
  47   TR_NAME *peer; /* who did this req come from? */
  48 };
  49
  50
  51 typedef struct trps_instance TRPS_INSTANCE;
  52
  53 typedef enum trp_connection_status {
  54   TRP_CONNECTION_DOWN=0,
  55   TRP_CONNECTION_UP,
  56 } TRP_CONNECTION_STATUS;
  57
  58 typedef struct trp_connection TRP_CONNECTION;
  59 struct trp_connection {
  60   pthread_mutex_t mutex; /* protects status attribute */
  61   TRP_CONNECTION *next;
  62   pthread_t *thread; /* thread servicing this connection */
  63   int fd;
  64   TR_NAME *gssname;
  65   TR_NAME *peer;
  66   gss_ctx_id_t *gssctx;
  67   TRP_CONNECTION_STATUS status;
  68 };
  69
  70 typedef TRP_RC (*TRPS_MSG_FUNC)(TRPS_INSTANCE *, TRP_CONNECTION *, TR_MSG *);
  71 typedef void (*TRP_RESP_FUNC)();
  72 /*typedef int (*TRP_AUTH_FUNC)(gss_name_t client_name, TR_NAME *display_name, void *cookie);*/
  73 typedef client_cb_fn TRP_AUTH_FUNC;
  74
  75 /* TRP Client Instance Data */
  76 typedef struct trpc_instance TRPC_INSTANCE;
  77 struct trpc_instance {
  78   TRPC_INSTANCE *next;
  79   TR_NAME *gssname;
  80   char *server;
  81   unsigned int port;
  82   TRP_CONNECTION *conn;
  83   TR_MQ *mq; /* msgs from master to trpc */
  84   DH *dh;                       /* Client's DH struct with priv and pub keys */
  85 };
  86
  87 /* TRP Server Instance Data */
  88 struct trps_instance {
  89   char *hostname;
  90   unsigned int port;
  91   TRP_AUTH_FUNC auth_handler;
  92   TRPS_MSG_FUNC msg_handler;
  93   void *cookie;
  94   TRP_CONNECTION *conn; /* connections from peers */
  95   TRPC_INSTANCE *trpc; /* connections to peers */
  96   TR_MQ *mq; /* incoming message queue */
  97   TRP_PTABLE *ptable; /* peer table */
  98   TRP_RTABLE *rtable; /* route table */
  99   struct timeval connect_interval; /* interval between connection refreshes */
 100   struct timeval update_interval; /* interval between scheduled updates */
 101   struct timeval sweep_interval; /* interval between route table sweeps */
 102 };
 103
 104
 105 TRP_CONNECTION *trp_connection_new(TALLOC_CTX *mem_ctx);
 106 void trp_connection_free(TRP_CONNECTION *conn);
 107 void trp_connection_close(TRP_CONNECTION *conn);
 108 int trp_connection_lock(TRP_CONNECTION *conn);
 109 int trp_connection_unlock(TRP_CONNECTION *conn);
 110 int trp_connection_get_fd(TRP_CONNECTION *conn);
 111 void trp_connection_set_fd(TRP_CONNECTION *conn, int fd);
 112 TR_NAME *trp_connection_get_peer(TRP_CONNECTION *conn);
 113 TR_NAME *trp_connection_get_gssname(TRP_CONNECTION *conn);
 114 void trp_connection_set_gssname(TRP_CONNECTION *conn, TR_NAME *gssname);
 115 gss_ctx_id_t *trp_connection_get_gssctx(TRP_CONNECTION *conn);
 116 void trp_connection_set_gssctx(TRP_CONNECTION *conn, gss_ctx_id_t *gssctx);
 117 TRP_CONNECTION_STATUS trp_connection_get_status(TRP_CONNECTION *conn);
 118 pthread_t *trp_connection_get_thread(TRP_CONNECTION *conn);
 119 void trp_connection_set_thread(TRP_CONNECTION *conn, pthread_t *thread);
 120 TRP_CONNECTION *trp_connection_get_next(TRP_CONNECTION *conn);
 121 TRP_CONNECTION *trp_connection_remove(TRP_CONNECTION *conn, TRP_CONNECTION *remove);
 122 void trp_connection_append(TRP_CONNECTION *conn, TRP_CONNECTION *new);
 123 int trp_connection_auth(TRP_CONNECTION *conn, TRP_AUTH_FUNC auth_callback, void *callback_data);
 124 TRP_CONNECTION *trp_connection_accept(TALLOC_CTX *mem_ctx, int listen, TR_NAME *gssname);
 125 TRP_RC trp_connection_initiate(TRP_CONNECTION *conn, char *server, unsigned int port);
 126
 127 TRPC_INSTANCE *trpc_new (TALLOC_CTX *mem_ctx);
 128 void trpc_free (TRPC_INSTANCE *trpc);
 129 TRP_CONNECTION *trpc_get_conn(TRPC_INSTANCE *trpc);
 130 void trpc_set_conn(TRPC_INSTANCE *trpc, TRP_CONNECTION *conn);
 131 TRPC_INSTANCE *trpc_get_next(TRPC_INSTANCE *trpc);
 132 void trpc_set_next(TRPC_INSTANCE *trpc, TRPC_INSTANCE *next);
 133 TRPC_INSTANCE *trpc_remove(TRPC_INSTANCE *trpc, TRPC_INSTANCE *remove);
 134 void trpc_append(TRPC_INSTANCE *trpc, TRPC_INSTANCE *new);
 135 char *trpc_get_server(TRPC_INSTANCE *trpc);
 136 void trpc_set_server(TRPC_INSTANCE *trpc, char *server);
 137 TR_NAME *trpc_get_gssname(TRPC_INSTANCE *trpc);
 138 void trpc_set_gssname(TRPC_INSTANCE *trpc, TR_NAME *gssname);
 139 unsigned int trpc_get_port(TRPC_INSTANCE *trpc);
 140 void trpc_set_port(TRPC_INSTANCE *trpc, unsigned int port);
 141 DH *trpc_get_dh(TRPC_INSTANCE *trpc);
 142 void trpc_set_dh(TRPC_INSTANCE *trpc, DH *dh);
 143 TRP_CONNECTION_STATUS trpc_get_status(TRPC_INSTANCE *trpc);
 144 TR_MQ *trpc_get_mq(TRPC_INSTANCE *trpc);
 145 void trpc_set_mq(TRPC_INSTANCE *trpc, TR_MQ *mq);
 146 void trpc_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
 147 TR_MQ_MSG *trpc_mq_pop(TRPC_INSTANCE *trpc);
 148 void trpc_master_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
 149 TR_MQ_MSG *trpc_master_mq_pop(TRPC_INSTANCE *trpc);
 150 TRP_RC trpc_connect(TRPC_INSTANCE *trpc);
 151 TRP_RC trpc_send_msg(TRPC_INSTANCE *trpc, const char *msg_content);
 152
 153 TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx);
 154 void trps_free (TRPS_INSTANCE *trps);
 155 void trps_set_connect_interval(TRPS_INSTANCE *trps, unsigned int interval);
 156 unsigned int trps_get_connect_interval(TRPS_INSTANCE *trps);
 157 void trps_set_update_interval(TRPS_INSTANCE *trps, unsigned int interval);
 158 unsigned int trps_get_update_interval(TRPS_INSTANCE *trps);
 159 void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval);
 160 unsigned int trps_get_sweep_interval(TRPS_INSTANCE *trps);
 161 TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer);
 162 TRP_RC trps_send_msg (TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg);
 163 void trps_add_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *new);
 164 void trps_remove_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *remove);
 165 void trps_add_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *trpc);
 166 void trps_remove_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *remove);
 167 int trps_get_listener(TRPS_INSTANCE *trps,
 168                       TRPS_MSG_FUNC msg_handler,
 169                       TRP_AUTH_FUNC auth_handler,
 170                       const char *hostname,
 171                       unsigned int port,
 172                       void *cookie);
 173 TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps);
 174 void trps_mq_append(TRPS_INSTANCE *trps, TR_MQ_MSG *msg);
 175 void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn);
 176 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg);
 177 TRP_RENTRY *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer);
 178 TRP_RENTRY *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
 179 TR_NAME *trps_get_next_hop(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
 180 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps);
 181 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer);
 182 TRP_PEER *trps_get_peer(TRPS_INSTANCE *trps, TR_NAME *gssname);
 183 TRP_RC trps_scheduled_update(TRPS_INSTANCE *trps);
 184
 185 #endif /* TRP_INTERNAL_H */