Amazing-Mirror/fragattacks
1
0
Fork 0
mirror of https://github.com/vanhoefm/fragattacks.git synced 2025-01-31 09:14:03 -05:00
Code Issues Projects Releases Wiki Activity

WPS ER: Only activate PBC mode on single AP

Browse Source 
Verify that the UUID given to wps_er_pbc command is known and only
activate PBC mode on the matching AP. The UUID can be that of the
AP or the station/Enrollee.
This commit is contained in:
Jouni Malinen 2010-10-25 21:41:10 +03:00 committed by Jouni Malinen
parent 3e7533b399
commit 20ebd9c4b9
2 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/wps/wps_er.c
View File

@ -62,11 +62,15 @@ static void wps_er_sta_event(struct wps_context *wps, struct wps_er_sta *sta,
} }
static struct wps_er_sta * wps_er_sta_get(struct wps_er_ap *ap, const u8 *addr) static struct wps_er_sta * wps_er_sta_get(struct wps_er_ap *ap, const u8 *addr,
const u8 *uuid)
{ {
struct wps_er_sta *sta; struct wps_er_sta *sta;
dl_list_for_each(sta, &ap->sta, struct wps_er_sta, list) { dl_list_for_each(sta, &ap->sta, struct wps_er_sta, list) {
if (os_memcmp(sta->addr, addr, ETH_ALEN) == 0) if ((addr == NULL ||
os_memcmp(sta->addr, addr, ETH_ALEN) == 0) &&
(uuid == NULL ||
os_memcmp(uuid, sta->uuid, WPS_UUID_LEN) == 0))
return sta; return sta;
} }
return NULL; return NULL;
@ -649,7 +653,7 @@ static struct wps_er_sta * wps_er_add_sta_data(struct wps_er_ap *ap,
struct wps_parse_attr *attr, struct wps_parse_attr *attr,
int probe_req) int probe_req)
{ {
struct wps_er_sta *sta = wps_er_sta_get(ap, addr); struct wps_er_sta *sta = wps_er_sta_get(ap, addr, NULL);
int new_sta = 0; int new_sta = 0;
int m1; int m1;
@ -1418,6 +1422,10 @@ void wps_er_set_sel_reg(struct wps_er *er, int sel_reg, u16 dev_passwd_id,
data.set_sel_reg.state = WPS_ER_SET_SEL_REG_START; data.set_sel_reg.state = WPS_ER_SET_SEL_REG_START;
dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) { dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
if (er->set_sel_reg_uuid_filter &&
os_memcmp(ap->uuid, er->set_sel_reg_uuid_filter,
WPS_UUID_LEN) != 0)
continue;
data.set_sel_reg.uuid = ap->uuid; data.set_sel_reg.uuid = ap->uuid;
er->wps->event_cb(er->wps->cb_ctx, er->wps->event_cb(er->wps->cb_ctx,
WPS_EV_ER_SET_SELECTED_REGISTRAR, &data); WPS_EV_ER_SET_SELECTED_REGISTRAR, &data);
@ -1430,6 +1438,9 @@ void wps_er_set_sel_reg(struct wps_er *er, int sel_reg, u16 dev_passwd_id,
int wps_er_pbc(struct wps_er *er, const u8 *uuid) int wps_er_pbc(struct wps_er *er, const u8 *uuid)
{ {
int res;
struct wps_er_ap *ap;
if (er == NULL || er->wps == NULL) if (er == NULL || er->wps == NULL)
return -1; return -1;
@ -1439,13 +1450,24 @@ int wps_er_pbc(struct wps_er *er, const u8 *uuid)
return -1; return -1;
} }
/* ap = wps_er_ap_get(er, NULL, uuid);
* TODO: Should enable PBC mode only in a single AP based on which AP if (ap == NULL) {
* the Enrollee (uuid) is using. Now, we may end up enabling multiple struct wps_er_sta *sta = NULL;
* APs in PBC mode which could result in session overlap at the dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
* Enrollee. sta = wps_er_sta_get(ap, NULL, uuid);
*/ if (sta) {
if (wps_registrar_button_pushed(er->wps->registrar)) uuid = ap->uuid;
break;
}
}
if (sta == NULL)
return -1; /* Unknown UUID */
}
er->set_sel_reg_uuid_filter = uuid;
res = wps_registrar_button_pushed(er->wps->registrar);
er->set_sel_reg_uuid_filter = NULL;
if (res)
return -1; return -1;
return 0; return 0;

1
src/wps/wps_er.h
View File

@ -92,6 +92,7 @@ struct wps_er {
void *deinit_done_ctx; void *deinit_done_ctx;
struct in_addr filter_addr; struct in_addr filter_addr;
int skip_set_sel_reg; int skip_set_sel_reg;
const u8 *set_sel_reg_uuid_filter;
}; };