usteer: drop usteer_ubus_notify_client_disassoc
[project/usteer.git] / local_node.c
index e69ec2157585d471c09bbbbe3822bcced15d1533..c79fd95b68281d7255e29a38d619052324a81b40 100644 (file)
@@ -71,6 +71,7 @@ usteer_free_node(struct ubus_context *ctx, struct usteer_local_node *ln)
        usteer_local_node_pending_bss_tm_free(ln);
        usteer_local_node_state_reset(ln);
        usteer_sta_node_cleanup(&ln->node);
+       usteer_measurement_report_node_cleanup(&ln->node);
        uloop_timeout_cancel(&ln->update);
        uloop_timeout_cancel(&ln->bss_tm_queries_timeout);
        avl_delete(&local_nodes, &ln->node.avl);
@@ -142,6 +143,136 @@ usteer_handle_bss_tm_query(struct usteer_local_node *ln, struct blob_attr *msg)
        return 1;
 }
 
+static int
+usteer_handle_bss_tm_response(struct usteer_local_node *ln, struct blob_attr *msg)
+{
+       enum {
+               BSS_TM_RESPONSE_ADDRESS,
+               BSS_TM_RESPONSE_STATUS_CODE,
+               __BSS_TM_RESPONSE_MAX
+       };
+       struct blobmsg_policy policy[__BSS_TM_RESPONSE_MAX] = {
+               [BSS_TM_RESPONSE_ADDRESS] = { .name = "address", .type = BLOBMSG_TYPE_STRING },
+               [BSS_TM_RESPONSE_STATUS_CODE] = { .name = "status-code", .type = BLOBMSG_TYPE_INT8 },
+       };
+       struct blob_attr *tb[__BSS_TM_RESPONSE_MAX];
+       struct sta_info *si;
+       struct sta *sta;
+       uint8_t *sta_addr;
+
+       blobmsg_parse(policy, __BSS_TM_RESPONSE_MAX, tb, blob_data(msg), blob_len(msg));
+
+       if (!tb[BSS_TM_RESPONSE_ADDRESS] || !tb[BSS_TM_RESPONSE_STATUS_CODE])
+               return 0;
+
+       sta_addr = (uint8_t *) ether_aton(blobmsg_get_string(tb[BSS_TM_RESPONSE_ADDRESS]));
+       if (!sta_addr)
+               return 0;
+
+       sta = usteer_sta_get(sta_addr, false);
+       if (!sta)
+               return 0;
+
+       si = usteer_sta_info_get(sta, &ln->node, false);
+       if (!si)
+               return 0;
+
+       si->bss_transition_response.status_code = blobmsg_get_u8(tb[BSS_TM_RESPONSE_STATUS_CODE]);
+       si->bss_transition_response.timestamp = current_time;
+
+       if (si->bss_transition_response.status_code) {
+               /* Cancel imminent kick in case BSS transition was rejected */
+               si->kick_time = 0;
+       }
+
+       return 0;
+}
+
+static int
+usteer_local_node_handle_beacon_report(struct usteer_local_node *ln, struct blob_attr *msg)
+{
+       enum {
+               BR_ADDRESS,
+               BR_BSSID,
+               BR_RCPI,
+               BR_RSNI,
+               __BR_MAX
+       };
+       struct blobmsg_policy policy[__BR_MAX] = {
+               [BR_ADDRESS] = { .name = "address", .type = BLOBMSG_TYPE_STRING },
+               [BR_BSSID] = { .name = "bssid", .type = BLOBMSG_TYPE_STRING },
+               [BR_RCPI] = { .name = "rcpi", .type = BLOBMSG_TYPE_INT16 },
+               [BR_RSNI] = { .name = "rsni", .type = BLOBMSG_TYPE_INT16 },
+       };
+       struct blob_attr *tb[__BR_MAX];
+       struct usteer_node *node;
+       uint8_t *addr;
+       struct sta *sta;
+
+       blobmsg_parse(policy, __BR_MAX, tb, blob_data(msg), blob_len(msg));
+       if (!tb[BR_ADDRESS] || !tb[BR_BSSID] || !tb[BR_RCPI] || !tb[BR_RSNI])
+               return 0;
+
+       addr = (uint8_t *) ether_aton(blobmsg_get_string(tb[BR_ADDRESS]));
+       if (!addr)
+               return 0;
+
+       sta = usteer_sta_get(addr, false);
+       if (!sta)
+               return 0;
+
+       addr = (uint8_t *) ether_aton(blobmsg_get_string(tb[BR_BSSID]));
+       if (!addr)
+               return 0;
+
+       node = usteer_node_by_bssid(addr);
+       if (!node)
+               return 0;
+
+       usteer_measurement_report_add(sta, node,
+                                     (uint8_t)blobmsg_get_u16(tb[BR_RCPI]),
+                                     (uint8_t)blobmsg_get_u16(tb[BR_RSNI]),
+                                     current_time);
+       return 0;
+}
+
+static int
+usteer_local_node_handle_link_measurement_report(struct usteer_local_node *ln, struct blob_attr *msg)
+{
+       enum {
+               LMR_ADDRESS,
+               LMR_RCPI,
+               LMR_RSNI,
+               __LMR_MAX
+       };
+       struct blobmsg_policy policy[__LMR_MAX] = {
+               [LMR_ADDRESS] = { .name = "address", .type = BLOBMSG_TYPE_STRING },
+               [LMR_RCPI] = { .name = "rcpi", .type = BLOBMSG_TYPE_INT16 },
+               [LMR_RSNI] = { .name = "rsni", .type = BLOBMSG_TYPE_INT16 },
+       };
+       struct blob_attr *tb[__LMR_MAX];
+       uint8_t *addr;
+       struct sta *sta;
+
+       blobmsg_parse(policy, __LMR_MAX, tb, blob_data(msg), blob_len(msg));
+       if (!tb[LMR_ADDRESS] || !tb[LMR_RCPI] || !tb[LMR_RSNI])
+               return 0;
+
+       addr = (uint8_t *) ether_aton(blobmsg_get_string(tb[LMR_ADDRESS]));
+       if (!addr)
+               return 0;
+
+       sta = usteer_sta_get(addr, false);
+       if (!sta)
+               return 0;
+
+       usteer_measurement_report_add(sta, &ln->node,
+                                     (uint8_t)blobmsg_get_u16(tb[LMR_RCPI]),
+                                     (uint8_t)blobmsg_get_u16(tb[LMR_RSNI]),
+                                     current_time);
+       return 0;
+}
+
 static int
 usteer_handle_event(struct ubus_context *ctx, struct ubus_object *obj,
                   struct ubus_request_data *req, const char *method,
@@ -177,6 +308,12 @@ usteer_handle_event(struct ubus_context *ctx, struct ubus_object *obj,
 
        if(!strcmp(method, "bss-transition-query")) {
                return usteer_handle_bss_tm_query(ln, msg);
+       } else if(!strcmp(method, "bss-transition-response")) {
+               return usteer_handle_bss_tm_response(ln, msg);
+       } else if(!strcmp(method, "beacon-report")) {
+               return usteer_local_node_handle_beacon_report(ln, msg);
+       } else if(!strcmp(method, "link-measurement-report")) {
+               return usteer_local_node_handle_link_measurement_report(ln, msg);
        }
 
        for (i = 0; i < ARRAY_SIZE(event_types); i++) {
@@ -249,28 +386,43 @@ usteer_local_node_assoc_update(struct sta_info *si, struct blob_attr *data)
 }
 
 static void
-usteer_local_node_update_sta_rrm(const uint8_t *addr, struct blob_attr *client_attr)
+usteer_local_node_update_sta_rrm_wnm(struct sta_info *si, struct blob_attr *client_attr)
 {
        static const struct blobmsg_policy rrm_policy = {
                .name = "rrm",
                .type = BLOBMSG_TYPE_ARRAY,
        };
-       struct blob_attr *sta_blob = NULL;
-       struct sta *sta;
+       static const struct blobmsg_policy ext_capa_policy = {
+               .name = "extended_capabilities",
+               .type = BLOBMSG_TYPE_ARRAY,
+       };
+       struct blob_attr *rrm_blob = NULL, *wnm_blob = NULL, *cur;
+       int rem;
+       int i = 0;
 
-       if (!addr)
+       /* RRM */
+       blobmsg_parse(&rrm_policy, 1, &rrm_blob, blobmsg_data(client_attr), blobmsg_data_len(client_attr));
+       if (!rrm_blob)
                return;
 
-       /* Don't create the STA */
-       sta = usteer_sta_get(addr, false);
-       if (!sta)
-               return;
+       si->rrm = blobmsg_get_u32(blobmsg_data(rrm_blob));
 
-       blobmsg_parse(&rrm_policy, 1, &sta_blob, blobmsg_data(client_attr), blobmsg_data_len(client_attr));
-       if (!sta_blob)
+       /* Extended Capabilities / WNM */
+       blobmsg_parse(&ext_capa_policy, 1, &wnm_blob, blobmsg_data(client_attr), blobmsg_data_len(client_attr));
+       if (!wnm_blob)
                return;
 
-       sta->rrm = blobmsg_get_u32(blobmsg_data(sta_blob));
+       blobmsg_for_each_attr(cur, wnm_blob, rem) {
+               if (blobmsg_type(cur) != BLOBMSG_TYPE_INT32)
+                       return;
+               
+               if (i == 2) {
+                       if (blobmsg_get_u32(cur) & (1 << 3))
+                               si->bss_transition = true;
+               }
+
+               i++;
+       }
 }
 
 static void
@@ -313,7 +465,7 @@ usteer_local_node_set_assoc(struct usteer_local_node *ln, struct blob_attr *cl)
                }
 
                /* Read RRM information */
-               usteer_local_node_update_sta_rrm(addr, cur);
+               usteer_local_node_update_sta_rrm_wnm(si, cur);
        }
 
        node->n_assoc = n_assoc;
@@ -362,12 +514,14 @@ usteer_local_node_status_cb(struct ubus_request *req, int type, struct blob_attr
                MSG_FREQ,
                MSG_CHANNEL,
                MSG_OP_CLASS,
+               MSG_BEACON_INTERVAL,
                __MSG_MAX,
        };
        static struct blobmsg_policy policy[__MSG_MAX] = {
                [MSG_FREQ] = { "freq", BLOBMSG_TYPE_INT32 },
                [MSG_CHANNEL] = { "channel", BLOBMSG_TYPE_INT32 },
                [MSG_OP_CLASS] = { "op_class", BLOBMSG_TYPE_INT32 },
+               [MSG_BEACON_INTERVAL] = { "beacon_interval", BLOBMSG_TYPE_INT32 },
        };
        struct blob_attr *tb[__MSG_MAX];
        struct usteer_local_node *ln;
@@ -381,8 +535,12 @@ usteer_local_node_status_cb(struct ubus_request *req, int type, struct blob_attr
                node->freq = blobmsg_get_u32(tb[MSG_FREQ]);
        if (tb[MSG_CHANNEL])
                node->channel = blobmsg_get_u32(tb[MSG_CHANNEL]);
-       if (tb[MSG_FREQ])
+       if (tb[MSG_OP_CLASS])
                node->op_class = blobmsg_get_u32(tb[MSG_OP_CLASS]);     
+
+       /* Local-Node */
+       if (tb[MSG_BEACON_INTERVAL])
+               ln->beacon_interval = blobmsg_get_u32(tb[MSG_BEACON_INTERVAL]);
 }
 
 static void
@@ -501,6 +659,33 @@ usteer_local_node_state_next(struct uloop_timeout *timeout)
        ubus_complete_request_async(ubus_ctx, &ln->req);
 }
 
+static void
+usteer_local_node_request_link_measurement(struct usteer_local_node *ln)
+{
+       unsigned int min_count = DIV_ROUND_UP(config.link_measurement_interval, config.local_sta_update);
+       struct usteer_node *node;
+       struct sta_info *si;
+
+       node = &ln->node;
+
+       if (ln->link_measurement_tries < min_count) {
+               ln->link_measurement_tries++;
+               return;
+       }
+       
+       ln->link_measurement_tries = 0;
+
+       if (!config.link_measurement_interval)
+               return;
+
+       list_for_each_entry(si, &node->sta_info, node_list) {
+               if (si->connected != STA_CONNECTED)
+                       continue;
+
+               usteer_ubus_trigger_link_measurement(si);
+       }
+}
+
 static void
 usteer_local_node_update(struct uloop_timeout *timeout)
 {
@@ -521,6 +706,9 @@ usteer_local_node_update(struct uloop_timeout *timeout)
        usteer_local_node_state_reset(ln);
        uloop_timeout_set(&ln->req_timer, 1);
        usteer_local_node_kick(ln);
+       usteer_band_steering_perform_steer(ln);
+       usteer_local_node_request_link_measurement(ln);
+
        uloop_timeout_set(timeout, config.local_sta_update);
 }
 
@@ -532,12 +720,13 @@ usteer_local_node_process_bss_tm_queries(struct uloop_timeout *timeout)
        struct usteer_node *node;
        struct sta_info *si;
        struct sta *sta;
-
-       uint8_t validity_period = 100; /* ~ 10 seconds */
+       uint8_t validity_period;
 
        ln = container_of(timeout, struct usteer_local_node, bss_tm_queries_timeout);
        node = &ln->node;
 
+       validity_period = 10000 / usteer_local_node_get_beacon_interval(ln); /* ~ 10 seconds */
+
        list_for_each_entry_safe(query, tmp, &ln->bss_tm_queries, list) {
                sta = usteer_sta_get(query->sta_addr, false);
                if (!sta)
@@ -547,7 +736,7 @@ usteer_local_node_process_bss_tm_queries(struct uloop_timeout *timeout)
                if (!si)
                        continue;
 
-               usteer_ubus_bss_transition_request(si, query->dialog_token, false, false, validity_period);
+               usteer_ubus_bss_transition_request(si, query->dialog_token, false, false, validity_period, NULL);
        }
 
        /* Free pending queries we can not handle */
@@ -578,6 +767,7 @@ usteer_get_node(struct ubus_context *ctx, const char *name)
        avl_insert(&local_nodes, &node->avl);
        kvlist_init(&ln->node_info, kvlist_blob_len);
        INIT_LIST_HEAD(&node->sta_info);
+       INIT_LIST_HEAD(&node->measurements);
 
        ln->bss_tm_queries_timeout.cb = usteer_local_node_process_bss_tm_queries;
        INIT_LIST_HEAD(&ln->bss_tm_queries);
@@ -623,6 +813,7 @@ usteer_check_node_enabled(struct usteer_local_node *ln)
                MSG(INFO, "Disconnecting from local node %s\n", usteer_node_name(&ln->node));
                usteer_local_node_state_reset(ln);
                usteer_sta_node_cleanup(&ln->node);
+               usteer_measurement_report_node_cleanup(&ln->node);
                uloop_timeout_cancel(&ln->update);
                ubus_unsubscribe(ubus_ctx, &ln->ev, ln->obj_id);
                return;
@@ -658,6 +849,7 @@ usteer_register_node(struct ubus_context *ctx, const char *name, uint32_t id)
 
        blob_buf_init(&b, 0);
        blobmsg_add_u8(&b, "neighbor_report", 1);
+       blobmsg_add_u8(&b, "link_measurement", 1);
        blobmsg_add_u8(&b, "beacon_report", 1);
        blobmsg_add_u8(&b, "bss_transition", 1);
        ubus_invoke(ctx, id, "bss_mgmt_enable", b.head, NULL, NULL, 1000);
@@ -709,6 +901,16 @@ node_list_cb(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv)
        usteer_register_node(ctx, obj->path, obj->id);
 }
 
+int
+usteer_local_node_get_beacon_interval(struct usteer_local_node *ln)
+{
+       /* Check if beacon-interval is not available (pre-21.02+) */
+       if (ln->beacon_interval < 1)
+               return 100;
+
+       return ln->beacon_interval;
+}
+
 void config_set_node_up_script(struct blob_attr *data)
 {
        const char *val;