1 /* Copyright 2011 NORDUnet A/S. All rights reserved.
2 See the file COPYING for licensing information. */
4 #if defined HAVE_CONFIG_H
10 #include <sys/socket.h>
11 #include <event2/event.h>
12 #include <radsec/radsec.h>
13 #include <radsec/radsec-impl.h>
19 /* Send one packet, the first in queue. */
21 _send (struct rs_connection *conn, int fd)
24 struct rs_packet *pkt = conn->out_queue;
27 assert (pkt->rpkt->data);
30 r = compat_send (fd, pkt->rpkt->data, pkt->rpkt->data_len, 0);
33 int sockerr = evutil_socket_geterror (pkt->conn->fd);
34 if (sockerr != EAGAIN)
35 return rs_err_conn_push_fl (pkt->conn, RSE_SOCKERR, __FILE__, __LINE__,
36 "%d: send: %d (%s)", fd, sockerr,
37 evutil_socket_error_to_string (sockerr));
40 assert (r == pkt->rpkt->data_len);
41 /* Unlink the packet. */
42 conn->out_queue = pkt->next;
44 /* If there are more packets in queue, add the write event again. */
45 if (pkt->conn->out_queue)
47 r = event_add (pkt->conn->wev, NULL);
49 return rs_err_conn_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
50 "event_add: %s", evutil_gai_strerror (r));
51 rs_debug (("%s: re-adding the write event\n", __func__));
57 /* Callback for conn->wev and conn->rev. FIXME: Rename.
59 USER_DATA contains connection for EV_READ and a packet for
60 EV_WRITE. This is because we don't have a connect/establish entry
61 point at the user level -- send implies connect so when we're
62 connected we need the packet to send. */
64 _evcb (evutil_socket_t fd, short what, void *user_data)
66 rs_debug (("%s: fd=%d what =", __func__, fd));
67 if (what & EV_TIMEOUT) rs_debug ((" TIMEOUT"));
68 if (what & EV_READ) rs_debug ((" READ"));
69 if (what & EV_WRITE) rs_debug ((" WRITE"));
74 /* Read a single UDP packet and stick it in USER_DATA. */
75 /* TODO: Verify that unsolicited packets are dropped. */
76 struct rs_packet *pkt = (struct rs_packet *) user_data;
82 pkt->rpkt->data = rs_malloc (pkt->conn->ctx, 4096);
83 if (pkt->rpkt->data == NULL)
85 rs_err_conn_push_fl (pkt->conn, RSE_NOMEM, __FILE__, __LINE__, NULL);
88 r = compat_recv (fd, pkt->rpkt->data, 4096, MSG_TRUNC);
91 int sockerr = evutil_socket_geterror (pkt->conn->fd);
92 if (sockerr == EAGAIN)
94 /* FIXME: Really shouldn't happen since we've been told
95 that fd is readable! */
96 rs_debug (("%s: EAGAIN reading UDP packet -- wot?"));
101 rs_err_conn_push_fl (pkt->conn, RSE_SOCKERR, __FILE__, __LINE__,
102 "%d: recv: %d (%s)", fd, sockerr,
103 evutil_socket_error_to_string (sockerr));
104 event_del (pkt->conn->tev);
107 event_del (pkt->conn->tev);
108 if (r < 20 || r > 4096) /* Short or long packet. */
110 rs_err_conn_push (pkt->conn, RSE_INVALID_PKT,
111 "invalid packet length: %d",
112 pkt->rpkt->data_len);
115 pkt->rpkt->data_len = (pkt->rpkt->data[2] << 8) + pkt->rpkt->data[3];
116 if (!rad_packet_ok (pkt->rpkt, 0))
118 rs_err_conn_push_fl (pkt->conn, RSE_FR, __FILE__, __LINE__,
119 "invalid packet: %s", fr_strerror ());
122 /* Hand over message to user. This changes ownership of pkt.
123 Don't touch it afterwards -- it might have been freed. */
124 if (pkt->conn->callbacks.received_cb)
125 pkt->conn->callbacks.received_cb (pkt, pkt->conn->user_data);
127 else if (what & EV_WRITE)
129 struct rs_packet *pkt = (struct rs_packet *) user_data;
133 if (!pkt->conn->is_connected)
134 event_on_connect (pkt->conn, pkt);
136 if (pkt->conn->out_queue)
137 if (_send (pkt->conn, fd) == RSE_OK)
138 if (pkt->conn->callbacks.sent_cb)
139 pkt->conn->callbacks.sent_cb (pkt->conn->user_data);
143 if (what & EV_TIMEOUT)
144 rs_debug (("%s: timeout on UDP event, shouldn't happen\n", __func__));
149 udp_init (struct rs_connection *conn, struct rs_packet *pkt)
153 conn->rev = event_new (conn->evb, conn->fd, EV_READ|EV_PERSIST, _evcb, NULL);
154 conn->wev = event_new (conn->evb, conn->fd, EV_WRITE, _evcb, NULL);
155 if (!conn->rev || !conn->wev)
159 event_free (conn->rev);
162 /* ENOMEM _or_ EINVAL but EINVAL only if we use EV_SIGNAL, at
163 least for now (libevent-2.0.5). */
164 return rs_err_conn_push_fl (conn, RSE_NOMEM, __FILE__, __LINE__, NULL);
170 udp_init_retransmit_timer (struct rs_connection *conn)
175 event_free (conn->tev);
176 conn->tev = evtimer_new (conn->evb, event_retransmit_timeout_cb, conn);
178 return rs_err_conn_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,