87 lines
2.3 KiB
C
87 lines
2.3 KiB
C
#include <linux/nl80211.h>
|
|
#include <netlink/genl/genl.h>
|
|
|
|
#include "netlink.h"
|
|
#include "neighbours.h"
|
|
|
|
static const char * neighbours_names[NL80211_STA_INFO_MAX + 1] = {
|
|
[NL80211_STA_INFO_SIGNAL] = "signal",
|
|
[NL80211_STA_INFO_INACTIVE_TIME] = "inactive",
|
|
};
|
|
|
|
static int station_neighbours_handler(struct nl_msg *msg, void *arg) {
|
|
struct json_object *neighbour, *json = (struct json_object *) arg;
|
|
|
|
neighbour = json_object_new_object();
|
|
if (!neighbour)
|
|
goto abort;
|
|
|
|
struct nlattr *tb[NL80211_ATTR_MAX + 1];
|
|
|
|
struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
|
|
struct nlattr *station_info = nla_find(genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NL80211_ATTR_STA_INFO);
|
|
|
|
nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL);
|
|
|
|
if (!station_info) {
|
|
fputs("respondd-module-wifi: station data missing in netlink message\n", stderr);
|
|
json_object_put(neighbour);
|
|
goto abort;
|
|
}
|
|
|
|
char mac_addr[20];
|
|
|
|
if (!tb[NL80211_ATTR_MAC]) {
|
|
json_object_put(neighbour);
|
|
goto abort;
|
|
}
|
|
mac_addr_n2a(mac_addr, nla_data(tb[NL80211_ATTR_MAC]));
|
|
|
|
int rem;
|
|
struct nlattr *nla;
|
|
nla_for_each_nested(nla, station_info, rem) {
|
|
int type = nla_type(nla);
|
|
|
|
if (type > NL80211_STA_INFO_MAX)
|
|
continue;
|
|
|
|
if (!neighbours_names[type])
|
|
continue;
|
|
|
|
struct json_object *data_json = NULL;
|
|
switch (nla_len(nla)) {
|
|
case sizeof(uint64_t):
|
|
data_json = json_object_new_int64(nla_get_u64(nla));
|
|
break;
|
|
case sizeof(uint32_t):
|
|
data_json = json_object_new_int(nla_get_u32(nla));
|
|
break;
|
|
case sizeof(uint8_t):
|
|
data_json = json_object_new_int(nla_get_u8(nla));
|
|
break;
|
|
default:
|
|
fprintf(stderr, "respondd-module-wifi: Unexpected NL attribute length: %d\n", nla_len(nla));
|
|
}
|
|
if (data_json)
|
|
json_object_object_add(neighbour, neighbours_names[type], data_json);
|
|
}
|
|
json_object_object_add(json, mac_addr, neighbour);
|
|
|
|
abort:
|
|
return NL_SKIP;
|
|
}
|
|
|
|
bool get_neighbours(struct json_object *result, int ifx) {
|
|
struct json_object *neighbours;
|
|
neighbours = json_object_new_object();
|
|
if (!neighbours)
|
|
return false;
|
|
if(!nl_send_dump(station_neighbours_handler, neighbours, NL80211_CMD_GET_STATION, ifx)) {
|
|
json_object_put(neighbours);
|
|
return false;
|
|
}
|
|
json_object_object_add(result, "neighbours", neighbours);
|
|
return true;
|
|
|
|
}
|