2 #include <cgreen/cgreen.h>
3 #include <freeradius/libradius.h>
4 #include "radsec/radsec.h"
5 #include "radsec/request.h"
8 #define true 1 /* FIXME: Bug report cgreen. */
11 #define FREERADIUS_DICT "/usr/share/freeradius/dictionary"
14 authenticate (struct rs_connection *conn, const char *user, const char *pw)
16 struct rs_request *req;
17 struct rs_packet *msg, *resp;
19 assert_true (rs_request_create (conn, &req) == 0);
20 assert_true (rs_packet_create_authn_request (conn, &msg, user, pw) == 0);
21 rs_request_add_reqpkt (req, msg);
22 assert_true (rs_request_send (req, &resp) == 0);
23 //printf ("%s\n", rs_err_msg (rs_err_conn_pop (conn), 1));
24 assert_true (rs_packet_frpkt (resp)->code == PW_AUTHENTICATION_ACK);
26 rs_request_destroy (req);
30 send_more_than_one_msg_in_one_packet (struct rs_connection *conn)
32 struct rs_packet *msg0, *msg1;
34 assert_true (rs_packet_create_authn_request (conn, &msg0, NULL, NULL) == 0);
35 assert_true (rs_packet_create_authn_request (conn, &msg1, NULL, NULL) == 0);
36 assert_true (rs_packet_send (msg0, NULL) == 0);
37 assert_true (rs_packet_send (msg1, NULL) == 0);
41 send_large_packet (struct rs_connection *conn)
43 struct rs_packet *msg0;
44 struct rs_attr *attr_x;
49 assert_true (buf != NULL);
50 memset (buf, 0, 4096);
52 assert_true (rs_packet_create (conn, &msg0) == 0);
53 /* 16 chunks --> heap corruption in evbuffer_drain detected by free() */
54 for (f = 0; f < 15; f++)
56 memset (buf, 'a' + f, 252);
57 rs_attr_create (conn, &attr_x, "EAP-Message", buf);
58 rs_packet_add_attr (msg0, attr_x);
60 assert_true (rs_packet_send (msg0, NULL) == 0);
63 /* ************************************************************ */
74 struct rs_context *ctx;
75 struct rs_connection *conn;
77 setup.config_file = "test.conf";
78 setup.config_name = "test-udp-auth";
79 setup.username = "molgan";
80 setup.pw = "password";
82 assert_true (rs_context_create (&ctx, FREERADIUS_DICT) == 0);
83 assert_true (rs_context_read_config (ctx, setup.config_file) == 0);
84 assert_true (rs_conn_create (ctx, &conn, setup.config_name) == 0);
86 authenticate (conn, setup.username, setup.pw);
88 rs_conn_destroy (conn);
89 rs_context_destroy (ctx);
93 test_buffering_cb (const uint8_t *buf, ssize_t len)
95 /* "Exactly one RADIUS packet is encapsulated in the UDP Data field"
100 assert_true (len >= 20);
101 assert_true (len <= 4096);
102 assert_true ((buf[2] << 8) + buf[3] == len);
109 struct rs_context *ctx;
110 struct rs_connection *conn;
111 struct timeval timeout;
112 struct polldata *polldata;
114 assert_true (rs_context_create (&ctx, FREERADIUS_DICT) == 0);
115 assert_true (rs_context_read_config (ctx, "test.conf") == 0);
116 assert_true (rs_conn_create (ctx, &conn, "test-udp-buffering") == 0);
119 timeout.tv_usec = 150000;
120 polldata = udp_server ("11820", &timeout, test_buffering_cb);
121 assert_true (polldata != NULL);
123 send_more_than_one_msg_in_one_packet (conn);
124 assert_true (udp_poll (polldata) > 0);
125 assert_true (udp_poll (polldata) > 0);
127 send_large_packet (conn);
128 assert_true (udp_poll (polldata) > 0);
130 udp_free_polldata (polldata);
131 rs_conn_destroy (conn);
132 rs_context_destroy (ctx);
135 /* ************************************************************ */
137 setup_auth (TestSuite *ts)
139 add_test (ts, test_auth);
143 setup_buffering (TestSuite *ts)
145 add_test (ts, test_buffering);
149 main (int argc, char *argv[])
151 TestSuite *ts = create_test_suite ();
154 setup_buffering (ts);
157 return run_single_test (ts, argv[1], create_text_reporter ());
159 return run_test_suite (ts, create_text_reporter ());