1 /* Copyright 2010, 2011 NORDUnet A/S. All rights reserved.
2 See the file COPYING for licensing information. */
4 #if defined HAVE_CONFIG_H
10 #include <event2/event.h>
11 #include <event2/bufferevent.h>
12 #include <radsec/radsec.h>
13 #include <radsec/radsec-impl.h>
21 conn_close (struct rs_connection **connp)
26 r = rs_conn_destroy (*connp);
33 conn_user_dispatch_p (const struct rs_connection *conn)
37 return (conn->callbacks.connected_cb ||
38 conn->callbacks.disconnected_cb ||
39 conn->callbacks.received_cb ||
40 conn->callbacks.sent_cb);
44 rs_conn_create (struct rs_context *ctx, struct rs_connection **conn,
47 struct rs_connection *c;
49 c = (struct rs_connection *) malloc (sizeof(struct rs_connection));
51 return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__, NULL);
53 memset (c, 0, sizeof(struct rs_connection));
58 struct rs_realm *r = rs_conf_find_realm (ctx, config);
64 c->peers = r->peers; /* FIXME: Copy instead? */
65 for (p = c->peers; p; p = p->next)
67 c->timeout.tv_sec = r->timeout;
68 c->tryagain = r->retries;
72 c->realm = rs_malloc (ctx, sizeof (struct rs_realm));
74 return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__,
76 memset (c->realm, 0, sizeof (struct rs_realm));
86 rs_conn_set_type (struct rs_connection *conn, rs_conn_type_t type)
90 conn->realm->type = type;
94 rs_conn_add_listener (struct rs_connection *conn, rs_conn_type_t type,
95 const char *hostname, int port)
97 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
102 rs_conn_disconnect (struct rs_connection *conn)
108 err = evutil_closesocket (conn->fd);
114 rs_conn_destroy (struct rs_connection *conn)
120 /* NOTE: conn->realm is owned by context. */
121 /* NOTE: conn->peers is owned by context. */
123 if (conn->is_connected)
124 err = rs_conn_disconnect (conn);
126 event_free (conn->tev);
128 bufferevent_free (conn->bev);
130 event_base_free (conn->evb);
132 /* TODO: free tls_ctx */
133 /* TODO: free tls_ssl */
139 rs_conn_set_eventbase (struct rs_connection *conn, struct event_base *eb)
141 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
145 rs_conn_set_callbacks (struct rs_connection *conn, struct rs_conn_callbacks *cb)
148 memcpy (&conn->callbacks, cb, sizeof (conn->callbacks));
152 rs_conn_del_callbacks (struct rs_connection *conn)
155 memset (&conn->callbacks, 0, sizeof (conn->callbacks));
158 struct rs_conn_callbacks *
159 rs_conn_get_callbacks(struct rs_connection *conn)
162 return &conn->callbacks;
166 rs_conn_select_peer (struct rs_connection *conn, const char *name)
168 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
172 rs_conn_get_current_peer (struct rs_connection *conn, const char *name,
175 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
178 int rs_conn_fd (struct rs_connection *conn)
181 assert (conn->active_peer);
186 _rcb (struct rs_packet *packet, void *user_data)
188 struct rs_packet *pkt = (struct rs_packet *) user_data;
192 pkt->flags |= rs_packet_received_flag;
194 bufferevent_disable (pkt->conn->bev, EV_WRITE|EV_READ);
196 event_del (pkt->conn->rev);
199 /* Special function used in libradsec blocking dispatching mode,
200 i.e. with socket set to block on read/write and with no libradsec
201 callbacks registered.
203 For any other use of libradsec, a the received_cb callback should
204 be registered in the callbacks member of struct rs_connection.
206 On successful reception of a RADIUS message it will be verified
207 against REQ_MSG, if !NULL.
209 If PKT_OUT is !NULL it will upon return point at a pointer to a
210 struct rs_packet containing the message.
212 If anything goes wrong or if the read times out (TODO: explain),
213 PKT_OUT will not be changed and one or more errors are pushed on
214 the connection (available through rs_err_conn_pop()). */
216 rs_conn_receive_packet (struct rs_connection *conn,
217 struct rs_packet *req_msg,
218 struct rs_packet **pkt_out)
221 struct rs_packet *pkt = NULL;
224 assert (conn->realm);
225 assert (!conn_user_dispatch_p (conn)); /* Dispatching mode only. */
227 if (rs_packet_create (conn, &pkt))
232 assert (conn->active_peer);
233 assert (conn->fd >= 0);
235 conn->callbacks.received_cb = _rcb;
236 conn->user_data = pkt;
237 pkt->flags &= ~rs_packet_received_flag;
241 bufferevent_setwatermark (conn->bev, EV_READ, RS_HEADER_LEN, 0);
242 bufferevent_setcb (conn->bev, tcp_read_cb, NULL, tcp_event_cb, pkt);
243 bufferevent_enable (conn->bev, EV_READ);
247 err = event_add (conn->rev, NULL);
249 return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
251 evutil_gai_strerror (err));
255 rs_debug (("%s: entering event loop\n", __func__));
256 err = event_base_dispatch (conn->evb);
257 conn->callbacks.received_cb = NULL;
259 return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
260 "event_base_dispatch: %s",
261 evutil_gai_strerror (err));
262 rs_debug (("%s: event loop done\n", __func__));
264 if ((pkt->flags & rs_packet_received_flag) == 0
266 && packet_verify_response (pkt->conn, pkt, req_msg) != RSE_OK))
268 assert (rs_err_conn_peek_code (pkt->conn));
269 return rs_err_conn_peek_code (conn);
278 rs_conn_set_timeout(struct rs_connection *conn, struct timeval *tv)
286 conn_activate_timeout (struct rs_connection *conn)
291 if (conn->timeout.tv_sec || conn->timeout.tv_usec)
293 rs_debug (("%s: activating timer: %d.%d\n", __func__,
294 conn->timeout.tv_sec, conn->timeout.tv_usec));
295 if (evtimer_add (conn->tev, &conn->timeout))
296 return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
297 "evtimer_add: %d", errno);