Line data Source code
1 : /*
2 : * hostapd / Test method for vendor specific (expanded) EAP type
3 : * Copyright (c) 2005-2007, Jouni Malinen <j@w1.fi>
4 : *
5 : * This software may be distributed under the terms of the BSD license.
6 : * See README for more details.
7 : */
8 :
9 : #include "includes.h"
10 :
11 : #include "common.h"
12 : #include "eap_i.h"
13 :
14 :
15 : #define EAP_VENDOR_ID EAP_VENDOR_HOSTAP
16 : #define EAP_VENDOR_TYPE 0xfcfbfaf9
17 :
18 :
19 : struct eap_vendor_test_data {
20 : enum { INIT, CONFIRM, SUCCESS, FAILURE } state;
21 : };
22 :
23 :
24 16 : static const char * eap_vendor_test_state_txt(int state)
25 : {
26 16 : switch (state) {
27 : case INIT:
28 4 : return "INIT";
29 : case CONFIRM:
30 8 : return "CONFIRM";
31 : case SUCCESS:
32 4 : return "SUCCESS";
33 : case FAILURE:
34 0 : return "FAILURE";
35 : default:
36 0 : return "?";
37 : }
38 : }
39 :
40 :
41 8 : static void eap_vendor_test_state(struct eap_vendor_test_data *data,
42 : int state)
43 : {
44 16 : wpa_printf(MSG_DEBUG, "EAP-VENDOR-TEST: %s -> %s",
45 8 : eap_vendor_test_state_txt(data->state),
46 : eap_vendor_test_state_txt(state));
47 8 : data->state = state;
48 8 : }
49 :
50 :
51 5 : static void * eap_vendor_test_init(struct eap_sm *sm)
52 : {
53 : struct eap_vendor_test_data *data;
54 :
55 5 : data = os_zalloc(sizeof(*data));
56 5 : if (data == NULL)
57 0 : return NULL;
58 5 : data->state = INIT;
59 :
60 5 : return data;
61 : }
62 :
63 :
64 5 : static void eap_vendor_test_reset(struct eap_sm *sm, void *priv)
65 : {
66 5 : struct eap_vendor_test_data *data = priv;
67 5 : os_free(data);
68 5 : }
69 :
70 :
71 9 : static struct wpabuf * eap_vendor_test_buildReq(struct eap_sm *sm, void *priv,
72 : u8 id)
73 : {
74 9 : struct eap_vendor_test_data *data = priv;
75 : struct wpabuf *req;
76 :
77 9 : req = eap_msg_alloc(EAP_VENDOR_ID, EAP_VENDOR_TYPE, 1,
78 : EAP_CODE_REQUEST, id);
79 9 : if (req == NULL) {
80 0 : wpa_printf(MSG_ERROR, "EAP-VENDOR-TEST: Failed to allocate "
81 : "memory for request");
82 0 : return NULL;
83 : }
84 :
85 9 : wpabuf_put_u8(req, data->state == INIT ? 1 : 3);
86 :
87 9 : return req;
88 : }
89 :
90 :
91 8 : static Boolean eap_vendor_test_check(struct eap_sm *sm, void *priv,
92 : struct wpabuf *respData)
93 : {
94 : const u8 *pos;
95 : size_t len;
96 :
97 8 : pos = eap_hdr_validate(EAP_VENDOR_ID, EAP_VENDOR_TYPE, respData, &len);
98 8 : if (pos == NULL || len < 1) {
99 0 : wpa_printf(MSG_INFO, "EAP-VENDOR-TEST: Invalid frame");
100 0 : return TRUE;
101 : }
102 :
103 8 : return FALSE;
104 : }
105 :
106 :
107 8 : static void eap_vendor_test_process(struct eap_sm *sm, void *priv,
108 : struct wpabuf *respData)
109 : {
110 8 : struct eap_vendor_test_data *data = priv;
111 : const u8 *pos;
112 : size_t len;
113 :
114 8 : pos = eap_hdr_validate(EAP_VENDOR_ID, EAP_VENDOR_TYPE, respData, &len);
115 8 : if (pos == NULL || len < 1)
116 8 : return;
117 :
118 8 : if (data->state == INIT) {
119 4 : if (*pos == 2)
120 4 : eap_vendor_test_state(data, CONFIRM);
121 : else
122 0 : eap_vendor_test_state(data, FAILURE);
123 4 : } else if (data->state == CONFIRM) {
124 4 : if (*pos == 4)
125 4 : eap_vendor_test_state(data, SUCCESS);
126 : else
127 0 : eap_vendor_test_state(data, FAILURE);
128 : } else
129 0 : eap_vendor_test_state(data, FAILURE);
130 : }
131 :
132 :
133 8 : static Boolean eap_vendor_test_isDone(struct eap_sm *sm, void *priv)
134 : {
135 8 : struct eap_vendor_test_data *data = priv;
136 8 : return data->state == SUCCESS;
137 : }
138 :
139 :
140 4 : static u8 * eap_vendor_test_getKey(struct eap_sm *sm, void *priv, size_t *len)
141 : {
142 4 : struct eap_vendor_test_data *data = priv;
143 : u8 *key;
144 4 : const int key_len = 64;
145 :
146 4 : if (data->state != SUCCESS)
147 0 : return NULL;
148 :
149 4 : key = os_malloc(key_len);
150 4 : if (key == NULL)
151 0 : return NULL;
152 :
153 4 : os_memset(key, 0x11, key_len / 2);
154 4 : os_memset(key + key_len / 2, 0x22, key_len / 2);
155 4 : *len = key_len;
156 :
157 4 : return key;
158 : }
159 :
160 :
161 4 : static Boolean eap_vendor_test_isSuccess(struct eap_sm *sm, void *priv)
162 : {
163 4 : struct eap_vendor_test_data *data = priv;
164 4 : return data->state == SUCCESS;
165 : }
166 :
167 :
168 25 : int eap_server_vendor_test_register(void)
169 : {
170 : struct eap_method *eap;
171 : int ret;
172 :
173 25 : eap = eap_server_method_alloc(EAP_SERVER_METHOD_INTERFACE_VERSION,
174 : EAP_VENDOR_ID, EAP_VENDOR_TYPE,
175 : "VENDOR-TEST");
176 25 : if (eap == NULL)
177 0 : return -1;
178 :
179 25 : eap->init = eap_vendor_test_init;
180 25 : eap->reset = eap_vendor_test_reset;
181 25 : eap->buildReq = eap_vendor_test_buildReq;
182 25 : eap->check = eap_vendor_test_check;
183 25 : eap->process = eap_vendor_test_process;
184 25 : eap->isDone = eap_vendor_test_isDone;
185 25 : eap->getKey = eap_vendor_test_getKey;
186 25 : eap->isSuccess = eap_vendor_test_isSuccess;
187 :
188 25 : ret = eap_server_method_register(eap);
189 25 : if (ret)
190 0 : eap_server_method_free(eap);
191 25 : return ret;
192 : }
|