[Devel] [PATCH hci-8.1 v10 11/13] ve: add bpf_prog failcounter

[Konstantin Khorenko]

From: Vladimir Riabchun <vladimir.riabchun at virtuozzo.com>

Count hits of the per-VE BPF program limit in ve.failcount.

https://virtuozzo.atlassian.net/browse/VSTOR-135520

Feature: per-ve failcounters
Signed-off-by: Vladimir Riabchun <vladimir.riabchun at virtuozzo.com>
Signed-off-by: Konstantin Khorenko <khorenko at virtuozzo.com>
---
 include/linux/ve.h   | 1 +
 kernel/bpf/syscall.c | 1 +
 kernel/ve/ve.c       | 3 +++
 3 files changed, 5 insertions(+)

diff --git a/include/linux/ve.h b/include/linux/ve.h
index 5d94470692f72..b4c9e628cb333 100644
--- a/include/linux/ve.h
+++ b/include/linux/ve.h
@@ -88,6 +88,7 @@ struct ve_struct {
 	int			netif_max_nr;
 
 	atomic_t		bpf_prog_avail_nr;
+	atomic_t		bpf_prog_failcount;
 	int			bpf_prog_max_nr;
 
 	atomic64_t		_uevent_seqnum;
diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index c94d4240e3d3d..8498bc9df7ea6 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -2891,6 +2891,7 @@ static int bpf_prog_load(union bpf_attr *attr, bpfptr_t uattr, u32 uattr_size)
 	if (!bpf_cap && type == BPF_PROG_TYPE_CGROUP_DEVICE) {
 		load_ve = get_exec_env();
 		if (atomic_dec_if_positive(&load_ve->bpf_prog_avail_nr) < 0) {
+			ve_failcount_inc(load_ve, bpf_prog);
 			load_ve = NULL;
 			err = -ENOSPC;
 			goto put_token;
diff --git a/kernel/ve/ve.c b/kernel/ve/ve.c
index 8505c1dae4a18..be45975b74045 100644
--- a/kernel/ve/ve.c
+++ b/kernel/ve/ve.c
@@ -106,6 +106,7 @@ struct ve_struct ve0 = {
 	.netif_failcount	= ATOMIC_INIT(0),
 	.netif_max_nr		= INT_MAX,
 	.bpf_prog_avail_nr	= ATOMIC_INIT(INT_MAX),
+	.bpf_prog_failcount	= ATOMIC_INIT(0),
 	.bpf_prog_max_nr	= INT_MAX,
 	.fsync_enable		= FSYNC_FILTERED,
 	._randomize_va_space	=
@@ -803,6 +804,7 @@ static struct cgroup_subsys_state *ve_create(struct cgroup_subsys_state *parent_
 	ve->netif_max_nr = NETIF_MAX_NR_DEFAULT;
 
 	atomic_set(&ve->bpf_prog_avail_nr, BPF_PROG_MAX_NR_DEFAULT);
+	atomic_set(&ve->bpf_prog_failcount, 0);
 	ve->bpf_prog_max_nr = BPF_PROG_MAX_NR_DEFAULT;
 
 	err = ve_log_init(ve);
@@ -1093,6 +1095,7 @@ static const struct ve_failcount_entry {
 	VE_FC_ENTRY(netns),
 	VE_FC_ENTRY(mnt),
 	VE_FC_ENTRY(netif),
+	VE_FC_ENTRY(bpf_prog),
 	{}
 };
 

-- 
2.43.0