Improved node traffic counters.

[meshlink] / src / protocol_key.c

diff --git a/src/protocol_key.c b/src/protocol_key.c
index 25dbc55e5c04b7533e9eb0f6c5eeb62214cc648e..768b6371e4a03338cc98f93b06595d4fe3726da9 100644 (file)
--- a/src/protocol_key.c
+++ b/src/protocol_key.c
@@ -345,6 +345,10 @@ bool req_key_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
                        return true;
                }
 
+               size_t len = strlen(request);
+               from->in_forward += len;
+               to->out_forward += len;
+
                send_request(mesh, to->nexthop->connection, NULL, "%s", request);
        }
 
@@ -411,6 +415,11 @@ bool ans_key_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
                        return false;
                }
 
+               /* TODO: find a good way to avoid the use of strlen() */
+               size_t len = strlen(request);
+               from->in_forward += len;
+               to->out_forward += len;
+
                /* Append the known UDP address of the from node, if we have a confirmed one */
                if(!*address && from->status.udp_confirmed && from->address.sa.sa_family != AF_UNSPEC) {
                        char *reflexive_address, *reflexive_port;