1 /* See the file COPYING for licensing information. */
5 #include <freeradius/libradius.h>
6 #include <event2/event.h>
7 #include <event2/bufferevent.h>
9 #include "libradsec-impl.h"
15 _packet_create (struct rs_connection *conn, struct rs_packet **pkt_out)
24 return rs_conn_err_push (conn, RSE_NOMEM, __func__);
27 p = (struct rs_packet *) malloc (sizeof (struct rs_packet));
31 return rs_conn_err_push (conn, RSE_NOMEM, __func__);
33 memset (p, 0, sizeof (struct rs_packet));
42 rs_packet_create_acc_request (struct rs_connection *conn,
43 struct rs_packet **pkt_out,
44 const char *user_name, const char *user_pw)
46 struct rs_packet *pkt;
49 if (_packet_create (conn, pkt_out))
52 pkt->rpkt->code = PW_AUTHENTICATION_REQUEST;
54 if (rs_attr_create (conn, &attr, "User-Name", user_name))
56 rs_packet_add_attr (pkt, attr);
58 if (rs_attr_create (conn, &attr, "User-Password", user_pw))
60 /* FIXME: need this too? rad_pwencode(user_pw, &pwlen, SECRET, reqauth) */
61 rs_packet_add_attr (pkt, attr);
67 _event_cb (struct bufferevent *bev, short events, void *ctx)
69 struct rs_packet *pkt = (struct rs_packet *) ctx;
70 struct rs_connection *conn;
75 assert (pkt->conn->active_peer);
77 p = conn->active_peer;
80 if (events & BEV_EVENT_CONNECTED)
84 fprintf (stderr, "%s: connected\n", __func__);
86 rad_encode (pkt->rpkt, NULL, pkt->conn->active_peer->secret);
89 fprintf (stderr, "%s: about to send this to %s:\n", __func__, "<fixme>");
92 if (bufferevent_write(bev, pkt->rpkt->data, pkt->rpkt->data_len))
93 rs_conn_err_push_fl (pkt->conn, RSE_EVENT, __FILE__, __LINE__,
95 /* Packet will be freed in write callback. */
97 else if (events & BEV_EVENT_ERROR)
98 rs_conn_err_push_fl (pkt->conn, RSE_CONNERR, __FILE__, __LINE__, NULL);
102 rs_packet_destroy(struct rs_packet *pkt)
104 rad_free (&pkt->rpkt);
105 rs_free (pkt->conn->ctx, pkt);
109 _write_cb (struct bufferevent *bev, void *ctx)
111 struct rs_packet *pkt = (struct rs_packet *) ctx;
116 fprintf (stderr, "%s: packet written, breaking event loop\n", __func__);
118 if (event_base_loopbreak (pkt->conn->evb) < 0)
119 abort (); /* FIXME */
120 rs_packet_destroy (pkt);
124 _read_cb (struct bufferevent *bev, void *ctx)
126 struct rs_packet *pkt = (struct rs_packet *) ctx;
131 if (!pkt->hdr_read_flag)
133 n = bufferevent_read (pkt->conn->bev, pkt->hdr, RS_HEADER_LEN;
134 if (n == RS_HEADER_LEN)
136 uint16_t len = (pkt->hdr[2] << 8) + pkt->hdr[3];
137 uint8_t *buf = rs_malloc (pkt->conn->ctx, len);
139 pkt->hdr_read_flag = 1;
142 rs_conn_err_push_fl (pkt->conn, RSE_NOMEM, __FILE__,
144 abort (); /* FIXME: recovering takes reading of packet */
146 pkt->rpkt->data = buf;
147 pkt->rpkt->data_len = len;
148 bufferevent_setwatermark (pkt->conn->bev, EV_READ,
149 len - RS_HEADER_LEN, 0);
151 fprintf (stderr, "%s: packet header read, pkt len=%d\n", __func__,
156 return; /* Buffer frozen, i suppose. Let's hope it thaws. */
159 assert (n < RS_HEADER_LEN);
160 return; /* Need more to complete header. */
164 printf ("%s: trying to read %d octets of packet data\n", __func__, pkt->rpkt->data_len - RS_HEADER_LEN;
165 n = bufferevent_read (pkt->conn->bev, pkt->rpkt->data,
166 pkt->rpkt->data_len - RS_HEADER_LEN);
167 printf ("%s: read %d octets of packet data\n", __func__, n);
168 if (n == pkt->rpkt->data_len - RS_HEADER_LEN)
170 bufferevent_disable (pkt->conn->bev, EV_READ);
171 pkt->hdr_read_flag = 0;
172 memset (pkt->hdr, 0, sizeof(*pkt->hdr));
174 fprintf (stderr, "%s: complete packet read\n", __func__);
176 if (event_base_loopbreak (pkt->conn->evb) < 0)
177 abort (); /* FIXME */
182 _init_evb (struct rs_connection *conn)
187 event_enable_debug_mode ();
189 conn->evb = event_base_new ();
191 return rs_conn_err_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
198 _init_socket (struct rs_connection *conn, struct rs_peer *p)
203 p->s = socket (p->addr->ai_family, p->addr->ai_socktype,
204 p->addr->ai_protocol);
206 return rs_conn_err_push_fl (conn, RSE_SOME_ERROR, __FILE__, __LINE__,
212 static struct rs_peer *
213 _pick_peer (struct rs_connection *conn)
215 if (!conn->active_peer)
216 conn->active_peer = conn->peers;
217 return conn->active_peer;
221 _init_bev (struct rs_connection *conn, struct rs_peer *peer)
225 conn->bev = bufferevent_socket_new (conn->evb, peer->s, 0);
227 return rs_conn_err_push_fl (conn, RSE_EVENT, __FILE__, __LINE__,
228 "bufferevent_socket_new");
234 _do_connect (struct rs_peer *p)
236 if (bufferevent_socket_connect (p->conn->bev, p->addr->ai_addr,
237 p->addr->ai_addrlen) < 0)
238 rs_conn_err_push_fl (p->conn, RSE_EVENT, __FILE__, __LINE__,
239 "bufferevent_socket_connect");
241 p->is_connecting = 1;
245 _conn_open(struct rs_connection *conn, struct rs_packet *pkt)
249 if (_init_evb (conn))
252 p = _pick_peer (conn);
254 return rs_conn_err_push_fl (conn, RSE_NOPEER, __FILE__, __LINE__, NULL);
256 if (_init_socket (conn, p))
259 if (_init_bev (conn, p))
261 bufferevent_setcb (conn->bev, _read_cb, _write_cb, _event_cb, pkt);
263 if (!p->is_connected)
264 if (!p->is_connecting)
271 rs_packet_send (struct rs_connection *conn, struct rs_packet *pkt, void *data)
276 if (_conn_open (conn, pkt))
280 assert (conn->active_peer);
281 assert (conn->active_peer->s >= 0);
283 event_base_dispatch (conn->evb);
286 fprintf (stderr, "%s: event loop done\n", __func__);
287 assert (event_base_got_break(conn->evb));
294 rs_packet_receive(struct rs_connection *conn, struct rs_packet **pkt_out)
296 struct rs_packet *pkt;
300 if (_packet_create (conn, pkt_out))
305 if (_conn_open (conn, pkt))
309 assert (conn->active_peer);
310 assert (conn->active_peer->s >= 0);
312 bufferevent_setwatermark (conn->bev, EV_READ, RS_HEADER_LEN, 0);
313 bufferevent_enable (conn->bev, EV_READ);
314 event_base_dispatch (conn->evb);
316 fprintf (stderr, "%s: event loop done\n", __func__);
317 assert (event_base_got_break(conn->evb));
321 fprintf (stderr, "%s: got this:\n", __func__);
322 rs_dump_packet (pkt);
329 rs_packet_add_attr(struct rs_packet *pkt, struct rs_attr *attr)
331 pairadd (&pkt->rpkt->vps, attr->vp);