1 files changed, 10 insertions, 0 deletions

diff --git a/kernel/bpf/tnum.c b/kernel/bpf/tnum.c
index 1f4bf68c12db..938d41211be7 100644
--- a/kernel/bpf/tnum.c
+++ b/kernel/bpf/tnum.c
@@ -43,6 +43,16 @@ struct tnum tnum_rshift(struct tnum a, u8 shift)
         return TNUM(a.value >> shift, a.mask >> shift);
 }
 
+struct tnum tnum_arshift(struct tnum a, u8 min_shift)
+{
+        /* if a.value is negative, arithmetic shifting by minimum shift
+         * will have larger negative offset compared to more shifting.
+         * If a.value is nonnegative, arithmetic shifting by minimum shift
+         * will have larger positive offset compare to more shifting.
+         */
+        return TNUM((s64)a.value >> min_shift, (s64)a.mask >> min_shift);
+}
+
 struct tnum tnum_add(struct tnum a, struct tnum b)
 {
         u64 sm, sv, sigma, chi, mu;