1 /* Copyright 2010-2013 NORDUnet A/S. All rights reserved.
2 See LICENSE for licensing information. */
4 #if defined HAVE_CONFIG_H
12 #include <event2/event.h>
13 #include <event2/bufferevent.h>
14 #include <radsec/radsec.h>
15 #include <radsec/radsec-impl.h>
23 conn_user_dispatch_p (const struct rs_connection *conn)
27 return (conn->callbacks.connected_cb ||
28 conn->callbacks.disconnected_cb ||
29 conn->callbacks.received_cb ||
30 conn->callbacks.sent_cb);
35 conn_activate_timeout (struct rs_connection *conn)
40 if (conn->timeout.tv_sec || conn->timeout.tv_usec)
42 rs_debug (("%s: activating timer: %d.%d\n", __func__,
43 conn->timeout.tv_sec, conn->timeout.tv_usec));
44 if (evtimer_add (conn->tev, &conn->timeout))
45 return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
46 "evtimer_add: %d", errno);
52 conn_type_tls (const struct rs_connection *conn)
54 return conn->realm->type == RS_CONN_TYPE_TLS
55 || conn->realm->type == RS_CONN_TYPE_DTLS;
59 conn_cred_psk (const struct rs_connection *conn)
61 return conn->realm->transport_cred &&
62 conn->realm->transport_cred->type == RS_CRED_TLS_PSK;
66 /* Public functions. */
68 rs_conn_create (struct rs_context *ctx,
69 struct rs_connection **conn,
72 struct rs_connection *c;
74 c = (struct rs_connection *) malloc (sizeof(struct rs_connection));
76 return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__, NULL);
78 memset (c, 0, sizeof(struct rs_connection));
83 struct rs_realm *r = rs_conf_find_realm (ctx, config);
89 c->peers = r->peers; /* FIXME: Copy instead? */
90 for (p = c->peers; p; p = p->next)
92 c->timeout.tv_sec = r->timeout;
93 c->tryagain = r->retries;
97 c->realm = rs_malloc (ctx, sizeof (struct rs_realm));
99 return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__,
101 memset (c->realm, 0, sizeof (struct rs_realm));
111 rs_conn_set_type (struct rs_connection *conn, rs_conn_type_t type)
114 assert (conn->realm);
115 conn->realm->type = type;
119 rs_conn_add_listener (struct rs_connection *conn,
121 const char *hostname,
124 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
129 rs_conn_disconnect (struct rs_connection *conn)
135 if (conn->is_connected)
136 event_on_disconnect (conn);
140 bufferevent_free (conn->bev);
145 event_free (conn->rev);
150 event_free (conn->wev);
154 err = evutil_closesocket (conn->fd);
160 rs_conn_destroy (struct rs_connection *conn)
166 /* NOTE: conn->realm is owned by context. */
167 /* NOTE: conn->peers is owned by context. */
169 if (conn->is_connected)
170 err = rs_conn_disconnect (conn);
172 #if defined (RS_ENABLE_TLS)
173 if (conn->tls_ssl) /* FIXME: Free SSL strucxt in rs_conn_disconnect? */
174 SSL_free (conn->tls_ssl);
176 SSL_CTX_free (conn->tls_ctx);
180 event_free (conn->tev);
182 bufferevent_free (conn->bev);
184 event_free (conn->rev);
186 event_free (conn->wev);
188 event_base_free (conn->evb);
190 rs_free (conn->ctx, conn);
196 rs_conn_set_eventbase (struct rs_connection *conn, struct event_base *eb)
198 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
202 rs_conn_set_callbacks (struct rs_connection *conn, struct rs_conn_callbacks *cb)
205 memcpy (&conn->callbacks, cb, sizeof (conn->callbacks));
209 rs_conn_del_callbacks (struct rs_connection *conn)
212 memset (&conn->callbacks, 0, sizeof (conn->callbacks));
215 struct rs_conn_callbacks *
216 rs_conn_get_callbacks(struct rs_connection *conn)
219 return &conn->callbacks;
223 rs_conn_select_peer (struct rs_connection *conn, const char *name)
225 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
229 rs_conn_get_current_peer (struct rs_connection *conn,
233 return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__, NULL);
236 int rs_conn_fd (struct rs_connection *conn)
239 assert (conn->active_peer);
244 _rcb (struct rs_packet *packet, void *user_data)
246 struct rs_packet *pkt = (struct rs_packet *) user_data;
250 pkt->flags |= RS_PACKET_RECEIVED;
252 bufferevent_disable (pkt->conn->bev, EV_WRITE|EV_READ);
254 event_del (pkt->conn->rev);
258 rs_conn_receive_packet (struct rs_connection *conn,
259 struct rs_packet *req_msg,
260 struct rs_packet **pkt_out)
263 struct rs_packet *pkt = NULL;
266 assert (conn->realm);
267 assert (!conn_user_dispatch_p (conn)); /* Blocking mode only. */
269 if (rs_packet_create (conn, &pkt))
273 assert (conn->fd >= 0);
275 conn->callbacks.received_cb = _rcb;
276 conn->user_data = pkt;
277 pkt->flags &= ~RS_PACKET_RECEIVED;
279 if (conn->bev) /* TCP. */
281 bufferevent_setwatermark (conn->bev, EV_READ, RS_HEADER_LEN, 0);
282 bufferevent_setcb (conn->bev, tcp_read_cb, NULL, tcp_event_cb, pkt);
283 bufferevent_enable (conn->bev, EV_READ);
287 /* Put fresh packet in user_data for the callback and enable the
289 event_assign (conn->rev, conn->evb, event_get_fd (conn->rev),
290 EV_READ, event_get_callback (conn->rev), pkt);
291 err = event_add (conn->rev, NULL);
293 return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
295 evutil_gai_strerror (err));
297 /* Activate retransmission timer. */
298 conn_activate_timeout (pkt->conn);
301 rs_debug (("%s: entering event loop\n", __func__));
302 err = event_base_dispatch (conn->evb);
303 conn->callbacks.received_cb = NULL;
305 return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
306 "event_base_dispatch: %s",
307 evutil_gai_strerror (err));
308 rs_debug (("%s: event loop done\n", __func__));
310 if ((pkt->flags & RS_PACKET_RECEIVED) != 0)
312 /* If the caller passed a request, check the response. */
314 err = packet_verify_response (pkt->conn, pkt, req_msg);
316 /* If the response was OK and the caller wants it, hand it
317 over, else free it. */
318 if (err == RSE_OK && pkt_out)
321 rs_packet_destroy (pkt);
324 err = rs_err_conn_peek_code (pkt->conn);
330 rs_conn_set_timeout(struct rs_connection *conn, struct timeval *tv)