3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-21 22:09:23 +01:00

nl80211util: use nl80211_parse_attrs/nested for get_key_seq

This really reduces the number of checks needed and overall function
length by using the helper APIs.
This commit is contained in:
James Prestwood 2024-06-25 06:06:19 -07:00 committed by Denis Kenzior
parent 5c824cc24e
commit 77cf621f15

View File

@ -184,6 +184,7 @@ static attr_handler handler_for_nl80211(int type)
return extract_iovec; return extract_iovec;
case NL80211_ATTR_WIPHY_BANDS: case NL80211_ATTR_WIPHY_BANDS:
case NL80211_ATTR_SURVEY_INFO: case NL80211_ATTR_SURVEY_INFO:
case NL80211_ATTR_KEY:
return extract_nested; return extract_nested;
case NL80211_ATTR_KEY_IDX: case NL80211_ATTR_KEY_IDX:
return extract_u8; return extract_u8;
@ -208,6 +209,18 @@ static attr_handler handler_for_survey_info(int type)
return NULL; return NULL;
} }
static attr_handler handler_for_key(int type)
{
switch (type) {
case NL80211_KEY_SEQ:
return extract_iovec;
default:
break;
}
return NULL;
}
struct attr_entry { struct attr_entry {
uint16_t type; uint16_t type;
void *data; void *data;
@ -322,6 +335,9 @@ int nl80211_parse_nested(struct l_genl_attr *attr, int type, int tag, ...)
case NL80211_ATTR_SURVEY_INFO: case NL80211_ATTR_SURVEY_INFO:
handler = handler_for_survey_info; handler = handler_for_survey_info;
break; break;
case NL80211_ATTR_KEY:
handler = handler_for_key;
break;
default: default:
return -ENOTSUP; return -ENOTSUP;
} }
@ -568,46 +584,29 @@ struct l_genl_msg *nl80211_build_get_key(uint32_t ifindex, uint8_t key_index)
const void *nl80211_parse_get_key_seq(struct l_genl_msg *msg) const void *nl80211_parse_get_key_seq(struct l_genl_msg *msg)
{ {
struct l_genl_attr attr, nested; struct l_genl_attr nested;
uint16_t type, len; struct iovec iov;
const void *data;
if (l_genl_msg_get_error(msg) < 0 || !l_genl_attr_init(&attr, msg)) { if (l_genl_msg_get_error(msg) < 0 ||
nl80211_parse_attrs(msg, NL80211_ATTR_KEY, &nested,
NL80211_ATTR_UNSPEC) < 0) {
l_error("GET_KEY failed for the GTK: %i", l_error("GET_KEY failed for the GTK: %i",
l_genl_msg_get_error(msg)); l_genl_msg_get_error(msg));
return NULL; return NULL;
} }
while (l_genl_attr_next(&attr, &type, &len, &data)) { if (nl80211_parse_nested(&nested, NL80211_ATTR_KEY, NL80211_KEY_SEQ,
if (type != NL80211_ATTR_KEY) &iov, NL80211_ATTR_UNSPEC)) {
continue;
break;
}
if (type != NL80211_ATTR_KEY || !l_genl_attr_recurse(&attr, &nested)) {
l_error("Can't recurse into ATTR_KEY in GET_KEY reply"); l_error("Can't recurse into ATTR_KEY in GET_KEY reply");
return NULL; return NULL;
} }
while (l_genl_attr_next(&nested, &type, &len, &data)) { if (iov.iov_len != 6) {
if (type != NL80211_KEY_SEQ)
continue;
break;
}
if (type != NL80211_KEY_SEQ) {
l_error("KEY_SEQ not returned in GET_KEY reply");
return NULL;
}
if (len != 6) {
l_error("KEY_SEQ length != 6 in GET_KEY reply"); l_error("KEY_SEQ length != 6 in GET_KEY reply");
return NULL; return NULL;
} }
return data; return iov.iov_base;
} }
struct l_genl_msg *nl80211_build_cmd_frame(uint32_t ifindex, struct l_genl_msg *nl80211_build_cmd_frame(uint32_t ifindex,