You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
276 lines
7.6 KiB
276 lines
7.6 KiB
#include <net/if.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
#include <stdbool.h>
|
|
|
|
#include <netlink/genl/genl.h>
|
|
#include <netlink/genl/family.h>
|
|
#include <netlink/genl/ctrl.h>
|
|
#include <netlink/msg.h>
|
|
#include <netlink/attr.h>
|
|
|
|
#include "nl80211.h"
|
|
#include "iw.h"
|
|
|
|
struct link_result {
|
|
uint8_t bssid[8];
|
|
bool link_found;
|
|
bool anything_found;
|
|
};
|
|
|
|
static struct link_result lr = { .link_found = false };
|
|
|
|
static int link_bss_handler(struct nl_msg *msg, void *arg)
|
|
{
|
|
struct nlattr *tb[NL80211_ATTR_MAX + 1];
|
|
struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
|
|
struct nlattr *bss[NL80211_BSS_MAX + 1];
|
|
static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
|
|
[NL80211_BSS_TSF] = { .type = NLA_U64 },
|
|
[NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
|
|
[NL80211_BSS_BSSID] = { },
|
|
[NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
|
|
[NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
|
|
[NL80211_BSS_INFORMATION_ELEMENTS] = { },
|
|
[NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
|
|
[NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
|
|
[NL80211_BSS_STATUS] = { .type = NLA_U32 },
|
|
};
|
|
struct link_result *result = arg;
|
|
char mac_addr[20], dev[20];
|
|
|
|
nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
|
|
genlmsg_attrlen(gnlh, 0), NULL);
|
|
|
|
if (!tb[NL80211_ATTR_BSS]) {
|
|
fprintf(stderr, "bss info missing!\n");
|
|
return NL_SKIP;
|
|
}
|
|
if (nla_parse_nested(bss, NL80211_BSS_MAX,
|
|
tb[NL80211_ATTR_BSS],
|
|
bss_policy)) {
|
|
fprintf(stderr, "failed to parse nested attributes!\n");
|
|
return NL_SKIP;
|
|
}
|
|
|
|
if (!bss[NL80211_BSS_BSSID])
|
|
return NL_SKIP;
|
|
|
|
if (!bss[NL80211_BSS_STATUS])
|
|
return NL_SKIP;
|
|
|
|
mac_addr_n2a(mac_addr, nla_data(bss[NL80211_BSS_BSSID]));
|
|
if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);
|
|
|
|
switch (nla_get_u32(bss[NL80211_BSS_STATUS])) {
|
|
case NL80211_BSS_STATUS_ASSOCIATED:
|
|
printf("Connected to %s (on %s)\n", mac_addr, dev);
|
|
break;
|
|
case NL80211_BSS_STATUS_AUTHENTICATED:
|
|
printf("Authenticated with %s (on %s)\n", mac_addr, dev);
|
|
return NL_SKIP;
|
|
case NL80211_BSS_STATUS_IBSS_JOINED:
|
|
printf("Joined IBSS %s (on %s)\n", mac_addr, dev);
|
|
break;
|
|
default:
|
|
return NL_SKIP;
|
|
}
|
|
|
|
result->anything_found = true;
|
|
|
|
if (bss[NL80211_BSS_INFORMATION_ELEMENTS])
|
|
print_ies(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
|
|
nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
|
|
false, PRINT_LINK);
|
|
|
|
if (bss[NL80211_BSS_FREQUENCY])
|
|
printf("\tfreq: %d\n",
|
|
nla_get_u32(bss[NL80211_BSS_FREQUENCY]));
|
|
|
|
if (nla_get_u32(bss[NL80211_BSS_STATUS]) != NL80211_BSS_STATUS_ASSOCIATED)
|
|
return NL_SKIP;
|
|
|
|
/* only in the assoc case do we want more info from station get */
|
|
result->link_found = true;
|
|
memcpy(result->bssid, nla_data(bss[NL80211_BSS_BSSID]), 6);
|
|
return NL_SKIP;
|
|
}
|
|
|
|
static int handle_scan_for_link(struct nl80211_state *state,
|
|
struct nl_cb *cb,
|
|
struct nl_msg *msg,
|
|
int argc, char **argv,
|
|
enum id_input id)
|
|
{
|
|
if (argc > 0)
|
|
return 1;
|
|
|
|
nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, link_bss_handler, &lr);
|
|
return 0;
|
|
}
|
|
|
|
static int print_link_sta(struct nl_msg *msg, void *arg)
|
|
{
|
|
struct nlattr *tb[NL80211_ATTR_MAX + 1];
|
|
struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
|
|
struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
|
|
struct nlattr *binfo[NL80211_STA_BSS_PARAM_MAX + 1];
|
|
static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
|
|
[NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
|
|
[NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
|
|
[NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
|
|
[NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
|
|
[NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
|
|
[NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 },
|
|
[NL80211_STA_INFO_TX_BITRATE] = { .type = NLA_NESTED },
|
|
[NL80211_STA_INFO_LLID] = { .type = NLA_U16 },
|
|
[NL80211_STA_INFO_PLID] = { .type = NLA_U16 },
|
|
[NL80211_STA_INFO_PLINK_STATE] = { .type = NLA_U8 },
|
|
};
|
|
static struct nla_policy bss_policy[NL80211_STA_BSS_PARAM_MAX + 1] = {
|
|
[NL80211_STA_BSS_PARAM_CTS_PROT] = { .type = NLA_FLAG },
|
|
[NL80211_STA_BSS_PARAM_SHORT_PREAMBLE] = { .type = NLA_FLAG },
|
|
[NL80211_STA_BSS_PARAM_SHORT_SLOT_TIME] = { .type = NLA_FLAG },
|
|
[NL80211_STA_BSS_PARAM_DTIM_PERIOD] = { .type = NLA_U8 },
|
|
[NL80211_STA_BSS_PARAM_BEACON_INTERVAL] = { .type = NLA_U16 },
|
|
};
|
|
|
|
nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
|
|
genlmsg_attrlen(gnlh, 0), NULL);
|
|
|
|
if (!tb[NL80211_ATTR_STA_INFO]) {
|
|
fprintf(stderr, "sta stats missing!\n");
|
|
return NL_SKIP;
|
|
}
|
|
if (nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
|
|
tb[NL80211_ATTR_STA_INFO],
|
|
stats_policy)) {
|
|
fprintf(stderr, "failed to parse nested attributes!\n");
|
|
return NL_SKIP;
|
|
}
|
|
|
|
if (sinfo[NL80211_STA_INFO_RX_BYTES] && sinfo[NL80211_STA_INFO_RX_PACKETS])
|
|
printf("\tRX: %u bytes (%u packets)\n",
|
|
nla_get_u32(sinfo[NL80211_STA_INFO_RX_BYTES]),
|
|
nla_get_u32(sinfo[NL80211_STA_INFO_RX_PACKETS]));
|
|
if (sinfo[NL80211_STA_INFO_TX_BYTES] && sinfo[NL80211_STA_INFO_TX_PACKETS])
|
|
printf("\tTX: %u bytes (%u packets)\n",
|
|
nla_get_u32(sinfo[NL80211_STA_INFO_TX_BYTES]),
|
|
nla_get_u32(sinfo[NL80211_STA_INFO_TX_PACKETS]));
|
|
if (sinfo[NL80211_STA_INFO_SIGNAL])
|
|
printf("\tsignal: %d dBm\n",
|
|
(int8_t)nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]));
|
|
|
|
if (sinfo[NL80211_STA_INFO_TX_BITRATE]) {
|
|
char buf[100];
|
|
|
|
parse_bitrate(sinfo[NL80211_STA_INFO_TX_BITRATE], buf, sizeof(buf));
|
|
printf("\ttx bitrate: %s\n", buf);
|
|
}
|
|
|
|
if (sinfo[NL80211_STA_INFO_BSS_PARAM]) {
|
|
if (nla_parse_nested(binfo, NL80211_STA_BSS_PARAM_MAX,
|
|
sinfo[NL80211_STA_INFO_BSS_PARAM],
|
|
bss_policy)) {
|
|
fprintf(stderr, "failed to parse nested bss parameters!\n");
|
|
} else {
|
|
char *delim = "";
|
|
printf("\n\tbss flags:\t");
|
|
if (binfo[NL80211_STA_BSS_PARAM_CTS_PROT]) {
|
|
printf("CTS-protection");
|
|
delim = " ";
|
|
}
|
|
if (binfo[NL80211_STA_BSS_PARAM_SHORT_PREAMBLE]) {
|
|
printf("%sshort-preamble", delim);
|
|
delim = " ";
|
|
}
|
|
if (binfo[NL80211_STA_BSS_PARAM_SHORT_SLOT_TIME])
|
|
printf("%sshort-slot-time", delim);
|
|
printf("\n\tdtim period:\t%d",
|
|
nla_get_u8(binfo[NL80211_STA_BSS_PARAM_DTIM_PERIOD]));
|
|
printf("\n\tbeacon int:\t%d",
|
|
nla_get_u16(binfo[NL80211_STA_BSS_PARAM_BEACON_INTERVAL]));
|
|
printf("\n");
|
|
}
|
|
}
|
|
|
|
return NL_SKIP;
|
|
}
|
|
|
|
static int handle_link_sta(struct nl80211_state *state,
|
|
struct nl_cb *cb,
|
|
struct nl_msg *msg,
|
|
int argc, char **argv,
|
|
enum id_input id)
|
|
{
|
|
unsigned char mac_addr[ETH_ALEN];
|
|
|
|
if (argc < 1)
|
|
return 1;
|
|
|
|
if (mac_addr_a2n(mac_addr, argv[0])) {
|
|
fprintf(stderr, "invalid mac address\n");
|
|
return 2;
|
|
}
|
|
|
|
argc--;
|
|
argv++;
|
|
|
|
if (argc)
|
|
return 1;
|
|
|
|
NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr);
|
|
|
|
nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_link_sta, NULL);
|
|
|
|
return 0;
|
|
nla_put_failure:
|
|
return -ENOBUFS;
|
|
}
|
|
|
|
static int handle_link(struct nl80211_state *state, struct nl_cb *cb,
|
|
struct nl_msg *msg, int argc, char **argv,
|
|
enum id_input id)
|
|
{
|
|
char *link_argv[] = {
|
|
NULL,
|
|
"link",
|
|
"get_bss",
|
|
NULL,
|
|
};
|
|
char *station_argv[] = {
|
|
NULL,
|
|
"link",
|
|
"get_sta",
|
|
NULL,
|
|
NULL,
|
|
};
|
|
char bssid_buf[3*6];
|
|
int err;
|
|
|
|
link_argv[0] = argv[0];
|
|
err = handle_cmd(state, id, 3, link_argv);
|
|
if (err)
|
|
return err;
|
|
|
|
if (!lr.link_found) {
|
|
if (!lr.anything_found)
|
|
printf("Not connected.\n");
|
|
return 0;
|
|
}
|
|
|
|
mac_addr_n2a(bssid_buf, lr.bssid);
|
|
bssid_buf[17] = '\0';
|
|
|
|
station_argv[0] = argv[0];
|
|
station_argv[3] = bssid_buf;
|
|
return handle_cmd(state, id, 4, station_argv);
|
|
}
|
|
TOPLEVEL(link, NULL, 0, 0, CIB_NETDEV, handle_link,
|
|
"Print information about the current link, if any.");
|
|
HIDDEN(link, get_sta, "", NL80211_CMD_GET_STATION, 0,
|
|
CIB_NETDEV, handle_link_sta);
|
|
HIDDEN(link, get_bss, NULL, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
|
|
CIB_NETDEV, handle_scan_for_link);
|