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_UNKNOWN,
  57 } TRP_CONNECTION_STATUS;
  58
  59 typedef struct trp_connection TRP_CONNECTION;
  60 struct trp_connection {
  61   pthread_mutex_t mutex; /* protects status attribute */
  62   TRP_CONNECTION *next;
  63   pthread_t *thread; /* thread servicing this connection */
  64   int fd;
  65   TR_NAME *gssname;
  66   TR_NAME *peer;
  67   gss_ctx_id_t *gssctx;
  68   TRP_CONNECTION_STATUS status;
  69   void (*status_change_cb)(TRP_CONNECTION *conn, void *cookie);
  70   void *status_change_cookie;
  71 };
  72
  73 typedef TRP_RC (*TRPS_MSG_FUNC)(TRPS_INSTANCE *, TRP_CONNECTION *, TR_MSG *);
  74 typedef void (*TRP_RESP_FUNC)();
  75 /*typedef int (*TRP_AUTH_FUNC)(gss_name_t client_name, TR_NAME *display_name, void *cookie);*/
  76 typedef client_cb_fn TRP_AUTH_FUNC;
  77
  78 /* TRP Client Instance Data */
  79 typedef struct trpc_instance TRPC_INSTANCE;
  80 struct trpc_instance {
  81   TRPC_INSTANCE *next;
  82   TR_NAME *gssname;
  83   char *server;
  84   unsigned int port;
  85   TRP_CONNECTION *conn;
  86   TR_MQ *mq; /* msgs from master to trpc */
  87   DH *dh;                       /* Client's DH struct with priv and pub keys */
  88 };
  89
  90 /* TRP Server Instance Data */
  91 struct trps_instance {
  92   char *hostname;
  93   unsigned int port;
  94   TRP_AUTH_FUNC auth_handler;
  95   TRPS_MSG_FUNC msg_handler;
  96   void *cookie;
  97   TRP_CONNECTION *conn; /* connections from peers */
  98   TRPC_INSTANCE *trpc; /* connections to peers */
  99   TR_MQ *mq; /* incoming message queue */
 100   TRP_PTABLE *ptable; /* peer table */
 101   TRP_RTABLE *rtable; /* route table */
 102   struct timeval connect_interval; /* interval between connection refreshes */
 103   struct timeval update_interval; /* interval between scheduled updates */
 104   struct timeval sweep_interval; /* interval between route table sweeps */
 105 };
 106
 107
 108 TRP_CONNECTION *trp_connection_new(TALLOC_CTX *mem_ctx);
 109 void trp_connection_free(TRP_CONNECTION *conn);
 110 void trp_connection_close(TRP_CONNECTION *conn);
 111 int trp_connection_lock(TRP_CONNECTION *conn);
 112 int trp_connection_unlock(TRP_CONNECTION *conn);
 113 int trp_connection_get_fd(TRP_CONNECTION *conn);
 114 void trp_connection_set_fd(TRP_CONNECTION *conn, int fd);
 115 TR_NAME *trp_connection_get_peer(TRP_CONNECTION *conn);
 116 TR_NAME *trp_connection_get_gssname(TRP_CONNECTION *conn);
 117 void trp_connection_set_gssname(TRP_CONNECTION *conn, TR_NAME *gssname);
 118 gss_ctx_id_t *trp_connection_get_gssctx(TRP_CONNECTION *conn);
 119 void trp_connection_set_gssctx(TRP_CONNECTION *conn, gss_ctx_id_t *gssctx);
 120 TRP_CONNECTION_STATUS trp_connection_get_status(TRP_CONNECTION *conn);
 121 pthread_t *trp_connection_get_thread(TRP_CONNECTION *conn);
 122 void trp_connection_set_thread(TRP_CONNECTION *conn, pthread_t *thread);
 123 TRP_CONNECTION *trp_connection_get_next(TRP_CONNECTION *conn);
 124 TRP_CONNECTION *trp_connection_remove(TRP_CONNECTION *conn, TRP_CONNECTION *remove);
 125 void trp_connection_append(TRP_CONNECTION *conn, TRP_CONNECTION *new);
 126 int trp_connection_auth(TRP_CONNECTION *conn, TRP_AUTH_FUNC auth_callback, void *callback_data);
 127 TRP_CONNECTION *trp_connection_accept(TALLOC_CTX *mem_ctx, int listen, TR_NAME *gssname);
 128 TRP_RC trp_connection_initiate(TRP_CONNECTION *conn, char *server, unsigned int port);
 129
 130 TRPC_INSTANCE *trpc_new (TALLOC_CTX *mem_ctx);
 131 void trpc_free (TRPC_INSTANCE *trpc);
 132 TRP_CONNECTION *trpc_get_conn(TRPC_INSTANCE *trpc);
 133 void trpc_set_conn(TRPC_INSTANCE *trpc, TRP_CONNECTION *conn);
 134 TRPC_INSTANCE *trpc_get_next(TRPC_INSTANCE *trpc);
 135 void trpc_set_next(TRPC_INSTANCE *trpc, TRPC_INSTANCE *next);
 136 TRPC_INSTANCE *trpc_remove(TRPC_INSTANCE *trpc, TRPC_INSTANCE *remove);
 137 void trpc_append(TRPC_INSTANCE *trpc, TRPC_INSTANCE *new);
 138 char *trpc_get_server(TRPC_INSTANCE *trpc);
 139 void trpc_set_server(TRPC_INSTANCE *trpc, char *server);
 140 TR_NAME *trpc_get_gssname(TRPC_INSTANCE *trpc);
 141 void trpc_set_gssname(TRPC_INSTANCE *trpc, TR_NAME *gssname);
 142 unsigned int trpc_get_port(TRPC_INSTANCE *trpc);
 143 void trpc_set_port(TRPC_INSTANCE *trpc, unsigned int port);
 144 DH *trpc_get_dh(TRPC_INSTANCE *trpc);
 145 void trpc_set_dh(TRPC_INSTANCE *trpc, DH *dh);
 146 TRP_CONNECTION_STATUS trpc_get_status(TRPC_INSTANCE *trpc);
 147 TR_MQ *trpc_get_mq(TRPC_INSTANCE *trpc);
 148 void trpc_set_mq(TRPC_INSTANCE *trpc, TR_MQ *mq);
 149 void trpc_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
 150 TR_MQ_MSG *trpc_mq_pop(TRPC_INSTANCE *trpc);
 151 void trpc_master_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
 152 TR_MQ_MSG *trpc_master_mq_pop(TRPC_INSTANCE *trpc);
 153 TRP_RC trpc_connect(TRPC_INSTANCE *trpc);
 154 TRP_RC trpc_send_msg(TRPC_INSTANCE *trpc, const char *msg_content);
 155
 156 TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx);
 157 void trps_free (TRPS_INSTANCE *trps);
 158 TRP_RC trps_init_rtable(TRPS_INSTANCE *trps);
 159 void trps_clear_rtable(TRPS_INSTANCE *trps);
 160 void trps_set_connect_interval(TRPS_INSTANCE *trps, unsigned int interval);
 161 unsigned int trps_get_connect_interval(TRPS_INSTANCE *trps);
 162 void trps_set_update_interval(TRPS_INSTANCE *trps, unsigned int interval);
 163 unsigned int trps_get_update_interval(TRPS_INSTANCE *trps);
 164 void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval);
 165 unsigned int trps_get_sweep_interval(TRPS_INSTANCE *trps);
 166 TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer);
 167 TRP_RC trps_send_msg (TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg);
 168 void trps_add_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *new);
 169 void trps_remove_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *remove);
 170 void trps_add_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *trpc);
 171 void trps_remove_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *remove);
 172 int trps_get_listener(TRPS_INSTANCE *trps,
 173                       TRPS_MSG_FUNC msg_handler,
 174                       TRP_AUTH_FUNC auth_handler,
 175                       const char *hostname,
 176                       unsigned int port,
 177                       void *cookie);
 178 TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps);
 179 void trps_mq_append(TRPS_INSTANCE *trps, TR_MQ_MSG *msg);
 180 void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn);
 181 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps);
 182 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg);
 183 TRP_RENTRY *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer);
 184 TRP_RENTRY *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
 185 TR_NAME *trps_get_next_hop(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
 186 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps);
 187 TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_RENTRY *route);
 188 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer);
 189 TRP_PEER *trps_get_peer(TRPS_INSTANCE *trps, TR_NAME *gssname);
 190 TRP_RC trps_scheduled_update(TRPS_INSTANCE *trps);
 191 int trps_peer_connected(TRPS_INSTANCE *trps, TRP_PEER *peer);
 192 #endif /* TRP_INTERNAL_H */