mirror of
https://github.com/berlin-open-wireless-lab/DAWN.git
synced 2025-03-09 15:40:12 +00:00
datastorage: AP, client, probe, auth entry and MAC list converted to linked list datastorage: functions adjusted to take pointers as parameters datastorage: optimised sort and search functions added mac_utils: struct dawn_mac added and comparisons adjusted general: adjust code to call new datastorage functions test_storage: large scale 100 AP, 3000 client, 70k probe added
813 lines
28 KiB
C
813 lines
28 KiB
C
#include <libubus.h>
|
|
|
|
#include "dawn_uci.h"
|
|
#include "datastorage.h"
|
|
#include "ubus.h"
|
|
#include "msghandler.h"
|
|
|
|
|
|
static struct blob_buf network_buf;
|
|
static struct blob_buf data_buf;
|
|
|
|
enum {
|
|
NETWORK_METHOD,
|
|
NETWORK_DATA,
|
|
__NETWORK_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy network_policy[__NETWORK_MAX] = {
|
|
[NETWORK_METHOD] = {.name = "method", .type = BLOBMSG_TYPE_STRING},
|
|
[NETWORK_DATA] = {.name = "data", .type = BLOBMSG_TYPE_STRING},
|
|
};
|
|
|
|
enum {
|
|
HOSTAPD_NOTIFY_BSSID_ADDR,
|
|
HOSTAPD_NOTIFY_CLIENT_ADDR,
|
|
__HOSTAPD_NOTIFY_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy hostapd_notify_policy[__HOSTAPD_NOTIFY_MAX] = {
|
|
[HOSTAPD_NOTIFY_BSSID_ADDR] = {.name = "bssid", .type = BLOBMSG_TYPE_STRING},
|
|
[HOSTAPD_NOTIFY_CLIENT_ADDR] = {.name = "address", .type = BLOBMSG_TYPE_STRING},
|
|
};
|
|
|
|
enum {
|
|
PROB_BSSID_ADDR,
|
|
PROB_CLIENT_ADDR,
|
|
PROB_TARGET_ADDR,
|
|
PROB_SIGNAL,
|
|
PROB_FREQ,
|
|
PROB_HT_CAPABILITIES,
|
|
PROB_VHT_CAPABILITIES,
|
|
PROB_RCPI,
|
|
PROB_RSNI,
|
|
__PROB_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy prob_policy[__PROB_MAX] = {
|
|
[PROB_BSSID_ADDR] = {.name = "bssid", .type = BLOBMSG_TYPE_STRING},
|
|
[PROB_CLIENT_ADDR] = {.name = "address", .type = BLOBMSG_TYPE_STRING},
|
|
[PROB_TARGET_ADDR] = {.name = "target", .type = BLOBMSG_TYPE_STRING},
|
|
[PROB_SIGNAL] = {.name = "signal", .type = BLOBMSG_TYPE_INT32},
|
|
[PROB_FREQ] = {.name = "freq", .type = BLOBMSG_TYPE_INT32},
|
|
[PROB_HT_CAPABILITIES] = {.name = "ht_capabilities", .type = BLOBMSG_TYPE_TABLE},
|
|
[PROB_VHT_CAPABILITIES] = {.name = "vht_capabilities", .type = BLOBMSG_TYPE_TABLE},
|
|
[PROB_RCPI] = {.name = "rcpi", .type = BLOBMSG_TYPE_INT32},
|
|
[PROB_RSNI] = {.name = "rsni", .type = BLOBMSG_TYPE_INT32},
|
|
};
|
|
|
|
enum {
|
|
CLIENT_TABLE,
|
|
CLIENT_TABLE_BSSID,
|
|
CLIENT_TABLE_SSID,
|
|
CLIENT_TABLE_FREQ,
|
|
CLIENT_TABLE_HT,
|
|
CLIENT_TABLE_VHT,
|
|
CLIENT_TABLE_CHAN_UTIL,
|
|
CLIENT_TABLE_NUM_STA,
|
|
CLIENT_TABLE_COL_DOMAIN,
|
|
CLIENT_TABLE_BANDWIDTH,
|
|
CLIENT_TABLE_WEIGHT,
|
|
CLIENT_TABLE_NEIGHBOR,
|
|
CLIENT_TABLE_IFACE,
|
|
CLIENT_TABLE_HOSTNAME,
|
|
__CLIENT_TABLE_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy client_table_policy[__CLIENT_TABLE_MAX] = {
|
|
[CLIENT_TABLE] = {.name = "clients", .type = BLOBMSG_TYPE_TABLE},
|
|
[CLIENT_TABLE_BSSID] = {.name = "bssid", .type = BLOBMSG_TYPE_STRING},
|
|
[CLIENT_TABLE_SSID] = {.name = "ssid", .type = BLOBMSG_TYPE_STRING},
|
|
[CLIENT_TABLE_FREQ] = {.name = "freq", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_HT] = {.name = "ht_supported", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_TABLE_VHT] = {.name = "vht_supported", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_TABLE_CHAN_UTIL] = {.name = "channel_utilization", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_NUM_STA] = {.name = "num_sta", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_COL_DOMAIN] = {.name = "collision_domain", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_BANDWIDTH] = {.name = "bandwidth", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_WEIGHT] = {.name = "ap_weight", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_TABLE_NEIGHBOR] = {.name = "neighbor_report", .type = BLOBMSG_TYPE_STRING},
|
|
[CLIENT_TABLE_IFACE] = {.name = "iface", .type = BLOBMSG_TYPE_STRING},
|
|
[CLIENT_TABLE_HOSTNAME] = {.name = "hostname", .type = BLOBMSG_TYPE_STRING},
|
|
};
|
|
|
|
enum {
|
|
CLIENT_SIGNATURE,
|
|
CLIENT_AUTH,
|
|
CLIENT_ASSOC,
|
|
CLIENT_AUTHORIZED,
|
|
CLIENT_PREAUTH,
|
|
CLIENT_WDS,
|
|
CLIENT_WMM,
|
|
CLIENT_HT,
|
|
CLIENT_VHT,
|
|
CLIENT_WPS,
|
|
CLIENT_MFP,
|
|
CLIENT_AID,
|
|
CLIENT_RRM,
|
|
__CLIENT_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy client_policy[__CLIENT_MAX] = {
|
|
[CLIENT_SIGNATURE] = {.name = "signature", .type = BLOBMSG_TYPE_STRING},
|
|
[CLIENT_AUTH] = {.name = "auth", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_ASSOC] = {.name = "assoc", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_AUTHORIZED] = {.name = "authorized", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_PREAUTH] = {.name = "preauth", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_WDS] = {.name = "wds", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_WMM] = {.name = "wmm", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_HT] = {.name = "ht", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_VHT] = {.name = "vht", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_WPS] = {.name = "wps", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_MFP] = {.name = "mfp", .type = BLOBMSG_TYPE_INT8},
|
|
[CLIENT_AID] = {.name = "aid", .type = BLOBMSG_TYPE_INT32},
|
|
[CLIENT_RRM] = {.name = "rrm", .type = BLOBMSG_TYPE_ARRAY},
|
|
};
|
|
|
|
static int handle_set_probe(struct blob_attr* msg);
|
|
|
|
static int handle_uci_config(struct blob_attr* msg);
|
|
|
|
|
|
int parse_to_hostapd_notify(struct blob_attr* msg, hostapd_notify_entry* notify_req) {
|
|
struct blob_attr* tb[__HOSTAPD_NOTIFY_MAX];
|
|
|
|
blobmsg_parse(hostapd_notify_policy, __HOSTAPD_NOTIFY_MAX, tb, blob_data(msg), blob_len(msg));
|
|
|
|
if (hwaddr_aton(blobmsg_data(tb[HOSTAPD_NOTIFY_BSSID_ADDR]), notify_req->bssid_addr.u8))
|
|
return UBUS_STATUS_INVALID_ARGUMENT;
|
|
|
|
if (hwaddr_aton(blobmsg_data(tb[HOSTAPD_NOTIFY_CLIENT_ADDR]), notify_req->client_addr.u8))
|
|
return UBUS_STATUS_INVALID_ARGUMENT;
|
|
|
|
return 0;
|
|
}
|
|
|
|
probe_entry *parse_to_probe_req(struct blob_attr* msg) {
|
|
struct blob_attr* tb[__PROB_MAX];
|
|
|
|
probe_entry* prob_req = malloc(sizeof(probe_entry));
|
|
if (prob_req == NULL)
|
|
{
|
|
fprintf(stderr, "malloc of probe_entry failed!\n");
|
|
return NULL;
|
|
}
|
|
|
|
blobmsg_parse(prob_policy, __PROB_MAX, tb, blob_data(msg), blob_len(msg));
|
|
|
|
if (hwaddr_aton(blobmsg_data(tb[PROB_BSSID_ADDR]), prob_req->bssid_addr.u8))
|
|
{
|
|
free(prob_req);
|
|
return NULL;
|
|
}
|
|
|
|
if (hwaddr_aton(blobmsg_data(tb[PROB_CLIENT_ADDR]), prob_req->client_addr.u8))
|
|
{
|
|
free(prob_req);
|
|
return NULL;
|
|
}
|
|
|
|
if (hwaddr_aton(blobmsg_data(tb[PROB_TARGET_ADDR]), prob_req->target_addr.u8))
|
|
{
|
|
free(prob_req);
|
|
return NULL;
|
|
}
|
|
|
|
if (tb[PROB_SIGNAL]) {
|
|
prob_req->signal = blobmsg_get_u32(tb[PROB_SIGNAL]);
|
|
}
|
|
|
|
if (tb[PROB_FREQ]) {
|
|
prob_req->freq = blobmsg_get_u32(tb[PROB_FREQ]);
|
|
}
|
|
|
|
if (tb[PROB_RCPI]) {
|
|
prob_req->rcpi = blobmsg_get_u32(tb[PROB_RCPI]);
|
|
}
|
|
else {
|
|
prob_req->rcpi = -1;
|
|
}
|
|
|
|
if (tb[PROB_RSNI]) {
|
|
prob_req->rsni = blobmsg_get_u32(tb[PROB_RSNI]);
|
|
}
|
|
else {
|
|
prob_req->rsni = -1;
|
|
}
|
|
|
|
if (tb[PROB_HT_CAPABILITIES]) {
|
|
prob_req->ht_capabilities = true;
|
|
}
|
|
else
|
|
{
|
|
prob_req->ht_capabilities = false;
|
|
}
|
|
|
|
if (tb[PROB_VHT_CAPABILITIES]) {
|
|
prob_req->vht_capabilities = true;
|
|
}
|
|
else
|
|
{
|
|
prob_req->vht_capabilities = false;
|
|
}
|
|
|
|
return prob_req;
|
|
}
|
|
|
|
int handle_deauth_req(struct blob_attr* msg) {
|
|
|
|
hostapd_notify_entry notify_req;
|
|
parse_to_hostapd_notify(msg, ¬ify_req);
|
|
|
|
pthread_mutex_lock(&client_array_mutex);
|
|
|
|
client* client_entry = client_array_get_client(notify_req.client_addr);
|
|
if (client_entry != NULL)
|
|
client_array_delete(client_entry, false);
|
|
|
|
pthread_mutex_unlock(&client_array_mutex);
|
|
|
|
printf("[WC] Deauth: %s\n", "deauth");
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int handle_set_probe(struct blob_attr* msg) {
|
|
|
|
hostapd_notify_entry notify_req;
|
|
parse_to_hostapd_notify(msg, ¬ify_req);
|
|
|
|
probe_array_set_all_probe_count(notify_req.client_addr, dawn_metric.min_probe_count);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int handle_network_msg(char* msg) {
|
|
struct blob_attr* tb[__NETWORK_MAX];
|
|
char* method;
|
|
char* data;
|
|
|
|
blob_buf_init(&network_buf, 0);
|
|
blobmsg_add_json_from_string(&network_buf, msg);
|
|
|
|
blobmsg_parse(network_policy, __NETWORK_MAX, tb, blob_data(network_buf.head), blob_len(network_buf.head));
|
|
|
|
if (!tb[NETWORK_METHOD] || !tb[NETWORK_DATA]) {
|
|
return -1;
|
|
}
|
|
|
|
method = blobmsg_data(tb[NETWORK_METHOD]);
|
|
data = blobmsg_data(tb[NETWORK_DATA]);
|
|
|
|
printf("Network Method new: %s : %s\n", method, msg);
|
|
|
|
blob_buf_init(&data_buf, 0);
|
|
blobmsg_add_json_from_string(&data_buf, data);
|
|
|
|
if (!data_buf.head) {
|
|
return -1;
|
|
}
|
|
|
|
if (blob_len(data_buf.head) <= 0) {
|
|
return -1;
|
|
}
|
|
|
|
if (strlen(method) < 2) {
|
|
return -1;
|
|
}
|
|
|
|
// add inactive death...
|
|
|
|
// TODO: strncmp() look wrong - should all tests be for n = 5 characters? Shorthand checks?
|
|
if (strncmp(method, "probe", 5) == 0) {
|
|
probe_entry *entry = parse_to_probe_req(data_buf.head);
|
|
if (entry != NULL) {
|
|
entry->time = time(0);
|
|
if (entry != insert_to_array(entry, false, false, false)) // use 802.11k values
|
|
{
|
|
// insert found an existing entry, rather than linking in our new one
|
|
free(entry);
|
|
}
|
|
}
|
|
}
|
|
else if (strncmp(method, "clients", 5) == 0) {
|
|
parse_to_clients(data_buf.head, 0, 0);
|
|
}
|
|
else if (strncmp(method, "deauth", 5) == 0) {
|
|
printf("METHOD DEAUTH\n");
|
|
handle_deauth_req(data_buf.head);
|
|
}
|
|
else if (strncmp(method, "setprobe", 5) == 0) {
|
|
printf("HANDLING SET PROBE!\n");
|
|
handle_set_probe(data_buf.head);
|
|
}
|
|
else if (strncmp(method, "addmac", 5) == 0) {
|
|
parse_add_mac_to_file(data_buf.head);
|
|
}
|
|
else if (strncmp(method, "macfile", 5) == 0) {
|
|
parse_add_mac_to_file(data_buf.head);
|
|
}
|
|
else if (strncmp(method, "uci", 2) == 0) {
|
|
printf("HANDLING UCI!\n");
|
|
handle_uci_config(data_buf.head);
|
|
}
|
|
else if (strncmp(method, "beacon-report", 12) == 0) {
|
|
// TODO: Check beacon report stuff
|
|
|
|
//printf("HANDLING BEACON REPORT NETWORK!\n");
|
|
//printf("The Method for beacon-report is: %s\n", method);
|
|
// ignore beacon reports send via network!, use probe functions for it
|
|
//probe_entry entry; // for now just stay at probe entry stuff...
|
|
//parse_to_beacon_rep(data_buf.head, &entry, true);
|
|
}
|
|
else
|
|
{
|
|
printf("No method fonud for: %s\n", method);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static uint8_t dump_rrm_data(void* data, int len, int type) //modify from examples/blobmsg-example.c in libubox
|
|
{
|
|
uint32_t ret = 0;
|
|
switch (type) {
|
|
case BLOBMSG_TYPE_INT32:
|
|
ret = *(uint32_t*)data;
|
|
break;
|
|
default:
|
|
fprintf(stderr, "wrong type of rrm array\n");
|
|
}
|
|
return (uint8_t)ret;
|
|
}
|
|
|
|
static uint8_t
|
|
dump_rrm_table(struct blob_attr* head, int len) //modify from examples/blobmsg-example.c in libubox
|
|
{
|
|
struct blob_attr* attr;
|
|
uint8_t ret = 0;
|
|
|
|
__blob_for_each_attr(attr, head, len) {
|
|
ret = dump_rrm_data(blobmsg_data(attr), blobmsg_data_len(attr), blob_id(attr));
|
|
return ret;// get the first rrm byte
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
// TOOD: Refactor this!
|
|
static void
|
|
dump_client(struct blob_attr** tb, struct dawn_mac client_addr, const char* bssid_addr, uint32_t freq, uint8_t ht_supported,
|
|
uint8_t vht_supported) {
|
|
client *client_entry = malloc(sizeof(struct client_s));
|
|
if (client_entry == NULL)
|
|
{
|
|
// MUSTDO: Error handling?
|
|
return;
|
|
}
|
|
|
|
hwaddr_aton(bssid_addr, client_entry->bssid_addr.u8);
|
|
client_entry->client_addr = client_addr;
|
|
client_entry->freq = freq;
|
|
client_entry->ht_supported = ht_supported;
|
|
client_entry->vht_supported = vht_supported;
|
|
|
|
if (tb[CLIENT_AUTH]) {
|
|
client_entry->auth = blobmsg_get_u8(tb[CLIENT_AUTH]);
|
|
}
|
|
if (tb[CLIENT_ASSOC]) {
|
|
client_entry->assoc = blobmsg_get_u8(tb[CLIENT_ASSOC]);
|
|
}
|
|
if (tb[CLIENT_AUTHORIZED]) {
|
|
client_entry->authorized = blobmsg_get_u8(tb[CLIENT_AUTHORIZED]);
|
|
}
|
|
if (tb[CLIENT_PREAUTH]) {
|
|
client_entry->preauth = blobmsg_get_u8(tb[CLIENT_PREAUTH]);
|
|
}
|
|
if (tb[CLIENT_WDS]) {
|
|
client_entry->wds = blobmsg_get_u8(tb[CLIENT_WDS]);
|
|
}
|
|
if (tb[CLIENT_WMM]) {
|
|
client_entry->wmm = blobmsg_get_u8(tb[CLIENT_WMM]);
|
|
}
|
|
if (tb[CLIENT_HT]) {
|
|
client_entry->ht = blobmsg_get_u8(tb[CLIENT_HT]);
|
|
}
|
|
if (tb[CLIENT_VHT]) {
|
|
client_entry->vht = blobmsg_get_u8(tb[CLIENT_VHT]);
|
|
}
|
|
if (tb[CLIENT_WPS]) {
|
|
client_entry->wps = blobmsg_get_u8(tb[CLIENT_WPS]);
|
|
}
|
|
if (tb[CLIENT_MFP]) {
|
|
client_entry->mfp = blobmsg_get_u8(tb[CLIENT_MFP]);
|
|
}
|
|
if (tb[CLIENT_AID]) {
|
|
client_entry->aid = blobmsg_get_u32(tb[CLIENT_AID]);
|
|
}
|
|
/* RRM Caps */
|
|
if (tb[CLIENT_RRM]) {
|
|
client_entry->rrm_enabled_capa = dump_rrm_table(blobmsg_data(tb[CLIENT_RRM]),
|
|
blobmsg_data_len(tb[CLIENT_RRM]));// get the first byte from rrm array
|
|
//ap_entry.ap_weight = blobmsg_get_u32(tb[CLIENT_TABLE_RRM]);
|
|
}
|
|
else {
|
|
client_entry->rrm_enabled_capa = 0;
|
|
//ap_entry.ap_weight = 0;
|
|
}
|
|
|
|
// copy signature
|
|
if (tb[CLIENT_SIGNATURE]) {
|
|
strncpy(client_entry->signature, blobmsg_data(tb[CLIENT_SIGNATURE]), SIGNATURE_LEN * sizeof(char));
|
|
}
|
|
else
|
|
{
|
|
memset(client_entry->signature, 0, SIGNATURE_LEN);
|
|
}
|
|
|
|
client_entry->time = time(0);
|
|
|
|
pthread_mutex_lock(&client_array_mutex);
|
|
insert_client_to_array(client_entry);
|
|
pthread_mutex_unlock(&client_array_mutex);
|
|
}
|
|
|
|
static int
|
|
dump_client_table(struct blob_attr* head, int len, const char* bssid_addr, uint32_t freq, uint8_t ht_supported,
|
|
uint8_t vht_supported) {
|
|
struct blob_attr* attr;
|
|
struct blobmsg_hdr* hdr;
|
|
int station_count = 0;
|
|
|
|
__blob_for_each_attr(attr, head, len)
|
|
{
|
|
hdr = blob_data(attr);
|
|
|
|
struct blob_attr* tb[__CLIENT_MAX];
|
|
blobmsg_parse(client_policy, __CLIENT_MAX, tb, blobmsg_data(attr), blobmsg_len(attr));
|
|
//char* str = blobmsg_format_json_indent(attr, true, -1);
|
|
|
|
int tmp_int_mac[ETH_ALEN];
|
|
struct dawn_mac tmp_mac;
|
|
sscanf((char*)hdr->name, MACSTR, STR2MAC(tmp_int_mac));
|
|
for (int i = 0; i < ETH_ALEN; ++i)
|
|
tmp_mac.u8[i] = (uint8_t)tmp_int_mac[i];
|
|
|
|
dump_client(tb, tmp_mac, bssid_addr, freq, ht_supported, vht_supported);
|
|
station_count++;
|
|
}
|
|
return station_count;
|
|
}
|
|
|
|
int parse_to_clients(struct blob_attr* msg, int do_kick, uint32_t id) {
|
|
struct blob_attr* tb[__CLIENT_TABLE_MAX];
|
|
|
|
if (!msg) {
|
|
return -1;
|
|
}
|
|
|
|
if (!blob_data(msg)) {
|
|
return -1;
|
|
}
|
|
|
|
if (blob_len(msg) <= 0) {
|
|
return -1;
|
|
}
|
|
|
|
blobmsg_parse(client_table_policy, __CLIENT_TABLE_MAX, tb, blob_data(msg), blob_len(msg));
|
|
|
|
if (tb[CLIENT_TABLE] && tb[CLIENT_TABLE_BSSID] && tb[CLIENT_TABLE_FREQ]) {
|
|
int num_stations = 0;
|
|
num_stations = dump_client_table(blobmsg_data(tb[CLIENT_TABLE]), blobmsg_data_len(tb[CLIENT_TABLE]),
|
|
blobmsg_data(tb[CLIENT_TABLE_BSSID]), blobmsg_get_u32(tb[CLIENT_TABLE_FREQ]),
|
|
blobmsg_get_u8(tb[CLIENT_TABLE_HT]), blobmsg_get_u8(tb[CLIENT_TABLE_VHT]));
|
|
ap *ap_entry = malloc(sizeof(struct ap_s));
|
|
hwaddr_aton(blobmsg_data(tb[CLIENT_TABLE_BSSID]), ap_entry->bssid_addr.u8);
|
|
ap_entry->freq = blobmsg_get_u32(tb[CLIENT_TABLE_FREQ]);
|
|
|
|
if (tb[CLIENT_TABLE_HT]) {
|
|
ap_entry->ht_support = blobmsg_get_u8(tb[CLIENT_TABLE_HT]);
|
|
}
|
|
else {
|
|
ap_entry->ht_support = false;
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_VHT]) {
|
|
ap_entry->vht_support = blobmsg_get_u8(tb[CLIENT_TABLE_VHT]);
|
|
}
|
|
else
|
|
{
|
|
ap_entry->vht_support = false;
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_CHAN_UTIL]) {
|
|
ap_entry->channel_utilization = blobmsg_get_u32(tb[CLIENT_TABLE_CHAN_UTIL]);
|
|
}
|
|
else // if this is not existing set to 0? //TODO: Consider setting to a value that will not mislead eval_probe_metric(), eg dawn_metric.chan_util_val?
|
|
{
|
|
ap_entry->channel_utilization = 0;
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_SSID]) {
|
|
strcpy((char*)ap_entry->ssid, blobmsg_get_string(tb[CLIENT_TABLE_SSID]));
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_COL_DOMAIN]) {
|
|
ap_entry->collision_domain = blobmsg_get_u32(tb[CLIENT_TABLE_COL_DOMAIN]);
|
|
}
|
|
else {
|
|
ap_entry->collision_domain = -1;
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_BANDWIDTH]) {
|
|
ap_entry->bandwidth = blobmsg_get_u32(tb[CLIENT_TABLE_BANDWIDTH]);
|
|
}
|
|
else {
|
|
ap_entry->bandwidth = -1;
|
|
}
|
|
|
|
ap_entry->station_count = num_stations;
|
|
|
|
if (tb[CLIENT_TABLE_WEIGHT]) {
|
|
ap_entry->ap_weight = blobmsg_get_u32(tb[CLIENT_TABLE_WEIGHT]);
|
|
}
|
|
else {
|
|
ap_entry->ap_weight = 0;
|
|
}
|
|
|
|
|
|
if (tb[CLIENT_TABLE_NEIGHBOR]) {
|
|
strncpy(ap_entry->neighbor_report, blobmsg_get_string(tb[CLIENT_TABLE_NEIGHBOR]), NEIGHBOR_REPORT_LEN);
|
|
}
|
|
else {
|
|
ap_entry->neighbor_report[0] = '\0';
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_IFACE]) {
|
|
strncpy(ap_entry->iface, blobmsg_get_string(tb[CLIENT_TABLE_IFACE]), MAX_INTERFACE_NAME);
|
|
}
|
|
else {
|
|
ap_entry->iface[0] = '\0';
|
|
}
|
|
|
|
if (tb[CLIENT_TABLE_HOSTNAME]) {
|
|
strncpy(ap_entry->hostname, blobmsg_get_string(tb[CLIENT_TABLE_HOSTNAME]), HOST_NAME_MAX);
|
|
}
|
|
else {
|
|
ap_entry->hostname[0] = '\0';
|
|
}
|
|
|
|
ap_entry->time = time(0);
|
|
insert_to_ap_array(ap_entry);
|
|
|
|
if (do_kick && dawn_metric.kicking) {
|
|
update_iw_info(ap_entry->bssid_addr);
|
|
kick_clients(ap_entry, id);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
enum {
|
|
UCI_TABLE_METRIC,
|
|
UCI_TABLE_TIMES,
|
|
__UCI_TABLE_MAX
|
|
};
|
|
|
|
enum {
|
|
UCI_HT_SUPPORT,
|
|
UCI_VHT_SUPPORT,
|
|
UCI_NO_HT_SUPPORT,
|
|
UCI_NO_VHT_SUPPORT,
|
|
UCI_RSSI,
|
|
UCI_LOW_RSSI,
|
|
UCI_FREQ,
|
|
UCI_CHAN_UTIL,
|
|
UCI_MAX_CHAN_UTIL,
|
|
UCI_RSSI_VAL,
|
|
UCI_LOW_RSSI_VAL,
|
|
UCI_CHAN_UTIL_VAL,
|
|
UCI_MAX_CHAN_UTIL_VAL,
|
|
UCI_MIN_PROBE_COUNT,
|
|
UCI_BANDWIDTH_THRESHOLD,
|
|
UCI_USE_STATION_COUNT,
|
|
UCI_MAX_STATION_DIFF,
|
|
UCI_EVAL_PROBE_REQ,
|
|
UCI_EVAL_AUTH_REQ,
|
|
UCI_EVAL_ASSOC_REQ,
|
|
UCI_KICKING,
|
|
UCI_DENY_AUTH_REASON,
|
|
UCI_DENY_ASSOC_REASON,
|
|
UCI_USE_DRIVER_RECOG,
|
|
UCI_MIN_NUMBER_TO_KICK,
|
|
UCI_CHAN_UTIL_AVG_PERIOD,
|
|
UCI_SET_HOSTAPD_NR,
|
|
UCI_OP_CLASS,
|
|
UCI_DURATION,
|
|
UCI_MODE,
|
|
UCI_SCAN_CHANNEL,
|
|
__UCI_METIC_MAX
|
|
};
|
|
|
|
enum {
|
|
UCI_UPDATE_CLIENT,
|
|
UCI_DENIED_REQ_THRESHOLD,
|
|
UCI_REMOVE_CLIENT,
|
|
UCI_REMOVE_PROBE,
|
|
UCI_REMOVE_AP,
|
|
UCI_UPDATE_HOSTAPD,
|
|
UCI_UPDATE_TCP_CON,
|
|
UCI_UPDATE_CHAN_UTIL,
|
|
UCI_UPDATE_BEACON_REPORTS,
|
|
__UCI_TIMES_MAX,
|
|
};
|
|
|
|
static const struct blobmsg_policy uci_table_policy[__UCI_TABLE_MAX] = {
|
|
[UCI_TABLE_METRIC] = {.name = "metric", .type = BLOBMSG_TYPE_TABLE},
|
|
[UCI_TABLE_TIMES] = {.name = "times", .type = BLOBMSG_TYPE_TABLE}
|
|
};
|
|
|
|
static const struct blobmsg_policy uci_metric_policy[__UCI_METIC_MAX] = {
|
|
[UCI_HT_SUPPORT] = {.name = "ht_support", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_VHT_SUPPORT] = {.name = "vht_support", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_NO_HT_SUPPORT] = {.name = "no_ht_support", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_NO_VHT_SUPPORT] = {.name = "no_vht_support", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_RSSI] = {.name = "rssi", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_LOW_RSSI] = {.name = "low_rssi", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_FREQ] = {.name = "freq", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_CHAN_UTIL] = {.name = "chan_util", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MAX_CHAN_UTIL] = {.name = "max_chan_util", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_RSSI_VAL] = {.name = "rssi_val", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_LOW_RSSI_VAL] = {.name = "low_rssi_val", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_CHAN_UTIL_VAL] = {.name = "chan_util_val", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MAX_CHAN_UTIL_VAL] = {.name = "max_chan_util_val", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MIN_PROBE_COUNT] = {.name = "min_probe_count", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_BANDWIDTH_THRESHOLD] = {.name = "bandwidth_threshold", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_USE_STATION_COUNT] = {.name = "use_station_count", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MAX_STATION_DIFF] = {.name = "max_station_diff", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_EVAL_PROBE_REQ] = {.name = "eval_probe_req", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_EVAL_AUTH_REQ] = {.name = "eval_auth_req", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_EVAL_ASSOC_REQ] = {.name = "eval_assoc_req", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_KICKING] = {.name = "kicking", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_DENY_AUTH_REASON] = {.name = "deny_auth_reason", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_DENY_ASSOC_REASON] = {.name = "deny_assoc_reason", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_USE_DRIVER_RECOG] = {.name = "use_driver_recog", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MIN_NUMBER_TO_KICK] = {.name = "min_number_to_kick", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_CHAN_UTIL_AVG_PERIOD] = {.name = "chan_util_avg_period", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_SET_HOSTAPD_NR] = {.name = "set_hostapd_nr", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_OP_CLASS] = {.name = "op_class", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_DURATION] = {.name = "duration", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_MODE] = {.name = "mode", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_SCAN_CHANNEL] = {.name = "mode", .type = BLOBMSG_TYPE_INT32},
|
|
};
|
|
|
|
static const struct blobmsg_policy uci_times_policy[__UCI_TIMES_MAX] = {
|
|
[UCI_UPDATE_CLIENT] = {.name = "update_client", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_DENIED_REQ_THRESHOLD] = {.name = "denied_req_threshold", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_REMOVE_CLIENT] = {.name = "remove_client", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_REMOVE_PROBE] = {.name = "remove_probe", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_REMOVE_AP] = {.name = "remove_ap", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_UPDATE_HOSTAPD] = {.name = "update_hostapd", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_UPDATE_TCP_CON] = {.name = "update_tcp_con", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_UPDATE_CHAN_UTIL] = {.name = "update_chan_util", .type = BLOBMSG_TYPE_INT32},
|
|
[UCI_UPDATE_BEACON_REPORTS] = {.name = "update_beacon_reports", .type = BLOBMSG_TYPE_INT32},
|
|
};
|
|
|
|
static int handle_uci_config(struct blob_attr* msg) {
|
|
|
|
struct blob_attr* tb[__UCI_TABLE_MAX];
|
|
blobmsg_parse(uci_table_policy, __UCI_TABLE_MAX, tb, blob_data(msg), blob_len(msg));
|
|
|
|
struct blob_attr* tb_metric[__UCI_METIC_MAX];
|
|
blobmsg_parse(uci_metric_policy, __UCI_METIC_MAX, tb_metric, blobmsg_data(tb[UCI_TABLE_METRIC]), blobmsg_len(tb[UCI_TABLE_METRIC]));
|
|
|
|
// TODO: Magic number?
|
|
char cmd_buffer[1024];
|
|
sprintf(cmd_buffer, "dawn.@metric[0].ht_support=%d", blobmsg_get_u32(tb_metric[UCI_HT_SUPPORT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].vht_support=%d", blobmsg_get_u32(tb_metric[UCI_VHT_SUPPORT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].no_ht_support=%d", blobmsg_get_u32(tb_metric[UCI_NO_HT_SUPPORT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].no_vht_support=%d", blobmsg_get_u32(tb_metric[UCI_NO_VHT_SUPPORT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].rssi=%d", blobmsg_get_u32(tb_metric[UCI_RSSI]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].low_rssi=%d", blobmsg_get_u32(tb_metric[UCI_LOW_RSSI]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].freq=%d", blobmsg_get_u32(tb_metric[UCI_FREQ]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].chan_util=%d", blobmsg_get_u32(tb_metric[UCI_CHAN_UTIL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].rssi_val=%d", blobmsg_get_u32(tb_metric[UCI_RSSI_VAL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].low_rssi_val=%d", blobmsg_get_u32(tb_metric[UCI_LOW_RSSI_VAL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].chan_util_val=%d", blobmsg_get_u32(tb_metric[UCI_CHAN_UTIL_VAL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].max_chan_util=%d", blobmsg_get_u32(tb_metric[UCI_MAX_CHAN_UTIL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].max_chan_util_val=%d", blobmsg_get_u32(tb_metric[UCI_MAX_CHAN_UTIL_VAL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].min_probe_count=%d", blobmsg_get_u32(tb_metric[UCI_MIN_PROBE_COUNT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].bandwidth_threshold=%d", blobmsg_get_u32(tb_metric[UCI_BANDWIDTH_THRESHOLD]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].use_station_count=%d", blobmsg_get_u32(tb_metric[UCI_USE_STATION_COUNT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].max_station_diff=%d", blobmsg_get_u32(tb_metric[UCI_MAX_STATION_DIFF]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].eval_probe_req=%d", blobmsg_get_u32(tb_metric[UCI_EVAL_PROBE_REQ]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].eval_auth_req=%d", blobmsg_get_u32(tb_metric[UCI_EVAL_AUTH_REQ]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].evalcd_assoc_req=%d", blobmsg_get_u32(tb_metric[UCI_EVAL_ASSOC_REQ]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].kicking=%d", blobmsg_get_u32(tb_metric[UCI_KICKING]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].deny_auth_reason=%d", blobmsg_get_u32(tb_metric[UCI_DENY_AUTH_REASON]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].deny_assoc_reason=%d", blobmsg_get_u32(tb_metric[UCI_DENY_ASSOC_REASON]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].use_driver_recog=%d", blobmsg_get_u32(tb_metric[UCI_USE_DRIVER_RECOG]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].min_number_to_kick=%d", blobmsg_get_u32(tb_metric[UCI_MIN_NUMBER_TO_KICK]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].chan_util_avg_period=%d", blobmsg_get_u32(tb_metric[UCI_CHAN_UTIL_AVG_PERIOD]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].set_hostapd_nr=%d", blobmsg_get_u32(tb_metric[UCI_SET_HOSTAPD_NR]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].op_class=%d", blobmsg_get_u32(tb_metric[UCI_OP_CLASS]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].duration=%d", blobmsg_get_u32(tb_metric[UCI_DURATION]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].mode=%d", blobmsg_get_u32(tb_metric[UCI_MODE]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@metric[0].scan_channel=%d", blobmsg_get_u32(tb_metric[UCI_SCAN_CHANNEL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
struct blob_attr* tb_times[__UCI_TIMES_MAX];
|
|
blobmsg_parse(uci_times_policy, __UCI_TIMES_MAX, tb_times, blobmsg_data(tb[UCI_TABLE_TIMES]), blobmsg_len(tb[UCI_TABLE_TIMES]));
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].update_client=%d", blobmsg_get_u32(tb_times[UCI_UPDATE_CLIENT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].denied_req_threshold=%d", blobmsg_get_u32(tb_times[UCI_DENIED_REQ_THRESHOLD]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].remove_client=%d", blobmsg_get_u32(tb_times[UCI_REMOVE_CLIENT]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].remove_probe=%d", blobmsg_get_u32(tb_times[UCI_REMOVE_PROBE]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].remove_ap=%d", blobmsg_get_u32(tb_times[UCI_REMOVE_AP]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].update_hostapd=%d", blobmsg_get_u32(tb_times[UCI_UPDATE_HOSTAPD]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].update_tcp_con=%d", blobmsg_get_u32(tb_times[UCI_UPDATE_TCP_CON]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].update_chan_util=%d", blobmsg_get_u32(tb_times[UCI_UPDATE_CHAN_UTIL]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
sprintf(cmd_buffer, "dawn.@times[0].update_beacon_reports=%d", blobmsg_get_u32(tb_times[UCI_UPDATE_BEACON_REPORTS]));
|
|
uci_set_network(cmd_buffer);
|
|
|
|
uci_reset();
|
|
dawn_metric = uci_get_dawn_metric();
|
|
timeout_config = uci_get_time_config();
|
|
|
|
return 0;
|
|
}
|