Fix memory freeing bugs. Seems stable, even through loss of connections.
[trust_router.git] / 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 <tr_mq.h>
9 #include <tr_msg.h>
10 #include <trp_ptable.h>
11 #include <trp_rtable.h>
12 #include <trust_router/trp.h>
13
14 /* info records */
15 /* TRP update record types */
16 typedef struct trp_inforec_route {
17   TR_NAME *comm;
18   TR_NAME *realm;
19   TR_NAME *trust_router;
20   TR_NAME *next_hop;
21   unsigned int metric;
22   unsigned int interval;
23 } TRP_INFOREC_ROUTE;
24
25 /* TODO: define struct trp_msg_info_community */
26
27 typedef union trp_inforec_data {
28   TRP_INFOREC_ROUTE *route;
29   /* TRP_INFOREC_COMM *comm; */
30 } TRP_INFOREC_DATA;
31
32 struct trp_inforec {
33   TRP_INFOREC *next;
34   TRP_INFOREC_TYPE type;
35   TRP_INFOREC_DATA data; /* contains pointer to one of the record types */
36 };
37
38 struct trp_update {
39   TRP_INFOREC *records;
40   TR_NAME *peer; /* who did this update come from? */
41 };
42
43 struct trp_req {
44   TR_NAME *comm;
45   TR_NAME *realm;
46   TR_NAME *peer; /* who did this req come from? */
47 };
48
49
50 typedef struct trps_instance TRPS_INSTANCE;
51
52 typedef enum trp_connection_status {
53   TRP_CONNECTION_CLOSED=0,
54   TRP_CONNECTION_DOWN,  
55   TRP_CONNECTION_AUTHORIZING,  
56   TRP_CONNECTION_UP,
57   TRP_CONNECTION_UNKNOWN,
58 } TRP_CONNECTION_STATUS;
59
60 typedef struct trp_connection TRP_CONNECTION;
61 struct trp_connection {
62   pthread_mutex_t mutex; /* protects status attribute */
63   TRP_CONNECTION *next;
64   pthread_t *thread; /* thread servicing this connection */
65   int fd;
66   TR_NAME *gssname;
67   TR_NAME *peer; /* TODO: why is there a peer and a gssname? jlr */
68   gss_ctx_id_t *gssctx;
69   TRP_CONNECTION_STATUS status;
70   void (*status_change_cb)(TRP_CONNECTION *conn, void *cookie);
71   void *status_change_cookie;
72 };
73
74 typedef TRP_RC (*TRPS_MSG_FUNC)(TRPS_INSTANCE *, TRP_CONNECTION *, TR_MSG *);
75 typedef void (*TRP_RESP_FUNC)();
76 /*typedef int (*TRP_AUTH_FUNC)(gss_name_t client_name, TR_NAME *display_name, void *cookie);*/
77 typedef client_cb_fn TRP_AUTH_FUNC;
78
79 /* TRP Client Instance Data */
80 typedef struct trpc_instance TRPC_INSTANCE;
81 struct trpc_instance {
82   TRPC_INSTANCE *next;
83   TR_NAME *gssname;
84   char *server;
85   unsigned int port;
86   TRP_CONNECTION *conn;
87   TR_MQ *mq; /* msgs from master to trpc */
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 typedef enum trp_update_type {
108   TRP_UPDATE_SCHEDULED=0,
109   TRP_UPDATE_TRIGGERED
110 } TRP_UPDATE_TYPE;
111
112 TRP_CONNECTION *trp_connection_new(TALLOC_CTX *mem_ctx);
113 void trp_connection_free(TRP_CONNECTION *conn);
114 void trp_connection_close(TRP_CONNECTION *conn);
115 int trp_connection_lock(TRP_CONNECTION *conn);
116 int trp_connection_unlock(TRP_CONNECTION *conn);
117 int trp_connection_get_fd(TRP_CONNECTION *conn);
118 void trp_connection_set_fd(TRP_CONNECTION *conn, int fd);
119 TR_NAME *trp_connection_get_peer(TRP_CONNECTION *conn);
120 TR_NAME *trp_connection_get_gssname(TRP_CONNECTION *conn);
121 void trp_connection_set_gssname(TRP_CONNECTION *conn, TR_NAME *gssname);
122 gss_ctx_id_t *trp_connection_get_gssctx(TRP_CONNECTION *conn);
123 void trp_connection_set_gssctx(TRP_CONNECTION *conn, gss_ctx_id_t *gssctx);
124 TRP_CONNECTION_STATUS trp_connection_get_status(TRP_CONNECTION *conn);
125 pthread_t *trp_connection_get_thread(TRP_CONNECTION *conn);
126 void trp_connection_set_thread(TRP_CONNECTION *conn, pthread_t *thread);
127 TRP_CONNECTION *trp_connection_get_next(TRP_CONNECTION *conn);
128 TRP_CONNECTION *trp_connection_remove(TRP_CONNECTION *conn, TRP_CONNECTION *remove);
129 void trp_connection_append(TRP_CONNECTION *conn, TRP_CONNECTION *new);
130 int trp_connection_auth(TRP_CONNECTION *conn, TRP_AUTH_FUNC auth_callback, void *callback_data);
131 TRP_CONNECTION *trp_connection_accept(TALLOC_CTX *mem_ctx, int listen, TR_NAME *gssname);
132 TRP_RC trp_connection_initiate(TRP_CONNECTION *conn, char *server, unsigned int port);
133
134 TRPC_INSTANCE *trpc_new (TALLOC_CTX *mem_ctx);
135 void trpc_free (TRPC_INSTANCE *trpc);
136 TRP_CONNECTION *trpc_get_conn(TRPC_INSTANCE *trpc);
137 void trpc_set_conn(TRPC_INSTANCE *trpc, TRP_CONNECTION *conn);
138 TRPC_INSTANCE *trpc_get_next(TRPC_INSTANCE *trpc);
139 void trpc_set_next(TRPC_INSTANCE *trpc, TRPC_INSTANCE *next);
140 TRPC_INSTANCE *trpc_remove(TRPC_INSTANCE *trpc, TRPC_INSTANCE *remove);
141 void trpc_append(TRPC_INSTANCE *trpc, TRPC_INSTANCE *new);
142 char *trpc_get_server(TRPC_INSTANCE *trpc);
143 void trpc_set_server(TRPC_INSTANCE *trpc, char *server);
144 TR_NAME *trpc_get_gssname(TRPC_INSTANCE *trpc);
145 void trpc_set_gssname(TRPC_INSTANCE *trpc, TR_NAME *gssname);
146 unsigned int trpc_get_port(TRPC_INSTANCE *trpc);
147 void trpc_set_port(TRPC_INSTANCE *trpc, unsigned int port);
148 TRP_CONNECTION_STATUS trpc_get_status(TRPC_INSTANCE *trpc);
149 TR_MQ *trpc_get_mq(TRPC_INSTANCE *trpc);
150 void trpc_set_mq(TRPC_INSTANCE *trpc, TR_MQ *mq);
151 void trpc_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
152 TR_MQ_MSG *trpc_mq_pop(TRPC_INSTANCE *trpc);
153 void trpc_master_mq_append(TRPC_INSTANCE *trpc, TR_MQ_MSG *msg);
154 TR_MQ_MSG *trpc_master_mq_pop(TRPC_INSTANCE *trpc);
155 TRP_RC trpc_connect(TRPC_INSTANCE *trpc);
156 TRP_RC trpc_send_msg(TRPC_INSTANCE *trpc, const char *msg_content);
157
158 TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx);
159 void trps_free (TRPS_INSTANCE *trps);
160 TRP_RC trps_init_rtable(TRPS_INSTANCE *trps);
161 void trps_clear_rtable(TRPS_INSTANCE *trps);
162 void trps_set_connect_interval(TRPS_INSTANCE *trps, unsigned int interval);
163 unsigned int trps_get_connect_interval(TRPS_INSTANCE *trps);
164 void trps_set_update_interval(TRPS_INSTANCE *trps, unsigned int interval);
165 unsigned int trps_get_update_interval(TRPS_INSTANCE *trps);
166 void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval);
167 unsigned int trps_get_sweep_interval(TRPS_INSTANCE *trps);
168 TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer);
169 TRP_RC trps_send_msg (TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg);
170 void trps_add_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *new);
171 void trps_remove_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *remove);
172 void trps_add_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *trpc);
173 void trps_remove_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *remove);
174 int trps_get_listener(TRPS_INSTANCE *trps,
175                       TRPS_MSG_FUNC msg_handler,
176                       TRP_AUTH_FUNC auth_handler,
177                       const char *hostname,
178                       unsigned int port,
179                       void *cookie);
180 TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps);
181 void trps_mq_append(TRPS_INSTANCE *trps, TR_MQ_MSG *msg);
182 void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn);
183 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps);
184 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg);
185 TRP_ROUTE *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer);
186 TRP_ROUTE *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
187 TR_NAME *trps_get_next_hop(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm);
188 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps);
189 TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_ROUTE *route);
190 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer);
191 TRP_PEER *trps_get_peer(TRPS_INSTANCE *trps, TR_NAME *gssname);
192 TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE type);
193 int trps_peer_connected(TRPS_INSTANCE *trps, TRP_PEER *peer);
194 #endif /* TRP_INTERNAL_H */