1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
#pragma once
#include <arpa/inet.h>
#include <ifaddrs.h>
#include <chrono>
#include <cstdlib>
#include "common.hpp"
#include "components/logger.hpp"
#include "errors.hpp"
#include "settings.hpp"
#include "utils/math.hpp"
#if WITH_LIBNL
#include <net/if.h>
struct nl_msg;
struct nlattr;
#else
#include <iwlib.h>
/*
* wirless_tools 29 (and possibly earlier) redefines 'inline' in iwlib.h
* With clang this leads to a conflict in the POLYBAR_NS macro
* wirless_tools 30 doesn't have that issue anymore
*/
#ifdef inline
#undef inline
#endif
#endif
POLYBAR_NS
class file_descriptor;
namespace net {
DEFINE_ERROR(network_error);
bool is_interface_valid(const string& ifname);
std::pair<string, bool> get_canonical_interface(const string& ifname);
bool is_wireless_interface(const string& ifname);
std::string find_wireless_interface();
std::string find_wired_interface();
// types {{{
struct quality_range {
int val{0};
int max{0};
int percentage() const {
if (val < 0) {
return std::max(std::min(std::abs(math_util::percentage(val, max, -20)), 100), 0);
}
return std::max(std::min(math_util::percentage(val, 0, max), 100), 0);
}
};
using bytes_t = unsigned int;
struct link_activity {
bytes_t transmitted{0};
bytes_t received{0};
std::chrono::steady_clock::time_point time;
};
struct link_status {
string ip;
string ip6;
string mac;
link_activity previous{};
link_activity current{};
};
// }}}
// class : network {{{
class network {
public:
explicit network(string interface);
virtual ~network() {}
virtual bool query(bool accumulate = false);
virtual bool connected() const = 0;
virtual bool ping() const;
string ip() const;
string ip6() const;
string mac() const;
string downspeed(int minwidth = 3, const string& unit = "B/s") const;
string upspeed(int minwidth = 3, const string& unit = "B/s") const;
string netspeed(int minwidth = 3, const string& unit = "B/s") const;
void set_unknown_up(bool unknown = true);
protected:
void check_tuntap_or_bridge();
bool test_interface() const;
string format_speedrate(float bytes_diff, int minwidth, const string& unit) const;
void query_ip6();
const logger& m_log;
unique_ptr<file_descriptor> m_socketfd;
link_status m_status{};
string m_interface;
bool m_tuntap{false};
bool m_bridge{false};
bool m_unknown_up{false};
};
// }}}
// class : wired_network {{{
class wired_network : public network {
public:
explicit wired_network(string interface) : network(interface) {}
bool query(bool accumulate = false) override;
bool connected() const override;
string linkspeed() const;
private:
int m_linkspeed{0};
};
// }}}
#if WITH_LIBNL
// class : wireless_network {{{
class wireless_network : public network {
public:
wireless_network(string interface) : network(interface), m_ifid(if_nametoindex(interface.c_str())){};
bool query(bool accumulate = false) override;
bool connected() const override;
string essid() const;
int signal() const;
int quality() const;
protected:
static int scan_cb(struct nl_msg* msg, void* instance);
bool associated_or_joined(struct nlattr** bss);
void parse_essid(struct nlattr** bss);
void parse_frequency(struct nlattr** bss);
void parse_quality(struct nlattr** bss);
void parse_signal(struct nlattr** bss);
private:
unsigned int m_ifid{};
string m_essid{};
int m_frequency{};
quality_range m_signalstrength{};
quality_range m_linkquality{};
};
// }}}
#else
// class : wireless_network {{{
class wireless_network : public network {
public:
wireless_network(string interface) : network(interface) {}
bool query(bool accumulate = false) override;
bool connected() const override;
string essid() const;
int signal() const;
int quality() const;
protected:
void query_essid(const int& socket_fd);
void query_quality(const int& socket_fd);
private:
shared_ptr<wireless_info> m_info{};
string m_essid{};
quality_range m_signalstrength{};
quality_range m_linkquality{};
};
// }}}
#endif
using wireless_t = unique_ptr<wireless_network>;
using wired_t = unique_ptr<wired_network>;
} // namespace net
POLYBAR_NS_END
|
