xref: /dragonfly/contrib/wpa_supplicant/src/eap_server/eap_server_vendor_test.c (revision 3a84a4273475ed07d0ab1c2dfeffdfedef35d9cd)
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 
eap_vendor_test_state_txt(int state)24 static const char * eap_vendor_test_state_txt(int state)
25 {
26           switch (state) {
27           case INIT:
28                     return "INIT";
29           case CONFIRM:
30                     return "CONFIRM";
31           case SUCCESS:
32                     return "SUCCESS";
33           case FAILURE:
34                     return "FAILURE";
35           default:
36                     return "?";
37           }
38 }
39 
40 
eap_vendor_test_state(struct eap_vendor_test_data * data,int state)41 static void eap_vendor_test_state(struct eap_vendor_test_data *data,
42                                           int state)
43 {
44           wpa_printf(MSG_DEBUG, "EAP-VENDOR-TEST: %s -> %s",
45                        eap_vendor_test_state_txt(data->state),
46                        eap_vendor_test_state_txt(state));
47           data->state = state;
48 }
49 
50 
eap_vendor_test_init(struct eap_sm * sm)51 static void * eap_vendor_test_init(struct eap_sm *sm)
52 {
53           struct eap_vendor_test_data *data;
54 
55           data = os_zalloc(sizeof(*data));
56           if (data == NULL)
57                     return NULL;
58           data->state = INIT;
59 
60           return data;
61 }
62 
63 
eap_vendor_test_reset(struct eap_sm * sm,void * priv)64 static void eap_vendor_test_reset(struct eap_sm *sm, void *priv)
65 {
66           struct eap_vendor_test_data *data = priv;
67           os_free(data);
68 }
69 
70 
eap_vendor_test_buildReq(struct eap_sm * sm,void * priv,u8 id)71 static struct wpabuf * eap_vendor_test_buildReq(struct eap_sm *sm, void *priv,
72                                                             u8 id)
73 {
74           struct eap_vendor_test_data *data = priv;
75           struct wpabuf *req;
76 
77           req = eap_msg_alloc(EAP_VENDOR_ID, EAP_VENDOR_TYPE, 1,
78                                   EAP_CODE_REQUEST, id);
79           if (req == NULL) {
80                     wpa_printf(MSG_ERROR, "EAP-VENDOR-TEST: Failed to allocate "
81                                  "memory for request");
82                     return NULL;
83           }
84 
85           wpabuf_put_u8(req, data->state == INIT ? 1 : 3);
86 
87           return req;
88 }
89 
90 
eap_vendor_test_check(struct eap_sm * sm,void * priv,struct wpabuf * respData)91 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           pos = eap_hdr_validate(EAP_VENDOR_ID, EAP_VENDOR_TYPE, respData, &len);
98           if (pos == NULL || len < 1) {
99                     wpa_printf(MSG_INFO, "EAP-VENDOR-TEST: Invalid frame");
100                     return TRUE;
101           }
102 
103           return FALSE;
104 }
105 
106 
eap_vendor_test_process(struct eap_sm * sm,void * priv,struct wpabuf * respData)107 static void eap_vendor_test_process(struct eap_sm *sm, void *priv,
108                                             struct wpabuf *respData)
109 {
110           struct eap_vendor_test_data *data = priv;
111           const u8 *pos;
112           size_t len;
113 
114           pos = eap_hdr_validate(EAP_VENDOR_ID, EAP_VENDOR_TYPE, respData, &len);
115           if (pos == NULL || len < 1)
116                     return;
117 
118           if (data->state == INIT) {
119                     if (*pos == 2)
120                               eap_vendor_test_state(data, CONFIRM);
121                     else
122                               eap_vendor_test_state(data, FAILURE);
123           } else if (data->state == CONFIRM) {
124                     if (*pos == 4)
125                               eap_vendor_test_state(data, SUCCESS);
126                     else
127                               eap_vendor_test_state(data, FAILURE);
128           } else
129                     eap_vendor_test_state(data, FAILURE);
130 }
131 
132 
eap_vendor_test_isDone(struct eap_sm * sm,void * priv)133 static Boolean eap_vendor_test_isDone(struct eap_sm *sm, void *priv)
134 {
135           struct eap_vendor_test_data *data = priv;
136           return data->state == SUCCESS;
137 }
138 
139 
eap_vendor_test_getKey(struct eap_sm * sm,void * priv,size_t * len)140 static u8 * eap_vendor_test_getKey(struct eap_sm *sm, void *priv, size_t *len)
141 {
142           struct eap_vendor_test_data *data = priv;
143           u8 *key;
144           const int key_len = 64;
145 
146           if (data->state != SUCCESS)
147                     return NULL;
148 
149           key = os_malloc(key_len);
150           if (key == NULL)
151                     return NULL;
152 
153           os_memset(key, 0x11, key_len / 2);
154           os_memset(key + key_len / 2, 0x22, key_len / 2);
155           *len = key_len;
156 
157           return key;
158 }
159 
160 
eap_vendor_test_isSuccess(struct eap_sm * sm,void * priv)161 static Boolean eap_vendor_test_isSuccess(struct eap_sm *sm, void *priv)
162 {
163           struct eap_vendor_test_data *data = priv;
164           return data->state == SUCCESS;
165 }
166 
167 
eap_server_vendor_test_register(void)168 int eap_server_vendor_test_register(void)
169 {
170           struct eap_method *eap;
171 
172           eap = eap_server_method_alloc(EAP_SERVER_METHOD_INTERFACE_VERSION,
173                                               EAP_VENDOR_ID, EAP_VENDOR_TYPE,
174                                               "VENDOR-TEST");
175           if (eap == NULL)
176                     return -1;
177 
178           eap->init = eap_vendor_test_init;
179           eap->reset = eap_vendor_test_reset;
180           eap->buildReq = eap_vendor_test_buildReq;
181           eap->check = eap_vendor_test_check;
182           eap->process = eap_vendor_test_process;
183           eap->isDone = eap_vendor_test_isDone;
184           eap->getKey = eap_vendor_test_getKey;
185           eap->isSuccess = eap_vendor_test_isSuccess;
186 
187           return eap_server_method_register(eap);
188 }
189