addded netlink and iw based wifi scanner for linux
This commit is contained in:
@@ -65,7 +65,7 @@ else()
|
|||||||
# system specific compiler flags
|
# system specific compiler flags
|
||||||
ADD_DEFINITIONS(
|
ADD_DEFINITIONS(
|
||||||
|
|
||||||
-std=gnu++11
|
-std=gnu++11
|
||||||
|
|
||||||
-Wall
|
-Wall
|
||||||
-Werror=return-type
|
-Werror=return-type
|
||||||
@@ -95,11 +95,16 @@ ADD_EXECUTABLE(
|
|||||||
${SOURCES}
|
${SOURCES}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
SET(EXTRA_LIBS ${EXTRA_LIBS} nl-genl-3 nl-3)
|
||||||
|
INCLUDE_DIRECTORIES(/usr/include/libnl3/)
|
||||||
|
SET(EXTRA_LIBS ${EXTRA_LIBS} iw)
|
||||||
|
|
||||||
# needed external libraries
|
# needed external libraries
|
||||||
TARGET_LINK_LIBRARIES(
|
TARGET_LINK_LIBRARIES(
|
||||||
${PROJECT_NAME}
|
${PROJECT_NAME}
|
||||||
gtest
|
gtest
|
||||||
pthread
|
pthread
|
||||||
|
${EXTRA_LIBS}
|
||||||
)
|
)
|
||||||
|
|
||||||
SET(CMAKE_C_COMPILER ${CMAKE_CXX_COMPILER})
|
SET(CMAKE_C_COMPILER ${CMAKE_CXX_COMPILER})
|
||||||
|
|||||||
15
main.cpp
15
main.cpp
@@ -10,8 +10,17 @@ class Test : public GridPoint {
|
|||||||
|
|
||||||
#include "tests/Tests.h"
|
#include "tests/Tests.h"
|
||||||
|
|
||||||
|
#include "sensors/radio/scan/WiFiScanLinux.h"
|
||||||
|
void wifi() {
|
||||||
|
const std::string dev = "wlp0s20u2u1";
|
||||||
|
WiFiScanLinux scanner(dev);
|
||||||
|
scanner.scan();
|
||||||
|
}
|
||||||
|
|
||||||
int main(int argc, char** argv) {
|
int main(int argc, char** argv) {
|
||||||
|
|
||||||
|
wifi(); return 0;
|
||||||
|
|
||||||
|
|
||||||
#ifdef WITH_TESTS
|
#ifdef WITH_TESTS
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
::testing::InitGoogleTest(&argc, argv);
|
||||||
@@ -25,8 +34,8 @@ int main(int argc, char** argv) {
|
|||||||
//::testing::GTEST_FLAG(filter) = "*Grid.*";
|
//::testing::GTEST_FLAG(filter) = "*Grid.*";
|
||||||
//::testing::GTEST_FLAG(filter) = "*Dijkstra.*";
|
//::testing::GTEST_FLAG(filter) = "*Dijkstra.*";
|
||||||
|
|
||||||
//::testing::GTEST_FLAG(filter) = "*LogDistanceCeilingModelBeacon*";
|
//::testing::GTEST_FLAG(filter) = "*LogDistanceCeilingModelBeacon*";
|
||||||
//::testing::GTEST_FLAG(filter) = "*WiFiOptimizer*";
|
//::testing::GTEST_FLAG(filter) = "*WiFiOptimizer*";
|
||||||
|
|
||||||
|
|
||||||
//::testing::GTEST_FLAG(filter) = "*Offline.readWrite*";
|
//::testing::GTEST_FLAG(filter) = "*Offline.readWrite*";
|
||||||
@@ -41,7 +50,7 @@ int main(int argc, char** argv) {
|
|||||||
//::testing::GTEST_FLAG(filter) = "*BVH*";
|
//::testing::GTEST_FLAG(filter) = "*BVH*";
|
||||||
|
|
||||||
|
|
||||||
//::testing::GTEST_FLAG(filter) = "*Barometer*";
|
//::testing::GTEST_FLAG(filter) = "*Barometer*";
|
||||||
//::testing::GTEST_FLAG(filter) = "*GridWalk2RelPressure*";
|
//::testing::GTEST_FLAG(filter) = "*GridWalk2RelPressure*";
|
||||||
|
|
||||||
//::testing::GTEST_FLAG(filter) = "Heading*";
|
//::testing::GTEST_FLAG(filter) = "Heading*";
|
||||||
|
|||||||
@@ -47,27 +47,27 @@ public:
|
|||||||
MACAddress(const std::string& str) {
|
MACAddress(const std::string& str) {
|
||||||
|
|
||||||
// sanity check
|
// sanity check
|
||||||
if (str.size() == 17 ){
|
if (str.size() == 17 ){
|
||||||
mac = 0; // all zeros
|
mac = 0; // all zeros
|
||||||
fields.h5 = hexWordToInt(str[ 0], str[ 1]);
|
fields.h5 = hexWordToInt(str[ 0], str[ 1]);
|
||||||
fields.h4 = hexWordToInt(str[ 3], str[ 4]);
|
fields.h4 = hexWordToInt(str[ 3], str[ 4]);
|
||||||
fields.h3 = hexWordToInt(str[ 6], str[ 7]);
|
fields.h3 = hexWordToInt(str[ 6], str[ 7]);
|
||||||
fields.h2 = hexWordToInt(str[ 9], str[10]);
|
fields.h2 = hexWordToInt(str[ 9], str[10]);
|
||||||
fields.h1 = hexWordToInt(str[12], str[13]);
|
fields.h1 = hexWordToInt(str[12], str[13]);
|
||||||
fields.h0 = hexWordToInt(str[15], str[16]);
|
fields.h0 = hexWordToInt(str[15], str[16]);
|
||||||
}
|
}
|
||||||
else if (str.size() == 12){
|
else if (str.size() == 12){
|
||||||
mac = 0; // all zeros
|
mac = 0; // all zeros
|
||||||
fields.h5 = hexWordToInt(str[ 0], str[ 1]);
|
fields.h5 = hexWordToInt(str[ 0], str[ 1]);
|
||||||
fields.h4 = hexWordToInt(str[ 2], str[ 3]);
|
fields.h4 = hexWordToInt(str[ 2], str[ 3]);
|
||||||
fields.h3 = hexWordToInt(str[ 4], str[ 5]);
|
fields.h3 = hexWordToInt(str[ 4], str[ 5]);
|
||||||
fields.h2 = hexWordToInt(str[ 6], str[ 7]);
|
fields.h2 = hexWordToInt(str[ 6], str[ 7]);
|
||||||
fields.h1 = hexWordToInt(str[ 8], str[ 9]);
|
fields.h1 = hexWordToInt(str[ 8], str[ 9]);
|
||||||
fields.h0 = hexWordToInt(str[10], str[11]);
|
fields.h0 = hexWordToInt(str[10], str[11]);
|
||||||
}
|
}
|
||||||
else{
|
else{
|
||||||
throw Exception("invalid hex string length. must be 17 or 12 (without :)");
|
throw Exception("invalid hex string length. must be 17 or 12 (without :)");
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
30
sensors/radio/scan/WiFiChannels.h
Normal file
30
sensors/radio/scan/WiFiChannels.h
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
#ifndef INDOOR_WIFICHANNELS_H
|
||||||
|
#define INDOOR_WIFICHANNELS_H
|
||||||
|
|
||||||
|
class WiFiChannels {
|
||||||
|
|
||||||
|
public:
|
||||||
|
|
||||||
|
static int freqToChannel(const int freq) {
|
||||||
|
switch(freq) {
|
||||||
|
case 2412: return 1;
|
||||||
|
case 2417: return 2;
|
||||||
|
case 2422: return 3;
|
||||||
|
case 2427: return 4;
|
||||||
|
case 2432: return 5;
|
||||||
|
case 2437: return 6;
|
||||||
|
case 2442: return 7;
|
||||||
|
case 2447: return 8;
|
||||||
|
case 2452: return 9;
|
||||||
|
case 2457: return 10;
|
||||||
|
case 2462: return 11;
|
||||||
|
case 2467: return 12;
|
||||||
|
case 2472: return 13;
|
||||||
|
case 2484: return 14;
|
||||||
|
}
|
||||||
|
throw "error";
|
||||||
|
}
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // INDOOR_WIFICHANNELS_H
|
||||||
15
sensors/radio/scan/WiFiScan.h
Normal file
15
sensors/radio/scan/WiFiScan.h
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
#ifndef INDOOR_WIFI_SCAN_H
|
||||||
|
#define INDOOR_WIFI_SCAN_H
|
||||||
|
|
||||||
|
#include "../WiFiMeasurements.h"
|
||||||
|
|
||||||
|
class WiFiScan {
|
||||||
|
|
||||||
|
public:
|
||||||
|
|
||||||
|
/** scan for nearby access points */
|
||||||
|
virtual WiFiMeasurements scan() = 0;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif //INDOOR_WIFI_SCAN_H
|
||||||
419
sensors/radio/scan/WiFiScanLinux.h
Normal file
419
sensors/radio/scan/WiFiScanLinux.h
Normal file
@@ -0,0 +1,419 @@
|
|||||||
|
#ifndef INDOOR_WIFISCANNER_LINUX_H
|
||||||
|
#define INDOOR_WIFISCANNER_LINUX_H
|
||||||
|
|
||||||
|
/*
|
||||||
|
* scan_access_points.c: Prints all detected access points with wlan0 using NL80211 (netlink).
|
||||||
|
*
|
||||||
|
* Only works on network interfaces whose drivers are compatible with Netlink. Test this by running `iw list`.
|
||||||
|
*
|
||||||
|
* Since only privileged users may submit NL80211_CMD_TRIGGER_SCAN, you'll have to run the compiled program as root.
|
||||||
|
*
|
||||||
|
* Build with: gcc $(pkg-config --cflags --libs libnl-genl-3.0) scan_access_points.c
|
||||||
|
*
|
||||||
|
* Raspbian prerequisites:
|
||||||
|
* sudo apt-get install libnl-genl-3-dev
|
||||||
|
*
|
||||||
|
* Resources:
|
||||||
|
* http://git.kernel.org/cgit/linux/kernel/git/jberg/iw.git/tree/scan.c
|
||||||
|
* http://stackoverflow.com/questions/21601521/how-to-use-the-libnl-library-to-trigger-nl80211-commands
|
||||||
|
* http://stackoverflow.com/questions/23760780/how-to-send-single-channel-scan-request-to-libnl-and-receive-single-
|
||||||
|
*/
|
||||||
|
#include <errno.h>
|
||||||
|
#include <netlink/errno.h>
|
||||||
|
#include <netlink/netlink.h>
|
||||||
|
#include <netlink/genl/genl.h>
|
||||||
|
#include <netlink/genl/ctrl.h>
|
||||||
|
#include <linux/nl80211.h>
|
||||||
|
#include <net/if.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <ctype.h>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#include "WiFiScan.h"
|
||||||
|
|
||||||
|
class WiFiScanLinux : public WiFiScan {
|
||||||
|
|
||||||
|
struct trigger_results {
|
||||||
|
int done;
|
||||||
|
int aborted;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
struct handler_args { // For family_handler() and nl_get_multicast_id().
|
||||||
|
const char *group;
|
||||||
|
int id;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
static int error_handler(struct sockaddr_nl* nla, struct nlmsgerr* err, void* arg) {
|
||||||
|
(void) nla;
|
||||||
|
// Callback for errors.
|
||||||
|
printf("error_handler() called.\n");
|
||||||
|
int* ret = (int*) arg;
|
||||||
|
*ret = err->error;
|
||||||
|
return NL_STOP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int finish_handler(struct nl_msg* msg, void* arg) {
|
||||||
|
(void) msg;
|
||||||
|
// Callback for NL_CB_FINISH.
|
||||||
|
int* ret = (int*) arg;
|
||||||
|
*ret = 0;
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int ack_handler(struct nl_msg* msg, void* arg) {
|
||||||
|
(void) msg;
|
||||||
|
// Callback for NL_CB_ACK.
|
||||||
|
int* ret = (int*)arg;
|
||||||
|
*ret = 0;
|
||||||
|
return NL_STOP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int no_seq_check(struct nl_msg* msg, void* arg) {
|
||||||
|
(void) msg;
|
||||||
|
(void) arg;
|
||||||
|
// Callback for NL_CB_SEQ_CHECK.
|
||||||
|
return NL_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int family_handler(struct nl_msg* msg, void* arg) {
|
||||||
|
// Callback for NL_CB_VALID within nl_get_multicast_id(). From http://sourcecodebrowser.com/iw/0.9.14/genl_8c.html.
|
||||||
|
struct handler_args* grp = (struct handler_args*) arg;
|
||||||
|
struct nlattr *tb[CTRL_ATTR_MAX + 1];
|
||||||
|
struct genlmsghdr* gnlh = (struct genlmsghdr*) nlmsg_data(nlmsg_hdr(msg));
|
||||||
|
struct nlattr *mcgrp;
|
||||||
|
int rem_mcgrp;
|
||||||
|
|
||||||
|
nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL);
|
||||||
|
|
||||||
|
if (!tb[CTRL_ATTR_MCAST_GROUPS]) return NL_SKIP;
|
||||||
|
|
||||||
|
nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], rem_mcgrp) { // This is a loop.
|
||||||
|
struct nlattr* tb_mcgrp[CTRL_ATTR_MCAST_GRP_MAX + 1];
|
||||||
|
|
||||||
|
nla_parse(tb_mcgrp, CTRL_ATTR_MCAST_GRP_MAX, (struct nlattr*) nla_data(mcgrp), nla_len(mcgrp), NULL);
|
||||||
|
|
||||||
|
if (!tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME] || !tb_mcgrp[CTRL_ATTR_MCAST_GRP_ID]) continue;
|
||||||
|
if (strncmp((const char*) nla_data(tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME]), grp->group,
|
||||||
|
nla_len(tb_mcgrp[CTRL_ATTR_MCAST_GRP_NAME]))) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
grp->id = nla_get_u32(tb_mcgrp[CTRL_ATTR_MCAST_GRP_ID]);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int nl_get_multicast_id(struct nl_sock *sock, const char *family, const char *group) {
|
||||||
|
// From http://sourcecodebrowser.com/iw/0.9.14/genl_8c.html.
|
||||||
|
struct nl_msg *msg;
|
||||||
|
struct nl_cb *cb;
|
||||||
|
int ret, ctrlid;
|
||||||
|
struct handler_args grp = { .group = group, .id = -ENOENT, };
|
||||||
|
|
||||||
|
msg = nlmsg_alloc();
|
||||||
|
if (!msg) return -ENOMEM;
|
||||||
|
|
||||||
|
cb = nl_cb_alloc(NL_CB_DEFAULT);
|
||||||
|
if (!cb) {
|
||||||
|
ret = -ENOMEM;
|
||||||
|
goto out_fail_cb;
|
||||||
|
}
|
||||||
|
|
||||||
|
ctrlid = genl_ctrl_resolve(sock, "nlctrl");
|
||||||
|
|
||||||
|
genlmsg_put(msg, 0, 0, ctrlid, 0, 0, CTRL_CMD_GETFAMILY, 0);
|
||||||
|
|
||||||
|
ret = -ENOBUFS;
|
||||||
|
NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
|
||||||
|
|
||||||
|
ret = nl_send_auto_complete(sock, msg);
|
||||||
|
if (ret < 0) goto out;
|
||||||
|
|
||||||
|
ret = 1;
|
||||||
|
|
||||||
|
nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &ret);
|
||||||
|
nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &ret);
|
||||||
|
nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, family_handler, &grp);
|
||||||
|
|
||||||
|
while (ret > 0) nl_recvmsgs(sock, cb);
|
||||||
|
|
||||||
|
if (ret == 0) ret = grp.id;
|
||||||
|
|
||||||
|
nla_put_failure:
|
||||||
|
out:
|
||||||
|
nl_cb_put(cb);
|
||||||
|
out_fail_cb:
|
||||||
|
nlmsg_free(msg);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void mac_addr_n2a(char *mac_addr, unsigned char *arg) {
|
||||||
|
// From http://git.kernel.org/cgit/linux/kernel/git/jberg/iw.git/tree/util.c.
|
||||||
|
int i, l;
|
||||||
|
|
||||||
|
l = 0;
|
||||||
|
for (i = 0; i < 6; i++) {
|
||||||
|
if (i == 0) {
|
||||||
|
sprintf(mac_addr+l, "%02x", arg[i]);
|
||||||
|
l += 2;
|
||||||
|
} else {
|
||||||
|
sprintf(mac_addr+l, ":%02x", arg[i]);
|
||||||
|
l += 3;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void print_ssid(unsigned char *ie, int ielen) {
|
||||||
|
uint8_t len;
|
||||||
|
uint8_t *data;
|
||||||
|
int i;
|
||||||
|
|
||||||
|
while (ielen >= 2 && ielen >= ie[1]) {
|
||||||
|
if (ie[0] == 0 && ie[1] >= 0 && ie[1] <= 32) {
|
||||||
|
len = ie[1];
|
||||||
|
data = ie + 2;
|
||||||
|
for (i = 0; i < len; i++) {
|
||||||
|
if (isprint(data[i]) && data[i] != ' ' && data[i] != '\\') printf("%c", data[i]);
|
||||||
|
else if (data[i] == ' ' && (i != 0 && i != len -1)) printf(" ");
|
||||||
|
else printf("\\x%.2x", data[i]);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
ielen -= ie[1] + 2;
|
||||||
|
ie += ie[1] + 2;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int callback_trigger(struct nl_msg *msg, void *arg) {
|
||||||
|
|
||||||
|
// Called by the kernel when the scan is done or has been aborted.
|
||||||
|
struct genlmsghdr* gnlh = (struct genlmsghdr*) nlmsg_data(nlmsg_hdr(msg));
|
||||||
|
struct trigger_results* results = (struct trigger_results*) arg;
|
||||||
|
|
||||||
|
//printf("Got something.\n");
|
||||||
|
//printf("%d\n", arg);
|
||||||
|
//nl_msg_dump(msg, stdout);
|
||||||
|
|
||||||
|
if (gnlh->cmd == NL80211_CMD_SCAN_ABORTED) {
|
||||||
|
printf("Got NL80211_CMD_SCAN_ABORTED.\n");
|
||||||
|
results->done = 1;
|
||||||
|
results->aborted = 1;
|
||||||
|
} else if (gnlh->cmd == NL80211_CMD_NEW_SCAN_RESULTS) {
|
||||||
|
printf("Got NL80211_CMD_NEW_SCAN_RESULTS.\n");
|
||||||
|
results->done = 1;
|
||||||
|
results->aborted = 0;
|
||||||
|
} // else probably an uninteresting multicast message.
|
||||||
|
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int callback_dump(struct nl_msg* msg, void* arg) {
|
||||||
|
|
||||||
|
(void) arg;
|
||||||
|
|
||||||
|
// Called by the kernel with a dump of the successful scan's data. Called for each SSID.
|
||||||
|
struct genlmsghdr* gnlh = (struct genlmsghdr*) nlmsg_data(nlmsg_hdr(msg));
|
||||||
|
char mac_addr[20];
|
||||||
|
struct nlattr *tb[NL80211_ATTR_MAX + 1];
|
||||||
|
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 },
|
||||||
|
// [NL80211_BSS_SEEN_MS_AGO] = { .type = NLA_U32 },
|
||||||
|
// [NL80211_BSS_BEACON_IES] = { },
|
||||||
|
// };
|
||||||
|
static struct nla_policy bss_policy[NL80211_BSS_MAX + 1];
|
||||||
|
memset(&bss_policy, 0, sizeof(bss_policy));
|
||||||
|
bss_policy[NL80211_BSS_TSF].type = NLA_U64;
|
||||||
|
bss_policy[NL80211_BSS_FREQUENCY].type = NLA_U32;
|
||||||
|
bss_policy[NL80211_BSS_BSSID];// = { };
|
||||||
|
bss_policy[NL80211_BSS_BEACON_INTERVAL].type = NLA_U16;
|
||||||
|
bss_policy[NL80211_BSS_CAPABILITY].type = NLA_U16;
|
||||||
|
bss_policy[NL80211_BSS_INFORMATION_ELEMENTS];// = { };
|
||||||
|
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;
|
||||||
|
bss_policy[NL80211_BSS_SEEN_MS_AGO].type = NLA_U32;
|
||||||
|
bss_policy[NL80211_BSS_BEACON_IES];// = { };
|
||||||
|
|
||||||
|
|
||||||
|
// Parse and error check.
|
||||||
|
nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL);
|
||||||
|
if (!tb[NL80211_ATTR_BSS]) {
|
||||||
|
printf("bss info missing!\n");
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS], bss_policy)) {
|
||||||
|
printf("failed to parse nested attributes!\n");
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
if (!bss[NL80211_BSS_BSSID]) return NL_SKIP;
|
||||||
|
if (!bss[NL80211_BSS_INFORMATION_ELEMENTS]) return NL_SKIP;
|
||||||
|
|
||||||
|
const uint64_t seen_ago_ms = nla_get_u32(bss[NL80211_BSS_SEEN_MS_AGO]);
|
||||||
|
const int rssi = (nla_get_s32(bss[NL80211_BSS_SIGNAL_MBM])) / 100.0f;
|
||||||
|
|
||||||
|
// Start printing.
|
||||||
|
mac_addr_n2a(mac_addr, (unsigned char*) nla_data(bss[NL80211_BSS_BSSID]));
|
||||||
|
printf("%s, ", mac_addr);
|
||||||
|
printf("%d MHz, ", nla_get_u32(bss[NL80211_BSS_FREQUENCY]));
|
||||||
|
print_ssid((unsigned char*) nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]), nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]));
|
||||||
|
printf(" %d ms", seen_ago_ms);
|
||||||
|
printf(" %d dBm", rssi);
|
||||||
|
printf("\n");
|
||||||
|
|
||||||
|
return NL_SKIP;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int do_scan_trigger(struct nl_sock *socket, int if_index, int driver_id) {
|
||||||
|
// Starts the scan and waits for it to finish. Does not return until the scan is done or has been aborted.
|
||||||
|
struct trigger_results results = { .done = 0, .aborted = 0 };
|
||||||
|
struct nl_msg *msg;
|
||||||
|
struct nl_cb *cb;
|
||||||
|
struct nl_msg *ssids_to_scan;
|
||||||
|
int err;
|
||||||
|
int ret;
|
||||||
|
int mcid = nl_get_multicast_id(socket, "nl80211", "scan");
|
||||||
|
nl_socket_add_membership(socket, mcid); // Without this, callback_trigger() won't be called.
|
||||||
|
|
||||||
|
// Allocate the messages and callback handler.
|
||||||
|
msg = nlmsg_alloc();
|
||||||
|
if (!msg) {
|
||||||
|
printf("ERROR: Failed to allocate netlink message for msg.\n");
|
||||||
|
return -ENOMEM;
|
||||||
|
}
|
||||||
|
ssids_to_scan = nlmsg_alloc();
|
||||||
|
if (!ssids_to_scan) {
|
||||||
|
printf("ERROR: Failed to allocate netlink message for ssids_to_scan.\n");
|
||||||
|
nlmsg_free(msg);
|
||||||
|
return -ENOMEM;
|
||||||
|
}
|
||||||
|
cb = nl_cb_alloc(NL_CB_DEFAULT);
|
||||||
|
if (!cb) {
|
||||||
|
printf("ERROR: Failed to allocate netlink callbacks.\n");
|
||||||
|
nlmsg_free(msg);
|
||||||
|
nlmsg_free(ssids_to_scan);
|
||||||
|
return -ENOMEM;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setup the messages and callback handler.
|
||||||
|
genlmsg_put(msg, 0, 0, driver_id, 0, 0, NL80211_CMD_TRIGGER_SCAN, 0); // Setup which command to run.
|
||||||
|
nla_put_u32(msg, NL80211_ATTR_IFINDEX, if_index); // Add message attribute, which interface to use.
|
||||||
|
nla_put(ssids_to_scan, 1, 0, ""); // Scan all SSIDs.
|
||||||
|
nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids_to_scan); // Add message attribute, which SSIDs to scan for.
|
||||||
|
nlmsg_free(ssids_to_scan); // Copied to `msg` above, no longer need this.
|
||||||
|
nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, callback_trigger, &results); // Add the callback.
|
||||||
|
nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
|
||||||
|
nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
|
||||||
|
nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
|
||||||
|
nl_cb_set(cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, no_seq_check, NULL); // No sequence checking for multicast messages.
|
||||||
|
|
||||||
|
// Send NL80211_CMD_TRIGGER_SCAN to start the scan. The kernel may reply with NL80211_CMD_NEW_SCAN_RESULTS on
|
||||||
|
// success or NL80211_CMD_SCAN_ABORTED if another scan was started by another process.
|
||||||
|
err = 1;
|
||||||
|
ret = nl_send_auto(socket, msg); // Send the message.
|
||||||
|
printf("NL80211_CMD_TRIGGER_SCAN sent %d bytes to the kernel.\n", ret);
|
||||||
|
printf("Waiting for scan to complete...\n");
|
||||||
|
while (err > 0) ret = nl_recvmsgs(socket, cb); // First wait for ack_handler(). This helps with basic errors.
|
||||||
|
if (err < 0) {
|
||||||
|
printf("WARNING: err has a value of %d.\n", err);
|
||||||
|
}
|
||||||
|
if (ret < 0) {
|
||||||
|
printf("ERROR: nl_recvmsgs() returned %d (%s).\n", ret, nl_geterror(-ret));
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
while (!results.done) nl_recvmsgs(socket, cb); // Now wait until the scan is done or aborted.
|
||||||
|
if (results.aborted) {
|
||||||
|
printf("ERROR: Kernel aborted scan.\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("Scan is done.\n");
|
||||||
|
|
||||||
|
// Cleanup.
|
||||||
|
nlmsg_free(msg);
|
||||||
|
nl_cb_put(cb);
|
||||||
|
nl_socket_drop_membership(socket, mcid); // No longer need this.
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int if_index;
|
||||||
|
struct nl_sock* socket;
|
||||||
|
int driver_id;
|
||||||
|
|
||||||
|
public:
|
||||||
|
|
||||||
|
WiFiScanLinux(const std::string& devName) {
|
||||||
|
|
||||||
|
if_index = if_nametoindex("wlp0s20u2u1"); // Use this wireless interface for scanning.
|
||||||
|
|
||||||
|
|
||||||
|
// Open socket to kernel.
|
||||||
|
socket = nl_socket_alloc(); // Allocate new netlink socket in memory.
|
||||||
|
genl_connect(socket); // Create file descriptor and bind socket.
|
||||||
|
driver_id = genl_ctrl_resolve(socket, "nl80211"); // Find the nl80211 driver ID.
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/** triger WiFiScan and fetch the result */
|
||||||
|
WiFiMeasurements scan() {
|
||||||
|
|
||||||
|
// Issue NL80211_CMD_TRIGGER_SCAN to the kernel and wait for it to finish.
|
||||||
|
int err = do_scan_trigger(socket, if_index, driver_id);
|
||||||
|
if (err != 0) {
|
||||||
|
printf("do_scan_trigger() failed with %d.\n", err);
|
||||||
|
throw "error";
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now get info for all SSIDs detected.
|
||||||
|
struct nl_msg *msg = nlmsg_alloc(); // Allocate a message.
|
||||||
|
genlmsg_put(msg, 0, 0, driver_id, 0, NLM_F_DUMP, NL80211_CMD_GET_SCAN, 0); // Setup which command to run.
|
||||||
|
nla_put_u32(msg, NL80211_ATTR_IFINDEX, if_index); // Add message attribute, which interface to use.
|
||||||
|
nl_socket_modify_cb(socket, NL_CB_VALID, NL_CB_CUSTOM, callback_dump, NULL); // Add the callback.
|
||||||
|
int ret = nl_send_auto(socket, msg); // Send the message.
|
||||||
|
printf("NL80211_CMD_GET_SCAN sent %d bytes to the kernel.\n", ret);
|
||||||
|
ret = nl_recvmsgs_default(socket); // Retrieve the kernel's answer. callback_dump() prints SSIDs to stdout.
|
||||||
|
nlmsg_free(msg);
|
||||||
|
if (ret < 0) {
|
||||||
|
printf("ERROR: nl_recvmsgs_default() returned %d (%s).\n", ret, nl_geterror(-ret));
|
||||||
|
throw "error";
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO
|
||||||
|
WiFiMeasurements mes;
|
||||||
|
return mes;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
// gcc -I /usr/include/libnl3 main.c -lnl-genl-3 -lnl-3
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#endif // INDOOR_WIFISCANNER_LINUX_H
|
||||||
100
sensors/radio/scan/WiFiScanLinux_IW.h
Normal file
100
sensors/radio/scan/WiFiScanLinux_IW.h
Normal file
@@ -0,0 +1,100 @@
|
|||||||
|
#ifndef INDOOR_WIFI_SCAN_LINUX_H
|
||||||
|
#define INDOOR_WIFI_SCAN_LINUX_H
|
||||||
|
|
||||||
|
#include "WiFiChannels.h"
|
||||||
|
#include "WiFiScan.h"
|
||||||
|
#include "../WiFiMeasurements.h"
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <time.h>
|
||||||
|
#include <iwlib.h>
|
||||||
|
#include <string>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
class WiFiScanLinux : public WiFiScan {
|
||||||
|
|
||||||
|
private:
|
||||||
|
|
||||||
|
wireless_scan_head head;
|
||||||
|
wireless_scan *result;
|
||||||
|
iwrange range;
|
||||||
|
int sock;
|
||||||
|
|
||||||
|
std::string dev;
|
||||||
|
|
||||||
|
public:
|
||||||
|
|
||||||
|
WiFiScanLinux(const std::string& dev) : dev(dev) {
|
||||||
|
|
||||||
|
/* Open socket to kernel */
|
||||||
|
sock = iw_sockets_open();
|
||||||
|
std::cout << sock << std::endl;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
WiFiMeasurements scan() override {
|
||||||
|
|
||||||
|
WiFiMeasurements res;
|
||||||
|
char* dev = (char*) this->dev.c_str();
|
||||||
|
|
||||||
|
/* Get some metadata to use for scanning */
|
||||||
|
if (iw_get_range_info(sock, dev, &range) < 0) {
|
||||||
|
printf("Error during iw_get_range_info. Aborting.\n");
|
||||||
|
exit(2);
|
||||||
|
}
|
||||||
|
if (range.we_version_compiled < 14) {
|
||||||
|
printf("scanning not supported");
|
||||||
|
exit(2);
|
||||||
|
}
|
||||||
|
|
||||||
|
// // params
|
||||||
|
// struct iwreq request;
|
||||||
|
// request.u.param.flags = IW_SCAN_DEFAULT;
|
||||||
|
// request.u.param.value = 0;
|
||||||
|
// if (iw_set_ext(sock, dev, SIOCSIWSCAN, &request) == -1) {
|
||||||
|
// printf("iw_set_ext(SIOCSIWSCAN)");
|
||||||
|
// exit(EXIT_FAILURE);
|
||||||
|
// }
|
||||||
|
|
||||||
|
/* Perform the scan */
|
||||||
|
if (iw_scan(sock, dev, range.we_version_compiled, &head) < 0) {
|
||||||
|
printf("Error during iw_scan. Aborting.\n");
|
||||||
|
exit(2);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Traverse the results */
|
||||||
|
result = head.result;
|
||||||
|
while (NULL != result) {
|
||||||
|
|
||||||
|
// access-point's MAC
|
||||||
|
const uint8_t* macPtr = (const uint8_t*) result->ap_addr.sa_data;
|
||||||
|
const uint64_t macLng = ((uint64_t)macPtr[5]<<40)|((uint64_t)macPtr[4]<<32)|((uint64_t)macPtr[3]<<24)|((uint64_t)macPtr[2]<<16)|((uint64_t)macPtr[1]<<8)|((uint64_t)macPtr[0]<<0);
|
||||||
|
const MACAddress mac(macLng);
|
||||||
|
|
||||||
|
const int8_t rssi = result->stats.qual.level;
|
||||||
|
const std::string ssid = result->b.essid;
|
||||||
|
|
||||||
|
result->b.
|
||||||
|
|
||||||
|
const int freq = (result->b.freq)/10e5;
|
||||||
|
const int channel = WiFiChannels::freqToChannel(freq);
|
||||||
|
//std::cout << ssid << "\t" << "freq: " << freq << "\t" << "rssi: " << (int) (rssi) << " dBm" << std::endl;
|
||||||
|
//printf("%s - %d\n", result->b.essid,
|
||||||
|
result = result->next;
|
||||||
|
|
||||||
|
std::cout << mac.asString() << "\t" << ssid << "\t" << channel << "\t" << (int)rssi << " dBm" << std::endl;
|
||||||
|
|
||||||
|
AccessPoint ap(mac, ssid);
|
||||||
|
|
||||||
|
WiFiMeasurement mes(ap, rssi);
|
||||||
|
res.entries.push_back(mes);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return res;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif //INDOOR_WIFI_SCAN_LINUX_H
|
||||||
Reference in New Issue
Block a user