Byte/packet counters for in-tunnel and transport streams
are now initialized and updated as needed.

To be exported via netlink.

Signed-off-by: Antonio Quartulli <anto...@openvpn.net>
---
 drivers/net/ovpn/Makefile |  1 +
 drivers/net/ovpn/io.c     | 12 +++++++++++-
 drivers/net/ovpn/peer.c   |  2 ++
 drivers/net/ovpn/peer.h   |  5 +++++
 drivers/net/ovpn/stats.c  | 21 +++++++++++++++++++++
 drivers/net/ovpn/stats.h  | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 6 files changed, 87 insertions(+), 1 deletion(-)

diff --git a/drivers/net/ovpn/Makefile b/drivers/net/ovpn/Makefile
index 
38c9fdca0e2e8e4af3c369ceb3971b58ab52d77b..04c3345807c5d759daf65cc80a290f784dbf5588
 100644
--- a/drivers/net/ovpn/Makefile
+++ b/drivers/net/ovpn/Makefile
@@ -17,4 +17,5 @@ ovpn-y += netlink-gen.o
 ovpn-y += peer.o
 ovpn-y += pktid.o
 ovpn-y += socket.o
+ovpn-y += stats.o
 ovpn-y += udp.o
diff --git a/drivers/net/ovpn/io.c b/drivers/net/ovpn/io.c
index 
d0b410535ac340a53f010d0b2f20430b26bb012d..50dc2e4c03f01b02bdf616473b755b6e1e6b57f7
 100644
--- a/drivers/net/ovpn/io.c
+++ b/drivers/net/ovpn/io.c
@@ -12,6 +12,7 @@
 #include <linux/skbuff.h>
 #include <net/gro_cells.h>
 #include <net/gso.h>
+#include <net/ip.h>
 
 #include "ovpnpriv.h"
 #include "peer.h"
@@ -55,9 +56,11 @@ static void ovpn_netdev_write(struct ovpn_peer *peer, struct 
sk_buff *skb)
        /* cause packet to be "received" by the interface */
        pkt_len = skb->len;
        ret = gro_cells_receive(&peer->ovpn->gro_cells, skb);
-       if (likely(ret == NET_RX_SUCCESS))
+       if (likely(ret == NET_RX_SUCCESS)) {
                /* update RX stats with the size of decrypted packet */
+               ovpn_peer_stats_increment_rx(&peer->vpn_stats, pkt_len);
                dev_sw_netstats_rx_add(peer->ovpn->dev, pkt_len);
+       }
 }
 
 void ovpn_decrypt_post(void *data, int ret)
@@ -158,6 +161,8 @@ void ovpn_recv(struct ovpn_peer *peer, struct sk_buff *skb)
        struct ovpn_crypto_key_slot *ks;
        u8 key_id;
 
+       ovpn_peer_stats_increment_rx(&peer->link_stats, skb->len);
+
        /* get the key slot matching the key ID in the received packet */
        key_id = ovpn_key_id_from_skb(skb);
        ks = ovpn_crypto_key_id_to_slot(&peer->crypto, key_id);
@@ -181,6 +186,7 @@ void ovpn_encrypt_post(void *data, int ret)
        struct sk_buff *skb = data;
        struct ovpn_socket *sock;
        struct ovpn_peer *peer;
+       unsigned int orig_len;
 
        /* encryption is happening asynchronously. This function will be
         * called later by the crypto callback with a proper return value
@@ -206,6 +212,7 @@ void ovpn_encrypt_post(void *data, int ret)
                goto err;
 
        skb_mark_not_on_list(skb);
+       orig_len = skb->len;
 
        rcu_read_lock();
        sock = rcu_dereference(peer->sock);
@@ -220,6 +227,8 @@ void ovpn_encrypt_post(void *data, int ret)
                /* no transport configured yet */
                goto err_unlock;
        }
+
+       ovpn_peer_stats_increment_tx(&peer->link_stats, orig_len);
        /* skb passed down the stack - don't free it */
        skb = NULL;
 err_unlock:
@@ -341,6 +350,7 @@ netdev_tx_t ovpn_net_xmit(struct sk_buff *skb, struct 
net_device *dev)
                goto drop;
        }
 
+       ovpn_peer_stats_increment_tx(&peer->vpn_stats, skb->len);
        ovpn_send(ovpn, skb_list.next, peer);
 
        return NETDEV_TX_OK;
diff --git a/drivers/net/ovpn/peer.c b/drivers/net/ovpn/peer.c
index 
69522f063e1882412f662d98af94ea9a058d2e74..1f4148475c8b93aabc72694da5af97ae00a3a59d
 100644
--- a/drivers/net/ovpn/peer.c
+++ b/drivers/net/ovpn/peer.c
@@ -61,6 +61,8 @@ struct ovpn_peer *ovpn_peer_new(struct ovpn_priv *ovpn, u32 
id)
        ovpn_crypto_state_init(&peer->crypto);
        spin_lock_init(&peer->lock);
        kref_init(&peer->refcount);
+       ovpn_peer_stats_init(&peer->vpn_stats);
+       ovpn_peer_stats_init(&peer->link_stats);
 
        ret = dst_cache_init(&peer->dst_cache, GFP_KERNEL);
        if (ret < 0) {
diff --git a/drivers/net/ovpn/peer.h b/drivers/net/ovpn/peer.h
index 
a9113a969f94d66fa17208d563d0bbd255c23fa9..2453d39ce327c6d174cfb35fe5430865b32c2efe
 100644
--- a/drivers/net/ovpn/peer.h
+++ b/drivers/net/ovpn/peer.h
@@ -14,6 +14,7 @@
 
 #include "crypto.h"
 #include "socket.h"
+#include "stats.h"
 
 /**
  * struct ovpn_peer - the main remote peer object
@@ -27,6 +28,8 @@
  * @crypto: the crypto configuration (ciphers, keys, etc..)
  * @dst_cache: cache for dst_entry used to send to peer
  * @bind: remote peer binding
+ * @vpn_stats: per-peer in-VPN TX/RX stats
+ * @link_stats: per-peer link/transport TX/RX stats
  * @delete_reason: why peer was deleted (i.e. timeout, transport error, ..)
  * @lock: protects binding to peer (bind)
  * @refcount: reference counter
@@ -45,6 +48,8 @@ struct ovpn_peer {
        struct ovpn_crypto_state crypto;
        struct dst_cache dst_cache;
        struct ovpn_bind __rcu *bind;
+       struct ovpn_peer_stats vpn_stats;
+       struct ovpn_peer_stats link_stats;
        enum ovpn_del_peer_reason delete_reason;
        spinlock_t lock; /* protects bind */
        struct kref refcount;
diff --git a/drivers/net/ovpn/stats.c b/drivers/net/ovpn/stats.c
new file mode 100644
index 
0000000000000000000000000000000000000000..d637143473bb913647c79832fd9eb3ebfd9efb59
--- /dev/null
+++ b/drivers/net/ovpn/stats.c
@@ -0,0 +1,21 @@
+// SPDX-License-Identifier: GPL-2.0
+/*  OpenVPN data channel offload
+ *
+ *  Copyright (C) 2020-2025 OpenVPN, Inc.
+ *
+ *  Author:    James Yonan <ja...@openvpn.net>
+ *             Antonio Quartulli <anto...@openvpn.net>
+ */
+
+#include <linux/atomic.h>
+
+#include "stats.h"
+
+void ovpn_peer_stats_init(struct ovpn_peer_stats *ps)
+{
+       atomic64_set(&ps->rx.bytes, 0);
+       atomic64_set(&ps->rx.packets, 0);
+
+       atomic64_set(&ps->tx.bytes, 0);
+       atomic64_set(&ps->tx.packets, 0);
+}
diff --git a/drivers/net/ovpn/stats.h b/drivers/net/ovpn/stats.h
new file mode 100644
index 
0000000000000000000000000000000000000000..53433d8b6c33160845de2ae1ca38e85cf31950b7
--- /dev/null
+++ b/drivers/net/ovpn/stats.h
@@ -0,0 +1,47 @@
+/* SPDX-License-Identifier: GPL-2.0-only */
+/*  OpenVPN data channel offload
+ *
+ *  Copyright (C) 2020-2025 OpenVPN, Inc.
+ *
+ *  Author:    James Yonan <ja...@openvpn.net>
+ *             Antonio Quartulli <anto...@openvpn.net>
+ *             Lev Stipakov <l...@openvpn.net>
+ */
+
+#ifndef _NET_OVPN_OVPNSTATS_H_
+#define _NET_OVPN_OVPNSTATS_H_
+
+/* one stat */
+struct ovpn_peer_stat {
+       atomic64_t bytes;
+       atomic64_t packets;
+};
+
+/* rx and tx stats combined */
+struct ovpn_peer_stats {
+       struct ovpn_peer_stat rx;
+       struct ovpn_peer_stat tx;
+};
+
+void ovpn_peer_stats_init(struct ovpn_peer_stats *ps);
+
+static inline void ovpn_peer_stats_increment(struct ovpn_peer_stat *stat,
+                                            const unsigned int n)
+{
+       atomic64_add(n, &stat->bytes);
+       atomic64_inc(&stat->packets);
+}
+
+static inline void ovpn_peer_stats_increment_rx(struct ovpn_peer_stats *stats,
+                                               const unsigned int n)
+{
+       ovpn_peer_stats_increment(&stats->rx, n);
+}
+
+static inline void ovpn_peer_stats_increment_tx(struct ovpn_peer_stats *stats,
+                                               const unsigned int n)
+{
+       ovpn_peer_stats_increment(&stats->tx, n);
+}
+
+#endif /* _NET_OVPN_OVPNSTATS_H_ */

-- 
2.45.3


Reply via email to