waybar/src/modules/network.cpp

149 lines
4.6 KiB
C++
Raw Normal View History

2018-08-09 16:38:24 +02:00
#include "modules/network.hpp"
2018-08-16 14:29:41 +02:00
#include <iostream>
2018-08-09 16:38:24 +02:00
waybar::modules::Network::Network(Json::Value config)
2018-08-16 14:29:41 +02:00
: config_(std::move(config)),
ifid_(if_nametoindex(config_["interface"].asCString()))
2018-08-09 16:38:24 +02:00
{
2018-08-16 14:29:41 +02:00
if (ifid_ == 0) {
2018-08-13 14:05:13 +02:00
throw std::runtime_error("Can't found network interface");
2018-08-16 14:29:41 +02:00
}
label_.set_name("network");
uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 30;
thread_ = [this, interval] {
getInfo();
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Network::update));
2018-08-16 14:29:41 +02:00
thread_.sleep_for(chrono::seconds(interval));
2018-08-09 16:38:24 +02:00
};
};
auto waybar::modules::Network::update() -> void
{
2018-08-16 14:29:41 +02:00
auto format = config_["format"] ? config_["format"].asString() : "{essid}";
label_.set_text(fmt::format(format,
fmt::arg("essid", essid_),
fmt::arg("signaldBm", signal_strength_dbm_),
fmt::arg("signalStrength", signal_strength_)
2018-08-09 22:05:15 +02:00
));
2018-08-09 16:38:24 +02:00
}
2018-08-16 14:29:41 +02:00
int waybar::modules::Network::scanCb(struct nl_msg *msg, void *data) {
2018-08-09 16:38:24 +02:00
auto net = static_cast<waybar::modules::Network *>(data);
auto gnlh = static_cast<genlmsghdr *>(nlmsg_data(nlmsg_hdr(msg)));
struct nlattr* tb[NL80211_ATTR_MAX + 1];
struct nlattr* bss[NL80211_BSS_MAX + 1];
struct nla_policy bss_policy[NL80211_BSS_MAX + 1]{};
bss_policy[NL80211_BSS_TSF].type = NLA_U64;
bss_policy[NL80211_BSS_FREQUENCY].type = NLA_U32;
bss_policy[NL80211_BSS_BSSID].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_BEACON_INTERVAL].type = NLA_U16;
bss_policy[NL80211_BSS_CAPABILITY].type = NLA_U16;
bss_policy[NL80211_BSS_INFORMATION_ELEMENTS].type = NLA_UNSPEC;
bss_policy[NL80211_BSS_SIGNAL_MBM].type = NLA_U32;
bss_policy[NL80211_BSS_SIGNAL_UNSPEC].type = NLA_U8;
bss_policy[NL80211_BSS_STATUS].type = NLA_U32;
2018-08-16 14:29:41 +02:00
if (nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), nullptr) < 0) {
2018-08-09 16:38:24 +02:00
return NL_SKIP;
2018-08-16 14:29:41 +02:00
}
if (tb[NL80211_ATTR_BSS] == nullptr) {
2018-08-09 16:38:24 +02:00
return NL_SKIP;
2018-08-16 14:29:41 +02:00
}
if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS], bss_policy) != 0) {
2018-08-09 16:38:24 +02:00
return NL_SKIP;
2018-08-16 14:29:41 +02:00
}
if (!net->associatedOrJoined(bss)) {
2018-08-09 16:38:24 +02:00
return NL_SKIP;
2018-08-16 14:29:41 +02:00
}
net->parseEssid(bss);
net->parseSignal(bss);
// TODO(someone): parse quality
2018-08-09 16:38:24 +02:00
return NL_SKIP;
}
2018-08-16 14:29:41 +02:00
void waybar::modules::Network::parseEssid(struct nlattr **bss)
2018-08-09 16:38:24 +02:00
{
2018-08-16 14:29:41 +02:00
essid_.clear();
2018-08-09 16:38:24 +02:00
if (bss[NL80211_BSS_INFORMATION_ELEMENTS] != nullptr) {
auto ies =
static_cast<char*>(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]));
auto ies_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
const auto hdr_len = 2;
while (ies_len > hdr_len && ies[0] != 0) {
ies_len -= ies[1] + hdr_len;
ies += ies[1] + hdr_len;
}
if (ies_len > hdr_len && ies_len > ies[1] + hdr_len) {
auto essid_begin = ies + hdr_len;
auto essid_end = essid_begin + ies[1];
2018-08-16 14:29:41 +02:00
std::copy(essid_begin, essid_end, std::back_inserter(essid_));
2018-08-09 16:38:24 +02:00
}
}
}
2018-08-16 14:29:41 +02:00
void waybar::modules::Network::parseSignal(struct nlattr **bss) {
2018-08-09 22:05:15 +02:00
if (bss[NL80211_BSS_SIGNAL_MBM] != nullptr) {
// signalstrength in dBm
2018-08-16 14:29:41 +02:00
signal_strength_dbm_ =
2018-08-09 22:05:15 +02:00
static_cast<int>(nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM])) / 100;
// WiFi-hardware usually operates in the range -90 to -20dBm.
const int hardwareMax = -20;
const int hardwareMin = -90;
2018-08-16 14:29:41 +02:00
signal_strength_ = ((signal_strength_dbm_ - hardwareMin)
/ double{hardwareMax - hardwareMin}) * 100;
2018-08-09 22:05:15 +02:00
}
}
2018-08-16 14:29:41 +02:00
bool waybar::modules::Network::associatedOrJoined(struct nlattr** bss)
2018-08-09 16:38:24 +02:00
{
2018-08-16 14:29:41 +02:00
if (bss[NL80211_BSS_STATUS] == nullptr) {
2018-08-09 16:38:24 +02:00
return false;
2018-08-16 14:29:41 +02:00
}
2018-08-09 16:38:24 +02:00
auto status = nla_get_u32(bss[NL80211_BSS_STATUS]);
switch (status) {
case NL80211_BSS_STATUS_ASSOCIATED:
case NL80211_BSS_STATUS_IBSS_JOINED:
case NL80211_BSS_STATUS_AUTHENTICATED:
return true;
default:
return false;
}
}
2018-08-16 14:29:41 +02:00
auto waybar::modules::Network::getInfo() -> void
2018-08-09 16:38:24 +02:00
{
struct nl_sock *sk = nl_socket_alloc();
if (genl_connect(sk) != 0) {
nl_socket_free(sk);
return;
}
2018-08-16 14:29:41 +02:00
if (nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, scanCb, this) < 0) {
2018-08-09 16:38:24 +02:00
nl_socket_free(sk);
return;
}
const int nl80211_id = genl_ctrl_resolve(sk, "nl80211");
if (nl80211_id < 0) {
nl_socket_free(sk);
return;
}
struct nl_msg *msg = nlmsg_alloc();
2018-08-16 14:29:41 +02:00
if (msg == nullptr) {
2018-08-09 16:38:24 +02:00
nl_socket_free(sk);
return;
}
2018-08-16 14:29:41 +02:00
if (genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, nl80211_id, 0, NLM_F_DUMP,
NL80211_CMD_GET_SCAN, 0) == nullptr
|| nla_put_u32(msg, NL80211_ATTR_IFINDEX, ifid_) < 0) {
2018-08-09 16:38:24 +02:00
nlmsg_free(msg);
return;
}
nl_send_sync(sk, msg);
nl_socket_free(sk);
}
waybar::modules::Network::operator Gtk::Widget &() {
2018-08-16 14:29:41 +02:00
return label_;
2018-08-09 16:38:24 +02:00
}